mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-27 01:05:15 +00:00
61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
package pivotshort
|
|
|
|
import (
|
|
"github.com/c9s/bbgo/pkg/bbgo"
|
|
"github.com/c9s/bbgo/pkg/indicator"
|
|
"github.com/c9s/bbgo/pkg/types"
|
|
)
|
|
|
|
type TrendEMA struct {
|
|
types.IntervalWindow
|
|
|
|
// MaxGradient is the maximum gradient allowed for the entry.
|
|
MaxGradient float64 `json:"maxGradient"`
|
|
MinGradient float64 `json:"minGradient"`
|
|
|
|
trendEWMA *indicator.EWMA
|
|
|
|
trendEWMALast, trendEWMACurrent, trendGradient float64
|
|
}
|
|
|
|
func (s *TrendEMA) Bind(session *bbgo.ExchangeSession, orderExecutor *bbgo.GeneralOrderExecutor) {
|
|
symbol := orderExecutor.Position().Symbol
|
|
s.trendEWMA = session.StandardIndicatorSet(symbol).EWMA(s.IntervalWindow)
|
|
session.MarketDataStream.OnStart(func() {
|
|
if s.trendEWMA.Length() > 1 {
|
|
s.trendEWMALast = s.trendEWMA.Values[s.trendEWMA.Length()-2]
|
|
s.trendEWMACurrent = s.trendEWMA.Last()
|
|
}
|
|
})
|
|
session.MarketDataStream.OnKLineClosed(types.KLineWith(symbol, s.Interval, func(kline types.KLine) {
|
|
s.trendEWMALast = s.trendEWMACurrent
|
|
s.trendEWMACurrent = s.trendEWMA.Last()
|
|
}))
|
|
}
|
|
|
|
func (s *TrendEMA) Gradient() float64 {
|
|
return s.trendGradient
|
|
}
|
|
|
|
func (s *TrendEMA) GradientAllowed() bool {
|
|
if s.trendEWMALast > 0.0 && s.trendEWMACurrent > 0.0 {
|
|
s.trendGradient = s.trendEWMALast / s.trendEWMACurrent
|
|
}
|
|
|
|
log.Infof("trendEMA %+v current=%f last=%f gradient=%f", s, s.trendEWMACurrent, s.trendEWMALast, s.trendGradient)
|
|
|
|
if s.trendGradient == .0 {
|
|
return false
|
|
}
|
|
|
|
if s.MaxGradient > 0.0 && s.trendGradient < s.MaxGradient {
|
|
return true
|
|
}
|
|
|
|
if s.MinGradient > 0.0 && s.trendGradient > s.MinGradient {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|