Fix color passthrough, make overlays a `dict`

basic_vlm_display
Tyler Goodlet 2021-09-26 17:14:30 -04:00
parent 39fb2ee85d
commit dbdd7b6497
1 changed files with 6 additions and 3 deletions

View File

@ -56,6 +56,7 @@ from ._style import (
) )
from ..data.feed import Feed from ..data.feed import Feed
from ..data._source import Symbol from ..data._source import Symbol
from ..data._sharedmem import ShmArray
from ..log import get_logger from ..log import get_logger
from ._interaction import ChartView from ._interaction import ChartView
from ._forms import FieldsForm from ._forms import FieldsForm
@ -632,7 +633,8 @@ class ChartPlotWidget(pg.PlotWidget):
'ohlc': array, 'ohlc': array,
} }
self._graphics = {} # registry of underlying graphics self._graphics = {} # registry of underlying graphics
self._overlays = set() # registry of overlay curve names # registry of overlay curve names
self._overlays: dict[str, ShmArray] = {}
self._feeds: dict[Symbol, Feed] = {} self._feeds: dict[Symbol, Feed] = {}
@ -808,8 +810,9 @@ class ChartPlotWidget(pg.PlotWidget):
the input array ``data``. the input array ``data``.
""" """
color = color or self.pen_color or 'default_light'
pdi_kwargs.update({ pdi_kwargs.update({
'color': color or self.pen_color or 'default_light' 'color': color
}) })
data_key = array_key or name data_key = array_key or name
@ -856,7 +859,7 @@ class ChartPlotWidget(pg.PlotWidget):
if overlay: if overlay:
anchor_at = ('bottom', 'left') anchor_at = ('bottom', 'left')
self._overlays.add(name) self._overlays[name] = None
else: else:
anchor_at = ('top', 'left') anchor_at = ('top', 'left')