vtkKdTreePointLocator.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
 
#ifndef vtkKdTreePointLocator_h
#define vtkKdTreePointLocator_h
 
#include "vtkAbstractPointLocator.h"
#include "vtkCommonDataModelModule.h" // For export macro
 
#include <memory> // For std::unique_ptr
 
VTK_ABI_NAMESPACE_BEGIN
class vtkIdList;
 
class VTKCOMMONDATAMODEL_EXPORT vtkKdTreePointLocator : public vtkAbstractPointLocator
{
public:
  vtkTypeMacro(vtkKdTreePointLocator, vtkAbstractPointLocator);
  static vtkKdTreePointLocator* New();
  void PrintSelf(ostream& os, vtkIndent indent) override;

  vtkIdType FindClosestPoint(const double x[3]) override;

  vtkIdType FindClosestPointWithinRadius(double radius, const double x[3], double& dist2) override;

  void FindClosestNPoints(int N, const double x[3], vtkIdList* result) override;

  void FindPointsWithinRadius(double R, const double x[3], vtkIdList* result) override;

  double FindNPointsInShell(int N, const double x[3], vtkDist2TupleArray& results,
    double minDist2 = -0.1, bool sort = true, vtkDoubleArray* petals = nullptr) override;

  void FindPointsInShell(const double x[3], vtkDist2TupleArray& results, double minDist2,
    double maxDist2, bool sort = true, vtkDoubleArray* petals = nullptr) override;


  vtkSetClampMacro(NumberOfPointsPerLeaf, int, 1, VTK_INT_MAX);
  vtkGetMacro(NumberOfPointsPerLeaf, int);


  void FreeSearchStructure() override;
  void BuildLocator() override;
  void ForceBuildLocator() override;
  void GenerateRepresentation(int level, vtkPolyData* pd) override;

  int GetDimension();
 
protected:
  vtkKdTreePointLocator();
  ~vtkKdTreePointLocator() override;
 
  void BuildLocatorInternal() override;
 
  int NumberOfPointsPerLeaf = 10;
 
  struct vtkInternals;
  std::unique_ptr<vtkInternals> Internals;
 
private:
  vtkKdTreePointLocator(const vtkKdTreePointLocator&) = delete;
  void operator=(const vtkKdTreePointLocator&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif