45 lines
1.0 KiB
Go
45 lines
1.0 KiB
Go
|
package scmaker
|
||
|
|
|
||
|
|
import (
|
||
|
|
"git.qtrade.icu/lychiyu/bbgo/pkg/fixedpoint"
|
||
|
|
indicatorv2 "git.qtrade.icu/lychiyu/bbgo/pkg/indicator/v2"
|
||
|
|
"git.qtrade.icu/lychiyu/bbgo/pkg/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
type IntensityStream struct {
|
||
|
|
*types.Float64Series
|
||
|
|
|
||
|
|
Buy, Sell *indicatorv2.RMAStream
|
||
|
|
window int
|
||
|
|
}
|
||
|
|
|
||
|
|
func Intensity(source indicatorv2.KLineSubscription, window int) *IntensityStream {
|
||
|
|
s := &IntensityStream{
|
||
|
|
Float64Series: types.NewFloat64Series(),
|
||
|
|
window: window,
|
||
|
|
|
||
|
|
Buy: indicatorv2.RMA2(types.NewFloat64Series(), window, false),
|
||
|
|
Sell: indicatorv2.RMA2(types.NewFloat64Series(), window, false),
|
||
|
|
}
|
||
|
|
|
||
|
|
threshold := fixedpoint.NewFromFloat(100.0)
|
||
|
|
source.AddSubscriber(func(k types.KLine) {
|
||
|
|
volume := k.Volume.Float64()
|
||
|
|
|
||
|
|
// ignore zero volume events or <= 10usd events
|
||
|
|
if volume == 0.0 || k.Close.Mul(k.Volume).Compare(threshold) <= 0 {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
c := k.Close.Compare(k.Open)
|
||
|
|
if c > 0 {
|
||
|
|
s.Buy.PushAndEmit(volume)
|
||
|
|
} else if c < 0 {
|
||
|
|
s.Sell.PushAndEmit(volume)
|
||
|
|
}
|
||
|
|
s.Float64Series.PushAndEmit(k.High.Sub(k.Low).Float64())
|
||
|
|
})
|
||
|
|
|
||
|
|
return s
|
||
|
|
}
|