vtk-m2/vtkm/exec/ConnectivityExplicit.h
Kenneth Moreland 5b47354058 Rename CellType to CellShape.
We have been using the term "shape" in the cell set and connectivity
classes. To be consistent, use the term "shape" for the geometric
identify of the cell everywhere.
2015-08-27 16:25:59 -06:00

103 lines
3.2 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 2015 Sandia Corporation.
// Copyright 2015 UT-Battelle, LLC.
// Copyright 2015 Los Alamos National Security.
//
// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
// 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_exec_ConnectivityExplicit_h
#define vtk_m_exec_ConnectivityExplicit_h
#include <vtkm/CellShape.h>
#include <vtkm/Types.h>
#include <vtkm/exec/internal/VecFromPortal.h>
namespace vtkm {
namespace exec {
template<typename ShapePortalType,
typename NumIndicesPortalType,
typename ConnectivityPortalType,
typename IndexOffsetPortalType
>
class ConnectivityExplicit
{
public:
ConnectivityExplicit() {}
ConnectivityExplicit(const ShapePortalType& shapePortal,
const NumIndicesPortalType& numIndicesPortal,
const ConnectivityPortalType& connPortal,
const IndexOffsetPortalType& indexOffsetPortal
)
: Shapes(shapePortal),
NumIndices(numIndicesPortal),
Connectivity(connPortal),
IndexOffset(indexOffsetPortal)
{
}
VTKM_EXEC_EXPORT
vtkm::Id GetNumberOfElements() const
{
return this->Shapes.GetNumberOfValues();
}
VTKM_EXEC_EXPORT
vtkm::IdComponent GetNumberOfIndices(vtkm::Id index) const
{
// Should the NumIndices array be typed as vtkm::IdComponent instead of
// vtkm::Id? (NumIndices is really defined in
// vtkm::cont::internal::ConnectivityExplicitInternals.)
return static_cast<vtkm::IdComponent>(this->NumIndices.Get(index));
}
VTKM_EXEC_EXPORT
vtkm::CellShapeId GetCellShape(vtkm::Id index) const
{
// Likewise, should Shapes be vtkm::Id or something smaller?
return static_cast<vtkm::CellShapeId>(this->Shapes.Get(index));
}
typedef vtkm::exec::internal::VecFromPortal<ConnectivityPortalType>
IndicesType;
/// Returns a Vec-like object containing the indices for the given index.
/// The object returned is not an actual array, but rather an object that
/// loads the indices lazily out of the connectivity array. This prevents
/// us from having to know the number of indices at compile time.
///
VTKM_EXEC_EXPORT
IndicesType GetIndices(vtkm::Id index) const
{
vtkm::Id offset = this->IndexOffset.Get(index);
vtkm::IdComponent length = this->GetNumberOfIndices(index);
return IndicesType(this->Connectivity, length, offset);
}
private:
ShapePortalType Shapes;
NumIndicesPortalType NumIndices;
ConnectivityPortalType Connectivity;
IndexOffsetPortalType IndexOffset;
};
} // namespace exec
} // namespace vtkm
#endif // vtk_m_exec_ConnectivityExplicit_h