exampleParamCurve3dDigitization.cpp

Go to the documentation of this file.

#include <iostream>
#include <iterator>
#include "DGtal/base/Common.h"
#include "DGtal/helpers/StdDefs.h"
#include "ConfigExamples.h"

#include "DGtal/geometry/curves/parametric/EllipticHelix.h"
#include "DGtal/geometry/curves/parametric/NaiveParametricCurveDigitizer3D.h"
 
#ifdef DGTAL_WITH_POLYSCOPE_VIEWER
  #include "DGtal/io/viewers/PolyscopeViewer.h"
#endif

 
using namespace std;
using namespace DGtal;
using namespace Z3i;
using namespace functors;
 
template <typename T>
inline
unsigned char findMainAxis ( const T & curve, const long double & t )
{
 RealPoint value;
 value[0] = std::abs ( curve.xp ( t )[0] );
 value[1] = std::abs ( curve.xp ( t )[1] );
 value[2] = std::abs ( curve.xp ( t )[2] );
 
 if ( value[0] >= value[1] && value[0] >= value[2] )
  return 0;
 else if ( value[1] >= value[0] && value[1] >= value[2] )
  return 1;
 else
  return 2;
}

int main()
{
 typedef EllipticHelix < Space > MyHelix;
 typedef NaiveParametricCurveDigitizer3D < MyHelix >  DigitizerHelix;
 typedef NaiveParametricCurveDigitizer3D < MyHelix >::DigitalCurve MyDigitalCurve;
 typedef NaiveParametricCurveDigitizer3D < MyHelix >::MetaData MyMetaData;
 trace.info() << "exampleParamCurve3dDigitization" << endl;
 
 PolyscopeViewer<> viewer;

 MyDigitalCurve digitalCurve;
 MyMetaData metaData;
 MyHelix helix( 15, 10, 1 );
 DigitizerHelix digitize;
 digitize.init ( M_PI / 2., ( MyHelix::getPeriod() * 10. ) + M_PI / 2., 0.0001 );
 digitize.attach ( &helix );

 digitize.digitize( back_insert_iterator < MyDigitalCurve> ( digitalCurve ), back_insert_iterator < MyMetaData > ( metaData ) );
 
 trace.info() << "Number of points: " << digitalCurve.size () << " number of metadata: " << metaData.size () << endl;

 for ( unsigned int i = 0; i < digitalCurve.size ( ); i++ )
 {
  if ( findMainAxis ( helix, metaData.at ( i ).first ) == 0 )
   viewer <<  Color ( 255, 0, 0, 128 );
  if ( findMainAxis ( helix, metaData.at ( i ).first ) == 1 )
   viewer << Color ( 0, 255, 0, 128 );
  if ( findMainAxis ( helix, metaData.at ( i ).first ) == 2 )
   viewer << Color ( 0, 0, 255, 128 );
  viewer << digitalCurve.at ( i );
 }
 
 viewer.show();
 return 0;
}
//                                                                           //