mirror of
https://github.com/luxfi/container.git
synced 2026-09-24 00:24:34 +00:00
zeekayandHanzo Dev
adf4508709
Modules declare go 1.27.1 and the build images follow. go fix applied the modernizers: any, WaitGroup.Go, range over int, and a promoted-field key in a struct literal. omitzero is not applied, since it changes what the JSON says. Co-authored-by: Hanzo Dev <[email protected]>
59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
// Copyright (C) 2019-2025, Lux Industries, Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package sampler
|
|
|
|
// uniformResample allows for sampling over a uniform distribution without
|
|
// replacement.
|
|
//
|
|
// Sampling is performed by sampling with replacement and resampling if a
|
|
// duplicate is sampled.
|
|
//
|
|
// Initialization takes O(1) time.
|
|
//
|
|
// Sampling is performed in O(count) time and O(count) space.
|
|
type uniformResample struct {
|
|
rng *rng
|
|
length uint64
|
|
drawn map[uint64]struct{}
|
|
}
|
|
|
|
func (s *uniformResample) Initialize(length uint64) {
|
|
s.length = length
|
|
s.drawn = make(map[uint64]struct{})
|
|
}
|
|
|
|
func (s *uniformResample) Sample(count int) ([]uint64, bool) {
|
|
s.Reset()
|
|
|
|
results := make([]uint64, count)
|
|
for i := range count {
|
|
ret, hasNext := s.Next()
|
|
if !hasNext {
|
|
return nil, false
|
|
}
|
|
results[i] = ret
|
|
}
|
|
return results, true
|
|
}
|
|
|
|
func (s *uniformResample) Reset() {
|
|
clear(s.drawn)
|
|
}
|
|
|
|
func (s *uniformResample) Next() (uint64, bool) {
|
|
i := uint64(len(s.drawn))
|
|
if i >= s.length {
|
|
return 0, false
|
|
}
|
|
|
|
for {
|
|
draw := s.rng.Uint64Inclusive(s.length - 1)
|
|
if _, ok := s.drawn[draw]; ok {
|
|
continue
|
|
}
|
|
s.drawn[draw] = struct{}{}
|
|
return draw, true
|
|
}
|
|
}
|