use hashbrown::HashMap;
use structs::Row;
use traits::AdjustmentValue;
#[derive(Clone, Default, Debug)]
pub(crate) struct Rows {
rows: HashMap<u32, Row>,
}
impl Rows {
pub(crate) fn has_sheet_data(&self) -> bool {
!self.rows.is_empty()
}
pub(crate) fn get_row_dimensions(&self) -> Vec<&Row> {
self.rows.values().collect()
}
pub(crate) fn get_row_dimensions_mut(&mut self) -> Vec<&mut Row> {
self.rows.values_mut().collect()
}
pub(crate) fn get_row_dimensions_to_hashmap(&self) -> &HashMap<u32, Row> {
&self.rows
}
pub(crate) fn get_row_dimensions_to_hashmap_mut(&mut self) -> &mut HashMap<u32, Row> {
&mut self.rows
}
pub(crate) fn get_row_dimension(&self, row: &u32) -> Option<&Row> {
self.rows.get(row)
}
pub(crate) fn get_row_dimension_mut(&mut self, row: &u32) -> &mut Row {
self.rows.entry(row.to_owned()).or_insert_with(|| {
let mut obj = Row::default();
obj.set_row_num(*row);
obj
})
}
pub(crate) fn set_row_dimension(&mut self, value: Row) -> &mut Self {
let row = value.get_row_num();
self.rows.insert(row.to_owned(), value);
self
}
pub(crate) fn rebuild_map(&mut self) {
self.rows = self
.get_row_dimensions_to_hashmap_mut()
.iter_mut()
.map(|(_, row)| (*row.get_row_num(), std::mem::take(row)))
.collect();
}
}
impl AdjustmentValue for Rows {
fn adjustment_insert_value(&mut self, root_num: &u32, offset_num: &u32) {
for row_dimension in self.get_row_dimensions_mut() {
row_dimension.adjustment_insert_value(root_num, offset_num);
}
self.rebuild_map();
}
fn adjustment_remove_value(&mut self, root_num: &u32, offset_num: &u32) {
self.get_row_dimensions_to_hashmap_mut()
.retain(|_, k| !(k.is_remove_value(root_num, offset_num)));
for row_dimension in self.get_row_dimensions_mut() {
row_dimension.adjustment_remove_value(root_num, offset_num);
}
self.rebuild_map();
}
}