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