feat: dc out controller.

This commit is contained in:
Ivan Li 2022-08-30 22:36:22 +08:00
parent 84eb469b8b
commit 2a8e7db4b3
3 changed files with 307 additions and 30 deletions

248
src/dc_out_controller.rs Normal file
View File

@ -0,0 +1,248 @@
use std::sync::{Arc, Mutex};
use embedded_hal::digital::v2::{OutputPin, PinState};
use embedded_svc::event_bus::{EventBus, Postbox};
use esp_idf_hal::gpio::{Gpio6, Output};
use esp_idf_svc::eventloop::{
Background, EspBackgroundEventLoop, EspEventFetchData, EspEventLoop, EspEventPostData,
EspSubscription, EspTypedEventDeserializer, EspTypedEventSerializer, EspTypedEventSource, User,
};
use esp_idf_sys::c_types;
use log::{info, warn};
use serde_json::json;
use crate::{voltage_detection::VoltageDetectionWorker, VOLTAGE_EVENTLOOP};
const WAITING_OFF_SECONDS: u8 = 60;
pub static mut DC_OUT_STATE_EVENT_LOOP: Option<
EspEventLoop<esp_idf_svc::eventloop::User<Background>>,
> = None;
static mut INSTANCE: Option<DcOutController> = None;
#[derive(Debug, Clone, Copy, PartialEq)]
pub enum DcOutStatus {
WaitingOn(u8),
On,
Off,
WaitingOff,
TurningOff(u8),
}
#[derive(Debug, Clone, Copy)]
pub struct DcOutControllerState {
pub status: DcOutStatus,
pub pin_state: PinState,
}
impl DcOutControllerState {
pub fn new() -> Self {
Self {
status: DcOutStatus::On,
pin_state: PinState::Low,
}
}
fn handle_adapter_down(&mut self) {
info!("status: {:?}", self);
match self.status {
DcOutStatus::On => {
self.status = DcOutStatus::WaitingOff;
}
DcOutStatus::WaitingOn(_) => {
self.status = DcOutStatus::Off;
}
DcOutStatus::TurningOff(seconds) => {
let seconds = seconds - 1;
if seconds > 0 {
self.status = DcOutStatus::TurningOff(seconds);
} else {
self.status = DcOutStatus::Off;
}
}
_ => {}
};
}
fn turn_off(&mut self) {
match self.status {
DcOutStatus::On => {
self.status = DcOutStatus::TurningOff(WAITING_OFF_SECONDS);
}
DcOutStatus::WaitingOff => {
self.status = DcOutStatus::TurningOff(WAITING_OFF_SECONDS);
}
DcOutStatus::WaitingOn(_) => {
self.status = DcOutStatus::Off;
}
DcOutStatus::TurningOff(seconds) => {
let seconds = seconds - 1;
if seconds > 0 {
self.status = DcOutStatus::TurningOff(seconds);
} else {
self.status = DcOutStatus::Off;
}
}
_ => {}
};
}
fn turn_on(&mut self) {
match self.status {
DcOutStatus::WaitingOff => {
self.status = DcOutStatus::On;
}
DcOutStatus::Off => {
self.status = DcOutStatus::WaitingOn(WAITING_OFF_SECONDS);
}
DcOutStatus::WaitingOn(seconds) => {
let seconds = seconds - 1;
if seconds > 0 {
self.status = DcOutStatus::WaitingOn(seconds);
} else {
self.status = DcOutStatus::On;
}
}
DcOutStatus::TurningOff(seconds) => {
let seconds = seconds - 1;
if seconds > 0 {
self.status = DcOutStatus::TurningOff(seconds);
} else {
self.status = DcOutStatus::On;
}
}
_ => {}
};
}
pub fn to_json(&self) -> String {
let status = match self.status {
DcOutStatus::WaitingOn(_) => "WaitingOn",
DcOutStatus::On => "On",
DcOutStatus::Off => "Off",
DcOutStatus::WaitingOff => "WaitingOff",
DcOutStatus::TurningOff(_) => "TurningOff",
};
let pin_state = match self.pin_state {
PinState::Low => "Low",
PinState::High => "High",
};
let seconds: i16 = match self.status {
DcOutStatus::WaitingOn(seconds) => seconds.into(),
DcOutStatus::TurningOff(seconds) => seconds.into(),
_ => -1,
};
json!({ "status": status, "pin_state": pin_state, "seconds": seconds })
.to_string()
}
}
impl EspTypedEventSource for DcOutControllerState {
fn source() -> *const c_types::c_char {
b"VOLTAGES\0".as_ptr() as *const _
}
}
impl EspTypedEventSerializer<DcOutControllerState> for DcOutControllerState {
fn serialize<R>(
event: &DcOutControllerState,
f: impl for<'a> FnOnce(&'a EspEventPostData) -> R,
) -> R {
f(&unsafe { EspEventPostData::new(Self::source(), Self::event_id(), event) })
}
}
impl EspTypedEventDeserializer<DcOutControllerState> for DcOutControllerState {
fn deserialize<R>(
data: &EspEventFetchData,
f: &mut impl for<'a> FnMut(&'a DcOutControllerState) -> R,
) -> R {
f(unsafe { data.as_payload() })
}
}
pub struct DcOutController {
pub state: DcOutControllerState,
voltage_subscription: Option<EspSubscription<User<Background>>>,
}
impl DcOutController {
fn new() -> Self {
Self {
state: DcOutControllerState::new(),
voltage_subscription: None,
}
}
pub fn get_instance() -> anyhow::Result<Self> {
// if let Some(instance) = unsafe { &mut INSTANCE } {
// anyhow::Ok(*instance)
// } else {
let instance = Self::new();
// unsafe { INSTANCE = Some(instance) };
match EspBackgroundEventLoop::new(&Default::default()) {
Ok(eventloop) => unsafe { DC_OUT_STATE_EVENT_LOOP = Some(eventloop) },
Err(err) => anyhow::bail!("Init Event Loop failed. {:?}", err),
}
anyhow::Ok(instance)
// }
}
pub fn watch(&mut self) -> anyhow::Result<()> {
let state = Arc::new(Mutex::new(self.state));
if let Some(event_loop) = unsafe { VOLTAGE_EVENTLOOP.as_mut() } {
let voltage_subscription = event_loop
.subscribe(
move |obj: &VoltageDetectionWorker| match state.lock().as_mut() {
Ok(state) => {
if obj.adapter_voltage < 1000 {
if obj.battery_voltage < 1000 {
state.turn_off();
} else {
state.handle_adapter_down();
}
} else {
state.turn_on();
}
let ref state = **state;
if let Err(err) = Self::output_ctl(*state) {
warn!("Put Control Pin State Failed. {}", err);
}
if let Some(event_loop) = unsafe { DC_OUT_STATE_EVENT_LOOP.as_mut() } {
if let Err(err) = event_loop.post(state, None) {
warn!("Post DC Out Status Failed. {}", err);
}
} else {
warn!("DC_OUT_STATE_EVENT_LOOP is None");
}
info!("status: {:?}", state);
}
Err(err) => warn!("Can not lock state. {}", err),
},
)
.map_err(|err| anyhow::anyhow!("Subscribe Voltage Failed. {}", err))?;
self.voltage_subscription = Some(voltage_subscription);
} else {
anyhow::bail!("Voltage Event Loop is None");
}
anyhow::Ok(())
}
fn output_ctl(mut state: DcOutControllerState) -> anyhow::Result<()> {
let mut pin = unsafe { Gpio6::<Output>::new() }
.into_output()
.map_err(|err| anyhow::anyhow!("Make Gpio6 Into output Failed. {}", err))?;
if DcOutStatus::Off == state.status && state.pin_state == PinState::Low {
pin.set_high()
.map_err(|err| anyhow::anyhow!("Set DC Output Control Pin High Failed. {}", err))?;
state.pin_state = PinState::High;
} else if DcOutStatus::On == state.status && state.pin_state == PinState::High {
pin.set_low()
.map_err(|err| anyhow::anyhow!("Set DC Output Control Pin Low Failed. {}", err))?;
state.pin_state = PinState::Low;
}
return anyhow::Ok(());
}
}

View File

@ -1,6 +1,6 @@
use embedded_svc::event_bus::EventBus; use embedded_svc::event_bus::EventBus;
use esp_idf_svc::eventloop::{Background, EspBackgroundEventLoop, EspEventLoop}; use esp_idf_svc::eventloop::{Background, EspBackgroundEventLoop, EspEventLoop};
use esp_idf_sys::{self as _, EspError}; use esp_idf_sys::{self as _};
use log::*; use log::*;
use std::{ use std::{
env, env,
@ -10,17 +10,22 @@ use std::{
mod beep; mod beep;
mod blink; mod blink;
mod dc_out_controller;
mod message_queue; mod message_queue;
mod time; mod time;
mod voltage_detection; mod voltage_detection;
mod wifi; mod wifi;
use crate::{beep::{Beep, ringtone}, voltage_detection::VoltageDetectionWorker}; use crate::{
beep::{ringtone, Beep},
dc_out_controller::{DcOutController, DcOutControllerState, DC_OUT_STATE_EVENT_LOOP},
voltage_detection::VoltageDetectionWorker,
};
use crate::{ use crate::{
message_queue::MessageQueue, time::Time, voltage_detection::VoltageDetection, wifi::Internet, message_queue::MessageQueue, time::Time, voltage_detection::VoltageDetection, wifi::Internet,
}; };
static mut EVENTLOOP: Option<EspEventLoop<esp_idf_svc::eventloop::User<Background>>> = None; static mut VOLTAGE_EVENTLOOP: Option<EspEventLoop<esp_idf_svc::eventloop::User<Background>>> = None;
fn main() { fn main() {
env::set_var("DEFMT_LOG", "trace"); env::set_var("DEFMT_LOG", "trace");
@ -47,7 +52,7 @@ fn main() {
info!("About to start a background event loop"); info!("About to start a background event loop");
match EspBackgroundEventLoop::new(&Default::default()) { match EspBackgroundEventLoop::new(&Default::default()) {
Ok(eventloop) => unsafe { EVENTLOOP = Some(eventloop) }, Ok(eventloop) => unsafe { VOLTAGE_EVENTLOOP = Some(eventloop) },
Err(err) => error!("Init Event Loop failed. {:?}", err), Err(err) => error!("Init Event Loop failed. {:?}", err),
}; };
@ -59,30 +64,6 @@ fn main() {
); );
blink.play(); blink.play();
}); });
let mut beep = Beep::new();
let _subscription;
if let Some(eventloop) = unsafe { EVENTLOOP.as_mut() } {
_subscription = eventloop
.subscribe(move |message: &VoltageDetectionWorker| {
info!("Event Loop Value");
if let Ok(json_str) = serde_json::to_string(&message) {
if let Err(err) = mq.publish("voltage", json_str.as_bytes()) {
warn!("Can not publish message to MQTT. {}", err);
}
}
if message.battery_voltage < 1000 {
beep.play(ringtone::BATTERY_LOW).expect("Can not beep.");
} else if message.adapter_voltage < 1000 {
beep.play(ringtone::ADAPTER_DOWN).expect("Can not beep.");
} else {
beep.play(ringtone::SILENCE).expect("Can not beep.");
}
})
.expect(" Listening Event Loop Failed");
}
let mut voltage_detection = VoltageDetection::new(); let mut voltage_detection = VoltageDetection::new();
@ -97,6 +78,54 @@ fn main() {
let mut time = Time::new(); let mut time = Time::new();
time.sync().unwrap(); time.sync().unwrap();
let mut beep = Beep::new();
// let _voltage_subscription;
// if let Some(eventloop) = unsafe { VOLTAGE_EVENTLOOP.as_mut() } {
// _voltage_subscription = eventloop
// .subscribe(move |message: &VoltageDetectionWorker| {
// if let Ok(json_str) = serde_json::to_string(&message) {
// if let Err(err) = mq.publish("voltage", json_str.as_bytes()) {
// warn!("Can not publish message to MQTT. {}", err);
// }
// }
// })
// .expect(" Listening Event Loop Failed");
// }
let mut dc_out_controller =
DcOutController::get_instance().expect("Can not get DcOutController instance");
dc_out_controller
.watch()
.expect("Can not watch for dc_out_controller");
let _dc_out_state_subscription;
if let Some(eventloop) = unsafe { DC_OUT_STATE_EVENT_LOOP.as_mut() } {
_dc_out_state_subscription = eventloop
.subscribe(move |message: &DcOutControllerState| {
info!("Event Loop Value");
match message.status {
dc_out_controller::DcOutStatus::WaitingOff => {
beep.play(ringtone::ADAPTER_DOWN).expect("Can not beep.")
}
dc_out_controller::DcOutStatus::WaitingOn(_) => {
beep.play(ringtone::BATTERY_LOW).expect("Can not beep.")
}
dc_out_controller::DcOutStatus::TurningOff(_) => {
beep.play(ringtone::SHUTDOWN).expect("Can not beep.")
}
_ => {beep.play(ringtone::SILENCE).expect("Can not beep.")}
}
if let Err(err) = mq.publish("dcOut", message.to_json().as_bytes()) {
warn!("Can not publish message to MQTT. {}", err);
}
})
.expect(" Listening Event Loop Failed");
} else {
error!("DC_OUT_STATE_EVENT_LOOP is None");
}
loop { loop {
sleep(Duration::from_millis(1000)); sleep(Duration::from_millis(1000));
} }

View File

@ -26,7 +26,7 @@ use esp_idf_sys::c_types;
use log::{info, warn}; use log::{info, warn};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use crate::EVENTLOOP; use crate::VOLTAGE_EVENTLOOP;
const ADAPTER_OFFSET: f32 = 12002f32 / 900f32; const ADAPTER_OFFSET: f32 = 12002f32 / 900f32;
const BATTERY_OFFSET: f32 = 12002f32 / 900f32; const BATTERY_OFFSET: f32 = 12002f32 / 900f32;
@ -64,7 +64,7 @@ impl VoltageDetection {
"Adapter: {},\tBattery: {},\t Output: {}", "Adapter: {},\tBattery: {},\t Output: {}",
worker.adapter_voltage, worker.battery_voltage, worker.output_voltage worker.adapter_voltage, worker.battery_voltage, worker.output_voltage
); );
if let Some(eventloop) = unsafe { EVENTLOOP.as_mut() } { if let Some(eventloop) = unsafe { VOLTAGE_EVENTLOOP.as_mut() } {
if let Err(err) = eventloop.post( if let Err(err) = eventloop.post(
&mut VoltageDetectionWorker { &mut VoltageDetectionWorker {
adapter_voltage: worker.adapter_voltage, adapter_voltage: worker.adapter_voltage,