tiny.simd.VectorXoshiro

1pub fn VectorXoshiro(comptime D: type) type {2    requireU64(D);3    return struct {4        state: [4]D.Vector,56        const Self = @This();7        const DF: type = D.rebind(f64);89        pub fn init(seed: u64) Self {10            return initThread(seed, 0);11        }1213        pub fn initThread(seed: u64, thread_number: u64) Self {14            var scalar = Xoshiro.init(seed);15            var stream: u64 = 0;16            while (stream < thread_number) : (stream += 1) scalar.longJump();17            var state: [4]D.Vector = @splat(@splat(0));18            inline for (0..D.lane_count) |lane| {19                const scalar_state = scalar.getState();20                inline for (0..4) |index| state[index][lane] = scalar_state[index];21                scalar.jump();22            }23            return .{ .state = state };24        }2526        pub fn next(self: *Self) D.Vector {27            return update(&self.state);28        }2930        pub fn fill(self: *Self, output: []u64) void {31            var index: usize = 0;32            while (index + D.lane_count <= output.len) : (index += D.lane_count) {33                const value: [D.lane_count]u64 = self.next();34                @memcpy(output[index .. index + D.lane_count], &value);35            }36            if (index != output.len) {37                const value: [D.lane_count]u64 = self.next();38                @memcpy(output[index..], value[0 .. output.len - index]);39            }40        }4142        pub fn uniform(self: *Self) DF.Vector {43            return toUniform(self.next());44        }4546        pub fn fillUniform(self: *Self, output: []f64) void {47            var index: usize = 0;48            while (index + D.lane_count <= output.len) : (index += D.lane_count) {49                const value: [D.lane_count]f64 = self.uniform();50                @memcpy(output[index .. index + D.lane_count], &value);51            }52            if (index != output.len) {53                const value: [D.lane_count]f64 = self.uniform();54                @memcpy(output[index..], value[0 .. output.len - index]);55            }56        }5758        pub fn getState(self: Self) [4]D.Vector {59            return self.state;60        }6162        pub fn setState(self: *Self, state: [4]D.Vector) void {63            self.state = state;64        }6566        pub fn stateSize() usize {67            return 4 * D.lane_count;68        }6970        fn update(state: *[4]D.Vector) D.Vector {71            const result = rotate.rotateRightSame(D, state[0] +% state[3], 41) +% state[0];72            const temporary = shift.shiftLeft(D, 17, state[1]);73            state[2] ^= state[0];74            state[3] ^= state[1];75            state[1] ^= state[2];76            state[0] ^= state[3];77            state[2] ^= temporary;78            state[3] = rotate.rotateRightSame(D, state[3], 19);79            return result;80        }8182        fn toUniform(value: D.Vector) DF.Vector {83            const bits = shift.shiftRight(D, 11, value);84            var result: DF.Vector = undefined;85            inline for (0..D.lane_count) |index| {86                result[index] = @as(f64, @floatFromInt(bits[index])) * uniform_scale;87            }88            return result;89        }90    };91}

1pub const VectorXoshiro = random.VectorXoshiro;