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_()
