bloeys/nset
1
0
Fork 0
mirror of https://github.com/bloeys/nset.git synced 2025-12-29 06:28:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Inline for speeeed

Browse Source
This commit is contained in:
bloeys
2022-06-10 11:15:16 +04:00
parent 52e80d82e4
commit 82ab26e069
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
nset.go
View File

@ -38,8 +38,13 @@ type NSet[T IntsIf] struct {
func (n *NSet[T]) Add(x T) { func (n *NSet[T]) Add(x T) {
bucket := n.GetBucketFromValue(x) //NOTE: The inlines here improve addition times by a good amount (several seconds)
unitIndex := n.GetStorageUnitIndex(x) //NOTE: inline of n.GetBucketFromValue(x)
bucket := &n.Buckets[BucketType(x>>n.shiftAmount)]
//NOTE: This is an inline of n.GetStorageUnitIndex(x)
unitIndex := uint32(((x << BucketIndexingBits) >> BucketIndexingBits) / StorageTypeBits)
if unitIndex >= bucket.StorageUnitCount { if unitIndex >= bucket.StorageUnitCount {
storageUnitsToAdd := unitIndex - bucket.StorageUnitCount + 1 storageUnitsToAdd := unitIndex - bucket.StorageUnitCount + 1
@ -49,7 +54,8 @@ func (n *NSet[T]) Add(x T) {
bucket.StorageUnitCount += storageUnitsToAdd bucket.StorageUnitCount += storageUnitsToAdd
} }
bucket.Data[unitIndex] |= n.GetBitMask(x) //NOTE: Inline of n.GetBitMask(x)
bucket.Data[unitIndex] |= 1 << (((x << BucketIndexingBits) >> BucketIndexingBits) % StorageTypeBits)
} }
func (n *NSet[T]) Remove(x T) { func (n *NSet[T]) Remove(x T) {
@ -109,8 +115,7 @@ func (n *NSet[T]) GetStorageUnitIndex(x T) uint32 {
//unit and bit mask. This is done by shifting left by 4 which removes the top 'n' bits, //unit and bit mask. This is done by shifting left by 4 which removes the top 'n' bits,
//then shifting right by 4 which puts the bits back to their original place, but now //then shifting right by 4 which puts the bits back to their original place, but now
//the top 'n' bits are zeros. //the top 'n' bits are zeros.
return uint32( return uint32(((x << BucketIndexingBits) >> BucketIndexingBits) / StorageTypeBits)
((x << BucketIndexingBits) >> BucketIndexingBits) / StorageTypeBits)
} }
func (n *NSet[T]) GetBitMask(x T) StorageType { func (n *NSet[T]) GetBitMask(x T) StorageType {