#include "GeneratedStructure.h"
#include "GenerationTemplate.h"


GeneratedStructure::GeneratedStructure(GenerationTemplate* t, Framework::Vec3<int> originPos, Framework::Vec3<int> size, Framework::Vec3<int> minAffectedPos)
	: ReferenceCounter(),
	size(size),
	minAffectedPos(minAffectedPos),
	originPos(originPos),
	t(t)
{
	blockIds = new int[size.x * size.y * size.z];
	blocks = new Block * [size.x * size.y * size.z];
	memset(blockIds, 0, sizeof(int) * size.x * size.y * size.z);
	memset(blocks, 0, sizeof(Block*) * size.x * size.y * size.z);
}

GeneratedStructure::~GeneratedStructure()
{
	for (int i = 0; i < size.x * size.y * size.z; i++)
	{
		if (blocks[i])
			blocks[i]->release();
	}
	delete[] blockIds;
	delete[] blocks;
	t->release();
}

void GeneratedStructure::setBlockAt(Framework::Either<Block*, int> block, Framework::Vec3<int> localPos)
{
	assert(localPos.x >= 0 && localPos.y >= 0 && localPos.z >= 0 && localPos.x < size.x&& localPos.y < size.y&& localPos.z < size.z);
	int index = ((localPos.x * size.y) + localPos.y) * size.z + localPos.z;
	if (block.isA())
		blocks[index] = block;
	else
		blockIds[index] = block;
}

bool GeneratedStructure::isBlockAffected(Framework::Vec3<int> location) const
{
	Framework::Vec3<int> localPos = location - minAffectedPos;
	if (localPos.x >= 0 && localPos.y >= 0 && localPos.z >= 0 && localPos.x < size.x && localPos.y < size.y && localPos.z < size.z)
	{
		int index = ((localPos.x * size.y) + localPos.y) * size.z + localPos.z;
		return blocks[index] || blockIds[index];
	}
	return 0;
}

Framework::Either<Block*, int> GeneratedStructure::generateBlockAt(Framework::Vec3<int> location) const
{
	Framework::Vec3<int> localPos = location - minAffectedPos;
	if (localPos.x >= 0 && localPos.y >= 0 && localPos.z >= 0 && localPos.x < size.x && localPos.y < size.y && localPos.z < size.z)
	{
		int index = ((localPos.x * size.y) + localPos.y) * size.z + localPos.z;
		if (blocks[index])
			return blocks[index];
		return blockIds[index];
	}
	return 0;
}

Framework::Vec3<int> GeneratedStructure::getOriginPos() const
{
	return originPos;
}

GenerationTemplate* GeneratedStructure::zTemplate() const
{
	return t;
}