oxihuman_morph/
jaw_line_control.rs

1// Copyright (C) 2026 COOLJAPAN OU (Team KitaSan)
2// SPDX-License-Identifier: Apache-2.0
3
4#![allow(dead_code)]
5
6//! Jaw line control: adjusts definition, angle and width of the jawline.
7
8use std::f32::consts::FRAC_PI_4;
9
10#[allow(dead_code)]
11#[derive(Debug, Clone)]
12pub struct JawLineConfig {
13    pub min_val: f32,
14    pub max_val: f32,
15}
16
17#[allow(dead_code)]
18#[derive(Debug, Clone)]
19pub struct JawLineState {
20    pub definition: f32,
21    pub angle: f32,
22    pub width: f32,
23}
24
25#[allow(dead_code)]
26pub fn default_jaw_line_config() -> JawLineConfig {
27    JawLineConfig {
28        min_val: 0.0,
29        max_val: 1.0,
30    }
31}
32
33#[allow(dead_code)]
34pub fn new_jaw_line_state() -> JawLineState {
35    JawLineState {
36        definition: 0.5,
37        angle: 0.5,
38        width: 0.5,
39    }
40}
41
42#[allow(dead_code)]
43pub fn jl_set_definition(state: &mut JawLineState, cfg: &JawLineConfig, v: f32) {
44    state.definition = v.clamp(cfg.min_val, cfg.max_val);
45}
46
47#[allow(dead_code)]
48pub fn jl_set_angle(state: &mut JawLineState, cfg: &JawLineConfig, v: f32) {
49    state.angle = v.clamp(cfg.min_val, cfg.max_val);
50}
51
52#[allow(dead_code)]
53pub fn jl_set_width(state: &mut JawLineState, cfg: &JawLineConfig, v: f32) {
54    state.width = v.clamp(cfg.min_val, cfg.max_val);
55}
56
57#[allow(dead_code)]
58pub fn jl_reset(state: &mut JawLineState) {
59    *state = new_jaw_line_state();
60}
61
62#[allow(dead_code)]
63pub fn jl_gonial_angle(state: &JawLineState) -> f32 {
64    state.angle * FRAC_PI_4 + FRAC_PI_4
65}
66
67#[allow(dead_code)]
68pub fn jl_to_weights(state: &JawLineState) -> Vec<(String, f32)> {
69    vec![
70        ("jaw_definition".to_string(), state.definition),
71        ("jaw_angle".to_string(), state.angle),
72        ("jaw_width".to_string(), state.width),
73    ]
74}
75
76#[allow(dead_code)]
77pub fn jl_to_json(state: &JawLineState) -> String {
78    format!(
79        r#"{{"definition":{:.4},"angle":{:.4},"width":{:.4}}}"#,
80        state.definition, state.angle, state.width
81    )
82}
83
84#[allow(dead_code)]
85pub fn jl_blend(a: &JawLineState, b: &JawLineState, t: f32) -> JawLineState {
86    let t = t.clamp(0.0, 1.0);
87    JawLineState {
88        definition: a.definition + (b.definition - a.definition) * t,
89        angle: a.angle + (b.angle - a.angle) * t,
90        width: a.width + (b.width - a.width) * t,
91    }
92}
93
94#[allow(dead_code)]
95pub fn jl_sharpness(state: &JawLineState) -> f32 {
96    state.definition * (1.0 - state.width * 0.3)
97}
98
99#[cfg(test)]
100mod tests {
101    use super::*;
102
103    #[test]
104    fn test_default_config() {
105        let cfg = default_jaw_line_config();
106        assert!(cfg.min_val.abs() < 1e-6);
107    }
108
109    #[test]
110    fn test_new_state() {
111        let s = new_jaw_line_state();
112        assert!((s.definition - 0.5).abs() < 1e-6);
113    }
114
115    #[test]
116    fn test_set_definition_clamps() {
117        let cfg = default_jaw_line_config();
118        let mut s = new_jaw_line_state();
119        jl_set_definition(&mut s, &cfg, 5.0);
120        assert!((s.definition - 1.0).abs() < 1e-6);
121    }
122
123    #[test]
124    fn test_set_angle() {
125        let cfg = default_jaw_line_config();
126        let mut s = new_jaw_line_state();
127        jl_set_angle(&mut s, &cfg, 0.8);
128        assert!((s.angle - 0.8).abs() < 1e-6);
129    }
130
131    #[test]
132    fn test_set_width() {
133        let cfg = default_jaw_line_config();
134        let mut s = new_jaw_line_state();
135        jl_set_width(&mut s, &cfg, 0.7);
136        assert!((s.width - 0.7).abs() < 1e-6);
137    }
138
139    #[test]
140    fn test_reset() {
141        let cfg = default_jaw_line_config();
142        let mut s = new_jaw_line_state();
143        jl_set_definition(&mut s, &cfg, 0.9);
144        jl_reset(&mut s);
145        assert!((s.definition - 0.5).abs() < 1e-6);
146    }
147
148    #[test]
149    fn test_gonial_angle() {
150        let s = new_jaw_line_state();
151        let a = jl_gonial_angle(&s);
152        assert!(a > FRAC_PI_4);
153    }
154
155    #[test]
156    fn test_blend() {
157        let a = new_jaw_line_state();
158        let mut b = new_jaw_line_state();
159        b.definition = 1.0;
160        let mid = jl_blend(&a, &b, 0.5);
161        assert!((mid.definition - 0.75).abs() < 1e-6);
162    }
163
164    #[test]
165    fn test_to_weights() {
166        let s = new_jaw_line_state();
167        assert_eq!(jl_to_weights(&s).len(), 3);
168    }
169
170    #[test]
171    fn test_sharpness() {
172        let s = new_jaw_line_state();
173        assert!(jl_sharpness(&s) > 0.0);
174    }
175}
oxihuman_morph/jaw_line_control.rs

oxihuman_morph/
jaw_line_control.rs
