oxihuman-export 0.1.2

Export pipeline for OxiHuman — glTF, COLLADA, STL, and streaming formats
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

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

//! XYZRGB colored point cloud export.

#[allow(dead_code)]
pub struct XyzRgbPoint {
    pub x: f32,
    pub y: f32,
    pub z: f32,
    pub r: u8,
    pub g: u8,
    pub b: u8,
}

#[allow(dead_code)]
pub struct XyzRgbExport {
    pub points: Vec<XyzRgbPoint>,
}

#[allow(dead_code)]
pub fn new_xyzrgb_export() -> XyzRgbExport {
    XyzRgbExport { points: Vec::new() }
}

#[allow(dead_code)]
pub fn xyzrgb_add_point(exp: &mut XyzRgbExport, x: f32, y: f32, z: f32, r: u8, g: u8, b: u8) {
    exp.points.push(XyzRgbPoint { x, y, z, r, g, b });
}

#[allow(dead_code)]
pub fn xyzrgb_point_count(exp: &XyzRgbExport) -> usize {
    exp.points.len()
}

#[allow(dead_code)]
pub fn xyzrgb_to_string(exp: &XyzRgbExport) -> String {
    let mut out = String::new();
    for p in &exp.points {
        out.push_str(&format!("{} {} {} {} {} {}\n", p.x, p.y, p.z, p.r, p.g, p.b));
    }
    out
}

#[allow(dead_code)]
pub fn xyzrgb_avg_color(exp: &XyzRgbExport) -> [f32; 3] {
    let n = exp.points.len();
    if n == 0 { return [0.0; 3]; }
    let sum_r: f32 = exp.points.iter().map(|p| p.r as f32).sum();
    let sum_g: f32 = exp.points.iter().map(|p| p.g as f32).sum();
    let sum_b: f32 = exp.points.iter().map(|p| p.b as f32).sum();
    [sum_r / (n as f32 * 255.0), sum_g / (n as f32 * 255.0), sum_b / (n as f32 * 255.0)]
}

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

    #[test]
    fn test_new_empty() {
        assert_eq!(xyzrgb_point_count(&new_xyzrgb_export()), 0);
    }

    #[test]
    fn test_add_point() {
        let mut exp = new_xyzrgb_export();
        xyzrgb_add_point(&mut exp, 1.0, 2.0, 3.0, 255, 0, 0);
        assert_eq!(xyzrgb_point_count(&exp), 1);
    }

    #[test]
    fn test_to_string_contains_numbers() {
        let mut exp = new_xyzrgb_export();
        xyzrgb_add_point(&mut exp, 1.0, 2.0, 3.0, 128, 64, 32);
        let s = xyzrgb_to_string(&exp);
        assert!(s.contains("1"));
        assert!(s.contains("128"));
    }

    #[test]
    fn test_avg_color_white() {
        let mut exp = new_xyzrgb_export();
        xyzrgb_add_point(&mut exp, 0.0, 0.0, 0.0, 255, 255, 255);
        let avg = xyzrgb_avg_color(&exp);
        assert!((avg[0] - 1.0).abs() < 1e-4);
    }

    #[test]
    fn test_avg_color_empty() {
        let exp = new_xyzrgb_export();
        assert_eq!(xyzrgb_avg_color(&exp), [0.0; 3]);
    }

    #[test]
    fn test_multiple_points_count() {
        let mut exp = new_xyzrgb_export();
        for _ in 0..5 {
            xyzrgb_add_point(&mut exp, 0.0, 0.0, 0.0, 0, 0, 0);
        }
        assert_eq!(xyzrgb_point_count(&exp), 5);
    }

    #[test]
    fn test_to_string_empty() {
        let exp = new_xyzrgb_export();
        assert!(xyzrgb_to_string(&exp).is_empty());
    }

    #[test]
    fn test_avg_color_two_points() {
        let mut exp = new_xyzrgb_export();
        xyzrgb_add_point(&mut exp, 0.0, 0.0, 0.0, 0, 0, 0);
        xyzrgb_add_point(&mut exp, 0.0, 0.0, 0.0, 255, 0, 0);
        let avg = xyzrgb_avg_color(&exp);
        assert!((avg[0] - 0.5).abs() < 1e-4);
    }
}