TEST: Complete MeanShift segmentation scheme test.

Testing/Code/OBIA/CMakeLists.txt

@@ -149,6 +149,24 @@ ADD_TEST(obTvStreamingConnectedComponentSegmentationOBIAToVectorDataFilter ${OBI
       "SHAPE_Elongation>8"
       5 )
 
+#TODO add validation test
+ADD_TEST(obTuMeanShiftStreamingConnectedComponentSegmentationOBIAToVectorDataFilter ${OBIA_TESTS1}
+      otbMeanShiftStreamingConnectedComponentSegmentationOBIAToVectorDataFilter
+      ${INPUTDATA}/ROI_QB_MUL_4.tif
+      ${TEMP}/obTuMeanShiftStreamingConnectedComponentSegmentationOBIAToVectorDataFilter.shp
+      9 
+      9 
+      50 
+      10 
+      0.001
+      ""
+      "distance<10"
+      20
+      "SHAPE_Elongation>10"
+      5 )
+
+
+
 ADD_TEST(obTuLabelImageToLabelMapWithAdjacencyFilterNew ${OBIA_TESTS3}
   otbLabelImageToLabelMapWithAdjacencyFilterNew)
 
@@ -271,6 +289,7 @@ otbStatisticsAttributesLabelMapFilterNew.cxx
 otbVectorDataToLabelMapFilter.cxx
 otbVectorDataToLabelMapFilterNew.cxx
 otbStreamingConnectedComponentOBIATest.cxx
+otbMeanShiftStreamingConnectedComponentOBIATest.cxx
 otbHooverMatrixFilterNew.cxx
 otbHooverMatrixFilter.cxx
 otbHooverInstanceFilterNew.cxx

Testing/Code/OBIA/otbMeanShiftStreamingConnectedComponentOBIATest.cxx

+/*=========================================================================
+
+  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 "otbImage.h"
+#include "otbVectorImage.h"
+#include "otbVectorData.h"
+#include "otbImageFileReader.h"
+#include "otbVectorDataFileWriter.h"
+#include "otbMeanShiftImageFilter2.h"
+#include "otbStreamingConnectedComponentSegmentationOBIAToVectorDataFilter.h"
+
+typedef float InputPixelType;
+const unsigned int Dimension = 2;
+typedef double                                          KernelType;
+typedef otb::Image<unsigned int, Dimension>             LabelImageType;
+typedef otb::Image<unsigned int, Dimension>             MaskImageType;
+
+typedef otb::VectorData<double, Dimension>          VectorDataType;
+typedef VectorDataType::Pointer                     VectorDataPointerType;
+typedef otb::VectorDataFileWriter<VectorDataType>   VectorDataFileWriterType;
+typedef VectorDataFileWriterType::Pointer           VectorDataFileWriterPointerType;
+
+typedef otb::VectorImage<InputPixelType, Dimension>     ImageType;
+typedef otb::ImageFileReader<ImageType>                 ReaderType;
+
+
+
+
+
+typedef otb::StreamingConnectedComponentSegmentationOBIAToVectorDataFilter
+  < ImageType,
+    LabelImageType,
+    MaskImageType,
+    VectorDataType >  ConnectedComponentSegmentationOBIAToVectorDataFilterType;
+
+typedef otb::MeanShiftImageFilter2<ImageType, ImageType,ImageType,KernelType> MeanShiftFilterType;
+
+
+int otbMeanShiftStreamingConnectedComponentSegmentationOBIAToVectorDataFilter(int argc, char * argv[])
+{
+
+  /* mean shift parameters */
+
+  const char * infname = argv[1];
+  const char * outputFilename = argv[2];
+
+  /* mean shift parameters */
+  const unsigned int spatialRadius = atoi(argv[3]);
+  const unsigned int spectralRadius = atoi(argv[4]);
+  const double spectralbandwidth = atof(argv[5]);
+  const double spatialbandwidth = atof(argv[6]);
+  const double threshold = atof(argv[7]);
+
+  /* conencted component parameters */
+
+  const char * maskexpression = argv[8];
+  const char * segmentationexpression = argv[9];
+  unsigned int minobjectsize = atoi(argv[10]);
+  const char * obiaexpression = argv[11];
+  unsigned int nbstreams = atoi(argv[12]);
+
+  // add meanshift options
+
+
+  // Instantiating object
+  MeanShiftFilterType::Pointer meanShiftFilter = MeanShiftFilterType::New();
+  ReaderType::Pointer reader = ReaderType::New();
+
+  reader->SetFileName(infname);
+
+  //define square radius
+  ImageType::SizeType radius;
+  radius[0] = spatialRadius;
+  radius[1] = spatialRadius;
+
+  meanShiftFilter->SetSpatialRadius(radius);
+
+  radius[0] = spectralRadius;
+  radius[1] = spectralRadius;
+  meanShiftFilter->SetRangeRadius(radius);
+  meanShiftFilter->SetSpectralBandwidth(spectralbandwidth);
+  meanShiftFilter->SetSpatialBandwidth(spatialbandwidth);
+  meanShiftFilter->SetThreshold(threshold);
+
+  meanShiftFilter->SetInput(reader->GetOutput());
+  meanShiftFilter->GetRangeOutput();
+
+  ConnectedComponentSegmentationOBIAToVectorDataFilterType::FilterType::Pointer
+      connected = ConnectedComponentSegmentationOBIAToVectorDataFilterType::FilterType::New();
+  connected->GetFilter()->SetInput(meanShiftFilter->GetRangeOutput());
+
+  connected->GetFilter()->SetMaskExpression(maskexpression);
+  connected->GetFilter()->SetConnectedComponentExpression(segmentationexpression);
+  connected->GetFilter()->SetMinimumObjectSize(minobjectsize);
+  connected->GetFilter()->SetOBIAExpression(obiaexpression);
+
+  connected->GetStreamer()->SetNumberOfDivisionsStrippedStreaming(nbstreams);
+  connected->Update();
+
+  VectorDataFileWriterPointerType vdwriter = VectorDataFileWriterType::New();
+  vdwriter->SetInput(connected->GetFilter()->GetOutputVectorData());
+  vdwriter->SetFileName(outputFilename);
+  vdwriter->Update();
+
+  return EXIT_SUCCESS;
+}

Testing/Code/OBIA/otbOBIATests1.cxx

@@ -53,6 +53,7 @@ REGISTER_TEST(otbVectorDataToLabelMapFilterNew);
 REGISTER_TEST(otbVectorDataToLabelMapFilter);
 REGISTER_TEST(otbStreamingConnectedComponentSegmentationOBIAToVectorDataFilterNew);
 REGISTER_TEST(otbStreamingConnectedComponentSegmentationOBIAToVectorDataFilter);
+REGISTER_TEST(otbMeanShiftStreamingConnectedComponentSegmentationOBIAToVectorDataFilter);
 REGISTER_TEST(otbHooverMatrixFilterNew);
 REGISTER_TEST(otbHooverMatrixFilter);
 REGISTER_TEST(otbHooverInstanceFilterNew);