diff --git a/tests/TestSurfaceExtractor.cpp b/tests/TestSurfaceExtractor.cpp index b061e569..241b69b4 100644 --- a/tests/TestSurfaceExtractor.cpp +++ b/tests/TestSurfaceExtractor.cpp @@ -32,6 +32,32 @@ freely, subject to the following restrictions: using namespace PolyVox; +// Test our ability to modify the behaviour of the SurfaceExtractor. This simple example only modifies +// the threshold (and actually this can be achieved by passing a parameter to the constructor of the +// DefaultSurfaceExtractionController) but you could implement custom behaviour in the other members +// if you wanted too. Actually, it's not clear if this ability is really useful because I can't think +// what you'd modify apart from the threshold but the ability is at least available. Also, the +// DefaultSurfaceExtractionController is templatised whereas this exmple shows that controllers don't +// have to be. +class CustomSurfaceExtractionController +{ +public: + float convertToDensity(float voxel) + { + return voxel; + } + + float convertToMaterial(float voxel) + { + return 1; + } + + float getThreshold(void) + { + return 50.0f; + } +}; + // These 'writeDensityValueToVoxel' functions provide a unified interface for writting densities to primative and class voxel types. // They are conceptually the inverse of the 'convertToDensity' function used by the SurfaceExtractor. They probably shouldn't be part // of PolyVox, but they might be usful to other tests so we cold move them into a 'Tests.h' or something in the future. @@ -96,6 +122,30 @@ void testForType(SurfaceMesh& result) extractor.execute(); } +void testCustomController(SurfaceMesh& result) +{ + const int32_t uVolumeSideLength = 32; + + //Create empty volume + SimpleVolume volData(Region(Vector3DInt32(0,0,0), Vector3DInt32(uVolumeSideLength-1, uVolumeSideLength-1, uVolumeSideLength-1))); + + for (int32_t z = 0; z < uVolumeSideLength; z++) + { + for (int32_t y = 0; y < uVolumeSideLength; y++) + { + for (int32_t x = 0; x < uVolumeSideLength; x++) + { + float voxelValue = x + y + z; + volData.setVoxelAt(x, y, z, voxelValue); + } + } + } + + CustomSurfaceExtractionController controller; + SurfaceExtractor< SimpleVolume, CustomSurfaceExtractionController > extractor(&volData, volData.getEnclosingRegion(), &result, controller); + extractor.execute(); +} + void TestSurfaceExtractor::testExecute() { const static uint32_t uExpectedVertices = 4731; @@ -106,6 +156,7 @@ void TestSurfaceExtractor::testExecute() SurfaceMesh mesh; + //Run the test for various voxel types. testForType(mesh); QCOMPARE(mesh.getNoOfVertices(), uExpectedVertices); QCOMPARE(mesh.getNoOfIndices(), uExpectedIndices); @@ -155,6 +206,12 @@ void TestSurfaceExtractor::testExecute() QCOMPARE(mesh.getNoOfVertices(), uExpectedVertices); QCOMPARE(mesh.getNoOfIndices(), uExpectedIndices); QCOMPARE(mesh.getVertices()[uMaterialToCheck].getMaterial(), fExpectedMaterial); + + //Test whether the CustomSurfaceExtractor works. + testCustomController(mesh); + QCOMPARE(mesh.getNoOfVertices(), uExpectedVertices); + QCOMPARE(mesh.getNoOfIndices(), uExpectedIndices); + QCOMPARE(mesh.getVertices()[uMaterialToCheck].getMaterial(), fNoMaterial); } QTEST_MAIN(TestSurfaceExtractor)