use crate::solid_sweep::outline_signed_distance;
use crate::topology::{CoedgeRecord, EdgeRecord, FaceRecord, LoopRecord, ShellRecord, VertexRecord};
use crate::{make_line, make_revolution, BrepSolid, KernelRefusal, KernelStage, NurbsCurve, NurbsSurface, OrRefuse, OutlinePiece, SolidMotion, Vec3, Vec4};
use std::collections::HashMap;
use std::f64::consts::{PI, TAU};
#[derive(Clone, Debug)]
pub enum SolidTool {
Polyhedron { vertices: Vec<Vec3>, faces: Vec<Vec<usize>> },
Revolved { axis_origin: Vec3, axis_direction: Vec3, radial: Vec3, meridian: Vec<OutlinePiece> },
}
const PCURVE_SAMPLES: usize = 576;
pub const SOLID_SWEEP_WHOLE_REFUSAL: &str = "sweepSolid";
fn refuse(what: &'static str, message: String) -> KernelRefusal {
KernelRefusal::unsupported(KernelStage::Classify, what, format!("{SOLID_SWEEP_WHOLE_REFUSAL}: {message}"))
}
fn input(what: &'static str, message: String) -> KernelRefusal {
KernelRefusal::input(KernelStage::Collect, what, format!("{SOLID_SWEEP_WHOLE_REFUSAL}: {message}"))
}
fn frag<T>(result: Result<T, String>, what: &'static str) -> Result<T, KernelRefusal> {
result.or_refuse(KernelStage::Fragment, what)
}
#[derive(Clone, Copy, Debug)]
struct Screw {
a: Vec3,
e: Vec3,
omega: f64,
c: f64,
c2: f64,
t0: f64,
t1: f64,
frame: Option<Frame>,
frozen: Option<f64>,
extent: f64,
}
#[derive(Clone, Copy, Debug)]
struct Frame {
control: [Vec3; 4],
up: Vec3,
start_tangent: Vec3,
}
impl Frame {
fn point(&self, s: f64) -> Vec3 {
let [p0, p1, p2, p3] = self.control;
let r = 1.0 - s;
p0.scale(r * r * r).add(p1.scale(3.0 * r * r * s)).add(p2.scale(3.0 * r * s * s)).add(p3.scale(s * s * s))
}
fn first(&self, s: f64) -> Vec3 {
let [p0, p1, p2, p3] = self.control;
let r = 1.0 - s;
p1.sub(p0).scale(3.0 * r * r).add(p2.sub(p1).scale(6.0 * r * s)).add(p3.sub(p2).scale(3.0 * s * s))
}
fn second(&self, s: f64) -> Vec3 {
let [p0, p1, p2, p3] = self.control;
p2.sub(p1.scale(2.0)).add(p0).scale(6.0 * (1.0 - s)).add(p3.sub(p2.scale(2.0)).add(p1).scale(6.0 * s))
}
fn angle(&self, s: f64) -> f64 {
let t = self.first(s);
self.up.dot(self.start_tangent.cross(t)).atan2(self.start_tangent.dot(t))
}
fn angle_rate(&self, s: f64) -> f64 {
let (d1, d2) = (self.first(s), self.second(s));
self.up.dot(d1.cross(d2)) / d1.dot(d1)
}
}
#[derive(Clone, Copy, Debug)]
pub struct RisingTurn {
pub axis_origin: Vec3,
pub axis_direction: Vec3,
pub rise: [f64; 2],
pub from: f64,
pub to: f64,
}
impl Screw {
fn from_motion(motion: &SolidMotion) -> Result<Self, KernelRefusal> {
let [step] = motion.steps.as_slice() else {
return Err(refuse("motion_steps", "only a single screw, turn or slide is swept whole here".into()));
};
let e = step
.axis_direction
.normalized()
.map_err(|_| input("axis", "the motion's axis has no direction".into()))?;
if !(motion.from.is_finite() && motion.to.is_finite() && motion.to > motion.from) {
return Err(input("range", "the motion's range must run forward".into()));
}
if step.turn == 0.0 && step.advance == 0.0 {
return Err(input("still", "the motion does not move".into()));
}
Ok(Self { a: step.axis_origin, e, omega: step.turn, c: step.advance, c2: 0.0, t0: motion.from, t1: motion.to, frame: None, frozen: None, extent: 0.0 })
}
fn from_rising(motion: &RisingTurn) -> Result<Self, KernelRefusal> {
let e = motion.axis_direction.normalized().map_err(|_| input("axis", "the motion's axis has no direction".into()))?;
if !(motion.from.is_finite() && motion.to.is_finite() && motion.to > motion.from) {
return Err(input("range", "the motion's range must run forward".into()));
}
let rate = |t: f64| motion.rise[0] + 2.0 * motion.rise[1] * t;
if rate(motion.from) * rate(motion.to) <= 0.0 {
return Err(refuse("rise_turns", "the rise stops or turns back within the motion".into()));
}
Ok(Self { a: motion.axis_origin, e, omega: 1.0, c: motion.rise[0], c2: motion.rise[1], t0: motion.from, t1: motion.to, frame: None, frozen: None, extent: 0.0 })
}
fn advance(&self, t: f64) -> f64 {
self.c * t + self.c2 * t * t
}
fn rate_bound(&self) -> f64 {
(self.c + 2.0 * self.c2 * self.t0).abs().max((self.c + 2.0 * self.c2 * self.t1).abs())
}
fn apply(&self, t: f64, x: Vec3) -> Vec3 {
if let Some(frame) = &self.frame {
return frame.point(t).add(x.sub(self.a).rotate_about(frame.up, frame.angle(t)));
}
self.a.add(x.sub(self.a).rotate_about(self.e, self.omega * t)).add(self.e.scale(self.advance(t)))
}
fn invert(&self, t: f64, y: Vec3) -> Vec3 {
if let Some(frame) = &self.frame {
return self.a.add(y.sub(frame.point(t)).rotate_about(frame.up, -frame.angle(t)));
}
self.a.add(y.sub(self.e.scale(self.advance(t))).sub(self.a).rotate_about(self.e, -self.omega * t))
}
fn velocity(&self, x: Vec3) -> Vec3 {
self.velocity_at(self.frozen.unwrap_or(0.5 * (self.t0 + self.t1)), x)
}
fn velocity_at(&self, t: f64, x: Vec3) -> Vec3 {
if let Some(frame) = &self.frame {
return frame.start_tangent.scale(frame.first(t).length()).add(frame.up.cross(x.sub(self.a)).scale(frame.angle_rate(t)));
}
self.e.cross(x.sub(self.a)).scale(self.omega).add(self.e.scale(self.c + 2.0 * self.c2 * t))
}
fn jet(&self, t: f64, x: Vec3) -> (Vec3, Vec3, Vec3) {
let r = x.sub(self.a).rotate_about(self.e, self.omega * t);
let axial = self.e.scale(r.dot(self.e));
let radial = r.sub(axial);
let value = self.a.add(r).add(self.e.scale(self.advance(t)));
let first = self.e.cross(radial).scale(self.omega).add(self.e.scale(self.c + 2.0 * self.c2 * t));
let second = radial.scale(-self.omega * self.omega).add(self.e.scale(2.0 * self.c2));
(value, first, second)
}
fn move_point(&self, t: f64, p: &Vec4) -> Result<Vec4, KernelRefusal> {
let x = frag(p.point(), "control_point")?;
Ok(Vec4::from_point(self.apply(t, x), p.w))
}
fn move_curve(&self, t: f64, curve: &NurbsCurve) -> Result<NurbsCurve, KernelRefusal> {
let points = curve.control_points.iter().map(|p| self.move_point(t, p)).collect::<Result<Vec<_>, _>>()?;
frag(NurbsCurve::new(curve.degree, curve.knots.clone(), points), "moved_curve")
}
fn move_surface(&self, t: f64, surface: &NurbsSurface) -> Result<NurbsSurface, KernelRefusal> {
let mut rows = Vec::with_capacity(surface.control_points.len());
for row in &surface.control_points {
rows.push(row.iter().map(|p| self.move_point(t, p)).collect::<Result<Vec<_>, _>>()?);
}
frag(
NurbsSurface::new(surface.degree_u, surface.degree_v, surface.knots_u.clone(), surface.knots_v.clone(), rows),
"moved_surface",
)
}
}
fn row_times(screw: &Screw) -> Vec<f64> {
let turned = (screw.omega * (screw.t1 - screw.t0)).abs();
let rows = ((turned / (0.5 * PI)).ceil() as usize).max(1);
let at = |k: usize| match k {
0 => screw.t0,
k if k == rows => screw.t1,
k => screw.t0 + (screw.t1 - screw.t0) * (k as f64 - 0.381966011250105) / rows as f64,
};
(0..=rows).map(at).collect()
}
fn carry_segments(screw: &Screw, ta: f64, tb: f64) -> usize {
let turned = (screw.omega * (tb - ta)).abs();
((turned / TAU * 48.0).ceil() as usize).max(4)
}
fn carry_knots(segments: usize) -> Vec<f64> {
let mut knots = vec![0.0; 6];
for k in 1..segments {
knots.extend([k as f64 / segments as f64; 3]);
}
knots.extend([1.0; 6]);
knots
}
fn carry_points(screw: &Screw, x: Vec3, ta: f64, tb: f64, segments: usize) -> Vec<Vec3> {
let h = (tb - ta) / segments as f64;
let at = |k: usize| screw.jet(ta + h * k as f64, x);
let piece = |k: usize| {
let ((f0, d0, a0), (f1, d1, a1)) = (at(k), at(k + 1));
[
f0,
f0.add(d0.scale(h / 5.0)),
f0.add(d0.scale(2.0 * h / 5.0)).add(a0.scale(h * h / 20.0)),
f1.sub(d1.scale(2.0 * h / 5.0)).add(a1.scale(h * h / 20.0)),
f1.sub(d1.scale(h / 5.0)),
f1,
]
};
let mut points = Vec::with_capacity(3 * segments + 3);
let first = piece(0);
points.extend_from_slice(&first[..3]);
for k in 0..segments - 1 {
let (here, next) = (piece(k), piece(k + 1));
points.push(here[3]);
points.push(here[4].scale(2.0).sub(here[3]));
points.push(next[2]);
}
let last = piece(segments - 1);
points.extend_from_slice(&last[3..]);
points
}
fn carry_curve(screw: &Screw, x: Vec3, ta: f64, tb: f64, segments: usize) -> Result<NurbsCurve, KernelRefusal> {
frag(
NurbsCurve::new(5, carry_knots(segments), carry_points(screw, x, ta, tb, segments).into_iter().map(|p| Vec4::from_point(p, 1.0)).collect()),
"carry_curve",
)
}
struct ToolCurve {
curve: NurbsCurve,
start: usize,
end: usize,
degenerate: bool,
samples: Option<Vec<Vec3>>,
}
const LOFT_SAMPLES: usize = 128;
struct Cap {
surface: NurbsSurface,
front: bool,
coedges: Vec<(usize, bool, NurbsCurve)>,
}
struct Split {
points: Vec<Vec3>,
curves: Vec<ToolCurve>,
caps: Vec<Cap>,
scale: f64,
shape: Shape,
}
enum Shape {
Planes(Vec<(Vec3, f64)>),
Revolved { origin: Vec3, axis: Vec3, meridian: Vec<OutlinePiece> },
}
impl Split {
fn distance(&self, x: Vec3) -> f64 {
match &self.shape {
Shape::Planes(planes) => planes.iter().map(|(n, d)| n.dot(x) - d).fold(f64::NEG_INFINITY, f64::max),
Shape::Revolved { origin, axis, meridian } => {
let r = x.sub(*origin);
let z = r.dot(*axis);
let rho = r.sub(axis.scale(z)).length();
outline_signed_distance(meridian, [rho, z])
}
}
}
fn grazing(&self) -> Result<Vec<(usize, bool)>, KernelRefusal> {
let mut uses: HashMap<usize, Vec<(bool, bool)>> = HashMap::new();
for cap in &self.caps {
for &(curve, forward, _) in &cap.coedges {
uses.entry(curve).or_default().push((forward, cap.front));
}
}
let mut grazing = Vec::new();
for (curve, list) in uses {
if self.curves[curve].degenerate {
continue;
}
match list.as_slice() {
[(fa, front_a), (fb, front_b)] => {
if fa == fb {
return Err(refuse("split_topology", "two caps run one boundary curve the same way".into()));
}
if front_a != front_b {
grazing.push((curve, if *front_a { *fb } else { *fa }));
}
}
_ => return Err(refuse("split_topology", format!("a boundary curve is used {} times", list.len()))),
}
}
grazing.sort_unstable();
Ok(grazing)
}
}
fn uv_line(a: [f64; 2], b: [f64; 2]) -> Result<NurbsCurve, KernelRefusal> {
frag(make_line(Vec3::new(a[0], a[1], 0.0), Vec3::new(b[0], b[1], 0.0)), "pcurve")
}
fn plane_patch(origin: Vec3, u: Vec3, v: Vec3, points: &[Vec3]) -> Result<NurbsSurface, KernelRefusal> {
let (mut lo, mut hi) = ([f64::INFINITY; 2], [f64::NEG_INFINITY; 2]);
for p in points {
let q = [p.sub(origin).dot(u), p.sub(origin).dot(v)];
for k in 0..2 {
lo[k] = lo[k].min(q[k]);
hi[k] = hi[k].max(q[k]);
}
}
let at = |s: f64, t: f64| Vec4::from_point(origin.add(u.scale(s)).add(v.scale(t)), 1.0);
frag(
NurbsSurface::new(
1,
1,
vec![lo[0], lo[0], hi[0], hi[0]],
vec![lo[1], lo[1], hi[1], hi[1]],
vec![vec![at(lo[0], lo[1]), at(lo[0], hi[1])], vec![at(hi[0], lo[1]), at(hi[0], hi[1])]],
),
"plane_surface",
)
}
fn plane_pcurve(origin: Vec3, u: Vec3, v: Vec3, curve: &NurbsCurve, forward: bool) -> Result<NurbsCurve, KernelRefusal> {
let mut points = Vec::with_capacity(curve.control_points.len());
for p in &curve.control_points {
let x = frag(p.point(), "control_point")?.sub(origin);
points.push(Vec4::from_point(Vec3::new(x.dot(u), x.dot(v), 0.0), p.w));
}
let pcurve = frag(NurbsCurve::new(curve.degree, curve.knots.clone(), points), "plane_pcurve")?;
if forward { Ok(pcurve) } else { frag(pcurve.reversed(), "plane_pcurve") }
}
fn split_polyhedron(screw: &Screw, vertices: &[Vec3], faces: &[Vec<usize>]) -> Result<Split, KernelRefusal> {
if vertices.len() < 4 || faces.len() < 4 {
return Err(input("polyhedron", "a polyhedron needs at least four corners and four faces".into()));
}
let centroid = vertices.iter().fold(Vec3::new(0.0, 0.0, 0.0), |s, &p| s.add(p)).scale(1.0 / vertices.len() as f64);
let scale = vertices.iter().map(|p| p.sub(centroid).length()).fold(0.0f64, f64::max).max(1e-300);
let tolerance = 1e-12 * scale.max(screw.a.sub(centroid).length());
let mut loops: Vec<Vec<usize>> = Vec::with_capacity(faces.len());
let mut planes = Vec::with_capacity(faces.len());
for face in faces {
if face.len() < 3 || face.iter().any(|&i| i >= vertices.len()) {
return Err(input("face", "a face needs three or more existing corners".into()));
}
let mut normal = Vec3::new(0.0, 0.0, 0.0);
for k in 0..face.len() {
let (p, q) = (vertices[face[k]], vertices[face[(k + 1) % face.len()]]);
normal = normal.add(p.sub(centroid).cross(q.sub(centroid)));
}
let mut normal = normal.normalized().map_err(|_| input("face", "a face has no area".into()))?;
let mut face = face.clone();
if normal.dot(vertices[face[0]].sub(centroid)) < 0.0 {
normal = normal.scale(-1.0);
face.reverse();
}
let offset = normal.dot(vertices[face[0]]);
for p in vertices {
if normal.dot(*p) - offset > 1e-9 * scale {
return Err(refuse("not_convex", "the polyhedron is not convex".into()));
}
}
for &i in &face {
if (normal.dot(vertices[i]) - offset).abs() > 1e-9 * scale {
return Err(input("face", "a face's corners are not in one plane".into()));
}
}
loops.push(face);
planes.push((normal, offset));
}
let g = |face: usize, x: Vec3| -> f64 {
let value = screw.velocity(x).dot(planes[face].0);
if value.abs() <= tolerance * screw.velocity(x).length().max(1.0) { 0.0 } else { value }
};
let mut owner: HashMap<(usize, usize), usize> = HashMap::new();
for (index, face) in loops.iter().enumerate() {
for k in 0..face.len() {
owner.insert((face[k], face[(k + 1) % face.len()]), index);
}
}
for &(i, j) in owner.keys() {
if !owner.contains_key(&(j, i)) {
return Err(input("closed", "the polyhedron's faces do not close".into()));
}
}
let mut points: Vec<Vec3> = vertices.to_vec();
let mut edge_splits: HashMap<(usize, usize), Vec<usize>> = HashMap::new();
let mut keys: Vec<(usize, usize)> = owner.keys().copied().filter(|&(i, j)| i < j).collect();
keys.sort_unstable();
for (i, j) in keys {
let mut params: Vec<f64> = Vec::new();
for f in [owner[&(i, j)], owner[&(j, i)]] {
let (gi, gj) = (g(f, vertices[i]), g(f, vertices[j]));
if gi * gj < 0.0 {
params.push(gi / (gi - gj));
}
}
params.sort_by(|a, b| a.total_cmp(b));
params.dedup_by(|a, b| (*a - *b).abs() < 1e-12);
let mut list = Vec::new();
for s in params {
points.push(vertices[i].add(vertices[j].sub(vertices[i]).scale(s)));
list.push(points.len() - 1);
}
edge_splits.insert((i, j), list);
}
let splits_along = |i: usize, j: usize| -> Vec<usize> {
if i < j {
edge_splits[&(i, j)].clone()
} else {
edge_splits[&(j, i)].iter().rev().copied().collect()
}
};
let mut rings: Vec<(Vec<usize>, usize, bool)> = Vec::new();
for (index, face) in loops.iter().enumerate() {
let mut ring: Vec<usize> = Vec::new();
for k in 0..face.len() {
ring.push(face[k]);
ring.extend(splits_along(face[k], face[(k + 1) % face.len()]));
}
let values: Vec<f64> = ring.iter().map(|&p| g(index, points[p])).collect();
let positive = values.iter().any(|&v| v > 0.0);
let negative = values.iter().any(|&v| v < 0.0);
if !positive && !negative {
return Err(refuse("face_in_own_plane", format!("face {index} moves in its own plane (its sweep is flat)")));
}
if !(positive && negative) {
rings.push((ring, index, positive));
continue;
}
let zeros: Vec<usize> = (0..ring.len()).filter(|&k| values[k] == 0.0).collect();
if zeros.len() != 2 {
return Err(refuse("grazing_split", format!("face {index} meets its grazing line {} times", zeros.len())));
}
let (za, zb) = (zeros[0], zeros[1]);
if za + 1 == zb || (zb + 1) % ring.len() == za {
return Err(refuse("grazing_split", format!("face {index} is split along one of its own edges")));
}
let first: Vec<usize> = (za..=zb).map(|k| ring[k]).collect();
let second: Vec<usize> = (zb..ring.len()).chain(0..=za).map(|k| ring[k]).collect();
let first_front = values[za + 1] > 0.0;
rings.push((first, index, first_front));
rings.push((second, index, !first_front));
}
if screw.c2 != 0.0 {
for (ring, face, front) in &rings {
for &p in ring {
for t in [screw.t0, screw.t1] {
let value = screw.velocity_at(t, points[p]).dot(planes[*face].0);
let slack = tolerance * screw.velocity_at(t, points[p]).length().max(1.0);
if (*front && value < -slack) || (!*front && value > slack) {
return Err(refuse("rising_sides", format!("face {face} changes side as the rise grows")));
}
}
}
}
}
let mut curves: Vec<ToolCurve> = Vec::new();
let mut by_pair: HashMap<(usize, usize), usize> = HashMap::new();
let mut caps = Vec::with_capacity(rings.len());
for (ring, face, front) in rings {
let normal = planes[face].0;
let origin = points[ring[0]];
let u = frag(points[ring[1]].sub(origin).normalized(), "plane_axis")?;
let v = normal.cross(u);
let corners: Vec<Vec3> = ring.iter().map(|&p| points[p]).collect();
let surface = plane_patch(origin, u, v, &corners)?;
let mut coedges = Vec::with_capacity(ring.len());
for k in 0..ring.len() {
let (p, q) = (ring[k], ring[(k + 1) % ring.len()]);
let key = (p.min(q), p.max(q));
let index = match by_pair.get(&key) {
Some(&index) => index,
None => {
curves.push(ToolCurve {
curve: frag(make_line(points[key.0], points[key.1]), "line")?,
start: key.0,
end: key.1,
degenerate: false,
samples: None,
});
by_pair.insert(key, curves.len() - 1);
curves.len() - 1
}
};
let forward = p == key.0;
coedges.push((index, forward, plane_pcurve(origin, u, v, &curves[index].curve, forward)?));
}
caps.push(Cap { surface, front, coedges });
}
Ok(Split { points, curves, caps, scale, shape: Shape::Planes(planes) })
}
fn harmonic(g: impl Fn(f64) -> f64) -> (f64, f64, f64) {
let (g0, g90, g180) = (g(0.0), g(0.5 * PI), g(PI));
let c = 0.5 * (g0 + g180);
(0.5 * (g0 - g180), g90 - c, c)
}
fn piece_point(piece: &OutlinePiece, s: f64) -> [f64; 2] {
match *piece {
OutlinePiece::Line { from, to } => [from[0] + (to[0] - from[0]) * s, from[1] + (to[1] - from[1]) * s],
OutlinePiece::Arc { centre, radius, start, end } => {
let angle = start + (end - start) * s;
[centre[0] + radius * angle.cos(), centre[1] + radius * angle.sin()]
}
}
}
fn split_revolved(
screw: &Screw,
origin: Vec3,
axis: Vec3,
radial: Vec3,
meridian: &[OutlinePiece],
) -> Result<Split, KernelRefusal> {
let w = axis.normalized().map_err(|_| input("axis", "the tool's axis has no direction".into()))?;
let u1 = radial.sub(w.scale(radial.dot(w))).normalized().map_err(|_| input("radial", "the radial direction lies along the axis".into()))?;
let u2 = w.cross(u1);
if let [OutlinePiece::Arc { centre, radius, start, end }] = meridian {
if ((end - start).abs() - TAU).abs() < 1e-12 && centre[0] > *radius {
return split_torus(screw, origin, w, u1, u2, *centre, *radius, meridian);
}
}
let at = |rho: f64, z: f64, psi: f64| origin.add(w.scale(z)).add(u1.scale(rho * psi.cos())).add(u2.scale(rho * psi.sin()));
let scale = meridian
.iter()
.flat_map(|p| [piece_point(p, 0.0), piece_point(p, 1.0)])
.map(|q| q[0].abs().max(q[1].abs()))
.fold(0.0f64, f64::max)
.max(1e-300);
let on_axis = |q: [f64; 2]| q[0].abs() <= 1e-12 * scale;
let mut area = 0.0;
for piece in meridian {
for k in 0..16 {
let (p, q) = (piece_point(piece, k as f64 / 16.0), piece_point(piece, (k + 1) as f64 / 16.0));
area += p[0] * q[1] - q[0] * p[1];
}
}
let reversed_meridian: Vec<OutlinePiece>;
let meridian: &[OutlinePiece] = if area > 0.0 {
meridian
} else {
reversed_meridian = meridian
.iter()
.rev()
.map(|piece| match *piece {
OutlinePiece::Line { from, to } => OutlinePiece::Line { from: to, to: from },
OutlinePiece::Arc { centre, radius, start, end } => OutlinePiece::Arc { centre, radius, start: end, end: start },
})
.collect();
&reversed_meridian
};
let ccw = true;
let normal = |piece: &OutlinePiece, s: f64| -> [f64; 2] {
let (p, q) = (piece_point(piece, (s - 1e-6).max(0.0)), piece_point(piece, (s + 1e-6).min(1.0)));
let (dx, dz) = (q[0] - p[0], q[1] - p[1]);
let length = dx.hypot(dz);
if ccw { [dz / length, -dx / length] } else { [-dz / length, dx / length] }
};
let g = |piece: &OutlinePiece, s: f64, psi: f64| -> f64 {
let m = piece_point(piece, s);
let n = normal(piece, s);
let x = at(m[0], m[1], psi);
let normal = w.scale(n[1]).add(u1.scale(n[0] * psi.cos())).add(u2.scale(n[0] * psi.sin()));
screw.velocity(x).dot(normal)
};
let faces: Vec<usize> = (0..meridian.len())
.filter(|&i| !(matches!(meridian[i], OutlinePiece::Line { .. }) && on_axis(piece_point(&meridian[i], 0.0)) && on_axis(piece_point(&meridian[i], 1.0))))
.collect();
enum Kind {
Ruled(Vec<f64>),
Moving,
Sphere { centre: Vec3, pole_at_start: bool },
Ball { centre: Vec3, radius: f64 },
TorusZone { psi_a: f64, psi_b: f64, at_end: bool },
}
let branches = |piece: &OutlinePiece, s: f64| -> Option<(f64, f64)> {
let reference = {
let (a, b, _) = harmonic(|psi| g(piece, 0.0, psi));
b.atan2(a)
};
let (a, b, c) = harmonic(|psi| g(piece, s, psi));
let r = a.hypot(b);
if !(c.abs() < r * (1.0 - 1e-9)) {
return None;
}
let base = reference + (b.atan2(a) - reference + PI).rem_euclid(TAU) - PI;
let spread = (-c / r).acos();
Some((base - spread, base + spread))
};
let mut kinds: Vec<Kind> = Vec::with_capacity(faces.len());
for &i in &faces {
let piece = &meridian[i];
match *piece {
OutlinePiece::Line { .. } => {
let roots = |s: f64| -> Vec<f64> {
let (a, b, c) = harmonic(|psi| g(piece, s, psi));
let r = a.hypot(b);
if c.abs() >= r * (1.0 - 1e-12) {
return Vec::new();
}
let base = b.atan2(a);
let spread = (-c / r).acos();
let mut list = vec![(base - spread).rem_euclid(TAU), (base + spread).rem_euclid(TAU)];
list.sort_by(|x, y| x.total_cmp(y));
list
};
let alive = |s: f64| {
let (a, b, c) = harmonic(|psi| g(piece, s, psi));
a.hypot(b).max(c.abs()) > 1e-12 * scale * screw.velocity(origin).length().max(screw.omega.abs() * scale).max(screw.rate_bound())
};
let readings: Vec<Vec<f64>> = [0.0, 0.5, 1.0].into_iter().filter(|&t| alive(t)).map(roots).collect();
if readings.is_empty() {
return Err(refuse("revolved_grazing", format!("meridian piece {i} grazes all over")));
}
let r0 = readings[0].clone();
let still = readings.iter().all(|r| r.len() == r0.len() && r.iter().zip(&r0).all(|(x, y)| (x - y).abs() < 1e-10));
if still {
kinds.push(Kind::Ruled(r0));
continue;
}
let mut previous: Option<(f64, f64)> = None;
for k in 0..=64 {
let Some(pair) = branches(piece, k as f64 / 64.0) else {
return Err(refuse("revolved_grazing", format!("the grazing curves on meridian piece {i} do not run rim to rim")));
};
if previous.is_some_and(|p| (p.0 - pair.0).abs() > 0.5 || (p.1 - pair.1).abs() > 0.5) {
return Err(refuse("revolved_grazing", format!("the grazing curves on meridian piece {i} turn too fast")));
}
previous = Some(pair);
}
kinds.push(Kind::Moving);
}
OutlinePiece::Arc { centre, radius, start, end } => {
let (p0, p1) = (piece_point(piece, 0.0), piece_point(piece, 1.0));
let span = (end - start).abs();
let centre_on_axis = on_axis(centre);
let pole_at_start = on_axis(p0);
let equator = |p: [f64; 2]| (p[1] - centre[1]).abs() <= 1e-12 * scale && (p[0] - radius).abs() <= 1e-12 * scale;
if centre_on_axis && (span - PI).abs() < 1e-12 && pole_at_start && on_axis(p1) {
kinds.push(Kind::Ball { centre: origin.add(w.scale(centre[1])), radius });
continue;
}
if centre[0] > radius * (1.0 + 1e-9) && end > start && span < PI {
let at_rim = |v: f64| [centre[0] + radius * v.cos(), centre[1] + radius * v.sin()];
let g_tube = |psi: f64, v: f64| -> f64 {
let m = at_rim(v);
let x = at(m[0], m[1], psi);
let normal = w.scale(v.sin()).add(u1.scale(v.cos() * psi.cos())).add(u2.scale(v.cos() * psi.sin()));
screw.velocity(x).dot(normal)
};
let in_zone = |psi: f64| -> Result<Option<f64>, KernelRefusal> {
let (a, b, c) = harmonic(|v| g_tube(psi, v));
let r = a.hypot(b);
if !(c.abs() < r) {
return Ok(None);
}
let base = b.atan2(a);
let spread = (-c / r).acos();
let inside: Vec<f64> = [base - spread, base + spread]
.into_iter()
.map(|v| start + (v - start).rem_euclid(TAU))
.filter(|&v| v <= end)
.collect();
match inside.as_slice() {
[] => Ok(None),
[v] => Ok(Some(*v)),
_ => Err(refuse("torus_zone", format!("meridian piece {i}: the grazing curve crosses the rounded corner twice at one angle"))),
}
};
let count = 720;
let has: Vec<bool> = (0..count).map(|k| in_zone(TAU * k as f64 / count as f64).map(|r| r.is_some())).collect::<Result<_, _>>()?;
let rises: Vec<usize> = (0..count).filter(|&k| has[k] && !has[(k + count - 1) % count]).collect();
let falls: Vec<usize> = (0..count).filter(|&k| !has[k] && has[(k + count - 1) % count]).collect();
let ([rise], [fall]) = (rises.as_slice(), falls.as_slice()) else {
return Err(refuse("torus_zone", format!("meridian piece {i}: the grazing curve does not cross the rounded corner once")));
};
let estimate = |k: usize| TAU * (k as f64 - 0.5) / count as f64;
let (ea, eb) = (estimate(*rise), estimate(*fall));
let sample = |k: usize| TAU * (k % count) as f64 / count as f64;
let probe_v = in_zone(sample(*rise))?.ok_or_else(|| refuse("torus_zone", "the grazing curve's crossing is not found".into()))?;
let at_end = (probe_v - end).abs() < (probe_v - start).abs();
let rim_v = if at_end { end } else { start };
let (a, b, c) = harmonic(|psi| g_tube(psi, rim_v));
let r = a.hypot(b);
let base = b.atan2(a);
let spread = (-c / r).acos();
let roots = [base - spread, base + spread];
let nearest = |e: f64| roots.iter().copied().min_by(|x, y| ((x - e + PI).rem_euclid(TAU) - PI).abs().total_cmp(&((y - e + PI).rem_euclid(TAU) - PI).abs())).expect("two");
let (psi_a, psi_b) = (nearest(ea).rem_euclid(TAU), nearest(eb).rem_euclid(TAU));
let psi_b = if psi_b <= psi_a { psi_b + TAU } else { psi_b };
let leave_v = in_zone(sample(*fall + count - 1))?.ok_or_else(|| refuse("torus_zone", "the grazing curve's crossing is not found".into()))?;
if ((leave_v - end).abs() < (leave_v - start).abs()) != at_end {
return Err(refuse("torus_zone", format!("meridian piece {i}: the grazing curve crosses the rounded corner from rim to rim")));
}
kinds.push(Kind::TorusZone { psi_a, psi_b, at_end });
continue;
}
if !(centre_on_axis && (span - 0.5 * PI).abs() < 1e-12 && (pole_at_start && equator(p1) || on_axis(p1) && equator(p0))) {
return Err(refuse("revolved_arc", format!("meridian piece {i}: only a quarter circle from a pole to an equator is swept here")));
}
kinds.push(Kind::Sphere { centre: origin.add(w.scale(centre[1])), pole_at_start });
}
}
}
let count = meridian.len();
let junction = |k: usize| piece_point(&meridian[k], 1.0); let face_of = |piece: usize| faces.iter().position(|&f| f == piece);
let mut rim_angles: HashMap<usize, Vec<f64>> = HashMap::new();
for k in 0..count {
if on_axis(junction(k)) {
continue;
}
let mut angles: Vec<f64> = Vec::new();
for (piece, s_end) in [(k, 1.0), ((k + 1) % count, 0.0)] {
let Some(f) = face_of(piece) else { continue };
match &kinds[f] {
Kind::Ruled(roots) => angles.extend(roots.iter().copied()),
Kind::Ball { .. } => {}
Kind::TorusZone { psi_a, psi_b, .. } => angles.extend([psi_a.rem_euclid(TAU), psi_b.rem_euclid(TAU)]),
Kind::Moving => {
let (lower, upper) = branches(&meridian[piece], s_end).expect("checked");
angles.extend([lower.rem_euclid(TAU), upper.rem_euclid(TAU)]);
}
Kind::Sphere { centre, .. } => {
let d = screw.velocity(*centre).cross(w);
let base = d.dot(u2).atan2(d.dot(u1)).rem_euclid(TAU);
angles.extend([base, (base + PI).rem_euclid(TAU)]);
}
}
}
angles.sort_by(|a, b| a.total_cmp(b));
let mut merged: Vec<f64> = Vec::new();
for a in angles {
if merged.last().is_some_and(|&b| (a - b).abs() < 1e-9) || merged.first().is_some_and(|&b| (b + TAU - a).abs() < 1e-9) {
continue;
}
merged.push(a);
}
rim_angles.insert(k, merged);
}
for (f, &piece) in faces.iter().enumerate() {
for (k, s_end) in [((piece + count - 1) % count, 0.0), (piece, 1.0)] {
let own: Vec<f64> = match &kinds[f] {
Kind::Ruled(roots) => roots.clone(),
Kind::Moving => {
let (lower, upper) = branches(&meridian[piece], s_end).expect("checked");
let mut list = vec![lower.rem_euclid(TAU), upper.rem_euclid(TAU)];
list.sort_by(|a, b| a.total_cmp(b));
list
}
Kind::TorusZone { psi_a, psi_b, .. } => {
let mut list = vec![psi_a.rem_euclid(TAU), psi_b.rem_euclid(TAU)];
list.sort_by(|a, b| a.total_cmp(b));
list
}
Kind::Sphere { .. } | Kind::Ball { .. } => continue,
};
if let Some(angles) = rim_angles.get(&k) {
if own.is_empty() {
continue;
}
if angles.len() != own.len() || angles.iter().zip(&own).any(|(a, b)| (a - b).abs() > 1e-9) {
return Err(refuse("revolved_rims", format!("meridian piece {piece} splits at other angles than its rim")));
}
}
}
}
let mut points: Vec<Vec3> = Vec::new();
let mut curves: Vec<ToolCurve> = Vec::new();
let mut caps: Vec<Cap> = Vec::new();
let mut rim_points: HashMap<(usize, usize), usize> = HashMap::new();
let mut rim_arcs: HashMap<(usize, usize), usize> = HashMap::new();
let mut rim_keys: Vec<usize> = rim_angles.keys().copied().collect();
rim_keys.sort_unstable();
for &k in &rim_keys {
let q = junction(k);
let angles = rim_angles[&k].clone();
let angles = if angles.is_empty() { vec![0.0] } else { angles };
for (j, &a) in angles.iter().enumerate() {
points.push(at(q[0], q[1], a));
rim_points.insert((k, j), points.len() - 1);
}
for j in 0..angles.len() {
let a = angles[j];
let b = if j + 1 < angles.len() { angles[j + 1] } else { angles[0] + TAU };
let base = at(q[0], q[1], a);
let surface = frag(make_revolution(origin, w, &frag(make_line(base, base.add(w)), "line")?, b - a), "rim")?;
curves.push(ToolCurve {
curve: frag(surface.iso_curve_v(0.0), "rim_arc")?,
start: rim_points[&(k, j)],
end: rim_points[&(k, (j + 1) % angles.len())],
degenerate: false,
samples: Some((0..=LOFT_SAMPLES).map(|i| at(q[0], q[1], a + (b - a) * i as f64 / LOFT_SAMPLES as f64)).collect()),
});
rim_arcs.insert((k, j), curves.len() - 1);
}
rim_angles.insert(k, angles);
}
let arc_index = |k: usize, angle: f64| -> Result<usize, KernelRefusal> {
rim_angles[&k]
.iter()
.position(|&a| ((a - angle).rem_euclid(TAU)).min((angle - a).rem_euclid(TAU)) < 1e-9)
.ok_or_else(|| refuse("revolved_rims", "a patch corner is not on its rim's split".into()))
};
let sign_at = |piece: &OutlinePiece, psi: f64| g(piece, 0.5, psi) > 0.0;
for (f, &piece_index) in faces.iter().enumerate() {
let piece = &meridian[piece_index];
let below = (piece_index + count - 1) % count; let above = piece_index; let (m0, m1) = (piece_point(piece, 0.0), piece_point(piece, 1.0));
match &kinds[f] {
Kind::Ruled(roots) if roots.is_empty() => {
let flat = (m0[1] - m1[1]).abs() <= 1e-12 * scale;
if !flat || !(on_axis(m0) || on_axis(m1)) {
return Err(refuse("revolved_side", format!("meridian piece {piece_index}: a whole side or ring is not swept here")));
}
let rim = if on_axis(m0) { above } else { below };
let front = sign_at(piece, 0.0);
if (0..64).any(|k| sign_at(piece, TAU * k as f64 / 64.0) != front) {
return Err(refuse("revolved_side", format!("meridian piece {piece_index}: a flat end meets its grazing line")));
}
let n = normal(piece, 0.5);
let outward = w.scale(n[1].signum());
let centre = origin.add(w.scale(m0[1]));
let (u, v) = (u1, outward.cross(u1));
let arcs = rim_angles[&rim].len();
let mut coedges = Vec::with_capacity(arcs);
let mut corners = Vec::new();
let order: Vec<usize> = if n[1] > 0.0 { (0..arcs).collect() } else { (0..arcs).rev().collect() };
for j in order {
let c = rim_arcs[&(rim, j)];
let forward = n[1] > 0.0;
coedges.push((c, forward, plane_pcurve(centre, u, v, &curves[c].curve, forward)?));
corners.push(points[curves[c].start]);
}
let r = junction(rim)[0];
corners.extend([centre.add(u.scale(r)).add(v.scale(r)), centre.sub(u.scale(r)).sub(v.scale(r))]);
caps.push(Cap { surface: plane_patch(centre, u, v, &corners)?, front, coedges });
}
Kind::Ruled(roots) => {
if on_axis(m0) || on_axis(m1) {
return Err(refuse("revolved_side", format!("meridian piece {piece_index}: a split cone to a point is not swept here")));
}
let mut generators = Vec::with_capacity(roots.len());
for &root in roots {
let (a, b) = (rim_points[&(below, arc_index(below, root)?)], rim_points[&(above, arc_index(above, root)?)]);
let line_samples = (0..=LOFT_SAMPLES).map(|i| points[a].add(points[b].sub(points[a]).scale(i as f64 / LOFT_SAMPLES as f64))).collect();
curves.push(ToolCurve { curve: frag(make_line(points[a], points[b]), "generator")?, start: a, end: b, degenerate: false, samples: Some(line_samples) });
generators.push(curves.len() - 1);
}
for j in 0..roots.len() {
let (alpha, beta) = (roots[j], if j + 1 < roots.len() { roots[j + 1] } else { roots[0] + TAU });
let generatrix = frag(make_line(at(m0[0], m0[1], alpha), at(m1[0], m1[1], alpha)), "generatrix")?;
let surface = frag(make_revolution(origin, w, &generatrix, beta - alpha), "patch")?;
let front = sign_at(piece, 0.5 * (alpha + beta));
let (lower, upper) = (rim_arcs[&(below, arc_index(below, alpha)?)], rim_arcs[&(above, arc_index(above, alpha)?)]);
let (g_alpha, g_beta) = (generators[j], generators[(j + 1) % roots.len()]);
let coedges = vec![
(lower, true, uv_line([0.0, 0.0], [1.0, 0.0])?),
(g_beta, true, uv_line([1.0, 0.0], [1.0, 1.0])?),
(upper, false, uv_line([1.0, 1.0], [0.0, 1.0])?),
(g_alpha, false, uv_line([0.0, 1.0], [0.0, 0.0])?),
];
caps.push(Cap { surface, front, coedges });
}
}
Kind::Moving => {
if on_axis(m0) || on_axis(m1) {
return Err(refuse("revolved_side", format!("meridian piece {piece_index}: a split cone to a point is not swept here")));
}
let n = 128;
let params: Vec<f64> = (0..=n).map(|i| i as f64 / n as f64).collect();
let pairs: Vec<(f64, f64)> = params.iter().map(|&t| branches(piece, t).expect("checked")).collect();
let pick = |pair: (f64, f64), which: usize| if which == 0 { pair.0 } else { pair.1 };
let mut branch_curves = [0usize; 2];
for which in 0..2 {
let (psi0, psi1) = (pick(pairs[0], which), pick(pairs[n], which));
let start = rim_points[&(below, arc_index(below, psi0)?)];
let end = rim_points[&(above, arc_index(above, psi1)?)];
let mut samples: Vec<Vec3> = params
.iter()
.zip(&pairs)
.map(|(&t, &pair)| {
let m = piece_point(piece, t);
at(m[0], m[1], pick(pair, which))
})
.collect();
samples[0] = points[start];
samples[n] = points[end];
let mut exact: Vec<Vec3> = (0..=LOFT_SAMPLES)
.map(|i| {
let t = i as f64 / LOFT_SAMPLES as f64;
let m = piece_point(piece, t);
at(m[0], m[1], pick(branches(piece, t).expect("checked"), which))
})
.collect();
exact[0] = points[start];
exact[LOFT_SAMPLES] = points[end];
curves.push(ToolCurve { curve: frag(crate::interpolate_curve(&samples, 5, ¶ms), "branch")?, start, end, degenerate: false, samples: Some(exact) });
branch_curves[which] = curves.len() - 1;
}
for (lo_which, hi_which, hi_turn) in [(0usize, 1usize, 0.0), (1, 0, TAU)] {
let lo = |i: usize| pick(pairs[i], lo_which);
let hi = |i: usize| pick(pairs[i], hi_which) + hi_turn;
let alpha = (0..=n).map(lo).fold(f64::INFINITY, f64::min) - 1e-3;
let theta = (0..=n).map(hi).fold(f64::NEG_INFINITY, f64::max) + 1e-3 - alpha;
if theta >= TAU {
return Err(refuse("revolved_side", format!("meridian piece {piece_index}: a side patch wraps all the way round")));
}
let generatrix = frag(make_line(at(m0[0], m0[1], alpha), at(m1[0], m1[1], alpha)), "generatrix")?;
let surface = frag(make_revolution(origin, w, &generatrix, theta), "patch")?;
let spans = (surface.knots_u.len() - 6) / 2 + 1;
let u_of = |psi: f64| crate::circle_angle_to_parameter(spans, theta, psi - alpha);
let angle_near = |x: Vec3, guess: f64| -> f64 {
let r = x.sub(origin);
let a = r.dot(u2).atan2(r.dot(u1));
guess + (a - guess + PI).rem_euclid(TAU) - PI
};
let front = sign_at(piece, 0.5 * (lo(n / 2) + hi(n / 2)));
let rim_pcurve = |arc: usize, v: f64| -> Result<NurbsCurve, KernelRefusal> {
let curve = &curves[arc].curve;
let [d0, d1] = frag(curve.domain(), "domain")?;
let ts: Vec<f64> = (0..=PCURVE_SAMPLES).map(|j| d0 + (d1 - d0) * j as f64 / PCURVE_SAMPLES as f64).collect();
let mut uv = Vec::with_capacity(ts.len());
for &t in &ts {
let x = frag(curve.evaluate(t), "rim")?;
uv.push(Vec3::new(u_of(angle_near(x, alpha + 0.5 * theta)), v, 0.0));
}
frag(crate::interpolate_curve(&uv, 5, &ts), "rim_pcurve")
};
let branch_pcurve = |which: usize, _turn: f64| -> Result<NurbsCurve, KernelRefusal> {
let curve = &curves[branch_curves[which]].curve;
let [d0, d1] = frag(curve.domain(), "domain")?;
let (span_rho, span_z) = (m1[0] - m0[0], m1[1] - m0[1]);
let length2 = span_rho * span_rho + span_z * span_z;
let ts: Vec<f64> = (0..=PCURVE_SAMPLES).map(|j| d0 + (d1 - d0) * j as f64 / PCURVE_SAMPLES as f64).collect();
let mut uv = Vec::with_capacity(ts.len());
for &t in &ts {
let x = frag(curve.evaluate(t), "branch")?;
let r = x.sub(origin);
let z = r.dot(w);
let rho = r.sub(w.scale(z)).length();
let along = ((rho - m0[0]) * span_rho + (z - m0[1]) * span_z) / length2;
uv.push(Vec3::new(u_of(angle_near(x, alpha + 0.5 * theta)), along, 0.0));
}
frag(crate::interpolate_curve(&uv, 5, &ts), "branch_pcurve")
};
let bottom = rim_arcs[&(below, arc_index(below, lo(0))?)];
let top = rim_arcs[&(above, arc_index(above, lo(n))?)];
let coedges = vec![
(bottom, true, rim_pcurve(bottom, 0.0)?),
(branch_curves[hi_which], true, branch_pcurve(hi_which, hi_turn)?),
(top, false, frag(rim_pcurve(top, 1.0)?.reversed(), "reversed")?),
(branch_curves[lo_which], false, frag(branch_pcurve(lo_which, 0.0)?.reversed(), "reversed")?),
];
caps.push(Cap { surface, front, coedges });
}
}
Kind::TorusZone { psi_a, psi_b, at_end } => {
let OutlinePiece::Arc { centre, radius, start, end } = *piece else { unreachable!() };
let (psi_a, psi_b, at_end) = (*psi_a, *psi_b, *at_end);
let tube_centre = |psi: f64| at(centre[0], centre[1], psi);
let tube_axis = |psi: f64| u1.scale(psi.cos()).add(u2.scale(psi.sin())).cross(w);
let tube_point = |psi: f64, v: f64| at(centre[0] + radius * v.cos(), centre[1] + radius * v.sin(), psi);
let normal_at = |psi: f64, v: f64| w.scale(v.sin()).add(u1.scale(v.cos() * psi.cos())).add(u2.scale(v.cos() * psi.sin()));
let root = |psi: f64| -> Option<f64> {
let (a, b, c) = harmonic(|v| screw.velocity(tube_point(psi, v)).dot(normal_at(psi, v)));
let r = a.hypot(b);
if !(c.abs() < r) {
return None;
}
let base = b.atan2(a);
let spread = (-c / r).acos();
[base - spread, base + spread].into_iter().map(|v| start + (v - start).rem_euclid(TAU)).find(|&v| v <= end + 1e-12)
};
let generatrix = |psi: f64| -> Result<NurbsCurve, KernelRefusal> {
let from = tube_point(psi, start);
let surface = frag(make_revolution(tube_centre(psi), tube_axis(psi), &frag(make_line(from, from.add(tube_axis(psi))), "line")?, end - start), "tube")?;
frag(surface.iso_curve_v(0.0), "tube")
};
let (rim_k, other_k) = if at_end { (above, below) } else { (below, above) };
let ka = rim_points[&(rim_k, arc_index(rim_k, psi_a)?)];
let kb = rim_points[&(rim_k, arc_index(rim_k, psi_b)?)];
let n = 256;
let params: Vec<f64> = (0..=n).map(|j| j as f64 / n as f64).collect();
let psi_of = |t: f64| psi_a + (psi_b - psi_a) * 0.5 * (1.0 - (PI * t).cos());
let mut xyz = Vec::with_capacity(n + 1);
for j in 0..=n {
let psi = psi_of(params[j]);
let v = if j == 0 || j == n { if at_end { end } else { start } } else {
root(psi).ok_or_else(|| refuse("torus_zone", "the grazing curve leaves the rounded corner".into()))?
};
xyz.push(tube_point(psi, v));
}
xyz[0] = points[ka];
xyz[n] = points[kb];
let mut exact = Vec::with_capacity(LOFT_SAMPLES + 1);
for i in 0..=LOFT_SAMPLES {
let psi = psi_of(i as f64 / LOFT_SAMPLES as f64);
let v = if i == 0 || i == LOFT_SAMPLES { if at_end { end } else { start } } else {
root(psi).ok_or_else(|| refuse("torus_zone", "the grazing curve leaves the rounded corner".into()))?
};
exact.push(tube_point(psi, v));
}
exact[0] = points[ka];
exact[LOFT_SAMPLES] = points[kb];
curves.push(ToolCurve { curve: frag(crate::interpolate_curve(&xyz, 5, ¶ms), "torus_k")?, start: ka, end: kb, degenerate: false, samples: Some(exact) });
let k_curve = curves.len() - 1;
let mut cut = [0usize; 2];
for (slot, psi) in [(0, psi_a), (1, psi_b)] {
let (lo, hi) = (rim_points[&(below, arc_index(below, psi)?)], rim_points[&(above, arc_index(above, psi)?)]);
curves.push(ToolCurve { curve: generatrix(psi)?, start: lo, end: hi, degenerate: false, samples: None });
cut[slot] = curves.len() - 1;
}
let label = |psi: f64, v: f64| screw.velocity(tube_point(psi, v)).dot(normal_at(psi, v)) > 0.0;
let patch = |alpha: f64, beta: f64| -> Result<(NurbsSurface, usize, usize), KernelRefusal> {
let g = generatrix(alpha)?;
let surface = frag(make_revolution(origin, w, &g, beta - alpha), "torus_patch")?;
let spans_u = (surface.knots_u.len() - 6) / 2 + 1;
let spans_v = (g.knots.len() - 6) / 2 + 1;
Ok((surface, spans_u, spans_v))
};
let inverse_pcurve = |curve: &NurbsCurve, alpha: f64, beta: f64, spans_u: usize, spans_v: usize, forward: bool| -> Result<NurbsCurve, KernelRefusal> {
let [d0, d1] = frag(curve.domain(), "domain")?;
let ts: Vec<f64> = (0..=PCURVE_SAMPLES).map(|j| d0 + (d1 - d0) * j as f64 / PCURVE_SAMPLES as f64).collect();
let mut uv = Vec::with_capacity(ts.len());
for &t in &ts {
let x = frag(curve.evaluate(t), "point")?;
let r = x.sub(origin);
let mid = 0.5 * (alpha + beta);
let psi = mid + (r.dot(u2).atan2(r.dot(u1)) - mid + PI).rem_euclid(TAU) - PI;
let q = x.sub(tube_centre(psi));
let radial = u1.scale(psi.cos()).add(u2.scale(psi.sin()));
let vmid = 0.5 * (start + end);
let v = vmid + (q.dot(w).atan2(q.dot(radial)) - vmid + PI).rem_euclid(TAU) - PI;
uv.push(Vec3::new(
crate::circle_angle_to_parameter(spans_u, beta - alpha, psi - alpha),
crate::circle_angle_to_parameter(spans_v, end - start, v - start),
0.0,
));
}
let fitted = frag(crate::interpolate_curve(&uv, 5, &ts), "torus_pcurve")?;
if forward { Ok(fitted) } else { frag(fitted.reversed(), "reversed") }
};
{
let (surface, _, _) = patch(psi_b, psi_a + TAU)?;
let (bottom, top) = (rim_arcs[&(below, arc_index(below, psi_b)?)], rim_arcs[&(above, arc_index(above, psi_b)?)]);
let coedges = vec![
(bottom, true, uv_line([0.0, 0.0], [1.0, 0.0])?),
(cut[0], true, uv_line([1.0, 0.0], [1.0, 1.0])?),
(top, false, uv_line([1.0, 1.0], [0.0, 1.0])?),
(cut[1], false, uv_line([0.0, 1.0], [0.0, 0.0])?),
];
let mid = psi_b + 0.5 * (psi_a + TAU - psi_b);
caps.push(Cap { surface, front: label(mid, 0.5 * (start + end)), coedges });
}
let (surface, spans_u, spans_v) = patch(psi_a, psi_b)?;
let rim_arc = rim_arcs[&(rim_k, arc_index(rim_k, psi_a)?)];
let other_arc = rim_arcs[&(other_k, arc_index(other_k, psi_a)?)];
let mid = 0.5 * (psi_a + psi_b);
let v_k = root(mid).ok_or_else(|| refuse("torus_zone", "no grazing point mid-way".into()))?;
let k_pc = |forward: bool| inverse_pcurve(&curves[k_curve].curve, psi_a, psi_b, spans_u, spans_v, forward);
let lens_v = if at_end { 0.5 * (v_k + end) } else { 0.5 * (v_k + start) };
let rest_v = if at_end { 0.5 * (v_k + start) } else { 0.5 * (v_k + end) };
if at_end {
let lens = vec![(k_curve, true, k_pc(true)?), (rim_arc, false, uv_line([1.0, 1.0], [0.0, 1.0])?)];
caps.push(Cap { surface: surface.clone(), front: label(mid, lens_v), coedges: lens });
let rest = vec![
(other_arc, true, uv_line([0.0, 0.0], [1.0, 0.0])?),
(cut[1], true, uv_line([1.0, 0.0], [1.0, 1.0])?),
(k_curve, false, k_pc(false)?),
(cut[0], false, uv_line([0.0, 1.0], [0.0, 0.0])?),
];
caps.push(Cap { surface, front: label(mid, rest_v), coedges: rest });
} else {
let lens = vec![(rim_arc, true, uv_line([0.0, 0.0], [1.0, 0.0])?), (k_curve, false, k_pc(false)?)];
caps.push(Cap { surface: surface.clone(), front: label(mid, lens_v), coedges: lens });
let rest = vec![
(k_curve, true, k_pc(true)?),
(cut[1], true, uv_line([1.0, 0.0], [1.0, 1.0])?),
(other_arc, false, uv_line([1.0, 1.0], [0.0, 1.0])?),
(cut[0], false, uv_line([0.0, 1.0], [0.0, 0.0])?),
];
caps.push(Cap { surface, front: label(mid, rest_v), coedges: rest });
}
}
Kind::Ball { centre, radius } => {
let m = frag(screw.velocity(*centre).normalized(), "ball_still")?;
let d = frag(w.cross(m).normalized().or_else(|_| u1.cross(m).normalized()), "ball_axis")?;
let (e1, e2) = (centre.add(d.scale(*radius)), centre.sub(d.scale(*radius)));
points.push(e1);
points.push(e2);
let (i1, i2) = (points.len() - 2, points.len() - 1);
let half = frag(make_revolution(*centre, m, &frag(make_line(e1, e1.add(m)), "line")?, PI), "great")?;
let k_a = frag(half.iso_curve_v(0.0), "great")?;
curves.push(ToolCurve { curve: k_a.clone(), start: i1, end: i2, degenerate: false, samples: None });
let k_a_index = curves.len() - 1;
let [a0, a1] = frag(k_a.domain(), "domain")?;
let h = frag(k_a.evaluate(0.5 * (a0 + a1)), "middle")?.sub(*centre);
let mut k_b_index: Option<usize> = None;
for front in [true, false] {
let toward = if front { m } else { m.scale(-1.0) };
let da = if d.cross(h).dot(toward) > 0.0 { d } else { d.scale(-1.0) };
let mut generatrix = k_a.clone();
let mut surface = frag(make_revolution(*centre, da, &generatrix, PI), "hemisphere")?;
let outward = |s: &NurbsSurface| -> Result<bool, KernelRefusal> {
let p = frag(s.evaluate(0.5, 0.5), "probe")?;
let su = frag(s.evaluate(0.5 + 1e-6, 0.5), "probe")?.sub(frag(s.evaluate(0.5 - 1e-6, 0.5), "probe")?);
let sv = frag(s.evaluate(0.5, 0.5 + 1e-6), "probe")?.sub(frag(s.evaluate(0.5, 0.5 - 1e-6), "probe")?);
Ok(su.cross(sv).dot(p.sub(*centre)) > 0.0)
};
let mut forward = true;
if !outward(&surface)? {
generatrix = frag(generatrix.reversed(), "reversed")?;
surface = frag(make_revolution(*centre, da, &generatrix, PI), "hemisphere")?;
forward = false;
}
let (start, end) = if forward { (i1, i2) } else { (i2, i1) };
let [g0, g1] = frag(generatrix.domain(), "domain")?;
let k_b = match k_b_index {
Some(index) => index,
None => {
curves.push(ToolCurve { curve: frag(surface.iso_curve_u(1.0), "great")?, start, end, degenerate: false, samples: None });
k_b_index = Some(curves.len() - 1);
curves.len() - 1
}
};
let k_b_forward = curves[k_b].start == start;
let k_b_pcurve = uv_line([1.0, g0], [1.0, g1])?;
let poles = [start, end].map(|p| {
curves.push(ToolCurve { curve: make_line(points[p], points[p]).expect("point"), start: p, end: p, degenerate: true, samples: None });
curves.len() - 1
});
let coedges = vec![
(poles[0], true, uv_line([0.0, g0], [1.0, g0])?),
(k_b, k_b_forward, k_b_pcurve),
(poles[1], true, uv_line([1.0, g1], [0.0, g1])?),
(k_a_index, !forward, uv_line([0.0, g1], [0.0, g0])?),
];
caps.push(Cap { surface, front, coedges });
}
}
Kind::Sphere { centre, pole_at_start } => {
let rim = if *pole_at_start { above } else { below };
let angles = rim_angles[&rim].clone();
if angles.len() != 2 {
return Err(refuse("revolved_rims", "a hemisphere's rim is not split in two".into()));
}
let pole_side = w.scale(if *pole_at_start { m0[1] - m1[1] } else { m1[1] - m0[1] }.signum());
let m = screw.velocity(*centre);
let e1 = points[rim_points[&(rim, 0)]];
let d = frag(e1.sub(*centre).normalized(), "lune_axis")?;
let mut great: Option<usize> = None;
for j in 0..2 {
let arc = rim_arcs[&(rim, j)];
let [a0, a1] = frag(curves[arc].curve.domain(), "domain")?;
let middle = frag(curves[arc].curve.evaluate(0.5 * (a0 + a1)), "middle")?;
let hv = middle.sub(*centre);
let da = if d.cross(hv).dot(pole_side) > 0.0 { d } else { d.scale(-1.0) };
let theta = (-m.dot(hv)).atan2(m.dot(da.cross(hv))).rem_euclid(TAU);
let theta = if theta > PI { theta - PI } else { theta };
if !(theta > 1e-9 && theta < PI - 1e-9) {
return Err(refuse("revolved_lune", "the great circle meets the rim's plane edge-on".into()));
}
let mut generatrix = curves[arc].curve.clone();
let mut surface = frag(make_revolution(*centre, da, &generatrix, theta), "lune")?;
let probe = |s: &NurbsSurface| -> Result<bool, KernelRefusal> {
let p = frag(s.evaluate(0.5, 0.5), "probe")?;
let su = frag(s.evaluate(0.5 + 1e-6, 0.5), "probe")?.sub(frag(s.evaluate(0.5 - 1e-6, 0.5), "probe")?);
let sv = frag(s.evaluate(0.5, 0.5 + 1e-6), "probe")?.sub(frag(s.evaluate(0.5, 0.5 - 1e-6), "probe")?);
Ok(su.cross(sv).dot(p.sub(*centre)) > 0.0)
};
let mut forward = true;
if !probe(&surface)? {
generatrix = frag(generatrix.reversed(), "reversed")?;
surface = frag(make_revolution(*centre, da, &generatrix, theta), "lune")?;
forward = false;
}
let (start, end) = if forward { (curves[arc].start, curves[arc].end) } else { (curves[arc].end, curves[arc].start) };
let [g0, g1] = frag(generatrix.domain(), "domain")?;
let k_curve = match great {
Some(index) => index,
None => {
curves.push(ToolCurve { curve: frag(surface.iso_curve_u(1.0), "great")?, start, end, degenerate: false, samples: None });
great = Some(curves.len() - 1);
curves.len() - 1
}
};
let k_forward = curves[k_curve].start == start;
let poles = [start, end].map(|p| {
curves.push(ToolCurve { curve: make_line(points[p], points[p]).expect("point"), start: p, end: p, degenerate: true, samples: None });
curves.len() - 1
});
let k_pcurve = uv_line([1.0, g0], [1.0, g1])?;
let coedges = vec![
(poles[0], true, uv_line([0.0, g0], [1.0, g0])?),
(k_curve, k_forward, k_pcurve),
(poles[1], true, uv_line([1.0, g1], [0.0, g1])?),
(arc, !forward, uv_line([0.0, g1], [0.0, g0])?),
];
let mid_psi = if j == 0 { 0.5 * (angles[0] + angles[1]) } else { 0.5 * (angles[1] + angles[0] + TAU) };
let front = m.dot(at(1.0, 0.0, mid_psi).sub(origin)) > 0.0;
caps.push(Cap { surface, front, coedges });
}
}
}
}
Ok(Split {
points,
curves,
caps,
scale,
shape: Shape::Revolved { origin, axis: w, meridian: meridian.to_vec() },
})
}
#[allow(clippy::too_many_arguments)]
fn split_torus(
screw: &Screw,
origin: Vec3,
w: Vec3,
u1: Vec3,
u2: Vec3,
centre: [f64; 2],
radius: f64,
meridian: &[OutlinePiece],
) -> Result<Split, KernelRefusal> {
let (rho_c, z_c) = (centre[0], centre[1]);
let radial = |psi: f64| u1.scale(psi.cos()).add(u2.scale(psi.sin()));
let tube_centre = |psi: f64| origin.add(w.scale(z_c)).add(radial(psi).scale(rho_c));
let point = |psi: f64, v: f64| tube_centre(psi).add(radial(psi).scale(radius * v.cos())).add(w.scale(radius * v.sin()));
let g = |psi: f64, v: f64| screw.velocity(point(psi, v)).dot(radial(psi).scale(v.cos()).add(w.scale(v.sin())));
let reference = {
let (a, b, _) = harmonic(|v| g(0.0, v));
b.atan2(a)
};
let branches = |psi: f64| -> Option<(f64, f64)> {
let (a, b, c) = harmonic(|v| g(psi, v));
let r = a.hypot(b);
if !(c.abs() < r * (1.0 - 1e-9)) {
return None;
}
let base = reference + (b.atan2(a) - reference + PI).rem_euclid(TAU) - PI;
let spread = (-c / r).acos();
Some((base - spread, base + spread))
};
let n = 128;
let halves = [(0.0, PI), (PI, TAU)];
let mut samples: Vec<Vec<(f64, f64, f64)>> = Vec::new(); for &(a, b) in &halves {
let mut list = Vec::with_capacity(n + 1);
let mut previous: Option<(f64, f64)> = None;
for i in 0..=n {
let psi = a + (b - a) * i as f64 / n as f64;
let Some(pair) = branches(psi) else {
return Err(refuse("ring_grazing", "the ring's grazing curves do not run all the way round it".into()));
};
if previous.is_some_and(|p| (p.0 - pair.0).abs() > 0.5 || (p.1 - pair.1).abs() > 0.5) {
return Err(refuse("ring_grazing", "the ring's grazing curves turn too fast".into()));
}
previous = Some(pair);
list.push((psi, pair.0, pair.1));
}
samples.push(list);
}
let params: Vec<f64> = (0..=n).map(|i| i as f64 / n as f64).collect();
let mut points: Vec<Vec3> = Vec::new();
let mut curves: Vec<ToolCurve> = Vec::new();
let mut corner: HashMap<(usize, usize), usize> = HashMap::new(); for seam in 0..2 {
let (psi, (lower, upper)) = if seam == 0 { (0.0, (samples[0][0].1, samples[0][0].2)) } else { (PI, (samples[1][0].1, samples[1][0].2)) };
for (branch, v) in [(0, lower), (1, upper)] {
points.push(point(psi, v));
corner.insert((seam, branch), points.len() - 1);
}
}
let seam_of_end = |half: usize, at_end: bool| -> usize { if at_end { 1 - half } else { half } };
let mut branch_curve: HashMap<(usize, usize), usize> = HashMap::new(); for half in 0..2 {
for branch in 0..2 {
let mut xyz: Vec<Vec3> = samples[half].iter().map(|&(psi, lo, hi)| point(psi, if branch == 0 { lo } else { hi })).collect();
let (start, end) = (corner[&(seam_of_end(half, false), branch)], corner[&(seam_of_end(half, true), branch)]);
xyz[0] = points[start];
xyz[n] = points[end];
curves.push(ToolCurve { curve: frag(crate::interpolate_curve(&xyz, 5, ¶ms), "ring_branch")?, start, end, degenerate: false, samples: None });
branch_curve.insert((half, branch), curves.len() - 1);
}
}
let tube_arc = |psi: f64, v0: f64, v1: f64| -> Result<NurbsCurve, KernelRefusal> {
let start = point(psi, v0);
let axis = radial(psi).cross(w);
let surface = frag(make_revolution(tube_centre(psi), axis, &frag(make_line(start, start.add(axis)), "line")?, v1 - v0), "tube_arc")?;
frag(surface.iso_curve_v(0.0), "tube_arc")
};
let mut seam_arc: HashMap<(usize, usize), usize> = HashMap::new(); for seam in 0..2 {
let psi = if seam == 0 { 0.0 } else { PI };
let (lower, upper) = (samples[seam][0].1, samples[seam][0].2);
let (lo_point, hi_point) = (corner[&(seam, 0)], corner[&(seam, 1)]);
curves.push(ToolCurve { curve: tube_arc(psi, lower, upper)?, start: lo_point, end: hi_point, degenerate: false, samples: None });
seam_arc.insert((seam, 0), curves.len() - 1);
curves.push(ToolCurve { curve: tube_arc(psi, upper, lower + TAU)?, start: hi_point, end: lo_point, degenerate: false, samples: None });
seam_arc.insert((seam, 1), curves.len() - 1);
}
let mut caps = Vec::new();
for half in 0..2 {
let (psi_a, psi_b) = halves[half];
for band in 0..2 {
let (lo_branch, hi_branch, hi_turn) = if band == 0 { (0usize, 1usize, 0.0) } else { (1, 0, TAU) };
let pick = |s: &(f64, f64, f64), b: usize| if b == 0 { s.1 } else { s.2 };
let v_low = samples[half].iter().map(|s| pick(s, lo_branch)).fold(f64::INFINITY, f64::min) - 1e-3;
let v_high = samples[half].iter().map(|s| pick(s, hi_branch) + hi_turn).fold(f64::NEG_INFINITY, f64::max) + 1e-3;
let theta_v = v_high - v_low;
if theta_v >= TAU {
return Err(refuse("ring_grazing", "a band of the ring wraps all the way round its tube".into()));
}
let generatrix = tube_arc(psi_a, v_low, v_high)?;
let surface = frag(make_revolution(origin, w, &generatrix, psi_b - psi_a), "ring_band")?;
let spans_u = (surface.knots_u.len() - 6) / 2 + 1;
let spans_v = (generatrix.knots.len() - 6) / 2 + 1;
let uv_of = |psi: f64, v: f64| {
Vec3::new(
crate::circle_angle_to_parameter(spans_u, psi_b - psi_a, psi - psi_a),
crate::circle_angle_to_parameter(spans_v, theta_v, v - v_low),
0.0,
)
};
let angles_of = |x: Vec3| -> (f64, f64) {
let r = x.sub(origin);
let psi = r.dot(u2).atan2(r.dot(u1));
let psi = 0.5 * (psi_a + psi_b) + (psi - 0.5 * (psi_a + psi_b) + PI).rem_euclid(TAU) - PI;
let q = x.sub(tube_centre(psi));
let v = q.dot(w).atan2(q.dot(radial(psi)));
let mid = 0.5 * (v_low + v_high);
(psi, mid + (v - mid + PI).rem_euclid(TAU) - PI)
};
let inverse_pcurve = |curve: &NurbsCurve| -> Result<NurbsCurve, KernelRefusal> {
let [d0, d1] = frag(curve.domain(), "domain")?;
let ts: Vec<f64> = (0..=PCURVE_SAMPLES).map(|j| d0 + (d1 - d0) * j as f64 / PCURVE_SAMPLES as f64).collect();
let mut uv = Vec::with_capacity(ts.len());
for &t in &ts {
let (psi, v) = angles_of(frag(curve.evaluate(t), "point")?);
uv.push(uv_of(psi, v));
}
frag(crate::interpolate_curve(&uv, 5, &ts), "ring_pcurve")
};
let branch_pcurve = |b: usize, _turn: f64| -> Result<NurbsCurve, KernelRefusal> { inverse_pcurve(&curves[branch_curve[&(half, b)]].curve) };
let lower_edge = branch_curve[&(half, lo_branch)];
let upper_edge = branch_curve[&(half, hi_branch)];
let right = seam_arc[&(seam_of_end(half, true), band)];
let left = seam_arc[&(seam_of_end(half, false), band)];
let coedges = vec![
(lower_edge, true, branch_pcurve(lo_branch, 0.0)?),
(right, true, inverse_pcurve(&curves[right].curve)?),
(upper_edge, false, frag(branch_pcurve(hi_branch, hi_turn)?.reversed(), "reversed")?),
(left, false, frag(inverse_pcurve(&curves[left].curve)?.reversed(), "reversed")?),
];
let mid = &samples[half][n / 2];
let front = g(mid.0, 0.5 * (pick(mid, lo_branch) + pick(mid, hi_branch) + hi_turn)) > 0.0;
caps.push(Cap { surface, front, coedges });
}
}
Ok(Split {
points,
curves,
caps,
scale: rho_c + radius,
shape: Shape::Revolved { origin, axis: w, meridian: meridian.to_vec() },
})
}
struct Shell {
vertices: Vec<VertexRecord>,
edges: Vec<EdgeRecord>,
faces: Vec<FaceRecord>,
next: u64,
}
impl Shell {
fn new() -> Self {
Self { vertices: Vec::new(), edges: Vec::new(), faces: Vec::new(), next: 1 }
}
fn id(&mut self) -> u64 {
self.next += 1;
self.next
}
fn vertex(&mut self, point: Vec3) -> u64 {
let id = self.id();
self.vertices.push(VertexRecord { id, point });
id
}
fn edge(&mut self, curve: NurbsCurve, start: u64, end: u64, degenerate: bool) -> Result<u64, KernelRefusal> {
let domain = frag(curve.domain(), "edge_domain")?;
let id = self.id();
self.edges.push(EdgeRecord { id, curve, t0: domain[0], t1: domain[1], start_vertex_id: start, end_vertex_id: end, degenerate, name: None });
Ok(id)
}
fn face(&mut self, surface: NurbsSurface, coedges: Vec<(u64, bool, NurbsCurve)>) {
let mut records = Vec::with_capacity(coedges.len());
for (edge, forward, pcurve) in coedges {
let id = self.id();
records.push(CoedgeRecord { id, edge_id: edge, forward, pcurve });
}
let loop_id = self.id();
let id = self.id();
self.faces.push(FaceRecord { id, surface, same_sense: true, loops: vec![LoopRecord { id: loop_id, coedges: records }], name: None });
}
fn solid(mut self) -> BrepSolid {
let shell = self.id();
let id = self.id();
let edges = self.edges.iter().filter(|e| !e.degenerate).count() as i64;
let euler = self.vertices.len() as i64 - edges + self.faces.len() as i64;
let genus = (2 - euler) / 2;
BrepSolid {
mass_properties_cache: Default::default(),
id,
vertices: self.vertices,
edges: self.edges,
shells: vec![ShellRecord { id: shell, faces: self.faces }],
genus,
kind: Default::default(),
cells: Default::default(),
}
}
}
pub fn sweep_solid(tool: &SolidTool, motion: &SolidMotion) -> Result<BrepSolid, KernelRefusal> {
sweep_with(&Screw::from_motion(motion)?, tool)
}
fn sweep_with(screw: &Screw, tool: &SolidTool) -> Result<BrepSolid, KernelRefusal> {
let screw = *screw;
if let SolidTool::Revolved { axis_origin, axis_direction, meridian, .. } = tool {
if let Some(body) = ball_across_axis(&screw, *axis_origin, *axis_direction, meridian)? {
return Ok(body);
}
}
let split = match tool {
SolidTool::Polyhedron { vertices, faces } => split_polyhedron(&screw, vertices, faces)?,
SolidTool::Revolved { axis_origin, axis_direction, radial, meridian } => {
split_revolved(&screw, *axis_origin, *axis_direction, *radial, meridian)?
}
};
let grazing = split.grazing()?;
if grazing.is_empty() {
return Err(refuse("no_grazing", "the tool has no grazing loop under this motion".into()));
}
let times = row_times(&screw);
let last = times.len() - 1;
let mut shell = Shell::new();
let mut vertex_at: HashMap<(usize, usize), u64> = HashMap::new();
let mut edge_at: HashMap<(usize, usize), u64> = HashMap::new();
let mut at_time = |shell: &mut Shell, j: usize, p: usize| -> u64 {
*vertex_at.entry((j, p)).or_insert_with(|| shell.vertex(screw.apply(times[j], split.points[p])))
};
let mut time_edge = |shell: &mut Shell, j: usize, c: usize, vertex: &mut dyn FnMut(&mut Shell, usize, usize) -> u64| -> Result<u64, KernelRefusal> {
if let Some(&edge) = edge_at.get(&(j, c)) {
return Ok(edge);
}
let curve = &split.curves[c];
let (a, b) = (vertex(shell, j, curve.start), vertex(shell, j, curve.end));
let edge = shell.edge(screw.move_curve(times[j], &curve.curve)?, a, b, curve.degenerate)?;
edge_at.insert((j, c), edge);
Ok(edge)
};
for cap in &split.caps {
let j = if cap.front { last } else { 0 };
let surface = screw.move_surface(times[j], &cap.surface)?;
let mut coedges = Vec::with_capacity(cap.coedges.len());
for (c, forward, pcurve) in &cap.coedges {
let edge = time_edge(&mut shell, j, *c, &mut at_time)?;
coedges.push((edge, *forward, pcurve.clone()));
}
shell.face(surface, coedges);
}
for row in 0..last {
let (ta, tb) = (times[row], times[row + 1]);
let segments = carry_segments(&screw, ta, tb);
let knots_v = carry_knots(segments);
let mut orbits: HashMap<usize, u64> = HashMap::new();
for &(c, _) in &grazing {
for r in [split.curves[c].start, split.curves[c].end] {
if orbits.contains_key(&r) {
continue;
}
let (v0, v1) = (at_time(&mut shell, row, r), at_time(&mut shell, row + 1, r));
let edge = shell.edge(carry_curve(&screw, split.points[r], ta, tb, segments)?, v0, v1, false)?;
orbits.insert(r, edge);
}
}
for &(c, back_forward) in &grazing {
let tool_curve = &split.curves[c];
let along = if back_forward { frag(tool_curve.curve.reversed(), "reversed")? } else { tool_curve.curve.clone() };
let (from, to) = if back_forward { (tool_curve.end, tool_curve.start) } else { (tool_curve.start, tool_curve.end) };
let mut rows = Vec::with_capacity(along.control_points.len());
for p in &along.control_points {
let x = frag(p.point(), "control_point")?;
rows.push(carry_points(&screw, x, ta, tb, segments).into_iter().map(|q| Vec4::from_point(q, p.w)).collect::<Vec<_>>());
}
let surface = frag(NurbsSurface::new(along.degree, 5, along.knots.clone(), knots_v.clone(), rows), "wall")?;
let [u0, u1] = frag(along.domain(), "domain")?;
let bottom = time_edge(&mut shell, row, c, &mut at_time)?;
let top = time_edge(&mut shell, row + 1, c, &mut at_time)?;
let coedges = vec![
(bottom, !back_forward, uv_line([u0, 0.0], [u1, 0.0])?),
(orbits[&to], true, uv_line([u1, 0.0], [u1, 1.0])?),
(top, back_forward, uv_line([u1, 1.0], [u0, 1.0])?),
(orbits[&from], false, uv_line([u0, 1.0], [u0, 0.0])?),
];
shell.face(surface, coedges);
}
}
let body = shell.solid();
let issues = body.validate();
if !issues.is_empty() {
return Err(refuse("assembly", format!("the assembled sweep is not valid: {:?}", issues.first())));
}
check_single_pass(&screw, &split, &body)?;
Ok(body)
}
pub fn sweep_solid_clipped(tool: &SolidTool, motion: &SolidMotion, clip: &crate::SweepClip) -> Result<BrepSolid, KernelRefusal> {
let screw = Screw::from_motion(motion)?;
if let (SolidTool::Polyhedron { vertices, faces }, Some(wedge)) = (tool, clip.angle) {
let axis = clip.axis_direction.normalized().map_err(|_| input("clip_axis", "the clip's axis has no direction".into()))?;
let same_axis = screw.c == 0.0 && screw.c2 == 0.0
&& screw.e.cross(axis).length() < 1e-12
&& screw.a.sub(clip.axis_origin).cross(axis).length() < 1e-9 * (1.0 + clip.radius[1]);
if same_axis {
return turn_shadow(&screw, vertices, faces, clip, axis, wedge);
}
}
if let SolidTool::Polyhedron { vertices, faces } = tool {
if let Some((vertices, faces)) = trim_to_clip(&screw, vertices, faces, clip)? {
return sweep_solid(&SolidTool::Polyhedron { vertices, faces }, motion);
}
}
sweep_solid(tool, motion)
}
pub fn sweep_solid_rising(tool: &SolidTool, motion: &RisingTurn, clip: &crate::SweepClip) -> Result<BrepSolid, KernelRefusal> {
let screw = Screw::from_rising(motion)?;
let SolidTool::Polyhedron { vertices, faces } = tool else {
return Err(refuse("rising_tool", "a growing rise sweeps a polyhedron only".into()));
};
let (vertices, faces) = trim_to_clip(&screw, vertices, faces, clip)?.unwrap_or_else(|| (vertices.clone(), faces.clone()));
sweep_with(&screw, &SolidTool::Polyhedron { vertices, faces })
}
pub fn sweep_solid_parts(tool: &SolidTool, motion: &SolidMotion) -> Result<Vec<BrepSolid>, KernelRefusal> {
let folded = match sweep_solid(tool, motion) {
Ok(body) => return Ok(vec![body]),
Err(error) if error.message.contains("passes over a point twice") => error,
Err(error) => return Err(error),
};
let SolidTool::Polyhedron { vertices, faces } = tool else { return Err(folded) };
let screw = Screw::from_motion(motion)?;
if screw.omega == 0.0 {
return Err(folded);
}
let centroid = vertices.iter().fold(Vec3::new(0.0, 0.0, 0.0), |a, &p| a.add(p)).scale(1.0 / vertices.len() as f64);
let mut planes: Vec<(Vec3, f64)> = Vec::new();
for face in faces {
let (p, q, r) = (vertices[face[0]], vertices[face[1]], vertices[face[2]]);
let mut n = frag(q.sub(p).cross(r.sub(p)).normalized(), "face_normal")?;
if n.dot(p.sub(centroid)) < 0.0 {
n = n.scale(-1.0);
}
planes.push((n, n.dot(p)));
}
for (n, d) in planes.clone() {
if n.dot(screw.e).abs() > 1e-12 {
continue;
}
let foot = screw.a.add(n.scale(d - n.dot(screw.a)));
let radial = foot.sub(screw.a).sub(screw.e.scale(foot.sub(screw.a).dot(screw.e)));
if radial.dot(n) <= 0.0 {
continue; }
let cut = match screw.e.cross(radial).normalized() {
Ok(m) => m,
Err(_) => continue,
};
let offset = cut.dot(screw.a);
let mut parts = Vec::with_capacity(2);
for sense in [1.0, -1.0] {
let mut half = planes.clone();
half.push((cut.scale(sense), offset * sense));
let (corners, loops) = polyhedron_of_planes(&half)?;
if corners.len() < 4 || loops.len() < 4 {
parts.clear();
break;
}
match sweep_solid(&SolidTool::Polyhedron { vertices: corners, faces: loops }, motion) {
Ok(body) => parts.push(body),
Err(_) => {
parts.clear();
break;
}
}
}
if parts.len() == 2 {
return Ok(parts);
}
}
Err(folded)
}
fn trim_to_clip(screw: &Screw, vertices: &[Vec3], faces: &[Vec<usize>], clip: &crate::SweepClip) -> Result<Option<(Vec<Vec3>, Vec<Vec<usize>>)>, KernelRefusal> {
let Ok(axis) = clip.axis_direction.normalized() else { return Ok(None) };
let same_axis = screw.omega != 0.0
&& screw.e.cross(axis).length() < 1e-12
&& screw.a.sub(clip.axis_origin).cross(axis).length() < 1e-9 * (1.0 + clip.radius[1]);
if !same_axis || !clip.radius[1].is_finite() {
return Ok(None);
}
let ez = axis;
let ex = Vec3::new(1.0, 0.0, 0.0).sub(ez.scale(ez.x)).normalized().or_else(|_| Vec3::new(0.0, 1.0, 0.0).sub(ez.scale(ez.y)).normalized()).map_err(|e| input("clip_axis", e))?;
let ey = ez.cross(ex);
let flat = |p: Vec3| {
let r = p.sub(clip.axis_origin);
[r.dot(ex), r.dot(ey)]
};
let radius = clip.radius[1] * 1.02;
let shadow: Vec<[f64; 2]> = vertices.iter().map(|&p| flat(p)).collect();
if shadow.iter().all(|q| q[0].hypot(q[1]) <= radius) {
return Ok(None);
}
let mut points = shadow.clone();
points.sort_by(|a, b| a[0].total_cmp(&b[0]).then(a[1].total_cmp(&b[1])));
let cross = |o: [f64; 2], a: [f64; 2], b: [f64; 2]| (a[0] - o[0]) * (b[1] - o[1]) - (a[1] - o[1]) * (b[0] - o[0]);
let mut hull: Vec<[f64; 2]> = Vec::new();
for pass in 0..2 {
let start = hull.len();
let ordered: Vec<[f64; 2]> = if pass == 0 { points.clone() } else { points.iter().rev().copied().collect() };
for p in ordered {
while hull.len() >= start + 2 && cross(hull[hull.len() - 2], hull[hull.len() - 1], p) <= 0.0 {
hull.pop();
}
hull.push(p);
}
hull.pop();
}
let m = hull.len();
let inside = |q: [f64; 2]| q[0].hypot(q[1]) <= radius;
let mut nearest = f64::INFINITY;
let mut wraps = true;
for k in 0..m {
let (a, b) = (hull[k], hull[(k + 1) % m]);
let d = [b[0] - a[0], b[1] - a[1]];
let t = (-(a[0] * d[0] + a[1] * d[1]) / (d[0] * d[0] + d[1] * d[1])).clamp(0.0, 1.0);
nearest = nearest.min((a[0] + d[0] * t).hypot(a[1] + d[1] * t));
if cross(a, b, [0.0, 0.0]) < 0.0 {
wraps = false;
}
}
if wraps || nearest >= radius || (0..m).all(|k| inside(hull[k])) {
return Ok(None);
}
let mut leave: Vec<f64> = Vec::new();
let mut enter: Vec<f64> = Vec::new();
for k in 0..m {
let (a, b) = (hull[k], hull[(k + 1) % m]);
let d = [b[0] - a[0], b[1] - a[1]];
let (qa, qb, qc) = (d[0] * d[0] + d[1] * d[1], 2.0 * (a[0] * d[0] + a[1] * d[1]), a[0] * a[0] + a[1] * a[1] - radius * radius);
let disc = qb * qb - 4.0 * qa * qc;
if disc <= 0.0 {
continue;
}
for t in [(-qb - disc.sqrt()) / (2.0 * qa), (-qb + disc.sqrt()) / (2.0 * qa)] {
if t > 0.0 && t < 1.0 {
let q = [a[0] + d[0] * t, a[1] + d[1] * t];
let outward = 2.0 * (q[0] * d[0] + q[1] * d[1]) > 0.0;
(if outward { &mut leave } else { &mut enter }).push(q[1].atan2(q[0]));
}
}
}
let ([a], [b]) = (leave.as_slice(), enter.as_slice()) else {
return Err(refuse("clip_trim", "the tool's shadow leaves the clip's disc more than once".into()));
};
let width = (b - a).rem_euclid(TAU);
if width >= 0.9 * PI {
return Err(refuse("clip_trim", "the tool's shadow spans too much of the clip's circle to be trimmed".into()));
}
let mut planes: Vec<(Vec3, f64)> = Vec::new();
let centroid = vertices.iter().fold(Vec3::new(0.0, 0.0, 0.0), |s, &p| s.add(p)).scale(1.0 / vertices.len() as f64);
for face in faces {
let (p, q, r) = (vertices[face[0]], vertices[face[1]], vertices[face[2]]);
let mut n = frag(q.sub(p).cross(r.sub(p)).normalized(), "face_normal")?;
if n.dot(p.sub(centroid)) < 0.0 {
n = n.scale(-1.0);
}
planes.push((n, n.dot(p)));
}
for angle in [*a, *b] {
let n = ex.scale(angle.cos()).add(ey.scale(angle.sin()));
planes.push((n, n.dot(clip.axis_origin) + radius));
}
Ok(Some(polyhedron_of_planes(&planes)?))
}
fn polyhedron_of_planes(planes: &[(Vec3, f64)]) -> Result<(Vec<Vec3>, Vec<Vec<usize>>), KernelRefusal> {
let k = planes.len();
let mut corners: Vec<Vec3> = Vec::new();
let scale = planes.iter().map(|p| p.1.abs()).fold(1.0f64, f64::max);
for i in 0..k {
for j in i + 1..k {
for l in j + 1..k {
let (a, b, c) = (planes[i].0, planes[j].0, planes[l].0);
let det = a.dot(b.cross(c));
if det.abs() < 1e-12 {
continue;
}
let p = b.cross(c).scale(planes[i].1).add(c.cross(a).scale(planes[j].1)).add(a.cross(b).scale(planes[l].1)).scale(1.0 / det);
if planes.iter().all(|(n, d)| n.dot(p) - d <= 1e-9 * scale) && !corners.iter().any(|q| q.sub(p).length() <= 1e-9 * scale) {
corners.push(p);
}
}
}
}
let mut faces = Vec::new();
for (n, d) in planes {
let on: Vec<usize> = (0..corners.len()).filter(|&c| (n.dot(corners[c]) - d).abs() <= 1e-9 * scale).collect();
if on.len() < 3 {
continue;
}
let middle = on.iter().fold(Vec3::new(0.0, 0.0, 0.0), |s, &c| s.add(corners[c])).scale(1.0 / on.len() as f64);
let u = frag(corners[on[0]].sub(middle).normalized(), "face_axis")?;
let v = n.cross(u);
let mut ordered = on.clone();
ordered.sort_by(|&p, &q| {
let (rp, rq) = (corners[p].sub(middle), corners[q].sub(middle));
rp.dot(v).atan2(rp.dot(u)).total_cmp(&rq.dot(v).atan2(rq.dot(u)))
});
faces.push(ordered);
}
Ok((corners, faces))
}
fn turn_shadow(screw: &Screw, vertices: &[Vec3], faces: &[Vec<usize>], clip: &crate::SweepClip, axis: Vec3, wedge: [f64; 2]) -> Result<BrepSolid, KernelRefusal> {
let ez = axis;
let ex = clip.reference.sub(ez.scale(clip.reference.dot(ez))).normalized().map_err(|_| input("clip_reference", "the clip's reference lies along its axis".into()))?;
let ey = ez.cross(ex);
let rate = screw.omega * screw.e.dot(ez).signum();
let local = |p: Vec3| {
let r = p.sub(clip.axis_origin);
[r.dot(ex), r.dot(ey), r.dot(ez)]
};
let corners: Vec<[f64; 3]> = vertices.iter().map(|&p| local(p)).collect();
let mid = corners.iter().fold([0.0, 0.0], |s, c| [s[0] + c[0], s[1] + c[1]]);
let mid_angle = mid[1].atan2(mid[0]);
let angles: Vec<f64> = corners.iter().map(|c| mid_angle + (c[1].atan2(c[0]) - mid_angle + PI).rem_euclid(TAU) - PI).collect();
let (low, high) = (angles.iter().copied().fold(f64::INFINITY, f64::min), angles.iter().copied().fold(f64::NEG_INFINITY, f64::max));
if high - low >= PI {
return Err(refuse("turn_shadow", "the tool reaches round the turn's axis".into()));
}
let (start, end) = (rate * screw.t0, rate * screw.t1);
let (first, last) = if rate > 0.0 { (start, end) } else { (end, start) };
let reference = 0.5 * (wedge[0] + wedge[1]);
let near = |a: f64| reference + (a - reference + PI).rem_euclid(TAU) - PI;
let clear = near(high + first) < wedge[0] - 1e-9 && near(low + last) > wedge[1] + 1e-9 && (last - first) < TAU;
if !clear {
return Err(refuse("turn_shadow", "the turned tool's ends reach the clip's wedge; its sweep there is not its turned shadow".into()));
}
let profile = polyhedron_shadow(&corners, faces)?;
let (c0, s0) = (wedge[0].cos(), wedge[0].sin());
let world = |rho: f64, z: f64| clip.axis_origin.add(ez.scale(z)).add(ex.scale(rho * c0)).add(ey.scale(rho * s0));
let mut curves = Vec::with_capacity(profile.len());
for piece in &profile {
let points: Vec<Vec4> = piece.iter().map(|&(rho, z, w)| Vec4::from_point(world(rho, z), w)).collect();
let curve = if points.len() == 2 {
frag(make_line(frag(points[0].point(), "point")?, frag(points[1].point(), "point")?), "line")?
} else {
frag(NurbsCurve::new(2, vec![0.0, 0.0, 0.0, 1.0, 1.0, 1.0], points), "conic")?
};
curves.push(curve);
}
frag(crate::revolve_profile_brep(&curves, clip.axis_origin, ez, wedge[1] - wedge[0]), "turn_shadow")
}
type ShadowPiece = Vec<(f64, f64, f64)>;
fn polyhedron_shadow(corners: &[[f64; 3]], faces: &[Vec<usize>]) -> Result<Vec<ShadowPiece>, KernelRefusal> {
let zs: Vec<f64> = corners.iter().map(|c| c[2]).collect();
let (z_low, z_high) = (zs.iter().copied().fold(f64::INFINITY, f64::min), zs.iter().copied().fold(f64::NEG_INFINITY, f64::max));
let scale = corners.iter().map(|c| c[0].hypot(c[1]).max(c[2].abs())).fold(1.0f64, f64::max);
let mut edge_faces: HashMap<(usize, usize), Vec<usize>> = HashMap::new();
for (f, face) in faces.iter().enumerate() {
for k in 0..face.len() {
let (i, j) = (face[k], face[(k + 1) % face.len()]);
edge_faces.entry((i.min(j), i.max(j))).or_default().push(f);
}
}
let mut edges: Vec<(usize, usize)> = edge_faces.keys().copied().collect();
edges.sort_unstable();
let planes: Vec<([f64; 3], f64)> = faces
.iter()
.map(|face| {
let (a, b, c) = (corners[face[0]], corners[face[1]], corners[face[2]]);
let (u, v) = ([b[0] - a[0], b[1] - a[1], b[2] - a[2]], [c[0] - a[0], c[1] - a[1], c[2] - a[2]]);
let n = [u[1] * v[2] - u[2] * v[1], u[2] * v[0] - u[0] * v[2], u[0] * v[1] - u[1] * v[0]];
(n, n[0] * a[0] + n[1] * a[1] + n[2] * a[2])
})
.collect();
#[derive(Clone, Copy, PartialEq, Debug)]
enum Source {
Edge(usize),
Face(usize),
}
let edge_at = |k: usize, z: f64| -> [f64; 2] {
let (a, b) = (corners[edges[k].0], corners[edges[k].1]);
let t = (z - a[2]) / (b[2] - a[2]);
[a[0] + (b[0] - a[0]) * t, a[1] + (b[1] - a[1]) * t]
};
let crossing = |z: f64| -> Vec<usize> {
(0..edges.len()).filter(|&k| (corners[edges[k].0][2] - z) * (corners[edges[k].1][2] - z) < 0.0).collect()
};
let farthest = |z: f64| -> (Source, f64) {
let mut best = (Source::Edge(0), f64::NEG_INFINITY);
for k in crossing(z) {
let p = edge_at(k, z);
let r = p[0].hypot(p[1]);
if r > best.1 {
best = (Source::Edge(k), r);
}
}
best
};
let nearest = |z: f64| -> Result<(Source, f64), KernelRefusal> {
let list = crossing(z);
let points: Vec<[f64; 2]> = list.iter().map(|&k| edge_at(k, z)).collect();
let m = points.len();
if m < 3 {
return Err(refuse("turn_shadow", "a section of the tool is not a polygon".into()));
}
let centre = points.iter().fold([0.0, 0.0], |s, p| [s[0] + p[0] / m as f64, s[1] + p[1] / m as f64]);
let mut order: Vec<usize> = (0..m).collect();
order.sort_by(|&a, &b| {
let (pa, pb) = (points[a], points[b]);
(pa[1] - centre[1]).atan2(pa[0] - centre[0]).total_cmp(&(pb[1] - centre[1]).atan2(pb[0] - centre[0]))
});
let mut inside = true;
let mut best = (Source::Edge(0), f64::INFINITY);
for k in 0..m {
let (ia, ib) = (order[k], order[(k + 1) % m]);
let (a, b) = (points[ia], points[ib]);
let d = [b[0] - a[0], b[1] - a[1]];
if d[0] * (0.0 - a[1]) - d[1] * (0.0 - a[0]) < 0.0 {
inside = false;
}
let length2 = d[0] * d[0] + d[1] * d[1];
let t = -(a[0] * d[0] + a[1] * d[1]) / length2;
let (source, r) = if t > 0.0 && t < 1.0 {
let (ea, eb) = (edges[list[ia]], edges[list[ib]]);
let shared = edge_faces[&ea].iter().find(|f| edge_faces[&eb].contains(f)).copied();
let Some(face) = shared else {
return Err(refuse("turn_shadow", "a section side does not lie on one face".into()));
};
let q = [a[0] + d[0] * t, a[1] + d[1] * t];
(Source::Face(face), q[0].hypot(q[1]))
} else {
let q = if t <= 0.0 { a } else { b };
(Source::Edge(list[if t <= 0.0 { ia } else { ib }]), q[0].hypot(q[1]))
};
if r < best.1 {
best = (source, r);
}
}
if inside {
return Err(refuse("turn_shadow", "the turn's axis passes through the tool".into()));
}
Ok(best)
};
let value = |source: Source, z: f64| -> f64 {
match source {
Source::Edge(k) => {
let p = edge_at(k, z);
p[0].hypot(p[1])
}
Source::Face(f) => {
let (n, d) = planes[f];
(d - n[2] * z).abs() / n[0].hypot(n[1])
}
}
};
let mut heights: Vec<f64> = zs.clone();
heights.sort_by(|a, b| a.total_cmp(b));
heights.dedup_by(|a, b| (*a - *b).abs() <= 1e-12 * scale);
let mut pieces_far: Vec<(f64, f64, Source)> = Vec::new();
let mut pieces_near: Vec<(f64, f64, Source)> = Vec::new();
for window in heights.windows(2) {
let (za, zb) = (window[0], window[1]);
let samples = 400;
let inner = |k: usize| za + (zb - za) * (k as f64 + 0.5) / samples as f64;
for near in [false, true] {
let source_at = |z: f64| -> Result<Source, KernelRefusal> { Ok(if near { nearest(z)?.0 } else { farthest(z).0 }) };
let mut runs: Vec<(f64, Source)> = vec![(za, source_at(inner(0))?)];
for k in 1..samples {
let (z0, z1) = (inner(k - 1), inner(k));
let (s0, s1) = (source_at(z0)?, source_at(z1)?);
if s0 == s1 {
continue;
}
let (mut lo, mut hi) = (z0, z1);
for _ in 0..200 {
let midz = 0.5 * (lo + hi);
if source_at(midz)? == s0 { lo = midz } else { hi = midz }
}
runs.push((0.5 * (lo + hi), s1));
}
let list = if near { &mut pieces_near } else { &mut pieces_far };
for k in 0..runs.len() {
let z_end = if k + 1 < runs.len() { runs[k + 1].0 } else { zb };
list.push((runs[k].0, z_end, runs[k].1));
}
}
}
let piece = |source: Source, za: f64, zb: f64| -> Vec<ShadowPiece> {
let (ra, rb) = (value(source, za), value(source, zb));
if let Source::Face(..) = source {
return vec![vec![(ra, za, 1.0), (rb, zb, 1.0)]];
}
let rho = |z: f64| value(source, z);
let slope = |z: f64| {
let h = 1e-6 * (zb - za).abs().max(1e-300);
(rho(z + h) - rho(z - h)) / (2.0 * h)
};
let parts = 4;
let mut out = Vec::new();
for k in 0..parts {
let (z0, z1) = (za + (zb - za) * k as f64 / parts as f64, za + (zb - za) * (k + 1) as f64 / parts as f64);
let (a, b) = ((rho(z0), z0), (rho(z1), z1));
let (ta, tb) = ((slope(z0), 1.0), (slope(z1), 1.0));
let det = ta.0 * (-tb.1) - ta.1 * (-tb.0);
if det.abs() < 1e-12 {
out.push(vec![(a.0, a.1, 1.0), (b.0, b.1, 1.0)]);
continue;
}
let rhs = (b.0 - a.0, b.1 - a.1);
let s = (rhs.0 * (-tb.1) - rhs.1 * (-tb.0)) / det;
let c = (a.0 + ta.0 * s, a.1 + ta.1 * s);
let m = (0.5 * (a.0 + b.0), 0.5 * (a.1 + b.1));
let on = |u: f64| {
let p = (m.0 + (c.0 - m.0) * u, m.1 + (c.1 - m.1) * u);
p.0 - rho(p.1)
};
let (mut lo, mut hi) = (0.0, 1.0);
let sign_lo = on(lo);
for _ in 0..200 {
let mid = 0.5 * (lo + hi);
if (on(mid) > 0.0) == (sign_lo > 0.0) { lo = mid } else { hi = mid }
}
let u = 0.5 * (lo + hi);
let weight = u / (1.0 - u);
out.push(vec![(a.0, a.1, 1.0), (c.0, c.1, weight), (b.0, b.1, 1.0)]);
}
out
};
let mut outline: Vec<ShadowPiece> = Vec::new();
for &(za, zb, source) in &pieces_near {
outline.extend(piece(source, za, zb));
}
let (near_top, far_top) = (value(pieces_near.last().expect("near").2, z_high), value(pieces_far.last().expect("far").2, z_high));
if far_top - near_top > 1e-12 * scale {
outline.push(vec![(near_top, z_high, 1.0), (far_top, z_high, 1.0)]);
}
for &(za, zb, source) in pieces_far.iter().rev() {
for mut arc in piece(source, za, zb).into_iter().rev() {
arc.reverse();
outline.push(arc);
}
}
let (near_bottom, far_bottom) = (value(pieces_near[0].2, z_low), value(pieces_far[0].2, z_low));
if far_bottom - near_bottom > 1e-12 * scale {
outline.push(vec![(far_bottom, z_low, 1.0), (near_bottom, z_low, 1.0)]);
}
for k in 0..outline.len() {
let next = (k + 1) % outline.len();
let end = *outline[k].last().expect("piece");
outline[next][0] = (end.0, end.1, 1.0);
}
Ok(outline)
}
fn ball_across_axis(screw: &Screw, origin: Vec3, axis: Vec3, meridian: &[OutlinePiece]) -> Result<Option<BrepSolid>, KernelRefusal> {
let [OutlinePiece::Arc { centre, radius, start, end }, OutlinePiece::Line { .. }] = meridian else { return Ok(None) };
let (centre, r) = (*centre, *radius);
if centre[0].abs() > 1e-12 * r || ((end - start).abs() - PI).abs() > 1e-12 || screw.c != 0.0 || screw.c2 != 0.0 || screw.omega == 0.0 {
return Ok(None);
}
let w = axis.normalized().map_err(|_| input("axis", "the tool's axis has no direction".into()))?;
let centre_tool = origin.add(w.scale(centre[1]));
let turned = screw.omega * (screw.t1 - screw.t0);
if turned.abs() >= TAU - 1e-9 {
return Ok(None);
}
let e = screw.e.scale(screw.omega.signum()); let c0 = screw.apply(screw.t0, centre_tool);
let c1 = screw.apply(screw.t1, centre_tool);
let foot = |c: Vec3| screw.a.add(e.scale(c.sub(screw.a).dot(e)));
let rho = c0.sub(foot(c0)).length();
if rho >= r * (1.0 - 1e-9) || rho <= 1e-9 * r {
return Ok(None);
}
let h0 = frag(c0.sub(foot(c0)).normalized(), "radial")?;
let height = (r * r - rho * rho).sqrt();
let (north, south) = (foot(c0).add(e.scale(height)), foot(c0).sub(e.scale(height)));
let arc = |about: Vec3, k: Vec3, from: Vec3, angle: f64| -> Result<NurbsCurve, KernelRefusal> {
let surface = frag(make_revolution(about, k, &frag(make_line(from, from.add(k)), "line")?, angle), "arc")?;
frag(surface.iso_curve_v(0.0), "arc")
};
let alpha = height.atan2(-rho);
let outer = arc(c0, h0.cross(e), south, 2.0 * alpha)?;
let wall = frag(make_revolution(screw.a, e, &outer, turned.abs()), "wall")?;
let outer_end = frag(wall.iso_curve_u(1.0), "wall_end")?;
let mid = c0.add(c1).scale(0.5);
let towards = frag(mid.sub(foot(mid)).normalized(), "bisector")?;
let rho_mid = mid.sub(foot(mid)).length();
let beta = height.atan2(rho_mid); let meet = arc(mid, e.cross(towards), south, 2.0 * beta)?;
{
let [d0, d1] = frag(meet.domain(), "domain")?;
let middle = frag(meet.evaluate(0.5 * (d0 + d1)), "middle")?;
if middle.sub(foot(middle)).dot(towards) > 0.0 {
return Err(refuse("ball_across_axis", "the spheres' common circle runs the wrong way round".into()));
}
}
let cap = |c: Vec3, inward: Vec3, side: Vec3, other: Vec3| -> Result<(NurbsSurface, Box<dyn Fn(Vec3) -> Vec3>), KernelRefusal> {
let toward_south = frag(inward.add(side.scale(-0.3)).normalized(), "pole")?;
let polar = toward_south.scale(-1.0);
let seam = frag(side.sub(polar.scale(side.dot(polar))).normalized(), "seam")?;
for k in 0..=64 {
let a = PI * k as f64 / 64.0;
let q = c.add(polar.scale(-r * a.cos())).add(seam.scale(r * a.sin()));
if q.sub(c).dot(side) < 0.0 && q.sub(other).length() >= r * (1.0 - 1e-9) {
return Err(refuse("ball_across_axis", "no seam place for a cap".into()));
}
}
let south_pole = c.sub(polar.scale(r));
if south_pole.sub(other).length() >= r * (1.0 - 1e-6) {
return Err(refuse("ball_across_axis", "no pole place for a cap".into()));
}
let generatrix = arc(c, polar.scale(-1.0).cross(seam), south_pole, PI)?;
let surface = frag(make_revolution(c, polar, &generatrix, TAU), "cap")?;
let spans_u = (surface.knots_u.len() - 6) / 2 + 1;
let spans_v = (generatrix.knots.len() - 6) / 2 + 1;
let around = polar.cross(seam);
let inverse = move |x: Vec3| -> Vec3 {
let d = x.sub(c).scale(1.0 / r);
let latitude = d.dot(polar).clamp(-1.0, 1.0).asin();
let longitude = d.dot(around).atan2(d.dot(seam)).rem_euclid(TAU);
Vec3::new(
crate::circle_angle_to_parameter(spans_u, TAU, longitude),
crate::circle_angle_to_parameter(spans_v, PI, latitude + 0.5 * PI),
0.0,
)
};
Ok((surface, Box::new(inverse)))
};
let moving0 = e.cross(h0);
let h1 = frag(c1.sub(foot(c1)).normalized(), "radial")?;
let moving1 = e.cross(h1);
let (back, back_inverse) = cap(c0, h0.scale(-1.0), moving0.scale(1.0), c1)?;
let (front, front_inverse) = cap(c1, h1.scale(-1.0), moving1.scale(-1.0), c0)?;
let pcurve = |inverse: &dyn Fn(Vec3) -> Vec3, curve: &NurbsCurve, forward: bool| -> Result<NurbsCurve, KernelRefusal> {
let [d0, d1] = frag(curve.domain(), "domain")?;
let ts: Vec<f64> = (0..=PCURVE_SAMPLES).map(|j| d0 + (d1 - d0) * j as f64 / PCURVE_SAMPLES as f64).collect();
let uv: Vec<Vec3> = ts.iter().map(|&t| curve.evaluate(t).map(|x| inverse(x))).collect::<Result<_, _>>().map_err(|e| refuse("pcurve", e))?;
let fitted = frag(crate::interpolate_curve(&uv, 5, &ts), "cap_pcurve")?;
if forward { Ok(fitted) } else { frag(fitted.reversed(), "reversed") }
};
let mut shell = Shell::new();
let (vn, vs) = (shell.vertex(north), shell.vertex(south));
let k0 = shell.edge(outer.clone(), vs, vn, false)?;
let k1 = shell.edge(outer_end.clone(), vs, vn, false)?;
let c = shell.edge(meet.clone(), vs, vn, false)?;
let ds = shell.edge(frag(make_line(south, south), "point")?, vs, vs, true)?;
let dn = shell.edge(frag(make_line(north, north), "point")?, vn, vn, true)?;
let wall_outward = {
let p = frag(wall.evaluate(0.5, 0.5), "probe")?;
let su = frag(wall.evaluate(0.5 + 1e-6, 0.5), "probe")?.sub(frag(wall.evaluate(0.5 - 1e-6, 0.5), "probe")?);
let sv = frag(wall.evaluate(0.5, 0.5 + 1e-6), "probe")?.sub(frag(wall.evaluate(0.5, 0.5 - 1e-6), "probe")?);
su.cross(sv).dot(p.sub(foot(p))) > 0.0
};
let [g0, g1] = frag(outer.domain(), "domain")?;
let wall_loop = vec![
(ds, true, uv_line([0.0, g0], [1.0, g0])?),
(k1, true, uv_line([1.0, g0], [1.0, g1])?),
(dn, true, uv_line([1.0, g1], [0.0, g1])?),
(k0, false, uv_line([0.0, g1], [0.0, g0])?),
];
let reversed = |list: Vec<(u64, bool, NurbsCurve)>| -> Result<Vec<(u64, bool, NurbsCurve)>, KernelRefusal> {
list.into_iter().rev().map(|(e, f, p)| Ok((e, !f, frag(p.reversed(), "reversed")?))).collect()
};
shell.face(wall.clone(), if wall_outward { wall_loop } else { reversed(wall_loop)? });
for (surface, inverse, k_edge, k_curve) in [(back, &back_inverse, k0, &outer), (front, &front_inverse, k1, &outer_end)] {
let mut lp = vec![(k_edge, true, pcurve(inverse.as_ref(), k_curve, true)?), (c, false, pcurve(inverse.as_ref(), &meet, false)?)];
let mut area = 0.0;
for (_, _, pc) in &lp {
let [d0, d1] = frag(pc.domain(), "domain")?;
for j in 0..64 {
let (a, b) = (frag(pc.evaluate(d0 + (d1 - d0) * j as f64 / 64.0), "uv")?, frag(pc.evaluate(d0 + (d1 - d0) * (j + 1) as f64 / 64.0), "uv")?);
area += a.x * b.y - b.x * a.y;
}
}
if area < 0.0 {
lp = reversed(lp)?;
}
shell.face(surface, lp);
}
let body = shell.solid();
let issues = body.validate();
if !issues.is_empty() {
return Err(refuse("assembly", format!("the turned ball's sweep is not valid: {:?}", issues.first())));
}
let probe = Split { points: Vec::new(), curves: Vec::new(), caps: Vec::new(), scale: r + rho, shape: Shape::Revolved { origin, axis: w, meridian: meridian.to_vec() } };
check_single_pass(screw, &probe, &body)?;
Ok(Some(body))
}
#[derive(Clone, Copy, Debug)]
pub struct FrameCurve {
pub control: [Vec3; 4],
pub up: Vec3,
}
pub fn sweep_solid_along(tool: &SolidTool, path: &FrameCurve) -> Result<BrepSolid, KernelRefusal> {
let up = path.up.normalized().map_err(|_| input("up", "the frame's up direction is zero".into()))?;
let base = path.control[0];
if path.control.iter().any(|p| p.sub(base).dot(up).abs() > 1e-9 * (1.0 + p.sub(base).length())) {
return Err(input("plane", "the curve is not in a plane square to up".into()));
}
let tangent = path.control[1].sub(base);
let tangent = if tangent.length() > 0.0 { tangent } else { path.control[2].sub(base) };
let start_tangent = tangent.normalized().map_err(|_| input("tangent", "the curve has no start direction".into()))?;
let frame = Frame { control: path.control, up, start_tangent };
let (mut speed, mut rate, mut extent) = (0.0f64, 0.0f64, 0.0f64);
let mut previous = 0.0;
for k in 0..=512 {
let t = k as f64 / 512.0;
let first = frame.first(t);
if first.length() <= 1e-12 {
return Err(input("tangent", "the curve stops (zero tangent)".into()));
}
speed = speed.max(first.length());
rate = rate.max(frame.angle_rate(t).abs());
extent = extent.max(frame.point(t).sub(base).length());
let angle = frame.angle(t);
if (angle - previous).abs() > 0.5 {
return Err(refuse("frame_turns", "the frame turns half way round".into()));
}
previous = angle;
}
let screw = Screw { a: base, e: up, omega: rate, c: speed, c2: 0.0, t0: 0.0, t1: 1.0, frame: Some(frame), frozen: None, extent };
let SolidTool::Revolved { axis_origin, axis_direction, radial, meridian } = tool else {
return Err(refuse("along_tool", "a frame along a curve carries a solid of revolution only".into()));
};
let axis = axis_direction.normalized().map_err(|_| input("axis", "the tool's axis has no direction".into()))?;
let mut last = None;
for k in 0..8 {
let reference = radial.rotate_about(axis, PI * k as f64 / 4.0);
match lofted(&screw, *axis_origin, axis, reference, meridian) {
Err(error) if error.message.contains("structure") => last = Some(error),
other => return other,
}
}
Err(last.expect("tried"))
}
pub fn sweep_prism_along(outline: &[[f64; 2]], from: f64, to: f64, path: &FrameCurve) -> Result<BrepSolid, KernelRefusal> {
let up = path.up.normalized().map_err(|_| input("up", "the frame's up direction is zero".into()))?;
let base = path.control[0];
if path.control.iter().any(|p| p.sub(base).dot(up).abs() > 1e-9 * (1.0 + p.sub(base).length())) {
return Err(input("plane", "the curve is not in a plane square to up".into()));
}
let tangent = path.control[1].sub(base);
let tangent = if tangent.length() > 0.0 { tangent } else { path.control[2].sub(base) };
let e1 = tangent.normalized().map_err(|_| input("tangent", "the curve has no start direction".into()))?;
let e2 = up.cross(e1);
let frame = Frame { control: path.control, up, start_tangent: e1 };
if outline.len() < 3 || !(to > from) {
return Err(input("prism", "a prism needs three corners and a positive height".into()));
}
let n = outline.len();
let mut corners = outline.to_vec();
let area: f64 = (0..n).map(|k| corners[k][0] * corners[(k + 1) % n][1] - corners[(k + 1) % n][0] * corners[k][1]).sum();
if area < 0.0 {
corners.reverse();
}
for k in 0..n {
let (a, b, c) = (corners[k], corners[(k + 1) % n], corners[(k + 2) % n]);
if (b[0] - a[0]) * (c[1] - b[1]) - (b[1] - a[1]) * (c[0] - b[0]) <= 0.0 {
return Err(input("prism", "the outline is not convex".into()));
}
}
let scale = corners.iter().map(|c| c[0].hypot(c[1])).fold(1.0f64, f64::max);
let flat = |p: Vec3| [p.sub(base).dot(e1), p.sub(base).dot(e2)];
let place = |s: f64, q: [f64; 2]| -> [f64; 2] {
let o = flat(frame.point(s));
let (sn, cs) = frame.angle(s).sin_cos();
[o[0] + cs * q[0] - sn * q[1], o[1] + sn * q[0] + cs * q[1]]
};
let local = |s: f64, p: [f64; 2]| -> [f64; 2] {
let o = flat(frame.point(s));
let (sn, cs) = frame.angle(s).sin_cos();
let d = [p[0] - o[0], p[1] - o[1]];
[cs * d[0] + sn * d[1], -sn * d[0] + cs * d[1]]
};
let edges: Vec<([f64; 2], f64)> = (0..n)
.map(|k| {
let (a, b) = (corners[k], corners[(k + 1) % n]);
let d = [b[0] - a[0], b[1] - a[1]];
let length = d[0].hypot(d[1]);
let normal = [d[1] / length, -d[0] / length];
(normal, normal[0] * a[0] + normal[1] * a[1])
})
.collect();
let inside = |q: [f64; 2]| edges.iter().map(|(nn, d)| nn[0] * q[0] + nn[1] * q[1] - d).fold(f64::NEG_INFINITY, f64::max);
let depth = |p: [f64; 2]| -> f64 {
let samples = 2000;
let f = |s: f64| inside(local(s, p));
let mut best = (f64::INFINITY, 0usize);
let values: Vec<f64> = (0..=samples).map(|k| f(k as f64 / samples as f64)).collect();
for (k, &v) in values.iter().enumerate() {
if v < best.0 {
best = (v, k);
}
}
let mut least = best.0;
for k in 0..=samples {
let left = if k == 0 { f64::INFINITY } else { values[k - 1] };
let right = if k == samples { f64::INFINITY } else { values[k + 1] };
if values[k] <= left && values[k] <= right && values[k] < best.0 + 1e-3 * scale {
let (mut lo, mut hi) = (((k as f64) - 1.0).max(0.0) / samples as f64, ((k as f64) + 1.0).min(samples as f64) / samples as f64);
for _ in 0..80 {
let (m1, m2) = (lo + (hi - lo) / 3.0, hi - (hi - lo) / 3.0);
if f(m1) < f(m2) { hi = m2 } else { lo = m1 }
}
least = least.min(f(0.5 * (lo + hi)));
}
}
least
};
#[derive(Clone, Copy)]
enum Path {
Corner(usize),
Graze(usize, f64, f64),
End(usize, bool),
}
let speed_ratio = |s: f64| frame.angle_rate(s); let graze_point = |k: usize, s: f64| -> Option<f64> {
let (a, b) = (corners[k], corners[(k + 1) % n]);
let (nn, _) = edges[k];
let kappa = speed_ratio(s) / frame.first(s).length();
let rot = |x: [f64; 2]| -x[1] * nn[0] + x[0] * nn[1]; let (ga, gb) = if nn[0].abs() < 1e-15 { (rot(a), rot(b)) } else { (nn[0] + kappa * rot(a), nn[0] + kappa * rot(b)) };
if ga == gb {
return None;
}
let u = ga / (ga - gb);
(u > 0.0 && u < 1.0).then_some(u)
};
let mut paths: Vec<Path> = (0..n).map(Path::Corner).collect();
for k in 0..n {
let steps = 2000;
let mut run: Option<f64> = None;
for j in 0..=steps {
let s = j as f64 / steps as f64;
let on = graze_point(k, s).is_some();
match (on, run) {
(true, None) => run = Some(s),
(false, Some(s0)) => {
paths.push(Path::Graze(k, s0, s - 1.0 / steps as f64));
run = None;
}
_ => {}
}
}
if let Some(s0) = run {
paths.push(Path::Graze(k, s0, 1.0));
}
paths.push(Path::End(k, false));
paths.push(Path::End(k, true));
}
let eval = |path: Path, t: f64| -> [f64; 2] {
match path {
Path::Corner(k) => place(t, corners[k]),
Path::Graze(k, s0, s1) => {
let s = s0 + (s1 - s0) * t;
let u = graze_point(k, s).unwrap_or(if t < 0.5 { 0.0 } else { 1.0 });
let (a, b) = (corners[k], corners[(k + 1) % n]);
place(s, [a[0] + (b[0] - a[0]) * u, a[1] + (b[1] - a[1]) * u])
}
Path::End(k, at_end) => {
let (a, b) = (corners[k], corners[(k + 1) % n]);
place(if at_end { 1.0 } else { 0.0 }, [a[0] + (b[0] - a[0]) * t, a[1] + (b[1] - a[1]) * t])
}
}
};
let m = 400;
let polylines: Vec<Vec<[f64; 2]>> = paths.iter().map(|&p| (0..=m).map(|i| eval(p, i as f64 / m as f64)).collect()).collect();
let mut cuts: Vec<Vec<f64>> = vec![vec![0.0, 1.0]; paths.len()];
let tol = 1e-10 * scale;
for a in 0..paths.len() {
for b in a + 1..paths.len() {
for i in 0..m {
for j in 0..m {
let (p0, p1, q0, q1) = (polylines[a][i], polylines[a][i + 1], polylines[b][j], polylines[b][j + 1]);
let d1 = [p1[0] - p0[0], p1[1] - p0[1]];
let d2 = [q1[0] - q0[0], q1[1] - q0[1]];
let det = d1[0] * d2[1] - d1[1] * d2[0];
if det.abs() < 1e-300 {
continue;
}
let r = [q0[0] - p0[0], q0[1] - p0[1]];
let (x, y) = ((r[0] * d2[1] - r[1] * d2[0]) / det, (r[0] * d1[1] - r[1] * d1[0]) / det);
if !(-1e-9..=1.0 + 1e-9).contains(&x) || !(-1e-9..=1.0 + 1e-9).contains(&y) {
continue;
}
let (mut ta, mut tb) = ((i as f64 + x) / m as f64, (j as f64 + y) / m as f64);
for _ in 0..50 {
let (pa, pb) = (eval(paths[a], ta), eval(paths[b], tb));
let h = 1e-7;
let da = { let q = eval(paths[a], (ta + h).min(1.0)); let p = eval(paths[a], (ta - h).max(0.0)); let w = (ta + h).min(1.0) - (ta - h).max(0.0); [(q[0] - p[0]) / w, (q[1] - p[1]) / w] };
let db = { let q = eval(paths[b], (tb + h).min(1.0)); let p = eval(paths[b], (tb - h).max(0.0)); let w = (tb + h).min(1.0) - (tb - h).max(0.0); [(q[0] - p[0]) / w, (q[1] - p[1]) / w] };
let f = [pa[0] - pb[0], pa[1] - pb[1]];
let det = -da[0] * db[1] + da[1] * db[0];
if det.abs() < 1e-300 {
break;
}
let dta = (-f[0] * db[1] + f[1] * db[0]) / det;
let dtb = (da[0] * f[1] - da[1] * f[0]) / det;
ta = (ta - dta).clamp(0.0, 1.0);
tb = (tb - dtb).clamp(0.0, 1.0);
if dta.abs() + dtb.abs() < 1e-15 {
break;
}
}
let (pa, pb) = (eval(paths[a], ta), eval(paths[b], tb));
if (pa[0] - pb[0]).hypot(pa[1] - pb[1]) < tol {
cuts[a].push(ta);
cuts[b].push(tb);
}
}
}
}
}
for c in &mut cuts {
c.sort_by(|x, y| x.total_cmp(y));
c.dedup_by(|x, y| (*x - *y).abs() < 1e-9);
}
struct Piece {
path: usize,
t0: f64,
t1: f64,
}
let eps = 1e-7 * scale;
let mut pieces: Vec<Piece> = Vec::new();
for (index, c) in cuts.iter().enumerate() {
for w in c.windows(2) {
let (t0, t1) = (w[0], w[1]);
if t1 - t0 < 1e-9 {
continue;
}
let (pa, pb) = (eval(paths[index], t0), eval(paths[index], t1));
if (pa[0] - pb[0]).hypot(pa[1] - pb[1]) < 1e-9 * scale {
continue;
}
let tm = 0.5 * (t0 + t1);
let (p, q) = (eval(paths[index], tm), eval(paths[index], (tm + 1e-6).min(1.0)));
let d = [q[0] - p[0], q[1] - p[1]];
let length = d[0].hypot(d[1]);
if length == 0.0 {
continue;
}
let left = [p[0] - d[1] / length * eps, p[1] + d[0] / length * eps];
let right = [p[0] + d[1] / length * eps, p[1] - d[0] / length * eps];
let (l, r) = (depth(left) < -1e-12 * scale, depth(right) < -1e-12 * scale);
match (l, r) {
(true, false) => pieces.push(Piece { path: index, t0, t1 }),
(false, true) => pieces.push(Piece { path: index, t0: t1, t1: t0 }),
_ => {}
}
}
}
if pieces.is_empty() {
return Err(refuse("prism_along", "no boundary found".into()));
}
let mut order = vec![0usize];
let mut used = vec![false; pieces.len()];
used[0] = true;
loop {
let last = &pieces[*order.last().expect("order")];
let end = eval(paths[last.path], last.t1);
let start = eval(paths[pieces[order[0]].path], pieces[order[0]].t0);
if order.len() > 1 && (end[0] - start[0]).hypot(end[1] - start[1]) < 1e-8 * scale {
break;
}
let next = (0..pieces.len()).filter(|&k| !used[k]).min_by(|&x, &y| {
let (px, py) = (eval(paths[pieces[x].path], pieces[x].t0), eval(paths[pieces[y].path], pieces[y].t0));
(px[0] - end[0]).hypot(px[1] - end[1]).total_cmp(&(py[0] - end[0]).hypot(py[1] - end[1]))
});
let Some(next) = next else {
return Err(refuse("prism_along", "the boundary does not close".into()));
};
let p = eval(paths[pieces[next].path], pieces[next].t0);
if (p[0] - end[0]).hypot(p[1] - end[1]) > 1e-8 * scale {
return Err(refuse("prism_along", format!("the boundary does not close (gap {:e})", (p[0] - end[0]).hypot(p[1] - end[1]))));
}
used[next] = true;
order.push(next);
}
if used.iter().any(|u| !u) {
return Err(refuse("prism_along", "the swept region is not one simple loop".into()));
}
let world = |q: [f64; 2], h: f64| base.add(e1.scale(q[0])).add(e2.scale(q[1])).add(up.scale(h));
let mut loop_points: Vec<[f64; 2]> = order.iter().map(|&k| eval(paths[pieces[k].path], pieces[k].t0)).collect();
loop_points.push(loop_points[0]);
let count = order.len();
let mut shell = Shell::new();
let (bottom_v, top_v): (Vec<u64>, Vec<u64>) = (0..count).map(|k| (shell.vertex(world(loop_points[k], from)), shell.vertex(world(loop_points[k], to)))).unzip();
let mut bottom_e = Vec::with_capacity(count);
let mut top_e = Vec::with_capacity(count);
let mut curves2d = Vec::with_capacity(count);
for (slot, &k) in order.iter().enumerate() {
let piece = &pieces[k];
let straight = matches!(paths[piece.path], Path::End(..));
let next = (slot + 1) % count;
let curve_at = |h: f64| -> Result<NurbsCurve, KernelRefusal> {
if straight {
return frag(make_line(world(loop_points[slot], h), world(loop_points[next], h)), "line");
}
let samples = 192;
let ts: Vec<f64> = (0..=samples).map(|i| i as f64 / samples as f64).collect();
let mut points: Vec<Vec3> = ts.iter().map(|&t| world(eval(paths[piece.path], piece.t0 + (piece.t1 - piece.t0) * t), h)).collect();
points[0] = world(loop_points[slot], h);
points[samples] = world(loop_points[next], h);
frag(crate::interpolate_curve(&points, 5, &ts), "plane_path")
};
let (cb, ct) = (curve_at(from)?, curve_at(to)?);
bottom_e.push(shell.edge(cb.clone(), bottom_v[slot], bottom_v[next], false)?);
top_e.push(shell.edge(ct.clone(), top_v[slot], top_v[next], false)?);
curves2d.push((cb, ct));
}
let verticals: Vec<u64> = (0..count).map(|k| shell.edge(make_line(world(loop_points[k], from), world(loop_points[k], to)).expect("line"), bottom_v[k], top_v[k], false)).collect::<Result<_, _>>()?;
let top_points: Vec<Vec3> = loop_points.iter().map(|&q| world(q, to)).collect();
let top_surface = plane_patch(world([0.0, 0.0], to), e1, e2, &top_points)?;
let mut coedges = Vec::with_capacity(count);
for k in 0..count {
coedges.push((top_e[k], true, plane_pcurve(world([0.0, 0.0], to), e1, e2, &curves2d[k].1, true)?));
}
shell.face(top_surface, coedges);
let bottom_points: Vec<Vec3> = loop_points.iter().map(|&q| world(q, from)).collect();
let bottom_surface = plane_patch(world([0.0, 0.0], from), e2, e1, &bottom_points)?;
let mut coedges = Vec::with_capacity(count);
for k in (0..count).rev() {
coedges.push((bottom_e[k], false, plane_pcurve(world([0.0, 0.0], from), e2, e1, &curves2d[k].0, false)?));
}
shell.face(bottom_surface, coedges);
for k in 0..count {
let next = (k + 1) % count;
let (cb, _) = &curves2d[k];
let rows: Vec<Vec<Vec4>> = cb
.control_points
.iter()
.map(|p| {
let x = p.point().expect("point");
vec![Vec4::from_point(x, p.w), Vec4::from_point(x.add(up.scale(to - from)), p.w)]
})
.collect();
let surface = frag(NurbsSurface::new(cb.degree, 1, cb.knots.clone(), vec![0.0, 0.0, 1.0, 1.0], rows), "side")?;
let [u0, u1] = frag(cb.domain(), "domain")?;
let coedges = vec![
(bottom_e[k], true, uv_line([u0, 0.0], [u1, 0.0])?),
(verticals[next], true, uv_line([u1, 0.0], [u1, 1.0])?),
(top_e[k], false, uv_line([u1, 1.0], [u0, 1.0])?),
(verticals[k], false, uv_line([u0, 1.0], [u0, 0.0])?),
];
shell.face(surface, coedges);
}
let body = shell.solid();
let issues = body.validate();
if !issues.is_empty() {
return Err(refuse("assembly", format!("the swept prism is not valid: {:?}", issues.first())));
}
Ok(body)
}
fn projected_pcurve(surface: &NurbsSurface, curve: &NurbsCurve, forward: bool) -> Result<NurbsCurve, KernelRefusal> {
let [d0, d1] = frag(curve.domain(), "domain")?;
let ts: Vec<f64> = (0..=PCURVE_SAMPLES).map(|j| d0 + (d1 - d0) * j as f64 / PCURVE_SAMPLES as f64).collect();
let mut uv = Vec::with_capacity(ts.len());
for &t in &ts {
let x = frag(curve.evaluate(t), "point")?;
let foot = frag(crate::project_point_to_surface(surface, x), "project")?;
if foot.distance > 1e-6 * (1.0 + x.length()) {
return Err(refuse("loft_cap", format!("a lofted edge leaves its cap by {} at {x:?}", foot.distance)));
}
uv.push(Vec3::new(foot.u, foot.v, 0.0));
}
let fitted = frag(crate::interpolate_curve(&uv, 5, &ts), "projected_pcurve")?;
if forward { Ok(fitted) } else { frag(fitted.reversed(), "reversed") }
}
type Signature = (Vec<(usize, usize, bool)>, Vec<(bool, Vec<(usize, bool)>)>, Vec<(usize, bool)>);
fn signature(split: &Split) -> Result<Signature, KernelRefusal> {
Ok((
split.curves.iter().map(|c| (c.start, c.end, c.degenerate)).collect(),
split.caps.iter().map(|cap| (cap.front, cap.coedges.iter().map(|&(c, f, _)| (c, f)).collect())).collect(),
split.grazing()?,
))
}
fn lofted(screw: &Screw, origin: Vec3, axis: Vec3, radial: Vec3, meridian: &[OutlinePiece]) -> Result<BrepSolid, KernelRefusal> {
let split_at = |t: f64| -> Result<Split, KernelRefusal> {
let mut frozen = *screw;
frozen.frozen = Some(t);
split_revolved(&frozen, origin, axis, radial, meridian)
};
let lambdas: Vec<f64> = (0..=LOFT_SAMPLES).map(|i| i as f64 / LOFT_SAMPLES as f64).collect();
let samples_of = |split: &Split, c: usize, back_forward: bool, t: f64| -> Result<Vec<Vec3>, KernelRefusal> {
let Some(list) = &split.curves[c].samples else {
return Err(refuse("loft_samples", "a grazing curve has no exact samples to loft".into()));
};
let mut world: Vec<Vec3> = list.iter().map(|&x| screw.apply(t, x)).collect();
if back_forward {
world.reverse();
}
Ok(world)
};
let mut count = 63usize;
loop {
let times: Vec<f64> = (0..=count).map(|j| j as f64 / count as f64).collect();
let splits: Vec<Split> = times.iter().map(|&t| split_at(t)).collect::<Result<_, _>>()?;
let shape = signature(&splits[0])?;
for split in &splits[1..] {
if signature(split)? != shape {
return Err(refuse("loft_structure", "the grazing loop changes its structure along the curve".into()));
}
}
let grazing = shape.2.clone();
if grazing.is_empty() {
return Err(refuse("no_grazing", "the tool has no grazing loop under this motion".into()));
}
let scale = splits[0].scale + screw.extent;
let mut walls = Vec::with_capacity(grazing.len());
for &(c, back_forward) in &grazing {
let mut rows: Vec<NurbsCurve> = Vec::with_capacity(times.len());
for (j, split) in splits.iter().enumerate() {
let points = samples_of(split, c, back_forward, times[j])?;
rows.push(frag(crate::interpolate_curve(&points, 5, &lambdas), "loft_row")?);
}
let knots_u = rows[0].knots.clone();
let mut net: Vec<Vec<Vec4>> = Vec::with_capacity(rows[0].control_points.len());
let mut knots_v = Vec::new();
for k in 0..rows[0].control_points.len() {
let column: Vec<Vec3> = rows.iter().map(|row| row.control_points[k].point()).collect::<Result<_, _>>().map_err(|e| refuse("loft", e))?;
let across = frag(crate::interpolate_curve(&column, 5, ×), "loft_column")?;
knots_v = across.knots.clone();
net.push(across.control_points.clone());
}
let surface = frag(NurbsSurface::new(5, 5, knots_u, knots_v, net), "loft_wall")?;
walls.push((c, back_forward, surface, rows.first().cloned().expect("rows"), rows.last().cloned().expect("rows")));
}
let mut worst = 0.0f64;
for j in 0..count {
let t = 0.5 * (times[j] + times[j + 1]);
let split = split_at(t)?;
if signature(&split)? != shape {
return Err(refuse("loft_structure", "the grazing loop changes its structure along the curve".into()));
}
for (c, back_forward, surface, _, _) in &walls {
let exact = samples_of(&split, *c, *back_forward, t)?;
for (i, x) in exact.iter().enumerate() {
worst = worst.max(frag(surface.evaluate(lambdas[i], t), "loft_check")?.sub(*x).length());
}
}
}
if worst > 1e-10 * scale {
if count > 1000 {
return Err(refuse("loft_accuracy", format!("the lofted wall stays {worst:e} off the grazing loop")));
}
count = 2 * count + 1;
continue;
}
let mut paths: HashMap<usize, Vec<Vec3>> = HashMap::new();
for &(c, _) in &grazing {
for r in [splits[0].curves[c].start, splits[0].curves[c].end] {
paths.entry(r).or_insert_with(|| splits.iter().zip(×).map(|(split, &t)| screw.apply(t, split.points[r])).collect());
}
}
return assemble_lofted(screw, &splits[0], &splits[count], &walls, ×, &paths);
}
}
#[allow(clippy::too_many_arguments)]
fn assemble_lofted(
screw: &Screw,
first: &Split,
last: &Split,
walls: &[(usize, bool, NurbsSurface, NurbsCurve, NurbsCurve)],
times: &[f64],
paths: &HashMap<usize, Vec<Vec3>>,
) -> Result<BrepSolid, KernelRefusal> {
let mut shell = Shell::new();
let ends = [(first, screw.t0), (last, screw.t1)];
let mut vertex_at: HashMap<(usize, usize), u64> = HashMap::new();
let mut vertex = |shell: &mut Shell, end: usize, p: usize| -> u64 {
*vertex_at.entry((end, p)).or_insert_with(|| shell.vertex(screw.apply(ends[end].1, ends[end].0.points[p])))
};
let grazing: HashMap<usize, usize> = walls.iter().enumerate().map(|(index, w)| (w.0, index)).collect();
let mut edge_at: HashMap<(usize, usize), u64> = HashMap::new();
let mut edge_curve: HashMap<(usize, usize), NurbsCurve> = HashMap::new();
for end in 0..2 {
let (split, t) = ends[end];
for (c, tool_curve) in split.curves.iter().enumerate() {
let used = split.caps.iter().any(|cap| cap.front == (end == 1) && cap.coedges.iter().any(|e| e.0 == c));
if !used {
continue;
}
let curve = match grazing.get(&c) {
Some(&index) => {
let (_, back_forward, _, row0, row1) = &walls[index];
let row = if end == 0 { row0 } else { row1 };
if *back_forward { frag(row.reversed(), "reversed")? } else { row.clone() }
}
None => screw.move_curve(t, &tool_curve.curve)?,
};
let (a, b) = (vertex(&mut shell, end, tool_curve.start), vertex(&mut shell, end, tool_curve.end));
let edge = shell.edge(curve.clone(), a, b, tool_curve.degenerate)?;
edge_at.insert((end, c), edge);
edge_curve.insert((end, c), curve);
}
}
for end in 0..2 {
let (split, t) = ends[end];
for cap in split.caps.iter().filter(|cap| cap.front == (end == 1)) {
let surface = screw.move_surface(t, &cap.surface)?;
let mut coedges = Vec::with_capacity(cap.coedges.len());
for (c, forward, pcurve) in &cap.coedges {
let pcurve = if grazing.contains_key(c) {
match projected_pcurve(&surface, &edge_curve[&(end, *c)], *forward) {
Ok(p) => p,
Err(error) => {
let samples = split.curves[*c].samples.as_ref().map(|list| {
list.iter()
.map(|&x| crate::project_point_to_surface(&surface, screw.apply(t, x)).map(|f| f.distance).unwrap_or(f64::NAN))
.fold(0.0f64, f64::max)
});
let line = split.curves[*c].curve.degree;
return Err(refuse("loft_cap", format!("{} (curve {c} of degree {line}, end {end}, its exact samples off the cap by {samples:?}, cap degrees {}x{})", error.message, surface.degree_u, surface.degree_v)));
}
}
} else {
pcurve.clone()
};
coedges.push((edge_at[&(end, *c)], *forward, pcurve));
}
shell.face(surface, coedges);
}
}
let mut orbits: HashMap<usize, u64> = HashMap::new();
let mut keys: Vec<usize> = paths.keys().copied().collect();
keys.sort_unstable();
for r in keys {
let curve = frag(crate::interpolate_curve(&paths[&r], 5, times), "orbit")?;
let (a, b) = (vertex(&mut shell, 0, r), vertex(&mut shell, 1, r));
orbits.insert(r, shell.edge(curve, a, b, false)?);
}
for (c, back_forward, surface, _, _) in walls {
let tool_curve = &first.curves[*c];
let (from, to) = if *back_forward { (tool_curve.end, tool_curve.start) } else { (tool_curve.start, tool_curve.end) };
let coedges = vec![
(edge_at[&(0, *c)], !*back_forward, uv_line([0.0, 0.0], [1.0, 0.0])?),
(orbits[&to], true, uv_line([1.0, 0.0], [1.0, 1.0])?),
(edge_at[&(1, *c)], *back_forward, uv_line([1.0, 1.0], [0.0, 1.0])?),
(orbits[&from], false, uv_line([0.0, 1.0], [0.0, 0.0])?),
];
shell.face(surface.clone(), coedges);
}
let body = shell.solid();
let issues = body.validate();
if !issues.is_empty() {
return Err(refuse("assembly", format!("the lofted sweep is not valid: {:?}", issues.first())));
}
check_single_pass(screw, first, &body)?;
Ok(body)
}
fn covered(screw: &Screw, split: &Split, y: Vec3, margin: f64) -> Result<bool, KernelRefusal> {
let reach = y.sub(screw.a).length() + split.scale + screw.extent;
let speed = (screw.omega.abs() * reach + screw.rate_bound()).max(1e-300);
let floor = (screw.t1 - screw.t0) * 1e-7;
let mut t = screw.t0;
let mut steps = 0usize;
loop {
let d = split.distance(screw.invert(t, y));
if d < -margin {
return Ok(true);
}
if t >= screw.t1 {
return Ok(false);
}
if steps > 2_000_000 {
return Err(refuse("probe_budget", "the one-pass check could not walk the motion past a point near the sweep".into()));
}
t = (t + ((d.abs() + margin) / speed).max(floor)).min(screw.t1);
steps += 1;
}
}
fn check_single_pass(screw: &Screw, split: &Split, body: &BrepSolid) -> Result<(), KernelRefusal> {
let step = 1e-6 * split.scale;
let margin = 1e-9 * split.scale;
for face in body.shells.iter().flat_map(|s| &s.faces) {
let surface = &face.surface;
let mut corners: Vec<[f64; 2]> = Vec::new();
for coedge in &face.loops[0].coedges {
let domain = frag(coedge.pcurve.domain(), "domain")?;
for k in 0..4 {
let p = frag(coedge.pcurve.evaluate(domain[0] + (domain[1] - domain[0]) * k as f64 / 4.0), "pcurve")?;
corners.push([p.x, p.y]);
}
}
let middle = corners.iter().fold([0.0, 0.0], |s, c| [s[0] + c[0], s[1] + c[1]]);
let middle = [middle[0] / corners.len() as f64, middle[1] / corners.len() as f64];
let mut probes: Vec<[f64; 2]> = vec![middle];
for c in &corners {
for w in [0.25, 0.5, 0.75, 0.9, 0.99] {
probes.push([middle[0] + (c[0] - middle[0]) * w, middle[1] + (c[1] - middle[1]) * w]);
}
}
if surface.degree_v == 5 {
let [u0, u1] = frag(surface.domain_u(), "domain")?;
for i in 1..8 {
for j in 1..64 {
probes.push([u0 + (u1 - u0) * i as f64 / 8.0, j as f64 / 64.0]);
}
}
}
let ([u0, u1], [v0, v1]) = (frag(surface.domain_u(), "domain")?, frag(surface.domain_v(), "domain")?);
let mut ring: Vec<[f64; 2]> = Vec::new();
for coedge in &face.loops[0].coedges {
let domain = frag(coedge.pcurve.domain(), "domain")?;
for k in 0..32 {
let p = frag(coedge.pcurve.evaluate(domain[0] + (domain[1] - domain[0]) * k as f64 / 32.0), "pcurve")?;
ring.push([p.x, p.y]);
}
}
let (lo, hi) = ring.iter().fold(([f64::INFINITY; 2], [f64::NEG_INFINITY; 2]), |(lo, hi), q| ([lo[0].min(q[0]), lo[1].min(q[1])], [hi[0].max(q[0]), hi[1].max(q[1])]));
for i in 1..12 {
for j in 1..12 {
probes.push([lo[0] + (hi[0] - lo[0]) * i as f64 / 12.0, lo[1] + (hi[1] - lo[1]) * j as f64 / 12.0]);
}
}
let inside = |q: [f64; 2]| -> bool {
let mut crossings = 0;
for k in 0..ring.len() {
let (a, b) = (ring[k], ring[(k + 1) % ring.len()]);
if (a[1] > q[1]) != (b[1] > q[1]) && q[0] < a[0] + (q[1] - a[1]) * (b[0] - a[0]) / (b[1] - a[1]) {
crossings += 1;
}
}
crossings % 2 == 1
};
for [u, v] in probes.into_iter().filter(|&q| inside(q)) {
let p = frag(surface.evaluate(u, v), "probe")?;
let (hu, hv) = (1e-6 * (u1 - u0), 1e-6 * (v1 - v0));
let du = frag(surface.evaluate((u + hu).min(u1), v), "probe")?.sub(frag(surface.evaluate((u - hu).max(u0), v), "probe")?);
let dv = frag(surface.evaluate(u, (v + hv).min(v1)), "probe")?.sub(frag(surface.evaluate(u, (v - hv).max(v0)), "probe")?);
let Ok(n) = du.cross(dv).normalized() else { continue };
let n = if face.same_sense { n } else { n.scale(-1.0) };
if covered(screw, split, p.add(n.scale(step)), margin)? {
return Err(refuse("folds", format!("the sweep passes over a point twice (just outside its own face at {p:?}); not one pass")));
}
}
}
Ok(())
}