lcd1602-driver 0.3.0

use embedded_hal::delay::DelayNs;

use crate::{
    command::{CommandSet, DataWidth, Font, LineMode, MoveDirection, RAMType, ShiftType, State},
    lcd::Lcd,
    sender::SendCommand,
    state::LcdState,
};

/// [`Config`] is the init config of a [`Lcd`]
#[derive(Default)]
pub struct Config {
    state: LcdState,

    // This two state is for automatic correction
    font_set_by_user: Option<Font>,
    line_mode_set_by_user: Option<LineMode>,
}

#[allow(missing_docs)]
impl Config {
    pub fn get_backlight(&self) -> State {
        self.state.get_backlight()
    }

    pub fn set_backlight(mut self, backlight: State) -> Self {
        self.state.set_backlight(backlight);
        self
    }

    pub fn get_data_width(&self) -> DataWidth {
        self.state.get_data_width()
    }

    pub fn set_data_width(mut self, data_width: DataWidth) -> Self {
        self.state.set_data_width(data_width);
        self
    }

    pub fn get_line_mode(&self) -> LineMode {
        self.state.get_line_mode()
    }

    pub fn set_line_mode(mut self, line_mode: LineMode) -> Self {
        self.line_mode_set_by_user = Some(line_mode);

        // automatically Font & LineMode compatible check
        self.state.set_font(match line_mode {
            LineMode::OneLine => self.font_set_by_user.unwrap_or_default(),
            LineMode::TwoLine => Font::Font5x8,
        });

        self.state.set_line_mode(line_mode);
        self
    }

    pub fn get_line_capacity(&self) -> u8 {
        self.state.get_line_capacity()
    }

    pub fn get_font(&self) -> Font {
        self.state.get_font()
    }

    pub fn set_font(mut self, font: Font) -> Self {
        self.font_set_by_user = Some(font);

        // automatically Font & LineMode compatible check
        self.state.set_line_mode(match font {
            Font::Font5x8 => self.line_mode_set_by_user.unwrap_or_default(),
            Font::Font5x11 => LineMode::OneLine,
        });

        self.state.set_font(font);
        self
    }

    pub fn get_display_state(&self) -> State {
        self.state.get_display_state()
    }

    pub fn set_display_state(mut self, display: State) -> Self {
        self.state.set_display_state(display);
        self
    }

    pub fn get_cursor_state(&self) -> State {
        self.state.get_cursor_state()
    }

    pub fn set_cursor_state(mut self, cursor: State) -> Self {
        self.state.set_cursor_state(cursor);
        self
    }

    pub fn get_cursor_blink(&self) -> State {
        self.state.get_cursor_blink()
    }

    pub fn set_cursor_blink(mut self, blink: State) -> Self {
        self.state.set_cursor_blink(blink);
        self
    }

    pub fn get_direction(&self) -> MoveDirection {
        self.state.get_direction()
    }

    pub fn set_direction(mut self, dir: MoveDirection) -> Self {
        self.state.set_direction(dir);
        self
    }

    pub fn get_shift_type(&self) -> ShiftType {
        self.state.get_shift_type()
    }

    pub fn set_shift_type(mut self, shift: ShiftType) -> Self {
        self.state.set_shift_type(shift);
        self
    }

    pub fn get_cursor_pos(&self) -> (u8, u8) {
        self.state.get_cursor_pos()
    }

    pub fn set_cursor_pos(mut self, pos: (u8, u8)) -> Self {
        self.state.set_cursor_pos(pos);
        self
    }

    pub fn get_display_offset(&self) -> u8 {
        self.state.get_display_offset()
    }

    pub fn set_display_offset(mut self, offset: u8) -> Self {
        self.state.set_display_offset(offset);
        self
    }

    pub fn get_ram_type(&self) -> RAMType {
        self.state.get_ram_type()
    }

    pub fn set_ram_type(mut self, ram_type: RAMType) -> Self {
        self.state.set_ram_type(ram_type);
        self
    }
}

impl<'a, 'b, Sender, Delayer, const READABLE: bool> Lcd<'a, 'b, Sender, Delayer, READABLE>
where
    Sender: SendCommand<Delayer, READABLE>,
    Delayer: DelayNs,
{
    /// Create a [`Lcd`] driver, and init LCD hardware
    ///
    /// If we don't specify `poll_interval_us`,  
    /// then default value will be 40 us for Write-Only mode, and 10 us for Read-Write mode
    pub fn new(
        sender: &'a mut Sender,
        delayer: &'b mut Delayer,
        config: Config,
        poll_interval_us: Option<u32>,
    ) -> Self {
        let poll_interval_us = if !READABLE {
            poll_interval_us.unwrap_or_default().max(40)
        } else {
            poll_interval_us.unwrap_or_default().max(10)
        };

        let state = config.state;

        // in initialization process, we'd better use "raw command", to strictly follow datasheet

        // only first 2 or 3 commands are different between 4 pin and 8 pin mode
        match state.get_data_width() {
            DataWidth::Bit4 => {
                sender.delay_and_send(
                    CommandSet::HalfFunctionSet,
                    delayer,
                    poll_interval_us.max(40_000),
                );

                sender.delay_and_send(
                    CommandSet::FunctionSet(
                        DataWidth::Bit4,
                        state.get_line_mode(),
                        state.get_font(),
                    ),
                    delayer,
                    poll_interval_us,
                );

                sender.delay_and_send(
                    CommandSet::FunctionSet(
                        DataWidth::Bit4,
                        state.get_line_mode(),
                        state.get_font(),
                    ),
                    delayer,
                    poll_interval_us,
                );
            }

            DataWidth::Bit8 => {
                sender.delay_and_send(
                    CommandSet::FunctionSet(
                        DataWidth::Bit8,
                        state.get_line_mode(),
                        state.get_font(),
                    ),
                    delayer,
                    poll_interval_us.max(40_000),
                );

                sender.delay_and_send(
                    CommandSet::FunctionSet(
                        DataWidth::Bit8,
                        state.get_line_mode(),
                        state.get_font(),
                    ),
                    delayer,
                    poll_interval_us,
                );
            }
        }

        sender.wait_and_send(
            CommandSet::DisplayOnOff {
                display: state.get_display_state(),
                cursor: state.get_cursor_state(),
                cursor_blink: state.get_cursor_blink(),
            },
            delayer,
            poll_interval_us,
        );

        sender.wait_and_send(CommandSet::ClearDisplay, delayer, poll_interval_us);

        sender.wait_and_send(
            CommandSet::EntryModeSet(state.get_direction(), state.get_shift_type()),
            delayer,
            poll_interval_us,
        );

        // set backlight after LCD init
        sender.set_actual_backlight(state.get_backlight());

        Lcd {
            sender,
            delayer,
            state,
            poll_interval_us,
        }
    }
}