#include "UnitTestPCH.h"

#include <FindDegenerates.h>


using namespace std;
using namespace Assimp;

class FindDegeneratesProcessTest : public ::testing::Test
{
public:

	virtual void SetUp();
	virtual void TearDown();

protected:

	aiMesh* mesh;
	FindDegeneratesProcess* process;
};

// ------------------------------------------------------------------------------------------------
void FindDegeneratesProcessTest::SetUp()
{
	mesh = new aiMesh();
	process = new FindDegeneratesProcess();

	mesh->mNumFaces = 1000;
	mesh->mFaces = new aiFace[1000];

	mesh->mNumVertices = 5000*2;
	mesh->mVertices = new aiVector3D[5000*2];

	for (unsigned int i = 0; i < 5000; ++i) {
		mesh->mVertices[i] = mesh->mVertices[i+5000] = aiVector3D((float)i);
	}

	mesh->mPrimitiveTypes = aiPrimitiveType_LINE | aiPrimitiveType_POINT |
	aiPrimitiveType_POLYGON | aiPrimitiveType_TRIANGLE;

	unsigned int numOut = 0, numFaces = 0;
	for (unsigned int i = 0; i < 1000; ++i) {
		aiFace& f = mesh->mFaces[i];
	f.mNumIndices = (i % 5)+1; // between 1 and 5
	f.mIndices = new unsigned int[f.mNumIndices];
	bool had = false;
	for (unsigned int n = 0; n < f.mNumIndices;++n) {
		// FIXME
#if 0
		// some duplicate indices
		if ( n && n == (i / 200)+1) {
			f.mIndices[n] = f.mIndices[n-1];
			had = true;
		}
		// and some duplicate vertices
#endif
		if (n && i % 2 && 0 == n % 2) {
			f.mIndices[n] = f.mIndices[n-1]+5000;
			had = true;
		}
		else {
			f.mIndices[n] = numOut++;
			}
		}
		if (!had)
			++numFaces;
	}
	mesh->mNumUVComponents[0] = numOut;
	mesh->mNumUVComponents[1] = numFaces;
}

// ------------------------------------------------------------------------------------------------
void FindDegeneratesProcessTest::TearDown()
{
	delete mesh;
	delete process;
}

// ------------------------------------------------------------------------------------------------
TEST_F(FindDegeneratesProcessTest, testDegeneratesDetection)
{
	process->EnableInstantRemoval(false);
	process->ExecuteOnMesh(mesh);

	unsigned int out = 0;
	for (unsigned int i = 0; i < 1000; ++i) {
		aiFace& f = mesh->mFaces[i];
		out += f.mNumIndices;
	}

	EXPECT_EQ(1000U, mesh->mNumFaces);
	EXPECT_EQ(10000U, mesh->mNumVertices);
	EXPECT_EQ(out, mesh->mNumUVComponents[0]);
	EXPECT_EQ(static_cast<unsigned int>(
	              aiPrimitiveType_LINE | aiPrimitiveType_POINT |
	              aiPrimitiveType_POLYGON | aiPrimitiveType_TRIANGLE),
	          mesh->mPrimitiveTypes);
}

// ------------------------------------------------------------------------------------------------
TEST_F(FindDegeneratesProcessTest, testDegeneratesRemoval)
{
	process->EnableInstantRemoval(true);
	process->ExecuteOnMesh(mesh);

	EXPECT_EQ(mesh->mNumUVComponents[1], mesh->mNumFaces);
}