Nareshkumar Rao
7 months ago
3 changed files with 252 additions and 64 deletions
@ -1,35 +1,127 @@ |
|||
use std::{error::Error, time::Duration}; |
|||
|
|||
use dht11::Dht11Sensor; |
|||
use linux_realtime::ThreadAttributes; |
|||
|
|||
mod dht11; |
|||
|
|||
fn main() -> Result<(), Box<dyn Error>> { |
|||
let mut sensor = Dht11Sensor::new(2)?; |
|||
let t = linux_realtime::JoinHandle::spawn( |
|||
ThreadAttributes { |
|||
stack_size: libc::PTHREAD_STACK_MIN, |
|||
scheduler_policy: linux_realtime::SchedulerPolicy::Fifo, |
|||
thread_priority: 99, |
|||
}, |
|||
move || -> Result<(), Box<dyn Error>> { |
|||
use std::{error::Error, io, thread, time::Duration}; |
|||
|
|||
use ads1x1x::{Ads1x1x, ChannelSelection, DynamicOneShot}; |
|||
use nb::block; |
|||
use rppal::{ |
|||
gpio::{Gpio, OutputPin}, |
|||
i2c::I2c, |
|||
}; |
|||
|
|||
type GenericResult<T> = Result<T, Box<dyn Error>>; |
|||
|
|||
struct ProgramState { |
|||
water_pin: OutputPin, |
|||
light_pin: OutputPin, |
|||
fan_pin: OutputPin, |
|||
} |
|||
|
|||
fn setup_program_state() -> GenericResult<ProgramState> { |
|||
let mut water_pin = Gpio::new()?.get(22)?.into_output(); |
|||
let mut light_pin = Gpio::new()?.get(17)?.into_output(); |
|||
let mut fan_pin = Gpio::new()?.get(27)?.into_output(); |
|||
|
|||
water_pin.set_high(); |
|||
light_pin.set_high(); |
|||
fan_pin.set_high(); |
|||
|
|||
Ok(ProgramState { |
|||
water_pin, |
|||
light_pin, |
|||
fan_pin, |
|||
}) |
|||
} |
|||
|
|||
fn main() -> GenericResult<()> { |
|||
let mut program_state: ProgramState = setup_program_state()?; |
|||
|
|||
loop { |
|||
let mut rtclock = linux_realtime::Clock::new()?; |
|||
match sensor.read() { |
|||
Ok(data) => println!( |
|||
"Humidity: {}; Temperature: {}", |
|||
data.humidity, data.temperature |
|||
), |
|||
Err(msg) => println!("Error: {}", msg), |
|||
println!("Enter command: "); |
|||
let mut input_string = String::new(); |
|||
io::stdin().read_line(&mut input_string)?; |
|||
|
|||
match process_command(input_string, &mut program_state) { |
|||
Ok(to_exit) => { |
|||
if to_exit { |
|||
break; |
|||
} |
|||
for n in 1..10 { |
|||
rtclock.sleep(Duration::from_millis(500))?; |
|||
} |
|||
Err(err) => println!("Error: {}", err), |
|||
} |
|||
}, |
|||
)?; |
|||
_ = t.join(); |
|||
} |
|||
|
|||
Ok(()) |
|||
} |
|||
|
|||
fn process_command(command: String, program_state: &mut ProgramState) -> GenericResult<bool> { |
|||
let command = command |
|||
.strip_suffix('\n') |
|||
.ok_or("Couldn't strip whitespace")?; |
|||
let args = command.split(' ').collect::<Vec<_>>(); |
|||
|
|||
let main_command = *args.first().ok_or("No command found")?; |
|||
match main_command { |
|||
"water" => toggle_water(&args, program_state)?, |
|||
"fan" => toggle_fan(program_state)?, |
|||
"light" => toggle_light(program_state)?, |
|||
"ana" => analogue_command(&args, program_state)?, |
|||
"exit" => return Ok(true), |
|||
_ => return Err(format!("Main command '{}' invalid", main_command).into()), |
|||
} |
|||
|
|||
Ok(false) |
|||
} |
|||
|
|||
fn analogue_command(args: &[&str], program_state: &mut ProgramState) -> GenericResult<()> { |
|||
let pin = args.get(1).ok_or("No pin given")?; |
|||
let pin: u8 = pin.parse()?; |
|||
|
|||
let adc_target = rppal::i2c::I2c::new()?; |
|||
let mut adc_target = |
|||
Ads1x1x::new_ads1115(adc_target, ads1x1x::SlaveAddr::Alternative(false, false)); |
|||
let channel: ChannelSelection = match pin { |
|||
0 => ChannelSelection::SingleA0, |
|||
1 => ChannelSelection::SingleA1, |
|||
2 => ChannelSelection::SingleA2, |
|||
3 => ChannelSelection::SingleA3, |
|||
_ => ChannelSelection::SingleA0, |
|||
}; |
|||
loop { |
|||
let value = block!(adc_target.read(channel)).map_err(|e| format!("{:?}", e))?; |
|||
let value = value as f32; |
|||
let voltage = value / i16::MAX as f32 * 2.048; |
|||
let resistance = (3.3 / voltage - 1.) * 9_700.; |
|||
println!("Value: {}", value); |
|||
println!("resistance: {}", resistance); |
|||
let temp = 1. / ((1. / 298.15) + (1. / 3950. * f32::ln(resistance / 10_000.))) - 273.15; |
|||
println!("Temp: {}", temp); |
|||
//println!("Value of pin {} is: {}", pin, value);
|
|||
thread::sleep(Duration::from_millis(2000)); |
|||
} |
|||
Ok(()) |
|||
} |
|||
|
|||
fn toggle_water(args: &[&str], program_state: &mut ProgramState) -> GenericResult<()> { |
|||
if let Some(duration) = args.get(1) { |
|||
let duration: u64 = duration.parse()?; |
|||
println!("Turning on water for {} milliseconds", duration); |
|||
program_state.water_pin.set_low(); |
|||
thread::sleep(Duration::from_millis(duration)); |
|||
program_state.water_pin.set_high(); |
|||
} else { |
|||
program_state.water_pin.toggle(); |
|||
println!("Toggling water output"); |
|||
} |
|||
Ok(()) |
|||
} |
|||
|
|||
fn toggle_light(program_state: &mut ProgramState) -> GenericResult<()> { |
|||
program_state.light_pin.toggle(); |
|||
println!("Toggling light output"); |
|||
Ok(()) |
|||
} |
|||
|
|||
fn toggle_fan(program_state: &mut ProgramState) -> GenericResult<()> { |
|||
program_state.fan_pin.toggle(); |
|||
println!("Toggling fan output"); |
|||
Ok(()) |
|||
} |
|||
|
|||
Loading…
Reference in new issue