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 _MITK_POINT_LOCATOR__H__
#define _MITK_POINT_LOCATOR__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