LinearCoefficientVisitor.h

Go to the documentation of this file.

#pragma once
 
#include <stack>
 
#include "storm/storage/expressions/Expression.h"
#include "storm/storage/expressions/ExpressionVisitor.h"
#include "storm/storage/expressions/SimpleValuation.h"
#include "storm/storage/expressions/Variable.h"
 
namespace storm {
namespace expressions {
class LinearCoefficientVisitor : public ExpressionVisitor {
   public:
    struct VariableCoefficients {
       public:
        VariableCoefficients(double constantPart = 0);
 
        VariableCoefficients(VariableCoefficients const& other) = default;
        VariableCoefficients& operator=(VariableCoefficients const& other) = default;
        VariableCoefficients(VariableCoefficients&& other) = default;
        VariableCoefficients& operator=(VariableCoefficients&& other) = default;
 
        VariableCoefficients& operator+=(VariableCoefficients&& other);
        VariableCoefficients& operator-=(VariableCoefficients&& other);
        VariableCoefficients& operator*=(VariableCoefficients&& other);
        VariableCoefficients& operator/=(VariableCoefficients&& other);
 
        void negate();
        void setCoefficient(storm::expressions::Variable const& variable, double coefficient);
        double getCoefficient(storm::expressions::Variable const& variable);
        double getConstantPart() const;

        void separateVariablesFromConstantPart(VariableCoefficients& rhs);
 
        // Propagate the iterators to variable-coefficient pairs.
        std::map<storm::expressions::Variable, double>::const_iterator begin() const;
        std::map<storm::expressions::Variable, double>::const_iterator end() const;
 
        size_t size() const;
 
       private:
        std::map<storm::expressions::Variable, double> variableToCoefficientMapping;
        double constantPart;
    };

    LinearCoefficientVisitor() = default;

    VariableCoefficients getLinearCoefficients(Expression const& expression);
 
    virtual boost::any visit(IfThenElseExpression const& expression, boost::any const& data) override;
    virtual boost::any visit(BinaryBooleanFunctionExpression const& expression, boost::any const& data) override;
    virtual boost::any visit(BinaryNumericalFunctionExpression const& expression, boost::any const& data) override;
    virtual boost::any visit(BinaryRelationExpression const& expression, boost::any const& data) override;
    virtual boost::any visit(VariableExpression const& expression, boost::any const& data) override;
    virtual boost::any visit(UnaryBooleanFunctionExpression const& expression, boost::any const& data) override;
    virtual boost::any visit(UnaryNumericalFunctionExpression const& expression, boost::any const& data) override;
    virtual boost::any visit(BooleanLiteralExpression const& expression, boost::any const& data) override;
    virtual boost::any visit(IntegerLiteralExpression const& expression, boost::any const& data) override;
    virtual boost::any visit(RationalLiteralExpression const& expression, boost::any const& data) override;
};
}  // namespace expressions
}  // namespace storm