1 //===- DiagTool.h - Classes for defining diagtool tools -------------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // This file implements the boilerplate for defining diagtool tools. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #ifndef DIAGTOOL_DIAGTOOL_H 15 #define DIAGTOOL_DIAGTOOL_H 16 17 #include "llvm/ADT/StringRef.h" 18 #include "llvm/Support/ManagedStatic.h" 19 #include "llvm/Support/raw_ostream.h" 20 #include <string> 21 22 23 namespace diagtool { 24 25 class DiagTool { 26 const std::string cmd; 27 const std::string description; 28 public: 29 DiagTool(llvm::StringRef toolCmd, llvm::StringRef toolDesc); 30 virtual ~DiagTool(); 31 32 llvm::StringRef getName() const { return cmd; } 33 llvm::StringRef getDescription() const { return description; } 34 35 virtual int run(unsigned argc, char *argv[], llvm::raw_ostream &out) = 0; 36 }; 37 38 class DiagTools { 39 void *tools; 40 public: 41 DiagTools(); 42 ~DiagTools(); 43 44 DiagTool *getTool(llvm::StringRef toolCmd); 45 void registerTool(DiagTool *tool); 46 void printCommands(llvm::raw_ostream &out); 47 }; 48 49 extern llvm::ManagedStatic<DiagTools> diagTools; 50 51 template <typename DIAGTOOL> 52 class RegisterDiagTool { 53 public: 54 RegisterDiagTool() { diagTools->registerTool(new DIAGTOOL()); } 55 }; 56 57 } // end diagtool namespace 58 59 #define DEF_DIAGTOOL(NAME, DESC, CLSNAME)\ 60 namespace {\ 61 class CLSNAME : public diagtool::DiagTool {\ 62 public:\ 63 CLSNAME() : DiagTool(NAME, DESC) {}\ 64 virtual ~CLSNAME() {}\ 65 virtual int run(unsigned argc, char *argv[], llvm::raw_ostream &out);\ 66 };\ 67 diagtool::RegisterDiagTool<CLSNAME> Register##CLSNAME;\ 68 } 69 70 #endif 71
