mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-08 13:23:51 +00:00
With the major revision 2.0 of VTK-m, many items previously marked as deprecated were removed. If updating to a new version of VTK-m, it is recommended to first update to VTK-m 1.9, which will include the deprecated features but provide warnings (with the right compiler) that will point to the replacement code. Once the deprecations have been fixed, updating to 2.0 should be smoother.
208 lines
8.3 KiB
C++
208 lines
8.3 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_Derivative_h
|
|
#define vtk_m_exec_Derivative_h
|
|
|
|
#include <vtkm/CellShape.h>
|
|
#include <vtkm/ErrorCode.h>
|
|
#include <vtkm/VecAxisAlignedPointCoordinates.h>
|
|
#include <vtkm/VecTraits.h>
|
|
|
|
#include <vtkm/exec/CellInterpolate.h>
|
|
#include <vtkm/exec/FunctorBase.h>
|
|
|
|
#include <lcl/lcl.h>
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace exec
|
|
{
|
|
|
|
//-----------------------------------------------------------------------------
|
|
namespace internal
|
|
{
|
|
|
|
template <typename LclCellShapeTag,
|
|
typename FieldVecType,
|
|
typename WorldCoordType,
|
|
typename ParametricCoordType>
|
|
VTKM_EXEC vtkm::ErrorCode CellDerivativeImpl(
|
|
LclCellShapeTag tag,
|
|
const FieldVecType& field,
|
|
const WorldCoordType& wCoords,
|
|
const ParametricCoordType& pcoords,
|
|
vtkm::Vec<typename FieldVecType::ComponentType, 3>& result)
|
|
{
|
|
result = { 0 };
|
|
if ((field.GetNumberOfComponents() != tag.numberOfPoints()) ||
|
|
(wCoords.GetNumberOfComponents() != tag.numberOfPoints()))
|
|
{
|
|
return vtkm::ErrorCode::InvalidNumberOfPoints;
|
|
}
|
|
|
|
using FieldType = typename FieldVecType::ComponentType;
|
|
|
|
auto fieldNumComponents = vtkm::VecTraits<FieldType>::GetNumberOfComponents(field[0]);
|
|
auto status = lcl::derivative(tag,
|
|
lcl::makeFieldAccessorNestedSOA(wCoords, 3),
|
|
lcl::makeFieldAccessorNestedSOA(field, fieldNumComponents),
|
|
pcoords,
|
|
result[0],
|
|
result[1],
|
|
result[2]);
|
|
return vtkm::internal::LclErrorToVtkmError(status);
|
|
}
|
|
|
|
} // namespace internal
|
|
|
|
template <typename FieldVecType,
|
|
typename WorldCoordType,
|
|
typename ParametricCoordType,
|
|
typename CellShapeTag>
|
|
VTKM_EXEC vtkm::ErrorCode CellDerivative(const FieldVecType& field,
|
|
const WorldCoordType& wCoords,
|
|
const vtkm::Vec<ParametricCoordType, 3>& pcoords,
|
|
CellShapeTag shape,
|
|
vtkm::Vec<typename FieldVecType::ComponentType, 3>& result)
|
|
{
|
|
return internal::CellDerivativeImpl(
|
|
vtkm::internal::make_LclCellShapeTag(shape), field, wCoords, pcoords, result);
|
|
}
|
|
|
|
template <typename FieldVecType, typename WorldCoordType, typename ParametricCoordType>
|
|
VTKM_EXEC vtkm::ErrorCode CellDerivative(const FieldVecType&,
|
|
const WorldCoordType&,
|
|
const vtkm::Vec<ParametricCoordType, 3>&,
|
|
vtkm::CellShapeTagEmpty,
|
|
vtkm::Vec<typename FieldVecType::ComponentType, 3>& result)
|
|
{
|
|
result = { 0 };
|
|
return vtkm::ErrorCode::OperationOnEmptyCell;
|
|
}
|
|
|
|
template <typename FieldVecType, typename WorldCoordType, typename ParametricCoordType>
|
|
VTKM_EXEC vtkm::ErrorCode CellDerivative(const FieldVecType& field,
|
|
const WorldCoordType& wCoords,
|
|
const vtkm::Vec<ParametricCoordType, 3>& pcoords,
|
|
vtkm::CellShapeTagPolyLine,
|
|
vtkm::Vec<typename FieldVecType::ComponentType, 3>& result)
|
|
{
|
|
vtkm::IdComponent numPoints = field.GetNumberOfComponents();
|
|
if (numPoints != wCoords.GetNumberOfComponents())
|
|
{
|
|
result = { 0 };
|
|
return vtkm::ErrorCode::InvalidNumberOfPoints;
|
|
}
|
|
|
|
switch (numPoints)
|
|
{
|
|
case 1:
|
|
return CellDerivative(field, wCoords, pcoords, vtkm::CellShapeTagVertex(), result);
|
|
case 2:
|
|
return CellDerivative(field, wCoords, pcoords, vtkm::CellShapeTagLine(), result);
|
|
}
|
|
|
|
auto dt = static_cast<ParametricCoordType>(1) / static_cast<ParametricCoordType>(numPoints - 1);
|
|
auto idx = static_cast<vtkm::IdComponent>(vtkm::Ceil(pcoords[0] / dt));
|
|
if (idx == 0)
|
|
{
|
|
idx = 1;
|
|
}
|
|
if (idx > numPoints - 1)
|
|
{
|
|
idx = numPoints - 1;
|
|
}
|
|
|
|
auto lineField = vtkm::make_Vec(field[idx - 1], field[idx]);
|
|
auto lineWCoords = vtkm::make_Vec(wCoords[idx - 1], wCoords[idx]);
|
|
auto pc = (pcoords[0] - static_cast<ParametricCoordType>(idx) * dt) / dt;
|
|
return internal::CellDerivativeImpl(lcl::Line{}, lineField, lineWCoords, &pc, result);
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
template <typename FieldVecType, typename WorldCoordType, typename ParametricCoordType>
|
|
VTKM_EXEC vtkm::ErrorCode CellDerivative(const FieldVecType& field,
|
|
const WorldCoordType& wCoords,
|
|
const vtkm::Vec<ParametricCoordType, 3>& pcoords,
|
|
vtkm::CellShapeTagPolygon,
|
|
vtkm::Vec<typename FieldVecType::ComponentType, 3>& result)
|
|
{
|
|
const vtkm::IdComponent numPoints = field.GetNumberOfComponents();
|
|
if ((numPoints <= 0) || (numPoints != wCoords.GetNumberOfComponents()))
|
|
{
|
|
result = { 0 };
|
|
return vtkm::ErrorCode::InvalidNumberOfPoints;
|
|
}
|
|
|
|
switch (field.GetNumberOfComponents())
|
|
{
|
|
case 1:
|
|
return CellDerivative(field, wCoords, pcoords, vtkm::CellShapeTagVertex(), result);
|
|
case 2:
|
|
return CellDerivative(field, wCoords, pcoords, vtkm::CellShapeTagLine(), result);
|
|
default:
|
|
return internal::CellDerivativeImpl(lcl::Polygon(numPoints), field, wCoords, pcoords, result);
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
template <typename FieldVecType, typename ParametricCoordType>
|
|
VTKM_EXEC vtkm::ErrorCode CellDerivative(const FieldVecType& field,
|
|
const vtkm::VecAxisAlignedPointCoordinates<2>& wCoords,
|
|
const vtkm::Vec<ParametricCoordType, 3>& pcoords,
|
|
vtkm::CellShapeTagQuad,
|
|
vtkm::Vec<typename FieldVecType::ComponentType, 3>& result)
|
|
{
|
|
return internal::CellDerivativeImpl(lcl::Pixel{}, field, wCoords, pcoords, result);
|
|
}
|
|
|
|
template <typename FieldVecType, typename ParametricCoordType>
|
|
VTKM_EXEC vtkm::ErrorCode CellDerivative(const FieldVecType& field,
|
|
const vtkm::VecAxisAlignedPointCoordinates<3>& wCoords,
|
|
const vtkm::Vec<ParametricCoordType, 3>& pcoords,
|
|
vtkm::CellShapeTagHexahedron,
|
|
vtkm::Vec<typename FieldVecType::ComponentType, 3>& result)
|
|
{
|
|
return internal::CellDerivativeImpl(lcl::Voxel{}, field, wCoords, pcoords, result);
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
/// \brief Take the derivative (get the gradient) of a point field in a cell.
|
|
///
|
|
/// Given the point field values for each node and the parametric coordinates
|
|
/// of a point within the cell, finds the derivative with respect to each
|
|
/// coordinate (i.e. the gradient) at that point. The derivative is not always
|
|
/// constant in some "linear" cells.
|
|
///
|
|
template <typename FieldVecType, typename WorldCoordType, typename ParametricCoordType>
|
|
VTKM_EXEC vtkm::ErrorCode CellDerivative(const FieldVecType& pointFieldValues,
|
|
const WorldCoordType& worldCoordinateValues,
|
|
const vtkm::Vec<ParametricCoordType, 3>& parametricCoords,
|
|
vtkm::CellShapeTagGeneric shape,
|
|
vtkm::Vec<typename FieldVecType::ComponentType, 3>& result)
|
|
{
|
|
vtkm::ErrorCode status;
|
|
switch (shape.Id)
|
|
{
|
|
vtkmGenericCellShapeMacro(
|
|
status = CellDerivative(
|
|
pointFieldValues, worldCoordinateValues, parametricCoords, CellShapeTag(), result));
|
|
default:
|
|
result = { 0 };
|
|
status = vtkm::ErrorCode::InvalidShapeId;
|
|
}
|
|
return status;
|
|
}
|
|
|
|
}
|
|
} // namespace vtkm::exec
|
|
|
|
#endif //vtk_m_exec_Derivative_h
|