add isl_aff_mod_val
[platform/upstream/isl.git] / interface / extract_interface.cc
index 89baf81..5aa6294 100644 (file)
@@ -31,6 +31,8 @@
  * Sven Verdoolaege.
  */ 
 
+#include "isl_config.h"
+
 #include <assert.h>
 #include <iostream>
 #include <llvm/Support/raw_ostream.h>
 #include <clang/Basic/TargetOptions.h>
 #include <clang/Basic/TargetInfo.h>
 #include <clang/Basic/Version.h>
+#include <clang/Driver/Compilation.h>
+#include <clang/Driver/Driver.h>
+#include <clang/Driver/Tool.h>
 #include <clang/Frontend/CompilerInstance.h>
 #include <clang/Frontend/CompilerInvocation.h>
+#ifdef HAVE_BASIC_DIAGNOSTICOPTIONS_H
+#include <clang/Basic/DiagnosticOptions.h>
+#else
 #include <clang/Frontend/DiagnosticOptions.h>
+#endif
 #include <clang/Frontend/TextDiagnosticPrinter.h>
 #include <clang/Frontend/Utils.h>
 #include <clang/Lex/HeaderSearch.h>
 #include <clang/Parse/ParseAST.h>
 #include <clang/Sema/Sema.h>
 
-#include "isl_config.h"
 #include "extract_interface.h"
 #include "python.h"
 
 using namespace std;
 using namespace clang;
+using namespace clang::driver;
 
 static llvm::cl::opt<string> InputFilename(llvm::cl::Positional,
                        llvm::cl::Required, llvm::cl::desc("<input file>"));
@@ -106,14 +115,14 @@ struct MyASTConsumer : public ASTConsumer {
        set<RecordDecl *> types;
        set<FunctionDecl *> functions;
 
-       virtual void HandleTopLevelDecl(DeclGroupRef D) {
+       virtual HandleTopLevelDeclReturn HandleTopLevelDecl(DeclGroupRef D) {
                Decl *decl;
 
                if (!D.isSingleDecl())
-                       return;
+                       return HandleTopLevelDeclContinue;
                decl = D.getSingleDecl();
                if (!is_exported(decl))
-                       return;
+                       return HandleTopLevelDeclContinue;
                switch (decl->getKind()) {
                case Decl::Record:
                        types.insert(cast<RecordDecl>(decl));
@@ -124,24 +133,158 @@ struct MyASTConsumer : public ASTConsumer {
                default:
                        break;
                }
+               return HandleTopLevelDeclContinue;
        }
 };
 
+#ifdef USE_ARRAYREF
+
+#ifdef HAVE_CXXISPRODUCTION
+static Driver *construct_driver(const char *binary, DiagnosticsEngine &Diags)
+{
+       return new Driver(binary, llvm::sys::getDefaultTargetTriple(),
+                           "", false, false, Diags);
+}
+#elif defined(HAVE_ISPRODUCTION)
+static Driver *construct_driver(const char *binary, DiagnosticsEngine &Diags)
+{
+       return new Driver(binary, llvm::sys::getDefaultTargetTriple(),
+                           "", false, Diags);
+}
+#else
+static Driver *construct_driver(const char *binary, DiagnosticsEngine &Diags)
+{
+       return new Driver(binary, llvm::sys::getDefaultTargetTriple(),
+                           "", Diags);
+}
+#endif
+
+/* Create a CompilerInvocation object that stores the command line
+ * arguments constructed by the driver.
+ * The arguments are mainly useful for setting up the system include
+ * paths on newer clangs and on some platforms.
+ */
+static CompilerInvocation *construct_invocation(const char *filename,
+       DiagnosticsEngine &Diags)
+{
+       const char *binary = CLANG_PREFIX"/bin/clang";
+       const llvm::OwningPtr<Driver> driver(construct_driver(binary, Diags));
+       std::vector<const char *> Argv;
+       Argv.push_back(binary);
+       Argv.push_back(filename);
+       const llvm::OwningPtr<Compilation> compilation(
+               driver->BuildCompilation(llvm::ArrayRef<const char *>(Argv)));
+       JobList &Jobs = compilation->getJobs();
+
+       Command *cmd = cast<Command>(*Jobs.begin());
+       if (strcmp(cmd->getCreator().getName(), "clang"))
+               return NULL;
+
+       const ArgStringList *args = &cmd->getArguments();
+
+       CompilerInvocation *invocation = new CompilerInvocation;
+       CompilerInvocation::CreateFromArgs(*invocation, args->data() + 1,
+                                               args->data() + args->size(),
+                                               Diags);
+       return invocation;
+}
+
+#else
+
+static CompilerInvocation *construct_invocation(const char *filename,
+       DiagnosticsEngine &Diags)
+{
+       return NULL;
+}
+
+#endif
+
+#ifdef HAVE_BASIC_DIAGNOSTICOPTIONS_H
+
+static TextDiagnosticPrinter *construct_printer(void)
+{
+       return new TextDiagnosticPrinter(llvm::errs(), new DiagnosticOptions());
+}
+
+#else
+
+static TextDiagnosticPrinter *construct_printer(void)
+{
+       DiagnosticOptions DO;
+       return new TextDiagnosticPrinter(llvm::errs(), DO);
+}
+
+#endif
+
+#ifdef CREATETARGETINFO_TAKES_POINTER
+
+static TargetInfo *create_target_info(CompilerInstance *Clang,
+       DiagnosticsEngine &Diags)
+{
+       TargetOptions &TO = Clang->getTargetOpts();
+       TO.Triple = llvm::sys::getDefaultTargetTriple();
+       return TargetInfo::CreateTargetInfo(Diags, &TO);
+}
+
+#else
+
+static TargetInfo *create_target_info(CompilerInstance *Clang,
+       DiagnosticsEngine &Diags)
+{
+       TargetOptions &TO = Clang->getTargetOpts();
+       TO.Triple = llvm::sys::getDefaultTargetTriple();
+       return TargetInfo::CreateTargetInfo(Diags, TO);
+}
+
+#endif
+
+#ifdef CREATEDIAGNOSTICS_TAKES_ARG
+
+static void create_diagnostics(CompilerInstance *Clang)
+{
+       Clang->createDiagnostics(0, NULL, construct_printer());
+}
+
+#else
+
+static void create_diagnostics(CompilerInstance *Clang)
+{
+       Clang->createDiagnostics(construct_printer());
+}
+
+#endif
+
+#ifdef ADDPATH_TAKES_4_ARGUMENTS
+
+void add_path(HeaderSearchOptions &HSO, string Path)
+{
+       HSO.AddPath(Path, frontend::Angled, false, false);
+}
+
+#else
+
+void add_path(HeaderSearchOptions &HSO, string Path)
+{
+       HSO.AddPath(Path, frontend::Angled, true, false, false);
+}
+
+#endif
+
 int main(int argc, char *argv[])
 {
        llvm::cl::ParseCommandLineOptions(argc, argv);
 
        CompilerInstance *Clang = new CompilerInstance();
-       DiagnosticOptions DO;
-       Clang->createDiagnostics(0, NULL,
-                               new TextDiagnosticPrinter(llvm::errs(), DO));
+       create_diagnostics(Clang);
        DiagnosticsEngine &Diags = Clang->getDiagnostics();
        Diags.setSuppressSystemWarnings(true);
+       CompilerInvocation *invocation =
+               construct_invocation(InputFilename.c_str(), Diags);
+       if (invocation)
+               Clang->setInvocation(invocation);
        Clang->createFileManager();
        Clang->createSourceManager(Clang->getFileManager());
-       TargetOptions TO;
-       TO.Triple = llvm::sys::getDefaultTargetTriple();
-       TargetInfo *target = TargetInfo::CreateTargetInfo(Diags, TO);
+       TargetInfo *target = create_target_info(Clang, Diags);
        Clang->setTarget(target);
        CompilerInvocation::setLangDefaults(Clang->getLangOpts(), IK_C,
                                            LangStandard::lang_unspecified);
@@ -151,7 +294,7 @@ int main(int argc, char *argv[])
        HSO.ResourceDir = ResourceDir;
 
        for (int i = 0; i < Includes.size(); ++i)
-               HSO.AddPath(Includes[i], frontend::Angled, true, false, false);
+               add_path(HSO, Includes[i]);
 
        PO.addMacroDef("__isl_give=__attribute__((annotate(\"isl_give\")))");
        PO.addMacroDef("__isl_keep=__attribute__((annotate(\"isl_keep\")))");