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