ghostty/pkg/pixman/image.zig

const std = @import("std");
const c = @import("c.zig").c;
const pixman = @import("main.zig");

pub const Image = opaque {
    pub fn createBitsNoClear(
        format: pixman.FormatCode,
        width: c_int,
        height: c_int,
        bits: [*]u32,
        stride: c_int,
    ) pixman.Error!*Image {
        return @as(?*Image, @ptrCast(c.pixman_image_create_bits_no_clear(
            @intFromEnum(format),
            width,
            height,
            bits,
            stride,
        ))) orelse return pixman.Error.PixmanFailure;
    }

    pub fn createSolidFill(
        color: pixman.Color,
    ) pixman.Error!*Image {
        return @as(?*Image, @ptrCast(c.pixman_image_create_solid_fill(
            @ptrCast(&color),
        ))) orelse return pixman.Error.PixmanFailure;
    }

    pub fn unref(self: *Image) bool {
        return c.pixman_image_unref(@ptrCast(self)) == 1;
    }

    /// A variant of getDataUnsafe that sets the length of the slice to
    /// height * stride. Its possible the buffer is larger but this is the
    /// known safe values. If you KNOW the buffer is larger you can use the
    /// unsafe variant.
    pub fn getData(self: *Image) []u32 {
        const height = self.getHeight();
        const stride = self.getStride();
        const ptr = self.getDataUnsafe();
        const len = @as(usize, @intCast(height * stride));
        return ptr[0..len];
    }

    pub fn getDataUnsafe(self: *Image) [*]u32 {
        return c.pixman_image_get_data(@ptrCast(self));
    }

    pub fn getHeight(self: *Image) c_int {
        return c.pixman_image_get_height(@ptrCast(self));
    }

    pub fn getWidth(self: *Image) c_int {
        return c.pixman_image_get_width(@ptrCast(self));
    }

    pub fn getStride(self: *Image) c_int {
        return c.pixman_image_get_stride(@ptrCast(self));
    }

    pub fn fillBoxes(
        self: *Image,
        op: pixman.Op,
        color: pixman.Color,
        boxes: []const pixman.Box32,
    ) pixman.Error!void {
        if (c.pixman_image_fill_boxes(
            @intFromEnum(op),
            @ptrCast(self),
            @ptrCast(&color),
            @intCast(boxes.len),
            @ptrCast(boxes.ptr),
        ) == 0) return pixman.Error.PixmanFailure;
    }

    pub fn fillRectangles(
        self: *Image,
        op: pixman.Op,
        color: pixman.Color,
        rects: []const pixman.Rectangle16,
    ) pixman.Error!void {
        if (c.pixman_image_fill_rectangles(
            @intFromEnum(op),
            @ptrCast(self),
            @ptrCast(&color),
            @intCast(rects.len),
            @ptrCast(rects.ptr),
        ) == 0) return pixman.Error.PixmanFailure;
    }

    pub fn rasterizeTrapezoid(
        self: *Image,
        trap: pixman.Trapezoid,
        x_off: c_int,
        y_off: c_int,
    ) void {
        c.pixman_rasterize_trapezoid(
            @ptrCast(self),
            @ptrCast(&trap),
            x_off,
            y_off,
        );
    }

    pub fn composite(
        self: *Image,
        op: pixman.Op,
        src: *Image,
        mask: ?*Image,
        src_x: i16,
        src_y: i16,
        mask_x: i16,
        mask_y: i16,
        dest_x: i16,
        dest_y: i16,
        width: u16,
        height: u16,
    ) void {
        c.pixman_image_composite(
            @intFromEnum(op),
            @ptrCast(src),
            @ptrCast(mask),
            @ptrCast(self),
            src_x,
            src_y,
            mask_x,
            mask_y,
            dest_x,
            dest_y,
            width,
            height,
        );
    }

    pub fn compositeTriangles(
        self: *Image,
        op: pixman.Op,
        src: *Image,
        mask_format: pixman.FormatCode,
        x_src: c_int,
        y_src: c_int,
        x_dst: c_int,
        y_dst: c_int,
        tris: []const pixman.Triangle,
    ) void {
        c.pixman_composite_triangles(
            @intFromEnum(op),
            @ptrCast(src),
            @ptrCast(self),
            @intFromEnum(mask_format),
            x_src,
            y_src,
            x_dst,
            y_dst,
            @intCast(tris.len),
            @ptrCast(tris.ptr),
        );
    }
};

test "create and destroy" {
    const testing = std.testing;
    const alloc = testing.allocator;

    const width = 10;
    const height = 10;
    const format: pixman.FormatCode = .g1;
    const stride = format.strideForWidth(width);

    const len = height * @as(usize, @intCast(stride));
    const data = try alloc.alloc(u32, len);
    defer alloc.free(data);
    @memset(data, 0);
    const img = try Image.createBitsNoClear(.g1, width, height, data.ptr, stride);
    try testing.expectEqual(@as(c_int, height), img.getHeight());
    try testing.expectEqual(@as(c_int, stride), img.getStride());
    try testing.expect(img.getData().len == height * stride);
    try testing.expect(img.unref());
}

test "fill boxes a1" {
    const testing = std.testing;
    const alloc = testing.allocator;

    // Dimensions
    const width = 100;
    const height = 100;
    const format: pixman.FormatCode = .a1;
    const stride = format.strideForWidth(width);

    // Image
    const len = height * @as(usize, @intCast(stride));
    const data = try alloc.alloc(u32, len);
    defer alloc.free(data);
    @memset(data, 0);
    const img = try Image.createBitsNoClear(format, width, height, data.ptr, stride);
    defer _ = img.unref();

    // Fill
    const color: pixman.Color = .{ .red = 0xFFFF, .green = 0xFFFF, .blue = 0xFFFF, .alpha = 0xFFFF };
    const boxes = &[_]pixman.Box32{
        .{
            .x1 = 0,
            .y1 = 0,
            .x2 = width,
            .y2 = height,
        },
    };
    try img.fillBoxes(.src, color, boxes);
}