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 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: $RCSfile: itkVTKPolyDataWriterTest2.cxx,v $
Language: C++
Date: $Date: 2009-06-12 15:00:17 $
Version: $Revision: 1.2 $
Copyright (c) Insight Software Consortium. All rights reserved.
See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notices for more information.
=========================================================================*/
#ifdef _MSC_VER
#pragma warning ( disable : 4786 )
#endif
#include "itkMesh.h"
#include "itkRegularSphereMeshSource.h"
#include "itkDefaultStaticMeshTraits.h"
#include "itkVTKPolyDataWriter.h"
#include <iostream>
int itkVTKPolyDataWriterTest2( int argc, char * argv [] )
{
if( argc != 2 )
{
std::cerr << "Usage: itkVTKPolyDataWriter outputFileName" << std::endl;
return EXIT_FAILURE;
}
typedef itk::Mesh<float, 3> MeshType;
typedef itk::RegularSphereMeshSource< MeshType > SphereMeshSourceType;
SphereMeshSourceType::Pointer mySphereMeshSource = SphereMeshSourceType::New();
typedef SphereMeshSourceType::PointType PointType;
typedef SphereMeshSourceType::VectorType VectorType;
PointType center;
center.Fill( 7.4 );
const double radius = 1.5;
const double tolerance = 1e-5;
VectorType scale;
scale.Fill( radius );
mySphereMeshSource->SetCenter( center );
mySphereMeshSource->SetResolution( 1 );
mySphereMeshSource->SetScale( scale );
mySphereMeshSource->Modified();
try
{
mySphereMeshSource->Update();
}
catch( itk::ExceptionObject & excp )
{
std::cerr << "Error during Update() " << std::endl;
std::cerr << excp << std::endl;
}
std::cout << "mySphereMeshSource: " << mySphereMeshSource;
MeshType::Pointer myMesh = mySphereMeshSource->GetOutput();
PointType pt;
pt.Fill( 0. );
bool testPassed = true;
std::cout << "Testing itk::RegularSphereMeshSource "<< std::endl;
for( unsigned int i=0; i< myMesh->GetNumberOfPoints(); i++)
{
myMesh->GetPoint(i, &pt);
std::cout << "Point[" << i << "]: " << pt << std::endl;
const double distanceToCenter = pt.EuclideanDistanceTo( center );
if( vnl_math_abs( distanceToCenter - radius ) > tolerance )
{
std::cerr << "Distance to center " << distanceToCenter;
std::cerr << " is too different from radius " << radius << std::endl;
testPassed = false;
}
}
typedef MeshType::CellsContainerPointer CellsContainerPointer;
typedef MeshType::CellsContainerIterator CellsContainerIterator;
typedef MeshType::CellType CellType;
CellsContainerPointer cells = myMesh->GetCells();
unsigned faceId = 0;
MeshType::CellsContainerIterator cellsItr = cells->Begin();
while( cellsItr != cells->End() )
{
CellType * cellPointer = cellsItr.Value();
if( cellPointer->GetType() != 1 )
{
const unsigned int numberOfPoints = cellPointer->GetNumberOfPoints();
std::cout <<"Face " << faceId << " has " << numberOfPoints <<" points" << std::endl;
if( numberOfPoints != 3 )
{
std::cerr << "Face with wrong number of points" << std::endl;
testPassed = false;
}
}
++cellsItr;
++faceId;
}
typedef itk::VTKPolyDataWriter<MeshType> WriterType;
WriterType::Pointer writer = WriterType::New();
writer->SetInput( myMesh );
writer->SetFileName( argv[1] );
writer->Write();
if( !testPassed )
{
std::cout << "Test FAILED! "<< std::endl;
return EXIT_FAILURE;
}
std::cout << "Test PASSED! "<< std::endl;
return EXIT_SUCCESS;
}
|