Fix input reading bug, module subdivision

This commit is contained in:
Marc Riera 2023-03-02 00:20:31 +01:00
parent 52a4d6eb62
commit 4c18e70ce8
6 changed files with 59 additions and 34 deletions

View file

@ -0,0 +1,36 @@
use crate::state::ControllerState;
mod dgoc44u;
#[derive(PartialEq)]
pub enum ControllerModel {
NONE,
DGOC44U,
TYPE2,
}
pub fn set_model(state: &ControllerState) -> ControllerModel {
if state.button_right {
println!("Selected controller DGOC44-U.");
return ControllerModel::DGOC44U;
}
else if state.button_up {
println!("Selected controller TCPP-20009.");
return ControllerModel::TYPE2;
}
else {
println!("No controller selected.");
return ControllerModel::NONE;
}
}
pub fn set_state(state: &mut ControllerState, model: &ControllerModel) {
match model {
ControllerModel::DGOC44U => {
dgoc44u::update_gadget(state);
}
ControllerModel::TYPE2 => {
}
ControllerModel::NONE => (),
}
}

View file

@ -0,0 +1,13 @@
use crate::state::ControllerState;
const POWER_NOTCHES: [u8; 6] = [0x81, 0x6D, 0x54, 0x3F, 0x21, 0x00];
const BRAKE_NOTCHES: [u8; 10] = [0x79, 0x8A, 0x94, 0x9A, 0xA2, 0xA8, 0xAF, 0xB2, 0xB5, 0xB9];
pub fn update_gadget(state: &mut ControllerState) {
// Calculate data for handles
let power = POWER_NOTCHES[state.power as usize];
let brake = BRAKE_NOTCHES[state.brake as usize];
// Assemble data and send it to gadget
let data = [brake, power, 0, 0, 0, 0];
}

View file

@ -0,0 +1,81 @@
use std::fs::File;
use std::io::Write;
use std::io::Result;
use evdev::Device;
use evdev::Key;
use crate::state::ControllerState;
const USED_KEYS: [Key; 26] = [Key::KEY_0, Key::KEY_1, Key::KEY_2, Key::KEY_3, Key::KEY_4, Key::KEY_5,
Key::KEY_B, Key::KEY_C, Key::KEY_D, Key::KEY_E, Key::KEY_F, Key::KEY_G, Key::KEY_H, Key::KEY_I, Key::KEY_J, Key::KEY_P,
Key::KEY_SPACE, Key::KEY_ENTER, Key::KEY_A, Key::KEY_Z, Key::KEY_X, Key::KEY_S, Key::KEY_UP, Key::KEY_DOWN, Key::KEY_LEFT, Key::KEY_RIGHT];
pub fn init() -> Result<[Device; 2]> {
let d1 = Device::open("/dev/input/event1")?;
let d2 = Device::open("/dev/input/event2")?;
Ok([d1, d2])
}
pub fn get_state(state: &mut ControllerState, dev: &[Device; 2]) {
for d in dev {
if let Ok(key_vals) = d.get_key_state() {
for key in USED_KEYS {
if d.supported_keys().map_or(true, |k| k.contains(key)) {
read_input(state, key, key_vals.contains(key));
}
}
}
}
}
fn read_input(controller: &mut ControllerState, key: Key, value: bool) {
// Save input status to object for processing
match key{
Key::KEY_0=>if value {controller.power = 0},
Key::KEY_1=>if value {controller.power = 1},
Key::KEY_2=>if value {controller.power = 2},
Key::KEY_3=>if value {controller.power = 3},
Key::KEY_4=>if value {controller.power = 4},
Key::KEY_5=>if value {controller.power = 5},
Key::KEY_B=>if value {controller.brake = 0},
Key::KEY_C=>if value {controller.brake = 1},
Key::KEY_D=>if value {controller.brake = 2},
Key::KEY_E=>if value {controller.brake = 3},
Key::KEY_F=>if value {controller.brake = 4},
Key::KEY_G=>if value {controller.brake = 5},
Key::KEY_H=>if value {controller.brake = 6},
Key::KEY_I=>if value {controller.brake = 7},
Key::KEY_J=>if value {controller.brake = 8},
Key::KEY_P=>if value {controller.brake = 9},
Key::KEY_SPACE=>controller.button_select = value,
Key::KEY_ENTER=>controller.button_start = value,
Key::KEY_A=>controller.button_a = value,
Key::KEY_Z=>controller.button_b = value,
Key::KEY_X=>controller.button_c = value,
Key::KEY_S=>controller.button_d = value,
Key::KEY_UP=>controller.button_up = value,
Key::KEY_DOWN=>controller.button_down = value,
Key::KEY_LEFT=>controller.button_left = value,
Key::KEY_RIGHT=>controller.button_right = value,
_=>(),
}
}
pub fn set_lamp(status: bool) {
if let Ok(mut out) = File::create("/sys/class/leds/led2/brightness") {
out.write(if status {b"1"} else {b"0"}).ok();
}
/*else {
println!("WARNING: Could not set door lamp status!")
}*/
}
pub fn set_rumble(status: bool) {
if let Ok(mut out) = File::create("/sys/class/leds/led1/brightness") {
out.write(if status {b"1"} else {b"0"}).ok();
}
/*else {
println!("WARNING: Could not set rumble motor status!")
}*/
}