testParallelIntegralInvariantEstimator.cpp

 
#include <iostream>
#include <type_traits>
#include "DGtal/base/Common.h"
 
#include "DGtal/shapes/implicit/ImplicitBall.h"
 
#include "DGtal/shapes/GaussDigitizer.h"
#include "DGtal/topology/LightImplicitDigitalSurface.h"
#include "DGtal/topology/DigitalSurface.h"
#include "DGtal/graph/DepthFirstVisitor.h"
#include "DGtal/graph/GraphVisitorRange.h"
 
#include "DGtal/geometry/surfaces/estimation/IIGeometricFunctors.h"
#include "DGtal/geometry/surfaces/estimation/IntegralInvariantVolumeEstimator.h"
#include "DGtal/geometry/surfaces/estimation/ParallelIIEstimator.h"
#include "DGtal/kernel/domains/DomainSplitter.h"
 
 
 
 
using namespace DGtal;
 
// Functions for testing class IntegralInvariantVolumeEstimator and IIGeometricFunctor.
 
bool testCurvature2dP ( double h )
{
  typedef ImplicitBall<Z2i::Space> ImplicitShape;
  typedef GaussDigitizer<Z2i::Space, ImplicitShape> DigitalShape;
  typedef LightImplicitDigitalSurface<Z2i::KSpace,DigitalShape> Boundary;
  typedef DigitalSurface< Boundary > MyDigitalSurface;
  typedef DepthFirstVisitor< MyDigitalSurface > Visitor;
  typedef GraphVisitorRange< Visitor > VisitorRange;
  typedef VisitorRange::ConstIterator VisitorConstIterator;
 
  typedef functors::IICurvatureFunctor<Z2i::Space> MyIICurvatureFunctor;
  typedef IntegralInvariantVolumeEstimator< Z2i::KSpace, DigitalShape, MyIICurvatureFunctor > MyIICurvatureEstimator;
 
  typedef RegularDomainSplitter<HyperRectDomain<Z2i::Space>> Splitter;
  typedef ParallelIIEstimator<MyIICurvatureEstimator, Splitter> MyIICurvatureEstimatorP;
 
  typedef MyIICurvatureEstimator::Quantity Value;
 
  static_assert(std::is_same_v<typename MyIICurvatureEstimator::Quantity, typename MyIICurvatureEstimatorP::Quantity>);
 
  double re = 10;
  double radius = 15;
 
  trace.beginBlock( "[PARALLEL] Shape initialisation ..." );
 
  ImplicitShape ishape( Z2i::RealPoint( 0, 0 ), radius );
  DigitalShape dshape;
  dshape.attach( ishape );
  dshape.init( Z2i::RealPoint( -20.0, -20.0 ), Z2i::RealPoint( 20.0, 20.0 ), h );
 
  Z2i::KSpace K;
  if ( !K.init( dshape.getLowerBound(), dshape.getUpperBound(), true ) )
  {
    trace.error() << "Problem with Khalimsky space" << std::endl;
    return false;
  }
 
  Z2i::KSpace::Surfel bel = Surfaces<Z2i::KSpace>::findABel( K, dshape, 10000 );
  Boundary boundary( K, dshape, SurfelAdjacency<Z2i::KSpace::dimension>( true ), bel );
  MyDigitalSurface surf ( boundary );
 
  trace.endBlock();
 
  trace.beginBlock( "Curvature estimator initialisation ...");
 
  // Visitor ranges are typically unique and single pass. We need
  // to create one for each estimator in this case
  VisitorRange range( new Visitor( surf, *surf.begin() ));
  VisitorConstIterator ibegin = range.begin();
  VisitorConstIterator iend = range.end();
  // Parallel iterations
  VisitorRange rangeP( new Visitor( surf, *surf.begin() ));
  VisitorConstIterator ibeginP = rangeP.begin();
  VisitorConstIterator iendP = rangeP.end();
 
  MyIICurvatureFunctor curvatureFunctor;
  curvatureFunctor.init( h, re );
 
  MyIICurvatureEstimator  curvatureEstimator (    curvatureFunctor);
  curvatureEstimator.attach( K, dshape );
  curvatureEstimator.setParams( re/h );
  curvatureEstimator.init( h, ibegin, iend );
 
  // Parallel version expects a domain splitter instance and a number of
  // threads. Subsequent arguments are forwarded to the underlying estimator.
  Splitter splitter;
  MyIICurvatureEstimatorP curvatureEstimatorP( splitter, 4, curvatureFunctor );
  curvatureEstimatorP.attach( K, dshape );
  curvatureEstimatorP.setParams( re/h );
  curvatureEstimatorP.init( h, ibeginP, iendP );
 
  trace.endBlock();
 
  trace.beginBlock( "Curvature estimator evaluation ...");
 
  std::vector< Value > results, resultsP;
  std::back_insert_iterator< std::vector< Value > > resultsIt ( results  );
  std::back_insert_iterator< std::vector< Value > > resultsItP( resultsP );
 
  curvatureEstimator .eval( ibegin , iend , resultsIt  );
  curvatureEstimatorP.eval( ibeginP, iendP, resultsItP );
 
  trace.endBlock();
 
  trace.beginBlock ( "Comparing results of integral invariant 2D curvature ..." );
 
  unsigned int rsize  = results.size();
  unsigned int rsizeP = resultsP.size();
 
  if (rsize != rsizeP)
  {
    trace.error() << "Size mismatch between parallel and non-parallel versions: " << rsize << " / " << rsizeP;
    trace.endBlock();
    return false;
  }
 
  for ( unsigned int i = 0; i < rsize; ++i )
  {
    if (std::abs(results[i] - resultsP[i]) >= 1e-2)
    {
      trace.error() << "Result mismatch between parallel and non-parallel versions at voxel " << i << ": " << results[i] << " / " << resultsP[i] << "\n";
      trace.endBlock();
      return false;
    }
  }
  trace.endBlock();
  return true;
}
 
// Standard services - public :
int main( int /*argc*/, char** /*argv*/ )
{
    trace.beginBlock ( "Testing class ParrallelIIEstimator with IntegralInvariantVolumeEstimator in 2d" );
 
    bool res = testCurvature2dP( 0.05 );
    trace.emphase() << ( res ? "Passed." : "Error." ) << std::endl;
    trace.endBlock();
    return res ? 0 : 1;
}