tiny.simd.MaskedWeakTwoMul

1pub fn MaskedWeakTwoMul(comptime bits: usize) type {2    validateBits(u32, bits);3    return struct {4        key_value: u32 = 0,56        const Self = @This();7        pub const Lane: type = u32;8        pub const mask: u32 = maskValue(u32, bits);910        pub fn init() Self {11            return .{};12        }1314        pub fn initKey(key_value: u32) Self {15            assertMasked(u32, bits, key_value);16            return .{ .key_value = key_value };17        }1819        pub fn initSeed(engine: *const random.AesCtrEngine, seed: u64) Self {20            return .{ .key_value = maybeMask(u32, bits, @truncate(engine.generate(seed, 0))) };21        }2223        pub fn hash(self: Self, input: u32) u32 {24            assertMasked(u32, bits, input);25            var result = input ^ self.key_value;26            result ^= result >> 16;27            result *%= 0x21f0_aaad;28            result = maybeMask(u32, bits, result);29            result ^= result >> 15;30            result *%= 0xf35a_2d97;31            result = maybeMask(u32, bits, result);32            result ^= result >> 15;33            assertMasked(u32, bits, result);34            return result;35        }3637        pub fn oneVec(self: Self, comptime D: type, input: D.Vector) D.Vector {38            requireTag(D, u32);39            assertMaskedVector(D, bits, input);40            var result = input ^ @as(D.Vector, @splat(self.key_value));41            result ^= shift.shiftRight(D, 16, result);42            result = arithmetic.mul(D, result, @splat(0x21f0_aaad));43            result = maybeMaskVector(D, bits, result);44            result ^= shift.shiftRight(D, 15, result);45            result = arithmetic.mul(D, result, @splat(0xf35a_2d97));46            result = maybeMaskVector(D, bits, result);47            result ^= shift.shiftRight(D, 15, result);48            assertMaskedVector(D, bits, result);49            return result;50        }5152        pub fn twoVec(self: Self, comptime D: type, first: *D.Vector, second: *D.Vector) void {53            first.* = self.oneVec(D, first.*);54            second.* = self.oneVec(D, second.*);55        }56    };57}

1pub const MaskedWeakTwoMul = hash.MaskedWeakTwoMul;