package indicator

import (
	"math"

	"github.com/c9s/bbgo/pkg/datatype/floats"
	"github.com/c9s/bbgo/pkg/types"
)

// Refer: https://jamesgoulding.com/Research_II/Ehlers/Ehlers%20(Optimal%20Tracking%20Filters).doc
// Ehler's Optimal Tracking Filter, an alpha-beta filter, also called g-h filter

//go:generate callbackgen -type GHFilter
type GHFilter struct {
	types.SeriesBase
	types.IntervalWindow
	a               float64 // maneuverability uncertainty
	b               float64 // measurement uncertainty
	lastMeasurement float64
	Values          floats.Slice

	UpdateCallbacks []func(value float64)
}

func (inc *GHFilter) Update(value float64) {
	inc.update(value, math.Abs(value-inc.lastMeasurement))
}

func (inc *GHFilter) update(value, uncertainty float64) {
	if len(inc.Values) == 0 {
		inc.a = 0
		inc.b = uncertainty / 2
		inc.lastMeasurement = value
		inc.Values.Push(value)
		return
	}
	multiplier := 2.0 / float64(1+inc.Window) // EMA multiplier
	inc.a = multiplier*(value-inc.lastMeasurement) + (1-multiplier)*inc.a
	inc.b = multiplier*uncertainty/2 + (1-multiplier)*inc.b
	lambda := inc.a / inc.b
	lambda2 := lambda * lambda
	alpha := (-lambda2 + math.Sqrt(lambda2*lambda2+16*lambda2)) / 8
	filtered := alpha*value + (1-alpha)*inc.Values.Last(0)
	inc.Values.Push(filtered)
	inc.lastMeasurement = value
}

func (inc *GHFilter) Length() int {
	return inc.Values.Length()
}

func (inc *GHFilter) Last(i int) float64 {
	return inc.Values.Last(i)
}

func (inc *GHFilter) Index(i int) float64 {
	return inc.Last(i)
}

// interfaces implementation check
var _ Simple = &GHFilter{}
var _ types.SeriesExtend = &GHFilter{}

func (inc *GHFilter) PushK(k types.KLine) {
	inc.update(k.Close.Float64(), k.High.Float64()-k.Low.Float64())
}