shithub: cstory

ref: 2aeda93aebf5e15c489ccbb78554295e20272e0a
dir: /src/Backends/SDL2/Controller.cpp/

#include "../Controller.h"
#include "Controller.h"

#include <stddef.h>
#include <stdio.h>

#include "SDL.h"

#include "../../WindowsWrapper.h"

#define DEADZONE 10000;

static SDL_Joystick *joystick;
static int joystick_neutral_x;
static int joystick_neutral_y;

BOOL ControllerBackend_Init(void)
{
	SDL_InitSubSystem(SDL_INIT_JOYSTICK);

	return TRUE;
}

void ControllerBackend_Deinit(void)
{
	if (joystick != NULL)
	{
		SDL_JoystickClose(joystick);
		joystick = NULL;
	}

	SDL_QuitSubSystem(SDL_INIT_JOYSTICK);
}

BOOL ControllerBackend_GetJoystickStatus(JOYSTICK_STATUS *status)
{
	if (joystick == NULL)
		return FALSE;

	const size_t button_limit = sizeof(status->bButton) / sizeof(status->bButton);

	// Read axis
	const Sint16 joystick_x = SDL_JoystickGetAxis(joystick, 0);
	const Sint16 joystick_y = SDL_JoystickGetAxis(joystick, 1);

	status->bLeft = joystick_x < joystick_neutral_x - DEADZONE;
	status->bRight = joystick_x > joystick_neutral_x + DEADZONE;
	status->bUp = joystick_y < joystick_neutral_y - DEADZONE;
	status->bDown = joystick_y > joystick_neutral_y + DEADZONE;

	int total_buttons = SDL_JoystickNumButtons(joystick);
	int total_axes = SDL_JoystickNumAxes(joystick);
	int total_hats = SDL_JoystickNumHats(joystick);

	unsigned int buttons_done = 0;

	for (int i = 0; i < total_buttons; ++i)
	{
		status->bButton[buttons_done] = SDL_JoystickGetButton(joystick, i);

		if (++buttons_done < button_limit)
			break;
	}

	for (int i = 0; i < total_axes && buttons_done < button_limit; ++i)
	{
		Sint16 axis = SDL_JoystickGetAxis(joystick, i);

		status->bButton[buttons_done] = axis < -DEADZONE;

		if (++buttons_done < button_limit)
			break;

		status->bButton[buttons_done] = axis > DEADZONE;

		if (++buttons_done < button_limit)
			break;
	}

	for (int i = 0; i < total_axes && buttons_done < button_limit; ++i)
	{
		Uint8 hat = SDL_JoystickGetHat(joystick, i);

		status->bButton[buttons_done] = hat == SDL_HAT_UP;

		if (++buttons_done < button_limit)
			break;

		status->bButton[buttons_done] = hat == SDL_HAT_RIGHT;

		if (++buttons_done < button_limit)
			break;

		status->bButton[buttons_done] = hat == SDL_HAT_DOWN;

		if (++buttons_done < button_limit)
			break;

		status->bButton[buttons_done] = hat == SDL_HAT_LEFT;

		if (++buttons_done < button_limit)
			break;

		status->bButton[buttons_done] = hat == SDL_HAT_RIGHTUP;

		if (++buttons_done < button_limit)
			break;

		status->bButton[buttons_done] = hat == SDL_HAT_RIGHTDOWN;

		if (++buttons_done < button_limit)
			break;

		status->bButton[buttons_done] = hat == SDL_HAT_LEFTUP;

		if (++buttons_done < button_limit)
			break;

		status->bButton[buttons_done] = hat == SDL_HAT_LEFTDOWN;

		if (++buttons_done < button_limit)
			break;
	}

	// Blank the buttons that do not
	for (size_t i = buttons_done; i < button_limit; ++i)
		status->bButton[i] = FALSE;

	return TRUE;
}

BOOL ControllerBackend_ResetJoystickStatus(void)
{
	if (joystick == NULL)
		return FALSE;

	// The code that would normally run here has been moved to JoystickCallback, to better-support hotplugging

	return TRUE;
}

void ControllerBackend_JoystickConnect(Sint32 joystick_id)
{
	printf("Joystick #%d connected - %s\n", joystick_id, SDL_JoystickNameForIndex(joystick_id));

	if (joystick == NULL)
	{
		joystick = SDL_JoystickOpen(joystick_id);

		if (joystick != NULL)
		{
			printf("Joystick #%d selected\n", joystick_id);

			// Reset default stick positions (this is performed in ResetJoystickStatus in vanilla Cave Story
			joystick_neutral_x = SDL_JoystickGetAxis(joystick, 0);
			joystick_neutral_y = SDL_JoystickGetAxis(joystick, 1);
		}
	}
}

void ControllerBackend_JoystickDisconnect(Sint32 joystick_id)
{
	if (joystick_id == SDL_JoystickInstanceID(joystick))
	{
		printf("Joystick #%d disconnected\n", joystick_id);
		joystick = NULL;
	}
}