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

//! `brightness-contrast` — linear brightness shift and contrast slope
//! around mid-gray over `Raster|Sprite` (pass-through). Operates in
//! non-premultiplied sRGB; alpha is preserved.

use std::sync::Arc;

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

use crate::nodes::common::{
    raster_or_sprite_output, read_number_or, unwrap_raster_or_sprite, wrap_raster_like,
    ACCEPTS_RASTER_OR_SPRITE,
};

struct BrightnessContrastNode {
    brightness: f32,
    contrast: f32,
}

impl Node for BrightnessContrastNode {
    fn op_name(&self) -> &'static str {
        "brightness-contrast"
    }
    fn inputs(&self) -> &[PortSpec] {
        static SPECS: &[PortSpec] = &[PortSpec {
            name: "input",
            accepts: ACCEPTS_RASTER_OR_SPRITE,
            optional: false,
        }];
        SPECS
    }
    fn output(&self, input_kinds: &[Option<PortKind>]) -> PortKind {
        raster_or_sprite_output(input_kinds)
    }
    fn eval(
        &self,
        _ctx: &EvalCtx<'_>,
        inputs: &[Option<PortValue>],
    ) -> Result<PortValue, EvalError> {
        let input = inputs[0]
            .as_ref()
            .ok_or_else(|| EvalError::MissingInput("input".into()))?;
        let (src, kind) = unwrap_raster_or_sprite(input, "input")?;
        let slope = 1.0 + self.contrast;
        let offset = self.brightness;
        let mut out = RasterBuf::new(src.width, src.height);
        for i in (0..src.pixels.len()).step_by(4) {
            let a = src.pixels[i + 3] as f32 / 255.0;
            if a <= 0.0 {
                continue; // transparent stays transparent (zeroed)
            }
            let r = (src.pixels[i] as f32 / 255.0) / a;
            let g = (src.pixels[i + 1] as f32 / 255.0) / a;
            let b = (src.pixels[i + 2] as f32 / 255.0) / a;
            let nr = ((r - 0.5) * slope + 0.5 + offset).clamp(0.0, 1.0);
            let ng = ((g - 0.5) * slope + 0.5 + offset).clamp(0.0, 1.0);
            let nb = ((b - 0.5) * slope + 0.5 + offset).clamp(0.0, 1.0);
            out.pixels[i] = (nr * a * 255.0).round() as u8;
            out.pixels[i + 1] = (ng * a * 255.0).round() as u8;
            out.pixels[i + 2] = (nb * a * 255.0).round() as u8;
            out.pixels[i + 3] = src.pixels[i + 3];
        }
        Ok(wrap_raster_like(Arc::new(out), kind))
    }
    fn param_hash(&self, h: &mut Xxh3) {
        h.update(b"brightness-contrast");
        h.update(&self.brightness.to_le_bytes());
        h.update(&self.contrast.to_le_bytes());
    }
}

pub(super) struct BrightnessContrastFactory;
impl NodeFactory for BrightnessContrastFactory {
    fn op_name(&self) -> &'static str {
        "brightness-contrast"
    }
    fn build(
        &self,
        fields: &serde_json::Map<String, Value>,
        ctx: &FactoryCtx<'_>,
    ) -> Result<BuiltNode, FactoryError> {
        let input = take_input_ref(fields, "input")?;
        let brightness = read_number_or(fields, "brightness", ctx, 0.0)? as f32;
        let contrast = read_number_or(fields, "contrast", ctx, 0.0)? as f32;
        Ok(BuiltNode {
            node: Box::new(BrightnessContrastNode {
                brightness,
                contrast,
            }),
            connections: vec![Connection {
                port: "input".into(),
                src: input,
            }],
        })
    }
    fn schema(&self) -> Value {
        serde_json::json!({
            "description": "Linear brightness shift and contrast slope around mid-gray. Both in [-1, 1], 0 = no change.",
            "properties": {
                "input": schema_frag::node_ref(),
                "brightness": { "type": "number", "minimum": -1.0, "maximum": 1.0, "default": 0.0 },
                "contrast": { "type": "number", "minimum": -1.0, "maximum": 1.0, "default": 0.0 },
            },
            "required": ["input"],
        })
    }
}

ezu_graph::submit_node!(BrightnessContrastFactory);