lay down netplay code

src/backend/core/mmio/PIF/Device.cpp

@@ -1,14 +1,17 @@
 #include <PIF.hpp>
 #include <MupenMovie.hpp>
+#include <PIF/Device.hpp>
+#include <Netplay.hpp>
 #include "log.hpp"
+#include <SDL2/SDL_keyboard.h>
 
 namespace n64 {
+JoybusDevice players[4]{};
+
 void PIF::InitDevices(SaveType saveType) {
-  joybusDevices[0].type = JOYBUS_CONTROLLER;
-  joybusDevices[0].accessoryType = ACCESSORY_MEMPACK;
-  for (int i = 1; i < 4; i++) { //TODO: make this configurable
-    joybusDevices[i].type = JOYBUS_NONE; 
-    joybusDevices[i].accessoryType = ACCESSORY_NONE;
+  for (int i = 0; i < 4; i++) { //TODO: make this configurable
+    joybusDevices[i].type = JOYBUS_CONTROLLER;
+    joybusDevices[i].accessoryType = ACCESSORY_MEMPACK;
   }
 
   if (saveType == SAVE_EEPROM_4k) {
@@ -21,6 +24,111 @@ void PIF::InitDevices(SaveType saveType) {
   joybusDevices[5].type = JOYBUS_NONE;
 }
 
+#define GET_BUTTON(gamecontroller, i) SDL_GameControllerGetButton(gamecontroller, i)
+#define GET_AXIS(gamecontroller, axis) SDL_GameControllerGetAxis(gamecontroller, axis)
+
+template <int port>
+void PIF::PollController() {
+  if(gamepadConnected) {
+    bool A = GET_BUTTON(gamepad, SDL_CONTROLLER_BUTTON_A);
+    bool B = GET_BUTTON(gamepad, SDL_CONTROLLER_BUTTON_X);
+    bool Z = GET_AXIS(gamepad, SDL_CONTROLLER_AXIS_TRIGGERLEFT) == SDL_JOYSTICK_AXIS_MAX;
+    bool START = GET_BUTTON(gamepad, SDL_CONTROLLER_BUTTON_START);
+    bool DUP = GET_BUTTON(gamepad, SDL_CONTROLLER_BUTTON_DPAD_UP);
+    bool DDOWN = GET_BUTTON(gamepad, SDL_CONTROLLER_BUTTON_DPAD_DOWN);
+    bool DLEFT = GET_BUTTON(gamepad, SDL_CONTROLLER_BUTTON_DPAD_LEFT);
+    bool DRIGHT = GET_BUTTON(gamepad, SDL_CONTROLLER_BUTTON_DPAD_RIGHT);
+    bool L = GET_BUTTON(gamepad, SDL_CONTROLLER_BUTTON_LEFTSHOULDER);
+    bool R = GET_BUTTON(gamepad, SDL_CONTROLLER_BUTTON_RIGHTSHOULDER);
+    bool CUP = GET_AXIS(gamepad, SDL_CONTROLLER_AXIS_RIGHTY) <= -127;
+    bool CDOWN = GET_AXIS(gamepad, SDL_CONTROLLER_AXIS_RIGHTY) >= 127;
+    bool CLEFT = GET_AXIS(gamepad, SDL_CONTROLLER_AXIS_RIGHTX) <= -127;
+    bool CRIGHT = GET_AXIS(gamepad, SDL_CONTROLLER_AXIS_RIGHTX) >= 127;
+
+    players[port].controller.a = A;
+    players[port].controller.b = B;
+    players[port].controller.z = Z;
+    players[port].controller.start = START;
+    players[port].controller.dp_up = DUP;
+    players[port].controller.dp_down = DDOWN;
+    players[port].controller.dp_left = DLEFT;
+    players[port].controller.dp_right = DRIGHT;
+    players[port].controller.joy_reset = L && R && START;
+    players[port].controller.l = L;
+    players[port].controller.r = R;
+    players[port].controller.c_up = CUP;
+    players[port].controller.c_down = CDOWN;
+    players[port].controller.c_left = CLEFT;
+    players[port].controller.c_right = CRIGHT;
+
+    float xclamped = GET_AXIS(gamepad, SDL_CONTROLLER_AXIS_LEFTX);
+    if(xclamped < 0) {
+      xclamped /= float(std::abs(SDL_JOYSTICK_AXIS_MIN));
+    } else {
+      xclamped /= SDL_JOYSTICK_AXIS_MAX;
+    }
+
+    xclamped *= 86;
+
+    float yclamped = GET_AXIS(gamepad, SDL_CONTROLLER_AXIS_LEFTY);
+    if(yclamped < 0) {
+      yclamped /= float(std::abs(SDL_JOYSTICK_AXIS_MIN));
+    } else {
+      yclamped /= SDL_JOYSTICK_AXIS_MAX;
+    }
+
+    yclamped *= 86;
+
+    players[port].controller.joy_x = xclamped;
+    players[port].controller.joy_y = -yclamped;
+
+    if (players[port].controller.joy_reset) {
+      players[port].controller.start = false;
+      players[port].controller.joy_x = 0;
+      players[port].controller.joy_y = 0;
+    }
+  } else {
+    const uint8_t* state = SDL_GetKeyboardState(nullptr);
+    players[port].controller.a = state[SDL_SCANCODE_X];
+    players[port].controller.b = state[SDL_SCANCODE_C];
+    players[port].controller.z = state[SDL_SCANCODE_Z];
+    players[port].controller.start = state[SDL_SCANCODE_RETURN];
+    players[port].controller.dp_up = state[SDL_SCANCODE_PAGEUP];
+    players[port].controller.dp_down = state[SDL_SCANCODE_PAGEDOWN];
+    players[port].controller.dp_left = state[SDL_SCANCODE_HOME];
+    players[port].controller.dp_right = state[SDL_SCANCODE_END];
+    players[port].controller.joy_reset = state[SDL_SCANCODE_RETURN] && state[SDL_SCANCODE_A] && state[SDL_SCANCODE_S];
+    players[port].controller.l = state[SDL_SCANCODE_A];
+    players[port].controller.r = state[SDL_SCANCODE_S];
+    players[port].controller.c_up = state[SDL_SCANCODE_I];
+    players[port].controller.c_down = state[SDL_SCANCODE_K];
+    players[port].controller.c_left = state[SDL_SCANCODE_J];
+    players[port].controller.c_right = state[SDL_SCANCODE_L];
+
+    s16 xaxis = 0, yaxis = 0;
+    if (state[SDL_SCANCODE_LEFT]) {
+      xaxis = -86;
+    } else if (state[SDL_SCANCODE_RIGHT]) {
+      xaxis = 86;
+    }
+
+    if (state[SDL_SCANCODE_DOWN]) {
+      yaxis = -86;
+    } else if (state[SDL_SCANCODE_UP]) {
+      yaxis = 86;
+    }
+
+    players[port].controller.joy_x = xaxis;
+    players[port].controller.joy_y = yaxis;
+
+    if (players[port].controller.joy_reset) {
+      players[port].controller.start = false;
+      players[port].controller.joy_x = 0;
+      players[port].controller.joy_y = 0;
+    }
+  }
+}
+
 void PIF::ControllerID(u8 *res) const {
   if (channel < 6) {
     switch (joybusDevices[channel].type) {