darkfi/bin/darkwallet/src/ui/vector_art/shape.rs

/* This file is part of DarkFi (https://dark.fi)
 *
 * Copyright (C) 2020-2024 Dyne.org foundation
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

use crate::{
    error::{Error, Result},
    expr::{Op, SExprCode, SExprMachine, SExprVal},
    gfx::{GfxBufferId, GfxDrawCall, GfxDrawInstruction, GfxDrawMesh, Rectangle, Vertex},
    mesh::Color,
    prop::{PropertyPtr, PropertyUint32, Role},
    util::enumerate,
    ExecutorPtr,
};

#[derive(Debug)]
pub struct ShapeVertex {
    x: SExprCode,
    y: SExprCode,
    color: Color,
}

impl ShapeVertex {
    pub fn new(x: SExprCode, y: SExprCode, color: Color) -> Self {
        Self { x, y, color }
    }

    pub fn from_xy(x: f32, y: f32, color: Color) -> Self {
        Self { x: vec![Op::ConstFloat32(x)], y: vec![Op::ConstFloat32(y)], color }
    }

    pub fn scale(mut self, scale: f32) -> Self {
        let last_x = self.x.pop().unwrap();
        let last_y = self.y.pop().unwrap();
        let mut x = self.x;
        x.push(Op::Mul((Box::new(Op::ConstFloat32(scale)), Box::new(last_x))));
        let mut y = self.y;
        y.push(Op::Mul((Box::new(Op::ConstFloat32(scale)), Box::new(last_y))));
        Self { x, y, color: self.color }
    }
}

#[derive(Debug)]
pub struct VectorShape {
    pub verts: Vec<ShapeVertex>,
    pub indices: Vec<u16>,
}

impl VectorShape {
    pub fn new() -> Self {
        Self { verts: vec![], indices: vec![] }
    }

    pub fn eval(&self, w: f32, h: f32) -> Result<Vec<Vertex>> {
        let mut verts = vec![];
        for shape_vert in &self.verts {
            let mut pos = [0.; 2];
            for (i, shape_X) in [(0, &shape_vert.x), (1, &shape_vert.y)] {
                let mut machine = SExprMachine {
                    globals: vec![
                        ("w".to_string(), SExprVal::Float32(w)),
                        ("h".to_string(), SExprVal::Float32(h)),
                    ],
                    stmts: shape_X,
                };
                pos[i] = machine.call()?.as_f32()?;
            }

            let vert = Vertex { pos, color: shape_vert.color.clone(), uv: [0., 0.] };
            verts.push(vert);
        }
        Ok(verts)
    }

    pub fn add_filled_box(
        &mut self,
        x1: SExprCode,
        y1: SExprCode,
        x2: SExprCode,
        y2: SExprCode,
        color: Color,
    ) {
        self.add_gradient_box(x1, y1, x2, y2, [color.clone(), color.clone(), color.clone(), color])
    }

    /// Colors go clockwise from top-left
    pub fn add_gradient_box(
        &mut self,
        x1: SExprCode,
        y1: SExprCode,
        x2: SExprCode,
        y2: SExprCode,
        color: [Color; 4],
    ) {
        let mut verts = vec![
            ShapeVertex::new(x1.clone(), y1.clone(), color[0]),
            ShapeVertex::new(x2.clone(), y1.clone(), color[1]),
            ShapeVertex::new(x1.clone(), y2.clone(), color[3]),
            ShapeVertex::new(x2, y2, color[2]),
        ];
        let i = self.verts.len() as u16;
        let mut indices = vec![i + 0, i + 2, i + 1, i + 1, i + 2, i + 3];
        self.verts.append(&mut verts);
        self.indices.append(&mut indices);
    }

    // s-expr surgery
    fn sexpr_add(mut x: SExprCode, border_px: f32) -> Option<SExprCode> {
        let eqn = x.pop()?;
        x.push(Op::Add((Box::new(eqn), Box::new(Op::ConstFloat32(border_px)))));
        Some(x)
    }

    pub fn add_outline(
        &mut self,
        x1: SExprCode,
        y1: SExprCode,
        x2: SExprCode,
        y2: SExprCode,
        border_px: f32,
        color: Color,
    ) {
        // LHS
        self.add_filled_box(
            x1.clone(),
            y1.clone(),
            Self::sexpr_add(x1.clone(), border_px).unwrap(),
            y2.clone(),
            color.clone(),
        );
        // THS
        self.add_filled_box(
            x1.clone(),
            y1.clone(),
            x2.clone(),
            Self::sexpr_add(y1.clone(), border_px).unwrap(),
            color.clone(),
        );
        // RHS
        self.add_filled_box(
            Self::sexpr_add(x2.clone(), -border_px).unwrap(),
            y1.clone(),
            x2.clone(),
            y2.clone(),
            color.clone(),
        );
        // BHS
        self.add_filled_box(
            x1.clone(),
            Self::sexpr_add(y2.clone(), -border_px).unwrap(),
            x2.clone(),
            y2.clone(),
            color.clone(),
        );
    }

    pub fn scaled(self, scale: f32) -> Self {
        Self {
            verts: self.verts.into_iter().map(|v| v.scale(scale)).collect(),
            indices: self.indices,
        }
    }
}