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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134

use svd_rs::Peripheral;

use super::{new_node, Config, Element, Encode, EncodeChildren, EncodeError, XMLNode};
use crate::{
    config::{DerivableSorting, Sorting},
    svd::Device,
};

impl Encode for Device {
    type Error = EncodeError;

    fn encode_with_config(&self, config: &Config) -> Result<Element, EncodeError> {
        let mut elem = Element::new("device");
        if let Some(v) = &self.vendor {
            elem.children.push(new_node("vendor", v.clone()));
        }
        if let Some(v) = &self.vendor_id {
            elem.children.push(new_node("vendorID", v.clone()));
        }

        elem.children.push(new_node("name", self.name.clone()));

        if let Some(v) = &self.series {
            elem.children.push(new_node("series", v.clone()));
        }

        elem.children
            .push(new_node("version", self.version.clone()));

        elem.children
            .push(new_node("description", self.description.clone()));

        if let Some(v) = &self.license_text {
            elem.children.push(new_node("licenseText", v.clone()));
        }

        if let Some(v) = &self.cpu {
            elem.children
                .push(XMLNode::Element(v.encode_with_config(config)?));
        }

        if let Some(v) = &self.header_system_filename {
            elem.children
                .push(new_node("headerSystemFilename", v.clone()));
        }

        if let Some(v) = &self.header_definitions_prefix {
            elem.children
                .push(new_node("headerDefinitionsPrefix", v.clone()));
        }

        elem.children.push(new_node(
            "addressUnitBits",
            format!("{}", self.address_unit_bits),
        ));

        elem.children
            .push(new_node("width", format!("{}", self.width)));

        elem.children.extend(
            self.default_register_properties
                .encode_with_config(config)?,
        );

        let peripherals: Result<Vec<_>, _> =
            if config.peripheral_sorting == DerivableSorting::Unchanged(None) {
                self.peripherals
                    .iter()
                    .map(|peripheral| peripheral.encode_node_with_config(config))
                    .collect()
            } else {
                sort_derived_peripherals(&self.peripherals, config.peripheral_sorting)
                    .into_iter()
                    .map(|peripheral| peripheral.encode_node_with_config(config))
                    .collect()
            };

        elem.children.push({
            let mut e = Element::new("peripherals");
            e.children = peripherals?;
            XMLNode::Element(e)
        });

        elem.attributes
            .insert(String::from("schemaVersion"), self.schema_version.clone());
        elem.attributes
            .insert(String::from("xmlns:xs"), self.xmlns_xs.clone());
        elem.attributes.insert(
            String::from("xs:noNamespaceSchemaLocation"),
            self.no_namespace_schema_location.clone(),
        );

        Ok(elem)
    }
}

fn sort_peripherals(refs: &mut [&Peripheral], sorting: Option<Sorting>) {
    if let Some(sorting) = sorting {
        match sorting {
            Sorting::Offset => refs.sort_by_key(|p| p.base_address),
            Sorting::OffsetReversed => {
                refs.sort_by_key(|p| -(p.base_address as i32));
            }
            Sorting::Name => refs.sort_by_key(|p| &p.name),
        }
    }
}

fn sort_derived_peripherals(
    peripherals: &[Peripheral],
    sorting: DerivableSorting,
) -> Vec<&Peripheral> {
    match sorting {
        DerivableSorting::Unchanged(sorting) => {
            let mut refs = peripherals.iter().collect::<Vec<_>>();
            sort_peripherals(&mut refs, sorting);
            refs
        }
        DerivableSorting::DeriveLast(sorting) => {
            let mut common_refs = peripherals
                .iter()
                .filter(|p| p.derived_from.is_none())
                .collect::<Vec<_>>();
            let mut derived_refs = peripherals
                .iter()
                .filter(|p| p.derived_from.is_some())
                .collect::<Vec<_>>();
            sort_peripherals(&mut common_refs, sorting);
            sort_peripherals(&mut derived_refs, sorting);
            common_refs.extend(derived_refs);
            common_refs
        }
    }
}