Add support for Neon VEXT (vector extract) shuffles.
[oota-llvm.git] / lib / Target / ARM / ARMISelLowering.h
1 //===-- ARMISelLowering.h - ARM DAG Lowering Interface ----------*- C++ -*-===//
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 the interfaces that ARM uses to lower LLVM code into a
11 // selection DAG.
12 //
13 //===----------------------------------------------------------------------===//
14
15 #ifndef ARMISELLOWERING_H
16 #define ARMISELLOWERING_H
17
18 #include "ARMSubtarget.h"
19 #include "llvm/Target/TargetLowering.h"
20 #include "llvm/CodeGen/SelectionDAG.h"
21 #include "llvm/CodeGen/CallingConvLower.h"
22 #include <vector>
23
24 namespace llvm {
25   class ARMConstantPoolValue;
26
27   namespace ARMISD {
28     // ARM Specific DAG Nodes
29     enum NodeType {
30       // Start the numbering where the builtin ops and target ops leave off.
31       FIRST_NUMBER = ISD::BUILTIN_OP_END,
32
33       Wrapper,      // Wrapper - A wrapper node for TargetConstantPool,
34                     // TargetExternalSymbol, and TargetGlobalAddress.
35       WrapperJT,    // WrapperJT - A wrapper node for TargetJumpTable
36
37       CALL,         // Function call.
38       CALL_PRED,    // Function call that's predicable.
39       CALL_NOLINK,  // Function call with branch not branch-and-link.
40       tCALL,        // Thumb function call.
41       BRCOND,       // Conditional branch.
42       BR_JT,        // Jumptable branch.
43       BR2_JT,       // Jumptable branch (2 level - jumptable entry is a jump).
44       RET_FLAG,     // Return with a flag operand.
45
46       PIC_ADD,      // Add with a PC operand and a PIC label.
47
48       CMP,          // ARM compare instructions.
49       CMPZ,         // ARM compare that sets only Z flag.
50       CMPFP,        // ARM VFP compare instruction, sets FPSCR.
51       CMPFPw0,      // ARM VFP compare against zero instruction, sets FPSCR.
52       FMSTAT,       // ARM fmstat instruction.
53       CMOV,         // ARM conditional move instructions.
54       CNEG,         // ARM conditional negate instructions.
55
56       FTOSI,        // FP to sint within a FP register.
57       FTOUI,        // FP to uint within a FP register.
58       SITOF,        // sint to FP within a FP register.
59       UITOF,        // uint to FP within a FP register.
60
61       SRL_FLAG,     // V,Flag = srl_flag X -> srl X, 1 + save carry out.
62       SRA_FLAG,     // V,Flag = sra_flag X -> sra X, 1 + save carry out.
63       RRX,          // V = RRX X, Flag     -> srl X, 1 + shift in carry flag.
64
65       FMRRD,        // double to two gprs.
66       FMDRR,        // Two gprs to double.
67
68       EH_SJLJ_SETJMP,    // SjLj exception handling setjmp.
69       EH_SJLJ_LONGJMP,   // SjLj exception handling longjmp.
70
71       THREAD_POINTER,
72
73       DYN_ALLOC,    // Dynamic allocation on the stack.
74
75       VCEQ,         // Vector compare equal.
76       VCGE,         // Vector compare greater than or equal.
77       VCGEU,        // Vector compare unsigned greater than or equal.
78       VCGT,         // Vector compare greater than.
79       VCGTU,        // Vector compare unsigned greater than.
80       VTST,         // Vector test bits.
81
82       // Vector shift by immediate:
83       VSHL,         // ...left
84       VSHRs,        // ...right (signed)
85       VSHRu,        // ...right (unsigned)
86       VSHLLs,       // ...left long (signed)
87       VSHLLu,       // ...left long (unsigned)
88       VSHLLi,       // ...left long (with maximum shift count)
89       VSHRN,        // ...right narrow
90
91       // Vector rounding shift by immediate:
92       VRSHRs,       // ...right (signed)
93       VRSHRu,       // ...right (unsigned)
94       VRSHRN,       // ...right narrow
95
96       // Vector saturating shift by immediate:
97       VQSHLs,       // ...left (signed)
98       VQSHLu,       // ...left (unsigned)
99       VQSHLsu,      // ...left (signed to unsigned)
100       VQSHRNs,      // ...right narrow (signed)
101       VQSHRNu,      // ...right narrow (unsigned)
102       VQSHRNsu,     // ...right narrow (signed to unsigned)
103
104       // Vector saturating rounding shift by immediate:
105       VQRSHRNs,     // ...right narrow (signed)
106       VQRSHRNu,     // ...right narrow (unsigned)
107       VQRSHRNsu,    // ...right narrow (signed to unsigned)
108
109       // Vector shift and insert:
110       VSLI,         // ...left
111       VSRI,         // ...right
112
113       // Vector get lane (VMOV scalar to ARM core register)
114       // (These are used for 8- and 16-bit element types only.)
115       VGETLANEu,    // zero-extend vector extract element
116       VGETLANEs,    // sign-extend vector extract element
117
118       // Vector duplicate:
119       VDUP,
120       VDUPLANE,
121
122       // Vector load/store with (de)interleaving
123       VLD2D,
124       VLD3D,
125       VLD4D,
126       VST2D,
127       VST3D,
128       VST4D,
129
130       // Vector shuffles:
131       VEXT,         // extract
132       VREV64,       // reverse elements within 64-bit doublewords
133       VREV32,       // reverse elements within 32-bit words
134       VREV16        // reverse elements within 16-bit halfwords
135     };
136   }
137
138   /// Define some predicates that are used for node matching.
139   namespace ARM {
140     /// getVMOVImm - If this is a build_vector of constants which can be
141     /// formed by using a VMOV instruction of the specified element size,
142     /// return the constant being splatted.  The ByteSize field indicates the
143     /// number of bytes of each element [1248].
144     SDValue getVMOVImm(SDNode *N, unsigned ByteSize, SelectionDAG &DAG);
145   }
146
147   //===--------------------------------------------------------------------===//
148   //  ARMTargetLowering - ARM Implementation of the TargetLowering interface
149
150   class ARMTargetLowering : public TargetLowering {
151     int VarArgsFrameIndex;            // FrameIndex for start of varargs area.
152   public:
153     explicit ARMTargetLowering(TargetMachine &TM);
154
155     virtual SDValue LowerOperation(SDValue Op, SelectionDAG &DAG);
156
157     /// ReplaceNodeResults - Replace the results of node with an illegal result
158     /// type with new values built out of custom code.
159     ///
160     virtual void ReplaceNodeResults(SDNode *N, SmallVectorImpl<SDValue>&Results,
161                                     SelectionDAG &DAG);
162
163     virtual SDValue PerformDAGCombine(SDNode *N, DAGCombinerInfo &DCI) const;
164
165     virtual const char *getTargetNodeName(unsigned Opcode) const;
166
167     virtual MachineBasicBlock *EmitInstrWithCustomInserter(MachineInstr *MI,
168                                                   MachineBasicBlock *MBB) const;
169
170     /// allowsUnalignedMemoryAccesses - Returns true if the target allows
171     /// unaligned memory accesses. of the specified type.
172     /// FIXME: Add getOptimalMemOpType to implement memcpy with NEON?
173     virtual bool allowsUnalignedMemoryAccesses(EVT VT) const;
174
175     /// isLegalAddressingMode - Return true if the addressing mode represented
176     /// by AM is legal for this target, for a load/store of the specified type.
177     virtual bool isLegalAddressingMode(const AddrMode &AM, const Type *Ty)const;
178     bool isLegalT2ScaledAddressingMode(const AddrMode &AM, EVT VT) const;
179
180     /// getPreIndexedAddressParts - returns true by value, base pointer and
181     /// offset pointer and addressing mode by reference if the node's address
182     /// can be legally represented as pre-indexed load / store address.
183     virtual bool getPreIndexedAddressParts(SDNode *N, SDValue &Base,
184                                            SDValue &Offset,
185                                            ISD::MemIndexedMode &AM,
186                                            SelectionDAG &DAG) const;
187
188     /// getPostIndexedAddressParts - returns true by value, base pointer and
189     /// offset pointer and addressing mode by reference if this node can be
190     /// combined with a load / store to form a post-indexed load / store.
191     virtual bool getPostIndexedAddressParts(SDNode *N, SDNode *Op,
192                                             SDValue &Base, SDValue &Offset,
193                                             ISD::MemIndexedMode &AM,
194                                             SelectionDAG &DAG) const;
195
196     virtual void computeMaskedBitsForTargetNode(const SDValue Op,
197                                                 const APInt &Mask,
198                                                 APInt &KnownZero,
199                                                 APInt &KnownOne,
200                                                 const SelectionDAG &DAG,
201                                                 unsigned Depth) const;
202
203
204     ConstraintType getConstraintType(const std::string &Constraint) const;
205     std::pair<unsigned, const TargetRegisterClass*>
206       getRegForInlineAsmConstraint(const std::string &Constraint,
207                                    EVT VT) const;
208     std::vector<unsigned>
209     getRegClassForInlineAsmConstraint(const std::string &Constraint,
210                                       EVT VT) const;
211
212     /// LowerAsmOperandForConstraint - Lower the specified operand into the Ops
213     /// vector.  If it is invalid, don't add anything to Ops. If hasMemory is
214     /// true it means one of the asm constraint of the inline asm instruction
215     /// being processed is 'm'.
216     virtual void LowerAsmOperandForConstraint(SDValue Op,
217                                               char ConstraintLetter,
218                                               bool hasMemory,
219                                               std::vector<SDValue> &Ops,
220                                               SelectionDAG &DAG) const;
221
222     virtual const ARMSubtarget* getSubtarget() {
223       return Subtarget;
224     }
225
226     /// getFunctionAlignment - Return the Log2 alignment of this function.
227     virtual unsigned getFunctionAlignment(const Function *F) const;
228
229   private:
230     /// Subtarget - Keep a pointer to the ARMSubtarget around so that we can
231     /// make the right decision when generating code for different targets.
232     const ARMSubtarget *Subtarget;
233
234     /// ARMPCLabelIndex - Keep track of the number of ARM PC labels created.
235     ///
236     unsigned ARMPCLabelIndex;
237
238     void addTypeForNEON(EVT VT, EVT PromotedLdStVT, EVT PromotedBitwiseVT);
239     void addDRTypeForNEON(EVT VT);
240     void addQRTypeForNEON(EVT VT);
241
242     typedef SmallVector<std::pair<unsigned, SDValue>, 8> RegsToPassVector;
243     void PassF64ArgInRegs(DebugLoc dl, SelectionDAG &DAG,
244                           SDValue Chain, SDValue &Arg,
245                           RegsToPassVector &RegsToPass,
246                           CCValAssign &VA, CCValAssign &NextVA,
247                           SDValue &StackPtr,
248                           SmallVector<SDValue, 8> &MemOpChains,
249                           ISD::ArgFlagsTy Flags);
250     SDValue GetF64FormalArgument(CCValAssign &VA, CCValAssign &NextVA,
251                                  SDValue &Root, SelectionDAG &DAG, DebugLoc dl);
252
253     CCAssignFn *CCAssignFnForNode(unsigned CC, bool Return, bool isVarArg) const;
254     SDValue LowerMemOpCallTo(SDValue Chain, SDValue StackPtr, SDValue Arg,
255                              DebugLoc dl, SelectionDAG &DAG,
256                              const CCValAssign &VA,
257                              ISD::ArgFlagsTy Flags);
258     SDValue LowerINTRINSIC_W_CHAIN(SDValue Op, SelectionDAG &DAG);
259     SDValue LowerINTRINSIC_WO_CHAIN(SDValue Op, SelectionDAG &DAG);
260     SDValue LowerGlobalAddressDarwin(SDValue Op, SelectionDAG &DAG);
261     SDValue LowerGlobalAddressELF(SDValue Op, SelectionDAG &DAG);
262     SDValue LowerGlobalTLSAddress(SDValue Op, SelectionDAG &DAG);
263     SDValue LowerToTLSGeneralDynamicModel(GlobalAddressSDNode *GA,
264                                             SelectionDAG &DAG);
265     SDValue LowerToTLSExecModels(GlobalAddressSDNode *GA,
266                                    SelectionDAG &DAG);
267     SDValue LowerGLOBAL_OFFSET_TABLE(SDValue Op, SelectionDAG &DAG);
268     SDValue LowerBR_JT(SDValue Op, SelectionDAG &DAG);
269     SDValue LowerFRAMEADDR(SDValue Op, SelectionDAG &DAG);
270     SDValue LowerDYNAMIC_STACKALLOC(SDValue Op, SelectionDAG &DAG);
271
272     SDValue EmitTargetCodeForMemcpy(SelectionDAG &DAG, DebugLoc dl,
273                                       SDValue Chain,
274                                       SDValue Dst, SDValue Src,
275                                       SDValue Size, unsigned Align,
276                                       bool AlwaysInline,
277                                       const Value *DstSV, uint64_t DstSVOff,
278                                       const Value *SrcSV, uint64_t SrcSVOff);
279     SDValue LowerCallResult(SDValue Chain, SDValue InFlag,
280                             unsigned CallConv, bool isVarArg,
281                             const SmallVectorImpl<ISD::InputArg> &Ins,
282                             DebugLoc dl, SelectionDAG &DAG,
283                             SmallVectorImpl<SDValue> &InVals);
284
285     virtual SDValue
286       LowerFormalArguments(SDValue Chain,
287                            unsigned CallConv, bool isVarArg,
288                            const SmallVectorImpl<ISD::InputArg> &Ins,
289                            DebugLoc dl, SelectionDAG &DAG,
290                            SmallVectorImpl<SDValue> &InVals);
291
292     virtual SDValue
293       LowerCall(SDValue Chain, SDValue Callee,
294                 unsigned CallConv, bool isVarArg,
295                 bool isTailCall,
296                 const SmallVectorImpl<ISD::OutputArg> &Outs,
297                 const SmallVectorImpl<ISD::InputArg> &Ins,
298                 DebugLoc dl, SelectionDAG &DAG,
299                 SmallVectorImpl<SDValue> &InVals);
300
301     virtual SDValue
302       LowerReturn(SDValue Chain,
303                   unsigned CallConv, bool isVarArg,
304                   const SmallVectorImpl<ISD::OutputArg> &Outs,
305                   DebugLoc dl, SelectionDAG &DAG);
306   };
307 }
308
309 #endif  // ARMISELLOWERING_H