1use crate::{KernelRefusal, KernelStage, OrRefuse};
2use crate::fit::{solve_dense, solve_dense_shared};
3use crate::topology::{BrepSolid, CoedgeRecord, EdgeRecord, FaceRecord, LoopRecord, VertexRecord};
4use crate::{
5 interpolate_curve, measure_edge_against_pcurve_image,
6 measure_surface_fit_against_pointwise_offset, offset_construction_band, solid_model_scale,
7 vertex_endpoint_gap, vertex_tolerance_from_edges, KnotVector, MeasuredTolerance, NurbsCurve,
8 NurbsSurface, OffsetEvaluator, OffsetNormal, Vec2, Vec3, Vec4,
9};
10use rustc_hash::FxHashMap as HashMap;
11use serde::Serialize;
12
13fn domains(surface: &NurbsSurface) -> Result<([f64; 2], [f64; 2]), KernelRefusal> {
14 Ok((
15 KnotVector::new(surface.knots_u.clone(), surface.degree_u).or_refuse(KernelStage::Refine, "new")?.domain(),
16 KnotVector::new(surface.knots_v.clone(), surface.degree_v).or_refuse(KernelStage::Refine, "new")?.domain(),
17 ))
18}
19
20fn stable_face_normal(face: &FaceRecord, u: f64, v: f64) -> Result<Vec3, KernelRefusal> {
24 OffsetEvaluator::new(
25 "offset_surface",
26 &face.surface,
27 OffsetNormal::FaceStable {
28 same_sense: face.same_sense,
29 },
30 )
31 .normal(u, v)
32 .or_refuse(KernelStage::Refine, "normal")
33}
34
35fn greville_parameters(knots: &KnotVector) -> Vec<f64> {
36 let mut parameters = (0..knots.control_point_count())
37 .map(|index| {
38 knots.knots[index + 1..=index + knots.degree]
39 .iter()
40 .sum::<f64>()
41 / knots.degree as f64
42 })
43 .collect::<Vec<_>>();
44 let domain = knots.domain();
45 parameters[0] = domain[0];
46 *parameters.last_mut().unwrap() = domain[1];
47 parameters
48}
49
50fn collocation_matrix(knots: &KnotVector, parameters: &[f64], weights: &[f64]) -> Vec<Vec<f64>> {
54 parameters
55 .iter()
56 .map(|parameter| {
57 let mut row = vec![0.0; knots.control_point_count()];
58 let span = knots.find_span(*parameter);
59 for (offset, value) in knots
60 .basis_functions(span, *parameter)
61 .into_iter()
62 .enumerate()
63 {
64 let index = span - knots.degree + offset;
65 row[index] = value * weights[index];
66 }
67 let denominator: f64 = row.iter().sum();
68 if denominator.abs() > 0.0 {
69 for value in &mut row {
70 *value /= denominator;
71 }
72 }
73 row
74 })
75 .collect()
76}
77
78fn separable_weights(weights: &[Vec<f64>]) -> Option<(Vec<f64>, Vec<f64>)> {
82 let first_row = weights.first()?;
83 let anchor = *first_row.first()?;
84 if anchor.abs() <= 1e-12 {
85 return None;
86 }
87 let a: Vec<f64> = weights.iter().map(|row| row[0]).collect();
88 let b: Vec<f64> = first_row.iter().map(|w| w / anchor).collect();
89 for (i, row) in weights.iter().enumerate() {
90 for (j, &w) in row.iter().enumerate() {
91 if (w - a[i] * b[j]).abs() > 1e-10 * (1.0 + w.abs()) {
92 return None;
93 }
94 }
95 }
96 Some((a, b))
97}
98
99fn interpolate_tensor(
105 knot_u: &KnotVector,
106 knot_v: &KnotVector,
107 parameters_u: &[f64],
108 parameters_v: &[f64],
109 samples: &[Vec<Vec3>],
110 weights: &[Vec<f64>],
111) -> Result<Vec<Vec<Vec4>>, KernelRefusal> {
112 let count_u = parameters_u.len();
113 let count_v = parameters_v.len();
114 if let Some((weights_u, weights_v)) = separable_weights(weights) {
115 let matrix_u = collocation_matrix(knot_u, parameters_u, &weights_u);
116 let matrix_v = collocation_matrix(knot_v, parameters_v, &weights_v);
117 let mut intermediate = vec![vec![Vec3::default(); count_v]; count_u];
124 for column in 0..count_v {
125 let side = |axis: fn(Vec3) -> f64| samples.iter().map(|row| axis(row[column])).collect::<Vec<_>>();
126 let [x, y, z] = solve_dense_shared(matrix_u.clone(), [side(|point| point.x), side(|point| point.y), side(|point| point.z)])
127 .or_refuse(KernelStage::Refine, "solve_axis")?;
128 for row in 0..count_u {
129 intermediate[row][column] = Vec3::new(x[row], y[row], z[row]);
130 }
131 }
132 let mut controls = vec![vec![Vec4::from_point(Vec3::default(), 1.0); count_v]; count_u];
133 for row in 0..count_u {
134 let side = |axis: fn(Vec3) -> f64| intermediate[row].iter().copied().map(axis).collect::<Vec<_>>();
135 let [x, y, z] = solve_dense_shared(matrix_v.clone(), [side(|point| point.x), side(|point| point.y), side(|point| point.z)])
136 .or_refuse(KernelStage::Refine, "solve_axis")?;
137 for column in 0..count_v {
138 controls[row][column] = Vec4::from_point(
139 Vec3::new(x[column], y[column], z[column]),
140 weights[row][column],
141 );
142 }
143 }
144 return Ok(controls);
145 }
146
147 let unknowns = count_u * count_v;
150 let mut matrix = vec![vec![0.0; unknowns]; unknowns];
151 for (k, &u) in parameters_u.iter().enumerate() {
152 let span_u = knot_u.find_span(u);
153 let basis_u = knot_u.basis_functions(span_u, u);
154 for (l, &v) in parameters_v.iter().enumerate() {
155 let span_v = knot_v.find_span(v);
156 let basis_v = knot_v.basis_functions(span_v, v);
157 let row = &mut matrix[k * count_v + l];
158 let mut denominator = 0.0;
159 for (du, value_u) in basis_u.iter().enumerate() {
160 let i = span_u - knot_u.degree + du;
161 for (dv, value_v) in basis_v.iter().enumerate() {
162 let j = span_v - knot_v.degree + dv;
163 let entry = value_u * value_v * weights[i][j];
164 row[i * count_v + j] = entry;
165 denominator += entry;
166 }
167 }
168 if denominator.abs() > 0.0 {
169 for value in row.iter_mut() {
170 *value /= denominator;
171 }
172 }
173 }
174 }
175 let solve_axis = |axis: fn(Vec3) -> f64| {
176 solve_dense(
177 matrix.clone(),
178 samples
179 .iter()
180 .flat_map(|row| row.iter().copied().map(axis))
181 .collect(),
182 )
183 };
184 let x = solve_axis(|point| point.x).or_refuse(KernelStage::Refine, "solve_axis")?;
185 let y = solve_axis(|point| point.y).or_refuse(KernelStage::Refine, "solve_axis")?;
186 let z = solve_axis(|point| point.z).or_refuse(KernelStage::Refine, "solve_axis")?;
187 let mut controls = vec![vec![Vec4::from_point(Vec3::default(), 1.0); count_v]; count_u];
188 for row in 0..count_u {
189 for column in 0..count_v {
190 let index = row * count_v + column;
191 controls[row][column] = Vec4::from_point(
192 Vec3::new(x[index], y[index], z[index]),
193 weights[row][column],
194 );
195 }
196 }
197 Ok(controls)
198}
199
200#[derive(Clone, Copy, Debug, Default, PartialEq, Eq)]
210pub enum OffsetSurfaceLane {
211 Affine,
215 Analytic,
221 #[default]
225 Fit,
226 Reparameterised,
232}
233
234#[derive(Clone, Debug)]
237pub struct MeasuredOffsetSurface {
238 pub surface: NurbsSurface,
239 pub lane: OffsetSurfaceLane,
240 pub fit: Option<MeasuredTolerance>,
248}
249
250#[derive(Clone, Copy, Debug, PartialEq)]
262pub struct CarrierExtension {
263 pub u_min: f64,
264 pub u_max: f64,
265 pub v_min: f64,
266 pub v_max: f64,
267}
268
269impl CarrierExtension {
270 pub const NONE: CarrierExtension = CarrierExtension {
271 u_min: 0.0,
272 u_max: 0.0,
273 v_min: 0.0,
274 v_max: 0.0,
275 };
276
277 pub fn uniform(amount: f64) -> CarrierExtension {
278 CarrierExtension {
279 u_min: amount,
280 u_max: amount,
281 v_min: amount,
282 v_max: amount,
283 }
284 }
285
286 pub fn is_active(&self) -> bool {
287 self.u_min > 0.0 || self.u_max > 0.0 || self.v_min > 0.0 || self.v_max > 0.0
288 }
289
290 fn extends_v(&self) -> bool {
291 self.v_min > 0.0 || self.v_max > 0.0
292 }
293}
294
295const EXACT_TRIM_BOX_DEFAULT: bool = true;
304
305fn exact_trim_box_on() -> bool {
306 switch!(live "BREP_OFFSET_EXACT_TRIM_BOX", EXACT_TRIM_BOX_DEFAULT)
307}
308
309
310const TRIM_EXTREME_STEPS: usize = 64;
314
315fn refine_trim_extreme(
324 pcurve: &NurbsCurve,
325 samples: &[(f64, Vec3)],
326 axis: usize,
327 sign: f64,
328) -> Result<f64, KernelRefusal> {
329 let read = |uv: Vec3| sign * if axis == 0 { uv.x } else { uv.y };
330 let at = |parameter: f64| -> Result<f64, KernelRefusal> {
331 Ok(read(pcurve.evaluate(parameter).or_refuse(KernelStage::Refine, "evaluate")?))
332 };
333 let best = (0..samples.len())
334 .max_by(|&a, &b| read(samples[a].1).total_cmp(&read(samples[b].1)))
335 .expect("a pcurve has samples");
336 let mut found = read(samples[best].1);
337 let (mut a, mut b) = (samples[best.saturating_sub(1)].0, samples[(best + 1).min(samples.len() - 1)].0);
338 const GOLDEN: f64 = 0.618_033_988_749_894_9;
339 let mut c = b - GOLDEN * (b - a);
340 let mut d = a + GOLDEN * (b - a);
341 let (mut fc, mut fd) = (at(c)?, at(d)?);
342 for _ in 0..TRIM_EXTREME_STEPS {
343 found = found.max(fc).max(fd);
344 if fc >= fd {
345 (b, d, fd) = (d, c, fc);
346 c = b - GOLDEN * (b - a);
347 fc = at(c)?;
348 } else {
349 (a, c, fc) = (c, d, fd);
350 d = a + GOLDEN * (b - a);
351 fd = at(d)?;
352 }
353 }
354 Ok(sign * found.max(fc).max(fd))
355}
356
357fn trim_domain_fractions(
362 face: &FaceRecord,
363 [u0, u1]: [f64; 2],
364 [v0, v1]: [f64; 2],
365) -> Result<([f64; 2], [f64; 2]), KernelRefusal> {
366 let mut low = Vec2 {
367 x: f64::INFINITY,
368 y: f64::INFINITY,
369 };
370 let mut high = Vec2 {
371 x: f64::NEG_INFINITY,
372 y: f64::NEG_INFINITY,
373 };
374 for coedge in face
375 .loops
376 .iter()
377 .flat_map(|loop_record| &loop_record.coedges)
378 {
379 let [p0, p1] = coedge.pcurve.domain().or_refuse(KernelStage::Refine, "domain")?;
380 let mut samples = Vec::with_capacity(25);
381 for sample in 0..=24 {
382 let parameter = p0 + (p1 - p0) * sample as f64 / 24.0;
383 let uv = coedge
384 .pcurve
385 .evaluate(parameter).or_refuse(KernelStage::Refine, "evaluate")?;
386 low.x = low.x.min(uv.x);
387 low.y = low.y.min(uv.y);
388 high.x = high.x.max(uv.x);
389 high.y = high.y.max(uv.y);
390 samples.push((parameter, uv));
391 }
392 if coedge.pcurve.degree > 1 && exact_trim_box_on() {
393 low.x = low.x.min(refine_trim_extreme(&coedge.pcurve, &samples, 0, -1.0)?);
394 high.x = high.x.max(refine_trim_extreme(&coedge.pcurve, &samples, 0, 1.0)?);
395 low.y = low.y.min(refine_trim_extreme(&coedge.pcurve, &samples, 1, -1.0)?);
396 high.y = high.y.max(refine_trim_extreme(&coedge.pcurve, &samples, 1, 1.0)?);
397 }
398 }
399 let fraction = |value: f64, start: f64, end: f64| {
400 let span = end - start;
401 if span.abs() <= f64::EPSILON || !value.is_finite() {
402 return None;
403 }
404 Some(((value - start) / span).clamp(0.0, 1.0))
405 };
406 let u = match (fraction(low.x, u0, u1), fraction(high.x, u0, u1)) {
407 (Some(a), Some(b)) if b - a > 1e-6 => [a, b],
408 _ => [0.0, 1.0],
409 };
410 let v = match (fraction(low.y, v0, v1), fraction(high.y, v0, v1)) {
411 (Some(a), Some(b)) if b - a > 1e-6 => [a, b],
412 _ => [0.0, 1.0],
413 };
414 Ok((u, v))
415}
416
417fn domain_end_moves(grow_min: f64, grow_max: f64, [ta, tb]: [f64; 2]) -> (f64, f64) {
427 let span = tb - ta;
428 if span <= 1e-6 {
429 return (grow_min, grow_max);
430 }
431 let rate = (grow_min + grow_max) / span;
432 let back = grow_min + ta * rate;
433 let forward = (1.0 - ta) * rate - grow_min;
434 (back.max(0.0), forward.max(0.0))
435}
436
437pub fn offset_surface(
441 face: &FaceRecord,
442 distance: f64,
443 planar_extension: f64,
444) -> Result<NurbsSurface, KernelRefusal> {
445 offset_surface_with_lane(face, distance, &CarrierExtension::uniform(planar_extension))
446 .map(|(surface, _)| surface)
447}
448
449pub fn offset_surface_measured(
458 face: &FaceRecord,
459 distance: f64,
460 planar_extension: f64,
461 band: f64,
462) -> Result<MeasuredOffsetSurface, KernelRefusal> {
463 offset_surface_measured_sided(
464 face,
465 distance,
466 &CarrierExtension::uniform(planar_extension),
467 band,
468 )
469}
470
471pub fn offset_surface_measured_sided(
473 face: &FaceRecord,
474 distance: f64,
475 extension: &CarrierExtension,
476 band: f64,
477) -> Result<MeasuredOffsetSurface, KernelRefusal> {
478 offset_surface_measured_sided_mode(face, distance, extension, band,
479 std::env::var("BREP_OFFSET_SPHERE_READING").map_or(true, |v| v != "0" && v != "off"))
480}
481
482fn offset_surface_measured_sided_mode(
483 face: &FaceRecord, distance: f64, extension: &CarrierExtension, band: f64, reuse: bool,
484) -> Result<MeasuredOffsetSurface, KernelRefusal> {
485 let mut sphere_reading = None;
486 let (surface, lane, extended) = offset_surface_constructed(face, distance, extension, &mut sphere_reading)?;
487 let fit = match lane {
488 OffsetSurfaceLane::Affine => Some(MeasuredTolerance::exact(band)),
489 OffsetSurfaceLane::Analytic if reuse && sphere_reading.is_some() => {
494 if crate::profile::profile_enabled() {
498 profile_line!("offset.sphere_reading.profile reused=1");
499 }
500 Some(MeasuredTolerance::new(sphere_reading.unwrap(), band))
501 }
502 OffsetSurfaceLane::Analytic | OffsetSurfaceLane::Fit => {
503 Some(measure_surface_fit_against_pointwise_offset(
504 extended.as_ref().unwrap_or(&face.surface),
505 face.same_sense,
506 &surface,
507 distance,
508 band,
509 )?)
510 }
511 OffsetSurfaceLane::Reparameterised => None,
512 };
513 Ok(MeasuredOffsetSurface { surface, lane, fit })
514}
515
516fn sphere_offset_surface(face: &FaceRecord, distance: f64, reading: &mut Option<f64>) -> Result<Option<NurbsSurface>, KernelRefusal> {
545 if std::env::var_os("BREP_NO_ANALYTIC_OFFSET").is_some() {
549 return Ok(None);
550 }
551 let source = &face.surface;
552 let Some((centre, radius)) = source.analytic().and_then(|shape| shape.sphere_geometry()) else {
553 return Ok(None);
554 };
555 let ([u0, u1], [v0, v1]) = domains(source)?;
556 let Ok(sample) = OffsetEvaluator::new(
563 "offset_surface",
564 source,
565 OffsetNormal::FaceStable {
566 same_sense: face.same_sense,
567 },
568 )
569 .at((u0 + u1) / 2.0, (v0 + v1) / 2.0, -distance)
570 else {
571 return Ok(None);
572 };
573 let source_radial = sample.source.sub(centre);
574 let offset_radial = sample.point.sub(centre);
575 let offset_radius = offset_radial.length();
576 let scale = radius.max(1.0);
577 if radius <= 1e-12
578 || offset_radius <= 1e-9 * scale
579 || offset_radial.dot(source_radial) <= 0.0
582 || ((offset_radius - radius).abs() - distance.abs()).abs() > 1e-9 * scale
585 {
586 return Ok(None);
587 }
588 let factor = offset_radius / radius;
589 let Ok(controls) = source
592 .control_points
593 .iter()
594 .map(|row| {
595 row.iter()
596 .map(|control| {
597 Ok(Vec4::from_point(
598 centre.add(control.point().or_refuse(KernelStage::Refine, "point")?.sub(centre).scale(factor)),
599 control.w,
600 ))
601 })
602 .collect::<Result<Vec<_>, KernelRefusal>>()
603 })
604 .collect::<Result<Vec<Vec<_>>, KernelRefusal>>()
605 else {
606 return Ok(None);
607 };
608 let Ok(surface) = NurbsSurface::new(
609 source.degree_u,
610 source.degree_v,
611 source.knots_u.clone(),
612 source.knots_v.clone(),
613 controls,
614 ) else {
615 return Ok(None);
616 };
617 match crate::offset_measure::sphere_fit_reading(
621 source,
622 face.same_sense,
623 &surface,
624 distance,
625 1e-9 * scale,
626 ) {
627 Ok(measured) if measured.within_band() => {
628 *reading = Some(measured.deviation());
629 Ok(Some(surface))
630 },
631 _ => Ok(None),
632 }
633}
634
635fn revolved_arc_offset_surface(
668 face: &FaceRecord,
669 distance: f64,
670) -> Result<Option<NurbsSurface>, KernelRefusal> {
671 if std::env::var_os("BREP_NO_ANALYTIC_OFFSET").is_some() {
672 return Ok(None);
673 }
674 let source = &face.surface;
675 match source.analytic() {
676 Some(crate::AnalyticSurface::Revolution { .. } | crate::AnalyticSurface::Torus { .. }) => {}
677 _ => return Ok(None),
678 }
679 let rows = &source.control_points;
680 let Some(first_row) = rows.first() else {
681 return Ok(None);
682 };
683 if rows.iter().flatten().any(|control| !(control.w > 0.0)) {
684 return Ok(None);
685 }
686 let Ok(generatrix) = NurbsCurve::new(source.degree_v, source.knots_v.clone(), first_row.clone())
687 else {
688 return Ok(None);
689 };
690 let [t0, t1] = generatrix.domain().or_refuse(KernelStage::Refine, "domain")?;
691 let at = |fraction: f64| generatrix.evaluate(t0 + (t1 - t0) * fraction);
692 let (a, b, e) = (at(0.0).or_refuse(KernelStage::Refine, "at")?, at(1.0 / 3.0).or_refuse(KernelStage::Refine, "at")?, at(2.0 / 3.0).or_refuse(KernelStage::Refine, "at")?);
694 let (ab, ae) = (b.sub(a), e.sub(a));
695 let normal = ab.cross(ae);
696 let normal_length_squared = normal.dot(normal);
697 let Some(centre) = crate::circumcenter(a, b, e) else { return Ok(None); };
698 let radius = a.sub(centre).length();
699 let scale = (centre.length() + radius).max(1.0);
700 let band = 1e-9 * scale;
701 let unit_normal = normal.scale(1.0 / normal_length_squared.sqrt());
702 let samples = 16 * generatrix.control_points.len();
703 for index in 0..=samples {
704 let point = at(index as f64 / samples as f64).or_refuse(KernelStage::Refine, "at")?;
705 let radial = point.sub(centre);
706 if (radial.length() - radius).abs() > band || radial.dot(unit_normal).abs() > band {
707 return Ok(None);
708 }
709 }
710 let controls = first_row
713 .iter()
714 .map(|control| control.point().or_refuse(KernelStage::Refine, "point"))
715 .collect::<Result<Vec<_>, KernelRefusal>>()?;
716 let origin = controls[0];
717 let mut best = (0usize, 0usize, 0.0f64);
718 for first in 1..controls.len() {
719 for second in first + 1..controls.len() {
720 let area = controls[first]
721 .sub(origin)
722 .cross(controls[second].sub(origin))
723 .length();
724 if area > best.2 {
725 best = (first, second, area);
726 }
727 }
728 }
729 let (first, second, area) = best;
730 if area <= 1e-9 * scale * scale {
731 return Ok(None);
732 }
733 let (along_first, along_second) = (controls[first].sub(origin), controls[second].sub(origin));
734 let offset_from_origin = centre.sub(origin);
735 let (g11, g12, g22) = (
736 along_first.dot(along_first),
737 along_first.dot(along_second),
738 along_second.dot(along_second),
739 );
740 let determinant = g11 * g22 - g12 * g12;
741 let (r1, r2) = (along_first.dot(offset_from_origin), along_second.dot(offset_from_origin));
742 let s = (r1 * g22 - r2 * g12) / determinant;
743 let t = (r2 * g11 - r1 * g12) / determinant;
744 if origin
745 .add(along_first.scale(s))
746 .add(along_second.scale(t))
747 .sub(centre)
748 .length()
749 > band
750 {
751 return Ok(None);
752 }
753 let ([u0, _], [v0, v1]) = domains(source)?;
756 let Ok(sample) = OffsetEvaluator::new(
757 "offset_surface",
758 source,
759 OffsetNormal::FaceStable {
760 same_sense: face.same_sense,
761 },
762 )
763 .at(u0, (v0 + v1) / 2.0, -distance)
764 else {
765 return Ok(None);
766 };
767 let source_radial = sample.source.sub(centre);
768 let offset_radial = sample.point.sub(centre);
769 let offset_radius = offset_radial.length();
770 if (source_radial.length() - radius).abs() > band
771 || offset_radius <= band
772 || offset_radial.dot(source_radial) <= 0.0
773 || ((offset_radius - radius).abs() - distance.abs()).abs() > band
774 {
775 return Ok(None);
776 }
777 let factor = offset_radius / radius;
778 let Ok(offset_rows) = rows
779 .iter()
780 .map(|row| {
781 let points = row.iter().map(|control| control.point().or_refuse(KernelStage::Refine, "point")).collect::<Result<Vec<_>, KernelRefusal>>()?;
782 let row_origin = points[0];
783 let row_centre = row_origin
784 .add(points[first].sub(row_origin).scale(s))
785 .add(points[second].sub(row_origin).scale(t));
786 Ok(points
787 .iter()
788 .zip(row)
789 .map(|(point, control)| {
790 Vec4::from_point(row_centre.add(point.sub(row_centre).scale(factor)), control.w)
791 })
792 .collect::<Vec<_>>())
793 })
794 .collect::<Result<Vec<Vec<_>>, KernelRefusal>>()
795 else {
796 return Ok(None);
797 };
798 let Ok(surface) = NurbsSurface::new(
799 source.degree_u,
800 source.degree_v,
801 source.knots_u.clone(),
802 source.knots_v.clone(),
803 offset_rows,
804 ) else {
805 return Ok(None);
806 };
807 match measure_surface_fit_against_pointwise_offset(
808 source,
809 face.same_sense,
810 &surface,
811 distance,
812 band,
813 ) {
814 Ok(measured) if measured.within_band() => Ok(Some(surface)),
815 _ => Ok(None),
816 }
817}
818
819pub(crate) fn exact_cylinder_offsets_on() -> bool {
824 switch!("BREP_OFFSET_EXACT_CYLINDER", true)
825}
826
827
828pub(crate) fn cylinder_offset_factor(
835 surface: &NurbsSurface,
836 same_sense: bool,
837 distance: f64,
838) -> Result<Option<(Vec3, Vec3, f64, f64)>, KernelRefusal> {
839 if std::env::var_os("BREP_NO_ANALYTIC_OFFSET").is_some() {
840 return Ok(None);
841 }
842 let Some(crate::AnalyticSurface::RuledRevolution { frame, rho0, rho1, height }) = surface.analytic() else {
843 return Ok(None);
844 };
845 let radius = 0.5 * (rho0 + rho1);
846 if !(radius > 0.0) || (rho0 - rho1).abs() > 1e-9 * radius.max(1.0) {
847 return Ok(None);
848 }
849 let Ok(axis) = frame.axis.normalized() else {
850 return Ok(None);
851 };
852 let origin = frame.origin;
853 let band = 1e-9 * (origin.length() + radius + height.abs()).max(1.0);
854 let perpendicular = |point: Vec3| {
855 let along = point.sub(origin);
856 along.sub(axis.scale(along.dot(axis)))
857 };
858 let [u0, u1] = surface.domain_u().or_refuse(KernelStage::Refine, "domain_u")?;
859 let [v0, v1] = surface.domain_v().or_refuse(KernelStage::Refine, "domain_v")?;
860 let Ok(sample) = OffsetEvaluator::new("offset_surface", surface, OffsetNormal::FaceStable { same_sense })
861 .at((u0 + u1) * 0.5, (v0 + v1) * 0.5, -distance)
862 else {
863 return Ok(None);
864 };
865 let (source_radial, offset_radial) = (perpendicular(sample.source), perpendicular(sample.point));
866 let offset_radius = offset_radial.length();
867 if (source_radial.length() - radius).abs() > band
868 || offset_radius <= band
869 || offset_radial.dot(source_radial) <= 0.0
870 || ((offset_radius - radius).abs() - distance.abs()).abs() > band
871 || sample.point.sub(sample.source).dot(axis).abs() > band
872 {
873 return Ok(None);
874 }
875 Ok(Some((origin, axis, radius, offset_radius / radius)))
876}
877
878fn cylinder_offset_surface(
891 reference: &NurbsSurface,
892 same_sense: bool,
893 distance: f64,
894) -> Result<Option<NurbsSurface>, KernelRefusal> {
895 let Some((origin, axis, radius, factor)) = cylinder_offset_factor(reference, same_sense, distance)? else {
896 return Ok(None);
897 };
898 if reference.control_points.iter().flatten().any(|control| !(control.w > 0.0)) {
899 return Ok(None);
900 }
901 let Ok(rows) = reference
902 .control_points
903 .iter()
904 .map(|row| {
905 row.iter()
906 .map(|control| {
907 let point = control.point()?;
908 let foot = origin.add(axis.scale(point.sub(origin).dot(axis)));
909 Ok(Vec4::from_point(foot.add(point.sub(foot).scale(factor)), control.w))
910 })
911 .collect::<Result<Vec<_>, String>>()
912 })
913 .collect::<Result<Vec<Vec<_>>, String>>()
914 else {
915 return Ok(None);
916 };
917 let Ok(surface) = NurbsSurface::new(
918 reference.degree_u,
919 reference.degree_v,
920 reference.knots_u.clone(),
921 reference.knots_v.clone(),
922 rows,
923 ) else {
924 return Ok(None);
925 };
926 let band = 1e-9 * (origin.length() + radius * factor.max(1.0)).max(1.0);
932 let evaluator = OffsetEvaluator::new("offset_surface", reference, OffsetNormal::FaceStable { same_sense });
933 let [u0, u1] = reference.domain_u().or_refuse(KernelStage::Refine, "domain_u")?;
934 let [v0, v1] = reference.domain_v().or_refuse(KernelStage::Refine, "domain_v")?;
935 for i in 0..=4 {
936 for j in 0..=2 {
937 let (u, v) = (u0 + (u1 - u0) * i as f64 / 4.0, v0 + (v1 - v0) * j as f64 / 2.0);
938 let (Ok(want), Ok(got)) = (evaluator.at(u, v, -distance), surface.evaluate(u, v)) else {
939 return Ok(None);
940 };
941 if !(want.point.sub(got).length() <= band) {
942 return Ok(None);
943 }
944 }
945 }
946 Ok(Some(surface))
947}
948
949pub(crate) fn offset_surface_with_lane(
950 face: &FaceRecord,
951 distance: f64,
952 extension: &CarrierExtension,
953) -> Result<(NurbsSurface, OffsetSurfaceLane), KernelRefusal> {
954 offset_surface_constructed(face, distance, extension, &mut None).map(|(surface, lane, _)| (surface, lane))
955}
956
957pub(crate) fn transpose_offset_surface(surface: &NurbsSurface) -> Result<NurbsSurface, KernelRefusal> {
961 let controls = (0..surface.control_points[0].len())
962 .map(|column| surface.control_points.iter().map(|row| row[column]).collect())
963 .collect();
964 NurbsSurface::new(surface.degree_v, surface.degree_u, surface.knots_v.clone(),
965 surface.knots_u.clone(), controls).or_refuse(KernelStage::Refine, "offset_transpose")
966}
967
968pub(crate) fn apex_end_reach_enabled() -> bool {
971 switch!("BREP_OFFSET_APEX_END_REACH", true)
972}
973
974
975fn offset_surface_constructed(
976 face: &FaceRecord,
977 distance: f64,
978 extension: &CarrierExtension,
979 sphere_reading: &mut Option<f64>,
980) -> Result<(NurbsSurface, OffsetSurfaceLane, Option<NurbsSurface>), KernelRefusal> {
981 let source = &face.surface;
982 let extension_active = extension.is_active();
983 let normalize_ruling = std::env::var("BREP_OFFSET_FIT_RESIDUAL").as_deref() != Ok("0")
989 && source.degree_u == 1 && source.control_points.len() == 2;
990 let normalize_sphere = std::env::var("BREP_OFFSET_FOLD_CHART").as_deref() != Ok("0")
991 && source.degree_u == 2 && source.degree_v == 2 && source.analytic().is_none();
992 if normalize_ruling || normalize_sphere {
993 let transposed = transpose_offset_surface(source)?;
994 let ruled = normalize_ruling && matches!(transposed.analytic(), Some(crate::AnalyticSurface::RuledRevolution { .. }));
995 let spherical = normalize_sphere && transposed.analytic().and_then(|shape| shape.sphere_geometry()).is_some();
996 if ruled || spherical {
997 let mut turned = face.clone();
998 turned.surface = transposed;
999 turned.same_sense = !turned.same_sense;
1000 for coedge in turned.loops.iter_mut().flat_map(|lp| &mut lp.coedges) {
1001 let controls = coedge.pcurve.control_points.iter().map(|point| {
1002 let mut swapped = *point;
1003 std::mem::swap(&mut swapped.x, &mut swapped.y);
1004 swapped
1005 }).collect();
1006 coedge.pcurve = NurbsCurve::new(coedge.pcurve.degree,
1007 coedge.pcurve.knots.clone(), controls).or_refuse(KernelStage::Refine, "offset_transpose_pcurve")?;
1008 }
1009 let turned_extension = CarrierExtension {
1010 u_min: extension.v_min, u_max: extension.v_max,
1011 v_min: extension.u_min, v_max: extension.u_max,
1012 };
1013 let (surface, lane, extended) = offset_surface_constructed(&turned, distance, &turned_extension, &mut None)?;
1014 return Ok((transpose_offset_surface(&surface)?, lane,
1015 extended.as_ref().map(transpose_offset_surface).transpose()?));
1016 }
1017 }
1018 if source.is_affine().or_refuse(KernelStage::Refine, "is_affine")? {
1019 let ([u0, u1], [v0, v1]) = domains(source)?;
1020 let normal = stable_face_normal(face, (u0 + u1) / 2.0, (v0 + v1) / 2.0)?;
1021 let shift = normal.scale(-distance);
1022 let mut points = source
1023 .control_points
1024 .iter()
1025 .map(|row| {
1026 row.iter()
1027 .map(|control| Ok(control.point().or_refuse(KernelStage::Refine, "point")?.add(shift)))
1028 .collect::<Result<Vec<_>, KernelRefusal>>()
1029 })
1030 .collect::<Result<Vec<_>, KernelRefusal>>()?;
1031 if extension_active {
1032 let p00 = points[0][0];
1033 let p01 = points[0][1];
1034 let p10 = points[1][0];
1035 let direction_u = p10.sub(p00).normalized().or_refuse(KernelStage::Refine, "normalized")?;
1036 let direction_v = p01.sub(p00).normalized().or_refuse(KernelStage::Refine, "normalized")?;
1037 let (trim_u, trim_v) = trim_domain_fractions(face, [u0, u1], [v0, v1])?;
1043 let (back_u, forward_u) = domain_end_moves(extension.u_min, extension.u_max, trim_u);
1044 let (back_v, forward_v) = domain_end_moves(extension.v_min, extension.v_max, trim_v);
1045 points[0][0] = p00
1046 .sub(direction_u.scale(back_u))
1047 .sub(direction_v.scale(back_v));
1048 points[0][1] = p01
1049 .sub(direction_u.scale(back_u))
1050 .add(direction_v.scale(forward_v));
1051 points[1][0] = p10
1052 .add(direction_u.scale(forward_u))
1053 .sub(direction_v.scale(back_v));
1054 points[1][1] = points[1][1]
1055 .add(direction_u.scale(forward_u))
1056 .add(direction_v.scale(forward_v));
1057 }
1058 let controls = points
1059 .into_iter()
1060 .enumerate()
1061 .map(|(row, points)| {
1062 points
1063 .into_iter()
1064 .enumerate()
1065 .map(|(column, point)| {
1066 Vec4::from_point(point, source.control_points[row][column].w)
1067 })
1068 .collect()
1069 })
1070 .collect();
1071 let lane = if extension_active {
1075 OffsetSurfaceLane::Reparameterised
1076 } else {
1077 OffsetSurfaceLane::Affine
1078 };
1079 return Ok((
1080 NurbsSurface::new(
1081 source.degree_u,
1082 source.degree_v,
1083 source.knots_u.clone(),
1084 source.knots_v.clone(),
1085 controls,
1086 ).or_refuse(KernelStage::Refine, "new")?,
1087 lane,
1088 None,
1089 ));
1090 }
1091
1092 if let Some(surface) = sphere_offset_surface(face, distance, sphere_reading)? {
1102 return Ok((surface, OffsetSurfaceLane::Analytic, None));
1103 }
1104 if let Some(surface) = revolved_arc_offset_surface(face, distance)? {
1107 return Ok((surface, OffsetSurfaceLane::Analytic, None));
1108 }
1109
1110 let knot_u = KnotVector::new(source.knots_u.clone(), source.degree_u).or_refuse(KernelStage::Refine, "new")?;
1111 let knot_v = KnotVector::new(source.knots_v.clone(), source.degree_v).or_refuse(KernelStage::Refine, "new")?;
1112 let parameters_u = greville_parameters(&knot_u);
1113 let parameters_v = greville_parameters(&knot_v);
1114 let evaluator = OffsetEvaluator::new(
1120 "offset_surface",
1121 source,
1122 OffsetNormal::FaceStable {
1123 same_sense: face.same_sense,
1124 },
1125 );
1126 let mut samples = Vec::new();
1127 for &u in ¶meters_u {
1128 let mut row = Vec::new();
1129 for &v in ¶meters_v {
1130 row.push(evaluator.at(u, v, -distance).or_refuse(KernelStage::Refine, "at")?.point);
1131 }
1132 samples.push(row);
1133 }
1134 let mut reparameterised = false;
1149 let mut extended_source: Option<NurbsSurface> = None;
1150 if parameters_v.len() == 2 && parameters_u.len() >= 2 {
1157 let centroid = |column: usize| {
1158 let mut sum = Vec3::default();
1159 for row in &samples {
1160 sum = sum.add(row[column]);
1161 }
1162 sum.scale(1.0 / samples.len() as f64)
1163 };
1164 let near_centroid = centroid(0);
1165 let far_centroid = centroid(1);
1166 let mut inverted = true;
1167 let mut pinch_fraction = 0.0f64;
1168 let mut near_mean = 0.0f64;
1169 let mut far_mean = 0.0f64;
1170 for row in &samples {
1171 let near_radial = row[0].sub(near_centroid);
1172 let far_radial = row[1].sub(far_centroid);
1173 let near_len = near_radial.length();
1174 let far_len = far_radial.length();
1175 if near_len <= 1e-9 || far_len <= 1e-9 {
1176 inverted = false;
1177 break;
1178 }
1179 if near_radial.dot(far_radial) >= 0.0 {
1180 inverted = false;
1181 break;
1182 }
1183 pinch_fraction += near_len / (near_len + far_len) / samples.len() as f64;
1184 near_mean += near_len / samples.len() as f64;
1185 far_mean += far_len / samples.len() as f64;
1186 }
1187 if inverted {
1188 let retrim_far = far_mean <= near_mean;
1189 let (_, trim_v) = trim_domain_fractions(face, knot_u.domain(), knot_v.domain())?;
1213 let trim_reaches_the_pinch = if retrim_far {
1214 trim_v[1] > pinch_fraction + 1e-9
1215 } else {
1216 trim_v[0] < pinch_fraction - 1e-9
1217 };
1218 if trim_reaches_the_pinch {
1219 reparameterised = true;
1222 for row in &mut samples {
1223 let near = row[0];
1224 let far = row[1];
1225 let pinch = near.add(far.sub(near).scale(pinch_fraction));
1226 if retrim_far {
1227 row[1] = pinch;
1228 } else {
1229 row[0] = pinch;
1230 }
1231 }
1232 }
1233 }
1234 if extension.extends_v() && !inverted {
1254 let apex_end_reach = apex_end_reach_enabled();
1263 let mut min_ruling = f64::MAX;
1264 let mut back_allowance = f64::MAX;
1265 let mut forward_allowance = f64::MAX;
1266 let mut extendable = true;
1267 for row in &samples {
1268 let ruling = row[1].sub(row[0]);
1269 let length = ruling.length();
1270 min_ruling = min_ruling.min(length);
1271 let near_radial = row[0].sub(near_centroid).length();
1275 let far_radial = row[1].sub(far_centroid).length();
1276 if (far_radial - near_radial).abs() > 1e-9 {
1277 let apex_at = near_radial / (near_radial - far_radial);
1278 if apex_end_reach && apex_at.abs() <= 1e-9 {
1279 back_allowance = 0.0;
1280 continue;
1281 }
1282 if apex_end_reach && (apex_at - 1.0).abs() <= 1e-9 {
1283 forward_allowance = 0.0;
1284 continue;
1285 }
1286 if (-1e-9..=1.0 + 1e-9).contains(&apex_at) {
1287 extendable = false;
1289 break;
1290 }
1291 if apex_at < 0.0 {
1292 back_allowance = back_allowance.min(0.9 * -apex_at);
1293 } else {
1294 forward_allowance = forward_allowance.min(0.9 * (apex_at - 1.0));
1295 }
1296 }
1297 }
1298 let mut source_back = f64::MAX;
1306 let mut source_forward = f64::MAX;
1307 if extendable {
1308 let source_centroid = |column: usize| {
1309 let mut sum = Vec3::default();
1310 for row in &source.control_points {
1311 if let Ok(point) = row[column].point() {
1312 sum = sum.add(point);
1313 }
1314 }
1315 sum.scale(1.0 / source.control_points.len() as f64)
1316 };
1317 let (near_source, far_source) = (source_centroid(0), source_centroid(1));
1318 for row in &source.control_points {
1319 let (near, far) = (row[0].point().or_refuse(KernelStage::Refine, "point")?, row[1].point().or_refuse(KernelStage::Refine, "point")?);
1320 let near_radial = near.sub(near_source).length();
1321 let far_radial = far.sub(far_source).length();
1322 if (far_radial - near_radial).abs() > 1e-9 {
1323 let apex_at = near_radial / (near_radial - far_radial);
1324 if apex_end_reach && apex_at.abs() <= 1e-9 {
1325 source_back = 0.0;
1326 continue;
1327 }
1328 if apex_end_reach && (apex_at - 1.0).abs() <= 1e-9 {
1329 source_forward = 0.0;
1330 continue;
1331 }
1332 if (-1e-9..=1.0 + 1e-9).contains(&apex_at) {
1333 extendable = false;
1334 break;
1335 }
1336 if apex_at < 0.0 {
1337 source_back = source_back.min(0.9 * -apex_at);
1338 } else {
1339 source_forward = source_forward.min(0.9 * (apex_at - 1.0));
1340 }
1341 }
1342 }
1343 }
1344 let (back_allowance, forward_allowance) =
1345 (back_allowance.min(source_back), forward_allowance.min(source_forward));
1346 if extendable && min_ruling > 1e-9 {
1347 let (_, trim_v) = trim_domain_fractions(face, knot_u.domain(), knot_v.domain())?;
1354 let (back_world, forward_world) =
1355 domain_end_moves(extension.v_min, extension.v_max, trim_v);
1356 let (requested_back, requested_forward) = (back_world / min_ruling, forward_world / min_ruling);
1357 let apex_end = analytic_offset_apex(source, &samples, offset_fit_contract(source)).and_then(
1369 |(at, apex)| {
1370 if at > 1.0 && at - 1.0 <= requested_forward && at - 1.0 <= source_forward {
1371 Some((false, at - 1.0, apex))
1372 } else if at < 0.0 && -at <= requested_back && -at <= source_back {
1373 Some((true, -at, apex))
1374 } else {
1375 None
1376 }
1377 },
1378 );
1379 let back = match apex_end {
1380 Some((true, reach, _)) => reach,
1381 _ => requested_back.min(back_allowance),
1382 };
1383 let forward = match apex_end {
1384 Some((false, reach, _)) => reach,
1385 _ => requested_forward.min(forward_allowance),
1386 };
1387 let stretched = samples
1388 .iter()
1389 .map(|row| {
1390 let ruling = row[1].sub(row[0]);
1391 let (near, far) = (row[0].sub(ruling.scale(back)), row[1].add(ruling.scale(forward)));
1392 match apex_end {
1393 Some((true, _, apex)) => vec![apex, far],
1394 Some((false, _, apex)) => vec![near, apex],
1395 None => vec![near, far],
1396 }
1397 })
1398 .collect::<Vec<_>>();
1399 let extended = extend_along_rulings(source, back, forward)?;
1400 let extended_offsets = OffsetEvaluator::new(
1401 "offset_surface",
1402 &extended,
1403 OffsetNormal::FaceStable {
1404 same_sense: face.same_sense,
1405 },
1406 );
1407 let [v0, v1] = knot_v.domain();
1408 let mut developability = 0.0f64;
1409 for (row, &u) in stretched.iter().zip(¶meters_u) {
1410 for fraction in [0.0, 0.5, 1.0] {
1411 let exact = extended_offsets.at(u, v0 + (v1 - v0) * fraction, -distance).or_refuse(KernelStage::Refine, "at")?.point;
1412 let along = row[0].add(row[1].sub(row[0]).scale(fraction));
1413 developability = developability.max(exact.sub(along).length());
1414 }
1415 }
1416 let contract = offset_fit_contract(source);
1417 let stretch_kept = developability <= contract;
1418 samples = if stretch_kept {
1419 stretched
1420 } else {
1421 parameters_u
1422 .iter()
1423 .map(|&u| {
1424 parameters_v
1425 .iter()
1426 .map(|&v| Ok(extended_offsets.at(u, v, -distance).or_refuse(KernelStage::Refine, "at")?.point))
1427 .collect::<Result<Vec<_>, KernelRefusal>>()
1428 })
1429 .collect::<Result<Vec<_>, KernelRefusal>>()?
1430 };
1431 if let Ok(path) = std::env::var("BREP_OFFSET_FIT_TRACE") {
1432 use std::io::Write;
1433 let line = format!(
1434 "extension deg=({},{}) net=({},{}) d={distance:.6} back={back:.6} forward={forward:.6} \
1435 developability={developability:.3e} contract={contract:.3e} lane={} thread={} args={}\n",
1436 source.degree_u,
1437 source.degree_v,
1438 source.control_points.len(),
1439 source.control_points[0].len(),
1440 if stretch_kept { "stretch" } else { "pointwise" },
1441 std::thread::current().name().unwrap_or("-"),
1442 std::env::args().skip(1).collect::<Vec<_>>().join(" "),
1443 );
1444 if let Ok(mut file) = std::fs::OpenOptions::new().create(true).append(true).open(path) {
1445 let _ = file.write_all(line.as_bytes());
1446 }
1447 }
1448 extended_source = Some(extended);
1449 }
1450 }
1451 }
1452 let reference = extended_source.as_ref().unwrap_or(source);
1453 if !reparameterised && exact_cylinder_offsets_on() {
1459 if let Some(surface) = cylinder_offset_surface(reference, face.same_sense, distance)? {
1460 return Ok((surface, OffsetSurfaceLane::Analytic, extended_source));
1461 }
1462 }
1463 let weights = reference
1464 .control_points
1465 .iter()
1466 .map(|row| row.iter().map(|point| point.w).collect::<Vec<_>>())
1467 .collect::<Vec<_>>();
1468 let fitted = NurbsSurface::new(
1469 source.degree_u,
1470 source.degree_v,
1471 source.knots_u.clone(),
1472 source.knots_v.clone(),
1473 interpolate_tensor(
1474 &knot_u,
1475 &knot_v,
1476 ¶meters_u,
1477 ¶meters_v,
1478 &samples,
1479 &weights,
1480 )?,
1481 ).or_refuse(KernelStage::Refine, "new")?;
1482 if reparameterised {
1483 return Ok((fitted, OffsetSurfaceLane::Reparameterised, None));
1484 }
1485 let reference_offsets = OffsetEvaluator::new(
1486 "offset_surface",
1487 reference,
1488 OffsetNormal::FaceStable {
1489 same_sense: face.same_sense,
1490 },
1491 );
1492 Ok((
1493 refine_offset_fit(reference, &reference_offsets, distance, fitted)?,
1494 OffsetSurfaceLane::Fit,
1495 extended_source,
1496 ))
1497}
1498
1499
1500fn analytic_offset_apex(source: &NurbsSurface, samples: &[Vec<Vec3>], contract: f64) -> Option<(f64, Vec3)> {
1509 let Some(crate::AnalyticSurface::RuledRevolution { frame, .. }) = crate::analytic_surface::recognize(source) else {
1510 return None;
1511 };
1512 let radial = |point: Vec3| {
1513 let offset = point.sub(frame.origin);
1514 offset.sub(frame.axis.scale(offset.dot(frame.axis)))
1515 };
1516 let mut crossings = Vec::with_capacity(samples.len());
1517 for row in samples {
1518 let (near, far) = (radial(row[0]), radial(row[1]));
1519 let step = far.sub(near);
1520 let at = -near.dot(step) / step.dot(step);
1521 if !at.is_finite() {
1522 return None;
1523 }
1524 crossings.push((at, row[0].add(row[1].sub(row[0]).scale(at))));
1525 }
1526 let count = crossings.len() as f64;
1529 let at = crossings.iter().map(|(at, _)| at).sum::<f64>() / count;
1530 let first = crossings.first()?.1;
1531 let spread = crossings.iter().fold(Vec3::default(), |sum, (_, point)| sum.add(point.sub(first)));
1532 let centre = first.add(spread.scale(1.0 / count));
1533 let apex = frame.origin.add(frame.axis.scale(centre.sub(frame.origin).dot(frame.axis)));
1534 crossings
1537 .iter()
1538 .all(|(_, point)| radial(*point).length() <= contract && point.sub(apex).length() <= contract)
1539 .then_some((at, apex))
1540}
1541
1542fn extend_along_rulings(surface: &NurbsSurface, back: f64, forward: f64) -> Result<NurbsSurface, KernelRefusal> {
1549 let rows = surface
1550 .control_points
1551 .iter()
1552 .map(|row| {
1553 let step = row[1].add(row[0].scale(-1.0));
1554 vec![row[0].add(step.scale(-back)), row[1].add(step.scale(forward))]
1555 })
1556 .collect::<Vec<_>>();
1557 if rows.iter().flatten().any(|point| point.w <= 1e-12) {
1558 return Err(KernelRefusal::unsupported(KernelStage::Refine, "offset_ruled_extension_weight", format!(
1559 "offset_surface: extending a ruled face by ({back:.6}, {forward:.6}) of its rulings \
1560 carries a weight through zero; the extended face is not this surface"
1561 )));
1562 }
1563 NurbsSurface::new(surface.degree_u, 1, surface.knots_u.clone(), surface.knots_v.clone(), rows)
1564 .or_refuse(KernelStage::Refine, "new")
1565}
1566
1567fn offset_fit_contract(source: &NurbsSurface) -> f64 {
1570 let scale = crate::model_scale(
1571 source
1572 .control_points
1573 .iter()
1574 .flatten()
1575 .filter_map(|point| point.point().ok()),
1576 );
1577 crate::KernelTolerances::for_scale(scale, 1e-7).intersection_fit
1578}
1579
1580const OFFSET_FIT_REFINEMENTS: usize = 6;
1582
1583const OFFSET_FIT_MAX_CONTROLS_SEPARABLE: usize = 16_384;
1586const OFFSET_FIT_MAX_CONTROLS_TENSOR: usize = 900;
1587
1588#[derive(Clone, Debug)]
1593struct OffsetFitDeviation {
1594 along_u: f64,
1595 along_v: f64,
1596 between: f64,
1597 u_sites: Vec<(f64, f64)>,
1598 v_sites: Vec<(f64, f64)>,
1599 between_uv: (f64, f64),
1600}
1601
1602impl OffsetFitDeviation {
1603 fn worst(&self) -> f64 {
1604 self.along_u.max(self.along_v).max(self.between)
1605 }
1606
1607 fn refinement_sites(&self, bar: f64) -> (Vec<f64>, Vec<f64>) {
1608 let mut u = self.u_sites.iter().filter(|(_, miss)| *miss > bar)
1609 .map(|(parameter, _)| *parameter).collect::<Vec<_>>();
1610 let mut v = self.v_sites.iter().filter(|(_, miss)| *miss > bar)
1611 .map(|(parameter, _)| *parameter).collect::<Vec<_>>();
1612 if u.is_empty() && v.is_empty() && self.between > bar {
1613 u.push(self.between_uv.0);
1614 v.push(self.between_uv.1);
1615 }
1616 (u, v)
1617 }
1618}
1619
1620fn greville_midpoints(parameters: &[f64]) -> Vec<f64> {
1621 parameters
1622 .windows(2)
1623 .filter(|pair| pair[1] - pair[0] > 1e-12)
1624 .map(|pair| 0.5 * (pair[0] + pair[1]))
1625 .collect()
1626}
1627
1628fn offset_fit_deviation(
1629 fitted: &NurbsSurface,
1630 evaluator: &OffsetEvaluator,
1631 distance: f64,
1632) -> Result<OffsetFitDeviation, KernelRefusal> {
1633 let parameters_u = greville_parameters(&KnotVector::new(fitted.knots_u.clone(), fitted.degree_u).or_refuse(KernelStage::Refine, "new")?);
1634 let parameters_v = greville_parameters(&KnotVector::new(fitted.knots_v.clone(), fitted.degree_v).or_refuse(KernelStage::Refine, "new")?);
1635 let (middle_u, middle_v) = (greville_midpoints(¶meters_u), greville_midpoints(¶meters_v));
1636 let residual_locus = std::env::var("BREP_OFFSET_FIT_RESIDUAL").as_deref() != Ok("0");
1637 let gap = |u: f64, v: f64| -> Result<f64, KernelRefusal> {
1638 Ok(fitted.evaluate(u, v).or_refuse(KernelStage::Refine, "evaluate")?.sub(evaluator.at(u, v, -distance).or_refuse(KernelStage::Refine, "at")?.point).length())
1639 };
1640 let mut deviation = OffsetFitDeviation {
1641 along_u: 0.0,
1642 along_v: 0.0,
1643 between: 0.0,
1644 u_sites: Vec::new(),
1645 v_sites: if residual_locus { middle_v.iter().map(|&v| (v, 0.0)).collect() } else { Vec::new() },
1646 between_uv: (0.0, 0.0),
1647 };
1648 for &u in &middle_u {
1649 let mut along = 0.0f64;
1650 for &v in ¶meters_v {
1651 let miss = gap(u, v)?;
1652 along = along.max(miss);
1653 deviation.along_u = deviation.along_u.max(miss);
1654 }
1655 if residual_locus { deviation.u_sites.push((u, along)); }
1656 for &v in &middle_v {
1657 let miss = gap(u, v)?;
1658 if miss > deviation.between { deviation.between_uv = (u, v); }
1659 deviation.between = deviation.between.max(miss);
1660 }
1661 }
1662 for &u in ¶meters_u {
1663 for (index, &v) in middle_v.iter().enumerate() {
1664 let miss = gap(u, v)?;
1665 deviation.along_v = deviation.along_v.max(miss);
1666 if residual_locus { deviation.v_sites[index].1 = deviation.v_sites[index].1.max(miss); }
1667 }
1668 }
1669 Ok(deviation)
1670}
1671
1672fn raise_linear_directions(
1682 surface: &NurbsSurface,
1683 in_u: bool,
1684 in_v: bool,
1685) -> Result<Option<NurbsSurface>, KernelRefusal> {
1686 fn raise(knots: &[f64], row: &[Vec4]) -> Option<(Vec<f64>, Vec<Vec4>)> {
1687 let count = row.len();
1689 if knots.len() != count + 2 {
1690 return None;
1691 }
1692 let breaks = &knots[1..=count];
1693 if breaks.windows(2).any(|pair| pair[1] - pair[0] <= 1e-12) {
1694 return None;
1695 }
1696 let mut raised_knots = vec![breaks[0]; 4];
1697 for &interior in &breaks[1..count - 1] {
1698 raised_knots.extend([interior; 3]);
1699 }
1700 raised_knots.extend([breaks[count - 1]; 4]);
1701 let mut points = Vec::with_capacity(3 * (count - 1) + 1);
1702 for pair in row.windows(2) {
1703 let step = pair[1].add(pair[0].scale(-1.0));
1704 points.push(pair[0]);
1705 points.push(pair[0].add(step.scale(1.0 / 3.0)));
1706 points.push(pair[0].add(step.scale(2.0 / 3.0)));
1707 }
1708 points.push(row[count - 1]);
1709 Some((raised_knots, points))
1710 }
1711 let mut knots_u = surface.knots_u.clone();
1712 let mut knots_v = surface.knots_v.clone();
1713 let mut degree_u = surface.degree_u;
1714 let mut degree_v = surface.degree_v;
1715 let mut net = surface.control_points.clone();
1716 if in_v {
1717 if degree_v != 1 {
1718 return Ok(None);
1719 }
1720 let mut rows = Vec::with_capacity(net.len());
1721 for row in &net {
1722 let Some((knots, points)) = raise(&surface.knots_v, row) else {
1723 return Ok(None);
1724 };
1725 knots_v = knots;
1726 rows.push(points);
1727 }
1728 net = rows;
1729 degree_v = 3;
1730 }
1731 if in_u {
1732 if degree_u != 1 {
1733 return Ok(None);
1734 }
1735 let count_v = net[0].len();
1736 let mut columns = Vec::with_capacity(count_v);
1737 for column in 0..count_v {
1738 let row = net.iter().map(|row| row[column]).collect::<Vec<_>>();
1739 let Some((knots, points)) = raise(&surface.knots_u, &row) else {
1740 return Ok(None);
1741 };
1742 knots_u = knots;
1743 columns.push(points);
1744 }
1745 let count_u = columns[0].len();
1746 net = (0..count_u)
1747 .map(|row| columns.iter().map(|column| column[row]).collect())
1748 .collect();
1749 degree_u = 3;
1750 }
1751 Ok(Some(NurbsSurface::new(degree_u, degree_v, knots_u, knots_v, net).or_refuse(KernelStage::Refine, "new")?))
1752}
1753
1754fn halve_knot_spans(surface: &NurbsSurface, in_u: bool, in_v: bool) -> Result<NurbsSurface, KernelRefusal> {
1758 fn midpoints(knots: &[f64], degree: usize) -> Vec<f64> {
1759 let last = knots.len() - 1 - degree;
1760 knots[degree..=last]
1761 .windows(2)
1762 .filter(|pair| pair[1] - pair[0] > 1e-12)
1763 .map(|pair| 0.5 * (pair[0] + pair[1]))
1764 .collect()
1765 }
1766 insert_offset_fit_knots(surface,
1767 &if in_u { midpoints(&surface.knots_u, surface.degree_u) } else { Vec::new() },
1768 &if in_v { midpoints(&surface.knots_v, surface.degree_v) } else { Vec::new() })
1769}
1770
1771fn insert_offset_fit_knots(surface: &NurbsSurface, u: &[f64], v: &[f64]) -> Result<NurbsSurface, KernelRefusal> {
1776 let refine = |degree: usize, knots: &[f64], net: Vec<Vec<Vec4>>, inserted: &[f64]| -> Result<(Vec<f64>, Vec<Vec<Vec4>>), KernelRefusal> {
1777 let mut inserted = inserted.to_vec();
1778 inserted.sort_by(f64::total_cmp);
1779 inserted.dedup_by(|a, b| (*a - *b).abs() <= 1e-12);
1780 inserted.retain(|parameter| knots.iter().filter(|knot| (**knot - *parameter).abs() <= 1e-12).count() < degree);
1781 let mut refined_knots = knots.to_vec();
1782 let mut rows = Vec::with_capacity(net.len());
1783 for row in net {
1784 let mut curve = NurbsCurve::new(degree, knots.to_vec(), row).or_refuse(KernelStage::Refine, "new")?;
1785 for ¶meter in &inserted {
1786 curve = curve.insert_knot(parameter, 1).or_refuse(KernelStage::Refine, "insert_knot")?;
1787 }
1788 refined_knots = curve.knots.clone();
1789 rows.push(curve.control_points);
1790 }
1791 Ok((refined_knots, rows))
1792 };
1793 let mut knots_u = surface.knots_u.clone();
1794 let mut knots_v = surface.knots_v.clone();
1795 let mut net = surface.control_points.clone();
1796 if !v.is_empty() {
1797 let (knots, rows) = refine(surface.degree_v, &knots_v, net, v)?;
1798 knots_v = knots;
1799 net = rows;
1800 }
1801 if !u.is_empty() {
1802 let count_v = net[0].len();
1804 let columns = (0..count_v)
1805 .map(|column| net.iter().map(|row| row[column]).collect::<Vec<_>>())
1806 .collect::<Vec<_>>();
1807 let (knots, refined) = refine(surface.degree_u, &knots_u, columns, u)?;
1808 knots_u = knots;
1809 let count_u = refined[0].len();
1810 net = (0..count_u)
1811 .map(|row| refined.iter().map(|column| column[row]).collect())
1812 .collect();
1813 }
1814 NurbsSurface::new(surface.degree_u, surface.degree_v, knots_u, knots_v, net).or_refuse(KernelStage::Refine, "new")
1815}
1816
1817fn refine_offset_fit(
1853 source: &NurbsSurface,
1854 evaluator: &OffsetEvaluator,
1855 distance: f64,
1856 fitted: NurbsSurface,
1857) -> Result<NurbsSurface, KernelRefusal> {
1858 let scale = crate::model_scale(
1859 source
1860 .control_points
1861 .iter()
1862 .flatten()
1863 .filter_map(|point| point.point().ok()),
1864 );
1865 let bar = crate::KernelTolerances::for_scale(scale, 1e-7).intersection_fit;
1866 let trace = std::env::var("BREP_OFFSET_FIT_TRACE").ok();
1867 let started = web_time::Instant::now();
1868 let enabled = std::env::var("BREP_OFFSET_FIT_REFINE").as_deref() != Ok("0");
1869 let Ok(initial) = offset_fit_deviation(&fitted, evaluator, distance) else {
1870 return Ok(fitted);
1871 };
1872 let mut best = fitted;
1873 let mut best_deviation = initial.clone();
1874 let mut refined_source = source.clone();
1875 let mut rounds = 0usize;
1876 let mut residual_mode = false;
1877 let mut stop = "under bar";
1878 if enabled {
1879 while best_deviation.worst() > bar {
1880 if rounds == OFFSET_FIT_REFINEMENTS {
1881 stop = "refinement limit";
1882 break;
1883 }
1884 let (along_u, along_v) = (best_deviation.along_u > bar, best_deviation.along_v > bar);
1885 let (in_u, in_v) = if along_u || along_v {
1886 (along_u, along_v)
1887 } else {
1888 (true, true)
1889 };
1890 let (raise_u, raise_v) = (in_u && refined_source.degree_u == 1, in_v && refined_source.degree_v == 1);
1891 let raised = if raise_u || raise_v {
1892 raise_linear_directions(&refined_source, raise_u, raise_v)?
1893 } else {
1894 None
1895 };
1896 let (halve_u, halve_v) = match raised {
1897 Some(_) => (in_u && !raise_u, in_v && !raise_v),
1898 None => (in_u, in_v),
1899 };
1900 let raised = raised.unwrap_or_else(|| refined_source.clone());
1901 let candidate_source = if halve_u || halve_v {
1902 if residual_mode {
1903 let (u, v) = best_deviation.refinement_sites(bar);
1904 insert_offset_fit_knots(&raised,
1905 if halve_u { &u } else { &[] }, if halve_v { &v } else { &[] })?
1906 } else {
1907 halve_knot_spans(&raised, halve_u, halve_v)?
1908 }
1909 } else {
1910 raised
1911 };
1912 let weights = candidate_source
1913 .control_points
1914 .iter()
1915 .map(|row| row.iter().map(|point| point.w).collect::<Vec<_>>())
1916 .collect::<Vec<_>>();
1917 let controls = weights.len() * weights[0].len();
1918 let ceiling = if separable_weights(&weights).is_some() {
1919 OFFSET_FIT_MAX_CONTROLS_SEPARABLE
1920 } else {
1921 OFFSET_FIT_MAX_CONTROLS_TENSOR
1922 };
1923 if controls > ceiling {
1924 stop = "net ceiling";
1925 break;
1926 }
1927 let knot_u = KnotVector::new(candidate_source.knots_u.clone(), candidate_source.degree_u).or_refuse(KernelStage::Refine, "new")?;
1928 let knot_v = KnotVector::new(candidate_source.knots_v.clone(), candidate_source.degree_v).or_refuse(KernelStage::Refine, "new")?;
1929 let parameters_u = greville_parameters(&knot_u);
1930 let parameters_v = greville_parameters(&knot_v);
1931 let samples = parameters_u
1932 .iter()
1933 .map(|&u| {
1934 parameters_v
1935 .iter()
1936 .map(|&v| Ok(evaluator.at(u, v, -distance).or_refuse(KernelStage::Refine, "at")?.point))
1937 .collect::<Result<Vec<_>, KernelRefusal>>()
1938 })
1939 .collect::<Result<Vec<_>, KernelRefusal>>();
1940 let Ok(samples) = samples else {
1941 stop = "unevaluable sample";
1942 break;
1943 };
1944 let candidate = NurbsSurface::new(
1945 candidate_source.degree_u,
1946 candidate_source.degree_v,
1947 candidate_source.knots_u.clone(),
1948 candidate_source.knots_v.clone(),
1949 interpolate_tensor(&knot_u, &knot_v, ¶meters_u, ¶meters_v, &samples, &weights)?,
1950 ).or_refuse(KernelStage::Refine, "new")?;
1951 let Ok(deviation) = offset_fit_deviation(&candidate, evaluator, distance) else {
1952 stop = "unmeasurable round";
1953 break;
1954 };
1955 rounds += 1;
1956 if deviation.worst() >= best_deviation.worst() {
1957 let first_quadratic_split = rounds == 1
1964 && ((halve_u && source.degree_u == 2 && source.knots_u.len() == 6)
1965 || (halve_v && source.degree_v == 2 && source.knots_v.len() == 6));
1966 if first_quadratic_split {
1967 refined_source = candidate_source;
1968 continue;
1969 }
1970 if !residual_mode && best_deviation.worst() > offset_construction_band(scale)
1975 && std::env::var("BREP_OFFSET_FIT_RESIDUAL").as_deref() != Ok("0")
1976 {
1977 residual_mode = true;
1978 continue;
1979 }
1980 stop = "stopped improving";
1981 break;
1982 }
1983 best = candidate;
1984 best_deviation = deviation;
1985 refined_source = candidate_source;
1986 }
1987 } else if best_deviation.worst() > bar {
1988 stop = "refinement disabled";
1989 }
1990 let band = offset_construction_band(scale);
1995 let refused = enabled && best_deviation.worst() > band;
1996 if let Some(path) = trace {
1997 use std::io::Write;
1998 let caller = std::env::args().skip(1).collect::<Vec<_>>().join(" ");
1999 let thread = std::thread::current().name().unwrap_or("-").to_string();
2000 let line = format!(
2001 "fit deg=({},{}) net=({},{})->({},{}) d={distance:.6} scale={scale:.4} bar={bar:.3e} \
2002 initial=(u {:.3e}, v {:.3e}, uv {:.3e}) final={:.3e} rounds={rounds} stop={stop} \
2003 refused={refused} seconds={:.4} thread={thread} args={caller}\n",
2004 source.degree_u,
2005 source.degree_v,
2006 source.control_points.len(),
2007 source.control_points[0].len(),
2008 best.control_points.len(),
2009 best.control_points[0].len(),
2010 initial.along_u,
2011 initial.along_v,
2012 initial.between,
2013 best_deviation.worst(),
2014 started.elapsed().as_secs_f64(),
2015 );
2016 if let Ok(mut file) = std::fs::OpenOptions::new().create(true).append(true).open(path) {
2017 let _ = file.write_all(line.as_bytes());
2018 }
2019 }
2020 if refused {
2021 return Err(KernelRefusal::non_convergence(KernelStage::Refine, "offset_carrier_fit", format!(
2022 "offset_surface: the fitted offset carrier stays {:.3e} off its pointwise offset at \
2023 distance {distance} after {rounds} refinement round(s) ({stop}), past the offset \
2024 construction band {band:.3e}",
2025 best_deviation.worst()
2026 )));
2027 }
2028 Ok(best)
2029}
2030
2031pub(crate) fn unify_offset_sheet_bases(
2041 first: &NurbsSurface,
2042 second: &NurbsSurface,
2043) -> Result<(NurbsSurface, NurbsSurface), KernelRefusal> {
2044 fn raise_to(surface: &NurbsSurface, degree_u: usize, degree_v: usize) -> Result<NurbsSurface, KernelRefusal> {
2045 let raise_u = surface.degree_u == 1 && degree_u == 3;
2046 let raise_v = surface.degree_v == 1 && degree_v == 3;
2047 if !raise_u && !raise_v {
2048 return Ok(surface.clone());
2049 }
2050 raise_linear_directions(surface, raise_u, raise_v)?
2051 .ok_or_else(|| {
2052 KernelRefusal::internal(
2053 KernelStage::Refine,
2054 "offset_raise_linear",
2055 "offset sheets: a linear direction could not be raised",
2056 )
2057 })
2058 }
2059 let degree_u = first.degree_u.max(second.degree_u);
2060 let degree_v = first.degree_v.max(second.degree_v);
2061 let mut first = raise_to(first, degree_u, degree_v)?;
2062 let mut second = raise_to(second, degree_u, degree_v)?;
2063 if first.degree_u != second.degree_u || first.degree_v != second.degree_v {
2064 return Err(KernelRefusal::internal(KernelStage::Refine, "offset_common_basis", format!(
2065 "offset sheets: degrees ({},{}) and ({},{}) have no common basis",
2066 first.degree_u, first.degree_v, second.degree_u, second.degree_v
2067 )));
2068 }
2069 fn missing(from: &[f64], into: &[f64]) -> Vec<f64> {
2071 let mut extra = Vec::new();
2072 let mut index = 0usize;
2073 for &knot in from {
2074 while index < into.len() && into[index] < knot - 1e-12 {
2075 index += 1;
2076 }
2077 if index < into.len() && (into[index] - knot).abs() <= 1e-12 {
2078 index += 1;
2079 } else {
2080 extra.push(knot);
2081 }
2082 }
2083 extra
2084 }
2085 fn insert(surface: &NurbsSurface, knots: &[f64], along_u: bool) -> Result<NurbsSurface, KernelRefusal> {
2086 if knots.is_empty() {
2087 return Ok(surface.clone());
2088 }
2089 let (degree, vector) = if along_u {
2090 (surface.degree_u, &surface.knots_u)
2091 } else {
2092 (surface.degree_v, &surface.knots_v)
2093 };
2094 let lines = if along_u {
2095 (0..surface.control_points[0].len())
2096 .map(|column| surface.control_points.iter().map(|row| row[column]).collect::<Vec<_>>())
2097 .collect::<Vec<_>>()
2098 } else {
2099 surface.control_points.clone()
2100 };
2101 let mut refined_knots = vector.clone();
2102 let mut refined = Vec::with_capacity(lines.len());
2103 for line in lines {
2104 let mut curve = NurbsCurve::new(degree, vector.clone(), line).or_refuse(KernelStage::Refine, "new")?;
2105 for &knot in knots {
2106 curve = curve.insert_knot(knot, 1).or_refuse(KernelStage::Refine, "insert_knot")?;
2107 }
2108 refined_knots = curve.knots.clone();
2109 refined.push(curve.control_points);
2110 }
2111 let net = if along_u {
2112 (0..refined[0].len())
2113 .map(|row| refined.iter().map(|column| column[row]).collect())
2114 .collect()
2115 } else {
2116 refined
2117 };
2118 let (knots_u, knots_v) = if along_u {
2119 (refined_knots, surface.knots_v.clone())
2120 } else {
2121 (surface.knots_u.clone(), refined_knots)
2122 };
2123 NurbsSurface::new(surface.degree_u, surface.degree_v, knots_u, knots_v, net).or_refuse(KernelStage::Refine, "new")
2124 }
2125 let (to_first_u, to_second_u) = (
2126 missing(&second.knots_u, &first.knots_u),
2127 missing(&first.knots_u, &second.knots_u),
2128 );
2129 first = insert(&first, &to_first_u, true)?;
2130 second = insert(&second, &to_second_u, true)?;
2131 let (to_first_v, to_second_v) = (
2132 missing(&second.knots_v, &first.knots_v),
2133 missing(&first.knots_v, &second.knots_v),
2134 );
2135 first = insert(&first, &to_first_v, false)?;
2136 second = insert(&second, &to_second_v, false)?;
2137 Ok((first, second))
2138}
2139
2140fn carrier_edge_curve(
2162 surface: &NurbsSurface,
2163 pcurve: &NurbsCurve,
2164 points: &[Vec3],
2165 parameters: &[f64],
2166 fit_tolerance: f64,
2167) -> Result<(NurbsCurve, f64, f64), KernelRefusal> {
2168 if affine_image_trusted() && surface.is_affine().or_refuse(KernelStage::Refine, "is_affine")? {
2176 let [q0, q1] = pcurve.domain().or_refuse(KernelStage::Refine, "domain")?;
2177 if (q1 - q0).is_finite() && (q1 - q0).abs() > 0.0 {
2178 let curve = crate::image_curve::affine_image_curve(surface, pcurve).or_refuse(KernelStage::Refine, "affine_image_curve")?;
2179 return Ok((curve, q0, q1));
2180 }
2181 }
2182 let image = if env_read!("BREP_CARRIER_POLYLINE").as_deref() == Some("1") {
2183 Err("BREP_CARRIER_POLYLINE set".to_string())
2184 } else {
2185 crate::image_curve::image_curve(surface, pcurve, fit_tolerance, "offset_face_carrier")
2186 };
2187 match image {
2188 Ok(image) if image.t0 <= image.t1 => Ok((image.curve, image.t0, image.t1)),
2189 Ok(image) => {
2190 let [start, end] = image.curve.domain().or_refuse(KernelStage::Refine, "domain")?;
2193 let curve = image.curve.reversed().or_refuse(KernelStage::Refine, "reversed")?;
2194 Ok((curve, start + end - image.t0, start + end - image.t1))
2195 }
2196 Err(_) => {
2197 let curve = interpolate_curve(points, 1, parameters).or_refuse(KernelStage::Refine, "interpolate_curve")?;
2198 let [t0, t1] = curve.domain().or_refuse(KernelStage::Refine, "domain")?;
2199 Ok((curve, t0, t1))
2200 }
2201 }
2202}
2203
2204fn affine_image_trusted() -> bool {
2205 switch!("BREP_CARRIER_AFFINE_IMAGE", true)
2206}
2207
2208
2209fn lazy_carrier_trace(
2217 surface: &NurbsSurface,
2218 pcurve: &NurbsCurve,
2219 collapse_tolerance: f64,
2220 fit_tolerance: f64,
2221) -> Option<(Vec<Vec3>, (NurbsCurve, f64, f64))> {
2222 if !switch!("BREP_OFFSET_LAZY_CARRIER_TRACE", true) {
2223 return None;
2224 }
2225 if pcurve.degree == 1 && pcurve.control_points.len() == 2 {
2228 let (a, b) = (pcurve.control_points[0], pcurve.control_points[1]);
2229 if (surface.degree_u <= 1 && surface.degree_v <= 1)
2230 || (a.w == b.w && a.y == b.y && surface.degree_u <= 1)
2231 || (a.w == b.w && a.x == b.x && surface.degree_v <= 1)
2232 {
2233 return None;
2234 }
2235 }
2236 fn knots_ok(knots: &[f64], degree: usize, count: usize) -> bool {
2237 degree <= 3 && count <= 64
2238 && KnotVector::new(knots.to_vec(), degree).ok().is_some_and(|k| k.control_point_count() == count)
2239 && knots.iter().all(|k| k.is_finite() && k.abs() <= 1e6)
2240 && knots.windows(2).all(|k| k[0] == k[1] || k[1] - k[0] >= 1e-6)
2241 }
2242 let bounded = |p: &Vec4| (1e-3..=1e3).contains(&p.w)
2243 && [p.x, p.y, p.z].iter().all(|x| x.is_finite() && x.abs() <= 1e12);
2244 let columns = surface.control_points.first()?.len();
2245 if !knots_ok(&surface.knots_u, surface.degree_u, surface.control_points.len())
2246 || !knots_ok(&surface.knots_v, surface.degree_v, columns)
2247 || surface.control_points.iter().any(|row| row.len() != columns || row.iter().any(|p| !bounded(p)))
2248 || !knots_ok(&pcurve.knots, pcurve.degree, pcurve.control_points.len())
2249 || pcurve.control_points.iter().any(|p| !bounded(p))
2250 {
2251 return None;
2252 }
2253 let ([u0, u1], [v0, v1]) = (surface.domain_u().ok()?, surface.domain_v().ok()?);
2254 let uv = pcurve.control_points.iter().map(|p| p.point().ok()).collect::<Option<Vec<_>>>()?;
2255 if uv.iter().any(|p| p.x < u0 || p.x > u1 || p.y < v0 || p.y > v1) {
2256 return None;
2257 }
2258 if !surface.is_affine().ok()? && !uv.iter().all(|p| p.x == uv[0].x) && !uv.iter().all(|p| p.y == uv[0].y) {
2260 return None;
2261 }
2262 let [start, end] = pcurve.domain().ok()?;
2263 let evaluate = |fraction: f64| {
2264 let uv = pcurve.evaluate(start + (end - start) * fraction).ok()?;
2265 surface.evaluate(uv.x, uv.y).ok()
2266 };
2267 let first = evaluate(0.0)?;
2268 let last = evaluate(1.0)?;
2269 let samples = [evaluate(0.25)?, evaluate(0.5)?, evaluate(0.75)?];
2270 let deviation = samples.iter().enumerate().map(|(index, point)| {
2271 point.sub(first.add(last.sub(first).scale((index + 1) as f64 * 0.25))).length()
2272 }).fold(0.0, f64::max);
2273 if last.sub(first).length() <= collapse_tolerance
2274 && !(deviation > 5e-4 && samples[1].sub(first).length() > collapse_tolerance)
2275 {
2276 return None;
2277 }
2278 let image = carrier_edge_curve(surface, pcurve, &[], &[], fit_tolerance).ok()?;
2281 Some((vec![first, samples[1], last], image))
2282}
2283
2284fn mapped_pcurve_polyline(
2285 surface: &NurbsSurface,
2286 pcurve: &NurbsCurve,
2287 degenerate: bool,
2288) -> Result<(Vec<Vec3>, Vec<f64>), KernelRefusal> {
2289 let [start, end] = pcurve.domain().or_refuse(KernelStage::Refine, "domain")?;
2290 let evaluate = |fraction: f64| -> Result<Vec3, KernelRefusal> {
2291 let uv = pcurve.evaluate(start + (end - start) * fraction).or_refuse(KernelStage::Refine, "evaluate")?;
2292 surface.evaluate(uv.x, uv.y).or_refuse(KernelStage::Refine, "evaluate")
2293 };
2294 let first = evaluate(0.0)?;
2295 let last = evaluate(1.0)?;
2296 if degenerate {
2297 return Ok((vec![first, last], vec![0.0, 1.0]));
2298 }
2299 fn append(
2300 evaluate: &impl Fn(f64) -> Result<Vec3, KernelRefusal>,
2301 a_fraction: f64,
2302 a: Vec3,
2303 b_fraction: f64,
2304 b: Vec3,
2305 depth: usize,
2306 parameters: &mut Vec<f64>,
2307 points: &mut Vec<Vec3>,
2308 ) -> Result<(), KernelRefusal> {
2309 let fractions =
2310 [0.25, 0.5, 0.75].map(|local| a_fraction + (b_fraction - a_fraction) * local);
2311 let samples = fractions
2312 .map(evaluate)
2313 .into_iter()
2314 .collect::<Result<Vec<_>, KernelRefusal>>()?;
2315 let deviation = samples
2316 .iter()
2317 .enumerate()
2318 .map(|(index, point)| {
2319 point
2320 .sub(a.add(b.sub(a).scale((index + 1) as f64 * 0.25)))
2321 .length()
2322 })
2323 .fold(0.0, f64::max);
2324 if deviation <= 5e-4 || depth >= 10 {
2325 parameters.push(b_fraction);
2326 points.push(b);
2327 return Ok(());
2328 }
2329 append(
2330 evaluate,
2331 a_fraction,
2332 a,
2333 fractions[1],
2334 samples[1],
2335 depth + 1,
2336 parameters,
2337 points,
2338 )?;
2339 append(
2340 evaluate,
2341 fractions[1],
2342 samples[1],
2343 b_fraction,
2344 b,
2345 depth + 1,
2346 parameters,
2347 points,
2348 )
2349 }
2350 let mut parameters = vec![0.0];
2351 let mut points = vec![first];
2352 append(
2353 &evaluate,
2354 0.0,
2355 first,
2356 1.0,
2357 last,
2358 0,
2359 &mut parameters,
2360 &mut points,
2361 )?;
2362 Ok((points, parameters))
2363}
2364
2365#[derive(Clone, Debug)]
2379pub struct CarrierDeviation {
2380 pub band: f64,
2383 pub lane: OffsetSurfaceLane,
2385 pub surface: Option<MeasuredTolerance>,
2387 pub edges: Vec<(u64, MeasuredTolerance)>,
2392 pub vertices: Vec<(u64, f64)>,
2396}
2397
2398impl CarrierDeviation {
2399 pub fn worst(&self) -> Option<MeasuredTolerance> {
2402 MeasuredTolerance::worst(
2403 self.surface
2404 .into_iter()
2405 .chain(self.edges.iter().map(|(_, measured)| *measured))
2406 .chain(
2407 self.vertices
2408 .iter()
2409 .map(|(_, gap)| MeasuredTolerance::new(*gap, self.band)),
2410 ),
2411 )
2412 }
2413
2414 pub fn exceedances(&self) -> Vec<String> {
2417 let mut out = Vec::new();
2418 if let Some(surface) = self.surface {
2419 if surface.exceeds_band() {
2420 out.push(format!("carrier surface fit {}", surface.describe()));
2421 }
2422 }
2423 for (id, measured) in &self.edges {
2424 if measured.exceeds_band() {
2425 out.push(format!("edge {id} {}", measured.describe()));
2426 }
2427 }
2428 for (id, gap) in &self.vertices {
2429 let measured = MeasuredTolerance::new(*gap, self.band);
2430 if measured.exceeds_band() {
2431 out.push(format!("vertex {id} {}", measured.describe()));
2432 }
2433 }
2434 out
2435 }
2436}
2437
2438#[derive(Clone, Debug, Serialize)]
2439pub struct OffsetFaceCarrier {
2440 pub vertices: Vec<VertexRecord>,
2441 pub edges: Vec<EdgeRecord>,
2442 pub face: FaceRecord,
2443 #[serde(skip)]
2451 pub deviation: Option<CarrierDeviation>,
2452 #[serde(skip)]
2456 pub lane: OffsetSurfaceLane,
2457}
2458
2459fn claim_vertex_image(
2470 source_id: u64,
2471 point: Vec3,
2472 split: f64,
2473 vertex_images: &mut HashMap<u64, Vec<u64>>,
2474 vertices: &mut Vec<VertexRecord>,
2475 next_id: &mut u64,
2476) -> u64 {
2477 if let Some(ids) = vertex_images.get(&source_id) {
2478 let near = ids.iter().copied().find(|id| {
2479 vertices
2480 .iter()
2481 .find(|vertex| vertex.id == *id)
2482 .is_some_and(|vertex| vertex.point.sub(point).length() <= split)
2483 });
2484 if let Some(id) = near {
2485 return id;
2486 }
2487 }
2488 let id = *next_id;
2489 *next_id += 1;
2490 vertices.push(VertexRecord { id, point });
2491 vertex_images.entry(source_id).or_default().push(id);
2492 id
2493}
2494
2495pub fn offset_face_carrier(
2496 solid: &BrepSolid,
2497 face_id: u64,
2498 distance: f64,
2499 planar_extension: f64,
2500) -> Result<OffsetFaceCarrier, KernelRefusal> {
2501 offset_face_carrier_impl(
2502 solid,
2503 face_id,
2504 distance,
2505 &CarrierExtension::uniform(planar_extension),
2506 false,
2507 )
2508}
2509
2510pub fn offset_face_carrier_sided(
2512 solid: &BrepSolid,
2513 face_id: u64,
2514 distance: f64,
2515 extension: &CarrierExtension,
2516) -> Result<OffsetFaceCarrier, KernelRefusal> {
2517 offset_face_carrier_impl(solid, face_id, distance, extension, false)
2518}
2519
2520pub fn offset_face_carrier_measured(
2540 solid: &BrepSolid,
2541 face_id: u64,
2542 distance: f64,
2543 planar_extension: f64,
2544) -> Result<OffsetFaceCarrier, KernelRefusal> {
2545 offset_face_carrier_impl(
2546 solid,
2547 face_id,
2548 distance,
2549 &CarrierExtension::uniform(planar_extension),
2550 true,
2551 )
2552}
2553
2554fn measure_carrier(
2565 surface: &NurbsSurface,
2566 vertices: &[VertexRecord],
2567 edges: &[EdgeRecord],
2568 loops: &[LoopRecord],
2569 lane: OffsetSurfaceLane,
2570 surface_fit: Option<MeasuredTolerance>,
2571 band: f64,
2572) -> Result<CarrierDeviation, KernelRefusal> {
2573 let edge_by_id: HashMap<u64, &EdgeRecord> = edges.iter().map(|edge| (edge.id, edge)).collect();
2574 let mut per_edge: HashMap<u64, MeasuredTolerance> = HashMap::default();
2578 for loop_record in loops {
2579 for coedge in &loop_record.coedges {
2580 let Some(edge) = edge_by_id.get(&coedge.edge_id) else {
2581 continue;
2582 };
2583 let measured = measure_edge_against_pcurve_image(
2584 surface,
2585 &coedge.pcurve,
2586 edge,
2587 coedge.forward,
2588 band,
2589 )?;
2590 per_edge
2591 .entry(edge.id)
2592 .and_modify(|existing| *existing = existing.worse_of(measured))
2593 .or_insert(measured);
2594 }
2595 }
2596
2597 let mut measured_edges: Vec<(u64, MeasuredTolerance)> = per_edge.into_iter().collect();
2598 measured_edges.sort_by_key(|(id, _)| *id);
2599 let deviation_of: HashMap<u64, f64> = measured_edges
2600 .iter()
2601 .map(|(id, measured)| (*id, measured.deviation()))
2602 .collect();
2603
2604 let mut ends_at: HashMap<u64, Vec<(&EdgeRecord, f64)>> = HashMap::default();
2608 for edge in edges {
2609 ends_at
2610 .entry(edge.start_vertex_id)
2611 .or_default()
2612 .push((edge, edge.t0));
2613 ends_at
2614 .entry(edge.end_vertex_id)
2615 .or_default()
2616 .push((edge, edge.t1));
2617 }
2618
2619 let mut measured_vertices = Vec::with_capacity(vertices.len());
2620 for vertex in vertices {
2621 let mut gaps = Vec::new();
2622 let mut incident = Vec::new();
2623 for (edge, parameter) in ends_at.get(&vertex.id).into_iter().flatten() {
2624 gaps.push(vertex_endpoint_gap(
2625 vertex.point,
2626 edge.curve.evaluate(*parameter).or_refuse(KernelStage::Refine, "evaluate")?,
2627 ));
2628 incident.push(deviation_of.get(&edge.id).copied().unwrap_or(0.0));
2629 }
2630 measured_vertices.push((vertex.id, vertex_tolerance_from_edges(gaps, incident)));
2631 }
2632
2633 Ok(CarrierDeviation {
2634 band,
2635 lane,
2636 surface: surface_fit,
2637 edges: measured_edges,
2638 vertices: measured_vertices,
2639 })
2640}
2641
2642fn offset_face_carrier_impl(
2643 solid: &BrepSolid,
2644 face_id: u64,
2645 distance: f64,
2646 extension: &CarrierExtension,
2647 measure: bool,
2648) -> Result<OffsetFaceCarrier, KernelRefusal> {
2649 let source = solid
2650 .shells
2651 .iter()
2652 .flat_map(|shell| &shell.faces)
2653 .find(|face| face.id == face_id)
2654 .ok_or_else(|| format!("offset_face_carrier: missing face {face_id}")).or_refuse(KernelStage::Refine, "offset_missing_face")?;
2655 let band = offset_construction_band(solid_model_scale(solid));
2663 let fit_tolerance =
2670 crate::KernelTolerances::for_scale(solid_model_scale(solid), 1e-7).intersection_fit;
2671 let (surface, lane, surface_fit) = if measure {
2672 let measured = offset_surface_measured_sided(source, distance, extension, band)?;
2673 (measured.surface, measured.lane, measured.fit)
2674 } else {
2675 let (surface, lane) = offset_surface_with_lane(source, distance, extension)?;
2676 (surface, lane, None)
2677 };
2678 let source_edges = solid
2679 .edges
2680 .iter()
2681 .map(|edge| (edge.id, edge))
2682 .collect::<HashMap<_, _>>();
2683 let source_vertices = solid
2684 .vertices
2685 .iter()
2686 .map(|vertex| (vertex.id, vertex))
2687 .collect::<HashMap<_, _>>();
2688 let mut vertices = Vec::new();
2689 let mut vertex_images = HashMap::default();
2690 let vertex_split = (distance.abs() * 0.1).max(band);
2694 let mut edges = Vec::new();
2695 let mut edge_images = HashMap::default();
2696 let mut loops = Vec::new();
2697 let mut next_id = 1u64;
2698
2699 for source_loop in &source.loops {
2700 let mut coedges = Vec::new();
2701 for source_coedge in &source_loop.coedges {
2702 let source_edge = source_edges
2703 .get(&source_coedge.edge_id)
2704 .ok_or_else(|| "offset_face_carrier: missing source edge".to_string()).or_refuse(KernelStage::Refine, "offset_missing_edge")?;
2705 let (source_start, source_end) = if source_coedge.forward {
2706 (source_edge.start_vertex_id, source_edge.end_vertex_id)
2707 } else {
2708 (source_edge.end_vertex_id, source_edge.start_vertex_id)
2709 };
2710 if !source_vertices.contains_key(&source_start)
2711 || !source_vertices.contains_key(&source_end)
2712 {
2713 return Err(KernelRefusal::internal(KernelStage::Refine, "offset_missing_vertex", "offset_face_carrier: missing source vertex"));
2714 }
2715 let collapse_tolerance = 1e-6f64.max(distance.abs() * 1e-2);
2733 let (points, parameters, exact_image) = match lazy_carrier_trace(
2734 &surface, &source_coedge.pcurve, collapse_tolerance, fit_tolerance,
2735 ) {
2736 Some((points, image)) => (points, Vec::new(), Some(image)),
2737 None => {
2738 let (points, parameters) = mapped_pcurve_polyline(&surface, &source_coedge.pcurve, false)?;
2739 (points, parameters, None)
2740 }
2741 };
2742 let image_collapsed = points
2743 .iter()
2744 .all(|point| point.sub(points[0]).length() <= collapse_tolerance);
2745 let (edge_id, forward) =
2746 if let Some((edge_id, edge_start_vertex_id, creator_forward)) =
2747 edge_images.get(&source_edge.id)
2748 {
2749 if source_start == source_end {
2750 (*edge_id, source_coedge.forward == *creator_forward)
2773 } else {
2774 (
2777 *edge_id,
2778 vertex_images
2779 .get(&source_start)
2780 .is_some_and(|ids: &Vec<u64>| ids.contains(edge_start_vertex_id)),
2781 )
2782 }
2783 } else {
2784 let (curve, t0, t1) = if image_collapsed {
2785 let curve = NurbsCurve::new(
2786 1,
2787 vec![0.0, 0.0, 1.0, 1.0],
2788 vec![
2789 Vec4::from_point(points[0], 1.0),
2790 Vec4::from_point(points[0], 1.0),
2791 ],
2792 ).or_refuse(KernelStage::Refine, "new")?;
2793 (curve, 0.0, 1.0)
2794 } else if let Some(image) = exact_image {
2795 image
2796 } else {
2797 carrier_edge_curve(
2798 &surface,
2799 &source_coedge.pcurve,
2800 &points,
2801 ¶meters,
2802 fit_tolerance,
2803 )?
2804 };
2805 let start_vertex_id = claim_vertex_image(
2806 source_start,
2807 points[0],
2808 vertex_split,
2809 &mut vertex_images,
2810 &mut vertices,
2811 &mut next_id,
2812 );
2813 let end_vertex_id = claim_vertex_image(
2814 source_end,
2815 points[points.len() - 1],
2816 vertex_split,
2817 &mut vertex_images,
2818 &mut vertices,
2819 &mut next_id,
2820 );
2821 let id = next_id;
2822 next_id += 1;
2823 edges.push(EdgeRecord {
2824 id,
2825 curve,
2826 t0,
2827 t1,
2828 start_vertex_id,
2829 end_vertex_id,
2830 degenerate: image_collapsed,
2844 name: source_edge
2848 .name
2849 .as_ref()
2850 .map(|name| format!("{name}_Offset")),
2851 });
2852 edge_images.insert(
2853 source_edge.id,
2854 (id, start_vertex_id, source_coedge.forward),
2855 );
2856 (id, true)
2857 };
2858 let id = next_id;
2859 next_id += 1;
2860 coedges.push(CoedgeRecord {
2861 id,
2862 edge_id,
2863 forward,
2864 pcurve: source_coedge.pcurve.clone(),
2865 });
2866 }
2867 let id = next_id;
2868 next_id += 1;
2869 loops.push(LoopRecord { id, coedges });
2870 }
2871 let deviation = if measure {
2872 Some(measure_carrier(
2873 &surface,
2874 &vertices,
2875 &edges,
2876 &loops,
2877 lane,
2878 surface_fit,
2879 band,
2880 )?)
2881 } else {
2882 None
2883 };
2884 Ok(OffsetFaceCarrier {
2885 vertices,
2886 edges,
2887 face: FaceRecord {
2888 id: next_id,
2889 surface,
2890 same_sense: source.same_sense,
2891 loops,
2892 name: source.name.as_ref().map(|name| format!("{name}_Offset")),
2893 },
2894 deviation,
2895 lane,
2896 })
2897}
2898
2899
2900
2901
2902