vtkJumpAndWalkCellLocator.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 vtkJumpAndWalkCellLocator_h
#define vtkJumpAndWalkCellLocator_h

#include "vtkAbstractCellLocator.h"
#include "vtkAbstractPointLocator.h"  // For vtkAbstractPointLocator
#include "vtkCommonDataModelModule.h" // For export macro
#include "vtkSMPThreadLocal.h"        // For vtkSMPThreadLocal
#include "vtkSmartPointer.h"          // For vtkSmartPointer

#include <unordered_set> // For unordered_set

VTK_ABI_NAMESPACE_BEGIN

class vtkAbstractPointLocator;

class VTKCOMMONDATAMODEL_EXPORT vtkJumpAndWalkCellLocator : public vtkAbstractCellLocator
{
public:

  vtkTypeMacro(vtkJumpAndWalkCellLocator, vtkAbstractCellLocator);
  void PrintSelf(ostream& os, vtkIndent indent) override;
  static vtkJumpAndWalkCellLocator* New();

  // Reuse any superclass signatures that we don't override.
  using vtkAbstractCellLocator::FindCell;
  using vtkAbstractCellLocator::FindClosestPointWithinRadius;


  vtkSetSmartPointerMacro(PointLocator, vtkAbstractPointLocator);
  vtkGetSmartPointerMacro(PointLocator, vtkAbstractPointLocator);


  vtkSetClampMacro(NumberOfClosestPoints, unsigned int, 1, 1000);
  vtkGetMacro(NumberOfClosestPoints, unsigned int);

  vtkIdType FindCell(double x[3], double tol2, vtkGenericCell* genCell, int& subId,
    double pcoords[3], double* weights) override;

  vtkIdType FindClosestPointWithinRadius(double x[3], double radius, double closestPoint[3],
    vtkGenericCell* genCell, vtkIdType& cellId, int& subId, double& dist2, int& inside) override;


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

  void ShallowCopy(vtkAbstractCellLocator* locator) override;

protected:
  vtkJumpAndWalkCellLocator();
  ~vtkJumpAndWalkCellLocator() override;

  void BuildLocatorInternal() override;

  void ReportReferences(vtkGarbageCollector*) override;

private:
  vtkJumpAndWalkCellLocator(const vtkJumpAndWalkCellLocator&) = delete;
  void operator=(const vtkJumpAndWalkCellLocator&) = delete;

  vtkIdType FindCellWalk(vtkIdList* cellIds, double x[3], double tol2, vtkGenericCell* genCell,
    int& subId, double pcoords[3], double* weights, std::unordered_set<vtkIdType>& visitedCellIds,
    vtkIdList* ptIds, vtkIdList* neighbors);

  // Bounding box of the whole dataset
  double Bounds[6] = { VTK_DOUBLE_MAX, VTK_DOUBLE_MIN, VTK_DOUBLE_MAX, VTK_DOUBLE_MIN,
    VTK_DOUBLE_MAX, VTK_DOUBLE_MIN };
  unsigned int NumberOfClosestPoints = 1;

  vtkSmartPointer<vtkAbstractPointLocator> PointLocator = nullptr;

  struct ThreadLocalData;
  vtkSMPThreadLocal<ThreadLocalData> TLData;
};
VTK_ABI_NAMESPACE_END

#endif // vtkJumpAndWalkCellLocator_h