seolpyo-mplchart 1.4.1__py3-none-any.whl → 2.1.0__py3-none-any.whl

seolpyo_mplchart/_chart/base/h_data.py

@@ -0,0 +1,138 @@
+from matplotlib.axes import Axes
+import numpy as np
+import pandas as pd
+
+from ..._config import ConfigData
+
+
+class Base:
+    CONFIG: ConfigData
+    df: pd.DataFrame
+
+    key_date = 'date'
+    key_open, key_high, key_low, key_close = ('open', 'high', 'low', 'close')
+    key_volume = 'volume'
+
+    ax_price: Axes
+    index_list: list[int] = []
+
+    set_segments: callable
+    axis: callable
+
+    def get_default_xlim(self):
+        """
+        get_default_xlim.
+
+        space = int(self.index_list[-1] / 20)
+
+        Returns:
+            (int, int): (-space, self.index_list[-1]+space)
+        """
+        # print(f'{self.index_list[-1]=}')
+        space = int(self.index_list[-1] / 20)
+        return (-space, self.index_list[-1]+space)
+
+    def set_variables(self):
+        self.index_list.clear()
+        self.index_list = self.df.index.tolist()
+        self.xmin, self.xmax = (0, self.index_list[-1])
+
+        self.chart_price_ymax = round(self.df['high'].max() * 1.3, self.CONFIG.UNIT.digit+2)
+        if self.key_volume:
+            self.chart_volume_ymax = round(self.df['volume'].max() * 1.3, self.CONFIG.UNIT.digit_volume+2)
+        else:
+            self.chart_volume_ymax = 10
+
+        if not self.CONFIG.MA.ma_list:
+            self.CONFIG.MA.ma_list = []
+        else:
+            self.CONFIG.MA.ma_list = sorted(self.CONFIG.MA.ma_list)
+
+        return
+
+    def set_data(self, df: pd.DataFrame, *, change_xlim=True):
+        """
+        `if change_xlim`: change xlim with `self.get_default_xlim()` value
+
+        `if not change_xlim`: Keep the current xlim
+        """
+        self.df = self._convert_df(df)
+
+        self._add_columns()
+        # print(f'{self.df.columns=}')
+
+        self.set_variables()
+
+        self.set_segments()
+
+        if change_xlim:
+            xmin, xmax = self.get_default_xlim()
+            self.axis(xmin, xmax=xmax)
+        return
+    
+    def _convert_df(self, df: pd.DataFrame):
+        keys = {
+            self.key_date: 'date',
+            self.key_open: 'open',
+            self.key_high: 'high',
+            self.key_low: 'low',
+            self.key_close: 'close',
+            self.key_volume: 'volume',
+        }
+        df.rename(columns=keys, inplace=True)
+
+        # df column 추출
+        if self.key_volume:
+            df = df[['date', 'open', 'high', 'low', 'close', 'volume']].copy()
+        else:
+            df = df[['date', 'open', 'high', 'low', 'close',]].copy()
+            df['volume'] = 0
+        df.loc[:, 'ymax_volume'] = df['volume'] * 1.2
+
+        # 오름차순 정렬
+        df = df.sort_values(['date'])
+        df = df.reset_index(drop=True)
+
+        return df
+
+    def _add_columns(self):
+        # 전일 종가 추가
+        self.df['pre_close'] = self.df['close'].shift(1).fillna(0)
+        # 거래정지인 경우 전일종가 적용
+        self.df.loc[self.df['close'] == 0, 'close'] = self.df['pre_close']
+        # 종가만 유효한 경우 종가로 통일
+        self.df.loc[(self.df['close'] != 0) & (self.df['open'] == 0), ['open', 'high', 'low']] = self.df['close']
+
+        # 가격이동평균선 계산
+        for ma in self.CONFIG.MA.ma_list:
+            self.df[f'ma{ma}'] = self.df['close'].rolling(ma).mean()
+
+        # 세그먼트 생성을 위한 column 추가
+        self.df['x'] = self.df.index + 0.5
+        self.df['left_candle'] = self.df['x'] - self.CONFIG.CANDLE.half_width
+        self.df['right_candle'] = self.df['x'] + self.CONFIG.CANDLE.half_width
+        self.df['left_volume'] = self.df['x'] - self.CONFIG.VOLUME.half_width
+        self.df['right_volume'] = self.df['x'] + self.CONFIG.VOLUME.half_width
+        self.df['zero'] = 0
+
+        self.df['is_up'] = np.where(self.df['open'] < self.df['close'], True, False)
+        self.df['top_candle'] = np.where(self.df['is_up'], self.df['close'], self.df['open'])
+        self.df['bottom_candle'] = np.where(self.df['is_up'], self.df['open'], self.df['close'])
+
+        return
+
+
+class DataMixin(Base):
+    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
+
+    chart_price_ymax: float
+    chart_volume_ymax: float
+    xmin: int
+    xmax: int
+

seolpyo_mplchart/_chart/base/test.py

@@ -0,0 +1,58 @@
+import sys
+from pathlib import Path
+name_pkg = 'seolpyo_mplchart'
+path_pkg = Path(__file__)
+while path_pkg.name != name_pkg:
+    path_pkg = path_pkg.parent
+sys.path = [path_pkg.parent.__str__()] + sys.path
+
+import json
+
+import pandas as pd
+import matplotlib.pyplot as plt
+
+from seolpyo_mplchart._utils.theme import set_theme
+from seolpyo_mplchart._chart.base import Chart
+
+
+path_file = path_pkg / 'sample' / 'samsung.txt'
+with open(path_file, 'r', encoding='utf-8') as txt:
+    data = json.load(txt)
+df = pd.DataFrame(data[:])
+
+
+class C(Chart):
+    # limit_wick = 200
+    t = 'light'
+    # watermark = ''
+    def __init__(self):
+        super().__init__()
+        self.figure.canvas.mpl_connect('button_press_event', lambda x: self.theme(x))
+    
+    def theme(self, e):
+        btn = getattr(e, 'button')
+        # print(f'{str(btn)=}')
+        if str(btn) == '3':
+            # print('refresh')
+            if self.t == 'light':
+                self.t = 'dark'
+            else:
+                self.t = 'light'
+            # print(f'{self.t=}')
+            self.CONFIG = set_theme(self.CONFIG, theme=self.t)
+            self.refresh()
+        return
+
+
+def run():
+    chart = C()
+    chart.set_data(df)
+    plt.show()
+    plt.close()
+    return
+
+
+if __name__ == '__main__':
+    run()
+
+

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
+