2 files changed, 27 insertions, 0 deletions
diff --git a/include/llvm/TableGen/Main.h b/include/llvm/TableGen/Main.h
index deaef4a990..663779f176 100644
--- a/include/llvm/TableGen/Main.h
+++ b/include/llvm/TableGen/Main.h
@@ -21,6 +21,14 @@ class TableGenAction;
 /// Run the table generator, performing the specified Action on parsed records.
 int TableGenMain(char *argv0, TableGenAction &Action);
 
+class RecordKeeper;
+class raw_ostream;
+typedef bool TableGenMainFn(raw_ostream &OS, RecordKeeper &Records);
+
+/// Perform the action using Records, and write output to OS.
+/// \returns true on error, false otherwise
+int TableGenMain(char *argv0, TableGenMainFn *MainFn);
+
 }
 
 #endif
diff --git a/lib/TableGen/Main.cpp b/lib/TableGen/Main.cpp
index 7aeef563b8..83bc66f460 100644
--- a/lib/TableGen/Main.cpp
+++ b/lib/TableGen/Main.cpp
@@ -22,6 +22,7 @@
 #include "llvm/Support/ToolOutputFile.h"
 #include "llvm/Support/system_error.h"
 #include "llvm/TableGen/Error.h"
+#include "llvm/TableGen/Main.h"
 #include "llvm/TableGen/Record.h"
 #include "llvm/TableGen/TableGenAction.h"
 #include <algorithm>
@@ -47,8 +48,26 @@ namespace {
               cl::value_desc("directory"), cl::Prefix);
 }
 
+namespace {
+// XXX: this is a crutch for transitioning to the new TableGenMain API
+// (with a TableGenMainFn* instead of a pointless class).
+class StubTransitionalTableGenAction : public TableGenAction {
+  TableGenMainFn *MainFn;
+public:
+  StubTransitionalTableGenAction(TableGenMainFn *M) : MainFn(M) {}
+  bool operator()(raw_ostream &OS, RecordKeeper &Records) {
+    return MainFn(OS, Records);
+  }
+};
+}
+
 namespace llvm {
 
+int TableGenMain(char *argv0, TableGenMainFn *MainFn) {
+  StubTransitionalTableGenAction Action(MainFn);
+  return TableGenMain(argv0, Action);
+}
+
 int TableGenMain(char *argv0, TableGenAction &Action) {
   RecordKeeper Records;