00001 #ifndef SimulationRewardCollector_H 00002 #define SimulationRewardCollector_H 00003 #include <cmath> 00004 #include <vector> 00005 00006 00007 #include "solverUtils.h" 00008 00009 using namespace momdp; 00010 using namespace std; 00011 namespace momdp 00012 { 00013 00014 class SimulationRewardCollector 00015 { 00016 private: 00017 double globalRew; 00018 double totalVar; 00019 00020 vector<double> expRewardRecord; 00021 00022 SolverParams p; 00023 00024 bool legendPrint; 00025 00026 void calculateConfidenceInterval(); 00027 public: 00028 double globalExpRew; 00029 double confInterval; 00030 00031 SimulationRewardCollector(void); 00032 virtual ~SimulationRewardCollector(void); 00033 00034 void setup(SolverParams& p); 00035 void addEntry(int currSim, double reward, double expReward); 00036 void printReward(int currSim); 00037 void printFinalReward(); 00038 00039 00040 00041 }; 00042 } 00043 00044 #endif 00045