package strategies

import (
	"gitee.com/quant1x/engine/factors"
	"gitee.com/quant1x/engine/models"
	"gitee.com/quant1x/engine/smart"
	"gitee.com/quant1x/engine/utils"
	"gitee.com/quant1x/gotdx/securities"
	"gitee.com/quant1x/gox/util/treemap"
	. "gitee.com/quant1x/pandas/formula"
	"gitee.com/quant1x/pandas/stat"
)

const (
	// MaximumResultDays 结果最大天数
	MaximumResultDays int = 3
)

func init() {
	err := models.Register(ModelNo1{})
	if err != nil {
		panic(err)
	}
}

// ModelNo1 1号模型
//
//	FormulaNo1 3天内5天线上穿10天线，10天线上穿20天线的个股
//	count(cross(MA(c,5),MA(c,10)),3)>=1 and count(cross(MA(c,10),MA(c,20)),3)>=1
type ModelNo1 struct {
}

func (m ModelNo1) Code() models.ModelKind {
	return models.ModelHousNo1
}

func (m ModelNo1) Name() string {
	return models.MapStrategies[m.Code()].Name
}

func (m ModelNo1) OrderFlag() string {
	return models.OrderFlagTail
}

func (m ModelNo1) Filter(snapshot models.QuoteSnapshot) bool {
	return GeneralFilter(snapshot)
}

func (m ModelNo1) Sort(snapshots []models.QuoteSnapshot) models.SortedStatus {
	return models.SortDefault
}

func (m ModelNo1) Evaluate(securityCode string, result *treemap.Map) {
	history := smart.GetL5History(securityCode)
	if history == nil {
		return
	}
	snapshot := models.GetTickFromMemory(securityCode)
	if snapshot == nil {
		return
	}

	// 取出昨日的数据
	lastNo1 := history.Last.No1
	r1MA5 := lastNo1.MA5
	r1MA10 := lastNo1.MA10
	r1MA20 := lastNo1.MA20
	// 取出今日的半成品数据
	today := history.Payloads.No1.Increase(*snapshot).(*factors.HousNo1)
	ma5 := today.MA5
	ma10 := today.MA10
	ma20 := today.MA20

	// 组织series
	s5 := stat.NewSeries(r1MA5, ma5)
	s10 := stat.NewSeries(r1MA10, ma10)
	s20 := stat.NewSeries(r1MA20, ma20)

	// 两个金叉
	c1 := CROSS(s5, s10)
	c2 := CROSS(s10, s20)
	// 横向对比
	d := c1.And(c2)
	s := utils.BoolIndexOf(d, -1)
	if s {
		price := snapshot.Price
		date := snapshot.Date
		result.Put(securityCode, models.ResultInfo{Code: securityCode,
			Name:         securities.GetStockName(securityCode),
			Date:         date,
			Rate:         0.00,
			Buy:          price,
			Sell:         price * 1.05,
			StrategyCode: m.Code(),
			StrategyName: m.Name()})
	}
}