merge_scoring_function_total_order.h

#ifndef MERGE_AND_SHRINK_MERGE_SCORING_FUNCTION_TOTAL_ORDER_H
#define MERGE_AND_SHRINK_MERGE_SCORING_FUNCTION_TOTAL_ORDER_H
 
#include "downward/merge_and_shrink/merge_scoring_function.h"
 
#include <memory>
 
namespace utils {
class RandomNumberGenerator;
}
 
namespace merge_and_shrink {
enum class AtomicTSOrder { REVERSE_LEVEL, LEVEL, RANDOM };
enum class ProductTSOrder { OLD_TO_NEW, NEW_TO_OLD, RANDOM };
 
class MergeScoringFunctionTotalOrder : public MergeScoringFunction {
    AtomicTSOrder atomic_ts_order;
    ProductTSOrder product_ts_order;
    bool atomic_before_product;
    int random_seed; // only for dump options
    std::shared_ptr<utils::RandomNumberGenerator> rng;
    std::vector<std::pair<int, int>> merge_candidate_order;
 
    virtual std::string name() const override;
    virtual void
    dump_function_specific_options(utils::LogProxy& log) const override;
 
public:
    explicit MergeScoringFunctionTotalOrder(
        AtomicTSOrder atomic_ts_order,
        ProductTSOrder product_ts_order,
        bool atomic_before_product,
        int random_seed);
    virtual std::vector<double> compute_scores(
        const FactoredTransitionSystem& fts,
        const std::vector<std::pair<int, int>>& merge_candidates) override;
    virtual void initialize(const TaskProxy& task_proxy) override;
 
    virtual bool requires_init_distances() const override { return false; }
 
    virtual bool requires_goal_distances() const override { return false; }
};
} // namespace merge_and_shrink
 
#endif