Something went wrong. Try again.
Firmware for the b-parasite board, but in Rust.
Something went wrong. Try again.
4.2 kB · 141 lines
Rust
at ci-fix
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142use embassy_futures::join::join;use embassy_nrf::{ Peri, gpio::{Level, Output, OutputDrive}, peripherals, pwm, saadc::{self, ChannelConfig, Saadc},};use embassy_time::Timer;use sachy_battery::BatteryDischargeProfile;use sachy_fmt::{info, unwrap};use static_cell::ConstStaticCell;
use crate::{ Irqs, constants::{DISCARGE_PROFILES, DRY_COEFFS, WET_COEFFS}, state::{ADC_MEASUREMENT, AdcMeasurements, START_MEASUREMENTS},};
const VREF: f32 = 3.6;
#[inline]fn calculate_polynomial(coeffs: &[f32; 3], val: f32) -> f32 { coeffs[0] + (coeffs[1] * val) + (coeffs[2] * (val * val))}
#[inline]fn calculate_soil_moisture(bat: f32, soil: i16) -> f32 { let dry = calculate_polynomial(&DRY_COEFFS, bat); let wet = calculate_polynomial(&WET_COEFFS, bat);
info!("WUH: dry {}, wet {}, soil {}", dry, wet, soil);
(((soil as f32) - dry) / (wet - dry)).clamp(0.0, 1.0)}
#[inline]fn calculate_lux(voltage: f32) -> f32 { const LUX_SUN: f32 = 10000.0; const CURRENT_SUN: f32 = 3.59e-3; const PHOTO_RESISTOR: f32 = 470.0;
let current = voltage / PHOTO_RESISTOR;
LUX_SUN * current / CURRENT_SUN}
#[inline]fn to_volts(sample: i16, reference: f32) -> f32 { ((sample.max(0) as f32) * reference) / 1024.0}
fn init_pwm<'scope>( pwm: Peri<'scope, peripherals::PWM0>, ch0: Peri<'scope, peripherals::P0_05>,) -> pwm::SimplePwm<'scope> { let mut config = pwm::SimpleConfig::default(); config.prescaler = pwm::Prescaler::Div1; let pwm_ctrl = pwm::SimplePwm::new_1ch(pwm, ch0, &config); pwm_ctrl.set_period(2_000_000);
pwm_ctrl}
fn init_saadc<'scope>( saadc: Peri<'scope, peripherals::SAADC>, light_pin: Peri<'scope, peripherals::P0_02>, soil_pin: Peri<'scope, peripherals::P0_03>,) -> Saadc<'scope, 3> { let light_config = ChannelConfig::single_ended(light_pin);
let mut soil_config = ChannelConfig::single_ended(soil_pin); soil_config.reference = saadc::Reference::VDD1_4;
let bat_config = ChannelConfig::single_ended(saadc::VddInput);
let mut saadc_config = saadc::Config::default(); saadc_config.resolution = saadc::Resolution::_10bit; saadc_config.oversample = saadc::Oversample::Over16x;
Saadc::new( saadc, Irqs, saadc_config, [soil_config, light_config, bat_config], )}
#[embassy_executor::task]pub async fn task( mut saadc: Peri<'static, peripherals::SAADC>, mut light_pin: Peri<'static, peripherals::P0_02>, mut soil_pin: Peri<'static, peripherals::P0_03>, mut photo_pin: Peri<'static, peripherals::P0_29>, mut pwm: Peri<'static, peripherals::PWM0>, mut pin5: Peri<'static, peripherals::P0_05>,) { static ADC_SAMPLE_BUFFER: ConstStaticCell<[i16; 3]> = ConstStaticCell::new([0; 3]);
let sample_buf = ADC_SAMPLE_BUFFER.take(); let mut measure = unwrap!(START_MEASUREMENTS.receiver());
loop { measure.changed().await;
let mut photo_ctrl = Output::new(photo_pin.reborrow(), Level::Low, OutputDrive::Standard); let mut pwm_ctrl = init_pwm(pwm.reborrow(), pin5.reborrow()); let mut saadc = init_saadc(saadc.reborrow(), light_pin.reborrow(), soil_pin.reborrow());
photo_ctrl.set_high(); pwm_ctrl.enable(); pwm_ctrl.set_duty(0, pwm::DutyCycle::normal(4));
join(saadc.calibrate(), Timer::after_millis(30)).await;
saadc.sample(sample_buf).await;
photo_ctrl.set_low(); pwm_ctrl.set_duty(0, pwm::DutyCycle::normal(0));
drop((pwm_ctrl, saadc, photo_ctrl));
let [soil, light, bat] = *sample_buf;
let bat_volt = to_volts(bat, VREF);
let (soil, light, bat_pct) = ( calculate_soil_moisture(bat_volt, soil), calculate_lux(to_volts(light, VREF)).max(0.0), BatteryDischargeProfile::calc_pct_from_profile_range( bat_volt, DISCARGE_PROFILES.iter(), ), );
let measurements = AdcMeasurements::new(bat_pct, bat_volt, soil, light);
info!("Soil {}, Light {}, Bat {}", soil, light, bat_pct);
ADC_MEASUREMENT.signal(measurements); }}