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
|
/*=========================================================================
Program: ORFEO Toolbox
Language: C++
Date: $Date$
Version: $Revision$
Copyright (c) Centre National d'Etudes Spatiales. All rights reserved.
See OTBCopyright.txt 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.
=========================================================================*/
#include "otbPointSetExtractROI.h"
#include "itkPointSet.h"
#include "otbRandomPointSetSource.h"
int otbPointSetExtractROITest(int, char*[])
{
typedef float PixelType;
typedef itk::PointSet<PixelType, 2> PointSetType;
typedef PointSetType::PointsContainer PointsContainerType;
typedef PointSetType::PointType PointType;
typedef otb::RandomPointSetSource<PointSetType> PointSetSource;
PointSetSource::Pointer pointSetSource = PointSetSource::New();
pointSetSource->SetNumberOfPoints(1000);
PointSetType::PointType maxPoint;
maxPoint[0] = 1000;
maxPoint[1] = 1000;
pointSetSource->SetMaxPoint(maxPoint);
pointSetSource->SetSeed(152002);
pointSetSource->Update();
std::cout << "Input PointSet has " << pointSetSource->GetOutput()->GetNumberOfPoints();
std::cout << " points " << std::endl;
typedef otb::PointSetExtractROI<PointSetType, PointSetType> FilterType;
FilterType::Pointer filter = FilterType::New();
filter->SetInput(pointSetSource->GetOutput());
filter->SetStartX(450);
filter->SetStartY(450);
filter->SetSizeX(100);
filter->SetSizeY(100);
filter->Update();
std::cout << "Keeping points from " << filter->GetStartX() << ", " << filter->GetStartY() << std::endl;
std::cout << "with a region size of " << filter->GetSizeX() << ", " << filter->GetSizeY() << std::endl;
//Output the result
// Get the Smart Pointer to the Filter Output
PointSetType::Pointer outputPointSet = filter->GetOutput();
std::cout << "Output PointSet has " << outputPointSet->GetNumberOfPoints();
std::cout << " points " << std::endl;
// Get the the point container
PointSetType::PointsContainerPointer
extractedPoints = outputPointSet->GetPoints();
PointsContainerType::ConstIterator it = extractedPoints->Begin();
while (it != extractedPoints->End())
{
PointType p = it.Value();
std::cout.width(5); std::cout << p[0] << ", ";
std::cout.width(5); std::cout << p[1] << std::endl;
++it;
}
// All objects should be automatically destroyed at this point
if (outputPointSet->GetNumberOfPoints() != 6)
{
std::cout << "Wrong number of point in the output" << std::endl;
return EXIT_FAILURE;
}
return EXIT_SUCCESS;
}
|