From bc3646604de6aa91029fca78d4852ba145edf6f3 Mon Sep 17 00:00:00 2001
From: sjunges <sebastian.junges@gmail.com>
Date: Wed, 16 Aug 2017 21:03:21 +0200
Subject: [PATCH] Extend ModelType with an entry for POMDPs

---
 src/storm/models/ModelType.cpp | 5 +++++
 src/storm/models/ModelType.h   | 2 +-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/storm/models/ModelType.cpp b/src/storm/models/ModelType.cpp
index 46bc06dc9..956c9baed 100644
--- a/src/storm/models/ModelType.cpp
+++ b/src/storm/models/ModelType.cpp
@@ -17,6 +17,8 @@ namespace storm {
                 return ModelType::MarkovAutomaton;
             } else if (type == "S2PG") {
                 return ModelType::S2pg;
+            } else if (type == "Pomdp") {
+                return ModelType::Pomdp;
             } else {
                 STORM_LOG_THROW(false, storm::exceptions::InvalidTypeException, "Type " << type << "not known.");
             }
@@ -39,6 +41,9 @@ namespace storm {
                 case ModelType::S2pg:
                     os << "S2PG";
                     break;
+                case ModelType::Pomdp:
+                    os << "POMDP";
+                    break;
                 default:
                     STORM_LOG_THROW(false, storm::exceptions::InvalidTypeException, "Unknown model type.");
             }
diff --git a/src/storm/models/ModelType.h b/src/storm/models/ModelType.h
index 40d4da220..b183a598d 100644
--- a/src/storm/models/ModelType.h
+++ b/src/storm/models/ModelType.h
@@ -7,7 +7,7 @@ namespace storm {
     namespace models {
         // All supported model types.
         enum class ModelType {
-            Dtmc, Ctmc, Mdp, MarkovAutomaton, S2pg
+            Dtmc, Ctmc, Mdp, MarkovAutomaton, S2pg, Pomdp
         };
 
         ModelType getModelType(std::string const& type);