policy.h

#ifndef PROBFD_POLICY_H
#define PROBFD_POLICY_H
 
#include "probfd/multi_policy.h"
 
#include <functional>
#include <optional>
#include <vector>
 
namespace probfd {
 
template <typename State, typename Action>
class Policy : public MultiPolicy<State, Action> {
public:
    virtual ~Policy() = default;
 
    virtual std::optional<PolicyDecision<Action>>
    get_decision(const State& state) const = 0;
 
    std::vector<PolicyDecision<Action>>
    get_decisions(const State& state) const override
    {
        std::optional decision = this->get_decision(state);
        std::vector<PolicyDecision<Action>> decisions;
        if (decision) decisions.emplace_back(std::move(*decision));
        return decisions;
    }
 
    virtual void print(
        std::ostream& out,
        std::function<void(const State&, std::ostream&)> state_printer,
        std::function<void(const Action&, std::ostream&)> action_printer) = 0;
};
 
} // namespace probfd
 
#endif