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>(
161 a: u32,
162 b: u32,
163 f: impl FnOnce(&BrepSolid, &BrepSolid) -> Result<T, String>,
164) -> Result<T, String> {
165 SOLID_REGISTRY.with(|registry| {
166 let registry = registry.borrow();
167 let sa = registry
168 .solids
169 .get(&a)
170 .ok_or_else(|| format!("unknown solid handle {a}"))?;
171 let sb = registry
172 .solids
173 .get(&b)
174 .ok_or_else(|| format!("unknown solid handle {b}"))?;
175 f(sa, sb)
176 })
177}
178
179pub(crate) fn replace_registered_solid(handle: u32, solid: BrepSolid) -> Result<(), String> {
186 SOLID_REGISTRY.with(|registry| {
187 let mut registry = registry.borrow_mut();
188 match registry.solids.get_mut(&handle) {
189 Some(slot) => {
190 *slot = solid;
191 Ok(())
192 }
193 None => Err(format!("unknown solid handle {handle}")),
194 }
195 })
196}
197
198pub(crate) fn free_registered_solid(handle: u32) {
202 SOLID_REGISTRY.with(|registry| {
203 registry.borrow_mut().solids.remove(&handle);
204 });
205}
206
207#[wasm_bindgen]
210pub fn register_solid_buffer(data: &[f64], names_json: &str) -> Result<u32, JsValue> {
211 let names: SolidNames = if names_json.trim().is_empty() {
212 SolidNames::default()
213 } else {
214 serde_json::from_str(names_json).map_err(|error| javascript_error(error.to_string()))?
215 };
216 let solid = decode_solid(data, &names).map_err(javascript_error)?;
217 let solid = TopologyArena::from_brep(&solid)
218 .and_then(|arena| arena.to_brep())
219 .map_err(javascript_error)?;
220 let issues = solid.validate();
221 if !issues.is_empty() {
222 return Err(javascript_error(format!(
223 "register_solid_buffer: invalid topology: {issues:?}"
224 )));
225 }
226 Ok(register_solid_value(solid))
227}
228
229#[wasm_bindgen]
231pub fn register_solid_json(solid_json: &str) -> Result<u32, JsValue> {
232 let solid: BrepSolid =
233 serde_json::from_str(solid_json).map_err(|error| javascript_error(error.to_string()))?;
234 let solid = TopologyArena::from_brep(&solid)
235 .and_then(|arena| arena.to_brep())
236 .map_err(javascript_error)?;
237 let issues = solid.validate();
238 if !issues.is_empty() {
239 return Err(javascript_error(format!(
240 "register_solid_json: invalid topology: {issues:?}"
241 )));
242 }
243 Ok(register_solid_value(solid))
244}
245
246#[wasm_bindgen]
248pub fn free_solid(handle: u32) {
249 SOLID_REGISTRY.with(|registry| {
250 registry.borrow_mut().solids.remove(&handle);
251 });
252}
253
254#[wasm_bindgen]
256pub fn registered_solid_count() -> usize {
257 SOLID_REGISTRY.with(|registry| registry.borrow().solids.len())
258}
259
260#[wasm_bindgen]
264pub fn boolean_handle(
265 a: u32,
266 b: u32,
267 operation: &str,
268 tolerance: f64,
269 merge_coplanar_faces: bool,
270) -> Result<u32, JsValue> {
271 let operation = match operation {
272 "union" => BooleanOperation::Union,
273 "intersect" => BooleanOperation::Intersect,
274 "subtract" => BooleanOperation::Subtract,
275 _ => return Err(javascript_error("unknown boolean operation".into())),
276 };
277 let result = SOLID_REGISTRY.with(|registry| {
278 let registry = registry.borrow();
279 let sa = registry
280 .solids
281 .get(&a)
282 .ok_or_else(|| javascript_error(format!("unknown solid handle {a}")))?;
283 let sb = registry
284 .solids
285 .get(&b)
286 .ok_or_else(|| javascript_error(format!("unknown solid handle {b}")))?;
287 boolean_operation(
288 sa,
289 sb,
290 operation,
291 &BooleanOptions {
292 tolerance,
293 merge_coplanar_faces,
294 ..BooleanOptions::default()
295 },
296 )
297 .map_err(javascript_error)
298 })?;
299 Ok(register_solid_value(result))
300}
301
302#[wasm_bindgen]
305pub fn tessellate_handle(handle: u32, chord_tolerance: f64) -> Result<WasmMeshBuffers, JsValue> {
306 with_registered_solid(handle, |solid| {
307 let mesh = tessellate_brep_watertight(solid, chord_tolerance).map_err(javascript_error)?;
308 Ok(WasmMeshBuffers {
309 positions: mesh.positions,
310 normals: mesh.normals,
311 indices: mesh.indices,
312 face_ids: mesh.face_ids,
313 })
314 })
315}
316
317#[wasm_bindgen]
319pub fn mass_properties_handle(handle: u32) -> Result<String, JsValue> {
320 with_registered_solid(handle, |solid| {
321 let properties = solid_mass_properties(solid).map_err(javascript_error)?;
322 serde_json::to_string(&properties).map_err(|error| javascript_error(error.to_string()))
323 })
324}
325
326#[wasm_bindgen]
330pub fn transform_handle(
331 handle: u32,
332 matrix: &[f64],
333 reverse_orientation: bool,
334) -> Result<u32, JsValue> {
335 let matrix: [f64; 16] = matrix
336 .try_into()
337 .map_err(|_| javascript_error("transform matrix must contain 16 values".into()))?;
338 let transform = AffineTransform::new(matrix).map_err(javascript_error)?;
339 let result = with_registered_solid(handle, |solid| {
340 transform_brep(solid, transform, reverse_orientation).map_err(javascript_error)
341 })?;
342 Ok(register_solid_value(result))
343}
344
345#[wasm_bindgen]
348pub fn face_names_handle(handle: u32) -> Result<String, JsValue> {
349 with_registered_solid(handle, |solid| {
350 let mut map: Vec<(u64, Option<String>)> = Vec::new();
351 for shell in &solid.shells {
352 for face in &shell.faces {
353 map.push((face.id, face.name.clone()));
354 }
355 }
356 serde_json::to_string(&map).map_err(|error| javascript_error(error.to_string()))
357 })
358}
359
360pub fn boolean_handle_native(
372 a: u32,
373 b: u32,
374 operation: BooleanOperation,
375 options: &BooleanOptions,
376) -> Result<u32, String> {
377 let result =
378 with_two_registered_solids(a, b, |sa, sb| boolean_operation(sa, sb, operation, options))?;
379 Ok(register_solid_value(result))
380}