5 Commits

Author SHA1 Message Date
55928776b4 filter added 2022-09-17 00:04:53 +02:00
84cc7a9e05 Count and CountCmp added 2022-09-07 22:12:33 +02:00
cdb1cf3e03 more helper functions 2022-09-06 01:58:19 +02:00
4f3cf3f537 removed channel dependency 2022-08-28 15:03:40 +02:00
2124bff9de added Search function 2022-08-25 14:03:50 +02:00
11 changed files with 147 additions and 55 deletions

22
contains.go Normal file
View File

@ -0,0 +1,22 @@
package slices
func IndexOf[T comparable](slice []T, value T) int {
return IndexOfCmp(slice, value, DefaultEqualityComparator[T])
}
func IndexOfCmp[T comparable](slice []T, value T, cmp EqualityComparator[T]) int {
for i, v := range slice {
if cmp(v, value) {
return i
}
}
return -1
}
func Contains[T comparable](slice []T, value T) bool {
return ContainsCmp(slice, value, DefaultEqualityComparator[T])
}
func ContainsCmp[T comparable](slice []T, value T, cmp EqualityComparator[T]) bool {
return IndexOfCmp(slice, value, cmp) != -1
}

15
count.go Normal file
View File

@ -0,0 +1,15 @@
package slices
func Count[T comparable](slice []T, value T) int {
return CountCmp(slice, value, DefaultEqualityComparator[T])
}
func CountCmp[T comparable](slice []T, value T, cmp EqualityComparator[T]) int {
c := 0
for _, v := range slice {
if cmp(v, value) {
c++
}
}
return c
}

11
each.go Normal file
View File

@ -0,0 +1,11 @@
package slices
func Each[T any](slice []T, f func(T)) {
EachIndex(slice, func(_ int, v T) { f(v) })
}
func EachIndex[T any](slice []T, f func(int, T)) {
for i, v := range slice {
f(i, v)
}
}

11
filter.go Normal file
View File

@ -0,0 +1,11 @@
package slices
func Filter[T any](slice []T, f func(T) bool) []T {
ret := make([]T, 0, len(slice))
for _, v := range slice {
if f(v) {
ret = append(ret, v)
}
}
return ret
}

9
flat.go Normal file
View File

@ -0,0 +1,9 @@
package slices
func Flat[T any](s ...[]T) []T {
out := make([]T, 0)
for _, v := range s {
out = append(out, v...)
}
return out
}

2
go.mod
View File

@ -1,5 +1,3 @@
module git.milar.in/milarin/slices module git.milar.in/milarin/slices
go 1.19 go 1.19
require git.milar.in/milarin/channel v0.0.7

9
map.go Normal file
View File

@ -0,0 +1,9 @@
package slices
func Map[I, O any](slice []I, mapper func(I) O) []O {
ret := make([]O, 0, len(slice))
for _, v := range slice {
ret = append(ret, mapper(v))
}
return ret
}

16
of.go Normal file
View File

@ -0,0 +1,16 @@
package slices
func Of[T any](values ...T) []T {
return values
}
// OfMap returns a slice containing the return values of the unmapper function
// applied to any key-value pair in m
// The order is random
func OfMap[K comparable, V, T any](m map[K]V, unmapper func(K, V) T) []T {
out := make([]T, 0, len(m))
for k, v := range m {
out = append(out, unmapper(k, v))
}
return out
}

11
search.go Normal file
View File

@ -0,0 +1,11 @@
package slices
func Search[T any](slice []T, f func(a, b T) T) T {
if len(slice) == 0 {
return *new(T)
}
value := slice[0]
Each(slice, func(v T) { value = f(value, v) })
return value
}

View File

@ -1,52 +0,0 @@
package slices
import (
"git.milar.in/milarin/channel"
)
func IndexOf[T comparable](slice []T, value T) int {
return IndexOfCmp(slice, value, DefaultEqualityComparator[T])
}
func IndexOfCmp[T comparable](slice []T, value T, cmp EqualityComparator[T]) int {
for i, v := range slice {
if cmp(v, value) {
return i
}
}
return -1
}
func Contains[T comparable](slice []T, value T) bool {
return ContainsCmp(slice, value, DefaultEqualityComparator[T])
}
func ContainsCmp[T comparable](slice []T, value T, cmp EqualityComparator[T]) bool {
return IndexOfCmp(slice, value, cmp) != -1
}
func Map[I, O any](slice []I, mapper func(I) O) []O {
ret := make([]O, 0, len(slice))
for _, v := range slice {
ret = append(ret, mapper(v))
}
return ret
}
func MapParallel[I, O any](slice []I, mapper func(I) O) []O {
return channel.ToSlice(channel.Map(channel.Of(slice...), mapper))
}
func MapParallelWithRunner[I, O any](slice []I, runner channel.Runner, mapper func(I) O) []O {
return channel.ToSlice(channel.MapWithRunner(channel.Of(slice...), runner, mapper))
}
func Each[T any](slice []T, f func(T)) {
EachIndex(slice, func(_ int, v T) { f(v) })
}
func EachIndex[T any](slice []T, f func(int, T)) {
for i, v := range slice {
f(i, v)
}
}

42
to.go Normal file
View File

@ -0,0 +1,42 @@
package slices
import (
"container/list"
)
// Deref returns a slice containing all dereferenced values of s.
// The returned slice will be a dereferenced and continuous block of memory.
// Nil pointers are ignored.
func Deref[T any](s []*T) []T {
out := make([]T, 0, len(s))
Each(s, func(v *T) {
if v != nil {
out = append(out, *v)
}
})
return out
}
// ToList returns a list.List containing all values of s
func ToList[T any](s []T) *list.List {
l := list.New()
Each(s, func(value T) { l.PushBack(value) })
return l
}
// ToMap returns a map containing all values of s.
// The map key-value pairs are determined by mapper
func ToMap[T any, K comparable, V any](s []T, mapper func(T) (K, V)) map[K]V {
m := map[K]V{}
Each(s, func(value T) {
k, v := mapper(value)
m[k] = v
})
return m
}
// ToStructMap returns a struct{} map containing all values of s as keys.
// It is a shorthand for ToMap(s, func(value T) (T, struct{}) { return value, struct{}{} })
func ToStructMap[T comparable](s []T) map[T]struct{} {
return ToMap(s, func(value T) (T, struct{}) { return value, struct{}{} })
}