imageGridCurveEstimator.cpp

#include <iostream>
#include <fstream>
#include <algorithm>

#include "DGtal/base/Common.h"
#include "DGtal/helpers/StdDefs.h"
#include "ConfigExamples.h"

#include "DGtal/base/BasicFunctors.h"
#include "DGtal/kernel/BasicPointPredicates.h"
#include "DGtal/io/readers/PGMReader.h"
#include "DGtal/images/ImageContainerBySTLVector.h"

#include "DGtal/topology/helpers/Surfaces.h"

#include "DGtal/geometry/curves/estimation/DSSLengthEstimator.h"

//display
#include "DGtal/io/boards/Board2D.h"

//segmentation
#include "DGtal/geometry/curves/GreedySegmentation.h"


using namespace DGtal;
using namespace DGtal::functors;

int main()
{
  //image import
  typedef DGtal::ImageContainerBySTLVector< Z2i::Domain, int> Image;
  std::string filename =  examplesPath + "samples/contourS.pgm";
  Image image = DGtal::PGMReader<Image>::importPGM(filename); 

  //predicate from the image
  typedef DGtal::functors::IntervalThresholder<Image::Value> Binarizer;
  Binarizer b(1, 135); 
  PointFunctorPredicate<Image,Binarizer> predicate(image, b); 

  Z2i::KSpace ks;                                            //Khalimsky space 
  ks.init( image.domain().lowerBound(), image.domain().upperBound(), true );
  SurfelAdjacency<2> sAdj( true );                           //adjacency

  //extraction of all the contours
  std::vector< std::vector< Z2i::SCell > > contours;
  Surfaces<Z2i::KSpace>
    ::extractAll2DSCellContours( contours, ks, sAdj, predicate );

  if (contours.size() > 0)
  {
    
    //init grid curve from the first retrieved contour
    Z2i::Curve c;
    c.initFromSCellsVector( contours.at(1) );  

    //range of points
    typedef Z2i::Curve::PointsRange Range; 
    Range r = c.getPointsRange(); 

    //length estimation based on a DSS segmentation
    DSSLengthEstimator< Range::ConstCirculator > DSSlength;
    DSSlength.init(1, r.c(), r.c());
    trace.info() << "Length: " << DSSlength.eval() << std::endl; 

    //DSS segmentation display
    typedef ArithmeticalDSSComputer<Range::ConstCirculator,int,4> SegmentComputer;
    typedef GreedySegmentation<SegmentComputer> Segmentation;

    Segmentation theSegmentation( r.c(), r.c(), SegmentComputer() );
    Segmentation::SegmentComputerIterator i = theSegmentation.begin();
    Segmentation::SegmentComputerIterator end = theSegmentation.end();
    
    DGtal::Board2D aBoard;
    aBoard << SetMode("PointVector", "Grid");
    for ( ; i != end; ++i) {
      SegmentComputer::Primitive dss = i->primitive(); 
      aBoard << SetMode(dss.className(), "Points") << dss; 
      aBoard << SetMode(dss.className(), "BoundingBox") << dss; 
    } 
    aBoard.saveEPS("DisplayDSSSegmentationTuto3.eps");
  
  } else trace.info() << "no contour" << std::endl; 
  
  return 0;

}