mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-19 18:45:43 +00:00
77568789ea
These changes are basically to support some upcoming changes to ArrayHandleZip. The major additions are an implementation of VecTraits for Pair and an overloaded << operator to ostream for Pair. I also had to declare the operator<< for Pair to be in Types.h. Under some circumstances the operator has to either be declared before the template is declared or declared in the vtkm namespace. (The reasons are described at <http://clang.llvm.org/compatibility.html#dep_lookup> but I still don't understand.) I tried adding it to the vtkm namespace, but that caused several of the other operator<< to fail. Since there is no way to guarantee that Pair.h is declared before, say, ArrayHandle.h, I moved the implementation to Types.h. Since I was moving operator<< to Types.h, I went ahead and moved the TypeTraits and VecTraits to their respective headers. Since Pair is declared (but not implemented) in Types.h, these templated classes can be implemented without including Pair.h.
147 lines
4.6 KiB
C++
147 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 2014 Sandia Corporation.
|
|
// Copyright 2014 UT-Battelle, LLC.
|
|
// Copyright 2014 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_TypeTraits_h
|
|
#define vtk_m_TypeTraits_h
|
|
|
|
#include <vtkm/Types.h>
|
|
|
|
namespace vtkm {
|
|
|
|
/// Tag used to identify types that aren't Real, Integer, Scalar or Vector.
|
|
///
|
|
struct TypeTraitsUnknownTag {};
|
|
|
|
/// Tag used to identify types that store real (floating-point) numbers. A
|
|
/// TypeTraits class will typedef this class to NumericTag if it stores real
|
|
/// numbers (or vectors of real numbers).
|
|
///
|
|
struct TypeTraitsRealTag {};
|
|
|
|
/// Tag used to identify types that store integer numbers. A TypeTraits class
|
|
/// will typedef this class to NumericTag if it stores integer numbers (or
|
|
/// vectors of integers).
|
|
///
|
|
struct TypeTraitsIntegerTag {};
|
|
|
|
/// Tag used to identify 0 dimensional types (scalars). Scalars can also be
|
|
/// treated like vectors when used with VecTraits. A TypeTraits class will
|
|
/// typedef this class to DimensionalityTag.
|
|
///
|
|
struct TypeTraitsScalarTag {};
|
|
|
|
/// Tag used to identify 1 dimensional types (vectors). A TypeTraits class will
|
|
/// typedef this class to DimensionalityTag.
|
|
///
|
|
struct TypeTraitsVectorTag {};
|
|
|
|
|
|
/// The TypeTraits class provides helpful compile-time information about the
|
|
/// basic types used in VTKm (and a few others for convienience). The majority
|
|
/// of TypeTraits contents are typedefs to tags that can be used to easily
|
|
/// override behavior of called functions.
|
|
///
|
|
template<typename T>
|
|
class TypeTraits
|
|
{
|
|
public:
|
|
/// \brief A tag to determing whether the type is integer or real.
|
|
///
|
|
/// This tag is either TypeTraitsRealTag or TypeTraitsIntegerTag.
|
|
typedef TypeTraitsUnknownTag NumericTag;
|
|
|
|
/// \brief A tag to determine whether the type has multiple components.
|
|
///
|
|
/// This tag is either TypeTraitsScalarTag or TypeTraitsVectorTag. Scalars can
|
|
/// also be treated as vectors.
|
|
typedef TypeTraitsUnknownTag DimensionalityTag;
|
|
|
|
VTKM_EXEC_CONT_EXPORT static T ZeroInitialization() { return T(); }
|
|
};
|
|
|
|
// Const types should have the same traits as their non-const counterparts.
|
|
//
|
|
template<typename T>
|
|
struct TypeTraits<const T> : TypeTraits<T>
|
|
{ };
|
|
|
|
#define VTKM_BASIC_REAL_TYPE(T) \
|
|
template<> struct TypeTraits<T> { \
|
|
typedef TypeTraitsRealTag NumericTag; \
|
|
typedef TypeTraitsScalarTag DimensionalityTag; \
|
|
VTKM_EXEC_CONT_EXPORT static T ZeroInitialization() { return T(); } \
|
|
}
|
|
|
|
#define VTKM_BASIC_INTEGER_TYPE(T) \
|
|
template<> struct TypeTraits<T> { \
|
|
typedef TypeTraitsIntegerTag NumericTag; \
|
|
typedef TypeTraitsScalarTag DimensionalityTag; \
|
|
VTKM_EXEC_CONT_EXPORT static T ZeroInitialization() { return T(); } \
|
|
}
|
|
|
|
/// Traits for basic C++ types.
|
|
///
|
|
|
|
VTKM_BASIC_REAL_TYPE(vtkm::Float32);
|
|
VTKM_BASIC_REAL_TYPE(vtkm::Float64);
|
|
VTKM_BASIC_INTEGER_TYPE(vtkm::Int8);
|
|
VTKM_BASIC_INTEGER_TYPE(vtkm::UInt8);
|
|
VTKM_BASIC_INTEGER_TYPE(vtkm::Int16);
|
|
VTKM_BASIC_INTEGER_TYPE(vtkm::UInt16);
|
|
VTKM_BASIC_INTEGER_TYPE(vtkm::Int32);
|
|
VTKM_BASIC_INTEGER_TYPE(vtkm::UInt32);
|
|
VTKM_BASIC_INTEGER_TYPE(vtkm::Int64);
|
|
VTKM_BASIC_INTEGER_TYPE(vtkm::UInt64);
|
|
|
|
#undef VTKM_BASIC_REAL_TYPE
|
|
#undef VTKM_BASIC_INTEGER_TYPE
|
|
|
|
/// Traits for Vec types.
|
|
///
|
|
template<typename T, vtkm::IdComponent Size>
|
|
struct TypeTraits<vtkm::Vec<T,Size> >
|
|
{
|
|
typedef typename vtkm::TypeTraits<T>::NumericTag NumericTag;
|
|
typedef TypeTraitsVectorTag DimensionalityTag;
|
|
|
|
VTKM_EXEC_CONT_EXPORT
|
|
static vtkm::Vec<T,Size> ZeroInitialization()
|
|
{ return vtkm::Vec<T,Size>( (T()) ); }
|
|
};
|
|
|
|
/// \brief Traits for Pair types.
|
|
///
|
|
template<typename T, typename U>
|
|
struct TypeTraits<vtkm::Pair<T,U> >
|
|
{
|
|
typedef TypeTraitsUnknownTag NumericTag;
|
|
typedef TypeTraitsScalarTag DimensionalityTag;
|
|
|
|
VTKM_EXEC_CONT_EXPORT
|
|
static vtkm::Pair<T,U> ZeroInitialization()
|
|
{
|
|
return vtkm::Pair<T,U>(TypeTraits<T>::ZeroInitialization(),
|
|
TypeTraits<U>::ZeroInitialization());
|
|
}
|
|
};
|
|
|
|
} // namespace vtkm
|
|
|
|
#endif //vtk_m_TypeTraits_h
|