umya_spreadsheet/structs/
data_fields.rs

1// dataFields
2use crate::reader::driver::*;
3use crate::structs::DataField;
4use crate::writer::driver::*;
5use quick_xml::events::{BytesStart, Event};
6use quick_xml::Reader;
7use quick_xml::Writer;
8use std::io::Cursor;
9use thin_vec::ThinVec;
10
11#[derive(Clone, Default, Debug)]
12pub struct DataFields {
13    list: ThinVec<DataField>,
14}
15impl DataFields {
16    #[inline]
17    pub fn get_list(&self) -> &[DataField] {
18        &self.list
19    }
20
21    #[inline]
22    pub fn get_list_mut(&mut self) -> &mut ThinVec<DataField> {
23        &mut self.list
24    }
25
26    #[inline]
27    pub fn add_list_mut(&mut self, value: DataField) -> &mut Self {
28        self.list.push(value);
29        self
30    }
31
32    #[inline]
33    pub(crate) fn set_attributes<R: std::io::BufRead>(
34        &mut self,
35        reader: &mut Reader<R>,
36        e: &BytesStart,
37    ) {
38        xml_read_loop!(
39            reader,
40            Event::Empty(ref e) => {
41                if e.name().into_inner() == b"dataField" {
42                    let mut obj = DataField::default();
43                    obj.set_attributes(reader, e);
44                    self.add_list_mut(obj);
45                }
46            },
47            Event::End(ref e) => {
48                if e.name().into_inner() == b"dataFields" {
49                    return
50                }
51            },
52            Event::Eof => panic!("Error: Could not find {} end element", "dataFields")
53        );
54    }
55
56    #[inline]
57    pub(crate) fn write_to(&self, writer: &mut Writer<Cursor<Vec<u8>>>) {
58        // dataFields
59        write_start_tag(
60            writer,
61            "dataFields",
62            vec![("count", &self.list.len().to_string())],
63            false,
64        );
65
66        // dataField
67        for sheet_view in &self.list {
68            sheet_view.write_to(writer);
69        }
70
71        write_end_tag(writer, "dataFields");
72    }
73}