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

class BaseMapper;
enum class AddressType;

class AddressHasher
{
public:
	size_t operator()(const uint32_t& addr) const 
	{
		//Quick hash for addresses
		return addr;
	}
};

class LabelManager
{
private:
	unordered_map<uint32_t, string, AddressHasher> _codeLabels;
	unordered_map<uint32_t, string, AddressHasher> _codeComments;	
	unordered_map<string, uint32_t> _codeLabelReverseLookup;

	shared_ptr<BaseMapper> _mapper;

	int32_t GetLabelAddress(uint32_t absoluteAddr, AddressType addressType);
	int32_t GetLabelAddress(uint16_t relativeAddr);

public:
	LabelManager(shared_ptr<BaseMapper> mapper);

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

	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);

	bool ContainsLabel(string &label);

	bool HasLabelOrComment(uint16_t relativeAddr);
	bool HasLabelOrComment(uint32_t absoluteAddr, AddressType addressType);
};