use crate::{dim::Dim, prelude::DynDenseStoMut};
#[derive(Debug, PartialEq)]
pub struct DrMatrixRowConstructor<'a, DS> {
data: &'a mut DS,
dim: &'a mut Dim<[usize; 2]>,
inserted_elems: usize,
}
impl<'a, DS> DrMatrixRowConstructor<'a, DS>
where
DS: DynDenseStoMut,
{
pub(crate) fn new(data: &'a mut DS, dim: &'a mut Dim<[usize; 2]>) -> Self {
DrMatrixRowConstructor {
data,
dim,
inserted_elems: 0,
}
}
pub fn clone_values_from_row(mut self, row: &[DS::Item]) -> Self
where
DS::Item: Clone,
{
self.inserted_elems += row.len();
self.data.extend_from_clone(row);
self
}
pub fn commit(mut self) {
assert!(
self.inserted_elems == self.dim.cols(),
"The number of inserted elements must be equal the number of columns of `Self`."
);
self.inserted_elems = 0;
*self.dim.rows_mut() += 1;
}
pub fn copy_values_from_row(mut self, row: &[DS::Item]) -> Self
where
DS::Item: Copy,
{
self.inserted_elems += row.len();
self.data.extend(row);
self
}
pub fn push_value(mut self, value: DS::Item) -> Self {
self.data.push(value);
self.inserted_elems += 1;
self
}
}
impl<'a, DS> Drop for DrMatrixRowConstructor<'a, DS> {
fn drop(&mut self) {
if self.inserted_elems > 0 {
panic!(
"Every single nonempty instance of `DrMatrixRowConstructor` must
end with a call to the `commit` method."
);
}
}
}