odeint.hpp

Go to the documentation of this file.

// Copyright (C) 2022 Petter Nilsson. MIT License.
 
#pragma once
 
#include <tuple>
 
#include <boost/numeric/odeint/algebra/operations_dispatcher.hpp>
 
#include "smooth/manifolds.hpp"
 
SMOOTH_BEGIN_NAMESPACE
 
namespace detail {
struct BoostOdeintOps
{
  template<typename... Fac>
  struct scale_sum
  {
    const std::tuple<Fac...> m_alpha;
 
    explicit inline scale_sum(Fac... alpha) noexcept : m_alpha(alpha...) {}
 
    template<typename... Ts, std::size_t... Is>
    inline auto helper(std::index_sequence<Is...>, const Ts &... as) noexcept
    {
      // plus 1 since alpha1 = 1 is not included in Ts...
      return ((std::get<Is + 1>(m_alpha) * as) + ...);
    }
 
    template<Manifold T1, Manifold T2, typename... Ts>
      requires(std::is_same_v<T1, T2> && std::conjunction_v<std::is_same<Tangent<T1>, Ts>...>)
    inline void operator()(T1 & y, const T2 & x, const Ts &... as) noexcept
    {
      y = smooth::rplus(x, helper(std::make_index_sequence<sizeof...(Ts)>(), as...));
    }
 
    using result_type = void;
  };
 
  // \cond
  // clang-format off
  template<typename Fac1, typename Fac2 = Fac1>
  using scale_sum2 = scale_sum<Fac1, Fac2>;
 
  template<typename Fac1, typename Fac2 = Fac1, typename Fac3 = Fac2>
  using scale_sum3 = scale_sum<Fac1, Fac2, Fac3>;
 
  template<typename Fac1, typename Fac2 = Fac1, typename Fac3 = Fac2, typename Fac4 = Fac3>
  using scale_sum4 = scale_sum<Fac1, Fac2, Fac3, Fac4>;
 
  template<typename Fac1, typename Fac2 = Fac1, typename Fac3 = Fac2, typename Fac4 = Fac3, typename Fac5 = Fac4>
  using scale_sum5 = scale_sum<Fac1, Fac2, Fac3, Fac4, Fac5>;
 
  template<typename Fac1, typename Fac2 = Fac1, typename Fac3 = Fac2, typename Fac4 = Fac3, typename Fac5 = Fac4, typename Fac6 = Fac5> //NOLINT
  using scale_sum6 = scale_sum<Fac1, Fac2, Fac3, Fac4, Fac5, Fac6>;
 
  template<typename Fac1, typename Fac2 = Fac1, typename Fac3 = Fac2, typename Fac4 = Fac3, typename Fac5 = Fac4, typename Fac6 = Fac5, typename Fac7 = Fac6> //NOLINT
  using scale_sum7 = scale_sum<Fac1, Fac2, Fac3, Fac4, Fac5, Fac6, Fac7>;
 
  template<typename Fac1, typename Fac2 = Fac1, typename Fac3 = Fac2, typename Fac4 = Fac3, typename Fac5 = Fac4, typename Fac6 = Fac5, typename Fac7 = Fac6, typename Fac8 = Fac7> //NOLINT
  using scale_sum8 = scale_sum<Fac1, Fac2, Fac3, Fac4, Fac5, Fac6, Fac7, Fac8>;
 
  template<typename Fac1, typename Fac2 = Fac1, typename Fac3 = Fac2, typename Fac4 = Fac3, typename Fac5 = Fac4, typename Fac6 = Fac5, typename Fac7 = Fac6, typename Fac8 = Fac7, typename Fac9 = Fac8> //NOLINT
  using scale_sum9 = scale_sum<Fac1, Fac2, Fac3, Fac4, Fac5, Fac6, Fac7, Fac8, Fac9>;
 
  template<typename Fac1, typename Fac2 = Fac1, typename Fac3 = Fac2, typename Fac4 = Fac3, typename Fac5 = Fac4, typename Fac6 = Fac5, typename Fac7 = Fac6, typename Fac8 = Fac7, typename Fac9 = Fac8, typename Fac10 = Fac9> //NOLINT
  using scale_sum10 = scale_sum<Fac1, Fac2, Fac3, Fac4, Fac5, Fac6, Fac7, Fac8, Fac9, Fac10>;
 
  template<typename Fac1, typename Fac2 = Fac1, typename Fac3 = Fac2, typename Fac4 = Fac3, typename Fac5 = Fac4, typename Fac6 = Fac5, typename Fac7 = Fac6, typename Fac8 = Fac7, typename Fac9 = Fac8, typename Fac10 = Fac9, typename Fac11 = Fac10> //NOLINT
  using scale_sum11 = scale_sum<Fac1, Fac2, Fac3, Fac4, Fac5, Fac6, Fac7, Fac8, Fac9, Fac10, Fac11>;
 
  template<typename Fac1, typename Fac2 = Fac1, typename Fac3 = Fac2, typename Fac4 = Fac3, typename Fac5 = Fac4, typename Fac6 = Fac5, typename Fac7 = Fac6, typename Fac8 = Fac7, typename Fac9 = Fac8, typename Fac10 = Fac9, typename Fac11 = Fac10, typename Fac12 = Fac11> //NOLINT
  using scale_sum12 = scale_sum<Fac1, Fac2, Fac3, Fac4, Fac5, Fac6, Fac7, Fac8, Fac9, Fac10, Fac11, Fac12>;
 
  template<typename Fac1, typename Fac2 = Fac1, typename Fac3 = Fac2, typename Fac4 = Fac3, typename Fac5 = Fac4, typename Fac6 = Fac5, typename Fac7 = Fac6, typename Fac8 = Fac7, typename Fac9 = Fac8, typename Fac10 = Fac9, typename Fac11 = Fac10, typename Fac12 = Fac11, typename Fac13 = Fac12> //NOLINT
  using scale_sum13 = scale_sum<Fac1, Fac2, Fac3, Fac4, Fac5, Fac6, Fac7, Fac8, Fac9, Fac10, Fac11, Fac12, Fac13>;
 
  template<typename Fac1, typename Fac2 = Fac1, typename Fac3 = Fac2, typename Fac4 = Fac3, typename Fac5 = Fac4, typename Fac6 = Fac5, typename Fac7 = Fac6, typename Fac8 = Fac7, typename Fac9 = Fac8, typename Fac10 = Fac9, typename Fac11 = Fac10, typename Fac12 = Fac11, typename Fac13 = Fac12, typename Fac14 = Fac13> //NOLINT
  using scale_sum14 = scale_sum<Fac1, Fac2, Fac3, Fac4, Fac5, Fac6, Fac7, Fac8, Fac9, Fac10, Fac11, Fac12, Fac13, Fac14>; //NOLINT
  // clang-format on
  // \endcond
};
 
}  // namespace detail
 
SMOOTH_END_NAMESPACE
 
// \cond
template<smooth::Manifold G>
struct boost::numeric::odeint::operations_dispatcher_sfinae<G, void>
{
  using operations_type = ::smooth::detail::BoostOdeintOps;
};
// \endcond