class Message(QWidget): def __init__(self, parent): QWidget.__init__(self, parent) self.layout = QTextLayout() self.layout.setFont(self.font()) self.layout.setCacheEnabled(True) self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Preferred) self.last_layout_rect = None def setText(self, text): self.layout.setText(text) self.last_layout_rect = None self.update() def sizeHint(self): return QSize(10, 10) def do_layout(self): ly = self.layout ly.beginLayout() w = self.width() - 5 height = 0 leading = self.fontMetrics().leading() while True: line = ly.createLine() if not line.isValid(): break line.setLineWidth(w) height += leading line.setPosition(QPointF(5, height)) height += line.height() ly.endLayout() def paintEvent(self, ev): if self.last_layout_rect != self.rect(): self.do_layout() p = QPainter(self) br = self.layout.boundingRect() y = 0 if br.height() < self.height(): y = (self.height() - br.height()) / 2 self.layout.draw(p, QPointF(0, y))
def viewItemDrawText(self, painter, option, rect, text, color, searchText=''): # noqa C901 ''' @note: most of codes taken from QCommonStylePrivate::viewItemDrawText added highlighting and simplified for single-line textlayouts @param painter QPainter @param option QStyleOptionViewItem @param rect QRect @param text QString @param color QColor @param searchText QString @return: int ''' if not text: return 0 fontMetrics = QFontMetrics(painter.font()) elidedText = fontMetrics.elidedText(text, option.textElideMode, rect.width()) textOption = QTextOption() textOption.setWrapMode(QTextOption.NoWrap) textOption.setAlignment(QStyle.visualAlignment(textOption.textDirection(), option.displayAlignment)) textLayout = QTextLayout() textLayout.setFont(painter.font()) textLayout.setText(elidedText) textLayout.setTextOption(textOption) if searchText: # QList<int> delimiters = [] searchStrings = [ item.strip() for item in searchText.split(' ') ] searchStrings = [ item for item in searchStrings if item ] # Look for longer parts first searchStrings.sort(key=lambda x: len(x), reverse=True) text0 = text.lower() for string in searchStrings: string0 = string.lower() delimiter = text0.find(string0) while delimiter != -1: start = delimiter end = delimiter + len(string) alreadyContains = False for idx in range(0, len(delimiters), 2): dStart = delimiters[idx] dEnd = delimiters[idx+1] if dStart <= start and dEnd >= end: alreadyContains = True break if not alreadyContains: delimiters.append(start) delimiters.append(end) delimiter = text0.find(string0, end) # We need to sort delimiters to properly paint all parts that user typed delimiters.sort() # If we don't find any match, just paint it withoutany highlight if delimiters and len(delimiters) % 2 == 0: highlightParts = [] # QList<QTextLayout::FormatRange> while delimiters: highlightedPart = QTextLayout.FormatRange() start = delimiters.pop(0) end = delimiters.pop(0) highlightedPart.start = start highlightedPart.length = end - start highlightedPart.format.setFontWeight(QFont.Bold) highlightedPart.format.setUnderlineStyle(QTextCharFormat.SingleUnderline) highlightParts.append(highlightedPart) textLayout.setAdditionalFormats(highlightParts) # do layout self._s_viewItemDrawText(textLayout, rect.width()) if textLayout.lineCount() <= 0: return 0 textLine = textLayout.lineAt(0) # if elidedText after highlighting is longer than available width then # re-elide it and redo layout diff = textLine.naturalTextWidth() - rect.width() if diff > 0: # TODO: ? elidedText = fontMetrics.elidedText(elidedText, option.textElideMode, rect.width() - diff) textLayout.setText(elidedText) # redo layout self._s_viewItemDrawText(textLayout, rect.width()) if textLayout.lineCount() <= 0: return 0 textLine = textLayout.lineAt(0) # draw line painter.setPen(color) # qreal width = max(rect.width(), textLayout.lineAt(0).width()) # QRect layoutRect = QStyle.alignedRect(option.direction, option.displayAlignment, QSize(int(width), int(textLine.height())), rect) # QPointF position = layoutRect.topLeft() textLine.draw(painter, position) return int(min(rect.width(), textLayout.lineAt(0).naturalTextWidth()))