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

seolpyo_mplchart/_chart/slider/h_data.py

@@ -0,0 +1,102 @@
+from matplotlib.axes import Axes
+from matplotlib.text import Text
+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_slider: Axes
+
+    index_list: list[int] = []
+
+    artist_text_slider: Text
+
+    axis: callable
+    get_default_xlim: callable
+    set_variables: callable
+
+    _set_slider_collection: callable
+    vxmin: int
+    vxmax: int
+
+
+class SliderMixin(Base):
+    min_distance = 5
+
+    def get_default_xlim(self):
+        xmax = self.index_list[-1] + 1
+        xmin = xmax - 120
+        if xmin < 0:
+            xmin = 0
+        return (xmin, xmax)
+
+    def set_variables(self):
+        super().set_variables()
+
+        self._set_slider()
+        return
+
+    def _set_slider(self):
+        # print('_set_slider')
+        self._set_slider_xtick()
+
+        xmax = self.index_list[-1]
+        # 슬라이더 xlim
+        xdistance = round(xmax / 30)
+        self.slider_xmin, self.slider_xmax = (-xdistance, xmax+xdistance)
+        self.ax_slider.set_xlim(self.slider_xmin, self.slider_xmax)
+
+        # 네비게이터 경계선 두께
+        self._nav_width = round((self.slider_xmax - self.slider_xmin) / 250, 2)
+
+        # 슬라이더 ylim
+        ymin, ymax = (self.df['low'].min(), self.df['high'].max())
+        ysub = ymax - ymin
+        ydistance = round(ysub / 5, self.CONFIG.UNIT.digit+2)
+        self.slider_ymin, self.slider_ymax = (ymin-ydistance, ymax+ydistance)
+        self.ax_slider.set_ylim(self.slider_ymin, self.slider_ymax)
+
+        self._set_slider_collection()
+        self._set_slider_text_position()
+
+        return
+
+    def _set_slider_text_position(self):
+        # 슬라이더 텍스트 y
+        self.artist_text_slider.set_y(self.df['high'].max())
+        return
+
+    def _set_slider_xtick(self):
+        indices = [0, self.index_list[-1]]
+        # print(f'{indices=}')
+
+        date_list = [self.df.iloc[idx]['date'] for idx in indices]
+        # print(f'{date_list=}')
+        # xtick 설정, major tick과 겹쳐서 무시되는 것 방지
+        self.ax_slider.set_xticks([idx+0.01 for idx in indices], labels=date_list, minor=True)
+
+        labels = self.ax_slider.get_xticklabels(minor=True)
+        # print(f'{labels=}')
+        for label, align in zip(labels, ['center', 'center']):
+            # 라벨 텍스트 정렬
+            label.set_horizontalalignment(align)
+        return
+
+
+class DataMixin(SliderMixin):
+    min_distance = 5
+    _nav_width: float
+
+    slider_xmin: int
+    slider_xmax: int
+    slider_ymin: float
+    slider_ymax: float
+

seolpyo_mplchart/_chart/slider/test.py

@@ -0,0 +1,71 @@
+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._config import SLIDERCONFIG
+from seolpyo_mplchart._chart.slider 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[:800])
+
+
+class C(Chart):
+    limit_candle = 100
+    limit_wick = 300
+    limit_volume = 10
+    limit_ma = 200
+    t = 'light'
+    # watermark = ''
+    def __init__(self):
+        super().__init__()
+        # super().__init__(config=set_theme(SLIDERCONFIG, theme='dark'))
+        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.slider_top = True
+                self.t = 'dark'
+                # self.CONFIG.MA.linewidth = 1
+                self.get_candle_segment = lambda **x: Chart.get_candle_segment(self, **x)
+                self.CONFIG.CANDLE.linewidth = 0.8
+            else:
+                self.slider_top = False
+                self.t = 'light'
+                # self.CONFIG.MA.linewidth = 3
+                self.get_candle_segment = self.get_bar_segment
+                self.CONFIG.CANDLE.linewidth = 1.3
+            # 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/test.py

@@ -0,0 +1,121 @@
+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
+
+
+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[:100])
+
+def test_base():
+    from seolpyo_mplchart._chart._base import Base
+    Base()
+    plt.show()
+    return
+
+def test_draw():
+    from seolpyo_mplchart._chart._draw import Chart
+    c = Chart()
+    # c.CONFIG.CANDLE.FACECOLOR.bull_fall = 'y'
+    # c.CONFIG.CANDLE.EDGECOLOR.bull_fall = 'k'
+    # c.CONFIG.CANDLE.FACECOLOR.bear_rise = 'k'
+    # c.CONFIG.CANDLE.EDGECOLOR.bear_rise = 'pink'
+    # c.CONFIG.VOLUME.EDGECOLOR.rise = 'k'
+    # c.CONFIG.AX.facecolor = 'k'
+    # c.CONFIG.AX.TICK.edgecolor = 'yellow'
+    # c.CONFIG.AX.TICK.fontcolor = 'pink'
+    # c.set_color()
+    # c.volume = None
+    c.set_data(df)
+    plt.show()
+    return
+
+
+def test_cursor():
+    from seolpyo_mplchart._chart._cursor._info import Chart
+    from seolpyo_mplchart._config import DEFAULTCONFIG_EN
+    c = Chart()
+    # c.CONFIG = DEFAULTCONFIG_EN
+    c.refresh()
+    c.fraction = True
+    c.set_data(df)
+    plt.show()
+    return
+
+def test_slider():
+    from seolpyo_mplchart._chart._slider import Chart
+    c = Chart()
+    c.set_data(df)
+    plt.show()
+    return
+
+
+if __name__ == '__main__':
+    # test_base()
+    # test_draw()
+    # test_cursor()
+    test_slider()
+
+
+import seolpyo_mplchart as mc
+
+
+# class Chart(mc.SliderChart):
+#     format_candleinfo = mc.format_candleinfo_ko + '\nCustom info: {ci}'
+#     format_volumeinfo = mc.format_volumeinfo_ko
+#     min_distance = 2
+
+#     def __init__(self, *args, **kwargs):
+#         super().__init__(*args, **kwargs)
+#         self.collection_candle.set_linewidth(1.5)
+#         return
+
+#     def get_info_kwargs(self, is_price, **kwargs):
+#         if is_price:
+#             kwargs['ci'] = 'You can add Custom text Info or Change text info.'
+#             kwargs['close'] = 'You can Change close price info.'
+#         return kwargs
+
+#     def get_candle_segment(self, *, x, left, right, top, bottom, is_up, high, low):
+#         if is_up:
+#             return (
+#                 (x, top), (right, top), (x, top),
+#                 (x, high),
+#                 (x, low),
+#                 (x, bottom), (left, bottom), (x, bottom),
+#             )
+#         else:
+#             return (
+#                 (x, bottom), (right, bottom), (x, bottom),
+#                 (x, high),
+#                 (x, low),
+#                 (x, top), (left, top), (x, top),
+#             )
+
+
+
+# C = Chart()
+# path_file = Path(__file__).parent / 'sample/samsung.txt'
+# # C.format_candleinfo = mc.format_candleinfo_ko
+# # C.format_volumeinfo = mc.format_volumeinfo_ko
+# # C.volume = None
+
+
+# with open(path_file, 'r', encoding='utf-8') as txt:
+#     data = json.load(txt)
+# df = pd.DataFrame(data[:100])
+
+# C.set_data(df)
+
+# mc.show()
+# mc.close()

seolpyo_mplchart/_config/__init__.py

@@ -0,0 +1,3 @@
+from .config import DEFAULTCONFIG, DEFAULTCONFIG_EN, ConfigData
+from .slider import SLIDERCONFIG, SLIDERCONFIG_EN, SliderConfigData
+

seolpyo_mplchart/_config/ax.py

@@ -0,0 +1,28 @@
+
+
+class Grid:
+    def __init__(self):
+        self.visible = True
+        self.linewidth = 0.7
+        self.color: str|tuple[float, float, float, float] = '#d0d0d0'
+        self.linestyle = '-'
+        self.dashes = (1, 0)
+        self.axis = 'both'
+
+GRID = Grid()
+
+class TickData:
+    def __init__(self):
+        self.edgecolor: str|tuple[float, float, float, float] = 'k'
+        self.fontcolor: str|tuple[float, float, float, float] = 'k'
+
+TICK = TickData()
+
+class AxData:
+    def __init__(self):
+        self.facecolor: str|tuple[float, float, float, float] = '#fafafa'
+        self.GRID = GRID
+        self.TICK = TICK
+
+AX = AxData()
+

seolpyo_mplchart/_config/candle.py

@@ -0,0 +1,31 @@
+
+
+class CandleFaceColorData:
+    def __init__(self):
+        self.bull_rise: str|tuple[float, float, float, float] = '#FF2400'
+        self.bull_fall: str|tuple[float, float, float, float] = 'w'
+        self.bear_fall: str|tuple[float, float, float, float] = '#1E90FF'
+        self.bear_rise: str|tuple[float, float, float, float] = 'w'
+
+CANDLEFACECOLOR = CandleFaceColorData()
+
+class CandleEdgeColorData:
+    def __init__(self):
+        self.bull_rise: str|tuple[float, float, float, float] = '#FF2400'
+        self.bull_fall: str|tuple[float, float, float, float] = '#FF2400'
+        self.bear_fall: str|tuple[float, float, float, float] = '#1E90FF'
+        self.bear_rise: str|tuple[float, float, float, float] = '#1E90FF'
+        self.doji: str|tuple[float, float, float, float] = 'k'
+
+CANDLEEDGECOLOR = CandleEdgeColorData()
+
+class CandleData:
+    def __init__(self):
+        self.half_width = 0.24
+        self.linewidth = 0.8
+        self.line_color: str|tuple[float, float, float, float] = 'k'
+        self.FACECOLOR = CANDLEFACECOLOR
+        self.EDGECOLOR = CANDLEEDGECOLOR
+
+CANDLE = CandleData()
+

seolpyo_mplchart/_config/config.py

@@ -0,0 +1,21 @@
+from . import figure, ax, candle, volume, ma, unit, cursor, format
+
+
+class ConfigData:
+    def __init__(self):
+        self.FIGURE = figure.FIGURE
+        self.UNIT = unit.UNIT
+        self.AX = ax.AX
+        self.CANDLE = candle.CANDLE
+        self.VOLUME = volume.VOLUME
+        self.MA = ma.MA
+        self.CURSOR = cursor.CURSOR
+        self.FORMAT = format.FORMAT
+
+DEFAULTCONFIG = ConfigData()
+
+DEFAULTCONFIG_EN = ConfigData()
+DEFAULTCONFIG_EN.UNIT = unit.UNIT_EN
+DEFAULTCONFIG_EN.MA = ma.MA_EN
+DEFAULTCONFIG_EN.FORMAT = format.FORMAT_EN
+

seolpyo_mplchart/_config/cursor.py

@@ -0,0 +1,49 @@
+
+
+class CrossLineData:
+    def __init__(self):
+        self.edgecolor = 'k'
+        self.linewidth = 1
+        self.linestyle = '-'
+
+CROSSLINE = CrossLineData()
+
+class BBoxData:
+    def __init__(self):
+        self.boxstyle = 'round'
+        self.facecolor = 'w'
+        self.edgecolor = 'k'
+
+BBOX = BBoxData()
+
+class Text:
+    def __init__(self):
+        self.color = 'k'
+        self.BBOX = BBOX
+
+    def to_dict(self):
+        data = {}
+        for k, v in self.__dict__.items():
+            if k == 'BBOX':
+                k = k.lower()
+                v = self.BBOX.__dict__
+            data[k] = v
+        return data
+
+TEXT = Text()
+
+class Box:
+    def __init__(self):
+        self.edgecolor = 'k'
+        self.linewidth = 1.2
+
+BOX = Box()
+
+class Cursor:
+    def __init__(self):
+        self.CROSSLINE = CROSSLINE
+        self.TEXT = TEXT
+        self.BOX = BOX
+
+CURSOR = Cursor()
+

seolpyo_mplchart/_config/figure.py

@@ -0,0 +1,40 @@
+
+
+class RatioData:
+    def __init__(self):
+        self.price = 5
+        self.volume = 5
+
+RATIO = RatioData()
+
+class WatermarkData:
+    def __init__(self):
+        self.alpha = 0.2
+        self.color = 'k'
+        self.fontsize = 20
+
+WATERMARK = WatermarkData()
+
+class AdjustData:
+    def __init__(self):
+        # 여백
+        self.top = 0.98
+        self.bottom = 0.05
+        self.left = 0.01
+        self.right = 0.93
+        # 플롯간 간격
+        self.wspace = 0
+        self.hspace = 0
+
+ADJUST = AdjustData()
+
+class FigureData:
+    def __init__(self):
+        self.facecolor: str|tuple[float, float, float, float] = '#fafafa'
+        self.figsize = (14, 7)
+        self.RATIO = RATIO
+        self.ADJUST = ADJUST
+        self.WATERMARK = WATERMARK
+
+FIGURE = FigureData()
+

seolpyo_mplchart/_config/format.py

@@ -0,0 +1,51 @@
+
+
+format_candleinfo_ko = """\
+{dt}
+
+종가:　 {close}
+등락률: {rate}
+대비:　 {compare}
+시가:　 {open}({rate_open})
+고가:　 {high}({rate_high})
+저가:　 {low}({rate_low})
+거래량: {volume}({rate_volume})\
+"""
+format_volumeinfo_ko = """\
+{dt}
+
+거래량:　　　 {volume}
+거래량증가율: {rate_volume}
+대비:　　　　 {compare}\
+"""
+
+class FormatData:
+    def __init__(self):
+        self.candle = format_candleinfo_ko
+        self.volume = format_volumeinfo_ko
+
+FORMAT = FormatData()
+
+format_candleinfo_en = """\
+{dt}
+
+close:      {close}
+rate:        {rate}
+compare: {compare}
+open:      {open}({rate_open})
+high:       {high}({rate_high})
+low:        {low}({rate_low})
+volume:  {volume}({rate_volume})\
+"""
+format_volumeinfo_en = """\
+{dt}
+
+volume:      {volume}
+volume rate: {rate_volume}
+compare:     {compare}\
+"""
+
+FORMAT_EN = FormatData()
+FORMAT_EN.candle = format_candleinfo_en
+FORMAT_EN.volume = format_volumeinfo_en
+

seolpyo_mplchart/_config/ma.py

@@ -0,0 +1,17 @@
+
+
+class MaData:
+    "https://matplotlib.org/stable/gallery/color/named_colors.html"
+    def __init__(self):
+        self.color_default: str|tuple[float, float, float, float] = 'k'
+        self.format = '{}일선'
+        self.linewidth = 1
+        self.ncol = 10
+        self.color_list: list[str|tuple[float, float, float, float]] = ['#8B00FF', '#008000', '#A0522D', '#008B8B', '#FF0080']
+        self.ma_list = (5, 20, 60, 120, 240)
+
+MA = MaData()
+
+MA_EN = MaData()
+MA_EN.format = 'ma{}'
+

seolpyo_mplchart/_config/slider/__init__.py

@@ -0,0 +1,2 @@
+from .config import SLIDERCONFIG, SLIDERCONFIG_EN, SliderConfigData
+

seolpyo_mplchart/_config/slider/config.py

@@ -0,0 +1,24 @@
+from .. import config
+from .figure import FIGURE, SliderFigureData
+from .nav import NAV
+
+
+class SliderData:
+    def __init__(self):
+        self.NAV = NAV
+
+SLIDER = SliderData()
+
+class SliderConfigData(config.ConfigData):
+    FIGURE: SliderFigureData
+    SLIDER: SliderData
+
+
+SLIDERCONFIG: SliderConfigData = config.DEFAULTCONFIG
+SLIDERCONFIG.FIGURE = FIGURE
+SLIDERCONFIG.SLIDER = SLIDER
+
+SLIDERCONFIG_EN: SliderConfigData = config.DEFAULTCONFIG_EN
+SLIDERCONFIG_EN.FIGURE = FIGURE
+SLIDERCONFIG_EN.SLIDER = SLIDER
+

seolpyo_mplchart/_config/slider/figure.py

@@ -0,0 +1,19 @@
+from .. import figure
+
+
+class RatioData:
+    def __init__(self):
+        self.price = 9
+        self.volume = 3
+        self.none = 2
+        self.slider = 1.5
+
+RATIO = RatioData()
+
+class SliderFigureData(figure.FigureData):
+    def __init__(self):
+        super().__init__()
+        self.RATIO: RatioData = RATIO
+
+FIGURE = SliderFigureData()
+

seolpyo_mplchart/_config/slider/nav.py

@@ -0,0 +1,10 @@
+
+
+class Nav:
+    def __init__(self):
+        self.edgecolor = '#2962FF'
+        self.facecolor = '#0000002E'
+        self.alpha = 0.18
+
+NAV = Nav()
+

seolpyo_mplchart/_config/unit.py

@@ -0,0 +1,19 @@
+from .utils import convert_unit, convert_unit_en
+
+
+class UnitData:
+    def __init__(self):
+        self.price = '원'
+        self.volume = '주'
+        self.digit = 0
+        self.digit_volume = 0
+        self.func = convert_unit
+
+UNIT = UnitData()
+
+UNIT_EN = UnitData()
+UNIT_EN.price = ' $'
+UNIT_EN.volume = ' Vol'
+UNIT_EN.digit = 2
+UNIT_EN.func = convert_unit_en
+

seolpyo_mplchart/_config/utils.py

@@ -0,0 +1,67 @@
+
+
+def convert_num(num):
+    if isinstance(num, float) and num % 1:
+        return num
+    return int(num)
+
+
+def float_to_str(num: float, *, digit=0, plus=False):
+    if 0 < digit:
+        num.__round__(digit)
+        text = f'{num:+,.{digit}f}' if plus else f'{num:,.{digit}f}'
+    else:
+        num = round(num, digit).__int__()
+        text = f'{num:+,}' if plus else f'{num:,}'
+    return text
+
+
+unit_ko = {
+    '경': 10_000_000_000_000_000,
+    '조':      1_000_000_000_000,
+    '억':            100_000_000,
+    '만':                 10_000,
+}
+def convert_unit(value, *, digit=0, word='원', unit_data: dict[str, int]=None):
+    if not unit_data:
+        unit_data = unit_ko
+    # print('ko')
+    # print(f'{value=:,}')
+    v = abs(value)
+    for unit, n in unit_data.items():
+        if n <= v:
+            # print(f'{n=:,}')
+            # print(f'{unit=}')
+            num = value / n
+            if word.startswith(' '):
+                return f'{float_to_str(num, digit=digit)}{unit}{word}'
+            return f'{float_to_str(num, digit=digit)}{unit} {word}'
+
+    if not value % 1:
+        value = int(value)
+    text = f'{float_to_str(value, digit=digit)}{word}'
+    # print(f'{text=}')
+    return text
+
+unit_en = {
+    'Qd': 1_000_000_000_000_000,
+    'T':      1_000_000_000_000,
+    'B':          1_000_000_000,
+    'M':              1_000_000,
+    'K':                  1_000,
+}
+def convert_unit_en(value, *, digit=0, word='$', unit_data: dict[str, int]=None):
+    if not unit_data:
+        unit_data = unit_en
+    # print('en')
+    # print(f'{value=:,}')
+    return convert_unit(value, digit=digit, word=word, unit_data=unit_data)
+
+
+if __name__ == '__main__':
+    a = 456.123
+    print(float_to_str(a))
+    print(float_to_str(a, 2))
+    print(float_to_str(a, 6))
+
+