2023-12-04 00:41:29 +01:00
|
|
|
//mod config;
|
2023-11-29 18:27:55 +01:00
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
use embedded_svc::wifi::{
|
2023-12-22 17:26:00 +01:00
|
|
|
AccessPointConfiguration, AuthMethod, ClientConfiguration, Configuration, AccessPointInfo,
|
2023-12-22 01:35:08 +01:00
|
|
|
};
|
2023-11-23 22:50:17 +01:00
|
|
|
use esp_idf_svc::eventloop::EspSystemEventLoop;
|
|
|
|
use esp_idf_svc::nvs::EspDefaultNvsPartition;
|
2023-12-22 01:35:08 +01:00
|
|
|
use esp_idf_svc::wifi::config::{ScanConfig, ScanType};
|
2023-12-22 17:26:00 +01:00
|
|
|
use esp_idf_svc::wifi::EspWifi;
|
2023-12-22 01:35:08 +01:00
|
|
|
use plant_ctrl2::sipo::ShiftRegister40;
|
2023-11-23 22:50:17 +01:00
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
use anyhow::anyhow;
|
2023-12-22 17:26:00 +01:00
|
|
|
use anyhow::{bail, Ok, Result};
|
2023-12-27 17:33:11 +01:00
|
|
|
use strum::EnumString;
|
2023-12-12 03:46:53 +01:00
|
|
|
use std::ffi::CString;
|
2023-12-04 00:41:29 +01:00
|
|
|
use std::fs::File;
|
2023-12-22 17:26:00 +01:00
|
|
|
use std::io::Read;
|
2023-12-12 03:46:53 +01:00
|
|
|
use std::path::Path;
|
|
|
|
use std::str::from_utf8;
|
2023-12-22 01:35:08 +01:00
|
|
|
use std::sync::{Mutex, Arc};
|
|
|
|
use std::time::Duration;
|
2023-11-23 22:50:17 +01:00
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
use chrono::{DateTime, NaiveDateTime, Utc};
|
2023-11-23 22:50:17 +01:00
|
|
|
use ds18b20::Ds18b20;
|
2023-12-23 01:59:00 +01:00
|
|
|
|
2023-11-23 22:50:17 +01:00
|
|
|
use embedded_hal::digital::v2::OutputPin;
|
2023-12-22 01:35:08 +01:00
|
|
|
use esp_idf_hal::adc::{attenuation, AdcChannelDriver, AdcDriver};
|
2023-12-22 17:26:00 +01:00
|
|
|
use esp_idf_hal::delay::Delay;
|
2023-12-22 01:35:08 +01:00
|
|
|
use esp_idf_hal::gpio::{AnyInputPin, Gpio39, Gpio4, Level, PinDriver};
|
|
|
|
use esp_idf_hal::pcnt::{
|
|
|
|
PcntChannel, PcntChannelConfig, PcntControlMode, PcntCountMode, PcntDriver, PinIndex,
|
|
|
|
};
|
|
|
|
use esp_idf_hal::prelude::Peripherals;
|
2023-11-23 22:50:17 +01:00
|
|
|
use esp_idf_hal::reset::ResetReason;
|
|
|
|
use esp_idf_svc::sntp::{self, SyncStatus};
|
|
|
|
use esp_idf_svc::systime::EspSystemTime;
|
2023-12-23 01:59:00 +01:00
|
|
|
use esp_idf_sys::{EspError, vTaskDelay};
|
2023-11-23 22:50:17 +01:00
|
|
|
use one_wire_bus::OneWire;
|
|
|
|
|
2023-12-27 17:33:11 +01:00
|
|
|
use crate::config::{self, WifiConfig, Config};
|
2023-12-04 00:41:29 +01:00
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
pub const PLANT_COUNT: usize = 8;
|
|
|
|
const PINS_PER_PLANT: usize = 5;
|
|
|
|
const PLANT_PUMP_OFFSET: usize = 0;
|
|
|
|
const PLANT_FAULT_OFFSET: usize = 1;
|
|
|
|
const PLANT_MOIST_PUMP_OFFSET: usize = 2;
|
|
|
|
const PLANT_MOIST_B_OFFSET: usize = 3;
|
|
|
|
const PLANT_MOIST_A_OFFSET: usize = 4;
|
2023-11-23 22:50:17 +01:00
|
|
|
|
2023-12-12 03:46:53 +01:00
|
|
|
const SPIFFS_PARTITION_NAME: &str = "storage";
|
|
|
|
const WIFI_CONFIG_FILE: &str = "/spiffs/wifi.cfg";
|
|
|
|
const CONFIG_FILE: &str = "/spiffs/config.cfg";
|
2023-11-23 22:50:17 +01:00
|
|
|
|
|
|
|
#[link_section = ".rtc.data"]
|
|
|
|
static mut LAST_WATERING_TIMESTAMP: [i64; PLANT_COUNT] = [0; PLANT_COUNT];
|
|
|
|
#[link_section = ".rtc.data"]
|
|
|
|
static mut CONSECUTIVE_WATERING_PLANT: [u32; PLANT_COUNT] = [0; PLANT_COUNT];
|
|
|
|
#[link_section = ".rtc.data"]
|
2023-12-22 01:35:08 +01:00
|
|
|
static mut LOW_VOLTAGE_DETECTED: bool = false;
|
2023-11-23 22:50:17 +01:00
|
|
|
|
|
|
|
pub struct BatteryState {
|
|
|
|
state_charge_percent: u8,
|
|
|
|
max_error_percent: u8,
|
|
|
|
remaining_milli_ampere_hour: u32,
|
|
|
|
max_milli_ampere_hour: u32,
|
2023-12-22 01:35:08 +01:00
|
|
|
design_milli_ampere_hour: u32,
|
2023-11-23 22:50:17 +01:00
|
|
|
voltage_milli_volt: u16,
|
|
|
|
average_current_milli_ampere: u16,
|
|
|
|
temperature_tenth_kelvin: u32,
|
|
|
|
average_time_to_empty_minute: u16,
|
|
|
|
average_time_to_full_minute: u16,
|
|
|
|
average_discharge_power_cycle_milli_watt: u16,
|
|
|
|
cycle_count: u16,
|
2023-12-22 01:35:08 +01:00
|
|
|
state_health_percent: u8,
|
2023-11-23 22:50:17 +01:00
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
pub struct FileSystemSizeInfo {
|
2023-12-12 03:46:53 +01:00
|
|
|
pub total_size: usize,
|
|
|
|
pub used_size: usize,
|
2023-12-22 01:35:08 +01:00
|
|
|
pub free_size: usize,
|
2023-12-12 03:46:53 +01:00
|
|
|
}
|
|
|
|
|
2023-12-27 17:33:11 +01:00
|
|
|
|
|
|
|
#[derive(strum::Display)]
|
|
|
|
pub enum ClearConfigType {
|
|
|
|
WifiConfig,
|
|
|
|
Config,
|
|
|
|
None
|
|
|
|
}
|
|
|
|
|
2023-11-29 18:27:55 +01:00
|
|
|
#[derive(Debug)]
|
2023-12-22 01:35:08 +01:00
|
|
|
pub enum Sensor {
|
2023-11-23 22:50:17 +01:00
|
|
|
A,
|
|
|
|
B,
|
2023-12-22 01:35:08 +01:00
|
|
|
PUMP,
|
2023-11-23 22:50:17 +01:00
|
|
|
}
|
2023-12-22 01:35:08 +01:00
|
|
|
pub trait PlantCtrlBoardInteraction {
|
2023-11-23 22:50:17 +01:00
|
|
|
fn time(&mut self) -> Result<chrono::DateTime<Utc>>;
|
2023-12-22 01:35:08 +01:00
|
|
|
fn wifi(&mut self, ssid: &str, password: Option<&str>, max_wait: u32) -> Result<()>;
|
|
|
|
fn sntp(&mut self, max_wait: u32) -> Result<chrono::DateTime<Utc>>;
|
2023-12-12 03:46:53 +01:00
|
|
|
fn mountFileSystem(&mut self) -> Result<()>;
|
|
|
|
fn fileSystemSize(&mut self) -> Result<FileSystemSizeInfo>;
|
2023-11-23 22:50:17 +01:00
|
|
|
|
|
|
|
fn battery_state(&mut self) -> Result<BatteryState>;
|
|
|
|
|
|
|
|
fn general_fault(&mut self, enable: bool);
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn is_day(&self) -> bool;
|
|
|
|
fn water_temperature_c(&mut self) -> Result<f32>;
|
|
|
|
fn tank_sensor_mv(&mut self) -> Result<u16>;
|
|
|
|
|
|
|
|
fn set_low_voltage_in_cycle(&mut self);
|
|
|
|
fn clear_low_voltage_in_cycle(&mut self);
|
2023-11-23 22:50:17 +01:00
|
|
|
fn low_voltage_in_cycle(&mut self) -> bool;
|
2023-12-22 01:35:08 +01:00
|
|
|
fn any_pump(&mut self, enabled: bool) -> Result<()>;
|
2023-11-23 22:50:17 +01:00
|
|
|
|
|
|
|
//keep state during deepsleep
|
2023-12-22 01:35:08 +01:00
|
|
|
fn light(&mut self, enable: bool) -> Result<()>;
|
2023-11-23 22:50:17 +01:00
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn measure_moisture_hz(&self, plant: usize, sensor: Sensor) -> Result<i32>;
|
|
|
|
fn pump(&self, plant: usize, enable: bool) -> Result<()>;
|
|
|
|
fn last_pump_time(&self, plant: usize) -> Result<chrono::DateTime<Utc>>;
|
|
|
|
fn store_last_pump_time(&mut self, plant: usize, time: chrono::DateTime<Utc>);
|
|
|
|
fn store_consecutive_pump_count(&mut self, plant: usize, count: u32);
|
|
|
|
fn consecutive_pump_count(&mut self, plant: usize) -> u32;
|
2023-11-23 22:50:17 +01:00
|
|
|
|
|
|
|
//keep state during deepsleep
|
2023-12-22 01:35:08 +01:00
|
|
|
fn fault(&self, plant: usize, enable: bool);
|
2023-11-29 18:27:55 +01:00
|
|
|
|
2023-12-12 03:46:53 +01:00
|
|
|
//config
|
|
|
|
fn is_config_reset(&mut self) -> bool;
|
2023-12-27 17:33:11 +01:00
|
|
|
fn remove_configs(&mut self) -> Result<ClearConfigType>;
|
2023-12-04 00:41:29 +01:00
|
|
|
fn get_config(&mut self) -> Result<config::Config>;
|
2023-12-27 17:33:11 +01:00
|
|
|
fn set_config(&mut self, wifi: &Config) -> Result<()>;
|
2023-12-12 03:46:53 +01:00
|
|
|
fn get_wifi(&mut self) -> Result<config::WifiConfig>;
|
|
|
|
fn set_wifi(&mut self, wifi: &WifiConfig) -> Result<()>;
|
2023-12-22 01:35:08 +01:00
|
|
|
fn wifi_ap(&mut self) -> Result<()>;
|
|
|
|
fn wifi_scan(&mut self) -> Result<Vec<AccessPointInfo>>;
|
2023-12-23 01:59:00 +01:00
|
|
|
fn test(&mut self) -> Result<()>;
|
2023-11-23 22:50:17 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
pub trait CreatePlantHal<'a> {
|
2023-12-22 01:35:08 +01:00
|
|
|
fn create() -> Result<Arc<Mutex<PlantCtrlBoard<'static>>>>;
|
2023-11-23 22:50:17 +01:00
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
pub struct PlantHal {}
|
|
|
|
|
|
|
|
impl CreatePlantHal<'_> for PlantHal {
|
2023-12-27 17:33:11 +01:00
|
|
|
fn create() -> Result<Mutex<PlantCtrlBoard<'static>>> {
|
2023-12-22 01:35:08 +01:00
|
|
|
let peripherals = Peripherals::take()?;
|
|
|
|
|
|
|
|
let mut clock = PinDriver::output(peripherals.pins.gpio21)?;
|
|
|
|
let mut latch = PinDriver::output(peripherals.pins.gpio22)?;
|
|
|
|
let mut data = PinDriver::output(peripherals.pins.gpio19)?;
|
|
|
|
|
2023-11-23 22:50:17 +01:00
|
|
|
let one_wire_pin = PinDriver::input_output_od(peripherals.pins.gpio4)?;
|
|
|
|
//TODO make to none if not possible to init
|
|
|
|
|
|
|
|
//init,reset rtc memory depending on cause
|
|
|
|
let reasons = ResetReason::get();
|
|
|
|
let reset_store = match reasons {
|
|
|
|
ResetReason::Software => false,
|
|
|
|
ResetReason::ExternalPin => false,
|
|
|
|
ResetReason::Watchdog => true,
|
|
|
|
ResetReason::Sdio => true,
|
|
|
|
ResetReason::Panic => true,
|
|
|
|
ResetReason::InterruptWatchdog => true,
|
|
|
|
ResetReason::PowerOn => true,
|
|
|
|
ResetReason::Unknown => true,
|
|
|
|
ResetReason::Brownout => true,
|
|
|
|
ResetReason::TaskWatchdog => true,
|
|
|
|
ResetReason::DeepSleep => false,
|
|
|
|
};
|
|
|
|
if reset_store {
|
|
|
|
println!("Clear and reinit RTC store");
|
|
|
|
unsafe {
|
|
|
|
LAST_WATERING_TIMESTAMP = [0; PLANT_COUNT];
|
|
|
|
CONSECUTIVE_WATERING_PLANT = [0; PLANT_COUNT];
|
|
|
|
LOW_VOLTAGE_DETECTED = false;
|
|
|
|
};
|
|
|
|
} else {
|
|
|
|
println!("Keeping RTC store");
|
|
|
|
}
|
|
|
|
|
|
|
|
let mut counter_unit1 = PcntDriver::new(
|
|
|
|
peripherals.pcnt0,
|
|
|
|
Some(peripherals.pins.gpio18),
|
|
|
|
Option::<AnyInputPin>::None,
|
|
|
|
Option::<AnyInputPin>::None,
|
|
|
|
Option::<AnyInputPin>::None,
|
|
|
|
)?;
|
|
|
|
|
2023-11-29 18:27:55 +01:00
|
|
|
println!("Channel config start");
|
|
|
|
|
2023-11-23 22:50:17 +01:00
|
|
|
counter_unit1.channel_config(
|
|
|
|
PcntChannel::Channel0,
|
|
|
|
PinIndex::Pin0,
|
|
|
|
PinIndex::Pin1,
|
|
|
|
&PcntChannelConfig {
|
|
|
|
lctrl_mode: PcntControlMode::Reverse,
|
|
|
|
hctrl_mode: PcntControlMode::Keep,
|
|
|
|
pos_mode: PcntCountMode::Decrement,
|
|
|
|
neg_mode: PcntCountMode::Increment,
|
|
|
|
counter_h_lim: i16::MAX,
|
|
|
|
counter_l_lim: 0,
|
|
|
|
},
|
|
|
|
)?;
|
|
|
|
|
2023-11-29 18:27:55 +01:00
|
|
|
println!("Setup filter");
|
|
|
|
|
2023-11-23 22:50:17 +01:00
|
|
|
//TODO validate filter value! currently max allowed value
|
|
|
|
counter_unit1.set_filter_value(1023)?;
|
|
|
|
counter_unit1.filter_enable()?;
|
|
|
|
|
2023-11-29 18:27:55 +01:00
|
|
|
println!("Wifi start");
|
2023-12-22 01:35:08 +01:00
|
|
|
|
2023-11-23 22:50:17 +01:00
|
|
|
let sys_loop = EspSystemEventLoop::take()?;
|
|
|
|
let nvs = EspDefaultNvsPartition::take()?;
|
2023-12-22 01:35:08 +01:00
|
|
|
let wifi_driver = EspWifi::new(peripherals.modem, sys_loop, Some(nvs))?;
|
2023-11-23 22:50:17 +01:00
|
|
|
|
2023-11-29 18:27:55 +01:00
|
|
|
let shift_register = ShiftRegister40::new(clock, latch, data);
|
|
|
|
let last_watering_timestamp = Mutex::new(unsafe { LAST_WATERING_TIMESTAMP });
|
|
|
|
let consecutive_watering_plant = Mutex::new(unsafe { CONSECUTIVE_WATERING_PLANT });
|
|
|
|
let low_voltage_detected = Mutex::new(unsafe { LOW_VOLTAGE_DETECTED });
|
2023-12-27 17:33:11 +01:00
|
|
|
let tank_driver = AdcDriver::new(peripherals.adc1, &esp_idf_hal::adc::config::Config::new())?;
|
2023-12-22 01:35:08 +01:00
|
|
|
let tank_channel: AdcChannelDriver<'_, { attenuation::DB_11 }, Gpio39> =
|
|
|
|
AdcChannelDriver::new(peripherals.pins.gpio39)?;
|
2023-11-29 18:27:55 +01:00
|
|
|
let solar_is_day = PinDriver::input(peripherals.pins.gpio25)?;
|
2023-12-12 03:46:53 +01:00
|
|
|
let boot_button = PinDriver::input(peripherals.pins.gpio0)?;
|
2023-11-29 18:27:55 +01:00
|
|
|
let light = PinDriver::output(peripherals.pins.gpio26)?;
|
|
|
|
let main_pump = PinDriver::output(peripherals.pins.gpio23)?;
|
|
|
|
let tank_power = PinDriver::output(peripherals.pins.gpio27)?;
|
|
|
|
let general_fault = PinDriver::output(peripherals.pins.gpio13)?;
|
2023-12-22 01:35:08 +01:00
|
|
|
let one_wire_bus = OneWire::new(one_wire_pin)
|
|
|
|
.map_err(|err| -> anyhow::Error { anyhow!("Missing attribute: {:?}", err) })?;
|
2023-11-29 18:27:55 +01:00
|
|
|
|
|
|
|
println!("After stuff");
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
|
2023-12-27 17:33:11 +01:00
|
|
|
let rv = Mutex::new(PlantCtrlBoard {
|
2023-12-23 01:59:00 +01:00
|
|
|
shift_register : shift_register,
|
2023-12-22 01:35:08 +01:00
|
|
|
last_watering_timestamp: last_watering_timestamp,
|
|
|
|
consecutive_watering_plant: consecutive_watering_plant,
|
|
|
|
low_voltage_detected: low_voltage_detected,
|
|
|
|
tank_driver: tank_driver,
|
2023-11-29 18:27:55 +01:00
|
|
|
tank_channel: tank_channel,
|
2023-12-22 01:35:08 +01:00
|
|
|
solar_is_day: solar_is_day,
|
|
|
|
boot_button: boot_button,
|
2023-11-29 18:27:55 +01:00
|
|
|
light: light,
|
|
|
|
main_pump: main_pump,
|
|
|
|
tank_power: tank_power,
|
|
|
|
general_fault: general_fault,
|
|
|
|
one_wire_bus: one_wire_bus,
|
2023-12-22 01:35:08 +01:00
|
|
|
signal_counter: counter_unit1,
|
|
|
|
wifi_driver: wifi_driver,
|
2023-12-27 17:33:11 +01:00
|
|
|
});
|
2023-12-22 01:35:08 +01:00
|
|
|
return Ok(rv);
|
2023-11-23 22:50:17 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
pub struct PlantCtrlBoard<'a> {
|
2023-12-23 01:59:00 +01:00
|
|
|
shift_register: ShiftRegister40<PinDriver<'a, esp_idf_hal::gpio::Gpio21, esp_idf_hal::gpio::Output>, PinDriver<'a, esp_idf_hal::gpio::Gpio22, esp_idf_hal::gpio::Output>, PinDriver<'a, esp_idf_hal::gpio::Gpio19, esp_idf_hal::gpio::Output>>,
|
2023-11-23 22:50:17 +01:00
|
|
|
consecutive_watering_plant: Mutex<[u32; PLANT_COUNT]>,
|
|
|
|
last_watering_timestamp: Mutex<[i64; PLANT_COUNT]>,
|
|
|
|
low_voltage_detected: Mutex<bool>,
|
|
|
|
tank_driver: AdcDriver<'a, esp_idf_hal::adc::ADC1>,
|
2023-12-22 01:35:08 +01:00
|
|
|
tank_channel: esp_idf_hal::adc::AdcChannelDriver<'a, { attenuation::DB_11 }, Gpio39>,
|
2023-11-23 22:50:17 +01:00
|
|
|
solar_is_day: PinDriver<'a, esp_idf_hal::gpio::Gpio25, esp_idf_hal::gpio::Input>,
|
2023-12-12 03:46:53 +01:00
|
|
|
boot_button: PinDriver<'a, esp_idf_hal::gpio::Gpio0, esp_idf_hal::gpio::Input>,
|
2023-11-23 22:50:17 +01:00
|
|
|
signal_counter: PcntDriver<'a>,
|
|
|
|
light: PinDriver<'a, esp_idf_hal::gpio::Gpio26, esp_idf_hal::gpio::Output>,
|
|
|
|
main_pump: PinDriver<'a, esp_idf_hal::gpio::Gpio23, esp_idf_hal::gpio::Output>,
|
|
|
|
tank_power: PinDriver<'a, esp_idf_hal::gpio::Gpio27, esp_idf_hal::gpio::Output>,
|
|
|
|
general_fault: PinDriver<'a, esp_idf_hal::gpio::Gpio13, esp_idf_hal::gpio::Output>,
|
2023-11-29 18:27:55 +01:00
|
|
|
pub wifi_driver: EspWifi<'a>,
|
2023-11-23 22:50:17 +01:00
|
|
|
one_wire_bus: OneWire<PinDriver<'a, Gpio4, esp_idf_hal::gpio::InputOutput>>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl PlantCtrlBoardInteraction for PlantCtrlBoard<'_> {
|
2023-12-22 01:35:08 +01:00
|
|
|
fn battery_state(&mut self) -> Result<BatteryState> {
|
2023-11-23 22:50:17 +01:00
|
|
|
todo!()
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn is_day(&self) -> bool {
|
2023-11-23 22:50:17 +01:00
|
|
|
return self.solar_is_day.get_level().into();
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn water_temperature_c(&mut self) -> Result<f32> {
|
2023-11-23 22:50:17 +01:00
|
|
|
let mut delay = Delay::new_default();
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
self.one_wire_bus
|
|
|
|
.reset(&mut delay)
|
|
|
|
.map_err(|err| -> anyhow::Error { anyhow!("Missing attribute: {:?}", err) })?;
|
2023-11-23 22:50:17 +01:00
|
|
|
let first = self.one_wire_bus.devices(false, &mut delay).next();
|
|
|
|
if first.is_none() {
|
|
|
|
bail!("Not found any one wire Ds18b20");
|
|
|
|
}
|
2023-12-22 01:35:08 +01:00
|
|
|
let device_address = first
|
|
|
|
.unwrap()
|
|
|
|
.map_err(|err| -> anyhow::Error { anyhow!("Missing attribute: {:?}", err) })?;
|
|
|
|
|
|
|
|
let water_temp_sensor = Ds18b20::new::<EspError>(device_address)
|
|
|
|
.map_err(|err| -> anyhow::Error { anyhow!("Missing attribute: {:?}", err) })?;
|
|
|
|
|
|
|
|
water_temp_sensor
|
|
|
|
.start_temp_measurement(&mut self.one_wire_bus, &mut delay)
|
|
|
|
.map_err(|err| -> anyhow::Error { anyhow!("Missing attribute: {:?}", err) })?;
|
|
|
|
ds18b20::Resolution::Bits12.delay_for_measurement_time(&mut delay);
|
|
|
|
let sensor_data = water_temp_sensor
|
|
|
|
.read_data(&mut self.one_wire_bus, &mut delay)
|
|
|
|
.map_err(|err| -> anyhow::Error { anyhow!("Missing attribute: {:?}", err) })?;
|
2023-11-23 22:50:17 +01:00
|
|
|
if sensor_data.temperature == 85_f32 {
|
|
|
|
bail!("Ds18b20 dummy temperature returned");
|
|
|
|
}
|
|
|
|
return Ok(sensor_data.temperature);
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn tank_sensor_mv(&mut self) -> Result<u16> {
|
2023-11-23 22:50:17 +01:00
|
|
|
let delay = Delay::new_default();
|
|
|
|
self.tank_power.set_high()?;
|
|
|
|
//let stabilize
|
|
|
|
delay.delay_ms(100);
|
|
|
|
let value = self.tank_driver.read(&mut self.tank_channel)?;
|
|
|
|
self.tank_power.set_low()?;
|
|
|
|
return Ok(value);
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn set_low_voltage_in_cycle(&mut self) {
|
2023-11-23 22:50:17 +01:00
|
|
|
*self.low_voltage_detected.get_mut().unwrap() = true;
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn clear_low_voltage_in_cycle(&mut self) {
|
2023-11-23 22:50:17 +01:00
|
|
|
*self.low_voltage_detected.get_mut().unwrap() = false;
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn light(&mut self, enable: bool) -> Result<()> {
|
2023-11-23 22:50:17 +01:00
|
|
|
self.light.set_state(enable.into())?;
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn pump(&self, plant: usize, enable: bool) -> Result<()> {
|
|
|
|
let index = plant * PINS_PER_PLANT + PLANT_PUMP_OFFSET;
|
2023-11-23 22:50:17 +01:00
|
|
|
//currently infailable error, keep for future as result anyway
|
2023-12-23 01:59:00 +01:00
|
|
|
self.shift_register.decompose()[index].set_state(enable.into()).unwrap();
|
2023-11-23 22:50:17 +01:00
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn last_pump_time(&self, plant: usize) -> Result<chrono::DateTime<Utc>> {
|
2023-11-23 22:50:17 +01:00
|
|
|
let ts = unsafe { LAST_WATERING_TIMESTAMP }[plant];
|
2023-12-22 01:35:08 +01:00
|
|
|
let timestamp = NaiveDateTime::from_timestamp_millis(ts)
|
|
|
|
.ok_or(anyhow!("could not convert timestamp"))?;
|
2023-11-23 22:50:17 +01:00
|
|
|
return Ok(DateTime::<Utc>::from_naive_utc_and_offset(timestamp, Utc));
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn store_last_pump_time(&mut self, plant: usize, time: chrono::DateTime<Utc>) {
|
2023-11-23 22:50:17 +01:00
|
|
|
self.last_watering_timestamp.get_mut().unwrap()[plant] = time.timestamp_millis();
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn store_consecutive_pump_count(&mut self, plant: usize, count: u32) {
|
2023-11-23 22:50:17 +01:00
|
|
|
self.consecutive_watering_plant.get_mut().unwrap()[plant] = count;
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn consecutive_pump_count(&mut self, plant: usize) -> u32 {
|
|
|
|
return self.consecutive_watering_plant.get_mut().unwrap()[plant];
|
2023-11-23 22:50:17 +01:00
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn fault(&self, plant: usize, enable: bool) {
|
|
|
|
let index = plant * PINS_PER_PLANT + PLANT_FAULT_OFFSET;
|
2023-12-23 01:59:00 +01:00
|
|
|
self.shift_register.decompose()[index].set_state(enable.into()).unwrap()
|
2023-11-23 22:50:17 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
fn low_voltage_in_cycle(&mut self) -> bool {
|
2023-12-22 01:35:08 +01:00
|
|
|
return *self.low_voltage_detected.get_mut().unwrap();
|
2023-11-23 22:50:17 +01:00
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn any_pump(&mut self, enable: bool) -> Result<()> {
|
2023-11-23 22:50:17 +01:00
|
|
|
return Ok(self.main_pump.set_state(enable.into()).unwrap());
|
|
|
|
}
|
|
|
|
|
|
|
|
fn time(&mut self) -> Result<chrono::DateTime<Utc>> {
|
2023-12-22 01:35:08 +01:00
|
|
|
let time = EspSystemTime {}.now().as_millis();
|
2023-11-23 22:50:17 +01:00
|
|
|
let smaller_time = time as i64;
|
2023-12-22 01:35:08 +01:00
|
|
|
let local_time = NaiveDateTime::from_timestamp_millis(smaller_time)
|
|
|
|
.ok_or(anyhow!("could not convert timestamp"))?;
|
2023-11-23 22:50:17 +01:00
|
|
|
return Ok(local_time.and_utc());
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn sntp(&mut self, max_wait_ms: u32) -> Result<chrono::DateTime<Utc>> {
|
2023-11-23 22:50:17 +01:00
|
|
|
let sntp = sntp::EspSntp::new_default()?;
|
|
|
|
let mut counter = 0;
|
2023-12-22 01:35:08 +01:00
|
|
|
while sntp.get_sync_status() != SyncStatus::Completed {
|
2023-11-23 22:50:17 +01:00
|
|
|
let delay = Delay::new_default();
|
|
|
|
delay.delay_ms(100);
|
|
|
|
counter += 100;
|
|
|
|
if counter > max_wait_ms {
|
|
|
|
bail!("Reached sntp timeout, aborting")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return self.time();
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn measure_moisture_hz(&self, plant: usize, sensor: Sensor) -> Result<i32> {
|
2023-11-23 22:50:17 +01:00
|
|
|
self.signal_counter.counter_pause()?;
|
|
|
|
self.signal_counter.counter_clear()?;
|
|
|
|
//
|
|
|
|
let offset = match sensor {
|
|
|
|
Sensor::A => PLANT_MOIST_A_OFFSET,
|
|
|
|
Sensor::B => PLANT_MOIST_B_OFFSET,
|
|
|
|
Sensor::PUMP => PLANT_MOIST_PUMP_OFFSET,
|
|
|
|
};
|
2023-12-22 01:35:08 +01:00
|
|
|
let index = plant * PINS_PER_PLANT + offset;
|
|
|
|
|
2023-11-23 22:50:17 +01:00
|
|
|
let delay = Delay::new_default();
|
|
|
|
let measurement = 100;
|
2023-12-22 01:35:08 +01:00
|
|
|
let factor = 1000 / 100;
|
2023-11-23 22:50:17 +01:00
|
|
|
|
2023-12-23 01:59:00 +01:00
|
|
|
self.shift_register.decompose()[index].set_high().unwrap();
|
2023-11-23 22:50:17 +01:00
|
|
|
//give some time to stabilize
|
|
|
|
delay.delay_ms(10);
|
|
|
|
self.signal_counter.counter_resume()?;
|
|
|
|
delay.delay_ms(measurement);
|
|
|
|
self.signal_counter.counter_pause()?;
|
2023-12-23 01:59:00 +01:00
|
|
|
self.shift_register.decompose()[index].set_low().unwrap();
|
2023-11-23 22:50:17 +01:00
|
|
|
let unscaled = self.signal_counter.get_counter_value()? as i32;
|
2023-12-22 01:35:08 +01:00
|
|
|
let hz = unscaled * factor;
|
2023-11-29 18:27:55 +01:00
|
|
|
println!("Measuring {:?} @ {} with {}", sensor, plant, hz);
|
|
|
|
return Ok(hz);
|
2023-11-23 22:50:17 +01:00
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn general_fault(&mut self, enable: bool) {
|
2023-11-23 22:50:17 +01:00
|
|
|
self.general_fault.set_state(enable.into()).unwrap();
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn wifi_ap(&mut self) -> Result<()> {
|
|
|
|
|
|
|
|
let apconfig = AccessPointConfiguration {
|
|
|
|
ssid: "PlantCtrl".into(),
|
|
|
|
auth_method: AuthMethod::None,
|
|
|
|
ssid_hidden: false,
|
|
|
|
..Default::default()
|
|
|
|
};
|
|
|
|
let clientconfig = ClientConfiguration::default();
|
|
|
|
self.wifi_driver.set_configuration(&Configuration::Mixed(clientconfig, apconfig))?;
|
|
|
|
self.wifi_driver.start()?;
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
|
|
|
fn wifi(&mut self, ssid: &str, password: Option<&str>, max_wait: u32) -> Result<()> {
|
|
|
|
match password {
|
2023-11-23 22:50:17 +01:00
|
|
|
Some(pw) => {
|
|
|
|
//TODO expect error due to invalid pw or similar! //call this during configuration and check if works, revert to config mode if not
|
2023-12-22 01:35:08 +01:00
|
|
|
self.wifi_driver.set_configuration(&Configuration::Client(
|
|
|
|
ClientConfiguration {
|
|
|
|
ssid: ssid.into(),
|
|
|
|
password: pw.into(),
|
|
|
|
..Default::default()
|
|
|
|
},
|
|
|
|
))?;
|
|
|
|
}
|
2023-11-23 22:50:17 +01:00
|
|
|
None => {
|
2023-12-22 01:35:08 +01:00
|
|
|
self.wifi_driver
|
|
|
|
.set_configuration(&Configuration::Client(ClientConfiguration {
|
|
|
|
ssid: ssid.into(),
|
|
|
|
auth_method: AuthMethod::None,
|
|
|
|
..Default::default()
|
|
|
|
}))
|
|
|
|
.unwrap();
|
|
|
|
}
|
2023-11-23 22:50:17 +01:00
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
self.wifi_driver.start()?;
|
|
|
|
self.wifi_driver.connect()?;
|
|
|
|
|
2023-11-23 22:50:17 +01:00
|
|
|
let delay = Delay::new_default();
|
|
|
|
let mut counter = 0_u32;
|
2023-12-22 01:35:08 +01:00
|
|
|
while !self.wifi_driver.is_connected()? {
|
2023-11-29 18:27:55 +01:00
|
|
|
println!("Waiting for station connection");
|
2023-11-23 22:50:17 +01:00
|
|
|
//TODO blink status?
|
|
|
|
delay.delay_ms(250);
|
|
|
|
counter += 250;
|
|
|
|
if counter > max_wait {
|
|
|
|
//ignore these errors, wifi will not be used this
|
|
|
|
self.wifi_driver.disconnect().unwrap_or(());
|
|
|
|
self.wifi_driver.stop().unwrap_or(());
|
|
|
|
bail!("Did not manage wifi connection within timeout");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
println!("Should be connected now");
|
2023-11-29 18:27:55 +01:00
|
|
|
|
|
|
|
while self.wifi_driver.is_up().unwrap() == false {
|
|
|
|
println!("Waiting for network being up");
|
|
|
|
delay.delay_ms(250);
|
|
|
|
counter += 250;
|
|
|
|
if counter > max_wait {
|
|
|
|
//ignore these errors, wifi will not be used this
|
|
|
|
self.wifi_driver.disconnect().unwrap_or(());
|
|
|
|
self.wifi_driver.stop().unwrap_or(());
|
|
|
|
bail!("Did not manage wifi connection within timeout");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
//update freertos registers ;)
|
2023-11-23 22:50:17 +01:00
|
|
|
let address = self.wifi_driver.sta_netif().get_ip_info().unwrap();
|
|
|
|
println!("IP info: {:?}", address);
|
|
|
|
return Ok(());
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn mountFileSystem(&mut self) -> Result<()> {
|
2023-12-12 03:46:53 +01:00
|
|
|
let base_path = CString::new("/spiffs")?;
|
|
|
|
let storage = CString::new(SPIFFS_PARTITION_NAME)?;
|
|
|
|
let conf = esp_idf_sys::esp_vfs_spiffs_conf_t {
|
|
|
|
base_path: base_path.as_ptr(),
|
|
|
|
partition_label: storage.as_ptr(),
|
|
|
|
max_files: 2,
|
|
|
|
format_if_mount_failed: true,
|
|
|
|
};
|
2023-12-04 00:41:29 +01:00
|
|
|
|
2023-12-12 03:46:53 +01:00
|
|
|
unsafe {
|
|
|
|
esp_idf_sys::esp!(esp_idf_sys::esp_vfs_spiffs_register(&conf))?;
|
|
|
|
Ok(())
|
|
|
|
}
|
2023-11-29 18:27:55 +01:00
|
|
|
}
|
|
|
|
|
2023-12-12 03:46:53 +01:00
|
|
|
fn fileSystemSize(&mut self) -> Result<FileSystemSizeInfo> {
|
|
|
|
let storage = CString::new(SPIFFS_PARTITION_NAME)?;
|
|
|
|
let mut total_size = 0;
|
|
|
|
let mut used_size = 0;
|
|
|
|
unsafe {
|
2023-12-22 01:35:08 +01:00
|
|
|
esp_idf_sys::esp!(esp_idf_sys::esp_spiffs_info(
|
|
|
|
storage.as_ptr(),
|
|
|
|
&mut total_size,
|
|
|
|
&mut used_size
|
|
|
|
))?;
|
2023-12-12 03:46:53 +01:00
|
|
|
}
|
2023-12-22 01:35:08 +01:00
|
|
|
return Ok(FileSystemSizeInfo {
|
|
|
|
total_size,
|
|
|
|
used_size,
|
|
|
|
free_size: total_size - used_size,
|
|
|
|
});
|
2023-12-12 03:46:53 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
fn is_config_reset(&mut self) -> bool {
|
|
|
|
return self.boot_button.get_level() == Level::Low;
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
|
2023-12-27 17:33:11 +01:00
|
|
|
|
|
|
|
fn remove_configs(&mut self) -> Result<ClearConfigType> {
|
2023-12-12 03:46:53 +01:00
|
|
|
let config = Path::new(CONFIG_FILE);
|
|
|
|
if config.exists() {
|
|
|
|
println!("Removing config");
|
2023-12-22 01:35:08 +01:00
|
|
|
std::fs::remove_file(config)?;
|
2023-12-27 17:33:11 +01:00
|
|
|
return Ok(ClearConfigType::Config);
|
2023-12-12 03:46:53 +01:00
|
|
|
}
|
2023-12-27 17:33:11 +01:00
|
|
|
|
|
|
|
let wifi_config = Path::new(WIFI_CONFIG_FILE);
|
|
|
|
if wifi_config.exists() {
|
|
|
|
println!("Removing wifi config");
|
|
|
|
std::fs::remove_file(wifi_config)?;
|
|
|
|
Ok(ClearConfigType::WifiConfig);
|
|
|
|
}
|
|
|
|
|
|
|
|
Ok((ClearConfigType::None));
|
|
|
|
|
|
|
|
|
2023-12-12 03:46:53 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
fn get_wifi(&mut self) -> Result<config::WifiConfig> {
|
|
|
|
let cfg = File::open(WIFI_CONFIG_FILE)?;
|
|
|
|
let config: WifiConfig = serde_json::from_reader(cfg)?;
|
|
|
|
return Ok(config);
|
|
|
|
}
|
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn set_wifi(&mut self, wifi: &WifiConfig) -> Result<()> {
|
2023-12-12 03:46:53 +01:00
|
|
|
let mut cfg = File::create(WIFI_CONFIG_FILE)?;
|
|
|
|
serde_json::to_writer(&mut cfg, &wifi)?;
|
|
|
|
println!("Wrote wifi config {}", wifi);
|
|
|
|
return Ok(());
|
|
|
|
}
|
|
|
|
|
|
|
|
fn get_config(&mut self) -> Result<config::Config> {
|
2023-12-27 17:33:11 +01:00
|
|
|
let cfg = File::open(CONFIG_FILE)?;
|
|
|
|
let config: Config = serde_json::from_reader(cfg)?;
|
|
|
|
return Ok(config);
|
|
|
|
}
|
2023-12-12 03:46:53 +01:00
|
|
|
|
2023-12-27 17:33:11 +01:00
|
|
|
fn set_config(&mut self, config: &Config) -> Result<()> {
|
|
|
|
let mut cfg = File::create(CONFIG_FILE)?;
|
|
|
|
serde_json::to_writer(&mut cfg, &config)?;
|
|
|
|
println!("Wrote config config {:?}", config);
|
|
|
|
return Ok(());
|
2023-12-12 03:46:53 +01:00
|
|
|
}
|
2023-11-23 22:50:17 +01:00
|
|
|
|
2023-12-22 01:35:08 +01:00
|
|
|
fn wifi_scan(&mut self) -> Result<Vec<AccessPointInfo>> {
|
2023-12-22 17:26:00 +01:00
|
|
|
//remove this parts
|
2023-12-22 01:35:08 +01:00
|
|
|
for i in 1..11 {
|
|
|
|
println!("Scanning channel {}", i);
|
|
|
|
self.wifi_driver.start_scan(&ScanConfig{
|
|
|
|
scan_type : ScanType::Passive(Duration::from_secs(1)),
|
|
|
|
show_hidden: false,
|
|
|
|
channel: Some(i),
|
|
|
|
..Default::default()
|
|
|
|
}, true)?;
|
|
|
|
let sr = self.wifi_driver.get_scan_result()?;
|
|
|
|
for r in sr.iter(){
|
|
|
|
println!("Found wifi {}", r.ssid);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-12-22 17:26:00 +01:00
|
|
|
self.wifi_driver.start_scan(&ScanConfig{
|
|
|
|
scan_type : ScanType::Passive(Duration::from_secs(1)),
|
|
|
|
show_hidden: false,
|
|
|
|
..Default::default()
|
|
|
|
}, true)?;
|
|
|
|
return Ok(self.wifi_driver.get_scan_result()?);
|
2023-12-22 01:35:08 +01:00
|
|
|
}
|
2023-12-23 01:59:00 +01:00
|
|
|
|
|
|
|
fn test(&mut self) -> Result<()> {
|
|
|
|
self.general_fault(true);
|
|
|
|
unsafe { vTaskDelay(100) };
|
|
|
|
self.general_fault(false);
|
|
|
|
unsafe { vTaskDelay(100) };
|
|
|
|
self.any_pump(true)?;
|
|
|
|
unsafe { vTaskDelay(500) };
|
|
|
|
self.any_pump(false)?;
|
|
|
|
unsafe { vTaskDelay(500) };
|
|
|
|
self.light(true)?;
|
|
|
|
unsafe { vTaskDelay(500) };
|
|
|
|
self.light(false)?;
|
|
|
|
unsafe { vTaskDelay(500) };
|
|
|
|
for i in 0 .. 8{
|
|
|
|
self.fault(i, true);
|
|
|
|
unsafe { vTaskDelay(500) };
|
|
|
|
self.fault(i, false);
|
|
|
|
unsafe { vTaskDelay(500) };
|
|
|
|
}
|
|
|
|
for i in 0 .. 8{
|
|
|
|
self.pump(i, true)?;
|
|
|
|
unsafe { vTaskDelay(500) };
|
|
|
|
self.pump(i, false)?;
|
|
|
|
unsafe { vTaskDelay(500) };
|
|
|
|
}
|
|
|
|
for i in 0 .. 8{
|
|
|
|
self.measure_moisture_hz(i, Sensor::A)?;
|
|
|
|
unsafe { vTaskDelay(500) };
|
|
|
|
}
|
|
|
|
for i in 0 .. 8{
|
|
|
|
self.measure_moisture_hz(i, Sensor::B)?;
|
|
|
|
unsafe { vTaskDelay(500) };
|
|
|
|
}
|
|
|
|
for i in 0 .. 8{
|
|
|
|
self.measure_moisture_hz(i, Sensor::PUMP)?;
|
|
|
|
unsafe { vTaskDelay(500) };
|
|
|
|
}
|
|
|
|
|
|
|
|
return Ok(());
|
|
|
|
}
|
2023-12-22 01:35:08 +01:00
|
|
|
}
|