Mesen-SX/Core/CheatManager.h

#pragma once
#include "stdafx.h"

class Console;

struct CheatCode
{
	uint32_t Address;
	uint8_t Value;
};

class CheatManager
{
private:
	Console* _console;
	bool _hasCheats = false;
	bool _bankHasCheats[0x100] = {};
	vector<CheatCode> _cheats;
	unordered_map<uint32_t, CheatCode> _cheatsByAddress;

	void AddCheat(CheatCode code);

public:
	CheatManager(Console* console);

	void AddStringCheat(string code);

	void SetCheats(vector<CheatCode> codes);
	void SetCheats(uint32_t codes[], uint32_t length);
	void ClearCheats(bool showMessage = true);

	vector<CheatCode> GetCheats();

	__forceinline void ApplyCheat(uint32_t addr, uint8_t& value);
};

__forceinline void CheatManager::ApplyCheat(uint32_t addr, uint8_t& value)
{
	if (_hasCheats && _bankHasCheats[addr >> 16])
	{
		auto result = _cheatsByAddress.find(addr);
		if (result != _cheatsByAddress.end())
		{
			value = result->second.Value;
		}
	}
}