CassandraParser.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_CASSANDRA_PARSER_HEADER_FILE
#define AI_TOOLBOX_CASSANDRA_PARSER_HEADER_FILE
 
#include <AIToolbox/Types.hpp>
#include <string>
#include <vector>
#include <unordered_map>
#include <functional>
 
namespace AIToolbox {
    class CassandraParser {
        public:
            using MDPVals = std::tuple<size_t, size_t, DumbMatrix3D, DumbMatrix3D, double>;
            using POMDPVals = std::tuple<size_t, size_t, size_t, DumbMatrix3D, DumbMatrix3D, DumbMatrix3D, double>;
 
            CassandraParser();
 
            MDPVals parseMDP(std::istream & input);
 
            POMDPVals parsePOMDP(std::istream & input);
 
        private:
            using IDMap = std::unordered_map<std::string, size_t>;
            using ActionMap = std::unordered_map<std::string, std::function<void(const std::string &)>>;
            using Tokens = std::vector<std::string>;
 
            void parseModelInfo(std::istream & input);
 
            size_t extractIDs(const std::string & line, IDMap & map);
 
            Tokens tokenize(const std::string & str, const char * list);
 
            std::vector<size_t> parseIndeces(const std::string & str, const IDMap & map, size_t max);
 
            std::vector<double> parseVector(Tokens::const_iterator begin, Tokens::const_iterator end, size_t N);
 
            std::vector<double> parseVector(const std::string & str, size_t N);
 
            void processMatrix(DumbMatrix3D & M, const IDMap & d1map, const IDMap & d3map);
 
            void processReward(DumbMatrix3D & R);
 
            // Storage for lines which are not empty and not used in the preamble.
            std::vector<std::string> lines_;
            size_t i_;
 
            // Storage for input preamble.
            // We need these so that we can store lambdas in the initMap_ with
            // the same signature.
            size_t S_, A_, O_;
            double discount_;
 
            // These are actions to perform for the preamble.
            ActionMap initMap_;
 
            // These contain the stringToken->id maps.
            IDMap stateMap_;
            IDMap actionMap_;
            IDMap observationMap_;
    };
}
 
#endif