src/2d.c

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 * Copyright 2025 Mike Becker. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *   1. Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 *
 *   2. Redistributions in binary form must reproduce the above copyright
 *      notice, this list of conditions and the following disclaimer in the
 *      documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include "ascension/2d.h"

#include "ascension/constants.h"
#include "ascension/shader.h"
#include "ascension/util.h"

#include <assert.h>

typedef struct asc_rectangle_shader_s {
    AscShaderProgram program;
    asc_uniform_loc color;
    asc_uniform_loc border_color;
    asc_uniform_loc size;
    asc_uniform_loc thickness;
    asc_uniform_loc radius;
} AscRectangleShader;

#define ASC_RECTANGLE_SHADER_FLAG_FILL          1
#define ASC_RECTANGLE_SHADER_FLAG_ROUND         2
#define ASC_RECTANGLE_SHADER_FLAG_BORDER        4

static void asc_rectangle_shader_init(AscShaderProgram *shader, int flags) {
    asc_shader_init_uniform_loc_nice(shader, AscRectangleShader, size);
    if (asc_test_flag(flags, ASC_RECTANGLE_SHADER_FLAG_FILL)) {
        asc_shader_init_uniform_loc_nice(shader, AscRectangleShader, color);
    }
    if (asc_test_flag(flags, ASC_RECTANGLE_SHADER_FLAG_BORDER)) {
        asc_shader_init_uniform_loc_nice(shader, AscRectangleShader, border_color);
        asc_shader_init_uniform_loc_nice(shader, AscRectangleShader, thickness);
    }
    if (asc_test_flag(flags, ASC_RECTANGLE_SHADER_FLAG_ROUND)) {
        asc_shader_init_uniform_loc_nice(shader, AscRectangleShader, radius);
    }
}

static AscShaderProgram *asc_rectangle_shader_create(int flags) {
    return asc_shader_create((AscShaderCodes){
        .vtx = {.source_file = "sprite_vtx.glsl"},
        .frag = {
            .source_file = "rectangle_frag.glsl",
            .preamble_code = (const char *[]) {
                "#define FILL",
                "#define ROUNDED_CORNERS",
                "#define BORDER"
            },
            .preamble_code_flags = flags,
        },
    }, sizeof(AscRectangleShader), asc_rectangle_shader_init, flags);
}

static void asc_rectangle_destroy(AscSceneNode *node) {
    asc_ptr_cast(AscRectangle, rectangle, node);
    asc_mesh_destroy(&rectangle->mesh);
}

static void asc_rectangle_update(AscSceneNode *node) {
    asc_ptr_cast(AscRectangle, rectangle, node);
    asc_mesh_plane_2d(&rectangle->mesh, .size = rectangle->size, .uv_scale = rectangle->size);
}

static void asc_rectangle_draw(const AscCamera *camera, const AscSceneNode *node) {
    asc_cptr_cast(AscRectangle, rectangle, node);
    const bool filled = rectangle->filled;
    const bool round = rectangle->radius > 0;
    const bool border = rectangle->thickness > 0;

    // Compute shader flags
    int flags = 0;
    asc_set_flag_if(flags, ASC_RECTANGLE_SHADER_FLAG_FILL, filled);
    asc_set_flag_if(flags, ASC_RECTANGLE_SHADER_FLAG_ROUND, round);
    asc_set_flag_if(flags, ASC_RECTANGLE_SHADER_FLAG_BORDER, border);

    // Look up and activate the shader
    const AscShaderProgram *shader = asc_shader_lookup(
        ASC_SHADER_RECTANGLE(flags),
        asc_rectangle_shader_create, flags
    );
    if (asc_shader_use(shader, camera)) return;
    asc_cptr_cast(AscRectangleShader, rect_shader, shader);

    // Upload uniforms
    asc_shader_upload_model_matrix(shader, node);

    if (filled) {
        asc_shader_upload_color(rect_shader->color, rectangle->color);
    }
    asc_shader_upload_vec2f(rect_shader->size, rectangle->size);

    if (border) {
        asc_shader_upload_float(rect_shader->thickness, rectangle->thickness);
        asc_shader_upload_color(rect_shader->border_color, rectangle->border_color);
    }
    if (round) {
        asc_shader_upload_float(rect_shader->radius, rectangle->radius);
    }

    // Draw mesh
    asc_mesh_draw_triangle_strip(&rectangle->mesh);
}

AscSceneNode *asc_rectangle_create(struct asc_rectangle_create_args args) {
    AscRectangle *rectangle = cxZallocDefault(sizeof(AscRectangle));

    float pos_x = (float) args.x;
    float pos_y = (float) args.y;
    rectangle->size.width = (float) args.width;
    rectangle->size.height = (float) args.height;

    rectangle->radius = (float)args.radius;
    rectangle->filled = args.filled;
    if (!args.filled && args.thickness == 0) {
        // when we do not fill the rectangle, we need a border
        rectangle->thickness = 1;
    } else {
        rectangle->thickness = args.thickness;
    }

    if (!args.filled && asc_memcmpz(&args.border_color, sizeof(args.border_color))) {
        // convenience fallback:
        // when we are drawing an outline but have no explicit border color,
        // use the main color
        args.border_color = args.color;
    }
    rectangle->color = args.color;
    rectangle->border_color = args.border_color;

    asc_ptr_cast(AscSceneNode, node, rectangle);
    asc_scene_node_init(node,
        ASC_SCENE_NODE_FUNCS(asc_rectangle),
        .pos2d = ASC_VEC2I(pos_x, pos_y),
        .origin2d = ASC_VEC2I(args.origin_x, args.origin_y),
        .render_group = (args.filled && args.color.alpha < 1) || args.border_color.alpha < 1
                             ? ASC_RENDER_GROUP_2D_BLEND
                             : ASC_RENDER_GROUP_2D_OPAQUE
    );
    return node;
}

void asc_rectangle_set_bounds(AscRectangle *rect, asc_rect bounds) {
    float x = (float) bounds.pos.x + rect->node.origin.x;
    float y = (float) bounds.pos.y + rect->node.origin.y;
    float width = (float) bounds.size.width;
    float height = (float) bounds.size.height;
    bool update_transform = false;
    bool update_mesh = false;
    update_transform |= asc_util_check_and_set(rect->node.position.x, x);
    update_transform |= asc_util_check_and_set(rect->node.position.y, y);
    update_mesh |= asc_util_check_and_set(rect->size.width, width);
    update_mesh |= asc_util_check_and_set(rect->size.height, height);
    if (update_mesh) {
        asc_scene_node_update(&rect->node);
    }
    if (update_transform) {
        asc_scene_node_update_transform(&rect->node);
    }
}


static void asc_ellipsis_destroy(AscSceneNode *node) {
    asc_ptr_cast(AscEllipsis, ellipsis, node);
    asc_mesh_destroy(&ellipsis->mesh);
}

static void asc_ellipsis_update(AscSceneNode *node) {
    asc_ptr_cast(AscEllipsis, ellipsis, node);
    asc_vec2f size = asc_vec2f_scale(ellipsis->radii, 2);
    asc_mesh_plane_2d(&ellipsis->mesh, .size = size, .uv_scale = size);
}


typedef struct asc_ellipsis_shader_s {
    AscShaderProgram program;
    asc_uniform_loc color;
    asc_uniform_loc border_color;
    asc_uniform_loc radii;
    asc_uniform_loc thickness;
} AscEllipsisShader;

#define ASC_ELLIPSIS_SHADER_FLAG_FILL          1
#define ASC_ELLIPSIS_SHADER_FLAG_BORDER        2

static void asc_ellipsis_shader_init(AscShaderProgram *shader, int flags) {
    asc_shader_init_uniform_loc_nice(shader, AscEllipsisShader, radii);
    if (asc_test_flag(flags, ASC_ELLIPSIS_SHADER_FLAG_FILL)) {
        asc_shader_init_uniform_loc_nice(shader, AscEllipsisShader, color);
    }
    if (asc_test_flag(flags, ASC_ELLIPSIS_SHADER_FLAG_BORDER)) {
        asc_shader_init_uniform_loc_nice(shader, AscEllipsisShader, thickness);
        asc_shader_init_uniform_loc_nice(shader, AscEllipsisShader, border_color);
    }
}

static AscShaderProgram *asc_ellipsis_shader_create(int flags) {
    return asc_shader_create((AscShaderCodes){
        .vtx = {.source_file = "sprite_vtx.glsl"},
        .frag = {
            .source_file = "ellipsis_frag.glsl",
            .preamble_code = (const char *[]) {
                "#define FILL",
                "#define BORDER"
            },
            .preamble_code_flags = flags,
        },
    }, sizeof(AscEllipsisShader), asc_ellipsis_shader_init, flags);
}

static void asc_ellipsis_draw(const AscCamera *camera, const AscSceneNode *node) {
    asc_cptr_cast(AscEllipsis, ellipsis, node);

    const bool filled = ellipsis->filled;
    const bool border = ellipsis->thickness > 0;

    // Compute shader flags
    int flags = 0;
    asc_set_flag_if(flags, ASC_ELLIPSIS_SHADER_FLAG_FILL, filled);
    asc_set_flag_if(flags, ASC_ELLIPSIS_SHADER_FLAG_BORDER, border);

    // Look up and activate the shader
    const AscShaderProgram *shader = asc_shader_lookup(
        ASC_SHADER_ELLIPSIS(flags),
        asc_ellipsis_shader_create, flags
    );
    if (asc_shader_use(shader, camera)) return;
    asc_cptr_cast(AscEllipsisShader, ellipsis_shader, shader);

    // Upload uniforms
    asc_shader_upload_model_matrix(shader, node);

    if (filled) {
        asc_shader_upload_color(ellipsis_shader->color, ellipsis->color);
    }
    asc_shader_upload_vec2f(ellipsis_shader->radii, ellipsis->radii);

    if (border) {
        asc_shader_upload_float(ellipsis_shader->thickness, ellipsis->thickness);
        asc_shader_upload_color(ellipsis_shader->border_color, ellipsis->border_color);
    }

    // Draw mesh
    asc_mesh_draw_triangle_strip(&ellipsis->mesh);
}

AscSceneNode *asc_ellipsis_create(struct asc_ellipsis_create_args args) {
    AscEllipsis *ellipsis = cxZallocDefault(sizeof(AscEllipsis));

    float pos_x, pos_y;
    if (args.bounds.size.width + args.bounds.size.height > 0) {
        ellipsis->radii.x = (float) args.bounds.size.width / 2.f;
        ellipsis->radii.y = (float) args.bounds.size.height / 2.f;
        pos_x = (float) args.bounds.pos.x + ellipsis->radii.x;
        pos_y = (float) args.bounds.pos.y + ellipsis->radii.y;
    } else {
        const unsigned cx = ASC_NONZERO_OR(args.x, args.center.x);
        const unsigned cy = ASC_NONZERO_OR(args.y, args.center.y);
        const unsigned rx = ASC_NONZERO_OR(args.radius, args.radius_x);
        const unsigned ry = ASC_NONZERO_OR(args.radius, args.radius_y);
        pos_x = (float) cx;
        pos_y = (float) cy;
        ellipsis->radii.x = (float) rx;
        ellipsis->radii.y = (float) ry;
    }

    ellipsis->filled = args.filled;
    if (!args.filled && args.thickness == 0) {
        // when we do not fill the ellipsis, we need a border
        ellipsis->thickness = 1;
    } else {
        ellipsis->thickness = args.thickness;
    }

    if (!args.filled && asc_memcmpz(&args.border_color, sizeof(asc_color))) {
        // convenience fallback:
        // when we are drawing an outline but have no explicit border color,
        // use the main color
        args.border_color = args.color;
    }
    ellipsis->color = args.color;
    ellipsis->border_color = args.border_color;

    asc_ptr_cast(AscSceneNode, node, ellipsis);
    asc_scene_node_init(node,
        ASC_SCENE_NODE_FUNCS(asc_ellipsis),
        .pos2d = ASC_VEC2I(pos_x, pos_y),
        .origin3d = ASC_VEC3F(ellipsis->radii.x, ellipsis->radii.y, 0), // use float to avoid cast
        .render_group = (args.filled && args.color.alpha < 1) || args.border_color.alpha < 1
                             ? ASC_RENDER_GROUP_2D_BLEND
                             : ASC_RENDER_GROUP_2D_OPAQUE
    );
    return node;
}