ezu-paint 0.1.0

Paint GIS features onto a hokusai surface for ezu
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

//! `gradient-conic` — `() -> Raster`. Sweep gradient around `center`.
//! Gradient parameter is the angle (clockwise from `start-angle`)
//! normalized to `[0, 1)`. `start-angle` is in degrees.

use std::sync::Arc;

use ezu_graph::{
    BuiltNode, CoordSpace, EvalCtx, EvalError, FactoryCtx, FactoryError, Node, NodeFactory,
    PortKind, PortSpec, PortValue, RasterBuf,
};
use serde_json::Value;
use xxhash_rust::xxh3::Xxh3;

use crate::nodes::common::{
    read_anchor, read_number_or, read_stops, read_xy, sample_stops, Anchor,
};
use crate::nodes::raster::gradient_common::pixel_to_user;

struct GradientConicNode {
    center: [f32; 2],
    start_angle: f32, // degrees
    stops: Vec<(f32, [f32; 4])>,
    anchor: Anchor,
}

impl Node for GradientConicNode {
    fn op_name(&self) -> &'static str {
        "gradient-conic"
    }
    fn inputs(&self) -> &[PortSpec] {
        &[]
    }
    fn output(&self) -> PortKind {
        PortKind::Raster
    }
    fn coord_space(&self) -> CoordSpace {
        match self.anchor {
            Anchor::Tile => CoordSpace::Tile,
            Anchor::World => CoordSpace::World,
        }
    }
    fn eval(&self, ctx: &EvalCtx<'_>, _: &[Option<PortValue>]) -> Result<PortValue, EvalError> {
        let size = ctx.canvas.padded_size();
        let mut out = RasterBuf::new(size, size);
        let start_rad = self.start_angle.to_radians();
        for y in 0..size {
            for x in 0..size {
                let (ux, uy) = pixel_to_user(x, y, ctx, self.anchor);
                let dx = ux - self.center[0];
                let dy = uy - self.center[1];
                // atan2 returns (-π, π]; subtract `start_angle` and
                // wrap to `[0, 2π)` for a clockwise sweep starting at
                // `start-angle` pointing right (+x).
                let ang = dy.atan2(dx) - start_rad;
                let t = ang.rem_euclid(std::f32::consts::TAU) / std::f32::consts::TAU;
                let c = sample_stops(&self.stops, t);
                let i = ((y * size + x) * 4) as usize;
                let a = c[3].clamp(0.0, 1.0);
                out.pixels[i] = (c[0].clamp(0.0, 1.0) * a * 255.0).round() as u8;
                out.pixels[i + 1] = (c[1].clamp(0.0, 1.0) * a * 255.0).round() as u8;
                out.pixels[i + 2] = (c[2].clamp(0.0, 1.0) * a * 255.0).round() as u8;
                out.pixels[i + 3] = (a * 255.0).round() as u8;
            }
        }
        Ok(PortValue::Raster(Arc::new(out)))
    }
    fn param_hash(&self, h: &mut Xxh3) {
        h.update(b"gradient-conic");
        h.update(&self.center[0].to_le_bytes());
        h.update(&self.center[1].to_le_bytes());
        h.update(&self.start_angle.to_le_bytes());
        h.update(&[self.anchor as u8]);
        for (t, c) in &self.stops {
            h.update(&t.to_le_bytes());
            for v in c {
                h.update(&v.to_le_bytes());
            }
        }
    }
}

pub(super) struct GradientConicFactory;
impl NodeFactory for GradientConicFactory {
    fn op_name(&self) -> &'static str {
        "gradient-conic"
    }
    fn build(
        &self,
        fields: &serde_json::Map<String, Value>,
        ctx: &FactoryCtx<'_>,
    ) -> Result<BuiltNode, FactoryError> {
        let center = read_xy(fields, "center", ctx, [0.5, 0.5])?;
        let start_angle = read_number_or(fields, "start-angle", ctx, 0.0)? as f32;
        let stops = read_stops(fields, "stops", ctx)?;
        let anchor = read_anchor(fields, "anchor", ctx)?;
        Ok(BuiltNode {
            node: Box::new(GradientConicNode {
                center,
                start_angle,
                stops,
                anchor,
            }),
            connections: vec![],
        })
    }
    fn schema(&self) -> Value {
        serde_json::json!({
            "description": "Conic (sweep) gradient around `center`. Gradient parameter is the clockwise angle from `start-angle` (degrees) normalized to [0, 1). 0° points along +x.",
            "properties": {
                "center": { "type": "array", "items": { "type": "number" }, "minItems": 2, "maxItems": 2, "default": [0.5, 0.5] },
                "start-angle": { "type": "number", "default": 0.0 },
                "stops": { "type": "array", "items": { "type": "array", "minItems": 2, "maxItems": 2 }, "minItems": 2 },
                "anchor": { "type": "string", "enum": ["tile", "world"], "default": "tile" },
            },
            "required": ["stops"],
        })
    }
}

ezu_graph::submit_node!(GradientConicFactory);