You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
1.8 KiB
Go
70 lines
1.8 KiB
Go
package lo
|
|
|
|
import "golang.org/x/exp/constraints"
|
|
|
|
// Range creates an array of numbers (positive and/or negative) with given length.
|
|
func Range(elementNum int) []int {
|
|
length := If(elementNum < 0, -elementNum).Else(elementNum)
|
|
result := make([]int, length)
|
|
step := If(elementNum < 0, -1).Else(1)
|
|
for i, j := 0, 0; i < length; i, j = i+1, j+step {
|
|
result[i] = j
|
|
}
|
|
return result
|
|
}
|
|
|
|
// RangeFrom creates an array of numbers from start with specified length.
|
|
func RangeFrom[T constraints.Integer | constraints.Float](start T, elementNum int) []T {
|
|
length := If(elementNum < 0, -elementNum).Else(elementNum)
|
|
result := make([]T, length)
|
|
step := If(elementNum < 0, -1).Else(1)
|
|
for i, j := 0, start; i < length; i, j = i+1, j+T(step) {
|
|
result[i] = j
|
|
}
|
|
return result
|
|
}
|
|
|
|
// RangeWithSteps creates an array of numbers (positive and/or negative) progressing from start up to, but not including end.
|
|
// step set to zero will return empty array.
|
|
func RangeWithSteps[T constraints.Integer | constraints.Float](start, end, step T) []T {
|
|
result := []T{}
|
|
if start == end || step == 0 {
|
|
return result
|
|
}
|
|
if start < end {
|
|
if step < 0 {
|
|
return result
|
|
}
|
|
for i := start; i < end; i += step {
|
|
result = append(result, i)
|
|
}
|
|
return result
|
|
}
|
|
if step > 0 {
|
|
return result
|
|
}
|
|
for i := start; i > end; i += step {
|
|
result = append(result, i)
|
|
}
|
|
return result
|
|
}
|
|
|
|
// Clamp clamps number within the inclusive lower and upper bounds.
|
|
func Clamp[T constraints.Ordered](value T, min T, max T) T {
|
|
if value < min {
|
|
return min
|
|
} else if value > max {
|
|
return max
|
|
}
|
|
return value
|
|
}
|
|
|
|
// Summarizes the values in a collection.
|
|
func SumBy[T any, R constraints.Float | constraints.Integer](collection []T, iteratee func(T) R) R {
|
|
var sum R = 0
|
|
for _, item := range collection {
|
|
sum = sum + iteratee(item)
|
|
}
|
|
return sum
|
|
}
|