added ostream for ShieldExpression

src/storm/logic/ShieldExpression.cpp

@@ -1,5 +1,7 @@
 #include "storm/logic/ShieldExpression.h"
 
+#include <iostream>
+
 namespace storm {
     namespace logic {
         ShieldExpression::ShieldExpression() {}
@@ -27,5 +29,25 @@ namespace storm {
         double ShieldExpression::getValue() const {
             return value;
         }
+
+        std::string ShieldExpression::typeToString() const {
+            switch(type) {
+                case storm::logic::ShieldingType::PostSafety: return "PostSafety";
+                case storm::logic::ShieldingType::PreSafety:  return "PreSafety";
+                case storm::logic::ShieldingType::Optimal:    return "Optimal";
+            }
+        }
+
+        std::string ShieldExpression::comparisonToString() const {
+            switch(comparison) {
+                case storm::logic::ShieldComparison::Absolute: return "gamma";
+                case storm::logic::ShieldComparison::Relative: return "lambda";
+            }
+        }
+
+        std::ostream& operator<<(std::ostream& out, ShieldExpression const& shieldExpression) {
+            out << "<" << shieldExpression.typeToString() << ", " << shieldExpression.comparisonToString() << "=" << std::to_string(shieldExpression.value) << ">";
+            return out;
+        }
     }
 }

src/storm/logic/ShieldExpression.h

@@ -1,6 +1,7 @@
 #pragma once
 
 #include <memory>
+#include <string>
 
 namespace storm {
     namespace logic {
@@ -25,6 +26,11 @@ namespace storm {
 
             double getValue() const;
 
+            std::string typeToString() const;
+            std::string comparisonToString() const;
+
+            friend std::ostream& operator<<(std::ostream& stream, ShieldExpression const& shieldExpression);
+
         private:
             ShieldingType type;
             ShieldComparison comparison;