#include "UnitTestPCH.h"

#include <assimp/scene.h>
#include <TriangulateProcess.h>


using namespace std;
using namespace Assimp;

class TriangulateProcessTest : public ::testing::Test
{
public:

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

protected:

    aiMesh* pcMesh;
    TriangulateProcess* piProcess;
};

void TriangulateProcessTest::SetUp()
{
    piProcess = new TriangulateProcess();
    pcMesh = new aiMesh();

    pcMesh->mNumFaces = 1000;
    pcMesh->mFaces = new aiFace[1000];
    pcMesh->mVertices = new aiVector3D[10000];

    pcMesh->mPrimitiveTypes = aiPrimitiveType_POINT | aiPrimitiveType_LINE |
    aiPrimitiveType_LINE | aiPrimitiveType_POLYGON;

    for (unsigned int m = 0, t = 0, q = 4; m < 1000; ++m)
    {
        ++t;
        aiFace& face = pcMesh->mFaces[m];
        face.mNumIndices = t;
        if (4 == t)
        {
            face.mNumIndices = q++;
            t = 0;

            if (10 == q)q = 4;
        }
        face.mIndices = new unsigned int[face.mNumIndices];
        for (unsigned int p = 0; p < face.mNumIndices; ++p)
        {
            face.mIndices[p] = pcMesh->mNumVertices;

        // construct fully convex input data in ccw winding, xy plane
            aiVector3D& v = pcMesh->mVertices[pcMesh->mNumVertices++];
            v.z = 0.f;
            v.x = cos (p * (float)(AI_MATH_TWO_PI)/face.mNumIndices);
            v.y = sin (p * (float)(AI_MATH_TWO_PI)/face.mNumIndices);
        }
    }
}

void TriangulateProcessTest::TearDown()
{
    delete piProcess;
    delete pcMesh;
}

TEST_F(TriangulateProcessTest, testTriangulation)
{
    piProcess->TriangulateMesh(pcMesh);

    for (unsigned int m = 0, t = 0, q = 4, max = 1000, idx = 0; m < max;++m)
    {
        ++t;
        aiFace& face = pcMesh->mFaces[m];
        if (4 == t)
        {
            t = 0;
            max += q-3;

            std::vector<bool> ait(q,false);

            for (unsigned int i = 0, tt = q-2; i < tt; ++i,++m)
            {
                aiFace& face = pcMesh->mFaces[m];
                EXPECT_EQ(3U, face.mNumIndices);

                for (unsigned int qqq = 0; qqq < face.mNumIndices; ++qqq)
                {
                    ait[face.mIndices[qqq]-idx] = true;
                }
            }
            for (std::vector<bool>::const_iterator it = ait.begin(); it != ait.end(); ++it)
            {
                EXPECT_TRUE(*it);
            }
            --m;
            idx+=q;
            if(++q == 10)q = 4;
        }
        else
        {
            EXPECT_EQ(t, face.mNumIndices);

            for (unsigned int i = 0; i < face.mNumIndices; ++i,++idx)
            {
                EXPECT_EQ(idx, face.mIndices[i]);
            }
        }
    }

    // we should have no valid normal vectors now necause we aren't a pure polygon mesh
    EXPECT_TRUE(pcMesh->mNormals == NULL);
}