Mesen-X/Core/ShortcutKeyHandler.cpp

#include "stdafx.h"
#include "ShortcutKeyHandler.h"
#include "EmulationSettings.h"
#include "ControlManager.h"
#include "VideoDecoder.h"
#include "VsControlManager.h"
#include "FDS.h"
#include "SaveStateManager.h"
#include "RewindManager.h"

ShortcutKeyHandler::ShortcutKeyHandler()
{
	_keySetIndex = 0;
	_stopThread = false;
	_thread = std::thread([=]() {
		while(!_stopThread) {
			ProcessKeys();
			std::this_thread::sleep_for(std::chrono::duration<int, std::milli>(50));
		}
	});
}

ShortcutKeyHandler::~ShortcutKeyHandler()
{
	_stopThread = true;
	_thread.join();
}

bool ShortcutKeyHandler::IsKeyPressed(EmulatorShortcut shortcut)
{
	KeyCombination comb = EmulationSettings::GetShortcutKey(shortcut, _keySetIndex);

	int keyCount = (comb.Key1 ? 1 : 0) + (comb.Key2 ? 1 : 0) + (comb.Key3 ? 1 : 0);

	if(keyCount == 0) {
		return false;
	}

	if(_pressedKeys.size() != keyCount) {
		//Only allow shortcuts that use as many keys as the number of keys pressed
		//e.g: Needed to prevent Shift-F1 from triggering a shortcut for F1
		return false;
	}

	return ControlManager::IsKeyPressed(comb.Key1) &&
		(comb.Key2 == 0 || ControlManager::IsKeyPressed(comb.Key2)) &&
		(comb.Key3 == 0 || ControlManager::IsKeyPressed(comb.Key3));
}

bool ShortcutKeyHandler::DetectKeyPress(EmulatorShortcut shortcut)
{
	if(IsKeyPressed(shortcut)) {
		_keysDown[_keySetIndex].emplace(shortcut);

		if(_prevKeysDown[_keySetIndex].find(shortcut) == _prevKeysDown[_keySetIndex].end()) {
			return true;
		}
	}
	return false;
}

bool ShortcutKeyHandler::DetectKeyRelease(EmulatorShortcut shortcut)
{
	if(!IsKeyPressed(shortcut)) {
		if(_prevKeysDown[_keySetIndex].find(shortcut) != _prevKeysDown[_keySetIndex].end()) {
			return true;
		}
	}
	return false;
}

void ShortcutKeyHandler::CheckMappedKeys()
{
	bool isNetplayClient = GameClient::Connected();
	bool isMovieActive = MovieManager::Playing() || MovieManager::Recording();
	bool needConfirm = EmulationSettings::CheckFlag(ConfirmExitResetPower);

	//Let the UI handle these shortcuts
	for(uint64_t i = (uint64_t)EmulatorShortcut::SwitchDiskSide; i <= (uint64_t)EmulatorShortcut::OpenTraceLogger; i++) {
		if(DetectKeyPress((EmulatorShortcut)i)) {
			void* param = (void*)i;
			MessageManager::SendNotification(ConsoleNotificationType::ExecuteShortcut, param);
		}
	}

	if(DetectKeyPress(EmulatorShortcut::FastForward)) {
		EmulationSettings::SetFlags(EmulationFlags::Turbo);
	} else if(DetectKeyRelease(EmulatorShortcut::FastForward)) {
		EmulationSettings::ClearFlags(EmulationFlags::Turbo);
	}

	if(VsControlManager::GetInstance() && !isNetplayClient && !isMovieActive) {
		VsControlManager* manager = VsControlManager::GetInstance();
		if(DetectKeyPress(EmulatorShortcut::VsServiceButton)) {
			manager->SetServiceButtonState(true);
		}
		if(DetectKeyRelease(EmulatorShortcut::VsServiceButton)) {
			manager->SetServiceButtonState(false);
		}
	}

	if(DetectKeyPress(EmulatorShortcut::InsertNextDisk) && !isNetplayClient && !isMovieActive) {
		FDS::InsertNextDisk();
	}

	if(DetectKeyPress(EmulatorShortcut::MoveToNextStateSlot)) {
		SaveStateManager::MoveToNextSlot();
	}

	if(DetectKeyPress(EmulatorShortcut::MoveToPreviousStateSlot)) {
		SaveStateManager::MoveToPreviousSlot();
	}

	if(DetectKeyPress(EmulatorShortcut::SaveState)) {
		SaveStateManager::SaveState();
	}

	if(DetectKeyPress(EmulatorShortcut::LoadState) && !isNetplayClient) {
		SaveStateManager::LoadState();
	}

	if(DetectKeyPress(EmulatorShortcut::ToggleCheats) && !isNetplayClient && !isMovieActive) {
		MessageManager::SendNotification(ConsoleNotificationType::ExecuteShortcut, (void*)EmulatorShortcut::ToggleCheats);
	}

	if(DetectKeyPress(EmulatorShortcut::ToggleAudio)) {
		MessageManager::SendNotification(ConsoleNotificationType::ExecuteShortcut, (void*)EmulatorShortcut::ToggleAudio);
	}

	if(IsKeyPressed(EmulatorShortcut::RunSingleFrame)) {
		if(EmulationSettings::CheckFlag(EmulationFlags::Paused)) {
			EmulationSettings::ClearFlags(EmulationFlags::Paused);
			Console::Pause();
			std::this_thread::sleep_for(std::chrono::duration<int, std::milli>(50));
			EmulationSettings::SetFlags(EmulationFlags::Paused);
			Console::Resume();
		} else {
			EmulationSettings::SetFlags(EmulationFlags::Paused);
		}
	}

	if(!isNetplayClient && !isMovieActive && !EmulationSettings::CheckFlag(NsfPlayerEnabled)) {
		if(DetectKeyPress(EmulatorShortcut::Rewind)) {
			RewindManager::StartRewinding();
		} else if(DetectKeyRelease(EmulatorShortcut::Rewind)) {
			RewindManager::StopRewinding();
		} else  if(DetectKeyPress(EmulatorShortcut::RewindTenSecs)) {
			RewindManager::RewindSeconds(10);
		} else if(DetectKeyPress(EmulatorShortcut::RewindOneMin)) {
			RewindManager::RewindSeconds(60);
		}
	}
}

void ShortcutKeyHandler::ProcessKeys()
{
	auto lock = _lock.AcquireSafe();
	ControlManager::RefreshKeyState();

	_pressedKeys = ControlManager::GetPressedKeys();

	for(int i = 0; i < 2; i++) {
		_keysDown[i].clear();
		_keySetIndex = i;
		CheckMappedKeys();
		_prevKeysDown[i] = _keysDown[i];
	}	
}