vtk-m/vtkm/BaseComponent.h

//============================================================================
//  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_BaseComponent_h
#define vtk_m_BaseComponent_h

#include <vtkm/Matrix.h>
#include <vtkm/TypeTraits.h>
#include <vtkm/VecTraits.h>

namespace vtkm
{

namespace detail
{

template <typename VecType, typename DimensionalityTag>
struct BaseComponentImpl;

template <typename VecType>
struct BaseComponentImpl<VecType, vtkm::TypeTraitsVectorTag>
{
private:
  using ComponentType = typename vtkm::VecTraits<VecType>::ComponentType;

public:
  using Type =
    typename BaseComponentImpl<ComponentType,
                               typename vtkm::TypeTraits<ComponentType>::DimensionalityTag>::Type;
};

template <typename VecType>
struct BaseComponentImpl<VecType, vtkm::TypeTraitsMatrixTag>
  : BaseComponentImpl<VecType, vtkm::TypeTraitsVectorTag>
{
};

template <typename ScalarType>
struct BaseComponentImpl<ScalarType, vtkm::TypeTraitsScalarTag>
{
  using Type = ScalarType;
};

} // namespace detail

// Finds the base component type of a Vec. If you have a Vec of Vecs, it will
// descend all Vecs until you get to the scalar type.
template <typename VecType>
struct BaseComponent
{
  using Type =
    typename detail::BaseComponentImpl<VecType,
                                       typename vtkm::TypeTraits<VecType>::DimensionalityTag>::Type;
};

} // namespace vtkm

#endif //vtk_m_BaseComponent_h