pikers
/
piker
3
0
Fork 0
Code Issues 4 Pull Requests 6 Packages Projects Releases Wiki Activity

Handle flat bar updates 

Flat bars have a rendering issue we work around by hacking values in `QLineF`
but we have to revert those on any last bar that is being updated in
real-time. Comment out candle implementations for now; we can get back
to it if/when the tinas unite. Oh, and make bars have a little space
between them.
bar_select
Tyler Goodlet 2020-07-07 10:44:55 -04:00
parent 6b1bdbe3ea
commit 9c1d64413e
1 changed files with 51 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
piker/ui/_graphics.py
View File

@ -1,7 +1,7 @@
""" """
Chart graphics for displaying a slew of different data types. Chart graphics for displaying a slew of different data types.
""" """
from typing import Dict, Any # from typing import Dict, Any
from enum import Enum from enum import Enum
from contextlib import contextmanager from contextlib import contextmanager
@ -18,7 +18,7 @@ from ._axes import YAxisLabel, XAxisLabel
# TODO: checkout pyqtgraph.PlotCurveItem.setCompositionMode # TODO: checkout pyqtgraph.PlotCurveItem.setCompositionMode
_mouse_rate_limit = 40 _mouse_rate_limit = 50
class CrossHairItem(pg.GraphicsObject): class CrossHairItem(pg.GraphicsObject):
@ -174,7 +174,8 @@ class BarItems(pg.GraphicsObject):
""" """
sigPlotChanged = QtCore.Signal(object) sigPlotChanged = QtCore.Signal(object)
w: float = 0.5 # 0.5 is no overlap between arms, 1.0 is full overlap
w: float = 0.4
bull_pen = pg.mkPen('#808080') bull_pen = pg.mkPen('#808080')
# XXX: tina mode, see below # XXX: tina mode, see below
@ -218,8 +219,10 @@ class BarItems(pg.GraphicsObject):
if low != high: if low != high:
hl = QLineF(index, low, index, high) hl = QLineF(index, low, index, high)
else: else:
# if we don't do it renders a weird rectangle? # XXX: if we don't do it renders a weird rectangle?
# see below too for handling this later...
hl = QLineF(low, low, low, low) hl = QLineF(low, low, low, low)
hl._flat = True
# open line # open line
o = QLineF(index - self.w, q['open'], index, q['open']) o = QLineF(index - self.w, q['open'], index, q['open'])
# close line # close line
@ -227,6 +230,7 @@ class BarItems(pg.GraphicsObject):
# indexing here is as per the below comments # indexing here is as per the below comments
lines[3*i:3*i+3] = (hl, o, c) lines[3*i:3*i+3] = (hl, o, c)
p.setPen(self.bull_pen) p.setPen(self.bull_pen)
p.drawLines(*lines) p.drawLines(*lines)
# if not _tina_mode: # piker mode # if not _tina_mode: # piker mode
@ -263,7 +267,10 @@ class BarItems(pg.GraphicsObject):
does) so this "should" be simpler and faster. does) so this "should" be simpler and faster.
""" """
# do we really need to verify the entire past data set? # do we really need to verify the entire past data set?
last = array['close'][-1] # last = array['close'][-1]
# index, time, open, high, low, close, volume
index, time, _, _, _, close, _ = array[-1]
last = close
body, larm, rarm = self.lines[-3:] body, larm, rarm = self.lines[-3:]
# XXX: is there a faster way to modify this? # XXX: is there a faster way to modify this?
@ -279,11 +286,18 @@ class BarItems(pg.GraphicsObject):
if last > high: if last > high:
high = last high = last
if getattr(body, '_flat', None) and low != high:
# if the bar was flat it likely does not have
# the index set correctly due to a rendering bug
# see above
body.setLine(index, low, index, high)
body._flat = False
else:
body.setLine(body.x1(), low, body.x2(), high) body.setLine(body.x1(), low, body.x2(), high)
# draw the pic # draw the pic
with self.painter() as p: with self.painter() as p:
p.setPen(self.bull_brush) p.setPen(self.bull_pen)
p.drawLines(*self.lines) p.drawLines(*self.lines)
# trigger re-render # trigger re-render
@ -305,37 +319,43 @@ class BarItems(pg.GraphicsObject):
return QtCore.QRectF(self.picture.boundingRect()) return QtCore.QRectF(self.picture.boundingRect())
class CandlestickItems(BarItems): # XXX: when we get back to enabling tina mode
# class CandlestickItems(BarItems):
w2 = 0.7 # w2 = 0.7
line_pen = pg.mkPen('#000000') # line_pen = pg.mkPen('#000000')
bull_brush = pg.mkBrush('#00ff00') # bull_brush = pg.mkBrush('#00ff00')
bear_brush = pg.mkBrush('#ff0000') # bear_brush = pg.mkBrush('#ff0000')
def _generate(self, p): # def _generate(self, p):
rects = np.array( # rects = np.array(
[ # [
QtCore.QRectF(q.id - self.w, q.open, self.w2, q.close - q.open) # QtCore.QRectF(
for q in Quotes # q.id - self.w,
] # q.open,
) # self.w2,
# q.close - q.open
# )
# for q in Quotes
# ]
# )
p.setPen(self.line_pen) # p.setPen(self.line_pen)
p.drawLines( # p.drawLines(
[QtCore.QLineF(q.id, q.low, q.id, q.high) # [QtCore.QLineF(q.id, q.low, q.id, q.high)
for q in Quotes] # for q in Quotes]
) # )
p.setBrush(self.bull_brush) # p.setBrush(self.bull_brush)
p.drawRects(*rects[Quotes.close > Quotes.open]) # p.drawRects(*rects[Quotes.close > Quotes.open])
p.setBrush(self.bear_brush) # p.setBrush(self.bear_brush)
p.drawRects(*rects[Quotes.close < Quotes.open]) # p.drawRects(*rects[Quotes.close < Quotes.open])
class ChartType(Enum): class ChartType(Enum):
"""Bar type to graphics class map. """Bar type to graphics class map.
""" """
BAR = BarItems BAR = BarItems
CANDLESTICK = CandlestickItems # CANDLESTICK = CandlestickItems
LINE = pg.PlotDataItem LINE = pg.PlotDataItem