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);
1727a7e6055SDimitry Andric   VariableDbgInfos.clear();
173f22ef01cSRoman Divacky   if (RegInfo) {
174f22ef01cSRoman Divacky     RegInfo->~MachineRegisterInfo();
175f22ef01cSRoman Divacky     Allocator.Deallocate(RegInfo);
176f22ef01cSRoman Divacky   }
177f22ef01cSRoman Divacky   if (MFInfo) {
178f22ef01cSRoman Divacky     MFInfo->~MachineFunctionInfo();
179f22ef01cSRoman Divacky     Allocator.Deallocate(MFInfo);
180f22ef01cSRoman Divacky   }
1817ae0e2c9SDimitry Andric 
1827ae0e2c9SDimitry Andric   FrameInfo->~MachineFrameInfo();
1837ae0e2c9SDimitry Andric   Allocator.Deallocate(FrameInfo);
1847ae0e2c9SDimitry Andric 
1857ae0e2c9SDimitry Andric   ConstantPool->~MachineConstantPool();
1867ae0e2c9SDimitry Andric   Allocator.Deallocate(ConstantPool);
187f22ef01cSRoman Divacky 
188f22ef01cSRoman Divacky   if (JumpTableInfo) {
189f22ef01cSRoman Divacky     JumpTableInfo->~MachineJumpTableInfo();
190f22ef01cSRoman Divacky     Allocator.Deallocate(JumpTableInfo);
191f22ef01cSRoman Divacky   }
1927d523365SDimitry Andric 
1937d523365SDimitry Andric   if (WinEHInfo) {
1947d523365SDimitry Andric     WinEHInfo->~WinEHFuncInfo();
1957d523365SDimitry Andric     Allocator.Deallocate(WinEHInfo);
1967d523365SDimitry Andric   }
197f22ef01cSRoman Divacky }
198f22ef01cSRoman Divacky 
199875ed548SDimitry Andric const DataLayout &MachineFunction::getDataLayout() const {
200875ed548SDimitry Andric   return Fn->getParent()->getDataLayout();
201875ed548SDimitry Andric }
202875ed548SDimitry Andric 
2038f0fd8f6SDimitry Andric /// Get the JumpTableInfo for this function.
2048f0fd8f6SDimitry Andric /// If it does not already exist, allocate one.
205f22ef01cSRoman Divacky MachineJumpTableInfo *MachineFunction::
206f22ef01cSRoman Divacky getOrCreateJumpTableInfo(unsigned EntryKind) {
207f22ef01cSRoman Divacky   if (JumpTableInfo) return JumpTableInfo;
208f22ef01cSRoman Divacky 
209f22ef01cSRoman Divacky   JumpTableInfo = new (Allocator)
210f22ef01cSRoman Divacky     MachineJumpTableInfo((MachineJumpTableInfo::JTEntryKind)EntryKind);
211f22ef01cSRoman Divacky   return JumpTableInfo;
212f22ef01cSRoman Divacky }
213f22ef01cSRoman Divacky 
21491bc56edSDimitry Andric /// Should we be emitting segmented stack stuff for the function
2158c24ff90SDimitry Andric bool MachineFunction::shouldSplitStack() const {
21691bc56edSDimitry Andric   return getFunction()->hasFnAttribute("split-stack");
21791bc56edSDimitry Andric }
21891bc56edSDimitry Andric 
2198f0fd8f6SDimitry Andric /// This discards all of the MachineBasicBlock numbers and recomputes them.
2208f0fd8f6SDimitry Andric /// This guarantees that the MBB numbers are sequential, dense, and match the
2218f0fd8f6SDimitry Andric /// ordering of the blocks within the function.  If a specific MachineBasicBlock
2228f0fd8f6SDimitry Andric /// is specified, only that block and those after it are renumbered.
223f22ef01cSRoman Divacky void MachineFunction::RenumberBlocks(MachineBasicBlock *MBB) {
224f22ef01cSRoman Divacky   if (empty()) { MBBNumbering.clear(); return; }
225f22ef01cSRoman Divacky   MachineFunction::iterator MBBI, E = end();
22691bc56edSDimitry Andric   if (MBB == nullptr)
227f22ef01cSRoman Divacky     MBBI = begin();
228f22ef01cSRoman Divacky   else
2297d523365SDimitry Andric     MBBI = MBB->getIterator();
230f22ef01cSRoman Divacky 
231f22ef01cSRoman Divacky   // Figure out the block number this should have.
232f22ef01cSRoman Divacky   unsigned BlockNo = 0;
233f22ef01cSRoman Divacky   if (MBBI != begin())
23491bc56edSDimitry Andric     BlockNo = std::prev(MBBI)->getNumber() + 1;
235f22ef01cSRoman Divacky 
236f22ef01cSRoman Divacky   for (; MBBI != E; ++MBBI, ++BlockNo) {
237f22ef01cSRoman Divacky     if (MBBI->getNumber() != (int)BlockNo) {
238f22ef01cSRoman Divacky       // Remove use of the old number.
239f22ef01cSRoman Divacky       if (MBBI->getNumber() != -1) {
240f22ef01cSRoman Divacky         assert(MBBNumbering[MBBI->getNumber()] == &*MBBI &&
241f22ef01cSRoman Divacky                "MBB number mismatch!");
24291bc56edSDimitry Andric         MBBNumbering[MBBI->getNumber()] = nullptr;
243f22ef01cSRoman Divacky       }
244f22ef01cSRoman Divacky 
245f22ef01cSRoman Divacky       // If BlockNo is already taken, set that block's number to -1.
246f22ef01cSRoman Divacky       if (MBBNumbering[BlockNo])
247f22ef01cSRoman Divacky         MBBNumbering[BlockNo]->setNumber(-1);
248f22ef01cSRoman Divacky 
2497d523365SDimitry Andric       MBBNumbering[BlockNo] = &*MBBI;
250f22ef01cSRoman Divacky       MBBI->setNumber(BlockNo);
251f22ef01cSRoman Divacky     }
252f22ef01cSRoman Divacky   }
253f22ef01cSRoman Divacky 
254f22ef01cSRoman Divacky   // Okay, all the blocks are renumbered.  If we have compactified the block
255f22ef01cSRoman Divacky   // numbering, shrink MBBNumbering now.
256f22ef01cSRoman Divacky   assert(BlockNo <= MBBNumbering.size() && "Mismatch!");
257f22ef01cSRoman Divacky   MBBNumbering.resize(BlockNo);
258f22ef01cSRoman Divacky }
259f22ef01cSRoman Divacky 
2608f0fd8f6SDimitry Andric /// Allocate a new MachineInstr. Use this instead of `new MachineInstr'.
2613ca95b02SDimitry Andric MachineInstr *MachineFunction::CreateMachineInstr(const MCInstrDesc &MCID,
2623ca95b02SDimitry Andric                                                   const DebugLoc &DL,
2633ca95b02SDimitry Andric                                                   bool NoImp) {
264f22ef01cSRoman Divacky   return new (InstructionRecycler.Allocate<MachineInstr>(Allocator))
265139f7f9bSDimitry Andric     MachineInstr(*this, MCID, DL, NoImp);
266f22ef01cSRoman Divacky }
267f22ef01cSRoman Divacky 
2688f0fd8f6SDimitry Andric /// Create a new MachineInstr which is a copy of the 'Orig' instruction,
2698f0fd8f6SDimitry Andric /// identical in all ways except the instruction has no parent, prev, or next.
270f22ef01cSRoman Divacky MachineInstr *
271f22ef01cSRoman Divacky MachineFunction::CloneMachineInstr(const MachineInstr *Orig) {
272f22ef01cSRoman Divacky   return new (InstructionRecycler.Allocate<MachineInstr>(Allocator))
273f22ef01cSRoman Divacky              MachineInstr(*this, *Orig);
274f22ef01cSRoman Divacky }
275f22ef01cSRoman Divacky 
2768f0fd8f6SDimitry Andric /// Delete the given MachineInstr.
277f22ef01cSRoman Divacky ///
278139f7f9bSDimitry Andric /// This function also serves as the MachineInstr destructor - the real
279139f7f9bSDimitry Andric /// ~MachineInstr() destructor must be empty.
280f22ef01cSRoman Divacky void
281f22ef01cSRoman Divacky MachineFunction::DeleteMachineInstr(MachineInstr *MI) {
282139f7f9bSDimitry Andric   // Strip it for parts. The operand array and the MI object itself are
283139f7f9bSDimitry Andric   // independently recyclable.
284139f7f9bSDimitry Andric   if (MI->Operands)
285139f7f9bSDimitry Andric     deallocateOperandArray(MI->CapOperands, MI->Operands);
286139f7f9bSDimitry Andric   // Don't call ~MachineInstr() which must be trivial anyway because
287139f7f9bSDimitry Andric   // ~MachineFunction drops whole lists of MachineInstrs wihout calling their
288139f7f9bSDimitry Andric   // destructors.
289f22ef01cSRoman Divacky   InstructionRecycler.Deallocate(Allocator, MI);
290f22ef01cSRoman Divacky }
291f22ef01cSRoman Divacky 
2928f0fd8f6SDimitry Andric /// Allocate a new MachineBasicBlock. Use this instead of
2938f0fd8f6SDimitry Andric /// `new MachineBasicBlock'.
294f22ef01cSRoman Divacky MachineBasicBlock *
295f22ef01cSRoman Divacky MachineFunction::CreateMachineBasicBlock(const BasicBlock *bb) {
296f22ef01cSRoman Divacky   return new (BasicBlockRecycler.Allocate<MachineBasicBlock>(Allocator))
297f22ef01cSRoman Divacky              MachineBasicBlock(*this, bb);
298f22ef01cSRoman Divacky }
299f22ef01cSRoman Divacky 
3008f0fd8f6SDimitry Andric /// Delete the given MachineBasicBlock.
301f22ef01cSRoman Divacky void
302f22ef01cSRoman Divacky MachineFunction::DeleteMachineBasicBlock(MachineBasicBlock *MBB) {
303f22ef01cSRoman Divacky   assert(MBB->getParent() == this && "MBB parent mismatch!");
304f22ef01cSRoman Divacky   MBB->~MachineBasicBlock();
305f22ef01cSRoman Divacky   BasicBlockRecycler.Deallocate(Allocator, MBB);
306f22ef01cSRoman Divacky }
307f22ef01cSRoman Divacky 
3083ca95b02SDimitry Andric MachineMemOperand *MachineFunction::getMachineMemOperand(
3093ca95b02SDimitry Andric     MachinePointerInfo PtrInfo, MachineMemOperand::Flags f, uint64_t s,
310d88c1a5aSDimitry Andric     unsigned base_alignment, const AAMDNodes &AAInfo, const MDNode *Ranges,
311d88c1a5aSDimitry Andric     SynchronizationScope SynchScope, AtomicOrdering Ordering,
312d88c1a5aSDimitry Andric     AtomicOrdering FailureOrdering) {
3133ca95b02SDimitry Andric   return new (Allocator)
314d88c1a5aSDimitry Andric       MachineMemOperand(PtrInfo, f, s, base_alignment, AAInfo, Ranges,
315d88c1a5aSDimitry Andric                         SynchScope, Ordering, FailureOrdering);
316f22ef01cSRoman Divacky }
317f22ef01cSRoman Divacky 
318f22ef01cSRoman Divacky MachineMemOperand *
319f22ef01cSRoman Divacky MachineFunction::getMachineMemOperand(const MachineMemOperand *MMO,
320f22ef01cSRoman Divacky                                       int64_t Offset, uint64_t Size) {
32191bc56edSDimitry Andric   if (MMO->getValue())
322f22ef01cSRoman Divacky     return new (Allocator)
3232754fe60SDimitry Andric                MachineMemOperand(MachinePointerInfo(MMO->getValue(),
3242754fe60SDimitry Andric                                                     MMO->getOffset()+Offset),
325d88c1a5aSDimitry Andric                                  MMO->getFlags(), Size, MMO->getBaseAlignment(),
326d88c1a5aSDimitry Andric                                  AAMDNodes(), nullptr, MMO->getSynchScope(),
327d88c1a5aSDimitry Andric                                  MMO->getOrdering(), MMO->getFailureOrdering());
32891bc56edSDimitry Andric   return new (Allocator)
32991bc56edSDimitry Andric              MachineMemOperand(MachinePointerInfo(MMO->getPseudoValue(),
33091bc56edSDimitry Andric                                                   MMO->getOffset()+Offset),
331d88c1a5aSDimitry Andric                                MMO->getFlags(), Size, MMO->getBaseAlignment(),
332d88c1a5aSDimitry Andric                                AAMDNodes(), nullptr, MMO->getSynchScope(),
333d88c1a5aSDimitry Andric                                MMO->getOrdering(), MMO->getFailureOrdering());
334f22ef01cSRoman Divacky }
335f22ef01cSRoman Divacky 
336f22ef01cSRoman Divacky MachineInstr::mmo_iterator
337f22ef01cSRoman Divacky MachineFunction::allocateMemRefsArray(unsigned long Num) {
338f22ef01cSRoman Divacky   return Allocator.Allocate<MachineMemOperand *>(Num);
339f22ef01cSRoman Divacky }
340f22ef01cSRoman Divacky 
341f22ef01cSRoman Divacky std::pair<MachineInstr::mmo_iterator, MachineInstr::mmo_iterator>
342f22ef01cSRoman Divacky MachineFunction::extractLoadMemRefs(MachineInstr::mmo_iterator Begin,
343f22ef01cSRoman Divacky                                     MachineInstr::mmo_iterator End) {
344f22ef01cSRoman Divacky   // Count the number of load mem refs.
345f22ef01cSRoman Divacky   unsigned Num = 0;
346f22ef01cSRoman Divacky   for (MachineInstr::mmo_iterator I = Begin; I != End; ++I)
347f22ef01cSRoman Divacky     if ((*I)->isLoad())
348f22ef01cSRoman Divacky       ++Num;
349f22ef01cSRoman Divacky 
350f22ef01cSRoman Divacky   // Allocate a new array and populate it with the load information.
351f22ef01cSRoman Divacky   MachineInstr::mmo_iterator Result = allocateMemRefsArray(Num);
352f22ef01cSRoman Divacky   unsigned Index = 0;
353f22ef01cSRoman Divacky   for (MachineInstr::mmo_iterator I = Begin; I != End; ++I) {
354f22ef01cSRoman Divacky     if ((*I)->isLoad()) {
355f22ef01cSRoman Divacky       if (!(*I)->isStore())
356f22ef01cSRoman Divacky         // Reuse the MMO.
357f22ef01cSRoman Divacky         Result[Index] = *I;
358f22ef01cSRoman Divacky       else {
359f22ef01cSRoman Divacky         // Clone the MMO and unset the store flag.
360f22ef01cSRoman Divacky         MachineMemOperand *JustLoad =
3612754fe60SDimitry Andric           getMachineMemOperand((*I)->getPointerInfo(),
362f22ef01cSRoman Divacky                                (*I)->getFlags() & ~MachineMemOperand::MOStore,
3632754fe60SDimitry Andric                                (*I)->getSize(), (*I)->getBaseAlignment(),
364d88c1a5aSDimitry Andric                                (*I)->getAAInfo(), nullptr,
365d88c1a5aSDimitry Andric                                (*I)->getSynchScope(), (*I)->getOrdering(),
366d88c1a5aSDimitry Andric                                (*I)->getFailureOrdering());
367f22ef01cSRoman Divacky         Result[Index] = JustLoad;
368f22ef01cSRoman Divacky       }
369f22ef01cSRoman Divacky       ++Index;
370f22ef01cSRoman Divacky     }
371f22ef01cSRoman Divacky   }
372f22ef01cSRoman Divacky   return std::make_pair(Result, Result + Num);
373f22ef01cSRoman Divacky }
374f22ef01cSRoman Divacky 
375f22ef01cSRoman Divacky std::pair<MachineInstr::mmo_iterator, MachineInstr::mmo_iterator>
376f22ef01cSRoman Divacky MachineFunction::extractStoreMemRefs(MachineInstr::mmo_iterator Begin,
377f22ef01cSRoman Divacky                                      MachineInstr::mmo_iterator End) {
378f22ef01cSRoman Divacky   // Count the number of load mem refs.
379f22ef01cSRoman Divacky   unsigned Num = 0;
380f22ef01cSRoman Divacky   for (MachineInstr::mmo_iterator I = Begin; I != End; ++I)
381f22ef01cSRoman Divacky     if ((*I)->isStore())
382f22ef01cSRoman Divacky       ++Num;
383f22ef01cSRoman Divacky 
384f22ef01cSRoman Divacky   // Allocate a new array and populate it with the store information.
385f22ef01cSRoman Divacky   MachineInstr::mmo_iterator Result = allocateMemRefsArray(Num);
386f22ef01cSRoman Divacky   unsigned Index = 0;
387f22ef01cSRoman Divacky   for (MachineInstr::mmo_iterator I = Begin; I != End; ++I) {
388f22ef01cSRoman Divacky     if ((*I)->isStore()) {
389f22ef01cSRoman Divacky       if (!(*I)->isLoad())
390f22ef01cSRoman Divacky         // Reuse the MMO.
391f22ef01cSRoman Divacky         Result[Index] = *I;
392f22ef01cSRoman Divacky       else {
393f22ef01cSRoman Divacky         // Clone the MMO and unset the load flag.
394f22ef01cSRoman Divacky         MachineMemOperand *JustStore =
3952754fe60SDimitry Andric           getMachineMemOperand((*I)->getPointerInfo(),
396f22ef01cSRoman Divacky                                (*I)->getFlags() & ~MachineMemOperand::MOLoad,
3972754fe60SDimitry Andric                                (*I)->getSize(), (*I)->getBaseAlignment(),
398d88c1a5aSDimitry Andric                                (*I)->getAAInfo(), nullptr,
399d88c1a5aSDimitry Andric                                (*I)->getSynchScope(), (*I)->getOrdering(),
400d88c1a5aSDimitry Andric                                (*I)->getFailureOrdering());
401f22ef01cSRoman Divacky         Result[Index] = JustStore;
402f22ef01cSRoman Divacky       }
403f22ef01cSRoman Divacky       ++Index;
404f22ef01cSRoman Divacky     }
405f22ef01cSRoman Divacky   }
406f22ef01cSRoman Divacky   return std::make_pair(Result, Result + Num);
407f22ef01cSRoman Divacky }
408f22ef01cSRoman Divacky 
4097d523365SDimitry Andric const char *MachineFunction::createExternalSymbolName(StringRef Name) {
4107d523365SDimitry Andric   char *Dest = Allocator.Allocate<char>(Name.size() + 1);
4117d523365SDimitry Andric   std::copy(Name.begin(), Name.end(), Dest);
4127d523365SDimitry Andric   Dest[Name.size()] = 0;
4137d523365SDimitry Andric   return Dest;
4147d523365SDimitry Andric }
4157d523365SDimitry Andric 
4163861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
4173ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineFunction::dump() const {
418f22ef01cSRoman Divacky   print(dbgs());
419f22ef01cSRoman Divacky }
4203861d79fSDimitry Andric #endif
4213861d79fSDimitry Andric 
4223861d79fSDimitry Andric StringRef MachineFunction::getName() const {
4233861d79fSDimitry Andric   assert(getFunction() && "No function!");
4243861d79fSDimitry Andric   return getFunction()->getName();
4253861d79fSDimitry Andric }
426f22ef01cSRoman Divacky 
4273ca95b02SDimitry Andric void MachineFunction::print(raw_ostream &OS, const SlotIndexes *Indexes) const {
4283861d79fSDimitry Andric   OS << "# Machine code for function " << getName() << ": ";
4293ca95b02SDimitry Andric   getProperties().print(OS);
430d88c1a5aSDimitry Andric   OS << '\n';
431f22ef01cSRoman Divacky 
432f22ef01cSRoman Divacky   // Print Frame Information
433f22ef01cSRoman Divacky   FrameInfo->print(*this, OS);
434f22ef01cSRoman Divacky 
435f22ef01cSRoman Divacky   // Print JumpTable Information
436f22ef01cSRoman Divacky   if (JumpTableInfo)
437f22ef01cSRoman Divacky     JumpTableInfo->print(OS);
438f22ef01cSRoman Divacky 
439f22ef01cSRoman Divacky   // Print Constant Pool
440f22ef01cSRoman Divacky   ConstantPool->print(OS);
441f22ef01cSRoman Divacky 
44239d628a0SDimitry Andric   const TargetRegisterInfo *TRI = getSubtarget().getRegisterInfo();
443f22ef01cSRoman Divacky 
444f22ef01cSRoman Divacky   if (RegInfo && !RegInfo->livein_empty()) {
445f22ef01cSRoman Divacky     OS << "Function Live Ins: ";
446f22ef01cSRoman Divacky     for (MachineRegisterInfo::livein_iterator
447f22ef01cSRoman Divacky          I = RegInfo->livein_begin(), E = RegInfo->livein_end(); I != E; ++I) {
448bd5abe19SDimitry Andric       OS << PrintReg(I->first, TRI);
449f22ef01cSRoman Divacky       if (I->second)
450bd5abe19SDimitry Andric         OS << " in " << PrintReg(I->second, TRI);
45191bc56edSDimitry Andric       if (std::next(I) != E)
452f22ef01cSRoman Divacky         OS << ", ";
453f22ef01cSRoman Divacky     }
454f22ef01cSRoman Divacky     OS << '\n';
455f22ef01cSRoman Divacky   }
456f22ef01cSRoman Divacky 
4573dac3a9bSDimitry Andric   ModuleSlotTracker MST(getFunction()->getParent());
4583dac3a9bSDimitry Andric   MST.incorporateFunction(*getFunction());
45991bc56edSDimitry Andric   for (const auto &BB : *this) {
460f22ef01cSRoman Divacky     OS << '\n';
4613dac3a9bSDimitry Andric     BB.print(OS, MST, Indexes);
462f22ef01cSRoman Divacky   }
463f22ef01cSRoman Divacky 
4643861d79fSDimitry Andric   OS << "\n# End machine code for function " << getName() << ".\n\n";
465f22ef01cSRoman Divacky }
466f22ef01cSRoman Divacky 
467f22ef01cSRoman Divacky namespace llvm {
468f22ef01cSRoman Divacky   template<>
469f22ef01cSRoman Divacky   struct DOTGraphTraits<const MachineFunction*> : public DefaultDOTGraphTraits {
470f22ef01cSRoman Divacky 
471f22ef01cSRoman Divacky   DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {}
472f22ef01cSRoman Divacky 
473f22ef01cSRoman Divacky     static std::string getGraphName(const MachineFunction *F) {
474ff0cc061SDimitry Andric       return ("CFG for '" + F->getName() + "' function").str();
475f22ef01cSRoman Divacky     }
476f22ef01cSRoman Divacky 
477f22ef01cSRoman Divacky     std::string getNodeLabel(const MachineBasicBlock *Node,
478f22ef01cSRoman Divacky                              const MachineFunction *Graph) {
479f22ef01cSRoman Divacky       std::string OutStr;
480f22ef01cSRoman Divacky       {
481f22ef01cSRoman Divacky         raw_string_ostream OSS(OutStr);
482f22ef01cSRoman Divacky 
4832754fe60SDimitry Andric         if (isSimple()) {
4842754fe60SDimitry Andric           OSS << "BB#" << Node->getNumber();
4852754fe60SDimitry Andric           if (const BasicBlock *BB = Node->getBasicBlock())
4862754fe60SDimitry Andric             OSS << ": " << BB->getName();
4872754fe60SDimitry Andric         } else
488f22ef01cSRoman Divacky           Node->print(OSS);
489f22ef01cSRoman Divacky       }
490f22ef01cSRoman Divacky 
491f22ef01cSRoman Divacky       if (OutStr[0] == '\n') OutStr.erase(OutStr.begin());
492f22ef01cSRoman Divacky 
493f22ef01cSRoman Divacky       // Process string output to make it nicer...
494f22ef01cSRoman Divacky       for (unsigned i = 0; i != OutStr.length(); ++i)
495f22ef01cSRoman Divacky         if (OutStr[i] == '\n') {                            // Left justify
496f22ef01cSRoman Divacky           OutStr[i] = '\\';
497f22ef01cSRoman Divacky           OutStr.insert(OutStr.begin()+i+1, 'l');
498f22ef01cSRoman Divacky         }
499f22ef01cSRoman Divacky       return OutStr;
500f22ef01cSRoman Divacky     }
501f22ef01cSRoman Divacky   };
5023dac3a9bSDimitry Andric }
503f22ef01cSRoman Divacky 
504f22ef01cSRoman Divacky void MachineFunction::viewCFG() const
505f22ef01cSRoman Divacky {
506f22ef01cSRoman Divacky #ifndef NDEBUG
5073861d79fSDimitry Andric   ViewGraph(this, "mf" + getName());
508f22ef01cSRoman Divacky #else
509ffd1746dSEd Schouten   errs() << "MachineFunction::viewCFG is only available in debug builds on "
510f22ef01cSRoman Divacky          << "systems with Graphviz or gv!\n";
511f22ef01cSRoman Divacky #endif // NDEBUG
512f22ef01cSRoman Divacky }
513f22ef01cSRoman Divacky 
514f22ef01cSRoman Divacky void MachineFunction::viewCFGOnly() const
515f22ef01cSRoman Divacky {
516f22ef01cSRoman Divacky #ifndef NDEBUG
5173861d79fSDimitry Andric   ViewGraph(this, "mf" + getName(), true);
518f22ef01cSRoman Divacky #else
519ffd1746dSEd Schouten   errs() << "MachineFunction::viewCFGOnly is only available in debug builds on "
520f22ef01cSRoman Divacky          << "systems with Graphviz or gv!\n";
521f22ef01cSRoman Divacky #endif // NDEBUG
522f22ef01cSRoman Divacky }
523f22ef01cSRoman Divacky 
5248f0fd8f6SDimitry Andric /// Add the specified physical register as a live-in value and
525f22ef01cSRoman Divacky /// create a corresponding virtual register for it.
526f22ef01cSRoman Divacky unsigned MachineFunction::addLiveIn(unsigned PReg,
527dd6029ffSDimitry Andric                                     const TargetRegisterClass *RC) {
528f22ef01cSRoman Divacky   MachineRegisterInfo &MRI = getRegInfo();
529f22ef01cSRoman Divacky   unsigned VReg = MRI.getLiveInVirtReg(PReg);
530f22ef01cSRoman Divacky   if (VReg) {
53191bc56edSDimitry Andric     const TargetRegisterClass *VRegRC = MRI.getRegClass(VReg);
53291bc56edSDimitry Andric     (void)VRegRC;
53391bc56edSDimitry Andric     // A physical register can be added several times.
53491bc56edSDimitry Andric     // Between two calls, the register class of the related virtual register
53591bc56edSDimitry Andric     // may have been constrained to match some operation constraints.
53691bc56edSDimitry Andric     // In that case, check that the current register class includes the
53791bc56edSDimitry Andric     // physical register and is a sub class of the specified RC.
53891bc56edSDimitry Andric     assert((VRegRC == RC || (VRegRC->contains(PReg) &&
53991bc56edSDimitry Andric                              RC->hasSubClassEq(VRegRC))) &&
54091bc56edSDimitry Andric             "Register class mismatch!");
541f22ef01cSRoman Divacky     return VReg;
542f22ef01cSRoman Divacky   }
543f22ef01cSRoman Divacky   VReg = MRI.createVirtualRegister(RC);
544f22ef01cSRoman Divacky   MRI.addLiveIn(PReg, VReg);
545f22ef01cSRoman Divacky   return VReg;
546f22ef01cSRoman Divacky }
547f22ef01cSRoman Divacky 
5488f0fd8f6SDimitry Andric /// Return the MCSymbol for the specified non-empty jump table.
549f22ef01cSRoman Divacky /// If isLinkerPrivate is specified, an 'l' label is returned, otherwise a
550f22ef01cSRoman Divacky /// normal 'L' label is returned.
551f22ef01cSRoman Divacky MCSymbol *MachineFunction::getJTISymbol(unsigned JTI, MCContext &Ctx,
552f22ef01cSRoman Divacky                                         bool isLinkerPrivate) const {
553875ed548SDimitry Andric   const DataLayout &DL = getDataLayout();
554f22ef01cSRoman Divacky   assert(JumpTableInfo && "No jump tables");
555f22ef01cSRoman Divacky   assert(JTI < JumpTableInfo->getJumpTables().size() && "Invalid JTI!");
556f22ef01cSRoman Divacky 
557d88c1a5aSDimitry Andric   StringRef Prefix = isLinkerPrivate ? DL.getLinkerPrivateGlobalPrefix()
558875ed548SDimitry Andric                                      : DL.getPrivateGlobalPrefix();
559f22ef01cSRoman Divacky   SmallString<60> Name;
560f22ef01cSRoman Divacky   raw_svector_ostream(Name)
561f22ef01cSRoman Divacky     << Prefix << "JTI" << getFunctionNumber() << '_' << JTI;
562ff0cc061SDimitry Andric   return Ctx.getOrCreateSymbol(Name);
563f22ef01cSRoman Divacky }
564f22ef01cSRoman Divacky 
5658f0fd8f6SDimitry Andric /// Return a function-local symbol to represent the PIC base.
5662754fe60SDimitry Andric MCSymbol *MachineFunction::getPICBaseSymbol() const {
567875ed548SDimitry Andric   const DataLayout &DL = getDataLayout();
568875ed548SDimitry Andric   return Ctx.getOrCreateSymbol(Twine(DL.getPrivateGlobalPrefix()) +
5692754fe60SDimitry Andric                                Twine(getFunctionNumber()) + "$pb");
5702754fe60SDimitry Andric }
571f22ef01cSRoman Divacky 
572d88c1a5aSDimitry Andric /// \name Exception Handling
573d88c1a5aSDimitry Andric /// \{
574d88c1a5aSDimitry Andric 
575d88c1a5aSDimitry Andric LandingPadInfo &
576d88c1a5aSDimitry Andric MachineFunction::getOrCreateLandingPadInfo(MachineBasicBlock *LandingPad) {
577d88c1a5aSDimitry Andric   unsigned N = LandingPads.size();
578d88c1a5aSDimitry Andric   for (unsigned i = 0; i < N; ++i) {
579d88c1a5aSDimitry Andric     LandingPadInfo &LP = LandingPads[i];
580d88c1a5aSDimitry Andric     if (LP.LandingPadBlock == LandingPad)
581d88c1a5aSDimitry Andric       return LP;
582d88c1a5aSDimitry Andric   }
583d88c1a5aSDimitry Andric 
584d88c1a5aSDimitry Andric   LandingPads.push_back(LandingPadInfo(LandingPad));
585d88c1a5aSDimitry Andric   return LandingPads[N];
586d88c1a5aSDimitry Andric }
587d88c1a5aSDimitry Andric 
588d88c1a5aSDimitry Andric void MachineFunction::addInvoke(MachineBasicBlock *LandingPad,
589d88c1a5aSDimitry Andric                                 MCSymbol *BeginLabel, MCSymbol *EndLabel) {
590d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
591d88c1a5aSDimitry Andric   LP.BeginLabels.push_back(BeginLabel);
592d88c1a5aSDimitry Andric   LP.EndLabels.push_back(EndLabel);
593d88c1a5aSDimitry Andric }
594d88c1a5aSDimitry Andric 
595d88c1a5aSDimitry Andric MCSymbol *MachineFunction::addLandingPad(MachineBasicBlock *LandingPad) {
596d88c1a5aSDimitry Andric   MCSymbol *LandingPadLabel = Ctx.createTempSymbol();
597d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
598d88c1a5aSDimitry Andric   LP.LandingPadLabel = LandingPadLabel;
599d88c1a5aSDimitry Andric   return LandingPadLabel;
600d88c1a5aSDimitry Andric }
601d88c1a5aSDimitry Andric 
602d88c1a5aSDimitry Andric void MachineFunction::addCatchTypeInfo(MachineBasicBlock *LandingPad,
603d88c1a5aSDimitry Andric                                        ArrayRef<const GlobalValue *> TyInfo) {
604d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
605d88c1a5aSDimitry Andric   for (unsigned N = TyInfo.size(); N; --N)
606d88c1a5aSDimitry Andric     LP.TypeIds.push_back(getTypeIDFor(TyInfo[N - 1]));
607d88c1a5aSDimitry Andric }
608d88c1a5aSDimitry Andric 
609d88c1a5aSDimitry Andric void MachineFunction::addFilterTypeInfo(MachineBasicBlock *LandingPad,
610d88c1a5aSDimitry Andric                                         ArrayRef<const GlobalValue *> TyInfo) {
611d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
612d88c1a5aSDimitry Andric   std::vector<unsigned> IdsInFilter(TyInfo.size());
613d88c1a5aSDimitry Andric   for (unsigned I = 0, E = TyInfo.size(); I != E; ++I)
614d88c1a5aSDimitry Andric     IdsInFilter[I] = getTypeIDFor(TyInfo[I]);
615d88c1a5aSDimitry Andric   LP.TypeIds.push_back(getFilterIDFor(IdsInFilter));
616d88c1a5aSDimitry Andric }
617d88c1a5aSDimitry Andric 
618d88c1a5aSDimitry Andric void MachineFunction::tidyLandingPads(DenseMap<MCSymbol*, uintptr_t> *LPMap) {
619d88c1a5aSDimitry Andric   for (unsigned i = 0; i != LandingPads.size(); ) {
620d88c1a5aSDimitry Andric     LandingPadInfo &LandingPad = LandingPads[i];
621d88c1a5aSDimitry Andric     if (LandingPad.LandingPadLabel &&
622d88c1a5aSDimitry Andric         !LandingPad.LandingPadLabel->isDefined() &&
623d88c1a5aSDimitry Andric         (!LPMap || (*LPMap)[LandingPad.LandingPadLabel] == 0))
624d88c1a5aSDimitry Andric       LandingPad.LandingPadLabel = nullptr;
625d88c1a5aSDimitry Andric 
626d88c1a5aSDimitry Andric     // Special case: we *should* emit LPs with null LP MBB. This indicates
627d88c1a5aSDimitry Andric     // "nounwind" case.
628d88c1a5aSDimitry Andric     if (!LandingPad.LandingPadLabel && LandingPad.LandingPadBlock) {
629d88c1a5aSDimitry Andric       LandingPads.erase(LandingPads.begin() + i);
630d88c1a5aSDimitry Andric       continue;
631d88c1a5aSDimitry Andric     }
632d88c1a5aSDimitry Andric 
633d88c1a5aSDimitry Andric     for (unsigned j = 0, e = LandingPads[i].BeginLabels.size(); j != e; ++j) {
634d88c1a5aSDimitry Andric       MCSymbol *BeginLabel = LandingPad.BeginLabels[j];
635d88c1a5aSDimitry Andric       MCSymbol *EndLabel = LandingPad.EndLabels[j];
636d88c1a5aSDimitry Andric       if ((BeginLabel->isDefined() ||
637d88c1a5aSDimitry Andric            (LPMap && (*LPMap)[BeginLabel] != 0)) &&
638d88c1a5aSDimitry Andric           (EndLabel->isDefined() ||
639d88c1a5aSDimitry Andric            (LPMap && (*LPMap)[EndLabel] != 0))) continue;
640d88c1a5aSDimitry Andric 
641d88c1a5aSDimitry Andric       LandingPad.BeginLabels.erase(LandingPad.BeginLabels.begin() + j);
642d88c1a5aSDimitry Andric       LandingPad.EndLabels.erase(LandingPad.EndLabels.begin() + j);
643d88c1a5aSDimitry Andric       --j;
644d88c1a5aSDimitry Andric       --e;
645d88c1a5aSDimitry Andric     }
646d88c1a5aSDimitry Andric 
647d88c1a5aSDimitry Andric     // Remove landing pads with no try-ranges.
648d88c1a5aSDimitry Andric     if (LandingPads[i].BeginLabels.empty()) {
649d88c1a5aSDimitry Andric       LandingPads.erase(LandingPads.begin() + i);
650d88c1a5aSDimitry Andric       continue;
651d88c1a5aSDimitry Andric     }
652d88c1a5aSDimitry Andric 
653d88c1a5aSDimitry Andric     // If there is no landing pad, ensure that the list of typeids is empty.
654d88c1a5aSDimitry Andric     // If the only typeid is a cleanup, this is the same as having no typeids.
655d88c1a5aSDimitry Andric     if (!LandingPad.LandingPadBlock ||
656d88c1a5aSDimitry Andric         (LandingPad.TypeIds.size() == 1 && !LandingPad.TypeIds[0]))
657d88c1a5aSDimitry Andric       LandingPad.TypeIds.clear();
658d88c1a5aSDimitry Andric     ++i;
659d88c1a5aSDimitry Andric   }
660d88c1a5aSDimitry Andric }
661d88c1a5aSDimitry Andric 
662d88c1a5aSDimitry Andric void MachineFunction::addCleanup(MachineBasicBlock *LandingPad) {
663d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
664d88c1a5aSDimitry Andric   LP.TypeIds.push_back(0);
665d88c1a5aSDimitry Andric }
666d88c1a5aSDimitry Andric 
667d88c1a5aSDimitry Andric void MachineFunction::addSEHCatchHandler(MachineBasicBlock *LandingPad,
668d88c1a5aSDimitry Andric                                          const Function *Filter,
669d88c1a5aSDimitry Andric                                          const BlockAddress *RecoverBA) {
670d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
671d88c1a5aSDimitry Andric   SEHHandler Handler;
672d88c1a5aSDimitry Andric   Handler.FilterOrFinally = Filter;
673d88c1a5aSDimitry Andric   Handler.RecoverBA = RecoverBA;
674d88c1a5aSDimitry Andric   LP.SEHHandlers.push_back(Handler);
675d88c1a5aSDimitry Andric }
676d88c1a5aSDimitry Andric 
677d88c1a5aSDimitry Andric void MachineFunction::addSEHCleanupHandler(MachineBasicBlock *LandingPad,
678d88c1a5aSDimitry Andric                                            const Function *Cleanup) {
679d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
680d88c1a5aSDimitry Andric   SEHHandler Handler;
681d88c1a5aSDimitry Andric   Handler.FilterOrFinally = Cleanup;
682d88c1a5aSDimitry Andric   Handler.RecoverBA = nullptr;
683d88c1a5aSDimitry Andric   LP.SEHHandlers.push_back(Handler);
684d88c1a5aSDimitry Andric }
685d88c1a5aSDimitry Andric 
686d88c1a5aSDimitry Andric void MachineFunction::setCallSiteLandingPad(MCSymbol *Sym,
687d88c1a5aSDimitry Andric                                             ArrayRef<unsigned> Sites) {
688d88c1a5aSDimitry Andric   LPadToCallSiteMap[Sym].append(Sites.begin(), Sites.end());
689d88c1a5aSDimitry Andric }
690d88c1a5aSDimitry Andric 
691d88c1a5aSDimitry Andric unsigned MachineFunction::getTypeIDFor(const GlobalValue *TI) {
692d88c1a5aSDimitry Andric   for (unsigned i = 0, N = TypeInfos.size(); i != N; ++i)
693d88c1a5aSDimitry Andric     if (TypeInfos[i] == TI) return i + 1;
694d88c1a5aSDimitry Andric 
695d88c1a5aSDimitry Andric   TypeInfos.push_back(TI);
696d88c1a5aSDimitry Andric   return TypeInfos.size();
697d88c1a5aSDimitry Andric }
698d88c1a5aSDimitry Andric 
699d88c1a5aSDimitry Andric int MachineFunction::getFilterIDFor(std::vector<unsigned> &TyIds) {
700d88c1a5aSDimitry Andric   // If the new filter coincides with the tail of an existing filter, then
701d88c1a5aSDimitry Andric   // re-use the existing filter.  Folding filters more than this requires
702d88c1a5aSDimitry Andric   // re-ordering filters and/or their elements - probably not worth it.
703d88c1a5aSDimitry Andric   for (std::vector<unsigned>::iterator I = FilterEnds.begin(),
704d88c1a5aSDimitry Andric        E = FilterEnds.end(); I != E; ++I) {
705d88c1a5aSDimitry Andric     unsigned i = *I, j = TyIds.size();
706d88c1a5aSDimitry Andric 
707d88c1a5aSDimitry Andric     while (i && j)
708d88c1a5aSDimitry Andric       if (FilterIds[--i] != TyIds[--j])
709d88c1a5aSDimitry Andric         goto try_next;
710d88c1a5aSDimitry Andric 
711d88c1a5aSDimitry Andric     if (!j)
712d88c1a5aSDimitry Andric       // The new filter coincides with range [i, end) of the existing filter.
713d88c1a5aSDimitry Andric       return -(1 + i);
714d88c1a5aSDimitry Andric 
715d88c1a5aSDimitry Andric try_next:;
716d88c1a5aSDimitry Andric   }
717d88c1a5aSDimitry Andric 
718d88c1a5aSDimitry Andric   // Add the new filter.
719d88c1a5aSDimitry Andric   int FilterID = -(1 + FilterIds.size());
720d88c1a5aSDimitry Andric   FilterIds.reserve(FilterIds.size() + TyIds.size() + 1);
721d88c1a5aSDimitry Andric   FilterIds.insert(FilterIds.end(), TyIds.begin(), TyIds.end());
722d88c1a5aSDimitry Andric   FilterEnds.push_back(FilterIds.size());
723d88c1a5aSDimitry Andric   FilterIds.push_back(0); // terminator
724d88c1a5aSDimitry Andric   return FilterID;
725d88c1a5aSDimitry Andric }
726d88c1a5aSDimitry Andric 
727d88c1a5aSDimitry Andric void llvm::addLandingPadInfo(const LandingPadInst &I, MachineBasicBlock &MBB) {
728d88c1a5aSDimitry Andric   MachineFunction &MF = *MBB.getParent();
729d88c1a5aSDimitry Andric   if (const auto *PF = dyn_cast<Function>(
730d88c1a5aSDimitry Andric           I.getParent()->getParent()->getPersonalityFn()->stripPointerCasts()))
731d88c1a5aSDimitry Andric     MF.getMMI().addPersonality(PF);
732d88c1a5aSDimitry Andric 
733d88c1a5aSDimitry Andric   if (I.isCleanup())
734d88c1a5aSDimitry Andric     MF.addCleanup(&MBB);
735d88c1a5aSDimitry Andric 
736d88c1a5aSDimitry Andric   // FIXME: New EH - Add the clauses in reverse order. This isn't 100% correct,
737d88c1a5aSDimitry Andric   //        but we need to do it this way because of how the DWARF EH emitter
738d88c1a5aSDimitry Andric   //        processes the clauses.
739d88c1a5aSDimitry Andric   for (unsigned i = I.getNumClauses(); i != 0; --i) {
740d88c1a5aSDimitry Andric     Value *Val = I.getClause(i - 1);
741d88c1a5aSDimitry Andric     if (I.isCatch(i - 1)) {
742d88c1a5aSDimitry Andric       MF.addCatchTypeInfo(&MBB,
743d88c1a5aSDimitry Andric                           dyn_cast<GlobalValue>(Val->stripPointerCasts()));
744d88c1a5aSDimitry Andric     } else {
745d88c1a5aSDimitry Andric       // Add filters in a list.
746d88c1a5aSDimitry Andric       Constant *CVal = cast<Constant>(Val);
747d88c1a5aSDimitry Andric       SmallVector<const GlobalValue *, 4> FilterList;
748d88c1a5aSDimitry Andric       for (User::op_iterator II = CVal->op_begin(), IE = CVal->op_end();
749d88c1a5aSDimitry Andric            II != IE; ++II)
750d88c1a5aSDimitry Andric         FilterList.push_back(cast<GlobalValue>((*II)->stripPointerCasts()));
751d88c1a5aSDimitry Andric 
752d88c1a5aSDimitry Andric       MF.addFilterTypeInfo(&MBB, FilterList);
753d88c1a5aSDimitry Andric     }
754d88c1a5aSDimitry Andric   }
755d88c1a5aSDimitry Andric }
756d88c1a5aSDimitry Andric 
757d88c1a5aSDimitry Andric /// \}
758d88c1a5aSDimitry Andric 
759f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
760f22ef01cSRoman Divacky //  MachineFrameInfo implementation
761f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
762f22ef01cSRoman Divacky 
7638f0fd8f6SDimitry Andric /// Make sure the function is at least Align bytes aligned.
764139f7f9bSDimitry Andric void MachineFrameInfo::ensureMaxAlignment(unsigned Align) {
7653ca95b02SDimitry Andric   if (!StackRealignable)
76639d628a0SDimitry Andric     assert(Align <= StackAlignment &&
767139f7f9bSDimitry Andric            "For targets without stack realignment, Align is out of limit!");
768139f7f9bSDimitry Andric   if (MaxAlignment < Align) MaxAlignment = Align;
769139f7f9bSDimitry Andric }
770139f7f9bSDimitry Andric 
7718f0fd8f6SDimitry Andric /// Clamp the alignment if requested and emit a warning.
772139f7f9bSDimitry Andric static inline unsigned clampStackAlignment(bool ShouldClamp, unsigned Align,
773139f7f9bSDimitry Andric                                            unsigned StackAlign) {
774139f7f9bSDimitry Andric   if (!ShouldClamp || Align <= StackAlign)
775139f7f9bSDimitry Andric     return Align;
776139f7f9bSDimitry Andric   DEBUG(dbgs() << "Warning: requested alignment " << Align
777139f7f9bSDimitry Andric                << " exceeds the stack alignment " << StackAlign
778139f7f9bSDimitry Andric                << " when stack realignment is off" << '\n');
779139f7f9bSDimitry Andric   return StackAlign;
780139f7f9bSDimitry Andric }
781139f7f9bSDimitry Andric 
7828f0fd8f6SDimitry Andric /// Create a new statically sized stack object, returning a nonnegative
7838f0fd8f6SDimitry Andric /// identifier to represent it.
784139f7f9bSDimitry Andric int MachineFrameInfo::CreateStackObject(uint64_t Size, unsigned Alignment,
78591bc56edSDimitry Andric                       bool isSS, const AllocaInst *Alloca) {
786139f7f9bSDimitry Andric   assert(Size != 0 && "Cannot allocate zero size stack objects!");
7873ca95b02SDimitry Andric   Alignment = clampStackAlignment(!StackRealignable, Alignment, StackAlignment);
78839d628a0SDimitry Andric   Objects.push_back(StackObject(Size, Alignment, 0, false, isSS, Alloca,
78939d628a0SDimitry Andric                                 !isSS));
790139f7f9bSDimitry Andric   int Index = (int)Objects.size() - NumFixedObjects - 1;
791139f7f9bSDimitry Andric   assert(Index >= 0 && "Bad frame index!");
792139f7f9bSDimitry Andric   ensureMaxAlignment(Alignment);
793139f7f9bSDimitry Andric   return Index;
794139f7f9bSDimitry Andric }
795139f7f9bSDimitry Andric 
7968f0fd8f6SDimitry Andric /// Create a new statically sized stack object that represents a spill slot,
7978f0fd8f6SDimitry Andric /// returning a nonnegative identifier to represent it.
798139f7f9bSDimitry Andric int MachineFrameInfo::CreateSpillStackObject(uint64_t Size,
799139f7f9bSDimitry Andric                                              unsigned Alignment) {
8003ca95b02SDimitry Andric   Alignment = clampStackAlignment(!StackRealignable, Alignment, StackAlignment);
80191bc56edSDimitry Andric   CreateStackObject(Size, Alignment, true);
802139f7f9bSDimitry Andric   int Index = (int)Objects.size() - NumFixedObjects - 1;
803139f7f9bSDimitry Andric   ensureMaxAlignment(Alignment);
804139f7f9bSDimitry Andric   return Index;
805139f7f9bSDimitry Andric }
806139f7f9bSDimitry Andric 
8078f0fd8f6SDimitry Andric /// Notify the MachineFrameInfo object that a variable sized object has been
8088f0fd8f6SDimitry Andric /// created. This must be created whenever a variable sized object is created,
8098f0fd8f6SDimitry Andric /// whether or not the index returned is actually used.
810*4f00c8c6SDimitry Andric int MachineFrameInfo::CreateVariableSizedObject(unsigned Alignment,
811*4f00c8c6SDimitry Andric                                                 const AllocaInst *Alloca) {
812139f7f9bSDimitry Andric   HasVarSizedObjects = true;
8133ca95b02SDimitry Andric   Alignment = clampStackAlignment(!StackRealignable, Alignment, StackAlignment);
81439d628a0SDimitry Andric   Objects.push_back(StackObject(0, Alignment, 0, false, false, Alloca, true));
815139f7f9bSDimitry Andric   ensureMaxAlignment(Alignment);
816139f7f9bSDimitry Andric   return (int)Objects.size()-NumFixedObjects-1;
817139f7f9bSDimitry Andric }
818139f7f9bSDimitry Andric 
8198f0fd8f6SDimitry Andric /// Create a new object at a fixed location on the stack.
820f22ef01cSRoman Divacky /// All fixed objects should be created before other objects are created for
821f22ef01cSRoman Divacky /// efficiency. By default, fixed objects are immutable. This returns an
822f22ef01cSRoman Divacky /// index with a negative value.
823f22ef01cSRoman Divacky int MachineFrameInfo::CreateFixedObject(uint64_t Size, int64_t SPOffset,
82439d628a0SDimitry Andric                                         bool Immutable, bool isAliased) {
825f22ef01cSRoman Divacky   assert(Size != 0 && "Cannot allocate zero size fixed stack objects!");
826ffd1746dSEd Schouten   // The alignment of the frame index can be determined from its offset from
827ffd1746dSEd Schouten   // the incoming frame position.  If the frame object is at offset 32 and
828ffd1746dSEd Schouten   // the stack is guaranteed to be 16-byte aligned, then we know that the
8293ca95b02SDimitry Andric   // object is 16-byte aligned. Note that unlike the non-fixed case, if the
8303ca95b02SDimitry Andric   // stack needs realignment, we can't assume that the stack will in fact be
8313ca95b02SDimitry Andric   // aligned.
8323ca95b02SDimitry Andric   unsigned Align = MinAlign(SPOffset, ForcedRealign ? 1 : StackAlignment);
8333ca95b02SDimitry Andric   Align = clampStackAlignment(!StackRealignable, Align, StackAlignment);
834ffd1746dSEd Schouten   Objects.insert(Objects.begin(), StackObject(Size, Align, SPOffset, Immutable,
8353861d79fSDimitry Andric                                               /*isSS*/   false,
83639d628a0SDimitry Andric                                               /*Alloca*/ nullptr, isAliased));
837f22ef01cSRoman Divacky   return -++NumFixedObjects;
838f22ef01cSRoman Divacky }
839f22ef01cSRoman Divacky 
8408f0fd8f6SDimitry Andric /// Create a spill slot at a fixed location on the stack.
8418f0fd8f6SDimitry Andric /// Returns an index with a negative value.
84291bc56edSDimitry Andric int MachineFrameInfo::CreateFixedSpillStackObject(uint64_t Size,
843d88c1a5aSDimitry Andric                                                   int64_t SPOffset,
844d88c1a5aSDimitry Andric                                                   bool Immutable) {
8453ca95b02SDimitry Andric   unsigned Align = MinAlign(SPOffset, ForcedRealign ? 1 : StackAlignment);
8463ca95b02SDimitry Andric   Align = clampStackAlignment(!StackRealignable, Align, StackAlignment);
847d88c1a5aSDimitry Andric   Objects.insert(Objects.begin(), StackObject(Size, Align, SPOffset, Immutable,
84891bc56edSDimitry Andric                                               /*isSS*/ true,
84939d628a0SDimitry Andric                                               /*Alloca*/ nullptr,
85039d628a0SDimitry Andric                                               /*isAliased*/ false));
85191bc56edSDimitry Andric   return -++NumFixedObjects;
85291bc56edSDimitry Andric }
853f22ef01cSRoman Divacky 
85497bc6c73SDimitry Andric BitVector MachineFrameInfo::getPristineRegs(const MachineFunction &MF) const {
85597bc6c73SDimitry Andric   const TargetRegisterInfo *TRI = MF.getSubtarget().getRegisterInfo();
856f22ef01cSRoman Divacky   BitVector BV(TRI->getNumRegs());
857f22ef01cSRoman Divacky 
858f22ef01cSRoman Divacky   // Before CSI is calculated, no registers are considered pristine. They can be
859f22ef01cSRoman Divacky   // freely used and PEI will make sure they are saved.
860f22ef01cSRoman Divacky   if (!isCalleeSavedInfoValid())
861f22ef01cSRoman Divacky     return BV;
862f22ef01cSRoman Divacky 
8637a7e6055SDimitry Andric   const MachineRegisterInfo &MRI = MF.getRegInfo();
8647a7e6055SDimitry Andric   for (const MCPhysReg *CSR = MRI.getCalleeSavedRegs(); CSR && *CSR;
8657a7e6055SDimitry Andric        ++CSR)
866f22ef01cSRoman Divacky     BV.set(*CSR);
867f22ef01cSRoman Divacky 
86897bc6c73SDimitry Andric   // Saved CSRs are not pristine.
8697d523365SDimitry Andric   for (auto &I : getCalleeSavedInfo())
8707d523365SDimitry Andric     for (MCSubRegIterator S(I.getReg(), TRI, true); S.isValid(); ++S)
8717d523365SDimitry Andric       BV.reset(*S);
872f22ef01cSRoman Divacky 
873f22ef01cSRoman Divacky   return BV;
874f22ef01cSRoman Divacky }
875f22ef01cSRoman Divacky 
876139f7f9bSDimitry Andric unsigned MachineFrameInfo::estimateStackSize(const MachineFunction &MF) const {
87739d628a0SDimitry Andric   const TargetFrameLowering *TFI = MF.getSubtarget().getFrameLowering();
87839d628a0SDimitry Andric   const TargetRegisterInfo *RegInfo = MF.getSubtarget().getRegisterInfo();
879139f7f9bSDimitry Andric   unsigned MaxAlign = getMaxAlignment();
880139f7f9bSDimitry Andric   int Offset = 0;
881139f7f9bSDimitry Andric 
882139f7f9bSDimitry Andric   // This code is very, very similar to PEI::calculateFrameObjectOffsets().
883139f7f9bSDimitry Andric   // It really should be refactored to share code. Until then, changes
884139f7f9bSDimitry Andric   // should keep in mind that there's tight coupling between the two.
885139f7f9bSDimitry Andric 
886139f7f9bSDimitry Andric   for (int i = getObjectIndexBegin(); i != 0; ++i) {
887139f7f9bSDimitry Andric     int FixedOff = -getObjectOffset(i);
888139f7f9bSDimitry Andric     if (FixedOff > Offset) Offset = FixedOff;
889139f7f9bSDimitry Andric   }
890139f7f9bSDimitry Andric   for (unsigned i = 0, e = getObjectIndexEnd(); i != e; ++i) {
891139f7f9bSDimitry Andric     if (isDeadObjectIndex(i))
892139f7f9bSDimitry Andric       continue;
893139f7f9bSDimitry Andric     Offset += getObjectSize(i);
894139f7f9bSDimitry Andric     unsigned Align = getObjectAlignment(i);
895139f7f9bSDimitry Andric     // Adjust to alignment boundary
896139f7f9bSDimitry Andric     Offset = (Offset+Align-1)/Align*Align;
897139f7f9bSDimitry Andric 
898139f7f9bSDimitry Andric     MaxAlign = std::max(Align, MaxAlign);
899139f7f9bSDimitry Andric   }
900139f7f9bSDimitry Andric 
901139f7f9bSDimitry Andric   if (adjustsStack() && TFI->hasReservedCallFrame(MF))
902139f7f9bSDimitry Andric     Offset += getMaxCallFrameSize();
903139f7f9bSDimitry Andric 
904139f7f9bSDimitry Andric   // Round up the size to a multiple of the alignment.  If the function has
905139f7f9bSDimitry Andric   // any calls or alloca's, align to the target's StackAlignment value to
906139f7f9bSDimitry Andric   // ensure that the callee's frame or the alloca data is suitably aligned;
907139f7f9bSDimitry Andric   // otherwise, for leaf functions, align to the TransientStackAlignment
908139f7f9bSDimitry Andric   // value.
909139f7f9bSDimitry Andric   unsigned StackAlign;
910139f7f9bSDimitry Andric   if (adjustsStack() || hasVarSizedObjects() ||
911139f7f9bSDimitry Andric       (RegInfo->needsStackRealignment(MF) && getObjectIndexEnd() != 0))
912139f7f9bSDimitry Andric     StackAlign = TFI->getStackAlignment();
913139f7f9bSDimitry Andric   else
914139f7f9bSDimitry Andric     StackAlign = TFI->getTransientStackAlignment();
915139f7f9bSDimitry Andric 
916139f7f9bSDimitry Andric   // If the frame pointer is eliminated, all frame offsets will be relative to
917139f7f9bSDimitry Andric   // SP not FP. Align to MaxAlign so this works.
918139f7f9bSDimitry Andric   StackAlign = std::max(StackAlign, MaxAlign);
919139f7f9bSDimitry Andric   unsigned AlignMask = StackAlign - 1;
920139f7f9bSDimitry Andric   Offset = (Offset + AlignMask) & ~uint64_t(AlignMask);
921139f7f9bSDimitry Andric 
922139f7f9bSDimitry Andric   return (unsigned)Offset;
923139f7f9bSDimitry Andric }
924f22ef01cSRoman Divacky 
925f22ef01cSRoman Divacky void MachineFrameInfo::print(const MachineFunction &MF, raw_ostream &OS) const{
926f22ef01cSRoman Divacky   if (Objects.empty()) return;
927f22ef01cSRoman Divacky 
92839d628a0SDimitry Andric   const TargetFrameLowering *FI = MF.getSubtarget().getFrameLowering();
929f22ef01cSRoman Divacky   int ValOffset = (FI ? FI->getOffsetOfLocalArea() : 0);
930f22ef01cSRoman Divacky 
931f22ef01cSRoman Divacky   OS << "Frame Objects:\n";
932f22ef01cSRoman Divacky 
933f22ef01cSRoman Divacky   for (unsigned i = 0, e = Objects.size(); i != e; ++i) {
934f22ef01cSRoman Divacky     const StackObject &SO = Objects[i];
935f22ef01cSRoman Divacky     OS << "  fi#" << (int)(i-NumFixedObjects) << ": ";
936f22ef01cSRoman Divacky     if (SO.Size == ~0ULL) {
937f22ef01cSRoman Divacky       OS << "dead\n";
938f22ef01cSRoman Divacky       continue;
939f22ef01cSRoman Divacky     }
940f22ef01cSRoman Divacky     if (SO.Size == 0)
941f22ef01cSRoman Divacky       OS << "variable sized";
942f22ef01cSRoman Divacky     else
943f22ef01cSRoman Divacky       OS << "size=" << SO.Size;
944f22ef01cSRoman Divacky     OS << ", align=" << SO.Alignment;
945f22ef01cSRoman Divacky 
946f22ef01cSRoman Divacky     if (i < NumFixedObjects)
947f22ef01cSRoman Divacky       OS << ", fixed";
948f22ef01cSRoman Divacky     if (i < NumFixedObjects || SO.SPOffset != -1) {
949f22ef01cSRoman Divacky       int64_t Off = SO.SPOffset - ValOffset;
950f22ef01cSRoman Divacky       OS << ", at location [SP";
951f22ef01cSRoman Divacky       if (Off > 0)
952f22ef01cSRoman Divacky         OS << "+" << Off;
953f22ef01cSRoman Divacky       else if (Off < 0)
954f22ef01cSRoman Divacky         OS << Off;
955f22ef01cSRoman Divacky       OS << "]";
956f22ef01cSRoman Divacky     }
957f22ef01cSRoman Divacky     OS << "\n";
958f22ef01cSRoman Divacky   }
959f22ef01cSRoman Divacky }
960f22ef01cSRoman Divacky 
9613861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
9627a7e6055SDimitry Andric LLVM_DUMP_METHOD void MachineFrameInfo::dump(const MachineFunction &MF) const {
963f22ef01cSRoman Divacky   print(MF, dbgs());
964f22ef01cSRoman Divacky }
9653861d79fSDimitry Andric #endif
966f22ef01cSRoman Divacky 
967f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
968f22ef01cSRoman Divacky //  MachineJumpTableInfo implementation
969f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
970f22ef01cSRoman Divacky 
9718f0fd8f6SDimitry Andric /// Return the size of each entry in the jump table.
9723861d79fSDimitry Andric unsigned MachineJumpTableInfo::getEntrySize(const DataLayout &TD) const {
973f22ef01cSRoman Divacky   // The size of a jump table entry is 4 bytes unless the entry is just the
974f22ef01cSRoman Divacky   // address of a block, in which case it is the pointer size.
975f22ef01cSRoman Divacky   switch (getEntryKind()) {
976f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_BlockAddress:
977f22ef01cSRoman Divacky     return TD.getPointerSize();
978dff0c46cSDimitry Andric   case MachineJumpTableInfo::EK_GPRel64BlockAddress:
979dff0c46cSDimitry Andric     return 8;
980f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_GPRel32BlockAddress:
981f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_LabelDifference32:
982f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_Custom32:
983f22ef01cSRoman Divacky     return 4;
984f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_Inline:
985f22ef01cSRoman Divacky     return 0;
986f22ef01cSRoman Divacky   }
987dff0c46cSDimitry Andric   llvm_unreachable("Unknown jump table encoding!");
988f22ef01cSRoman Divacky }
989f22ef01cSRoman Divacky 
9908f0fd8f6SDimitry Andric /// Return the alignment of each entry in the jump table.
9913861d79fSDimitry Andric unsigned MachineJumpTableInfo::getEntryAlignment(const DataLayout &TD) const {
992f22ef01cSRoman Divacky   // The alignment of a jump table entry is the alignment of int32 unless the
993f22ef01cSRoman Divacky   // entry is just the address of a block, in which case it is the pointer
994f22ef01cSRoman Divacky   // alignment.
995f22ef01cSRoman Divacky   switch (getEntryKind()) {
996f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_BlockAddress:
997f22ef01cSRoman Divacky     return TD.getPointerABIAlignment();
998dff0c46cSDimitry Andric   case MachineJumpTableInfo::EK_GPRel64BlockAddress:
999dff0c46cSDimitry Andric     return TD.getABIIntegerTypeAlignment(64);
1000f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_GPRel32BlockAddress:
1001f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_LabelDifference32:
1002f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_Custom32:
1003f22ef01cSRoman Divacky     return TD.getABIIntegerTypeAlignment(32);
1004f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_Inline:
1005f22ef01cSRoman Divacky     return 1;
1006f22ef01cSRoman Divacky   }
1007dff0c46cSDimitry Andric   llvm_unreachable("Unknown jump table encoding!");
1008f22ef01cSRoman Divacky }
1009f22ef01cSRoman Divacky 
10108f0fd8f6SDimitry Andric /// Create a new jump table entry in the jump table info.
1011f22ef01cSRoman Divacky unsigned MachineJumpTableInfo::createJumpTableIndex(
1012f22ef01cSRoman Divacky                                const std::vector<MachineBasicBlock*> &DestBBs) {
1013f22ef01cSRoman Divacky   assert(!DestBBs.empty() && "Cannot create an empty jump table!");
1014f22ef01cSRoman Divacky   JumpTables.push_back(MachineJumpTableEntry(DestBBs));
1015f22ef01cSRoman Divacky   return JumpTables.size()-1;
1016f22ef01cSRoman Divacky }
1017f22ef01cSRoman Divacky 
10188f0fd8f6SDimitry Andric /// If Old is the target of any jump tables, update the jump tables to branch
10198f0fd8f6SDimitry Andric /// to New instead.
1020f22ef01cSRoman Divacky bool MachineJumpTableInfo::ReplaceMBBInJumpTables(MachineBasicBlock *Old,
1021f22ef01cSRoman Divacky                                                   MachineBasicBlock *New) {
1022f22ef01cSRoman Divacky   assert(Old != New && "Not making a change?");
1023f22ef01cSRoman Divacky   bool MadeChange = false;
1024f22ef01cSRoman Divacky   for (size_t i = 0, e = JumpTables.size(); i != e; ++i)
1025f22ef01cSRoman Divacky     ReplaceMBBInJumpTable(i, Old, New);
1026f22ef01cSRoman Divacky   return MadeChange;
1027f22ef01cSRoman Divacky }
1028f22ef01cSRoman Divacky 
10298f0fd8f6SDimitry Andric /// If Old is a target of the jump tables, update the jump table to branch to
10308f0fd8f6SDimitry Andric /// New instead.
1031f22ef01cSRoman Divacky bool MachineJumpTableInfo::ReplaceMBBInJumpTable(unsigned Idx,
1032f22ef01cSRoman Divacky                                                  MachineBasicBlock *Old,
1033f22ef01cSRoman Divacky                                                  MachineBasicBlock *New) {
1034f22ef01cSRoman Divacky   assert(Old != New && "Not making a change?");
1035f22ef01cSRoman Divacky   bool MadeChange = false;
1036f22ef01cSRoman Divacky   MachineJumpTableEntry &JTE = JumpTables[Idx];
1037f22ef01cSRoman Divacky   for (size_t j = 0, e = JTE.MBBs.size(); j != e; ++j)
1038f22ef01cSRoman Divacky     if (JTE.MBBs[j] == Old) {
1039f22ef01cSRoman Divacky       JTE.MBBs[j] = New;
1040f22ef01cSRoman Divacky       MadeChange = true;
1041f22ef01cSRoman Divacky     }
1042f22ef01cSRoman Divacky   return MadeChange;
1043f22ef01cSRoman Divacky }
1044f22ef01cSRoman Divacky 
1045f22ef01cSRoman Divacky void MachineJumpTableInfo::print(raw_ostream &OS) const {
1046f22ef01cSRoman Divacky   if (JumpTables.empty()) return;
1047f22ef01cSRoman Divacky 
1048f22ef01cSRoman Divacky   OS << "Jump Tables:\n";
1049f22ef01cSRoman Divacky 
1050f22ef01cSRoman Divacky   for (unsigned i = 0, e = JumpTables.size(); i != e; ++i) {
1051f22ef01cSRoman Divacky     OS << "  jt#" << i << ": ";
1052f22ef01cSRoman Divacky     for (unsigned j = 0, f = JumpTables[i].MBBs.size(); j != f; ++j)
1053f22ef01cSRoman Divacky       OS << " BB#" << JumpTables[i].MBBs[j]->getNumber();
1054f22ef01cSRoman Divacky   }
1055f22ef01cSRoman Divacky 
1056f22ef01cSRoman Divacky   OS << '\n';
1057f22ef01cSRoman Divacky }
1058f22ef01cSRoman Divacky 
10593861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
10603ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineJumpTableInfo::dump() const { print(dbgs()); }
10613861d79fSDimitry Andric #endif
1062f22ef01cSRoman Divacky 
1063f22ef01cSRoman Divacky 
1064f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
1065f22ef01cSRoman Divacky //  MachineConstantPool implementation
1066f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
1067f22ef01cSRoman Divacky 
1068dff0c46cSDimitry Andric void MachineConstantPoolValue::anchor() { }
1069dff0c46cSDimitry Andric 
10706122f3e6SDimitry Andric Type *MachineConstantPoolEntry::getType() const {
1071f22ef01cSRoman Divacky   if (isMachineConstantPoolEntry())
1072f22ef01cSRoman Divacky     return Val.MachineCPVal->getType();
1073f22ef01cSRoman Divacky   return Val.ConstVal->getType();
1074f22ef01cSRoman Divacky }
1075f22ef01cSRoman Divacky 
10767d523365SDimitry Andric bool MachineConstantPoolEntry::needsRelocation() const {
1077f22ef01cSRoman Divacky   if (isMachineConstantPoolEntry())
10787d523365SDimitry Andric     return true;
10797d523365SDimitry Andric   return Val.ConstVal->needsRelocation();
1080f22ef01cSRoman Divacky }
1081f22ef01cSRoman Divacky 
108291bc56edSDimitry Andric SectionKind
108391bc56edSDimitry Andric MachineConstantPoolEntry::getSectionKind(const DataLayout *DL) const {
10847d523365SDimitry Andric   if (needsRelocation())
10857d523365SDimitry Andric     return SectionKind::getReadOnlyWithRel();
108691bc56edSDimitry Andric   switch (DL->getTypeAllocSize(getType())) {
108791bc56edSDimitry Andric   case 4:
10887d523365SDimitry Andric     return SectionKind::getMergeableConst4();
108991bc56edSDimitry Andric   case 8:
10907d523365SDimitry Andric     return SectionKind::getMergeableConst8();
109191bc56edSDimitry Andric   case 16:
10927d523365SDimitry Andric     return SectionKind::getMergeableConst16();
10933ca95b02SDimitry Andric   case 32:
10943ca95b02SDimitry Andric     return SectionKind::getMergeableConst32();
109591bc56edSDimitry Andric   default:
10967d523365SDimitry Andric     return SectionKind::getReadOnly();
109791bc56edSDimitry Andric   }
109891bc56edSDimitry Andric }
109991bc56edSDimitry Andric 
1100f22ef01cSRoman Divacky MachineConstantPool::~MachineConstantPool() {
1101d88c1a5aSDimitry Andric   // A constant may be a member of both Constants and MachineCPVsSharingEntries,
1102d88c1a5aSDimitry Andric   // so keep track of which we've deleted to avoid double deletions.
1103d88c1a5aSDimitry Andric   DenseSet<MachineConstantPoolValue*> Deleted;
1104f22ef01cSRoman Divacky   for (unsigned i = 0, e = Constants.size(); i != e; ++i)
1105d88c1a5aSDimitry Andric     if (Constants[i].isMachineConstantPoolEntry()) {
1106d88c1a5aSDimitry Andric       Deleted.insert(Constants[i].Val.MachineCPVal);
1107f22ef01cSRoman Divacky       delete Constants[i].Val.MachineCPVal;
1108d88c1a5aSDimitry Andric     }
1109dd6029ffSDimitry Andric   for (DenseSet<MachineConstantPoolValue*>::iterator I =
1110dd6029ffSDimitry Andric        MachineCPVsSharingEntries.begin(), E = MachineCPVsSharingEntries.end();
1111d88c1a5aSDimitry Andric        I != E; ++I) {
1112d88c1a5aSDimitry Andric     if (Deleted.count(*I) == 0)
1113dd6029ffSDimitry Andric       delete *I;
1114f22ef01cSRoman Divacky   }
1115d88c1a5aSDimitry Andric }
1116f22ef01cSRoman Divacky 
11178f0fd8f6SDimitry Andric /// Test whether the given two constants can be allocated the same constant pool
11188f0fd8f6SDimitry Andric /// entry.
1119f22ef01cSRoman Divacky static bool CanShareConstantPoolEntry(const Constant *A, const Constant *B,
1120875ed548SDimitry Andric                                       const DataLayout &DL) {
1121f22ef01cSRoman Divacky   // Handle the trivial case quickly.
1122f22ef01cSRoman Divacky   if (A == B) return true;
1123f22ef01cSRoman Divacky 
1124f22ef01cSRoman Divacky   // If they have the same type but weren't the same constant, quickly
1125f22ef01cSRoman Divacky   // reject them.
1126f22ef01cSRoman Divacky   if (A->getType() == B->getType()) return false;
1127f22ef01cSRoman Divacky 
1128dff0c46cSDimitry Andric   // We can't handle structs or arrays.
1129dff0c46cSDimitry Andric   if (isa<StructType>(A->getType()) || isa<ArrayType>(A->getType()) ||
1130dff0c46cSDimitry Andric       isa<StructType>(B->getType()) || isa<ArrayType>(B->getType()))
1131dff0c46cSDimitry Andric     return false;
1132dff0c46cSDimitry Andric 
1133f22ef01cSRoman Divacky   // For now, only support constants with the same size.
1134875ed548SDimitry Andric   uint64_t StoreSize = DL.getTypeStoreSize(A->getType());
1135875ed548SDimitry Andric   if (StoreSize != DL.getTypeStoreSize(B->getType()) || StoreSize > 128)
1136f22ef01cSRoman Divacky     return false;
1137f22ef01cSRoman Divacky 
1138dff0c46cSDimitry Andric   Type *IntTy = IntegerType::get(A->getContext(), StoreSize*8);
1139f22ef01cSRoman Divacky 
1140dff0c46cSDimitry Andric   // Try constant folding a bitcast of both instructions to an integer.  If we
1141dff0c46cSDimitry Andric   // get two identical ConstantInt's, then we are good to share them.  We use
1142dff0c46cSDimitry Andric   // the constant folding APIs to do this so that we get the benefit of
11433861d79fSDimitry Andric   // DataLayout.
1144dff0c46cSDimitry Andric   if (isa<PointerType>(A->getType()))
11453ca95b02SDimitry Andric     A = ConstantFoldCastOperand(Instruction::PtrToInt,
11463ca95b02SDimitry Andric                                 const_cast<Constant *>(A), IntTy, DL);
1147dff0c46cSDimitry Andric   else if (A->getType() != IntTy)
11483ca95b02SDimitry Andric     A = ConstantFoldCastOperand(Instruction::BitCast, const_cast<Constant *>(A),
11493ca95b02SDimitry Andric                                 IntTy, DL);
1150dff0c46cSDimitry Andric   if (isa<PointerType>(B->getType()))
11513ca95b02SDimitry Andric     B = ConstantFoldCastOperand(Instruction::PtrToInt,
11523ca95b02SDimitry Andric                                 const_cast<Constant *>(B), IntTy, DL);
1153dff0c46cSDimitry Andric   else if (B->getType() != IntTy)
11543ca95b02SDimitry Andric     B = ConstantFoldCastOperand(Instruction::BitCast, const_cast<Constant *>(B),
11553ca95b02SDimitry Andric                                 IntTy, DL);
1156f22ef01cSRoman Divacky 
1157dff0c46cSDimitry Andric   return A == B;
1158f22ef01cSRoman Divacky }
1159f22ef01cSRoman Divacky 
11608f0fd8f6SDimitry Andric /// Create a new entry in the constant pool or return an existing one.
11618f0fd8f6SDimitry Andric /// User must specify the log2 of the minimum required alignment for the object.
1162f22ef01cSRoman Divacky unsigned MachineConstantPool::getConstantPoolIndex(const Constant *C,
1163f22ef01cSRoman Divacky                                                    unsigned Alignment) {
1164f22ef01cSRoman Divacky   assert(Alignment && "Alignment must be specified!");
1165f22ef01cSRoman Divacky   if (Alignment > PoolAlignment) PoolAlignment = Alignment;
1166f22ef01cSRoman Divacky 
1167f22ef01cSRoman Divacky   // Check to see if we already have this constant.
1168f22ef01cSRoman Divacky   //
1169f22ef01cSRoman Divacky   // FIXME, this could be made much more efficient for large constant pools.
1170f22ef01cSRoman Divacky   for (unsigned i = 0, e = Constants.size(); i != e; ++i)
1171f22ef01cSRoman Divacky     if (!Constants[i].isMachineConstantPoolEntry() &&
1172875ed548SDimitry Andric         CanShareConstantPoolEntry(Constants[i].Val.ConstVal, C, DL)) {
1173f22ef01cSRoman Divacky       if ((unsigned)Constants[i].getAlignment() < Alignment)
1174f22ef01cSRoman Divacky         Constants[i].Alignment = Alignment;
1175f22ef01cSRoman Divacky       return i;
1176f22ef01cSRoman Divacky     }
1177f22ef01cSRoman Divacky 
1178f22ef01cSRoman Divacky   Constants.push_back(MachineConstantPoolEntry(C, Alignment));
1179f22ef01cSRoman Divacky   return Constants.size()-1;
1180f22ef01cSRoman Divacky }
1181f22ef01cSRoman Divacky 
1182f22ef01cSRoman Divacky unsigned MachineConstantPool::getConstantPoolIndex(MachineConstantPoolValue *V,
1183f22ef01cSRoman Divacky                                                    unsigned Alignment) {
1184f22ef01cSRoman Divacky   assert(Alignment && "Alignment must be specified!");
1185f22ef01cSRoman Divacky   if (Alignment > PoolAlignment) PoolAlignment = Alignment;
1186f22ef01cSRoman Divacky 
1187f22ef01cSRoman Divacky   // Check to see if we already have this constant.
1188f22ef01cSRoman Divacky   //
1189f22ef01cSRoman Divacky   // FIXME, this could be made much more efficient for large constant pools.
1190f22ef01cSRoman Divacky   int Idx = V->getExistingMachineCPValue(this, Alignment);
1191dd6029ffSDimitry Andric   if (Idx != -1) {
1192dd6029ffSDimitry Andric     MachineCPVsSharingEntries.insert(V);
1193f22ef01cSRoman Divacky     return (unsigned)Idx;
1194dd6029ffSDimitry Andric   }
1195f22ef01cSRoman Divacky 
1196f22ef01cSRoman Divacky   Constants.push_back(MachineConstantPoolEntry(V, Alignment));
1197f22ef01cSRoman Divacky   return Constants.size()-1;
1198f22ef01cSRoman Divacky }
1199f22ef01cSRoman Divacky 
1200f22ef01cSRoman Divacky void MachineConstantPool::print(raw_ostream &OS) const {
1201f22ef01cSRoman Divacky   if (Constants.empty()) return;
1202f22ef01cSRoman Divacky 
1203f22ef01cSRoman Divacky   OS << "Constant Pool:\n";
1204f22ef01cSRoman Divacky   for (unsigned i = 0, e = Constants.size(); i != e; ++i) {
1205f22ef01cSRoman Divacky     OS << "  cp#" << i << ": ";
1206f22ef01cSRoman Divacky     if (Constants[i].isMachineConstantPoolEntry())
1207f22ef01cSRoman Divacky       Constants[i].Val.MachineCPVal->print(OS);
1208f22ef01cSRoman Divacky     else
120991bc56edSDimitry Andric       Constants[i].Val.ConstVal->printAsOperand(OS, /*PrintType=*/false);
1210f22ef01cSRoman Divacky     OS << ", align=" << Constants[i].getAlignment();
1211f22ef01cSRoman Divacky     OS << "\n";
1212f22ef01cSRoman Divacky   }
1213f22ef01cSRoman Divacky }
1214f22ef01cSRoman Divacky 
12153861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
12163ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineConstantPool::dump() const { print(dbgs()); }
12173861d79fSDimitry Andric #endif
1218