package models

import (
	"gitee.com/quant1x/engine/factors"
	"gitee.com/quant1x/gox/api"
	"gitee.com/quant1x/num"
	"gitee.com/quant1x/ta-lib/linear"
	"sort"
)

// ResultInfo 策略结果
type ResultInfo struct {
	Code           string  `name:"证券代码" dataframe:"code"`
	Name           string  `name:"证券名称" dataframe:"name"`
	Date           string  `name:"信号日期" dataframe:"date"`
	TurnZ          float64 `name:"开盘换手Z" dataframe:"turn_z"`
	Rate           float64 `name:"涨跌幅%" dataframe:"rate"`
	Buy            float64 `name:"委托价格" dataframe:"buy"`
	Sell           float64 `name:"目标价格" dataframe:"sell"`
	StrategyCode   uint64  `name:"策略编码" dataframe:"strategy_code"`
	StrategyName   string  `name:"策略名称" dataframe:"strategy_name"`
	BlockType      string  `name:"板块类型" dataframe:"block_type"`
	BlockCode      string  `name:"板块代码" dataframe:"block_code"`
	BlockName      string  `name:"板块名称" dataframe:"block_name"`
	BlockRate      float64 `name:"板块涨幅%" dataframe:"block_rate"`
	BlockTop       int     `name:"板块排名" dataframe:"block_top"`
	BlockRank      int     `name:"个股排名" dataframe:"block_rank"`
	BlockZhangTing string  `name:"板块涨停数" dataframe:"block_zhangting"`
	BlockDescribe  string  `name:"涨/跌/平" dataframe:"block_describe"`
	BlockTopCode   string  `name:"领涨股代码" dataframe:"block_top_code"`
	BlockTopName   string  `name:"领涨股名称" dataframe:"block_top_name"`
	BlockTopRate   float64 `name:"领涨股涨幅%" dataframe:"block_top_rate"`
	Tendency       string  `name:"短线趋势" dataframe:"tendency"`
}

// Predict 预测趋势
func (this *ResultInfo) Predict() {
	N := 3
	df := factors.BasicKLine(this.Code)
	if df.Nrow() < N+1 {
		return
	}
	limit := api.RangeFinite(-N)
	OPEN := df.Col("open").Select(limit)
	CLOSE := df.Col("close").Select(limit)
	HIGH := df.Col("high").Select(limit)
	LOW := df.Col("low").Select(limit)
	lastClose := num.AnyToFloat64(CLOSE.IndexOf(-1))
	po := linear.CurveRegression(OPEN).IndexOf(-1).(num.DType)
	pc := linear.CurveRegression(CLOSE).IndexOf(-1).(num.DType)
	ph := linear.CurveRegression(HIGH).IndexOf(-1).(num.DType)
	pl := linear.CurveRegression(LOW).IndexOf(-1).(num.DType)
	if po > lastClose {
		this.Tendency = "高开"
	} else if po == lastClose {
		this.Tendency = "平开"
	} else {
		this.Tendency = "低开"
	}
	if pl > ph {
		this.Tendency += ",冲高回落"
	} else if pl > pc {
		this.Tendency += ",探底回升"
	} else if pc < pl {
		this.Tendency += ",趋势向下"
	} else {
		this.Tendency += ",短线向好"
	}

	fs := []float64{float64(po), float64(pc), float64(ph), float64(pl)}
	sort.Float64s(fs)

	_ = lastClose
}