You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
2.7 KiB
51 lines
2.7 KiB
#pragma once
|
|
|
|
#include <sstream>
|
|
#include <unordered_map>
|
|
|
|
#include "storm/storage/expressions/Variable.h"
|
|
#include "storm/storage/expressions/ExpressionVisitor.h"
|
|
|
|
namespace storm {
|
|
namespace expressions {
|
|
class Expression;
|
|
|
|
enum class ToCppTranslationMode {
|
|
KeepType, CastDouble, CastRationalNumber, CastRationalFunction
|
|
};
|
|
|
|
class ToCppTranslationOptions {
|
|
public:
|
|
ToCppTranslationOptions(std::unordered_map<storm::expressions::Variable, std::string> const& prefixes, std::unordered_map<storm::expressions::Variable, std::string> const& names, ToCppTranslationMode mode = ToCppTranslationMode::KeepType);
|
|
|
|
std::unordered_map<storm::expressions::Variable, std::string> const& getPrefixes() const;
|
|
std::unordered_map<storm::expressions::Variable, std::string> const& getNames() const;
|
|
ToCppTranslationMode const& getMode() const;
|
|
|
|
private:
|
|
std::reference_wrapper<std::unordered_map<storm::expressions::Variable, std::string> const> prefixes;
|
|
std::reference_wrapper<std::unordered_map<storm::expressions::Variable, std::string> const> names;
|
|
ToCppTranslationMode mode;
|
|
};
|
|
|
|
class ToCppVisitor : public ExpressionVisitor {
|
|
public:
|
|
std::string translate(storm::expressions::Expression const& expression, ToCppTranslationOptions const& options);
|
|
|
|
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;
|
|
|
|
private:
|
|
std::stringstream stream;
|
|
};
|
|
|
|
}
|
|
}
|