policy_picker.h

#ifndef PROBFD_ALGORITHMS_POLICY_PICKER_H
#define PROBFD_ALGORITHMS_POLICY_PICKER_H
 
#include "probfd/types.h"
 
#include <optional>
#include <ostream>
#include <vector>
 
// Forward Declarations
namespace probfd {
template <typename>
class Distribution;
template <typename>
struct Transition;
template <typename, typename>
class MDP;
} // namespace probfd
 
namespace probfd::algorithms {
class StateProperties;
}
 
namespace probfd::algorithms {
template <typename State, typename Action>
class PolicyPicker {
public:
    virtual ~PolicyPicker() = default;
 
    virtual int pick_index(
        MDP<State, Action>& mdp,
        std::optional<Action> incumbent_greedy,
        const std::vector<Transition<Action>>& greedy_transitions,
        StateProperties& properties) = 0;
 
    virtual void print_statistics(std::ostream&) {}
};
 
} // namespace probfd::algorithms
 
#endif // PROBFD_ALGORITHMS_POLICY_PICKER_H