Tidy up. Simplify logic. No functional change intended.
[oota-llvm.git] / lib / Target / Mips / MipsISelDAGToDAG.cpp
1 //===-- MipsISelDAGToDAG.cpp - A dag to dag inst selector for Mips --------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file defines an instruction selector for the MIPS target.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #define DEBUG_TYPE "mips-isel"
15 #include "Mips.h"
16 #include "MipsMachineFunction.h"
17 #include "MipsRegisterInfo.h"
18 #include "MipsSubtarget.h"
19 #include "MipsTargetMachine.h"
20 #include "llvm/GlobalValue.h"
21 #include "llvm/Instructions.h"
22 #include "llvm/Intrinsics.h"
23 #include "llvm/Support/CFG.h"
24 #include "llvm/Type.h"
25 #include "llvm/CodeGen/MachineConstantPool.h"
26 #include "llvm/CodeGen/MachineFunction.h"
27 #include "llvm/CodeGen/MachineFrameInfo.h"
28 #include "llvm/CodeGen/MachineInstrBuilder.h"
29 #include "llvm/CodeGen/MachineRegisterInfo.h"
30 #include "llvm/CodeGen/SelectionDAGISel.h"
31 #include "llvm/Target/TargetMachine.h"
32 #include "llvm/Support/Debug.h"
33 #include "llvm/Support/ErrorHandling.h"
34 #include "llvm/Support/raw_ostream.h"
35 using namespace llvm;
36
37 //===----------------------------------------------------------------------===//
38 // Instruction Selector Implementation
39 //===----------------------------------------------------------------------===//
40
41 //===----------------------------------------------------------------------===//
42 // MipsDAGToDAGISel - MIPS specific code to select MIPS machine
43 // instructions for SelectionDAG operations.
44 //===----------------------------------------------------------------------===//
45 namespace {
46
47 class MipsDAGToDAGISel : public SelectionDAGISel {
48
49   /// TM - Keep a reference to MipsTargetMachine.
50   MipsTargetMachine &TM;
51
52   /// Subtarget - Keep a pointer to the MipsSubtarget around so that we can
53   /// make the right decision when generating code for different targets.
54   const MipsSubtarget &Subtarget;
55
56 public:
57   explicit MipsDAGToDAGISel(MipsTargetMachine &tm) :
58   SelectionDAGISel(tm),
59   TM(tm), Subtarget(tm.getSubtarget<MipsSubtarget>()) {}
60
61   // Pass Name
62   virtual const char *getPassName() const {
63     return "MIPS DAG->DAG Pattern Instruction Selection";
64   }
65
66
67 private:
68   // Include the pieces autogenerated from the target description.
69   #include "MipsGenDAGISel.inc"
70
71   /// getTargetMachine - Return a reference to the TargetMachine, casted
72   /// to the target-specific type.
73   const MipsTargetMachine &getTargetMachine() {
74     return static_cast<const MipsTargetMachine &>(TM);
75   }
76
77   /// getInstrInfo - Return a reference to the TargetInstrInfo, casted
78   /// to the target-specific type.
79   const MipsInstrInfo *getInstrInfo() {
80     return getTargetMachine().getInstrInfo();
81   }
82
83   SDNode *getGlobalBaseReg();
84   SDNode *Select(SDNode *N);
85
86   // Complex Pattern.
87   bool SelectAddr(SDValue N, SDValue &Base, SDValue &Offset);
88
89   // getImm - Return a target constant with the specified value.
90   inline SDValue getImm(const SDNode *Node, unsigned Imm) {
91     return CurDAG->getTargetConstant(Imm, Node->getValueType(0));
92   }
93
94   virtual bool SelectInlineAsmMemoryOperand(const SDValue &Op,
95                                             char ConstraintCode,
96                                             std::vector<SDValue> &OutOps);
97 };
98
99 }
100
101
102 /// getGlobalBaseReg - Output the instructions required to put the
103 /// GOT address into a register.
104 SDNode *MipsDAGToDAGISel::getGlobalBaseReg() {
105   unsigned GlobalBaseReg = getInstrInfo()->getGlobalBaseReg(MF);
106   return CurDAG->getRegister(GlobalBaseReg, TLI.getPointerTy()).getNode();
107 }
108
109 /// ComplexPattern used on MipsInstrInfo
110 /// Used on Mips Load/Store instructions
111 bool MipsDAGToDAGISel::
112 SelectAddr(SDValue Addr, SDValue &Base, SDValue &Offset) {
113   EVT ValTy = Addr.getValueType();
114   unsigned GPReg = ValTy == MVT::i32 ? Mips::GP : Mips::GP_64;
115
116   // if Address is FI, get the TargetFrameIndex.
117   if (FrameIndexSDNode *FIN = dyn_cast<FrameIndexSDNode>(Addr)) {
118     Base   = CurDAG->getTargetFrameIndex(FIN->getIndex(), ValTy);
119     Offset = CurDAG->getTargetConstant(0, ValTy);
120     return true;
121   }
122
123   // on PIC code Load GA
124   if (Addr.getOpcode() == MipsISD::Wrapper) {
125     Base   = CurDAG->getRegister(GPReg, ValTy);
126     Offset = Addr.getOperand(0);
127     return true;
128   }
129
130   if (TM.getRelocationModel() != Reloc::PIC_) {
131     if ((Addr.getOpcode() == ISD::TargetExternalSymbol ||
132         Addr.getOpcode() == ISD::TargetGlobalAddress))
133       return false;
134   }
135
136   // Addresses of the form FI+const or FI|const
137   if (CurDAG->isBaseWithConstantOffset(Addr)) {
138     ConstantSDNode *CN = dyn_cast<ConstantSDNode>(Addr.getOperand(1));
139     if (isInt<16>(CN->getSExtValue())) {
140
141       // If the first operand is a FI, get the TargetFI Node
142       if (FrameIndexSDNode *FIN = dyn_cast<FrameIndexSDNode>
143                                   (Addr.getOperand(0)))
144         Base = CurDAG->getTargetFrameIndex(FIN->getIndex(), ValTy);
145       else
146         Base = Addr.getOperand(0);
147
148       Offset = CurDAG->getTargetConstant(CN->getZExtValue(), ValTy);
149       return true;
150     }
151   }
152
153   // Operand is a result from an ADD.
154   if (Addr.getOpcode() == ISD::ADD) {
155     // When loading from constant pools, load the lower address part in
156     // the instruction itself. Example, instead of:
157     //  lui $2, %hi($CPI1_0)
158     //  addiu $2, $2, %lo($CPI1_0)
159     //  lwc1 $f0, 0($2)
160     // Generate:
161     //  lui $2, %hi($CPI1_0)
162     //  lwc1 $f0, %lo($CPI1_0)($2)
163     if (Addr.getOperand(1).getOpcode() == MipsISD::Lo) {
164       SDValue LoVal = Addr.getOperand(1);
165       if (isa<ConstantPoolSDNode>(LoVal.getOperand(0)) || 
166           isa<GlobalAddressSDNode>(LoVal.getOperand(0))) {
167         Base = Addr.getOperand(0);
168         Offset = LoVal.getOperand(0);
169         return true;
170       }
171     }
172   }
173
174   Base   = Addr;
175   Offset = CurDAG->getTargetConstant(0, ValTy);
176   return true;
177 }
178
179 /// Select instructions not customized! Used for
180 /// expanded, promoted and normal instructions
181 SDNode* MipsDAGToDAGISel::Select(SDNode *Node) {
182   unsigned Opcode = Node->getOpcode();
183   DebugLoc dl = Node->getDebugLoc();
184
185   // Dump information about the Node being selected
186   DEBUG(errs() << "Selecting: "; Node->dump(CurDAG); errs() << "\n");
187
188   // If we have a custom node, we already have selected!
189   if (Node->isMachineOpcode()) {
190     DEBUG(errs() << "== "; Node->dump(CurDAG); errs() << "\n");
191     return NULL;
192   }
193
194   ///
195   // Instruction Selection not handled by the auto-generated
196   // tablegen selection should be handled here.
197   ///
198   switch(Opcode) {
199     default: break;
200
201     case ISD::SUBE:
202     case ISD::ADDE: {
203       SDValue InFlag = Node->getOperand(2), CmpLHS;
204       unsigned Opc = InFlag.getOpcode(); (void)Opc;
205       assert(((Opc == ISD::ADDC || Opc == ISD::ADDE) ||
206               (Opc == ISD::SUBC || Opc == ISD::SUBE)) &&
207              "(ADD|SUB)E flag operand must come from (ADD|SUB)C/E insn");
208
209       unsigned MOp;
210       if (Opcode == ISD::ADDE) {
211         CmpLHS = InFlag.getValue(0);
212         MOp = Mips::ADDu;
213       } else {
214         CmpLHS = InFlag.getOperand(0);
215         MOp = Mips::SUBu;
216       }
217
218       SDValue Ops[] = { CmpLHS, InFlag.getOperand(1) };
219
220       SDValue LHS = Node->getOperand(0);
221       SDValue RHS = Node->getOperand(1);
222
223       EVT VT = LHS.getValueType();
224       SDNode *Carry = CurDAG->getMachineNode(Mips::SLTu, dl, VT, Ops, 2);
225       SDNode *AddCarry = CurDAG->getMachineNode(Mips::ADDu, dl, VT,
226                                                 SDValue(Carry,0), RHS);
227
228       return CurDAG->SelectNodeTo(Node, MOp, VT, MVT::Glue,
229                                   LHS, SDValue(AddCarry,0));
230     }
231
232     /// Mul with two results
233     case ISD::SMUL_LOHI:
234     case ISD::UMUL_LOHI: {
235       assert(Node->getValueType(0) != MVT::i64 &&
236              "64-bit multiplication with two results not handled.");
237       SDValue Op1 = Node->getOperand(0);
238       SDValue Op2 = Node->getOperand(1);
239
240       unsigned Op;
241       Op = (Opcode == ISD::UMUL_LOHI ? Mips::MULTu : Mips::MULT);
242
243       SDNode *Mul = CurDAG->getMachineNode(Op, dl, MVT::Glue, Op1, Op2);
244
245       SDValue InFlag = SDValue(Mul, 0);
246       SDNode *Lo = CurDAG->getMachineNode(Mips::MFLO, dl, MVT::i32,
247                                           MVT::Glue, InFlag);
248       InFlag = SDValue(Lo,1);
249       SDNode *Hi = CurDAG->getMachineNode(Mips::MFHI, dl, MVT::i32, InFlag);
250
251       if (!SDValue(Node, 0).use_empty())
252         ReplaceUses(SDValue(Node, 0), SDValue(Lo,0));
253
254       if (!SDValue(Node, 1).use_empty())
255         ReplaceUses(SDValue(Node, 1), SDValue(Hi,0));
256
257       return NULL;
258     }
259
260     /// Special Muls
261     case ISD::MUL:
262       // Mips32 has a 32-bit three operand mul instruction.
263       if (Subtarget.hasMips32() && Node->getValueType(0) == MVT::i32)
264         break;
265     case ISD::MULHS:
266     case ISD::MULHU: {
267       assert((Opcode == ISD::MUL || Node->getValueType(0) != MVT::i64) &&
268              "64-bit MULH* not handled.");
269       EVT Ty = Node->getValueType(0);
270       SDValue MulOp1 = Node->getOperand(0);
271       SDValue MulOp2 = Node->getOperand(1);
272
273       unsigned MulOp  = (Opcode == ISD::MULHU ?
274                          Mips::MULTu :
275                          (Ty == MVT::i32 ? Mips::MULT : Mips::DMULT));
276       SDNode *MulNode = CurDAG->getMachineNode(MulOp, dl,
277                                                MVT::Glue, MulOp1, MulOp2);
278
279       SDValue InFlag = SDValue(MulNode, 0);
280
281       if (Opcode == ISD::MUL) {
282         unsigned Opc = (Ty == MVT::i32 ? Mips::MFLO : Mips::MFLO64);
283         return CurDAG->getMachineNode(Opc, dl, Ty, InFlag);
284       }
285       else
286         return CurDAG->getMachineNode(Mips::MFHI, dl, MVT::i32, InFlag);
287     }
288
289     // Get target GOT address.
290     case ISD::GLOBAL_OFFSET_TABLE:
291       return getGlobalBaseReg();
292
293     case ISD::ConstantFP: {
294       ConstantFPSDNode *CN = dyn_cast<ConstantFPSDNode>(Node);
295       if (Node->getValueType(0) == MVT::f64 && CN->isExactlyValue(+0.0)) {
296         SDValue Zero = CurDAG->getCopyFromReg(CurDAG->getEntryNode(), dl,
297                                               Mips::ZERO, MVT::i32);
298         return CurDAG->getMachineNode(Mips::BuildPairF64, dl, MVT::f64, Zero,
299                                       Zero);
300       }
301       break;
302     }
303
304     case MipsISD::ThreadPointer: {
305       EVT PtrVT = TLI.getPointerTy();
306       unsigned RdhwrOpc, SrcReg, DestReg;
307
308       if (PtrVT == MVT::i32) {
309         RdhwrOpc = Mips::RDHWR;
310         SrcReg = Mips::HWR29;
311         DestReg = Mips::V1;
312       } else {
313         RdhwrOpc = Mips::RDHWR64;
314         SrcReg = Mips::HWR29_64;
315         DestReg = Mips::V1_64;
316       }
317       
318       SDNode *Rdhwr =
319         CurDAG->getMachineNode(RdhwrOpc, Node->getDebugLoc(),
320                                Node->getValueType(0),
321                                CurDAG->getRegister(SrcReg, PtrVT));
322       SDValue Chain = CurDAG->getCopyToReg(CurDAG->getEntryNode(), dl, DestReg,
323                                            SDValue(Rdhwr, 0));
324       SDValue ResNode = CurDAG->getCopyFromReg(Chain, dl, DestReg, PtrVT);
325       ReplaceUses(SDValue(Node, 0), ResNode);
326       return ResNode.getNode();
327     }
328   }
329
330   // Select the default instruction
331   SDNode *ResNode = SelectCode(Node);
332
333   DEBUG(errs() << "=> ");
334   if (ResNode == NULL || ResNode == Node)
335     DEBUG(Node->dump(CurDAG));
336   else
337     DEBUG(ResNode->dump(CurDAG));
338   DEBUG(errs() << "\n");
339   return ResNode;
340 }
341
342 bool MipsDAGToDAGISel::
343 SelectInlineAsmMemoryOperand(const SDValue &Op, char ConstraintCode,
344                              std::vector<SDValue> &OutOps) {
345   assert(ConstraintCode == 'm' && "unexpected asm memory constraint");
346   OutOps.push_back(Op);
347   return false;
348 }
349
350 /// createMipsISelDag - This pass converts a legalized DAG into a
351 /// MIPS-specific DAG, ready for instruction scheduling.
352 FunctionPass *llvm::createMipsISelDag(MipsTargetMachine &TM) {
353   return new MipsDAGToDAGISel(TM);
354 }