seolpyo-mplchart 2.0.0.3__py3-none-any.whl → 2.1.0.4__py3-none-any.whl

seolpyo_mplchart/_chart/cursor/__init__.py

@@ -0,0 +1,125 @@
+from matplotlib.axes import Axes
+from matplotlib.collections import LineCollection
+from matplotlib.text import Text
+from matplotlib.axes import Axes
+import numpy as np
+import pandas as pd
+
+from ..base import Chart as BaseChart, Figure, ConfigData
+
+from .b_artist import ArtistMixin
+from .c_draw import DrawMixin
+from .d_segment import SegmentMixin
+from .e_axis import AxisMixin
+from .g_event import EventMixin
+from .h_data import DataMixin
+
+
+class CursorMixin(
+    ArtistMixin,
+    DrawMixin,
+    SegmentMixin,
+    AxisMixin,
+    EventMixin,
+    DataMixin
+):
+    pass
+
+
+class Chart(CursorMixin, BaseChart):
+    limit_candle = 400
+    limit_wick = 2_000
+    candle_on_ma = True
+    fraction = False
+
+    key_date = 'date'
+    key_open, key_high, key_low, key_close = ('open', 'high', 'low', 'close')
+    key_volume = 'volume'
+
+    index_list: list[int] = []
+
+    df: pd.DataFrame
+
+    CONFIG: ConfigData
+
+    figure: Figure
+    ax_legend: Axes
+    ax_price: Axes
+    ax_volume: Axes
+
+    artist_watermark: Text
+    collection_candle: LineCollection
+    collection_volume: LineCollection
+    collection_ma: LineCollection
+
+    in_chart = False
+    in_chart_price = False
+    in_chart_volume = False
+
+    in_candle = False
+    in_volume = False
+
+    collection_price_crossline: LineCollection
+    collection_volume_crossline: LineCollection
+
+    artist_label_x: Text = None
+    artist_label_y: Text = None
+
+    collection_box_price: LineCollection
+    collection_box_volume: LineCollection
+
+    artist_info_candle: Text
+    artist_info_volume: Text
+
+    ###
+
+    segment_volume: np.ndarray
+    segment_volume_wick: np.ndarray
+    facecolor_volume: np.ndarray
+    edgecolor_volume: np.ndarray
+
+    segment_candle: np.ndarray
+    segment_candle_wick: np.ndarray
+    segment_priceline: np.ndarray
+    facecolor_candle: np.ndarray
+    edgecolor_candle: np.ndarray
+
+    segment_ma: np.ndarray
+    edgecolor_ma: np.ndarray
+
+    price_ymin: int
+    price_ymax: int
+    volume_ymax: int
+
+    chart_price_ymax: float
+    chart_volume_ymax: float
+
+    vxmin: int
+    vxmax: int
+
+    v0: int
+    v1: int
+    vmiddle: int
+
+    min_height_box_candle: float
+    min_height_box_volume: float
+
+    ###
+
+    _visible_ma: set[int] = set()
+    _edgecolor_ma = []
+
+    _background = None
+    _background_background = None
+    _creating_background = False
+
+    _length_text: int
+
+    _in_mouse_move = False
+
+    def refresh(self):
+        super().refresh()
+
+        self._set_length_text()
+        return
+

seolpyo_mplchart/_chart/cursor/b_artist.py

@@ -0,0 +1,130 @@
+from matplotlib.axes import Axes
+from matplotlib.collections import LineCollection
+from matplotlib.text import Text
+
+from ..._config import ConfigData
+from ..base.a_canvas import Figure
+
+
+class Base:
+    CONFIG: ConfigData
+
+    figure: Figure
+    ax_price: Axes
+    ax_volume: Axes
+
+    add_artists: callable
+    set_artists: callable
+
+
+class CrossLineMixin(Base):
+    def _add_crosslines(self):
+        kwargs = {'segments': [], 'animated': True}
+
+        self.collection_price_crossline = LineCollection(**kwargs)
+        self.ax_price.add_artist(self.collection_price_crossline)
+
+        self.collection_volume_crossline = LineCollection(**kwargs)
+        self.ax_volume.add_artist(self.collection_volume_crossline)
+        return
+
+    def _set_crosslines(self):
+        kwargs = self.CONFIG.CURSOR.CROSSLINE.__dict__
+        kwargs.update({'segments': [], 'animated': True})
+
+        self.collection_price_crossline.set(**kwargs)
+        self.collection_volume_crossline.set(**kwargs)
+        return
+
+
+class LabelMixin(Base):
+    def _add_artist_labels(self):
+        kwargs = {'text': '', 'animated': True,}
+
+        self.artist_label_x = Text(**kwargs)
+        self.figure.add_artist(self.artist_label_x)
+
+        self.artist_label_y = Text(**kwargs)
+        self.figure.add_artist(self.artist_label_y)
+        return
+
+    def _set_artist_labels(self):
+        kwargs = self.CONFIG.CURSOR.TEXT.to_dict()
+        kwargs.update({'text': ' ', 'animated': True, 'horizontalalignment': 'center', 'verticalalignment': 'center', 'clip_on':True})
+
+        self.artist_label_x.set(**kwargs)
+        self.artist_label_y.set(**kwargs)
+        return
+
+
+class BoxMixin(Base):
+    def _add_box_collections(self):
+        kwargs = {'segments': [], 'animated': True,}
+
+        self.collection_box_price = LineCollection(**kwargs)
+        self.ax_price.add_artist(self.collection_box_price)
+        self.collection_box_volume = LineCollection(**kwargs)
+        self.ax_volume.add_artist(self.collection_box_volume)
+        return
+
+    def _set_box_collections(self):
+        kwargs = self.CONFIG.CURSOR.BOX.__dict__
+        kwargs.update({'segments': [], 'animated': True,})
+
+        self.collection_box_price.set(**kwargs)
+        self.collection_box_volume.set(**kwargs)
+        return
+
+
+class InfoMixin(Base):
+    def _add_info_texts(self):
+        kwargs = {'text': '', 'animated': True, 'horizontalalignment': 'left', 'verticalalignment': 'top',}
+
+        self.artist_info_candle = Text(**kwargs)
+        self.ax_price.add_artist(self.artist_info_candle)
+        self.artist_info_volume = Text(**kwargs)
+        self.ax_volume.add_artist(self.artist_info_volume)
+        return
+
+    def _set_info_texts(self):
+        kwargs = self.CONFIG.CURSOR.TEXT.to_dict()
+        kwargs.update({'text': '', 'animated': True, 'horizontalalignment': 'left', 'verticalalignment': 'top',})
+
+        self.artist_info_candle.set(**kwargs)
+        self.artist_info_volume.set(**kwargs)
+        return
+
+
+class ArtistMixin(CrossLineMixin, LabelMixin, BoxMixin, InfoMixin):
+    collection_price_crossline: LineCollection
+    collection_volume_crossline: LineCollection
+
+    artist_label_x: Text = None
+    artist_label_y: Text = None
+
+    collection_box_price: LineCollection
+    collection_box_volume: LineCollection
+
+    artist_info_candle: Text
+    artist_info_volume: Text
+
+    def add_artists(self):
+        super().add_artists()
+
+        self._add_crosslines()
+        self._add_artist_labels()
+        self._add_box_collections()
+        self._add_info_texts()
+
+        self.set_artists()
+        return
+
+    def set_artists(self):
+        super().set_artists()
+
+        self._set_crosslines()
+        self._set_artist_labels()
+        self._set_box_collections()
+        self._set_info_texts()
+        return
+

seolpyo_mplchart/_chart/cursor/c_draw.py

@@ -0,0 +1,96 @@
+from matplotlib.axes import Axes
+from matplotlib.collections import LineCollection
+from matplotlib.text import Text
+import pandas as pd
+
+from ..._config import ConfigData
+from ..._chart.base.a_canvas import Figure
+
+
+class Base:
+    CONFIG: ConfigData
+
+    df: pd.DataFrame
+
+    watermark: str
+
+    figure: Figure
+    ax_legend: Axes
+    ax_price: Axes
+    ax_volume: Axes
+    artist_watermark: Text
+    collection_candle: LineCollection
+    collection_volume: LineCollection
+    collection_ma: LineCollection
+
+    collection_price_crossline: LineCollection
+    collection_volume_crossline: LineCollection
+
+    artist_label_x: Text
+    artist_label_y: Text
+
+    collection_box_price: LineCollection
+    collection_box_volume: LineCollection
+
+    artist_info_candle: Text
+    artist_info_volume: Text
+
+    in_chart_price: bool
+    in_chart_volume: bool
+
+
+class CrosslineMixin(Base):
+    def _draw_crossline(self):
+        renderer = self.figure.canvas.renderer
+        for artist in [
+            self.collection_price_crossline,
+            self.collection_volume_crossline,
+        ]:
+            artist.draw(renderer)
+        return
+
+
+class LabelMixin(Base):
+    def _draw_label_x(self):
+        artist = self.artist_label_x
+        renderer = self.figure.canvas.renderer
+
+        artist.draw(renderer)
+        # print(f'{artist.get_position()=}')
+        return
+
+    def _draw_label_y(self):
+        artist = self.artist_label_y
+        renderer = self.figure.canvas.renderer
+
+        artist.draw(renderer)
+        return
+
+
+class BoxMixin(Base):
+    def _draw_box_candle(self):
+        renderer = self.figure.canvas.renderer
+        self.collection_box_price.draw(renderer)
+        return
+
+    def _draw_box_volume(self):
+        renderer = self.figure.canvas.renderer
+        self.collection_box_volume.draw(renderer)
+        return
+
+
+class InfoMixin(Base):
+    def _draw_info_candle(self):
+        renderer = self.figure.canvas.renderer
+        self.artist_info_candle.draw(renderer)
+        return
+
+    def _draw_info_volume(self):
+        renderer = self.figure.canvas.renderer
+        self.artist_info_volume.draw(renderer)
+        return
+
+
+class DrawMixin(CrosslineMixin, LabelMixin, BoxMixin, InfoMixin):
+    pass
+

seolpyo_mplchart/_chart/cursor/d_segment.py

@@ -0,0 +1,359 @@
+from fractions import Fraction
+
+from matplotlib.axes import Axes
+from matplotlib.collections import LineCollection
+from matplotlib.text import Text
+from matplotlib.backend_bases import MouseEvent
+import pandas as pd
+
+from ..._config import ConfigData
+from ..._utils.nums import float_to_str
+from ..base.a_canvas import Figure
+
+
+class Base:
+    CONFIG: ConfigData
+
+    key_volume: str
+    _length_text: int
+    df: pd.DataFrame
+
+    watermark: str
+
+    v0: int
+    v1: int
+    vmiddle: int
+    vxmin: int
+    vxmax: int
+    price_ymin: int
+    price_ymax: int
+    volume_ymax: int
+
+    figure: Figure
+    ax_legend: Axes
+    ax_price: Axes
+    ax_volume: Axes
+    artist_watermark: Text
+    collection_candle: LineCollection
+    collection_volume: LineCollection
+    collection_ma: LineCollection
+
+    collection_price_crossline: LineCollection
+    collection_volume_crossline: LineCollection
+
+    artist_label_x: Text
+    artist_label_y: Text
+
+    collection_box_price: LineCollection
+    collection_box_volume: LineCollection
+
+    artist_info_candle: Text
+    artist_info_volume: Text
+
+    in_chart_price: bool
+    in_chart_volume: bool
+
+
+class CrosslineMixin(Base):
+    def _set_crossline(self, e: MouseEvent,):
+        x, y = (e.xdata, e.ydata)
+
+        if self.in_chart_price:
+            seg = [
+                [
+                    (x, self.price_ymin),
+                    (x, self.price_ymax),
+                ],
+                [
+                    (self.vxmin, y),
+                    (self.vxmax, y),
+                ]
+            ]
+            self.collection_price_crossline.set_segments(seg)
+            seg = [
+                [
+                    (x, 0),
+                    (x, self.volume_ymax),
+                ]
+            ]
+            self.collection_volume_crossline.set_segments(seg)
+        elif self.in_chart_volume:
+            seg = [
+                [
+                    (x, self.price_ymin),
+                    (x, self.price_ymax),
+                ]
+            ]
+            self.collection_price_crossline.set_segments(seg)
+            seg = [
+                [
+                    (x, 0),
+                    (x, self.volume_ymax),
+                ],
+                [
+                    (self.vxmin, y),
+                    (self.vxmax, y)
+                ]
+            ]
+            self.collection_volume_crossline.set_segments(seg)
+
+        return
+
+
+class LabelMixin(Base):
+    def _set_label_x(self, e: MouseEvent):
+        xdata, ydata = (e.xdata, e.xdata)
+        # print(f'{(x, xdata)=}')
+
+        if xdata < 0 or xdata is None:
+            return
+
+        try:
+            text = self.df.iloc[int(xdata)]['date']
+        except:
+            return
+        # print(f'{text=}')
+
+        display_coords = e.inaxes.transData.transform((xdata, ydata))
+        figure_coords = self.figure.transFigure.inverted()\
+            .transform(display_coords)
+        # print(f'{figure_coords=}')
+
+        artist = self.artist_label_x
+
+        artist.set_text(text)
+        artist.set_x(figure_coords[0])
+        self._set_label_x_position()
+        return 1
+
+    def _set_label_x_position(self):
+        artist = self.artist_label_x
+        renderer = self.figure.canvas.renderer
+
+        # Axes 하단 경계 좌표
+        boundary = self.ax_volume.get_position()\
+            .y0
+        # print(f'{y0=}')
+
+        if not artist.get_text():
+            artist.set_text(' ')
+
+        # Text bbox 너비
+        bbox = artist.get_bbox_patch()\
+            .get_window_extent(renderer)
+        bbox_size = bbox.height
+        # 밀어야 하는 값
+        fig_size = self.figure.bbox.height
+        offset = (bbox_size + 10) / fig_size
+        # print(f'{box_width_fig=}')
+
+        # x축 값(가격 또는 거래량)
+        # self.artist_label_y.set_x(x1)
+        y = boundary - (offset / 2)
+        # print(f'{(x1, x)=}')
+        artist.set_y(y)
+        return
+
+    def _set_label_y(self, e: MouseEvent, *, is_price_chart):
+        xdata, ydata = (e.xdata, e.ydata)
+        artist = self.artist_label_y
+
+        if is_price_chart:
+            text = float_to_str(ydata, digit=self.CONFIG.UNIT.digit) + self.CONFIG.UNIT.price
+        else:
+            text = float_to_str(ydata, digit=self.CONFIG.UNIT.digit_volume) + self.CONFIG.UNIT.volume
+
+        display_coords = e.inaxes.transData.transform((xdata, ydata))
+        figure_coords = self.figure.transFigure.inverted().transform(display_coords)
+
+        artist.set_text(text)
+        artist.set_y(figure_coords[1])
+        self._set_label_y_position()
+        return
+
+    def _set_label_y_position(self):
+        artist = self.artist_label_y
+        renderer = self.figure.canvas.renderer
+
+        # Axes 우측 경계 좌표
+        boundary = self.ax_volume.get_position()\
+            .x1
+        # print(f'{boundary=}')
+
+        if not artist.get_text():
+            artist.set_text(' ')
+
+        # Text bbox 너비
+        bbox = artist.get_bbox_patch()\
+            .get_window_extent(renderer)
+        bbox_size = bbox.width
+        # 밀어야 하는 값
+        fig_size = self.figure.bbox.width
+        offset = (bbox_size + 8) / fig_size
+        # print(f'{box_width_fig=}')
+
+        # x축 값(가격 또는 거래량)
+        # self.artist_label_y.set_x(x1)
+        x = boundary + (offset / 2)
+        # print(f'{(x1, x)=}')
+        artist.set_x(x)
+        return
+
+
+class BoxMixin(Base):
+    def _set_box_candle(self, segment):
+        self.collection_box_price.set_segments(segment)
+        return
+
+    def _set_box_volume(self, segment):
+        self.collection_box_volume.set_segments(segment)
+        return
+
+
+class InfoMixin(Base):
+    fraction = False
+
+    def _set_info_candle(self, ind):
+        text = self._get_info(ind, is_price=True)
+        self.artist_info_candle.set_text(text)
+
+        # 정보 텍스트를 중앙에 몰리게 설정할 수도 있지만,
+        # 그런 경우 차트를 가리므로 좌우 끝단에 위치하도록 설정
+        if self.vmiddle < ind:
+            self.artist_info_candle.set_x(self.v0)
+        else:
+            # self.artist_info_candle.set_x(self.vmax - self.x_distance)
+            # self.artist_info_candle.set_horizontalalignment('right')
+            # 텍스트박스 크기 가져오기
+            bbox = self.artist_info_candle.get_window_extent()\
+                .transformed(self.ax_price.transData.inverted())
+            width = bbox.x1 - bbox.x0
+            self.artist_info_candle.set_x(self.v1 - width)
+
+        self.artist_info_candle.draw(self.figure.canvas.renderer)
+        return
+
+    def _set_info_volume(self, ind):
+        text = self._get_info(ind, is_price=False)
+        self.artist_info_volume.set_text(text)
+
+        if self.vmiddle < ind:
+            self.artist_info_volume.set_x(self.v0)
+        else:
+            # self.artist_info_volume.set_x(self.vmax - self.x_distance)
+            # self.artist_info_volume.set_horizontalalignment('right')
+            # 텍스트박스 크기 가져오기
+            bbox = self.artist_info_volume.get_window_extent()\
+                .transformed(self.ax_price.transData.inverted())
+            width = bbox.x1 - bbox.x0
+            self.artist_info_volume.set_x(self.v1 - width)
+
+        self.artist_info_volume.draw(self.figure.canvas.renderer)
+        return
+
+    def get_info_kwargs(self, is_price: bool, **kwargs)-> dict:
+        """
+        get text info kwargs
+
+        Args:
+            is_price (bool): is price chart info or not
+
+        Returns:
+            dict[str, any]: text info kwargs
+        """
+        return kwargs
+
+    def _get_info(self, ind: int, is_price=True):
+        # print(f'{self._length_text=}')
+        series = self.df.iloc[ind]
+
+        dt = series['date']
+        if not self.key_volume:
+            v, vr = ('-', '-')
+        else:
+            v, vr = series.loc[['volume', 'rate_volume']]
+            # print(f'{self.CONFIG.UNIT.digit_volume=}')
+            v = float_to_str(v, digit=self.CONFIG.UNIT.digit_volume)
+            # if not v % 1:
+            #     v = int(v)
+            vr = f'{vr:+06,.2f}'
+
+        if is_price:
+            o, h, l, c = (series['open'], series['high'], series['low'], series['close'])
+            rate, compare = (series['rate'], series['compare'])
+            r = f'{rate:+06,.2f}'
+            Or, hr, lr = (series['rate_open'], series['rate_high'], series['rate_low'])
+
+            if self.fraction:
+                data = {}
+                c = round(c, self.CONFIG.UNIT.digit)
+                for value, key in [
+                    [c, 'close'],
+                    [compare, 'compare'],
+                    [o, 'open'],
+                    [h, 'high'],
+                    [l, 'low'],
+                ]:
+                    div = divmod(value, 1)
+                    if div[1]:
+                        if div[0]:
+                            data[key] = f'{float_to_str(div[0])} {Fraction((div[1]))}'
+                        else:
+                            data[key] = f'　 {Fraction((div[1]))}'
+                    else:
+                        data[key] = float_to_str(div[0])
+                # print(f'{data=}')
+
+                kwargs = self.get_info_kwargs(
+                    is_price=is_price,
+                    dt=dt,
+                    close=f'{data["close"]:>{self._length_text}}{self.CONFIG.UNIT.price}',
+                    rate=f'{r:>{self._length_text}}%',
+                    compare=f'{data["compare"]:>{self._length_text}}{self.CONFIG.UNIT.price}',
+                    open=f'{data["open"]:>{self._length_text}}{self.CONFIG.UNIT.price}', rate_open=f'{Or:+06,.2f}%',
+                    high=f'{data["high"]:>{self._length_text}}{self.CONFIG.UNIT.price}', rate_high=f'{hr:+06,.2f}%',
+                    low=f'{data["low"]:>{self._length_text}}{self.CONFIG.UNIT.price}', rate_low=f'{lr:+06,.2f}%',
+                    volume=f'{v:>{self._length_text}}{self.CONFIG.UNIT.volume}', rate_volume=f'{vr}%',
+                )
+                text = self.CONFIG.FORMAT.candle.format(**kwargs)
+            else:
+                o, h, l, c = (
+                    float_to_str(o, digit=self.CONFIG.UNIT.digit),
+                    float_to_str(h, digit=self.CONFIG.UNIT.digit),
+                    float_to_str(l, digit=self.CONFIG.UNIT.digit),
+                    float_to_str(c, digit=self.CONFIG.UNIT.digit),
+                )
+                com = float_to_str(compare, digit=self.CONFIG.UNIT.digit, plus=True)
+
+                kwargs = self.get_info_kwargs(
+                    is_price=is_price,
+                    dt=dt,
+                    close=f'{c:>{self._length_text}}{self.CONFIG.UNIT.price}',
+                    rate=f'{r:>{self._length_text}}%',
+                    compare=f'{com:>{self._length_text}}{self.CONFIG.UNIT.price}',
+                    open=f'{o:>{self._length_text}}{self.CONFIG.UNIT.price}', rate_open=f'{Or:+06,.2f}%',
+                    high=f'{h:>{self._length_text}}{self.CONFIG.UNIT.price}', rate_high=f'{hr:+06,.2f}%',
+                    low=f'{l:>{self._length_text}}{self.CONFIG.UNIT.price}', rate_low=f'{lr:+06,.2f}%',
+                    volume=f'{v:>{self._length_text}}{self.CONFIG.UNIT.volume}', rate_volume=f'{vr}%',
+                )
+                text = self.CONFIG.FORMAT.candle.format(**kwargs)
+        elif self.key_volume:
+            compare = self.df.loc[ind, 'compare_volume']
+            com = float_to_str(compare, digit=self.CONFIG.UNIT.digit_volume, plus=True)
+            kwargs = self.get_info_kwargs(
+                is_price=is_price,
+                dt=dt,
+                volume=f'{v:>{self._length_text}}{self.CONFIG.UNIT.volume}',
+                rate_volume=f'{vr:>{self._length_text}}%',
+                compare=f'{com:>{self._length_text}}{self.CONFIG.UNIT.volume}',
+            )
+            text = self.CONFIG.FORMAT.volume.format(**kwargs)
+        else:
+            text = ''
+
+        return text
+
+
+class SegmentMixin(CrosslineMixin, LabelMixin, BoxMixin, InfoMixin):
+    fraction = False
+