oxihuman_morph/
jaw_clench_control.rs

1// Copyright (C) 2026 COOLJAPAN OU (Team KitaSan) / SPDX-License-Identifier: Apache-2.0
2#![allow(dead_code)]
3
4//! Jaw clench (masseter bulge) control — bilateral and unilateral clenching.
5
6use std::f32::consts::PI;
7
8/// Config.
9#[allow(dead_code)]
10#[derive(Debug, Clone, PartialEq)]
11pub struct JawClenchConfig {
12    pub max_bulge_m: f32,
13}
14
15impl Default for JawClenchConfig {
16    fn default() -> Self {
17        Self { max_bulge_m: 0.007 }
18    }
19}
20
21/// State.
22#[allow(dead_code)]
23#[derive(Debug, Clone, Default)]
24pub struct JawClenchState {
25    /// Left masseter clench, 0..=1.
26    pub left: f32,
27    /// Right masseter clench, 0..=1.
28    pub right: f32,
29}
30
31#[allow(dead_code)]
32pub fn new_jaw_clench_state() -> JawClenchState {
33    JawClenchState::default()
34}
35
36#[allow(dead_code)]
37pub fn default_jaw_clench_config() -> JawClenchConfig {
38    JawClenchConfig::default()
39}
40
41/// Clench side enum.
42#[allow(dead_code)]
43#[derive(Debug, Clone, Copy, PartialEq, Eq)]
44pub enum ClenchSide {
45    Left,
46    Right,
47}
48
49#[allow(dead_code)]
50pub fn jcl_set(state: &mut JawClenchState, side: ClenchSide, v: f32) {
51    let v = v.clamp(0.0, 1.0);
52    match side {
53        ClenchSide::Left => state.left = v,
54        ClenchSide::Right => state.right = v,
55    }
56}
57
58#[allow(dead_code)]
59pub fn jcl_set_both(state: &mut JawClenchState, v: f32) {
60    let v = v.clamp(0.0, 1.0);
61    state.left = v;
62    state.right = v;
63}
64
65#[allow(dead_code)]
66pub fn jcl_reset(state: &mut JawClenchState) {
67    *state = JawClenchState::default();
68}
69
70#[allow(dead_code)]
71pub fn jcl_is_neutral(state: &JawClenchState) -> bool {
72    state.left < 1e-4 && state.right < 1e-4
73}
74
75#[allow(dead_code)]
76pub fn jcl_asymmetry(state: &JawClenchState) -> f32 {
77    (state.left - state.right).abs()
78}
79
80/// Bite force approximation (0..=1, normalised).
81#[allow(dead_code)]
82pub fn jcl_bite_force(state: &JawClenchState) -> f32 {
83    ((state.left + state.right) * 0.5).min(1.0)
84}
85
86/// Temporal muscle pressure angle in radians.
87#[allow(dead_code)]
88pub fn jcl_temporal_angle_rad(state: &JawClenchState) -> f32 {
89    jcl_bite_force(state) * (PI / 8.0)
90}
91
92#[allow(dead_code)]
93pub fn jcl_to_weights(state: &JawClenchState, cfg: &JawClenchConfig) -> [f32; 2] {
94    [state.left * cfg.max_bulge_m, state.right * cfg.max_bulge_m]
95}
96
97#[allow(dead_code)]
98pub fn jcl_blend(a: &JawClenchState, b: &JawClenchState, t: f32) -> JawClenchState {
99    let t = t.clamp(0.0, 1.0);
100    let inv = 1.0 - t;
101    JawClenchState {
102        left: a.left * inv + b.left * t,
103        right: a.right * inv + b.right * t,
104    }
105}
106
107#[allow(dead_code)]
108pub fn jcl_to_json(state: &JawClenchState) -> String {
109    format!(
110        "{{\"left\":{:.4},\"right\":{:.4}}}",
111        state.left, state.right
112    )
113}
114
115#[cfg(test)]
116mod tests {
117    use super::*;
118
119    #[test]
120    fn default_neutral() {
121        assert!(jcl_is_neutral(&new_jaw_clench_state()));
122    }
123
124    #[test]
125    fn set_clamps_high() {
126        let mut s = new_jaw_clench_state();
127        jcl_set(&mut s, ClenchSide::Left, 5.0);
128        assert!((s.left - 1.0).abs() < 1e-6);
129    }
130
131    #[test]
132    fn set_clamps_low() {
133        let mut s = new_jaw_clench_state();
134        jcl_set(&mut s, ClenchSide::Right, -1.0);
135        assert!(s.right < 1e-6);
136    }
137
138    #[test]
139    fn set_both_equal() {
140        let mut s = new_jaw_clench_state();
141        jcl_set_both(&mut s, 0.6);
142        assert!((s.left - s.right).abs() < 1e-6);
143    }
144
145    #[test]
146    fn reset_clears() {
147        let mut s = new_jaw_clench_state();
148        jcl_set_both(&mut s, 0.9);
149        jcl_reset(&mut s);
150        assert!(jcl_is_neutral(&s));
151    }
152
153    #[test]
154    fn bite_force_average() {
155        let mut s = new_jaw_clench_state();
156        jcl_set_both(&mut s, 1.0);
157        assert!((jcl_bite_force(&s) - 1.0).abs() < 1e-6);
158    }
159
160    #[test]
161    fn temporal_angle_positive_when_clenched() {
162        let mut s = new_jaw_clench_state();
163        jcl_set_both(&mut s, 1.0);
164        assert!(jcl_temporal_angle_rad(&s) > 0.0);
165    }
166
167    #[test]
168    fn weights_proportional() {
169        let cfg = default_jaw_clench_config();
170        let mut s = new_jaw_clench_state();
171        jcl_set_both(&mut s, 1.0);
172        let w = jcl_to_weights(&s, &cfg);
173        assert!((w[0] - cfg.max_bulge_m).abs() < 1e-6);
174    }
175
176    #[test]
177    fn blend_midpoint() {
178        let mut b = new_jaw_clench_state();
179        jcl_set_both(&mut b, 1.0);
180        let r = jcl_blend(&new_jaw_clench_state(), &b, 0.5);
181        assert!((r.left - 0.5).abs() < 1e-5);
182    }
183
184    #[test]
185    fn json_has_keys() {
186        let j = jcl_to_json(&new_jaw_clench_state());
187        assert!(j.contains("left") && j.contains("right"));
188    }
189}
oxihuman_morph/jaw_clench_control.rs

oxihuman_morph/
jaw_clench_control.rs
