mirror of
https://github.com/marcriera/ddgo-pnp-controller.git
synced 2025-04-18 09:39:28 +02:00
97 lines
3.4 KiB
Rust
97 lines
3.4 KiB
Rust
use std::fs::File;
|
|
use std::io::Write;
|
|
use std::io::Result;
|
|
|
|
use evdev::Device;
|
|
use evdev::Key;
|
|
|
|
#[derive(Default)]
|
|
pub struct ControllerState {
|
|
pub power: u8,
|
|
pub brake: u8,
|
|
pub button_select: bool,
|
|
pub button_start: bool,
|
|
pub button_a: bool,
|
|
pub button_b: bool,
|
|
pub button_c: bool,
|
|
pub button_d: bool,
|
|
pub button_up: bool,
|
|
pub button_down: bool,
|
|
pub button_left: bool,
|
|
pub button_right: bool,
|
|
pub lamp: bool,
|
|
pub rumble: bool,
|
|
}
|
|
|
|
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!")
|
|
}*/
|
|
}
|