ooza/src/machine.rs

use g_code::{
    command,
    emit::Token,
    parse::{ast::Snippet, token::Field},
};

/// Whether the tool is active (i.e. cutting)
#[derive(Copy, Clone, PartialEq, Eq, Debug)]
pub enum Tool {
    Off,
    On,
}

impl std::ops::Not for Tool {
    type Output = Self;
    fn not(self) -> Self {
        match self {
            Self::Off => Self::On,
            Self::On => Self::Off,
        }
    }
}

/// The distance mode for movement commands
#[derive(Copy, Clone, PartialEq, Eq, Debug)]
pub enum Distance {
    Absolute,
    Relative,
}

impl std::ops::Not for Distance {
    type Output = Self;
    fn not(self) -> Self {
        match self {
            Self::Absolute => Self::Relative,
            Self::Relative => Self::Absolute,
        }
    }
}

/// Generic machine state simulation, assuming nothing is known about the machine when initialized.
/// This is used to reduce output GCode verbosity and run repetitive actions.
#[derive(Debug)]
pub struct Machine<'input> {
    pub(crate) tool_state: Option<Tool>,
    pub(crate) distance_mode: Option<Distance>,
    pub(crate) tool_on_action: Option<Snippet<'input>>,
    pub(crate) tool_off_action: Option<Snippet<'input>>,
    pub(crate) program_begin_sequence: Option<Snippet<'input>>,
    pub(crate) program_end_sequence: Option<Snippet<'input>>,
}

impl<'input> Machine<'input> {
    /// Output gcode to turn the tool on.
    pub fn tool_on<'a>(&'a mut self) -> Vec<Token> {
        if self.tool_state == Some(Tool::Off) || self.tool_state == None {
            self.tool_state = Some(Tool::On);
            self.tool_on_action
                .iter()
                .flat_map(|s| s.iter_fields())
                .map(|f: &Field| Token::from(f))
                .collect()
        } else {
            vec![]
        }
    }

    /// Output gcode to turn the tool off.
    pub fn tool_off<'a>(&'a mut self) -> Vec<Token> {
        if self.tool_state == Some(Tool::On) || self.tool_state == None {
            self.tool_state = Some(Tool::Off);
            self.tool_on_action
                .iter()
                .flat_map(|s| s.iter_fields())
                .map(|f: &Field| Token::from(f))
                .collect()
        } else {
            vec![]
        }
    }

    pub fn program_begin<'a>(&'a self) -> Vec<Token> {
        self.program_begin_sequence
            .iter()
            .flat_map(|s| s.iter_fields())
            .map(|f: &Field| Token::from(f))
            .collect()
    }
    pub fn program_end<'a>(&'a self) -> Vec<Token> {
        self.program_end_sequence
            .iter()
            .flat_map(|s| s.iter_fields())
            .map(|f: &Field| Token::from(f))
            .collect()
    }

    /// Output absolute distance field if mode was relative or unknown.
    pub fn absolute(&mut self) -> Vec<Token> {
        if self.distance_mode == Some(Distance::Relative) || self.distance_mode == None {
            self.distance_mode = Some(Distance::Absolute);
            command!(AbsoluteDistanceMode {}).as_token_vec()
        } else {
            vec![]
        }
    }

    /// Output relative distance field if mode was absolute or unknown.
    pub fn relative(&mut self) -> Vec<Token> {
        if self.distance_mode == Some(Distance::Absolute) || self.distance_mode == None {
            self.distance_mode = Some(Distance::Relative);
            command!(RelativeDistanceMode {}).as_token_vec()
        } else {
            vec![]
        }
    }
}