def __init__(self, text='', width=0, font=None, img=None, max_height=100, align=Qt.AlignCenter): self.layouts = [] self._position = Point(0, 0) self.leading = self.line_spacing = 0 if font is not None: fm = QFontMetrics(font, img) self.leading = fm.leading() self.line_spacing = fm.lineSpacing() for text in text.split('<br>') if text else (): text, formats = parse_text_formatting(sanitize(text)) l = QTextLayout(unescape_formatting(text), font, img) l.setAdditionalFormats(formats) to = QTextOption(align) to.setWrapMode(QTextOption.WrapAtWordBoundaryOrAnywhere) l.setTextOption(to) l.beginLayout() height = 0 while height + 3*self.leading < max_height: line = l.createLine() if not line.isValid(): break line.setLineWidth(width) height += self.leading line.setPosition(QPointF(0, height)) height += line.height() max_height -= height l.endLayout() if self.layouts: self.layouts.append(self.leading) else: self._position = Point(l.position().x(), l.position().y()) self.layouts.append(l) if self.layouts: self.layouts.append(self.leading)
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()))