use glam::{Vec2, Vec3};
#[inline(always)]
pub fn sdf_parabola_segment(p: Vec3, width: f32, para_height: f32, half_depth: f32) -> f32 {
let px = p.x.abs();
let py = p.y;
let w = width;
let h = para_height;
let y_arch = if px <= w {
h * ((px / w) * -(px / w) + 1.0)
} else {
0.0
};
let inside = px <= w && py >= 0.0 && py <= y_arch;
let ww = w * w;
let mut t = super::clamp_total(px, 0.0, w);
for _ in 0..8 {
let ft = h * (1.0 - t * t / ww);
let dft = -2.0 * h * t / ww;
let ex = px - t;
let ey = py - ft;
let f = -ex + ey * dft;
let df = 1.0 + dft * dft + ey * (-2.0 * h / ww);
if df.abs() > 1e-10 {
t = super::clamp_total(t - f / df, 0.0, w);
}
}
let closest_y = h * (1.0 - t * t / ww);
let d_para = Vec2::new(px - t, py - closest_y).length();
let d_base = if px <= w {
py.abs()
} else {
Vec2::new(px - w, py).length()
};
let d_unsigned = d_para.min(d_base);
let d_2d = if inside { -d_unsigned } else { d_unsigned };
let d_z = p.z.abs() - half_depth;
let ext = Vec2::new(d_2d.max(0.0), d_z.max(0.0));
d_2d.max(d_z).min(0.0) + ext.length()
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_parabola_segment_inside() {
let d = sdf_parabola_segment(Vec3::new(0.0, 0.3, 0.0), 1.0, 1.0, 0.5);
assert!(d < 0.0, "Interior point should be inside, got {}", d);
}
#[test]
fn test_parabola_segment_far_outside() {
let d = sdf_parabola_segment(Vec3::new(5.0, 0.0, 0.0), 1.0, 1.0, 0.5);
assert!(d > 0.0, "Far point should be outside, got {}", d);
}
#[test]
fn test_parabola_segment_symmetry_x() {
let d1 = sdf_parabola_segment(Vec3::new(0.3, 0.2, 0.1), 1.0, 1.0, 0.5);
let d2 = sdf_parabola_segment(Vec3::new(-0.3, 0.2, 0.1), 1.0, 1.0, 0.5);
assert!((d1 - d2).abs() < 0.001, "Should be symmetric in X");
}
#[test]
fn test_parabola_segment_symmetry_z() {
let d1 = sdf_parabola_segment(Vec3::new(0.2, 0.3, 0.2), 1.0, 1.0, 0.5);
let d2 = sdf_parabola_segment(Vec3::new(0.2, 0.3, -0.2), 1.0, 1.0, 0.5);
assert!((d1 - d2).abs() < 0.001, "Should be symmetric in Z");
}
}