package fmaker import ( "fmt" "time" "github.com/c9s/bbgo/pkg/datatype/floats" "github.com/c9s/bbgo/pkg/indicator" "github.com/c9s/bbgo/pkg/types" ) //go:generate callbackgen -type S3 type S3 struct { types.IntervalWindow // Values Values floats.Slice EndTime time.Time UpdateCallbacks []func(val float64) } func (inc *S3) Last() float64 { if len(inc.Values) == 0 { return 0.0 } return inc.Values[len(inc.Values)-1] } func (inc *S3) CalculateAndUpdate(klines []types.KLine) { if len(klines) < inc.Window { return } var end = len(klines) - 1 var lastKLine = klines[end] if inc.EndTime != zeroTime && lastKLine.GetEndTime().Before(inc.EndTime) { return } var recentT = klines[end-(inc.Window-1) : end+1] val, err := calculateS3(recentT, indicator.KLineClosePriceMapper, indicator.KLineOpenPriceMapper) if err != nil { log.WithError(err).Error("can not calculate") return } inc.Values.Push(val) if len(inc.Values) > indicator.MaxNumOfVOL { inc.Values = inc.Values[indicator.MaxNumOfVOLTruncateSize-1:] } inc.EndTime = klines[end].GetEndTime().Time() inc.EmitUpdate(val) } func (inc *S3) handleKLineWindowUpdate(interval types.Interval, window types.KLineWindow) { if inc.Interval != interval { return } inc.CalculateAndUpdate(window) } func (inc *S3) Bind(updater indicator.KLineWindowUpdater) { updater.OnKLineWindowUpdate(inc.handleKLineWindowUpdate) } func calculateS3(klines []types.KLine, valClose KLineValueMapper, valOpen KLineValueMapper) (float64, error) { window := 2 length := len(klines) if length == 0 || length < window { return 0., fmt.Errorf("insufficient elements for calculating with window = %d", window) } var closes floats.Slice var opens floats.Slice for _, k := range klines { closes.Push(valClose(k)) opens.Push(valOpen(k)) } prevC := closes.Index(1) currO := opens.Index(0) alpha := currO / prevC return alpha, nil }