Cvc5ExpressionAdapter.h

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <unordered_map>
#include <vector>
 
#include "storm-config.h"
 
// Include the headers of cvc5 only if it is available.
#ifdef STORM_HAVE_CVC5
#include <cvc5/cvc5.h>
#endif
 
#include "storm/adapters/RationalNumberAdapter.h"
#include "storm/storage/expressions/ExpressionVisitor.h"
#include "storm/storage/expressions/Variable.h"
 
namespace storm::expressions {
class BaseExpression;
}
 
namespace storm::adapters {
 
#ifdef STORM_HAVE_CVC5
class Cvc5ExpressionAdapter : public storm::expressions::ExpressionVisitor {
   public:
    Cvc5ExpressionAdapter(storm::expressions::ExpressionManager& manager, cvc5::Solver& solver);

    cvc5::Term translateExpression(storm::expressions::Expression const& expression);

    cvc5::Term translateExpression(storm::expressions::Variable const& variable);

    storm::expressions::Expression translateExpression(cvc5::Term const& term);

    storm::expressions::Variable const& getVariable(cvc5::Term const& constant);
 
    virtual boost::any visit(storm::expressions::BinaryBooleanFunctionExpression const& expression, boost::any const& data) override;
 
    virtual boost::any visit(storm::expressions::BinaryNumericalFunctionExpression const& expression, boost::any const& data) override;
 
    virtual boost::any visit(storm::expressions::BinaryRelationExpression const& expression, boost::any const& data) override;
 
    virtual boost::any visit(storm::expressions::BooleanLiteralExpression const& expression, boost::any const& data) override;
 
    virtual boost::any visit(storm::expressions::RationalLiteralExpression const& expression, boost::any const& data) override;
 
    virtual boost::any visit(storm::expressions::IntegerLiteralExpression const& expression, boost::any const& data) override;
 
    virtual boost::any visit(storm::expressions::UnaryBooleanFunctionExpression const& expression, boost::any const& data) override;
 
    virtual boost::any visit(storm::expressions::UnaryNumericalFunctionExpression const& expression, boost::any const& data) override;
 
    virtual boost::any visit(storm::expressions::IfThenElseExpression const& expression, boost::any const& data) override;
 
    virtual boost::any visit(storm::expressions::VariableExpression const& expression, boost::any const& data) override;
 
   private:
    cvc5::Term createVariable(storm::expressions::Variable const& variable);

    cvc5::Term createRational(storm::RationalNumber const& value);

    cvc5::Term convertToRealIfExpected(cvc5::Term const& term, bool const& isRealExpected) const;

    void unifyNumericalTypes(cvc5::Term& leftTerm, cvc5::Term& rightTerm) const;
 
    // The manager that can be used to build expressions.
    storm::expressions::ExpressionManager& manager;
 
    // The manager of the solver the adapter translates for, which is where terms and sorts are built.
    cvc5::TermManager& termManager;
 
    // A mapping from variables to their cvc5 equivalent.
    std::unordered_map<storm::expressions::Variable, cvc5::Term> variableToExpressionMapping;
 
    // A mapping from cvc5 constants to the corresponding variables.
    std::unordered_map<cvc5::Term, storm::expressions::Variable> expressionToVariableMapping;
 
    // A cache of already translated constraints. Only valid during the translation of one expression.
    std::unordered_map<storm::expressions::BaseExpression const*, cvc5::Term> expressionCache;
};
#endif
}  // namespace storm::adapters