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 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 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);