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

seolpyo_mplchart/_chart/cursor/h_data.py

@@ -0,0 +1,61 @@
+import pandas as pd
+
+from ..._config import ConfigData
+
+class Base:
+    CONFIG: ConfigData
+
+    key_volume: str
+    df: pd.DataFrame
+
+    _add_columns: callable
+    set_variables: callable
+
+
+class Mixin(Base):
+    def _add_columns(self):
+        super()._add_columns()
+
+        self.df['compare'] = (self.df['close'] - self.df['pre_close']).fillna(0)
+        self.df['rate'] = (self.df['compare'] * 100 / self.df['pre_close']).__round__(2).fillna(0)
+        self.df['rate_open'] = ((self.df['open'] - self.df['pre_close']) * 100 / self.df['pre_close']).__round__(2).fillna(0)
+        self.df['rate_high'] = ((self.df['high'] - self.df['pre_close']) * 100 / self.df['pre_close']).__round__(2).fillna(0)
+        self.df['rate_low'] = ((self.df['low'] - self.df['pre_close']) * 100 / self.df['pre_close']).__round__(2).fillna(0)
+        if self.key_volume:
+            self.df['pre_volume'] = self.df['volume'].shift(1)
+            self.df['compare_volume'] = (self.df['volume'] - self.df['pre_volume']).fillna(0)
+            self.df['rate_volume'] = (self.df['compare_volume'] * 100 / self.df['pre_volume']).__round__(2).fillna(0)
+
+        self.df['space_box_candle'] = (self.df['high'] - self.df['low']) / 5
+        self.df['box_candle_top'] = self.df['high'] + self.df['space_box_candle']
+        self.df['box_candle_bottom'] = self.df['low'] - self.df['space_box_candle']
+        self.df['box_candle_height'] = self.df['box_candle_top'] - self.df['box_candle_bottom']
+
+        if self.key_volume:
+            self.df['box_volume_top'] = self.df['volume'] * 1.15
+
+        return
+
+    def set_variables(self):
+        super().set_variables()
+
+        self._set_length_text()
+        return
+
+    def _set_length_text(self):
+        func = lambda x: len(self.CONFIG.UNIT.func(x, digit=self.CONFIG.UNIT.digit, word=self.CONFIG.UNIT.price))
+        self._length_text = self.df['high'].apply(func).max()
+
+        if self.key_volume:
+            func = lambda x: len(self.CONFIG.UNIT.func(x, digit=self.CONFIG.UNIT.digit_volume, word=self.CONFIG.UNIT.volume))
+            lenth_volume = self.df['volume'].apply(func).max()
+            # print(f'{self._length_text=}')
+            # print(f'{lenth_volume=}')
+            if self._length_text < lenth_volume:
+                self._length_text = lenth_volume
+        return
+
+
+class DataMixin(Mixin):
+    _length_text: int
+

seolpyo_mplchart/_chart/cursor/test.py

@@ -0,0 +1,69 @@
+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.cursor import Chart
+from seolpyo_mplchart._config import DEFAULTCONFIG, DEFAULTCONFIG_EN
+
+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[:300])
+
+
+class C(Chart):
+    # limit_candle = 200
+    # limit_wick = 200
+    t = 'light'
+    # watermark = ''
+    def __init__(self):
+        super().__init__()
+        self.figure.canvas.mpl_connect('button_press_event', lambda x: self.theme(x))
+        self.CONFIG.FORMAT.candle += '\nCustom: {custom}'
+
+    def get_info_kwargs(self, is_price, **kwargs):
+        kwargs = super().get_info_kwargs(is_price, **kwargs)
+        kwargs['close'] = 'Cusotom close value'
+        kwargs['custom'] = 'this is custom add info kwargs'
+        return kwargs
+
+    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'
+                self.CONFIG = set_theme(DEFAULTCONFIG_EN, theme=self.t)
+            else:
+                self.t = 'light'
+                self.CONFIG = set_theme(DEFAULTCONFIG, theme=self.t)
+            # print(f'{self.t=}')
+            # self.CONFIG.MA.ma_list = []
+            self.refresh()
+        return
+
+
+def run():
+    chart = C()
+    chart.set_data(df)
+    plt.show()
+    plt.close()
+    return
+
+
+if __name__ == '__main__':
+    run()
+
+

seolpyo_mplchart/_chart/slider/__init__.py

@@ -0,0 +1,169 @@
+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 ..._config import SLIDERCONFIG, SliderConfigData
+from ..base import Chart as BaseChart
+from ..cursor import CursorMixin
+
+from .a_canvas import CanvasMixin, Figure
+from .b_artist import ArtistMixin
+from .c_draw import DrawMixin
+from .d_segment import SegmentMixin
+from .e_axis import AxisMixin
+from .f_background import BackgroundMixin
+from .g_event import EventMixin
+from .h_data import DataMixin
+
+
+class SliderMixin(
+    CanvasMixin,
+    ArtistMixin,
+    DrawMixin,
+    SegmentMixin,
+    AxisMixin,
+    BackgroundMixin,
+    EventMixin,
+    DataMixin,
+):
+    pass
+
+
+class Chart(SliderMixin, CursorMixin, BaseChart):
+    candle_on_ma = True
+    slider_top = True
+    fraction = False
+
+    limit_candle = 400
+    limit_wick = 2_000
+    limit_volume = 200
+    limit_ma = 8_000
+
+    min_distance = 5
+
+    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: SliderConfigData
+
+    figure: Figure
+    ax_legend: Axes
+    ax_price: Axes
+    ax_volume: Axes
+
+    ax_slider: Axes
+    ax_none: Axes
+    _ax_slider_top: Axes
+    _ax_slider_bottom: Axes
+
+    artist_watermark: Text
+    collection_candle: LineCollection
+    collection_volume: LineCollection
+    collection_ma: LineCollection
+
+    collection_slider: LineCollection
+    collection_nav: LineCollection
+    collection_slider_vline: LineCollection
+    artist_text_slider: Text
+
+    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
+
+    v0: int
+    v1: int
+    vmiddle: int
+
+    min_height_box_candle: float
+    min_height_box_volume: float
+
+    _length_text: int
+
+    _in_mouse_move = False
+
+    ###
+
+    segment_nav: np.ndarray
+    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
+
+    _nav_width: float
+
+    slider_xmin: int
+    slider_xmax: int
+    slider_ymin: float
+    slider_ymax: float
+
+    in_chart = False
+    in_slider = False
+    in_chart_price = False
+    in_chart_volume = False
+
+    is_move_chart = False
+    is_click_slider = False
+    is_click_chart = False
+    x_click: float
+
+    click_nav_left = False
+    click_nav_right = False
+
+    navcoordinate: tuple[int, int]
+
+    ###
+
+    _visible_ma: set[int] = set()
+    _edgecolor_ma = []
+
+    _background = None
+    _creating_background = False
+
+    def __init__(self, config=SLIDERCONFIG):
+        self.CONFIG = config
+        super().__init__()
+        return
+
+    def refresh(self):
+        self._set_slider_artists()
+        super().refresh()
+        return
+

seolpyo_mplchart/_chart/slider/a_canvas.py

@@ -0,0 +1,260 @@
+import matplotlib.pyplot as plt
+from matplotlib.axes import Axes
+from matplotlib.collections import LineCollection
+from matplotlib.text import Text
+
+from ..._config import SLIDERCONFIG, SliderConfigData
+from ..base.a_canvas import CanvasMixin as BaseMixin, Figure
+
+
+class Base(BaseMixin):
+    CONFIG: SliderConfigData
+
+    def add_axes(self):
+        if not self.figure:
+            self.figure, *_ = plt.subplots(
+                7, # row 수
+                figsize=self.CONFIG.FIGURE.figsize, # 기본 크기
+                height_ratios=(
+                    0,
+                    0,
+                    0,
+                    self.CONFIG.FIGURE.RATIO.price,
+                    self.CONFIG.FIGURE.RATIO.volume,
+                    0,
+                    0,
+                ) # row 크기 비율
+            )
+
+        self._ax_slider_top, self.ax_none_top, self.ax_legend, self.ax_price, self.ax_volume, self.ax_none_bottom, self._ax_slider_bottom = self.figure.axes
+
+        return
+
+
+class FigureMixin(Base):
+    key_volume: str
+    ax_legend: Axes
+
+    slider_top = True
+
+    def _set_figure_ratios(self):
+        gs = self.figure.axes[0].get_subplotspec().get_gridspec()
+
+        ratio_volume = self.CONFIG.FIGURE.RATIO.volume
+        if not self.key_volume:
+            ratio_volume = 0
+
+        legend = self.ax_legend.get_legend()
+        if not legend:
+            if self.slider_top:
+                ratios = [
+                    self.CONFIG.FIGURE.RATIO.slider,
+                    0,
+                    0,
+                    self.CONFIG.FIGURE.RATIO.price, ratio_volume,
+                    0,
+                    0,
+                ]
+            else:
+                ratios = [
+                    0,
+                    0,
+                    0,
+                    self.CONFIG.FIGURE.RATIO.price, ratio_volume,
+                    self.CONFIG.FIGURE.RATIO.none,
+                    self.CONFIG.FIGURE.RATIO.slider,
+                ]
+        else:
+            fig_heihgt = self.figure.get_figheight()
+            fig_px = fig_heihgt * (1-self.CONFIG.FIGURE.ADJUST.hspace*2) * self.figure.dpi
+            # print(f'{(fig_heihgt, fig_px)=}')
+
+            # Legend에 Axes 높이 맞추기
+            bbox = legend.get_window_extent().transformed(self.figure.transFigure.inverted())
+            ax_pos = self.ax_legend.get_position()
+            self.ax_legend.set_position([ax_pos.x0, ax_pos.y0, ax_pos.width, bbox.height])
+
+            legend_height = bbox.height
+            legend_px = legend.get_window_extent().height
+            # print(f'{(legend_height, legend_px)=}')
+
+            chart_px = fig_px - legend_height
+            chart_ratio = self.CONFIG.FIGURE.RATIO.price + ratio_volume
+
+            # print(f'{self.CONFIG.FIGURE.RATIO.__dict__=}')
+            ratio_none = 0 if self.slider_top else self.CONFIG.FIGURE.RATIO.none
+            chart_ratio += self.CONFIG.FIGURE.RATIO.slider
+            div_chart = chart_px / chart_ratio
+            price_px = div_chart * self.CONFIG.FIGURE.RATIO.price
+            volume_px = div_chart * ratio_volume
+            slider_px = div_chart * self.CONFIG.FIGURE.RATIO.slider
+            none_px = div_chart * ratio_none
+            # print(f'{none_px=}')
+
+            if self.slider_top:
+                # 차트 비율
+                ratios = [
+                    slider_px,
+                    0,
+                    legend_px * 3,
+                    price_px, volume_px,
+                    0,
+                    0,
+                ]
+            else:
+                ratios = [
+                    0,
+                    0,
+                    legend_px * 1.2,
+                    price_px, volume_px,
+                    none_px,
+                    slider_px,
+                ]
+
+        # print(f'{ratios=}')
+        gs.set_height_ratios(ratios)
+
+        self.figure.tight_layout()
+
+        # 플롯간 간격 설정(Configure subplots)
+        self.figure.subplots_adjust(**self.CONFIG.FIGURE.ADJUST.__dict__)
+
+        return
+
+    def _set_figure(self):
+        self.figure.canvas.manager.set_window_title('Seolpyo MPLChart')
+
+        # print(f'{self.CONFIG.FIGURE.RATIO.volume=}')
+        # print(f'{gs.get_height_ratios()=}')
+
+        self._set_figure_ratios()
+
+        self.figure.set_facecolor(self.CONFIG.FIGURE.facecolor)
+        return
+
+
+class AxesMixin(Base):
+    slider_top = True
+
+    ax_slider: Axes = None
+
+    collection_slider: LineCollection
+    collection_nav: LineCollection
+    collection_slider_vline: LineCollection
+    artist_text_slider: Text
+
+    def _set_axes(self):
+        super()._set_axes()
+
+        if self.slider_top:
+            ax_slider = self._ax_slider_top
+        else:
+            ax_slider = self._ax_slider_bottom
+
+        if self.ax_slider and ax_slider is not self.ax_slider:
+            # print('move artist')
+            # ax_slider 위치가 변경된 경우 artist 이동하기
+            artists: list[LineCollection|Text] = [
+                self.collection_slider,
+                self.collection_nav,
+                self.collection_slider_vline,
+                self.artist_text_slider,
+            ]
+            for artist in artists:
+                artist.remove()
+                artist.set_transform(ax_slider.transData)
+                ax_slider.add_artist(artist)
+            # axis
+            ax_slider.set_xlim(*self.ax_slider.get_xlim())
+            ax_slider.set_ylim(*self.ax_slider.get_ylim())
+            # tick label
+            ax_slider.set_xticks(self.ax_slider.get_xticks(minor=True), minor=True)
+            ax_slider.set_xticklabels(self.ax_slider.get_xticklabels(minor=True), minor=True)
+
+        self.ax_slider = ax_slider
+
+        self._ax_slider_top.set_label('top slider ax')
+        self._ax_slider_bottom.set_label('bottom slider ax')
+        self.ax_none_top.set_label('top none ax')
+        self.ax_none_bottom.set_label('bottom none ax')
+
+        self.ax_slider.set_label('slider ax')
+
+        for ax in (self.ax_none_top, self.ax_none_bottom):
+            ax.set_animated(True)
+            ax.set_axis_off()
+
+        self._set_axes_slider()
+
+        return
+
+    def _set_axes_slider(self):
+        # print(f'{self.slider_top=}')
+        formatter = lambda x, _: self.CONFIG.UNIT.func(
+            x,
+            word=self.CONFIG.UNIT.price,
+            digit=self.CONFIG.UNIT.digit+2
+        )
+        # 공통 설정
+        for ax in (self._ax_slider_top, self._ax_slider_bottom):
+            ax.yaxis.set_major_formatter(formatter)
+            # x tick 외부 눈금 표시하지 않기
+            ax.xaxis.set_ticks_position('none')
+            # x tick label 제거
+            ax.set_xticklabels([])
+            # y tick 눈금 표시하지 않기
+            ax.yaxis.set_ticks_position('none')
+
+            # 차트 영역 배경 색상
+            ax.set_facecolor(self.CONFIG.AX.facecolor)
+
+            # Axes 외곽선 색 변경(틱 색과 일치)
+            for i in ['top', 'bottom', 'left', 'right']:
+                ax.spines[i].set_color(self.CONFIG.AX.TICK.edgecolor)
+            # 틱 색상
+            ax.tick_params('both', colors=self.CONFIG.AX.TICK.edgecolor)
+            # 틱 라벨 색상
+            ticklabels: list[Text] = ax.get_xticklabels() + ax.get_yticklabels()
+            for ticklabel in ticklabels:
+                ticklabel.set_color(self.CONFIG.AX.TICK.fontcolor)
+
+            # Axes grid(구분선, 격자) 그리기
+            # 어째서인지 grid의 zorder 값을 선언해도 1.6을 값으로 한다.
+            ax.grid(**self.CONFIG.AX.GRID.__dict__)
+
+            # major tick mark 길이를 0으로 만들어 튀어나오지 않게 하기
+            ax.tick_params('x', which='major', length=0)
+            # minor tick mark 색상 변경
+            ax.tick_params('x', which='minor', colors=self.CONFIG.AX.TICK.edgecolor)
+            # 틱 라벨 색상
+            ticklabels: list[Text] = ax.get_xticklabels(minor=True) + ax.get_yticklabels()
+            for ticklabel in ticklabels:
+                ticklabel.set_color(self.CONFIG.AX.TICK.fontcolor)
+
+            ax.yaxis.set_ticks_position('right')
+        # 상단 슬라이더의 x tick 하단 설정
+        self._ax_slider_top.xaxis.set_ticks_position('bottom')
+        # 하단 슬라이더의 x tick 상단 설정
+        self._ax_slider_bottom.xaxis.set_ticks_position('top')
+
+        return
+
+
+class CanvasMixin(FigureMixin, AxesMixin):
+    slider_top = True
+
+    figure: Figure
+
+    ax_legend: Axes
+    ax_price: Axes
+    ax_volume: Axes
+    ax_slider: Axes
+    ax_none: Axes
+    _ax_slider_top: Axes
+    _ax_slider_bottom: Axes
+
+    def __init__(self, config=SLIDERCONFIG):
+        super().__init__(config=config)
+        return
+
+

seolpyo_mplchart/_chart/slider/b_artist.py

@@ -0,0 +1,91 @@
+from matplotlib.axes import Axes
+from matplotlib.collections import LineCollection
+from matplotlib.text import Text
+
+from ..._config import SliderConfigData
+
+
+class Base:
+    CONFIG: SliderConfigData
+
+    ax_slider: Axes
+    ax_price: Axes
+    ax_volume: Axes
+    ax_legend: Axes
+
+    add_artists: callable
+    set_artists: callable
+
+
+class CollectionMixin(Base):
+    def add_artists(self):
+        super().add_artists()
+
+        self.collection_slider = LineCollection([])
+        self.collection_nav = LineCollection([])
+        self.collection_slider_vline = LineCollection([])
+        self.artist_text_slider = Text('')
+
+        for artist in [
+            self.collection_slider,
+            self.collection_nav,
+            self.collection_slider_vline,
+            self.artist_text_slider
+        ]:
+            self.ax_slider.add_artist(artist)
+        return
+
+    def set_artists(self):
+        super().set_artists()
+
+        self._set_slider_artists()
+        self._set_slider_test()
+        return
+
+    def _set_slider_artists(self):
+        self.collection_slider.set_animated(True)
+        self.collection_slider.set_transform(self.ax_slider.transData)
+
+        color_overay = self.CONFIG.SLIDER.NAV.facecolor
+        color_line = self.CONFIG.SLIDER.NAV.edgecolor
+        colors = [color_overay, color_overay, color_line, color_line]
+        # print(f'{colors=}')
+        self.collection_nav.set_linewidth(0.1)
+        self.collection_nav.set_facecolor(colors)
+
+        # facecolor alpha값 제거
+        colors = []
+        for c in self.collection_nav.get_facecolor():
+            # print(c)
+            colors.append(c[:3])
+        self.collection_nav.set_facecolor(colors)
+
+        alpha = self.CONFIG.SLIDER.NAV.alpha
+        self.collection_nav.set_alpha([alpha, alpha, 1, 1])
+
+        self.collection_nav.set_edgecolor([(0, 0, 0, 0) for _ in colors])
+        self.collection_nav.set_animated(True)
+        self.collection_nav.set_transform(self.ax_slider.transData)
+
+        kwargs = self.CONFIG.CURSOR.CROSSLINE.__dict__
+        kwargs.update({'segments': [], 'animated': True})
+        self.collection_slider_vline.set(**kwargs)
+
+        kwargs = self.CONFIG.CURSOR.TEXT.to_dict()
+        kwargs.update({'text': ' ', 'animated': True})
+        self.artist_text_slider.set(**kwargs)
+        return
+
+    def _set_slider_test(self):
+        kwargs = self.CONFIG.CURSOR.TEXT.to_dict()
+        kwargs.update(animated=True, verticalalignment='top', horizontalalignment='center')
+        self.artist_text_slider.set(**kwargs)
+        return
+
+
+class ArtistMixin(CollectionMixin):
+    collection_slider: LineCollection
+    collection_nav: LineCollection
+    collection_slider_vline: LineCollection
+    artist_text_slider: Text
+

seolpyo_mplchart/_chart/slider/c_draw.py

@@ -0,0 +1,54 @@
+from matplotlib.axes import Axes
+from matplotlib.collections import LineCollection
+from matplotlib.text import Text
+
+from .a_canvas import Figure
+
+
+class Base:
+    figure: Figure
+
+    ax_slider: Axes
+
+    collection_slider: LineCollection
+    collection_nav: LineCollection
+    collection_slider_vline: LineCollection
+    artist_text_slider: Text
+
+
+class SliderMixin(Base):
+    def _draw_ax_slider(self):
+        renderer = self.figure.canvas.renderer
+
+        self.ax_slider.xaxis.draw(renderer)
+        self.ax_slider.yaxis.draw(renderer)
+
+        self.collection_slider.draw(renderer)
+        c = self.collection_slider.get_edgecolor()
+        # print(f'{c=}')
+        s = self.collection_slider.get_segments()
+        # print(f'{s=}')
+        return
+
+    def _draw_nav(self):
+        renderer = self.figure.canvas.renderer
+
+        self.collection_nav.draw(renderer)
+        return
+
+    def _draw_slider_vline(self):
+        renderer = self.figure.canvas.renderer
+
+        self.collection_slider_vline.draw(renderer)
+        return
+
+    def _draw_slider_text(self):
+        renderer = self.figure.canvas.renderer
+
+        self.artist_text_slider.draw(renderer)
+        return
+
+
+class DrawMixin(SliderMixin):
+    pass
+