oxihuman_morph/
chin_recess_control.rs1#![allow(dead_code)]
4
5#[allow(dead_code)]
9#[derive(Debug, Clone)]
10pub struct ChinRecessConfig {
11 pub max_recess: f32,
12 pub max_protrusion: f32,
13}
14
15#[allow(dead_code)]
17#[derive(Debug, Clone)]
18pub struct ChinRecessState {
19 pub recess: f32,
20 pub protrusion: f32,
21 pub vertical_offset: f32,
22}
23
24#[allow(dead_code)]
25pub fn default_chin_recess_config() -> ChinRecessConfig {
26 ChinRecessConfig {
27 max_recess: 1.0,
28 max_protrusion: 1.0,
29 }
30}
31
32#[allow(dead_code)]
33pub fn new_chin_recess_state() -> ChinRecessState {
34 ChinRecessState {
35 recess: 0.0,
36 protrusion: 0.0,
37 vertical_offset: 0.0,
38 }
39}
40
41#[allow(dead_code)]
42pub fn cr_set_recess(state: &mut ChinRecessState, cfg: &ChinRecessConfig, v: f32) {
43 state.recess = v.clamp(0.0, cfg.max_recess);
44 state.protrusion = 0.0;
45}
46
47#[allow(dead_code)]
48pub fn cr_set_protrusion(state: &mut ChinRecessState, cfg: &ChinRecessConfig, v: f32) {
49 state.protrusion = v.clamp(0.0, cfg.max_protrusion);
50 state.recess = 0.0;
51}
52
53#[allow(dead_code)]
54pub fn cr_set_vertical(state: &mut ChinRecessState, v: f32) {
55 state.vertical_offset = v.clamp(-1.0, 1.0);
56}
57
58#[allow(dead_code)]
59pub fn cr_reset(state: &mut ChinRecessState) {
60 *state = new_chin_recess_state();
61}
62
63#[allow(dead_code)]
64pub fn cr_is_neutral(state: &ChinRecessState) -> bool {
65 state.recess.abs() < 1e-6 && state.protrusion.abs() < 1e-6 && state.vertical_offset.abs() < 1e-6
66}
67
68#[allow(dead_code)]
69pub fn cr_net_offset(state: &ChinRecessState) -> f32 {
70 state.protrusion - state.recess
71}
72
73#[allow(dead_code)]
74pub fn cr_blend(a: &ChinRecessState, b: &ChinRecessState, t: f32) -> ChinRecessState {
75 let t = t.clamp(0.0, 1.0);
76 ChinRecessState {
77 recess: a.recess + (b.recess - a.recess) * t,
78 protrusion: a.protrusion + (b.protrusion - a.protrusion) * t,
79 vertical_offset: a.vertical_offset + (b.vertical_offset - a.vertical_offset) * t,
80 }
81}
82
83#[allow(dead_code)]
84pub fn cr_to_weights(state: &ChinRecessState) -> Vec<(String, f32)> {
85 vec![
86 ("chin_recess".to_string(), state.recess),
87 ("chin_protrusion".to_string(), state.protrusion),
88 ("chin_vertical_offset".to_string(), state.vertical_offset),
89 ]
90}
91
92#[allow(dead_code)]
93pub fn cr_to_json(state: &ChinRecessState) -> String {
94 format!(
95 r#"{{"recess":{:.4},"protrusion":{:.4},"vertical_offset":{:.4}}}"#,
96 state.recess, state.protrusion, state.vertical_offset
97 )
98}
99
100#[cfg(test)]
101mod tests {
102 use super::*;
103
104 #[test]
105 fn default_config() {
106 let cfg = default_chin_recess_config();
107 assert!((cfg.max_recess - 1.0).abs() < 1e-6);
108 }
109
110 #[test]
111 fn new_state_neutral() {
112 let s = new_chin_recess_state();
113 assert!(cr_is_neutral(&s));
114 }
115
116 #[test]
117 fn set_recess_clears_protrusion() {
118 let cfg = default_chin_recess_config();
119 let mut s = new_chin_recess_state();
120 cr_set_protrusion(&mut s, &cfg, 0.5);
121 cr_set_recess(&mut s, &cfg, 0.3);
122 assert_eq!(s.protrusion, 0.0);
123 assert!((s.recess - 0.3).abs() < 1e-6);
124 }
125
126 #[test]
127 fn set_protrusion_clears_recess() {
128 let cfg = default_chin_recess_config();
129 let mut s = new_chin_recess_state();
130 cr_set_recess(&mut s, &cfg, 0.3);
131 cr_set_protrusion(&mut s, &cfg, 0.7);
132 assert_eq!(s.recess, 0.0);
133 assert!((s.protrusion - 0.7).abs() < 1e-6);
134 }
135
136 #[test]
137 fn set_vertical_clamps() {
138 let mut s = new_chin_recess_state();
139 cr_set_vertical(&mut s, 5.0);
140 assert!((s.vertical_offset - 1.0).abs() < 1e-6);
141 }
142
143 #[test]
144 fn net_offset_positive() {
145 let cfg = default_chin_recess_config();
146 let mut s = new_chin_recess_state();
147 cr_set_protrusion(&mut s, &cfg, 0.8);
148 assert!((cr_net_offset(&s) - 0.8).abs() < 1e-6);
149 }
150
151 #[test]
152 fn reset_clears() {
153 let cfg = default_chin_recess_config();
154 let mut s = new_chin_recess_state();
155 cr_set_recess(&mut s, &cfg, 0.5);
156 cr_reset(&mut s);
157 assert!(cr_is_neutral(&s));
158 }
159
160 #[test]
161 fn blend_midpoint() {
162 let a = new_chin_recess_state();
163 let cfg = default_chin_recess_config();
164 let mut b = new_chin_recess_state();
165 cr_set_recess(&mut b, &cfg, 1.0);
166 let mid = cr_blend(&a, &b, 0.5);
167 assert!((mid.recess - 0.5).abs() < 1e-6);
168 }
169
170 #[test]
171 fn to_weights_count() {
172 let s = new_chin_recess_state();
173 assert_eq!(cr_to_weights(&s).len(), 3);
174 }
175
176 #[test]
177 fn to_json_fields() {
178 let s = new_chin_recess_state();
179 let j = cr_to_json(&s);
180 assert!(j.contains("recess"));
181 assert!(j.contains("vertical_offset"));
182 }
183}