Mesen/Core/LabelManager.h

#pragma once
#include "stdafx.h"
#include <unordered_map>
#include <functional>
using std::unordered_map;

class BaseMapper;
enum class AddressType;

class LabelManager
{
private:
	unordered_map<uint32_t, string, std::function<size_t(const uint32_t &addr)>> _codeLabels = unordered_map<uint32_t, string, std::function<size_t(const uint32_t &addr)>>(10000, [](const uint32_t & addr) { return addr; });
	unordered_map<uint32_t, string, std::function<size_t(const uint32_t &addr)>> _codeComments = unordered_map<uint32_t, string, std::function<size_t(const uint32_t &addr)>>(10000, [](const uint32_t & addr) { return addr; });
	unordered_map<string, uint32_t> _codeLabelReverseLookup;

	shared_ptr<BaseMapper> _mapper;

	int32_t GetLabelAddress(uint16_t relativeAddr, bool checkRegisters);

public:
	LabelManager(shared_ptr<BaseMapper> mapper);

	void SetLabel(uint32_t address, AddressType addressType, string label, string comment);
	
	int32_t GetLabelRelativeAddress(string label);

	string GetLabel(uint16_t relativeAddr, bool checkRegisters);
	string GetComment(uint16_t relativeAddr);
	void GetLabelAndComment(uint16_t relativeAddr, string &label, string &comment);
};