[cxxmodules][wip] Add SetupModules call to load all available PCMs

Author: Teemperor

core/base/src/TROOT.cxx

@@ -2035,6 +2035,8 @@ void TROOT::InitInterpreter()
 
    // Enable autoloading
    fInterpreter->EnableAutoLoading();
+   if (fInterpreter->getCI()->getLangOpts().Modules)
+     fInterpreter->SetupModules();
 }
 
 ////////////////////////////////////////////////////////////////////////////////

core/meta/inc/TInterpreter.h

@@ -162,6 +162,7 @@ class TInterpreter : public TNamed {
                                void (*histaddFunc)(const char* line)) = 0;
    virtual void     Reset() = 0;
    virtual void     ResetAll() = 0;
+   virtual void     SetupModules() = 0;
    virtual void     ResetGlobals() = 0;
    virtual void     ResetGlobalVar(void *obj) = 0;
    virtual void     RewindDictionary() = 0;

core/metacling/src/TCling.cxx

@@ -3768,6 +3768,12 @@ void TCling::UpdateListOfDataMembers(TClass* cl) const
 {
 }
 
+void TCling::SetupModules() {
+  assert (fInterpreter->getCI()->getLangOpts().Modules &&
+          "Should only be called with modules enabled!");
+  fInterpreter->setupModules();
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 /// Create list of pointers to method arguments for TMethod m.
 

core/metacling/src/TCling.h

@@ -234,6 +234,7 @@ class TCling : public TInterpreter {
    void    CreateListOfMethodArgs(TFunction* m) const;
    void    UpdateListOfMethods(TClass* cl) const;
    void    UpdateListOfDataMembers(TClass* cl) const;
+   virtual void SetupModules();
 
    virtual DeclId_t GetDataMember(ClassInfo_t *cl, const char *name) const;
    virtual DeclId_t GetDataMemberAtAddr(const void *addr) const;

interpreter/cling/include/cling/Interpreter/Interpreter.h

@@ -648,6 +648,8 @@ namespace cling {
     int getDefaultOptLevel() const { return m_OptLevel; }
     void setDefaultOptLevel(int optLevel) { m_OptLevel = optLevel; }
 
+    void setupModules();
+
     clang::CompilerInstance* getCI() const;
     clang::CompilerInstance* getCIOrNull() const;
     clang::Sema& getSema() const;

interpreter/cling/lib/Interpreter/Interpreter.cpp

@@ -60,6 +60,9 @@
 
 #include <string>
 #include <vector>
+#include <fstream>
+#include <iostream>
+#include <sstream>
 
 using namespace clang;
 
@@ -1473,6 +1476,77 @@ namespace cling {
     m_DynamicLookupEnabled = value;
   }
 
+  void Interpreter::setupModules() {
+    clang::CompilerInstance* CI = getCI();
+    std::stringstream declarations;
+    if (CI->getLangOpts().Modules && CI->getLangOpts().CurrentModule.empty()) {
+
+       clang::HeaderSearch &headerSearch = CI->getPreprocessor().getHeaderSearchInfo();
+       clang::ModuleMap &moduleMap = headerSearch.getModuleMap();
+
+       llvm::SetVector<clang::Module *> modules;
+       auto& PP = getParser().getPreprocessor();
+
+       for (auto MI = moduleMap.module_begin(), end = moduleMap.module_end(); MI != end; MI++) {
+          clang::Module* Module = MI->second;
+          HeaderSearchOptions &HSOpts =
+              PP.getHeaderSearchInfo().getHeaderSearchOpts();
+
+          std::string ModuleFileName;
+          bool LoadFromPrebuiltModulePath = false;
+          // We try to load the module from the prebuilt module paths. If not
+          // successful, we then try to find it in the module cache.
+          if (!HSOpts.PrebuiltModulePaths.empty()) {
+            // Load the module from the prebuilt module path.
+            ModuleFileName = PP.getHeaderSearchInfo().getModuleFileName(
+                Module->Name, "", /*UsePrebuiltPath*/ true);
+            if (!ModuleFileName.empty())
+              LoadFromPrebuiltModulePath = true;
+          }
+          if (!LoadFromPrebuiltModulePath && Module) {
+            // Load the module from the module cache.
+            ModuleFileName = PP.getHeaderSearchInfo().getModuleFileName(Module);
+          }
+          bool Exists = false;
+          {
+            std::ifstream f(ModuleFileName);
+            Exists = f.good();
+          }
+          if (Exists)
+            modules.insert(Module);
+       }
+
+       for (size_t i = 0; i < modules.size(); ++i) {
+          clang::Module *M = modules[i];
+          for (clang::Module *subModule : M->submodules()) modules.insert(subModule);
+       }
+
+       // Now we collect all header files from the previously collected modules.
+       std::set<StringRef> moduleHeaders;
+       for (clang::Module *module : modules) {
+          for (int i = 0; i < 4; i++) {
+             auto &headerList = module->Headers[i];
+             for (const clang::Module::Header &moduleHeader : headerList) {
+                moduleHeaders.insert(moduleHeader.NameAsWritten);
+             }
+          }
+       }
+
+       for (StringRef H : moduleHeaders) {
+          if (H == "GL/glew.h")
+             continue;
+          if (H == "GL/glxew.h")
+             continue;
+          if (H.endswith(".inc"))
+             continue;
+          if (H == "TException.h")
+              continue;
+          declarations << "#include \"" << H << "\"\n";
+       }
+    }
+    declare(declarations.str());
+  }
+
   Interpreter::ExecutionResult
   Interpreter::executeTransaction(Transaction& T) {
     assert(!isInSyntaxOnlyMode() && "Running on what?");