state_space.h

#ifndef PROBFD_STATE_SPACE_H
#define PROBFD_STATE_SPACE_H
 
#include "probfd/type_traits.h"
#include "probfd/types.h"
 
#include <vector>
 
// Forward Declarations
namespace probfd {
template <typename>
class Distribution;
template <typename>
struct Transition;
} // namespace probfd
 
namespace probfd {
 
template <typename State, typename Action>
class StateSpace {
protected:
    using TransitionType = Transition<Action>;
 
public:
    virtual ~StateSpace() = default;
 
    virtual StateID get_state_id(param_type<State> state) = 0;
 
    virtual State get_state(StateID state_id) = 0;
 
    virtual void generate_applicable_actions(
        param_type<State> state,
        std::vector<Action>& result) = 0;
 
    virtual void generate_action_transitions(
        param_type<State> state,
        param_type<Action> action,
        Distribution<StateID>& result) = 0;
 
    virtual void generate_all_transitions(
        param_type<State> state,
        std::vector<Action>& aops,
        std::vector<Distribution<StateID>>& successors) = 0;
 
    virtual void generate_all_transitions(
        param_type<State> state,
        std::vector<TransitionType>& transitions) = 0;
};
 
} // namespace probfd
 
#endif