1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123
|
/*=========================================================================
*
* Copyright Insight Software Consortium
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0.txt
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*=========================================================================*/
#include "itkFastMarchingQuadEdgeMeshFilterBase.h"
#include "itkQuadEdgeMeshExtendedTraits.h"
#include "itkRegularSphereMeshSource.h"
#include "itkFastMarchingNumberOfElementsStoppingCriterion.h"
int itkFastMarchingQuadEdgeMeshFilterWithNumberOfElementsTest( int , char * [] )
{
typedef float PixelType;
typedef double CoordType;
const unsigned int Dimension = 3;
typedef itk::QuadEdgeMeshExtendedTraits <
PixelType, // type of data for vertices
Dimension, // geometrical dimension of space
2, // Mac topological dimension of a cell
CoordType, // type for point coordinate
CoordType, // type for interpolation weight
PixelType, // type of data for cell
bool, // type of data for primal edges
bool // type of data for dual edges
> Traits;
typedef itk::QuadEdgeMesh< PixelType, Dimension, Traits > MeshType;
typedef itk::FastMarchingQuadEdgeMeshFilterBase< MeshType, MeshType > FastMarchingType;
MeshType::PointType center;
center.Fill( 0. );
typedef itk::RegularSphereMeshSource< MeshType > SphereSourceType;
SphereSourceType::Pointer sphere_filter = SphereSourceType::New();
sphere_filter->SetCenter( center );
sphere_filter->SetResolution( 5 );
sphere_filter->Update();
MeshType::Pointer sphere_output = sphere_filter->GetOutput();
MeshType::PointsContainerConstPointer points =
sphere_output->GetPoints();
MeshType::PointsContainerConstIterator p_it = points->Begin();
MeshType::PointsContainerConstIterator p_end = points->End();
while( p_it != p_end )
{
sphere_output->SetPointData( p_it->Index(), 1. );
++p_it;
}
typedef FastMarchingType::NodePairType NodePairType;
// typedef FastMarchingType::NodeContainerType NodeContainerType;
typedef FastMarchingType::NodePairContainerType NodePairContainerType;
NodePairContainerType::Pointer trial = NodePairContainerType::New();
NodePairType node_pair( 0, 1. );
trial->push_back( node_pair );
typedef itk::FastMarchingNumberOfElementsStoppingCriterion< MeshType, MeshType > CriterionType;
CriterionType::Pointer criterion = CriterionType::New();
criterion->SetTargetNumberOfElements( 100 );
FastMarchingType::Pointer fmm_filter = FastMarchingType::New();
fmm_filter->SetInput( sphere_output );
fmm_filter->SetTrialPoints( trial );
fmm_filter->SetStoppingCriterion( criterion );
try
{
fmm_filter->Update();
}
catch( itk::ExceptionObject & excep )
{
std::cerr << "Exception caught !" << std::endl;
std::cerr << excep << std::endl;
return EXIT_FAILURE;
}
MeshType::Pointer output = fmm_filter->GetOutput();
MeshType::PointDataContainerPointer pointdata = output->GetPointData();
MeshType::PointDataContainerIterator it = pointdata->Begin();
unsigned int counter = 0;
while( it != pointdata->End() )
{
if( it->Value() > 0. )
{
++counter;
}
++it;
}
if( counter >= 100 )
{
return EXIT_SUCCESS;
}
else
{
return EXIT_FAILURE;
}
}
|