max_tensor_order.hpp

Go to the documentation of this file.

/* This file is part of OpenKalman, a header-only C++ library for
 * Kalman filters and other recursive filters.
 *
 * Copyright (c) 2019-2023 Christopher Lee Ogden <ogden@gatech.edu>
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */

#ifndef OPENKALMAN_MAX_TENSOR_ORDER_HPP
#define OPENKALMAN_MAX_TENSOR_ORDER_HPP


namespace OpenKalman
{
  namespace detail
  {
    template<std::size_t i, typename T>
    constexpr std::size_t max_tensor_order_impl(std::size_t result = 0)
    {
      if constexpr (i == 0) return result;
      else if constexpr (dimension_size_of_index_is<T, i - 1, 1>) return max_tensor_order_impl<i - 1, T>(result);
      else if constexpr (dimension_size_of_index_is<T, i - 1, 0>) return 0;
      else return max_tensor_order_impl<i - 1, T>(result + 1);
    }
  } // namespace detail


#ifdef __cpp_concepts
  template<typename T>
#else
  template<typename T, typename = void>
#endif
  struct max_tensor_order
    : std::integral_constant<std::size_t, indexible<T> ? dynamic_size : 0> {};


#ifdef __cpp_concepts
  template<typename T> requires (index_count_v<T> != dynamic_size)
  struct max_tensor_order<T>
#else
  template<typename T>
  struct max_tensor_order<T, std::enable_if_t<index_count<T>::value != dynamic_size>>
#endif
    : std::integral_constant<std::size_t, detail::max_tensor_order_impl<index_count_v<T>, T>()> {};


  template<typename T>
  static constexpr std::size_t max_tensor_order_v = max_tensor_order<T>::value;


} // namespace OpenKalman

#endif //OPENKALMAN_MAX_TENSOR_ORDER_HPP