50 lines
1,013 B
Go
50 lines
1,013 B
Go
package sequence
|
|
|
|
import (
|
|
"math"
|
|
"math/rand"
|
|
)
|
|
|
|
// Random is a random number sequence generator.
|
|
type Random struct {
|
|
rnd *rand.Rand
|
|
scale *float64
|
|
average *float64
|
|
len *int
|
|
}
|
|
|
|
// Len returns the number of elements that will be generated.
|
|
func (r *Random) Len() int {
|
|
if r.len != nil {
|
|
return *r.len
|
|
}
|
|
return math.MaxInt32
|
|
}
|
|
|
|
// GetValue returns the value.
|
|
func (r *Random) GetValue(_ int) float64 {
|
|
if r.average != nil && r.scale != nil {
|
|
return *r.average + *r.scale - (r.rnd.Float64() * (2 * *r.scale))
|
|
} else if r.scale != nil {
|
|
return r.rnd.Float64() * *r.scale
|
|
}
|
|
return r.rnd.Float64()
|
|
}
|
|
|
|
// WithLen sets a maximum len
|
|
func (r *Random) WithLen(length int) *Random {
|
|
r.len = &length
|
|
return r
|
|
}
|
|
|
|
// WithScale sets the scale and returns the Random.
|
|
func (r *Random) WithScale(scale float64) *Random {
|
|
r.scale = &scale
|
|
return r
|
|
}
|
|
|
|
// WithAverage sets the average and returns the Random.
|
|
func (r *Random) WithAverage(average float64) *Random {
|
|
r.average = &average
|
|
return r
|
|
}
|