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
|
/*=========================================================================
*
* 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 "itkFastMarchingThresholdStoppingCriterion.h"
#include "itkMeshFileWriter.h"
int itkFastMarchingQuadEdgeMeshFilterBaseTest( 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, 0. );
trial->push_back( node_pair );
typedef itk::FastMarchingThresholdStoppingCriterion< MeshType, MeshType >
CriterionType;
CriterionType::Pointer criterion = CriterionType::New();
criterion->SetThreshold( 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;
}
typedef itk::MeshFileWriter< MeshType > WriterType;
WriterType::Pointer writer = WriterType::New();
writer->SetInput( fmm_filter->GetOutput() );
writer->SetFileName( "itkFastMarchingQuadEdgeMeshFilterBase.vtk" );
writer->Update();
return EXIT_SUCCESS;
}
|