OpenDiablo2/d2core/d2stats/diablo2stats/stat_value.go

package diablo2stats

import (
	"github.com/OpenDiablo2/OpenDiablo2/d2core/d2stats"
)

// static check that Diablo2StatValue implements StatValue
var _ d2stats.StatValue = &Diablo2StatValue{}

// Diablo2StatValue is a diablo 2 implementation of a stat value
type Diablo2StatValue struct {
	number      float64
	stringerFn  func(d2stats.StatValue) string
	numberType  d2stats.StatNumberType
	combineType d2stats.ValueCombineType
}

// NumberType returns the stat value type
func (sv *Diablo2StatValue) NumberType() d2stats.StatNumberType {
	return sv.numberType
}

// CombineType returns the stat value combination type
func (sv *Diablo2StatValue) CombineType() d2stats.ValueCombineType {
	return sv.combineType
}

// Clone returns a deep copy of the stat value
func (sv Diablo2StatValue) Clone() d2stats.StatValue {
	clone := &Diablo2StatValue{}

	switch sv.numberType {
	case d2stats.StatValueInt:
		clone.SetInt(sv.Int())
	case d2stats.StatValueFloat:
		clone.SetFloat(sv.Float())
	}

	clone.stringerFn = sv.stringerFn

	return clone
}

// Int returns the integer version of the stat value
func (sv *Diablo2StatValue) Int() int {
	return int(sv.number)
}

// String returns a string version of the value
func (sv *Diablo2StatValue) String() string {
	return sv.stringerFn(sv)
}

// Float returns a float64 version of the value
func (sv *Diablo2StatValue) Float() float64 {
	return sv.number
}

// SetInt sets the stat value using an int
func (sv *Diablo2StatValue) SetInt(i int) d2stats.StatValue {
	sv.number = float64(i)

	return sv
}

// SetFloat sets the stat value using a float64
func (sv *Diablo2StatValue) SetFloat(f float64) d2stats.StatValue {
	sv.number = f

	return sv
}

// Stringer returns the string evaluation function
func (sv *Diablo2StatValue) Stringer() func(d2stats.StatValue) string {
	return sv.stringerFn
}

// SetStringer sets the string evaluation function
func (sv *Diablo2StatValue) SetStringer(f func(d2stats.StatValue) string) d2stats.StatValue {
	sv.stringerFn = f
	return sv
}