mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-20 02:55:47 +00:00
5db762ee71
The `From` and `To` nomenclature for topology mapping has been confusing for both users and developers, especially at lower levels where the intention of mapping attributes from one element to another is easily conflated with the concept of mapping indices (which maps in the exact opposite direction). These identifiers have been renamed to `VisitTopology` and `IncidentTopology` to clarify the direction of the mapping. The order in which these template parameters are specified for `WorkletMapTopology` have also been reversed, since eventually there may be more than one `IncidentTopology`, and having `IncidentTopology` at the end will allow us to replace it with a variadic template parameter pack in the future. Other implementation details supporting these worklets, include `Fetch` tags, `Connectivity` classes, and methods on the various `CellSet` classes (such as `PrepareForInput` have also reversed their template arguments. These will need to be cautiously updated. The convenience implementations of `WorkletMapTopology` have been renamed for clarity as follows: ``` WorkletMapPointToCell --> WorkletVisitCellsWithPoints WorkletMapCellToPoint --> WorkletVisitPointsWithCells ``` The `ControlSignature` tags have been renamed as follows: ``` FieldInTo --> FieldInVisit FieldInFrom --> FieldInMap FromCount --> IncidentElementCount FromIndices --> IncidentElementIndices ```
131 lines
4.6 KiB
C++
131 lines
4.6 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.
|
|
//
|
|
// Copyright 2018 National Technology & Engineering Solutions of Sandia, LLC (NTESS).
|
|
// Copyright 2018 UT-Battelle, LLC.
|
|
// Copyright 2018 Los Alamos National Security.
|
|
//
|
|
// Under the terms of Contract DE-NA0003525 with NTESS,
|
|
// the U.S. Government retains certain rights in this software.
|
|
//
|
|
// Under the terms of Contract DE-AC52-06NA25396 with Los Alamos National
|
|
// Laboratory (LANL), the U.S. Government retains certain rights in
|
|
// this software.
|
|
//============================================================================
|
|
|
|
#ifndef vtk_m_worklet_MeshQuality_h
|
|
#define vtk_m_worklet_MeshQuality_h
|
|
|
|
#include "vtkm/exec/CellMeasure.h"
|
|
#include "vtkm/exec/cellmetrics/CellDiagonalRatioMetric.h"
|
|
#include "vtkm/exec/cellmetrics/CellEdgeRatioMetric.h"
|
|
#include "vtkm/worklet/WorkletMapTopology.h"
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace worklet
|
|
{
|
|
|
|
/**
|
|
* Worklet that computes mesh quality metric values for each cell in
|
|
* the input mesh. A metric is specified per cell type in the calling filter,
|
|
* and this metric is invoked over all cells of that cell type. An array of
|
|
* the computed metric values (one per cell) is returned as output.
|
|
*/
|
|
template <typename MetricTagType>
|
|
class MeshQuality : public vtkm::worklet::WorkletVisitCellsWithPoints
|
|
{
|
|
public:
|
|
using ControlSignature = void(CellSetIn cellset,
|
|
WholeArrayIn counts,
|
|
WholeArrayIn metrics,
|
|
FieldInPoint pointCoords,
|
|
FieldOutCell metricOut);
|
|
using ExecutionSignature = void(CellShape, PointCount, _2, _3, _4, _5);
|
|
using InputDomain = _1;
|
|
|
|
template <typename CellShapeType,
|
|
typename PointCoordVecType,
|
|
typename CountsArrayType,
|
|
typename MetricsArrayType,
|
|
typename OutType>
|
|
VTKM_EXEC void operator()(CellShapeType shape,
|
|
const vtkm::IdComponent& numPoints,
|
|
const CountsArrayType& counts,
|
|
const MetricsArrayType& metrics,
|
|
const PointCoordVecType& pts,
|
|
OutType& metricValue) const
|
|
{
|
|
printf("shape.Id: %u\n", shape.Id);
|
|
vtkm::UInt8 thisId = shape.Id;
|
|
if (shape.Id == vtkm::CELL_SHAPE_POLYGON)
|
|
{
|
|
if (numPoints == 3)
|
|
thisId = vtkm::CELL_SHAPE_TRIANGLE;
|
|
else if (numPoints == 4)
|
|
thisId = vtkm::CELL_SHAPE_QUAD;
|
|
}
|
|
switch (thisId)
|
|
{
|
|
vtkmGenericCellShapeMacro(
|
|
metricValue = this->ComputeMetric<OutType>(
|
|
numPoints, pts, counts.Get(shape.Id), CellShapeTag(), metrics.Get(CellShapeTag().Id)));
|
|
default:
|
|
this->RaiseError("Asked for metric of unknown cell type.");
|
|
metricValue = OutType(0.0);
|
|
}
|
|
}
|
|
|
|
protected:
|
|
template <typename OutType,
|
|
typename PointCoordVecType,
|
|
typename CellShapeType,
|
|
typename CellMetricType>
|
|
VTKM_EXEC OutType ComputeMetric(const vtkm::IdComponent& numPts,
|
|
const PointCoordVecType& pts,
|
|
const vtkm::Id& numShapes,
|
|
CellShapeType tag,
|
|
CellMetricType metric) const
|
|
{
|
|
UNUSED(numShapes);
|
|
constexpr vtkm::IdComponent dims = vtkm::CellTraits<CellShapeType>::TOPOLOGICAL_DIMENSIONS;
|
|
|
|
//Only compute the metric for 2D and 3D shapes; return 0 otherwise
|
|
OutType metricValue = OutType(0.0);
|
|
if (dims > 0)
|
|
{
|
|
switch (metric)
|
|
{
|
|
case MetricTagType::DIAGONAL_RATIO:
|
|
metricValue =
|
|
vtkm::exec::cellmetrics::CellDiagonalRatioMetric<OutType>(numPts, pts, tag, *this);
|
|
break;
|
|
case MetricTagType::EDGE_RATIO:
|
|
metricValue =
|
|
vtkm::exec::cellmetrics::CellEdgeRatioMetric<OutType>(numPts, pts, tag, *this);
|
|
break;
|
|
case MetricTagType::VOLUME:
|
|
metricValue = vtkm::exec::CellMeasure<OutType>(numPts, pts, tag, *this);
|
|
break;
|
|
case MetricTagType::EMPTY:
|
|
break;
|
|
default:
|
|
//Only call metric function if a metric is specified for this shape type
|
|
this->RaiseError("Asked for unknown metric.");
|
|
}
|
|
}
|
|
|
|
return metricValue;
|
|
}
|
|
};
|
|
|
|
} // namespace worklet
|
|
} // namespace vtkm
|
|
|
|
#endif // vtk_m_worklet_MeshQuality_h
|