1use super::*;
2
3#[wasm_bindgen]
4pub struct WasmNurbsCurve {
5 pub(crate) curve: NurbsCurve,
6}
7
8#[wasm_bindgen]
9pub struct WasmNurbsSurface {
10 pub(crate) surface: NurbsSurface,
11}
12
13#[wasm_bindgen]
17pub struct WasmSolidBuffer {
18 pub(crate) data: Vec<f64>,
19 pub(crate) names_json: String,
20 pub(crate) metadata_json: String,
21}
22
23#[wasm_bindgen]
24impl WasmSolidBuffer {
25 pub fn take_data(&mut self) -> Vec<f64> {
26 std::mem::take(&mut self.data)
27 }
28
29 pub fn names_json(&self) -> String {
30 self.names_json.clone()
31 }
32
33 pub fn metadata_json(&self) -> String {
36 self.metadata_json.clone()
37 }
38}
39
40pub(crate) fn solid_buffer(solid: &BrepSolid, metadata_json: String) -> Result<WasmSolidBuffer, JsValue> {
41 let (data, names) = encode_solid(solid).map_err(javascript_error)?;
42 Ok(WasmSolidBuffer {
43 data,
44 names_json: serde_json::to_string(&names)
45 .map_err(|error| javascript_error(error.to_string()))?,
46 metadata_json,
47 })
48}
49
50#[wasm_bindgen]
52pub struct WasmMeshBuffers {
53 pub(crate) positions: Vec<f64>,
54 pub(crate) normals: Vec<f64>,
55 pub(crate) indices: Vec<u32>,
56 pub(crate) face_ids: Vec<u32>,
57}
58
59#[wasm_bindgen]
60impl WasmMeshBuffers {
61 pub fn take_positions(&mut self) -> Vec<f64> {
62 std::mem::take(&mut self.positions)
63 }
64
65 pub fn take_normals(&mut self) -> Vec<f64> {
66 std::mem::take(&mut self.normals)
67 }
68
69 pub fn take_indices(&mut self) -> Vec<u32> {
70 std::mem::take(&mut self.indices)
71 }
72
73 pub fn take_face_ids(&mut self) -> Vec<u32> {
74 std::mem::take(&mut self.face_ids)
75 }
76}
77
78thread_local! {
95 static SOLID_REGISTRY: std::cell::RefCell<SolidRegistry> =
96 std::cell::RefCell::new(SolidRegistry::new());
97}
98
99pub(crate) struct SolidRegistry {
100 pub(crate) next: u32,
101 pub(crate) solids: std::collections::HashMap<u32, BrepSolid>,
102}
103
104impl SolidRegistry {
105 fn new() -> Self {
106 Self {
107 next: 1,
108 solids: std::collections::HashMap::new(),
109 }
110 }
111 fn insert(&mut self, solid: BrepSolid) -> u32 {
112 let handle = self.next;
113 self.next = self.next.wrapping_add(1).max(1);
114 self.solids.insert(handle, solid);
115 handle
116 }
117}
118
119pub(crate) fn register_solid_value(solid: BrepSolid) -> u32 {
120 SOLID_REGISTRY.with(|registry| registry.borrow_mut().insert(solid))
121}
122
123pub(crate) fn with_registered_solid<T>(
125 handle: u32,
126 f: impl FnOnce(&BrepSolid) -> Result<T, JsValue>,
127) -> Result<T, JsValue> {
128 SOLID_REGISTRY.with(|registry| {
129 let registry = registry.borrow();
130 let solid = registry
131 .solids
132 .get(&handle)
133 .ok_or_else(|| javascript_error(format!("unknown solid handle {handle}")))?;
134 f(solid)
135 })
136}
137
138pub(crate) fn with_registered_solid_str<T>(
143 handle: u32,
144 f: impl FnOnce(&BrepSolid) -> Result<T, String>,
145) -> Result<T, String> {
146 SOLID_REGISTRY.with(|registry| {
147 let registry = registry.borrow();
148 let solid = registry
149 .solids
150 .get(&handle)
151 .ok_or_else(|| format!("unknown solid handle {handle}"))?;
152 f(solid)
153 })
154}
155
156pub(crate) fn with_two_registered_solids<T, E: From<crate::KernelRefusal>>(
163 a: u32,
164 b: u32,
165 f: impl FnOnce(&BrepSolid, &BrepSolid) -> Result<T, E>,
166) -> Result<T, E> {
167 SOLID_REGISTRY.with(|registry| {
168 let registry = registry.borrow();
169 let handle = |id: u32| -> Result<&BrepSolid, E> {
170 registry.solids.get(&id).ok_or_else(|| {
171 crate::KernelRefusal::input(
172 crate::KernelStage::Collect,
173 "solid_handle",
174 format!("unknown solid handle {id}"),
175 )
176 .into()
177 })
178 };
179 let sa = handle(a)?;
180 let sb = handle(b)?;
181 f(sa, sb)
182 })
183}
184
185pub(crate) fn replace_registered_solid(handle: u32, solid: BrepSolid) -> Result<(), String> {
192 SOLID_REGISTRY.with(|registry| {
193 let mut registry = registry.borrow_mut();
194 match registry.solids.get_mut(&handle) {
195 Some(slot) => {
196 *slot = solid;
197 Ok(())
198 }
199 None => Err(format!("unknown solid handle {handle}")),
200 }
201 })
202}
203
204pub(crate) fn free_registered_solid(handle: u32) {
208 SOLID_REGISTRY.with(|registry| {
209 registry.borrow_mut().solids.remove(&handle);
210 });
211}
212
213#[wasm_bindgen]
216pub fn register_solid_buffer(data: &[f64], names_json: &str) -> Result<u32, JsValue> {
217 let solid = decode_solid_buffer(data, names_json)?;
218 let solid = validate_imported_solid(solid, "register_solid_buffer")?;
219 Ok(register_solid_value(solid))
220}
221
222#[wasm_bindgen]
224pub fn register_solid_json(solid_json: &str) -> Result<u32, JsValue> {
225 let solid: BrepSolid =
226 serde_json::from_str(solid_json).map_err(|error| javascript_error(error.to_string()))?;
227 let solid = validate_imported_solid(solid, "register_solid_json")?;
228 Ok(register_solid_value(solid))
229}
230
231#[wasm_bindgen]
233pub fn free_solid(handle: u32) {
234 SOLID_REGISTRY.with(|registry| {
235 registry.borrow_mut().solids.remove(&handle);
236 });
237}
238
239pub fn registered_solid_clone(handle: u32) -> Result<BrepSolid, String> {
244 with_registered_solid_str(handle, |solid| Ok(solid.clone()))
245}
246
247#[wasm_bindgen]
249pub fn registered_solid_count() -> usize {
250 SOLID_REGISTRY.with(|registry| registry.borrow().solids.len())
251}
252
253#[wasm_bindgen]
257pub fn boolean_handle(
258 a: u32,
259 b: u32,
260 operation: &str,
261 tolerance: f64,
262 merge_coplanar_faces: bool,
263) -> Result<u32, JsValue> {
264 let operation = match operation {
265 "union" => BooleanOperation::Union,
266 "intersect" => BooleanOperation::Intersect,
267 "subtract" => BooleanOperation::Subtract,
268 _ => return Err(javascript_error("unknown boolean operation".into())),
269 };
270 let result = SOLID_REGISTRY.with(|registry| {
271 let registry = registry.borrow();
272 let sa = registry
273 .solids
274 .get(&a)
275 .ok_or_else(|| javascript_error(format!("unknown solid handle {a}")))?;
276 let sb = registry
277 .solids
278 .get(&b)
279 .ok_or_else(|| javascript_error(format!("unknown solid handle {b}")))?;
280 boolean_operation(
281 sa,
282 sb,
283 operation,
284 &BooleanOptions {
285 tolerance,
286 merge_coplanar_faces,
287 ..BooleanOptions::default()
288 },
289 )
290 .map_err(crate::abi::geometry::javascript_refusal)
291 })?;
292 Ok(register_solid_value(result))
293}
294
295#[wasm_bindgen]
298pub fn tessellate_handle(handle: u32, chord_tolerance: f64) -> Result<WasmMeshBuffers, JsValue> {
299 with_registered_solid(handle, |solid| {
300 let mesh = tessellate_brep_watertight(solid, chord_tolerance).map_err(javascript_error)?;
301 Ok(WasmMeshBuffers {
302 positions: mesh.positions,
303 normals: mesh.normals,
304 indices: mesh.indices,
305 face_ids: mesh.face_ids,
306 })
307 })
308}
309
310#[wasm_bindgen]
312pub fn mass_properties_handle(handle: u32) -> Result<String, JsValue> {
313 with_registered_solid(handle, |solid| {
314 let properties = solid_mass_properties(solid).map_err(javascript_error)?;
315 serde_json::to_string(&properties).map_err(|error| javascript_error(error.to_string()))
316 })
317}
318
319#[wasm_bindgen]
323pub fn transform_handle(
324 handle: u32,
325 matrix: &[f64],
326 reverse_orientation: bool,
327) -> Result<u32, JsValue> {
328 let matrix: [f64; 16] = matrix
329 .try_into()
330 .map_err(|_| javascript_error("transform matrix must contain 16 values".into()))?;
331 let transform = AffineTransform::new(matrix).map_err(javascript_error)?;
332 let result = with_registered_solid(handle, |solid| {
333 transform_brep(solid, transform, reverse_orientation).map_err(javascript_error)
334 })?;
335 Ok(register_solid_value(result))
336}
337
338#[wasm_bindgen]
341pub fn face_names_handle(handle: u32) -> Result<String, JsValue> {
342 with_registered_solid(handle, |solid| {
343 let mut map: Vec<(u64, Option<String>)> = Vec::new();
344 for shell in &solid.shells {
345 for face in &shell.faces {
346 map.push((face.id, face.name.clone()));
347 }
348 }
349 serde_json::to_string(&map).map_err(|error| javascript_error(error.to_string()))
350 })
351}
352
353pub fn boolean_handle_native(
365 a: u32,
366 b: u32,
367 operation: BooleanOperation,
368 options: &BooleanOptions,
369) -> Result<u32, String> {
370 let result =
371 with_two_registered_solids(a, b, |sa, sb| boolean_operation(sa, sb, operation, options))?;
372 Ok(register_solid_value(result))
373}