hikyuu 2.6.8__py3-none-manylinux2014_aarch64.whl

hikyuu/draw/elder.py

@@ -0,0 +1,135 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+# cp936
+#
+# The MIT License (MIT)
+#
+# Copyright (c) 2010-2017 fasiondog
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+#===============================================================================
+# History:
+# 1. 20100224, Added by fasiondog
+#===============================================================================
+"""
+绘制亚历山大.艾尔德交易系统图形
+参见：《走进我的交易室》（2007年 地震出版社） Alexander Elder
+"""
+from pylab import plot
+from numpy import mean
+
+from hikyuu import (Query, constant, Indicator, CLOSE, EMA, MACD, VIGOR, SAFTYLOSS, CVAL, PRICELIST)
+from .drawplot import (
+    create_figure, show_gcf, ax_draw_macd2, adjust_axes_show, ax_set_locator_formatter
+)
+
+
+def _find_ema_coefficient(closes, emas, number=66, percent=0.95):
+    """计算EMA通道系数。
+    在《走进我的交易室》中，艾尔德介绍的价格通道为：
+        通道上轨 ＝ EMA ＋ EMA＊通道系数
+        通道下轨 ＝ EMA － EMA＊通道系数
+    其中一条绘制得恰到好处的通道应能将绝大多数价格包含在内，一般调节通道系数使其能够包含95％的价格
+    参数：closes：收盘价序列
+          emas：收盘价对应的EMA序列
+          number: 以最近多少天的数据来计算，即取最后N天的数据作为计算标准
+          percent：通道包含多少的价格，如0.95表示通道将包含95%的价格
+    """
+    assert len(closes) == len(emas), "数据长度不等"
+    assert number >= 1, "Number必须大于0"
+
+    tmp_closes = closes[-number:]
+    tmp_emas = emas[-number:]
+    dev = [abs(tmp_closes[i] - tmp_emas[i]) for i in range(len(tmp_closes))]
+    sortdev = sorted(dev)
+    x = int(number * percent)
+    x = x if len(dev) - 1 > x else len(dev) - 1
+    ix = dev.index(sortdev[x])
+    return float(dev[ix]) / tmp_emas[ix] if tmp_emas[ix] != 0.0 else 0.0
+
+
+def _draw_ema_pipe(axes, kdata, ema, n=22, w=0.10):
+    emas = [i for i in ema]
+    p = _find_ema_coefficient(
+        list(CLOSE(kdata)), emas, number=66, percent=0.95
+    ) if w == 'auto' else w
+    emas_high = [i + i * p for i in emas]
+    emas_low = [i - i * p for i in emas]
+    emas_len = len(emas)
+
+    PRICELIST(emas).plot(
+        axes=axes, color='b', linestyle='-', label='%s(%s)  %.2f' % (ema.name, n, emas[-1]), kref=kdata
+    )
+    PRICELIST(emas_high).plot(
+        axes=axes,
+        color='b',
+        linestyle=':',
+        label='U%s(%.2f%%) %.2f' % (ema.name, p * 100, emas_high[-1]),
+        kref=kdata
+    )
+    PRICELIST(emas_low).plot(
+        axes=axes,
+        color='b',
+        linestyle=':',
+        label='L%s(%.2f%%) %.2f' % (ema.name, p * 100, emas_low[-1]),
+        kref=kdata
+    )
+    #axes.plot(emas, '-b', label='%s(%s)  %.2f'%(ema.name, n, emas[-1]))
+    #axes.plot(emas_high, ':b', label='U%s(%.2f%%) %.2f'%(ema.name, p*100, emas_high[-1]))
+    #axes.plot(emas_low, ':b', label='L%s(%.2f%%) %.2f'%(ema.name, p*100, emas_low[-1]))
+    fy1 = [i for i in emas_low]
+    fy2 = [i for i in emas_high]
+    axes.fill_between(range(emas_len), fy1, fy2, alpha=0.2, color='y')
+
+
+def draw(stock, query=Query(-130), ma_n=22, ma_w='auto', vigor_n=13):
+    """绘制亚历山大.艾尔德交易系统图形"""
+    kdata = stock.get_kdata(query)
+    close = CLOSE(kdata)
+    ema = EMA(close, ma_n)
+    sf = SAFTYLOSS(close, 10, 3, 2.0)
+    vigor = VIGOR(kdata, vigor_n)
+
+    ax1, ax2, ax3 = create_figure(3)
+    kdata.plot(axes=ax1)
+    _draw_ema_pipe(ax1, kdata, ema, n=ma_n, w=ma_w)
+    sf.plot(axes=ax1, color='y', legend_on=True, kref=kdata)
+
+    ax_draw_macd2(ax2, ema, kdata)
+
+    vigor.plot(axes=ax3, marker='.', color='r', zero_on=True, legend_on=False, text_on=True, kref=kdata)
+    u = [i for i in vigor if i > 0 and i != constant.null_price]
+    l = [i for i in vigor if i < 0]
+    umean = mean(u)
+    umax = max(u)
+    lmean = mean(l)
+    lmin = min(l)
+    up = int(umax / umean)
+    lp = int(lmin / lmean)
+    for i in range(up):
+        CVAL(close, umean * (i + 1)).plot(axes=ax3, color='r', linestyle='--', kref=kdata)
+
+    for i in range(lp):
+        CVAL(close, lmean * (i + 1)).plot(axes=ax3, color='g', linestyle='--', kref=kdata)
+
+    ax1.set_xlim((0, len(kdata)))
+    ax_set_locator_formatter(ax1, kdata.get_datetime_list(), kdata.get_query().ktype)
+    adjust_axes_show([ax1, ax2, ax3])
+    return show_gcf()

hikyuu/draw/elder.pyi

@@ -0,0 +1,44 @@
+"""
+
+绘制亚历山大.艾尔德交易系统图形
+参见：《走进我的交易室》（2007年 地震出版社） Alexander Elder
+"""
+from __future__ import annotations
+import hikyuu.cpp.core39
+from hikyuu.cpp.core39 import CVAL
+from hikyuu.cpp.core39 import EMA
+from hikyuu.cpp.core39 import Indicator
+from hikyuu.cpp.core39 import MACD
+from hikyuu.cpp.core39 import PRICELIST
+from hikyuu.cpp.core39 import Query
+from hikyuu.cpp.core39 import SAFTYLOSS
+from hikyuu.cpp.core39 import VIGOR
+from hikyuu.draw.drawplot import adjust_axes_show
+from hikyuu.draw.drawplot import ax_draw_macd2
+from hikyuu.draw.drawplot import ax_set_locator_formatter
+from hikyuu.draw.drawplot import create_figure
+from hikyuu.draw.drawplot import show_gcf
+from matplotlib.pyplot import plot
+from numpy import mean
+__all__: list[str] = ['CLOSE', 'CVAL', 'EMA', 'Indicator', 'MACD', 'PRICELIST', 'Query', 'SAFTYLOSS', 'VIGOR', 'adjust_axes_show', 'ax_draw_macd2', 'ax_set_locator_formatter', 'constant', 'create_figure', 'draw', 'mean', 'plot', 'show_gcf']
+def _draw_ema_pipe(axes, kdata, ema, n = 22, w = 0.1):
+    ...
+def _find_ema_coefficient(closes, emas, number = 66, percent = 0.95):
+    """
+    计算EMA通道系数。
+        在《走进我的交易室》中，艾尔德介绍的价格通道为：
+            通道上轨 ＝ EMA ＋ EMA＊通道系数
+            通道下轨 ＝ EMA － EMA＊通道系数
+        其中一条绘制得恰到好处的通道应能将绝大多数价格包含在内，一般调节通道系数使其能够包含95％的价格
+        参数：closes：收盘价序列
+              emas：收盘价对应的EMA序列
+              number: 以最近多少天的数据来计算，即取最后N天的数据作为计算标准
+              percent：通道包含多少的价格，如0.95表示通道将包含95%的价格
+        
+    """
+def draw(stock, query = ..., ma_n = 22, ma_w = 'auto', vigor_n = 13):
+    """
+    绘制亚历山大.艾尔德交易系统图形
+    """
+CLOSE: hikyuu.cpp.core39.Indicator  # value = Indicator{...
+constant: hikyuu.cpp.core39.Constant  # value = <hikyuu.cpp.core39.Constant object>

hikyuu/draw/kaufman.py

@@ -0,0 +1,192 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+# cp936
+#
+# The MIT License (MIT)
+#
+# Copyright (c) 2010-2017 fasiondog
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+# ===============================================================================
+# History:
+# 1. 20100224, Added by fasiondog
+# ===============================================================================
+"""
+绘制佩里.J.考夫曼（Perry J.Kaufman） 自适应移动平均系统(AMA)
+参见：《精明交易者》（2006年 广东经济出版社） 
+"""
+
+from hikyuu import (
+    Query, StockManager, AMA, STDEV, CVAL, PRICELIST, EMA, CLOSE, HIGH, LOW, OPEN, KDATA, POS, SG_Single, SG_Cross,
+    SG_Flex, RESULT
+)
+from .drawplot import (show_gcf, create_figure, ax_set_locator_formatter, adjust_axes_show, ax_draw_macd)
+
+
+def draw(
+    stock, query=Query(-130), n=10, filter_n=20, filter_p=0.1, sg_type="CROSS", show_high_low=False, arrow_style=1
+):
+    """绘制佩里.J.考夫曼（Perry J.Kaufman） 自适应移动平均系统(AMA)"""
+    kdata = stock.get_kdata(query)
+
+    ax1, ax2 = create_figure(2)
+    kdata.plot(axes=ax1)
+
+    cama = AMA(CLOSE(kdata), n=n)
+    cama.name = "CAMA"
+    cama.plot(axes=ax1, color='b', legend_on=True, kref=kdata)
+
+    hama = AMA(HIGH(kdata), n=n)
+    hama.name = "HAMA"
+    hstd = STDEV(hama, n)
+    lama = AMA(LOW(kdata), n=n)
+    lama.name = "LAMA"
+    lstd = STDEV(lama, n)
+    fy1 = list(lama - lstd)[lstd.discard:]
+    fy2 = list(hama + hstd)[hstd.discard:]
+    ax1.fill_between(range(lstd.discard, len(kdata)), fy1, fy2, alpha=0.2, color='y')
+
+    if show_high_low:
+        hama.plot(axes=ax1, color='r', legend_on=True, kref=kdata)
+        lama.plot(axes=ax1, color='g', legend_on=True, kref=kdata)
+
+    if sg_type == 'CROSS':
+        fast_op = AMA(CLOSE(), n=n)
+        slow_op = EMA(n=2 * n)(fast_op)
+        sg = SG_Cross(fast_op, slow_op)
+        sg.plot(axes=ax1, kdata=kdata)
+        ind = slow_op(KDATA(kdata))
+        ind.name = "EMA(CAMA)"
+        ind.plot(axes=ax1, color='m', legend_on=True, kref=kdata)
+
+    elif sg_type == 'SINGLE':
+        sg = SG_Single(cama, filter_n=filter_n, filter_p=filter_p)
+        sg.plot(axes=ax1, kdata=kdata)
+
+    else:
+        print("sg_type only in ('CORSS', 'SINGLE')")
+
+    cer = RESULT(cama, 1)
+    label = "ER(%s)" % cer[-1]
+    cer.plot(axes=ax2, color='b', marker='o', label=label, legend_on=False, text_on=True, kref=kdata)
+
+    c = CLOSE(kdata)
+    CVAL(c, 0.8).plot(axes=ax2, color='r', linestyle='--', kref=kdata)
+    CVAL(c, -0.6).plot(axes=ax2, color='r', linestyle='--', kref=kdata)
+    CVAL(c, -0.8).plot(axes=ax2, color='r', linestyle='--', kref=kdata)
+    CVAL(c, 0).plot(axes=ax2, color='k', linestyle='-', kref=kdata)
+    # ax2.hlines(0.8,0,len(kdata),color='r',linestyle='--')
+    # ax2.hlines(-0.6,0,len(kdata),color='r',linestyle='--')
+    # ax2.hlines(-0.8,0,len(kdata),color='r',linestyle='--')
+    # ax2.hlines(0,0,len(kdata))
+
+    ax1.set_xlim((0, len(kdata)))
+    ax_set_locator_formatter(ax1, kdata.get_datetime_list(), query.ktype)
+    adjust_axes_show([ax1, ax2])
+    return show_gcf()
+
+
+def draw2(
+    block,
+    query=Query(-130),
+    ama1=None,
+    ama2=None,
+    n=10,
+    filter_n=20,
+    filter_p=0.1,
+    sg_type='CROSS',
+    show_high_low=True,
+    arrow_style=1
+):
+    """绘制佩里.J.考夫曼（Perry J.Kaufman） 自适应移动平均系统(AMA)"""
+    if ama1 is None:
+        ama1 = AMA(CLOSE(), n=10, fast_n=2, slow_n=30)
+    sm = StockManager.instance()
+    if block.name == 'SZ':
+        kdata = sm['sz000001'].get_kdata(query)
+    elif block.name == 'GEM':
+        kdata = sm['sz399006'].get_kdata(query)
+    else:
+        kdata = sm['sh000001'].get_kdata(query)
+
+    ax1, ax2, ax3 = create_figure(3)
+    kdata.plot(axes=ax1)
+
+    cama = AMA(CLOSE(kdata), n=n)
+    cama.name = "CAMA"
+    cama.plot(axes=ax1, color='b', legend_on=True, kref=kdata)
+
+    hama = AMA(HIGH(kdata), n=n)
+    hama.name = "HAMA"
+    hstd = STDEV(hama, n)
+    lama = AMA(LOW(kdata), n=n)
+    lama.name = "LAMA"
+    lstd = STDEV(lama, n)
+    fy1 = list(lama - lstd)[lstd.discard:]
+    fy2 = list(hama + hstd)[hstd.discard:]
+    ax1.fill_between(range(lstd.discard, len(kdata)), fy1, fy2, alpha=0.2, color='y')
+
+    if show_high_low:
+        hama.plot(axes=ax1, color='r', legend_on=True, kref=kdata)
+        lama.plot(axes=ax1, color='g', legend_on=True, kref=kdata)
+
+    if sg_type == 'CROSS':
+        fast_op = AMA(CLOSE(), n=n)
+        slow_op = EMA(n=2 * n)(fast_op)
+        sg = SG_Cross(fast_op, slow_op)
+        sg.plot(axes=ax1, kdata=kdata)
+        ind = slow_op(KDATA(kdata))
+        ind.name = "EMA(CAMA)"
+        ind.plot(axes=ax1, color='m', legend_on=True, kref=kdata)
+
+    elif sg_type == 'SINGLE':
+        sg = SG_Single(cama, filter_n=filter_n, filter_p=filter_p)
+        sg.plot(axes=ax1, kdata=kdata)
+
+    else:
+        print("sg_type only in ('CORSS', 'SINGLE')")
+
+    a = POS(block, query, SG_Flex(AMA(CLOSE(), n=3), 6))
+    a.name = "POS(3)"
+    a.plot(axes=ax2, color='b', marker='.', legend_on=True, kref=kdata)
+    a = POS(block, query, SG_Flex(AMA(CLOSE(), n=30), 60))
+    a.name = "POS(30)"
+    a.plot(axes=ax2, color='g', marker='.', legend_on=True, kref=kdata)
+
+    c = CLOSE(kdata)
+    CVAL(c, 0.8).plot(axes=ax2, color='r', linestyle='--', kref=kdata)
+    CVAL(c, 0.2).plot(axes=ax2, color='r', linestyle='--', kref=kdata)
+
+    if ama1.name == "AMA":
+        cer = RESULT(cama, 1)
+        label = "ER(%s)" % cer[-1]
+        cer.plot(axes=ax3, color='b', marker='.', label=label, legend_on=False, text_on=True, kref=kdata)
+        CVAL(c, 0.8).plot(axes=ax3, color='r', linestyle='--', kref=kdata)
+        CVAL(c, -0.6).plot(axes=ax3, color='r', linestyle='--', kref=kdata)
+        CVAL(c, -0.8).plot(axes=ax3, color='r', linestyle='--', kref=kdata)
+        CVAL(c, 0).plot(axes=ax3, color='k', linestyle='-', kref=kdata)
+    else:
+        ax_draw_macd(ax2, kdata)
+    # ax2.set_ylim(-1, 1)
+
+    ax1.set_xlim((0, len(kdata)))
+    ax_set_locator_formatter(ax1, kdata.get_datetime_list(), query.ktype)
+    adjust_axes_show([ax1, ax2])
+    return show_gcf()

hikyuu/draw/kaufman.pyi

@@ -0,0 +1,38 @@
+"""
+
+绘制佩里.J.考夫曼（Perry J.Kaufman） 自适应移动平均系统(AMA)
+参见：《精明交易者》（2006年 广东经济出版社） 
+"""
+from __future__ import annotations
+import hikyuu.cpp.core39
+from hikyuu.cpp.core39 import AMA
+from hikyuu.cpp.core39 import CVAL
+from hikyuu.cpp.core39 import EMA
+from hikyuu.cpp.core39 import POS
+from hikyuu.cpp.core39 import PRICELIST
+from hikyuu.cpp.core39 import Query
+from hikyuu.cpp.core39 import RESULT
+from hikyuu.cpp.core39 import SG_Cross
+from hikyuu.cpp.core39 import SG_Flex
+from hikyuu.cpp.core39 import SG_Single
+from hikyuu.cpp.core39 import STDEV
+from hikyuu.cpp.core39 import StockManager
+from hikyuu.draw.drawplot import adjust_axes_show
+from hikyuu.draw.drawplot import ax_draw_macd
+from hikyuu.draw.drawplot import ax_set_locator_formatter
+from hikyuu.draw.drawplot import create_figure
+from hikyuu.draw.drawplot import show_gcf
+__all__: list[str] = ['AMA', 'CLOSE', 'CVAL', 'EMA', 'HIGH', 'KDATA', 'LOW', 'OPEN', 'POS', 'PRICELIST', 'Query', 'RESULT', 'SG_Cross', 'SG_Flex', 'SG_Single', 'STDEV', 'StockManager', 'adjust_axes_show', 'ax_draw_macd', 'ax_set_locator_formatter', 'create_figure', 'draw', 'draw2', 'show_gcf']
+def draw(stock, query = ..., n = 10, filter_n = 20, filter_p = 0.1, sg_type = 'CROSS', show_high_low = False, arrow_style = 1):
+    """
+    绘制佩里.J.考夫曼（Perry J.Kaufman） 自适应移动平均系统(AMA)
+    """
+def draw2(block, query = ..., ama1 = None, ama2 = None, n = 10, filter_n = 20, filter_p = 0.1, sg_type = 'CROSS', show_high_low = True, arrow_style = 1):
+    """
+    绘制佩里.J.考夫曼（Perry J.Kaufman） 自适应移动平均系统(AMA)
+    """
+CLOSE: hikyuu.cpp.core39.Indicator  # value = Indicator{...
+HIGH: hikyuu.cpp.core39.Indicator  # value = Indicator{...
+KDATA: hikyuu.cpp.core39.Indicator  # value = Indicator{...
+LOW: hikyuu.cpp.core39.Indicator  # value = Indicator{...
+OPEN: hikyuu.cpp.core39.Indicator  # value = Indicator{...

hikyuu/draw/volume.py

@@ -0,0 +1,122 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+# cp936
+#
+# The MIT License (MIT)
+#
+# Copyright (c) 2010-2017 fasiondog
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+#===============================================================================
+# History:
+# 1. 20100227, Added by fasiondog
+#===============================================================================
+"""
+绘制普通K线图 + 成交量（成交金额）
+"""
+from hikyuu.util.mylog import spend_time
+from hikyuu import Query
+from hikyuu.indicator import Indicator, IF, MA, CLOSE, VOL, CVAL, PRICELIST, SG_Cross
+from .drawplot import (
+    create_figure, get_current_draw_engine, ax_set_locator_formatter, adjust_axes_show, ax_draw_macd, show_gcf
+)
+
+
+def draw(stock, query=Query(-130), ma1_n=5, ma2_n=10, ma3_n=20, ma4_n=60, ma5_n=100, vma1_n=5, vma2_n=10):
+    """绘制普通K线图 + 成交量（成交金额）"""
+    kdata = stock.get_kdata(query)
+    close = CLOSE(kdata, )
+    ma1 = MA(close, ma1_n)
+    ma2 = MA(close, ma2_n)
+    ma3 = MA(close, ma3_n)
+    ma4 = MA(close, ma4_n)
+    ma5 = MA(close, ma5_n)
+
+    ax1, ax2 = create_figure(2)
+    kdata.plot(axes=ax1)
+    ma1.plot(axes=ax1, legend_on=True, kref=kdata)
+    ma2.plot(axes=ax1, legend_on=True, kref=kdata)
+    ma3.plot(axes=ax1, legend_on=True, kref=kdata)
+    ma4.plot(axes=ax1, legend_on=True, kref=kdata)
+    ma5.plot(axes=ax1, legend_on=True, kref=kdata)
+
+    sg = SG_Cross(MA(CLOSE(), n=ma1_n), MA(CLOSE(), n=ma2_n))
+    sg.to = kdata
+    sg.plot(axes=ax1, kdata=kdata)
+
+    vol = VOL(kdata)
+    total = len(kdata)
+
+    x1 = IF(kdata.close > kdata.open, vol, 0)
+    x2 = IF(kdata.close <= kdata.open, vol, 0)
+    x1.bar(axes=ax2, width=0.4, color='r', edgecolor='r', kref=kdata)
+    x2.bar(axes=ax2, width=0.4, color='g', edgecolor='g', kref=kdata)
+
+    vma1 = MA(vol, vma1_n)
+    vma2 = MA(vol, vma2_n)
+    vma1.plot(axes=ax2, legend_on=True, kref=kdata)
+    vma2.plot(axes=ax2, legend_on=True, kref=kdata)
+
+    if query.ktype == Query.WEEK and stock.market == 'SH' and stock.code == '000001':
+        CVAL(0.16e+009, total, color='b', linestyle='--', kref=kdata)
+        #ax2.hlines(0.16e+009,0,len(kdata),color='b',linestyle='--')
+
+    ax_set_locator_formatter(ax1, kdata.get_datetime_list(), kdata.get_query().ktype)
+    adjust_axes_show([ax1, ax2])
+    return show_gcf()
+
+
+def draw2(stock, query=Query(-130), ma1_n=7, ma2_n=20, ma3_n=30, ma4_n=42, ma5_n=100, vma1_n=5, vma2_n=10):
+    """绘制普通K线图 + 成交量（成交金额）+ MACD"""
+    kdata = stock.get_kdata(query)
+    close = CLOSE(kdata)
+    ma1 = MA(close, ma1_n)
+    ma2 = MA(close, ma2_n)
+    ma3 = MA(close, ma3_n)
+    ma4 = MA(close, ma4_n)
+    ma5 = MA(close, ma5_n)
+
+    ax1, ax2, ax3 = create_figure(3)
+    kdata.plot(axes=ax1)
+    ma1.plot(axes=ax1, legend_on=True, kref=kdata)
+    ma2.plot(axes=ax1, legend_on=True, kref=kdata)
+    ma3.plot(axes=ax1, legend_on=True, kref=kdata)
+    ma4.plot(axes=ax1, legend_on=True, kref=kdata)
+    ma5.plot(axes=ax1, legend_on=True, kref=kdata)
+
+    vol = VOL(kdata)
+    x1 = IF(kdata.close > kdata.open, vol, 0)
+    x2 = IF(kdata.close <= kdata.open, vol, 0)
+    x1.bar(axes=ax2, width=0.4, color='r', edgecolor='r', kref=kdata)
+    x2.bar(axes=ax2, width=0.4, color='g', edgecolor='g', kref=kdata)
+
+    vol.bar(axes=ax2, color='r', legend_on=True)
+
+    vma1 = MA(vol, vma1_n)
+    vma2 = MA(vol, vma2_n)
+    vma1.plot(axes=ax2, legend_on=True)
+    vma2.plot(axes=ax2, legend_on=True)
+
+    ax_draw_macd(ax3, kdata)
+
+    ax1.set_xlim((0, len(kdata)))
+    ax_set_locator_formatter(ax1, kdata.get_datetime_list(), kdata.get_query().ktype)
+    adjust_axes_show([ax1, ax2, ax3])
+    return show_gcf()

hikyuu/draw/volume.pyi

@@ -0,0 +1,31 @@
+"""
+
+绘制普通K线图 + 成交量（成交金额）
+"""
+from __future__ import annotations
+import hikyuu.cpp.core39
+from hikyuu.cpp.core39 import CVAL
+from hikyuu.cpp.core39 import IF
+from hikyuu.cpp.core39 import Indicator
+from hikyuu.cpp.core39 import MA
+from hikyuu.cpp.core39 import PRICELIST
+from hikyuu.cpp.core39 import Query
+from hikyuu.cpp.core39 import SG_Cross
+from hikyuu.draw.drawplot import adjust_axes_show
+from hikyuu.draw.drawplot import ax_draw_macd
+from hikyuu.draw.drawplot import ax_set_locator_formatter
+from hikyuu.draw.drawplot import create_figure
+from hikyuu.draw.drawplot import get_current_draw_engine
+from hikyuu.draw.drawplot import show_gcf
+from hikyuu.util.mylog import spend_time
+__all__: list[str] = ['CLOSE', 'CVAL', 'IF', 'Indicator', 'MA', 'PRICELIST', 'Query', 'SG_Cross', 'VOL', 'adjust_axes_show', 'ax_draw_macd', 'ax_set_locator_formatter', 'create_figure', 'draw', 'draw2', 'get_current_draw_engine', 'show_gcf', 'spend_time']
+def draw(stock, query = ..., ma1_n = 5, ma2_n = 10, ma3_n = 20, ma4_n = 60, ma5_n = 100, vma1_n = 5, vma2_n = 10):
+    """
+    绘制普通K线图 + 成交量（成交金额）
+    """
+def draw2(stock, query = ..., ma1_n = 7, ma2_n = 20, ma3_n = 30, ma4_n = 42, ma5_n = 100, vma1_n = 5, vma2_n = 10):
+    """
+    绘制普通K线图 + 成交量（成交金额）+ MACD
+    """
+CLOSE: hikyuu.cpp.core39.Indicator  # value = Indicator{...
+VOL: hikyuu.cpp.core39.Indicator  # value = Indicator{...

hikyuu/examples/Turtle_SG.py

@@ -0,0 +1,47 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+# cp936
+
+# ===============================================================================
+# Aothor: fasiondog
+# History: 20160407, Added by fasiondog
+# ===============================================================================
+
+from hikyuu import *
+
+
+class TurtleSignal(SignalBase):
+    def __init__(self, n=20):
+        super(TurtleSignal, self).__init__("TurtleSignal")
+        self.set_param("n", 20)
+
+    def _clone(self):
+        return TurtleSignal()
+
+    def _calculate(self, k):
+        n = self.get_param("n")
+        c = CLOSE(k)
+        h = REF(HHV(c, n), 1)  # 前n日高点
+        L = REF(LLV(c, n), 1)  # 前n日低点
+        for i in range(h.discard, len(k)):
+            if (c[i] >= h[i]):
+                self._add_buy_signal(k[i].datetime)
+            elif (c[i] <= L[i]):
+                self._add_sell_signal(k[i].datetime)
+
+
+if __name__ == "__main__":
+    from examples_init import *
+
+    sg = TurtleSignal()
+    s = get_stock("sh000001")
+    k = s.get_kdata(Query(-500))
+
+    # 只有设置交易对象时，才会开始实际计算
+    sg.to = k
+    dates = k.get_datetime_list()
+    for d in dates:
+        if (sg.should_buy(d)):
+            print("买入：%s" % d)
+        elif (sg.should_sell(d)):
+            print("卖出: %s" % d)

hikyuu/examples/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.0.1"

hikyuu/examples/examples_init.py

@@ -0,0 +1,33 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+# cp936
+
+#===============================================================================
+# 作者：fasiondog
+# 历史：1）20130128, Added by fasiondog
+#===============================================================================
+from hikyuu import *
+
+import os
+
+config_file = os.path.expanduser('~') + "/.hikyuu/hikyuu.ini"
+if not os.path.exists(config_file):
+    #检查老版本配置是否存在，如果存在可继续使用，否则异常终止
+    data_config_file = os.path.expanduser('~') + "/.hikyuu/data_dir.ini"
+    data_config = configparser.ConfigParser()
+    data_config.read(data_config_file)
+    data_dir = data_config['data_dir']['data_dir']
+    if sys.platform == 'win32':
+        config_file = data_dir + "\\hikyuu_win.ini"
+    else:
+        config_file = data_dir + "/hikyuu_linux.ini"
+    if not os.path.exists(config_file):
+        raise("未找到配置文件，请先使用数据导入工具导入数据（将自动生成配置文件）！！!")
+    
+    
+#starttime = time.time()
+#print "Loading Day Data ..."
+hikyuu_init(config_file)
+sm = StockManager.instance()
+#endtime = time.time()
+#print "%.2fs" % (endtime-starttime)