sericom-core 0.4.0

use super::Cell;
use std::ops::{Index, IndexMut};

/// Line is a wrapper around [`Vec<Cell>`] and represents a line within the [`ScreenBuffer`][`super::ScreenBuffer`].
#[derive(Clone, Debug)]
pub struct Line(Vec<Cell>);

impl Line {
    /// Create a new line with the length/size of `width`.
    ///
    /// Filled with [`Cell::default()`].
    pub fn new(width: usize) -> Self {
        Self(vec![Cell::default(); width])
    }

    /// Iterates over all the [`Cell`]s within the line and sets them to [`Cell::default()`].
    pub fn reset(&mut self) {
        self.0.iter_mut().for_each(|cell| *cell = Cell::default());
    }

    /// Iterates over the [`Cell`]s to index `idx` within [`Self`]
    /// and sets them to [`Cell::default()`].
    pub fn reset_to(&mut self, idx: usize) {
        self.0[..idx]
            .iter_mut()
            .for_each(|cell| *cell = Cell::default());
    }

    /// Iterates over the [`Cell`]s from index `idx` within [`Self`]
    /// to the end of [`Self`] and sets them to [`Cell::default()`].
    pub fn reset_from(&mut self, idx: usize) {
        self.0
            .iter_mut()
            .skip(idx)
            .for_each(|cell| *cell = Cell::default());
    }

    /// Sets the character in [`Cell`] at [`Self`]\[`idx`\] to `ch`.
    pub fn set_char(&mut self, idx: usize, ch: char) {
        self.0[idx].character = ch;
    }

    /// Util function to return the length of [`Self`].
    #[allow(clippy::len_without_is_empty)]
    pub const fn len(&self) -> usize {
        self.0.len()
    }

    /// Iterates over the [`Cell`]s and resets their selected state.
    pub fn clear_selection(&mut self) {
        self.0.iter_mut().for_each(|cell| cell.is_selected = false);
    }

    /// Returns a reference to [`Cell`] at `idx`.
    pub fn get_cell(&self, idx: usize) -> Option<&Cell> {
        self.0.get(idx)
    }

    /// Returns a mutable reference to [`Cell`] at `idx`.
    pub fn get_mut_cell(&mut self, idx: usize) -> Option<&mut Cell> {
        self.0.get_mut(idx)
    }
}

impl IntoIterator for Line {
    type Item = Cell;
    type IntoIter = std::vec::IntoIter<Self::Item>;

    fn into_iter(self) -> Self::IntoIter {
        self.0.into_iter()
    }
}

impl<'a> IntoIterator for &'a Line {
    type Item = &'a Cell;
    type IntoIter = std::slice::Iter<'a, Cell>;

    fn into_iter(self) -> Self::IntoIter {
        self.0.iter()
    }
}

impl<'a> IntoIterator for &'a mut Line {
    type Item = &'a mut Cell;
    type IntoIter = std::slice::IterMut<'a, Cell>;

    fn into_iter(self) -> Self::IntoIter {
        self.0.iter_mut()
    }
}

impl Index<usize> for Line {
    type Output = Cell;
    fn index(&self, index: usize) -> &Self::Output {
        &self.0[index]
    }
}

impl IndexMut<usize> for Line {
    fn index_mut(&mut self, index: usize) -> &mut Self::Output {
        &mut self.0[index]
    }
}