From 6f40f24b74ca61ecfcdd89630e21be560249bf08 Mon Sep 17 00:00:00 2001
From: sjunges <sebastian.junges@gmail.com>
Date: Thu, 9 Feb 2017 09:42:41 +0100
Subject: [PATCH] JANI operator to set level in assignment

---
 src/storm/storage/jani/Assignment.cpp | 6 +++++-
 src/storm/storage/jani/Assignment.h   | 7 ++++++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/storm/storage/jani/Assignment.cpp b/src/storm/storage/jani/Assignment.cpp
index 203d92933..19655a132 100644
--- a/src/storm/storage/jani/Assignment.cpp
+++ b/src/storm/storage/jani/Assignment.cpp
@@ -43,7 +43,11 @@ namespace storm  {
         int64_t Assignment::getLevel() const {
             return level;
         }
-        
+
+        void Assignment::setLevel(int64_t level) const {
+            this->level = level;
+        }
+
         bool Assignment::isLinear() const {
             storm::expressions::LinearityCheckVisitor linearityChecker;
             return linearityChecker.check(this->getAssignedExpression(), true);
diff --git a/src/storm/storage/jani/Assignment.h b/src/storm/storage/jani/Assignment.h
index 46c0a467e..a4e2dbd23 100644
--- a/src/storm/storage/jani/Assignment.h
+++ b/src/storm/storage/jani/Assignment.h
@@ -51,7 +51,12 @@ namespace storm {
              * Retrieves the level of the assignment.
              */
             int64_t getLevel() const;
-            
+
+            /*!
+             * Sets the level
+             */
+            void setLevel(int64_t level);
+
             /*!
              * Checks the assignment for linearity.
              */