vnpy/vn.training/SubPlot/分时手数子图.py

# -*- coding: utf-8 -*-





import numpy



#	import matplotlib.spines as spines
import matplotlib.ticker as ticker



import Public.Public as Public







class 分时手数子图:

	def __init__(self, parent, 目标日期, 绘图数据):
		'''

		'''
		self._parent= parent
		self._ilogger= Public.IndentLogger(logger=parent._ilogger._logger, indent=parent._ilogger._indent+1)

		# 原始数据
		任务参数= 绘图数据['任务参数']

		# 当日数据
		当日数据= 绘图数据['分时数据'][目标日期]
		self._本图日期= 目标日期
		#	self._日期对象= 当日数据['日期对象']
		#	self._时间常数= 当日数据['时间常数']

		#	个股行情
		#===================================================================================================

		# 分时行情是否存在
		self._个股行情有效= 当日数据['个股行情有效']

		# 日线数据
		self._个股平均成交= 当日数据['个股平均成交']	# 格式: 行情['平均手数']= {Public.计算均值(个股成交量[-n:]) for n in 个股量均参数}

		if self._个股行情有效:
			# 分时数据
			个股分时行情= 当日数据['个股分时行情']
			self._个股手数序列= 个股分时行情['手数序列']
			self._个股金额序列= 个股分时行情['金额序列']
			self._个股备注序列= 个股分时行情['备注序列']

			#	self._个股上涨标记= 个股分时行情['上涨标记']
			#	self._个股下跌标记= 个股分时行情['下跌标记']
			#	self._个股平盘标记= 个股分时行情['平盘标记']

			self._个股买盘标记= [True if 备注.startswith('买') else False for 备注 in self._个股备注序列]
			self._个股卖盘标记= [True if 备注.startswith('卖') else False for 备注 in self._个股备注序列]
			self._个股中性标记= [True if 备注.startswith('中') else False for 备注 in self._个股备注序列]

			# 分时衍生数据（均线等）
			self._个股坐标序列= parent._个股坐标序列
			self._个股平均手数= Public.计算均值(序列=[nr for nr in self._个股手数序列 if nr>0])

		#	TODO: 指数行情
		#===================================================================================================



		#	横轴参数、纵轴参数
		#===================================================================================================
		self._横轴参数= parent._横轴参数
		self._纵轴参数= None

		#	平面对象，留待后面初始化
		#===================================================================================================
		self._个股布局参数= None
		self._指数布局参数= None
		
		self._指数平面= None
		self._指数横轴= None
		self._指数纵轴= None

		self._个股平面= None
		self._个股横轴= None
		self._个股纵轴= None



	def 计算成交步进记录(self):
		'''
		本函数是 naive 的，只考虑本图。
		'''
		if self._个股行情有效:
			# 决定个股步进值
			个股平均手数= self._个股平均手数[0]
			个股步进= 25	# 步进代表主坐标的间隔距离

			while 个股平均手数/个股步进 > 1.0:
				个股步进= 个股步进*2
		else:
			个股步进= 25

		return {'个股步进': 个股步进}



	def 计算纵轴参数(self, 步进记录=None):
		'''

		'''
		纵轴参数= {}

		#	大小固定
		#=======================================================================================
		纵轴尺寸= 4.0
		纵轴倍率= 0.3
		纵轴参数['纵轴倍率']= 纵轴倍率
		纵轴参数['纵轴高度']= 纵轴尺寸 * 纵轴倍率

		if 步进记录 is None:
			步进记录= self.计算成交步进记录()

		#	个股部分
		#=======================================================================================
		步进= 步进记录['个股步进']

		# 坐标起点 与 坐标终点
		个股坐标起点= 0.0
		个股坐标终点= max(self._个股手数序列) if self._个股行情有效 else 步进*纵轴尺寸
		纵轴参数['个股坐标起点']= 个股坐标起点
		纵轴参数['个股坐标终点']= 个股坐标终点

		个股纵轴尺寸= max(个股坐标终点/步进, 纵轴尺寸)
		纵轴参数['个股纵轴尺寸']= 个股纵轴尺寸
		纵轴参数['个股纵轴高度']= 个股纵轴尺寸 * 纵轴倍率

		# 计算 坐标值 与 坐标点
		个股主坐标值= [步进 * i for i in range(1, 4)]
		个股副坐标值= [(步进/2.0) + 步进*i for i in range(4)]
		if 个股副坐标值[-1] > 纵轴参数['个股坐标终点']: del 个股副坐标值[-1]
		
		纵轴参数['个股主坐标值']= 个股主坐标值
		纵轴参数['个股副坐标值']= 个股副坐标值

		self._纵轴参数= 纵轴参数



	def 返回纵轴高度(self):
		'''

		'''
		return self._纵轴参数['纵轴高度']



	def 返回指数平面(self):
		'''

		'''
		return self._指数平面



	def 平面初始化(self, 图片对象, 子图偏移, 全图大小):
		'''

		'''
		子图横移, \
		子图纵移= 子图偏移
		
		全图宽度, \
		全图高度= 全图大小
		
		本图宽度= self._横轴参数['横轴宽度']
		指数平面高度= self._纵轴参数['纵轴高度']	# XXX: 以后指数平面可以有自己的高度
		个股平面高度= self._纵轴参数['个股纵轴高度']

		指数布局参数= (子图横移/全图宽度, 子图纵移/全图高度, 本图宽度/全图宽度, 指数平面高度/全图高度)
		个股布局参数= (子图横移/全图宽度, 子图纵移/全图高度, 本图宽度/全图宽度, 个股平面高度/全图高度)

		self._指数布局参数= 指数布局参数
		self._个股布局参数= 个股布局参数

		#	指数部分
		#=======================================================================================
		指数平面= 图片对象.add_axes(指数布局参数, axis_bgcolor='black')
		指数平面.set_frame_on(False)	# XXX
		指数平面.set_axisbelow(True)	# 网格线放在底层

		指数横轴= 指数平面.get_xaxis()
		指数纵轴= 指数平面.get_yaxis()

		self._指数平面= 指数平面
		self._指数横轴= 指数横轴
		self._指数纵轴= 指数纵轴

		self.设置指数横轴()
		self.设置指数纵轴()

		#	个股部分
		#=======================================================================================

		# XXX: 不用 twinx()，原因见分时价格子图。
		个股平面= 图片对象.add_axes(个股布局参数, axis_bgcolor='black')
		个股平面.set_frame_on(False)	# XXX
		个股平面.set_axisbelow(True)	# 网格线放在底层

		for 方位, 边框 in 个股平面.spines.items():
			边框.set_color(None)

		个股横轴= 个股平面.get_xaxis()
		个股纵轴= 个股平面.get_yaxis()

		self._个股平面= 个股平面
		self._个股横轴= 个股横轴
		self._个股纵轴= 个股纵轴

		self.设置个股横轴()
		self.设置个股纵轴()



	def 设置指数横轴(self):
		'''

		'''
		指数平面= self._指数平面
		指数横轴= self._指数横轴
		横轴参数= self._横轴参数

		指数横轴.grid(True, 'major', color='0.3', linestyle='solid', linewidth=0.2)
		指数横轴.grid(True, 'minor', color='0.3', linestyle='dotted', linewidth=0.15)

		指数横轴.set_ticks_position('none')

		坐标起点= 横轴参数['坐标起点']
		坐标终点= 横轴参数['坐标终点']

		xMajorLocator= 横轴参数['xMajorLocator']
		xMinorLocator= 横轴参数['xMinorLocator']

		指数平面.set_xlim(坐标起点, 坐标终点)
		指数横轴.set_major_locator(xMajorLocator)
		指数横轴.set_minor_locator(xMinorLocator)

		for 主坐标 in 指数平面.get_xticklabels(minor=False):
			主坐标.set_visible(False)
		for 副坐标 in 指数平面.get_xticklabels(minor=True):
			副坐标.set_visible(False)



	def 设置指数纵轴(self):
		'''

		'''
		指数平面= self._指数平面
		指数纵轴= self._指数纵轴

		#	指数纵轴.set_label_position('right')	# XXX: 不顶用
		指数纵轴.set_ticks_position('none')

		for 主坐标 in 指数平面.get_yticklabels(minor=False):
			主坐标.set_visible(False)
		for 副坐标 in 指数平面.get_yticklabels(minor=True):
			副坐标.set_visible(False)



	def 设置个股横轴(self):
		'''

		'''
		个股平面= self._个股平面
		个股横轴= self._个股横轴
		横轴参数= self._横轴参数

		个股横轴.set_ticks_position('none')

		坐标起点= 横轴参数['坐标起点']
		坐标终点= 横轴参数['坐标终点']

		xMajorLocator= 横轴参数['xMajorLocator']
		xMinorLocator= 横轴参数['xMinorLocator']
		xMajorFormatter= 横轴参数['xMajorFormatter']
		xMinorFormatter= 横轴参数['xMinorFormatter']

		个股平面.set_xlim(坐标起点, 坐标终点)
		个股横轴.set_major_locator(xMajorLocator)
		个股横轴.set_minor_locator(xMinorLocator)

		个股横轴.set_major_formatter(xMajorFormatter)
		个股横轴.set_minor_formatter(xMinorFormatter)

		for 主坐标 in 个股平面.get_xticklabels(minor=False):
			主坐标.set_fontsize(5)
			主坐标.set_horizontalalignment('right')
			主坐标.set_rotation('45')

		for 副坐标 in 个股平面.get_xticklabels(minor=True):
			副坐标.set_fontsize(4)
			副坐标.set_color('blue')
			副坐标.set_horizontalalignment('right')
			副坐标.set_rotation('45')



	def 设置个股纵轴(self):
		'''

		'''
		平面对象= self._个股平面
		个股纵轴= self._个股纵轴
		纵轴参数= self._纵轴参数

		#	个股纵轴.set_label_position('right')	# XXX: 不顶用
		个股纵轴.set_ticks_position('none')

		个股纵轴.grid(True, 'major', color='0.3', linestyle='solid', linewidth=0.2)
		个股纵轴.grid(True, 'minor', color='0.3', linestyle='dotted', linewidth=0.15)

		坐标起点= 纵轴参数['个股坐标起点']
		坐标终点= 纵轴参数['个股坐标终点']

		平面对象.set_ylim(坐标起点, 坐标终点)

		#	主坐标点
		#======================================================================================
		主坐标值= 纵轴参数['个股主坐标值']
		yMajorLocator= ticker.FixedLocator(numpy.array(主坐标值))
		个股纵轴.set_major_locator(yMajorLocator)

		#	def y_major_formatter(num, pos=None):
		#		return str(num)
		#	yMajorFormatter= ticker.FuncFormatter(y_major_formatter)
		#	个股纵轴.set_major_formatter(yMajorFormatter)

		for 主坐标 in 平面对象.get_yticklabels(minor=False):
			主坐标.set_visible(False)

		#	副坐标点
		#======================================================================================
		副坐标值= 纵轴参数['个股副坐标值']
		yMinorLocator= ticker.FixedLocator(numpy.array(副坐标值))
		个股纵轴.set_minor_locator(yMinorLocator)

		#	def y_minor_formatter(num, pos=None):
		#		return str(num)
		#	yMinorFormatter= ticker.FuncFormatter(y_minor_formatter)
		#	个股纵轴.set_minor_formatter(yMinorFormatter)

		for 副坐标 in 平面对象.get_yticklabels(minor=True):
			副坐标.set_visible(False)



	def 绘图(self):
		'''

		'''
		self.绘制辅助标记()
		self.绘制个股手数()



	def 绘制辅助标记(self):
		'''

		'''
		个股平面= self._个股平面

		标注位置组一= self._横轴参数['标注位置组一']
		主标手数= self._纵轴参数['个股主坐标值']
		副标手数= self._纵轴参数['个股副坐标值']

		for 横标 in 标注位置组一:
			for 纵标 in 主标手数:
				个股平面.text(横标+30.0, 纵标, str(int(纵标)), color='0.3', fontsize=5.0, zorder=0)
			for 纵标 in 副标手数:
				个股平面.text(横标+30.0, 纵标, str(int(纵标)), color='0.3', fontsize=5.0, zorder=0)



	def 绘制个股手数(self):
		'''

		'''
		个股平面= self._个股平面

		if self._个股行情有效:
			坐标序列= numpy.array(self._个股坐标序列)
			手数序列= numpy.array(self._个股手数序列)
			序列长度= len(手数序列)
			起点序列= numpy.zeros(序列长度)

			#	正向标记= numpy.array(self._个股上涨标记)
			#	负向标记= numpy.array(self._个股下跌标记)
			#	中性标记= numpy.array(self._个股平盘标记)

			正向标记= numpy.array(self._个股买盘标记)
			负向标记= numpy.array(self._个股卖盘标记)
			中性标记= numpy.array(self._个股中性标记)

			lwidth, alpha= (0.15, 1.0)

			if True in 正向标记:
				个股平面.vlines(坐标序列[正向标记], 起点序列[正向标记], 手数序列[正向标记], edgecolor='red', linewidth=lwidth, label='_nolegend_', alpha=alpha)
			if True in 负向标记:
				个股平面.vlines(坐标序列[负向标记], 起点序列[负向标记], 手数序列[负向标记], edgecolor='green', linewidth=lwidth, label='_nolegend_', alpha=alpha)
			if True in 中性标记:
				个股平面.vlines(坐标序列[中性标记], 起点序列[中性标记], 手数序列[中性标记], edgecolor='white', linewidth=lwidth, label='_nolegend_', alpha=alpha)

			# 绘制平均手数数值（直线）
			平均手数= self._个股平均手数[0]
			横轴参数= self._横轴参数
			横标起点= 横轴参数['坐标起点']
			横标终点= 横轴参数['坐标终点']

			个股平面.plot([横标起点, 横标终点], [平均手数, 平均手数], '-', color='yellow', linewidth=0.2, alpha=0.7)