mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-10-08 11:29:02 +00:00
222 lines
9.3 KiB
C++
222 lines
9.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.
|
|
///
|
|
/// @param[in] pointFieldValues A list of field values for each point in the cell. This
|
|
/// usually comes from a `FieldInPoint` argument in a
|
|
/// `vtkm::worklet::WorkletVisitCellsWithPoints`.
|
|
/// @param[in] worldCoordinateValues A list of world coordinates for each point in the cell. This
|
|
/// usually comes from a `FieldInPoint` argument in a
|
|
/// `vtkm::worklet::WorkletVisitCellsWithPoints` where the coordinate system is passed
|
|
/// into that argument.
|
|
/// @param[in] parametricCoords The parametric coordinates where you want to find the derivative.
|
|
/// @param[in] shape A tag of type `CellShapeTag*` to identify the shape of the cell.
|
|
/// This method is overloaded for different shape types.
|
|
/// @param[out] result Value to store the derivative/gradient. Because the derivative is taken
|
|
/// partially in the x, y, and z directions, the result is a `vtkm::Vec` of size 3 with the
|
|
/// component type the same as the field. If the field is itself a vector, you get a `Vec`
|
|
/// of `Vec`s.
|
|
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
|