umya_spreadsheet/structs/
column_items.rs

1// colItems
2use crate::reader::driver::*;
3use crate::structs::RowItem;
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 ColumnItems {
13    list: ThinVec<RowItem>,
14}
15impl ColumnItems {
16    #[inline]
17    pub fn get_list(&self) -> &[RowItem] {
18        &self.list
19    }
20
21    #[inline]
22    pub fn get_list_mut(&mut self) -> &mut ThinVec<RowItem> {
23        &mut self.list
24    }
25
26    #[inline]
27    pub fn add_list_mut(&mut self, value: RowItem) -> &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"i" {
42                    let mut obj = RowItem::default();
43                    obj.set_attributes(reader, e, true);
44                    self.add_list_mut(obj);
45                }
46            },
47            Event::Start(ref e) => {
48                if e.name().into_inner() == b"i" {
49                    let mut obj = RowItem::default();
50                    obj.set_attributes(reader, e, false);
51                    self.add_list_mut(obj);
52                }
53            },
54            Event::End(ref e) => {
55                if e.name().into_inner() == b"colItems" {
56                    return
57                }
58            },
59            Event::Eof => panic!("Error: Could not find {} end element", "colItems")
60        );
61    }
62
63    #[inline]
64    pub(crate) fn write_to(&self, writer: &mut Writer<Cursor<Vec<u8>>>) {
65        // colItems
66        write_start_tag(
67            writer,
68            "colItems",
69            vec![("count", &self.list.len().to_string())],
70            false,
71        );
72
73        // i
74        for i in &self.list {
75            i.write_to(writer);
76        }
77
78        write_end_tag(writer, "colItems");
79    }
80}