tiny.simd.WrapRowMirror

1pub const WrapRowMirror = struct {2    first_row: [*]const f32,3    last_row: [*]const f32,45    pub fn init(image: anytype, ysize: usize) WrapRowMirror {6        std.debug.assert(ysize != 0);7        return .{8            .first_row = image.constRow(0).ptr,9            .last_row = image.constRow(ysize - 1).ptr,10        };11    }1213    pub fn call(self: WrapRowMirror, row: [*]const f32, stride: i64) [*]const f32 {14        std.debug.assert(stride > 0);15        const first = @intFromPtr(self.first_row);16        const last = @intFromPtr(self.last_row);17        const address = @intFromPtr(row);18        const stride_elements: usize = @intCast(stride);19        if (address < first) {20            const distance_bytes = first - address;21            std.debug.assert(distance_bytes % @sizeOf(f32) == 0);22            const distance = distance_bytes / @sizeOf(f32);23            std.debug.assert(distance >= stride_elements);24            return self.first_row + (distance - stride_elements);25        }26        if (address > last) {27            const distance_bytes = address - last;28            std.debug.assert(distance_bytes % @sizeOf(f32) == 0);29            const distance = distance_bytes / @sizeOf(f32);30            std.debug.assert(distance >= stride_elements);31            return self.last_row - (distance - stride_elements);32        }33        return row;34    }35};

1pub const WrapRowMirror = image.WrapRowMirror;