use crate::structs::Row;
use crate::traits::AdjustmentValue;
use std::collections::HashMap;
#[derive(Clone, Default, Debug)]
pub(crate) struct Rows {
rows: HashMap<u32, Box<Row>>,
}
impl Rows {
#[inline]
pub(crate) fn has_sheet_data(&self) -> bool {
!self.rows.is_empty()
}
#[inline]
pub(crate) fn get_row_dimensions(&self) -> Vec<&Row> {
self.rows.values().map(Box::as_ref).collect()
}
#[inline]
pub(crate) fn get_row_dimensions_mut(&mut self) -> Vec<&mut Row> {
self.rows.values_mut().map(Box::as_mut).collect()
}
#[inline]
pub(crate) fn get_row_dimensions_to_hashmap(&self) -> &HashMap<u32, Box<Row>> {
&self.rows
}
#[inline]
pub(crate) fn get_row_dimensions_to_hashmap_mut(&mut self) -> &mut HashMap<u32, Box<Row>> {
&mut self.rows
}
#[inline]
pub(crate) fn get_row_dimension(&self, row: &u32) -> Option<&Row> {
self.rows.get(row).map(Box::as_ref)
}
#[inline]
pub(crate) fn get_row_dimension_mut(&mut self, row: &u32) -> &mut Row {
self.rows.entry(*row).or_insert_with(|| {
let mut obj = Row::default();
obj.set_row_num(*row);
Box::new(obj)
})
}
#[inline]
pub(crate) fn set_row_dimension(&mut self, value: Row) -> &mut Self {
let row = value.get_row_num();
self.rows.insert(*row, Box::new(value));
self
}
#[inline]
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();
}
}