exampleParamCurve3dDigitizationTransformationDecorator.cpp

Go to the documentation of this file.

#include <iostream>
#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"
#include "DGtal/geometry/curves/parametric/DecoratorParametricCurveTransformation.h"
#include "DGtal/images/RigidTransformation3D.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 ForwardRigidTransformation3D < Space, RealPoint, RealPoint, Identity > ForwardTrans;
 typedef DecoratorParametricCurveTransformation < MyHelix, ForwardTrans > MyRotatedCurve;
 typedef NaiveParametricCurveDigitizer3D < MyRotatedCurve >  Digitizer;
 typedef typename NaiveParametricCurveDigitizer3D < MyRotatedCurve >::DigitalCurve MyDigitalCurve;
 typedef NaiveParametricCurveDigitizer3D < MyRotatedCurve >::MetaData MyMetaData;
 trace.info() << "exampleParamCurve3dDigitizationTransformationDecorator" << endl;
 
 PolyscopeViewer<> viewer;
 
 MyMetaData metaData;

 MyDigitalCurve digitalCurve;
 MyHelix helix( 30, 20, 1 );

 double angle = M_PI/3.;
 RealVector axis ( 1., 0., 1. );
 ForwardTrans trans ( RealPoint ( 0, 0, 0 ), axis, angle, RealVector ( 0, 0, 0 ) );
 MyRotatedCurve rotCurve ( helix, trans );

 Digitizer digitize;
 digitize.attach ( &rotCurve );
 digitize.init ( 0, MyHelix::getPeriod() * 10., 0.0001 );
 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 ( rotCurve, metaData.at ( i ).first ) == 0 )
   viewer.drawColor ( Color ( 255, 0, 0, 128 ) );
  if ( findMainAxis ( rotCurve, metaData.at ( i ).first ) == 1 )
   viewer.drawColor ( Color ( 0, 255, 0, 128 ) );
  if ( findMainAxis ( rotCurve, metaData.at ( i ).first ) == 2 )
   viewer.drawColor ( Color ( 0, 0, 255, 128 ) );
  viewer << digitalCurve.at ( i );
 }
 
 viewer.show();
 return 0;
}
//                                                                           //