umya_spreadsheet/structs/
sheet_views.rs1use super::SheetView;
3use crate::reader::driver::*;
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 SheetViews {
13    sheet_view_list: ThinVec<SheetView>,
14}
15
16impl SheetViews {
17    #[inline]
18    pub fn get_sheet_view_list(&self) -> &[SheetView] {
19        &self.sheet_view_list
20    }
21
22    #[inline]
23    pub fn get_sheet_view_list_mut(&mut self) -> &mut ThinVec<SheetView> {
24        &mut self.sheet_view_list
25    }
26
27    #[inline]
28    pub fn add_sheet_view_list_mut(&mut self, value: SheetView) -> &mut Self {
29        self.sheet_view_list.push(value);
30        self
31    }
32
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"sheetView" {
42                    let mut obj = SheetView::default();
43                    obj.set_attributes(reader, e, true);
44                    self.add_sheet_view_list_mut(obj);
45                }
46            },
47            Event::Start(ref e) => {
48                if e.name().into_inner() == b"sheetView" {
49                    let mut obj = SheetView::default();
50                    obj.set_attributes(reader, e, false);
51                    self.add_sheet_view_list_mut(obj);
52                }
53            },
54            Event::End(ref e) => {
55                if e.name().into_inner() == b"sheetViews" {
56                    return
57                }
58            },
59            Event::Eof => panic!("Error: Could not find {} end element", "sheetViews")
60        );
61    }
62
63    pub(crate) fn write_to(&self, writer: &mut Writer<Cursor<Vec<u8>>>) {
64        if self.sheet_view_list.is_empty() {
66            return;
67        }
68        write_start_tag(writer, "sheetViews", vec![], false);
69
70        for sheet_view in &self.sheet_view_list {
72            sheet_view.write_to(writer);
73        }
74
75        write_end_tag(writer, "sheetViews");
76    }
77}