Driver.h

#ifndef ROSE_Source_Generation_Driver_H
#define ROSE_Source_Generation_Driver_H
 
#include <RoseFirst.h>
 
#include <set>
#include <map>
#include <string>
 
#include <boost/filesystem.hpp>
 
class SgProject;
class SgSourceFile;
class SgScopeStatement;
class SgSymbol;
class SgGlobal;
 
namespace Rose { namespace Source { namespace Generation {
 
class Driver {
  public:
    SgProject * project; 
 
  private:
    size_t file_id_counter;
    std::map<boost::filesystem::path, size_t> path_to_id_map;
    std::map<size_t, SgSourceFile *> id_to_file_map;
    std::map<SgSourceFile *, size_t> file_to_id_map;
    std::map<size_t, std::set<size_t> > file_id_to_accessible_file_id_map;
 
    void init(SgProject * project_);
    size_t add(SgSourceFile * file);
 
  public:
    Driver();
    Driver(SgProject * project_);
    Driver(std::vector<std::string> & args);
 
    void addCxxExtension(std::string) const;
 
    size_t create(const boost::filesystem::path & path);
    size_t add(const boost::filesystem::path & path);
 
    size_t getFileID(const boost::filesystem::path & path) const;
    size_t getFileID(SgSourceFile * source_file) const;
    size_t getFileID(SgScopeStatement * scope) const;
 
    SgGlobal * getGlobalScope(size_t file_id) const;
 
    SgSourceFile * getSourceFile(size_t file_id) const;
 
    void setUnparsedFile(size_t file_id) const;
 
    void setCompiledFile(size_t file_id) const;
 
    void addExternalHeader(size_t file_id, std::string header_name, bool is_system_header = true) const;
 
    void addPragmaDecl(size_t file_id, std::string str) const;
 
    template <typename ContainerFileID0, typename ContainerFileID1>
    void exportNameQualification(ContainerFileID0 const & unparsed_ids, ContainerFileID1 const & header_ids) const {  
      for (auto unparsed_id: unparsed_ids) {
        auto & extra_nodes_for_namequal_init = getSourceFile(unparsed_id)->get_extra_nodes_for_namequal_init();
        for (auto header_id: header_ids) {
          extra_nodes_for_namequal_init.push_back(getGlobalScope(header_id));
        }
      }
    }
};
 
} } }
 
#endif /* ROSE_Source_Generation_Driver_H */