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);

    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;

    void setUnparsedFile(size_t file_id) const;

    void setCompiledFile(size_t file_id) const;

    void addIncludeDirectives(size_t target_file_id, size_t header_file_id);

    void addExternalHeader(size_t file_id, std::string header_name, bool is_system_header = true);

    void addPragmaDecl(size_t file_id, std::string str);
};

} }

#endif /* Rose_CodeGen_Driver_h */