oxihuman-mesh 0.2.1

Mesh processing, topology, and geometry algorithms 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

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

//! Mesh warp: apply a displacement field to mesh vertices.

#[allow(dead_code)]
pub struct WarpField {
    pub displacements: Vec<[f32; 3]>,
}

#[allow(dead_code)]
pub fn new_warp_field(n: usize) -> WarpField {
    WarpField { displacements: vec![[0.0; 3]; n] }
}

#[allow(dead_code)]
pub fn wf_set(field: &mut WarpField, i: usize, d: [f32; 3]) {
    if i < field.displacements.len() {
        field.displacements[i] = d;
    }
}

#[allow(dead_code)]
#[allow(clippy::needless_range_loop)]
pub fn wf_apply(positions: &mut [[f32; 3]], field: &WarpField) {
    let n = positions.len().min(field.displacements.len());
    for i in 0..n {
        positions[i][0] += field.displacements[i][0];
        positions[i][1] += field.displacements[i][1];
        positions[i][2] += field.displacements[i][2];
    }
}

#[allow(dead_code)]
pub fn wf_magnitude_max(field: &WarpField) -> f32 {
    field.displacements.iter().map(|d| {
        (d[0] * d[0] + d[1] * d[1] + d[2] * d[2]).sqrt()
    }).fold(0.0f32, f32::max)
}

#[allow(dead_code)]
pub fn wf_magnitude_avg(field: &WarpField) -> f32 {
    let n = field.displacements.len();
    if n == 0 { return 0.0; }
    let sum: f32 = field.displacements.iter().map(|d| {
        (d[0] * d[0] + d[1] * d[1] + d[2] * d[2]).sqrt()
    }).sum();
    sum / n as f32
}

#[allow(dead_code)]
pub fn wf_scale(field: &mut WarpField, s: f32) {
    for d in &mut field.displacements {
        d[0] *= s; d[1] *= s; d[2] *= s;
    }
}

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

    #[test]
    fn test_zero_field_no_movement() {
        let field = new_warp_field(3);
        let mut positions = vec![[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]];
        let before = positions.clone();
        wf_apply(&mut positions, &field);
        assert_eq!(positions, before);
    }

    #[test]
    fn test_set_and_apply() {
        let mut field = new_warp_field(2);
        wf_set(&mut field, 0, [1.0, 0.0, 0.0]);
        let mut positions = vec![[0.0, 0.0, 0.0], [0.0, 0.0, 0.0]];
        wf_apply(&mut positions, &field);
        assert!((positions[0][0] - 1.0).abs() < 1e-5);
        assert!((positions[1][0]).abs() < 1e-5);
    }

    #[test]
    fn test_magnitude_max() {
        let mut field = new_warp_field(2);
        wf_set(&mut field, 0, [3.0, 4.0, 0.0]);
        wf_set(&mut field, 1, [1.0, 0.0, 0.0]);
        assert!((wf_magnitude_max(&field) - 5.0).abs() < 1e-4);
    }

    #[test]
    fn test_magnitude_avg() {
        let mut field = new_warp_field(2);
        wf_set(&mut field, 0, [3.0, 4.0, 0.0]);
        wf_set(&mut field, 1, [3.0, 4.0, 0.0]);
        assert!((wf_magnitude_avg(&field) - 5.0).abs() < 1e-4);
    }

    #[test]
    fn test_scale() {
        let mut field = new_warp_field(1);
        wf_set(&mut field, 0, [2.0, 0.0, 0.0]);
        wf_scale(&mut field, 3.0);
        assert!((field.displacements[0][0] - 6.0).abs() < 1e-5);
    }

    #[test]
    fn test_empty_field_magnitude() {
        let field = new_warp_field(0);
        assert_eq!(wf_magnitude_max(&field), 0.0);
        assert_eq!(wf_magnitude_avg(&field), 0.0);
    }

    #[test]
    fn test_set_out_of_bounds_ignored() {
        let mut field = new_warp_field(2);
        wf_set(&mut field, 99, [1.0, 1.0, 1.0]);
        assert_eq!(wf_magnitude_max(&field), 0.0);
    }

    #[test]
    fn test_scale_zero() {
        let mut field = new_warp_field(2);
        wf_set(&mut field, 0, [5.0, 5.0, 5.0]);
        wf_scale(&mut field, 0.0);
        assert_eq!(wf_magnitude_max(&field), 0.0);
    }
}