constant_diagonal_coefficient.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-2024 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_CONSTANT_DIAGONAL_COEFFICIENT_HPP
#define OPENKALMAN_CONSTANT_DIAGONAL_COEFFICIENT_HPP

#include <type_traits>

namespace OpenKalman::values
{
#ifdef __cpp_concepts
  template<indexible T>
#else
  template<typename T, typename = void>
#endif
  struct constant_diagonal_coefficient
  {
    explicit constexpr constant_diagonal_coefficient(const std::decay_t<T>&) {};
  };


  template<typename T>
  explicit constant_diagonal_coefficient(T&&) -> constant_diagonal_coefficient<std::decay_t<T>>;


#ifdef __cpp_concepts
  template<indexible T>
#else
  template<typename T>
#endif
  constexpr auto constant_diagonal_coefficient_v = constant_diagonal_coefficient<T>::value;


  namespace detail
  {
    template<typename T>
    constexpr auto const_value =
      std::decay_t<decltype(interface::indexible_object_traits<std::decay_t<T>>::get_constant(std::declval<T>()))>::value;


#ifdef __cpp_concepts
    template<typename T>
    concept has_static_constant_diagonal =
      values::fixed<typename interface::get_constant_diagonal_return_type<T>::type> or
      (values::fixed<typename interface::get_constant_return_type<T>::type> and
        (one_dimensional<T> or values::internal::near(const_value<T>, 0)));
#else
    template<typename T, typename = void>
    struct has_static_constant_diagonal_impl : std::false_type {};

    template<typename T>
    struct has_static_constant_diagonal_impl<T, std::enable_if_t<values::fixed<typename interface::get_constant_return_type<T>::type>>>
      : std::bool_constant<one_dimensional<T> or values::internal::near(const_value<T>, 0)> {};


    template<typename T>
    constexpr bool has_static_constant_diagonal =
      values::fixed<typename interface::get_constant_diagonal_return_type<T>::type> or
      has_static_constant_diagonal_impl<T>::value;
#endif
  } // namespace detail


#ifdef __cpp_concepts
  template<indexible T> requires detail::has_static_constant_diagonal<T>
  struct constant_diagonal_coefficient<T>
#else
  template<typename T>
  struct constant_diagonal_coefficient<T, std::enable_if_t<indexible<T> and detail::has_static_constant_diagonal<T>>>
#endif
  {
  private:

    using Trait = interface::indexible_object_traits<std::decay_t<T>>;

  public:

    constexpr constant_diagonal_coefficient() = default;

    explicit constexpr constant_diagonal_coefficient(const std::decay_t<T>&) {};

    using value_type = scalar_type_of_t<T>;

    using type = constant_diagonal_coefficient;

    static constexpr value_type value = []{
      if constexpr (values::scalar<typename interface::get_constant_diagonal_return_type<T>::type>)
        return std::decay_t<decltype(Trait::get_constant_diagonal(std::declval<T>()))>::value;
      else
        return std::decay_t<decltype(Trait::get_constant(std::declval<T>()))>::value;
    }();

    constexpr operator value_type() const { return value; }

    constexpr value_type operator()() const { return value; }
  };


#ifdef __cpp_concepts
  template<indexible T> requires (not detail::has_static_constant_diagonal<T>) and
    (values::dynamic<typename interface::get_constant_diagonal_return_type<T>::type> or one_dimensional<T>)
  struct constant_diagonal_coefficient<T>
#else
  template<typename T>
  struct constant_diagonal_coefficient<T, std::enable_if_t<indexible<T> and (not detail::has_static_constant_diagonal<T>) and
    (values::dynamic<typename interface::get_constant_diagonal_return_type<T>::type> or one_dimensional<T>)>>
#endif
  {
  private:

    using Trait = interface::indexible_object_traits<std::decay_t<T>>;

  public:

    explicit constexpr constant_diagonal_coefficient(const std::decay_t<T>& t) : m_value {[](const auto& t){
        if constexpr (values::scalar<typename interface::get_constant_diagonal_return_type<T>::type>)
          return values::to_number(Trait::get_constant_diagonal(t));
        else if constexpr (values::scalar<typename interface::get_constant_return_type<T>::type>)
          return values::to_number(Trait::get_constant(t));
        else
          return internal::get_singular_component(t);
      }(t)} {};

    using value_type = scalar_type_of_t<T>;

    using type = constant_diagonal_coefficient;

    constexpr operator value_type() const { return m_value; }

    constexpr value_type operator()() const { return m_value; }

  private:

    value_type m_value;
  };


} // namespace OpenKalman::values


namespace OpenKalman
{
  using values::constant_diagonal_coefficient;
  using values::constant_diagonal_coefficient_v;
}


#endif //OPENKALMAN_CONSTANT_DIAGONAL_COEFFICIENT_HPP