From 55b344c560159cac81ca2189bdc3f3b5f643eb6e Mon Sep 17 00:00:00 2001
From: Sebastian Junges <sebastian.junges@gmail.com>
Date: Sun, 24 May 2020 22:21:21 -0700
Subject: [PATCH] but state valuations as comments into drn

---
 src/storm/utility/DirectEncodingExporter.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/storm/utility/DirectEncodingExporter.cpp b/src/storm/utility/DirectEncodingExporter.cpp
index 5e1ce3231..2ea83e9d7 100644
--- a/src/storm/utility/DirectEncodingExporter.cpp
+++ b/src/storm/utility/DirectEncodingExporter.cpp
@@ -115,6 +115,10 @@ namespace storm {
                     }
                 }
                 os << std::endl;
+                // Write state valuations as comments
+                if(sparseModel->hasStateValuations()) {
+                    os << "//" << sparseModel->getStateValuations().getStateInfo(group) << std::endl;
+                }
 
                 // Write probabilities
                 typename storm::storage::SparseMatrix<ValueType>::index_type start = matrix.hasTrivialRowGrouping() ? group : matrix.getRowGroupIndices()[group];