167 lines
6.4 KiB
Python
167 lines
6.4 KiB
Python
from lightweight_charts.widgets import JupyterChart
|
|
|
|
class Panel:
|
|
"""
|
|
A class to represent a panel in a chart.
|
|
|
|
Attributes
|
|
----------
|
|
* ohlcv : tuple optional\n
|
|
(series, entries, exits, other_markers)
|
|
* histogram : list of tuples, optional.\n
|
|
[(series, name, color)]
|
|
* title : str, optional
|
|
The title of the panel. Default is None.
|
|
* right : list of tuples, optional
|
|
A list of line tuples in given scale.\n
|
|
[(series, name, entries, exits, other_markers)]
|
|
* left : list of tuples, optional
|
|
* middle1 : list of tuples, optional
|
|
* middle2 : list of tuples, optional
|
|
|
|
Examples
|
|
-------
|
|
```
|
|
# Example usage
|
|
pane1 = Panel(
|
|
ohlcv=(t1data.data["BAC"],), #(series, entries, exits, other_markers)
|
|
histogram=[(order_imbalance_allvolume, "oivol")], # [(series, name, "rgba(53, 94, 59, 0.6)")]
|
|
right=[], # [(series, name, entries, exits, other_markers)]
|
|
left=[(sma, "sma", short_signals, short_exits)],
|
|
middle1=[],
|
|
middle2=[],
|
|
)
|
|
|
|
```
|
|
"""
|
|
def __init__(self, ohlcv=None, right=None, left=None, middle1=None, middle2=None, histogram=None, title=None):
|
|
self.ohlcv = ohlcv if ohlcv is not None else []
|
|
self.right = right if right is not None else []
|
|
self.left = left if left is not None else []
|
|
self.middle1 = middle1 if middle1 is not None else []
|
|
self.middle2 = middle2 if middle2 is not None else []
|
|
self.histogram = histogram if histogram is not None else []
|
|
self.title = title
|
|
|
|
|
|
|
|
|
|
def chart(panes: list[Panel], sync=False, title='', size="m"):
|
|
"""
|
|
Function to fast render a chart with multiple panes. This function manipulates graphical
|
|
output or interfaces with an external framework to display charts with synchronized
|
|
scales if required.
|
|
|
|
Args:
|
|
-----
|
|
* panes (List[Pane]): A list of Pane instances to be rendered in the chart. Each Pane
|
|
can include various data series and configurations such as OHLCV,
|
|
histograms, and indicators positioned on left, right, or middle scales.
|
|
* sync (bool): If True, synchronize scales of all panes to have the same scale limits.
|
|
Default is False.
|
|
* title (str): Title of the chart. If not specified, defaults to an empty string.
|
|
|
|
* size (str): The size designation, which can be 's', 'm', or 'xl'. Defaults to'm'.
|
|
Expected values:
|
|
- 's' for small
|
|
- 'm' for medium
|
|
- 'xl' for extra large
|
|
|
|
Returns:
|
|
None: This function does not return a value. It performs operations to render graphical content.
|
|
|
|
Examples:
|
|
---------
|
|
```
|
|
pane1 = Pane(
|
|
ohlcv=(t1data.data["BAC"],),
|
|
histogram=[(order_imbalance_allvolume, "oivol")]
|
|
right=[],
|
|
left=[(sma, "sma", short_signals, short_exits)],
|
|
middle1=[],
|
|
middle2=[],
|
|
)
|
|
pane2 = Pane(
|
|
ohlcv=(t1data.data["BAC"],),
|
|
right=[],
|
|
left=[(sma, "sma_below", short_signals, short_exits)],
|
|
middle1=[],
|
|
middle2=[],
|
|
histogram=[(order_imbalance_sma, "oisma")]
|
|
)
|
|
ch = chart([pane1, pane2], sync=True, title="Chart", size="l")
|
|
```
|
|
|
|
Notes:
|
|
------
|
|
|
|
"""
|
|
size_to_dimensions = {
|
|
's': (800, 400),
|
|
'm': (1000, 600),
|
|
'l': (1300, 800)}
|
|
width, height = size_to_dimensions.get(size, (1000, 600))
|
|
height_ratio = 1 / len(panes)
|
|
main_title_set = False
|
|
for index, pane in enumerate(panes):
|
|
subchartX = None
|
|
if index == 0:
|
|
chartX = JupyterChart(width=width, height=height, inner_width=1, inner_height=height_ratio, leftScale=bool(pane.left))
|
|
active_chart = chartX
|
|
else:
|
|
subchartX = chartX.create_subchart(position='right', width=1, height=height_ratio, sync=sync, leftScale=bool(pane.left))
|
|
active_chart = subchartX
|
|
|
|
if pane.ohlcv is not None:
|
|
series, entries, exits, markers = (pane.ohlcv + (None,) * 4)[:4]
|
|
active_chart.set(series)
|
|
if entries is not None:
|
|
active_chart.markers_set(entries, "entries")
|
|
if exits is not None:
|
|
active_chart.markers_set(exits, "exits")
|
|
if markers is not None:
|
|
active_chart.markers_set(markers)
|
|
|
|
for tup in pane.histogram:
|
|
series, name, color, _, _ = (tup + (None, None, None, None, None))[:5]
|
|
if series is None:
|
|
continue
|
|
#conditionally include color
|
|
kwargs = {'name': name}
|
|
if color is not None:
|
|
kwargs['color'] = color
|
|
tmp = active_chart.create_histogram(**kwargs) #green transparent "rgba(53, 94, 59, 0.6)"
|
|
tmp.set(series)
|
|
|
|
if pane.title is not None:
|
|
active_chart.topbar.textbox("title",pane.title)
|
|
main_title_set = True if index==0 else False
|
|
|
|
#iterate over keys - they are all priceScaleId except of histogram and ohlcv
|
|
for att_name, att_value_tuple in vars(pane).items():
|
|
if att_name in ["ohlcv","histogram","title"]:
|
|
continue
|
|
for tup in att_value_tuple:
|
|
series, name, entries, exits, markers = (tup + (None, None, None, None, None))[:5]
|
|
if series is None:
|
|
continue
|
|
tmp = active_chart.create_line(name=name, priceScaleId=att_name)#, color="blue")
|
|
tmp.set(series)
|
|
|
|
if entries is not None:
|
|
tmp.markers_set(entries, "entries")
|
|
if exits is not None:
|
|
tmp.markers_set(exits, "exits")
|
|
if markers is not None:
|
|
tmp.markers_set(markers)
|
|
|
|
active_chart.legend(True)
|
|
active_chart.fit()
|
|
|
|
if not main_title_set:
|
|
chartX.topbar.textbox("title",title)
|
|
chartX.legend(True)
|
|
chartX.fit()
|
|
chartX.load()
|
|
return chartX
|