testBall3D.cpp

Go to the documentation of this file.

#include <iostream>
#include "DGtal/shapes/parametric/Ball3D.h"
#include "DGtal/helpers/StdDefs.h"
#include "DGtal/shapes/GaussDigitizer.h"
#include "DGtal/io/Color.h"
#include "DGtal/topology/SurfelAdjacency.h"
#include "DGtal/topology/DigitalSurface.h"
#include "DGtal/topology/helpers/BoundaryPredicate.h"
#include "DGtal/topology/SetOfSurfels.h"
#include "DGtal/io/colormaps/GradientColorMap.h"
#include "DGtal/topology/SCellsFunctors.h"
#include "DGtal/io/viewers/PolyscopeViewer.h"
 
 using namespace std;
 using namespace DGtal;
 using namespace Z3i;

// Standard services - public :
 
 
 int main()
 {
   // -------------------------------------------------------------------------- Type declaring
   typedef Space::RealPoint RealPoint;
   typedef Ball3D<Space> EuclideanShape;
   typedef GaussDigitizer<Space,EuclideanShape> DigitalShape;
 
   // -------------------------------------------------------------------------- Creating the shape
    RealPoint c1(0, 0, 0 );
    EuclideanShape ball1( c1, 12.2 );
 
   // -------------------------------------------------------------------------- GaussDigitizing
    DigitalShape dshape;
    dshape.attach( ball1 );
    RealPoint p1 =RealPoint( -15.0, -15.0, -15.0 );
    RealPoint p2 =RealPoint( 15.0, 15.0, 15.0 );
    dshape.init( RealPoint( p1 ), RealPoint( p2 ), 1.0);
    Domain domain = dshape.getDomain();
 
 
   // -------------------------------------------------------------------------- Khalimskhy
    KSpace K;
    bool space_ok = K.init( domain.lowerBound(), domain.upperBound(), true );
    if (!space_ok)
    {
      return 2;
    }
 
 
    // -------------------------------------------------------------------------- Other types
    typedef SurfelAdjacency<KSpace::dimension> MySurfelAdjacency;
    typedef KSpace::Surfel Surfel;
    typedef KSpace::SurfelSet SurfelSet;
    typedef SetOfSurfels< KSpace, SurfelSet > MySetOfSurfels;
 
 
    // -------------------------------------------------------------------------- Tracking the boudnadry
    MySurfelAdjacency surfAdj( true ); // interior in all directions.
    MySetOfSurfels theSetOfSurfels( K, surfAdj );
    Surfel bel = Surfaces<KSpace>::findABel( K, dshape, 1000 );
    Surfaces<KSpace>::trackBoundary( theSetOfSurfels.surfelSet(), K, surfAdj, dshape, bel );
 
 
 
    PolyscopeViewer<> viewer;
//-----------------------------------------------------------------------
// Looking for the min and max values
 
  double minCurv=1;
  double maxCurv=0;
  CanonicSCellEmbedder<KSpace> midpoint(K);
  for ( std::set<SCell>::iterator it = theSetOfSurfels.begin(), it_end = theSetOfSurfels.end();
  it != it_end; ++it)
  {
 
    RealPoint A = midpoint( *it );
    DGtal::StarShaped3D<Space>::AngularCoordinates Angles= ball1.parameter(A);
    double a =ball1.meanCurvature(Angles);
//    double a =ball1.gaussianCurvature(Angles);
 
    if(a>maxCurv)
    {
  maxCurv=a;
    }
    if(a<minCurv)
    {
  minCurv=a;
    }
  }
//-----------------------------------------------------------------------
//Drawing the ball && giving a color to the surfels( depending on the
//curvature)
 
  unsigned int nbSurfels = 0;
 
  viewer.allowReuseList = true;
  for ( std::set<SCell>::iterator it = theSetOfSurfels.begin(), it_end = theSetOfSurfels.end();
it != it_end; ++it, ++nbSurfels )
  {
    RealPoint A = midpoint( *it );
 
    DGtal::StarShaped3D<Space>::AngularCoordinates Angles= ball1.parameter(A);
    double curvature =ball1.meanCurvature(Angles);
    viewer << WithQuantity(*it, "Curvature", curvature);
  }
 
  viewer.show();
  return 0;
}
// //