assimp/test/unit/utRemoveComponent.cpp

149 lines
4.7 KiB
C++

#include "UnitTestPCH.h"
#include "utRemoveComponent.h"
CPPUNIT_TEST_SUITE_REGISTRATION (RemoveVCProcessTest);
void RemoveVCProcessTest :: setUp (void)
{
// construct the process
piProcess = new RemoveVCProcess();
pScene = new aiScene();
// fill the scene ..
pScene->mMeshes = new aiMesh*[pScene->mNumMeshes = 2];
pScene->mMeshes[0] = new aiMesh();
pScene->mMeshes[1] = new aiMesh();
pScene->mMeshes[0]->mNumVertices = 120;
pScene->mMeshes[0]->mVertices = new aiVector3D[120];
pScene->mMeshes[0]->mNormals = new aiVector3D[120];
pScene->mMeshes[0]->mTextureCoords[0] = new aiVector3D[120];
pScene->mMeshes[0]->mTextureCoords[1] = new aiVector3D[120];
pScene->mMeshes[0]->mTextureCoords[2] = new aiVector3D[120];
pScene->mMeshes[0]->mTextureCoords[3] = new aiVector3D[120];
pScene->mMeshes[1]->mNumVertices = 120;
pScene->mMeshes[1]->mVertices = new aiVector3D[120];
pScene->mAnimations = new aiAnimation*[pScene->mNumAnimations = 2];
pScene->mAnimations[0] = new aiAnimation();
pScene->mAnimations[1] = new aiAnimation();
pScene->mTextures = new aiTexture*[pScene->mNumTextures = 2];
pScene->mTextures[0] = new aiTexture();
pScene->mTextures[1] = new aiTexture();
pScene->mMaterials = new aiMaterial*[pScene->mNumMaterials = 2];
pScene->mMaterials[0] = new MaterialHelper();
pScene->mMaterials[1] = new MaterialHelper();
pScene->mLights = new aiLight*[pScene->mNumLights = 2];
pScene->mLights[0] = new aiLight();
pScene->mLights[1] = new aiLight();
pScene->mCameras = new aiCamera*[pScene->mNumCameras = 2];
pScene->mCameras[0] = new aiCamera();
pScene->mCameras[1] = new aiCamera();
// COMPILE TEST: MaterialHelper may no add any extra members,
// so we don't need a virtual destructor
char check[sizeof(MaterialHelper) == sizeof(aiMaterial) ? 10 : -1];
}
void RemoveVCProcessTest :: tearDown (void)
{
delete pScene;
delete piProcess;
}
void RemoveVCProcessTest::testMeshRemove (void)
{
piProcess->SetDeleteFlags(aiComponent_MESHES);
piProcess->Execute(pScene);
CPPUNIT_ASSERT(NULL == pScene->mMeshes && 0 == pScene->mNumMeshes);
CPPUNIT_ASSERT(pScene->mFlags == AI_SCENE_FLAGS_INCOMPLETE);
}
void RemoveVCProcessTest::testAnimRemove (void)
{
piProcess->SetDeleteFlags(aiComponent_ANIMATIONS);
piProcess->Execute(pScene);
CPPUNIT_ASSERT(NULL == pScene->mAnimations && 0 == pScene->mNumAnimations);
CPPUNIT_ASSERT(pScene->mFlags == 0);
}
void RemoveVCProcessTest::testMaterialRemove (void)
{
piProcess->SetDeleteFlags(aiComponent_MATERIALS);
piProcess->Execute(pScene);
// there should be one default material now ...
CPPUNIT_ASSERT(1 == pScene->mNumMaterials &&
pScene->mMeshes[0]->mMaterialIndex == 0 &&
pScene->mMeshes[1]->mMaterialIndex == 0);
CPPUNIT_ASSERT(pScene->mFlags == 0);
}
void RemoveVCProcessTest::testTextureRemove (void)
{
piProcess->SetDeleteFlags(aiComponent_TEXTURES);
piProcess->Execute(pScene);
CPPUNIT_ASSERT(NULL == pScene->mTextures && 0 == pScene->mNumTextures);
CPPUNIT_ASSERT(pScene->mFlags == 0);
}
void RemoveVCProcessTest::testCameraRemove (void)
{
piProcess->SetDeleteFlags(aiComponent_CAMERAS);
piProcess->Execute(pScene);
CPPUNIT_ASSERT(NULL == pScene->mCameras && 0 == pScene->mNumCameras);
CPPUNIT_ASSERT(pScene->mFlags == 0);
}
void RemoveVCProcessTest::testLightRemove (void)
{
piProcess->SetDeleteFlags(aiComponent_LIGHTS);
piProcess->Execute(pScene);
CPPUNIT_ASSERT(NULL == pScene->mLights && 0 == pScene->mNumLights);
CPPUNIT_ASSERT(pScene->mFlags == 0);
}
void RemoveVCProcessTest::testMeshComponentsRemoveA (void)
{
piProcess->SetDeleteFlags(aiComponent_TEXCOORDSn(1) | aiComponent_TEXCOORDSn(2) | aiComponent_TEXCOORDSn(3));
piProcess->Execute(pScene);
CPPUNIT_ASSERT(pScene->mMeshes[0]->mTextureCoords[0] &&
!pScene->mMeshes[0]->mTextureCoords[1] &&
!pScene->mMeshes[0]->mTextureCoords[2] &&
!pScene->mMeshes[0]->mTextureCoords[3]);
CPPUNIT_ASSERT(pScene->mFlags == 0);
}
void RemoveVCProcessTest::testMeshComponentsRemoveB (void)
{
piProcess->SetDeleteFlags(aiComponent_TEXCOORDSn(1) | aiComponent_NORMALS);
piProcess->Execute(pScene);
CPPUNIT_ASSERT(pScene->mMeshes[0]->mTextureCoords[0] &&
pScene->mMeshes[0]->mTextureCoords[1] &&
pScene->mMeshes[0]->mTextureCoords[2] && // shift forward ...
!pScene->mMeshes[0]->mTextureCoords[3] &&
!pScene->mMeshes[0]->mNormals);
CPPUNIT_ASSERT(pScene->mFlags == 0);
}
void RemoveVCProcessTest::testRemoveEverything (void)
{
piProcess->SetDeleteFlags(aiComponent_LIGHTS | aiComponent_ANIMATIONS |
aiComponent_MATERIALS | aiComponent_MESHES | aiComponent_CAMERAS | aiComponent_TEXTURES);
piProcess->Execute(pScene);
}