umya_spreadsheet/structs/
pivot_fields.rs

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

umya_spreadsheet/structs/
pivot_fields.rs
