merge_tree_factory.h

#ifndef MERGE_AND_SHRINK_MERGE_TREE_FACTORY_H
#define MERGE_AND_SHRINK_MERGE_TREE_FACTORY_H
 
#include <memory>
#include <string>
#include <vector>
 
class TaskProxy;
 
namespace utils {
class LogProxy;
class RandomNumberGenerator;
} // namespace utils
 
namespace merge_and_shrink {
class FactoredTransitionSystem;
class MergeTree;
enum class UpdateOption;
 
class MergeTreeFactory {
protected:
    std::shared_ptr<utils::RandomNumberGenerator> rng;
    UpdateOption update_option;
    virtual std::string name() const = 0;
    virtual void dump_tree_specific_options(utils::LogProxy&) const {}
 
public:
    MergeTreeFactory(int random_seed, UpdateOption update_option);
    virtual ~MergeTreeFactory() = default;
    void dump_options(utils::LogProxy& log) const;
    // Compute a merge tree for the given entire task.
    virtual std::unique_ptr<MergeTree>
    compute_merge_tree(const TaskProxy& task_proxy) = 0;
    /* Compute a merge tree for the given current factored transition,
       system, possibly for a subset of indices. */
    virtual std::unique_ptr<MergeTree> compute_merge_tree(
        const TaskProxy& task_proxy,
        const FactoredTransitionSystem& fts,
        const std::vector<int>& indices_subset);
    virtual bool requires_init_distances() const = 0;
    virtual bool requires_goal_distances() const = 0;
};
 
} // namespace merge_and_shrink
 
#endif