cubicalComplexThinning.cpp

Go to the documentation of this file.

#include "ConfigExamples.h"
#include "DGtal/helpers/StdDefs.h"
// Cellular grid
#include "DGtal/topology/CubicalComplex.h"
#include "DGtal/topology/ParDirCollapse.h"
// Shape construction
#include "DGtal/shapes/GaussDigitizer.h"
#include "DGtal/shapes/Shapes.h"
#include "DGtal/shapes/EuclideanShapesDecorator.h"
#include "DGtal/shapes/parametric/Flower2D.h"
// Drawing
#include "DGtal/io/boards/Board2D.h"
 
using namespace std;
using namespace DGtal;
using namespace Z2i;

 
template <typename CC, typename KSpace>
void getComplex ( CC & complex, KSpace & K )
{
  typedef Flower2D< Space > MyEuclideanShape;
  MyEuclideanShape shape( RealPoint( 0.0, 0.0 ), 16, 5, 5, M_PI_2/2. );
 
  typedef GaussDigitizer< Space, MyEuclideanShape > MyGaussDigitizer;
  MyGaussDigitizer digShape;
  digShape.attach( shape );
  digShape.init ( shape.getLowerBound(), shape.getUpperBound(), 1.0 );
  Domain domainShape = digShape.getDomain();
  DigitalSet aSet( domainShape );
  Shapes<Domain>::digitalShaper( aSet, digShape );
 
  K.init ( domainShape.lowerBound(), domainShape.upperBound(), true );
  complex.clear();
  complex.construct ( aSet );
}
 
template <typename CC>
void drawComplex ( Board2D & board, CC & complex )
{
    board.clear();
    typedef typename CC::CellMapConstIterator CellMapConstIterator;
    for ( Dimension d = 0; d <= 2; ++d )
        for ( CellMapConstIterator it = complex.begin( d ), itE = complex.end( d );
                it != itE; ++it )
        {
            if ( d == 0 )
                board << CustomStyle( it->first.className(),
                        new CustomColors( Color( 0, 0, 0 ),
                            Color( 0, 0, 0 ) ) );
            else if ( d == 1 )
                board << CustomStyle( it->first.className(),
                        new CustomColors( Color( 200, 0, 0 ),
                            Color( 100, 255, 100 ) ) );
            else
                board << CustomStyle( it->first.className(),
                        new CustomColors( Color( 0, 0, 200 ),
                            Color( 100, 255, 100 ) ) );
            board << it->first;
        }
}

//                                                                           //
 
 
int main( int, char** )
{
  typedef map<Cell, CubicalCellData>   Map;
  typedef CubicalComplex< KSpace, Map >     CC;
  Board2D board;
  KSpace K;
  CC complex ( K );
  ParDirCollapse < CC > thinning ( K );
  trace.beginBlock ( "ParDirCollapse -- 2 iterations." );
    getComplex< CC, KSpace > ( complex, K );
    drawComplex<CC> ( board, complex );
    board.saveEPS ( "ComplexBeforeThinning.eps" );
    thinning.attach ( &complex );

    thinning.eval ( 2 );
    drawComplex<CC> ( board, complex );
    board.saveEPS ( "ParDirCollapse_2.eps" );
  trace.endBlock();
 
  trace.beginBlock ( "ParDirCollapse -- collapseSurface." );
    getComplex< CC, KSpace > ( complex, K );
    thinning.attach ( &complex );
    thinning.collapseSurface ();
    drawComplex<CC> ( board, complex );
    board.saveEPS ( "ParDirCollapse_collapseSurface.eps" );
  trace.endBlock();
 
  trace.beginBlock ( "ParDirCollapse -- collapseIsthmus." );
    getComplex< CC, KSpace > ( complex, K );
    thinning.attach ( &complex );
    thinning.collapseIsthmus ();
    drawComplex<CC> ( board, complex );
    board.saveEPS ( "ParDirCollapse_collapseIsthmus.eps" );
  trace.endBlock();
  return 0;
}