TEST: add test for otb::SamplingRateCalculatorList

Modules/Learning/Sampling/src/CMakeLists.txt

 set(OTBSampling_SRC
   otbSamplingRateCalculator.cxx
+  otbSamplingRateCalculatorList.cxx
 )
 
 add_library(OTBSampling ${OTBSampling_SRC})

Modules/Learning/Sampling/test/CMakeLists.txt

@@ -6,6 +6,7 @@ otbOGRDataToSamplePositionFilterTest.cxx
 otbSamplingRateCalculatorTest.cxx
 otbOGRDataToClassStatisticsFilterTest.cxx
 otbImageSampleExtractorFilterTest.cxx
+otbSamplingRateCalculatorListTest.cxx
 )
 
 add_executable(otbSamplingTestDriver ${OTBSamplingTests})
@@ -92,3 +93,16 @@ otb_add_test(NAME leTvImageSampleExtractorFilterUpdate COMMAND otbSamplingTestDr
   otbImageSampleExtractorFilterUpdate
   ${INPUTDATA}/variousVectors.sqlite
   ${TEMP}/leTvImageSampleExtractorFilterUpdateTest.shp)
+
+# ---------------- SamplingRateCalculatorList ---------------------------------
+otb_add_test(NAME leTuSamplingRateCalculatorListNew COMMAND otbSamplingTestDriver
+            otbSamplingRateCalculatorListNew 
+  
+  )
+
+otb_add_test(NAME leTvSamplingRateCalculatorList COMMAND otbSamplingTestDriver
+  --compare-ascii ${NOTOL}
+  ${BASELINE_FILES}/leTvSamplingRateCalculatorList.txt
+  ${TEMP}/leTvSamplingRateCalculatorList.txt
+  otbSamplingRateCalculatorList
+  ${TEMP}/leTvSamplingRateCalculatorList.txt)

Modules/Learning/Sampling/test/otbSamplingRateCalculatorListTest.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 "otbSamplingRateCalculatorList.h"
+
+#include "otbStatisticsXMLFileReader.h"
+#include "itkVariableLengthVector.h"
+#include <fstream>
+
+int otbSamplingRateCalculatorListNew(int itkNotUsed(argc), char* itkNotUsed(argv) [])
+{
+  typedef otb::SamplingRateCalculatorList RateCalculatorListType;
+  
+  RateCalculatorListType::Pointer rateCalculator = RateCalculatorListType::New();
+  std::cout << rateCalculator << std::endl;
+  return EXIT_SUCCESS;
+}
+
+int otbSamplingRateCalculatorList(int itkNotUsed(argc), char* argv[])
+{
+  std::ofstream file(argv[1], std::ios::out | std::ios::trunc);
+
+  if (!file)
+    {
+    std::cout << " Couldn't open " << argv[1];
+    return EXIT_FAILURE;
+    }
+
+  itk::Indent indent_1(1);
+
+  typedef otb::SamplingRateCalculatorList::ClassCountMapType  ClassCountMapType;
+  typedef otb::SamplingRateCalculatorList::PartitionType      PartitionType;
+
+  PartitionType typeProportional = otb::SamplingRateCalculatorList::PROPORTIONAL;
+  PartitionType typeEqual = otb::SamplingRateCalculatorList::EQUAL;
+  PartitionType typeCustom = otb::SamplingRateCalculatorList::CUSTOM;
+
+  std::string c1("1");
+  std::string c2("2");
+  std::string c3("3");
+  std::string c4("4");
+
+  ClassCountMapType classCount1, classCount2, classCount3;
+  // input 1
+  classCount1[c1] = 104;
+  classCount1[c2] = 160;
+  classCount1[c3] = 211;
+  // input 2
+  classCount2[c1] = 98;
+  classCount2[c2] = 190;
+  classCount2[c3] = 178;
+  // input 3
+  classCount3[c1] = 130;
+  classCount3[c2] = 144;
+  classCount3[c4] = 250;
+
+  std::vector<unsigned long> nbSamplesCst;
+  nbSamplesCst.push_back(151);
+
+  std::vector<unsigned long> nbSamplesCstCustom;
+  nbSamplesCstCustom.push_back(70);
+  nbSamplesCstCustom.push_back(80);
+  nbSamplesCstCustom.push_back(90);
+
+  ClassCountMapType needed1;
+  needed1[c1] = 140;
+  needed1[c2] = 130;
+  needed1[c3] = 124;
+
+  std::vector<ClassCountMapType> nbByClass;
+  nbByClass.push_back(needed1);
+
+  ClassCountMapType needed2;
+  needed2[c1] = 67;
+  needed2[c2] = 55;
+  needed2[c3] = 72;
+
+  ClassCountMapType needed3;
+  needed3[c1] = 77;
+  needed3[c2] = 69;
+  needed3[c3] = 34;
+
+  ClassCountMapType needed4;
+  needed4[c1] = 86;
+  needed4[c2] = 43;
+  needed4[c4] = 98;
+
+  std::vector<ClassCountMapType> nbByClassCustom;
+  nbByClassCustom.push_back(needed2);
+  nbByClassCustom.push_back(needed3);
+  nbByClassCustom.push_back(needed4);
+
+  typedef otb::SamplingRateCalculatorList RateCalculatorListType;
+  RateCalculatorListType::Pointer rateCalcList = RateCalculatorListType::New();
+  rateCalcList->SetNthClassCount(0,classCount1);
+  rateCalcList->SetNthClassCount(2,classCount3);
+  rateCalcList->SetNthClassCount(1,classCount2);
+
+  file << "# Test the strategy : smallest - equal" << std::endl;
+  rateCalcList->SetMinimumNbOfSamplesByClass(typeEqual);
+  for (unsigned int i=0 ; i<rateCalcList->Size() ; i++)
+    {
+    file << "# Input "<< i << std::endl;
+    rateCalcList->GetNthElement(i)->Print(file, indent_1);
+    }
+
+  file << "# Test the strategy : smallest - custom" << std::endl;
+  rateCalcList->SetMinimumNbOfSamplesByClass(typeCustom);
+  for (unsigned int i=0 ; i<rateCalcList->Size() ; i++)
+    {
+    file << "# Input "<< i << std::endl;
+    rateCalcList->GetNthElement(i)->Print(file, indent_1);
+    }
+
+  file << "# Test the strategy : constant - proportional" << std::endl;
+  rateCalcList->SetNbOfSamplesAllClasses(nbSamplesCst,typeProportional);
+  for (unsigned int i=0 ; i<rateCalcList->Size() ; i++)
+    {
+    file << "# Input "<< i << std::endl;
+    rateCalcList->GetNthElement(i)->Print(file, indent_1);
+    }
+
+  file << "# Test the strategy : constant - equal" << std::endl;
+  rateCalcList->SetNbOfSamplesAllClasses(nbSamplesCst,typeEqual);
+  for (unsigned int i=0 ; i<rateCalcList->Size() ; i++)
+    {
+    file << "# Input "<< i << std::endl;
+    rateCalcList->GetNthElement(i)->Print(file, indent_1);
+    }
+
+  file << "# Test the strategy : constant - custom" << std::endl;
+  rateCalcList->SetNbOfSamplesAllClasses(nbSamplesCstCustom,typeCustom);
+  for (unsigned int i=0 ; i<rateCalcList->Size() ; i++)
+    {
+    file << "# Input "<< i << std::endl;
+    rateCalcList->GetNthElement(i)->Print(file, indent_1);
+    }
+
+  file << "# Test the strategy : byClass - proportional" << std::endl;
+  rateCalcList->SetNbOfSamplesByClass(nbByClass,typeProportional);
+  for (unsigned int i=0 ; i<rateCalcList->Size() ; i++)
+    {
+    file << "# Input "<< i << std::endl;
+    rateCalcList->GetNthElement(i)->Print(file, indent_1);
+    }
+
+  file << "# Test the strategy : byClass - equal" << std::endl;
+  rateCalcList->SetNbOfSamplesByClass(nbByClass,typeEqual);
+  for (unsigned int i=0 ; i<rateCalcList->Size() ; i++)
+    {
+    file << "# Input "<< i << std::endl;
+    rateCalcList->GetNthElement(i)->Print(file, indent_1);
+    }
+
+  file << "# Test the strategy : byClass - custom" << std::endl;
+  rateCalcList->SetNbOfSamplesByClass(nbByClassCustom,typeCustom);
+  for (unsigned int i=0 ; i<rateCalcList->Size() ; i++)
+    {
+    file << "# Input "<< i << std::endl;
+    rateCalcList->GetNthElement(i)->Print(file, indent_1);
+    }
+
+  file << "# Test the strategy : all - equal" << std::endl;
+  rateCalcList->SetAllSamples(typeEqual);
+  for (unsigned int i=0 ; i<rateCalcList->Size() ; i++)
+    {
+    file << "# Input "<< i << std::endl;
+    rateCalcList->GetNthElement(i)->Print(file, indent_1);
+    }
+
+  file.close();
+  return EXIT_SUCCESS;
+}

Modules/Learning/Sampling/test/otbSamplingTestDriver.cxx

@@ -11,4 +11,6 @@ void RegisterTests()
   REGISTER_TEST(otbImageSampleExtractorFilterNew);
   REGISTER_TEST(otbImageSampleExtractorFilter);
   REGISTER_TEST(otbImageSampleExtractorFilterUpdate);
+  REGISTER_TEST(otbSamplingRateCalculatorListNew);
+  REGISTER_TEST(otbSamplingRateCalculatorList);
 }