include/llvm/CodeGen/MachineInstrBuilder.h: Add addClobber() inline

[oota-llvm.git] / lib / Target / X86 / Printer.cpp

diff --git a/lib/Target/X86/Printer.cpp b/lib/Target/X86/Printer.cpp
index 1a42ed86f84058ca9bc73f4b3deef859c2cea7b4..fdb444e54eef44a82a667d96df5776ccabadd136 100644 (file)
--- a/lib/Target/X86/Printer.cpp
+++ b/lib/Target/X86/Printer.cpp
@@ -6,16 +6,63 @@
 //===----------------------------------------------------------------------===//
 
 #include "X86.h"
+#include "X86InstrInfo.h"
+#include "llvm/Pass.h"
+#include "llvm/Function.h"
+#include "llvm/Target/TargetMachine.h"
+#include "llvm/CodeGen/MachineFunction.h"
 #include <iostream>
 
-/// X86PrintCode - Print out the specified machine code function to the
-/// specified stream.  This function should work regardless of whether or not
-/// the function is in SSA form or not, although when in SSA form, we obviously
-/// don't care about being consumable by an assembler.
-///
-void X86PrintCode(const MachineFunction *MF, std::ostream &O) {
-  O << "x86 printing not implemented yet!\n";
+namespace {
+  struct Printer : public FunctionPass {
+    TargetMachine &TM;
+    std::ostream &O;
+
+    Printer(TargetMachine &tm, std::ostream &o) : TM(tm), O(o) {}
+
+    bool runOnFunction(Function &F);
+  };
+}
+
+/// runOnFunction - This uses the X86InstructionInfo::print method
+/// to print assembly for each instruction.
+bool Printer::runOnFunction (Function & F)
+{
+  static unsigned bbnumber = 0;
+  MachineFunction & MF = MachineFunction::get (&F);
+  const MachineInstrInfo & MII = TM.getInstrInfo ();
+  const X86InstrInfo & x86ii = dynamic_cast <const X86InstrInfo &> (MII);
+
+  O << "# x86 printing not implemented yet!\n";
 
-  // This should use the X86InstructionInfo::print method to print assembly for
-  // each instruction
+  // Print out labels for the function.
+  O << "\t.globl\t" << F.getName () << "\n";
+  O << "\t.type\t" << F.getName () << ", @function\n";
+  O << F.getName () << ":\n";
+
+  // Print out code for the function.
+  for (MachineFunction::const_iterator bb_i = MF.begin (), bb_e = MF.end ();
+       bb_i != bb_e; ++bb_i)
+    {
+      // Print a label for the basic block.
+      O << ".BB" << bbnumber++ << ":\n";
+      for (MachineBasicBlock::const_iterator i_i = bb_i->begin (), i_e =
+          bb_i->end (); i_i != i_e; ++i_i)
+       {
+         // Print the assembly for the instruction.
+         O << "\t";
+         x86ii.print (*i_i, O);
+       }
+    }
+
+  // We didn't modify anything.
+  return false;
+}
+
+/// createX86CodePrinterPass - Print out the specified machine code function to
+/// the specified stream.  This function should work regardless of whether or
+/// not the function is in SSA form or not.
+///
+Pass *createX86CodePrinterPass(TargetMachine &TM, std::ostream &O) {
+  return new Printer(TM, O);
 }