mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-19 18:45:43 +00:00
83 lines
2.4 KiB
C
83 lines
2.4 KiB
C
|
//============================================================================
|
||
|
// Copyright (c) Kitware, Inc.
|
||
|
// All rights reserved.
|
||
|
// See LICENSE.txt 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 notice for more information.
|
||
|
//============================================================================
|
||
|
#ifndef vtk_m_exec_CellLocatorPartitioned_h
|
||
|
#define vtk_m_exec_CellLocatorPartitioned_h
|
||
|
|
||
|
#include <vtkm/Types.h>
|
||
|
#include <vtkm/cont/ArrayHandle.h>
|
||
|
#include <vtkm/cont/CellLocatorGeneral.h>
|
||
|
|
||
|
namespace vtkm
|
||
|
{
|
||
|
namespace exec
|
||
|
{
|
||
|
class VTKM_ALWAYS_EXPORT CellLocatorPartitioned
|
||
|
{
|
||
|
private:
|
||
|
vtkm::cont::ArrayHandle<vtkm::cont::CellLocatorGeneral::ExecObjType>::ReadPortalType Locators;
|
||
|
vtkm::cont::ArrayHandle<vtkm::cont::ArrayHandle<vtkm::UInt8>::ReadPortalType>::ReadPortalType
|
||
|
Ghosts;
|
||
|
|
||
|
public:
|
||
|
VTKM_CONT CellLocatorPartitioned() = default;
|
||
|
VTKM_CONT CellLocatorPartitioned(
|
||
|
const vtkm::cont::ArrayHandle<vtkm::cont::CellLocatorGeneral::ExecObjType>::ReadPortalType&
|
||
|
locators,
|
||
|
vtkm::cont::ArrayHandle<vtkm::cont::ArrayHandle<vtkm::UInt8>::ReadPortalType>::ReadPortalType
|
||
|
ghosts)
|
||
|
: Locators(locators)
|
||
|
, Ghosts(ghosts)
|
||
|
{
|
||
|
}
|
||
|
|
||
|
VTKM_EXEC
|
||
|
vtkm::ErrorCode FindCell(const vtkm::Vec3f& point,
|
||
|
vtkm::Id& partitionId,
|
||
|
vtkm::Id& cellId,
|
||
|
vtkm::Vec3f& parametric) const
|
||
|
{
|
||
|
bool found = 0;
|
||
|
for (vtkm::Id partitionIndex = 0; partitionIndex < this->Locators.GetNumberOfValues();
|
||
|
++partitionIndex)
|
||
|
{
|
||
|
vtkm::Id cellIndex;
|
||
|
vtkm ::Vec3f parametricLocal;
|
||
|
vtkm ::ErrorCode status =
|
||
|
Locators.Get(partitionIndex).FindCell(point, cellIndex, parametricLocal);
|
||
|
if (status != vtkm ::ErrorCode ::Success)
|
||
|
{
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
if (Ghosts.Get(partitionIndex).Get(cellIndex) == 0)
|
||
|
{
|
||
|
partitionId = partitionIndex;
|
||
|
cellId = cellIndex;
|
||
|
parametric = parametricLocal;
|
||
|
found = true;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
if (found)
|
||
|
{
|
||
|
return vtkm::ErrorCode::Success;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
return vtkm::ErrorCode::CellNotFound;
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
} //namespace exec
|
||
|
} //namespace vtkm
|
||
|
|
||
|
#endif //vtk_m_exec_CellLocatorPartitioned_h
|