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
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkZeroFluxBoundaryConditionTest.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 "itkConstShapedNeighborhoodIterator.h"
#include "itkZeroFluxNeumannBoundaryCondition.h"
#include "itkImage.h"
void ZFBCTest_PrintNeighborhood( itk::ConstShapedNeighborhoodIterator<itk::Image<int, 2> > &p )
{
itk::ConstShapedNeighborhoodIterator<itk::Image<int, 2> >::ConstIterator
ci = p.Begin();
while (!ci.IsAtEnd())
{
std::cout << " "
<< ci.GetNeighborhoodIndex() << " -> "
<< ci.GetNeighborhoodOffset() << " = " << ci.Get() << std::endl;
ci++;
}
}
int itkZeroFluxBoundaryConditionTest(int, char* [] )
{
typedef itk::Image<int, 2> ImageType;
typedef itk::ConstShapedNeighborhoodIterator<ImageType> IteratorType;
typedef IteratorType::RadiusType RadiusType;
ImageType::Pointer img = ImageType::New();
ImageType::RegionType reg;
ImageType::RegionType::SizeType sz = {{ 5, 5 }};
ImageType::RegionType::IndexType idx = {{ 0, 0 }};
reg.SetSize(sz);
reg.SetIndex(idx);
img->SetRegions(reg);
img->Allocate();
ImageType::IndexType pos;
for (pos[1] = 0; pos[1] < 5; ++pos[1])
{
for (pos[0] = 0; pos[0] < 5; ++pos[0])
{
img->SetPixel(pos, pos[0] * 10 + pos[1]);
std::cout << img->GetPixel(pos) << " ";
}
std::cout << std::endl;
}
RadiusType rad;
RadiusType radTwo;
rad[0] = rad[1] = 1;
IteratorType it(rad, img, img->GetRequestedRegion());
itk::ZeroFluxNeumannBoundaryCondition<ImageType> bc;
it.OverrideBoundaryCondition(&bc);
itk::ConstShapedNeighborhoodIterator<ImageType>::OffsetType off;
off[0] = -1; off[1] = -1;
it.ActivateOffset(off);
off[0] = 0; off[1] = 0;
it.ActivateOffset(off);
off[0] = 1; off[1] = 1;
it.ActivateOffset(off);
pos[0] = pos[1] = 0;
it.SetLocation(pos);
for (it.GoToBegin(); !it.IsAtEnd(); ++it)
{
std::cout << "Index: " << it.GetIndex() << std::endl;
ZFBCTest_PrintNeighborhood(it);
}
// RadiusType radTwo;
radTwo[0] = radTwo[1] = 2;
IteratorType it2(radTwo, img, img->GetRequestedRegion());
it2.OverrideBoundaryCondition(&bc);
pos[0] = pos[1] = 0;
it2.SetLocation(pos);
for (it2.GoToBegin(); !it2.IsAtEnd(); ++it2)
{
ZFBCTest_PrintNeighborhood(it2);
}
return EXIT_SUCCESS;
}
|
