liftingToDot.hpp

Go to the documentation of this file.

 
/*
**  Copyright (C) - Triton
**
**  This program is under the terms of the Apache License 2.0.
*/
 
#ifndef TRITON_LIFTINGTODOT_HPP
#define TRITON_LIFTINGTODOT_HPP
 
#include <map>
#include <ostream>
#include <set>
#include <unordered_map>
#include <utility>
 
#include <triton/astContext.hpp>
#include <triton/dllexport.hpp>
#include <triton/symbolicEngine.hpp>
#include <triton/symbolicExpression.hpp>
 
 
 
namespace triton {
  namespace engines {
    namespace lifters {
 
      class LiftingToDot {
        private:
          triton::ast::SharedAstContext astCtxt;
 
          triton::engines::symbolic::SymbolicEngine* symbolic;
 
          triton::usize uniqueid;
 
          std::map<triton::ast::AbstractNode*, triton::engines::symbolic::SymbolicExpression*> information;
 
          std::unordered_map<triton::usize, triton::engines::symbolic::SharedSymbolicExpression> expressions;
 
          std::set<std::pair<triton::usize, triton::usize>> edges;
 
          std::set<std::pair<triton::usize, std::string>> nodes;
 
          void spreadInformation(std::ostream& stream);
 
          void defineLegend(std::ostream& stream);
 
          void iterateNodes(const triton::ast::SharedAbstractNode& root);
 
          void handleVariable(const triton::ast::SharedAbstractNode& parent, const triton::ast::SharedAbstractNode& var);
 
        public:
          TRITON_EXPORT LiftingToDot(const triton::ast::SharedAstContext& astCtxt, triton::engines::symbolic::SymbolicEngine* symbolic);
 
          TRITON_EXPORT std::ostream& liftToDot(std::ostream& stream, const triton::ast::SharedAbstractNode& node);
 
          TRITON_EXPORT std::ostream& liftToDot(std::ostream& stream, const triton::engines::symbolic::SharedSymbolicExpression& expr);
      };
 
    };
  };
};
 
#endif /* TRITON_LIFTINGTODOT_HPP */