oxihuman_morph/
brow_spacing_control.rs1#![allow(dead_code)]
5
6use std::f32::consts::FRAC_PI_4;
9
10#[allow(dead_code)]
11#[derive(Debug, Clone)]
12pub struct BrowSpacingConfig {
13 pub min_spacing: f32,
14 pub max_spacing: f32,
15}
16
17#[allow(dead_code)]
18#[derive(Debug, Clone)]
19pub struct BrowSpacingState {
20 pub spacing: f32,
21 pub arch_offset: f32,
22 pub symmetry: f32,
23}
24
25#[allow(dead_code)]
26pub fn default_brow_spacing_config() -> BrowSpacingConfig {
27 BrowSpacingConfig {
28 min_spacing: 0.0,
29 max_spacing: 1.0,
30 }
31}
32
33#[allow(dead_code)]
34pub fn new_brow_spacing_state() -> BrowSpacingState {
35 BrowSpacingState {
36 spacing: 0.5,
37 arch_offset: 0.0,
38 symmetry: 1.0,
39 }
40}
41
42#[allow(dead_code)]
43pub fn bs_set_spacing(state: &mut BrowSpacingState, cfg: &BrowSpacingConfig, v: f32) {
44 state.spacing = v.clamp(cfg.min_spacing, cfg.max_spacing);
45}
46
47#[allow(dead_code)]
48pub fn bs_set_arch_offset(state: &mut BrowSpacingState, v: f32) {
49 state.arch_offset = v.clamp(-1.0, 1.0);
50}
51
52#[allow(dead_code)]
53pub fn bs_set_symmetry(state: &mut BrowSpacingState, v: f32) {
54 state.symmetry = v.clamp(0.0, 1.0);
55}
56
57#[allow(dead_code)]
58pub fn bs_reset(state: &mut BrowSpacingState) {
59 *state = new_brow_spacing_state();
60}
61
62#[allow(dead_code)]
63pub fn bs_effective_left(state: &BrowSpacingState) -> f32 {
64 state.spacing * state.symmetry
65}
66
67#[allow(dead_code)]
68pub fn bs_effective_right(state: &BrowSpacingState) -> f32 {
69 state.spacing * (2.0 - state.symmetry)
70}
71
72#[allow(dead_code)]
73pub fn bs_to_weights(state: &BrowSpacingState) -> Vec<(String, f32)> {
74 vec![
75 ("brow_spacing".to_string(), state.spacing),
76 ("brow_arch_offset".to_string(), state.arch_offset),
77 ("brow_symmetry".to_string(), state.symmetry),
78 ]
79}
80
81#[allow(dead_code)]
82pub fn bs_to_json(state: &BrowSpacingState) -> String {
83 format!(
84 r#"{{"spacing":{:.4},"arch_offset":{:.4},"symmetry":{:.4}}}"#,
85 state.spacing, state.arch_offset, state.symmetry
86 )
87}
88
89#[allow(dead_code)]
90pub fn bs_blend(a: &BrowSpacingState, b: &BrowSpacingState, t: f32) -> BrowSpacingState {
91 let t = t.clamp(0.0, 1.0);
92 BrowSpacingState {
93 spacing: a.spacing + (b.spacing - a.spacing) * t,
94 arch_offset: a.arch_offset + (b.arch_offset - a.arch_offset) * t,
95 symmetry: a.symmetry + (b.symmetry - a.symmetry) * t,
96 }
97}
98
99#[allow(dead_code)]
101pub fn bs_arch_angle(state: &BrowSpacingState) -> f32 {
102 state.arch_offset * FRAC_PI_4
103}
104
105#[cfg(test)]
106mod tests {
107 use super::*;
108
109 #[test]
110 fn test_default_config() {
111 let cfg = default_brow_spacing_config();
112 assert!(cfg.min_spacing.abs() < 1e-6);
113 assert!((cfg.max_spacing - 1.0).abs() < 1e-6);
114 }
115
116 #[test]
117 fn test_new_state() {
118 let s = new_brow_spacing_state();
119 assert!((s.spacing - 0.5).abs() < 1e-6);
120 assert!((s.symmetry - 1.0).abs() < 1e-6);
121 }
122
123 #[test]
124 fn test_set_spacing_clamps() {
125 let cfg = default_brow_spacing_config();
126 let mut s = new_brow_spacing_state();
127 bs_set_spacing(&mut s, &cfg, 5.0);
128 assert!((s.spacing - 1.0).abs() < 1e-6);
129 }
130
131 #[test]
132 fn test_set_arch_offset() {
133 let mut s = new_brow_spacing_state();
134 bs_set_arch_offset(&mut s, 0.7);
135 assert!((s.arch_offset - 0.7).abs() < 1e-6);
136 }
137
138 #[test]
139 fn test_set_symmetry() {
140 let mut s = new_brow_spacing_state();
141 bs_set_symmetry(&mut s, 0.8);
142 assert!((s.symmetry - 0.8).abs() < 1e-6);
143 }
144
145 #[test]
146 fn test_reset() {
147 let cfg = default_brow_spacing_config();
148 let mut s = new_brow_spacing_state();
149 bs_set_spacing(&mut s, &cfg, 0.9);
150 bs_reset(&mut s);
151 assert!((s.spacing - 0.5).abs() < 1e-6);
152 }
153
154 #[test]
155 fn test_effective_sides() {
156 let s = new_brow_spacing_state();
157 let l = bs_effective_left(&s);
158 let r = bs_effective_right(&s);
159 assert!((l - r).abs() < 1e-6);
160 }
161
162 #[test]
163 fn test_to_weights() {
164 let s = new_brow_spacing_state();
165 assert_eq!(bs_to_weights(&s).len(), 3);
166 }
167
168 #[test]
169 fn test_blend() {
170 let a = new_brow_spacing_state();
171 let mut b = new_brow_spacing_state();
172 b.spacing = 1.0;
173 let mid = bs_blend(&a, &b, 0.5);
174 assert!((mid.spacing - 0.75).abs() < 1e-6);
175 }
176
177 #[test]
178 fn test_arch_angle() {
179 let mut s = new_brow_spacing_state();
180 s.arch_offset = 1.0;
181 let angle = bs_arch_angle(&s);
182 assert!((angle - FRAC_PI_4).abs() < 1e-6);
183 }
184}