O3prmrInterpreter.h

Go to the documentation of this file.

#ifndef SKOORINTERPRETER_H
#define SKOORINTERPRETER_H

#ifndef DOXYGEN_SHOULD_SKIP_THIS
#  include <agrum/PRM/inference/PRMInference.h>
#  include <agrum/PRM/o3prm/O3prmReader.h>
#  include <agrum/PRM/o3prmr/O3prmrContext.h>

#  include <agrum/core/errorsContainer.h>
#endif   // DOXYGEN_SHOULD_SKIP_THIS

namespace gum {

  namespace prm {

    namespace o3prmr {

      class SingleResult {
        public:
        std::string label;
        float       p;
      };

      class QueryResult {
        public:
        double                      time;
        std::string                 command;
        std::vector< SingleResult > values;
      };

      class O3prmrInterpreter {
        public:
        O3prmrInterpreter();
        ~O3prmrInterpreter();

        bool interpretFile(const std::string& filename);
        bool interpretLine(const std::string& line);

        const gum::prm::PRM< double >* prm() const;
        const gum::prm::PRMInference< double >* inference() const;
        const std::vector< QueryResult >& results() const;
        ErrorsContainer errorsContainer() const;

        O3prmrContext< double >* getContext() const;
        void                     setContext(O3prmrContext< double >* context);

        std::vector< std::string > getPaths() const;
        void                       addPath(std::string path);
        void                       clearPaths();

        bool isInSyntaxMode() const;
        void setSyntaxMode(bool f);

        bool isVerboseMode() const;
        void setVerboseMode(bool f);

        Size count() const;
        Size errors() const;
        Size warnings() const;
        ParseError error(Idx i) const;
        void showElegantErrors(std::ostream& o = std::cerr) const;
        void showElegantErrorsAndWarnings(std::ostream& o = std::cerr) const;
        void showErrorCounts(std::ostream& o = std::cerr) const;

        private:
        bool checkSemantic(O3prmrContext< double >* context);
        bool checkSetEngine(SetEngineCommand* command);
        bool checkSetGndEngine(SetGndEngineCommand* command);
        bool checkObserve(ObserveCommand< double >* command);
        bool checkUnobserve(UnobserveCommand< double >* command);
        bool checkQuery(QueryCommand< double >* command);

        bool interpret(O3prmrContext< double >* c);
        bool import(O3prmrContext< double >* context, std::string import);
        bool observe(const ObserveCommand< double >* command);
        bool unobserve(const UnobserveCommand< double >* command);
        void query(const QueryCommand< double >* command);
        void setEngine(const SetEngineCommand* command);
        void setGndEngine(const SetGndEngineCommand* command);

        std::string findSystemName(std::string& s);
        std::string findInstanceName(std::string&                         s,
                                     const gum::prm::PRMSystem< double >& sys);
        std::string
                                   findAttributeName(const std::string&                     s,
                                                     const gum::prm::PRMInstance< double >& instance);
        const PRMSystem< double >& system(std::string& ident);
        void generateInfEngine(const gum::prm::PRMSystem< double >& sys);

        void addError(std::string msg);
        void addWarning(std::string msg);

        std::string __readFile(const std::string& file);

        O3prmrContext< double >*                m_context;
        gum::prm::o3prm::O3prmReader< double >* m_reader;
        std::vector< std::string >              m_paths;
        gum::BayesNet< double >*                m_bn;
        // Don't delete this, the m_inf_map will do it
        gum::prm::PRMInference< double >*                                m_inf;
        HashTable< const PRMSystem< double >*, PRMInference< double >* > m_inf_map;
        std::string                                                      m_engine;
        std::string                m_bn_engine;
        std::vector< QueryResult > m_results;
        gum::ErrorsContainer       m_errors;
        bool                       m_syntax_flag;
        bool                       m_verbose;
        std::ostream&              m_log;
        int                        m_current_line;

        // gum::BayesNetInference<gum::prm::prm_float>* bn;

      };   // class O3prmrInterpreter

    }   // namespace o3prmr
  }     // namespace prm
}   // namespace gum

#endif   // GUM_SKOORINTERPRETER_H