Skip to main content

oxihuman_morph/
brow_spacing_control.rs

1// Copyright (C) 2026 COOLJAPAN OU (Team KitaSan)
2// SPDX-License-Identifier: Apache-2.0
3
4#![allow(dead_code)]
5
6//! Brow spacing control: adjusts the horizontal distance between eyebrows.
7
8use 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/// Compute arch angle from offset.
100#[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}