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
|
/*=========================================================================
*
* Copyright NumFOCUS
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0.txt
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*=========================================================================*/
#include "itkNarrowBand.h"
#include "itkTestingMacros.h"
int
itkNarrowBandTest(int, char *[])
{
unsigned int i;
using IndexType = unsigned int;
using DataType = float;
using BandNodeType = itk::BandNode<IndexType, DataType>;
using BandType = itk::NarrowBand<BandNodeType>;
using RegionType = BandType::RegionType;
auto band = BandType::New();
ITK_EXERCISE_BASIC_OBJECT_METHODS(band, NarrowBand, LightObject);
band->Reserve(100);
// Create nodes
BandNodeType node;
band->SetTotalRadius(10);
band->SetInnerRadius(5);
for (i = 0; i < 20; ++i)
{
node.m_Data = i * 5.0;
node.m_Index = i;
node.m_NodeState = 0;
// Fill the band
band->PushBack(node);
}
std::cout << "Band size: " << band->Size() << std::endl;
// Iterate over the band
using itType = BandType::ConstIterator;
itType it = band->Begin();
itType itend = band->End();
i = 0;
// BandNodeType *tmp;
for (; it != itend; ++it)
{
std::cout << "Node " << i << std::endl << "Index: " << it->m_Index << " Data: " << it->m_Data << std::endl;
i++;
}
// Split the band
std::vector<RegionType> regions;
regions = band->SplitBand(10);
// RegionType region;
using regionitType = std::vector<RegionType>::const_iterator;
regionitType regionsit = regions.begin();
regionitType regionsitend = regions.end();
std::cout << "Number of regions: " << regions.size() << std::endl;
i = 0;
for (; regionsit != regionsitend; ++regionsit)
{
std::cout << "Region " << i << std::endl;
for (; regions[i].Begin != regions[i].End; regions[i].Begin++)
{
std::cout << "Index: " << regions[i].Begin->m_Index << " Data: " << regions[i].Begin->m_Data << std::endl;
}
i++;
}
std::cout << "Test Passed. " << std::endl;
return EXIT_SUCCESS;
}
|