You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

32 lines
711 B

  1. #ifndef MRMC_PARSER_SPARSESTATEREWARDPARSER_H_
  2. #define MRMC_PARSER_SPARSESTATEREWARDPARSER_H_
  3. #include "boost/integer/integer_mask.hpp"
  4. #include "src/parser/Parser.h"
  5. #include <memory>
  6. #include <vector>
  7. namespace mrmc {
  8. namespace parser {
  9. /*!
  10. * @brief Load state reward file and return vector of state rewards.
  11. */
  12. class SparseStateRewardParser : Parser {
  13. public:
  14. SparseStateRewardParser(uint_fast64_t stateCount, std::string const &filename);
  15. std::shared_ptr<std::vector<double>> getStateRewards() {
  16. return this->stateRewards;
  17. }
  18. private:
  19. std::shared_ptr<std::vector<double>> stateRewards;
  20. };
  21. } // namespace parser
  22. } // namespace mrmc
  23. #endif /* MRMC_PARSER_SPARSESTATEREWARDPARSER_H_ */