#include "DimensionGenerator.h"
#include "Constants.h"
#include "Noise.h"
#include "NoBlock.h"
#include "RandNoise.h"
#include "Game.h"

#include <iostream>


DimensionGenerator::DimensionGenerator(int dimensionId, CaveGenerator* caveGenerator)
	: ReferenceCounter(),
	dimensionId(dimensionId),
	minTemplateAffectedPosition(0, 0, 0),
	maxTemplateAffectedPosition(0, 0, 0),
	caveGenerator(caveGenerator)
{
	StaticRegistry<DimensionGenerator>::INSTANCE.registerT(this, dimensionId);
}

DimensionGenerator::~DimensionGenerator()
{}

void DimensionGenerator::initialize(int seed)
{
	this->seed = seed + dimensionId;
	caveGenerator->initialize(this->seed);
}

BiomGenerator* DimensionGenerator::zBiomGenerator(int x, int y)
{
	double noise = zBiomNoise()->getNoise((double)x, (double)y, 0.0);
	double border = 0;
	BiomGenerator* gen = 0;
	auto genI = biomGenerators.begin();
	auto distI = biomDistribution.begin();
	do
	{
		border += (double)distI++;
		gen = genI++;
	} while (border < noise && (bool)distI && (bool)genI);
	return gen;
}

void DimensionGenerator::registerBiom(BiomGenerator* generator, double possibility)
{
	biomGenerators.add(generator);
	biomDistribution.add(possibility);
	for (auto t : generator->getTemplates())
	{
		minTemplateAffectedPosition.x = MIN(minTemplateAffectedPosition.x, t->getMinAffectedOffset().x);
		minTemplateAffectedPosition.y = MIN(minTemplateAffectedPosition.y, t->getMinAffectedOffset().y);
		minTemplateAffectedPosition.z = MIN(minTemplateAffectedPosition.z, t->getMinAffectedOffset().z);
		maxTemplateAffectedPosition.x = MAX(maxTemplateAffectedPosition.x, t->getMaxAffectedOffset().x);
		maxTemplateAffectedPosition.y = MAX(maxTemplateAffectedPosition.y, t->getMaxAffectedOffset().y);
		maxTemplateAffectedPosition.z = MAX(maxTemplateAffectedPosition.z, t->getMaxAffectedOffset().z);
	}
}

Framework::RCArray<GeneratedStructure>* DimensionGenerator::getGeneratedStructoresForArea(Framework::Vec3<int> minPos, Framework::Vec3<int> maxPos)
{
	Framework::RCArray<GeneratedStructure>* result = new Framework::RCArray<GeneratedStructure>();

	int minSearchX = minPos.x - maxTemplateAffectedPosition.x;
	int minSearchY = minPos.y - maxTemplateAffectedPosition.y;
	int minSearchZ = MAX(minPos.z - maxTemplateAffectedPosition.z, 0);
	int maxSearchX = maxPos.x - minTemplateAffectedPosition.x;
	int maxSearchY = maxPos.y - minTemplateAffectedPosition.y;
	int maxSearchZ = MIN(maxPos.z - minTemplateAffectedPosition.z, WORLD_HEIGHT - 1);

	Noise* structureNoise = zStructureNoise();
	for (int x = minSearchX; x <= maxSearchX; x++)
	{
		for (int y = minSearchY; y <= maxSearchY; y++)
		{
			BiomGenerator* biom = zBiomGenerator(x, y);
			int height = MIN_AIR_LEVEL + (int)(biom->zHeightMapNoise(seed)->getNoise((double)(x), (double)(y), 0.0) * (MAX_AIR_LEVEL - MIN_AIR_LEVEL));
			for (int z = minSearchZ; z <= maxSearchZ; z++)
			{
				if (z < height)
				{
					double rValue = structureNoise->getNoise((double)x, (double)y, (double)z);
					double probSum = 0;
					for (auto t : biom->getTemplates())
					{
						if (t->isGenerationPossable(Framework::Vec3<int>(x, y, z), height - z))
						{
							if (rValue - probSum <= t->getPropability())
							{
								result->add(t->generateAt(Framework::Vec3<int>(x, y, z), structureNoise, dimensionId));
								break;
							}
						}
						probSum += t->getPropability();
					}
				}
			}
		}
	}

	return result;
}

Chunk* DimensionGenerator::generateChunk(int centerX, int centerY)
{
	Framework::RCArray<GeneratedStructure>* structures = getGeneratedStructoresForArea(Framework::Vec3<int>(centerX - CHUNK_SIZE / 2, centerY - CHUNK_SIZE / 2, 0), Framework::Vec3<int>(centerX + CHUNK_SIZE / 2, centerY + CHUNK_SIZE / 2, WORLD_HEIGHT - 1));
	std::cout << "generating chunk " << centerX << ", " << centerY << "\n";
	CaveChunkGenerator* caveGen = caveGenerator->getGeneratorForChunk(centerX, centerY);
	Chunk* chunk = new Chunk(Framework::Punkt(centerX, centerY), dimensionId);
	for (int x = -CHUNK_SIZE / 2; x < CHUNK_SIZE / 2; x++)
	{
		for (int y = -CHUNK_SIZE / 2; y < CHUNK_SIZE / 2; y++)
		{
			BiomGenerator* biom = zBiomGenerator(x + centerX, y + centerY);
			// TODO: use Noise interpolator for height map between different bioms
			int height = MIN_AIR_LEVEL + (int)(biom->zHeightMapNoise(seed)->getNoise((double)(x + centerX), (double)(y + centerY), 0.0) * (MAX_AIR_LEVEL - MIN_AIR_LEVEL));
			int maxSurfaceHeight = (int)(MAX_SURFACE_HEIGHT * (1.f - (float)(height - MIN_AIR_LEVEL) / (float)(MAX_AIR_LEVEL - MIN_AIR_LEVEL)));
			int actualSurfaceHeight = (int)((float)maxSurfaceHeight * (1.f - VARIABLE_SURFACE_PART) + ((float)maxSurfaceHeight * VARIABLE_SURFACE_PART * (float)biom->zHeightMapNoise(seed)->getNoise((double)(x + centerX), (double)(y + centerY), 10.0)));
			for (int z = 0; z < WORLD_HEIGHT; z++)
			{
				Framework::Either<Block*, int> generated = BlockTypeEnum::AIR;
				bool structureAffected = 0;
				for (auto structure : *structures)
				{
					if (structure->isBlockAffected(Framework::Vec3<int>(x + centerX, y + centerY, z)))
					{
						generated = structure->generateBlockAt(Framework::Vec3<int>(x + centerX, y + centerY, z));
						structureAffected = 1;
						break;
					}
				}
				if (!structureAffected)
				{
					if (caveGen->isInCave(x + centerX, y + centerY, z))
						generated = biom->generateCaveBlock(x + centerX, y + centerY, z);
					else if (z < height && z >= height - actualSurfaceHeight)
						generated = biom->generateSurfaceBlock(x + centerX, y + centerY, z);
					else if (z < height)
						generated = biom->generateBelowSurfaceBlock(x + centerX, y + centerY, z);
				}
				if (generated.isA())
					chunk->putBlockAt(Framework::Vec3<int>(x + CHUNK_SIZE / 2, y + CHUNK_SIZE / 2, z), generated);
				else
					chunk->putBlockTypeAt(Framework::Vec3<int>(x + CHUNK_SIZE / 2, y + CHUNK_SIZE / 2, z), generated);
			}
		}
	}
	caveGen->release();
	structures->release();
	return chunk;
}

Framework::Either<Block*, int> DimensionGenerator::generateBlock(Framework::Vec3<int> location)
{
	Framework::RCArray<GeneratedStructure>* structures = getGeneratedStructoresForArea(location, location);
	BiomGenerator* biom = zBiomGenerator(location.x, location.y);
	// TODO: use Noise interpolator for height map between different bioms
	int height = MIN_AIR_LEVEL + (int)(biom->zHeightMapNoise(seed)->getNoise((double)(location.x), (double)(location.y), 0.0) * (MAX_AIR_LEVEL - MIN_AIR_LEVEL));
	int maxSurfaceHeight = (int)(MAX_SURFACE_HEIGHT * (1.f - (float)(height - MIN_AIR_LEVEL) / (float)(MAX_AIR_LEVEL - MIN_AIR_LEVEL)));
	int actualSurfaceHeight = (int)((float)maxSurfaceHeight * (1.f - VARIABLE_SURFACE_PART) + ((float)maxSurfaceHeight * VARIABLE_SURFACE_PART * (float)biom->zHeightMapNoise(seed)->getNoise((double)(location.x), (double)(location.y), 10.0)));
	for (auto structure : *structures)
	{
		if (structure->isBlockAffected(location))
		{
			auto generated = structure->generateBlockAt(location);
			structures->release();
			return generated;
		}
	}
	structures->release();
	if (location.z < height && location.z >= height - actualSurfaceHeight)
		return biom->generateSurfaceBlock(location.x, location.y, location.z);
	else if (location.z < height)
		return biom->generateBelowSurfaceBlock(location.x, location.y, location.z);
	return BlockTypeEnum::AIR;
}

bool DimensionGenerator::spawnStructure(Framework::Vec3<int> location, std::function<bool(GenerationTemplate* tmpl)> filter)
{
	BiomGenerator* biom = zBiomGenerator(location.x, location.y);
	for (auto t : biom->getTemplates())
	{
		if (filter(t))
		{
			RandNoise noise((int)time(0));
			GeneratedStructure *genStr = t->generateAt(location, &noise, dimensionId);
			if (genStr)
			{
				int minSearchX = location.x + t->getMinAffectedOffset().x;
				int minSearchY = location.y + t->getMinAffectedOffset().y;
				int minSearchZ = MAX(location.z + t->getMinAffectedOffset().z, 0);
				int maxSearchX = location.x + t->getMaxAffectedOffset().x;
				int maxSearchY = location.y + t->getMaxAffectedOffset().y;
				int maxSearchZ = MIN(location.z + t->getMaxAffectedOffset().z, WORLD_HEIGHT - 1);

				for (int x = minSearchX; x <= maxSearchX; x++)
				{
					for (int y = minSearchY; y <= maxSearchY; y++)
					{
						for (int z = minSearchZ; z <= maxSearchZ; z++)
						{
							if (genStr->isBlockAffected(Framework::Vec3<int>(x, y, z)))
							{
								auto gen = genStr->generateBlockAt(Framework::Vec3<int>(x, y, z));
								Game::INSTANCE->zDimension(dimensionId)->placeBlock(Framework::Vec3<int>(x, y, z), gen);
							}
						}
					}
				}
				genStr->release();
				return 1;
			}
		}
	}
	return 0;
}

int DimensionGenerator::getDimensionId() const
{
	return dimensionId;
}