oxihuman-viewer 0.2.1

wgpu/WebGPU rendering adapter for OxiHuman
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
128
129
130
131

// Copyright (C) 2026 COOLJAPAN OU (Team KitaSan)
// SPDX-License-Identifier: Apache-2.0
#![allow(dead_code)]

#[derive(Debug, Clone)]
pub struct HeatmapDepthView {
    pub near: f32,
    pub far: f32,
    pub enabled: bool,
    pub invert: bool,
}

pub fn new_heatmap_depth_view() -> HeatmapDepthView {
    HeatmapDepthView {
        near: 0.1,
        far: 100.0,
        enabled: false,
        invert: false,
    }
}

pub fn hdv_set_range(v: &mut HeatmapDepthView, near: f32, far: f32) {
    v.near = near.max(0.001);
    v.far = far.max(v.near + 0.001);
}

pub fn hdv_enable(v: &mut HeatmapDepthView) {
    v.enabled = true;
}

pub fn hdv_depth_to_color(v: &HeatmapDepthView, depth: f32) -> [f32; 3] {
    let range = (v.far - v.near).max(1e-6);
    let mut t = ((depth - v.near) / range).clamp(0.0, 1.0);
    if v.invert {
        t = 1.0 - t;
    }
    /* heatmap: blue->cyan->green->yellow->red */
    let r = (t * 2.0 - 1.0).clamp(0.0, 1.0);
    let g = (1.0 - (t * 2.0 - 1.0).abs()).clamp(0.0, 1.0);
    let b = (1.0 - t * 2.0).clamp(0.0, 1.0);
    [r, g, b]
}

pub fn hdv_is_enabled(v: &HeatmapDepthView) -> bool {
    v.enabled
}

pub fn hdv_to_json(v: &HeatmapDepthView) -> String {
    format!(
        r#"{{"near":{:.4},"far":{:.4},"enabled":{},"invert":{}}}"#,
        v.near, v.far, v.enabled, v.invert
    )
}

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

    #[test]
    fn test_new_defaults() {
        /* default near/far sensible */
        let v = new_heatmap_depth_view();
        assert!(v.near < v.far);
        assert!(!v.enabled);
    }

    #[test]
    fn test_set_range() {
        /* range stored correctly */
        let mut v = new_heatmap_depth_view();
        hdv_set_range(&mut v, 1.0, 50.0);
        assert!((v.near - 1.0).abs() < 1e-6);
        assert!((v.far - 50.0).abs() < 1e-6);
    }

    #[test]
    fn test_enable() {
        /* enable sets flag */
        let mut v = new_heatmap_depth_view();
        hdv_enable(&mut v);
        assert!(hdv_is_enabled(&v));
    }

    #[test]
    fn test_depth_to_color_near() {
        /* depth at near => cool color */
        let v = new_heatmap_depth_view();
        let c = hdv_depth_to_color(&v, v.near);
        /* near end should have blue component */
        assert!(c[2] >= c[0]);
    }

    #[test]
    fn test_depth_to_color_far() {
        /* depth at far => warm color */
        let v = new_heatmap_depth_view();
        let c = hdv_depth_to_color(&v, v.far);
        assert!(c[0] >= c[2]);
    }

    #[test]
    fn test_depth_to_color_out_of_range() {
        /* depth beyond far clamped */
        let v = new_heatmap_depth_view();
        let c = hdv_depth_to_color(&v, v.far * 2.0);
        assert!(c[0] <= 1.0);
        assert!(c[0] >= 0.0);
    }

    #[test]
    fn test_to_json_has_near() {
        /* JSON contains near field */
        let v = new_heatmap_depth_view();
        assert!(hdv_to_json(&v).contains("near"));
    }

    #[test]
    fn test_to_json_has_invert() {
        /* JSON contains invert field */
        let v = new_heatmap_depth_view();
        assert!(hdv_to_json(&v).contains("invert"));
    }

    #[test]
    fn test_clone() {
        /* clone independent */
        let v = new_heatmap_depth_view();
        let v2 = v.clone();
        assert_eq!(v.enabled, v2.enabled);
    }
}