umya_spreadsheet/structs/
column_fields.rs

1// colFields
2use crate::reader::driver::*;
3use crate::structs::Field;
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 ColumnFields {
13    list: ThinVec<Field>,
14}
15impl ColumnFields {
16    #[inline]
17    pub fn get_list(&self) -> &[Field] {
18        &self.list
19    }
20
21    #[inline]
22    pub fn get_list_mut(&mut self) -> &mut ThinVec<Field> {
23        &mut self.list
24    }
25
26    #[inline]
27    pub fn add_list_mut(&mut self, value: Field) -> &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"field" {
42                    let mut obj = Field::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"colFields" {
49                    return
50                }
51            },
52            Event::Eof => panic!("Error: Could not find {} end element", "colFields")
53        );
54    }
55
56    #[inline]
57    pub(crate) fn write_to(&self, writer: &mut Writer<Cursor<Vec<u8>>>) {
58        // colFields
59        write_start_tag(
60            writer,
61            "colFields",
62            vec![("count", &self.list.len().to_string())],
63            false,
64        );
65
66        // i
67        for i in &self.list {
68            i.write_to(writer);
69        }
70
71        write_end_tag(writer, "colFields");
72    }
73}