1 /* Title: RegClass.h -*- C++ -*-
2 Author: Ruchira Sasanka
4 Purpose: Contains machine independent methods for register coloring.
11 #include "llvm/CodeGen/IGNode.h"
12 #include "llvm/CodeGen/InterferenceGraph.h"
13 #include "llvm/Target/MachineRegInfo.h"
16 class MachineRegClassInfo;
18 typedef std::vector<unsigned> ReservedColorListType;
21 //-----------------------------------------------------------------------------
24 // Implements a machine independant register class.
26 // This is the class that contains all data structures and common algos
27 // for coloring a particular register class (e.g., int class, fp class).
28 // This class is hardware independent. This class accepts a hardware
29 // dependent description of machine registers (MachineRegInfo class) to
30 // get hardware specific info and to color an individual IG node.
32 // This class contains the InterferenceGraph (IG).
33 // Also it contains an IGNode stack that can be used for coloring.
34 // The class provides some easy access methods to the IG methods, since these
35 // methods are called thru a register class.
37 //-----------------------------------------------------------------------------
39 const Method *const Meth; // Method we are working on
40 const MachineRegClassInfo *const MRC; // corresponding MRC
41 const unsigned RegClassID; // my int ID
43 InterferenceGraph IG; // Interference graph - constructed by
44 // buildInterferenceGraph
45 std::stack<IGNode *> IGNodeStack; // the stack used for coloring
47 const ReservedColorListType *const ReservedColorList;
49 // for passing registers that are pre-allocated and cannot be used by the
50 // register allocator for this method.
54 // An array used for coloring each node. This array must be of size
55 // MRC->getNumOfAllRegs(). Allocated once in the constructor
59 //--------------------------- private methods ------------------------------
61 void pushAllIGNodes();
63 bool pushUnconstrainedIGNodes();
65 IGNode * getIGNodeWithMinSpillCost();
67 void colorIGNode(IGNode *const Node);
72 RegClass(const Method *const M,
73 const MachineRegClassInfo *const MRC,
74 const ReservedColorListType *const RCL = NULL);
76 ~RegClass() { delete[] IsColorUsedArr; };
78 inline void createInterferenceGraph()
81 inline InterferenceGraph &getIG() { return IG; }
83 inline const unsigned getID() const { return RegClassID; }
85 // main method called for coloring regs
89 inline unsigned getNumOfAvailRegs() const
90 { return MRC->getNumOfAvailRegs(); }
93 // --- following methods are provided to access the IG contained within this
94 // ---- RegClass easilly.
96 inline void addLRToIG(LiveRange *const LR)
99 inline void setInterference(const LiveRange *const LR1,
100 const LiveRange *const LR2)
101 { IG.setInterference(LR1, LR2); }
103 inline unsigned getInterference(const LiveRange *const LR1,
104 const LiveRange *const LR2) const
105 { return IG.getInterference(LR1, LR2); }
107 inline void mergeIGNodesOfLRs(const LiveRange *const LR1,
108 LiveRange *const LR2)
109 { IG.mergeIGNodesOfLRs(LR1, LR2); }
112 inline bool * getIsColorUsedArr() { return IsColorUsedArr; }
115 inline void printIGNodeList() const {
116 std::cerr << "IG Nodes for Register Class " << RegClassID << ":" << "\n";
117 IG.printIGNodeList();
120 inline void printIG() {
121 std::cerr << "IG for Register Class " << RegClassID << ":" << "\n";