# -*- coding: utf-8 -*-# Qt相关和十字光标from qtpy.QtGui import *from qtpy.QtCore import *from qtpy import QtWidgets, QtGui, QtCorefrom qtpy.QtWidgets import QApplication, QWidgetfrom uiCrosshair import Crosshairimport pyqtgraph as pg# 其他import numpy as npimport pandas as pdfrom functools import partialfrom datetime import datetime, timedeltafrom collections import deque# 字符串转换# ---------------------------------------------------------------------------------------try:    _fromUtf8 = QtCore.QString.fromUtf8except AttributeError:    def _fromUtf8(s):        return s######################################################################### 键盘鼠标功能########################################################################class KeyWraper(QtWidgets.QWidget):    """键盘鼠标功能支持的元类"""    # 初始化    # ----------------------------------------------------------------------    def __init__(self, parent=None):        QtWidgets.QWidget.__init__(self, parent)    # 重载方法keyPressEvent(self,event),即按键按下事件方法    # ----------------------------------------------------------------------    def keyPressEvent(self, event):        if event.key() == QtCore.Qt.Key_Up:            self.onUp()        elif event.key() == QtCore.Qt.Key_Down:            self.onDown()        elif event.key() == QtCore.Qt.Key_Left:            self.onLeft()        elif event.key() == QtCore.Qt.Key_Right:            self.onRight()        elif event.key() == QtCore.Qt.Key_PageUp:            self.onPre()        elif event.key() == QtCore.Qt.Key_PageDown:            self.onNxt()    # 重载方法mousePressEvent(self,event),即鼠标点击事件方法    # ----------------------------------------------------------------------    def mousePressEvent(self, event):        if event.button() == QtCore.Qt.RightButton:            self.onRClick(event.pos())        elif event.button() == QtCore.Qt.LeftButton:            self.onLClick(event.pos())    # 重载方法mouseReleaseEvent(self,event),即鼠标点击事件方法    # ----------------------------------------------------------------------    def mouseReleaseEvent(self, event):        if event.button() == QtCore.Qt.RightButton:            self.onRRelease(event.pos())        elif event.button() == QtCore.Qt.LeftButton:            self.onLRelease(event.pos())        self.releaseMouse()    # 重载方法wheelEvent(self,event),即滚轮事件方法    # ----------------------------------------------------------------------    def wheelEvent(self, event):        if event.delta() > 0:            self.onUp()        else:            self.onDown()    # 重载方法dragMoveEvent(self,event),即拖动事件方法    # ----------------------------------------------------------------------    def paintEvent(self, event):        self.onPaint()    # PgDown键    # ----------------------------------------------------------------------    def onNxt(self):        pass    # PgUp键    # ----------------------------------------------------------------------    def onPre(self):        pass    # 向上键和滚轮向上    # ----------------------------------------------------------------------    def onUp(self):        pass    # 向下键和滚轮向下    # ----------------------------------------------------------------------    def onDown(self):        pass    # 向左键    # ----------------------------------------------------------------------    def onLeft(self):        pass    # 向右键    # ----------------------------------------------------------------------    def onRight(self):        pass    # 鼠标左单击    # ----------------------------------------------------------------------    def onLClick(self, pos):        pass    # 鼠标右单击    # ----------------------------------------------------------------------    def onRClick(self, pos):        pass    # 鼠标左释放    # ----------------------------------------------------------------------    def onLRelease(self, pos):        pass    # 鼠标右释放    # ----------------------------------------------------------------------    def onRRelease(self, pos):        pass    # 画图    # ----------------------------------------------------------------------    def onPaint(self):        pass######################################################################### 选择缩放功能支持########################################################################class CustomViewBox(pg.ViewBox):    # ----------------------------------------------------------------------    def __init__(self, parent, *args, **kwds):        pg.ViewBox.__init__(self, *args, **kwds)        self.parent = parent        # 拖动放大模式        # self.setMouseMode(self.RectMode)    ## 右键自适应    # ----------------------------------------------------------------------    def mouseClickEvent(self, ev):        if ev.button() == QtCore.Qt.RightButton:            self.autoRange()    # 重载方法mousePressEvent(self,event),即鼠标点击事件方法    # ----------------------------------------------------------------------    def mousePressEvent(self, event):        pg.ViewBox.mousePressEvent(self, event)    # 重载方法mouseDragEvent(self,event),即拖动事件方法    def mouseDragEvent(self, ev, axis=None):        # if ev.start==True and ev.finish==False: ##判断拖拽事件是否结束        pos = ev.pos()        lastPos = ev.lastPos()        dif = pos - lastPos        rect = self.sceneBoundingRect()        pianyi = dif.x() * self.parent.countK * 2 / rect.width()        self.parent.index -= int(pianyi)        self.parent.index = max(self.parent.index, 60)        xMax = self.parent.index + self.parent.countK  ##        xMin = self.parent.index - self.parent.countK        if xMin < 0:            xMin = 0        # self.parent.plotAll(False, xMin, xMax) #注释原因：拖动事件不需要先绘制图形界面        pg.ViewBox.mouseDragEvent(self, ev, axis)    # ## 重载方法resizeEvent(self, ev)    def resizeEvent(self, ev):        self.linkedXChanged()        self.linkedYChanged()        self.updateAutoRange()        self.updateViewRange()        self._matrixNeedsUpdate = True        self.sigStateChanged.emit(self)        self.background.setRect(self.rect())        self.sigResized.emit(self)        self.parent.refreshHeight()######################################################################### 时间序列，横坐标支持########################################################################class MyStringAxis(pg.AxisItem):    """时间序列横坐标支持"""    # 初始化    # ----------------------------------------------------------------------    def __init__(self, xdict, *args, **kwargs):        pg.AxisItem.__init__(self, *args, **kwargs)        self.minVal = 0        self.maxVal = 0        self.xdict = xdict        self.x_values = np.asarray(xdict.keys())        self.x_strings = xdict.values()        self.setPen(color=(255, 255, 255, 255), width=0.8)        self.setStyle(tickFont=QFont("Roman times", 10, QFont.Bold), autoExpandTextSpace=True)    # 更新坐标映射表    # ----------------------------------------------------------------------    def update_xdict(self, xdict):        self.xdict.update(xdict)        self.x_values = np.asarray(self.xdict.keys())        self.x_strings = self.xdict.values()    # 将原始横坐标转换为时间字符串,第一个坐标包含日期    # ----------------------------------------------------------------------    def tickStrings(self, values, scale, spacing):        strings = []        for v in values:            vs = v * scale            if vs in self.x_values:                vstr = self.x_strings[np.abs(self.x_values - vs).argmin()]                if (isinstance(vstr, (str))):                    vstr = vstr                else:                    vstr = vstr.strftime('%Y-%m-%d %H:%M:%S')            else:                vstr = ""            strings.append(vstr)        return strings######################################################################### K线图形对象########################################################################class CandlestickItem(pg.GraphicsObject):    """K线图形对象"""    # 初始化    # ----------------------------------------------------------------------    def __init__(self, data):        """初始化"""        pg.GraphicsObject.__init__(self)        # 数据格式: [ (time, open, close, low, high),...]        self.data = data        # 只重画部分图形，大大提高界面更新速度        self.setFlag(self.ItemUsesExtendedStyleOption)        # 画笔和画刷        w = 0.4        self.offset = 0        self.low = 0        self.high = 1        self.picture = QtGui.QPicture()        self.pictures = []        self.bPen = pg.mkPen(color=(0, 240, 240, 255), width=w * 2)        self.bBrush = pg.mkBrush((0, 240, 240, 255))        self.rPen = pg.mkPen(color=(255, 60, 60, 255), width=w * 2)        self.rBrush = pg.mkBrush((255, 60, 60, 255))        self.rBrush.setStyle(Qt.NoBrush)        # 刷新K线        self.generatePicture(self.data)    # 画K线    # ----------------------------------------------------------------------    def generatePicture(self, data=None, redraw=False):        """重新生成图形对象"""        # 重画或者只更新最后一个K线        if redraw:            self.pictures = []        elif self.pictures:            self.pictures.pop()        w = 0.4        bPen = self.bPen        bBrush = self.bBrush        rPen = self.rPen        rBrush = self.rBrush        low, high = (data[0]['low'], data[0]['high']) if len(data) > 0 else (0, 1)        for (t, open0, close0, low0, high0) in data:            if t >= len(self.pictures):                tShift = t                low, high = (min(low, low0), max(high, high0))                picture = QtGui.QPicture()                p = QtGui.QPainter(picture)                # # 下跌蓝色（实心）, 上涨红色（空心）                pen, brush, pmin, pmax = (bPen, bBrush, close0, open0) \                    if open0 > close0 else (rPen, rBrush, open0, close0)                p.setPen(pen)                p.setBrush(brush)                # 画K线方块和上下影线                if pmin > low0:                    p.drawLine(QtCore.QPointF(tShift, low0), QtCore.QPointF(tShift, pmin))                if high0 > pmax:                    p.drawLine(QtCore.QPointF(tShift, pmax), QtCore.QPointF(tShift, high0))                p.drawRect(QtCore.QRectF(tShift - w, open0, w * 2, close0 - open0))                # if open0 == close0:                #     p.drawRect(QtCore.QPointF(tShift - w, open0), QtCore.QPointF(tShift + w, close0))                # else:                #     p.drawRect(QtCore.QRectF(tShift - w, open0, w * 2, close0 - open0))                # if pmin > low0:                #     p.drawLine(QtCore.QPointF(tShift, low0), QtCore.QPointF(tShift, pmin))                # if high0 > pmax:                #     p.drawLine(QtCore.QPointF(tShift, pmax), QtCore.QPointF(tShift, high0))                p.end()                self.pictures.append(picture)        self.low, self.high = low, high    # 手动重画    # ----------------------------------------------------------------------    def update(self):        if not self.scene() is None:            self.scene().update()    # 自动重画    # ----------------------------------------------------------------------    def paint(self, p, o, w):        rect = o.exposedRect        xmin, xmax = (max(0, int(rect.left())), min(len(self.pictures), int(rect.right())))        [p.drawPicture(0, 0, pic) for pic in self.pictures[xmin:xmax]]    # 定义边界    # ----------------------------------------------------------------------    def boundingRect(self):        return QtCore.QRectF(0, self.low, len(self.pictures), (self.high - self.low))########################################################################class KLineWidget(KeyWraper):    """用于显示价格走势图"""    # 保存K线数据的列表和Numpy Array对象    listBar = []    listVol = []    listHigh = []    listLow = []    listSig = []    listOpenInterest = []    arrows = []    # 是否完成了历史数据的读取    initCompleted = False    # ----------------------------------------------------------------------    def __init__(self, parent=None, name=None):        """Constructor"""        self.parent = parent        self.name = name        super(KLineWidget, self).__init__(parent)        # 当前序号        self.index = None  # 下标        self.countK = 60  # 显示的Ｋ线范围        self.oldsize=0#rectd的hieght        # 缓存数据        self.datas = []        self.listBar = []        self.listVol = []        self.listHigh = []        self.listLow = []        self.listSig = []        self.listOpenInterest = []        self.arrows = []        self.sars = []        # 所有K线上信号图        self.allColor = deque(['blue', 'green', 'yellow', 'white'])        self.sigData = {}        self.sigColor = {}        self.sigPlots = {}        # 初始化完成        self.initCompleted = False        # 调用函数        self.initUi()    # ----------------------------------------------------------------------    #  初始化相关    # ----------------------------------------------------------------------    def initUi(self):        """初始化界面"""        self.setWindowTitle(u'K线工具')        # 主图        self.pw = pg.PlotWidget()        # 界面布局        self.lay_KL = pg.GraphicsLayout(border=(100, 100, 100))        self.lay_KL.setContentsMargins(10, 10, 10, 10)        self.lay_KL.setSpacing(0)        self.lay_KL.setBorder(color=(255, 255, 255, 255), width=0.8)        self.lay_KL.setZValue(0)        self.lay_KL.setMinimumHeight(140)        self.KLtitle = self.lay_KL.addLabel(u'')        self.pw.setCentralItem(self.lay_KL)        # 设置横坐标        xdict = {}        self.axisTime = MyStringAxis(xdict, orientation='bottom')        # 初始化子图        self.initplotKline()        self.initplotVol()        self.initplotOI()        # 注册十字光标        self.crosshair = Crosshair(self.pw, self)        # 设置界面        self.vb = QtWidgets.QVBoxLayout()        self.vb.addWidget(self.pw)        self.setLayout(self.vb)        # 初始化完成        self.initCompleted = True        self.oldsize=self.rect().height()        self.customBox = {}    # ----------------------------------------------------------------------    def makePI(self, name):        """生成PlotItem对象"""        vb = CustomViewBox(self)        plotItem = pg.PlotItem(viewBox=vb, name=name, axisItems={
   'bottom': self.axisTime})        plotItem.setMenuEnabled(False)        plotItem.setClipToView(True)        plotItem.hideAxis('left')        plotItem.showAxis('right')        plotItem.setDownsampling(mode='peak')        plotItem.setRange(xRange=(0, 1), yRange=(0, 1))        plotItem.getAxis('right').setWidth(60)        plotItem.getAxis('right').setStyle(tickFont=QFont("Roman times", 10, QFont.Bold))        plotItem.getAxis('right').setPen(color=(255, 255, 255, 255), width=0.8)        plotItem.showGrid(True, True)        plotItem.hideButtons()        return plotItem    # ----------------------------------------------------------------------    def initplotVol(self):        """初始化成交量子图"""        self.pwVol = self.makePI('PlotVol' + self.name)        self.volume = CandlestickItem(self.listVol)        self.pwVol.addItem(self.volume)        self.pwVol.setMaximumHeight((self.rect().height()-30)/4)        self.pwVol.setMinimumHeight(12)        self.pwVol.setXLink('PlotOI' + self.name)        self.pwVol.hideAxis('bottom')        self.lay_KL.nextRow()        self.lay_KL.addItem(self.pwVol)        self.lay_KL.adjustSize()    # ----------------------------------------------------------------------    def initplotKline(self):        """初始化K线子图"""        self.pwKL = self.makePI('PlotKL' + self.name)        self.candle = CandlestickItem(self.listBar)        self.pwKL.addItem(self.candle)        self.pwKL.setXLink('PlotOI' + self.name)        self.pwKL.hideAxis('bottom')        self.pwKL.setMinimumHeight((self.rect().height()-30)/3)        self.lay_KL.nextRow()        self.lay_KL.addItem(self.pwKL)    # ----------------------------------------------------------------------    def initplotOI(self):        """初始化持仓量子图"""        self.pwOI = self.makePI('PlotOI' + self.name)        self.curveOI = self.pwOI.plot()        self.pwOI.setMaximumHeight((self.rect().height() - 30) / 4)        self.pwOI.setMinimumHeight(20)        self.lay_KL.nextRow()        self.lay_KL.addItem(self.pwOI)    # ----------------------------------------------------------------------    #  画图相关    # ----------------------------------------------------------------------    def plotVol(self, redraw=False, xmin=0, xmax=-1):        """重画成交量子图"""        if self.initCompleted:            self.volume.generatePicture(self.listVol[xmin:xmax], redraw)  # 画成交量子图    # ----------------------------------------------------------------------    def plotKline(self, redraw=False, xmin=0, xmax=-1):        """重画K线子图"""        if self.initCompleted:            self.candle.generatePicture(self.listBar[xmin:xmax], redraw)  # 画K线            self.plotMark()  # 显示开平仓信号位置    # ----------------------------------------------------------------------    def plotOI(self, xmin=0, xmax=-1):        """重画持仓量子图"""        if self.initCompleted:            # self.curveOI.setData(self.listOpenInterest[xmin:xmax]+[0], symbol='o', name="OpenInterest")            self.curveOI.setData(self.listOpenInterest[xmin:xmax] + [0],                                 name="OpenInterest" + self.name)  # 去除symbol='o'    def updateIndicatorSar(self, sarDatas, startIndex):        """更新sar指标   startIndex  从0到 len -1"""        if len(sarDatas) == 0:            return        while (len(self.sars) > (startIndex - 1)):            sar = self.sars[-1]            self.pwKL.removeItem(sar)            self.sars.remove(sar)        for i in range(startIndex, len(sarDatas)):            arrow = pg.ArrowItem(pos=(i, sarDatas[i]), angle=90, tipAngle=5, headLen=2, tailWidth=4, pen={
   'width': 1},                                 brush=(0, 0, 255))            defaultOpts = {                'pxMode': True,                'angle': -150,  ## If the angle is 0, the arrow points left                'pos': (0, 0),                'headLen': 10,                'tipAngle': 10,                'baseAngle': 0,                'tailLen': None,                'tailWidth': 3,                'pen': (200, 200, 200),                'brush': (255, 0, 0),            }            # arrow.setStyle(defaultOpts)            self.pwKL.addItem(arrow)            self.sars.append(arrow)    def addIndicatorSar(self, sarDatas):        """增加sar指标"""        if len(sarDatas) == 0:            return        for sar in self.sars:            self.pwKL.removeItem(sar)        # 画信号        for i in range(len(sarDatas)):            # arrow = pg.ArrowItem(pos=(i, sarDatas[i]),headLen=10,  angle=90, brush=(0, 0, 255))            arrow = pg.ArrowItem(pos=(i, sarDatas[i]), angle=90, tipAngle=5, headLen=2, tailWidth=4, pen={
   'width': 1},                                 brush=(0, 0, 255))            defaultOpts = {                'pxMode': True,                'angle': -150,  ## If the angle is 0, the arrow points left                'pos': (0, 0),                'headLen': 10,                'tipAngle': 10,                'baseAngle': 0,                'tailLen': None,                'tailWidth': 3,                'pen': (200, 200, 200),                'brush': (255, 0, 0),            }            # arrow.setStyle(defaultOpts)            self.pwKL.addItem(arrow)            self.sars.append(arrow)    # ----------------------------------------------------------------------    def addSig(self, sig):        """新增信号图"""        if sig in self.sigPlots:            self.pwKL.removeItem(self.sigPlots[sig])        self.sigPlots[sig] = self.pwKL.plot()        self.sigColor[sig] = self.allColor[0]        self.allColor.append(self.allColor.popleft())    # ----------------------------------------------------------------------    def showSig(self, datas):        """刷新信号图"""        for sig in self.sigPlots:            self.sigData[sig] = datas[sig]        [self.sigPlots[sig].setData(datas[sig], pen=self.sigColor[sig][0], name=sig) \         for sig in self.sigPlots]  # if sig in datas]    # ----------------------------------------------------------------------    def plotMark(self):        """显示开平仓信号"""        # 检查是否有数据        if len(self.datas) == 0:            return        for arrow in self.arrows:            self.pwKL.removeItem(arrow)        # 画买卖信号        for i in range(len(self.listSig)):            # 无信号            if self.listSig[i] == None:                continue            # 买信号            elif self.listSig[i] != None:                direction = self.listSig[i]["direction"]                offset = self.listSig[i]["offset"]                price = self.listSig[i]["price"]                if direction == "空" and offset == "开仓":                    # arrow = pg.ArrowItem(pos=(i, price),  angle=-90, brush=(255, 0, 0))                    arrow = pg.ArrowItem(pos=(i, price), angle=180, tipAngle=60, headLen=8, tailLen=3, tailWidth=5,                                         pen={
   'color': 'w', 'width': 1}, brush='r')                elif direction == "多" and offset == "开仓":                    # arrow = pg.ArrowItem(pos=(i, price),  angle=90, brush=(255, 0, 0))                    arrow = pg.ArrowItem(pos=(i, price), angle=180, tipAngle=60, headLen=8, tailLen=3, tailWidth=5,                                         pen={
   'color': 'w', 'width': 1}, brush='b')                elif direction == "空" and offset == "平仓":                    # arrow = pg.ArrowItem(pos=(i, price),  angle=-90, brush=(0, 0, 255))                    arrow = pg.ArrowItem(pos=(i, price), angle=0, tipAngle=40, headLen=8, tailLen=None, tailWidth=8,                                         pen={
   'color': 'w', 'width': 1}, brush='y')                elif direction == "多" and offset == "平仓":                    # arrow = pg.ArrowItem(pos=(i, price),  angle=90, brush=(0, 0, 255))                    arrow = pg.ArrowItem(pos=(i, price), angle=0, tipAngle=40, headLen=8, tailLen=None, tailWidth=8,                                         pen={
   'color': 'w', 'width': 1}, brush='y')            self.pwKL.addItem(arrow)            self.arrows.append(arrow)    # ----------------------------------------------------------------------    def updateAll(self):        """        手动更新所有K线图形，K线播放模式下需要        """        datas = self.datas        self.volume.update()        self.candle.update()        def update(view, low, high):            vRange = view.viewRange()            xmin = max(0, int(vRange[0][0]))            xmax = max(0, int(vRange[0][1]))            xmax = min(xmax, len(datas))            if len(datas) > 0 and xmax > xmin:                ymin = min(datas[xmin:xmax][low])                ymax = max(datas[xmin:xmax][high])                if  ymin and ymax:                   view.setRange(yRange=(ymin, ymax))            else:                view.setRange(yRange=(0, 1))        update(self.pwKL.getViewBox(), 'low', 'high')        update(self.pwVol.getViewBox(), 'volume', 'volume')        #update(self.pwOI.getViewBox(), 'openInterest', 'openInterest')        update(self.curveOI.getViewBox(), 'openInterest', 'openInterest')    # ----------------------------------------------------------------------    def plotAll(self, redraw=True, xMin=0, xMax=-1):        """        重画所有界面        redraw ：False=重画最后一根K线; True=重画所有        xMin,xMax : 数据范围        """        # xMax = len(self.datas) if xMax < 0 else xMax        # self.countK = xMax-xMin        # self.index = int((xMax+xMin)/2)        if redraw:            xmax = len(self.datas) if xMax < 0 else xMax            xmin=max(0,xmax-self.countK)            self.index = int((xmax + xmin) / 2)        self.pwOI.setLimits(xMin=xMin, xMax=xMax)        self.pwKL.setLimits(xMin=xMin, xMax=xMax)        self.pwVol.setLimits(xMin=xMin, xMax=xMax)        self.plotKline(redraw, xMin, xMax)  # K线图        self.plotVol(redraw, xMin, xMax)  # K线副图，成交量        self.plotOI(0, len(self.datas))  # K线副图，持仓量        self.refresh()    def refreshHeight(self):       # super.__init__(QResizeEvent)        # 如果窗口最大化，不修改比例，防御性设计        # if self.isMaximized():        #     return       if len(self.datas)!=0:            height =self.rect().height()            if height!=self.oldsize:                self.oldsize=height                height=(height-30)/4                self.pwKL.setMinimumHeight(height * 2-24)                self.pwVol.setMaximumHeight(height)                self.pwVol.setMinimumHeight(12)                self.pwOI.setMaximumHeight(height)                self.pwOI.setMinimumHeight(12)                #print height    # ----------------------------------------------------------------------    def refresh(self):        """        刷新三个子图的现实范围        """        datas = self.datas        minutes = int(self.countK / 2)        xmin = max(0, self.index - minutes)        xmax = xmin + 2 * minutes        self.pwOI.setRange(xRange=(xmin, xmax))        self.pwKL.setRange(xRange=(xmin, xmax))        self.pwVol.setRange(xRange=(xmin, xmax))    # ----------------------------------------------------------------------    #  快捷键相关    # ----------------------------------------------------------------------    def onNxt(self):        """跳转到下一个开平仓点"""        if len(self.listSig) > 0 and not self.index is None:            datalen = len(self.listSig)            self.index += 1            while self.index < datalen and self.listSig[self.index] == 0:                self.index += 1            self.refresh()            x = self.index            y = self.datas[x]['close']            self.crosshair.signal.emit((x, y))    # ----------------------------------------------------------------------    def onPre(self):        """跳转到上一个开平仓点"""        if len(self.listSig) > 0 and not self.index is None:            self.index -= 1            while self.index > 0 and self.listSig[self.index] == 0:                self.index -= 1            self.refresh()            x = self.index            y = self.datas[x]['close']            self.crosshair.signal.emit((x, y))    # ----------------------------------------------------------------------    def onDown(self):        """放大显示区间"""        self.countK = min(len(self.datas), int(self.countK * 1.2) + 1)        self.refresh()        if len(self.datas) > 0:            x = self.index - self.countK / 2 + 2 if int(                self.crosshair.xAxis) < self.index - self.countK / 2 + 2 else int(self.crosshair.xAxis)            x = self.index + self.countK / 2 - 2 if x > self.index + self.countK / 2 - 2 else x            x=min(x,len(self.datas)-1)            y = self.datas[x][2]            # y = self.datas[x]['close']            self.crosshair.signal.emit((x, y))    # ----------------------------------------------------------------------    def onUp(self):        """缩小显示区间"""        # self.countK = max(3,int(self.countK/1.2)-1)        self.countK = max(20, int(self.countK / 1.2) - 1)  # 最小显示k线范围20        self.refresh()        if len(self.datas) > 0:            x = self.index - self.countK / 2 + 2 if int(                self.crosshair.xAxis) < self.index - self.countK / 2 + 2 else int(self.crosshair.xAxis)            x = self.index + self.countK / 2 - 2 if x > self.index + self.countK / 2 - 2 else x            x = min(x, len(self.datas)-1)            y = self.datas[x]['close']            self.crosshair.signal.emit((x, y))    # ----------------------------------------------------------------------    def onLeft(self):        """向左移动"""        if len(self.datas) > 0 and int(self.crosshair.xAxis) > 2:            x = int(self.crosshair.xAxis) - 1            y = self.datas[x]['close']            if x <= self.index - self.countK / 2 + 2 and self.index > 1:                self.index -= 1                self.refresh()            self.crosshair.signal.emit((x, y))    # ----------------------------------------------------------------------    def onRight(self):        """向右移动"""        if len(self.datas) > 0 and int(self.crosshair.xAxis) < len(self.datas) - 1:            x = int(self.crosshair.xAxis) + 1            y = self.datas[x]['close']            if x >= self.index + int(self.countK / 2) - 2:                self.index += 1                self.refresh()            self.crosshair.signal.emit((x, y))    # ----------------------------------------------------------------------    # 界面回调相关    # ----------------------------------------------------------------------    def onPaint(self):        """界面刷新回调"""        view = self.pwKL.getViewBox()        vRange = view.viewRange()        # xmin = max(0,int(vRange[0][0]))        # xmax = max(0,int(vRange[0][1]))        # self.index  = int((xmin+xmax)/2)+1    # ----------------------------------------------------------------------    def resignData(self, datas):        """更新数据，用于Y坐标自适应"""        self.crosshair.datas = datas        def viewXRangeChanged(low, high, self):            vRange = self.viewRange()            xmin = max(0, int(vRange[0][0]))            xmax = max(0, int(vRange[0][1]))            xmax = min(xmax, len(datas))            if len(datas) > 0 and xmax > xmin:                ymin = min(datas[xmin:xmax][low])                ymax = max(datas[xmin:xmax][high])                if ymin and ymax:                   self.setRange(yRange=(ymin, ymax))            else:                self.setRange(yRange=(0, 1))        view = self.pwKL.getViewBox()        view.sigXRangeChanged.connect(partial(viewXRangeChanged, 'low', 'high'))        view = self.pwVol.getViewBox()        view.sigXRangeChanged.connect(partial(viewXRangeChanged, 'volume', 'volume'))        view = self.pwOI.getViewBox()        view.sigXRangeChanged.connect(partial(viewXRangeChanged, 'openInterest', 'openInterest'))    # ----------------------------------------------------------------------    # 数据相关    # ----------------------------------------------------------------------    def clearData(self):        """清空数据"""        # 清空数据，重新画图        self.time_index = []        self.listBar = []        self.listVol = []        self.listLow = []        self.listHigh = []        self.listOpenInterest = []        self.listSig = []        self.sigData = {}        self.arrows = []        self.datas = None    # ----------------------------------------------------------------------    def updateSig(self, sig):        """刷新买卖信号"""        self.listSig = sig        self.plotMark()    # ----------------------------------------------------------------------    def onBar(self, bar, nWindow=20):        """        新增K线数据,K线播放模式        nWindow : 最大数据窗口        """        # 是否需要更新K线        newBar = False if len(self.datas) > 0 and bar.datetime == self.datas[-1].datetime else True        nrecords = len(self.datas) if newBar else len(self.datas) - 1        bar.openInterest = np.random.randint(0,                                             3) if bar.openInterest == np.inf or bar.openInterest == -np.inf else bar.openInterest        recordVol = (nrecords, bar.volume, 0, 0, bar.volume) if bar.close < bar.open else (        nrecords, 0, bar.volume, 0, bar.volume)        if newBar and any(self.datas):            self.datas.resize(nrecords + 1, refcheck=0)            self.listBar.resize(nrecords + 1, refcheck=0)            self.listVol.resize(nrecords + 1, refcheck=0)        elif any(self.datas):            self.listLow.pop()            self.listHigh.pop()            self.listOpenInterest.pop()        if any(self.datas):            self.datas[-1] = (bar.datetime, bar.open, bar.close, bar.low, bar.high, bar.volume, bar.openInterest)            self.listBar[-1] = (nrecords, bar.open, bar.close, bar.low, bar.high)            self.listVol[-1] = recordVol        else:            self.datas = np.rec.array(                [(datetime, bar.open, bar.close, bar.low, bar.high, bar.volume, bar.openInterest)], \                names=('datetime', 'open', 'close', 'low', 'high', 'volume', 'openInterest'))            self.listBar = np.rec.array([(nrecords, bar.open, bar.close, bar.low, bar.high)], \                                        names=('datetime', 'open', 'close', 'low', 'high'))            self.listVol = np.rec.array([recordVol], names=('datetime', 'open', 'close', 'low', 'high'))            self.resignData(self.datas)        self.axisTime.update_xdict({nrecords: bar.datetime})        self.listLow.append(bar.low)        self.listHigh.append(bar.high)        self.listOpenInterest.append(bar.openInterest)        # if newBar:        #     xMax = nrecords + 1        #     xMin = max(0, xMax - self.countK)  # 最小显示区间        #     self.index=int(xMax+xMin)/2        # if self.pwKL.getViewBox().mousePress == True:        # print("sssssssssss")        # if self.pwKL.getViewBox().mousePress == False: ##self.mousePress==false,不在执行刷新操作        xMax = self.index + self.countK        xMin = self.index - self.countK        #xMin=(0,xMin)        if xMin<0:            xMin=0        if not newBar:            self.updateAll()        self.plotAll(False, xMin, xMax)        self.crosshair.signal.emit((None, None))    # ----------------------------------------------------------------------    def loadData(self, datas):        """        载入pandas.DataFrame数据        datas : 数据格式，cols : datetime, open, close, low, high        """        # 设置中心点时间        self.index = 0        # 绑定数据，更新横坐标映射，更新Y轴自适应函数，更新十字光标映射        datas['time_int'] = np.array(range(len(datas.index)))        self.datas = datas[['open', 'close', 'low', 'high', 'volume', 'openInterest']].to_records()        self.axisTime.xdict = {}        xdict = dict(enumerate(datas.index.tolist()))        self.axisTime.update_xdict(xdict)        self.resignData(self.datas)        # 更新画图用到的数据        self.listBar = datas[['time_int', 'open', 'close', 'low', 'high']].to_records(False)        self.listHigh = list(datas['high'])        self.listLow = list(datas['low'])        self.listOpenInterest = list(datas['openInterest'])        # 成交量颜色和涨跌同步，K线方向由涨跌决定        datas0 = pd.DataFrame()        datas0['open'] = datas.apply(lambda x: 0 if x['close'] >= x['open'] else x['volume'], axis=1)        datas0['close'] = datas.apply(lambda x: 0 if x['close'] < x['open'] else x['volume'], axis=1)        datas0['low'] = datas0['open']        datas0['high'] = datas0['close']        datas0['time_int'] = np.array(range(len(datas.index)))        self.listVol = datas0[['time_int', 'open', 'close', 'low', 'high']].to_records(False)        # 调用画图函数        self.plotAll(True, 0, len(self.datas))    def loadDataBar(self, datas):        """        载入pandas.DataFrame数据        datas : 数据格式，'symbol','vtSymbol','exchange','open','high','low','close','date','time1','datetime','volume','openInterest'        """        print(datas)        self.index = 0        # 绑定数据，更新横坐标映射，更新Y轴自适应函数，更新十字光标映射        datas['time_int'] = np.array(range(len(datas.index)))        self.datas = datas[['datetime', 'open', 'close', 'low', 'high', 'volume', 'openInterest']].to_records(False)        # self.datas = datas[['symbol','vtSymbol','exchange','open','high','low','close','date','time1','datetime','volume','openInterest']].to_records()        self.axisTime.xdict = {}        # xdict = dict(enumerate(datas.index.tolist()))        xdict = dict(enumerate(datas['datetime'].tolist()))        self.axisTime.update_xdict(xdict)        self.resignData(self.datas)        # 更新画图用到的数据        self.listBar = datas[['time_int', 'open', 'close', 'low', 'high']].to_records(False)        self.listHigh = list(datas['high'])        self.listLow = list(datas['low'])        self.listOpenInterest = list(datas['openInterest'])        # 成交量颜色和涨跌同步，K线方向由涨跌决定        datas0 = pd.DataFrame()        datas0['open'] = datas.apply(lambda x: 0 if x['close'] >= x['open'] else x['volume'], axis=1)        datas0['close'] = datas.apply(lambda x: 0 if x['close'] < x['open'] else x['volume'], axis=1)        datas0['low'] = datas0['open']        datas0['high'] = datas0['close']        datas0['time_int'] = np.array(range(len(datas.index)))        self.listVol = datas0[['time_int', 'open', 'close', 'low', 'high']].to_records(False)        # 调用画图函数        self.plotAll(True, 0, len(self.datas))    ##VtBarData 结构    def loadDataBarArray(self, record):        txtData = pd.DataFrame.from_records(record, index="datetime")        # txtData = txtData.rename(columns = {0:'symbol', 1:"vtSymbol",2:"exchange",3:"open",4:"high",5:"low",6:"close",7:"date",8:"time",9:"datetime",10:"volume",11:"openInterest"})        self.loadData(txtData)######################################################################### 功能测试########################################################################import sysif __name__ == '__main__':    app = QApplication(sys.argv)    # 界面设置    cfgfile = QtCore.QFile('css.qss')    cfgfile.open(QtCore.QFile.ReadOnly)    styleSheet = cfgfile.readAll()    styleSheet = unicode(styleSheet, encoding='utf8')    app.setStyleSheet(styleSheet)    # K线界面    # ui = KLineWidget()    # ui.show()    # ui.KLtitle.setText('rb1701',size='20pt')    # ui.loadData(pd.DataFrame.from_csv('data.csv'))    ui = KLineWidget(name="opt")    ui.show()    ui.KLtitle.setText('rb1701', size='20pt')    # txtData = pd.DataFrame.from_csv('D:/data/day/rb1101.txt',header=None,index_col=7)    txtData = pd.DataFrame.from_csv('E:/Data/day/rb1101.txt', header=None, index_col=7)    txtData = txtData.rename(        columns={0: 'symbol', 1: "vtSymbol", 2: "exchange", 3: "open", 4: "high", 5: "low", 6: "close", 7: "date",                 8: "time", 9: "datetime", 10: "volume", 11: "openInterest"})    ui.loadDataBar(txtData)    app.exec_()