114 lines
2.4 KiB
C++
114 lines
2.4 KiB
C++
#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);
|
|
}
|