-use sdl2::JoystickSubsystem;
-use common::Nanoseconds;
+use common::Point;
+use {hashmap, point};
use common::Radians;
-use common::Point2D;
use sdl2::HapticSubsystem;
+use sdl2::JoystickSubsystem;
use sdl2::event::Event;
use sdl2::haptic::Haptic;
use sdl2::joystick::Joystick;
use std::cell::RefCell;
use std::collections::HashMap;
use std::rc::Rc;
+use time::{Duration, prelude::*};
#[derive(Debug, Default)]
pub struct Button {
- pub time_pressed: Nanoseconds,
- pub time_released: Nanoseconds,
+ id: u8,
+ pub time_pressed: Duration,
+ pub time_released: Duration,
pub is_pressed: bool,
pub was_pressed: bool,
pub toggle: bool,
}
impl Button {
- fn update(&mut self, device: &Joystick, dt: Nanoseconds, btn: u8) {
+ fn update(&mut self, device: &Joystick, dt: Duration) {
self.was_pressed = self.is_pressed;
- self.is_pressed = match device.button(btn as u32) {
+ self.is_pressed = match device.button(self.id as u32) {
Ok(true) => {
if !self.was_pressed {
- self.time_pressed = 0;
+ self.time_pressed = 0.seconds();
self.toggle = !self.toggle;
}
self.time_pressed += dt;
}
Ok(false) => {
if self.was_pressed {
- self.time_released = 0;
+ self.time_released = 0.seconds();
}
self.time_released += dt;
false
}
- Err(_) => { panic!("invalid button {}", btn) }
+ Err(_) => { panic!("invalid button {}", self.id) }
}
}
}
#[derive(Debug, Default)]
pub struct Axis {
+ id: u8,
pub val: f32,
}
impl Axis {
#[allow(dead_code)]
- fn update(&mut self, device: &Joystick, _dt: Nanoseconds, axis: u8) {
- self.val = match device.axis(axis as u32) {
+ fn update(&mut self, device: &Joystick, _dt: Duration) {
+ self.val = match device.axis(self.id as u32) {
Ok(val) => val as f32 / 32768.0,
- Err(_) => panic!("invalid axis {}", axis),
+ Err(_) => panic!("invalid axis {}", self.id),
}
}
}
#[derive(Debug, Default)]
pub struct Stick {
+ idx: u8,
+ idy: u8,
pub x: f32,
pub y: f32,
pub a: Radians,
- pub len: f32,
}
impl Stick {
- fn update(&mut self, device: &Joystick, _dt: Nanoseconds, x_axis: u8, y_axis: u8) {
- self.x = match device.axis(x_axis as u32) {
+ fn update(&mut self, device: &Joystick, _dt: Duration) {
+ self.x = match device.axis(self.idx as u32) {
Ok(val) => val as f32 / 32768.0,
- Err(_) => panic!("invalid x axis {}", x_axis),
+ Err(_) => panic!("invalid x axis {}", self.idx),
};
- self.y = match device.axis(y_axis as u32) {
+ self.y = match device.axis(self.idy as u32) {
Ok(val) => val as f32 / 32768.0,
- Err(_) => panic!("invalid y axis {}", y_axis),
+ Err(_) => panic!("invalid y axis {}", self.idy),
};
self.a = Radians(self.y.atan2(self.x) as f64);
- self.len = {
- let x = (self.x / self.y).abs().min(1.0);
- let y = (self.y / self.x).abs().min(1.0);
- (self.x.powi(2) + self.y.powi(2)).sqrt() / (x.powi(2) + y.powi(2)).sqrt()
- }
}
- #[inline(always)] #[allow(dead_code)] fn up(&self) -> bool { self.y > 0.99 }
- #[inline(always)] #[allow(dead_code)] fn down(&self) -> bool { self.y < -0.99 }
- #[inline(always)] #[allow(dead_code)] fn left(&self) -> bool { self.x < -0.99 }
- #[inline(always)] #[allow(dead_code)] fn right(&self) -> bool { self.x > 0.99 }
+ #[inline(always)] #[allow(dead_code)] pub fn up(&self) -> bool { self.y < -0.99 }
+ #[inline(always)] #[allow(dead_code)] pub fn down(&self) -> bool { self.y > 0.99 }
+ #[inline(always)] #[allow(dead_code)] pub fn left(&self) -> bool { self.x < -0.99 }
+ #[inline(always)] #[allow(dead_code)] pub fn right(&self) -> bool { self.x > 0.99 }
- pub fn to_point(&self) -> Point2D<f64> {
- Point2D {
- x: self.x as f64,
- y: self.y as f64,
- }
+ pub fn to_axis_point(&self) -> Point<f64> {
+ point!(self.x as f64, self.y as f64)
}
- pub fn to_adjusted_point(&self) -> Point2D<f64> {
- Point2D::from(self.a) * self.len as f64
+ pub fn to_point(&self) -> Point<f64> {
+ let p = point!(self.x as f64, self.y as f64);
+ if p.length() > 1.0 {
+ p.normalized()
+ } else {
+ p
+ }
}
}
-impl From<&Stick> for Point2D<f64> {
+impl From<&Stick> for Point<f64> {
fn from(item: &Stick) -> Self {
Self {
x: item.x as f64,
}
}
+#[derive(Eq, PartialEq, Hash)]
+enum DeviceControls {
+ AxisLX,
+ AxisLY,
+ AxisRX,
+ AxisRY,
+ AxisL2,
+ AxisR2,
+ ButtonA,
+ ButtonB,
+ ButtonY,
+ ButtonX,
+ ButtonSelect,
+ ButtonStart,
+ ButtonHome,
+ ButtonL3,
+ ButtonR3,
+ ButtonL1,
+ ButtonR1,
+ ButtonL2,
+ ButtonR2,
+ ButtonUp,
+ ButtonDown,
+ ButtonLeft,
+ ButtonRight,
+}
+use self::DeviceControls::*;
+
+#[derive(Eq, PartialEq, Hash)]
+enum ActionControls {
+ MovementX,
+ MovementY,
+ AimX,
+ AimY,
+ Jump,
+ Shoot,
+ Start,
+}
+use self::ActionControls::*;
+
//#[derive(Debug)]
pub struct Controller {
pub device: Joystick,
impl Controller {
pub fn new(device: Joystick, haptic: Option<Rc<RefCell<Haptic>>>) -> Self {
- Controller {
+ let action_map = get_action_mapping();
+ let device_map = get_device_mapping(&device.name());
+ let mut ctrl = Controller {
device,
haptic,
mov: Default::default(),
jump: Default::default(),
start: Default::default(),
shoot: Default::default(),
- }
+ };
+ ctrl.set_mapping(&action_map, &device_map);
+ ctrl
}
- pub fn update(&mut self, dt: Nanoseconds) {
- self.mov.update(&self.device, dt, 0, 1); // left stick
- self.aim.update(&self.device, dt, 3, 4); // right stick
- self.jump.update(&self.device, dt, 4); // left shoulder
- self.shoot.update(&self.device, dt, 5); // right shoulder
- self.start.update(&self.device, dt, 9); // start
+ fn set_mapping(&mut self, action: &HashMap<ActionControls, DeviceControls>, device: &HashMap<DeviceControls, u8>) {
+ self.mov.idx = *action.get(&MovementX).map(|i| device.get(i)).flatten().unwrap();
+ self.mov.idy = *action.get(&MovementY).map(|i| device.get(i)).flatten().unwrap();
+ self.aim.idx = *action.get(&AimX).map(|i| device.get(i)).flatten().unwrap();
+ self.aim.idy = *action.get(&AimY).map(|i| device.get(i)).flatten().unwrap();
+ self.jump.id = *action.get(&Jump).map(|i| device.get(i)).flatten().unwrap();
+ self.shoot.id = *action.get(&Shoot).map(|i| device.get(i)).flatten().unwrap();
+ self.start.id = *action.get(&Start).map(|i| device.get(i)).flatten().unwrap();
+ }
+
+ pub fn update(&mut self, dt: Duration) {
+ self.mov.update(&self.device, dt);
+ self.aim.update(&self.device, dt);
+ self.jump.update(&self.device, dt);
+ self.shoot.update(&self.device, dt);
+ self.start.update(&self.device, dt);
}
/// strength [0 - 1]
- pub fn rumble(&self, strength: f32, duration_ms: u32) {
+ pub fn rumble(&self, strength: f32, duration: Duration) {
if let Some(h) = &self.haptic {
- h.borrow_mut().rumble_play(strength, duration_ms);
+ h.borrow_mut().rumble_play(strength, duration.whole_milliseconds() as u32);
}
}
}
+fn get_action_mapping() -> HashMap<ActionControls, DeviceControls> {
+ hashmap!(
+ MovementX => AxisLX,
+ MovementY => AxisLY,
+ AimX => AxisRX,
+ AimY => AxisRY,
+ Jump => ButtonA,
+ Shoot => ButtonR1,
+ Start => ButtonStart
+ )
+}
+
+fn get_device_mapping(device_name: &str) -> HashMap<DeviceControls, u8> {
+ match device_name {
+ "Sony PLAYSTATION(R)3 Controller" => hashmap!(
+ AxisLX => 0,
+ AxisLY => 1,
+ AxisRX => 3,
+ AxisRY => 4,
+ AxisL2 => 2,
+ AxisR2 => 5,
+ ButtonA => 0,
+ ButtonB => 1,
+ ButtonY => 3,
+ ButtonX => 2,
+ ButtonSelect => 8,
+ ButtonStart => 9,
+ ButtonHome => 10,
+ ButtonL3 => 11,
+ ButtonR3 => 12,
+ ButtonL1 => 4,
+ ButtonR1 => 5,
+ ButtonL2 => 6,
+ ButtonR2 => 7,
+ ButtonUp => 13,
+ ButtonDown => 14,
+ ButtonLeft => 15,
+ ButtonRight => 16
+ ),
+ _ => panic!("No controller mapping for device '{}'", device_name)
+ }
+}
+
//#[derive(Debug)]
pub struct ControllerManager {
pub joystick: JoystickSubsystem,
}
}
- pub fn update(&mut self, dt: Nanoseconds) {
+ pub fn update(&mut self, dt: Duration) {
self.controllers.iter().for_each(|(_, v)| v.borrow_mut().update(dt));
}
match event {
Event::JoyDeviceAdded { which, .. } => { self.add_device(*which) }
Event::JoyDeviceRemoved { which, .. } => { self.remove_device(*which) }
- Event::JoyButtonDown { which, button_idx, .. } => { println!("device {} button {} down!", which, button_idx) }
- Event::JoyButtonUp { which, button_idx, .. } => { println!("device {} button {} up!", which, button_idx) }
- Event::JoyAxisMotion { which, axis_idx, .. } => { println!("device {} axis motion {}!", which, axis_idx) }
+ // Event::JoyButtonDown { which, button_idx, .. } => { println!("device {} button {} down!", which, button_idx) }
+ // Event::JoyButtonUp { which, button_idx, .. } => { println!("device {} button {} up!", which, button_idx) }
+ // Event::JoyAxisMotion { which, axis_idx, .. } => { println!("device {} axis motion {}!", which, axis_idx) }
_ => {}
}
}