rangetype.hh

Go to the documentation of this file.

#pragma once
 
#include <array>
#include <type_traits>
 
 
namespace Dune {
 
// forward declarations
template <class T, int N>
class FieldVector;
 
template <class T, int N, int M>
class FieldMatrix;
 
} // end namespace Dune
 
namespace Dune::Vtk {
 
template <class F, class D>
using RangeType = std::decay_t<std::invoke_result_t<F,D>>;
 
namespace Impl {
 
  template <class T, int N>
  constexpr auto dimRangeImpl (Dune::FieldVector<T,N>)
    -> std::integral_constant<int, N> { return {}; }
 
  template <class T, int N, int M>
  constexpr auto dimRangeImpl (Dune::FieldMatrix<T,N,M>)
    -> std::integral_constant<int, N*M> { return {}; }
 
  constexpr auto dimRangeImpl (...)
    -> std::integral_constant<int, 1> { return {}; }
 
} // end namespace Impl
 
template <class T>
static constexpr int dimRange ()
{
  return decltype(Impl::dimRangeImpl(std::declval<T>()))::value;
}
 
namespace Impl {
 
  template <class T, int N, int M>
  constexpr std::array<int,2> rangeShapeImpl (Dune::FieldMatrix<T,N,M> const*) { return {N,M}; }
 
  constexpr std::array<int,2> rangeShapeImpl (...) { return {0,0}; }
 
} // end namespace Impl
 
template <class T>
constexpr std::array<int,2> rangeShape ()
{
  return Impl::rangeShapeImpl(static_cast<T const*>(nullptr));
}
 
} // end namespace Dune::Vtk