bbgo_origin/pkg/indicator/v2_multiply.go

package indicator

import "github.com/c9s/bbgo/pkg/datatype/floats"

type MultiplyStream struct {
	*Float64Series
	a, b floats.Slice
}

func Multiply(a, b Float64Source) *MultiplyStream {
	s := &MultiplyStream{
		Float64Series: NewFloat64Series(),
	}

	a.OnUpdate(func(v float64) {
		s.a.Push(v)
		s.calculate()
	})
	b.OnUpdate(func(v float64) {
		s.b.Push(v)
		s.calculate()
	})

	return s
}

func (s *MultiplyStream) calculate() {
	if s.a.Length() != s.b.Length() {
		return
	}

	if s.a.Length() > s.slice.Length() {
		var numNewElems = s.a.Length() - s.slice.Length()
		var tailA = s.a.Tail(numNewElems)
		var tailB = s.b.Tail(numNewElems)
		var tailC = tailA.Mul(tailB)
		for _, f := range tailC {
			s.slice.Push(f)
			s.EmitUpdate(f)
		}
	}
}