diff --git a/piker/ui/_graphics/_cursor.py b/piker/ui/_graphics/_cursor.py index d5d4575a..5d312f39 100644 --- a/piker/ui/_graphics/_cursor.py +++ b/piker/ui/_graphics/_cursor.py @@ -17,7 +17,7 @@ Mouse interaction graphics """ -from typing import Optional, Tuple +from typing import Optional, Tuple, Set, Dict import inspect import numpy as np @@ -198,7 +198,7 @@ class ContentsLabel(pg.LabelItem): self.setText(f"{name}: {data:.2f}") -class CrossHair(pg.GraphicsObject): +class Cursor(pg.GraphicsObject): def __init__( self, @@ -217,11 +217,21 @@ class CrossHair(pg.GraphicsObject): style=QtCore.Qt.DashLine, ) self.lsc = linkedsplitcharts - self.graphics = {} - self.plots = [] + self.graphics: Dict[str, pg.GraphicsObject] = {} + self.plots: List['PlotChartWidget'] = [] # type: ignore # noqa self.active_plot = None - self.digits = digits - self._lastx = None + self.digits: int = digits + self._datum_xy: Tuple[int, float] = (0, 0) + + self._hovered: Set[pg.GraphicsObject] = set() + self._trackers: Set[pg.GraphicsObject] = set() + + def add_hovered( + self, + item: pg.GraphicsObject, + ) -> None: + assert getattr(item, 'delete'), f"{item} must define a ``.delete()``" + self._hovered.add(item) def add_plot( self, @@ -293,7 +303,11 @@ class CrossHair(pg.GraphicsObject): ) -> LineDot: # if this plot contains curves add line dot "cursors" to denote # the current sample under the mouse - cursor = LineDot(curve, index=plot._ohlc[-1]['index'], plot=plot) + cursor = LineDot( + curve, + index=plot._ohlc[-1]['index'], + plot=plot + ) plot.addItem(cursor) self.graphics[plot].setdefault('cursors', []).append(cursor) return cursor @@ -336,15 +350,21 @@ class CrossHair(pg.GraphicsObject): # update y-range items self.graphics[plot]['hl'].setY(y) + self.graphics[self.active_plot]['yl'].update_label( abs_pos=pos, value=y ) # Update x if cursor changed after discretization calc # (this saves draw cycles on small mouse moves) - lastx = self._lastx + lastx, lasty = self._datum_xy ix = round(x) # since bars are centered around index + # update all trackers + for item in self._trackers: + # print(f'setting {item} with {(ix, y)}') + item.on_tracked_source(ix, y) + if ix != lastx: for plot, opts in self.graphics.items(): @@ -355,7 +375,6 @@ class CrossHair(pg.GraphicsObject): plot.update_contents_labels(ix) # update all subscribed curve dots - # first = plot._ohlc[0]['index'] for cursor in opts.get('cursors', ()): cursor.setIndex(ix) @@ -371,7 +390,7 @@ class CrossHair(pg.GraphicsObject): value=x, ) - self._lastx = ix + self._datum_xy = ix, y def boundingRect(self): try: