indicator: move EWMA2 to ewma2.go

This commit is contained in:
c9s 2023-05-29 21:42:22 +08:00
parent 8c7962f07f
commit 68570e1eeb
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
2 changed files with 46 additions and 41 deletions

46
pkg/indicator/ewma2.go Normal file
View File

@ -0,0 +1,46 @@
package indicator
import (
"github.com/c9s/bbgo/pkg/datatype/floats"
"github.com/c9s/bbgo/pkg/types"
)
//go:generate callbackgen -type EWMAStream
type EWMAStream struct {
Float64Updater
types.SeriesBase
slice floats.Slice
window int
multiplier float64
}
func EWMA2(source Float64Source, window int) *EWMAStream {
s := &EWMAStream{
window: window,
multiplier: 2.0 / float64(1+window),
}
s.SeriesBase.Series = s.slice
if sub, ok := source.(Float64Subscription); ok {
sub.AddSubscriber(s.calculateAndPush)
} else {
source.OnUpdate(s.calculateAndPush)
}
return s
}
func (s *EWMAStream) calculateAndPush(v float64) {
v2 := s.calculate(v)
s.slice.Push(v2)
s.EmitUpdate(v2)
}
func (s *EWMAStream) calculate(v float64) float64 {
last := s.slice.Last()
m := s.multiplier
return (1.0-m)*last + m*v
}

View File

@ -1,7 +1,6 @@
package indicator
import (
"github.com/c9s/bbgo/pkg/datatype/floats"
"github.com/c9s/bbgo/pkg/types"
)
@ -32,43 +31,3 @@ type Float64Subscription interface {
types.Series
AddSubscriber(f func(v float64))
}
//go:generate callbackgen -type EWMAStream
type EWMAStream struct {
Float64Updater
types.SeriesBase
slice floats.Slice
window int
multiplier float64
}
func EWMA2(source Float64Source, window int) *EWMAStream {
s := &EWMAStream{
window: window,
multiplier: 2.0 / float64(1+window),
}
s.SeriesBase.Series = s.slice
if sub, ok := source.(Float64Subscription); ok {
sub.AddSubscriber(s.calculateAndPush)
} else {
source.OnUpdate(s.calculateAndPush)
}
return s
}
func (s *EWMAStream) calculateAndPush(v float64) {
v2 := s.calculate(v)
s.slice.Push(v2)
s.EmitUpdate(v2)
}
func (s *EWMAStream) calculate(v float64) float64 {
last := s.slice.Last()
m := s.multiplier
return (1.0-m)*last + m*v
}