mitkPointLocator.h

Go to the documentation of this file.

/*=========================================================================
 
Program:   Medical Imaging & Interaction Toolkit
Language:  C++
Date:      $Date$
Version:   $Revision$
 
Copyright (c) German Cancer Research Center, Division of Medical and
Biological Informatics. All rights reserved.
See MITKCopyright.txt or https://www.mitk.org/copyright.html for details.
 
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE.  See the above copyright notices for more information.
 
=========================================================================*/

#ifndef _MITK_POINT_LOCATOR__H__
#define _MITK_POINT_LOCATOR__H__

#include <itkObject.h>
#include "MitkExtExports.h"

#include "mitkPointSet.h"

#include <vtkPoints.h>


//forward declarations
class vtkPointSet;
class ANNkd_tree;

namespace mitk
{  
  
class MitkExt_EXPORT PointLocator : public itk::Object
{
public:
    
  mitkClassMacro( PointLocator, Object );

  itkNewMacro( Self );
  
  typedef int IdType;
  typedef float 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 vtkFloatingPointType point[3] );
  
  IdType FindClosestPoint( vtkFloatingPointType x, vtkFloatingPointType y, vtkFloatingPointType 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 float*      MyANNpoint;
  typedef int         MyANNidx;
  typedef float       MyANNdist;
  typedef MyANNpoint* MyANNpointArray;
  typedef MyANNidx*   MyANNidxArray;
  typedef MyANNdist*  MyANNdistArray;
  
  PointLocator();
  
  ~PointLocator();
  
  void InitANN();
  
  void DestroyANN();  
  
  IdType FindClosestPoint( 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