1#![forbid(unsafe_code)]
2
3use std::collections::HashMap;
11use std::fmt;
12
13use axiolid_core::Interval;
14use axiolid_curve::{Curve2, Curve3};
15use axiolid_surface::Surface;
16use axiolid_topology::{audit_brep, BRep, BRepHealth, EdgeId, FaceId, LoopId};
17
18pub mod name;
20
21pub use name::{EdgeName, FaceName, Operand, SweptFace};
22
23macro_rules! geometry_id {
24 ($name:ident, $label:literal) => {
25 #[doc = concat!("Typed handle into the exact B-rep ", $label, " catalog.")]
26 #[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
27 pub struct $name(u32);
28
29 impl $name {
30 fn from_index(index: usize) -> Self {
31 Self(u32::try_from(index).expect("exact B-rep catalog exceeds u32 capacity"))
32 }
33
34 pub const fn index(self) -> usize {
36 self.0 as usize
37 }
38 }
39 };
40}
41
42geometry_id!(Curve3Id, "3D curve");
43geometry_id!(Curve2Id, "2D pcurve");
44geometry_id!(SurfaceId, "surface");
45
46pub type ExactTopology = BRep<Curve3Id, Curve2Id, SurfaceId>;
48
49#[derive(Debug, Clone, PartialEq)]
57pub struct ExactBRep {
58 topology: ExactTopology,
59 curves3: Vec<Curve3>,
60 curves2: Vec<Curve2>,
61 surfaces: Vec<Surface>,
62 edge_intervals: HashMap<EdgeId, Interval>,
63 pcurve_intervals: HashMap<(LoopId, usize), Interval>,
64 face_names: HashMap<FaceId, FaceName>,
65 edge_names: HashMap<EdgeId, EdgeName>,
66}
67
68impl ExactBRep {
69 pub fn face_name(&self, face: FaceId) -> Option<&FaceName> {
75 self.face_names.get(&face)
76 }
77
78 pub fn edge_name(&self, edge: EdgeId) -> Option<&EdgeName> {
80 self.edge_names.get(&edge)
81 }
82
83 pub fn face_by_name(&self, name: &FaceName) -> Option<FaceId> {
89 self.face_names
90 .iter()
91 .find(|(_, candidate)| *candidate == name)
92 .map(|(id, _)| *id)
93 }
94
95 pub fn edge_by_name(&self, name: &EdgeName) -> Option<EdgeId> {
97 self.edge_names
98 .iter()
99 .find(|(_, candidate)| *candidate == name)
100 .map(|(id, _)| *id)
101 }
102 pub fn name_caps(&mut self, start: Option<FaceName>, end: Option<FaceName>) {
109 let start_id = self.topology.face_id_at(0);
110 let end_id = self.topology.face_id_at(1);
111 if let (Some(name), Some(id)) = (start, start_id) {
112 self.face_names.insert(id, name);
113 }
114 if let (Some(name), Some(id)) = (end, end_id) {
115 self.face_names.insert(id, name);
116 }
117 }
118
119 pub fn topology(&self) -> &ExactTopology {
121 &self.topology
122 }
123
124 pub fn curves3(&self) -> &[Curve3] {
126 &self.curves3
127 }
128
129 pub fn curves2(&self) -> &[Curve2] {
131 &self.curves2
132 }
133
134 pub fn surfaces(&self) -> &[Surface] {
136 &self.surfaces
137 }
138
139 pub fn edge_interval(&self, edge: EdgeId) -> Option<Interval> {
141 self.edge_intervals.get(&edge).copied()
142 }
143
144 pub fn pcurve_interval(&self, loop_id: LoopId, use_index: usize) -> Option<Interval> {
146 self.pcurve_intervals.get(&(loop_id, use_index)).copied()
147 }
148}
149
150#[derive(Debug, Clone, Default, PartialEq)]
152pub struct ExactBRepBuilder {
153 topology: ExactTopology,
154 curves3: Vec<Curve3>,
155 curves2: Vec<Curve2>,
156 surfaces: Vec<Surface>,
157 edge_intervals: HashMap<EdgeId, Interval>,
158 pcurve_intervals: HashMap<(LoopId, usize), Interval>,
159 face_names: HashMap<FaceId, FaceName>,
160 edge_names: HashMap<EdgeId, EdgeName>,
161}
162
163impl ExactBRepBuilder {
164 pub fn set_face_name(&mut self, face: FaceId, name: FaceName) {
170 self.face_names.insert(face, name);
171 }
172
173 pub fn set_edge_name(&mut self, edge: EdgeId, name: EdgeName) {
175 self.edge_names.insert(edge, name);
176 }
177 pub fn try_reserve(
179 &mut self,
180 curves3: usize,
181 curves2: usize,
182 surfaces: usize,
183 edge_intervals: usize,
184 pcurve_intervals: usize,
185 ) -> Result<(), std::collections::TryReserveError> {
186 self.curves3.try_reserve_exact(curves3)?;
187 self.curves2.try_reserve_exact(curves2)?;
188 self.surfaces.try_reserve_exact(surfaces)?;
189 self.edge_intervals.try_reserve(edge_intervals)?;
190 self.pcurve_intervals.try_reserve(pcurve_intervals)?;
191 Ok(())
192 }
193
194 pub fn add_curve3(&mut self, curve: Curve3) -> Curve3Id {
196 let id = Curve3Id::from_index(self.curves3.len());
197 self.curves3.push(curve);
198 id
199 }
200
201 pub fn add_curve2(&mut self, curve: Curve2) -> Curve2Id {
203 let id = Curve2Id::from_index(self.curves2.len());
204 self.curves2.push(curve);
205 id
206 }
207
208 pub fn add_surface(&mut self, surface: Surface) -> SurfaceId {
210 let id = SurfaceId::from_index(self.surfaces.len());
211 self.surfaces.push(surface);
212 id
213 }
214
215 pub fn topology_mut(&mut self) -> &mut ExactTopology {
217 &mut self.topology
218 }
219
220 pub fn set_edge_interval(&mut self, edge: EdgeId, interval: Interval) {
222 self.edge_intervals.insert(edge, interval);
223 }
224
225 pub fn set_pcurve_interval(&mut self, loop_id: LoopId, use_index: usize, interval: Interval) {
227 self.pcurve_intervals.insert((loop_id, use_index), interval);
228 }
229
230 pub fn finish(self) -> Result<ExactBRep, ExactBRepError> {
232 validate(&self)?;
233 Ok(ExactBRep {
234 topology: self.topology,
235 curves3: self.curves3,
236 curves2: self.curves2,
237 surfaces: self.surfaces,
238 edge_intervals: self.edge_intervals,
239 pcurve_intervals: self.pcurve_intervals,
240 face_names: self.face_names,
241 edge_names: self.edge_names,
242 })
243 }
244}
245
246#[non_exhaustive]
248#[derive(Debug, Clone, PartialEq, Eq)]
249pub enum ExactBRepError {
250 Empty,
252 Topology(BRepHealth),
254 MissingEdgeCurve { edge_index: usize },
256 UnknownCurve3 { edge_index: usize },
258 MissingEdgeInterval { edge_index: usize },
260 InvalidEdgeInterval { edge_index: usize },
262 MissingPcurve { loop_index: usize, use_index: usize },
264 UnknownCurve2 { loop_index: usize, use_index: usize },
266 MissingPcurveInterval { loop_index: usize, use_index: usize },
268 InvalidPcurveInterval { loop_index: usize, use_index: usize },
270 MissingFaceSurface { face_index: usize },
272 UnknownSurface { face_index: usize },
274}
275
276impl fmt::Display for ExactBRepError {
277 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
278 match self {
279 Self::Empty => f.write_str("exact B-rep requires at least one face"),
280 Self::Topology(_) => f.write_str("exact B-rep topology is structurally invalid"),
281 Self::MissingEdgeCurve { edge_index } => {
282 write!(f, "edge {edge_index} lacks a 3D curve")
283 }
284 Self::UnknownCurve3 { edge_index } => {
285 write!(f, "edge {edge_index} has an unknown 3D curve")
286 }
287 Self::MissingEdgeInterval { edge_index } => {
288 write!(f, "edge {edge_index} lacks a curve interval")
289 }
290 Self::InvalidEdgeInterval { edge_index } => {
291 write!(f, "edge {edge_index} has an invalid curve interval")
292 }
293 Self::MissingPcurve {
294 loop_index,
295 use_index,
296 } => write!(f, "loop {loop_index} use {use_index} lacks a pcurve"),
297 Self::UnknownCurve2 {
298 loop_index,
299 use_index,
300 } => write!(f, "loop {loop_index} use {use_index} has an unknown pcurve"),
301 Self::MissingPcurveInterval {
302 loop_index,
303 use_index,
304 } => write!(
305 f,
306 "loop {loop_index} use {use_index} lacks a pcurve interval"
307 ),
308 Self::InvalidPcurveInterval {
309 loop_index,
310 use_index,
311 } => write!(
312 f,
313 "loop {loop_index} use {use_index} has an invalid pcurve interval"
314 ),
315 Self::MissingFaceSurface { face_index } => {
316 write!(f, "face {face_index} lacks a support surface")
317 }
318 Self::UnknownSurface { face_index } => {
319 write!(f, "face {face_index} has an unknown support surface")
320 }
321 }
322 }
323}
324
325impl std::error::Error for ExactBRepError {}
326
327fn validate(value: &ExactBRepBuilder) -> Result<(), ExactBRepError> {
328 if value.topology.faces().is_empty() {
329 return Err(ExactBRepError::Empty);
330 }
331 let health = audit_brep(&value.topology);
332 if !health.is_tessellable() {
333 return Err(ExactBRepError::Topology(health));
334 }
335 for (edge_index, edge) in value.topology.edges().iter().enumerate() {
336 let Some(curve) = edge.curve else {
337 return Err(ExactBRepError::MissingEdgeCurve { edge_index });
338 };
339 if curve.index() >= value.curves3.len() {
340 return Err(ExactBRepError::UnknownCurve3 { edge_index });
341 }
342 let Some(edge_id) = value.topology.edge_id_at(edge_index) else {
343 return Err(ExactBRepError::MissingEdgeInterval { edge_index });
344 };
345 let Some(interval) = value.edge_intervals.get(&edge_id) else {
346 return Err(ExactBRepError::MissingEdgeInterval { edge_index });
347 };
348 if !valid_interval(*interval) {
349 return Err(ExactBRepError::InvalidEdgeInterval { edge_index });
350 }
351 }
352 for (loop_index, loop_) in value.topology.loops().iter().enumerate() {
353 let Some(loop_id) = value.topology.loop_id_at(loop_index) else {
354 return Err(ExactBRepError::Topology(health));
355 };
356 for (use_index, use_) in loop_.edges.iter().enumerate() {
357 let Some(curve) = use_.pcurve else {
358 return Err(ExactBRepError::MissingPcurve {
359 loop_index,
360 use_index,
361 });
362 };
363 if curve.index() >= value.curves2.len() {
364 return Err(ExactBRepError::UnknownCurve2 {
365 loop_index,
366 use_index,
367 });
368 }
369 let Some(interval) = value.pcurve_intervals.get(&(loop_id, use_index)) else {
370 return Err(ExactBRepError::MissingPcurveInterval {
371 loop_index,
372 use_index,
373 });
374 };
375 if !valid_interval(*interval) {
376 return Err(ExactBRepError::InvalidPcurveInterval {
377 loop_index,
378 use_index,
379 });
380 }
381 }
382 }
383 for (face_index, face) in value.topology.faces().iter().enumerate() {
384 let Some(surface) = face.surface else {
385 return Err(ExactBRepError::MissingFaceSurface { face_index });
386 };
387 if surface.index() >= value.surfaces.len() {
388 return Err(ExactBRepError::UnknownSurface { face_index });
389 }
390 }
391 Ok(())
392}
393
394fn valid_interval(interval: Interval) -> bool {
395 interval.start.is_finite() && interval.end.is_finite() && interval.length() > 0.0
396}