[llvm-commits] [llvm] r44051 - /llvm/trunk/tools/llvm2cpp/CppWriter.cpp

[Chris Lattner]

Author: lattner
Date: Tue Nov 13 12:22:33 2007
New Revision: 44051

URL: http://llvm.org/viewvc/llvm-project?rev=44051&view=rev
Log:
Make llvm2cpp better, patch for PR1794, contributed by Zack Rusin.



Modified:
    llvm/trunk/tools/llvm2cpp/CppWriter.cpp

Modified: llvm/trunk/tools/llvm2cpp/CppWriter.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/tools/llvm2cpp/CppWriter.cpp?rev=44051&r1=44050&r2=44051&view=diff

==============================================================================
--- llvm/trunk/tools/llvm2cpp/CppWriter.cpp (original)
+++ llvm/trunk/tools/llvm2cpp/CppWriter.cpp Tue Nov 13 12:22:33 2007
@@ -44,6 +44,7 @@
   GenModule,
   GenContents,
   GenFunction,
+  GenFunctions,
   GenInline,
   GenVariable,
   GenType
@@ -53,13 +54,14 @@
   cl::desc("Choose what kind of output to generate"),
   cl::init(GenProgram),
   cl::values(
-    clEnumValN(GenProgram, "gen-program",  "Generate a complete program"),
-    clEnumValN(GenModule,  "gen-module",   "Generate a module definition"),
-    clEnumValN(GenContents,"gen-contents", "Generate contents of a module"),
-    clEnumValN(GenFunction,"gen-function", "Generate a function definition"),
-    clEnumValN(GenInline,  "gen-inline",   "Generate an inline function"),
-    clEnumValN(GenVariable,"gen-variable", "Generate a variable definition"),
-    clEnumValN(GenType,    "gen-type",     "Generate a type definition"),
+    clEnumValN(GenProgram,  "gen-program",   "Generate a complete program"),
+    clEnumValN(GenModule,   "gen-module",    "Generate a module definition"),
+    clEnumValN(GenContents, "gen-contents",  "Generate contents of a module"),
+    clEnumValN(GenFunction, "gen-function",  "Generate a function definition"),
+    clEnumValN(GenFunctions,"gen-functions", "Generate all function definitions"),
+    clEnumValN(GenInline,   "gen-inline",    "Generate an inline function"),
+    clEnumValN(GenVariable, "gen-variable",  "Generate a variable definition"),
+    clEnumValN(GenType,     "gen-type",      "Generate a type definition"),
     clEnumValEnd
   )
 );
@@ -103,6 +105,7 @@
   void printModule(const std::string& fname, const std::string& modName );
   void printContents(const std::string& fname, const std::string& modName );
   void printFunction(const std::string& fname, const std::string& funcName );
+  void printFunctions();
   void printInline(const std::string& fname, const std::string& funcName );
   void printVariable(const std::string& fname, const std::string& varName );
   void printType(const std::string& fname, const std::string& typeName );
@@ -1784,6 +1787,21 @@
   Out << "}\n";
 }
 
+void CppWriter::printFunctions() {
+  const Module::FunctionListType &funcs = TheModule->getFunctionList();
+  Module::const_iterator I  = funcs.begin();
+  Module::const_iterator IE = funcs.end();
+
+  for (; I != IE; ++I) {
+    const Function &func = *I;
+    if (!func.isDeclaration()) {
+      std::string name("define_");
+      name += func.getName();
+      printFunction(name, func.getName());
+    }
+  }
+}
+
 void CppWriter::printVariable(
   const std::string& fname,  /// Name of generated function
   const std::string& varName // Name of variable to generate
@@ -1835,7 +1853,8 @@
   std::string tgtname = NameToGenerate.getValue();
   if (GenerationType == GenModule || 
       GenerationType == GenContents || 
-      GenerationType == GenProgram) {
+      GenerationType == GenProgram ||
+      GenerationType == GenFunctions) {
     if (tgtname == "!bad!") {
       if (mod->getModuleIdentifier() == "-")
         tgtname = "<stdin>";
@@ -1867,6 +1886,9 @@
         fname = "makeLLVMFunction";
       W.printFunction(fname,tgtname);
       break;
+  case GenFunctions:
+      W.printFunctions();
+      break;
     case GenInline:
       if (fname.empty())
         fname = "makeLLVMInline";