mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 23:16:45 +00:00
4651af5767
* modernise: Run new gopls modernise tool against codebase
* Address shazbert's nits
* apichecker, gctcli: Simplify HTML scraping functions and improve depth limit handling
* refactor: Create minSyncInterval const and update order book limit handling for binance and binanceUS
* refactor: Various slice usage improvements and rename TODO
* tranches: Revert deleteByID changes due to performance decrease
Shazbert was a F1 driver in a past lifetime 🏎️
* tranches: Simply retrieve copy
Thanks to shazbert
* documentation: Sort contributors list by contributions
* tranches: Remove deadcode in deleteByID
94 lines
2.2 KiB
Go
94 lines
2.2 KiB
Go
package indicators
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
objects "github.com/d5/tengo/v2"
|
|
"github.com/thrasher-corp/gct-ta/indicators"
|
|
"github.com/thrasher-corp/gocryptotrader/gctscript/modules"
|
|
"github.com/thrasher-corp/gocryptotrader/gctscript/wrappers/validator"
|
|
)
|
|
|
|
// ObvModule volume indicator commands
|
|
var ObvModule = map[string]objects.Object{
|
|
"calculate": &objects.UserFunction{Name: "calculate", Value: obv},
|
|
}
|
|
|
|
// OnBalanceVolume is the string constant
|
|
const OnBalanceVolume = "On Balance Volume"
|
|
|
|
// OBV defines a custom On Balance Volume tengo indicator object type
|
|
type OBV struct {
|
|
objects.Array
|
|
}
|
|
|
|
// TypeName returns the name of the custom type.
|
|
func (o *OBV) TypeName() string {
|
|
return OnBalanceVolume
|
|
}
|
|
|
|
func obv(args ...objects.Object) (objects.Object, error) {
|
|
if len(args) != 1 {
|
|
return nil, objects.ErrWrongNumArguments
|
|
}
|
|
|
|
r := new(OBV)
|
|
if validator.IsTestExecution.Load() == true {
|
|
return r, nil
|
|
}
|
|
|
|
ohlcvInput := objects.ToInterface(args[0])
|
|
ohlcvInputData, valid := ohlcvInput.([]any)
|
|
if !valid {
|
|
return nil, fmt.Errorf(modules.ErrParameterConvertFailed, OHLCV)
|
|
}
|
|
|
|
ohlcvData := make([][]float64, 6)
|
|
var allErrors []string
|
|
for x := range ohlcvInputData {
|
|
t, ok := ohlcvInputData[x].([]any)
|
|
if !ok {
|
|
return nil, errors.New("ohlcvInputData type assert failed")
|
|
}
|
|
if len(t) < 6 {
|
|
return nil, errors.New("ohlcvInputData invalid data length")
|
|
}
|
|
value, err := toFloat64(t[2])
|
|
if err != nil {
|
|
allErrors = append(allErrors, err.Error())
|
|
}
|
|
ohlcvData[2] = append(ohlcvData[2], value)
|
|
|
|
value, err = toFloat64(t[3])
|
|
if err != nil {
|
|
allErrors = append(allErrors, err.Error())
|
|
}
|
|
ohlcvData[3] = append(ohlcvData[3], value)
|
|
|
|
value, err = toFloat64(t[4])
|
|
if err != nil {
|
|
allErrors = append(allErrors, err.Error())
|
|
}
|
|
ohlcvData[4] = append(ohlcvData[4], value)
|
|
|
|
value, err = toFloat64(t[5])
|
|
if err != nil {
|
|
allErrors = append(allErrors, err.Error())
|
|
}
|
|
ohlcvData[5] = append(ohlcvData[5], value)
|
|
}
|
|
|
|
if len(allErrors) > 0 {
|
|
return nil, errors.New(strings.Join(allErrors, ", "))
|
|
}
|
|
|
|
ret := indicators.OBV(ohlcvData[4], ohlcvData[5])
|
|
for x := range ret {
|
|
temp := &objects.Float{Value: ret[x]}
|
|
r.Value = append(r.Value, temp)
|
|
}
|
|
return r, nil
|
|
}
|