oxihuman_morph/
ear_tragus_control.rs1#![allow(dead_code)]
5
6#[allow(dead_code)]
10#[derive(Debug, Clone)]
11pub struct EarTragusConfig {
12 pub min_size: f32,
13 pub max_size: f32,
14}
15
16#[allow(dead_code)]
18#[derive(Debug, Clone)]
19pub struct EarTragusState {
20 pub left_size: f32,
21 pub right_size: f32,
22 pub protrusion: f32,
23}
24
25#[allow(dead_code)]
26pub fn default_ear_tragus_config() -> EarTragusConfig {
27 EarTragusConfig {
28 min_size: 0.0,
29 max_size: 1.0,
30 }
31}
32
33#[allow(dead_code)]
34pub fn new_ear_tragus_state() -> EarTragusState {
35 EarTragusState {
36 left_size: 0.5,
37 right_size: 0.5,
38 protrusion: 0.3,
39 }
40}
41
42#[allow(dead_code)]
43pub fn et_set_left(state: &mut EarTragusState, cfg: &EarTragusConfig, v: f32) {
44 state.left_size = v.clamp(cfg.min_size, cfg.max_size);
45}
46
47#[allow(dead_code)]
48pub fn et_set_right(state: &mut EarTragusState, cfg: &EarTragusConfig, v: f32) {
49 state.right_size = v.clamp(cfg.min_size, cfg.max_size);
50}
51
52#[allow(dead_code)]
53pub fn et_set_both(state: &mut EarTragusState, cfg: &EarTragusConfig, v: f32) {
54 let clamped = v.clamp(cfg.min_size, cfg.max_size);
55 state.left_size = clamped;
56 state.right_size = clamped;
57}
58
59#[allow(dead_code)]
60pub fn et_set_protrusion(state: &mut EarTragusState, v: f32) {
61 state.protrusion = v.clamp(0.0, 1.0);
62}
63
64#[allow(dead_code)]
65pub fn et_reset(state: &mut EarTragusState) {
66 *state = new_ear_tragus_state();
67}
68
69#[allow(dead_code)]
70pub fn et_to_weights(state: &EarTragusState) -> Vec<(String, f32)> {
71 vec![
72 ("ear_tragus_left".to_string(), state.left_size),
73 ("ear_tragus_right".to_string(), state.right_size),
74 ("ear_tragus_protrusion".to_string(), state.protrusion),
75 ]
76}
77
78#[allow(dead_code)]
79pub fn et_to_json(state: &EarTragusState) -> String {
80 format!(
81 r#"{{"left_size":{:.4},"right_size":{:.4},"protrusion":{:.4}}}"#,
82 state.left_size, state.right_size, state.protrusion
83 )
84}
85
86#[allow(dead_code)]
87pub fn et_blend(a: &EarTragusState, b: &EarTragusState, t: f32) -> EarTragusState {
88 let t = t.clamp(0.0, 1.0);
89 EarTragusState {
90 left_size: a.left_size + (b.left_size - a.left_size) * t,
91 right_size: a.right_size + (b.right_size - a.right_size) * t,
92 protrusion: a.protrusion + (b.protrusion - a.protrusion) * t,
93 }
94}
95
96#[cfg(test)]
97mod tests {
98 use super::*;
99
100 #[test]
101 fn test_default_config() {
102 let cfg = default_ear_tragus_config();
103 assert!(cfg.min_size.abs() < 1e-6);
104 assert!((cfg.max_size - 1.0).abs() < 1e-6);
105 }
106
107 #[test]
108 fn test_new_state() {
109 let s = new_ear_tragus_state();
110 assert!((s.left_size - 0.5).abs() < 1e-6);
111 assert!((s.protrusion - 0.3).abs() < 1e-6);
112 }
113
114 #[test]
115 fn test_set_left_clamps() {
116 let cfg = default_ear_tragus_config();
117 let mut s = new_ear_tragus_state();
118 et_set_left(&mut s, &cfg, 5.0);
119 assert!((s.left_size - 1.0).abs() < 1e-6);
120 }
121
122 #[test]
123 fn test_set_both() {
124 let cfg = default_ear_tragus_config();
125 let mut s = new_ear_tragus_state();
126 et_set_both(&mut s, &cfg, 0.7);
127 assert!((s.left_size - 0.7).abs() < 1e-6);
128 assert!((s.right_size - 0.7).abs() < 1e-6);
129 }
130
131 #[test]
132 fn test_set_protrusion() {
133 let mut s = new_ear_tragus_state();
134 et_set_protrusion(&mut s, 0.8);
135 assert!((s.protrusion - 0.8).abs() < 1e-6);
136 }
137
138 #[test]
139 fn test_reset() {
140 let cfg = default_ear_tragus_config();
141 let mut s = new_ear_tragus_state();
142 et_set_left(&mut s, &cfg, 0.9);
143 et_reset(&mut s);
144 assert!((s.left_size - 0.5).abs() < 1e-6);
145 }
146
147 #[test]
148 fn test_to_weights() {
149 let s = new_ear_tragus_state();
150 assert_eq!(et_to_weights(&s).len(), 3);
151 }
152
153 #[test]
154 fn test_to_json() {
155 let s = new_ear_tragus_state();
156 let j = et_to_json(&s);
157 assert!(j.contains("left_size"));
158 }
159
160 #[test]
161 fn test_blend() {
162 let a = new_ear_tragus_state();
163 let mut b = new_ear_tragus_state();
164 b.left_size = 1.0;
165 let mid = et_blend(&a, &b, 0.5);
166 assert!((mid.left_size - 0.75).abs() < 1e-6);
167 }
168
169 #[test]
170 fn test_set_right() {
171 let cfg = default_ear_tragus_config();
172 let mut s = new_ear_tragus_state();
173 et_set_right(&mut s, &cfg, 0.3);
174 assert!((s.right_size - 0.3).abs() < 1e-6);
175 }
176}