#pragma once

#include <robin_hood.h>
#include <string_view>

// BIDI stands for bi-directional
#define BIDI_LUT_DECL(name, aType, aCount, bType, bCount) \
	int gLut_##name##_A2B[aCount];                        \
	int gLut_##name##_B2A[bCount];                        \
	using name##AType = aType;                            \
	using name##BType = bType;                            \
	void InitializeLut##name()
#define BIDI_LUT_MAP_FOR(name)              \
	int* lutMappingA2B = gLut_##name##_A2B; \
	int* lutMappingB2A = gLut_##name##_B2A
#define BIDI_LUT_MAP(from, to) \
	lutMappingA2B[from] = to;  \
	lutMappingB2A[to] = from
#define BIDI_LUT_INIT(name) InitializeLut##name()
#define BIDI_LUT_A2B_LOOKUP(name, from) (name##BType)(gLut_##name##_A2B[from])
#define BIDI_LUT_B2A_LOOKUP(name, to) (name##AType)(gLut_##name##_B2A[to])

// Forward string lookup
#define FSTR_LUT_DECL(name, enumMinValue, enumMaxValue)             \
	constexpr int kLutMinVal_##name = enumMinValue;                 \
	const char* gLut_##name[(int)enumMaxValue - (int)enumMinValue]; \
	void InitializeLut##name()
#define FSTR_LUT_MAP_FOR(name)             \
	const char** lutMapping = gLut_##name; \
	int lutMappingMinValue = kLutMinVal_##name
#define FSTR_LUT_MAP(value, text) lutMapping[value - lutMappingMinValue] = text
#define FSTR_LUT_MAP_ENUM(enumValue) FSTR_LUT_MAP(enumValue, #enumValue)
#define FSTR_LUT_LOOKUP(name, enumValue) gLut_##name[enumValue - kLutMinVal_##name]
#define FSTR_LUT_INIT(name) InitializeLut##name()

// RSTR stands for reverse-string lookup
#define RSTR_LUT_DECL(name, enumMinValue, enumMaxValue)                                   \
	robin_hood::unordered_flat_map<std::string_view, decltype(enumMaxValue)> gLut_##name; \
	void InitializeLut##name()
#define RSTR_LUT_MAP_FOR(name) auto& lutMapping = gLut_##name;
#define RSTR_LUT_MAP(value, text) lutMapping.insert_or_assign(std::string_view(text), value);
#define RSTR_LUT(name) gLut_##name
#define BSTR_LUT_LOOKUP(name, string) gLut_##name.find(std::string_view(text))->second
#define RSTR_LUT_INIT(name) InitializeLut##name()

// BSTR stands for bi-directional string lookup
#define BSTR_LUT_DECL(name, enumMinValue, enumMaxValue)                                         \
	constexpr int kLutMinVal_##name = enumMinValue;                                             \
	const char* gLut_##name##_V2S[(int)enumMaxValue - (int)enumMinValue];                       \
	robin_hood::unordered_flat_map<std::string_view, decltype(enumMaxValue)> gLut_##name##_S2V; \
	void InitializeLut##name()
#define BSTR_LUT_MAP_FOR(name)                      \
	const char** lutMappingV2S = gLut_##name##_V2S; \
	auto& lutMappingS2V = gLut_##name##_S2V;        \
	int lutMappingMinValue = kLutMinVal_##name
#define BSTR_LUT_MAP(value, text)                     \
	lutMappingV2S[value - lutMappingMinValue] = text; \
	lutMappingS2V.insert_or_assign(std::string_view(text), value);
#define BSTR_LUT_MAP_ENUM(enumValue) BSTR_LUT_MAP(enumValue, #enumValue)
#define BSTR_LUT_V2S(name) gLut_##name##_V2S
#define BSTR_LUT_S2V(name) gLut_##name##_S2V
#define BSTR_LUT_V2S_LOOKUP(name, enumValue) gLut_##name##_V2S[enumValue - kLutMinVal_##name]
#define BSTR_LUT_S2V_LOOKUP(name, string) gLut_##name##_S2V.find(std::string_view(text))->second
#define BSTR_LUT_INIT(name) InitializeLut##name()