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
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkPathIteratorTest.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 <iostream>
#include <math.h>
#include "itkImage.h"
#include "itkImageRegionIterator.h"
#include "itkPolyLineParametricPath.h"
#include "itkPathIterator.h"
int itkPathIteratorTest(int, char*[])
{
typedef itk::Image<double, 2> ImageType;
typedef itk::PolyLineParametricPath<2> PathType;
typedef itk::PathIterator< ImageType, PathType > IterType;
typedef ImageType::IndexType IndexType;
typedef PathType::InputType PathInputType;
typedef PathType::VertexType VertexType;
typedef PathType::OffsetType OffsetType;
bool passed = true;
// Setup the image
std::cout << "Making a 64x64 white square centered in a 128x128 black image"<<std::endl;
ImageType::Pointer image = ImageType::New();
IndexType start;
start[0]=0;
start[1]=0;
ImageType::SizeType size;
size[0]=128;
size[1]=128;
ImageType::RegionType region;
region.SetSize(size);
region.SetIndex(start);
image->SetRegions(region);
double spacing[ ImageType::ImageDimension ];
spacing[0]=1.0;
spacing[1]=1.0;
image->SetSpacing(spacing);
image->Allocate();
typedef itk::ImageRegionIterator<ImageType> ImageItType;
ImageItType it( image, image->GetRequestedRegion() );
it.GoToBegin();
ImageType::PixelType storedValue;
IndexType pixelIndex;
while( !it.IsAtEnd() )
{
pixelIndex = it.GetIndex();
if( pixelIndex[0] >= int(size[0]/4) && pixelIndex[0] < int(size[0]*3/4) &&
pixelIndex[1] >= int(size[1]/4) && pixelIndex[1] < int(size[1]*3/4) )
{
it.Set(1.0);
}
else
{
it.Set(0.0);
}
++it;
}
// Retrieve and print the value stored at pixel index (32,32)
pixelIndex[0]=32;
pixelIndex[1]=32;
storedValue = image->GetPixel(pixelIndex);
std::cout << "The pixel at index (" << pixelIndex[0] << "," << pixelIndex[1]
<< ") has the value " << storedValue << ".\n" << std::endl;
// Setup the path
std::cout << "Making a square Path with v0 at (30,30) and v2 at (33,33)" << std::endl;
VertexType v;
PathType::Pointer path = PathType::New();
v.Fill(30);
path->AddVertex(v);
v[0]=33;
v[1]=30;
path->AddVertex(v);
v.Fill(33);
path->AddVertex(v);
v[0]=30;
v[1]=33;
path->AddVertex(v);
v.Fill(30);
path->AddVertex(v);
// Test the iterator
std::cout << "Creating an iterator to trace the PolyLineParametricPath" << std::endl;
IterType iter(image, path);
for( iter.GoToBegin(); !iter.IsAtEnd(); ++iter )
{
std::cout << "Path("<<iter.GetPathPosition()<<") @ "<<iter.GetIndex()<<" = "
<< iter.Get()<<"; Now inverting."<<std::endl;
iter.Set( 1.0 - iter.Get() );
}
if( int(.5+1000*iter.Get()) != 1000 )
{
std::cout << "PathIteratorTest: Set() Failed" << std::endl;
passed = false;
}
for( iter.GoToBegin(); !iter.IsAtEnd(); ++iter )
{
std::cout << "Path("<<iter.GetPathPosition()<<") @ "<<iter.GetIndex()<<" = "
<< iter.Get()<<std::endl;
}
std::cout << "Should still be at end: ";
std::cout << "Path("<<iter.GetPathPosition()<<") @ "<<iter.GetIndex()<<" = "<<
iter.Get()<<std::endl;
if( (iter.GetIndex())[0] != 30 || (iter.GetIndex())[1] != 30 )
{
std::cout << "PathIteratorTest: Failed to iterate over a closed loop" << std::endl;
passed = false;
}
if (passed)
{
std::cout << "PathIterator tests passed" << std::endl;
return EXIT_SUCCESS;
}
else
{
std::cout << "PathIterator tests failed" << std::endl;
return EXIT_FAILURE;
}
}
|