pi-frame-server/src/display.rs

use epd_waveshare::{epd7in3f::Epd7in3f, prelude::WaveshareDisplay};
use linux_embedded_hal::spidev::SpiModeFlags;
use linux_embedded_hal::spidev::SpidevOptions;
use linux_embedded_hal::{CdevPin, Delay, SpidevDevice};
use tracing::instrument;
use tracing::{debug, warn};

use anyhow::Result;
use linux_embedded_hal::gpio_cdev::{Chip, LineRequestFlags};

use crate::imageproc::DitheredImage;

pub trait EInkPanel {
    fn display(&mut self, buf: &DitheredImage) -> Result<()>;
}

pub struct Wrapper {
    spi: SpidevDevice,
    gpiochip: Chip,
    delay: Delay,
    panel: Epd7in3f<SpidevDevice, CdevPin, CdevPin, CdevPin, Delay>,
}

impl Wrapper {
    pub fn new() -> Result<Self> {
        let mut spi = SpidevDevice::open("/dev/spidev0.0")?;
        let spi_options = SpidevOptions::new()
            .bits_per_word(8)
            .max_speed_hz(10_000_000)
            .mode(SpiModeFlags::SPI_MODE_0)
            .build();
        spi.configure(&spi_options)?;

        let mut gpiochip = Chip::new("/dev/gpiochip0")?;
        let busy_pin = CdevPin::new(gpiochip.get_line(24)?.request(
            LineRequestFlags::INPUT,
            0,
            "frametool",
        )?)?;
        let dc_pin = CdevPin::new(gpiochip.get_line(25)?.request(
            LineRequestFlags::OUTPUT,
            0,
            "frametool",
        )?)?;
        let rst_pin = CdevPin::new(gpiochip.get_line(17)?.request(
            LineRequestFlags::OUTPUT,
            0,
            "frametool",
        )?)?;
        let mut delay = Delay {};
        let panel = Epd7in3f::new(&mut spi, busy_pin, dc_pin, rst_pin, &mut delay, None)?;

        Ok(Self {
            spi,
            gpiochip,
            delay,
            panel,
        })
    }
    pub fn test(&mut self) -> Result<()> {
        self.panel.show_7block(&mut self.spi, &mut self.delay)?;
        self.panel.sleep(&mut self.spi, &mut self.delay)?;
        Ok(())
    }
}

impl EInkPanel for Wrapper {
    #[instrument(skip_all)]
    fn display(&mut self, img: &DitheredImage) -> Result<()> {
        let buf = img.into_display_buffer();
        self.panel
            .update_and_display_frame(&mut self.spi, &buf, &mut self.delay)?;
        debug!("Finished updating frame");
        self.panel.sleep(&mut self.spi, &mut self.delay)?;
        debug!("Display entered sleep mode");
        Ok(())
    }
}

/// A Fake EInk display for testing purposes.
/// Saves the output as `display.bmp`
pub struct FakeEInk();
impl EInkPanel for FakeEInk {
    fn display(&mut self, img: &DitheredImage) -> Result<()> {
        warn!("Fake display was called: saving to display.bmp");
        img.into_rgbimage().save("display.bmp");

        Ok(())
    }
}