umya_spreadsheet/structs/office/excel/
reference_sequence.rs

1// xm:sqref
2use crate::reader::driver::*;
3use crate::structs::Coordinate;
4use crate::structs::Range;
5use crate::writer::driver::*;
6use quick_xml::events::{BytesStart, Event};
7use quick_xml::Reader;
8use quick_xml::Writer;
9use std::io::Cursor;
10use std::vec;
11use thin_vec::ThinVec;
12
13#[derive(Default, Debug, Clone)]
14pub struct ReferenceSequence {
15    value: ThinVec<Range>,
16}
17impl ReferenceSequence {
18    #[inline]
19    pub fn get_value(&self) -> &[Range] {
20        &self.value
21    }
22
23    #[inline]
24    pub fn get_value_mut(&mut self) -> &mut ThinVec<Range> {
25        &mut self.value
26    }
27
28    #[inline]
29    pub fn set_value(&mut self, value: impl Into<ThinVec<Range>>) -> &mut Self {
30        self.value = value.into();
31        self
32    }
33
34    #[inline]
35    pub fn add_value(&mut self, value: Range) -> &mut Self {
36        self.value.push(value);
37        self
38    }
39
40    #[inline]
41    pub fn remove_value(&mut self) -> &mut Self {
42        self.value.clear();
43        self
44    }
45
46    pub fn set_sqref<S: Into<String>>(&mut self, value: S) -> &mut Self {
47        value.into().split(' ').for_each(|range_value| {
48            let mut range = Range::default();
49            range.set_range(range_value);
50            self.value.push(range);
51        });
52        self
53    }
54
55    #[inline]
56    pub fn get_sqref(&self) -> String {
57        self.value
58            .iter()
59            .map(|range| range.get_range())
60            .collect::<Vec<String>>()
61            .join(" ")
62    }
63
64    pub(crate) fn set_attributes<R: std::io::BufRead>(
65        &mut self,
66        reader: &mut Reader<R>,
67        _e: &BytesStart,
68    ) {
69        let mut value: String = String::new();
70        let mut buf = Vec::new();
71        loop {
72            match reader.read_event_into(&mut buf) {
73                Ok(Event::Text(e)) => {
74                    value = e.unescape().unwrap().to_string();
75                }
76                Ok(Event::End(ref e)) => match e.name().into_inner() {
77                    b"xm:sqref" => {
78                        self.set_sqref(value);
79                        value = String::new();
80                        return;
81                    }
82                    _ => (),
83                },
84                Ok(Event::Eof) => panic!("Error: Could not find {} end element", "xm:sqref"),
85                Err(e) => panic!("Error at position {}: {:?}", reader.buffer_position(), e),
86                _ => (),
87            }
88            buf.clear();
89        }
90    }
91
92    #[inline]
93    pub(crate) fn write_to(&self, writer: &mut Writer<Cursor<Vec<u8>>>) {
94        write_start_tag(writer, "xm:sqref", vec![], false);
95        write_text_node(writer, &self.get_sqref());
96        write_end_tag(writer, "xm:sqref");
97    }
98}