bbgo/pkg/indicator/v2/stoch.go

65 lines
1.8 KiB
Go

package indicatorv2
import (
"git.qtrade.icu/lychiyu/bbgo/pkg/datatype/floats"
"git.qtrade.icu/lychiyu/bbgo/pkg/types"
)
const DPeriod int = 3
// Stochastic Oscillator
// - https://www.investopedia.com/terms/s/stochasticoscillator.asp
//
// The Stochastic Oscillator is a technical analysis indicator that is used to identify potential overbought or oversold conditions
// in a security's price. It is calculated by taking the current closing price of the security and comparing it to the high and low prices
// over a specified period of time. This comparison is then plotted as a line on the price chart, with values above 80 indicating overbought
// conditions and values below 20 indicating oversold conditions. The Stochastic Oscillator can be used by traders to identify potential
// entry and exit points for trades, or to confirm other technical analysis signals. It is typically used in conjunction with other indicators
// to provide a more comprehensive view of the security's price.
//go:generate callbackgen -type StochStream
type StochStream struct {
types.SeriesBase
K, D floats.Slice
window int
dPeriod int
highPrices, lowPrices *PriceStream
updateCallbacks []func(k, d float64)
}
// Stochastic Oscillator
func Stoch(source KLineSubscription, window, dPeriod int) *StochStream {
highPrices := HighPrices(source)
lowPrices := LowPrices(source)
s := &StochStream{
window: window,
dPeriod: dPeriod,
highPrices: highPrices,
lowPrices: lowPrices,
}
source.AddSubscriber(func(kLine types.KLine) {
lowest := s.lowPrices.Slice.Tail(s.window).Min()
highest := s.highPrices.Slice.Tail(s.window).Max()
var k = 50.0
var d = 0.0
if highest != lowest {
k = 100.0 * (kLine.Close.Float64() - lowest) / (highest - lowest)
}
s.K.Push(k)
d = s.K.Tail(s.dPeriod).Mean()
s.D.Push(d)
s.EmitUpdate(k, d)
})
return s
}