state_ranking_function.h

#ifndef PROBFD_PDBS_STATE_RANKING_FUNCTION_H
#define PROBFD_PDBS_STATE_RANKING_FUNCTION_H
 
#include "probfd/pdbs/assignment_enumerator.h"
#include "probfd/pdbs/types.h"
 
#include "downward/task_proxy.h"
 
#include <span>
#include <string>
#include <vector>
 
namespace probfd::pdbs {
 
class StateRankingFunction {
    Pattern pattern_;
    AssignmentEnumerator enumerator_;
 
public:
    StateRankingFunction() = default;
    
    StateRankingFunction(const VariablesProxy& variables, Pattern pattern);
 
    [[nodiscard]]
    unsigned int num_states() const;
 
    [[nodiscard]]
    unsigned int num_vars() const;
 
    [[nodiscard]]
    const Pattern& get_pattern() const;
 
    [[nodiscard]]
    const AssignmentEnumerator& get_enumerator() const;
 
    [[nodiscard]]
    long long int get_multiplier(int i) const;
 
    [[nodiscard]]
    int get_domain_size(int i) const;
 
    [[nodiscard]]
    StateRank get_abstract_rank(const State& state) const;
 
    [[nodiscard]]
    int rank_fact(int idx, int val) const;
 
    [[nodiscard]]
    std::vector<int> unrank(StateRank abstract_state) const;
 
    [[nodiscard]]
    int value_of(StateRank rank, int idx) const;
 
    bool next_rank(StateRank& s, std::span<int> mutable_variables) const;
};
 
class StateRankToString {
    const VariablesProxy variables_;
    const StateRankingFunction& state_mapper_;
 
public:
    explicit StateRankToString(
        VariablesProxy variables,
        const StateRankingFunction& state_mapper);
 
    std::string operator()(StateRank state) const;
};
 
} // namespace probfd::pdbs
 
#endif // PROBFD_PDBS_STATE_RANKING_FUNCTION_H