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 { |
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; |
||||
|
} |
||||
|
} |
||||
|
Err(err) => println!("Error: {}", err), |
||||
} |
} |
||||
for n in 1..10 { |
|
||||
rtclock.sleep(Duration::from_millis(500))?; |
|
||||
} |
} |
||||
|
|
||||
|
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(()) |
||||
} |
} |
||||
}, |
|
||||
)?; |
|
||||
_ = t.join(); |
|
||||
|
|
||||
|
fn toggle_fan(program_state: &mut ProgramState) -> GenericResult<()> { |
||||
|
program_state.fan_pin.toggle(); |
||||
|
println!("Toggling fan output"); |
||||
Ok(()) |
Ok(()) |
||||
} |
} |
||||
|
|||||
Loading…
Reference in new issue