PolicyInterface.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_POLICYINTERFACE_HEADER_FILE
#define AI_TOOLBOX_POLICYINTERFACE_HEADER_FILE
 
#include <AIToolbox/Types.hpp>
#include <AIToolbox/Seeder.hpp>
 
namespace AIToolbox {
    template <typename State, typename Sampling, typename Action>
    class PolicyInterface {
        public:
            PolicyInterface(State s, Action a);
 
            virtual ~PolicyInterface();
 
            virtual Action sampleAction(const Sampling & s) const = 0;
 
            virtual double getActionProbability(const Sampling & s, const Action & a) const = 0;
 
            const State & getS() const;
 
            const Action & getA() const;
 
        protected:
            State S;
            Action A;
 
            // This is mutable because sampling doesn't really change the policy
            mutable RandomEngine rand_;
    };
 
    template <typename State, typename Sampling, typename Action>
    PolicyInterface<State, Sampling, Action>::PolicyInterface(State s, Action a) :
            S(std::move(s)), A(std::move(a)), rand_(Seeder::getSeed()) {}
 
    template <typename State, typename Sampling, typename Action>
    PolicyInterface<State, Sampling, Action>::~PolicyInterface() {}
 
    template <typename State, typename Sampling, typename Action>
    const State & PolicyInterface<State, Sampling, Action>::getS() const { return S; }
 
    template <typename State, typename Sampling, typename Action>
    const Action & PolicyInterface<State, Sampling, Action>::getA() const { return A; }
 
    template <typename Action>
    class PolicyInterface<void, void, Action> {
        public:
            PolicyInterface(Action a);
 
            virtual ~PolicyInterface();
 
            virtual Action sampleAction() const = 0;
 
            virtual double getActionProbability(const Action & a) const = 0;
 
            const Action & getA() const;
 
        protected:
            Action A;
 
            // This is mutable because sampling doesn't really change the policy
            mutable RandomEngine rand_;
    };
 
    template <typename Action>
    PolicyInterface<void, void, Action>::PolicyInterface(Action a) :
            A(std::move(a)), rand_(Seeder::getSeed()) {}
 
    template <typename Action>
    PolicyInterface<void, void, Action>::~PolicyInterface() {}
 
    template <typename Action>
    const Action & PolicyInterface<void, void, Action>::getA() const { return A; }
}
 
#endif