From 0c724c9b685c265723b801b40eb128c08141538c Mon Sep 17 00:00:00 2001
From: Tobias Winkler <tobias.winkler@cs.rwth-aachen.de>
Date: Sun, 27 Sep 2020 19:55:56 +0200
Subject: [PATCH] added getExpression for formula

---
 src/logic/formulae.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/logic/formulae.cpp b/src/logic/formulae.cpp
index 3e4d773..69f87e3 100644
--- a/src/logic/formulae.cpp
+++ b/src/logic/formulae.cpp
@@ -54,7 +54,8 @@ void define_formulae(py::module& m) {
 
     // State Formulae
     py::class_<storm::logic::StateFormula, std::shared_ptr<storm::logic::StateFormula>> stateFormula(m, "StateFormula", "Formula about a state of an automaton", formula);
-    py::class_<storm::logic::AtomicExpressionFormula, std::shared_ptr<storm::logic::AtomicExpressionFormula>>(m, "AtomicExpressionFormula", "Formula with an atomic expression", stateFormula);
+    py::class_<storm::logic::AtomicExpressionFormula, std::shared_ptr<storm::logic::AtomicExpressionFormula>>(m, "AtomicExpressionFormula", "Formula with an atomic expression", stateFormula)
+        .def("get_expression", &storm::logic::AtomicExpressionFormula::getExpression);
     py::class_<storm::logic::AtomicLabelFormula, std::shared_ptr<storm::logic::AtomicLabelFormula>>(m, "AtomicLabelFormula", "Formula with an atomic label", stateFormula);
     py::class_<storm::logic::BooleanLiteralFormula, std::shared_ptr<storm::logic::BooleanLiteralFormula>>(m, "BooleanLiteralFormula", "Formula with a boolean literal", stateFormula)
             .def(py::init<bool>(),"truth value"_a);