Skip to main content

lib3mf_core/parser/
component_parser.rs

1use crate::error::{Lib3mfError, Result};
2use crate::model::{Component, Components};
3use crate::parser::xml_parser::{XmlParser, get_attribute, get_attribute_u32};
4use glam::Mat4;
5use quick_xml::events::Event;
6use std::io::BufRead;
7use std::str::FromStr;
8
9/// Parses a `<components>` element and its `<component>` children into a `Components`.
10pub fn parse_components<R: BufRead>(parser: &mut XmlParser<R>) -> Result<Components> {
11    let mut components = Vec::new();
12
13    loop {
14        match parser.read_next_event()? {
15            Event::Start(e) | Event::Empty(e) if e.name().as_ref() == b"component" => {
16                let object_id = crate::model::ResourceId(get_attribute_u32(&e, b"objectid")?);
17                let path = get_attribute(&e, b"path")
18                    .or_else(|| get_attribute(&e, b"p:path"))
19                    .map(|s| s.into_owned());
20                let uuid = crate::parser::xml_parser::get_attribute_uuid(&e)?;
21                let transform = if let Some(s) = get_attribute(&e, b"transform") {
22                    parse_transform(&s)?
23                } else {
24                    Mat4::IDENTITY
25                };
26                components.push(Component {
27                    object_id,
28                    path,
29                    uuid,
30                    transform,
31                });
32            }
33            Event::End(e) if e.name().as_ref() == b"components" => break,
34            Event::Eof => {
35                return Err(Lib3mfError::Validation(
36                    "Unexpected EOF in components".to_string(),
37                ));
38            }
39            _ => {}
40        }
41    }
42
43    Ok(Components { components })
44}
45
46/// Parses a 3MF transform matrix string (12 space-separated floats in column-major order) into a `Mat4`.
47pub fn parse_transform(s: &str) -> Result<Mat4> {
48    let parts: Vec<&str> = s.split_whitespace().collect();
49    if parts.len() != 12 {
50        return Err(Lib3mfError::Validation(format!(
51            "Invalid transform matrix: expected 12 values, got {}",
52            parts.len()
53        )));
54    }
55
56    let p: Result<Vec<f32>> = parts
57        .iter()
58        .map(|v| {
59            f32::from_str(v)
60                .map_err(|_| Lib3mfError::Validation(format!("Invalid float in transform: {}", v)))
61        })
62        .collect();
63    let p = p?;
64
65    // 3MF uses column-major order 4x3 matrix, last column is 0,0,0,1
66    Ok(Mat4::from_cols_array(&[
67        p[0], p[1], p[2], 0.0, p[3], p[4], p[5], 0.0, p[6], p[7], p[8], 0.0, p[9], p[10], p[11],
68        1.0,
69    ]))
70}