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
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkFloodFilledSpatialFunctionTest.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
/*
This file test the various inclusion strategies available
to itkFloodFilledSpatialFunctionConditionalIterator.
*/
#include <stdio.h>
#include "itkSize.h"
#include "itkIndex.h"
#include "itkImage.h"
#include "itkImageRegionIterator.h"
#include "itkPoint.h"
#include "itkSphereSpatialFunction.h"
#include "itkFloodFilledSpatialFunctionConditionalIterator.h"
int itkFloodFilledSpatialFunctionTest(int, char* [])
{
const unsigned int dim = 2;
// Image typedef
typedef itk::Image< bool, dim > ImageType;
typedef ImageType::SizeValueType SizeValueType;
typedef ImageType::SpacingValueType SpacingValueType;
typedef ImageType::PointValueType PointValueType;
// Image size and spacing parameters
SizeValueType sourceImageSize[] = { 5, 5};
SpacingValueType sourceImageSpacing[] = { 1.0,1.0 };
PointValueType sourceImageOrigin[] = { 0,0 };
// Creates the sourceImage (but doesn't set the size or allocate memory)
ImageType::Pointer sourceImage = ImageType::New();
sourceImage->SetOrigin(sourceImageOrigin);
sourceImage->SetSpacing(sourceImageSpacing);
// Create a size object native to the sourceImage type
ImageType::SizeType sourceImageSizeObject;
// Set the size object to the array defined earlier
sourceImageSizeObject.SetSize( sourceImageSize );
// Create a region object native to the sourceImage type
ImageType::RegionType largestPossibleRegion;
// Resize the region
largestPossibleRegion.SetSize( sourceImageSizeObject );
// Set the largest legal region size (i.e. the size of the whole sourceImage) to what we just defined
sourceImage->SetLargestPossibleRegion( largestPossibleRegion );
// Set the buffered region
sourceImage->SetBufferedRegion( largestPossibleRegion );
// Set the requested region
sourceImage->SetRequestedRegion( largestPossibleRegion );
// Now allocate memory for the sourceImage
sourceImage->Allocate();
// Loop over all available iterator strategies
for(int strat = 0; strat < 4; strat++)
{
// Initialize the image to hold all 0's
itk::ImageRegionIterator<ImageType> it =
itk::ImageRegionIterator<ImageType>(sourceImage, largestPossibleRegion);
for(it.GoToBegin(); !it.IsAtEnd(); ++it)
{
it.Set(0);
}
// Create and initialize a spatial function
typedef itk::SphereSpatialFunction<dim> FunctionType;
typedef FunctionType::InputType FunctionPositionType;
FunctionType::Pointer spatialFunc = FunctionType::New();
spatialFunc->SetRadius( 1.0 );
FunctionPositionType center;
center[0] = 2.5;
center[1] = 2.5;
spatialFunc->SetCenter(center);
// Create and initialize a spatial function iterator
ImageType::IndexType seedPos;
const ImageType::IndexValueType pos[] = {2,2};
seedPos.SetIndex(pos);
typedef itk::FloodFilledSpatialFunctionConditionalIterator
<ImageType, FunctionType> ItType;
ItType sfi = ItType(sourceImage, spatialFunc, seedPos);
switch(strat){
case 0:
{
sfi.SetOriginInclusionStrategy();
}
break;
case 1:
{
sfi.SetCenterInclusionStrategy();
}
break;
case 2:
{
sfi.SetCompleteInclusionStrategy();
}
break;
case 3:
{
sfi.SetIntersectInclusionStrategy();
}
} // end switch inclusion strategy
// Iterate through the entire image and set interior pixels to 1
for( sfi.GoToBegin(); !( sfi.IsAtEnd() ); ++sfi)
{
sfi.Set(1);
}
} // end loop over iterator strategies
return EXIT_SUCCESS;
}
|