variable_order_finder.h

#ifndef TASK_UTILS_VARIABLE_ORDER_FINDER_H
#define TASK_UTILS_VARIABLE_ORDER_FINDER_H
 
#include "downward/task_proxy.h"
 
#include <memory>
#include <vector>
 
namespace utils {
class LogProxy;
class RandomNumberGenerator;
} // namespace utils
 
namespace variable_order_finder {
enum VariableOrderType {
    CG_GOAL_LEVEL,
    CG_GOAL_RANDOM,
    GOAL_CG_LEVEL,
    RANDOM,
    LEVEL,
    REVERSE_LEVEL
};
 
extern void dump_variable_order_type(
    VariableOrderType variable_order_type,
    utils::LogProxy& log);
 
/*
  NOTE: VariableOrderFinder keeps a copy of the task proxy passed to the
  constructor. Therefore, users of the class must ensure that the underlying
  task lives at least as long as the variable order finder.
*/
class VariableOrderFinder {
    TaskProxy task_proxy;
    const VariableOrderType variable_order_type;
    std::vector<int> selected_vars;
    std::vector<int> remaining_vars;
    std::vector<bool> is_goal_variable;
    std::vector<bool> is_causal_predecessor;
 
    void select_next(int position, int var_no);
 
public:
    VariableOrderFinder(
        const TaskProxy& task_proxy,
        VariableOrderType variable_order_type,
        const std::shared_ptr<utils::RandomNumberGenerator>& rng = nullptr);
    ~VariableOrderFinder() = default;
    bool done() const;
    int next();
};
} // namespace variable_order_finder
 
#endif