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 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: $RCSfile: itkQuadEdgeMeshLinearParameterizationTest.cxx,v $
Language: C++
Date: $Date: 2009-12-19 20:56:32 $
Version: $Revision: 1.9 $
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.
=========================================================================*/
#include "itkVTKPolyDataReader.h"
#include "itkVTKPolyDataWriter.h"
#include "itkQuadEdgeMesh.h"
// NEW
#include "itkQuadEdgeMeshBorderTransform.h"
#include "itkQuadEdgeMeshParamMatrixCoefficients.h"
#ifdef QuadEdgePARAM_TAUCS
#include "TAUCSSparseSolverTraits.h"
#else
#include "VNLIterativeSparseSolverTraits.h"
#endif
#include "itkQuadEdgeMeshParam.h"
int itkQuadEdgeMeshLinearParameterizationTest( int argc, char* argv[] )
{
// ** ERROR MESSAGE AND HELP ** //
if( argc < 5 )
{
std::cout <<"Requires 4 arguments: " << std::endl;
std::cout <<"1-Input file name " << std::endl;
std::cout <<"2-Border Type" << std::endl;
std::cout <<" * 0: SQUARE" << std::endl;
std::cout <<" * 1: DISK" << std::endl;
std::cout <<"3-CoefficientType Type" << std::endl;
std::cout <<" * 0: OnesMatrixCoefficients" << std::endl;
std::cout <<" * 1: InverseEuclideanDistanceMatrixCoefficients" << std::endl;
std::cout <<" * 2: ConformalMatrixCoefficients" << std::endl;
std::cout <<" * 3: AuthalicMatrixCoefficients" << std::endl;
std::cout <<" * 4: HarmonicMatrixCoefficients" << std::endl;
std::cout <<"4-Output file name " << std::endl;
return EXIT_FAILURE;
}
// ** TYPEDEF **
typedef double Coord;
typedef itk::QuadEdgeMesh< Coord, 3 > MeshType;
typedef itk::VTKPolyDataReader< MeshType > ReaderType;
typedef itk::VTKPolyDataWriter< MeshType > WriterType;
typedef itk::QuadEdgeMeshBorderTransform< MeshType, MeshType > BorderTransformType;
#ifdef QuadEdgePARAM_TAUCS
typedef TAUCSSolverTraits< Coord > SolverTraits;
#else
typedef VNLIterativeSparseSolverTraits< Coord > SolverTraits;
#endif
typedef itk::QuadEdgeMeshParam< MeshType, MeshType, SolverTraits > ParametrizationType;
// ** READ THE FILE IN **
ReaderType::Pointer reader = ReaderType::New( );
reader->SetFileName( argv[1] );
try
{
reader->Update( );
}
catch( itk::ExceptionObject & excp )
{
std::cerr << "Exception thrown while reading the input file " << std::endl;
std::cerr << excp << std::endl;
return EXIT_FAILURE;
}
MeshType::Pointer mesh = reader->GetOutput( );
// ** CHOSE< COMPUTE AND SET BORDER TRANSFORM **
BorderTransformType::Pointer border_transform = BorderTransformType::New( );
border_transform->SetInput( mesh );
// two following line for coverage
border_transform->SetRadius( border_transform->GetRadius() );
border_transform->GetNameOfClass();
int border;
std::stringstream ssout( argv[2] );
ssout >>border;
switch( border ) // choose border type
{
case 0: // square shaped domain
border_transform->SetTransformType( BorderTransformType::SQUARE_BORDER_TRANSFORM );
break;
case 1: // disk shaped domain
border_transform->SetTransformType( BorderTransformType::DISK_BORDER_TRANSFORM );
break;
default: // handle .... user ....
std::cerr << "2nd argument must be " << std::endl;
std::cerr << "0 for SQUARE BORDER TRANSFORM or "
<< "1 for DISK BORDER TRANSFORM" << std::endl;
return EXIT_FAILURE;
}
std::cout << "Transform type is: " << border_transform->GetTransformType( );
std::cout << std::endl;
// ** CHOOSE AND SET BARYCENTRIC WEIGHTS **
itk::OnesMatrixCoefficients< MeshType > coeff0;
itk::InverseEuclideanDistanceMatrixCoefficients< MeshType > coeff1;
itk::ConformalMatrixCoefficients< MeshType > coeff2;
itk::AuthalicMatrixCoefficients< MeshType > coeff3;
itk::HarmonicMatrixCoefficients< MeshType > coeff4;
ParametrizationType::Pointer param = ParametrizationType::New( );
param->SetInput( mesh );
param->SetBorderTransform( border_transform );
int param_type;
std::stringstream ssout3( argv[3] );
ssout3 >> param_type;
switch( param_type )
{
case 0:
param->SetCoefficientsMethod( &coeff0 );
break;
case 1:
param->SetCoefficientsMethod( &coeff1 );
break;
case 2:
param->SetCoefficientsMethod( &coeff2 );
break;
case 3:
param->SetCoefficientsMethod( &coeff3 );
break;
case 4:
param->SetCoefficientsMethod( &coeff4 );
break;
default:
std::cerr << "3rd argument must be " << std::endl;
std::cerr << "0, 1, 2, 3 or 4" << std::endl;
std::cerr << "Here it is: " << param_type << std::endl;
return EXIT_FAILURE;
}
// ** PROCESS **
param->Update( );
MeshType::Pointer output = param->GetOutput( );
// ** WRITE OUTPUT **
WriterType::Pointer writer = WriterType::New( );
writer->SetInput( param->GetOutput( ) );
writer->SetFileName( argv[4] );
writer->Update( );
// GET OUT OF HERE AND GET (YET ANOTHER) COFFEE
return EXIT_SUCCESS;
}
|