go-chart/bollinger_band_series.go

133 lines
3.2 KiB
Go
Raw Permalink Normal View History

2016-07-15 18:02:50 +02:00
package chart
2017-02-03 20:26:53 +01:00
import (
"fmt"
"github.com/wcharczuk/go-chart/seq"
2017-02-03 20:26:53 +01:00
)
2016-07-15 18:02:50 +02:00
// BollingerBandsSeries draws bollinger bands for an inner series.
// Bollinger bands are defined by two lines, one at SMA+k*stddev, one at SMA-k*stdev.
type BollingerBandsSeries struct {
Name string
Style Style
2016-08-01 01:54:09 +02:00
YAxis YAxisType
2016-07-15 18:02:50 +02:00
2016-07-19 00:06:42 +02:00
Period int
2016-07-15 18:02:50 +02:00
K float64
InnerSeries ValuesProvider
2016-07-15 18:02:50 +02:00
valueBuffer *seq.Buffer
2016-07-15 18:02:50 +02:00
}
// GetName returns the name of the time series.
func (bbs BollingerBandsSeries) GetName() string {
return bbs.Name
}
// GetStyle returns the line style.
func (bbs BollingerBandsSeries) GetStyle() Style {
return bbs.Style
}
// GetYAxis returns which YAxis the series draws on.
2016-08-01 01:54:09 +02:00
func (bbs BollingerBandsSeries) GetYAxis() YAxisType {
2016-07-15 18:02:50 +02:00
return bbs.YAxis
}
2016-07-19 00:06:42 +02:00
// GetPeriod returns the window size.
func (bbs BollingerBandsSeries) GetPeriod() int {
if bbs.Period == 0 {
return DefaultSimpleMovingAveragePeriod
2016-07-15 18:02:50 +02:00
}
2016-07-19 00:06:42 +02:00
return bbs.Period
2016-07-15 18:02:50 +02:00
}
// GetK returns the K value, or the number of standard deviations above and below
// to band the simple moving average with.
// Typical K value is 2.0.
2016-07-15 18:17:51 +02:00
func (bbs BollingerBandsSeries) GetK(defaults ...float64) float64 {
if bbs.K == 0 {
if len(defaults) > 0 {
return defaults[0]
}
return 2.0
}
return bbs.K
}
2016-07-15 18:02:50 +02:00
// Len returns the number of elements in the series.
func (bbs BollingerBandsSeries) Len() int {
2016-07-15 18:02:50 +02:00
return bbs.InnerSeries.Len()
}
// GetBoundedValues gets the bounded value for the series.
func (bbs *BollingerBandsSeries) GetBoundedValues(index int) (x, y1, y2 float64) {
2016-07-15 18:02:50 +02:00
if bbs.InnerSeries == nil {
return
}
if bbs.valueBuffer == nil || index == 0 {
bbs.valueBuffer = seq.NewBufferWithCapacity(bbs.GetPeriod())
2016-07-15 18:02:50 +02:00
}
2016-07-19 00:06:42 +02:00
if bbs.valueBuffer.Len() >= bbs.GetPeriod() {
2016-07-15 18:02:50 +02:00
bbs.valueBuffer.Dequeue()
}
px, py := bbs.InnerSeries.GetValues(index)
2016-07-15 18:02:50 +02:00
bbs.valueBuffer.Enqueue(py)
x = px
ay := seq.New(bbs.valueBuffer).Average()
std := seq.New(bbs.valueBuffer).StdDev()
2016-07-15 18:02:50 +02:00
2016-07-15 18:17:51 +02:00
y1 = ay + (bbs.GetK() * std)
y2 = ay - (bbs.GetK() * std)
2016-07-15 18:02:50 +02:00
return
}
// GetBoundedLastValues returns the last bounded value for the series.
func (bbs *BollingerBandsSeries) GetBoundedLastValues() (x, y1, y2 float64) {
2016-07-15 22:27:45 +02:00
if bbs.InnerSeries == nil {
return
}
2016-07-19 00:06:42 +02:00
period := bbs.GetPeriod()
2016-07-15 22:27:45 +02:00
seriesLength := bbs.InnerSeries.Len()
2016-07-19 00:06:42 +02:00
startAt := seriesLength - period
2016-07-15 22:27:45 +02:00
if startAt < 0 {
startAt = 0
}
vb := seq.NewBufferWithCapacity(period)
2016-07-15 22:27:45 +02:00
for index := startAt; index < seriesLength; index++ {
xn, yn := bbs.InnerSeries.GetValues(index)
2016-07-15 22:27:45 +02:00
vb.Enqueue(yn)
x = xn
}
ay := seq.Seq{Provider: vb}.Average()
std := seq.Seq{Provider: vb}.StdDev()
2016-07-15 22:27:45 +02:00
y1 = ay + (bbs.GetK() * std)
y2 = ay - (bbs.GetK() * std)
return
}
2016-07-15 18:02:50 +02:00
// Render renders the series.
func (bbs *BollingerBandsSeries) Render(r Renderer, canvasBox Box, xrange, yrange Range, defaults Style) {
s := bbs.Style.InheritFrom(defaults.InheritFrom(Style{
StrokeWidth: 1.0,
StrokeColor: DefaultAxisColor.WithAlpha(64),
FillColor: DefaultAxisColor.WithAlpha(32),
}))
2016-07-30 01:36:29 +02:00
Draw.BoundedSeries(r, canvasBox, xrange, yrange, s, bbs, bbs.GetPeriod())
2016-07-15 18:02:50 +02:00
}
2017-02-03 20:26:53 +01:00
// Validate validates the series.
func (bbs BollingerBandsSeries) Validate() error {
if bbs.InnerSeries == nil {
return fmt.Errorf("bollinger bands series requires InnerSeries to be set")
}
return nil
}