From 3355b21b72926e8fc1060b9f53e072a17c222f6a Mon Sep 17 00:00:00 2001
From: Sebastian Junges <sebastian.junges@gmail.com>
Date: Mon, 8 Mar 2021 16:28:17 -0800
Subject: [PATCH] add timing info to storm-pomdp

---
 src/storm-pomdp-cli/storm-pomdp.cpp | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/storm-pomdp-cli/storm-pomdp.cpp b/src/storm-pomdp-cli/storm-pomdp.cpp
index ba2ed275d..5df3aeee6 100644
--- a/src/storm-pomdp-cli/storm-pomdp.cpp
+++ b/src/storm-pomdp-cli/storm-pomdp.cpp
@@ -475,12 +475,18 @@ int main(const int argc, const char** argv) {
         if (!optionsCorrect) {
             return -1;
         }
+        storm::utility::Stopwatch totalTimer(true);
         storm::cli::setUrgentOptions();
 
         // Invoke storm-pomdp with obtained settings
         storm::pomdp::cli::processOptions();
 
-        // All operations have now been performed, so we clean up everything and terminate.
+        totalTimer.stop();
+        if (storm::settings::getModule<storm::settings::modules::ResourceSettings>().isPrintTimeAndMemorySet()) {
+            storm::cli::printTimeAndMemoryStatistics(totalTimer.getTimeInMilliseconds());
+        }
+
+    // All operations have now been performed, so we clean up everything and terminate.
         storm::utility::cleanUp();
         return 0;
     // } catch (storm::exceptions::BaseException const &exception) {