use crate::fit::solve_dense;
use crate::topology::{BrepSolid, CoedgeRecord, EdgeRecord, FaceRecord, LoopRecord, VertexRecord};
use crate::{
interpolate_curve, measure_edge_against_pcurve_image,
measure_surface_fit_against_pointwise_offset, offset_construction_band, solid_model_scale,
vertex_endpoint_gap, vertex_tolerance_from_edges, KnotVector, MeasuredTolerance, NurbsCurve,
NurbsSurface, OffsetEvaluator, OffsetNormal, Vec2, Vec3, Vec4,
};
use rustc_hash::FxHashMap as HashMap;
use serde::Serialize;
fn domains(surface: &NurbsSurface) -> Result<([f64; 2], [f64; 2]), String> {
Ok((
KnotVector::new(surface.knots_u.clone(), surface.degree_u)?.domain(),
KnotVector::new(surface.knots_v.clone(), surface.degree_v)?.domain(),
))
}
fn stable_face_normal(face: &FaceRecord, u: f64, v: f64) -> Result<Vec3, String> {
OffsetEvaluator::new(
"offset_surface",
&face.surface,
OffsetNormal::FaceStable {
same_sense: face.same_sense,
},
)
.normal(u, v)
}
fn greville_parameters(knots: &KnotVector) -> Vec<f64> {
let mut parameters = (0..knots.control_point_count())
.map(|index| {
knots.knots[index + 1..=index + knots.degree]
.iter()
.sum::<f64>()
/ knots.degree as f64
})
.collect::<Vec<_>>();
let domain = knots.domain();
parameters[0] = domain[0];
*parameters.last_mut().unwrap() = domain[1];
parameters
}
fn collocation_matrix(knots: &KnotVector, parameters: &[f64], weights: &[f64]) -> Vec<Vec<f64>> {
parameters
.iter()
.map(|parameter| {
let mut row = vec![0.0; knots.control_point_count()];
let span = knots.find_span(*parameter);
for (offset, value) in knots
.basis_functions(span, *parameter)
.into_iter()
.enumerate()
{
let index = span - knots.degree + offset;
row[index] = value * weights[index];
}
let denominator: f64 = row.iter().sum();
if denominator.abs() > 0.0 {
for value in &mut row {
*value /= denominator;
}
}
row
})
.collect()
}
fn separable_weights(weights: &[Vec<f64>]) -> Option<(Vec<f64>, Vec<f64>)> {
let first_row = weights.first()?;
let anchor = *first_row.first()?;
if anchor.abs() <= 1e-12 {
return None;
}
let a: Vec<f64> = weights.iter().map(|row| row[0]).collect();
let b: Vec<f64> = first_row.iter().map(|w| w / anchor).collect();
for (i, row) in weights.iter().enumerate() {
for (j, &w) in row.iter().enumerate() {
if (w - a[i] * b[j]).abs() > 1e-10 * (1.0 + w.abs()) {
return None;
}
}
}
Some((a, b))
}
fn interpolate_tensor(
knot_u: &KnotVector,
knot_v: &KnotVector,
parameters_u: &[f64],
parameters_v: &[f64],
samples: &[Vec<Vec3>],
weights: &[Vec<f64>],
) -> Result<Vec<Vec<Vec4>>, String> {
let count_u = parameters_u.len();
let count_v = parameters_v.len();
if let Some((weights_u, weights_v)) = separable_weights(weights) {
let matrix_u = collocation_matrix(knot_u, parameters_u, &weights_u);
let matrix_v = collocation_matrix(knot_v, parameters_v, &weights_v);
let mut intermediate = vec![vec![Vec3::default(); count_v]; count_u];
for column in 0..count_v {
let solve_axis = |axis: fn(Vec3) -> f64| {
solve_dense(
matrix_u.clone(),
samples.iter().map(|row| axis(row[column])).collect(),
)
};
let x = solve_axis(|point| point.x)?;
let y = solve_axis(|point| point.y)?;
let z = solve_axis(|point| point.z)?;
for row in 0..count_u {
intermediate[row][column] = Vec3::new(x[row], y[row], z[row]);
}
}
let mut controls = vec![vec![Vec4::from_point(Vec3::default(), 1.0); count_v]; count_u];
for row in 0..count_u {
let solve_axis = |axis: fn(Vec3) -> f64| {
solve_dense(
matrix_v.clone(),
intermediate[row].iter().copied().map(axis).collect(),
)
};
let x = solve_axis(|point| point.x)?;
let y = solve_axis(|point| point.y)?;
let z = solve_axis(|point| point.z)?;
for column in 0..count_v {
controls[row][column] = Vec4::from_point(
Vec3::new(x[column], y[column], z[column]),
weights[row][column],
);
}
}
return Ok(controls);
}
let unknowns = count_u * count_v;
let mut matrix = vec![vec![0.0; unknowns]; unknowns];
for (k, &u) in parameters_u.iter().enumerate() {
let span_u = knot_u.find_span(u);
let basis_u = knot_u.basis_functions(span_u, u);
for (l, &v) in parameters_v.iter().enumerate() {
let span_v = knot_v.find_span(v);
let basis_v = knot_v.basis_functions(span_v, v);
let row = &mut matrix[k * count_v + l];
let mut denominator = 0.0;
for (du, value_u) in basis_u.iter().enumerate() {
let i = span_u - knot_u.degree + du;
for (dv, value_v) in basis_v.iter().enumerate() {
let j = span_v - knot_v.degree + dv;
let entry = value_u * value_v * weights[i][j];
row[i * count_v + j] = entry;
denominator += entry;
}
}
if denominator.abs() > 0.0 {
for value in row.iter_mut() {
*value /= denominator;
}
}
}
}
let solve_axis = |axis: fn(Vec3) -> f64| {
solve_dense(
matrix.clone(),
samples
.iter()
.flat_map(|row| row.iter().copied().map(axis))
.collect(),
)
};
let x = solve_axis(|point| point.x)?;
let y = solve_axis(|point| point.y)?;
let z = solve_axis(|point| point.z)?;
let mut controls = vec![vec![Vec4::from_point(Vec3::default(), 1.0); count_v]; count_u];
for row in 0..count_u {
for column in 0..count_v {
let index = row * count_v + column;
controls[row][column] = Vec4::from_point(
Vec3::new(x[index], y[index], z[index]),
weights[row][column],
);
}
}
Ok(controls)
}
#[derive(Clone, Copy, Debug, Default, PartialEq, Eq)]
pub enum OffsetSurfaceLane {
Affine,
#[default]
Fit,
Reparameterised,
}
#[derive(Clone, Debug)]
pub struct MeasuredOffsetSurface {
pub surface: NurbsSurface,
pub lane: OffsetSurfaceLane,
pub fit: Option<MeasuredTolerance>,
}
#[derive(Clone, Copy, Debug, PartialEq)]
pub struct CarrierExtension {
pub u_min: f64,
pub u_max: f64,
pub v_min: f64,
pub v_max: f64,
}
impl CarrierExtension {
pub const NONE: CarrierExtension = CarrierExtension {
u_min: 0.0,
u_max: 0.0,
v_min: 0.0,
v_max: 0.0,
};
pub fn uniform(amount: f64) -> CarrierExtension {
CarrierExtension {
u_min: amount,
u_max: amount,
v_min: amount,
v_max: amount,
}
}
pub fn is_active(&self) -> bool {
self.u_min > 0.0 || self.u_max > 0.0 || self.v_min > 0.0 || self.v_max > 0.0
}
fn extends_v(&self) -> bool {
self.v_min > 0.0 || self.v_max > 0.0
}
}
fn trim_domain_fractions(
face: &FaceRecord,
[u0, u1]: [f64; 2],
[v0, v1]: [f64; 2],
) -> Result<([f64; 2], [f64; 2]), String> {
let mut low = Vec2 {
x: f64::INFINITY,
y: f64::INFINITY,
};
let mut high = Vec2 {
x: f64::NEG_INFINITY,
y: f64::NEG_INFINITY,
};
for coedge in face
.loops
.iter()
.flat_map(|loop_record| &loop_record.coedges)
{
let [p0, p1] = coedge.pcurve.domain()?;
for sample in 0..=24 {
let uv = coedge
.pcurve
.evaluate(p0 + (p1 - p0) * sample as f64 / 24.0)?;
low.x = low.x.min(uv.x);
low.y = low.y.min(uv.y);
high.x = high.x.max(uv.x);
high.y = high.y.max(uv.y);
}
}
let fraction = |value: f64, start: f64, end: f64| {
let span = end - start;
if span.abs() <= f64::EPSILON || !value.is_finite() {
return None;
}
Some(((value - start) / span).clamp(0.0, 1.0))
};
let u = match (fraction(low.x, u0, u1), fraction(high.x, u0, u1)) {
(Some(a), Some(b)) if b - a > 1e-6 => [a, b],
_ => [0.0, 1.0],
};
let v = match (fraction(low.y, v0, v1), fraction(high.y, v0, v1)) {
(Some(a), Some(b)) if b - a > 1e-6 => [a, b],
_ => [0.0, 1.0],
};
Ok((u, v))
}
fn domain_end_moves(grow_min: f64, grow_max: f64, [ta, tb]: [f64; 2]) -> (f64, f64) {
let span = tb - ta;
if span <= 1e-6 {
return (grow_min, grow_max);
}
let rate = (grow_min + grow_max) / span;
let back = grow_min + ta * rate;
let forward = (1.0 - ta) * rate - grow_min;
(back.max(0.0), forward.max(0.0))
}
pub fn offset_surface(
face: &FaceRecord,
distance: f64,
planar_extension: f64,
) -> Result<NurbsSurface, String> {
offset_surface_with_lane(face, distance, &CarrierExtension::uniform(planar_extension))
.map(|(surface, _)| surface)
}
pub fn offset_surface_measured(
face: &FaceRecord,
distance: f64,
planar_extension: f64,
band: f64,
) -> Result<MeasuredOffsetSurface, String> {
offset_surface_measured_sided(
face,
distance,
&CarrierExtension::uniform(planar_extension),
band,
)
}
pub fn offset_surface_measured_sided(
face: &FaceRecord,
distance: f64,
extension: &CarrierExtension,
band: f64,
) -> Result<MeasuredOffsetSurface, String> {
let (surface, lane) = offset_surface_with_lane(face, distance, extension)?;
let fit = match lane {
OffsetSurfaceLane::Affine => Some(MeasuredTolerance::exact(band)),
OffsetSurfaceLane::Fit => Some(measure_surface_fit_against_pointwise_offset(
&face.surface,
face.same_sense,
&surface,
distance,
band,
)?),
OffsetSurfaceLane::Reparameterised => None,
};
Ok(MeasuredOffsetSurface { surface, lane, fit })
}
fn offset_surface_with_lane(
face: &FaceRecord,
distance: f64,
extension: &CarrierExtension,
) -> Result<(NurbsSurface, OffsetSurfaceLane), String> {
let source = &face.surface;
let extension_active = extension.is_active();
if source.is_affine()? {
let ([u0, u1], [v0, v1]) = domains(source)?;
let normal = stable_face_normal(face, (u0 + u1) / 2.0, (v0 + v1) / 2.0)?;
let shift = normal.scale(-distance);
let mut points = source
.control_points
.iter()
.map(|row| {
row.iter()
.map(|control| Ok(control.point()?.add(shift)))
.collect::<Result<Vec<_>, String>>()
})
.collect::<Result<Vec<_>, String>>()?;
if extension_active {
let p00 = points[0][0];
let p01 = points[0][1];
let p10 = points[1][0];
let direction_u = p10.sub(p00).normalized()?;
let direction_v = p01.sub(p00).normalized()?;
let (trim_u, trim_v) = trim_domain_fractions(face, [u0, u1], [v0, v1])?;
let (back_u, forward_u) = domain_end_moves(extension.u_min, extension.u_max, trim_u);
let (back_v, forward_v) = domain_end_moves(extension.v_min, extension.v_max, trim_v);
points[0][0] = p00
.sub(direction_u.scale(back_u))
.sub(direction_v.scale(back_v));
points[0][1] = p01
.sub(direction_u.scale(back_u))
.add(direction_v.scale(forward_v));
points[1][0] = p10
.add(direction_u.scale(forward_u))
.sub(direction_v.scale(back_v));
points[1][1] = points[1][1]
.add(direction_u.scale(forward_u))
.add(direction_v.scale(forward_v));
}
let controls = points
.into_iter()
.enumerate()
.map(|(row, points)| {
points
.into_iter()
.enumerate()
.map(|(column, point)| {
Vec4::from_point(point, source.control_points[row][column].w)
})
.collect()
})
.collect();
let lane = if extension_active {
OffsetSurfaceLane::Reparameterised
} else {
OffsetSurfaceLane::Affine
};
return Ok((
NurbsSurface::new(
source.degree_u,
source.degree_v,
source.knots_u.clone(),
source.knots_v.clone(),
controls,
)?,
lane,
));
}
let knot_u = KnotVector::new(source.knots_u.clone(), source.degree_u)?;
let knot_v = KnotVector::new(source.knots_v.clone(), source.degree_v)?;
let parameters_u = greville_parameters(&knot_u);
let parameters_v = greville_parameters(&knot_v);
let evaluator = OffsetEvaluator::new(
"offset_surface",
source,
OffsetNormal::FaceStable {
same_sense: face.same_sense,
},
);
let mut samples = Vec::new();
for &u in ¶meters_u {
let mut row = Vec::new();
for &v in ¶meters_v {
row.push(evaluator.at(u, v, -distance)?.point);
}
samples.push(row);
}
let mut reparameterised = false;
if parameters_v.len() == 2 && parameters_u.len() >= 3 {
let centroid = |column: usize| {
let mut sum = Vec3::default();
for row in &samples {
sum = sum.add(row[column]);
}
sum.scale(1.0 / samples.len() as f64)
};
let near_centroid = centroid(0);
let far_centroid = centroid(1);
let mut inverted = true;
let mut pinch_fraction = 0.0f64;
let mut near_mean = 0.0f64;
let mut far_mean = 0.0f64;
for row in &samples {
let near_radial = row[0].sub(near_centroid);
let far_radial = row[1].sub(far_centroid);
let near_len = near_radial.length();
let far_len = far_radial.length();
if near_len <= 1e-9 || far_len <= 1e-9 {
inverted = false;
break;
}
if near_radial.dot(far_radial) >= 0.0 {
inverted = false;
break;
}
pinch_fraction += near_len / (near_len + far_len) / samples.len() as f64;
near_mean += near_len / samples.len() as f64;
far_mean += far_len / samples.len() as f64;
}
if inverted {
reparameterised = true;
let retrim_far = far_mean <= near_mean;
for row in &mut samples {
let near = row[0];
let far = row[1];
let pinch = near.add(far.sub(near).scale(pinch_fraction));
if retrim_far {
row[1] = pinch;
} else {
row[0] = pinch;
}
}
}
if extension.extends_v() && !inverted {
let mut min_ruling = f64::MAX;
let mut back_allowance = f64::MAX;
let mut forward_allowance = f64::MAX;
let mut extendable = true;
for row in &samples {
let ruling = row[1].sub(row[0]);
let length = ruling.length();
min_ruling = min_ruling.min(length);
let near_radial = row[0].sub(near_centroid).length();
let far_radial = row[1].sub(far_centroid).length();
if (far_radial - near_radial).abs() > 1e-9 {
let apex_at = near_radial / (near_radial - far_radial);
if (-1e-9..=1.0 + 1e-9).contains(&apex_at) {
extendable = false;
break;
}
if apex_at < 0.0 {
back_allowance = back_allowance.min(0.9 * -apex_at);
} else {
forward_allowance = forward_allowance.min(0.9 * (apex_at - 1.0));
}
}
}
if extendable && min_ruling > 1e-9 {
reparameterised = true;
let (_, trim_v) = trim_domain_fractions(face, knot_u.domain(), knot_v.domain())?;
let (back_world, forward_world) =
domain_end_moves(extension.v_min, extension.v_max, trim_v);
let back = (back_world / min_ruling).min(back_allowance);
let forward = (forward_world / min_ruling).min(forward_allowance);
for row in &mut samples {
let ruling = row[1].sub(row[0]);
row[0] = row[0].sub(ruling.scale(back));
row[1] = row[1].add(ruling.scale(forward));
}
}
}
}
let weights = source
.control_points
.iter()
.map(|row| row.iter().map(|point| point.w).collect::<Vec<_>>())
.collect::<Vec<_>>();
let lane = if reparameterised {
OffsetSurfaceLane::Reparameterised
} else {
OffsetSurfaceLane::Fit
};
Ok((
NurbsSurface::new(
source.degree_u,
source.degree_v,
source.knots_u.clone(),
source.knots_v.clone(),
interpolate_tensor(
&knot_u,
&knot_v,
¶meters_u,
¶meters_v,
&samples,
&weights,
)?,
)?,
lane,
))
}
fn carrier_edge_curve(
surface: &NurbsSurface,
pcurve: &NurbsCurve,
points: &[Vec3],
parameters: &[f64],
fit_tolerance: f64,
) -> Result<(NurbsCurve, f64, f64), String> {
let image = if std::env::var("BREP_CARRIER_POLYLINE").as_deref() == Ok("1") {
Err("BREP_CARRIER_POLYLINE set".to_string())
} else {
crate::image_curve::image_curve(surface, pcurve, fit_tolerance, "offset_face_carrier")
};
match image {
Ok(image) if image.t0 <= image.t1 => Ok((image.curve, image.t0, image.t1)),
Ok(image) => {
let [start, end] = image.curve.domain()?;
let curve = image.curve.reversed()?;
Ok((curve, start + end - image.t0, start + end - image.t1))
}
Err(_) => {
let curve = interpolate_curve(points, 1, parameters)?;
let [t0, t1] = curve.domain()?;
Ok((curve, t0, t1))
}
}
}
fn mapped_pcurve_polyline(
surface: &NurbsSurface,
pcurve: &NurbsCurve,
degenerate: bool,
) -> Result<(Vec<Vec3>, Vec<f64>), String> {
let [start, end] = pcurve.domain()?;
let evaluate = |fraction: f64| {
let uv = pcurve.evaluate(start + (end - start) * fraction)?;
surface.evaluate(uv.x, uv.y)
};
let first = evaluate(0.0)?;
let last = evaluate(1.0)?;
if degenerate {
return Ok((vec![first, last], vec![0.0, 1.0]));
}
fn append(
evaluate: &impl Fn(f64) -> Result<Vec3, String>,
a_fraction: f64,
a: Vec3,
b_fraction: f64,
b: Vec3,
depth: usize,
parameters: &mut Vec<f64>,
points: &mut Vec<Vec3>,
) -> Result<(), String> {
let fractions =
[0.25, 0.5, 0.75].map(|local| a_fraction + (b_fraction - a_fraction) * local);
let samples = fractions
.map(evaluate)
.into_iter()
.collect::<Result<Vec<_>, String>>()?;
let deviation = samples
.iter()
.enumerate()
.map(|(index, point)| {
point
.sub(a.add(b.sub(a).scale((index + 1) as f64 * 0.25)))
.length()
})
.fold(0.0, f64::max);
if deviation <= 5e-4 || depth >= 10 {
parameters.push(b_fraction);
points.push(b);
return Ok(());
}
append(
evaluate,
a_fraction,
a,
fractions[1],
samples[1],
depth + 1,
parameters,
points,
)?;
append(
evaluate,
fractions[1],
samples[1],
b_fraction,
b,
depth + 1,
parameters,
points,
)
}
let mut parameters = vec![0.0];
let mut points = vec![first];
append(
&evaluate,
0.0,
first,
1.0,
last,
0,
&mut parameters,
&mut points,
)?;
Ok((points, parameters))
}
#[derive(Clone, Debug)]
pub struct CarrierDeviation {
pub band: f64,
pub lane: OffsetSurfaceLane,
pub surface: Option<MeasuredTolerance>,
pub edges: Vec<(u64, MeasuredTolerance)>,
pub vertices: Vec<(u64, f64)>,
}
impl CarrierDeviation {
pub fn worst(&self) -> Option<MeasuredTolerance> {
MeasuredTolerance::worst(
self.surface
.into_iter()
.chain(self.edges.iter().map(|(_, measured)| *measured))
.chain(
self.vertices
.iter()
.map(|(_, gap)| MeasuredTolerance::new(*gap, self.band)),
),
)
}
pub fn exceedances(&self) -> Vec<String> {
let mut out = Vec::new();
if let Some(surface) = self.surface {
if surface.exceeds_band() {
out.push(format!("carrier surface fit {}", surface.describe()));
}
}
for (id, measured) in &self.edges {
if measured.exceeds_band() {
out.push(format!("edge {id} {}", measured.describe()));
}
}
for (id, gap) in &self.vertices {
let measured = MeasuredTolerance::new(*gap, self.band);
if measured.exceeds_band() {
out.push(format!("vertex {id} {}", measured.describe()));
}
}
out
}
}
#[derive(Clone, Debug, Serialize)]
pub struct OffsetFaceCarrier {
pub vertices: Vec<VertexRecord>,
pub edges: Vec<EdgeRecord>,
pub face: FaceRecord,
#[serde(skip)]
pub deviation: Option<CarrierDeviation>,
}
fn claim_vertex_image(
source_id: u64,
point: Vec3,
vertex_images: &mut HashMap<u64, u64>,
vertices: &mut Vec<VertexRecord>,
next_id: &mut u64,
) -> u64 {
if let Some(id) = vertex_images.get(&source_id) {
return *id;
}
let id = *next_id;
*next_id += 1;
vertices.push(VertexRecord { id, point });
vertex_images.insert(source_id, id);
id
}
pub fn offset_face_carrier(
solid: &BrepSolid,
face_id: u64,
distance: f64,
planar_extension: f64,
) -> Result<OffsetFaceCarrier, String> {
offset_face_carrier_impl(
solid,
face_id,
distance,
&CarrierExtension::uniform(planar_extension),
false,
)
}
pub fn offset_face_carrier_sided(
solid: &BrepSolid,
face_id: u64,
distance: f64,
extension: &CarrierExtension,
) -> Result<OffsetFaceCarrier, String> {
offset_face_carrier_impl(solid, face_id, distance, extension, false)
}
pub fn offset_face_carrier_measured(
solid: &BrepSolid,
face_id: u64,
distance: f64,
planar_extension: f64,
) -> Result<OffsetFaceCarrier, String> {
offset_face_carrier_impl(
solid,
face_id,
distance,
&CarrierExtension::uniform(planar_extension),
true,
)
}
fn measure_carrier(
surface: &NurbsSurface,
vertices: &[VertexRecord],
edges: &[EdgeRecord],
loops: &[LoopRecord],
lane: OffsetSurfaceLane,
surface_fit: Option<MeasuredTolerance>,
band: f64,
) -> Result<CarrierDeviation, String> {
let edge_by_id: HashMap<u64, &EdgeRecord> = edges.iter().map(|edge| (edge.id, edge)).collect();
let mut per_edge: HashMap<u64, MeasuredTolerance> = HashMap::default();
for loop_record in loops {
for coedge in &loop_record.coedges {
let Some(edge) = edge_by_id.get(&coedge.edge_id) else {
continue;
};
let measured = measure_edge_against_pcurve_image(
surface,
&coedge.pcurve,
edge,
coedge.forward,
band,
)?;
per_edge
.entry(edge.id)
.and_modify(|existing| *existing = existing.worse_of(measured))
.or_insert(measured);
}
}
let mut measured_edges: Vec<(u64, MeasuredTolerance)> = per_edge.into_iter().collect();
measured_edges.sort_by_key(|(id, _)| *id);
let deviation_of: HashMap<u64, f64> = measured_edges
.iter()
.map(|(id, measured)| (*id, measured.deviation()))
.collect();
let mut ends_at: HashMap<u64, Vec<(&EdgeRecord, f64)>> = HashMap::default();
for edge in edges {
ends_at
.entry(edge.start_vertex_id)
.or_default()
.push((edge, edge.t0));
ends_at
.entry(edge.end_vertex_id)
.or_default()
.push((edge, edge.t1));
}
let mut measured_vertices = Vec::with_capacity(vertices.len());
for vertex in vertices {
let mut gaps = Vec::new();
let mut incident = Vec::new();
for (edge, parameter) in ends_at.get(&vertex.id).into_iter().flatten() {
gaps.push(vertex_endpoint_gap(
vertex.point,
edge.curve.evaluate(*parameter)?,
));
incident.push(deviation_of.get(&edge.id).copied().unwrap_or(0.0));
}
measured_vertices.push((vertex.id, vertex_tolerance_from_edges(gaps, incident)));
}
Ok(CarrierDeviation {
band,
lane,
surface: surface_fit,
edges: measured_edges,
vertices: measured_vertices,
})
}
fn offset_face_carrier_impl(
solid: &BrepSolid,
face_id: u64,
distance: f64,
extension: &CarrierExtension,
measure: bool,
) -> Result<OffsetFaceCarrier, String> {
let source = solid
.shells
.iter()
.flat_map(|shell| &shell.faces)
.find(|face| face.id == face_id)
.ok_or_else(|| format!("offset_face_carrier: missing face {face_id}"))?;
let band = offset_construction_band(solid_model_scale(solid));
let fit_tolerance =
crate::KernelTolerances::for_scale(solid_model_scale(solid), 1e-7).intersection_fit;
let (surface, lane, surface_fit) = if measure {
let measured = offset_surface_measured_sided(source, distance, extension, band)?;
(measured.surface, measured.lane, measured.fit)
} else {
let (surface, lane) = offset_surface_with_lane(source, distance, extension)?;
(surface, lane, None)
};
let source_edges = solid
.edges
.iter()
.map(|edge| (edge.id, edge))
.collect::<HashMap<_, _>>();
let source_vertices = solid
.vertices
.iter()
.map(|vertex| (vertex.id, vertex))
.collect::<HashMap<_, _>>();
let mut vertices = Vec::new();
let mut vertex_images = HashMap::default();
let mut edges = Vec::new();
let mut edge_images = HashMap::default();
let mut loops = Vec::new();
let mut next_id = 1u64;
for source_loop in &source.loops {
let mut coedges = Vec::new();
for source_coedge in &source_loop.coedges {
let source_edge = source_edges
.get(&source_coedge.edge_id)
.ok_or_else(|| "offset_face_carrier: missing source edge".to_string())?;
let (source_start, source_end) = if source_coedge.forward {
(source_edge.start_vertex_id, source_edge.end_vertex_id)
} else {
(source_edge.end_vertex_id, source_edge.start_vertex_id)
};
if !source_vertices.contains_key(&source_start)
|| !source_vertices.contains_key(&source_end)
{
return Err("offset_face_carrier: missing source vertex".into());
}
let (points, parameters) =
mapped_pcurve_polyline(&surface, &source_coedge.pcurve, false)?;
let collapse_tolerance = 1e-6f64.max(distance.abs() * 1e-2);
let image_collapsed = points
.iter()
.all(|point| point.sub(points[0]).length() <= collapse_tolerance);
let (edge_id, forward) =
if let Some((edge_id, edge_start_vertex_id, creator_forward)) =
edge_images.get(&source_edge.id)
{
if source_start == source_end {
(*edge_id, source_coedge.forward == *creator_forward)
} else {
(
*edge_id,
vertex_images.get(&source_start) == Some(edge_start_vertex_id),
)
}
} else {
let (curve, t0, t1) = if image_collapsed {
let curve = NurbsCurve::new(
1,
vec![0.0, 0.0, 1.0, 1.0],
vec![
Vec4::from_point(points[0], 1.0),
Vec4::from_point(points[0], 1.0),
],
)?;
(curve, 0.0, 1.0)
} else {
carrier_edge_curve(
&surface,
&source_coedge.pcurve,
&points,
¶meters,
fit_tolerance,
)?
};
let start_vertex_id = claim_vertex_image(
source_start,
points[0],
&mut vertex_images,
&mut vertices,
&mut next_id,
);
let end_vertex_id = claim_vertex_image(
source_end,
points[points.len() - 1],
&mut vertex_images,
&mut vertices,
&mut next_id,
);
let id = next_id;
next_id += 1;
edges.push(EdgeRecord {
id,
curve,
t0,
t1,
start_vertex_id,
end_vertex_id,
degenerate: source_edge.degenerate && image_collapsed,
name: source_edge
.name
.as_ref()
.map(|name| format!("{name}_Offset")),
});
edge_images.insert(
source_edge.id,
(id, start_vertex_id, source_coedge.forward),
);
(id, true)
};
let id = next_id;
next_id += 1;
coedges.push(CoedgeRecord {
id,
edge_id,
forward,
pcurve: source_coedge.pcurve.clone(),
});
}
let id = next_id;
next_id += 1;
loops.push(LoopRecord { id, coedges });
}
let deviation = if measure {
Some(measure_carrier(
&surface,
&vertices,
&edges,
&loops,
lane,
surface_fit,
band,
)?)
} else {
None
};
Ok(OffsetFaceCarrier {
vertices,
edges,
face: FaceRecord {
id: next_id,
surface,
same_sense: source.same_sense,
loops,
name: source.name.as_ref().map(|name| format!("{name}_Offset")),
},
deviation,
})
}
#[cfg(test)]
mod tests {
use super::*;
use crate::{make_box_brep, make_cylinder_brep};
#[test]
fn a_full_torus_carrier_reproduces_the_source_seam_senses() {
let solid =
crate::make_torus_brep(Vec3::default(), Vec3::new(0.0, 0.0, 1.0), 5.0, 1.5).unwrap();
assert!(
solid.validate().is_empty(),
"the source torus is the oracle and must be clean"
);
let face = &solid.shells[0].faces[0];
let source_senses: Vec<bool> = face
.loops
.iter()
.flat_map(|loop_record| &loop_record.coedges)
.map(|coedge| coedge.forward)
.collect();
assert_eq!(
source_senses,
vec![true, true, false, false],
"a full torus's two seam edges are each traversed both ways"
);
let carrier = offset_face_carrier_measured(&solid, face.id, 0.25, 0.0).unwrap();
let carrier_senses: Vec<bool> = carrier
.face
.loops
.iter()
.flat_map(|loop_record| &loop_record.coedges)
.map(|coedge| coedge.forward)
.collect();
assert_eq!(
carrier_senses, source_senses,
"the carrier copies the source topology, senses included"
);
let deviation = carrier.deviation.expect("measured");
for (id, measured) in &deviation.edges {
assert!(
measured.deviation() < 1e-3,
"carrier edge {id} deviates {:.3e} from its pcurve image",
measured.deviation()
);
}
}
#[test]
fn measuring_a_carrier_cannot_change_it() {
let cases: Vec<(&str, BrepSolid)> = vec![
("box", make_box_brep(Vec3::default(), 20.0, 20.0, 4.0).unwrap()),
(
"cylinder",
make_cylinder_brep(Vec3::default(), Vec3::new(0.0, 0.0, 1.0), 5.0, 12.0).unwrap(),
),
(
"cone",
crate::make_cone_brep(Vec3::default(), Vec3::new(0.0, 0.0, 1.0), 5.0, 2.5, 10.0)
.unwrap(),
),
(
"sphere",
crate::make_sphere_brep(Vec3::default(), 5.0, Vec3::new(0.0, 0.0, 1.0)).unwrap(),
),
];
for (label, solid) in cases {
let face_ids: Vec<u64> = solid
.shells
.iter()
.flat_map(|shell| &shell.faces)
.map(|face| face.id)
.collect();
for face_id in face_ids {
for distance in [0.25, -0.25] {
for extension in [0.0, 0.5] {
let plain = offset_face_carrier(&solid, face_id, distance, extension);
let measured =
offset_face_carrier_measured(&solid, face_id, distance, extension);
match (plain, measured) {
(Ok(plain), Ok(measured)) => assert_eq!(
serde_json::to_string(&plain).unwrap(),
serde_json::to_string(&measured).unwrap(),
"{label} face {face_id} d={distance} ext={extension} moved"
),
(Err(plain), Err(measured)) => assert_eq!(
plain, measured,
"{label} face {face_id} refusal text moved"
),
(plain, measured) => panic!(
"{label} face {face_id}: outcomes disagree ({:?} vs {:?})",
plain.is_ok(),
measured.is_ok()
),
}
}
}
}
}
}
#[test]
fn the_surface_lane_names_the_branch_that_ran() {
let plate = make_box_brep(Vec3::default(), 20.0, 20.0, 4.0).unwrap();
let plane = &plate.shells[0].faces[0];
assert_eq!(
offset_surface_measured(plane, 0.25, 0.0, 1e-2).unwrap().lane,
OffsetSurfaceLane::Affine
);
assert_eq!(
offset_surface_measured(plane, 0.25, 0.5, 1e-2).unwrap().lane,
OffsetSurfaceLane::Reparameterised,
"a grown plane no longer names the same point at the same (u, v)"
);
assert!(
offset_surface_measured(plane, 0.25, 0.5, 1e-2)
.unwrap()
.fit
.is_none(),
"a deliberate divergence is not reported as a fit error"
);
let ball = crate::make_sphere_brep(Vec3::default(), 5.0, Vec3::new(0.0, 0.0, 1.0)).unwrap();
let sphere = &ball.shells[0].faces[0];
let measured = offset_surface_measured(sphere, 0.25, 0.0, 1e-2).unwrap();
assert_eq!(measured.lane, OffsetSurfaceLane::Fit);
let fit = measured.fit.expect("a fit has an error to report");
assert!(
fit.deviation() > 0.0 && fit.deviation() < 1e-4,
"the sphere's collocation fit is small but not exact (got {:.3e})",
fit.deviation()
);
}
#[test]
fn affine_offset_is_exact_and_preserves_weights() {
let solid = make_box_brep(Vec3::default(), 4.0, 4.0, 4.0).unwrap();
let face = &solid.shells[0].faces[0];
let offset = offset_surface(face, 0.75, 0.0).unwrap();
let domain_u = KnotVector::new(face.surface.knots_u.clone(), 1)
.unwrap()
.domain();
let domain_v = KnotVector::new(face.surface.knots_v.clone(), 1)
.unwrap()
.domain();
let u = (domain_u[0] + domain_u[1]) / 2.0;
let v = (domain_v[0] + domain_v[1]) / 2.0;
let displacement = offset
.evaluate(u, v)
.unwrap()
.sub(face.surface.evaluate(u, v).unwrap());
assert!((displacement.length() - 0.75).abs() < 1e-12);
}
#[test]
fn curved_offset_carrier_maps_every_trim_to_new_surface() {
let solid =
make_cylinder_brep(Vec3::default(), Vec3::new(0.0, 0.0, 1.0), 2.0, 4.0).unwrap();
let side = &solid.shells[0].faces[0];
let carrier = offset_face_carrier(&solid, side.id, 0.5, 0.0).unwrap();
for coedge in carrier
.face
.loops
.iter()
.flat_map(|loop_record| &loop_record.coedges)
{
let edge = carrier
.edges
.iter()
.find(|edge| edge.id == coedge.edge_id)
.unwrap();
for fraction in [0.0, 0.3, 0.8, 1.0] {
let uv = coedge.pcurve.evaluate(fraction).unwrap();
let on_surface = carrier.face.surface.evaluate(uv.x, uv.y).unwrap();
let parameter = if coedge.forward {
edge.t0 + (edge.t1 - edge.t0) * fraction
} else {
edge.t1 - (edge.t1 - edge.t0) * fraction
};
assert!(
on_surface
.sub(edge.curve.evaluate(parameter).unwrap())
.length()
< 7e-4
);
}
}
}
}