Add a custom initialize hook for clang tools + minor fixes in CustomCompilationDatabase.h

llvm-svn: 160369

1 files changed, 7 insertions, 0 deletions

diff --git a/clang/lib/Tooling/CommandLineClangTool.cpp b/clang/lib/Tooling/CommandLineClangTool.cpp
index 8da2a335a55..32b2ac2857b 100644
--- a/clang/lib/Tooling/CommandLineClangTool.cpp
+++ b/clang/lib/Tooling/CommandLineClangTool.cpp
@@ -25,6 +25,10 @@
 #include "clang/Tooling/CommandLineClangTool.h"
 #include "clang/Tooling/Tooling.h"
 
+#ifdef USE_CUSTOM_TOOL_INIT
+#include "CustomToolInit.h"
+#endif
+
 using namespace clang::tooling;
 using namespace llvm;
 
@@ -58,6 +62,9 @@ CommandLineClangTool::CommandLineClangTool() :
 }
 
 void CommandLineClangTool::initialize(int argc, const char **argv) {
+#ifdef USE_CUSTOM_TOOL_INIT
+  customToolInit(argc, argv);
+#endif
   Compilations.reset(FixedCompilationDatabase::loadFromCommandLine(argc, argv));
   cl::ParseCommandLineOptions(argc, argv);
   if (!Compilations) {