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

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,15 @@
+
+
+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.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 NAVIGATOR
+
+
+class SliderData:
+    def __init__(self):
+        self.NAVIGATOR = NAVIGATOR
+
+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,20 @@
+from .. import figure
+
+
+class RatioData:
+    def __init__(self):
+        self.legend = 2
+        self.price = 18
+        self.volume = 5
+        self.slider = 3
+        self.none = 2
+
+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,9 @@
+
+
+class NavigatorData:
+    def __init__(self):
+        self.edgecolor = '#2962FF'
+        self.facecolor = '#0000002E'
+
+NAVIGATOR = NavigatorData()
+

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))
+
+

seolpyo_mplchart/_config/volume.py

@@ -0,0 +1,26 @@
+
+
+class VolumeFaceColorData:
+    def __init__(self):
+        self.rise: str|tuple[float, float, float, float] = '#F27663'
+        self.fall: str|tuple[float, float, float, float] = '#70B5F2'
+        self.doji: str|tuple[float, float, float, float] = '#BEBEBE'
+
+VOLUMEFACECOLOR = VolumeFaceColorData()
+
+class VolumeEdgeColorData:
+    def __init__(self):
+        self.rise: str|tuple[float, float, float, float] = '#F27663'
+        self.fall: str|tuple[float, float, float, float] = '#70B5F2'
+        self.doji: str|tuple[float, float, float, float] = '#BEBEBE'
+
+VOLUMEEDGECOLOR = VolumeEdgeColorData()
+
+class VolumeData:
+    def __init__(self):
+        self.half_width = 0.36
+        self.FACECOLOR = VOLUMEFACECOLOR
+        self.EDGECOLOR = VOLUMEEDGECOLOR
+
+VOLUME = VolumeData()
+

seolpyo_mplchart/_cursor.py

@@ -105,7 +105,7 @@ class CollectionMixin(BM):
 _set_key = {
     'compare', 'rate',
     'rate_open', 'rate_high', 'rate_low',
-    'compare_volume', 'rate_volume',
+    'volume_pre', 'compare_volume', 'rate_volume',
     'space_box_candle',
     'bottom_box_candle', 'top_box_candle',
     'max_box_volume',
@@ -136,13 +136,14 @@ class DataMixin(CollectionMixin):
                     raise Exception(f'"{i}" column not in DataFrame.\nadd column or set calc_info=True.')
         else:
             self.df['compare'] = (self.df[self.close] - self.df['close_pre']).fillna(0)
-            self.df['rate'] = (self.df['compare'] / self.df[self.close] * 100).__round__(2).fillna(0)
-            self.df['rate_open'] = ((self.df[self.Open] - self.df['close_pre']) / self.df[self.close] * 100).__round__(2).fillna(0)
-            self.df['rate_high'] = ((self.df[self.high] - self.df['close_pre']) / self.df[self.close] * 100).__round__(2).fillna(0)
-            self.df['rate_low'] = ((self.df[self.low] - self.df['close_pre']) / self.df[self.close] * 100).__round__(2).fillna(0)
+            self.df['rate'] = (self.df['compare'] * 100 / self.df['close_pre']).__round__(2).fillna(0)
+            self.df['rate_open'] = ((self.df[self.Open] - self.df['close_pre']) * 100 / self.df['close_pre']).__round__(2).fillna(0)
+            self.df['rate_high'] = ((self.df[self.high] - self.df['close_pre']) * 100 / self.df['close_pre']).__round__(2).fillna(0)
+            self.df['rate_low'] = ((self.df[self.low] - self.df['close_pre']) * 100 / self.df['close_pre']).__round__(2).fillna(0)
             if self.volume:
-                self.df['compare_volume'] = (self.df[self.volume] - self.df[self.volume].shift(1)).fillna(0)
-                self.df['rate_volume'] = (self.df['compare_volume'] / self.df[self.volume].shift(1) * 100).__round__(2).fillna(0)
+                self.df['volume_pre'] = self.df[self.volume].shift(1)
+                self.df['compare_volume'] = (self.df[self.volume] - self.df['volume_pre']).fillna(0)
+                self.df['rate_volume'] = (self.df['compare_volume'] * 100 / self.df['volume_pre']).__round__(2).fillna(0)
 
         self.df['space_box_candle'] = (self.df[self.high] - self.df[self.low]) / 5
         self.df['bottom_box_candle'] = self.df[self.low] - self.df['space_box_candle']
@@ -233,7 +234,7 @@ class CrossLineMixin(EventMixin):
 
         if self.in_price_chart or self.in_volume_chart:
             self._restore_region()
-            self._draw_crossline(e, self.in_price_chart)
+            self._draw_crossline(e)
             self.figure.canvas.blit()
         else:
             if self._erase_crossline():
@@ -248,10 +249,10 @@ class CrossLineMixin(EventMixin):
             return True
         return False
 
-    def _draw_crossline(self, e: MouseEvent, in_price_chart):
+    def _draw_crossline(self, e: MouseEvent):
         x, y = (e.xdata, e.ydata)
 
-        if in_price_chart:
+        if self.in_price_chart:
             self.collection_price_crossline.set_segments([((x, self.price_ymin), (x, self.price_ymax)), ((self.vxmin, y), (self.vxmax, y))])
             self.collection_volume_crossline.set_segments([((x, 0), (x, self.volume_ymax))])
         else:
@@ -262,14 +263,14 @@ class CrossLineMixin(EventMixin):
         self.collection_price_crossline.draw(renderer)
         self.collection_volume_crossline.draw(renderer)
 
-        self._draw_text_artist(e, in_price_chart)
+        self._draw_text_artist(e)
         return
 
-    def _draw_text_artist(self, e: MouseEvent, in_price_chart):
+    def _draw_text_artist(self, e: MouseEvent):
         x, y = (e.xdata, e.ydata)
 
         renderer = self.figure.canvas.renderer
-        if in_price_chart:
+        if self.in_price_chart:
             # 가격
             self.artist_text_price.set_text(f'{float_to_str(y, self.digit_price)}{self.unit_price}')
             self.artist_text_price.set_x(self.v0 if self.veighth < x else self.vsixth)
@@ -297,17 +298,20 @@ class CrossLineMixin(EventMixin):
 
 
 class BoxMixin(CrossLineMixin):
-    def _draw_crossline(self, e, in_price_chart):
-        super()._draw_crossline(e, in_price_chart)
-        self._draw_box_artist(e, in_price_chart)
+    def _draw_crossline(self, e):
+        super()._draw_crossline(e)
+        self._draw_box_artist(e)
         return
 
-    def _draw_box_artist(self, e: MouseEvent, in_price_chart):
+    def _draw_box_artist(self, e: MouseEvent):
         y = e.ydata
 
         renderer = self.figure.canvas.renderer
+
+        self.in_candle = False
+        self.in_volumebar = False
         if self.intx is not None:
-            if in_price_chart:
+            if self.in_price_chart:
                 # 박스 크기
                 high = self.df['top_box_candle'][self.intx]
                 low = self.df['bottom_box_candle'][self.intx]
@@ -317,21 +321,19 @@ class BoxMixin(CrossLineMixin):
                     high, low = (high+sub, low-sub)
 
                 # 커서가 캔들 사이에 있는지 확인
-                if high < y or y < low: self.in_candle = False
-                else:
+                if low <= y and y <= high:
                     # 캔들 강조
                     self.in_candle = True
                     x1, x2 = (self.intx-0.3, self.intx+1.3)
                     self.collection_price_box.set_segments([((x1, high), (x2, high), (x2, low), (x1, low), (x1, high))])
                     self.collection_price_box.draw(renderer)
-            elif self.volume:
+            elif self.in_volume_chart and self.volume:
                 # 거래량 강조
                 high = self.df['max_box_volume'][self.intx]
                 low = 0
                 if high < self.min_height_box_volume: high = self.min_height_box_volume
 
-                if high < y or y < low: self.in_volumebar = False
-                else:
+                if low <= y and y <= high:
                     self.in_volumebar = True
                     x1, x2 = (self.intx-0.3, self.intx+1.3)
                     self.collection_volume_box.set_segments([((x1, high), (x2, high), (x2, low), (x1, low), (x1, high))])
@@ -394,8 +396,8 @@ class InfoMixin(BoxMixin):
         self._length_text = lenth_high if lenth_volume < lenth_high else lenth_volume
         return
 
-    def _draw_box_artist(self, e, in_price_chart):
-        super()._draw_box_artist(e, in_price_chart)
+    def _draw_box_artist(self, e):
+        super()._draw_box_artist(e)
 
         if self.intx is not None:
             if self.in_candle: self._draw_candle_info_artist(e)

seolpyo_mplchart/_draw.py

@@ -251,18 +251,11 @@ class CandleSegmentMixin(DataMixin):
             tuple[tuple[float, float]]: candle segment
         """
         return (
-            (x, high),
-            (x, top),
-            (left, top),
-            (left, bottom),
-            (x, bottom),
-            (x, low),
-            (x, bottom),
-            (right, bottom),
-            (right, top),
-            (x, top),
-            (x, high),
             (x, top),
+            (left, top), (left, bottom),
+            (x, bottom), (x, low), (x, bottom),
+            (right, bottom), (right, top),
+            (x, top), (x, high)
         )
 
     def _create_candle_segments(self):
@@ -278,8 +271,7 @@ class CandleSegmentMixin(DataMixin):
             segment_candle.append(
                 self.get_candle_segment(
                     is_up=is_up,
-                    x=x,
-                    left=left, right=right,
+                    x=x, left=left, right=right,
                     top=top, bottom=bottom,
                     high=high, low=low,
                 )
@@ -419,11 +411,8 @@ class VolumeSegmentMixin(MaSegmentMixin):
             tuple[tuple[float, float]]: volume bar segment
         """
         return (
-            (left, top),
-            (left, 0),
-            (right, 0),
-            (right, top),
-            (left, top),
+            (left, 0), (left, top),
+            (right, top), (right, 0), 
         )
 
     def _create_volume_segments(self):

seolpyo_mplchart/_slider.py

@@ -131,7 +131,9 @@ class CollectionMixin(PlotMixin):
             keys.append('x')
             keys.append(f'ma{i}')
 
-        segment_slider = self.df[keys + ['x', self.close] ].values
+        series = self.df[keys + ['x', self.close]]
+        series['x'] = series['x'] - 0.5
+        segment_slider = series.values
         segment_slider = segment_slider.reshape(segment_slider.shape[0], len(self.list_ma)+1, 2).swapaxes(0, 1)
         self.collection_slider.set_segments(segment_slider)
         linewidth = [1 for _ in self.list_ma]
@@ -300,7 +302,7 @@ class MouseMoveMixin(BackgroundMixin):
             self.figure.canvas.blit()
         elif self.in_price_chart or self.in_volume_chart:
             self._restore_region()
-            self._draw_crossline(e, self.in_price_chart)
+            self._draw_crossline(e)
             self.figure.canvas.blit()
         else:
             if self._erase_crossline():
@@ -327,13 +329,16 @@ class MouseMoveMixin(BackgroundMixin):
         navleft, navright = self.navcoordinate
         if navleft == navright: return
 
-        x = e.xdata
-        leftmin, leftmax = (navleft-self._navLineWidth, navleft+self._navLineWidth_half)
-        rightmin, rightmax = (navright-self._navLineWidth_half, navright+self._navLineWidth)
+        x = e.xdata.__round__()
+
+        leftmin = navleft - self._navLineWidth
+        leftmax = navleft + self._navLineWidth_half
+        rightmin = navright - self._navLineWidth_half
+        rightmax = navright + self._navLineWidth
         if x < leftmin: self.figure.canvas.set_cursor(cursors.POINTER)
-        elif x < leftmax: self.figure.canvas.set_cursor(cursors.RESIZE_HORIZONTAL)
+        elif x <= leftmax: self.figure.canvas.set_cursor(cursors.RESIZE_HORIZONTAL)
         elif x < rightmin: self.figure.canvas.set_cursor(cursors.MOVE)
-        elif x < rightmax: self.figure.canvas.set_cursor(cursors.RESIZE_HORIZONTAL)
+        elif x <= rightmax: self.figure.canvas.set_cursor(cursors.RESIZE_HORIZONTAL)
         else: self.figure.canvas.set_cursor(cursors.POINTER)
         return
 
@@ -373,10 +378,10 @@ class MouseMoveMixin(BackgroundMixin):
                 self.artist_text_slider.draw(renderer)
         return
 
-    def _draw_crossline(self, e: MouseEvent, in_price_chart):
+    def _draw_crossline(self, e: MouseEvent):
         self.collection_slider_vline.set_segments([((e.xdata, self.slider_ymin), (e.xdata, self.slider_ymax))])
         self.collection_slider_vline.draw(self.figure.canvas.renderer)
-        return super()._draw_crossline(e, in_price_chart)
+        return super()._draw_crossline(e)
 
 
 class ClickMixin(MouseMoveMixin):
@@ -400,25 +405,26 @@ class ClickMixin(MouseMoveMixin):
         self.is_click_slider = True
         self.figure.canvas.set_cursor(cursors.RESIZE_HORIZONTAL)
 
-        x = e.xdata.__int__()
-        navmin, navmax = self.navcoordinate
-        
-        leftmax = navmin+self._navLineWidth_half
-        rightmin = navmax-self._navLineWidth_half
+        navleft, navright = self.navcoordinate
+        x = e.xdata.__round__()
+
+        leftmax = navleft + self._navLineWidth_half
+        rightmin = navright - self._navLineWidth_half
 
-        grater_than_left, less_then_right = (leftmax < x, x < rightmin)
+        grater_than_left = leftmax < x
+        less_then_right = x < rightmin
         if grater_than_left and less_then_right:
             self.is_move = True
             self.x_click = x
         else:
-            leftmin = navmin - self._navLineWidth
-            rightmax = navmax+self._navLineWidth
+            leftmin = navleft - self._navLineWidth
+            rightmax = navright + self._navLineWidth
             if not grater_than_left and leftmin <= x:
                 self.click_navleft = True
-                self.x_click = navmax
+                self.x_click = navright
             elif not less_then_right and x <= rightmax:
                 self.click_navright = True
-                self.x_click = navmin
+                self.x_click = navleft
             else:
                 self.x_click = x
         return
@@ -561,7 +567,7 @@ class ChartClickMixin(ReleaseMixin):
         elif self.in_price_chart or self.in_volume_chart:
             self._restore_region(self.is_click_chart)
             if not self.is_click_chart:
-                self._draw_crossline(e, self.in_price_chart)
+                self._draw_crossline(e)
             else: self._move_chart(e)
             self.figure.canvas.blit()
         else: