bherbruckandClaude Opus 5.5
ccde7a4837
- solvecraft-cli fuzz / fuzz-one: seeded random modelling sessions (boxes, extruded profiles with arcs, revolves; joins and cuts of boxes, cylinders and spheres, fillets, chamfers, shells, holes, patterns) on a grid that makes coincident faces and tangencies common, every step checked (command ok, every body a sound solid via the new Body::validity, volume moving the right way), plus the boolean volume identity on random primitives; one process per seed with a time limit; failures minimised to scripts - shell/offset: corners where a curved face meets another at an angle are refined onto the moved surfaces (Newton); edges that are neither lines nor arcs are rebuilt through their moved points; one shell per piece; faces that would cross say so (not supported yet) instead of leaving a cavity inside out; flat revolved faces count as planes; the unhealed body is tried when the healed one fails - cylinders start their seams at an odd angle (grid-aligned seams broke booleans along them) - partial revolves of profiles touching the axis build no zero-area face on the axis - booleans on bodies in several pieces retry a piece at a time - regression tests for each, and a fixed set of seeds in cargo test Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01AS5Tn3xieewUSg3uxrMVdL
704 lines
28 KiB
Rust
704 lines
28 KiB
Rust
use serde::Serialize;
|
|
use solvecraft_geom::{Mesh, Vec3};
|
|
use truck_meshalgo::prelude::*;
|
|
use truck_modeling as mt;
|
|
|
|
use crate::{KernelError, Result, guard};
|
|
|
|
pub(crate) type Solid = mt::Solid;
|
|
|
|
/// A solid body (one or more closed shells), or a mesh body (triangles only, like an imported
|
|
/// 3MF or STL: it renders, measures and exports, but B-rep features cannot use it). Cheap to
|
|
/// clone; never mutated in place.
|
|
#[derive(Clone, Debug)]
|
|
pub struct Body {
|
|
pub(crate) solid: std::sync::Arc<Solid>,
|
|
/// Set for mesh bodies (`solid` is then empty).
|
|
pub(crate) mesh: Option<std::sync::Arc<crate::meshbody::TriMesh>>,
|
|
/// Display colour (linear RGB 0..1) from an imported file; operations drop it.
|
|
pub(crate) color: Option<[f32; 3]>,
|
|
/// Opacity and face colours for export (set by the document; operations drop it).
|
|
pub(crate) paint: Option<std::sync::Arc<Paint>>,
|
|
/// Lines the user split faces along (Split Face), as polylines: export keeps the faces on
|
|
/// either side apart where it would otherwise join pieces of one surface.
|
|
pub(crate) splits: Option<std::sync::Arc<Vec<Vec<Vec3>>>>,
|
|
}
|
|
|
|
/// How a body looks beyond its colour, for export: its opacity and its faces' own colours.
|
|
#[derive(Clone, Debug, PartialEq)]
|
|
pub struct Paint {
|
|
/// 1 = opaque.
|
|
pub opacity: f32,
|
|
pub faces: Vec<FacePaint>,
|
|
}
|
|
|
|
/// A face's own colour: the B-rep face index (`face_iter` order), sRGB as 0..1, opacity.
|
|
#[derive(Clone, Debug, PartialEq)]
|
|
pub struct FacePaint {
|
|
pub face: usize,
|
|
pub color: [f32; 3],
|
|
pub opacity: f32,
|
|
}
|
|
|
|
/// A B-rep edge as seen by the rest of the application.
|
|
#[derive(Clone, Debug, Serialize)]
|
|
pub struct EdgeInfo {
|
|
pub index: usize,
|
|
/// Polyline approximation.
|
|
pub points: Vec<Vec3>,
|
|
/// Point halfway along the polyline (used to re-find the edge).
|
|
pub mid: Vec3,
|
|
pub length: f64,
|
|
}
|
|
|
|
/// A B-rep face.
|
|
#[derive(Clone, Debug, Serialize)]
|
|
pub struct FaceInfo {
|
|
pub index: usize,
|
|
pub area: f64,
|
|
pub centroid: Vec3,
|
|
/// Unit normal if the face is planar.
|
|
pub plane_normal: Option<Vec3>,
|
|
}
|
|
|
|
type MeshedShell = truck_topology::Shell<mt::Point3, PolylineCurve<mt::Point3>, Option<PolygonMesh>>;
|
|
|
|
/// Triangulate each shell (not the solid: truck's solid constructor panics on the open or
|
|
/// non-manifold shells an import can produce). Boundaries that lie on their surface only within
|
|
/// tolerance (imports, boolean results) are projected onto it.
|
|
fn mesh_shells(solid: &Solid, tol: f64) -> Vec<MeshedShell> {
|
|
solid
|
|
.boundaries()
|
|
.iter()
|
|
// The robust search tries the exact one first, so exact geometry costs nothing extra; a
|
|
// face can also come back as an empty mesh rather than none (boolean results with
|
|
// approximate intersection curves), so always use it.
|
|
.map(|sh| sh.robust_triangulation(tol))
|
|
.collect()
|
|
}
|
|
|
|
/// A face's triangles (corner positions and normals, in the mesh's winding) with every edge
|
|
/// whose middle sags more than `tol` from the surface split there, found through the mesh's
|
|
/// surface parameters. A split depends on the edge alone, so neighbours stay conforming.
|
|
/// `None` when nothing needs splitting or the parameters don't match the surface.
|
|
#[allow(clippy::type_complexity)]
|
|
fn refined(pm: &PolygonMesh, surf: &mt::Surface, tol: f64) -> Option<Vec<[(Vec3, Vec3); 3]>> {
|
|
use mt::{ParametricSurface, ParametricSurface3D};
|
|
let (pos, uvs, nor) = (pm.positions(), pm.uv_coords(), pm.normals());
|
|
if uvs.is_empty() {
|
|
return None;
|
|
}
|
|
type V = (Vec3, (f64, f64), Vec3);
|
|
let mut tris: Vec<[V; 3]> = Vec::new();
|
|
for tri in pm.faces().triangle_iter() {
|
|
let mut t = [(Vec3::ZERO, (0.0, 0.0), Vec3::Z); 3];
|
|
for (slot, v) in t.iter_mut().zip(tri.iter()) {
|
|
let p = pos.get(v.pos)?;
|
|
let uv = uvs.get(v.uv?)?;
|
|
let n = v.nor.and_then(|i| nor.get(i)).map(|n| Vec3::new(n.x, n.y, n.z)).unwrap_or(Vec3::Z);
|
|
*slot = (Vec3::new(p.x, p.y, p.z), (uv.x, uv.y), n);
|
|
}
|
|
tris.push(t);
|
|
}
|
|
// The parameters must be the surface's own.
|
|
let check = tris.iter().take(8).flat_map(|t| t.iter()).all(|(p, (u, v), _)| from_p3(surf.subs(*u, *v)).dist(*p) < tol * 4.0 + 1e-9);
|
|
// Bounded work: a face may grow to a few times its triangles (wild surfaces sag
|
|
// everywhere and are left as meshed).
|
|
let cap = (tris.len() * 8).clamp(2_000, 60_000);
|
|
if !check || tris.len() > 50_000 {
|
|
return None;
|
|
}
|
|
// Edges on the face's boundary are shared with the next face: never split them.
|
|
let k = |x: &V| ((x.1.0 * 1e9).round() as i64, (x.1.1 * 1e9).round() as i64);
|
|
let ek = |a: &V, b: &V| if k(a) < k(b) { (k(a), k(b)) } else { (k(b), k(a)) };
|
|
let mut changed = false;
|
|
for _ in 0..4 {
|
|
let mut uses: std::collections::HashMap<((i64, i64), (i64, i64)), u32> = std::collections::HashMap::new();
|
|
for [a, b, c] in &tris {
|
|
for e in [ek(a, b), ek(b, c), ek(c, a)] {
|
|
*uses.entry(e).or_default() += 1;
|
|
}
|
|
}
|
|
let at = |a: &V, b: &V| -> Option<V> {
|
|
if uses.get(&ek(a, b)).copied().unwrap_or(0) < 2 {
|
|
return None;
|
|
}
|
|
// Not at a degenerate point (a cone's apex: its parameters there are arbitrary).
|
|
let singular = |x: &V| {
|
|
let (du, dv) = (surf.uder(x.1.0, x.1.1), surf.vder(x.1.0, x.1.1));
|
|
let c = du.cross(dv);
|
|
c.x * c.x + c.y * c.y + c.z * c.z
|
|
<= 1e-12 * (du.x * du.x + du.y * du.y + du.z * du.z + dv.x * dv.x + dv.y * dv.y + dv.z * dv.z).powi(2) + 1e-300
|
|
};
|
|
if singular(a) || singular(b) {
|
|
return None;
|
|
}
|
|
let (u, v) = ((a.1.0 + b.1.0) / 2.0, (a.1.1 + b.1.1) / 2.0);
|
|
let q = from_p3(surf.subs(u, v));
|
|
// Within tolerance, or so far off that the parameters wrapped (a seam).
|
|
let sag = q.dist((a.0 + b.0) * 0.5);
|
|
if sag <= tol || sag >= 0.5 * a.0.dist(b.0) {
|
|
return None;
|
|
}
|
|
let n = surf.normal(u, v);
|
|
let mut n = Vec3::new(n.x, n.y, n.z);
|
|
// Keep the mesh's normal side.
|
|
if n.dot(a.2 + b.2) < 0.0 {
|
|
n = -n;
|
|
}
|
|
Some((q, (u, v), n))
|
|
};
|
|
let mut next = Vec::with_capacity(tris.len());
|
|
let mut any = false;
|
|
for t in &tris {
|
|
let [a, b, c] = t;
|
|
let m = [at(a, b), at(b, c), at(c, a)];
|
|
any |= m.iter().any(Option::is_some);
|
|
match m {
|
|
[None, None, None] => next.push(*t),
|
|
[Some(x), Some(y), Some(z)] => {
|
|
next.extend([[*a, x, z], [x, *b, y], [z, y, *c], [x, y, z]]);
|
|
}
|
|
[Some(x), None, None] => next.extend([[*a, x, *c], [x, *b, *c]]),
|
|
[None, Some(y), None] => next.extend([[*a, *b, y], [*a, y, *c]]),
|
|
[None, None, Some(z)] => next.extend([[*a, *b, z], [z, *b, *c]]),
|
|
[Some(x), Some(y), None] => next.extend([[*a, x, *c], [x, *b, y], [x, y, *c]]),
|
|
[None, Some(y), Some(z)] => next.extend([[*a, *b, z], [z, *b, y], [z, y, *c]]),
|
|
[Some(x), None, Some(z)] => next.extend([[*a, x, z], [x, *b, *c], [x, *c, z]]),
|
|
}
|
|
}
|
|
tris = next;
|
|
changed |= any;
|
|
if !any || tris.len() > cap {
|
|
break;
|
|
}
|
|
}
|
|
changed.then(|| tris.into_iter().map(|t| t.map(|(p, _, n)| (p, n))).collect())
|
|
}
|
|
|
|
/// Revolved surfaces are periodic in their angle; a face whose angles start at 0 is found at 0
|
|
/// or 2π, which confuses the boolean's loop projection. Each revolved face's surface is turned
|
|
/// (same geometry) so the face sits around angle π, well away from the wrap.
|
|
fn rephase_revolved(solid: Solid) -> Solid {
|
|
use mt::{BoundedCurve, ParametricCurve, SearchNearestParameter};
|
|
let r = guard("rephase", || {
|
|
let mut changed = false;
|
|
let mut shells = Vec::new();
|
|
for sh in solid.boundaries() {
|
|
let mut faces = Vec::new();
|
|
for f in sh.face_iter() {
|
|
let mt::Surface::RevolutedCurve(proc_) = f.surface() else {
|
|
faces.push(f.clone());
|
|
continue;
|
|
};
|
|
let surf = f.surface();
|
|
// The face's middle angle.
|
|
let mut acc = Vec3::ZERO;
|
|
let mut n = 0.0f64;
|
|
for w in f.absolute_boundaries() {
|
|
for e in w.edge_iter() {
|
|
let c = e.oriented_curve();
|
|
let (t0, t1) = c.range_tuple();
|
|
for k in 0..4 {
|
|
acc += from_p3(c.subs(t0 + (t1 - t0) * k as f64 / 4.0));
|
|
n += 1.0;
|
|
}
|
|
}
|
|
}
|
|
let Some((_, v)) = surf.search_nearest_parameter(p3(acc / n.max(1.0)), None, 100) else {
|
|
faces.push(f.clone());
|
|
continue;
|
|
};
|
|
let phi = v.rem_euclid(std::f64::consts::TAU) - std::f64::consts::PI;
|
|
if phi.abs() < 0.3 {
|
|
faces.push(f.clone());
|
|
continue;
|
|
}
|
|
let new_proc = proc_.map_ref(|rc| {
|
|
let (o, a) = (rc.origin(), rc.axis());
|
|
let rot = mt::Matrix4::from_translation(mt::EuclideanSpace::to_vec(o))
|
|
* mt::Matrix4::from_axis_angle(a, mt::Rad(phi))
|
|
* mt::Matrix4::from_translation(-mt::EuclideanSpace::to_vec(o));
|
|
mt::RevolutedCurve::by_revolution(mt::Transformed::transformed(rc.entity_curve(), rot), o, a)
|
|
});
|
|
match mt::Face::try_new(f.absolute_boundaries().clone(), mt::Surface::RevolutedCurve(new_proc)) {
|
|
Ok(mut nf) => {
|
|
if !f.orientation() {
|
|
nf.invert();
|
|
}
|
|
faces.push(nf);
|
|
changed = true;
|
|
}
|
|
Err(_) => faces.push(f.clone()),
|
|
}
|
|
}
|
|
shells.push(mt::Shell::from(faces));
|
|
}
|
|
if !changed {
|
|
return Ok(None);
|
|
}
|
|
Ok(Solid::try_new(shells).ok())
|
|
});
|
|
match r {
|
|
Ok(Some(s)) => s,
|
|
_ => solid,
|
|
}
|
|
}
|
|
|
|
pub(crate) fn p3(v: Vec3) -> mt::Point3 {
|
|
mt::Point3::new(v.x, v.y, v.z)
|
|
}
|
|
pub(crate) fn v3(v: Vec3) -> mt::Vector3 {
|
|
mt::Vector3::new(v.x, v.y, v.z)
|
|
}
|
|
pub(crate) fn from_p3(p: mt::Point3) -> Vec3 {
|
|
Vec3::new(p.x, p.y, p.z)
|
|
}
|
|
|
|
fn polyline_mid(pts: &[Vec3]) -> (Vec3, f64) {
|
|
let len: f64 = pts.windows(2).map(|w| w.first().zip(w.get(1)).map(|(a, b)| a.dist(*b)).unwrap_or(0.0)).sum();
|
|
let mut acc = 0.0;
|
|
for w in pts.windows(2) {
|
|
if let (Some(a), Some(b)) = (w.first(), w.get(1)) {
|
|
let l = a.dist(*b);
|
|
if acc + l >= len * 0.5 && l > 0.0 {
|
|
return (a.lerp(*b, (len * 0.5 - acc) / l), len);
|
|
}
|
|
acc += l;
|
|
}
|
|
}
|
|
(pts.first().copied().unwrap_or_default(), len)
|
|
}
|
|
|
|
impl Body {
|
|
/// The very same body (shared data), not just an equal-looking one.
|
|
pub fn same(&self, other: &Body) -> bool {
|
|
std::sync::Arc::ptr_eq(&self.solid, &other.solid)
|
|
&& match (&self.mesh, &other.mesh) {
|
|
(Some(a), Some(b)) => std::sync::Arc::ptr_eq(a, b),
|
|
(None, None) => true,
|
|
_ => false,
|
|
}
|
|
&& self.color == other.color
|
|
&& self.paint == other.paint
|
|
&& self.splits == other.splits
|
|
}
|
|
|
|
pub(crate) fn new(solid: Solid) -> Result<Body> {
|
|
if solid.boundaries().is_empty() {
|
|
return Err(KernelError::Failed("empty result".into()));
|
|
}
|
|
let mut solid = rephase_revolved(solid);
|
|
// Orient outward: a closed solid must have positive volume (meshed finely enough that
|
|
// every face comes out: a face missing from a coarse mesh can flip the sign).
|
|
let span = {
|
|
let mut lo = Vec3::new(f64::INFINITY, f64::INFINITY, f64::INFINITY);
|
|
let mut hi = -lo;
|
|
for v in solid.vertex_iter() {
|
|
let p = from_p3(v.point());
|
|
lo = Vec3::new(lo.x.min(p.x), lo.y.min(p.y), lo.z.min(p.z));
|
|
hi = Vec3::new(hi.x.max(p.x), hi.y.max(p.y), hi.z.max(p.z));
|
|
}
|
|
if lo.x.is_finite() { (hi - lo).len() } else { 1.0 }
|
|
};
|
|
let tol = (span * 0.01).clamp(1e-3, 1.0);
|
|
let v = guard("orient", || Ok(mesh_shells(&solid, tol).iter().map(|s| s.to_polygon().volume()).sum::<f64>()))?;
|
|
if v < 0.0 {
|
|
solid.not();
|
|
}
|
|
Ok(Body { solid: std::sync::Arc::new(solid), mesh: None, color: None, paint: None, splits: None })
|
|
}
|
|
|
|
/// A deep copy of the solid that can be mutated without affecting other bodies.
|
|
pub(crate) fn deep_copy(&self) -> Solid {
|
|
mt::builder::clone(&*self.solid)
|
|
}
|
|
|
|
/// Is this a mesh body (no B-rep)?
|
|
pub fn is_mesh(&self) -> bool {
|
|
self.mesh.is_some()
|
|
}
|
|
|
|
/// Error for operations that need a B-rep.
|
|
pub(crate) fn require_brep(&self, what: &str) -> Result<()> {
|
|
if self.is_mesh() {
|
|
Err(KernelError::Invalid(format!("{what} needs a solid body; mesh bodies can only be moved, measured and exported")))
|
|
} else {
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
/// Display colour from an imported file.
|
|
pub fn color(&self) -> Option<[f32; 3]> {
|
|
self.color
|
|
}
|
|
|
|
/// The same body with a display colour.
|
|
pub fn with_color(mut self, c: Option<[f32; 3]>) -> Body {
|
|
self.color = c.filter(|c| c.iter().all(|x| x.is_finite())).map(|c| c.map(|x| x.clamp(0.0, 1.0)));
|
|
self
|
|
}
|
|
|
|
/// Lines faces were split along on purpose (Split Face), as polylines.
|
|
pub fn split_lines(&self) -> Option<&std::sync::Arc<Vec<Vec<Vec3>>>> {
|
|
self.splits.as_ref()
|
|
}
|
|
|
|
pub(crate) fn split_keep(&self) -> &[Vec<Vec3>] {
|
|
self.splits.as_ref().map(|l| l.as_slice()).unwrap_or(&[])
|
|
}
|
|
|
|
/// The same body with the lines its faces were split along on purpose.
|
|
pub fn with_split_lines(mut self, s: Option<std::sync::Arc<Vec<Vec<Vec3>>>>) -> Body {
|
|
self.splits = s.filter(|l| !l.is_empty());
|
|
self
|
|
}
|
|
|
|
/// Opacity and face colours for export, if any.
|
|
pub fn paint(&self) -> Option<&Paint> {
|
|
self.paint.as_deref()
|
|
}
|
|
|
|
/// The same body with opacity and face colours for export (`None` or opaque with no face
|
|
/// colours clears them).
|
|
pub fn with_paint(mut self, p: Option<Paint>) -> Body {
|
|
let clean = |x: f32| if x.is_finite() { x.clamp(0.0, 1.0) } else { 1.0 };
|
|
self.paint = p
|
|
.map(|mut p| {
|
|
p.opacity = clean(p.opacity);
|
|
p.faces.retain(|f| f.color.iter().all(|x| x.is_finite()));
|
|
for f in &mut p.faces {
|
|
f.color = f.color.map(|x| x.clamp(0.0, 1.0));
|
|
f.opacity = clean(f.opacity);
|
|
}
|
|
p
|
|
})
|
|
.filter(|p| p.opacity < 1.0 || !p.faces.is_empty())
|
|
.map(std::sync::Arc::new);
|
|
self
|
|
}
|
|
|
|
/// Characteristic size (bounding box diagonal) for tolerances.
|
|
pub fn size(&self) -> f64 {
|
|
let mut b = solvecraft_geom::Aabb3::EMPTY;
|
|
if let Some(m) = &self.mesh {
|
|
for p in &m.positions {
|
|
b.add(*p);
|
|
}
|
|
}
|
|
for v in self.solid.vertex_iter() {
|
|
b.add(from_p3(v.point()));
|
|
}
|
|
let d = b.diagonal();
|
|
if d.is_finite() && d > 0.0 { d } else { 1.0 }
|
|
}
|
|
|
|
/// Distinct edges in a stable order (the order of `edges()` and `tessellate().edges`).
|
|
pub(crate) fn unique_edges(solid: &Solid) -> Vec<mt::Edge> {
|
|
let mut seen = std::collections::HashSet::new();
|
|
solid.edge_iter().filter(|e| seen.insert(e.id())).collect()
|
|
}
|
|
|
|
/// Triangulate with chord tolerance `tol` (mm). Triangles carry their face index.
|
|
pub fn tessellate(&self, tol: f64) -> Result<Mesh> {
|
|
self.tessellate_with(tol, false)
|
|
}
|
|
|
|
/// Like [`Body::tessellate`]; with `refine`, triangles that cut through curved faces are
|
|
/// split until within `tol` of the surface (for measuring).
|
|
pub fn tessellate_with(&self, tol: f64, refine: bool) -> Result<Mesh> {
|
|
let tol = if tol.is_finite() && tol > 0.0 { tol } else { 0.05 };
|
|
let tol = tol.max(self.size() * 1e-6);
|
|
if let Some(m) = &self.mesh {
|
|
return Ok(m.to_mesh());
|
|
}
|
|
guard("tessellate", || {
|
|
let meshed = mesh_shells(&self.solid, tol);
|
|
let mut out = Mesh::default();
|
|
let solid_faces: Vec<&mt::Face> = self.solid.face_iter().collect();
|
|
for (fi, face) in meshed.iter().flat_map(|s| s.face_iter()).enumerate() {
|
|
let Some(pm) = face.surface() else { continue };
|
|
let flip = !face.orientation();
|
|
let base = u32::try_from(out.positions.len()).unwrap_or(u32::MAX);
|
|
// Curved faces: split triangles that cut through the surface.
|
|
if refine
|
|
&& let Some(sf) = solid_faces.get(fi)
|
|
&& !matches!(sf.surface(), mt::Surface::Plane(_))
|
|
&& let Some(tris) = refined(&pm, &sf.surface(), tol)
|
|
{
|
|
let fid = u32::try_from(fi).unwrap_or(u32::MAX);
|
|
for t in tris {
|
|
let mut idx = [0u32; 3];
|
|
for (slot, (p, n)) in idx.iter_mut().zip(t) {
|
|
out.positions.push(p);
|
|
out.normals.push(if flip { -n } else { n });
|
|
*slot = u32::try_from(out.positions.len() - 1).unwrap_or(0);
|
|
}
|
|
out.triangles.push(if flip { [idx[0], idx[2], idx[1]] } else { idx });
|
|
out.tri_face.push(fid);
|
|
}
|
|
continue;
|
|
}
|
|
let pos = pm.positions();
|
|
let nor = pm.normals();
|
|
// Expand to per-corner vertices so normals stay per face (crisp edges).
|
|
for tri in pm.faces().triangle_iter() {
|
|
let mut idx = [0u32; 3];
|
|
for (k, v) in tri.iter().enumerate() {
|
|
let Some(p) = pos.get(v.pos) else { continue };
|
|
let n = v.nor.and_then(|ni| nor.get(ni)).map(|n| Vec3::new(n.x, n.y, n.z)).unwrap_or(Vec3::Z);
|
|
let n = if flip { -n } else { n };
|
|
out.positions.push(Vec3::new(p.x, p.y, p.z));
|
|
out.normals.push(n);
|
|
if let Some(slot) = idx.get_mut(k) {
|
|
*slot = u32::try_from(out.positions.len() - 1).unwrap_or(0);
|
|
}
|
|
}
|
|
let t = if flip { [idx[0], idx[2], idx[1]] } else { idx };
|
|
out.triangles.push(t);
|
|
out.tri_face.push(u32::try_from(fi).unwrap_or(u32::MAX));
|
|
}
|
|
let _ = base;
|
|
}
|
|
// Fill missing normals from geometry.
|
|
for t in &out.triangles {
|
|
let (Some(a), Some(b), Some(c)) =
|
|
(out.positions.get(t[0] as usize), out.positions.get(t[1] as usize), out.positions.get(t[2] as usize))
|
|
else {
|
|
continue;
|
|
};
|
|
let fnrm = (b - a).cross(c - a).normalized().unwrap_or(Vec3::Z);
|
|
for k in t {
|
|
if let Some(n) = out.normals.get_mut(*k as usize)
|
|
&& (n.dot(fnrm) < -0.2 || n.len() < 0.5)
|
|
{
|
|
*n = fnrm;
|
|
}
|
|
}
|
|
}
|
|
let mut seen = std::collections::HashMap::new();
|
|
for e in meshed.iter().flat_map(|s| s.edge_iter()) {
|
|
if seen.contains_key(&e.id()) {
|
|
continue;
|
|
}
|
|
seen.insert(e.id(), out.edges.len());
|
|
let c = e.oriented_curve();
|
|
out.edges.push(c.0.iter().map(|p| Vec3::new(p.x, p.y, p.z)).collect());
|
|
}
|
|
// Which faces each edge bounds.
|
|
out.edge_faces = vec![Vec::new(); out.edges.len()];
|
|
for (fi, face) in meshed.iter().flat_map(|s| s.face_iter()).enumerate() {
|
|
for e in face.edge_iter() {
|
|
if let Some(slot) = seen.get(&e.id()).and_then(|i| out.edge_faces.get_mut(*i)) {
|
|
let f = u32::try_from(fi).unwrap_or(u32::MAX);
|
|
if !slot.contains(&f) {
|
|
slot.push(f);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
Ok(out)
|
|
})
|
|
}
|
|
|
|
/// Faces with no triangles in a coarse tessellation (the mesher failed or gave up).
|
|
pub fn unmeshed_faces(&self) -> usize {
|
|
if self.is_mesh() {
|
|
return 0;
|
|
}
|
|
let Ok(m) = self.tessellate((self.size() * 1e-2).max(1e-3)) else { return self.face_count() };
|
|
let mut has = vec![false; self.face_count()];
|
|
for f in &m.tri_face {
|
|
if let Some(h) = has.get_mut(*f as usize) {
|
|
*h = true;
|
|
}
|
|
}
|
|
has.iter().filter(|h| !**h).count()
|
|
}
|
|
|
|
/// Display mesh: like [`Body::tessellate`], with seam edges flagged.
|
|
pub fn display_mesh(&self, tol: f64) -> Result<Mesh> {
|
|
let mut m = self.tessellate(tol)?;
|
|
if self.is_mesh() {
|
|
return Ok(m);
|
|
}
|
|
m.seams = crate::topo::seam_flags(self, &m);
|
|
Ok(m)
|
|
}
|
|
|
|
/// Edges with polylines (index = position in [`Body::tessellate`]'s `edges`).
|
|
pub fn edges(&self, tol: f64) -> Result<Vec<EdgeInfo>> {
|
|
let m = self.tessellate(tol)?;
|
|
Ok(m.edges
|
|
.into_iter()
|
|
.enumerate()
|
|
.map(|(index, points)| {
|
|
let (mid, length) = polyline_mid(&points);
|
|
EdgeInfo { index, points, mid, length }
|
|
})
|
|
.collect())
|
|
}
|
|
|
|
/// Faces with area and centroid.
|
|
pub fn faces(&self, tol: f64) -> Result<Vec<FaceInfo>> {
|
|
let m = self.tessellate(tol)?;
|
|
let nf = self.face_count();
|
|
let mut acc: Vec<(f64, Vec3, Vec3, bool)> = vec![(0.0, Vec3::ZERO, Vec3::ZERO, true); nf];
|
|
for (t, f) in m.triangles.iter().zip(&m.tri_face) {
|
|
let Some([a, b, c]) = m.tri(t) else { continue };
|
|
let cr = (b - a).cross(c - a);
|
|
let ar = cr.len() * 0.5;
|
|
if let Some(slot) = acc.get_mut(*f as usize) {
|
|
slot.0 += ar;
|
|
slot.1 += (a + b + c) * (ar / 3.0);
|
|
if let Some(n) = cr.normalized() {
|
|
if slot.2 == Vec3::ZERO {
|
|
slot.2 = n;
|
|
} else if slot.2.dot(n) < 1.0 - 1e-6 {
|
|
slot.3 = false;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
Ok(acc
|
|
.into_iter()
|
|
.enumerate()
|
|
.map(|(index, (area, c, n, planar))| FaceInfo {
|
|
index,
|
|
area,
|
|
centroid: if area > 0.0 { c / area } else { Vec3::ZERO },
|
|
plane_normal: (planar && n != Vec3::ZERO).then_some(n),
|
|
})
|
|
.collect())
|
|
}
|
|
|
|
/// Index of the edge closest to `p` (by polyline distance).
|
|
pub fn nearest_edge(&self, p: Vec3, tol: f64) -> Result<Option<(usize, f64)>> {
|
|
let edges = self.edges(tol)?;
|
|
let mut best: Option<(usize, f64)> = None;
|
|
for e in &edges {
|
|
for w in e.points.windows(2) {
|
|
if let (Some(a), Some(b)) = (w.first(), w.get(1)) {
|
|
let d = p.dist_to_segment(*a, *b);
|
|
if best.is_none_or(|(_, bd)| d < bd) {
|
|
best = Some((e.index, d));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
Ok(best)
|
|
}
|
|
|
|
/// Is the point inside the solid?
|
|
pub fn contains(&self, p: Vec3) -> bool {
|
|
self.tessellate(self.size() * 2e-3).map(|m| m.contains(p)).unwrap_or(false)
|
|
}
|
|
|
|
/// B-rep faces (a mesh body is one face; [`crate::measure`] counts its triangles).
|
|
pub fn face_count(&self) -> usize {
|
|
match &self.mesh {
|
|
Some(_) => 1,
|
|
None => self.solid.face_iter().count(),
|
|
}
|
|
}
|
|
/// B-rep edges (none for a mesh body).
|
|
pub fn edge_count(&self) -> usize {
|
|
match &self.mesh {
|
|
Some(_) => 0,
|
|
None => Self::unique_edges(&self.solid).len(),
|
|
}
|
|
}
|
|
/// B-rep vertices (none for a mesh body).
|
|
pub fn vertex_count(&self) -> usize {
|
|
if self.mesh.is_some() {
|
|
return 0;
|
|
}
|
|
let mut seen = std::collections::HashSet::new();
|
|
self.solid.vertex_iter().filter(|v| seen.insert(v.id())).count()
|
|
}
|
|
/// What is wrong with the body as a solid: shells that aren't closed and consistently
|
|
/// oriented, faces that don't mesh, a volume that isn't positive. Empty when it is sound.
|
|
pub fn validity(&self) -> Vec<String> {
|
|
if let Some(m) = &self.mesh {
|
|
return if crate::meshbody::is_closed(m) { Vec::new() } else { vec!["the mesh is not closed".into()] };
|
|
}
|
|
let mut out = Vec::new();
|
|
for (i, sh) in self.solid.boundaries().iter().enumerate() {
|
|
let c = sh.shell_condition();
|
|
if c != truck_topology::shell::ShellCondition::Closed {
|
|
out.push(format!("shell {i} is {c:?}, not closed"));
|
|
}
|
|
}
|
|
let missing = self.unmeshed_faces();
|
|
if missing > 0 {
|
|
out.push(format!("{missing} face(s) don't mesh"));
|
|
}
|
|
match crate::measure(self) {
|
|
Ok(m) if m.volume.is_finite() && m.volume > 0.0 => {}
|
|
Ok(m) => out.push(format!("volume {}", m.volume)),
|
|
Err(e) => out.push(format!("no measure: {e}")),
|
|
}
|
|
out
|
|
}
|
|
|
|
pub fn shell_count(&self) -> usize {
|
|
match &self.mesh {
|
|
Some(_) => 1,
|
|
None => self.solid.boundaries().len(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Body {
|
|
/// The separate pieces of a body: each outer shell with the voids inside it. A body in one
|
|
/// piece comes back as itself.
|
|
pub fn lumps(&self) -> Result<Vec<Body>> {
|
|
if self.mesh.is_some() || self.solid.boundaries().len() < 2 {
|
|
return Ok(vec![self.clone()]);
|
|
}
|
|
let tol = (self.size() * 1e-3).max(1e-3);
|
|
guard("lumps", || {
|
|
let shells: Vec<mt::Shell> = self.solid.boundaries().clone();
|
|
let meshes: Vec<(f64, Option<Mesh>)> = shells
|
|
.iter()
|
|
.map(|sh| {
|
|
let pm = sh.robust_triangulation(tol).to_polygon();
|
|
let v = pm.volume();
|
|
let mut m = Mesh::default();
|
|
for p in pm.positions() {
|
|
m.positions.push(Vec3::new(p.x, p.y, p.z));
|
|
}
|
|
for t in pm.faces().triangle_iter() {
|
|
let idx = |k: usize| t.get(k).map(|x| u32::try_from(x.pos).unwrap_or(0)).unwrap_or(0);
|
|
m.triangles.push([idx(0), idx(1), idx(2)]);
|
|
m.tri_face.push(0);
|
|
}
|
|
(v, Some(m))
|
|
})
|
|
.collect();
|
|
let outer: Vec<usize> = (0..shells.len()).filter(|i| meshes.get(*i).is_some_and(|m| m.0 > 0.0)).collect();
|
|
if outer.len() < 2 {
|
|
return Ok(vec![self.clone()]);
|
|
}
|
|
let mut groups: Vec<Vec<mt::Shell>> = outer.iter().filter_map(|i| shells.get(*i).cloned()).map(|s| vec![s]).collect();
|
|
for (i, sh) in shells.iter().enumerate() {
|
|
if outer.contains(&i) {
|
|
continue;
|
|
}
|
|
// A void goes with the outer shell around one of its points.
|
|
let probe = sh.vertex_iter().next().map(|v| from_p3(v.point()));
|
|
let owner = outer.iter().position(|o| {
|
|
let (Some(p), Some((_, Some(m)))) = (probe, meshes.get(*o)) else { return false };
|
|
m.contains(p)
|
|
});
|
|
if let Some(k) = owner
|
|
&& let Some(g) = groups.get_mut(k)
|
|
{
|
|
g.push(sh.clone());
|
|
}
|
|
}
|
|
groups.into_iter().map(|g| Solid::try_new(g).map_err(|e| KernelError::Failed(format!("lump: {e}"))).and_then(Body::new)).collect()
|
|
})
|
|
}
|
|
}
|