kirdy/src/laser_diode/current_sources.rs

use stm32f4xx_hal::{
    gpio::{gpioa::*, gpiob::*, gpiod::*, Alternate, Output, PushPull, Analog, PD9},
    hal::{blocking::spi::Write, digital::v2::OutputPin},
    pac::SPI2,
    spi::{NoMiso, Spi, TransferModeNormal},
};

use crate::device::sys_timer::sleep;
pub trait ChannelPins {
    type PdMonPin;
    type CurrentSourceLdoEn: OutputPin;
    type CurrentSourceShort: OutputPin;
    type Max5719Load: OutputPin;
    type Max5719Cs: OutputPin;
    type Max5719Spi: Write<u8>;
}

pub struct CurrentSourcePhyConstruct<C: ChannelPins> {
    pub max5719_spi: C::Max5719Spi,
    pub max5719_load: C::Max5719Load,
    pub max5719_cs: C::Max5719Cs,
    pub current_source_ldo_en: C::CurrentSourceLdoEn,
    pub current_source_short: C::CurrentSourceShort,
}
pub struct CurrentSourceSettings {
    pub output_current: f32,
}
pub struct CurrentSource<C: ChannelPins> {
    pub phy: CurrentSourcePhyConstruct<C>,
    pub setting: CurrentSourceSettings,
}

pub struct Channel0;

impl ChannelPins for Channel0 {
    type PdMonPin = PA3<Analog>;
    type CurrentSourceLdoEn = PD9<Output<PushPull>>;
    type CurrentSourceShort = PA4<Output<PushPull>>;
    type Max5719Load = PB14<Output<PushPull>>;
    type Max5719Cs = PD8<Output<PushPull>>;
    type Max5719Spi =
        Spi<SPI2, (PB10<Alternate<5>>, NoMiso, PB15<Alternate<5>>), TransferModeNormal>;
}

impl<C: ChannelPins> CurrentSource<C> {
    pub fn setup(&mut self) {
        let _ = self.phy.max5719_load.set_high();
        let _ = self.phy.max5719_cs.set_high();
        let _ = self.phy.current_source_ldo_en.set_high();
        sleep(10_u32);
        let _ = self.phy.current_source_short.set_high();
        sleep(10_u32);
    }

    pub fn set_current(
        &mut self,
        current: f32,
    ) -> Result<(), <<C as ChannelPins>::Max5719Spi as Write<u8>>::Error> {
        let _ = self.phy.max5719_load.set_high();
        let _ = self.phy.max5719_cs.set_low();
        self.setting.output_current = current * 10.0 / 0.75;
        let word = (((self.setting.output_current / 4.096) * 1048576.0) as u32) << 4;
        let mut buf = [
            ((word >> 16) & 0xFF) as u8,
            ((word >> 8) & 0xFF) as u8,
            ((word >> 0) & 0xFF) as u8,
        ];
        self.phy.max5719_spi.write(&mut buf)?;
        let _ = self.phy.max5719_cs.set_high();
        let _ = self.phy.max5719_load.set_low();
        Ok(())
    }
}