3 files changed, 29 insertions, 19 deletions
diff --git a/llvm/include/llvm/CodeGen/RegisterAllocation.h b/llvm/include/llvm/CodeGen/RegisterAllocation.h
index 161be18498b..e552fbe887a 100644
--- a/llvm/include/llvm/CodeGen/RegisterAllocation.h
+++ b/llvm/include/llvm/CodeGen/RegisterAllocation.h
@@ -9,16 +9,12 @@
 
 #include "llvm/Pass.h"
 class TargetMachine;
+class MethodPass;
 
 //----------------------------------------------------------------------------
 // Entry point for register allocation for a module
 //----------------------------------------------------------------------------
 
-class RegisterAllocation : public MethodPass {
-  TargetMachine &Target;
-public:
-  inline RegisterAllocation(TargetMachine &T) : Target(T) {}
-  bool runOnMethod(Method *M);
-};
+MethodPass *getRegisterAllocator(TargetMachine &T);
 
 #endif
diff --git a/llvm/lib/CodeGen/RegAlloc/PhyRegAlloc.cpp b/llvm/lib/CodeGen/RegAlloc/PhyRegAlloc.cpp
index 7efb469b457..46f045c5e27 100644
--- a/llvm/lib/CodeGen/RegAlloc/PhyRegAlloc.cpp
+++ b/llvm/lib/CodeGen/RegAlloc/PhyRegAlloc.cpp
@@ -32,21 +32,35 @@ cl::Enum<RegAllocDebugLevel_t> DEBUG_RA("dregalloc", cl::NoFlags,
   clEnumValN(RA_DEBUG_Verbose, "v", "enable extra debug output"), 0);
 
 
-bool RegisterAllocation::runOnMethod(Method *M) {
-  if (DEBUG_RA)
-    cerr << "\n******************** Method "<< M->getName()
-         << " ********************\n";
-    
-  MethodLiveVarInfo LVI(M);   // Analyze live varaibles
-  LVI.analyze();
+//----------------------------------------------------------------------------
+// RegisterAllocation pass front end...
+//----------------------------------------------------------------------------
+namespace {
+  class RegisterAllocator : public MethodPass {
+    TargetMachine &Target;
+  public:
+    inline RegisterAllocator(TargetMachine &T) : Target(T) {}
     
-  PhyRegAlloc PRA(M, Target, &LVI); // allocate registers
-  PRA.allocateRegisters();
-
-  if (DEBUG_RA) cerr << "\nRegister allocation complete!\n";
-  return false;
+    bool runOnMethod(Method *M) {
+      if (DEBUG_RA)
+        cerr << "\n******************** Method "<< M->getName()
+             << " ********************\n";
+      
+      MethodLiveVarInfo LVI(M);   // Analyze live varaibles
+      LVI.analyze();
+      
+      PhyRegAlloc PRA(M, Target, &LVI); // allocate registers
+      PRA.allocateRegisters();
+      
+      if (DEBUG_RA) cerr << "\nRegister allocation complete!\n";
+      return false;
+    }
+  };
 }
 
+MethodPass *getRegisterAllocator(TargetMachine &T) {
+  return new RegisterAllocator(T);
+}
 
 //----------------------------------------------------------------------------
 // Constructor: Init local composite objects and create register classes.
diff --git a/llvm/lib/Target/Sparc/Sparc.cpp b/llvm/lib/Target/Sparc/Sparc.cpp
index b4e069453cc..91e51d88d28 100644
--- a/llvm/lib/Target/Sparc/Sparc.cpp
+++ b/llvm/lib/Target/Sparc/Sparc.cpp
@@ -260,7 +260,7 @@ void UltraSparc::addPassesToEmitAssembly(PassManager &PM, std::ostream &Out) {
 
   //PM.add(new InstructionScheduling(*this));
 
-  PM.add(new RegisterAllocation(*this));
+  PM.add(getRegisterAllocator(*this));
   
   //PM.add(new OptimizeLeafProcedures());
   //PM.add(new DeleteFallThroughBranches());