From 1470d65586b28de659ad2a0309cee2668808c3c9 Mon Sep 17 00:00:00 2001
From: Stefan Pranger <stefan.pranger@student.tugraz.at>
Date: Wed, 16 Dec 2020 10:41:15 +0100
Subject: [PATCH] gameForumlas now gather referenced variables

---
 src/storm/logic/GameFormula.cpp | 4 ++++
 src/storm/logic/GameFormula.h   | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/src/storm/logic/GameFormula.cpp b/src/storm/logic/GameFormula.cpp
index 697a695a0..bdd74440c 100644
--- a/src/storm/logic/GameFormula.cpp
+++ b/src/storm/logic/GameFormula.cpp
@@ -20,6 +20,10 @@ namespace storm {
             return coalition;
         }
 
+        void GameFormula::gatherReferencedRewardModels(std::set<std::string>& referencedRewardModels) const {
+            this->getSubformula().gatherReferencedRewardModels(referencedRewardModels);
+        }
+
         boost::any GameFormula::accept(FormulaVisitor const& visitor, boost::any const& data) const {
             return visitor.visit(*this, data);
         }
diff --git a/src/storm/logic/GameFormula.h b/src/storm/logic/GameFormula.h
index 113b9e65f..719571767 100644
--- a/src/storm/logic/GameFormula.h
+++ b/src/storm/logic/GameFormula.h
@@ -20,6 +20,8 @@ namespace storm {
             Formula const& getSubformula() const;
             Coalition getCoalition() const;
 
+            virtual void gatherReferencedRewardModels(std::set<std::string>& referencedRewardModels) const;
+
             boost::any accept(FormulaVisitor const& visitor, boost::any const& data) const;
 
             std::ostream& writeToStream(std::ostream& out) const;