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

seolpyo_mplchart/_chart/base/a_canvas.py

@@ -0,0 +1,250 @@
+import matplotlib.pyplot as plt
+import matplotlib.style as mplstyle
+from matplotlib.artist import Artist
+from matplotlib.axes import Axes
+from matplotlib.backends.backend_agg import FigureCanvasAgg, RendererAgg
+from matplotlib.backend_bases import FigureManagerBase
+from matplotlib.figure import Figure as Fig
+from matplotlib.text import Text
+
+try: plt.switch_backend('TkAgg')
+except: pass
+
+# 한글 깨짐 문제 방지
+try: plt.rcParams['font.family'] ='Malgun Gothic'
+except: pass
+
+from ..._config import DEFAULTCONFIG, ConfigData
+
+
+mplstyle.use('fast')
+
+
+class Canvas(FigureCanvasAgg):
+    manager: FigureManagerBase
+    renderer = RendererAgg
+
+class Figure(Fig):
+    canvas: Canvas
+
+
+class Base:
+    CONFIG: ConfigData
+    figure: Figure = None
+
+    def add_axes(self):
+        if not self.figure:
+            self.figure, *_ = plt.subplots(
+                3, # row 수
+                figsize=self.CONFIG.FIGURE.figsize, # 기본 크기
+                height_ratios=(
+                    1,
+                    self.CONFIG.FIGURE.RATIO.price,
+                    self.CONFIG.FIGURE.RATIO.volume,
+                ) # row 크기 비율
+            )
+
+        self.ax_legend, self.ax_price, self.ax_volume = self.figure.axes
+        self.ax_legend.set_label('legend ax')
+        self.ax_price.set_label('price ax')
+        self.ax_volume.set_label('volume ax')
+
+        return
+
+
+class FigureMixin(Base):
+    key_volume: str
+    ax_legend: Axes
+
+    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:
+            ratios = [
+                0,
+                self.CONFIG.FIGURE.RATIO.price, ratio_volume
+            ]
+        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
+            div_chart = chart_px / chart_ratio
+            price_px = div_chart * self.CONFIG.FIGURE.RATIO.price
+            volume_px = div_chart * ratio_volume
+
+            # 차트 비율 변경
+            ratios = [
+                legend_px * 1.2,
+                price_px, volume_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(FigureMixin):
+    def _set_axes(self):
+        # ax 요소 animated 처리
+        for ax in self.figure.axes:
+            ax.patch.set_animated(True)
+
+            # ax 경계선
+            for spine in ax.spines.values():
+                spine.set_animated(True)
+
+            for axis in (ax.xaxis, ax.yaxis):
+                axis.set_animated(True)
+                axis.label.set_animated(True)
+                for tick in axis.get_major_ticks() + axis.get_minor_ticks():
+                    artists: list[Artist] = [
+                        tick.tick1line, tick.tick2line,
+                        tick.gridline,
+                        tick.label1, tick.label2
+                    ]
+                    for artist in artists:
+                        if artist is not None:
+                            artist.set_animated(True)
+
+        # y ticklabel foramt 설정
+        self.ax_price.yaxis.set_major_formatter(
+            lambda x, _: self.CONFIG.UNIT.func(
+                x,
+                word=self.CONFIG.UNIT.price,
+                digit=self.CONFIG.UNIT.digit+(0 if self.CONFIG.UNIT.digit % 1 else 2)
+            )
+        )
+        self.ax_volume.yaxis.set_major_formatter(
+            lambda x, _: self.CONFIG.UNIT.func(
+                x,
+                word=self.CONFIG.UNIT.volume,
+                digit=self.CONFIG.UNIT.digit_volume
+            )
+        )
+
+        if not self.key_volume:
+            # tick 그리지 않기
+            self.ax_volume.set_yticklabels([])
+            self.ax_volume.set_yticks([])
+
+        # 공통 설정
+        for ax in (self.ax_price, self.ax_volume):
+            ax.xaxis.set_animated(True)
+            ax.yaxis.set_animated(True)
+
+            # x tick 외부 눈금 표시하지 않기
+            ax.xaxis.set_ticks_position('none')
+            # x tick label 제거
+            ax.set_xticklabels([])
+            # y tick 위치를 우측으로 이동
+            ax.tick_params(left=False, right=True, labelleft=False, labelright=True)
+
+            # 차트 영역 배경 색상
+            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__)
+        
+        # 거래량 차트의 x tick 외부 눈금 표시하기
+        self.ax_volume.xaxis.set_ticks_position('bottom')
+        # major tick mark 길이를 0으로 만들어 튀어나오지 않게 하기
+        self.ax_volume.tick_params('x', which='major', length=0)
+        # minor tick mark 색상 변경
+        self.ax_volume.tick_params('x', which='minor', colors=self.CONFIG.AX.TICK.edgecolor)
+        return
+
+
+class LegendMixin(AxesMixin):
+    def _set_axes_legend(self):
+        # 이평선 라벨 axis 그리지 않기
+        self.ax_legend.set_axis_off()
+        self.ax_legend.xaxis.set_animated(True)
+        self.ax_legend.yaxis.set_animated(True)
+        self.ax_legend.set_animated(True)
+
+        # 이평선 라벨 Axes 배경색
+        legends = self.ax_legend.get_legend()
+        if legends:
+            legends.get_frame().set_facecolor(self.CONFIG.AX.facecolor)
+
+        # 이평선 라벨 Axes 테두리색
+        legends = self.ax_legend.get_legend()
+        if legends:
+            legends.get_frame().set_edgecolor(self.CONFIG.AX.TICK.edgecolor)
+
+        # 이평선 라벨 폰트 색상
+        fontcolor = self.CONFIG.AX.TICK.fontcolor
+        legends = self.ax_legend.get_legend()
+        if legends:
+            legend_labels: list[Text] = legends.texts
+            for i in legend_labels:
+                i.set_color(fontcolor)
+        return
+
+
+class CanvasMixin(LegendMixin):
+    figure: Figure
+    ax_legend: Axes
+    ax_price: Axes
+    ax_volume: Axes
+
+    def __init__(self, config=DEFAULTCONFIG):
+        # 기본 툴바 비활성화
+        plt.rcParams['toolbar'] = 'None'
+        # plt.rcParams['figure.dpi'] = 600
+
+        self.CONFIG = config
+        self.add_axes()
+        self.set_canvas()
+        return
+
+    def set_canvas(self):
+        self._set_axes()
+        self._set_axes_legend()
+        self._set_figure()
+        return
+

seolpyo_mplchart/_chart/base/b_artist.py

@@ -0,0 +1,143 @@
+from matplotlib.axes import Axes
+from matplotlib.collections import LineCollection
+from matplotlib.lines import Line2D
+from matplotlib.text import Text
+
+from ..._config import ConfigData
+
+
+class WatermarkMixin:
+    CONFIG: ConfigData
+
+    watermark = 'seolpyo mplchart'
+
+    ax_price: Axes
+
+    def _add_watermark(self):
+        self.artist_watermark = Text(
+            x=0.5, y=0.5, text='',
+            animated=True,
+            horizontalalignment='center', verticalalignment='center',
+            transform=self.ax_price.transAxes
+        )
+        self.ax_price.add_artist(self.artist_watermark)
+        return
+
+    def _set_watermark(self):
+        self.artist_watermark.set(animated=True, horizontalalignment='center', verticalalignment='center',)
+        self.artist_watermark.set_text(self.watermark)
+        self.artist_watermark.set_fontsize(self.CONFIG.FIGURE.WATERMARK.fontsize)
+        self.artist_watermark.set_color(self.CONFIG.FIGURE.WATERMARK.color)
+        self.artist_watermark.set_alpha(self.CONFIG.FIGURE.WATERMARK.alpha)
+        return
+
+
+class CollectionMixin:
+    ax_price: Axes
+    ax_volume: Axes
+    ax_legend: Axes
+
+    def _add_collections(self):
+        self.collection_candle = LineCollection([], animated=True, linewidths=0.8)
+        self.ax_price.add_collection(self.collection_candle)
+
+        self.collection_ma = LineCollection([], animated=True, linewidths=1)
+        self.ax_price.add_collection(self.collection_ma)
+
+        self.collection_volume = LineCollection([], animated=True, linewidths=1)
+        self.ax_volume.add_collection(self.collection_volume)
+        return
+
+    def set_collection_candle(self, segment, facecolors, edgecolors):
+        self.collection_candle.set_segments(segment)
+        self.collection_candle.set_facecolor(facecolors)
+        self.collection_candle.set_edgecolor(edgecolors)
+        self.collection_candle.set_transform(self.ax_price.transData)
+        return
+
+    def set_collection_volume(self, segment, facecolors, edgecolors):
+        self.collection_volume.set_segments(segment)
+        self.collection_volume.set_facecolor(facecolors)
+        self.collection_volume.set_edgecolor(edgecolors)
+        self.collection_volume.set_transform(self.ax_volume.transData)
+        self.collection_volume.set_antialiased(False)
+        return
+
+
+class MaMixin:
+    CONFIG: ConfigData
+    _visible_ma = set()
+
+    ax_legend: Axes
+    ax_price: Axes
+    collection_ma: LineCollection
+    _set_figure_ratios: callable
+
+    def set_collection_ma(self, segment, edgecolors):
+        self.collection_ma.set_segments(segment)
+        self.collection_ma.set_facecolor([])
+        self.collection_ma.set_edgecolor(edgecolors)
+        # print(self.collection_ma.get_linewidth())
+        self.collection_ma.set_linewidth(self.CONFIG.MA.linewidth)
+        self.collection_ma.set_transform(self.ax_price.transData)
+        return
+
+    def _set_legends(self):
+        legends = self.ax_legend.get_legend()
+        if legends:
+            legends.remove()
+
+        self._visible_ma.clear()
+
+        label_list, handle_list, edgecolor_list = ([], [], [])
+        # Legend Ax에 표시하는 선 segment
+        arr = [0, 1]
+        for n, ma in enumerate(self.CONFIG.MA.ma_list):
+            self._visible_ma.add(ma)
+            label_list.append(self.CONFIG.MA.format.format(ma))
+            try:
+                color = self.CONFIG.MA.color_list[n]
+            except:
+                color = self.CONFIG.MA.color_default
+            edgecolor_list.append(color)
+            handle_list.append(Line2D(arr, ydata=arr, color=color, linewidth=5, label=ma))
+
+        self.set_collection_ma([], edgecolors=edgecolor_list)
+
+        # 가격이동평균선 legend 생성
+        if handle_list:
+            legends = self.ax_legend.legend(
+                handle_list, label_list, loc='lower left', ncol=self.CONFIG.MA.ncol,
+                borderpad=0.55,
+                facecolor=self.CONFIG.AX.facecolor, edgecolor=self.CONFIG.AX.TICK.edgecolor,
+                labelcolor=self.CONFIG.AX.TICK.fontcolor,
+            )
+            for handle in legends.legend_handles:
+                # legend 클릭시 pick event가 발생할 수 있도록 설정
+                handle.set_picker(5)
+
+            # legend ax 크기 조정
+            self._set_figure_ratios()
+        return
+
+
+class ArtistMixin(WatermarkMixin, CollectionMixin, MaMixin):
+    artist_watermark: Text
+    collection_candle: LineCollection
+    collection_volume: LineCollection
+    collection_ma: LineCollection
+
+    _visible_ma: set[int] = set()
+
+    def add_artists(self):
+        self._add_watermark()
+        self._set_watermark()
+        self._add_collections()
+        self._set_legends()
+        return
+
+    def set_artists(self):
+        self._set_watermark()
+        self._set_legends()
+        return
+

seolpyo_mplchart/_chart/base/c_draw.py

@@ -0,0 +1,100 @@
+from matplotlib.axes import Axes
+from matplotlib.collections import LineCollection
+from matplotlib.text import Text
+
+from .a_canvas import Figure
+
+
+class Base:
+    candle_on_ma = True
+
+    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
+
+    def draw_chart(self):
+        self._draw_ax_price()
+        self._draw_ax_volume()
+        self._draw_chart_price()
+        self._draw_chart_volume()
+        return
+
+    def _draw_chart_volume(self):
+        renderer = self.figure.canvas.renderer
+
+        self.collection_volume.draw(renderer)
+        return
+
+    def _draw_chart_price(self):
+        renderer = self.figure.canvas.renderer
+        # print(f'{renderer=}')
+
+        # print(self.collection_candle.get_segments())
+        if self.candle_on_ma:
+            self.collection_ma.draw(renderer)
+            self.collection_candle.draw(renderer)
+        else:
+            self.collection_candle.draw(renderer)
+            self.collection_ma.draw(renderer)
+
+        if self.watermark:
+            if self.watermark != self.artist_watermark.get_text():
+                self.artist_watermark.set_text(self.watermark)
+            self.artist_watermark.draw(renderer)
+        return
+
+    def draw_artists(self):
+        return
+
+    def draw_background(self):
+        self._draw_ax(self.ax_price)
+        self._draw_ax(self.ax_volume)
+
+        legend = self.ax_legend.get_legend()
+        if legend:
+            legend.draw(self.figure.canvas.renderer)
+        return
+
+    def _draw_ax(self, ax: Axes):
+        renderer = self.figure.canvas.renderer
+
+        # ax 배경
+        ax.patch.draw(renderer)
+
+        # ax 외곽선
+        for spine in ax.spines.values():
+            spine.draw(renderer)
+
+        return
+
+    def _draw_ax_volume(self):
+        renderer = self.figure.canvas.renderer
+
+        # grid, tick, ticklabel
+        ax = self.ax_volume
+        for axis in (ax.xaxis, ax.yaxis):
+            axis.draw(renderer)
+        return
+
+    def _draw_ax_price(self):
+        renderer = self.figure.canvas.renderer
+        # print(f'{renderer=}')
+
+        # grid, tick, ticklabel
+        ax = self.ax_price
+        for axis in (ax.xaxis, ax.yaxis):
+            axis.draw(renderer)
+        return
+
+
+class DrawMixin(Base):
+    candle_on_ma = True
+