testArithDSS3d.cpp

#include <iostream>
#include <iterator>
#include <cstdio>
#include <cmath>
#include <fstream>
#include <vector>

#include "DGtal/base/Common.h"
#include "DGtal/base/Exceptions.h"
#include "DGtal/kernel/SpaceND.h"
#include "DGtal/kernel/domains/HyperRectDomain.h"
#include "DGtal/geometry/curves/ArithmeticalDSS3d.h"
#include "DGtal/geometry/curves/GreedySegmentation.h"

using namespace DGtal;
using namespace std;


// Functions for testing class ArithmeticalDSS3d.

bool testDSSreco()
{

  typedef PointVector<3,int> Point;
  typedef std::vector<Point>::iterator Iterator;
  typedef ArithmeticalDSS3d<Iterator,int,4> SegmentComputer;  
  
  std::vector<Point> sequence;
  sequence.push_back(Point(0,0,0));
  sequence.push_back(Point(1,0,0));
  sequence.push_back(Point(2,0,0));
  sequence.push_back(Point(2,1,0));
  sequence.push_back(Point(2,1,1));
  sequence.push_back(Point(3,1,1));
  sequence.push_back(Point(4,1,1));
  sequence.push_back(Point(4,2,1));
  sequence.push_back(Point(4,2,2));
  sequence.push_back(Point(5,2,2));
  sequence.push_back(Point(6,2,2));
  sequence.push_back(Point(6,3,2));
  sequence.push_back(Point(6,3,3));
  sequence.push_back(Point(6,4,3));
  sequence.push_back(Point(6,4,4));
  sequence.push_back(Point(6,5,4));
  
  // Adding step
  trace.beginBlock("Add points while it is possible and display the result");

  SegmentComputer algo;  
  Iterator i = sequence.begin();  
  algo.init(i);
  trace.info() << "init with " << (*i) << std::endl;

    while ( (algo.end() != sequence.end())
            && algo.extendForward()) {
      trace.info() << "extended with " << (*(--algo.end())) << std::endl;
    }
    

    trace.info() << algo << " " << algo.isValid() << std::endl;

    trace.endBlock();

  return ( algo.isValid() && (algo.end() == (sequence.begin()+13)) );  
}


bool testSegmentation()
{

  typedef PointVector<3,int> Point;
  typedef std::vector<Point>::iterator Iterator;
  typedef ArithmeticalDSS3d<Iterator,int,4> SegmentComputer;  
  typedef GreedySegmentation<SegmentComputer> Decomposition;

  std::vector<Point> sequence;
  sequence.push_back(Point(0,0,0));
  sequence.push_back(Point(1,0,0));
  sequence.push_back(Point(2,0,0));
  sequence.push_back(Point(2,1,0));
  sequence.push_back(Point(2,1,1));
  sequence.push_back(Point(3,1,1));
  sequence.push_back(Point(4,1,1));
  sequence.push_back(Point(4,2,1));
  sequence.push_back(Point(4,2,2));
  sequence.push_back(Point(5,2,2));
  sequence.push_back(Point(6,2,2));
  sequence.push_back(Point(6,3,2));
  sequence.push_back(Point(6,3,3));
  sequence.push_back(Point(6,4,3));
  sequence.push_back(Point(6,4,4));
  sequence.push_back(Point(6,5,4));
  
  //Segmentation
  trace.beginBlock("Segmentation test");
    
    SegmentComputer algo;
    Decomposition theDecomposition(sequence.begin(), sequence.end(), algo);
           
    unsigned int c = 0;
    Decomposition::SegmentComputerIterator i = theDecomposition.begin();
    for ( ; i != theDecomposition.end(); ++i) {
      SegmentComputer currentSegmentComputer(*i);
      trace.info() << currentSegmentComputer << std::endl;  //standard output
      c++;
    } 

  trace.endBlock();
  return (c==2);
}


void testArithmeticalDSS3dConceptChecking()
{
  typedef PointVector<3,int> Point;
  typedef std::vector<Point>::const_iterator ConstIterator; 
  typedef ArithmeticalDSS3d<ConstIterator,int,4> ArithDSS3d; 

  trace.beginBlock("Concept checking");

  BOOST_CONCEPT_ASSERT(( CForwardSegmentComputer<ArithDSS3d> ));
  
  trace.endBlock();

}



int main(int argc, char **argv)
{
  trace.beginBlock ( "Testing class ArithmeticalDSS" );
 
  testArithmeticalDSS3dConceptChecking();  
    
  trace.info() << "Args:";
  for ( int i = 0; i < argc; ++i )
    trace.info() << " " << argv[ i ];
  trace.info() << endl;

  bool res = testDSSreco() 
        && testSegmentation()

  ;
  trace.emphase() << ( res ? "Passed." : "Error." ) << endl;
  trace.endBlock();

  return res ? 0 : 1;

}