From 8fe3b7b1f885f41fe8116d76d665ea06fce3f07c Mon Sep 17 00:00:00 2001
From: Sebastian Junges <sebastian.junges@rwth-aachen.de>
Date: Wed, 10 Oct 2018 19:53:21 +0200
Subject: [PATCH] give edges a color to mark them from user side

---
 src/storm/storage/jani/Edge.cpp |  8 ++++++++
 src/storm/storage/jani/Edge.h   | 13 +++++++++++++
 2 files changed, 21 insertions(+)

diff --git a/src/storm/storage/jani/Edge.cpp b/src/storm/storage/jani/Edge.cpp
index 301c695a0..3ec065cf0 100644
--- a/src/storm/storage/jani/Edge.cpp
+++ b/src/storm/storage/jani/Edge.cpp
@@ -91,6 +91,14 @@ namespace storm {
         bool Edge::hasSilentAction() const {
             return actionIndex == Model::SILENT_ACTION_INDEX;
         }
+
+        uint64_t Edge::getColor() const {
+            return this->color;
+        }
+
+        void Edge::setColor(uint64_t newColor) {
+            this->color = newColor;
+        }
         
         boost::container::flat_set<storm::expressions::Variable> const& Edge::getWrittenGlobalVariables() const {
             return templateEdge->getWrittenGlobalVariables();
diff --git a/src/storm/storage/jani/Edge.h b/src/storm/storage/jani/Edge.h
index 099f5ca14..5af6ec38d 100644
--- a/src/storm/storage/jani/Edge.h
+++ b/src/storm/storage/jani/Edge.h
@@ -111,6 +111,16 @@ namespace storm {
              */
             bool usesAssignmentLevels(bool onlyTransient = false) const;
 
+            /*!
+             * Retrieves the color of the edge
+             */
+            uint64_t getColor() const;
+
+            /*!
+             * Sets the color of the edge
+             */
+            void setColor(uint64_t newColor);
+
             /*!
              *
              * @param localVars
@@ -151,6 +161,9 @@ namespace storm {
             
             /// The concrete destination objects.
             std::vector<EdgeDestination> destinations;
+
+            /// The color of the edge, used to persistently mark and identify specific edges (by the user)
+            uint64_t color;
         };
         
         std::ostream& operator<<(std::ostream& stream, Edge const& edge);