ElementaryConvolutionNormalVectorEstimator.h

 
#pragma once
 
#if defined(ElementaryConvolutionNormalVectorEstimator_RECURSES)
#error Recursive header files inclusion detected in ElementaryConvolutionNormalVectorEstimator.h
#else // defined(ElementaryConvolutionNormalVectorEstimator_RECURSES)
 
#define ElementaryConvolutionNormalVectorEstimator_RECURSES
 
#if !defined ElementaryConvolutionNormalVectorEstimator_h
 
#define ElementaryConvolutionNormalVectorEstimator_h
 
// Inclusions
#include <iostream>
#include <DGtal/base/Common.h>
#include <DGtal/topology/SCellsFunctors.h>
#include <vector>
 
 
namespace DGtal
{
  namespace functors
  {
  // template class ElementaryConvolutionNormalVectorEstimator
  template <typename TSurfel, typename TEmbedder>
  class ElementaryConvolutionNormalVectorEstimator
  {
  public:
 
    typedef TSurfel Surfel;
    typedef TEmbedder SCellEmbedder;
    typedef typename SCellEmbedder::RealPoint RealPoint;
    typedef RealPoint Quantity;
 
    ElementaryConvolutionNormalVectorEstimator(ConstAlias<SCellEmbedder> anEmbedder,
                                               const double h):
      myEmbedder(&anEmbedder), myH(h)
    {
    }
 
    void pushSurfel(const Surfel & aSurf,
                    const double aDistance)
    {
      RealPoint elementary;
      Dimension i = myEmbedder->space().sOrthDir ( aSurf );
      elementary[ i ] = myEmbedder->space().sDirect ( aSurf, i ) ? 1 : -1;
 
      myWeightedVector += aDistance *elementary;
    }
 
    Quantity eval( )
    {
      return myWeightedVector.getNormalized();
    }
 
    void reset()
    {
      myWeightedVector = RealPoint().diagonal(0.0);
    }
 
 
  private:
 
    const SCellEmbedder * myEmbedder;
 
    RealPoint myWeightedVector;
 
    //Grid step
    double myH;
 
  }; // end of class ElementaryConvolutionNormalVectorEstimator
  } //namespace functors
} // namespace DGtal
 
 
//                                                                           //
 
#endif // !defined ElementaryConvolutionNormalVectorEstimator_h
 
#undef ElementaryConvolutionNormalVectorEstimator_RECURSES
#endif // else defined(ElementaryConvolutionNormalVectorEstimator_RECURSES)