From 11fc1d29eb7dab8deb067ac23e48b5fa51c1b3a6 Mon Sep 17 00:00:00 2001
From: TimQu <tim.quatmann@cs.rwth-aachen.de>
Date: Wed, 6 Sep 2017 14:40:30 +0200
Subject: [PATCH] more output for statistics

---
 .../rewardbounded/MultiDimensionalRewardUnfolding.cpp      | 1 +
 .../rewardbounded/MultiDimensionalRewardUnfolding.h        | 7 +++++++
 2 files changed, 8 insertions(+)

diff --git a/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.cpp b/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.cpp
index b80727dfb..18c53b5af 100644
--- a/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.cpp
+++ b/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.cpp
@@ -315,6 +315,7 @@ namespace storm {
                 }
                 
                 swSetEpochClass.stop();
+                epochModelSizes.push_back(epochModel.epochMatrix.getRowGroupCount());
             }
             
             template<typename ValueType>
diff --git a/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.h b/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.h
index 8e864c21f..3e5d4924f 100644
--- a/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.h
+++ b/src/storm/modelchecker/multiobjective/rewardbounded/MultiDimensionalRewardUnfolding.h
@@ -60,6 +60,12 @@ namespace storm {
                     std::cout << "     findSolutions: " << swFindSol << " seconds." << std::endl;
                     std::cout << "   insertSolutions: " << swInsertSol << " seconds." << std::endl;
                     std::cout << "---------------------------------------------" << std::endl;
+                    std::cout << "      Product size: " << memoryProduct.getProduct().getNumberOfStates();
+                    std::cout << " Epoch model sizes: ";
+                    for (auto const& i : epochModelSizes) {
+                        std::cout << i << " ";
+                    }
+                    std::cout << std::endl;
                     std::cout << "---------------------------------------------" << std::endl;
                     std::cout << std::endl;
                     
@@ -156,6 +162,7 @@ namespace storm {
                 
                 
                 storm::utility::Stopwatch swInit, swFindSol, swInsertSol, swSetEpoch, swSetEpochClass;
+                std::vector<uint64_t> epochModelSizes;
             };
         }
     }