868d3335a1aa72e5509c0574b7c8ec06c5246a49
[oota-llvm.git] / lib / CodeGen / SelectionDAG / ScheduleDAG.cpp
1 //===---- ScheduleDAG.cpp - Implement the ScheduleDAG class ---------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file was developed by James M. Laskey and is distributed under the
6 // University of Illinois Open Source License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This implements a simple two pass scheduler.  The first pass attempts to push
11 // backward any lengthy instructions and critical paths.  The second pass packs
12 // instructions into semi-optimal time slots.
13 //
14 //===----------------------------------------------------------------------===//
15
16 #define DEBUG_TYPE "sched"
17 #include "llvm/Type.h"
18 #include "llvm/CodeGen/ScheduleDAG.h"
19 #include "llvm/CodeGen/MachineConstantPool.h"
20 #include "llvm/CodeGen/MachineFunction.h"
21 #include "llvm/CodeGen/SSARegMap.h"
22 #include "llvm/Target/TargetData.h"
23 #include "llvm/Target/TargetMachine.h"
24 #include "llvm/Target/TargetInstrInfo.h"
25 #include "llvm/Target/TargetLowering.h"
26 #include "llvm/Support/Debug.h"
27 #include "llvm/Support/MathExtras.h"
28 using namespace llvm;
29
30 /// BuildSchedUnits - Build SUnits from the selection dag that we are input.
31 /// This SUnit graph is similar to the SelectionDAG, but represents flagged
32 /// together nodes with a single SUnit.
33 void ScheduleDAG::BuildSchedUnits() {
34   // Reserve entries in the vector for each of the SUnits we are creating.  This
35   // ensure that reallocation of the vector won't happen, so SUnit*'s won't get
36   // invalidated.
37   SUnits.reserve(std::distance(DAG.allnodes_begin(), DAG.allnodes_end()));
38   
39   const InstrItineraryData &InstrItins = TM.getInstrItineraryData();
40   
41   for (SelectionDAG::allnodes_iterator NI = DAG.allnodes_begin(),
42        E = DAG.allnodes_end(); NI != E; ++NI) {
43     if (isPassiveNode(NI))  // Leaf node, e.g. a TargetImmediate.
44       continue;
45     
46     // If this node has already been processed, stop now.
47     if (SUnitMap[NI]) continue;
48     
49     SUnit *NodeSUnit = NewSUnit(NI);
50     
51     // See if anything is flagged to this node, if so, add them to flagged
52     // nodes.  Nodes can have at most one flag input and one flag output.  Flags
53     // are required the be the last operand and result of a node.
54     
55     // Scan up, adding flagged preds to FlaggedNodes.
56     SDNode *N = NI;
57     if (N->getNumOperands() &&
58         N->getOperand(N->getNumOperands()-1).getValueType() == MVT::Flag) {
59       do {
60         N = N->getOperand(N->getNumOperands()-1).Val;
61         NodeSUnit->FlaggedNodes.push_back(N);
62         SUnitMap[N] = NodeSUnit;
63       } while (N->getNumOperands() &&
64                N->getOperand(N->getNumOperands()-1).getValueType()== MVT::Flag);
65       std::reverse(NodeSUnit->FlaggedNodes.begin(),
66                    NodeSUnit->FlaggedNodes.end());
67     }
68     
69     // Scan down, adding this node and any flagged succs to FlaggedNodes if they
70     // have a user of the flag operand.
71     N = NI;
72     while (N->getValueType(N->getNumValues()-1) == MVT::Flag) {
73       SDOperand FlagVal(N, N->getNumValues()-1);
74       
75       // There are either zero or one users of the Flag result.
76       bool HasFlagUse = false;
77       for (SDNode::use_iterator UI = N->use_begin(), E = N->use_end(); 
78            UI != E; ++UI)
79         if (FlagVal.isOperand(*UI)) {
80           HasFlagUse = true;
81           NodeSUnit->FlaggedNodes.push_back(N);
82           SUnitMap[N] = NodeSUnit;
83           N = *UI;
84           break;
85         }
86       if (!HasFlagUse) break;
87     }
88     
89     // Now all flagged nodes are in FlaggedNodes and N is the bottom-most node.
90     // Update the SUnit
91     NodeSUnit->Node = N;
92     SUnitMap[N] = NodeSUnit;
93     
94     // Compute the latency for the node.  We use the sum of the latencies for
95     // all nodes flagged together into this SUnit.
96     if (InstrItins.isEmpty()) {
97       // No latency information.
98       NodeSUnit->Latency = 1;
99     } else {
100       NodeSUnit->Latency = 0;
101       if (N->isTargetOpcode()) {
102         unsigned SchedClass = TII->getSchedClass(N->getTargetOpcode());
103         InstrStage *S = InstrItins.begin(SchedClass);
104         InstrStage *E = InstrItins.end(SchedClass);
105         for (; S != E; ++S)
106           NodeSUnit->Latency += S->Cycles;
107       }
108       for (unsigned i = 0, e = NodeSUnit->FlaggedNodes.size(); i != e; ++i) {
109         SDNode *FNode = NodeSUnit->FlaggedNodes[i];
110         if (FNode->isTargetOpcode()) {
111           unsigned SchedClass = TII->getSchedClass(FNode->getTargetOpcode());
112           InstrStage *S = InstrItins.begin(SchedClass);
113           InstrStage *E = InstrItins.end(SchedClass);
114           for (; S != E; ++S)
115             NodeSUnit->Latency += S->Cycles;
116         }
117       }
118     }
119   }
120   
121   // Pass 2: add the preds, succs, etc.
122   for (unsigned su = 0, e = SUnits.size(); su != e; ++su) {
123     SUnit *SU = &SUnits[su];
124     SDNode *MainNode = SU->Node;
125     
126     if (MainNode->isTargetOpcode()) {
127       unsigned Opc = MainNode->getTargetOpcode();
128       for (unsigned i = 0, ee = TII->getNumOperands(Opc); i != ee; ++i) {
129         if (TII->getOperandConstraint(Opc, i, TOI::TIED_TO) != -1) {
130           SU->isTwoAddress = true;
131           break;
132         }
133       }
134       if (TII->isCommutableInstr(Opc))
135         SU->isCommutable = true;
136     }
137     
138     // Find all predecessors and successors of the group.
139     // Temporarily add N to make code simpler.
140     SU->FlaggedNodes.push_back(MainNode);
141     
142     for (unsigned n = 0, e = SU->FlaggedNodes.size(); n != e; ++n) {
143       SDNode *N = SU->FlaggedNodes[n];
144       
145       for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) {
146         SDNode *OpN = N->getOperand(i).Val;
147         if (isPassiveNode(OpN)) continue;   // Not scheduled.
148         SUnit *OpSU = SUnitMap[OpN];
149         assert(OpSU && "Node has no SUnit!");
150         if (OpSU == SU) continue;           // In the same group.
151
152         MVT::ValueType OpVT = N->getOperand(i).getValueType();
153         assert(OpVT != MVT::Flag && "Flagged nodes should be in same sunit!");
154         bool isChain = OpVT == MVT::Other;
155         
156         if (SU->addPred(OpSU, isChain)) {
157           if (!isChain) {
158             SU->NumPreds++;
159             SU->NumPredsLeft++;
160           } else {
161             SU->NumChainPredsLeft++;
162           }
163         }
164         if (OpSU->addSucc(SU, isChain)) {
165           if (!isChain) {
166             OpSU->NumSuccs++;
167             OpSU->NumSuccsLeft++;
168           } else {
169             OpSU->NumChainSuccsLeft++;
170           }
171         }
172       }
173     }
174     
175     // Remove MainNode from FlaggedNodes again.
176     SU->FlaggedNodes.pop_back();
177   }
178   
179   return;
180 }
181
182 static void CalculateDepths(SUnit &SU, unsigned Depth) {
183   if (SU.Depth == 0 || Depth > SU.Depth) {
184     SU.Depth = Depth;
185     for (SUnit::succ_iterator I = SU.Succs.begin(), E = SU.Succs.end();
186          I != E; ++I)
187       CalculateDepths(*I->first, Depth+1);
188   }
189 }
190
191 void ScheduleDAG::CalculateDepths() {
192   SUnit *Entry = SUnitMap[DAG.getEntryNode().Val];
193   ::CalculateDepths(*Entry, 0U);
194   for (unsigned i = 0, e = SUnits.size(); i != e; ++i)
195     if (SUnits[i].Preds.size() == 0 && &SUnits[i] != Entry) {
196       ::CalculateDepths(SUnits[i], 0U);
197     }
198 }
199
200 static void CalculateHeights(SUnit &SU, unsigned Height) {
201   if (SU.Height == 0 || Height > SU.Height) {
202     SU.Height = Height;
203     for (SUnit::pred_iterator I = SU.Preds.begin(), E = SU.Preds.end();
204          I != E; ++I)
205       CalculateHeights(*I->first, Height+1);
206   }
207 }
208 void ScheduleDAG::CalculateHeights() {
209   SUnit *Root = SUnitMap[DAG.getRoot().Val];
210   ::CalculateHeights(*Root, 0U);
211 }
212
213 /// CountResults - The results of target nodes have register or immediate
214 /// operands first, then an optional chain, and optional flag operands (which do
215 /// not go into the machine instrs.)
216 unsigned ScheduleDAG::CountResults(SDNode *Node) {
217   unsigned N = Node->getNumValues();
218   while (N && Node->getValueType(N - 1) == MVT::Flag)
219     --N;
220   if (N && Node->getValueType(N - 1) == MVT::Other)
221     --N;    // Skip over chain result.
222   return N;
223 }
224
225 /// CountOperands  The inputs to target nodes have any actual inputs first,
226 /// followed by an optional chain operand, then flag operands.  Compute the
227 /// number of actual operands that  will go into the machine instr.
228 unsigned ScheduleDAG::CountOperands(SDNode *Node) {
229   unsigned N = Node->getNumOperands();
230   while (N && Node->getOperand(N - 1).getValueType() == MVT::Flag)
231     --N;
232   if (N && Node->getOperand(N - 1).getValueType() == MVT::Other)
233     --N; // Ignore chain if it exists.
234   return N;
235 }
236
237 static const TargetRegisterClass *getInstrOperandRegClass(
238         const MRegisterInfo *MRI, 
239         const TargetInstrInfo *TII,
240         const TargetInstrDescriptor *II,
241         unsigned Op) {
242   if (Op >= II->numOperands) {
243     assert((II->Flags & M_VARIABLE_OPS)&& "Invalid operand # of instruction");
244     return NULL;
245   }
246   const TargetOperandInfo &toi = II->OpInfo[Op];
247   return (toi.Flags & M_LOOK_UP_PTR_REG_CLASS)
248          ? TII->getPointerRegClass() : MRI->getRegClass(toi.RegClass);
249 }
250
251 static unsigned CreateVirtualRegisters(const MRegisterInfo *MRI,
252                                        MachineInstr *MI,
253                                        unsigned NumResults,
254                                        SSARegMap *RegMap,
255                                        const TargetInstrInfo *TII,
256                                        const TargetInstrDescriptor &II) {
257   // Create the result registers for this node and add the result regs to
258   // the machine instruction.
259   unsigned ResultReg =
260     RegMap->createVirtualRegister(getInstrOperandRegClass(MRI, TII, &II, 0));
261   MI->addRegOperand(ResultReg, true);
262   for (unsigned i = 1; i != NumResults; ++i) {
263     const TargetRegisterClass *RC = getInstrOperandRegClass(MRI, TII, &II, i);
264     assert(RC && "Isn't a register operand!");
265     MI->addRegOperand(RegMap->createVirtualRegister(RC), true);
266   }
267   return ResultReg;
268 }
269
270 /// getVR - Return the virtual register corresponding to the specified result
271 /// of the specified node.
272 static unsigned getVR(SDOperand Op, DenseMap<SDNode*, unsigned> &VRBaseMap) {
273   DenseMap<SDNode*, unsigned>::iterator I = VRBaseMap.find(Op.Val);
274   assert(I != VRBaseMap.end() && "Node emitted out of order - late");
275   return I->second + Op.ResNo;
276 }
277
278
279 /// AddOperand - Add the specified operand to the specified machine instr.  II
280 /// specifies the instruction information for the node, and IIOpNum is the
281 /// operand number (in the II) that we are adding. IIOpNum and II are used for 
282 /// assertions only.
283 void ScheduleDAG::AddOperand(MachineInstr *MI, SDOperand Op,
284                              unsigned IIOpNum,
285                              const TargetInstrDescriptor *II,
286                              DenseMap<SDNode*, unsigned> &VRBaseMap) {
287   if (Op.isTargetOpcode()) {
288     // Note that this case is redundant with the final else block, but we
289     // include it because it is the most common and it makes the logic
290     // simpler here.
291     assert(Op.getValueType() != MVT::Other &&
292            Op.getValueType() != MVT::Flag &&
293            "Chain and flag operands should occur at end of operand list!");
294     
295     // Get/emit the operand.
296     unsigned VReg = getVR(Op, VRBaseMap);
297     MI->addRegOperand(VReg, false);
298     
299     // Verify that it is right.
300     assert(MRegisterInfo::isVirtualRegister(VReg) && "Not a vreg?");
301     if (II) {
302       const TargetRegisterClass *RC =
303                           getInstrOperandRegClass(MRI, TII, II, IIOpNum);
304       assert(RC && "Don't have operand info for this instruction!");
305       const TargetRegisterClass *VRC = RegMap->getRegClass(VReg);
306       if (VRC != RC) {
307         cerr << "Register class of operand and regclass of use don't agree!\n";
308 #ifndef NDEBUG
309         cerr << "Operand = " << IIOpNum << "\n";
310         cerr << "Op->Val = "; Op.Val->dump(0); cerr << "\n";
311         cerr << "MI = "; MI->print(cerr);
312         cerr << "VReg = " << VReg << "\n";
313         cerr << "VReg RegClass     size = " << VRC->getSize()
314           << ", align = " << VRC->getAlignment() << "\n";
315         cerr << "Expected RegClass size = " << RC->getSize()
316           << ", align = " << RC->getAlignment() << "\n";
317 #endif
318         cerr << "Fatal error, aborting.\n";
319         abort();
320       }
321     }
322   } else if (ConstantSDNode *C =
323              dyn_cast<ConstantSDNode>(Op)) {
324     MI->addImmOperand(C->getValue());
325   } else if (RegisterSDNode *R =
326              dyn_cast<RegisterSDNode>(Op)) {
327     MI->addRegOperand(R->getReg(), false);
328   } else if (GlobalAddressSDNode *TGA =
329              dyn_cast<GlobalAddressSDNode>(Op)) {
330     MI->addGlobalAddressOperand(TGA->getGlobal(), TGA->getOffset());
331   } else if (BasicBlockSDNode *BB =
332              dyn_cast<BasicBlockSDNode>(Op)) {
333     MI->addMachineBasicBlockOperand(BB->getBasicBlock());
334   } else if (FrameIndexSDNode *FI =
335              dyn_cast<FrameIndexSDNode>(Op)) {
336     MI->addFrameIndexOperand(FI->getIndex());
337   } else if (JumpTableSDNode *JT =
338              dyn_cast<JumpTableSDNode>(Op)) {
339     MI->addJumpTableIndexOperand(JT->getIndex());
340   } else if (ConstantPoolSDNode *CP = 
341              dyn_cast<ConstantPoolSDNode>(Op)) {
342     int Offset = CP->getOffset();
343     unsigned Align = CP->getAlignment();
344     const Type *Type = CP->getType();
345     // MachineConstantPool wants an explicit alignment.
346     if (Align == 0) {
347       Align = TM.getTargetData()->getPreferredTypeAlignmentShift(Type);
348       if (Align == 0) {
349         // Alignment of vector types.  FIXME!
350         Align = TM.getTargetData()->getTypeSize(Type);
351         Align = Log2_64(Align);
352       }
353     }
354     
355     unsigned Idx;
356     if (CP->isMachineConstantPoolEntry())
357       Idx = ConstPool->getConstantPoolIndex(CP->getMachineCPVal(), Align);
358     else
359       Idx = ConstPool->getConstantPoolIndex(CP->getConstVal(), Align);
360     MI->addConstantPoolIndexOperand(Idx, Offset);
361   } else if (ExternalSymbolSDNode *ES = 
362              dyn_cast<ExternalSymbolSDNode>(Op)) {
363     MI->addExternalSymbolOperand(ES->getSymbol());
364   } else {
365     assert(Op.getValueType() != MVT::Other &&
366            Op.getValueType() != MVT::Flag &&
367            "Chain and flag operands should occur at end of operand list!");
368     unsigned VReg = getVR(Op, VRBaseMap);
369     MI->addRegOperand(VReg, false);
370     
371     // Verify that it is right.
372     assert(MRegisterInfo::isVirtualRegister(VReg) && "Not a vreg?");
373     if (II) {
374       const TargetRegisterClass *RC =
375                             getInstrOperandRegClass(MRI, TII, II, IIOpNum);
376       assert(RC && "Don't have operand info for this instruction!");
377       assert(RegMap->getRegClass(VReg) == RC &&
378              "Register class of operand and regclass of use don't agree!");
379     }
380   }
381   
382 }
383
384
385 /// EmitNode - Generate machine code for an node and needed dependencies.
386 ///
387 void ScheduleDAG::EmitNode(SDNode *Node, 
388                            DenseMap<SDNode*, unsigned> &VRBaseMap) {
389   unsigned VRBase = 0;                 // First virtual register for node
390   
391   // If machine instruction
392   if (Node->isTargetOpcode()) {
393     unsigned Opc = Node->getTargetOpcode();
394     const TargetInstrDescriptor &II = TII->get(Opc);
395
396     unsigned NumResults = CountResults(Node);
397     unsigned NodeOperands = CountOperands(Node);
398     unsigned NumMIOperands = NodeOperands + NumResults;
399 #ifndef NDEBUG
400     assert((unsigned(II.numOperands) == NumMIOperands ||
401             (II.Flags & M_VARIABLE_OPS)) &&
402            "#operands for dag node doesn't match .td file!"); 
403 #endif
404
405     // Create the new machine instruction.
406     MachineInstr *MI = new MachineInstr(II);
407     
408     // Add result register values for things that are defined by this
409     // instruction.
410     
411     // If the node is only used by a CopyToReg and the dest reg is a vreg, use
412     // the CopyToReg'd destination register instead of creating a new vreg.
413     if (NumResults == 1) {
414       for (SDNode::use_iterator UI = Node->use_begin(), E = Node->use_end();
415            UI != E; ++UI) {
416         SDNode *Use = *UI;
417         if (Use->getOpcode() == ISD::CopyToReg && 
418             Use->getOperand(2).Val == Node) {
419           unsigned Reg = cast<RegisterSDNode>(Use->getOperand(1))->getReg();
420           if (MRegisterInfo::isVirtualRegister(Reg)) {
421             VRBase = Reg;
422             MI->addRegOperand(Reg, true);
423             break;
424           }
425         }
426       }
427     }
428     
429     // Otherwise, create new virtual registers.
430     if (NumResults && VRBase == 0)
431       VRBase = CreateVirtualRegisters(MRI, MI, NumResults, RegMap, TII, II);
432     
433     // Emit all of the actual operands of this instruction, adding them to the
434     // instruction as appropriate.
435     for (unsigned i = 0; i != NodeOperands; ++i)
436       AddOperand(MI, Node->getOperand(i), i+NumResults, &II, VRBaseMap);
437
438     // Commute node if it has been determined to be profitable.
439     if (CommuteSet.count(Node)) {
440       MachineInstr *NewMI = TII->commuteInstruction(MI);
441       if (NewMI == 0)
442         DOUT << "Sched: COMMUTING FAILED!\n";
443       else {
444         DOUT << "Sched: COMMUTED TO: " << *NewMI;
445         if (MI != NewMI) {
446           delete MI;
447           MI = NewMI;
448         }
449       }
450     }
451
452     // Now that we have emitted all operands, emit this instruction itself.
453     if ((II.Flags & M_USES_CUSTOM_DAG_SCHED_INSERTION) == 0) {
454       BB->insert(BB->end(), MI);
455     } else {
456       // Insert this instruction into the end of the basic block, potentially
457       // taking some custom action.
458       BB = DAG.getTargetLoweringInfo().InsertAtEndOfBasicBlock(MI, BB);
459     }
460   } else {
461     switch (Node->getOpcode()) {
462     default:
463 #ifndef NDEBUG
464       Node->dump();
465 #endif
466       assert(0 && "This target-independent node should have been selected!");
467     case ISD::EntryToken: // fall thru
468     case ISD::TokenFactor:
469     case ISD::LABEL:
470       break;
471     case ISD::CopyToReg: {
472       unsigned InReg;
473       if (RegisterSDNode *R = dyn_cast<RegisterSDNode>(Node->getOperand(2)))
474         InReg = R->getReg();
475       else
476         InReg = getVR(Node->getOperand(2), VRBaseMap);
477       unsigned DestReg = cast<RegisterSDNode>(Node->getOperand(1))->getReg();
478       if (InReg != DestReg)   // Coalesced away the copy?
479         MRI->copyRegToReg(*BB, BB->end(), DestReg, InReg,
480                           RegMap->getRegClass(InReg));
481       break;
482     }
483     case ISD::CopyFromReg: {
484       unsigned SrcReg = cast<RegisterSDNode>(Node->getOperand(1))->getReg();
485       if (MRegisterInfo::isVirtualRegister(SrcReg)) {
486         VRBase = SrcReg;  // Just use the input register directly!
487         break;
488       }
489
490       // If the node is only used by a CopyToReg and the dest reg is a vreg, use
491       // the CopyToReg'd destination register instead of creating a new vreg.
492       for (SDNode::use_iterator UI = Node->use_begin(), E = Node->use_end();
493            UI != E; ++UI) {
494         SDNode *Use = *UI;
495         if (Use->getOpcode() == ISD::CopyToReg && 
496             Use->getOperand(2).Val == Node) {
497           unsigned DestReg = cast<RegisterSDNode>(Use->getOperand(1))->getReg();
498           if (MRegisterInfo::isVirtualRegister(DestReg)) {
499             VRBase = DestReg;
500             break;
501           }
502         }
503       }
504
505       // Figure out the register class to create for the destreg.
506       const TargetRegisterClass *TRC = 0;
507       if (VRBase) {
508         TRC = RegMap->getRegClass(VRBase);
509       } else {
510
511         // Pick the register class of the right type that contains this physreg.
512         for (MRegisterInfo::regclass_iterator I = MRI->regclass_begin(),
513              E = MRI->regclass_end(); I != E; ++I)
514           if ((*I)->hasType(Node->getValueType(0)) &&
515               (*I)->contains(SrcReg)) {
516             TRC = *I;
517             break;
518           }
519         assert(TRC && "Couldn't find register class for reg copy!");
520       
521         // Create the reg, emit the copy.
522         VRBase = RegMap->createVirtualRegister(TRC);
523       }
524       MRI->copyRegToReg(*BB, BB->end(), VRBase, SrcReg, TRC);
525       break;
526     }
527     case ISD::INLINEASM: {
528       unsigned NumOps = Node->getNumOperands();
529       if (Node->getOperand(NumOps-1).getValueType() == MVT::Flag)
530         --NumOps;  // Ignore the flag operand.
531       
532       // Create the inline asm machine instruction.
533       MachineInstr *MI =
534         new MachineInstr(BB, TII->get(TargetInstrInfo::INLINEASM));
535
536       // Add the asm string as an external symbol operand.
537       const char *AsmStr =
538         cast<ExternalSymbolSDNode>(Node->getOperand(1))->getSymbol();
539       MI->addExternalSymbolOperand(AsmStr);
540       
541       // Add all of the operand registers to the instruction.
542       for (unsigned i = 2; i != NumOps;) {
543         unsigned Flags = cast<ConstantSDNode>(Node->getOperand(i))->getValue();
544         unsigned NumVals = Flags >> 3;
545         
546         MI->addImmOperand(Flags);
547         ++i;  // Skip the ID value.
548         
549         switch (Flags & 7) {
550         default: assert(0 && "Bad flags!");
551         case 1:  // Use of register.
552           for (; NumVals; --NumVals, ++i) {
553             unsigned Reg = cast<RegisterSDNode>(Node->getOperand(i))->getReg();
554             MI->addRegOperand(Reg, false);
555           }
556           break;
557         case 2:   // Def of register.
558           for (; NumVals; --NumVals, ++i) {
559             unsigned Reg = cast<RegisterSDNode>(Node->getOperand(i))->getReg();
560             MI->addRegOperand(Reg, true);
561           }
562           break;
563         case 3: { // Immediate.
564           assert(NumVals == 1 && "Unknown immediate value!");
565           if (ConstantSDNode *CS=dyn_cast<ConstantSDNode>(Node->getOperand(i))){
566             MI->addImmOperand(CS->getValue());
567           } else {
568             GlobalAddressSDNode *GA = 
569               cast<GlobalAddressSDNode>(Node->getOperand(i));
570             MI->addGlobalAddressOperand(GA->getGlobal(), GA->getOffset());
571           }
572           ++i;
573           break;
574         }
575         case 4:  // Addressing mode.
576           // The addressing mode has been selected, just add all of the
577           // operands to the machine instruction.
578           for (; NumVals; --NumVals, ++i)
579             AddOperand(MI, Node->getOperand(i), 0, 0, VRBaseMap);
580           break;
581         }
582       }
583       break;
584     }
585     }
586   }
587
588   assert(!VRBaseMap.count(Node) && "Node emitted out of order - early");
589   VRBaseMap[Node] = VRBase;
590 }
591
592 void ScheduleDAG::EmitNoop() {
593   TII->insertNoop(*BB, BB->end());
594 }
595
596 /// EmitSchedule - Emit the machine code in scheduled order.
597 void ScheduleDAG::EmitSchedule() {
598   // If this is the first basic block in the function, and if it has live ins
599   // that need to be copied into vregs, emit the copies into the top of the
600   // block before emitting the code for the block.
601   MachineFunction &MF = DAG.getMachineFunction();
602   if (&MF.front() == BB && MF.livein_begin() != MF.livein_end()) {
603     for (MachineFunction::livein_iterator LI = MF.livein_begin(),
604          E = MF.livein_end(); LI != E; ++LI)
605       if (LI->second)
606         MRI->copyRegToReg(*MF.begin(), MF.begin()->end(), LI->second,
607                           LI->first, RegMap->getRegClass(LI->second));
608   }
609   
610   
611   // Finally, emit the code for all of the scheduled instructions.
612   DenseMap<SDNode*, unsigned> VRBaseMap;
613   for (unsigned i = 0, e = Sequence.size(); i != e; i++) {
614     if (SUnit *SU = Sequence[i]) {
615       for (unsigned j = 0, ee = SU->FlaggedNodes.size(); j != ee; j++)
616         EmitNode(SU->FlaggedNodes[j], VRBaseMap);
617       EmitNode(SU->Node, VRBaseMap);
618     } else {
619       // Null SUnit* is a noop.
620       EmitNoop();
621     }
622   }
623 }
624
625 /// dump - dump the schedule.
626 void ScheduleDAG::dumpSchedule() const {
627   for (unsigned i = 0, e = Sequence.size(); i != e; i++) {
628     if (SUnit *SU = Sequence[i])
629       SU->dump(&DAG);
630     else
631       cerr << "**** NOOP ****\n";
632   }
633 }
634
635
636 /// Run - perform scheduling.
637 ///
638 MachineBasicBlock *ScheduleDAG::Run() {
639   TII = TM.getInstrInfo();
640   MRI = TM.getRegisterInfo();
641   RegMap = BB->getParent()->getSSARegMap();
642   ConstPool = BB->getParent()->getConstantPool();
643
644   Schedule();
645   return BB;
646 }
647
648 /// SUnit - Scheduling unit. It's an wrapper around either a single SDNode or
649 /// a group of nodes flagged together.
650 void SUnit::dump(const SelectionDAG *G) const {
651   cerr << "SU(" << NodeNum << "): ";
652   Node->dump(G);
653   cerr << "\n";
654   if (FlaggedNodes.size() != 0) {
655     for (unsigned i = 0, e = FlaggedNodes.size(); i != e; i++) {
656       cerr << "    ";
657       FlaggedNodes[i]->dump(G);
658       cerr << "\n";
659     }
660   }
661 }
662
663 void SUnit::dumpAll(const SelectionDAG *G) const {
664   dump(G);
665
666   cerr << "  # preds left       : " << NumPredsLeft << "\n";
667   cerr << "  # succs left       : " << NumSuccsLeft << "\n";
668   cerr << "  # chain preds left : " << NumChainPredsLeft << "\n";
669   cerr << "  # chain succs left : " << NumChainSuccsLeft << "\n";
670   cerr << "  Latency            : " << Latency << "\n";
671   cerr << "  Depth              : " << Depth << "\n";
672   cerr << "  Height             : " << Height << "\n";
673
674   if (Preds.size() != 0) {
675     cerr << "  Predecessors:\n";
676     for (SUnit::const_succ_iterator I = Preds.begin(), E = Preds.end();
677          I != E; ++I) {
678       if (I->second)
679         cerr << "   ch  #";
680       else
681         cerr << "   val #";
682       cerr << I->first << " - SU(" << I->first->NodeNum << ")\n";
683     }
684   }
685   if (Succs.size() != 0) {
686     cerr << "  Successors:\n";
687     for (SUnit::const_succ_iterator I = Succs.begin(), E = Succs.end();
688          I != E; ++I) {
689       if (I->second)
690         cerr << "   ch  #";
691       else
692         cerr << "   val #";
693       cerr << I->first << " - SU(" << I->first->NodeNum << ")\n";
694     }
695   }
696   cerr << "\n";
697 }