package formula

import (
	"gitee.com/quant1x/num"
	"gitee.com/quant1x/pandas"
)

// DMA 返回动态移动平均
//
//	求S的动态移动平均, A作平滑因子,必须 0<A<1 (此为核心函数，非指标）
//	用法:
//	DMA(X,A),求X的动态移动平均
//	算法:Y=A*X+(1-A)*Y',其中Y'表示上一周期Y值,A必须大于0且小于1.A支持变量
//	例如:
//	DMA(CLOSE,VOL/CAPITAL)表示求以换手率作平滑因子的平均价
func DMA(S pandas.Series, A any) []num.DType {
	switch N := A.(type) {
	case /*nil, */ int8, uint8, int16, uint16, int32, uint32, int64, uint64, int, uint, float32, float64 /*, bool, string*/ :
		x := S.EWM(pandas.EW{Alpha: 1 / num.Any2DType(N), Adjust: false}).Mean().DTypes()
		return x
	case []num.DType:
		s := S.DTypes()
		num.Fill(N, 1.0, true)
		Y := num.Repeat(num.DType(0), len(s))
		Y[0] = s[0]
		for i := 1; i < len(s); i++ {
			a := 1 / N[i]
			if num.DTypeIsNaN(a) {
				a = 1
			}
			Y[i] = a*s[i] + (1-a)*Y[i-1]
		}
		return Y
	case pandas.Series:
		s := S.DTypes()
		M := N.DTypes()
		num.Fill(M, 1.0, true)
		Y := num.Repeat(num.DType(0), len(s))
		Y[0] = s[0]
		for i := 1; i < len(s); i++ {
			a := 1 / M[i]
			if num.DTypeIsNaN(a) {
				a = 1
			}
			Y[i] = a*s[i] + (1-a)*Y[i-1]
		}
		return Y
	}
	return []num.DType{}
}