tiny.simd.TypedBins

1pub fn TypedBins(comptime capacity: usize, comptime Count: type) type {2    comptime requireBins(capacity, Count);34    return struct {5        counts: [capacity]Count = @splat(0),67        const Self = @This();8        pub const bin_count = capacity;9        pub const CountType = Count;1011        pub fn notify(self: *Self, bin_value: anytype) void {12            const bin_index = index(bin_value);13            self.counts[bin_index] +%= 1;14        }1516        pub fn incrementBy(self: *Self, bin_value: anytype, count: u32) void {17            const bin_index = index(bin_value);18            self.counts[bin_index] +%= countValue(count);19        }2021        pub fn bin(self: *const Self, bin_index: usize) Count {22            std.debug.assert(bin_index < capacity);23            return self.counts[bin_index];24        }2526        pub fn resetBin(self: *Self, bin_index: usize) void {27            std.debug.assert(bin_index < capacity);28            self.counts[bin_index] = 0;29        }3031        pub fn assimilate(self: *Self, other: *const Self) void {32            for (&self.counts, other.counts) |*count, other_count| {33                count.* +%= other_count;34            }35        }3637        pub fn firstNonzero(self: *const Self) usize {38            for (self.counts, 0..) |count, bin_index| {39                if (count != 0) return bin_index;40            }41            return capacity;42        }4344        pub fn lastNonzero(self: *const Self) usize {45            var bin_index = capacity;46            while (bin_index != 0) {47                bin_index -= 1;48                if (self.counts[bin_index] != 0) return bin_index;49            }50            return 0;51        }5253        pub fn numNonzero(self: *const Self) usize {54            var count: usize = 0;55            for (self.counts) |bin_count_value| count += @intFromBool(bin_count_value != 0);56            return count;57        }5859        pub fn modalBinIndex(self: *const Self) usize {60            var maximum: Count = 0;61            var maximum_index: usize = 0;62            for (self.counts, 0..) |count, bin_index| {63                if (count > maximum) {64                    maximum = count;65                    maximum_index = bin_index;66                }67            }68            return maximum_index;69        }7071        pub fn write(72            self: *const Self,73            writer: *std.Io.Writer,74            caption: []const u8,75            skip_zero: bool,76        ) std.Io.Writer.Error!void {77            try writer.print("\n{s} [{d}, modal idx {d}]\n", .{78                caption,79                capacity,80                self.modalBinIndex(),81            });82            const first = self.firstNonzero();83            if (first == capacity) return;84            const last = self.lastNonzero();85            for (first..last + 1) |bin_index| {86                const count = self.counts[bin_index];87                if (!skip_zero or count != 0) {88                    try writer.print(" {d:3}: {d}\n", .{ bin_index, count });89                }90            }91        }9293        pub fn reset(self: *Self) void {94            @memset(&self.counts, 0);95        }9697        fn index(bin_value: anytype) usize {98            const T = @TypeOf(bin_value);99            comptime switch (@typeInfo(T)) {100                .int, .comptime_int => {},101                else => @compileError("bin index must be an integer"),102            };103            switch (@typeInfo(T)) {104                .int => |info| if (info.signedness == .signed) {105                    std.debug.assert(bin_value >= 0);106                },107                .comptime_int => std.debug.assert(bin_value >= 0),108                else => unreachable,109            }110            const bin_index: usize = @intCast(bin_value);111            std.debug.assert(bin_index < capacity);112            return bin_index;113        }114115        fn countValue(count: u32) Count {116            if (@bitSizeOf(Count) >= @bitSizeOf(u32)) return @intCast(count);117            return @truncate(count);118        }119    };120}

1pub const TypedBins = stats.TypedBins;