bbgo/pkg/indicator/v2/macd.go

34 lines
824 B
Go
Raw Permalink Normal View History

package indicatorv2
import (
"git.qtrade.icu/lychiyu/bbgo/pkg/types"
)
type MACDStream struct {
*SubtractStream
shortWindow, longWindow, signalWindow int
FastEWMA, SlowEWMA, Signal *EWMAStream
Histogram *SubtractStream
}
func MACD2(source types.Float64Source, shortWindow, longWindow, signalWindow int) *MACDStream {
// bind and calculate these first
fastEWMA := EWMA2(source, shortWindow)
slowEWMA := EWMA2(source, longWindow)
macd := Subtract(fastEWMA, slowEWMA)
signal := EWMA2(macd, signalWindow)
histogram := Subtract(macd, signal)
return &MACDStream{
SubtractStream: macd,
shortWindow: shortWindow,
longWindow: longWindow,
signalWindow: signalWindow,
FastEWMA: fastEWMA,
SlowEWMA: slowEWMA,
Signal: signal,
Histogram: histogram,
}
}