thermostat/src/pins.rs

335 lines
9.8 KiB
Rust
Raw Normal View History

use stm32f4xx_hal::{
2020-04-11 03:05:05 +08:00
adc::Adc,
gpio::{
AF5, Alternate, AlternateOD, Analog, Floating, Input,
2020-03-13 00:26:14 +08:00
gpioa::*,
gpiob::*,
gpioc::*,
gpioe::*,
2020-03-13 02:24:57 +08:00
gpiof::*,
2020-03-13 00:26:14 +08:00
gpiog::*,
GpioExt,
Output, PushPull,
},
hal::{self, blocking::spi::Transfer, digital::v2::OutputPin},
i2c::I2c,
2020-09-11 05:17:31 +08:00
otg_fs::USB,
rcc::Clocks,
2020-03-13 00:26:14 +08:00
pwm::{self, PwmChannels},
spi::{Spi, NoMiso, TransferModeNormal},
pac::{
2020-09-11 05:17:31 +08:00
ADC1,
GPIOA, GPIOB, GPIOC, GPIOD, GPIOE, GPIOF, GPIOG,
I2C1,
2020-09-11 05:17:31 +08:00
OTG_FS_GLOBAL, OTG_FS_DEVICE, OTG_FS_PWRCLK,
SPI2, SPI4, SPI5,
TIM1, TIM3,
},
timer::Timer,
2020-03-20 05:21:17 +08:00
time::U32Ext,
2020-03-09 07:27:35 +08:00
};
use eeprom24x::{self, Eeprom24x};
2020-09-04 03:38:56 +08:00
use stm32_eth::EthPins;
2020-09-07 03:10:10 +08:00
use crate::{
channel::{Channel0, Channel1},
leds::Leds,
};
2020-03-09 07:27:35 +08:00
pub type Eeprom = Eeprom24x<
I2c<I2C1, (
PB8<AlternateOD<{ stm32f4xx_hal::gpio::AF4 }>>,
PB9<AlternateOD<{ stm32f4xx_hal::gpio::AF4 }>>
)>,
eeprom24x::page_size::B8,
eeprom24x::addr_size::OneByte
>;
2020-09-04 03:38:56 +08:00
pub type EthernetPins = EthPins<
PA1<Input<Floating>>,
PA7<Input<Floating>>,
PB11<Input<Floating>>,
PG13<Input<Floating>>,
PB13<Input<Floating>>,
PC4<Input<Floating>>,
PC5<Input<Floating>>,
>;
2020-05-13 05:16:57 +08:00
pub trait ChannelPins {
type DacSpi: Transfer<u8>;
type DacSync: OutputPin;
type Shdn: OutputPin;
2020-05-19 04:43:44 +08:00
type VRefPin;
2020-05-17 05:59:31 +08:00
type ItecPin;
2020-05-17 08:18:25 +08:00
type DacFeedbackPin;
2020-05-19 03:38:13 +08:00
type TecUMeasPin;
2020-05-13 05:16:57 +08:00
}
impl ChannelPins for Channel0 {
type DacSpi = Dac0Spi;
type DacSync = PE4<Output<PushPull>>;
type Shdn = PE10<Output<PushPull>>;
2020-05-19 04:43:44 +08:00
type VRefPin = PA0<Analog>;
2020-05-17 05:59:31 +08:00
type ItecPin = PA6<Analog>;
2020-05-17 08:18:25 +08:00
type DacFeedbackPin = PA4<Analog>;
2020-05-19 03:38:13 +08:00
type TecUMeasPin = PC2<Analog>;
2020-05-13 05:16:57 +08:00
}
impl ChannelPins for Channel1 {
type DacSpi = Dac1Spi;
type DacSync = PF6<Output<PushPull>>;
type Shdn = PE15<Output<PushPull>>;
2020-05-19 04:43:44 +08:00
type VRefPin = PA3<Analog>;
2020-05-17 05:59:31 +08:00
type ItecPin = PB0<Analog>;
2020-05-17 08:18:25 +08:00
type DacFeedbackPin = PA5<Analog>;
2020-05-19 03:38:13 +08:00
type TecUMeasPin = PC3<Analog>;
2020-05-13 05:16:57 +08:00
}
/// SPI peripheral used for communication with the ADC
pub type AdcSpi = Spi<SPI2, (PB10<Alternate<AF5>>, PB14<Alternate<AF5>>, PB15<Alternate<AF5>>), TransferModeNormal>;
2020-05-13 05:16:57 +08:00
pub type AdcNss = PB12<Output<PushPull>>;
type Dac0Spi = Spi<SPI4, (PE2<Alternate<AF5>>, NoMiso, PE6<Alternate<AF5>>), TransferModeNormal>;
type Dac1Spi = Spi<SPI5, (PF7<Alternate<AF5>>, NoMiso, PF9<Alternate<AF5>>), TransferModeNormal>;
2020-05-28 08:01:55 +08:00
pub type PinsAdc = Adc<ADC1>;
2020-03-13 02:24:57 +08:00
2020-05-13 05:16:57 +08:00
pub struct ChannelPinSet<C: ChannelPins> {
pub dac_spi: C::DacSpi,
pub dac_sync: C::DacSync,
pub shdn: C::Shdn,
2020-05-19 04:43:44 +08:00
pub vref_pin: C::VRefPin,
2020-05-17 05:59:31 +08:00
pub itec_pin: C::ItecPin,
2020-05-17 08:18:25 +08:00
pub dac_feedback_pin: C::DacFeedbackPin,
2020-05-19 03:38:13 +08:00
pub tec_u_meas_pin: C::TecUMeasPin,
2020-05-13 05:16:57 +08:00
}
pub struct Pins {
pub adc_spi: AdcSpi,
2020-05-13 05:16:57 +08:00
pub adc_nss: AdcNss,
2020-05-28 08:01:55 +08:00
pub pins_adc: PinsAdc,
2020-03-13 00:26:14 +08:00
pub pwm: PwmPins,
2020-05-13 05:16:57 +08:00
pub channel0: ChannelPinSet<Channel0>,
pub channel1: ChannelPinSet<Channel1>,
}
impl Pins {
/// Setup GPIO pins and configure MCU peripherals
2020-03-13 00:26:14 +08:00
pub fn setup(
clocks: Clocks,
2020-04-11 03:05:05 +08:00
tim1: TIM1, tim3: TIM3,
2020-09-07 03:10:10 +08:00
gpioa: GPIOA, gpiob: GPIOB, gpioc: GPIOC, gpiod: GPIOD, gpioe: GPIOE, gpiof: GPIOF, gpiog: GPIOG,
i2c1: I2C1,
2020-04-11 03:05:05 +08:00
spi2: SPI2, spi4: SPI4, spi5: SPI5,
2020-05-28 08:06:32 +08:00
adc1: ADC1,
2020-09-11 05:17:31 +08:00
otg_fs_global: OTG_FS_GLOBAL, otg_fs_device: OTG_FS_DEVICE, otg_fs_pwrclk: OTG_FS_PWRCLK,
) -> (Self, Leds, Eeprom, EthernetPins, USB) {
let gpioa = gpioa.split();
let gpiob = gpiob.split();
let gpioc = gpioc.split();
2020-09-07 03:10:10 +08:00
let gpiod = gpiod.split();
2020-03-13 00:26:14 +08:00
let gpioe = gpioe.split();
2020-03-13 02:24:57 +08:00
let gpiof = gpiof.split();
let gpiog = gpiog.split();
let adc_spi = Self::setup_spi_adc(clocks, spi2, gpiob.pb10, gpiob.pb14, gpiob.pb15);
let adc_nss = gpiob.pb12.into_push_pull_output();
2020-03-13 00:26:14 +08:00
2020-05-28 08:01:55 +08:00
let pins_adc = Adc::adc1(adc1, true, Default::default());
2020-05-19 03:38:13 +08:00
2020-03-21 07:33:31 +08:00
let pwm = PwmPins::setup(
clocks, tim1, tim3,
gpioc.pc6, gpioc.pc7,
gpioe.pe9, gpioe.pe11,
gpioe.pe13, gpioe.pe14
);
2020-03-13 02:24:57 +08:00
let (dac0_spi, dac0_sync) = Self::setup_dac0(
clocks, spi4,
gpioe.pe2, gpioe.pe4, gpioe.pe6
);
2020-03-21 07:33:31 +08:00
let mut shdn0 = gpioe.pe10.into_push_pull_output();
let _ = shdn0.set_low();
2020-05-19 04:43:44 +08:00
let vref0_pin = gpioa.pa0.into_analog();
2020-05-17 05:59:31 +08:00
let itec0_pin = gpioa.pa6.into_analog();
2020-05-17 08:18:25 +08:00
let dac_feedback0_pin = gpioa.pa4.into_analog();
2020-05-19 03:38:13 +08:00
let tec_u_meas0_pin = gpioc.pc2.into_analog();
2020-05-13 05:16:57 +08:00
let channel0 = ChannelPinSet {
dac_spi: dac0_spi,
dac_sync: dac0_sync,
shdn: shdn0,
2020-05-19 04:43:44 +08:00
vref_pin: vref0_pin,
2020-05-17 05:59:31 +08:00
itec_pin: itec0_pin,
2020-05-17 08:18:25 +08:00
dac_feedback_pin: dac_feedback0_pin,
2020-05-19 03:38:13 +08:00
tec_u_meas_pin: tec_u_meas0_pin,
2020-05-13 05:16:57 +08:00
};
2020-04-11 03:05:05 +08:00
2020-03-13 02:24:57 +08:00
let (dac1_spi, dac1_sync) = Self::setup_dac1(
clocks, spi5,
gpiof.pf7, gpiof.pf6, gpiof.pf9
);
2020-03-21 07:33:31 +08:00
let mut shdn1 = gpioe.pe15.into_push_pull_output();
let _ = shdn1.set_low();
2020-05-19 04:43:44 +08:00
let vref1_pin = gpioa.pa3.into_analog();
2020-05-17 05:59:31 +08:00
let itec1_pin = gpiob.pb0.into_analog();
2020-05-17 08:18:25 +08:00
let dac_feedback1_pin = gpioa.pa5.into_analog();
2020-05-19 03:38:13 +08:00
let tec_u_meas1_pin = gpioc.pc3.into_analog();
2020-05-13 05:16:57 +08:00
let channel1 = ChannelPinSet {
dac_spi: dac1_spi,
dac_sync: dac1_sync,
shdn: shdn1,
2020-05-19 04:43:44 +08:00
vref_pin: vref1_pin,
2020-05-17 05:59:31 +08:00
itec_pin: itec1_pin,
2020-05-17 08:18:25 +08:00
dac_feedback_pin: dac_feedback1_pin,
2020-05-19 03:38:13 +08:00
tec_u_meas_pin: tec_u_meas1_pin,
2020-05-13 05:16:57 +08:00
};
2020-03-13 00:26:14 +08:00
2020-09-04 03:38:56 +08:00
let pins = Pins {
2020-03-13 02:24:57 +08:00
adc_spi, adc_nss,
2020-05-28 08:06:32 +08:00
pins_adc,
2020-03-13 00:26:14 +08:00
pwm,
2020-05-13 05:16:57 +08:00
channel0,
channel1,
2020-09-04 03:38:56 +08:00
};
2020-09-07 03:10:10 +08:00
let leds = Leds::new(gpiod.pd9, gpiod.pd10.into_push_pull_output(), gpiod.pd11.into_push_pull_output());
let eeprom_scl = gpiob.pb8.into_alternate().set_open_drain();
let eeprom_sda = gpiob.pb9.into_alternate().set_open_drain();
let eeprom_i2c = I2c::new(i2c1, (eeprom_scl, eeprom_sda), 400.khz(), clocks);
let eeprom = Eeprom24x::new_24x02(eeprom_i2c, eeprom24x::SlaveAddr::default());
2020-09-04 03:38:56 +08:00
let eth_pins = EthPins {
ref_clk: gpioa.pa1,
crs: gpioa.pa7,
tx_en: gpiob.pb11,
tx_d0: gpiog.pg13,
tx_d1: gpiob.pb13,
rx_d0: gpioc.pc4,
rx_d1: gpioc.pc5,
};
2020-09-11 05:17:31 +08:00
let usb = USB {
usb_global: otg_fs_global,
usb_device: otg_fs_device,
usb_pwrclk: otg_fs_pwrclk,
pin_dm: gpioa.pa11.into_alternate(),
pin_dp: gpioa.pa12.into_alternate(),
2020-09-11 05:17:31 +08:00
hclk: clocks.hclk(),
};
(pins, leds, eeprom, eth_pins, usb)
}
/// Configure the GPIO pins for SPI operation, and initialize SPI
fn setup_spi_adc<M1, M2, M3>(
clocks: Clocks,
spi2: SPI2,
sck: PB10<M1>,
miso: PB14<M2>,
mosi: PB15<M3>,
) -> AdcSpi
{
let sck = sck.into_alternate();
let miso = miso.into_alternate();
let mosi = mosi.into_alternate();
Spi::new(
spi2,
(sck, miso, mosi),
crate::ad7172::SPI_MODE,
crate::ad7172::SPI_CLOCK,
clocks
)
}
2020-03-13 02:24:57 +08:00
fn setup_dac0<M1, M2, M3>(
clocks: Clocks, spi4: SPI4,
sclk: PE2<M1>, sync: PE4<M2>, sdin: PE6<M3>
2020-09-07 01:06:59 +08:00
) -> (Dac0Spi, <Channel0 as ChannelPins>::DacSync) {
let sclk = sclk.into_alternate();
let sdin = sdin.into_alternate();
let spi = Spi::new(
2020-03-13 02:24:57 +08:00
spi4,
(sclk, NoMiso {}, sdin),
2020-03-13 04:27:03 +08:00
crate::ad5680::SPI_MODE,
crate::ad5680::SPI_CLOCK,
2020-03-13 02:24:57 +08:00
clocks
);
let sync = sync.into_push_pull_output();
(spi, sync)
}
fn setup_dac1<M1, M2, M3>(
clocks: Clocks, spi5: SPI5,
sclk: PF7<M1>, sync: PF6<M2>, sdin: PF9<M3>
2020-09-07 01:06:59 +08:00
) -> (Dac1Spi, <Channel1 as ChannelPins>::DacSync) {
let sclk = sclk.into_alternate();
let sdin = sdin.into_alternate();
let spi = Spi::new(
2020-03-13 02:24:57 +08:00
spi5,
(sclk, NoMiso {}, sdin),
2020-03-13 04:27:03 +08:00
crate::ad5680::SPI_MODE,
crate::ad5680::SPI_CLOCK,
2020-03-13 02:24:57 +08:00
clocks
);
let sync = sync.into_push_pull_output();
(spi, sync)
}
2020-03-09 07:27:35 +08:00
}
2020-03-13 00:26:14 +08:00
pub struct PwmPins {
2020-03-20 01:34:57 +08:00
pub max_v0: PwmChannels<TIM3, pwm::C1>,
pub max_v1: PwmChannels<TIM3, pwm::C2>,
pub max_i_pos0: PwmChannels<TIM1, pwm::C1>,
pub max_i_pos1: PwmChannels<TIM1, pwm::C2>,
pub max_i_neg0: PwmChannels<TIM1, pwm::C3>,
pub max_i_neg1: PwmChannels<TIM1, pwm::C4>,
2020-03-13 00:26:14 +08:00
}
impl PwmPins {
fn setup<M1, M2, M3, M4, M5, M6>(
clocks: Clocks,
tim1: TIM1,
tim3: TIM3,
max_v0: PC6<M1>,
max_v1: PC7<M2>,
max_i_pos0: PE9<M3>,
max_i_pos1: PE11<M4>,
max_i_neg0: PE13<M5>,
max_i_neg1: PE14<M6>,
) -> PwmPins {
let freq = 20u32.khz();
2020-09-12 06:35:58 +08:00
fn init_pwm_pin<P: hal::PwmPin<Duty=u16>>(pin: &mut P) {
pin.set_duty(0);
pin.enable();
}
2020-03-13 00:26:14 +08:00
let channels = (
max_v0.into_alternate(),
max_v1.into_alternate(),
2020-03-13 00:26:14 +08:00
);
//let (mut max_v0, mut max_v1) = pwm::tim3(tim3, channels, clocks, freq);
let (mut max_v0, mut max_v1) = Timer::new(tim3, &clocks).pwm(channels, freq);
2020-09-12 06:35:58 +08:00
init_pwm_pin(&mut max_v0);
init_pwm_pin(&mut max_v1);
2020-03-13 00:26:14 +08:00
let channels = (
max_i_pos0.into_alternate(),
max_i_pos1.into_alternate(),
max_i_neg0.into_alternate(),
max_i_neg1.into_alternate(),
2020-03-13 00:26:14 +08:00
);
2020-09-12 06:35:58 +08:00
let (mut max_i_pos0, mut max_i_pos1, mut max_i_neg0, mut max_i_neg1) =
Timer::new(tim1, &clocks).pwm(channels, freq);
2020-09-12 06:35:58 +08:00
init_pwm_pin(&mut max_i_pos0);
init_pwm_pin(&mut max_i_neg0);
init_pwm_pin(&mut max_i_pos1);
init_pwm_pin(&mut max_i_neg1);
2020-03-13 00:26:14 +08:00
PwmPins {
max_v0, max_v1,
max_i_pos0, max_i_pos1,
max_i_neg0, max_i_neg1,
}
}
}