400 lines
10 KiB
Python
400 lines
10 KiB
Python
# piker: trading gear for hackers
|
|
# Copyright (C) 2018-present Tyler Goodlet (in stewardship of piker0)
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
"""
|
|
Lines for orders, alerts, L2.
|
|
|
|
"""
|
|
from typing import Tuple
|
|
|
|
import pyqtgraph as pg
|
|
from PyQt5 import QtCore, QtGui
|
|
from PyQt5.QtCore import QPointF
|
|
|
|
from .._style import (
|
|
hcolor,
|
|
_down_2_font_inches_we_like,
|
|
)
|
|
from .._axes import YSticky
|
|
|
|
|
|
class LevelLabel(YSticky):
|
|
|
|
_w_margin = 4
|
|
_h_margin = 3
|
|
level: float = 0
|
|
|
|
def __init__(
|
|
self,
|
|
chart,
|
|
*args,
|
|
color: str = 'bracket',
|
|
orient_v: str = 'bottom',
|
|
orient_h: str = 'left',
|
|
**kwargs
|
|
) -> None:
|
|
super().__init__(chart, *args, **kwargs)
|
|
|
|
# TODO: this is kinda cludgy
|
|
self._hcolor = None
|
|
self.color = color
|
|
|
|
# orientation around axis options
|
|
self._orient_v = orient_v
|
|
self._orient_h = orient_h
|
|
self._v_shift = {
|
|
'top': 1.,
|
|
'bottom': 0,
|
|
'middle': 1 / 2.
|
|
}[orient_v]
|
|
|
|
self._h_shift = {
|
|
'left': -1., 'right': 0
|
|
}[orient_h]
|
|
|
|
@property
|
|
def color(self):
|
|
return self._hcolor
|
|
|
|
@color.setter
|
|
def color(self, color: str) -> None:
|
|
self._hcolor = color
|
|
self._pen = self.pen = pg.mkPen(hcolor(color))
|
|
|
|
def update_label(
|
|
self,
|
|
abs_pos: QPointF, # scene coords
|
|
level: float, # data for text
|
|
offset: int = 1 # if have margins, k?
|
|
) -> None:
|
|
|
|
# write contents, type specific
|
|
self.set_label_str(level)
|
|
|
|
br = self.boundingRect()
|
|
h, w = br.height(), br.width()
|
|
|
|
# this triggers ``.paint()`` implicitly?
|
|
self.setPos(QPointF(
|
|
self._h_shift * w - offset,
|
|
abs_pos.y() - (self._v_shift * h) - offset
|
|
))
|
|
self.update()
|
|
|
|
self.level = level
|
|
|
|
def set_label_str(self, level: float):
|
|
# self.label_str = '{size} x {level:.{digits}f}'.format(
|
|
# size=self._size,
|
|
|
|
# this is read inside ``.paint()``
|
|
self.label_str = '{level:.{digits}f}'.format(
|
|
digits=self.digits,
|
|
level=level
|
|
).replace(',', ' ')
|
|
|
|
def size_hint(self) -> Tuple[None, None]:
|
|
return None, None
|
|
|
|
def draw(
|
|
self,
|
|
p: QtGui.QPainter,
|
|
rect: QtCore.QRectF
|
|
) -> None:
|
|
p.setPen(self._pen)
|
|
|
|
if self._orient_v == 'bottom':
|
|
lp, rp = rect.topLeft(), rect.topRight()
|
|
# p.drawLine(rect.topLeft(), rect.topRight())
|
|
elif self._orient_v == 'top':
|
|
lp, rp = rect.bottomLeft(), rect.bottomRight()
|
|
|
|
p.drawLine(lp.x(), lp.y(), rp.x(), rp.y())
|
|
|
|
def highlight(self, pen) -> None:
|
|
self._pen = pen
|
|
self.update()
|
|
|
|
def unhighlight(self):
|
|
self._pen = self.pen
|
|
self.update()
|
|
|
|
|
|
class L1Label(LevelLabel):
|
|
|
|
size: float = 0
|
|
size_digits: float = 3
|
|
|
|
text_flags = (
|
|
QtCore.Qt.TextDontClip
|
|
| QtCore.Qt.AlignLeft
|
|
)
|
|
|
|
def set_label_str(self, level: float) -> None:
|
|
"""Reimplement the label string write to include the level's order-queue's
|
|
size in the text, eg. 100 x 323.3.
|
|
|
|
"""
|
|
self.label_str = '{size:.{size_digits}f} x {level:,.{digits}f}'.format(
|
|
size_digits=self.size_digits,
|
|
size=self.size or 2,
|
|
digits=self.digits,
|
|
level=level
|
|
).replace(',', ' ')
|
|
|
|
|
|
class L1Labels:
|
|
"""Level 1 bid ask labels for dynamic update on price-axis.
|
|
|
|
"""
|
|
max_value: float = '100.0 x 100 000.00'
|
|
|
|
def __init__(
|
|
self,
|
|
chart: 'ChartPlotWidget', # noqa
|
|
digits: int = 2,
|
|
size_digits: int = 3,
|
|
font_size_inches: float = _down_2_font_inches_we_like,
|
|
) -> None:
|
|
|
|
self.chart = chart
|
|
|
|
self.bid_label = L1Label(
|
|
chart=chart,
|
|
parent=chart.getAxis('right'),
|
|
# TODO: pass this from symbol data
|
|
digits=digits,
|
|
opacity=1,
|
|
font_size_inches=font_size_inches,
|
|
bg_color='papas_special',
|
|
fg_color='bracket',
|
|
orient_v='bottom',
|
|
)
|
|
self.bid_label.size_digits = size_digits
|
|
self.bid_label._size_br_from_str(self.max_value)
|
|
|
|
self.ask_label = L1Label(
|
|
chart=chart,
|
|
parent=chart.getAxis('right'),
|
|
# TODO: pass this from symbol data
|
|
digits=digits,
|
|
opacity=1,
|
|
font_size_inches=font_size_inches,
|
|
bg_color='papas_special',
|
|
fg_color='bracket',
|
|
orient_v='top',
|
|
)
|
|
self.ask_label.size_digits = size_digits
|
|
self.ask_label._size_br_from_str(self.max_value)
|
|
|
|
|
|
# TODO: probably worth investigating if we can
|
|
# make .boundingRect() faster:
|
|
# https://stackoverflow.com/questions/26156486/determine-bounding-rect-of-line-in-qt
|
|
class LevelLine(pg.InfiniteLine):
|
|
|
|
# TODO: fill in these slots for orders
|
|
# .sigPositionChangeFinished.emit(self)
|
|
|
|
def __init__(
|
|
self,
|
|
chart: 'ChartPlotWidget', # type: ignore # noqa
|
|
label: LevelLabel,
|
|
color: str = 'default',
|
|
highlight_color: str = 'default_light',
|
|
hl_on_hover: bool = True,
|
|
**kwargs,
|
|
) -> None:
|
|
|
|
super().__init__(**kwargs)
|
|
self.label = label
|
|
|
|
self.sigPositionChanged.connect(self.set_level)
|
|
self._chart = chart
|
|
self._hoh = hl_on_hover
|
|
|
|
self._hcolor = None
|
|
self.color = color
|
|
|
|
self._track_cursor: bool = False
|
|
|
|
@property
|
|
def color(self):
|
|
return self._hcolor
|
|
|
|
@color.setter
|
|
def color(self, color: str) -> None:
|
|
self._hcolor = color
|
|
self.setPen(pg.mkPen(hcolor(color)))
|
|
|
|
# set hover pen to new color
|
|
pen = pg.mkPen(hcolor(color + '_light'))
|
|
# use slightly thicker highlight
|
|
pen.setWidth(2)
|
|
self.setHoverPen(pen)
|
|
|
|
def set_level(self, value: float) -> None:
|
|
self.label.update_from_data(0, self.value())
|
|
|
|
def on_tracked_source(
|
|
self,
|
|
x: int,
|
|
y: float
|
|
) -> None:
|
|
self.movable = True
|
|
self.setPos(y) # implictly calls ``.set_level()``
|
|
self.update()
|
|
|
|
def setMouseHover(self, hover: bool) -> None:
|
|
"""Mouse hover callback.
|
|
|
|
"""
|
|
# XXX: currently we'll just return if _hoh is False
|
|
if self.mouseHovering == hover:
|
|
return
|
|
|
|
self.mouseHovering = hover
|
|
|
|
chart = self._chart
|
|
|
|
if hover:
|
|
# highlight if so configured
|
|
if self._hoh:
|
|
self.currentPen = self.hoverPen
|
|
self.label.highlight(self.hoverPen)
|
|
|
|
# add us to cursor state
|
|
chart._cursor.add_hovered(self)
|
|
|
|
# TODO: hide y-crosshair?
|
|
# chart._cursor.graphics[chart]['hl'].hide()
|
|
|
|
else:
|
|
self.currentPen = self.pen
|
|
self.label.unhighlight()
|
|
|
|
chart._cursor._hovered.remove(self)
|
|
|
|
# highlight any attached label
|
|
|
|
# self.setCursor(QtCore.Qt.OpenHandCursor)
|
|
# self.setCursor(QtCore.Qt.DragMoveCursor)
|
|
self.update()
|
|
|
|
def mouseDragEvent(self, ev):
|
|
chart = self._chart
|
|
# hide y-crosshair
|
|
chart._cursor.graphics[chart]['hl'].hide()
|
|
|
|
# highlight
|
|
self.currentPen = self.hoverPen
|
|
self.label.highlight(self.hoverPen)
|
|
|
|
# normal tracking behavior
|
|
super().mouseDragEvent(ev)
|
|
|
|
# This is the final position in the drag
|
|
if ev.isFinish():
|
|
# show y-crosshair again
|
|
chart = self._chart
|
|
chart._cursor.graphics[chart]['hl'].show()
|
|
|
|
def mouseDoubleClickEvent(
|
|
self,
|
|
ev: QtGui.QMouseEvent,
|
|
) -> None:
|
|
print(f'double click {ev}')
|
|
|
|
# def mouseMoved(
|
|
# self,
|
|
# ev: Tuple[QtGui.QMouseEvent],
|
|
# ) -> None:
|
|
# pos = evt[0]
|
|
# print(pos)
|
|
|
|
def delete(self) -> None:
|
|
"""Remove this line from containing chart/view/scene.
|
|
|
|
"""
|
|
scene = self.scene()
|
|
if scene:
|
|
# self.label.parent.scene().removeItem(self.label)
|
|
scene.removeItem(self.label)
|
|
|
|
self._chart.plotItem.removeItem(self)
|
|
|
|
|
|
def level_line(
|
|
chart: 'ChartPlogWidget', # noqa
|
|
level: float,
|
|
digits: int = 1,
|
|
color: str = 'default',
|
|
|
|
# size 4 font on 4k screen scaled down, so small-ish.
|
|
font_size_inches: float = _down_2_font_inches_we_like,
|
|
|
|
show_label: bool = True,
|
|
|
|
# whether or not the line placed in view should highlight
|
|
# when moused over (aka "hovered")
|
|
hl_on_hover: bool = True,
|
|
|
|
**linelabelkwargs
|
|
) -> LevelLine:
|
|
"""Convenience routine to add a styled horizontal line to a plot.
|
|
|
|
"""
|
|
label = LevelLabel(
|
|
chart=chart,
|
|
parent=chart.getAxis('right'),
|
|
# TODO: pass this from symbol data
|
|
digits=digits,
|
|
opacity=0.666,
|
|
font_size_inches=font_size_inches,
|
|
color=color,
|
|
|
|
# TODO: make this take the view's bg pen
|
|
bg_color='papas_special',
|
|
fg_color=color,
|
|
**linelabelkwargs
|
|
)
|
|
label.update_from_data(0, level)
|
|
|
|
# TODO: can we somehow figure out a max value from the parent axis?
|
|
label._size_br_from_str(label.label_str)
|
|
|
|
line = LevelLine(
|
|
chart,
|
|
label,
|
|
color=color,
|
|
# lookup "highlight" equivalent
|
|
highlight_color=color + '_light',
|
|
movable=True,
|
|
angle=0,
|
|
hl_on_hover=hl_on_hover,
|
|
)
|
|
line.setValue(level)
|
|
|
|
# activate/draw label
|
|
line.setValue(level)
|
|
|
|
chart.plotItem.addItem(line)
|
|
|
|
if not show_label:
|
|
label.hide()
|
|
|
|
return line
|