testLambdaMST2D.cpp

Go to the documentation of this file.

#include <iostream>
#include <fstream>
#include <iterator>
#include <vector>
#include "DGtal/base/Common.h"
#include "ConfigTest.h"
#include "DGtal/helpers/StdDefs.h"
#include "DGtal/io/readers/PointListReader.h"
#include "DGtal/geometry/curves/ArithmeticalDSSComputer.h"
#include "DGtal/geometry/curves/estimation/LambdaMST2D.h"
#include "DGtal/geometry/curves/SaturatedSegmentation.h"


using namespace std;
using namespace DGtal;
using namespace Z2i;

// Functions for testing class LambdaMST2D.

class testLambdaMST2D
{
  typedef vector < Point > Range;
  typedef Range::const_iterator ConstIterator;
  typedef ArithmeticalDSSComputer < ConstIterator, int, 8 > SegmentComputer;
  typedef SaturatedSegmentation<SegmentComputer> Segmentation;
private:
  Range curve;
public:
  testLambdaMST2D ()
  {
    fstream inputStream;
    inputStream.open ( (testPath + "samples/sinus2D4.dat").c_str(), ios::in );
    try {
      curve = PointListReader<Point>::getPointsFromInputStream ( inputStream );
      if ( curve.size() == 0) throw IOException();
    } catch (DGtal::IOException & ioe) {
      trace.error() << "Size is null." << endl;
    }
  }
  bool lambda64ByPoint ()
  {
    Segmentation segmenter ( curve.begin(), curve.end(), SegmentComputer() );
    LambdaMST2D < Segmentation > lmst64;
    lmst64.attach ( segmenter );
    for ( ConstIterator it = curve.begin(); it != curve.end(); ++it )
      lmst64.eval ( *it );
    return true;
  }
  bool lambda64()
  {
    Segmentation segmenter ( curve.begin(), curve.end(), SegmentComputer() );
    LambdaMST2D < Segmentation > lmst64;
    lmst64.attach ( segmenter );
    lmst64.init ( curve.begin(), curve.end() );
    std::vector < RealVector > tangent;
    lmst64.eval < back_insert_iterator< vector < RealVector > > > ( curve.begin(), curve.end(),  back_inserter ( tangent ) );
    return true;
  }
};


// Standard services - public :

int main( int , char**  )
{
    bool res = true;
    testLambdaMST2D testLMST;
    trace.beginBlock ( "Testing LambdaMST2D" );
        trace.beginBlock ( "Testing point only calculation" );
          res &= testLMST.lambda64ByPoint();
        trace.endBlock();
        trace.beginBlock ( "Testing calculation for whole curve" );
           res &= testLMST.lambda64();
        trace.endBlock();
    trace.emphase() << ( res ? "Passed." : "Error." ) << endl;
    trace.endBlock();
    return res ? 0 : 1;
}
//                                                                           //