oxihuman_morph/
eye_inner_corner.rs1#![allow(dead_code)]
4
5#[allow(dead_code)]
9#[derive(Debug, Clone)]
10pub struct EyeInnerCornerConfig {
11 pub max_depth: f32,
12}
13
14#[allow(dead_code)]
16#[derive(Debug, Clone)]
17pub struct EyeInnerCornerState {
18 pub left_depth: f32,
19 pub right_depth: f32,
20 pub left_tilt: f32,
21 pub right_tilt: f32,
22}
23
24#[allow(dead_code)]
25pub fn default_eye_inner_corner_config() -> EyeInnerCornerConfig {
26 EyeInnerCornerConfig { max_depth: 1.0 }
27}
28
29#[allow(dead_code)]
30pub fn new_eye_inner_corner_state() -> EyeInnerCornerState {
31 EyeInnerCornerState {
32 left_depth: 0.0,
33 right_depth: 0.0,
34 left_tilt: 0.0,
35 right_tilt: 0.0,
36 }
37}
38
39#[allow(dead_code)]
40pub fn eic_set_depth_left(state: &mut EyeInnerCornerState, cfg: &EyeInnerCornerConfig, v: f32) {
41 state.left_depth = v.clamp(0.0, cfg.max_depth);
42}
43
44#[allow(dead_code)]
45pub fn eic_set_depth_right(state: &mut EyeInnerCornerState, cfg: &EyeInnerCornerConfig, v: f32) {
46 state.right_depth = v.clamp(0.0, cfg.max_depth);
47}
48
49#[allow(dead_code)]
50pub fn eic_set_both_depth(state: &mut EyeInnerCornerState, cfg: &EyeInnerCornerConfig, v: f32) {
51 let clamped = v.clamp(0.0, cfg.max_depth);
52 state.left_depth = clamped;
53 state.right_depth = clamped;
54}
55
56#[allow(dead_code)]
57pub fn eic_set_tilt(state: &mut EyeInnerCornerState, left: f32, right: f32) {
58 state.left_tilt = left.clamp(-1.0, 1.0);
59 state.right_tilt = right.clamp(-1.0, 1.0);
60}
61
62#[allow(dead_code)]
63pub fn eic_reset(state: &mut EyeInnerCornerState) {
64 *state = new_eye_inner_corner_state();
65}
66
67#[allow(dead_code)]
68pub fn eic_is_neutral(state: &EyeInnerCornerState) -> bool {
69 let vals = [
70 state.left_depth,
71 state.right_depth,
72 state.left_tilt,
73 state.right_tilt,
74 ];
75 !vals.is_empty() && vals.iter().all(|v| v.abs() < 1e-6)
76}
77
78#[allow(dead_code)]
79pub fn eic_average_depth(state: &EyeInnerCornerState) -> f32 {
80 (state.left_depth + state.right_depth) * 0.5
81}
82
83#[allow(dead_code)]
84pub fn eic_symmetry(state: &EyeInnerCornerState) -> f32 {
85 (state.left_depth - state.right_depth).abs()
86}
87
88#[allow(dead_code)]
89pub fn eic_blend(a: &EyeInnerCornerState, b: &EyeInnerCornerState, t: f32) -> EyeInnerCornerState {
90 let t = t.clamp(0.0, 1.0);
91 EyeInnerCornerState {
92 left_depth: a.left_depth + (b.left_depth - a.left_depth) * t,
93 right_depth: a.right_depth + (b.right_depth - a.right_depth) * t,
94 left_tilt: a.left_tilt + (b.left_tilt - a.left_tilt) * t,
95 right_tilt: a.right_tilt + (b.right_tilt - a.right_tilt) * t,
96 }
97}
98
99#[allow(dead_code)]
100pub fn eic_to_weights(state: &EyeInnerCornerState) -> Vec<(String, f32)> {
101 vec![
102 ("eye_inner_depth_l".to_string(), state.left_depth),
103 ("eye_inner_depth_r".to_string(), state.right_depth),
104 ("eye_inner_tilt_l".to_string(), state.left_tilt),
105 ("eye_inner_tilt_r".to_string(), state.right_tilt),
106 ]
107}
108
109#[allow(dead_code)]
110pub fn eic_to_json(state: &EyeInnerCornerState) -> String {
111 format!(
112 r#"{{"left_depth":{:.4},"right_depth":{:.4},"left_tilt":{:.4},"right_tilt":{:.4}}}"#,
113 state.left_depth, state.right_depth, state.left_tilt, state.right_tilt
114 )
115}
116
117#[cfg(test)]
118mod tests {
119 use super::*;
120
121 #[test]
122 fn default_config() {
123 let cfg = default_eye_inner_corner_config();
124 assert!((cfg.max_depth - 1.0).abs() < 1e-6);
125 }
126
127 #[test]
128 fn new_state_neutral() {
129 let s = new_eye_inner_corner_state();
130 assert!(eic_is_neutral(&s));
131 }
132
133 #[test]
134 fn set_depth_left_clamps() {
135 let cfg = default_eye_inner_corner_config();
136 let mut s = new_eye_inner_corner_state();
137 eic_set_depth_left(&mut s, &cfg, 5.0);
138 assert!((s.left_depth - 1.0).abs() < 1e-6);
139 }
140
141 #[test]
142 fn set_both_depth_equal() {
143 let cfg = default_eye_inner_corner_config();
144 let mut s = new_eye_inner_corner_state();
145 eic_set_both_depth(&mut s, &cfg, 0.7);
146 assert!((s.left_depth - 0.7).abs() < 1e-6);
147 assert!((s.right_depth - 0.7).abs() < 1e-6);
148 }
149
150 #[test]
151 fn set_tilt_clamped() {
152 let mut s = new_eye_inner_corner_state();
153 eic_set_tilt(&mut s, 2.0, -3.0);
154 assert!((s.left_tilt - 1.0).abs() < 1e-6);
155 assert!((s.right_tilt + 1.0).abs() < 1e-6);
156 }
157
158 #[test]
159 fn average_depth() {
160 let cfg = default_eye_inner_corner_config();
161 let mut s = new_eye_inner_corner_state();
162 eic_set_depth_left(&mut s, &cfg, 0.4);
163 eic_set_depth_right(&mut s, &cfg, 0.6);
164 assert!((eic_average_depth(&s) - 0.5).abs() < 1e-6);
165 }
166
167 #[test]
168 fn symmetry_zero_when_equal() {
169 let cfg = default_eye_inner_corner_config();
170 let mut s = new_eye_inner_corner_state();
171 eic_set_both_depth(&mut s, &cfg, 0.5);
172 assert!(eic_symmetry(&s) < 1e-6);
173 }
174
175 #[test]
176 fn reset_clears() {
177 let cfg = default_eye_inner_corner_config();
178 let mut s = new_eye_inner_corner_state();
179 eic_set_both_depth(&mut s, &cfg, 0.8);
180 eic_reset(&mut s);
181 assert!(eic_is_neutral(&s));
182 }
183
184 #[test]
185 fn blend_midpoint() {
186 let a = new_eye_inner_corner_state();
187 let cfg = default_eye_inner_corner_config();
188 let mut b = new_eye_inner_corner_state();
189 eic_set_both_depth(&mut b, &cfg, 1.0);
190 let mid = eic_blend(&a, &b, 0.5);
191 assert!((mid.left_depth - 0.5).abs() < 1e-6);
192 }
193
194 #[test]
195 fn to_weights_count() {
196 let s = new_eye_inner_corner_state();
197 assert_eq!(eic_to_weights(&s).len(), 4);
198 }
199
200 #[test]
201 fn to_json_fields() {
202 let s = new_eye_inner_corner_state();
203 let j = eic_to_json(&s);
204 assert!(j.contains("left_depth"));
205 assert!(j.contains("right_tilt"));
206 }
207}