/*******************************************************************************
Copyright (c) 2010 Matt Williams

This software is provided 'as-is', without any express or implied
warranty. In no event will the authors be held liable for any damages
arising from the use of this software.

Permission is granted to anyone to use this software for any purpose,
including commercial applications, and to alter it and redistribute it
freely, subject to the following restrictions:

    1. The origin of this software must not be misrepresented; you must not
    claim that you wrote the original software. If you use this software
    in a product, an acknowledgment in the product documentation would be
    appreciated but is not required.

    2. Altered source versions must be plainly marked as such, and must not be
    misrepresented as being the original software.

    3. This notice may not be removed or altered from any source
    distribution.
*******************************************************************************/

#include "TestSurfaceExtractor.h"

#include "PolyVoxCore/Density.h"
#include "PolyVoxCore/MaterialDensityPair.h"
#include "PolyVoxCore/SimpleVolume.h"
#include "PolyVoxCore/MarchingCubesSurfaceExtractor.h"

#include <QtTest>

using namespace PolyVox;

// Test our ability to modify the behaviour of the MarchingCubesSurfaceExtractor. This simple example only modifies
// the threshold (and actually this can be achieved by passing a parameter to the constructor of the
// DefaultMarchingCubesController) but you could implement custom behaviour in the other members
// if you wanted too. Actually, it's not clear if this ability is really useful because I can't think
// what you'd modify apart from the threshold but the ability is at least available. Also, the
// DefaultMarchingCubesController is templatised whereas this exmple shows that controllers don't
// have to be.
class CustomMarchingCubesController
{
public:
	typedef float DensityType;
	typedef float MaterialType;

	float convertToDensity(float voxel)
	{
		return voxel;
	}

	float convertToMaterial(float /*voxel*/)
	{
		return 1;
	}

	float blendMaterials(float /*a*/, float /*b*/, float /*weight*/)
	{
		return 1;
	}

	float getThreshold(void)
	{
		return 50.0f;
	}
};

// These 'writeDensityValueToVoxel' functions provide a unified interface for writting densities to primative and class voxel types.
// They are conceptually the inverse of the 'convertToDensity' function used by the MarchingCubesSurfaceExtractor. They probably shouldn't be part
// of PolyVox, but they might be usful to other tests so we cold move them into a 'Tests.h' or something in the future.
template<typename VoxelType>
void writeDensityValueToVoxel(int valueToWrite, VoxelType& voxel)
{
	voxel = valueToWrite;
}

template<>
void writeDensityValueToVoxel(int valueToWrite, Density8& voxel)
{
	voxel.setDensity(valueToWrite);
}

template<>
void writeDensityValueToVoxel(int valueToWrite, MaterialDensityPair88& voxel)
{
	voxel.setDensity(valueToWrite);
}

template<typename VoxelType>
void writeMaterialValueToVoxel(int /*valueToWrite*/, VoxelType& /*voxel*/)
{
	//Most types don't have a material
	return;
}

template<>
void writeMaterialValueToVoxel(int valueToWrite, MaterialDensityPair88& voxel)
{
	voxel.setMaterial(valueToWrite);
}

// Runs the surface extractor for a given type. 
template <typename VoxelType>
void testForType(SurfaceMesh<PositionMaterialNormal<VoxelType> >& result)
{
	const int32_t uVolumeSideLength = 32;

	//Create empty volume
	SimpleVolume<VoxelType> volData(Region(Vector3DInt32(0,0,0), Vector3DInt32(uVolumeSideLength-1, uVolumeSideLength-1, uVolumeSideLength-1)));

	for (int32_t z = 0; z < uVolumeSideLength; z++)
	{
		for (int32_t y = 0; y < uVolumeSideLength; y++)
		{
			for (int32_t x = 0; x < uVolumeSideLength; x++)
			{
				VoxelType voxelValue;
				//Create a density field which changes throughout the volume.
				writeDensityValueToVoxel<VoxelType>(x + y + z, voxelValue);
				//Two different materials in two halves of the volume
				writeMaterialValueToVoxel<VoxelType>(z > uVolumeSideLength / 2 ? 42 : 79, voxelValue);
				volData.setVoxelAt(x, y, z, voxelValue);
			}
		}
	}

	DefaultMarchingCubesController<VoxelType> controller;
	controller.setThreshold(50);
	MarchingCubesSurfaceExtractor< SimpleVolume<VoxelType> > extractor(&volData, volData.getEnclosingRegion(), &result, WrapModes::Border, VoxelType(), controller);
	extractor.execute();
}

void testCustomController(SurfaceMesh<PositionMaterialNormal<float> >& result)
{
	const int32_t uVolumeSideLength = 32;

	//Create empty volume
	SimpleVolume<float> volData(Region(Vector3DInt32(0,0,0), Vector3DInt32(uVolumeSideLength-1, uVolumeSideLength-1, uVolumeSideLength-1)));

	for (int32_t z = 0; z < uVolumeSideLength; z++)
	{
		for (int32_t y = 0; y < uVolumeSideLength; y++)
		{
			for (int32_t x = 0; x < uVolumeSideLength; x++)
			{
				float voxelValue = x + y + z;
				volData.setVoxelAt(x, y, z, voxelValue);
			}
		}
	}

	CustomMarchingCubesController controller;
	MarchingCubesSurfaceExtractor< SimpleVolume<float>, CustomMarchingCubesController > extractor(&volData, volData.getEnclosingRegion(), &result, WrapModes::Border, 0, controller);
	extractor.execute();
}

void TestSurfaceExtractor::testExecute()
{
	const static uint32_t uExpectedVertices = 4731;
	const static uint32_t uExpectedIndices = 12810;
	const static uint32_t uMaterialToCheck = 3000;
	const static float fExpectedMaterial = 42.0f;
	const static float fNoMaterial = 1.0f;

	SurfaceMesh<PositionMaterialNormal<int8_t> > int8Mesh;
	//Run the test for various voxel types.
	QBENCHMARK {
		testForType<int8_t>(int8Mesh);
	}
	QCOMPARE(int8Mesh.getNoOfVertices(), uExpectedVertices);
	QCOMPARE(int8Mesh.getNoOfIndices(), uExpectedIndices);
	QCOMPARE(int8Mesh.getVertices()[uMaterialToCheck].getMaterial(), static_cast<int8_t>(fNoMaterial));

	SurfaceMesh<PositionMaterialNormal<uint8_t> > uint8Mesh;
	testForType<uint8_t>(uint8Mesh);
	QCOMPARE(uint8Mesh.getNoOfVertices(), uExpectedVertices);
	QCOMPARE(uint8Mesh.getNoOfIndices(), uExpectedIndices);
	QCOMPARE(uint8Mesh.getVertices()[uMaterialToCheck].getMaterial(), static_cast<uint8_t>(fNoMaterial));

	SurfaceMesh<PositionMaterialNormal<int16_t> > int16Mesh;
	testForType<int16_t>(int16Mesh);
	QCOMPARE(int16Mesh.getNoOfVertices(), uExpectedVertices);
	QCOMPARE(int16Mesh.getNoOfIndices(), uExpectedIndices);
	QCOMPARE(int16Mesh.getVertices()[uMaterialToCheck].getMaterial(), static_cast<int16_t>(fNoMaterial));

	SurfaceMesh<PositionMaterialNormal<uint16_t> > uint16Mesh;
	testForType<uint16_t>(uint16Mesh);
	QCOMPARE(uint16Mesh.getNoOfVertices(), uExpectedVertices);
	QCOMPARE(uint16Mesh.getNoOfIndices(), uExpectedIndices);
	QCOMPARE(uint16Mesh.getVertices()[uMaterialToCheck].getMaterial(), static_cast<uint16_t>(fNoMaterial));

	SurfaceMesh<PositionMaterialNormal<int32_t> > int32Mesh;
	testForType<int32_t>(int32Mesh);
	QCOMPARE(int32Mesh.getNoOfVertices(), uExpectedVertices);
	QCOMPARE(int32Mesh.getNoOfIndices(), uExpectedIndices);
	QCOMPARE(int32Mesh.getVertices()[uMaterialToCheck].getMaterial(), static_cast<int32_t>(fNoMaterial));

	SurfaceMesh<PositionMaterialNormal<uint32_t> > uint32Mesh;
	testForType<uint32_t>(uint32Mesh);
	QCOMPARE(uint32Mesh.getNoOfVertices(), uExpectedVertices);
	QCOMPARE(uint32Mesh.getNoOfIndices(), uExpectedIndices);
	QCOMPARE(uint32Mesh.getVertices()[uMaterialToCheck].getMaterial(), static_cast<uint32_t>(fNoMaterial));

	SurfaceMesh<PositionMaterialNormal<float> > floatMesh;
	testForType<float>(floatMesh);
	QCOMPARE(floatMesh.getNoOfVertices(), uExpectedVertices);
	QCOMPARE(floatMesh.getNoOfIndices(), uExpectedIndices);
	QCOMPARE(floatMesh.getVertices()[uMaterialToCheck].getMaterial(), fNoMaterial);

	SurfaceMesh<PositionMaterialNormal<double> > doubleMesh;
	testForType<double>(doubleMesh);
	QCOMPARE(doubleMesh.getNoOfVertices(), uExpectedVertices);
	QCOMPARE(doubleMesh.getNoOfIndices(), uExpectedIndices);
	QCOMPARE(doubleMesh.getVertices()[uMaterialToCheck].getMaterial(), static_cast<double>(fNoMaterial));

	SurfaceMesh<PositionMaterialNormal<Density8> > densityMesh;
	testForType<Density8>(densityMesh);
	QCOMPARE(densityMesh.getNoOfVertices(), uExpectedVertices);
	QCOMPARE(densityMesh.getNoOfIndices(), uExpectedIndices);
	//QCOMPARE(densityMesh.getVertices()[uMaterialToCheck].getMaterial(), static_cast<uint8_t>(fNoMaterial));

	SurfaceMesh<PositionMaterialNormal<MaterialDensityPair88> > materialDensityMesh;
	testForType<MaterialDensityPair88>(materialDensityMesh);
	QCOMPARE(materialDensityMesh.getNoOfVertices(), uExpectedVertices);
	QCOMPARE(materialDensityMesh.getNoOfIndices(), uExpectedIndices);
	//QCOMPARE(materialDensityMesh.getVertices()[uMaterialToCheck].getMaterial(), static_cast<uint8_t>(fNoMaterial));

	//Test whether the CustomSurfaceExtractor works.
	testCustomController(floatMesh);
	QCOMPARE(floatMesh.getNoOfVertices(), uExpectedVertices);
	QCOMPARE(floatMesh.getNoOfIndices(), uExpectedIndices);
	QCOMPARE(floatMesh.getVertices()[uMaterialToCheck].getMaterial(), fNoMaterial);
}

QTEST_MAIN(TestSurfaceExtractor)