Function write_bgeo

splashsurf_lib/src/io/bgeo_format.rs:173–231

(bgeo: &BgeoFile, mut writer: W2)

    enable_compression: bool,
) -> Result<(), anyhow::Error> {
    fn write_bgeo<W2: io::Write>(bgeo: &BgeoFile, mut writer: W2) -> Result<(), anyhow::Error> {
        writer.write_all(&bgeo.header.magic_bytes)?;
        writer.write_all(&bgeo.header.version_char.to_be_bytes())?;
        writer.write_all(&bgeo.header.version.to_be_bytes())?;

        writer.write_all(&bgeo.header.num_points.to_be_bytes())?;
        writer.write_all(&bgeo.header.num_prims.to_be_bytes())?;
        writer.write_all(&bgeo.header.num_point_groups.to_be_bytes())?;
        writer.write_all(&bgeo.header.num_prim_groups.to_be_bytes())?;
        writer.write_all(&bgeo.header.num_point_attrib.to_be_bytes())?;
        writer.write_all(&bgeo.header.num_vertex_attrib.to_be_bytes())?;
        writer.write_all(&bgeo.header.num_prim_attrib.to_be_bytes())?;
        writer.write_all(&bgeo.header.num_attrib.to_be_bytes())?;

        // Write attribute definitions
        for attrib in &bgeo.attribute_definitions {
            writer.write_all(&(attrib.name.len() as u16).to_be_bytes())?;
            writer.write_all(attrib.name.as_bytes())?;
            writer.write_all(&(attrib.size as u16).to_be_bytes())?;
            writer.write_all(&(attrib.attr_type.to_i32()).to_be_bytes())?;
            for &default_val in &attrib.default_values {
                writer.write_all(&(default_val.to_be_bytes()))?;
            }
        }

        let special_attribs = [&bgeo.positions, &bgeo.weights];
        let attrib_iter = || {
            special_attribs
                .iter()
                .copied()
                .chain(bgeo.attribute_data.iter().map(|(_, s)| s))
        };

        let num_points = attrib_iter().map(|s| s.num_points()).max().unwrap();
        for i in 0..num_points {
            for attrib in attrib_iter() {
                // TODO: Use default values
                match attrib {
                    AttributeStorage::Int(v) => {
                        writer.write_all(&(v[i].to_be_bytes()))?;
                    }
                    AttributeStorage::Float(v) => {
                        writer.write_all(&(v[i].to_be_bytes()))?;
                    }
                    AttributeStorage::Vector(n, v) => {
                        for j in 0..*n {
                            writer.write_all(&(v[i * n + j].to_be_bytes()))?;
                        }
                    }
                }
            }
        }

        // End bytes
        writer.write_all(&0x00_u8.to_be_bytes())?;
        writer.write_all(&0xff_u8.to_be_bytes())?;

        Ok(())