// PieceModifier.cpp

// Implements the various classes descending from cPiece::cPieceModifier

#include "Globals.h"
#include "PieceModifier.h"
#include "../Noise/Noise.h"





// Constant that is added to seed
static const int SEED_OFFSET = 135 * 13;





// Emit a warning if the first param is true
#define CONDWARNING(ShouldLog, Fmt, ...) \
	do { \
		if (ShouldLog) \
		{ \
			LOGWARNING(Fmt, __VA_ARGS__); \
		} \
	} while (false)





////////////////////////////////////////////////////////////////////////////////
/** A modifier which is pseudo-randomly replacing blocks to other types and metas. */
class cPieceModifierRandomizeBlocks:
	public cPiece::cPieceModifier
{
public:
	cPieceModifierRandomizeBlocks(void) :
		m_Seed()
	{
	}

	virtual bool InitializeFromString(const AString & a_Params, bool a_LogWarnings) override
	{
		m_AllWeights = 0;
		AString Params = a_Params;


		/** BlocksToReplace parsing */
		auto idxPipe = Params.find('|');
		if (idxPipe != AString::npos)
		{
			AString blocksToReplaceStr = Params.substr(0, idxPipe);
			auto blocksToReplace = StringSplitAndTrim(blocksToReplaceStr, ",");
			for (size_t i = 0; i < blocksToReplace.size(); i++)
			{
				BLOCKTYPE blockType = static_cast<BLOCKTYPE>(BlockStringToType(blocksToReplace[i]));
				if ((blockType == E_BLOCK_AIR) && !NoCaseCompare(blocksToReplace[i], "Air"))
				{
					CONDWARNING(a_LogWarnings, "Cannot parse block type from string \"%s\"!", blocksToReplace[i].c_str());
					return false;
				}
				m_BlocksToReplace[blockType] = 1;
			}

			Params = Params.substr(idxPipe + 1, Params.length() - 1);
		}
		if (m_BlocksToReplace.size() == 0)
		{
			CONDWARNING(a_LogWarnings, "You must specify at least one block to replace%s!", "");
			return false;
		}


		/** Meta params parsing */
		auto idxSqBracketStart = Params.find('[');
		auto idxSqBracketStartLast = Params.find_last_of('[');

		bool isMultiMeta = false;
		if ((idxSqBracketStart != idxSqBracketStartLast) && (idxSqBracketStartLast != Params.length() - 1))
		{
			// Meta per block type
			isMultiMeta = true;
		}
		else
		{
			auto idxSqBracketEnd = Params.find(']');
			if ((idxSqBracketEnd == Params.length() - 1) && (idxSqBracketStart != AString::npos))
			{
				AString metaParamsStr = Params.substr(idxSqBracketStart + 1, Params.length() - idxSqBracketStart - 2);
				std::array<int, 4> metaParamsInt;
				if (!ParseMeta(metaParamsStr, metaParamsInt, a_LogWarnings))
				{
					return false;
				}

				m_MinMeta = metaParamsInt[0];
				m_MaxMeta = metaParamsInt[1];
				m_MinNoiseMeta = metaParamsInt[2];
				m_MaxNoiseMeta = metaParamsInt[3];

				Params = Params.substr(0, idxSqBracketStart);
			}
		}


		// BlocksToRandomize parsing
		auto BlocksToRandomize = StringSplitAndTrim(Params, ";");
		for (size_t i = 0; i < BlocksToRandomize.size(); i++)
		{
			AString block = BlocksToRandomize[i];

			cRandomizedBlock Block{};

			if (isMultiMeta)
			{
				auto sqBrStart = block.find('[');
				if (sqBrStart != AString::npos)
				{
					auto sqBrEnd = block.find(']');
					if (sqBrEnd != block.size() - 1)
					{
						CONDWARNING(a_LogWarnings, "If present, block meta params must be at the end of block to randomize definition \"%s\"!", block.c_str());
						return false;

					}
					AString metaParamsStr = block.substr(sqBrStart + 1, block.size() - sqBrStart - 2);

					std::array<int, 4> metaParamsInt;
					if (!ParseMeta(metaParamsStr, metaParamsInt, a_LogWarnings))
					{
						return false;
					}

					Block.m_MinMeta = metaParamsInt[0];
					Block.m_MaxMeta = metaParamsInt[1];
					Block.m_MinNoiseMeta = metaParamsInt[2];
					Block.m_MaxNoiseMeta = metaParamsInt[3];

					block = block.substr(0, sqBrStart);

				}
				// No meta randomization for this block
			}

			auto BlockParams = StringSplitAndTrim(block, ",");
			if (BlockParams.size() < 2)
			{
				CONDWARNING(a_LogWarnings, "Block weight is required param \"%s\"!", BlockParams[0].c_str());
				return false;
			}

			BLOCKTYPE BlockType = static_cast<BLOCKTYPE>(BlockStringToType(BlockParams[0]));
			int BlockWeight = 0;
			if ((BlockType != E_BLOCK_AIR) && !NoCaseCompare(BlockParams[0], "Air"))
			{
				// Failed to parse block type
				CONDWARNING(
					a_LogWarnings, "Cannot parse block type from string \"%s\"!",
					BlockParams[0].c_str());
				return false;
			}

			if (!StringToInteger(BlockParams[1], BlockWeight))
			{
				// Failed to parse the crop weight:
				CONDWARNING(
					a_LogWarnings,
					"Cannot parse block weight from string \"%s\"!",
					BlockParams[1].c_str());
				return false;
			}


			Block.m_Type = BlockType;
			Block.m_Weight = BlockWeight;
			m_AllWeights += BlockWeight;

			m_BlocksToRandomize.push_back(Block);
		}
		if (m_BlocksToRandomize.size() == 0)
		{
			CONDWARNING(a_LogWarnings, "You must specify at least one block to randomize%s!", "");
			return false;
		}

		return true;
	}





	bool ParseMeta(const AString & a_Meta, std::array<int, 4> & a_ParsedMeta, bool a_LogWarnings)
	{
		auto MetaParams = StringSplitAndTrim(a_Meta, ",");

		for (size_t i = 0; i < MetaParams.size(); i++)
		{
			int Value;
			if (!StringToInteger(MetaParams[i], Value))
			{
				// Failed to parse meta parameter from string:
				CONDWARNING(a_LogWarnings, "Cannot parse meta param from string \"%s\", meta: %s!", MetaParams[i].c_str(), a_Meta.c_str());
				return false;
			}

			if (i > 3)
			{
				CONDWARNING(a_LogWarnings, "Unsupported meta param \"%d\"!", Value);
				return false;
			}

			a_ParsedMeta[i] = Value;
		}

		if (MetaParams.size() == 1)
		{
			// Noise is not used for meta
			a_ParsedMeta[1] = a_ParsedMeta[0];
		}
		else if (MetaParams.size() == 2)
		{
			// Noise range is same as meta range
			a_ParsedMeta[2] = a_ParsedMeta[0];
			a_ParsedMeta[3] = a_ParsedMeta[1];
		}
		else if (MetaParams.size() == 3)
		{
			a_ParsedMeta[3] = a_ParsedMeta[1];
		}

		return true;
	}





	virtual void Modify(cBlockArea & a_Image, const Vector3i a_PiecePos, const int a_PieceRot) override
	{
		cNoise Noise(m_Seed);
		cNoise PieceNoise(Noise.IntNoise3DInt(a_PiecePos));

		size_t NumBlocks = a_Image.GetBlockCount();
		BLOCKTYPE * BlockTypes = a_Image.GetBlockTypes();
		BLOCKTYPE * BlockMetas = a_Image.GetBlockMetas();

		for (size_t i = 0; i < NumBlocks; i++)
		{
			if (m_BlocksToReplace.count(BlockTypes[i]))
			{
				float BlockRnd = PieceNoise.IntNoise2DInRange(a_PieceRot, static_cast<int>(i), 0, m_AllWeights);

				int weightDelta = 0;
				for (auto & blockToRnd : m_BlocksToRandomize)
				{
					weightDelta += blockToRnd.m_Weight;
					if (BlockRnd <= weightDelta)
					{
						BlockTypes[i] = blockToRnd.m_Type;

						// Per block meta params
						if (blockToRnd.m_MinMeta < blockToRnd.m_MaxMeta)
						{
							int BlockMetaRnd = std::clamp(static_cast<int>(PieceNoise.IntNoise2DInRange(a_PieceRot*2, static_cast<int>(i), blockToRnd.m_MinNoiseMeta, blockToRnd.m_MaxNoiseMeta)), blockToRnd.m_MinMeta, blockToRnd.m_MaxMeta);
							BlockMetas[i] = static_cast<NIBBLETYPE>(BlockMetaRnd);
						}
						else if ((blockToRnd.m_MaxMeta > -1) && (blockToRnd.m_MaxMeta == blockToRnd.m_MinMeta))
						{
							// Change meta if at least minimum meta was specified
							BlockMetas[i] = static_cast<NIBBLETYPE>(blockToRnd.m_MaxMeta);
						}
						break;
					}
				}

				// All blocks meta params
				if (m_MaxMeta > m_MinMeta)
				{
					int BlockMetaRnd = std::clamp(static_cast<int>(PieceNoise.IntNoise2DInRange(a_PieceRot*2, static_cast<int>(i), m_MinNoiseMeta, m_MaxNoiseMeta)), m_MinMeta, m_MaxMeta);
					BlockMetas[i] = static_cast<NIBBLETYPE>(BlockMetaRnd);
				}
				else if ((m_MaxMeta > -1) && (m_MaxMeta == m_MinMeta))
				{
					// Change meta if at least minimum meta was specified
					BlockMetas[i] = static_cast<NIBBLETYPE>(m_MaxMeta);
				}
			}
		}  // for i - BlockTypes[]
	}

	virtual void AssignSeed(int a_Seed) override
	{
		m_Seed = a_Seed + SEED_OFFSET;
	}
protected:
	int m_Seed;
	int m_AllWeights = 0;


	/** Block types of a blocks which are being replaced by this strategy */
	std::map<BLOCKTYPE, int> m_BlocksToReplace;

	/** Randomized blocks with their weights and meta params */
	cRandomizedBlocks m_BlocksToRandomize;

	/** Minimum meta to randomize */
	int m_MinMeta = 0;

	/** Maximum meta to randomize */
	int m_MaxMeta = -1;

	/** Maximum meta in noise range */
	int m_MaxNoiseMeta = 0;

	/** Minimum meta in noise range */
	int m_MinNoiseMeta = 0;
};





////////////////////////////////////////////////////////////////////////////////
// CreatePieceModifierFromString:

bool CreatePieceModifierFromString(const AString & a_Definition, std::shared_ptr<cPiece::cPieceModifiers> & a_Modifiers, bool a_LogWarnings)
{

	auto idxCurlyStart = a_Definition.find('{');
	auto idxCurlyFirstEnd = a_Definition.find('}');
	if ((idxCurlyStart == AString::npos) && (idxCurlyFirstEnd == AString::npos))
	{
		CONDWARNING(a_LogWarnings, "Piece metadata \"Modifiers\" needs at least one valid modifier definition \"%s\"!", a_Definition.c_str());
		return false;
	}

	auto modifiersStr = StringSplitAndTrim(a_Definition, "{");

	for (size_t i = 0; i < modifiersStr.size(); i++)
	{
		AString modifierStr = TrimString(modifiersStr[i]);

		if (modifierStr.size() == 0)
		{
			continue;
		}
		auto idxCurlyEnd = modifierStr.find('}');
		if (idxCurlyEnd == AString::npos)
		{
			CONDWARNING(a_LogWarnings, "Modifier definition must end with curly bracket \"%s\"!!", modifierStr.c_str());
			return false;
		}

		modifierStr = modifierStr.substr(0, idxCurlyEnd);

		// Break apart the modifier class, the first parameter before the first pipe char:
		auto idxPipe = modifierStr.find('|');
		if (idxPipe == AString::npos)
		{
			idxPipe = modifierStr.length();
		}
		AString ModifierClass = modifierStr.substr(0, idxPipe);

		// Create a modifier class based on the class string:
		cPiece::cPieceModifierPtr Modifier;
		if (NoCaseCompare(ModifierClass, "RandomizeBlocks") == 0)
		{
			Modifier = std::make_shared<cPieceModifierRandomizeBlocks>();
		}

		if (Modifier == nullptr)
		{
			CONDWARNING(a_LogWarnings, "Unknown modifier class \"%s\" %s!", ModifierClass.c_str(), modifierStr.c_str());
			return false;
		}

		// Initialize the modifier's parameters:
		AString Params;
		if (idxPipe < modifierStr.length())
		{
			Params = modifierStr.substr(idxPipe + 1);
		}

		if (!Modifier->InitializeFromString(Params, a_LogWarnings))
		{
			CONDWARNING(a_LogWarnings, "InitializeFromString error \"%s\" -- %!", Params.c_str(), modifierStr.c_str());
			return false;
		}

		a_Modifiers->push_back(Modifier);
	}

	return true;
}