probability_aware_pattern_database.h

#ifndef PROBFD_PDBS_PROBABILITY_AWARE_PATTERN_DATABASE_H
#define PROBFD_PDBS_PROBABILITY_AWARE_PATTERN_DATABASE_H
 
#include "probfd/pdbs/evaluators.h"
#include "probfd/pdbs/state_ranking_function.h"
#include "probfd/pdbs/types.h"
 
#include "probfd/heuristics/constant_evaluator.h"
 
#include "probfd/fdr_types.h"
 
#include <limits>
#include <vector>
 
namespace probfd {
class ProbabilisticTaskProxy;
}
 
namespace probfd::pdbs {
class ProjectionStateSpace;
}
 
namespace probfd::pdbs {
 
class ProbabilityAwarePatternDatabase {
    StateRankingFunction ranking_function_;
    std::vector<value_t> value_table_;
 
    ProbabilityAwarePatternDatabase(
        ProbabilisticTaskProxy task_proxy,
        Pattern pattern);
 
    explicit ProbabilityAwarePatternDatabase(
        StateRankingFunction ranking_function);
 
public:
    ProbabilityAwarePatternDatabase(
        StateRankingFunction ranking_function,
        std::vector<value_t> value_table);
 
    ProbabilityAwarePatternDatabase(
        ProbabilisticTaskProxy task_proxy,
        std::shared_ptr<FDRSimpleCostFunction> task_cost_function,
        Pattern pattern,
        const State& initial_state,
        bool operator_pruning = true,
        const StateRankEvaluator& heuristic =
            heuristics::BlindEvaluator<StateRank>(),
        double max_time = std::numeric_limits<double>::infinity());
 
    ProbabilityAwarePatternDatabase(
        ProjectionStateSpace& projection,
        StateRankingFunction ranking_function,
        StateRank initial_state,
        const StateRankEvaluator& heuristic =
            heuristics::BlindEvaluator<StateRank>(),
        double max_time = std::numeric_limits<double>::infinity());
 
    ProbabilityAwarePatternDatabase(
        ProbabilisticTaskProxy task_proxy,
        std::shared_ptr<FDRSimpleCostFunction> task_cost_function,
        const ::pdbs::PatternDatabase& pdb,
        const State& initial_state,
        bool operator_pruning = true,
        double max_time = std::numeric_limits<double>::infinity());
 
    ProbabilityAwarePatternDatabase(
        ProjectionStateSpace& projection,
        StateRankingFunction ranking_function,
        const ::pdbs::PatternDatabase& pdb,
        StateRank initial_state,
        double max_time = std::numeric_limits<double>::infinity());
 
    ProbabilityAwarePatternDatabase(
        ProbabilisticTaskProxy task_proxy,
        std::shared_ptr<FDRSimpleCostFunction> task_cost_function,
        const ProbabilityAwarePatternDatabase& pdb,
        int add_var,
        const State& initial_state,
        bool operator_pruning = true,
        double max_time = std::numeric_limits<double>::infinity());
 
    ProbabilityAwarePatternDatabase(
        ProjectionStateSpace& state_space,
        StateRankingFunction ranking_function,
        const ProbabilityAwarePatternDatabase& pdb,
        int add_var,
        StateRank initial_state,
        double max_time = std::numeric_limits<double>::infinity());
 
    ProbabilityAwarePatternDatabase(
        ProbabilisticTaskProxy task_proxy,
        std::shared_ptr<FDRSimpleCostFunction> task_cost_function,
        const ProbabilityAwarePatternDatabase& left,
        const ProbabilityAwarePatternDatabase& right,
        const State& initial_state,
        bool operator_pruning = true,
        double max_time = std::numeric_limits<double>::infinity());
 
    ProbabilityAwarePatternDatabase(
        ProjectionStateSpace& state_space,
        StateRankingFunction ranking_function,
        const ProbabilityAwarePatternDatabase& left,
        const ProbabilityAwarePatternDatabase& right,
        StateRank initial_state,
        double max_time = std::numeric_limits<double>::infinity());
 
    [[nodiscard]]
    const Pattern& get_pattern() const;
 
    [[nodiscard]]
    const StateRankingFunction& get_state_ranking_function() const;
 
    [[nodiscard]]
    const std::vector<value_t>& get_value_table() const;
 
    [[nodiscard]]
    unsigned int num_states() const;
 
    [[nodiscard]]
    StateRank get_abstract_state(const State& state) const;
 
    [[nodiscard]]
    value_t lookup_estimate(const State& s) const;
 
    [[nodiscard]]
    value_t lookup_estimate(StateRank s) const;
};
 
} // namespace probfd::pdbs
 
#endif // PROBFD_PDBS_PROBABILITY_AWARE_PATTERN_DATABASE_H