Driver.h

#ifndef ROSE_CodeGen_Driver_H
#define ROSE_CodeGen_Driver_H

#include <set>
#include <map>
#include <string>

#include <boost/filesystem.hpp>

class SgProject;
class SgSourceFile;
class SgScopeStatement;
class SgSymbol;
class SgGlobal;

namespace Rose { namespace CodeGen {

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