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

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
+

seolpyo_mplchart/_chart/slider/d_segment.py

@@ -0,0 +1,166 @@
+from matplotlib.axes import Axes
+from matplotlib.backend_bases import MouseEvent
+from matplotlib.collections import LineCollection
+from matplotlib.text import Text
+import pandas as pd
+import numpy as np
+
+from ..._config import SliderConfigData
+
+
+class Base:
+    CONFIG: SliderConfigData
+    df: pd.DataFrame
+
+    ax_slider: Axes
+    collection_slider: LineCollection
+    collection_nav: LineCollection
+    collection_slider_vline: LineCollection
+    artist_text_slider: Text
+
+    set_segments: callable
+    set_color_segments: callable
+
+    slider_ymin: float
+    slider_ymax: float
+
+    slider_xmin: float
+    slider_xmax: float
+    _nav_width: float
+
+
+class SliderMixin(Base):
+    def _set_slider_collection(self):
+        keys = []
+        for i in reversed(self.CONFIG.MA.ma_list):
+            keys.append('x')
+            keys.append(f'ma{i}')
+
+        series = self.df[keys + ['x', 'close']]
+        series['x'] = series['x'] - 0.5
+        segment_slider = series.values
+        sizes = [segment_slider.shape[0], len(self.CONFIG.MA.ma_list)+1, 2]
+        segment_slider = segment_slider.reshape(*sizes).swapaxes(0, 1)
+        self.collection_slider.set_segments(segment_slider)
+
+        linewidth = []
+        ma_colors = []
+        for n, _ in enumerate(self.CONFIG.MA.ma_list):
+            linewidth.append(0.9)
+            try:
+                ma_colors.append(self.CONFIG.MA.color_list[n])
+            except:
+                ma_colors.append(self.CONFIG.MA.color_default)
+
+        self.collection_slider.set_edgecolor(ma_colors + [self.CONFIG.CANDLE.line_color])
+
+        self.collection_slider.set_linewidth(linewidth + [2.4])
+        self.collection_slider.set_edgecolor(ma_colors + [self.CONFIG.CANDLE.line_color])
+        return
+
+    def _set_slider_color_collection(self):
+        ma_colors = []
+        for n, _ in enumerate(self.CONFIG.MA.ma_list):
+            try:
+                ma_colors.append(self.CONFIG.MA.color_list[n])
+            except:
+                ma_colors.append(self.CONFIG.MA.color_default)
+
+        self.collection_slider.set_edgecolor(ma_colors + [self.CONFIG.CANDLE.line_color])
+        return
+
+
+class VlineMixin(Base):
+    in_slider: bool
+
+    def _set_slider_vline(self, e: MouseEvent):
+        xdata = e.xdata
+        if xdata is None:
+            return
+        xdata = round(xdata, 2)
+        # print(f'{xdata=}')
+        if not self.in_slider:
+            xdata -= 0.5
+        
+        seg = [((xdata, self.slider_ymin), (xdata, self.slider_ymax))]
+        # print(f'{seg=}')
+        self.collection_slider_vline.set_segments(seg)
+        return
+
+    def _set_slider_text(self, e: MouseEvent):
+        xdata = e.xdata
+        if xdata is None:
+            return
+        idx = round(xdata)
+        if idx < 0:
+            return
+
+        try:
+            text = self.df.iloc[idx]['date']
+        except:
+            return
+
+        self.artist_text_slider.set_text(text)
+        self.artist_text_slider.set_x(round(xdata, 2))
+        return 1
+
+
+class NavMixin(Base):
+    segment_nav: np.ndarray
+
+    def _set_nav_segment(self, xmin, xmax):
+        xmin0 = xmin - self._nav_width
+        xmin1 = xmin
+        xmax0 = xmax
+        xmax1 = xmax + self._nav_width
+        seg = [
+            # 좌측 오버레이
+            (
+                (self.slider_xmin, self.slider_ymax),
+                (xmin0, self.slider_ymax),
+                (xmin0, self.slider_ymin),
+                (self.slider_xmin, self.slider_ymin),
+            ),
+            # 우측 오버레이
+            (
+                (xmax1, self.slider_ymax),
+                (self.slider_xmax, self.slider_ymax),
+                (self.slider_xmax, self.slider_ymin),
+                (xmax1, self.slider_ymin),
+            ),
+            # 좌측 네비게이터
+            (
+                (xmin1, self.slider_ymax),
+                (xmin1, self.slider_ymin),
+                (xmin0, self.slider_ymin),
+                (xmin0, self.slider_ymax),
+            ),
+            # 우측 네비게이터
+            (
+                (xmax0, self.slider_ymin),
+                (xmax0, self.slider_ymax),
+                (xmax1, self.slider_ymax),
+                (xmax1, self.slider_ymin),
+            ),
+        ]
+        self.segment_nav = np.array(seg)
+
+        self.collection_nav.set_segments(self.segment_nav)
+        return
+
+
+class SegmentMixin(SliderMixin, VlineMixin, NavMixin):
+    segment_nav: np.ndarray
+
+    def set_segments(self):
+        super().set_segments()
+
+        self._set_slider_collection()
+        self._set_slider_color_collection()
+        return
+
+    def set_color_segments(self):
+        super().set_color_segments()
+
+        self._set_slider_color_collection()
+        return

seolpyo_mplchart/_chart/slider/e_axis.py

@@ -0,0 +1,70 @@
+from matplotlib.collections import LineCollection
+import numpy as np
+
+
+class Base:
+    limit_volume = 200
+    limit_ma = 8_000
+
+    x_click: int
+    vxmin: int
+    vxmax: int
+
+    axis: callable
+    _set_nav_segment: callable
+    set_collections: callable
+
+    key_volume: str
+    segment_volume_wick: np.ndarray
+    _set_volume_collection_wick_segments: callable
+    _set_ma_collection_segments: callable
+    facecolor_volume: np.ndarray
+    edgecolor_volume: np.ndarray
+    collection_volume: LineCollection
+    collection_ma: LineCollection
+
+
+class Mixin(Base):
+    def axis(self, xmin, *, xmax):
+        super().axis(xmin, xmax=xmax)
+        self._set_nav_segment(self.vxmin, xmax=self.vxmax-1)
+        return
+
+    def _set_volume_collection_wick_segments(self, ind_start, ind_end):
+        if not self.key_volume or not self.x_click:
+            super()._set_volume_collection_wick_segments(ind_start, ind_end=ind_end)
+        else:
+            indsub = ind_end - ind_start
+            if indsub <= self.limit_volume:
+                super()._set_volume_collection_wick_segments(ind_start, ind_end=ind_end)
+            else:
+                # 일부 거래량만 그리기
+                seg_volume = self.segment_volume_wick[ind_start:ind_end]
+                values = seg_volume[:, 1, 1]
+                top_index = np.argsort(-values)[:self.limit_volume]
+                seg = seg_volume[top_index]
+                facecolors = self.facecolor_volume[ind_start:ind_end][top_index]
+                edgecolors = self.edgecolor_volume[ind_start:ind_end][top_index]
+                self.collection_volume.set_segments(seg)
+                self.collection_volume.set_linewidth(1.3)
+                self.collection_volume.set_facecolor(facecolors)
+                self.collection_volume.set_edgecolor(edgecolors)
+        return
+
+    def _set_ma_collection_segments(self, ind_start, ind_end):
+        if not self.x_click:
+            super()._set_ma_collection_segments(ind_start, ind_end=ind_end)
+        else:
+            indsub = ind_end - ind_start
+            if indsub <= self.limit_ma:
+                super()._set_ma_collection_segments(ind_start, ind_end=ind_end)
+            else:
+                # 이평선 그리지 않기
+                self.collection_ma.set_segments([])
+        return
+
+
+class AxisMixin(Mixin):
+    limit_volume = 200
+    limit_ma = 8_000
+

seolpyo_mplchart/_chart/slider/f_background.py

@@ -0,0 +1,37 @@
+from matplotlib.axes import Axes
+
+
+class Base:
+    _background = None
+    _background_background = None
+
+    _creating_background = False
+
+    draw_background: callable
+    _draw_ax: callable
+    _draw_ax_slider: callable
+    draw_artists: callable
+    _draw_nav: callable
+
+
+class SliderMixin(Base):
+    ax_slider: Axes
+
+    def draw_background(self):
+        self._draw_ax(self.ax_slider)
+        self._draw_ax_slider()
+
+        super().draw_background()
+        return
+
+    def draw_artists(self):
+        super().draw_artists()
+
+        self._draw_nav()
+        return
+
+
+class BackgroundMixin(SliderMixin):
+    _background = None
+    _creating_background = False
+