yog-cli 0.47.1

//! The designer's stand-in for the game's `DrawContext`: a display list of
//! `draw2d_*` calls (what yog-ui's renderer — or a mod's own render code —
//! issues) rasterized the way the Yog hosts draw them in game: coordinates
//! truncated to whole GUI pixels (`NativeDraw` casts with `(int)`), alpha
//! blending, Minecraft's font (see [`font`](super::font)) and textures.
//! Output is RGBA at `scale` physical pixels per GUI pixel, for the canvas
//! to show with nearest-neighbour scaling.

use std::collections::HashMap;
use std::sync::Arc;

use serde::{Deserialize, Serialize};

use yog_gfx::Color;
use yog_ui::text::LINE_H;

use super::assets::{Image, Resources};
use super::font::{Font, Glyph, styled};

/// One `draw2d_*` call (GUI pixels, colors `0xAARRGGBB`).
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(tag = "op", rename_all = "snake_case")]
pub enum DrawOp {
    Rect {
        x1: f32,
        y1: f32,
        x2: f32,
        y2: f32,
        color: u32,
    },
    Gradient {
        x1: f32,
        y1: f32,
        x2: f32,
        y2: f32,
        top: u32,
        bottom: u32,
    },
    Text {
        text: String,
        font: String,
        x: f32,
        y: f32,
        color: u32,
        shadow: bool,
    },
    /// `draw2d_text_scaled`: text `scale` times its size.
    TextScaled {
        text: String,
        font: String,
        x: f32,
        y: f32,
        scale: f32,
        color: u32,
        shadow: bool,
    },
    Item {
        id: String,
        x: f32,
        y: f32,
        size: f32,
    },
    /// `draw2d_blocks`: blocks in 3D (`x y z state` lines) as big as fits
    /// in `x, y, w, h`, turned `yaw`, tilted `pitch` (degrees).
    Blocks {
        list: String,
        x: f32,
        y: f32,
        w: f32,
        h: f32,
        yaw: f32,
        pitch: f32,
    },
    /// `draw2d_image`: a Yog image source into `x, y, w, h`; `region` `u v w h` (see [`images`](super::images)).
    Image {
        source: String,
        x: f32,
        y: f32,
        w: f32,
        h: f32,
        region: [f32; 4],
        tint: u32,
    },
    /// `draw2d_svg`: an SVG document into `x, y, w, h`.
    Svg {
        data: String,
        x: f32,
        y: f32,
        w: f32,
        h: f32,
    },
    /// `draw2d_entity`: a living entity standing in `x, y, w, h` — drawn
    /// here as its spawn egg (the designer has no entity models).
    Entity {
        entity: String,
        x: f32,
        y: f32,
        w: f32,
        h: f32,
    },
    Scissor {
        x: i32,
        y: i32,
        w: i32,
        h: i32,
    },
    ClearScissor,
}

/// An RGBA canvas `w × h` physical pixels.
pub struct Canvas<'r> {
    pub w: u32,
    pub h: u32,
    pub rgba: Vec<u8>,
    scale: f32,
    clip: Option<(i32, i32, i32, i32)>,
    res: &'r Resources,
    fonts: HashMap<String, Arc<Font>>,
}

/// Straight-alpha "over".
fn blend(dst: &mut [u8], src: [f32; 4]) {
    let sa = src[3];
    if sa <= 0.0 {
        return;
    }
    let d = Color::from_rgba8([dst[0], dst[1], dst[2], dst[3]]).to_f32();
    let oa = sa + d[3] * (1.0 - sa);
    let mut o = [0.0, 0.0, 0.0, oa];
    for i in 0..3 {
        o[i] = (src[i] * sa + d[i] * d[3] * (1.0 - sa)) / oa;
    }
    dst[..4].copy_from_slice(&Color::from_f32(o).to_rgba8());
}

fn argb(c: u32) -> [f32; 4] {
    Color(c).to_f32()
}

/// Text whose alpha is below this the game draws fully opaque (`Font.adjustColor`: `(c & 0xFC000000) == 0`).
const TEXT_MIN_ALPHA: u8 = 4;

/// The game draws text whose alpha is (almost) 0 fully opaque.
fn text_color(c: u32) -> u32 {
    let c = Color(c);
    if c.a() < TEXT_MIN_ALPHA { c.with_alpha(u8::MAX).0 } else { c.0 }
}

/// The game's text shadow: each channel a quarter of the text's (`rgb & 0xFCFCFC >> 2`).
fn shadow_color(c: u32) -> u32 {
    const QUARTER_MASK: u32 = 0x00FC_FCFC;
    const QUARTER_SHIFT: u32 = 2;
    (c & Color::BLACK.0) | ((c & QUARTER_MASK) >> QUARTER_SHIFT)
}

/// Where the game draws a line's underline and strikethrough: one pixel
/// above the line's bottom, and one through its middle.
const UNDERLINE: (f32, f32) = (LINE_H, LINE_H - 1.0);
const STRIKE: (f32, f32) = (LINE_H / 2.0, LINE_H / 2.0 - 1.0);

impl<'r> Canvas<'r> {
    pub fn new(gui_w: u32, gui_h: u32, scale: u32, res: &'r Resources) -> Self {
        let (w, h) = (gui_w * scale, gui_h * scale);
        Canvas {
            w,
            h,
            rgba: vec![0; w as usize * h as usize * 4],
            scale: scale as f32,
            clip: None,
            res,
            fonts: HashMap::new(),
        }
    }

    pub fn font(&mut self, id: &str) -> Arc<Font> {
        let res = self.res;
        self.fonts
            .entry(id.to_string())
            .or_insert_with(|| Font::load(res, id))
            .clone()
    }

    /// Physical pixel span of GUI `[a, b)`, inside the canvas (and the clip).
    fn span(&self, a: f32, b: f32, vertical: bool) -> (i32, i32) {
        let max = if vertical { self.h } else { self.w } as i32;
        let (mut lo, mut hi) = (
            (a * self.scale).floor() as i32,
            (b * self.scale).floor() as i32,
        );
        if let Some((cx, cy, cw, ch)) = self.clip {
            let (c0, c1) = if vertical {
                (cy, cy + ch)
            } else {
                (cx, cx + cw)
            };
            lo = lo.max((c0 as f32 * self.scale) as i32);
            hi = hi.min((c1 as f32 * self.scale) as i32);
        }
        (lo.max(0), hi.min(max))
    }

    fn px(&mut self, x: i32, y: i32) -> &mut [u8] {
        let i = ((y as u32 * self.w + x as u32) * 4) as usize;
        &mut self.rgba[i..i + 4]
    }

    /// `img` mapped onto the parallelogram `o`, `o + a`, `o + a + b`, `o + b`
    /// (GUI pixels; `a` runs along the image's x, `b` along its y), times
    /// `shade`.
    pub(super) fn quad(&mut self, img: &Image, o: [f32; 2], a: [f32; 2], b: [f32; 2], shade: f32) {
        let det = a[0] * b[1] - a[1] * b[0];
        if det.abs() < 1e-6 {
            return;
        }
        let xs = [o[0], o[0] + a[0], o[0] + b[0], o[0] + a[0] + b[0]];
        let ys = [o[1], o[1] + a[1], o[1] + b[1], o[1] + a[1] + b[1]];
        let min = |v: [f32; 4]| v.iter().copied().fold(f32::MAX, f32::min);
        let max = |v: [f32; 4]| v.iter().copied().fold(f32::MIN, f32::max);
        let (xa, xb) = self.span(min(xs), max(xs) + 1.0 / self.scale, false);
        let (ya, yb) = self.span(min(ys), max(ys) + 1.0 / self.scale, true);
        for py in ya..yb {
            let gy = (py as f32 + 0.5) / self.scale - o[1];
            for px in xa..xb {
                let gx = (px as f32 + 0.5) / self.scale - o[0];
                let u = (gx * b[1] - gy * b[0]) / det;
                let v = (a[0] * gy - a[1] * gx) / det;
                if !(0.0..1.0).contains(&u) || !(0.0..1.0).contains(&v) {
                    continue;
                }
                let t = img.pixel((u * img.width as f32) as u32, (v * img.width.min(img.height) as f32) as u32).to_rgba8();
                if t[3] == 0 {
                    continue;
                }
                let [r, g, b, a] = Color::from_rgba8(t).to_f32();
                let c = [r * shade, g * shade, b * shade, a];
                blend(self.px(px, py), c);
            }
        }
    }

    pub(super) fn resources(&self) -> &Resources {
        self.res
    }

    pub fn fill(&mut self, x1: f32, y1: f32, x2: f32, y2: f32, top: u32, bottom: u32) {
        let (x1, x2) = (
            (x1 as i32).min(x2 as i32) as f32,
            (x1 as i32).max(x2 as i32) as f32,
        );
        let (y1, y2) = (
            (y1 as i32).min(y2 as i32) as f32,
            (y1 as i32).max(y2 as i32) as f32,
        );
        let (xa, xb) = self.span(x1, x2, false);
        let (ya, yb) = self.span(y1, y2, true);
        let (t, b) = (argb(top), argb(bottom));
        let height = ((y2 - y1) * self.scale).max(1.0);
        for py in ya..yb {
            let f = ((py as f32 + 0.5 - y1 * self.scale) / height).clamp(0.0, 1.0);
            let c = [0, 1, 2, 3].map(|i| t[i] + (b[i] - t[i]) * f);
            for px in xa..xb {
                blend(self.px(px, py), c);
            }
        }
    }

    /// Draws `img` texels `(u, v)`–`(u + w·tw/…)` into GUI box `x, y, w, h`
    /// — UVs normalized against the declared texture size `tw × th`.
    #[allow(clippy::too_many_arguments)]
    pub(super) fn blit(
        &mut self,
        img: &Image,
        x: f32,
        y: f32,
        w: f32,
        h: f32,
        uv: [f32; 4],
        tint: [f32; 4],
    ) {
        let (xa, xb) = self.span(x, x + w, false);
        let (ya, yb) = self.span(y, y + h, true);
        let [u0, v0, u1, v1] = uv;
        for py in ya..yb {
            let fy = ((py as f32 + 0.5) / self.scale - y) / h;
            let ty = ((v0 + (v1 - v0) * fy) * img.height as f32).floor() as i64;
            if ty < 0 || ty >= img.height as i64 {
                continue;
            }
            for px in xa..xb {
                let fx = ((px as f32 + 0.5) / self.scale - x) / w;
                let tx = ((u0 + (u1 - u0) * fx) * img.width as f32).floor() as i64;
                if tx < 0 || tx >= img.width as i64 {
                    continue;
                }
                let t = img.pixel(tx as u32, ty as u32).to_rgba8();
                let t = Color::from_rgba8(t).to_f32();
                let c = [0, 1, 2, 3].map(|i| t[i] * tint[i]);
                blend(self.px(px, py), c);
            }
        }
    }

    fn glyph(&mut self, g: &Glyph, x: f32, y: f32, k: f32, color: u32) {
        let Some(b) = &g.bitmap else { return };
        let img = Image {
            width: b.w,
            height: b.h,
            rgba: b.rgba.clone(),
        };
        self.blit(
            &img,
            x,
            y + b.top * k,
            b.w as f32 / b.oversample * k,
            b.h as f32 / b.oversample * k,
            [0.0, 0.0, 1.0, 1.0],
            argb(color),
        );
    }

    /// `Font.drawInBatch`: the shadow pass (offset, color / 4), then the text.
    pub fn text(&mut self, text: &str, font: &str, x: f32, y: f32, color: u32, shadow: bool) {
        self.text_scaled(text, font, (x as i32) as f32, (y as i32) as f32, 1.0, color, shadow);
    }

    /// [`Self::text`] `k` times its size (the host scales the pose: no
    /// rounding to whole pixels).
    #[allow(clippy::too_many_arguments)]
    pub fn text_scaled(&mut self, text: &str, font: &str, x: f32, y: f32, k: f32, color: u32, shadow: bool) {
        let font = self.font(font);
        let chars = styled(text, text_color(color));
        let passes: &[bool] = if shadow { &[true, false] } else { &[false] };
        for &is_shadow in passes {
            let mut pen = x;
            for s in &chars {
                let g = font.glyph(s.ch);
                let c = if is_shadow {
                    shadow_color(s.color)
                } else {
                    s.color
                };
                let off = if is_shadow { g.shadow_offset * k } else { 0.0 };
                self.glyph(&g, pen + off, y + off, k, c);
                let mut adv = g.advance * k;
                if s.bold {
                    self.glyph(&g, pen + off + g.bold_offset * k, y + off, k, c);
                    adv += g.bold_offset * k;
                }
                if s.underline {
                    self.fill(
                        pen + off - 1.0,
                        y + off + UNDERLINE.0,
                        pen + off + adv,
                        y + off + UNDERLINE.1,
                        c,
                        c,
                    );
                }
                if s.strike {
                    self.fill(
                        pen + off - 1.0,
                        y + off + STRIKE.0,
                        pen + off + adv,
                        y + off + STRIKE.1,
                        c,
                        c,
                    );
                }
                pen += adv;
            }
        }
    }

    pub fn draw(&mut self, op: &DrawOp) {
        match op {
            DrawOp::Rect {
                x1,
                y1,
                x2,
                y2,
                color,
            } => self.fill(*x1, *y1, *x2, *y2, *color, *color),
            DrawOp::Gradient {
                x1,
                y1,
                x2,
                y2,
                top,
                bottom,
            } => self.fill(*x1, *y1, *x2, *y2, *top, *bottom),
            DrawOp::Text {
                text,
                font,
                x,
                y,
                color,
                shadow,
            } => self.text(text, font, *x, *y, *color, *shadow),
            DrawOp::TextScaled { text, font, x, y, scale, color, shadow } => {
                self.text_scaled(text, font, *x, *y, *scale, *color, *shadow)
            }
            DrawOp::Blocks { list, x, y, w, h, yaw, pitch } => {
                super::blocks3d::draw(self, list, [*x, *y, *w, *h], *yaw, *pitch)
            }
            DrawOp::Item { id, x, y, size } => {
                // As the game shows it in a slot (blocks and heads in 3D;
                // mod items: from their `content/` file).
                let source = format!("item:{id}");
                super::images::draw(self, &source, [*x, *y, *size, *size], [0.0; 4], Color::WHITE.0);
            }
            DrawOp::Image {
                source,
                x,
                y,
                w,
                h,
                region,
                tint,
            } => super::images::draw(self, source, [*x, *y, *w, *h], *region, *tint),
            DrawOp::Svg { data, x, y, w, h } => super::images::draw_svg(self, data, [*x, *y, *w, *h], Color::WHITE.0),
            DrawOp::Entity { entity, x, y, w, h } => super::images::draw_entity(self, entity, [*x, *y, *w, *h]),
            DrawOp::Scissor { x, y, w, h } => self.clip = Some((*x, *y, *w, *h)),
            DrawOp::ClearScissor => self.clip = None,
        }
    }

    pub fn png(&self) -> Result<Vec<u8>, String> {
        let mut out = Vec::new();
        let mut enc = png::Encoder::new(&mut out, self.w, self.h);
        enc.set_color(png::ColorType::Rgba);
        enc.set_depth(png::BitDepth::Eight);
        enc.set_compression(png::Compression::Fast);
        let mut writer = enc.write_header().map_err(|e| e.to_string())?;
        writer
            .write_image_data(&self.rgba)
            .map_err(|e| e.to_string())?;
        writer.finish().map_err(|e| e.to_string())?;
        Ok(out)
    }
}

/// The most physical pixels a canvas gets (256 MiB of RGBA).
const MAX_PIXELS: u64 = 1 << 26;

/// `scale` clamped to 1..=8, lowered until a `gui_w`×`gui_h` canvas fits [`MAX_PIXELS`].
pub fn fit_scale(gui_w: u32, gui_h: u32, scale: u32) -> u32 {
    let gui = u64::from(gui_w) * u64::from(gui_h);
    let mut scale = scale.clamp(1, 8);
    while scale > 1 && gui.saturating_mul(u64::from(scale * scale)) > MAX_PIXELS {
        scale -= 1;
    }
    scale
}

/// Rasterizes a display list.
pub fn rasterize<'r>(
    ops: &[DrawOp],
    gui_w: u32,
    gui_h: u32,
    scale: u32,
    res: &'r Resources,
) -> Canvas<'r> {
    let mut canvas = Canvas::new(gui_w, gui_h, fit_scale(gui_w, gui_h, scale), res);
    for op in ops {
        canvas.draw(op);
    }
    canvas
}

#[cfg(test)]
mod tests {
    use super::*;

    fn res() -> Resources {
        Resources {
            mod_root: None,
            vanilla: None,
        }
    }

    #[test]
    fn huge_canvases_lower_their_scale() {
        assert_eq!(fit_scale(400, 300, 3), 3);
        assert_eq!(fit_scale(400, 300, 0), 1);
        assert_eq!(fit_scale(4096, 4096, 8), 2);
        assert_eq!(fit_scale(u32::MAX, u32::MAX, 8), 1);
    }

    #[test]
    fn rects_truncate_to_gui_pixels_and_blend() {
        let r = res();
        let c = rasterize(
            &[DrawOp::Rect {
                x1: 1.7,
                y1: 0.0,
                x2: 3.2,
                y2: 1.0,
                color: 0xFF_FF0000,
            }],
            4,
            1,
            2,
            &r,
        );
        // x 1..3 → physical 2..6
        let px = |x: usize| &c.rgba[x * 4..x * 4 + 4];
        assert_eq!(px(1), &[0, 0, 0, 0]);
        assert_eq!(px(2), &[255, 0, 0, 255]);
        assert_eq!(px(5), &[255, 0, 0, 255]);
        assert_eq!(px(6), &[0, 0, 0, 0]);
    }

    #[test]
    fn scissor_clips() {
        let r = res();
        let ops = [
            DrawOp::Scissor {
                x: 0,
                y: 0,
                w: 1,
                h: 1,
            },
            DrawOp::Rect {
                x1: 0.0,
                y1: 0.0,
                x2: 2.0,
                y2: 1.0,
                color: 0xFF_FFFFFF,
            },
        ];
        let c = rasterize(&ops, 2, 1, 1, &r);
        assert_eq!(c.rgba[3], 255);
        assert_eq!(c.rgba[7], 0);
    }

    #[test]
    fn missing_font_draws_boxes_with_the_shadow_first() {
        let r = res();
        let mut c = Canvas::new(16, 10, 1, &r);
        c.text("A", "", 0.0, 0.0, 0xFF_FFFFFF, true);
        // Missing glyph box: top-left texel white, shadow at (1,1) under the text's own (1,1)? (1,1) is the box's inside → shadow shows.
        assert_eq!(&c.rgba[0..4], &[255, 255, 255, 255]);
        let i = (16 + 1) * 4;
        assert_eq!(&c.rgba[i..i + 4], &[63, 63, 63, 255]);
    }
}