def get_legend_items(self): plot = self.plot() if plot is None: return [] text_items = [] for item in plot.get_items(): if not isinstance(item, CurveItem) or not self.include_item(item): continue text = QTextDocument() text.setDefaultFont(self.font) text.setDefaultStyleSheet("div { color: %s; }" % self.color) text.setHtml("<div>%s</div>" % item.curveparam.label) text_items.append((text, item.pen(), item.brush(), item.symbol())) return text_items
class LabelItem(AbstractLabelItem): __implements__ = (IBasePlotItem, ISerializableType) def __init__(self, text=None, labelparam=None): self.text_string = "" if text is None else text self.text = QTextDocument() super(LabelItem, self).__init__(labelparam) def __reduce__(self): return (self.__class__, (self.text_string, self.labelparam)) def serialize(self, writer): """Serialize object to HDF5 writer""" super(LabelItem, self).serialize(writer) writer.write(self.text_string, group_name="text") def deserialize(self, reader): """Deserialize object from HDF5 reader""" super(LabelItem, self).deserialize(reader) self.set_text(reader.read("text", func=reader.read_unicode)) def types(self): return (IShapeItemType, ISerializableType) def set_pos(self, x, y): self.G = x, y self.labelparam.xg, self.labelparam.yg = x, y def get_plain_text(self): return to_text_string(self.text.toPlainText()) def set_text(self, text=None): if text is not None: self.text_string = text self.text.setHtml("<div>%s</div>" % self.text_string) def set_text_style(self, font=None, color=None): if font is not None: self.text.setDefaultFont(font) if color is not None: self.text.setDefaultStyleSheet("div { color: %s; }" % color) self.set_text() def get_text_rect(self): sz = self.text.size() return QRectF(0, 0, sz.width(), sz.height()) def update_text(self): pass def draw(self, painter, xMap, yMap, canvasRect): self.update_text() x, y = self.get_top_left(xMap, yMap, canvasRect) x0, y0 = self.get_origin(xMap, yMap, canvasRect) painter.save() self.marker.drawSymbols(painter, [QPointF(x0, y0)]) painter.restore() sz = self.text.size() self.draw_frame(painter, x, y, sz.width(), sz.height()) painter.setPen(QPen(QColor(self.labelparam.color))) painter.translate(x, y) self.text.drawContents(painter)
class FrameContentHover(FrameBase): """Frame used in css styling with fade in and fade out effect.""" sig_clicked = Signal() sig_entered = Signal() sig_left = Signal() def __init__(self, *args, **kwargs): """Frame used in css styling with fade in and fade out effect.""" super(FrameContentHover, self).__init__(*args, **kwargs) self.current_opacity = 0 self.max_frame = 100 self.max_opacity = 0.95 self.button_text = None self.label_icon = None self.label_text = None self.summary = '' # Widgets self.text_document = QTextDocument(self) self.timeline = QTimeLine(500) font = QFont() if sys.platform == 'darwin': font.setPointSize(12) elif os.name == 'nt': font.setPointSize(9) else: font.setPointSize(9) self.text_document.setDefaultFont(font) self.text_document.setMaximumBlockCount(5) self.text_document.setDocumentMargin(10) # Setup self.setWindowFlags(Qt.FramelessWindowHint) self.setAutoFillBackground(True) self.setWindowFlags(Qt.FramelessWindowHint) self.setMinimumSize(self.widget_size()) self.timeline.setFrameRange(0, self.max_frame) # Signals self.timeline.frameChanged.connect(self.set_opacity) def set_opacity(self, val): """Set the opacity via timeline.""" self.current_opacity = (val * 1.0 / self.max_frame) * self.max_opacity self.update() def widget_size(self): """Return hover frame size.""" bheight = 40 if self.button_text is None else self.button_text.height() width = SASS_VARIABLES.WIDGET_CONTENT_TOTAL_WIDTH height = SASS_VARIABLES.WIDGET_CONTENT_TOTAL_HEIGHT padding = SASS_VARIABLES.WIDGET_CONTENT_PADDING margin = 0 # SASS_VARIABLES.WIDGET_CONTENT_MARGIN return QSize( width - 2 * (padding + margin), height - 2 * margin - bheight ) def fade_in(self): """Fade in hover card with text.""" self.raise_() self.timeline.stop() self.timeline.setDirection(QTimeLine.Forward) self.timeline.start() def fade_out(self): """Fade out hover card with text.""" self.timeline.stop() self.timeline.setDirection(QTimeLine.Backward) self.timeline.start() def enterEvent(self, event): """Override Qt method.""" super(FrameContentHover, self).enterEvent(event) self.fade_in() self.sig_entered.emit() def leaveEvent(self, event): """Override Qt method.""" super(FrameContentHover, self).leaveEvent(event) self.fade_out() self.sig_left.emit() def mousePressEvent(self, event): """Override Qt method.""" self.sig_clicked.emit() def paintEvent(self, event): """Override Qt method.""" painter = QPainter(self) painter.setOpacity(self.current_opacity) max_width = ( SASS_VARIABLES.WIDGET_CONTENT_TOTAL_WIDTH - 2 * SASS_VARIABLES.WIDGET_CONTENT_PADDING - 2 * SASS_VARIABLES.WIDGET_CONTENT_MARGIN ) # Hover top br = self.label_icon.rect().bottomRight() tl = self.label_icon.rect().topLeft() + QPoint(1, 1) y = br.y() + self.label_text.height() - 2 # br_new = QPoint(br.x() + 2, y) - QPoint(1, 1) br_new = QPoint(max_width - 1, y) - QPoint(1, 1) rect_hover = QRect(tl, br_new) # 2 is the border pen = QPen(Qt.NoPen) brush = QBrush(Qt.SolidPattern) brush.setColor(QColor('#fff')) painter.setBrush(brush) painter.setPen(pen) painter.drawRect(rect_hover) font = self.font() font.setPointSize(10) painter.setFont(font) pen = QPen() pen.setColor(QColor('black')) painter.setPen(pen) td = self.text_document td.setPageSize(QSizeF(rect_hover.size())) td.setHtml(self.summary) td.drawContents(painter) self.raise_()