PrismMenuGameAbstractor.h

Go to the documentation of this file.

#pragma once
 
#include "storm/storage/dd/DdType.h"
 
#include "storm-gamebased-ar/abstraction/AbstractionInformation.h"
#include "storm-gamebased-ar/abstraction/MenuGame.h"
#include "storm-gamebased-ar/abstraction/MenuGameAbstractor.h"
#include "storm-gamebased-ar/abstraction/RefinementCommand.h"
#include "storm-gamebased-ar/abstraction/ValidBlockAbstractor.h"
#include "storm-gamebased-ar/abstraction/prism/ModuleAbstractor.h"
 
#include "storm/storage/dd/Add.h"
 
#include "storm/storage/expressions/Expression.h"
 
namespace storm {
namespace utility {
namespace solver {
class SmtSolverFactory;
}
}  // namespace utility
 
class Environment;
 
namespace models {
namespace symbolic {
template<storm::dd::DdType Type, typename ValueType>
class StochasticTwoPlayerGame;
}
}  // namespace models
 
namespace prism {
// Forward-declare concrete Program class.
class Program;
}  // namespace prism
}  // namespace storm
 
namespace storm::gbar {
namespace abstraction {
namespace prism {
 
template<storm::dd::DdType DdType, typename ValueType>
class PrismMenuGameAbstractor : public MenuGameAbstractor<DdType, ValueType> {
   public:
    PrismMenuGameAbstractor(storm::Environment const& env, storm::prism::Program const& program,
                            std::shared_ptr<storm::utility::solver::SmtSolverFactory> const& smtSolverFactory,
                            AbstractionOptions const& options = AbstractionOptions());
 
    PrismMenuGameAbstractor(PrismMenuGameAbstractor const&) = default;
    PrismMenuGameAbstractor& operator=(PrismMenuGameAbstractor const&) = default;
    PrismMenuGameAbstractor(PrismMenuGameAbstractor&&) = default;
    PrismMenuGameAbstractor& operator=(PrismMenuGameAbstractor&&) = default;
 
    storm::dd::DdManager<DdType> const& getDdManager() const override;

    MenuGame<DdType, ValueType> abstract() override;

    AbstractionInformation<DdType> const& getAbstractionInformation() const override;

    storm::expressions::Expression const& getGuard(uint64_t player1Choice) const override;

    virtual uint64_t getNumberOfUpdates(uint64_t player1Choice) const override;

    std::map<storm::expressions::Variable, storm::expressions::Expression> getVariableUpdates(uint64_t player1Choice, uint64_t auxiliaryChoice) const override;

    virtual std::set<storm::expressions::Variable> const& getAssignedVariables(uint64_t player1Choice) const override;

    std::pair<uint64_t, uint64_t> getPlayer1ChoiceRange() const override;

    storm::expressions::Expression getInitialExpression() const override;
 
    storm::dd::Bdd<DdType> getStates(storm::expressions::Expression const& expression) override;

    virtual void refine(RefinementCommand const& command) override;

    virtual void exportToDot(std::string const& filename, storm::dd::Bdd<DdType> const& highlightStates, storm::dd::Bdd<DdType> const& filter) const override;
 
    virtual uint64_t getNumberOfPredicates() const override;
 
    virtual void addTerminalStates(storm::expressions::Expression const& expression) override;
 
    virtual void notifyGuardsArePredicates() override;
 
   protected:
    using MenuGameAbstractor<DdType, ValueType>::exportToDot;
 
   private:
    std::unique_ptr<MenuGame<DdType, ValueType>> buildGame();
 
    // The concrete program this abstract program refers to.
    std::reference_wrapper<storm::prism::Program const> program;
 
    // A factory that can be used to create new SMT solvers.
    std::shared_ptr<storm::utility::solver::SmtSolverFactory> smtSolverFactory;
 
    // An object containing all information about the abstraction like predicates and the corresponding DDs.
    AbstractionInformation<DdType> abstractionInformation;
 
    // The abstract modules of the abstract program.
    std::vector<ModuleAbstractor<DdType, ValueType>> modules;
 
    // A state-set abstractor used to determine the initial states of the abstraction.
    StateSetAbstractor<DdType, ValueType> initialStateAbstractor;
 
    // A flag indicating whether the valid blocks need to be computed and the game restricted to these.
    bool restrictToValidBlocks;
 
    // An object that is used to compute the valid blocks.
    ValidBlockAbstractor<DdType> validBlockAbstractor;
 
    // An ADD characterizing the probabilities of commands and their updates.
    storm::dd::Add<DdType, ValueType> commandUpdateProbabilitiesAdd;
 
    // The current game-based abstraction.
    std::unique_ptr<MenuGame<DdType, ValueType>> currentGame;
 
    // A flag storing whether a refinement was performed.
    bool refinementPerformed;
 
    // A list of terminal state expressions.
    std::vector<storm::expressions::Expression> terminalStateExpressions;
};
}  // namespace prism
}  // namespace abstraction
}  // namespace storm::gbar