init

2026-08-24 20:53:21 +08:00
commit 6e78bba16a
287 changed files with 32443 additions and 0 deletions
--- a/old/s01/main.go
+++ b/old/s01/main.go
@@ -0,0 +1,163 @@
+package s01
+
+import (
+	"github.com/banbox/banbot/config"
+	"github.com/banbox/banbot/strat"
+	"github.com/banbox/banexg/log"
+	ta "github.com/banbox/banta"
+	"go.uber.org/zap"
+	"math"
+)
+
+/*
+此策略对应python原始策略名为PivotRev
+
+【s01_base】
+是最初版本的策略，所有标的使用固定的RSI差阈值：300周期9, 50周期20
+
+【s01_adp.50】~【s01_adp.100】
+增加了标的自适应的前20个高于固定阈值50%~100%的所有RSI值的均值，作为实际阈值。
+开单数量倍增，曲线稍微平滑一些，收益更高。
+测试最佳值是54%
+
+
+【s01_adpb.55】~【s01_adp.70】
+将bigTmp和smlTmp同时满足时才记录，前面是各自达到阈值就记录。
+改了后订单减少40%左右，收益减少30%左右。
+不采用
+
+后续考虑记录RSI300和RSI50的历史峰值，然后峰值排序，取Top50%的均值作为阈值。
+*/
+
+var (
+	RsiBig = 300
+	RsiSml = 50
+	ThsBig = float64(9)
+	ThsSml = float64(20)
+	AtrLen = 14
+
+	bigMin = float64(ThsBig) * 0.55
+	smlMin = float64(ThsSml) * 0.55
+
+	AvgLastNum = 20
+)
+
+type MoreType struct {
+	hit       bool
+	bigThsHis []float64
+	smlThsHis []float64
+	bigThs    float64
+	smlThs    float64
+	bigTmp    float64
+	smlTmp    float64
+}
+
+func (m *MoreType) close() {
+	if m.bigTmp > bigMin {
+		m.bigThsHis = append(m.bigThsHis, m.bigTmp)
+		if len(m.bigThsHis) > AvgLastNum {
+			m.bigThsHis = m.bigThsHis[len(m.bigThsHis)-AvgLastNum:]
+		}
+		sumVal := float64(0)
+		for _, val := range m.bigThsHis {
+			sumVal += val
+		}
+		m.bigThs = sumVal / float64(len(m.bigThsHis))
+	}
+	if m.smlTmp > smlMin {
+		m.smlThsHis = append(m.smlThsHis, m.smlTmp)
+		if len(m.smlThsHis) > AvgLastNum {
+			m.smlThsHis = m.smlThsHis[len(m.smlThsHis)-AvgLastNum:]
+		}
+		sumSml := float64(0)
+		for _, val := range m.smlThsHis {
+			sumSml += val
+		}
+		m.smlThs = sumSml / float64(len(m.smlThsHis))
+	}
+	m.bigTmp = -1
+	m.smlTmp = -1
+	m.hit = false
+}
+
+func (m *MoreType) check(big, sml float64) {
+	if m.bigTmp < 0 {
+		m.bigTmp = math.Abs(big)
+	} else {
+		m.bigTmp = max(m.bigTmp, math.Abs(big))
+	}
+	if m.smlTmp < 0 {
+		m.smlTmp = math.Abs(sml)
+	} else {
+		m.smlTmp = max(m.smlTmp, math.Abs(sml))
+	}
+}
+
+func MainV0(pol *config.RunPolicyConfig) *strat.TradeStrat {
+	return &strat.TradeStrat{
+		Version:      1,
+		WarmupNum:    RsiBig * 10,
+		MinTfScore:   0.75,
+		DrawDownExit: true,
+		OnStartUp: func(s *strat.StratJob) {
+			s.More = &MoreType{
+				bigThs: ThsBig,
+				smlThs: ThsSml,
+				bigTmp: -1,
+				smlTmp: -1,
+			}
+		},
+		OnBar: func(s *strat.StratJob) {
+			e := s.Env
+			ma5 := ta.EMA(e.Close, 5)
+			ma20 := ta.EMA(e.Close, 20)
+			maX := ma5.Cross(ma20)
+
+			rsiB := ta.RSI(e.Close, RsiBig).Get(0) - 50
+			rsiS := ta.RSI(e.Close, RsiSml).Get(0) - 50
+
+			atr := ta.ATR(e.High, e.Low, e.Close, AtrLen)
+			baseAtrLen := int(math.Round(float64(AtrLen) * 4.3))
+			atrB := ta.Lowest(ta.Highest(atr, AtrLen), baseAtrLen).Get(0)
+
+			if math.IsNaN(rsiB) || math.IsInf(rsiB, 0) {
+				// 过滤无效值
+				return
+			}
+
+			m, _ := s.More.(*MoreType)
+
+			if math.Abs(float64(maX)) == 1 {
+				// 均线交叉，顶底切换，允许开单
+				m.close()
+			}
+			m.check(rsiB, rsiS)
+			if m.hit {
+				// 已开单，禁止重复开单
+				return
+			}
+
+			if math.Abs(rsiB) <= m.bigThs || math.Abs(rsiS) <= m.smlThs {
+				// 未触发阈值，不开单
+				return
+			}
+
+			m.hit = true // 禁止同方向再次开单
+			isShort := rsiB > 0
+			odDirt := float64(1)
+			if isShort {
+				odDirt = -1
+			}
+			stopLoss := e.Close.Get(0) - max(atrB*3, atr.Get(0)*1.5)*odDirt
+			err := s.OpenOrder(&strat.EnterReq{
+				Tag:      "enter",
+				Short:    isShort,
+				StopLoss: stopLoss,
+			})
+			if err != nil {
+				log.Error("open order fail", zap.String("pair", s.Symbol.Symbol),
+					zap.String("stagy", s.Strat.Name), zap.Error(err))
+			}
+		},
+	}
+}
--- a/old/s01/v1.go
+++ b/old/s01/v1.go
@@ -0,0 +1,71 @@
+package s01
+
+import (
+	"github.com/banbox/banbot/config"
+	"github.com/banbox/banbot/strat"
+	"github.com/banbox/banexg/log"
+	ta "github.com/banbox/banta"
+	"go.uber.org/zap"
+	"math"
+)
+
+/*
+此策略对应python原始策略名为PivotRev
+*/
+
+func MainV1(pol *config.RunPolicyConfig) *strat.TradeStrat {
+	RsiBig = 60
+	RsiSml = 14
+	ThsBig = 20
+	ThsSml = 30
+	AtrLen = 14
+	return &strat.TradeStrat{
+		Version:      1,
+		WarmupNum:    RsiBig,
+		MinTfScore:   0.75,
+		DrawDownExit: true,
+		OnBar: func(s *strat.StratJob) {
+			e := s.Env
+			ma5 := ta.EMA(e.Close, 5)
+			ma20 := ta.EMA(e.Close, 20)
+			maX := ma5.Cross(ma20)
+
+			rsiB := ta.RSI(e.Close, RsiBig).Get(0) - 50
+			rsiS := ta.RSI(e.Close, RsiSml).Get(0) - 50
+
+			atr := ta.ATR(e.High, e.Low, e.Close, AtrLen)
+			basev1AtrLen := int(math.Round(float64(AtrLen) * 4.3))
+			atrB := ta.Lowest(ta.Highest(atr, AtrLen), basev1AtrLen).Get(0)
+
+			if math.Abs(float64(maX)) == 1 {
+				// 均线交叉，顶底切换，允许开单
+				s.More = nil
+			} else if s.More != nil {
+				// 禁止开单
+				return
+			}
+
+			if math.Abs(rsiB) <= ThsBig || math.Abs(rsiS) <= ThsSml {
+				// 未触发阈值，不开单
+				return
+			}
+
+			s.More = true // 禁止同方向再次开单
+			isShort := rsiB > 0
+			odDirt := float64(1)
+			if isShort {
+				odDirt = -1
+			}
+			stopLoss := e.Close.Get(0) - max(atrB*3, atr.Get(0)*1.5)*odDirt
+			err := s.OpenOrder(&strat.EnterReq{
+				Tag:      "enter",
+				Short:    isShort,
+				StopLoss: stopLoss,
+			})
+			if err != nil {
+				log.Error("open order fail", zap.String("pair", s.Symbol.Symbol),
+					zap.String("stagy", s.Strat.Name), zap.Error(err))
+			}
+		},
+	}
+}
--- a/old/s01/v2.go
+++ b/old/s01/v2.go
@@ -0,0 +1,120 @@
+package s01
+
+import (
+	"fmt"
+	"github.com/banbox/banbot/btime"
+	"github.com/banbox/banbot/config"
+	"github.com/banbox/banbot/orm/ormo"
+	"github.com/banbox/banbot/strat"
+	"github.com/banbox/banexg/log"
+	ta "github.com/banbox/banta"
+	"go.uber.org/zap"
+	"math"
+)
+
+type S1More struct {
+	JumpRates map[int64]float64
+	HitNum    int // 记录当前是否已开过单。对于单边行情，为避免重复开单，只有当rsi转折时才开单
+}
+
+func MainV2(pol *config.RunPolicyConfig) *strat.TradeStrat {
+	RsiSml = 14
+	ThsSml = 40
+	AtrLen = 14
+	return &strat.TradeStrat{
+		Version:      1,
+		WarmupNum:    RsiBig,
+		MinTfScore:   0.1,
+		DrawDownExit: true,
+		BatchInOut:   true,
+		OnStartUp: func(s *strat.StratJob) {
+			s.More = &S1More{
+				JumpRates: make(map[int64]float64),
+			}
+		},
+		OnBar: func(s *strat.StratJob) {
+			e := s.Env
+			rsi := ta.RSI(e.Close, RsiSml)
+			rsiS := rsi.Get(0) - 50
+			atr := ta.ATR(e.High, e.Low, e.Close, 14).Get(0)
+
+			m, _ := s.More.(*S1More)
+
+			// 检查是否RSI多次触发，且首次转折
+			openOK := m.HitNum > 1 && math.Abs(rsiS) < math.Abs(rsi.Get(1)-50)
+			if math.Abs(rsiS) < ThsSml && !openOK {
+				// rsi低于阈值，且不属于多次触发后转折
+				m.HitNum = 0
+				return
+			}
+			if rsiS > 0 && len(s.ShortOrders) > 0 || rsiS < 0 && len(s.LongOrders) > 0 {
+				// 已有入场订单，不重复开单
+				return
+			}
+			m.HitNum += 1
+			if m.HitNum > 1 && !openOK {
+				// 重复触发，只允许rsi调头时第二次入场
+				return
+			}
+			c := e.Close.Get(0)
+			stopLoss := c - atr*2
+			tag := "long"
+			if rsiS > 0 {
+				stopLoss = c + atr*2
+				tag = "short"
+			}
+			err := s.OpenOrder(&strat.EnterReq{
+				Tag:      tag,
+				Short:    rsiS > 0,
+				StopLoss: stopLoss,
+			})
+			if err != nil {
+				log.Error("open order fail", zap.String("pair", s.Symbol.Symbol),
+					zap.String("stagy", s.Strat.Name), zap.Error(err))
+			}
+		},
+		OnBatchJobs: func(jobs []*strat.StratJob) {
+			openNum := float64(0)
+			for _, j := range jobs {
+				if len(j.Entrys) > 0 {
+					openNum += 1
+				}
+			}
+			openRate := openNum / float64(len(jobs))
+			// 这里无需限制openRate的上限，前面已禁止单个趋势重复多次开单
+			if openRate < 0.2 {
+				for _, j := range jobs {
+					j.Entrys = nil
+				}
+				return
+			}
+			log.Info(fmt.Sprintf("%s open orders, num: %v, total: %v, rate: %.2f",
+				btime.ToDateStr(btime.TimeMS(), ""), openNum, len(jobs), openRate))
+		},
+		GetDrawDownExitRate: func(s *strat.StratJob, od *ormo.InOutOrder, maxChg float64) float64 {
+			more, _ := s.More.(*S1More)
+			jumpRate, ok := more.JumpRates[od.ID]
+			if !ok {
+				e := s.Env
+				jumpRate = (e.Close.Get(RsiSml)/e.Close.Get(0) - 1) * 0.3
+				more.JumpRates[od.ID] = jumpRate
+			}
+			var rate float64
+			switch {
+			case maxChg > jumpRate*15:
+				rate = 0.15
+			case maxChg > jumpRate*10:
+				rate = 0.17
+			case maxChg > jumpRate*3:
+				rate = 0.25
+			case maxChg > jumpRate*1:
+				rate = 0.37
+			case maxChg > jumpRate*0.5:
+				rate = 0.5
+			default:
+				rate = 0
+			}
+			return rate
+		},
+	}
+}
--- a/old/s01/v3.go
+++ b/old/s01/v3.go
@@ -0,0 +1,64 @@
+package s01
+
+import (
+	"github.com/banbox/banbot/config"
+	"github.com/banbox/banbot/core"
+	"github.com/banbox/banbot/strat"
+	ta "github.com/banbox/banta"
+	"math"
+	"slices"
+)
+
+func MainV3(pol *config.RunPolicyConfig) *strat.TradeStrat {
+	RsiSml = 100
+	watchBig := 5
+	watchSml := 3
+	threshold := float64(17)
+	return &strat.TradeStrat{
+		Version:    1,
+		WarmupNum:  RsiBig,
+		MinTfScore: 0.75,
+		//DrawDownExit: true,
+		OnBar: func(s *strat.StratJob) {
+			e := s.Env
+			rsi := ta.RSI(e.Close, RsiSml)
+			sml := ta.SMA(rsi, watchSml)
+			big := ta.SMA(rsi, watchBig)
+			atr := ta.ATR(e.High, e.Low, e.Close, 14).Get(0)
+			crx := sml.Cross(big)
+			if crx != 1 && crx != -1 {
+				return
+			}
+			rsiVal := float64(0)
+			if rsi.Get(watchBig) > 50 {
+				rsiVal = slices.Max(rsi.Range(0, watchBig*2))
+			} else {
+				rsiVal = slices.Min(rsi.Range(0, watchBig*2))
+			}
+			if math.Abs(rsiVal-50) < threshold {
+				return
+			}
+			c := e.Close.Get(0)
+			if crx == -1 {
+				_ = s.CloseOrders(&strat.ExitReq{
+					Tag:  "x_short",
+					Dirt: core.OdDirtShort,
+				})
+				_ = s.OpenOrder(&strat.EnterReq{
+					Tag:      "long",
+					StopLoss: c - atr*10,
+				})
+			} else if crx == 1 {
+				_ = s.CloseOrders(&strat.ExitReq{
+					Tag:  "x_long",
+					Dirt: core.OdDirtLong,
+				})
+				_ = s.OpenOrder(&strat.EnterReq{
+					Tag:      "short",
+					Short:    true,
+					StopLoss: c + atr*10,
+				})
+			}
+		},
+	}
+}