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
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkWarpHarmonicEnergyCalculatorTest.cxx
Language: C++
Date: $Date$
Version: $Revision$
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.
=========================================================================*/
#if defined(_MSC_VER)
#pragma warning ( disable : 4786 )
#endif
#include "itkOrientedImage.h"
#include "itkVector.h"
#include "itkWarpHarmonicEnergyCalculator.h"
#include "itkImageRegionIteratorWithIndex.h"
int itkWarpHarmonicEnergyCalculatorTest(int, char* [] )
{
// Define the dimension of the images
const unsigned int ImageDimension = 3;
typedef itk::Vector< double, ImageDimension > DeformationPixelType;
typedef unsigned char OutputPixelType;
// Declare the types of the images
typedef itk::Image<DeformationPixelType, ImageDimension> DeformationFieldType;
typedef itk::Image<OutputPixelType, ImageDimension> OutputImageType;
// Declare Iterator types apropriated for each image
typedef itk::ImageRegionIteratorWithIndex< DeformationFieldType > DeformationIteratorType;
typedef itk::ImageRegionIteratorWithIndex< OutputImageType > OutputIteratorType;
// Declare the type of the index to access images
typedef itk::Index<ImageDimension> IndexType;
// Declare the type of the size
typedef itk::Size<ImageDimension> SizeType;
// Declare the type of the Region
typedef itk::ImageRegion<ImageDimension> RegionType;
// Create two images
DeformationFieldType ::Pointer inputDeformationField = DeformationFieldType ::New();
// Define their size, and start index
SizeType size;
size[0] = 2;
size[1] = 2;
size[2] = 2;
IndexType start;
start[0] = 0;
start[1] = 0;
start[2] = 0;
RegionType region;
region.SetIndex( start );
region.SetSize( size );
// Initialize Image A
inputDeformationField->SetLargestPossibleRegion( region );
inputDeformationField->SetBufferedRegion( region );
inputDeformationField->SetRequestedRegion( region );
inputDeformationField->Allocate();
// Create one iterator for the Input Image (this is a light object)
DeformationIteratorType it( inputDeformationField,
inputDeformationField->GetBufferedRegion() );
// Initialize the content of Image A
DeformationPixelType vectorValue;
vectorValue.Fill( 5.0 ); // FIXME: replace with something more interesting...
it.GoToBegin();
while( !it.IsAtEnd() )
{
it.Set( vectorValue );
std::cout << it.Get() << std::endl;
++it;
}
// Declare the type for the calculator
typedef itk::WarpHarmonicEnergyCalculator< DeformationFieldType > CalculatorType;
// Create one Filter
CalculatorType::Pointer calculator = CalculatorType::New();
// Connect the input images
calculator->SetImage( inputDeformationField );
// Execute the calculator
calculator->Compute();
const double energy = calculator->GetHarmonicEnergy();
std::cout << energy << std::endl;
if ( vcl_abs(energy) > 1e-8 ) return EXIT_FAILURE;
return EXIT_SUCCESS;
}
|
