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

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

pub struct OpenPoseKeypoint {
    pub x: f32,
    pub y: f32,
    pub confidence: f32,
}

pub struct OpenPoseBody {
    pub keypoints: Vec<OpenPoseKeypoint>,
}

pub fn new_openpose_keypoint(x: f32, y: f32, conf: f32) -> OpenPoseKeypoint {
    OpenPoseKeypoint {
        x,
        y,
        confidence: conf,
    }
}

pub fn new_openpose_body() -> OpenPoseBody {
    OpenPoseBody {
        keypoints: Vec::new(),
    }
}

pub fn body_push_keypoint(b: &mut OpenPoseBody, kp: OpenPoseKeypoint) {
    b.keypoints.push(kp);
}

pub fn body_to_json(b: &OpenPoseBody) -> String {
    let inner: Vec<_> = b
        .keypoints
        .iter()
        .map(|kp| format!(r#"[{},{},{}]"#, kp.x, kp.y, kp.confidence))
        .collect();
    format!(r#"{{"keypoints":[{}]}}"#, inner.join(","))
}

static COCO_NAMES: &[&str] = &[
    "nose",
    "neck",
    "right_shoulder",
    "right_elbow",
    "right_wrist",
    "left_shoulder",
    "left_elbow",
    "left_wrist",
    "right_hip",
    "right_knee",
    "right_ankle",
    "left_hip",
    "left_knee",
    "left_ankle",
    "right_eye",
    "left_eye",
    "right_ear",
    "left_ear",
];

pub fn keypoint_name_coco(idx: usize) -> &'static str {
    COCO_NAMES.get(idx).copied().unwrap_or("unknown")
}

pub fn body_is_valid_coco(b: &OpenPoseBody) -> bool {
    b.keypoints.len() == 18
}

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

    #[test]
    fn test_new_openpose_body_empty() {
        /* starts empty */
        let b = new_openpose_body();
        assert_eq!(b.keypoints.len(), 0);
    }

    #[test]
    fn test_body_push_keypoint() {
        /* push increases count */
        let mut b = new_openpose_body();
        body_push_keypoint(&mut b, new_openpose_keypoint(0.5, 0.3, 0.9));
        assert_eq!(b.keypoints.len(), 1);
    }

    #[test]
    fn test_keypoint_name_coco_nose() {
        /* idx 0 = nose */
        assert_eq!(keypoint_name_coco(0), "nose");
    }

    #[test]
    fn test_body_is_valid_coco_false() {
        /* not 18 = invalid */
        let b = new_openpose_body();
        assert!(!body_is_valid_coco(&b));
    }

    #[test]
    fn test_body_to_json_not_empty() {
        /* json not empty */
        let mut b = new_openpose_body();
        body_push_keypoint(&mut b, new_openpose_keypoint(0.0, 0.0, 1.0));
        let j = body_to_json(&b);
        assert!(!j.is_empty());
    }
}