82 lines
3.3 KiB
Python
82 lines
3.3 KiB
Python
import copy
|
|
from typing import Dict, Optional
|
|
|
|
from ChanEnum import Chan_FX_TYPE, Chan_KLINE_DIR, Chan_BI_DIR
|
|
import ChanKLU
|
|
from ChanBI import ChanBI
|
|
|
|
class ChanSBI():
|
|
def __init__(self, start_bi: ChanBI, index, dir=Chan_BI_DIR.UP):
|
|
self.start_bi = start_bi
|
|
self.end_bi = None
|
|
self.index = index
|
|
self.dir = dir
|
|
self.high = start_bi.high
|
|
self.low = start_bi.low
|
|
self.pre = None
|
|
self.next = None
|
|
self.fx = Chan_FX_TYPE.UNKNOWN
|
|
self.bi_list = []
|
|
self.bi_list.append(start_bi)
|
|
self.has_fx_gap = False
|
|
def set_fx(self, fx):
|
|
self.fx = fx
|
|
def set_end_bi(self, bi):
|
|
self.end_bi = bi
|
|
def set_pre(self, sbi):
|
|
self.pre = sbi
|
|
def set_next(self, sbi):
|
|
self.next = sbi
|
|
def add_bi(self, bi):
|
|
self.bi_list.append(bi)
|
|
def check_fx(self):
|
|
if self.pre and self.next:
|
|
#print(self.pre.start_bi.start_time, self.start_bi.start_time, self.end_bi.end_time, self.next.start_bi.start_time, self.pre.high, self.high, self.next.high, self.pre.low, self.low, self.next.low, self.dir)
|
|
if self.high > self.pre.high and self.high > self.next.high:
|
|
self.fx = Chan_FX_TYPE.TOP
|
|
#print(self.start_bi.start_time, self.pre.start_bi.start_time, self.next.start_bi.start_time, self.fx)
|
|
if self.low > self.pre.high:
|
|
self.has_fx_gap = True
|
|
#print(self.start_bi.start_time, self.end_bi.end_time, self.pre.start_bi.start_time, self.next.start_bi.start_time, self.dir, self.has_fx_gap, self.fx)
|
|
return Chan_FX_TYPE.TOP
|
|
else:
|
|
if self.low < self.pre.low and self.low < self.next.low:
|
|
self.fx = Chan_FX_TYPE.BOTTOM
|
|
#print(self.start_bi.start_time, self.pre.start_bi.start_time, self.next.start_bi.start_time, self.fx)
|
|
if self.high < self.pre.low:
|
|
self.has_fx_gap = True
|
|
#print(self.start_bi.start_time, self.end_bi.end_time, self.pre.start_bi.start_time, self.next.start_bi.start_time, self.dir, self.has_fx_gap, self.fx)
|
|
return Chan_FX_TYPE.BOTTOM
|
|
return Chan_FX_TYPE.UNKNOWN
|
|
def check_bi_included(self, bi):
|
|
included = False
|
|
if self.high > bi.high:
|
|
# high大于，low小于，左包含
|
|
if self.low < bi.low:
|
|
included = True
|
|
# high大于，low大于，不包含
|
|
else:
|
|
# if self.low > bi.low
|
|
# high相等，右包含
|
|
included = False
|
|
else:
|
|
included = False
|
|
# high小于，low大于，右包含
|
|
if self.low > bi.low:
|
|
included = True
|
|
if included:
|
|
if self.pre:
|
|
if self.high > self.pre.high and self.low < self.pre.low:
|
|
included = True
|
|
if included:
|
|
self.add_bi(bi)
|
|
# gn>gn-1
|
|
if self.dir == Chan_BI_DIR.DOWN:
|
|
# UP -> max(dn)
|
|
self.low = bi.low
|
|
else:
|
|
# DOWN -> min(gn)
|
|
self.high = bi.high
|
|
#self.print(bi, "Z")
|
|
#print(self.start_bi.start_time, bi.start_time, included)
|
|
return included |