xsd_parser/pipeline/optimizer/
dynamic_to_choice.rs

1use crate::models::{
2    meta::{ComplexMeta, ElementMeta, ElementMode, GroupMeta, MetaType, MetaTypeVariant},
3    Ident,
4};
5use crate::traits::{NameBuilderExt as _, VecHelper};
6
7use super::Optimizer;
8
9impl Optimizer {
10    /// This will use a enum that contains all known variants of the dynamic
11    /// type instead of a dynamic box.
12    ///
13    /// # Examples
14    ///
15    /// Consider the following XML schema.
16    /// ```xml
17    #[doc = include_str!("../../../tests/optimizer/abstract.xsd")]
18    /// ```
19    ///
20    /// Without this optimization this will result in the following code:
21    /// ```rust
22    #[doc = include_str!("../../../tests/optimizer/expected0/convert_dynamic_to_choice.rs")]
23    /// ```
24    ///
25    /// With this optimization the following code is generated:
26    /// ```rust
27    #[doc = include_str!("../../../tests/optimizer/expected1/convert_dynamic_to_choice.rs")]
28    /// ```
29    pub fn convert_dynamic_to_choice(mut self) -> Self {
30        use std::collections::btree_map::Entry;
31
32        tracing::debug!("convert_dynamic_to_choice");
33
34        let idents = self
35            .types
36            .items
37            .iter()
38            .filter_map(|(ident, ty)| {
39                if matches!(&ty.variant, MetaTypeVariant::Dynamic(_)) {
40                    Some(ident)
41                } else {
42                    None
43                }
44            })
45            .cloned()
46            .collect::<Vec<_>>();
47
48        for ident in idents {
49            let content_name = self.types.name_builder().shared_name("Content").finish();
50            let content_ident = Ident::new(content_name).with_ns(ident.ns);
51
52            let mut si = GroupMeta::default();
53            let type_ = self.types.items.get(&ident).unwrap();
54            self.add_elements(&mut si, type_);
55
56            let type_ = self.types.items.get_mut(&ident).unwrap();
57            type_.variant = MetaTypeVariant::ComplexType(ComplexMeta {
58                content: Some(content_ident.clone()),
59                is_dynamic: true,
60                ..Default::default()
61            });
62
63            match self.types.items.entry(content_ident) {
64                Entry::Vacant(e) => {
65                    e.insert(MetaType::new(MetaTypeVariant::Choice(si)));
66                }
67                Entry::Occupied(_) => crate::unreachable!(),
68            }
69        }
70
71        self
72    }
73
74    fn add_elements(&self, group: &mut GroupMeta, ty: &MetaType) {
75        let form = ty.form();
76        let MetaTypeVariant::Dynamic(x) = &ty.variant else {
77            crate::unreachable!();
78        };
79
80        for derived in &x.derived_types {
81            let derived_ty = self.types.get_resolved_type(derived).unwrap();
82            if let MetaTypeVariant::Dynamic(_) = &derived_ty.variant {
83                self.add_elements(group, derived_ty);
84            } else {
85                group.elements.find_or_insert(derived.clone(), |ident| {
86                    ElementMeta::new(ident, derived.clone(), ElementMode::Element, form)
87                });
88            }
89        }
90    }
91}
xsd_parser/pipeline/optimizer/dynamic_to_choice.rs

xsd_parser/pipeline/optimizer/
dynamic_to_choice.rs
