1f22ef01cSRoman Divacky //===-- MachineFunction.cpp -----------------------------------------------===//
2f22ef01cSRoman Divacky //
3f22ef01cSRoman Divacky //                     The LLVM Compiler Infrastructure
4f22ef01cSRoman Divacky //
5f22ef01cSRoman Divacky // This file is distributed under the University of Illinois Open Source
6f22ef01cSRoman Divacky // License. See LICENSE.TXT for details.
7f22ef01cSRoman Divacky //
8f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
9f22ef01cSRoman Divacky //
10f22ef01cSRoman Divacky // Collect native machine code information for a function.  This allows
11f22ef01cSRoman Divacky // target-specific information about the generated code to be stored with each
12f22ef01cSRoman Divacky // function.
13f22ef01cSRoman Divacky //
14f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
15f22ef01cSRoman Divacky 
16f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineFunction.h"
17139f7f9bSDimitry Andric #include "llvm/ADT/STLExtras.h"
18139f7f9bSDimitry Andric #include "llvm/ADT/SmallString.h"
19139f7f9bSDimitry Andric #include "llvm/Analysis/ConstantFolding.h"
207d523365SDimitry Andric #include "llvm/Analysis/EHPersonalities.h"
21dff0c46cSDimitry Andric #include "llvm/CodeGen/MachineConstantPool.h"
22f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineFrameInfo.h"
238f0fd8f6SDimitry Andric #include "llvm/CodeGen/MachineFunctionInitializer.h"
24139f7f9bSDimitry Andric #include "llvm/CodeGen/MachineFunctionPass.h"
25f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineInstr.h"
26f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineJumpTableInfo.h"
27f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineModuleInfo.h"
28f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineRegisterInfo.h"
29f22ef01cSRoman Divacky #include "llvm/CodeGen/Passes.h"
307d523365SDimitry Andric #include "llvm/CodeGen/PseudoSourceValue.h"
317d523365SDimitry Andric #include "llvm/CodeGen/WinEHFuncInfo.h"
32139f7f9bSDimitry Andric #include "llvm/IR/DataLayout.h"
3391bc56edSDimitry Andric #include "llvm/IR/DebugInfo.h"
34139f7f9bSDimitry Andric #include "llvm/IR/Function.h"
35875ed548SDimitry Andric #include "llvm/IR/Module.h"
363dac3a9bSDimitry Andric #include "llvm/IR/ModuleSlotTracker.h"
37f22ef01cSRoman Divacky #include "llvm/MC/MCAsmInfo.h"
38f22ef01cSRoman Divacky #include "llvm/MC/MCContext.h"
39f22ef01cSRoman Divacky #include "llvm/Support/Debug.h"
40f22ef01cSRoman Divacky #include "llvm/Support/GraphWriter.h"
41f22ef01cSRoman Divacky #include "llvm/Support/raw_ostream.h"
42139f7f9bSDimitry Andric #include "llvm/Target/TargetFrameLowering.h"
43139f7f9bSDimitry Andric #include "llvm/Target/TargetLowering.h"
44139f7f9bSDimitry Andric #include "llvm/Target/TargetMachine.h"
4539d628a0SDimitry Andric #include "llvm/Target/TargetSubtargetInfo.h"
46f22ef01cSRoman Divacky using namespace llvm;
47f22ef01cSRoman Divacky 
4891bc56edSDimitry Andric #define DEBUG_TYPE "codegen"
4991bc56edSDimitry Andric 
507d523365SDimitry Andric static cl::opt<unsigned>
517d523365SDimitry Andric     AlignAllFunctions("align-all-functions",
527d523365SDimitry Andric                       cl::desc("Force the alignment of all functions."),
537d523365SDimitry Andric                       cl::init(0), cl::Hidden);
547d523365SDimitry Andric 
558f0fd8f6SDimitry Andric void MachineFunctionInitializer::anchor() {}
568f0fd8f6SDimitry Andric 
57d88c1a5aSDimitry Andric static const char *getPropertyName(MachineFunctionProperties::Property Prop) {
58d88c1a5aSDimitry Andric   typedef MachineFunctionProperties::Property P;
59d88c1a5aSDimitry Andric   switch(Prop) {
60d88c1a5aSDimitry Andric   case P::FailedISel: return "FailedISel";
61d88c1a5aSDimitry Andric   case P::IsSSA: return "IsSSA";
62d88c1a5aSDimitry Andric   case P::Legalized: return "Legalized";
63d88c1a5aSDimitry Andric   case P::NoPHIs: return "NoPHIs";
64d88c1a5aSDimitry Andric   case P::NoVRegs: return "NoVRegs";
65d88c1a5aSDimitry Andric   case P::RegBankSelected: return "RegBankSelected";
66d88c1a5aSDimitry Andric   case P::Selected: return "Selected";
67d88c1a5aSDimitry Andric   case P::TracksLiveness: return "TracksLiveness";
68d88c1a5aSDimitry Andric   }
69d88c1a5aSDimitry Andric   llvm_unreachable("Invalid machine function property");
70d88c1a5aSDimitry Andric }
71d88c1a5aSDimitry Andric 
72d88c1a5aSDimitry Andric void MachineFunctionProperties::print(raw_ostream &OS) const {
73d88c1a5aSDimitry Andric   const char *Separator = "";
74d88c1a5aSDimitry Andric   for (BitVector::size_type I = 0; I < Properties.size(); ++I) {
75d88c1a5aSDimitry Andric     if (!Properties[I])
763ca95b02SDimitry Andric       continue;
77d88c1a5aSDimitry Andric     OS << Separator << getPropertyName(static_cast<Property>(I));
78d88c1a5aSDimitry Andric     Separator = ", ";
793ca95b02SDimitry Andric   }
803ca95b02SDimitry Andric }
813ca95b02SDimitry Andric 
82f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
83f22ef01cSRoman Divacky // MachineFunction implementation
84f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
85f22ef01cSRoman Divacky 
868f0fd8f6SDimitry Andric // Out-of-line virtual method.
87f22ef01cSRoman Divacky MachineFunctionInfo::~MachineFunctionInfo() {}
88f22ef01cSRoman Divacky 
89d88c1a5aSDimitry Andric void ilist_alloc_traits<MachineBasicBlock>::deleteNode(MachineBasicBlock *MBB) {
90f22ef01cSRoman Divacky   MBB->getParent()->DeleteMachineBasicBlock(MBB);
91f22ef01cSRoman Divacky }
92f22ef01cSRoman Divacky 
933ca95b02SDimitry Andric static inline unsigned getFnStackAlignment(const TargetSubtargetInfo *STI,
943ca95b02SDimitry Andric                                            const Function *Fn) {
953ca95b02SDimitry Andric   if (Fn->hasFnAttribute(Attribute::StackAlignment))
963ca95b02SDimitry Andric     return Fn->getFnStackAlignment();
973ca95b02SDimitry Andric   return STI->getFrameLowering()->getStackAlignment();
983ca95b02SDimitry Andric }
993ca95b02SDimitry Andric 
100f22ef01cSRoman Divacky MachineFunction::MachineFunction(const Function *F, const TargetMachine &TM,
10139d628a0SDimitry Andric                                  unsigned FunctionNum, MachineModuleInfo &mmi)
102ff0cc061SDimitry Andric     : Fn(F), Target(TM), STI(TM.getSubtargetImpl(*F)), Ctx(mmi.getContext()),
10339d628a0SDimitry Andric       MMI(mmi) {
104d88c1a5aSDimitry Andric   FunctionNumber = FunctionNum;
105d88c1a5aSDimitry Andric   init();
106d88c1a5aSDimitry Andric }
107d88c1a5aSDimitry Andric 
108d88c1a5aSDimitry Andric void MachineFunction::init() {
1093ca95b02SDimitry Andric   // Assume the function starts in SSA form with correct liveness.
1103ca95b02SDimitry Andric   Properties.set(MachineFunctionProperties::Property::IsSSA);
1113ca95b02SDimitry Andric   Properties.set(MachineFunctionProperties::Property::TracksLiveness);
11239d628a0SDimitry Andric   if (STI->getRegisterInfo())
11339d628a0SDimitry Andric     RegInfo = new (Allocator) MachineRegisterInfo(this);
114f22ef01cSRoman Divacky   else
11591bc56edSDimitry Andric     RegInfo = nullptr;
116f785676fSDimitry Andric 
11791bc56edSDimitry Andric   MFInfo = nullptr;
1183ca95b02SDimitry Andric   // We can realign the stack if the target supports it and the user hasn't
1193ca95b02SDimitry Andric   // explicitly asked us not to.
1203ca95b02SDimitry Andric   bool CanRealignSP = STI->getFrameLowering()->isStackRealignable() &&
121d88c1a5aSDimitry Andric                       !Fn->hasFnAttribute("no-realign-stack");
1223ca95b02SDimitry Andric   FrameInfo = new (Allocator) MachineFrameInfo(
1233ca95b02SDimitry Andric       getFnStackAlignment(STI, Fn), /*StackRealignable=*/CanRealignSP,
1243ca95b02SDimitry Andric       /*ForceRealign=*/CanRealignSP &&
125d88c1a5aSDimitry Andric           Fn->hasFnAttribute(Attribute::StackAlignment));
126f785676fSDimitry Andric 
127ff0cc061SDimitry Andric   if (Fn->hasFnAttribute(Attribute::StackAlignment))
128ff0cc061SDimitry Andric     FrameInfo->ensureMaxAlignment(Fn->getFnStackAlignment());
129f785676fSDimitry Andric 
130875ed548SDimitry Andric   ConstantPool = new (Allocator) MachineConstantPool(getDataLayout());
13139d628a0SDimitry Andric   Alignment = STI->getTargetLowering()->getMinFunctionAlignment();
132f785676fSDimitry Andric 
133bd5abe19SDimitry Andric   // FIXME: Shouldn't use pref alignment if explicit alignment is set on Fn.
1347d523365SDimitry Andric   // FIXME: Use Function::optForSize().
135ff0cc061SDimitry Andric   if (!Fn->hasFnAttribute(Attribute::OptimizeForSize))
136bd5abe19SDimitry Andric     Alignment = std::max(Alignment,
13739d628a0SDimitry Andric                          STI->getTargetLowering()->getPrefFunctionAlignment());
138f785676fSDimitry Andric 
1397d523365SDimitry Andric   if (AlignAllFunctions)
1407d523365SDimitry Andric     Alignment = AlignAllFunctions;
1417d523365SDimitry Andric 
14291bc56edSDimitry Andric   JumpTableInfo = nullptr;
1437d523365SDimitry Andric 
1447d523365SDimitry Andric   if (isFuncletEHPersonality(classifyEHPersonality(
145d88c1a5aSDimitry Andric           Fn->hasPersonalityFn() ? Fn->getPersonalityFn() : nullptr))) {
1467d523365SDimitry Andric     WinEHInfo = new (Allocator) WinEHFuncInfo();
1477d523365SDimitry Andric   }
1487d523365SDimitry Andric 
149d88c1a5aSDimitry Andric   assert(Target.isCompatibleDataLayout(getDataLayout()) &&
1507d523365SDimitry Andric          "Can't create a MachineFunction using a Module with a "
1517d523365SDimitry Andric          "Target-incompatible DataLayout attached\n");
1527d523365SDimitry Andric 
1537d523365SDimitry Andric   PSVManager = llvm::make_unique<PseudoSourceValueManager>();
154f22ef01cSRoman Divacky }
155f22ef01cSRoman Divacky 
156f22ef01cSRoman Divacky MachineFunction::~MachineFunction() {
157d88c1a5aSDimitry Andric   clear();
158d88c1a5aSDimitry Andric }
159d88c1a5aSDimitry Andric 
160d88c1a5aSDimitry Andric void MachineFunction::clear() {
161d88c1a5aSDimitry Andric   Properties.reset();
162139f7f9bSDimitry Andric   // Don't call destructors on MachineInstr and MachineOperand. All of their
163139f7f9bSDimitry Andric   // memory comes from the BumpPtrAllocator which is about to be purged.
164139f7f9bSDimitry Andric   //
165139f7f9bSDimitry Andric   // Do call MachineBasicBlock destructors, it contains std::vectors.
166139f7f9bSDimitry Andric   for (iterator I = begin(), E = end(); I != E; I = BasicBlocks.erase(I))
167139f7f9bSDimitry Andric     I->Insts.clearAndLeakNodesUnsafely();
168139f7f9bSDimitry Andric 
169f22ef01cSRoman Divacky   InstructionRecycler.clear(Allocator);
170139f7f9bSDimitry Andric   OperandRecycler.clear(Allocator);
171f22ef01cSRoman Divacky   BasicBlockRecycler.clear(Allocator);
172f22ef01cSRoman Divacky   if (RegInfo) {
173f22ef01cSRoman Divacky     RegInfo->~MachineRegisterInfo();
174f22ef01cSRoman Divacky     Allocator.Deallocate(RegInfo);
175f22ef01cSRoman Divacky   }
176f22ef01cSRoman Divacky   if (MFInfo) {
177f22ef01cSRoman Divacky     MFInfo->~MachineFunctionInfo();
178f22ef01cSRoman Divacky     Allocator.Deallocate(MFInfo);
179f22ef01cSRoman Divacky   }
1807ae0e2c9SDimitry Andric 
1817ae0e2c9SDimitry Andric   FrameInfo->~MachineFrameInfo();
1827ae0e2c9SDimitry Andric   Allocator.Deallocate(FrameInfo);
1837ae0e2c9SDimitry Andric 
1847ae0e2c9SDimitry Andric   ConstantPool->~MachineConstantPool();
1857ae0e2c9SDimitry Andric   Allocator.Deallocate(ConstantPool);
186f22ef01cSRoman Divacky 
187f22ef01cSRoman Divacky   if (JumpTableInfo) {
188f22ef01cSRoman Divacky     JumpTableInfo->~MachineJumpTableInfo();
189f22ef01cSRoman Divacky     Allocator.Deallocate(JumpTableInfo);
190f22ef01cSRoman Divacky   }
1917d523365SDimitry Andric 
1927d523365SDimitry Andric   if (WinEHInfo) {
1937d523365SDimitry Andric     WinEHInfo->~WinEHFuncInfo();
1947d523365SDimitry Andric     Allocator.Deallocate(WinEHInfo);
1957d523365SDimitry Andric   }
196f22ef01cSRoman Divacky }
197f22ef01cSRoman Divacky 
198875ed548SDimitry Andric const DataLayout &MachineFunction::getDataLayout() const {
199875ed548SDimitry Andric   return Fn->getParent()->getDataLayout();
200875ed548SDimitry Andric }
201875ed548SDimitry Andric 
2028f0fd8f6SDimitry Andric /// Get the JumpTableInfo for this function.
2038f0fd8f6SDimitry Andric /// If it does not already exist, allocate one.
204f22ef01cSRoman Divacky MachineJumpTableInfo *MachineFunction::
205f22ef01cSRoman Divacky getOrCreateJumpTableInfo(unsigned EntryKind) {
206f22ef01cSRoman Divacky   if (JumpTableInfo) return JumpTableInfo;
207f22ef01cSRoman Divacky 
208f22ef01cSRoman Divacky   JumpTableInfo = new (Allocator)
209f22ef01cSRoman Divacky     MachineJumpTableInfo((MachineJumpTableInfo::JTEntryKind)EntryKind);
210f22ef01cSRoman Divacky   return JumpTableInfo;
211f22ef01cSRoman Divacky }
212f22ef01cSRoman Divacky 
21391bc56edSDimitry Andric /// Should we be emitting segmented stack stuff for the function
2148c24ff90SDimitry Andric bool MachineFunction::shouldSplitStack() const {
21591bc56edSDimitry Andric   return getFunction()->hasFnAttribute("split-stack");
21691bc56edSDimitry Andric }
21791bc56edSDimitry Andric 
2188f0fd8f6SDimitry Andric /// This discards all of the MachineBasicBlock numbers and recomputes them.
2198f0fd8f6SDimitry Andric /// This guarantees that the MBB numbers are sequential, dense, and match the
2208f0fd8f6SDimitry Andric /// ordering of the blocks within the function.  If a specific MachineBasicBlock
2218f0fd8f6SDimitry Andric /// is specified, only that block and those after it are renumbered.
222f22ef01cSRoman Divacky void MachineFunction::RenumberBlocks(MachineBasicBlock *MBB) {
223f22ef01cSRoman Divacky   if (empty()) { MBBNumbering.clear(); return; }
224f22ef01cSRoman Divacky   MachineFunction::iterator MBBI, E = end();
22591bc56edSDimitry Andric   if (MBB == nullptr)
226f22ef01cSRoman Divacky     MBBI = begin();
227f22ef01cSRoman Divacky   else
2287d523365SDimitry Andric     MBBI = MBB->getIterator();
229f22ef01cSRoman Divacky 
230f22ef01cSRoman Divacky   // Figure out the block number this should have.
231f22ef01cSRoman Divacky   unsigned BlockNo = 0;
232f22ef01cSRoman Divacky   if (MBBI != begin())
23391bc56edSDimitry Andric     BlockNo = std::prev(MBBI)->getNumber() + 1;
234f22ef01cSRoman Divacky 
235f22ef01cSRoman Divacky   for (; MBBI != E; ++MBBI, ++BlockNo) {
236f22ef01cSRoman Divacky     if (MBBI->getNumber() != (int)BlockNo) {
237f22ef01cSRoman Divacky       // Remove use of the old number.
238f22ef01cSRoman Divacky       if (MBBI->getNumber() != -1) {
239f22ef01cSRoman Divacky         assert(MBBNumbering[MBBI->getNumber()] == &*MBBI &&
240f22ef01cSRoman Divacky                "MBB number mismatch!");
24191bc56edSDimitry Andric         MBBNumbering[MBBI->getNumber()] = nullptr;
242f22ef01cSRoman Divacky       }
243f22ef01cSRoman Divacky 
244f22ef01cSRoman Divacky       // If BlockNo is already taken, set that block's number to -1.
245f22ef01cSRoman Divacky       if (MBBNumbering[BlockNo])
246f22ef01cSRoman Divacky         MBBNumbering[BlockNo]->setNumber(-1);
247f22ef01cSRoman Divacky 
2487d523365SDimitry Andric       MBBNumbering[BlockNo] = &*MBBI;
249f22ef01cSRoman Divacky       MBBI->setNumber(BlockNo);
250f22ef01cSRoman Divacky     }
251f22ef01cSRoman Divacky   }
252f22ef01cSRoman Divacky 
253f22ef01cSRoman Divacky   // Okay, all the blocks are renumbered.  If we have compactified the block
254f22ef01cSRoman Divacky   // numbering, shrink MBBNumbering now.
255f22ef01cSRoman Divacky   assert(BlockNo <= MBBNumbering.size() && "Mismatch!");
256f22ef01cSRoman Divacky   MBBNumbering.resize(BlockNo);
257f22ef01cSRoman Divacky }
258f22ef01cSRoman Divacky 
2598f0fd8f6SDimitry Andric /// Allocate a new MachineInstr. Use this instead of `new MachineInstr'.
2603ca95b02SDimitry Andric MachineInstr *MachineFunction::CreateMachineInstr(const MCInstrDesc &MCID,
2613ca95b02SDimitry Andric                                                   const DebugLoc &DL,
2623ca95b02SDimitry Andric                                                   bool NoImp) {
263f22ef01cSRoman Divacky   return new (InstructionRecycler.Allocate<MachineInstr>(Allocator))
264139f7f9bSDimitry Andric     MachineInstr(*this, MCID, DL, NoImp);
265f22ef01cSRoman Divacky }
266f22ef01cSRoman Divacky 
2678f0fd8f6SDimitry Andric /// Create a new MachineInstr which is a copy of the 'Orig' instruction,
2688f0fd8f6SDimitry Andric /// identical in all ways except the instruction has no parent, prev, or next.
269f22ef01cSRoman Divacky MachineInstr *
270f22ef01cSRoman Divacky MachineFunction::CloneMachineInstr(const MachineInstr *Orig) {
271f22ef01cSRoman Divacky   return new (InstructionRecycler.Allocate<MachineInstr>(Allocator))
272f22ef01cSRoman Divacky              MachineInstr(*this, *Orig);
273f22ef01cSRoman Divacky }
274f22ef01cSRoman Divacky 
2758f0fd8f6SDimitry Andric /// Delete the given MachineInstr.
276f22ef01cSRoman Divacky ///
277139f7f9bSDimitry Andric /// This function also serves as the MachineInstr destructor - the real
278139f7f9bSDimitry Andric /// ~MachineInstr() destructor must be empty.
279f22ef01cSRoman Divacky void
280f22ef01cSRoman Divacky MachineFunction::DeleteMachineInstr(MachineInstr *MI) {
281139f7f9bSDimitry Andric   // Strip it for parts. The operand array and the MI object itself are
282139f7f9bSDimitry Andric   // independently recyclable.
283139f7f9bSDimitry Andric   if (MI->Operands)
284139f7f9bSDimitry Andric     deallocateOperandArray(MI->CapOperands, MI->Operands);
285139f7f9bSDimitry Andric   // Don't call ~MachineInstr() which must be trivial anyway because
286139f7f9bSDimitry Andric   // ~MachineFunction drops whole lists of MachineInstrs wihout calling their
287139f7f9bSDimitry Andric   // destructors.
288f22ef01cSRoman Divacky   InstructionRecycler.Deallocate(Allocator, MI);
289f22ef01cSRoman Divacky }
290f22ef01cSRoman Divacky 
2918f0fd8f6SDimitry Andric /// Allocate a new MachineBasicBlock. Use this instead of
2928f0fd8f6SDimitry Andric /// `new MachineBasicBlock'.
293f22ef01cSRoman Divacky MachineBasicBlock *
294f22ef01cSRoman Divacky MachineFunction::CreateMachineBasicBlock(const BasicBlock *bb) {
295f22ef01cSRoman Divacky   return new (BasicBlockRecycler.Allocate<MachineBasicBlock>(Allocator))
296f22ef01cSRoman Divacky              MachineBasicBlock(*this, bb);
297f22ef01cSRoman Divacky }
298f22ef01cSRoman Divacky 
2998f0fd8f6SDimitry Andric /// Delete the given MachineBasicBlock.
300f22ef01cSRoman Divacky void
301f22ef01cSRoman Divacky MachineFunction::DeleteMachineBasicBlock(MachineBasicBlock *MBB) {
302f22ef01cSRoman Divacky   assert(MBB->getParent() == this && "MBB parent mismatch!");
303f22ef01cSRoman Divacky   MBB->~MachineBasicBlock();
304f22ef01cSRoman Divacky   BasicBlockRecycler.Deallocate(Allocator, MBB);
305f22ef01cSRoman Divacky }
306f22ef01cSRoman Divacky 
3073ca95b02SDimitry Andric MachineMemOperand *MachineFunction::getMachineMemOperand(
3083ca95b02SDimitry Andric     MachinePointerInfo PtrInfo, MachineMemOperand::Flags f, uint64_t s,
309d88c1a5aSDimitry Andric     unsigned base_alignment, const AAMDNodes &AAInfo, const MDNode *Ranges,
310d88c1a5aSDimitry Andric     SynchronizationScope SynchScope, AtomicOrdering Ordering,
311d88c1a5aSDimitry Andric     AtomicOrdering FailureOrdering) {
3123ca95b02SDimitry Andric   return new (Allocator)
313d88c1a5aSDimitry Andric       MachineMemOperand(PtrInfo, f, s, base_alignment, AAInfo, Ranges,
314d88c1a5aSDimitry Andric                         SynchScope, Ordering, FailureOrdering);
315f22ef01cSRoman Divacky }
316f22ef01cSRoman Divacky 
317f22ef01cSRoman Divacky MachineMemOperand *
318f22ef01cSRoman Divacky MachineFunction::getMachineMemOperand(const MachineMemOperand *MMO,
319f22ef01cSRoman Divacky                                       int64_t Offset, uint64_t Size) {
32091bc56edSDimitry Andric   if (MMO->getValue())
321f22ef01cSRoman Divacky     return new (Allocator)
3222754fe60SDimitry Andric                MachineMemOperand(MachinePointerInfo(MMO->getValue(),
3232754fe60SDimitry Andric                                                     MMO->getOffset()+Offset),
324d88c1a5aSDimitry Andric                                  MMO->getFlags(), Size, MMO->getBaseAlignment(),
325d88c1a5aSDimitry Andric                                  AAMDNodes(), nullptr, MMO->getSynchScope(),
326d88c1a5aSDimitry Andric                                  MMO->getOrdering(), MMO->getFailureOrdering());
32791bc56edSDimitry Andric   return new (Allocator)
32891bc56edSDimitry Andric              MachineMemOperand(MachinePointerInfo(MMO->getPseudoValue(),
32991bc56edSDimitry Andric                                                   MMO->getOffset()+Offset),
330d88c1a5aSDimitry Andric                                MMO->getFlags(), Size, MMO->getBaseAlignment(),
331d88c1a5aSDimitry Andric                                AAMDNodes(), nullptr, MMO->getSynchScope(),
332d88c1a5aSDimitry Andric                                MMO->getOrdering(), MMO->getFailureOrdering());
333f22ef01cSRoman Divacky }
334f22ef01cSRoman Divacky 
335f22ef01cSRoman Divacky MachineInstr::mmo_iterator
336f22ef01cSRoman Divacky MachineFunction::allocateMemRefsArray(unsigned long Num) {
337f22ef01cSRoman Divacky   return Allocator.Allocate<MachineMemOperand *>(Num);
338f22ef01cSRoman Divacky }
339f22ef01cSRoman Divacky 
340f22ef01cSRoman Divacky std::pair<MachineInstr::mmo_iterator, MachineInstr::mmo_iterator>
341f22ef01cSRoman Divacky MachineFunction::extractLoadMemRefs(MachineInstr::mmo_iterator Begin,
342f22ef01cSRoman Divacky                                     MachineInstr::mmo_iterator End) {
343f22ef01cSRoman Divacky   // Count the number of load mem refs.
344f22ef01cSRoman Divacky   unsigned Num = 0;
345f22ef01cSRoman Divacky   for (MachineInstr::mmo_iterator I = Begin; I != End; ++I)
346f22ef01cSRoman Divacky     if ((*I)->isLoad())
347f22ef01cSRoman Divacky       ++Num;
348f22ef01cSRoman Divacky 
349f22ef01cSRoman Divacky   // Allocate a new array and populate it with the load information.
350f22ef01cSRoman Divacky   MachineInstr::mmo_iterator Result = allocateMemRefsArray(Num);
351f22ef01cSRoman Divacky   unsigned Index = 0;
352f22ef01cSRoman Divacky   for (MachineInstr::mmo_iterator I = Begin; I != End; ++I) {
353f22ef01cSRoman Divacky     if ((*I)->isLoad()) {
354f22ef01cSRoman Divacky       if (!(*I)->isStore())
355f22ef01cSRoman Divacky         // Reuse the MMO.
356f22ef01cSRoman Divacky         Result[Index] = *I;
357f22ef01cSRoman Divacky       else {
358f22ef01cSRoman Divacky         // Clone the MMO and unset the store flag.
359f22ef01cSRoman Divacky         MachineMemOperand *JustLoad =
3602754fe60SDimitry Andric           getMachineMemOperand((*I)->getPointerInfo(),
361f22ef01cSRoman Divacky                                (*I)->getFlags() & ~MachineMemOperand::MOStore,
3622754fe60SDimitry Andric                                (*I)->getSize(), (*I)->getBaseAlignment(),
363d88c1a5aSDimitry Andric                                (*I)->getAAInfo(), nullptr,
364d88c1a5aSDimitry Andric                                (*I)->getSynchScope(), (*I)->getOrdering(),
365d88c1a5aSDimitry Andric                                (*I)->getFailureOrdering());
366f22ef01cSRoman Divacky         Result[Index] = JustLoad;
367f22ef01cSRoman Divacky       }
368f22ef01cSRoman Divacky       ++Index;
369f22ef01cSRoman Divacky     }
370f22ef01cSRoman Divacky   }
371f22ef01cSRoman Divacky   return std::make_pair(Result, Result + Num);
372f22ef01cSRoman Divacky }
373f22ef01cSRoman Divacky 
374f22ef01cSRoman Divacky std::pair<MachineInstr::mmo_iterator, MachineInstr::mmo_iterator>
375f22ef01cSRoman Divacky MachineFunction::extractStoreMemRefs(MachineInstr::mmo_iterator Begin,
376f22ef01cSRoman Divacky                                      MachineInstr::mmo_iterator End) {
377f22ef01cSRoman Divacky   // Count the number of load mem refs.
378f22ef01cSRoman Divacky   unsigned Num = 0;
379f22ef01cSRoman Divacky   for (MachineInstr::mmo_iterator I = Begin; I != End; ++I)
380f22ef01cSRoman Divacky     if ((*I)->isStore())
381f22ef01cSRoman Divacky       ++Num;
382f22ef01cSRoman Divacky 
383f22ef01cSRoman Divacky   // Allocate a new array and populate it with the store information.
384f22ef01cSRoman Divacky   MachineInstr::mmo_iterator Result = allocateMemRefsArray(Num);
385f22ef01cSRoman Divacky   unsigned Index = 0;
386f22ef01cSRoman Divacky   for (MachineInstr::mmo_iterator I = Begin; I != End; ++I) {
387f22ef01cSRoman Divacky     if ((*I)->isStore()) {
388f22ef01cSRoman Divacky       if (!(*I)->isLoad())
389f22ef01cSRoman Divacky         // Reuse the MMO.
390f22ef01cSRoman Divacky         Result[Index] = *I;
391f22ef01cSRoman Divacky       else {
392f22ef01cSRoman Divacky         // Clone the MMO and unset the load flag.
393f22ef01cSRoman Divacky         MachineMemOperand *JustStore =
3942754fe60SDimitry Andric           getMachineMemOperand((*I)->getPointerInfo(),
395f22ef01cSRoman Divacky                                (*I)->getFlags() & ~MachineMemOperand::MOLoad,
3962754fe60SDimitry Andric                                (*I)->getSize(), (*I)->getBaseAlignment(),
397d88c1a5aSDimitry Andric                                (*I)->getAAInfo(), nullptr,
398d88c1a5aSDimitry Andric                                (*I)->getSynchScope(), (*I)->getOrdering(),
399d88c1a5aSDimitry Andric                                (*I)->getFailureOrdering());
400f22ef01cSRoman Divacky         Result[Index] = JustStore;
401f22ef01cSRoman Divacky       }
402f22ef01cSRoman Divacky       ++Index;
403f22ef01cSRoman Divacky     }
404f22ef01cSRoman Divacky   }
405f22ef01cSRoman Divacky   return std::make_pair(Result, Result + Num);
406f22ef01cSRoman Divacky }
407f22ef01cSRoman Divacky 
4087d523365SDimitry Andric const char *MachineFunction::createExternalSymbolName(StringRef Name) {
4097d523365SDimitry Andric   char *Dest = Allocator.Allocate<char>(Name.size() + 1);
4107d523365SDimitry Andric   std::copy(Name.begin(), Name.end(), Dest);
4117d523365SDimitry Andric   Dest[Name.size()] = 0;
4127d523365SDimitry Andric   return Dest;
4137d523365SDimitry Andric }
4147d523365SDimitry Andric 
4153861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
4163ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineFunction::dump() const {
417f22ef01cSRoman Divacky   print(dbgs());
418f22ef01cSRoman Divacky }
4193861d79fSDimitry Andric #endif
4203861d79fSDimitry Andric 
4213861d79fSDimitry Andric StringRef MachineFunction::getName() const {
4223861d79fSDimitry Andric   assert(getFunction() && "No function!");
4233861d79fSDimitry Andric   return getFunction()->getName();
4243861d79fSDimitry Andric }
425f22ef01cSRoman Divacky 
4263ca95b02SDimitry Andric void MachineFunction::print(raw_ostream &OS, const SlotIndexes *Indexes) const {
4273861d79fSDimitry Andric   OS << "# Machine code for function " << getName() << ": ";
4283ca95b02SDimitry Andric   getProperties().print(OS);
429d88c1a5aSDimitry Andric   OS << '\n';
430f22ef01cSRoman Divacky 
431f22ef01cSRoman Divacky   // Print Frame Information
432f22ef01cSRoman Divacky   FrameInfo->print(*this, OS);
433f22ef01cSRoman Divacky 
434f22ef01cSRoman Divacky   // Print JumpTable Information
435f22ef01cSRoman Divacky   if (JumpTableInfo)
436f22ef01cSRoman Divacky     JumpTableInfo->print(OS);
437f22ef01cSRoman Divacky 
438f22ef01cSRoman Divacky   // Print Constant Pool
439f22ef01cSRoman Divacky   ConstantPool->print(OS);
440f22ef01cSRoman Divacky 
44139d628a0SDimitry Andric   const TargetRegisterInfo *TRI = getSubtarget().getRegisterInfo();
442f22ef01cSRoman Divacky 
443f22ef01cSRoman Divacky   if (RegInfo && !RegInfo->livein_empty()) {
444f22ef01cSRoman Divacky     OS << "Function Live Ins: ";
445f22ef01cSRoman Divacky     for (MachineRegisterInfo::livein_iterator
446f22ef01cSRoman Divacky          I = RegInfo->livein_begin(), E = RegInfo->livein_end(); I != E; ++I) {
447bd5abe19SDimitry Andric       OS << PrintReg(I->first, TRI);
448f22ef01cSRoman Divacky       if (I->second)
449bd5abe19SDimitry Andric         OS << " in " << PrintReg(I->second, TRI);
45091bc56edSDimitry Andric       if (std::next(I) != E)
451f22ef01cSRoman Divacky         OS << ", ";
452f22ef01cSRoman Divacky     }
453f22ef01cSRoman Divacky     OS << '\n';
454f22ef01cSRoman Divacky   }
455f22ef01cSRoman Divacky 
4563dac3a9bSDimitry Andric   ModuleSlotTracker MST(getFunction()->getParent());
4573dac3a9bSDimitry Andric   MST.incorporateFunction(*getFunction());
45891bc56edSDimitry Andric   for (const auto &BB : *this) {
459f22ef01cSRoman Divacky     OS << '\n';
4603dac3a9bSDimitry Andric     BB.print(OS, MST, Indexes);
461f22ef01cSRoman Divacky   }
462f22ef01cSRoman Divacky 
4633861d79fSDimitry Andric   OS << "\n# End machine code for function " << getName() << ".\n\n";
464f22ef01cSRoman Divacky }
465f22ef01cSRoman Divacky 
466f22ef01cSRoman Divacky namespace llvm {
467f22ef01cSRoman Divacky   template<>
468f22ef01cSRoman Divacky   struct DOTGraphTraits<const MachineFunction*> : public DefaultDOTGraphTraits {
469f22ef01cSRoman Divacky 
470f22ef01cSRoman Divacky   DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {}
471f22ef01cSRoman Divacky 
472f22ef01cSRoman Divacky     static std::string getGraphName(const MachineFunction *F) {
473ff0cc061SDimitry Andric       return ("CFG for '" + F->getName() + "' function").str();
474f22ef01cSRoman Divacky     }
475f22ef01cSRoman Divacky 
476f22ef01cSRoman Divacky     std::string getNodeLabel(const MachineBasicBlock *Node,
477f22ef01cSRoman Divacky                              const MachineFunction *Graph) {
478f22ef01cSRoman Divacky       std::string OutStr;
479f22ef01cSRoman Divacky       {
480f22ef01cSRoman Divacky         raw_string_ostream OSS(OutStr);
481f22ef01cSRoman Divacky 
4822754fe60SDimitry Andric         if (isSimple()) {
4832754fe60SDimitry Andric           OSS << "BB#" << Node->getNumber();
4842754fe60SDimitry Andric           if (const BasicBlock *BB = Node->getBasicBlock())
4852754fe60SDimitry Andric             OSS << ": " << BB->getName();
4862754fe60SDimitry Andric         } else
487f22ef01cSRoman Divacky           Node->print(OSS);
488f22ef01cSRoman Divacky       }
489f22ef01cSRoman Divacky 
490f22ef01cSRoman Divacky       if (OutStr[0] == '\n') OutStr.erase(OutStr.begin());
491f22ef01cSRoman Divacky 
492f22ef01cSRoman Divacky       // Process string output to make it nicer...
493f22ef01cSRoman Divacky       for (unsigned i = 0; i != OutStr.length(); ++i)
494f22ef01cSRoman Divacky         if (OutStr[i] == '\n') {                            // Left justify
495f22ef01cSRoman Divacky           OutStr[i] = '\\';
496f22ef01cSRoman Divacky           OutStr.insert(OutStr.begin()+i+1, 'l');
497f22ef01cSRoman Divacky         }
498f22ef01cSRoman Divacky       return OutStr;
499f22ef01cSRoman Divacky     }
500f22ef01cSRoman Divacky   };
5013dac3a9bSDimitry Andric }
502f22ef01cSRoman Divacky 
503f22ef01cSRoman Divacky void MachineFunction::viewCFG() const
504f22ef01cSRoman Divacky {
505f22ef01cSRoman Divacky #ifndef NDEBUG
5063861d79fSDimitry Andric   ViewGraph(this, "mf" + getName());
507f22ef01cSRoman Divacky #else
508ffd1746dSEd Schouten   errs() << "MachineFunction::viewCFG is only available in debug builds on "
509f22ef01cSRoman Divacky          << "systems with Graphviz or gv!\n";
510f22ef01cSRoman Divacky #endif // NDEBUG
511f22ef01cSRoman Divacky }
512f22ef01cSRoman Divacky 
513f22ef01cSRoman Divacky void MachineFunction::viewCFGOnly() const
514f22ef01cSRoman Divacky {
515f22ef01cSRoman Divacky #ifndef NDEBUG
5163861d79fSDimitry Andric   ViewGraph(this, "mf" + getName(), true);
517f22ef01cSRoman Divacky #else
518ffd1746dSEd Schouten   errs() << "MachineFunction::viewCFGOnly is only available in debug builds on "
519f22ef01cSRoman Divacky          << "systems with Graphviz or gv!\n";
520f22ef01cSRoman Divacky #endif // NDEBUG
521f22ef01cSRoman Divacky }
522f22ef01cSRoman Divacky 
5238f0fd8f6SDimitry Andric /// Add the specified physical register as a live-in value and
524f22ef01cSRoman Divacky /// create a corresponding virtual register for it.
525f22ef01cSRoman Divacky unsigned MachineFunction::addLiveIn(unsigned PReg,
526dd6029ffSDimitry Andric                                     const TargetRegisterClass *RC) {
527f22ef01cSRoman Divacky   MachineRegisterInfo &MRI = getRegInfo();
528f22ef01cSRoman Divacky   unsigned VReg = MRI.getLiveInVirtReg(PReg);
529f22ef01cSRoman Divacky   if (VReg) {
53091bc56edSDimitry Andric     const TargetRegisterClass *VRegRC = MRI.getRegClass(VReg);
53191bc56edSDimitry Andric     (void)VRegRC;
53291bc56edSDimitry Andric     // A physical register can be added several times.
53391bc56edSDimitry Andric     // Between two calls, the register class of the related virtual register
53491bc56edSDimitry Andric     // may have been constrained to match some operation constraints.
53591bc56edSDimitry Andric     // In that case, check that the current register class includes the
53691bc56edSDimitry Andric     // physical register and is a sub class of the specified RC.
53791bc56edSDimitry Andric     assert((VRegRC == RC || (VRegRC->contains(PReg) &&
53891bc56edSDimitry Andric                              RC->hasSubClassEq(VRegRC))) &&
53991bc56edSDimitry Andric             "Register class mismatch!");
540f22ef01cSRoman Divacky     return VReg;
541f22ef01cSRoman Divacky   }
542f22ef01cSRoman Divacky   VReg = MRI.createVirtualRegister(RC);
543f22ef01cSRoman Divacky   MRI.addLiveIn(PReg, VReg);
544f22ef01cSRoman Divacky   return VReg;
545f22ef01cSRoman Divacky }
546f22ef01cSRoman Divacky 
5478f0fd8f6SDimitry Andric /// Return the MCSymbol for the specified non-empty jump table.
548f22ef01cSRoman Divacky /// If isLinkerPrivate is specified, an 'l' label is returned, otherwise a
549f22ef01cSRoman Divacky /// normal 'L' label is returned.
550f22ef01cSRoman Divacky MCSymbol *MachineFunction::getJTISymbol(unsigned JTI, MCContext &Ctx,
551f22ef01cSRoman Divacky                                         bool isLinkerPrivate) const {
552875ed548SDimitry Andric   const DataLayout &DL = getDataLayout();
553f22ef01cSRoman Divacky   assert(JumpTableInfo && "No jump tables");
554f22ef01cSRoman Divacky   assert(JTI < JumpTableInfo->getJumpTables().size() && "Invalid JTI!");
555f22ef01cSRoman Divacky 
556d88c1a5aSDimitry Andric   StringRef Prefix = isLinkerPrivate ? DL.getLinkerPrivateGlobalPrefix()
557875ed548SDimitry Andric                                      : DL.getPrivateGlobalPrefix();
558f22ef01cSRoman Divacky   SmallString<60> Name;
559f22ef01cSRoman Divacky   raw_svector_ostream(Name)
560f22ef01cSRoman Divacky     << Prefix << "JTI" << getFunctionNumber() << '_' << JTI;
561ff0cc061SDimitry Andric   return Ctx.getOrCreateSymbol(Name);
562f22ef01cSRoman Divacky }
563f22ef01cSRoman Divacky 
5648f0fd8f6SDimitry Andric /// Return a function-local symbol to represent the PIC base.
5652754fe60SDimitry Andric MCSymbol *MachineFunction::getPICBaseSymbol() const {
566875ed548SDimitry Andric   const DataLayout &DL = getDataLayout();
567875ed548SDimitry Andric   return Ctx.getOrCreateSymbol(Twine(DL.getPrivateGlobalPrefix()) +
5682754fe60SDimitry Andric                                Twine(getFunctionNumber()) + "$pb");
5692754fe60SDimitry Andric }
570f22ef01cSRoman Divacky 
571d88c1a5aSDimitry Andric /// \name Exception Handling
572d88c1a5aSDimitry Andric /// \{
573d88c1a5aSDimitry Andric 
574d88c1a5aSDimitry Andric LandingPadInfo &
575d88c1a5aSDimitry Andric MachineFunction::getOrCreateLandingPadInfo(MachineBasicBlock *LandingPad) {
576d88c1a5aSDimitry Andric   unsigned N = LandingPads.size();
577d88c1a5aSDimitry Andric   for (unsigned i = 0; i < N; ++i) {
578d88c1a5aSDimitry Andric     LandingPadInfo &LP = LandingPads[i];
579d88c1a5aSDimitry Andric     if (LP.LandingPadBlock == LandingPad)
580d88c1a5aSDimitry Andric       return LP;
581d88c1a5aSDimitry Andric   }
582d88c1a5aSDimitry Andric 
583d88c1a5aSDimitry Andric   LandingPads.push_back(LandingPadInfo(LandingPad));
584d88c1a5aSDimitry Andric   return LandingPads[N];
585d88c1a5aSDimitry Andric }
586d88c1a5aSDimitry Andric 
587d88c1a5aSDimitry Andric void MachineFunction::addInvoke(MachineBasicBlock *LandingPad,
588d88c1a5aSDimitry Andric                                 MCSymbol *BeginLabel, MCSymbol *EndLabel) {
589d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
590d88c1a5aSDimitry Andric   LP.BeginLabels.push_back(BeginLabel);
591d88c1a5aSDimitry Andric   LP.EndLabels.push_back(EndLabel);
592d88c1a5aSDimitry Andric }
593d88c1a5aSDimitry Andric 
594d88c1a5aSDimitry Andric MCSymbol *MachineFunction::addLandingPad(MachineBasicBlock *LandingPad) {
595d88c1a5aSDimitry Andric   MCSymbol *LandingPadLabel = Ctx.createTempSymbol();
596d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
597d88c1a5aSDimitry Andric   LP.LandingPadLabel = LandingPadLabel;
598d88c1a5aSDimitry Andric   return LandingPadLabel;
599d88c1a5aSDimitry Andric }
600d88c1a5aSDimitry Andric 
601d88c1a5aSDimitry Andric void MachineFunction::addCatchTypeInfo(MachineBasicBlock *LandingPad,
602d88c1a5aSDimitry Andric                                        ArrayRef<const GlobalValue *> TyInfo) {
603d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
604d88c1a5aSDimitry Andric   for (unsigned N = TyInfo.size(); N; --N)
605d88c1a5aSDimitry Andric     LP.TypeIds.push_back(getTypeIDFor(TyInfo[N - 1]));
606d88c1a5aSDimitry Andric }
607d88c1a5aSDimitry Andric 
608d88c1a5aSDimitry Andric void MachineFunction::addFilterTypeInfo(MachineBasicBlock *LandingPad,
609d88c1a5aSDimitry Andric                                         ArrayRef<const GlobalValue *> TyInfo) {
610d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
611d88c1a5aSDimitry Andric   std::vector<unsigned> IdsInFilter(TyInfo.size());
612d88c1a5aSDimitry Andric   for (unsigned I = 0, E = TyInfo.size(); I != E; ++I)
613d88c1a5aSDimitry Andric     IdsInFilter[I] = getTypeIDFor(TyInfo[I]);
614d88c1a5aSDimitry Andric   LP.TypeIds.push_back(getFilterIDFor(IdsInFilter));
615d88c1a5aSDimitry Andric }
616d88c1a5aSDimitry Andric 
617d88c1a5aSDimitry Andric void MachineFunction::tidyLandingPads(DenseMap<MCSymbol*, uintptr_t> *LPMap) {
618d88c1a5aSDimitry Andric   for (unsigned i = 0; i != LandingPads.size(); ) {
619d88c1a5aSDimitry Andric     LandingPadInfo &LandingPad = LandingPads[i];
620d88c1a5aSDimitry Andric     if (LandingPad.LandingPadLabel &&
621d88c1a5aSDimitry Andric         !LandingPad.LandingPadLabel->isDefined() &&
622d88c1a5aSDimitry Andric         (!LPMap || (*LPMap)[LandingPad.LandingPadLabel] == 0))
623d88c1a5aSDimitry Andric       LandingPad.LandingPadLabel = nullptr;
624d88c1a5aSDimitry Andric 
625d88c1a5aSDimitry Andric     // Special case: we *should* emit LPs with null LP MBB. This indicates
626d88c1a5aSDimitry Andric     // "nounwind" case.
627d88c1a5aSDimitry Andric     if (!LandingPad.LandingPadLabel && LandingPad.LandingPadBlock) {
628d88c1a5aSDimitry Andric       LandingPads.erase(LandingPads.begin() + i);
629d88c1a5aSDimitry Andric       continue;
630d88c1a5aSDimitry Andric     }
631d88c1a5aSDimitry Andric 
632d88c1a5aSDimitry Andric     for (unsigned j = 0, e = LandingPads[i].BeginLabels.size(); j != e; ++j) {
633d88c1a5aSDimitry Andric       MCSymbol *BeginLabel = LandingPad.BeginLabels[j];
634d88c1a5aSDimitry Andric       MCSymbol *EndLabel = LandingPad.EndLabels[j];
635d88c1a5aSDimitry Andric       if ((BeginLabel->isDefined() ||
636d88c1a5aSDimitry Andric            (LPMap && (*LPMap)[BeginLabel] != 0)) &&
637d88c1a5aSDimitry Andric           (EndLabel->isDefined() ||
638d88c1a5aSDimitry Andric            (LPMap && (*LPMap)[EndLabel] != 0))) continue;
639d88c1a5aSDimitry Andric 
640d88c1a5aSDimitry Andric       LandingPad.BeginLabels.erase(LandingPad.BeginLabels.begin() + j);
641d88c1a5aSDimitry Andric       LandingPad.EndLabels.erase(LandingPad.EndLabels.begin() + j);
642d88c1a5aSDimitry Andric       --j;
643d88c1a5aSDimitry Andric       --e;
644d88c1a5aSDimitry Andric     }
645d88c1a5aSDimitry Andric 
646d88c1a5aSDimitry Andric     // Remove landing pads with no try-ranges.
647d88c1a5aSDimitry Andric     if (LandingPads[i].BeginLabels.empty()) {
648d88c1a5aSDimitry Andric       LandingPads.erase(LandingPads.begin() + i);
649d88c1a5aSDimitry Andric       continue;
650d88c1a5aSDimitry Andric     }
651d88c1a5aSDimitry Andric 
652d88c1a5aSDimitry Andric     // If there is no landing pad, ensure that the list of typeids is empty.
653d88c1a5aSDimitry Andric     // If the only typeid is a cleanup, this is the same as having no typeids.
654d88c1a5aSDimitry Andric     if (!LandingPad.LandingPadBlock ||
655d88c1a5aSDimitry Andric         (LandingPad.TypeIds.size() == 1 && !LandingPad.TypeIds[0]))
656d88c1a5aSDimitry Andric       LandingPad.TypeIds.clear();
657d88c1a5aSDimitry Andric     ++i;
658d88c1a5aSDimitry Andric   }
659d88c1a5aSDimitry Andric }
660d88c1a5aSDimitry Andric 
661d88c1a5aSDimitry Andric void MachineFunction::addCleanup(MachineBasicBlock *LandingPad) {
662d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
663d88c1a5aSDimitry Andric   LP.TypeIds.push_back(0);
664d88c1a5aSDimitry Andric }
665d88c1a5aSDimitry Andric 
666d88c1a5aSDimitry Andric void MachineFunction::addSEHCatchHandler(MachineBasicBlock *LandingPad,
667d88c1a5aSDimitry Andric                                          const Function *Filter,
668d88c1a5aSDimitry Andric                                          const BlockAddress *RecoverBA) {
669d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
670d88c1a5aSDimitry Andric   SEHHandler Handler;
671d88c1a5aSDimitry Andric   Handler.FilterOrFinally = Filter;
672d88c1a5aSDimitry Andric   Handler.RecoverBA = RecoverBA;
673d88c1a5aSDimitry Andric   LP.SEHHandlers.push_back(Handler);
674d88c1a5aSDimitry Andric }
675d88c1a5aSDimitry Andric 
676d88c1a5aSDimitry Andric void MachineFunction::addSEHCleanupHandler(MachineBasicBlock *LandingPad,
677d88c1a5aSDimitry Andric                                            const Function *Cleanup) {
678d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
679d88c1a5aSDimitry Andric   SEHHandler Handler;
680d88c1a5aSDimitry Andric   Handler.FilterOrFinally = Cleanup;
681d88c1a5aSDimitry Andric   Handler.RecoverBA = nullptr;
682d88c1a5aSDimitry Andric   LP.SEHHandlers.push_back(Handler);
683d88c1a5aSDimitry Andric }
684d88c1a5aSDimitry Andric 
685d88c1a5aSDimitry Andric void MachineFunction::setCallSiteLandingPad(MCSymbol *Sym,
686d88c1a5aSDimitry Andric                                             ArrayRef<unsigned> Sites) {
687d88c1a5aSDimitry Andric   LPadToCallSiteMap[Sym].append(Sites.begin(), Sites.end());
688d88c1a5aSDimitry Andric }
689d88c1a5aSDimitry Andric 
690d88c1a5aSDimitry Andric unsigned MachineFunction::getTypeIDFor(const GlobalValue *TI) {
691d88c1a5aSDimitry Andric   for (unsigned i = 0, N = TypeInfos.size(); i != N; ++i)
692d88c1a5aSDimitry Andric     if (TypeInfos[i] == TI) return i + 1;
693d88c1a5aSDimitry Andric 
694d88c1a5aSDimitry Andric   TypeInfos.push_back(TI);
695d88c1a5aSDimitry Andric   return TypeInfos.size();
696d88c1a5aSDimitry Andric }
697d88c1a5aSDimitry Andric 
698d88c1a5aSDimitry Andric int MachineFunction::getFilterIDFor(std::vector<unsigned> &TyIds) {
699d88c1a5aSDimitry Andric   // If the new filter coincides with the tail of an existing filter, then
700d88c1a5aSDimitry Andric   // re-use the existing filter.  Folding filters more than this requires
701d88c1a5aSDimitry Andric   // re-ordering filters and/or their elements - probably not worth it.
702d88c1a5aSDimitry Andric   for (std::vector<unsigned>::iterator I = FilterEnds.begin(),
703d88c1a5aSDimitry Andric        E = FilterEnds.end(); I != E; ++I) {
704d88c1a5aSDimitry Andric     unsigned i = *I, j = TyIds.size();
705d88c1a5aSDimitry Andric 
706d88c1a5aSDimitry Andric     while (i && j)
707d88c1a5aSDimitry Andric       if (FilterIds[--i] != TyIds[--j])
708d88c1a5aSDimitry Andric         goto try_next;
709d88c1a5aSDimitry Andric 
710d88c1a5aSDimitry Andric     if (!j)
711d88c1a5aSDimitry Andric       // The new filter coincides with range [i, end) of the existing filter.
712d88c1a5aSDimitry Andric       return -(1 + i);
713d88c1a5aSDimitry Andric 
714d88c1a5aSDimitry Andric try_next:;
715d88c1a5aSDimitry Andric   }
716d88c1a5aSDimitry Andric 
717d88c1a5aSDimitry Andric   // Add the new filter.
718d88c1a5aSDimitry Andric   int FilterID = -(1 + FilterIds.size());
719d88c1a5aSDimitry Andric   FilterIds.reserve(FilterIds.size() + TyIds.size() + 1);
720d88c1a5aSDimitry Andric   FilterIds.insert(FilterIds.end(), TyIds.begin(), TyIds.end());
721d88c1a5aSDimitry Andric   FilterEnds.push_back(FilterIds.size());
722d88c1a5aSDimitry Andric   FilterIds.push_back(0); // terminator
723d88c1a5aSDimitry Andric   return FilterID;
724d88c1a5aSDimitry Andric }
725d88c1a5aSDimitry Andric 
726d88c1a5aSDimitry Andric void llvm::addLandingPadInfo(const LandingPadInst &I, MachineBasicBlock &MBB) {
727d88c1a5aSDimitry Andric   MachineFunction &MF = *MBB.getParent();
728d88c1a5aSDimitry Andric   if (const auto *PF = dyn_cast<Function>(
729d88c1a5aSDimitry Andric           I.getParent()->getParent()->getPersonalityFn()->stripPointerCasts()))
730d88c1a5aSDimitry Andric     MF.getMMI().addPersonality(PF);
731d88c1a5aSDimitry Andric 
732d88c1a5aSDimitry Andric   if (I.isCleanup())
733d88c1a5aSDimitry Andric     MF.addCleanup(&MBB);
734d88c1a5aSDimitry Andric 
735d88c1a5aSDimitry Andric   // FIXME: New EH - Add the clauses in reverse order. This isn't 100% correct,
736d88c1a5aSDimitry Andric   //        but we need to do it this way because of how the DWARF EH emitter
737d88c1a5aSDimitry Andric   //        processes the clauses.
738d88c1a5aSDimitry Andric   for (unsigned i = I.getNumClauses(); i != 0; --i) {
739d88c1a5aSDimitry Andric     Value *Val = I.getClause(i - 1);
740d88c1a5aSDimitry Andric     if (I.isCatch(i - 1)) {
741d88c1a5aSDimitry Andric       MF.addCatchTypeInfo(&MBB,
742d88c1a5aSDimitry Andric                           dyn_cast<GlobalValue>(Val->stripPointerCasts()));
743d88c1a5aSDimitry Andric     } else {
744d88c1a5aSDimitry Andric       // Add filters in a list.
745d88c1a5aSDimitry Andric       Constant *CVal = cast<Constant>(Val);
746d88c1a5aSDimitry Andric       SmallVector<const GlobalValue *, 4> FilterList;
747d88c1a5aSDimitry Andric       for (User::op_iterator II = CVal->op_begin(), IE = CVal->op_end();
748d88c1a5aSDimitry Andric            II != IE; ++II)
749d88c1a5aSDimitry Andric         FilterList.push_back(cast<GlobalValue>((*II)->stripPointerCasts()));
750d88c1a5aSDimitry Andric 
751d88c1a5aSDimitry Andric       MF.addFilterTypeInfo(&MBB, FilterList);
752d88c1a5aSDimitry Andric     }
753d88c1a5aSDimitry Andric   }
754d88c1a5aSDimitry Andric }
755d88c1a5aSDimitry Andric 
756d88c1a5aSDimitry Andric /// \}
757d88c1a5aSDimitry Andric 
758f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
759f22ef01cSRoman Divacky //  MachineFrameInfo implementation
760f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
761f22ef01cSRoman Divacky 
7628f0fd8f6SDimitry Andric /// Make sure the function is at least Align bytes aligned.
763139f7f9bSDimitry Andric void MachineFrameInfo::ensureMaxAlignment(unsigned Align) {
7643ca95b02SDimitry Andric   if (!StackRealignable)
76539d628a0SDimitry Andric     assert(Align <= StackAlignment &&
766139f7f9bSDimitry Andric            "For targets without stack realignment, Align is out of limit!");
767139f7f9bSDimitry Andric   if (MaxAlignment < Align) MaxAlignment = Align;
768139f7f9bSDimitry Andric }
769139f7f9bSDimitry Andric 
7708f0fd8f6SDimitry Andric /// Clamp the alignment if requested and emit a warning.
771139f7f9bSDimitry Andric static inline unsigned clampStackAlignment(bool ShouldClamp, unsigned Align,
772139f7f9bSDimitry Andric                                            unsigned StackAlign) {
773139f7f9bSDimitry Andric   if (!ShouldClamp || Align <= StackAlign)
774139f7f9bSDimitry Andric     return Align;
775139f7f9bSDimitry Andric   DEBUG(dbgs() << "Warning: requested alignment " << Align
776139f7f9bSDimitry Andric                << " exceeds the stack alignment " << StackAlign
777139f7f9bSDimitry Andric                << " when stack realignment is off" << '\n');
778139f7f9bSDimitry Andric   return StackAlign;
779139f7f9bSDimitry Andric }
780139f7f9bSDimitry Andric 
7818f0fd8f6SDimitry Andric /// Create a new statically sized stack object, returning a nonnegative
7828f0fd8f6SDimitry Andric /// identifier to represent it.
783139f7f9bSDimitry Andric int MachineFrameInfo::CreateStackObject(uint64_t Size, unsigned Alignment,
78491bc56edSDimitry Andric                       bool isSS, const AllocaInst *Alloca) {
785139f7f9bSDimitry Andric   assert(Size != 0 && "Cannot allocate zero size stack objects!");
7863ca95b02SDimitry Andric   Alignment = clampStackAlignment(!StackRealignable, Alignment, StackAlignment);
78739d628a0SDimitry Andric   Objects.push_back(StackObject(Size, Alignment, 0, false, isSS, Alloca,
78839d628a0SDimitry Andric                                 !isSS));
789139f7f9bSDimitry Andric   int Index = (int)Objects.size() - NumFixedObjects - 1;
790139f7f9bSDimitry Andric   assert(Index >= 0 && "Bad frame index!");
791139f7f9bSDimitry Andric   ensureMaxAlignment(Alignment);
792139f7f9bSDimitry Andric   return Index;
793139f7f9bSDimitry Andric }
794139f7f9bSDimitry Andric 
7958f0fd8f6SDimitry Andric /// Create a new statically sized stack object that represents a spill slot,
7968f0fd8f6SDimitry Andric /// returning a nonnegative identifier to represent it.
797139f7f9bSDimitry Andric int MachineFrameInfo::CreateSpillStackObject(uint64_t Size,
798139f7f9bSDimitry Andric                                              unsigned Alignment) {
7993ca95b02SDimitry Andric   Alignment = clampStackAlignment(!StackRealignable, Alignment, StackAlignment);
80091bc56edSDimitry Andric   CreateStackObject(Size, Alignment, true);
801139f7f9bSDimitry Andric   int Index = (int)Objects.size() - NumFixedObjects - 1;
802139f7f9bSDimitry Andric   ensureMaxAlignment(Alignment);
803139f7f9bSDimitry Andric   return Index;
804139f7f9bSDimitry Andric }
805139f7f9bSDimitry Andric 
8068f0fd8f6SDimitry Andric /// Notify the MachineFrameInfo object that a variable sized object has been
8078f0fd8f6SDimitry Andric /// created. This must be created whenever a variable sized object is created,
8088f0fd8f6SDimitry Andric /// whether or not the index returned is actually used.
809*4f00c8c6SDimitry Andric int MachineFrameInfo::CreateVariableSizedObject(unsigned Alignment,
810*4f00c8c6SDimitry Andric                                                 const AllocaInst *Alloca) {
811139f7f9bSDimitry Andric   HasVarSizedObjects = true;
8123ca95b02SDimitry Andric   Alignment = clampStackAlignment(!StackRealignable, Alignment, StackAlignment);
81339d628a0SDimitry Andric   Objects.push_back(StackObject(0, Alignment, 0, false, false, Alloca, true));
814139f7f9bSDimitry Andric   ensureMaxAlignment(Alignment);
815139f7f9bSDimitry Andric   return (int)Objects.size()-NumFixedObjects-1;
816139f7f9bSDimitry Andric }
817139f7f9bSDimitry Andric 
8188f0fd8f6SDimitry Andric /// Create a new object at a fixed location on the stack.
819f22ef01cSRoman Divacky /// All fixed objects should be created before other objects are created for
820f22ef01cSRoman Divacky /// efficiency. By default, fixed objects are immutable. This returns an
821f22ef01cSRoman Divacky /// index with a negative value.
822f22ef01cSRoman Divacky int MachineFrameInfo::CreateFixedObject(uint64_t Size, int64_t SPOffset,
82339d628a0SDimitry Andric                                         bool Immutable, bool isAliased) {
824f22ef01cSRoman Divacky   assert(Size != 0 && "Cannot allocate zero size fixed stack objects!");
825ffd1746dSEd Schouten   // The alignment of the frame index can be determined from its offset from
826ffd1746dSEd Schouten   // the incoming frame position.  If the frame object is at offset 32 and
827ffd1746dSEd Schouten   // the stack is guaranteed to be 16-byte aligned, then we know that the
8283ca95b02SDimitry Andric   // object is 16-byte aligned. Note that unlike the non-fixed case, if the
8293ca95b02SDimitry Andric   // stack needs realignment, we can't assume that the stack will in fact be
8303ca95b02SDimitry Andric   // aligned.
8313ca95b02SDimitry Andric   unsigned Align = MinAlign(SPOffset, ForcedRealign ? 1 : StackAlignment);
8323ca95b02SDimitry Andric   Align = clampStackAlignment(!StackRealignable, Align, StackAlignment);
833ffd1746dSEd Schouten   Objects.insert(Objects.begin(), StackObject(Size, Align, SPOffset, Immutable,
8343861d79fSDimitry Andric                                               /*isSS*/   false,
83539d628a0SDimitry Andric                                               /*Alloca*/ nullptr, isAliased));
836f22ef01cSRoman Divacky   return -++NumFixedObjects;
837f22ef01cSRoman Divacky }
838f22ef01cSRoman Divacky 
8398f0fd8f6SDimitry Andric /// Create a spill slot at a fixed location on the stack.
8408f0fd8f6SDimitry Andric /// Returns an index with a negative value.
84191bc56edSDimitry Andric int MachineFrameInfo::CreateFixedSpillStackObject(uint64_t Size,
842d88c1a5aSDimitry Andric                                                   int64_t SPOffset,
843d88c1a5aSDimitry Andric                                                   bool Immutable) {
8443ca95b02SDimitry Andric   unsigned Align = MinAlign(SPOffset, ForcedRealign ? 1 : StackAlignment);
8453ca95b02SDimitry Andric   Align = clampStackAlignment(!StackRealignable, Align, StackAlignment);
846d88c1a5aSDimitry Andric   Objects.insert(Objects.begin(), StackObject(Size, Align, SPOffset, Immutable,
84791bc56edSDimitry Andric                                               /*isSS*/ true,
84839d628a0SDimitry Andric                                               /*Alloca*/ nullptr,
84939d628a0SDimitry Andric                                               /*isAliased*/ false));
85091bc56edSDimitry Andric   return -++NumFixedObjects;
85191bc56edSDimitry Andric }
852f22ef01cSRoman Divacky 
85397bc6c73SDimitry Andric BitVector MachineFrameInfo::getPristineRegs(const MachineFunction &MF) const {
85497bc6c73SDimitry Andric   const TargetRegisterInfo *TRI = MF.getSubtarget().getRegisterInfo();
855f22ef01cSRoman Divacky   BitVector BV(TRI->getNumRegs());
856f22ef01cSRoman Divacky 
857f22ef01cSRoman Divacky   // Before CSI is calculated, no registers are considered pristine. They can be
858f22ef01cSRoman Divacky   // freely used and PEI will make sure they are saved.
859f22ef01cSRoman Divacky   if (!isCalleeSavedInfoValid())
860f22ef01cSRoman Divacky     return BV;
861f22ef01cSRoman Divacky 
86297bc6c73SDimitry Andric   for (const MCPhysReg *CSR = TRI->getCalleeSavedRegs(&MF); CSR && *CSR; ++CSR)
863f22ef01cSRoman Divacky     BV.set(*CSR);
864f22ef01cSRoman Divacky 
86597bc6c73SDimitry Andric   // Saved CSRs are not pristine.
8667d523365SDimitry Andric   for (auto &I : getCalleeSavedInfo())
8677d523365SDimitry Andric     for (MCSubRegIterator S(I.getReg(), TRI, true); S.isValid(); ++S)
8687d523365SDimitry Andric       BV.reset(*S);
869f22ef01cSRoman Divacky 
870f22ef01cSRoman Divacky   return BV;
871f22ef01cSRoman Divacky }
872f22ef01cSRoman Divacky 
873139f7f9bSDimitry Andric unsigned MachineFrameInfo::estimateStackSize(const MachineFunction &MF) const {
87439d628a0SDimitry Andric   const TargetFrameLowering *TFI = MF.getSubtarget().getFrameLowering();
87539d628a0SDimitry Andric   const TargetRegisterInfo *RegInfo = MF.getSubtarget().getRegisterInfo();
876139f7f9bSDimitry Andric   unsigned MaxAlign = getMaxAlignment();
877139f7f9bSDimitry Andric   int Offset = 0;
878139f7f9bSDimitry Andric 
879139f7f9bSDimitry Andric   // This code is very, very similar to PEI::calculateFrameObjectOffsets().
880139f7f9bSDimitry Andric   // It really should be refactored to share code. Until then, changes
881139f7f9bSDimitry Andric   // should keep in mind that there's tight coupling between the two.
882139f7f9bSDimitry Andric 
883139f7f9bSDimitry Andric   for (int i = getObjectIndexBegin(); i != 0; ++i) {
884139f7f9bSDimitry Andric     int FixedOff = -getObjectOffset(i);
885139f7f9bSDimitry Andric     if (FixedOff > Offset) Offset = FixedOff;
886139f7f9bSDimitry Andric   }
887139f7f9bSDimitry Andric   for (unsigned i = 0, e = getObjectIndexEnd(); i != e; ++i) {
888139f7f9bSDimitry Andric     if (isDeadObjectIndex(i))
889139f7f9bSDimitry Andric       continue;
890139f7f9bSDimitry Andric     Offset += getObjectSize(i);
891139f7f9bSDimitry Andric     unsigned Align = getObjectAlignment(i);
892139f7f9bSDimitry Andric     // Adjust to alignment boundary
893139f7f9bSDimitry Andric     Offset = (Offset+Align-1)/Align*Align;
894139f7f9bSDimitry Andric 
895139f7f9bSDimitry Andric     MaxAlign = std::max(Align, MaxAlign);
896139f7f9bSDimitry Andric   }
897139f7f9bSDimitry Andric 
898139f7f9bSDimitry Andric   if (adjustsStack() && TFI->hasReservedCallFrame(MF))
899139f7f9bSDimitry Andric     Offset += getMaxCallFrameSize();
900139f7f9bSDimitry Andric 
901139f7f9bSDimitry Andric   // Round up the size to a multiple of the alignment.  If the function has
902139f7f9bSDimitry Andric   // any calls or alloca's, align to the target's StackAlignment value to
903139f7f9bSDimitry Andric   // ensure that the callee's frame or the alloca data is suitably aligned;
904139f7f9bSDimitry Andric   // otherwise, for leaf functions, align to the TransientStackAlignment
905139f7f9bSDimitry Andric   // value.
906139f7f9bSDimitry Andric   unsigned StackAlign;
907139f7f9bSDimitry Andric   if (adjustsStack() || hasVarSizedObjects() ||
908139f7f9bSDimitry Andric       (RegInfo->needsStackRealignment(MF) && getObjectIndexEnd() != 0))
909139f7f9bSDimitry Andric     StackAlign = TFI->getStackAlignment();
910139f7f9bSDimitry Andric   else
911139f7f9bSDimitry Andric     StackAlign = TFI->getTransientStackAlignment();
912139f7f9bSDimitry Andric 
913139f7f9bSDimitry Andric   // If the frame pointer is eliminated, all frame offsets will be relative to
914139f7f9bSDimitry Andric   // SP not FP. Align to MaxAlign so this works.
915139f7f9bSDimitry Andric   StackAlign = std::max(StackAlign, MaxAlign);
916139f7f9bSDimitry Andric   unsigned AlignMask = StackAlign - 1;
917139f7f9bSDimitry Andric   Offset = (Offset + AlignMask) & ~uint64_t(AlignMask);
918139f7f9bSDimitry Andric 
919139f7f9bSDimitry Andric   return (unsigned)Offset;
920139f7f9bSDimitry Andric }
921f22ef01cSRoman Divacky 
922f22ef01cSRoman Divacky void MachineFrameInfo::print(const MachineFunction &MF, raw_ostream &OS) const{
923f22ef01cSRoman Divacky   if (Objects.empty()) return;
924f22ef01cSRoman Divacky 
92539d628a0SDimitry Andric   const TargetFrameLowering *FI = MF.getSubtarget().getFrameLowering();
926f22ef01cSRoman Divacky   int ValOffset = (FI ? FI->getOffsetOfLocalArea() : 0);
927f22ef01cSRoman Divacky 
928f22ef01cSRoman Divacky   OS << "Frame Objects:\n";
929f22ef01cSRoman Divacky 
930f22ef01cSRoman Divacky   for (unsigned i = 0, e = Objects.size(); i != e; ++i) {
931f22ef01cSRoman Divacky     const StackObject &SO = Objects[i];
932f22ef01cSRoman Divacky     OS << "  fi#" << (int)(i-NumFixedObjects) << ": ";
933f22ef01cSRoman Divacky     if (SO.Size == ~0ULL) {
934f22ef01cSRoman Divacky       OS << "dead\n";
935f22ef01cSRoman Divacky       continue;
936f22ef01cSRoman Divacky     }
937f22ef01cSRoman Divacky     if (SO.Size == 0)
938f22ef01cSRoman Divacky       OS << "variable sized";
939f22ef01cSRoman Divacky     else
940f22ef01cSRoman Divacky       OS << "size=" << SO.Size;
941f22ef01cSRoman Divacky     OS << ", align=" << SO.Alignment;
942f22ef01cSRoman Divacky 
943f22ef01cSRoman Divacky     if (i < NumFixedObjects)
944f22ef01cSRoman Divacky       OS << ", fixed";
945f22ef01cSRoman Divacky     if (i < NumFixedObjects || SO.SPOffset != -1) {
946f22ef01cSRoman Divacky       int64_t Off = SO.SPOffset - ValOffset;
947f22ef01cSRoman Divacky       OS << ", at location [SP";
948f22ef01cSRoman Divacky       if (Off > 0)
949f22ef01cSRoman Divacky         OS << "+" << Off;
950f22ef01cSRoman Divacky       else if (Off < 0)
951f22ef01cSRoman Divacky         OS << Off;
952f22ef01cSRoman Divacky       OS << "]";
953f22ef01cSRoman Divacky     }
954f22ef01cSRoman Divacky     OS << "\n";
955f22ef01cSRoman Divacky   }
956f22ef01cSRoman Divacky }
957f22ef01cSRoman Divacky 
9583861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
959f22ef01cSRoman Divacky void MachineFrameInfo::dump(const MachineFunction &MF) const {
960f22ef01cSRoman Divacky   print(MF, dbgs());
961f22ef01cSRoman Divacky }
9623861d79fSDimitry Andric #endif
963f22ef01cSRoman Divacky 
964f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
965f22ef01cSRoman Divacky //  MachineJumpTableInfo implementation
966f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
967f22ef01cSRoman Divacky 
9688f0fd8f6SDimitry Andric /// Return the size of each entry in the jump table.
9693861d79fSDimitry Andric unsigned MachineJumpTableInfo::getEntrySize(const DataLayout &TD) const {
970f22ef01cSRoman Divacky   // The size of a jump table entry is 4 bytes unless the entry is just the
971f22ef01cSRoman Divacky   // address of a block, in which case it is the pointer size.
972f22ef01cSRoman Divacky   switch (getEntryKind()) {
973f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_BlockAddress:
974f22ef01cSRoman Divacky     return TD.getPointerSize();
975dff0c46cSDimitry Andric   case MachineJumpTableInfo::EK_GPRel64BlockAddress:
976dff0c46cSDimitry Andric     return 8;
977f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_GPRel32BlockAddress:
978f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_LabelDifference32:
979f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_Custom32:
980f22ef01cSRoman Divacky     return 4;
981f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_Inline:
982f22ef01cSRoman Divacky     return 0;
983f22ef01cSRoman Divacky   }
984dff0c46cSDimitry Andric   llvm_unreachable("Unknown jump table encoding!");
985f22ef01cSRoman Divacky }
986f22ef01cSRoman Divacky 
9878f0fd8f6SDimitry Andric /// Return the alignment of each entry in the jump table.
9883861d79fSDimitry Andric unsigned MachineJumpTableInfo::getEntryAlignment(const DataLayout &TD) const {
989f22ef01cSRoman Divacky   // The alignment of a jump table entry is the alignment of int32 unless the
990f22ef01cSRoman Divacky   // entry is just the address of a block, in which case it is the pointer
991f22ef01cSRoman Divacky   // alignment.
992f22ef01cSRoman Divacky   switch (getEntryKind()) {
993f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_BlockAddress:
994f22ef01cSRoman Divacky     return TD.getPointerABIAlignment();
995dff0c46cSDimitry Andric   case MachineJumpTableInfo::EK_GPRel64BlockAddress:
996dff0c46cSDimitry Andric     return TD.getABIIntegerTypeAlignment(64);
997f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_GPRel32BlockAddress:
998f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_LabelDifference32:
999f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_Custom32:
1000f22ef01cSRoman Divacky     return TD.getABIIntegerTypeAlignment(32);
1001f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_Inline:
1002f22ef01cSRoman Divacky     return 1;
1003f22ef01cSRoman Divacky   }
1004dff0c46cSDimitry Andric   llvm_unreachable("Unknown jump table encoding!");
1005f22ef01cSRoman Divacky }
1006f22ef01cSRoman Divacky 
10078f0fd8f6SDimitry Andric /// Create a new jump table entry in the jump table info.
1008f22ef01cSRoman Divacky unsigned MachineJumpTableInfo::createJumpTableIndex(
1009f22ef01cSRoman Divacky                                const std::vector<MachineBasicBlock*> &DestBBs) {
1010f22ef01cSRoman Divacky   assert(!DestBBs.empty() && "Cannot create an empty jump table!");
1011f22ef01cSRoman Divacky   JumpTables.push_back(MachineJumpTableEntry(DestBBs));
1012f22ef01cSRoman Divacky   return JumpTables.size()-1;
1013f22ef01cSRoman Divacky }
1014f22ef01cSRoman Divacky 
10158f0fd8f6SDimitry Andric /// If Old is the target of any jump tables, update the jump tables to branch
10168f0fd8f6SDimitry Andric /// to New instead.
1017f22ef01cSRoman Divacky bool MachineJumpTableInfo::ReplaceMBBInJumpTables(MachineBasicBlock *Old,
1018f22ef01cSRoman Divacky                                                   MachineBasicBlock *New) {
1019f22ef01cSRoman Divacky   assert(Old != New && "Not making a change?");
1020f22ef01cSRoman Divacky   bool MadeChange = false;
1021f22ef01cSRoman Divacky   for (size_t i = 0, e = JumpTables.size(); i != e; ++i)
1022f22ef01cSRoman Divacky     ReplaceMBBInJumpTable(i, Old, New);
1023f22ef01cSRoman Divacky   return MadeChange;
1024f22ef01cSRoman Divacky }
1025f22ef01cSRoman Divacky 
10268f0fd8f6SDimitry Andric /// If Old is a target of the jump tables, update the jump table to branch to
10278f0fd8f6SDimitry Andric /// New instead.
1028f22ef01cSRoman Divacky bool MachineJumpTableInfo::ReplaceMBBInJumpTable(unsigned Idx,
1029f22ef01cSRoman Divacky                                                  MachineBasicBlock *Old,
1030f22ef01cSRoman Divacky                                                  MachineBasicBlock *New) {
1031f22ef01cSRoman Divacky   assert(Old != New && "Not making a change?");
1032f22ef01cSRoman Divacky   bool MadeChange = false;
1033f22ef01cSRoman Divacky   MachineJumpTableEntry &JTE = JumpTables[Idx];
1034f22ef01cSRoman Divacky   for (size_t j = 0, e = JTE.MBBs.size(); j != e; ++j)
1035f22ef01cSRoman Divacky     if (JTE.MBBs[j] == Old) {
1036f22ef01cSRoman Divacky       JTE.MBBs[j] = New;
1037f22ef01cSRoman Divacky       MadeChange = true;
1038f22ef01cSRoman Divacky     }
1039f22ef01cSRoman Divacky   return MadeChange;
1040f22ef01cSRoman Divacky }
1041f22ef01cSRoman Divacky 
1042f22ef01cSRoman Divacky void MachineJumpTableInfo::print(raw_ostream &OS) const {
1043f22ef01cSRoman Divacky   if (JumpTables.empty()) return;
1044f22ef01cSRoman Divacky 
1045f22ef01cSRoman Divacky   OS << "Jump Tables:\n";
1046f22ef01cSRoman Divacky 
1047f22ef01cSRoman Divacky   for (unsigned i = 0, e = JumpTables.size(); i != e; ++i) {
1048f22ef01cSRoman Divacky     OS << "  jt#" << i << ": ";
1049f22ef01cSRoman Divacky     for (unsigned j = 0, f = JumpTables[i].MBBs.size(); j != f; ++j)
1050f22ef01cSRoman Divacky       OS << " BB#" << JumpTables[i].MBBs[j]->getNumber();
1051f22ef01cSRoman Divacky   }
1052f22ef01cSRoman Divacky 
1053f22ef01cSRoman Divacky   OS << '\n';
1054f22ef01cSRoman Divacky }
1055f22ef01cSRoman Divacky 
10563861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
10573ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineJumpTableInfo::dump() const { print(dbgs()); }
10583861d79fSDimitry Andric #endif
1059f22ef01cSRoman Divacky 
1060f22ef01cSRoman Divacky 
1061f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
1062f22ef01cSRoman Divacky //  MachineConstantPool implementation
1063f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
1064f22ef01cSRoman Divacky 
1065dff0c46cSDimitry Andric void MachineConstantPoolValue::anchor() { }
1066dff0c46cSDimitry Andric 
10676122f3e6SDimitry Andric Type *MachineConstantPoolEntry::getType() const {
1068f22ef01cSRoman Divacky   if (isMachineConstantPoolEntry())
1069f22ef01cSRoman Divacky     return Val.MachineCPVal->getType();
1070f22ef01cSRoman Divacky   return Val.ConstVal->getType();
1071f22ef01cSRoman Divacky }
1072f22ef01cSRoman Divacky 
10737d523365SDimitry Andric bool MachineConstantPoolEntry::needsRelocation() const {
1074f22ef01cSRoman Divacky   if (isMachineConstantPoolEntry())
10757d523365SDimitry Andric     return true;
10767d523365SDimitry Andric   return Val.ConstVal->needsRelocation();
1077f22ef01cSRoman Divacky }
1078f22ef01cSRoman Divacky 
107991bc56edSDimitry Andric SectionKind
108091bc56edSDimitry Andric MachineConstantPoolEntry::getSectionKind(const DataLayout *DL) const {
10817d523365SDimitry Andric   if (needsRelocation())
10827d523365SDimitry Andric     return SectionKind::getReadOnlyWithRel();
108391bc56edSDimitry Andric   switch (DL->getTypeAllocSize(getType())) {
108491bc56edSDimitry Andric   case 4:
10857d523365SDimitry Andric     return SectionKind::getMergeableConst4();
108691bc56edSDimitry Andric   case 8:
10877d523365SDimitry Andric     return SectionKind::getMergeableConst8();
108891bc56edSDimitry Andric   case 16:
10897d523365SDimitry Andric     return SectionKind::getMergeableConst16();
10903ca95b02SDimitry Andric   case 32:
10913ca95b02SDimitry Andric     return SectionKind::getMergeableConst32();
109291bc56edSDimitry Andric   default:
10937d523365SDimitry Andric     return SectionKind::getReadOnly();
109491bc56edSDimitry Andric   }
109591bc56edSDimitry Andric }
109691bc56edSDimitry Andric 
1097f22ef01cSRoman Divacky MachineConstantPool::~MachineConstantPool() {
1098d88c1a5aSDimitry Andric   // A constant may be a member of both Constants and MachineCPVsSharingEntries,
1099d88c1a5aSDimitry Andric   // so keep track of which we've deleted to avoid double deletions.
1100d88c1a5aSDimitry Andric   DenseSet<MachineConstantPoolValue*> Deleted;
1101f22ef01cSRoman Divacky   for (unsigned i = 0, e = Constants.size(); i != e; ++i)
1102d88c1a5aSDimitry Andric     if (Constants[i].isMachineConstantPoolEntry()) {
1103d88c1a5aSDimitry Andric       Deleted.insert(Constants[i].Val.MachineCPVal);
1104f22ef01cSRoman Divacky       delete Constants[i].Val.MachineCPVal;
1105d88c1a5aSDimitry Andric     }
1106dd6029ffSDimitry Andric   for (DenseSet<MachineConstantPoolValue*>::iterator I =
1107dd6029ffSDimitry Andric        MachineCPVsSharingEntries.begin(), E = MachineCPVsSharingEntries.end();
1108d88c1a5aSDimitry Andric        I != E; ++I) {
1109d88c1a5aSDimitry Andric     if (Deleted.count(*I) == 0)
1110dd6029ffSDimitry Andric       delete *I;
1111f22ef01cSRoman Divacky   }
1112d88c1a5aSDimitry Andric }
1113f22ef01cSRoman Divacky 
11148f0fd8f6SDimitry Andric /// Test whether the given two constants can be allocated the same constant pool
11158f0fd8f6SDimitry Andric /// entry.
1116f22ef01cSRoman Divacky static bool CanShareConstantPoolEntry(const Constant *A, const Constant *B,
1117875ed548SDimitry Andric                                       const DataLayout &DL) {
1118f22ef01cSRoman Divacky   // Handle the trivial case quickly.
1119f22ef01cSRoman Divacky   if (A == B) return true;
1120f22ef01cSRoman Divacky 
1121f22ef01cSRoman Divacky   // If they have the same type but weren't the same constant, quickly
1122f22ef01cSRoman Divacky   // reject them.
1123f22ef01cSRoman Divacky   if (A->getType() == B->getType()) return false;
1124f22ef01cSRoman Divacky 
1125dff0c46cSDimitry Andric   // We can't handle structs or arrays.
1126dff0c46cSDimitry Andric   if (isa<StructType>(A->getType()) || isa<ArrayType>(A->getType()) ||
1127dff0c46cSDimitry Andric       isa<StructType>(B->getType()) || isa<ArrayType>(B->getType()))
1128dff0c46cSDimitry Andric     return false;
1129dff0c46cSDimitry Andric 
1130f22ef01cSRoman Divacky   // For now, only support constants with the same size.
1131875ed548SDimitry Andric   uint64_t StoreSize = DL.getTypeStoreSize(A->getType());
1132875ed548SDimitry Andric   if (StoreSize != DL.getTypeStoreSize(B->getType()) || StoreSize > 128)
1133f22ef01cSRoman Divacky     return false;
1134f22ef01cSRoman Divacky 
1135dff0c46cSDimitry Andric   Type *IntTy = IntegerType::get(A->getContext(), StoreSize*8);
1136f22ef01cSRoman Divacky 
1137dff0c46cSDimitry Andric   // Try constant folding a bitcast of both instructions to an integer.  If we
1138dff0c46cSDimitry Andric   // get two identical ConstantInt's, then we are good to share them.  We use
1139dff0c46cSDimitry Andric   // the constant folding APIs to do this so that we get the benefit of
11403861d79fSDimitry Andric   // DataLayout.
1141dff0c46cSDimitry Andric   if (isa<PointerType>(A->getType()))
11423ca95b02SDimitry Andric     A = ConstantFoldCastOperand(Instruction::PtrToInt,
11433ca95b02SDimitry Andric                                 const_cast<Constant *>(A), IntTy, DL);
1144dff0c46cSDimitry Andric   else if (A->getType() != IntTy)
11453ca95b02SDimitry Andric     A = ConstantFoldCastOperand(Instruction::BitCast, const_cast<Constant *>(A),
11463ca95b02SDimitry Andric                                 IntTy, DL);
1147dff0c46cSDimitry Andric   if (isa<PointerType>(B->getType()))
11483ca95b02SDimitry Andric     B = ConstantFoldCastOperand(Instruction::PtrToInt,
11493ca95b02SDimitry Andric                                 const_cast<Constant *>(B), IntTy, DL);
1150dff0c46cSDimitry Andric   else if (B->getType() != IntTy)
11513ca95b02SDimitry Andric     B = ConstantFoldCastOperand(Instruction::BitCast, const_cast<Constant *>(B),
11523ca95b02SDimitry Andric                                 IntTy, DL);
1153f22ef01cSRoman Divacky 
1154dff0c46cSDimitry Andric   return A == B;
1155f22ef01cSRoman Divacky }
1156f22ef01cSRoman Divacky 
11578f0fd8f6SDimitry Andric /// Create a new entry in the constant pool or return an existing one.
11588f0fd8f6SDimitry Andric /// User must specify the log2 of the minimum required alignment for the object.
1159f22ef01cSRoman Divacky unsigned MachineConstantPool::getConstantPoolIndex(const Constant *C,
1160f22ef01cSRoman Divacky                                                    unsigned Alignment) {
1161f22ef01cSRoman Divacky   assert(Alignment && "Alignment must be specified!");
1162f22ef01cSRoman Divacky   if (Alignment > PoolAlignment) PoolAlignment = Alignment;
1163f22ef01cSRoman Divacky 
1164f22ef01cSRoman Divacky   // Check to see if we already have this constant.
1165f22ef01cSRoman Divacky   //
1166f22ef01cSRoman Divacky   // FIXME, this could be made much more efficient for large constant pools.
1167f22ef01cSRoman Divacky   for (unsigned i = 0, e = Constants.size(); i != e; ++i)
1168f22ef01cSRoman Divacky     if (!Constants[i].isMachineConstantPoolEntry() &&
1169875ed548SDimitry Andric         CanShareConstantPoolEntry(Constants[i].Val.ConstVal, C, DL)) {
1170f22ef01cSRoman Divacky       if ((unsigned)Constants[i].getAlignment() < Alignment)
1171f22ef01cSRoman Divacky         Constants[i].Alignment = Alignment;
1172f22ef01cSRoman Divacky       return i;
1173f22ef01cSRoman Divacky     }
1174f22ef01cSRoman Divacky 
1175f22ef01cSRoman Divacky   Constants.push_back(MachineConstantPoolEntry(C, Alignment));
1176f22ef01cSRoman Divacky   return Constants.size()-1;
1177f22ef01cSRoman Divacky }
1178f22ef01cSRoman Divacky 
1179f22ef01cSRoman Divacky unsigned MachineConstantPool::getConstantPoolIndex(MachineConstantPoolValue *V,
1180f22ef01cSRoman Divacky                                                    unsigned Alignment) {
1181f22ef01cSRoman Divacky   assert(Alignment && "Alignment must be specified!");
1182f22ef01cSRoman Divacky   if (Alignment > PoolAlignment) PoolAlignment = Alignment;
1183f22ef01cSRoman Divacky 
1184f22ef01cSRoman Divacky   // Check to see if we already have this constant.
1185f22ef01cSRoman Divacky   //
1186f22ef01cSRoman Divacky   // FIXME, this could be made much more efficient for large constant pools.
1187f22ef01cSRoman Divacky   int Idx = V->getExistingMachineCPValue(this, Alignment);
1188dd6029ffSDimitry Andric   if (Idx != -1) {
1189dd6029ffSDimitry Andric     MachineCPVsSharingEntries.insert(V);
1190f22ef01cSRoman Divacky     return (unsigned)Idx;
1191dd6029ffSDimitry Andric   }
1192f22ef01cSRoman Divacky 
1193f22ef01cSRoman Divacky   Constants.push_back(MachineConstantPoolEntry(V, Alignment));
1194f22ef01cSRoman Divacky   return Constants.size()-1;
1195f22ef01cSRoman Divacky }
1196f22ef01cSRoman Divacky 
1197f22ef01cSRoman Divacky void MachineConstantPool::print(raw_ostream &OS) const {
1198f22ef01cSRoman Divacky   if (Constants.empty()) return;
1199f22ef01cSRoman Divacky 
1200f22ef01cSRoman Divacky   OS << "Constant Pool:\n";
1201f22ef01cSRoman Divacky   for (unsigned i = 0, e = Constants.size(); i != e; ++i) {
1202f22ef01cSRoman Divacky     OS << "  cp#" << i << ": ";
1203f22ef01cSRoman Divacky     if (Constants[i].isMachineConstantPoolEntry())
1204f22ef01cSRoman Divacky       Constants[i].Val.MachineCPVal->print(OS);
1205f22ef01cSRoman Divacky     else
120691bc56edSDimitry Andric       Constants[i].Val.ConstVal->printAsOperand(OS, /*PrintType=*/false);
1207f22ef01cSRoman Divacky     OS << ", align=" << Constants[i].getAlignment();
1208f22ef01cSRoman Divacky     OS << "\n";
1209f22ef01cSRoman Divacky   }
1210f22ef01cSRoman Divacky }
1211f22ef01cSRoman Divacky 
12123861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
12133ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineConstantPool::dump() const { print(dbgs()); }
12143861d79fSDimitry Andric #endif
1215