前面一中简单介绍了kline-chart的图表程序,实际上这个程序最主要的功能不是显示K线,因为显示K线的程序太多了,没必要专门重写,这个程序最主要的功能是根据需要显示包含K线在内的各种指标,自己算的指标,或是不知道怎么算,但别人算出来的指标数据,甚至是买卖点,只要按一定格式存储进来,然后在配置文件中标记一下,即可以将K线,指标组合,何时买入,何时卖出的标记在图上显示出来,先给出一个自己计算MA20,MA60的指标,再加上一个存在的MACD指标数据(这个是在别处生成的,在这儿直接加载进来即可),组合在一起显示出来的效果如下:
一、分析结构:
从图上看,本程序应该有三个plot,第1个plot的大小是尽可能大,因此配置的参数max_height=0,第2个和第3个类似于股票软件中的副图指标,因此指定大小max_height=120.
然后每个plot中可以叠加不同的图形,如第1个plot中为一个K线图叠加了一个均线图,第2个plot中是macd图,第3个是成交量图
另外,前面也说了,均线是要自己计算的,成交量也是需要从K线中提取的,这些需要自己计算的一般在配置文件中均会提供回调函数,然后由该函数计算出数据以供画图,而macd图则是外部算好后以文件的形式提供给程序画出来,会在配置中提供对应的文件路径:
以下展示对应的配置:
当然,如果你只想显示2栏,最后一栏不想要的话,直接把plot[2]对应的数据都删掉即可。
程序稍有一点点复杂,后面看有没有机会将指标写成插件的形式,使得只想用的人直接写自己对应的代码即可。
二、简单介绍:
1、主调用:
这个相对简单,调用PySide6的组件QtWidgets,然后加载主窗口即可
if __name__ == '__main__':
init_logger("info")
logger.info(conf)
app = QtWidgets.QApplication(sys.argv)
w = MainWindow()
w.resize(1024, 768)
w.show()
app.exec()
2、图形核心代码:
继承于ChartBase的各种图形的实现类,如ChartCandle(K线图类),ChartLine(曲线图类,可用于显示均线等各种曲线),ChartMacd(专们用来显示macd图形),ChartVolume(显示成交量图形),ChartStraight(显示直接,射线,线段图形),ChartSignal(显示买卖信号图形)等
展示一下如何画K线的:
def _draw_bar_picture(self, ix: int, old_bar: DataItem, bar: DataItem) -> QtGui.QPicture:
""""""
# Create objects
candle_picture = QtGui.QPicture()
painter = QtGui.QPainter(candle_picture)
# 1 open
# 2 high
# 3 low
# 4 close
# Set painter color
if bar[4] > bar[1]: # bar.close_price >= bar.open_price:
painter.setPen(self._up_pen)
painter.setBrush(self._black_brush)
else:
painter.setPen(self._down_pen)
painter.setBrush(self._down_brush)
# Draw candle shadow
if bar[2] > bar[3]: # bar.high_price > bar.low_price:
painter.drawLine(
QtCore.QPointF(ix, bar[2]),
QtCore.QPointF(ix, bar[3])
)
# Draw candle body
if bar[1] == bar[4]: # bar.open_price == bar.close_price:
painter.drawLine(
QtCore.QPointF(ix - BAR_WIDTH, bar[1]),
QtCore.QPointF(ix + BAR_WIDTH, bar[1]),
)
else:
rect = QtCore.QRectF(
ix - BAR_WIDTH,
bar[1],
BAR_WIDTH * 2,
bar[4] - bar[1]
)
painter.drawRect(rect)
# Finish
painter.end()
return candle_picture
看到,如果 收盘>开盘,则设置上涨色,否则设置下跌色,然后再从最高画到最低,上下影线,然后通过QtCore.QRectF画K线的实体部分,当bar[1]==bar[4]时,即开盘等于收盘,就不需要画实体框,只需要横着画一条线即可painter.drawLine。
3、两个回调函数:
def calc_volumes(klines: list[KLine]):
bars = {}
for k in klines:
dt = datetime.fromtimestamp(k.time)
bars[dt] = [dt, k.volume]
return bars
从K线中获取成交量,然后丢给前端显示
def calc_ma20_60(klines: list[KLine]):
bars = {}
MA20, MA60 = MA(20), MA(60)
for k in klines:
dt = datetime.fromtimestamp(k.time)
MA20.input(k.close)
MA60.input(k.close)
bars[dt] = [dt, MA20.ma, MA60.ma]
return bars
计算出M20,M60值,然后丢给前端显示
前面两个函数是需要自己实现的,如果有更复杂的实现,比如kdj的计算,就需要自己按以上方法写一个,然后在配置文件中配上对应的函数名即可。
更多的细节可查看源代码,已经上传于资源中。https://download.csdn.net/download/luhouxiang/89026271
最后,说一下,写这个文章只是想找一些对量化交易有兴趣的朋友一起研究研究算法,顺便把这个程序完善一下,使之更好的服务于量化算法。
到此,本程序应该是够用了,下周有空估计会把MT5上的ZigZag即波峰波谷的代码研究一下,先转成python然后在这个图形上显示一下以便深入理解其原理。