TypeTraits.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_MDP_TYPE_TRAITS_HEADER_FILE
#define AI_TOOLBOX_MDP_TYPE_TRAITS_HEADER_FILE
 
#include <AIToolbox/TypeTraits.hpp>
#include <AIToolbox/MDP/Types.hpp>
 
namespace AIToolbox::MDP {
    template <typename M>
    concept IsGenerativeModel = AIToolbox::IsGenerativeModel<M> &&
                                HasIntegralStateSpace<M> &&
                                HasIntegralActionSpace<M> &&
                                HasFixedActionSpace<M>;
 
    template <typename M>
    concept IsModel = IsGenerativeModel<M> && requires (const M m, size_t s, size_t a) {
        { m.getTransitionProbability(s, a, s) } -> std::convertible_to<double>;
        { m.getExpectedReward(s, a, s) }        -> std::convertible_to<double>;
    };
 
    template <typename M>
    concept IsModelEigen = IsModel<M> && requires (const M m, size_t a) {
        m.getTransitionFunction(a);
        requires IsDerivedFromEigen<std::remove_cvref_t<decltype((m.getTransitionFunction(a)))>>;
 
        m.getRewardFunction();
        requires IsDerivedFromEigen<std::remove_cvref_t<decltype((m.getRewardFunction()))>>;
    };
 
    template <typename E>
    concept IsExperience = requires (const E e, size_t s, size_t a) {
        { e.getVisits(s, a, s) } -> std::convertible_to<long unsigned>;
        { e.getVisitsSum(s, a) } -> std::convertible_to<long unsigned>;
        { e.getReward(s, a) }    -> std::convertible_to<double>;
        { e.getM2(s, a) }        -> std::convertible_to<double>;
    };
 
    template <typename E>
    concept IsExperienceEigen = IsExperience<E> && requires (const E e, size_t a) {
        e.getVisitsSumTable();
        requires IsDerivedFromEigen<std::remove_cvref_t<decltype((e.getVisitsSumTable()))>>;
 
        e.getVisitsTable(a);
        requires IsDerivedFromEigen<std::remove_cvref_t<decltype((e.getVisitsTable(a)))>>;
 
        e.getRewardMatrix();
        requires IsDerivedFromEigen<std::remove_cvref_t<decltype((e.getRewardMatrix()))>>;
 
        e.getM2Matrix();
        requires IsDerivedFromEigen<std::remove_cvref_t<decltype((e.getM2Matrix()))>>;
    };
}
 
#endif