vtkOctreePointLocator.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-FileCopyrightText: Copyright (c) Sandia Corporation
// SPDX-License-Identifier: BSD-3-Clause
 
#ifndef vtkOctreePointLocator_h
#define vtkOctreePointLocator_h
 
#include "vtkAbstractPointLocator.h"
#include "vtkCommonDataModelModule.h" // For export macro
 
VTK_ABI_NAMESPACE_BEGIN
class vtkCellArray;
class vtkIdTypeArray;
class vtkOctreePointLocatorNode;
class vtkPoints;
class vtkPolyData;
 
class VTKCOMMONDATAMODEL_EXPORT vtkOctreePointLocator : public vtkAbstractPointLocator
{
public:
  vtkTypeMacro(vtkOctreePointLocator, vtkAbstractPointLocator);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
  static vtkOctreePointLocator* New();
 
 
  vtkSetMacro(MaximumPointsPerRegion, int);
  vtkGetMacro(MaximumPointsPerRegion, int);
 
 
  vtkSetMacro(CreateCubicOctants, int);
  vtkGetMacro(CreateCubicOctants, int);
 
 
  vtkGetMacro(FudgeFactor, double);
  vtkSetMacro(FudgeFactor, double);
 
 
  double* GetBounds() override;
  void GetBounds(double* bounds) override;
 
 
  vtkGetMacro(NumberOfLeafNodes, int);
 
  void GetRegionBounds(int regionID, double bounds[6]);
 
  void GetRegionDataBounds(int leafNodeID, double bounds[6]);
 
  int GetRegionContainingPoint(double x, double y, double z);
 
  void BuildLocator() override;
 
  void ForceBuildLocator() override;
 
 
  vtkIdType FindClosestPoint(const double x[3]) override;
  vtkIdType FindClosestPoint(double x, double y, double z, double& dist2);
 
  vtkIdType FindClosestPointWithinRadius(double radius, const double x[3], double& dist2) override;
 
 
  vtkIdType FindClosestPointInRegion(int regionId, double* x, double& dist2);
  vtkIdType FindClosestPointInRegion(int regionId, double x, double y, double z, double& dist2);
 
  void FindPointsWithinRadius(double radius, const double x[3], vtkIdList* result) override;
 
  void FindClosestNPoints(int N, const double x[3], vtkIdList* result) override;
 
  VTK_NEWINSTANCE vtkIdTypeArray* GetPointsInRegion(int leafNodeId);
 
  void FreeSearchStructure() override;
 
  void GenerateRepresentation(int level, vtkPolyData* pd) override;
 
  void FindPointsInArea(double* area, vtkIdTypeArray* ids, bool clearArray = true);
 
protected:
  vtkOctreePointLocator();
  ~vtkOctreePointLocator() override;
 
  void BuildLocatorInternal() override;
 
  vtkOctreePointLocatorNode* Top;
  vtkOctreePointLocatorNode** LeafNodeList; // indexed by region/node ID
 
  void BuildLeafNodeList(vtkOctreePointLocatorNode* node, int& index);
 
 
  int FindRegion(vtkOctreePointLocatorNode* node, float x, float y, float z);
  int FindRegion(vtkOctreePointLocatorNode* node, double x, double y, double z);
 
  static void SetDataBoundsToSpatialBounds(vtkOctreePointLocatorNode* node);
 
  static void DeleteAllDescendants(vtkOctreePointLocatorNode* octant);
 
  void FindPointsWithinRadius(
    vtkOctreePointLocatorNode* node, double radiusSquared, const double x[3], vtkIdList* ids);
 
  // Recursive helper for public FindPointsWithinRadius
  void AddAllPointsInRegion(vtkOctreePointLocatorNode* node, vtkIdList* ids);
 
  // Recursive helper for public FindPointsInArea
  void FindPointsInArea(vtkOctreePointLocatorNode* node, double* area, vtkIdTypeArray* ids);
 
  // Recursive helper for public FindPointsInArea
  void AddAllPointsInRegion(vtkOctreePointLocatorNode* node, vtkIdTypeArray* ids);
 
  void DivideRegion(vtkOctreePointLocatorNode* node, int* ordering, int level);
 
  int DivideTest(int size, int level);
 
  void AddPolys(vtkOctreePointLocatorNode* node, vtkPoints* pts, vtkCellArray* polys);
 
  int FindClosestPointInRegion_(int leafNodeId, double x, double y, double z, double& dist2);
 
  int FindClosestPointInSphere(
    double x, double y, double z, double radius, int skipRegion, double& dist2);
 
 
  int MaximumPointsPerRegion;
  int NumberOfLeafNodes;
 
  double FudgeFactor; // a very small distance, relative to the dataset's size
  int NumberOfLocatorPoints;
  float* LocatorPoints;
  int* LocatorIds;
 
  float MaxWidth;
 
  int CreateCubicOctants;
 
  vtkOctreePointLocator(const vtkOctreePointLocator&) = delete;
  void operator=(const vtkOctreePointLocator&) = delete;
};
VTK_ABI_NAMESPACE_END
#endif