marc/ddgo-pnp-controller
1
0
Fork 0
mirror of https://github.com/marcriera/ddgo-pnp-controller.git synced 2025-04-11 06:29:29 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add support for Train Mascon

Browse source
This commit is contained in:
Marc Riera 2023-04-13 20:22:12 +02:00
parent a77c2d5b6d
commit 77ad46ea5d
3 changed files with 80 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/controller/emulated.rs
View file

@ -10,6 +10,7 @@ mod dgoc44u;
mod tcpp20009;
mod tcpp20011;
mod sotp031201_p4b7;
mod sotp031201_p5b5;
mod vok00106;
const FFS_MOUNT: &str = "/tmp/ffs";
@ -23,6 +24,7 @@ pub enum ControllerModel {
TCPP20009,
TCPP20011,
SOTP031201P4B7,
SOTP031201P5B5,
VOK00106,
}
@ -57,6 +59,11 @@ pub fn set_model(state: &ControllerState) -> Option<ControllerModel> {
init_gadget(&sotp031201_p4b7::DEVICE_DESCRIPTOR, &sotp031201_p4b7::DESCRIPTORS, &sotp031201_p4b7::STRINGS);
return Some(ControllerModel::SOTP031201P4B7);
}
else if state.button_c && state.power == 2 {
println!("Selected controller SOTP-031201 (P5/B5 mode).");
init_gadget(&sotp031201_p5b5::DEVICE_DESCRIPTOR, &sotp031201_p5b5::DESCRIPTORS, &sotp031201_p5b5::STRINGS);
return Some(ControllerModel::SOTP031201P5B5);
}
/* else if state.button_a {
println!("Selected controller VOK-00106.");
return Some(ControllerModel::VOK00106);
@ -81,6 +88,9 @@ pub fn set_state(state: &mut ControllerState, model: &ControllerModel) {
ControllerModel::SOTP031201P4B7 => {
sotp031201_p4b7::update_gadget(state);
}
ControllerModel::SOTP031201P5B5 => {
sotp031201_p5b5::update_gadget(state);
}
ControllerModel::VOK00106 => {
vok00106::update_gadget(state);
}

2
src/controller/emulated/sotp031201_p4b7.rs
View file

@ -62,7 +62,7 @@ pub fn update_gadget(state: &mut ControllerState) {
if state.button_select { buttons2.insert(Buttons2::SELECT) }
// Assemble data and send it to endpoint
let data = [0x0, handle, buttons1.bits, buttons2.bits];
let data = [0x1, handle, buttons1.bits, buttons2.bits];
if let Ok(mut file) = File::create(ENDPOINT1) {
file.write(&data).ok();
}

69
src/controller/emulated/sotp031201_p5b5.rs Normal file
View file

@ -0,0 +1,69 @@
use std::fs::File;
use std::io::Write;
use bitflags::bitflags;
use crate::controller::physical::ControllerState;
use crate::controller::emulated::{DeviceDescriptor, ENDPOINT1};
pub const DESCRIPTORS: [u8; 48] = [0x01, 0x00, 0x00, 0x00, 0x30, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00,
0x09, 0x04, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00,
0x07, 0x05, 0x81, 0x03, 0x08, 0x00, 0x14,
0x09, 0x04, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00,
0x07, 0x05, 0x81, 0x03, 0x08, 0x00, 0x14];
pub const STRINGS: [u8; 16] = [0x02, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00];
pub const DEVICE_DESCRIPTOR: DeviceDescriptor = DeviceDescriptor{b_device_class: 0x00, b_device_sub_class: 0x0, id_vendor: 0x1C06, id_product: 0x77A7, i_manufacturer: "TAITO", i_product: "Densha de Go! Plug & Play (MTC P5/B5 mode)", i_serial_number: "SOTP-031201"};
const POWER_NOTCHES: [u8; 6] = [0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C];
const BRAKE_NOTCHES: [u8; 10] = [0x07, 0x06, 0x05, 0x04, 0x03, 0x02, 0x02, 0x02, 0x02, 0x01];
bitflags! {
struct Buttons1: u8 {
const NONE = 0;
const S = 1;
const D = 2;
const A = 4;
const A2 = 8;
const B = 16;
const C = 32;
}
struct Buttons2: u8 {
const NONE = 0;
const START = 1;
const SELECT = 2;
const UP = 4;
const DOWN = 8;
const LEFT = 16;
const RIGHT = 32;
}
}
pub fn update_gadget(state: &mut ControllerState) {
// Calculate data for handles
let mut handle = POWER_NOTCHES[state.power as usize];
if state.brake > 0 {
handle = BRAKE_NOTCHES[state.brake as usize];
}
// Calculate data for buttons 1
let mut buttons1 = Buttons1::NONE;
if state.button_a { buttons1.insert(Buttons1::A) }
if state.button_b { buttons1.insert(Buttons1::B) }
if state.button_c { buttons1.insert(Buttons1::C) }
if state.button_d { buttons1.insert(Buttons1::D) }
// Calculate data for buttons 2
let mut buttons2 = Buttons2::NONE;
if state.button_up { buttons2.insert(Buttons2::UP) }
if state.button_down { buttons2.insert(Buttons2::DOWN) }
if state.button_left { buttons2.insert(Buttons2::LEFT) }
if state.button_right { buttons2.insert(Buttons2::RIGHT) }
if state.button_start { buttons2.insert(Buttons2::START) }
if state.button_select { buttons2.insert(Buttons2::SELECT) }
// Assemble data and send it to endpoint
let data = [0x1, handle, buttons1.bits, buttons2.bits];
if let Ok(mut file) = File::create(ENDPOINT1) {
file.write(&data).ok();
}
}