mitkPointLocator.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitkPointLocator_h
#define mitkPointLocator_h
 
#include <MitkAlgorithmsExtExports.h>
#include <itkObject.h>
 
#include <mitkPointSet.h>
 
#include <vtkPoints.h>
 
// forward declarations
class vtkPointSet;
class ANNkd_tree;
 
namespace mitk
{
  class MITKALGORITHMSEXT_EXPORT PointLocator : public itk::Object
  {
  public:
    mitkClassMacroItkParent(PointLocator, itk::Object);
 
    itkFactorylessNewMacro(Self);
 
    itkCloneMacro(Self);
 
      typedef int IdType;
    typedef ScalarType DistanceType;
 
    typedef float PixelType;
    typedef double CoordRepType;
    typedef itk::DefaultStaticMeshTraits<PixelType, 3, 2, CoordRepType, CoordRepType, PixelType> MeshTraits;
    typedef itk::PointSet<PixelType, 3, MeshTraits> ITKPointSet;
 
    void SetPoints(vtkPointSet *points);
 
    void SetPoints(mitk::PointSet *points);
 
    void SetPoints(ITKPointSet *pointSet);
 
    IdType FindClosestPoint(const double point[3]);
 
    IdType FindClosestPoint(double x, double y, double z);
 
    IdType FindClosestPoint(mitk::PointSet::PointType point);
 
    DistanceType GetMinimalDistance(mitk::PointSet::PointType point);
 
    bool FindClosestPointAndDistance(mitk::PointSet::PointType point, IdType *id, DistanceType *dist);
 
  protected:
    //
    // Definition of a vector of ids
    //
    typedef std::vector<IdType> IdVectorType;
 
    //
    // ANN related typedefs, to prevent ANN from being in the global include path.
    // Please note, that these line are prone to failure, if the point type in
    // ANN changes. Please note also, that these typedefs are only used in the header
    // file. The implementation always refers to the original types
    //
    typedef double *MyANNpoint;
    typedef int MyANNidx;
    typedef double MyANNdist;
    typedef MyANNpoint *MyANNpointArray;
    typedef MyANNidx *MyANNidxArray;
    typedef MyANNdist *MyANNdistArray;
 
    PointLocator();
 
    ~PointLocator() override;
 
    void InitANN();
 
    void DestroyANN();
 
    IdType FindClosestANNPoint(const MyANNpoint &point);
 
    DistanceType GetMinimalDistance(const MyANNpoint &point);
 
    bool m_SearchTreeInitialized;
 
    IdVectorType m_IndexToPointIdContainer;
 
    vtkPoints *m_VtkPoints;
    mitk::PointSet *m_MitkPoints;
    ITKPointSet *m_ItkPoints;
 
    //
    // ANN related variables
    //
    unsigned int m_ANNK;
    unsigned int m_ANNDimension;
    double m_ANNEpsilon;
    MyANNpointArray m_ANNDataPoints;
    MyANNpoint m_ANNQueryPoint;
    MyANNidxArray m_ANNPointIndexes;
    MyANNdistArray m_ANNDistances;
    ANNkd_tree *m_ANNTree;
  };
}
 
#endif