ezu-paint 0.3.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
121
122
123
124
125
126
127

//! `transform` — `Features -> Features`. Apply an affine (translate +
//! rotate + scale) transform to every vertex. Rotation happens around
//! `pivot` (in feature-space coordinates, default = origin) before
//! the final translation. Scale is per-axis.

use ezu_features::ops::transform::transform;
use ezu_graph::{
    schema_frag, take_input_ref, BuiltNode, Connection, CoordSpace, EvalCtx, EvalError, FactoryCtx,
    FactoryError, Node, NodeFactory, PortKind, PortSpec, PortValue,
};
use serde_json::Value;
use xxhash_rust::xxh3::Xxh3;

use crate::nodes::common::{downcast_features, features_value, read_number_or, read_xy};

struct TransformNode {
    translate: (f64, f64),
    rotation_rad: f64,
    scale: (f64, f64),
    pivot: (f64, f64),
}

impl Node for TransformNode {
    fn op_name(&self) -> &'static str {
        "transform"
    }
    fn inputs(&self) -> &[PortSpec] {
        static SPECS: &[PortSpec] = &[PortSpec {
            name: "features",
            accepts: &[PortKind::Features],
            optional: false,
        }];
        SPECS
    }
    fn output(&self, _input_kinds: &[Option<PortKind>]) -> PortKind {
        PortKind::Features
    }
    fn coord_space(&self) -> CoordSpace {
        CoordSpace::Tile
    }
    fn eval(
        &self,
        _ctx: &EvalCtx<'_>,
        inputs: &[Option<PortValue>],
    ) -> Result<PortValue, EvalError> {
        let feats = downcast_features(
            inputs[0]
                .as_ref()
                .ok_or_else(|| EvalError::MissingInput("features".into()))?,
        )?;
        let mut points = feats.points.clone();
        let mut lines = feats.lines.clone();
        let mut polygons = feats.polygons.clone();
        transform(
            &mut points,
            &mut lines,
            &mut polygons,
            self.scale,
            self.rotation_rad,
            self.pivot,
            self.translate,
        );
        Ok(features_value(feats.extent, polygons, lines, points))
    }
    fn param_hash(&self, h: &mut Xxh3) {
        h.update(b"transform");
        h.update(&self.translate.0.to_le_bytes());
        h.update(&self.translate.1.to_le_bytes());
        h.update(&self.rotation_rad.to_le_bytes());
        h.update(&self.scale.0.to_le_bytes());
        h.update(&self.scale.1.to_le_bytes());
        h.update(&self.pivot.0.to_le_bytes());
        h.update(&self.pivot.1.to_le_bytes());
    }
}

pub(super) struct TransformFactory;
impl NodeFactory for TransformFactory {
    fn op_name(&self) -> &'static str {
        "transform"
    }
    fn build(
        &self,
        fields: &serde_json::Map<String, Value>,
        ctx: &FactoryCtx<'_>,
    ) -> Result<BuiltNode, FactoryError> {
        let features = take_input_ref(fields, "features")?;
        let tx = read_xy(fields, "translate", ctx, [0.0, 0.0])?;
        let rotation_deg = read_number_or(fields, "rotation-deg", ctx, 0.0)?;
        let scale_uniform = read_number_or(fields, "scale", ctx, 1.0)?;
        let scale_xy = read_xy(
            fields,
            "scale-xy",
            ctx,
            [scale_uniform as f32, scale_uniform as f32],
        )?;
        let pivot = read_xy(fields, "pivot", ctx, [0.0, 0.0])?;
        Ok(BuiltNode {
            node: Box::new(TransformNode {
                translate: (tx[0] as f64, tx[1] as f64),
                rotation_rad: rotation_deg.to_radians(),
                scale: (scale_xy[0] as f64, scale_xy[1] as f64),
                pivot: (pivot[0] as f64, pivot[1] as f64),
            }),
            connections: vec![Connection {
                port: "features".into(),
                src: features,
            }],
        })
    }
    fn schema(&self) -> Value {
        serde_json::json!({
            "description": "Translate / rotate / scale every input vertex. Rotation happens around `pivot` (defaults to the origin) before the final translation. `scale-xy` overrides the uniform `scale` when both are present.",
            "properties": {
                "features": schema_frag::node_ref(),
                "translate": { "type": "array", "items": { "type": "number" }, "minItems": 2, "maxItems": 2 },
                "rotation-deg": { "type": "number", "default": 0.0 },
                "scale": { "type": "number", "default": 1.0 },
                "scale-xy": { "type": "array", "items": { "type": "number" }, "minItems": 2, "maxItems": 2 },
                "pivot": { "type": "array", "items": { "type": "number" }, "minItems": 2, "maxItems": 2 },
            },
            "required": ["features"],
        })
    }
}

ezu_graph::submit_node!(TransformFactory);