ecitygml-io 0.0.2-alpha.5

IO operations for processing CityGML data.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

use crate::Error;
use crate::gml::codec::core::{
    deserialize_abstract_occupied_space, serialize_abstract_occupied_space,
};
use crate::gml::util::CityGmlElement;
use ecitygml_core::model::core::AsAbstractOccupiedSpace;
use ecitygml_core::model::water_body::WaterBody;
use ecitygml_core::model::water_body::values::{
    WaterBodyClassValue, WaterBodyFunctionValue, WaterBodyUsageValue,
};
use egml::io::codec::basic::GmlCode;
use egml::io::util::{
    Formatting, XmlNode, XmlNodeContent, extract_xml_element_spans, serialize_inner,
};
use egml::model::basic_types::Code;
use quick_xml::de;
use serde::{Deserialize, Serialize};

pub fn deserialize_water_body(xml_document: &[u8]) -> Result<WaterBody, Error> {
    let spans = extract_xml_element_spans(xml_document)?;
    let (abstract_occupied_space_result, parsed_result) = rayon::join(
        || deserialize_abstract_occupied_space(xml_document, &spans),
        || de::from_reader::<_, GmlWaterBody>(xml_document).map_err(Error::from),
    );
    let abstract_occupied_space = abstract_occupied_space_result?;
    let parsed = parsed_result?;
    let mut water_body = WaterBody::from_abstract_occupied_space(abstract_occupied_space);

    water_body.set_class_opt(parsed.class.map(Code::from).map(WaterBodyClassValue::from));
    water_body.set_functions(
        parsed
            .functions
            .into_iter()
            .map(Code::from)
            .map(WaterBodyFunctionValue::from)
            .collect(),
    );
    water_body.set_usages(
        parsed
            .usages
            .into_iter()
            .map(Code::from)
            .map(WaterBodyUsageValue::from)
            .collect(),
    );

    Ok(water_body)
}

pub fn serialize_water_body(
    water_body: &WaterBody,
    formatting: Formatting,
) -> Result<XmlNode, Error> {
    let mut xml_node_parts =
        serialize_abstract_occupied_space(water_body.abstract_occupied_space(), formatting)?;

    if let Some(raw) = serialize_inner(GmlWaterBody::from(water_body), formatting)? {
        xml_node_parts.content.push(XmlNodeContent::Raw(raw));
    }

    Ok(XmlNode::new(
        CityGmlElement::WaterBody.into(),
        xml_node_parts,
    ))
}

#[derive(Debug, Serialize, Deserialize, PartialEq, Clone)]
pub struct GmlWaterBody {
    #[serde(
        rename(serialize = "wtr:class", deserialize = "class"),
        skip_serializing_if = "Option::is_none"
    )]
    pub class: Option<GmlCode>,

    #[serde(rename(serialize = "wtr:function", deserialize = "function"), default)]
    pub functions: Vec<GmlCode>,

    #[serde(rename(serialize = "wtr:usage", deserialize = "usage"), default)]
    pub usages: Vec<GmlCode>,
}

impl From<&WaterBody> for GmlWaterBody {
    fn from(item: &WaterBody) -> Self {
        Self {
            class: item.class().map(WaterBodyClassValue::code).map(Into::into),
            functions: item
                .functions()
                .iter()
                .map(WaterBodyFunctionValue::code)
                .map(Into::into)
                .collect(),
            usages: item
                .usages()
                .iter()
                .map(WaterBodyUsageValue::code)
                .map(Into::into)
                .collect(),
        }
    }
}