use crate::grid::records::IntoRecords;
#[derive(Debug)]
pub struct LimitColumns<I> {
records: I,
limit: usize,
}
impl LimitColumns<()> {
pub fn new<I>(records: I, limit: usize) -> LimitColumns<I> {
LimitColumns { records, limit }
}
}
impl<I> IntoRecords for LimitColumns<I>
where
I: IntoRecords,
{
type Cell = I::Cell;
type IterColumns = LimitColumnsColumnsIter<<I::IterColumns as IntoIterator>::IntoIter>;
type IterRows = LimitColumnsIter<<I::IterRows as IntoIterator>::IntoIter>;
fn iter_rows(self) -> Self::IterRows {
LimitColumnsIter {
iter: self.records.iter_rows().into_iter(),
limit: self.limit,
}
}
}
#[derive(Debug)]
pub struct LimitColumnsIter<I> {
iter: I,
limit: usize,
}
impl<I> Iterator for LimitColumnsIter<I>
where
I: Iterator,
I::Item: IntoIterator,
{
type Item = LimitColumnsColumnsIter<<I::Item as IntoIterator>::IntoIter>;
fn next(&mut self) -> Option<Self::Item> {
let iter = self.iter.next()?;
let iter = LimitColumnsColumnsIter {
iter: iter.into_iter(),
limit: self.limit,
};
Some(iter)
}
}
#[derive(Debug)]
pub struct LimitColumnsColumnsIter<I> {
iter: I,
limit: usize,
}
impl<I> Iterator for LimitColumnsColumnsIter<I>
where
I: Iterator,
{
type Item = I::Item;
fn next(&mut self) -> Option<Self::Item> {
if self.limit == 0 {
return None;
}
self.limit -= 1;
self.iter.next()
}
}