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"
23139f7f9bSDimitry Andric #include "llvm/CodeGen/MachineFunctionPass.h"
24f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineInstr.h"
25f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineJumpTableInfo.h"
26f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineModuleInfo.h"
27f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineRegisterInfo.h"
28f22ef01cSRoman Divacky #include "llvm/CodeGen/Passes.h"
297d523365SDimitry Andric #include "llvm/CodeGen/PseudoSourceValue.h"
307d523365SDimitry Andric #include "llvm/CodeGen/WinEHFuncInfo.h"
31139f7f9bSDimitry Andric #include "llvm/IR/DataLayout.h"
3291bc56edSDimitry Andric #include "llvm/IR/DebugInfo.h"
33139f7f9bSDimitry Andric #include "llvm/IR/Function.h"
34875ed548SDimitry Andric #include "llvm/IR/Module.h"
353dac3a9bSDimitry Andric #include "llvm/IR/ModuleSlotTracker.h"
36f22ef01cSRoman Divacky #include "llvm/MC/MCAsmInfo.h"
37f22ef01cSRoman Divacky #include "llvm/MC/MCContext.h"
38f22ef01cSRoman Divacky #include "llvm/Support/Debug.h"
39f22ef01cSRoman Divacky #include "llvm/Support/GraphWriter.h"
40f22ef01cSRoman Divacky #include "llvm/Support/raw_ostream.h"
41139f7f9bSDimitry Andric #include "llvm/Target/TargetFrameLowering.h"
42139f7f9bSDimitry Andric #include "llvm/Target/TargetLowering.h"
43139f7f9bSDimitry Andric #include "llvm/Target/TargetMachine.h"
4439d628a0SDimitry Andric #include "llvm/Target/TargetSubtargetInfo.h"
45f22ef01cSRoman Divacky using namespace llvm;
46f22ef01cSRoman Divacky 
4791bc56edSDimitry Andric #define DEBUG_TYPE "codegen"
4891bc56edSDimitry Andric 
497d523365SDimitry Andric static cl::opt<unsigned>
507d523365SDimitry Andric     AlignAllFunctions("align-all-functions",
517d523365SDimitry Andric                       cl::desc("Force the alignment of all functions."),
527d523365SDimitry Andric                       cl::init(0), cl::Hidden);
537d523365SDimitry Andric 
54d88c1a5aSDimitry Andric static const char *getPropertyName(MachineFunctionProperties::Property Prop) {
55d88c1a5aSDimitry Andric   typedef MachineFunctionProperties::Property P;
56d88c1a5aSDimitry Andric   switch(Prop) {
57d88c1a5aSDimitry Andric   case P::FailedISel: return "FailedISel";
58d88c1a5aSDimitry Andric   case P::IsSSA: return "IsSSA";
59d88c1a5aSDimitry Andric   case P::Legalized: return "Legalized";
60d88c1a5aSDimitry Andric   case P::NoPHIs: return "NoPHIs";
61d88c1a5aSDimitry Andric   case P::NoVRegs: return "NoVRegs";
62d88c1a5aSDimitry Andric   case P::RegBankSelected: return "RegBankSelected";
63d88c1a5aSDimitry Andric   case P::Selected: return "Selected";
64d88c1a5aSDimitry Andric   case P::TracksLiveness: return "TracksLiveness";
65d88c1a5aSDimitry Andric   }
66d88c1a5aSDimitry Andric   llvm_unreachable("Invalid machine function property");
67d88c1a5aSDimitry Andric }
68d88c1a5aSDimitry Andric 
69d88c1a5aSDimitry Andric void MachineFunctionProperties::print(raw_ostream &OS) const {
70d88c1a5aSDimitry Andric   const char *Separator = "";
71d88c1a5aSDimitry Andric   for (BitVector::size_type I = 0; I < Properties.size(); ++I) {
72d88c1a5aSDimitry Andric     if (!Properties[I])
733ca95b02SDimitry Andric       continue;
74d88c1a5aSDimitry Andric     OS << Separator << getPropertyName(static_cast<Property>(I));
75d88c1a5aSDimitry Andric     Separator = ", ";
763ca95b02SDimitry Andric   }
773ca95b02SDimitry Andric }
783ca95b02SDimitry Andric 
79f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
80f22ef01cSRoman Divacky // MachineFunction implementation
81f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
82f22ef01cSRoman Divacky 
838f0fd8f6SDimitry Andric // Out-of-line virtual method.
84f22ef01cSRoman Divacky MachineFunctionInfo::~MachineFunctionInfo() {}
85f22ef01cSRoman Divacky 
86d88c1a5aSDimitry Andric void ilist_alloc_traits<MachineBasicBlock>::deleteNode(MachineBasicBlock *MBB) {
87f22ef01cSRoman Divacky   MBB->getParent()->DeleteMachineBasicBlock(MBB);
88f22ef01cSRoman Divacky }
89f22ef01cSRoman Divacky 
903ca95b02SDimitry Andric static inline unsigned getFnStackAlignment(const TargetSubtargetInfo *STI,
913ca95b02SDimitry Andric                                            const Function *Fn) {
923ca95b02SDimitry Andric   if (Fn->hasFnAttribute(Attribute::StackAlignment))
933ca95b02SDimitry Andric     return Fn->getFnStackAlignment();
943ca95b02SDimitry Andric   return STI->getFrameLowering()->getStackAlignment();
953ca95b02SDimitry Andric }
963ca95b02SDimitry Andric 
97f22ef01cSRoman Divacky MachineFunction::MachineFunction(const Function *F, const TargetMachine &TM,
9839d628a0SDimitry Andric                                  unsigned FunctionNum, MachineModuleInfo &mmi)
99ff0cc061SDimitry Andric     : Fn(F), Target(TM), STI(TM.getSubtargetImpl(*F)), Ctx(mmi.getContext()),
10039d628a0SDimitry Andric       MMI(mmi) {
101d88c1a5aSDimitry Andric   FunctionNumber = FunctionNum;
102d88c1a5aSDimitry Andric   init();
103d88c1a5aSDimitry Andric }
104d88c1a5aSDimitry Andric 
105d88c1a5aSDimitry Andric void MachineFunction::init() {
1063ca95b02SDimitry Andric   // Assume the function starts in SSA form with correct liveness.
1073ca95b02SDimitry Andric   Properties.set(MachineFunctionProperties::Property::IsSSA);
1083ca95b02SDimitry Andric   Properties.set(MachineFunctionProperties::Property::TracksLiveness);
10939d628a0SDimitry Andric   if (STI->getRegisterInfo())
11039d628a0SDimitry Andric     RegInfo = new (Allocator) MachineRegisterInfo(this);
111f22ef01cSRoman Divacky   else
11291bc56edSDimitry Andric     RegInfo = nullptr;
113f785676fSDimitry Andric 
11491bc56edSDimitry Andric   MFInfo = nullptr;
1153ca95b02SDimitry Andric   // We can realign the stack if the target supports it and the user hasn't
1163ca95b02SDimitry Andric   // explicitly asked us not to.
1173ca95b02SDimitry Andric   bool CanRealignSP = STI->getFrameLowering()->isStackRealignable() &&
118d88c1a5aSDimitry Andric                       !Fn->hasFnAttribute("no-realign-stack");
1193ca95b02SDimitry Andric   FrameInfo = new (Allocator) MachineFrameInfo(
1203ca95b02SDimitry Andric       getFnStackAlignment(STI, Fn), /*StackRealignable=*/CanRealignSP,
1213ca95b02SDimitry Andric       /*ForceRealign=*/CanRealignSP &&
122d88c1a5aSDimitry Andric           Fn->hasFnAttribute(Attribute::StackAlignment));
123f785676fSDimitry Andric 
124ff0cc061SDimitry Andric   if (Fn->hasFnAttribute(Attribute::StackAlignment))
125ff0cc061SDimitry Andric     FrameInfo->ensureMaxAlignment(Fn->getFnStackAlignment());
126f785676fSDimitry Andric 
127875ed548SDimitry Andric   ConstantPool = new (Allocator) MachineConstantPool(getDataLayout());
12839d628a0SDimitry Andric   Alignment = STI->getTargetLowering()->getMinFunctionAlignment();
129f785676fSDimitry Andric 
130bd5abe19SDimitry Andric   // FIXME: Shouldn't use pref alignment if explicit alignment is set on Fn.
1317d523365SDimitry Andric   // FIXME: Use Function::optForSize().
132ff0cc061SDimitry Andric   if (!Fn->hasFnAttribute(Attribute::OptimizeForSize))
133bd5abe19SDimitry Andric     Alignment = std::max(Alignment,
13439d628a0SDimitry Andric                          STI->getTargetLowering()->getPrefFunctionAlignment());
135f785676fSDimitry Andric 
1367d523365SDimitry Andric   if (AlignAllFunctions)
1377d523365SDimitry Andric     Alignment = AlignAllFunctions;
1387d523365SDimitry Andric 
13991bc56edSDimitry Andric   JumpTableInfo = nullptr;
1407d523365SDimitry Andric 
1417d523365SDimitry Andric   if (isFuncletEHPersonality(classifyEHPersonality(
142d88c1a5aSDimitry Andric           Fn->hasPersonalityFn() ? Fn->getPersonalityFn() : nullptr))) {
1437d523365SDimitry Andric     WinEHInfo = new (Allocator) WinEHFuncInfo();
1447d523365SDimitry Andric   }
1457d523365SDimitry Andric 
146d88c1a5aSDimitry Andric   assert(Target.isCompatibleDataLayout(getDataLayout()) &&
1477d523365SDimitry Andric          "Can't create a MachineFunction using a Module with a "
1487d523365SDimitry Andric          "Target-incompatible DataLayout attached\n");
1497d523365SDimitry Andric 
1507d523365SDimitry Andric   PSVManager = llvm::make_unique<PseudoSourceValueManager>();
151f22ef01cSRoman Divacky }
152f22ef01cSRoman Divacky 
153f22ef01cSRoman Divacky MachineFunction::~MachineFunction() {
154d88c1a5aSDimitry Andric   clear();
155d88c1a5aSDimitry Andric }
156d88c1a5aSDimitry Andric 
157d88c1a5aSDimitry Andric void MachineFunction::clear() {
158d88c1a5aSDimitry Andric   Properties.reset();
159139f7f9bSDimitry Andric   // Don't call destructors on MachineInstr and MachineOperand. All of their
160139f7f9bSDimitry Andric   // memory comes from the BumpPtrAllocator which is about to be purged.
161139f7f9bSDimitry Andric   //
162139f7f9bSDimitry Andric   // Do call MachineBasicBlock destructors, it contains std::vectors.
163139f7f9bSDimitry Andric   for (iterator I = begin(), E = end(); I != E; I = BasicBlocks.erase(I))
164139f7f9bSDimitry Andric     I->Insts.clearAndLeakNodesUnsafely();
165139f7f9bSDimitry Andric 
166f22ef01cSRoman Divacky   InstructionRecycler.clear(Allocator);
167139f7f9bSDimitry Andric   OperandRecycler.clear(Allocator);
168f22ef01cSRoman Divacky   BasicBlockRecycler.clear(Allocator);
1697a7e6055SDimitry Andric   VariableDbgInfos.clear();
170f22ef01cSRoman Divacky   if (RegInfo) {
171f22ef01cSRoman Divacky     RegInfo->~MachineRegisterInfo();
172f22ef01cSRoman Divacky     Allocator.Deallocate(RegInfo);
173f22ef01cSRoman Divacky   }
174f22ef01cSRoman Divacky   if (MFInfo) {
175f22ef01cSRoman Divacky     MFInfo->~MachineFunctionInfo();
176f22ef01cSRoman Divacky     Allocator.Deallocate(MFInfo);
177f22ef01cSRoman Divacky   }
1787ae0e2c9SDimitry Andric 
1797ae0e2c9SDimitry Andric   FrameInfo->~MachineFrameInfo();
1807ae0e2c9SDimitry Andric   Allocator.Deallocate(FrameInfo);
1817ae0e2c9SDimitry Andric 
1827ae0e2c9SDimitry Andric   ConstantPool->~MachineConstantPool();
1837ae0e2c9SDimitry Andric   Allocator.Deallocate(ConstantPool);
184f22ef01cSRoman Divacky 
185f22ef01cSRoman Divacky   if (JumpTableInfo) {
186f22ef01cSRoman Divacky     JumpTableInfo->~MachineJumpTableInfo();
187f22ef01cSRoman Divacky     Allocator.Deallocate(JumpTableInfo);
188f22ef01cSRoman Divacky   }
1897d523365SDimitry Andric 
1907d523365SDimitry Andric   if (WinEHInfo) {
1917d523365SDimitry Andric     WinEHInfo->~WinEHFuncInfo();
1927d523365SDimitry Andric     Allocator.Deallocate(WinEHInfo);
1937d523365SDimitry Andric   }
194f22ef01cSRoman Divacky }
195f22ef01cSRoman Divacky 
196875ed548SDimitry Andric const DataLayout &MachineFunction::getDataLayout() const {
197875ed548SDimitry Andric   return Fn->getParent()->getDataLayout();
198875ed548SDimitry Andric }
199875ed548SDimitry Andric 
2008f0fd8f6SDimitry Andric /// Get the JumpTableInfo for this function.
2018f0fd8f6SDimitry Andric /// If it does not already exist, allocate one.
202f22ef01cSRoman Divacky MachineJumpTableInfo *MachineFunction::
203f22ef01cSRoman Divacky getOrCreateJumpTableInfo(unsigned EntryKind) {
204f22ef01cSRoman Divacky   if (JumpTableInfo) return JumpTableInfo;
205f22ef01cSRoman Divacky 
206f22ef01cSRoman Divacky   JumpTableInfo = new (Allocator)
207f22ef01cSRoman Divacky     MachineJumpTableInfo((MachineJumpTableInfo::JTEntryKind)EntryKind);
208f22ef01cSRoman Divacky   return JumpTableInfo;
209f22ef01cSRoman Divacky }
210f22ef01cSRoman Divacky 
21191bc56edSDimitry Andric /// Should we be emitting segmented stack stuff for the function
2128c24ff90SDimitry Andric bool MachineFunction::shouldSplitStack() const {
21391bc56edSDimitry Andric   return getFunction()->hasFnAttribute("split-stack");
21491bc56edSDimitry Andric }
21591bc56edSDimitry Andric 
2168f0fd8f6SDimitry Andric /// This discards all of the MachineBasicBlock numbers and recomputes them.
2178f0fd8f6SDimitry Andric /// This guarantees that the MBB numbers are sequential, dense, and match the
2188f0fd8f6SDimitry Andric /// ordering of the blocks within the function.  If a specific MachineBasicBlock
2198f0fd8f6SDimitry Andric /// is specified, only that block and those after it are renumbered.
220f22ef01cSRoman Divacky void MachineFunction::RenumberBlocks(MachineBasicBlock *MBB) {
221f22ef01cSRoman Divacky   if (empty()) { MBBNumbering.clear(); return; }
222f22ef01cSRoman Divacky   MachineFunction::iterator MBBI, E = end();
22391bc56edSDimitry Andric   if (MBB == nullptr)
224f22ef01cSRoman Divacky     MBBI = begin();
225f22ef01cSRoman Divacky   else
2267d523365SDimitry Andric     MBBI = MBB->getIterator();
227f22ef01cSRoman Divacky 
228f22ef01cSRoman Divacky   // Figure out the block number this should have.
229f22ef01cSRoman Divacky   unsigned BlockNo = 0;
230f22ef01cSRoman Divacky   if (MBBI != begin())
23191bc56edSDimitry Andric     BlockNo = std::prev(MBBI)->getNumber() + 1;
232f22ef01cSRoman Divacky 
233f22ef01cSRoman Divacky   for (; MBBI != E; ++MBBI, ++BlockNo) {
234f22ef01cSRoman Divacky     if (MBBI->getNumber() != (int)BlockNo) {
235f22ef01cSRoman Divacky       // Remove use of the old number.
236f22ef01cSRoman Divacky       if (MBBI->getNumber() != -1) {
237f22ef01cSRoman Divacky         assert(MBBNumbering[MBBI->getNumber()] == &*MBBI &&
238f22ef01cSRoman Divacky                "MBB number mismatch!");
23991bc56edSDimitry Andric         MBBNumbering[MBBI->getNumber()] = nullptr;
240f22ef01cSRoman Divacky       }
241f22ef01cSRoman Divacky 
242f22ef01cSRoman Divacky       // If BlockNo is already taken, set that block's number to -1.
243f22ef01cSRoman Divacky       if (MBBNumbering[BlockNo])
244f22ef01cSRoman Divacky         MBBNumbering[BlockNo]->setNumber(-1);
245f22ef01cSRoman Divacky 
2467d523365SDimitry Andric       MBBNumbering[BlockNo] = &*MBBI;
247f22ef01cSRoman Divacky       MBBI->setNumber(BlockNo);
248f22ef01cSRoman Divacky     }
249f22ef01cSRoman Divacky   }
250f22ef01cSRoman Divacky 
251f22ef01cSRoman Divacky   // Okay, all the blocks are renumbered.  If we have compactified the block
252f22ef01cSRoman Divacky   // numbering, shrink MBBNumbering now.
253f22ef01cSRoman Divacky   assert(BlockNo <= MBBNumbering.size() && "Mismatch!");
254f22ef01cSRoman Divacky   MBBNumbering.resize(BlockNo);
255f22ef01cSRoman Divacky }
256f22ef01cSRoman Divacky 
2578f0fd8f6SDimitry Andric /// Allocate a new MachineInstr. Use this instead of `new MachineInstr'.
2583ca95b02SDimitry Andric MachineInstr *MachineFunction::CreateMachineInstr(const MCInstrDesc &MCID,
2593ca95b02SDimitry Andric                                                   const DebugLoc &DL,
2603ca95b02SDimitry Andric                                                   bool NoImp) {
261f22ef01cSRoman Divacky   return new (InstructionRecycler.Allocate<MachineInstr>(Allocator))
262139f7f9bSDimitry Andric     MachineInstr(*this, MCID, DL, NoImp);
263f22ef01cSRoman Divacky }
264f22ef01cSRoman Divacky 
2658f0fd8f6SDimitry Andric /// Create a new MachineInstr which is a copy of the 'Orig' instruction,
2668f0fd8f6SDimitry Andric /// identical in all ways except the instruction has no parent, prev, or next.
267f22ef01cSRoman Divacky MachineInstr *
268f22ef01cSRoman Divacky MachineFunction::CloneMachineInstr(const MachineInstr *Orig) {
269f22ef01cSRoman Divacky   return new (InstructionRecycler.Allocate<MachineInstr>(Allocator))
270f22ef01cSRoman Divacky              MachineInstr(*this, *Orig);
271f22ef01cSRoman Divacky }
272f22ef01cSRoman Divacky 
2738f0fd8f6SDimitry Andric /// Delete the given MachineInstr.
274f22ef01cSRoman Divacky ///
275139f7f9bSDimitry Andric /// This function also serves as the MachineInstr destructor - the real
276139f7f9bSDimitry Andric /// ~MachineInstr() destructor must be empty.
277f22ef01cSRoman Divacky void
278f22ef01cSRoman Divacky MachineFunction::DeleteMachineInstr(MachineInstr *MI) {
279139f7f9bSDimitry Andric   // Strip it for parts. The operand array and the MI object itself are
280139f7f9bSDimitry Andric   // independently recyclable.
281139f7f9bSDimitry Andric   if (MI->Operands)
282139f7f9bSDimitry Andric     deallocateOperandArray(MI->CapOperands, MI->Operands);
283139f7f9bSDimitry Andric   // Don't call ~MachineInstr() which must be trivial anyway because
284139f7f9bSDimitry Andric   // ~MachineFunction drops whole lists of MachineInstrs wihout calling their
285139f7f9bSDimitry Andric   // destructors.
286f22ef01cSRoman Divacky   InstructionRecycler.Deallocate(Allocator, MI);
287f22ef01cSRoman Divacky }
288f22ef01cSRoman Divacky 
2898f0fd8f6SDimitry Andric /// Allocate a new MachineBasicBlock. Use this instead of
2908f0fd8f6SDimitry Andric /// `new MachineBasicBlock'.
291f22ef01cSRoman Divacky MachineBasicBlock *
292f22ef01cSRoman Divacky MachineFunction::CreateMachineBasicBlock(const BasicBlock *bb) {
293f22ef01cSRoman Divacky   return new (BasicBlockRecycler.Allocate<MachineBasicBlock>(Allocator))
294f22ef01cSRoman Divacky              MachineBasicBlock(*this, bb);
295f22ef01cSRoman Divacky }
296f22ef01cSRoman Divacky 
2978f0fd8f6SDimitry Andric /// Delete the given MachineBasicBlock.
298f22ef01cSRoman Divacky void
299f22ef01cSRoman Divacky MachineFunction::DeleteMachineBasicBlock(MachineBasicBlock *MBB) {
300f22ef01cSRoman Divacky   assert(MBB->getParent() == this && "MBB parent mismatch!");
301f22ef01cSRoman Divacky   MBB->~MachineBasicBlock();
302f22ef01cSRoman Divacky   BasicBlockRecycler.Deallocate(Allocator, MBB);
303f22ef01cSRoman Divacky }
304f22ef01cSRoman Divacky 
3053ca95b02SDimitry Andric MachineMemOperand *MachineFunction::getMachineMemOperand(
3063ca95b02SDimitry Andric     MachinePointerInfo PtrInfo, MachineMemOperand::Flags f, uint64_t s,
307d88c1a5aSDimitry Andric     unsigned base_alignment, const AAMDNodes &AAInfo, const MDNode *Ranges,
308c4394386SDimitry Andric     SyncScope::ID SSID, AtomicOrdering Ordering,
309d88c1a5aSDimitry Andric     AtomicOrdering FailureOrdering) {
3103ca95b02SDimitry Andric   return new (Allocator)
311d88c1a5aSDimitry Andric       MachineMemOperand(PtrInfo, f, s, base_alignment, AAInfo, Ranges,
312c4394386SDimitry Andric                         SSID, Ordering, FailureOrdering);
313f22ef01cSRoman Divacky }
314f22ef01cSRoman Divacky 
315f22ef01cSRoman Divacky MachineMemOperand *
316f22ef01cSRoman Divacky MachineFunction::getMachineMemOperand(const MachineMemOperand *MMO,
317f22ef01cSRoman Divacky                                       int64_t Offset, uint64_t Size) {
31891bc56edSDimitry Andric   if (MMO->getValue())
319f22ef01cSRoman Divacky     return new (Allocator)
3202754fe60SDimitry Andric                MachineMemOperand(MachinePointerInfo(MMO->getValue(),
3212754fe60SDimitry Andric                                                     MMO->getOffset()+Offset),
322d88c1a5aSDimitry Andric                                  MMO->getFlags(), Size, MMO->getBaseAlignment(),
323c4394386SDimitry Andric                                  AAMDNodes(), nullptr, MMO->getSyncScopeID(),
324d88c1a5aSDimitry Andric                                  MMO->getOrdering(), MMO->getFailureOrdering());
32591bc56edSDimitry Andric   return new (Allocator)
32691bc56edSDimitry Andric              MachineMemOperand(MachinePointerInfo(MMO->getPseudoValue(),
32791bc56edSDimitry Andric                                                   MMO->getOffset()+Offset),
328d88c1a5aSDimitry Andric                                MMO->getFlags(), Size, MMO->getBaseAlignment(),
329c4394386SDimitry Andric                                AAMDNodes(), nullptr, MMO->getSyncScopeID(),
330d88c1a5aSDimitry Andric                                MMO->getOrdering(), MMO->getFailureOrdering());
331f22ef01cSRoman Divacky }
332f22ef01cSRoman Divacky 
333f22ef01cSRoman Divacky MachineInstr::mmo_iterator
334f22ef01cSRoman Divacky MachineFunction::allocateMemRefsArray(unsigned long Num) {
335f22ef01cSRoman Divacky   return Allocator.Allocate<MachineMemOperand *>(Num);
336f22ef01cSRoman Divacky }
337f22ef01cSRoman Divacky 
338f22ef01cSRoman Divacky std::pair<MachineInstr::mmo_iterator, MachineInstr::mmo_iterator>
339f22ef01cSRoman Divacky MachineFunction::extractLoadMemRefs(MachineInstr::mmo_iterator Begin,
340f22ef01cSRoman Divacky                                     MachineInstr::mmo_iterator End) {
341f22ef01cSRoman Divacky   // Count the number of load mem refs.
342f22ef01cSRoman Divacky   unsigned Num = 0;
343f22ef01cSRoman Divacky   for (MachineInstr::mmo_iterator I = Begin; I != End; ++I)
344f22ef01cSRoman Divacky     if ((*I)->isLoad())
345f22ef01cSRoman Divacky       ++Num;
346f22ef01cSRoman Divacky 
347f22ef01cSRoman Divacky   // Allocate a new array and populate it with the load information.
348f22ef01cSRoman Divacky   MachineInstr::mmo_iterator Result = allocateMemRefsArray(Num);
349f22ef01cSRoman Divacky   unsigned Index = 0;
350f22ef01cSRoman Divacky   for (MachineInstr::mmo_iterator I = Begin; I != End; ++I) {
351f22ef01cSRoman Divacky     if ((*I)->isLoad()) {
352f22ef01cSRoman Divacky       if (!(*I)->isStore())
353f22ef01cSRoman Divacky         // Reuse the MMO.
354f22ef01cSRoman Divacky         Result[Index] = *I;
355f22ef01cSRoman Divacky       else {
356f22ef01cSRoman Divacky         // Clone the MMO and unset the store flag.
357f22ef01cSRoman Divacky         MachineMemOperand *JustLoad =
3582754fe60SDimitry Andric           getMachineMemOperand((*I)->getPointerInfo(),
359f22ef01cSRoman Divacky                                (*I)->getFlags() & ~MachineMemOperand::MOStore,
3602754fe60SDimitry Andric                                (*I)->getSize(), (*I)->getBaseAlignment(),
361d88c1a5aSDimitry Andric                                (*I)->getAAInfo(), nullptr,
362c4394386SDimitry Andric                                (*I)->getSyncScopeID(), (*I)->getOrdering(),
363d88c1a5aSDimitry Andric                                (*I)->getFailureOrdering());
364f22ef01cSRoman Divacky         Result[Index] = JustLoad;
365f22ef01cSRoman Divacky       }
366f22ef01cSRoman Divacky       ++Index;
367f22ef01cSRoman Divacky     }
368f22ef01cSRoman Divacky   }
369f22ef01cSRoman Divacky   return std::make_pair(Result, Result + Num);
370f22ef01cSRoman Divacky }
371f22ef01cSRoman Divacky 
372f22ef01cSRoman Divacky std::pair<MachineInstr::mmo_iterator, MachineInstr::mmo_iterator>
373f22ef01cSRoman Divacky MachineFunction::extractStoreMemRefs(MachineInstr::mmo_iterator Begin,
374f22ef01cSRoman Divacky                                      MachineInstr::mmo_iterator End) {
375f22ef01cSRoman Divacky   // Count the number of load mem refs.
376f22ef01cSRoman Divacky   unsigned Num = 0;
377f22ef01cSRoman Divacky   for (MachineInstr::mmo_iterator I = Begin; I != End; ++I)
378f22ef01cSRoman Divacky     if ((*I)->isStore())
379f22ef01cSRoman Divacky       ++Num;
380f22ef01cSRoman Divacky 
381f22ef01cSRoman Divacky   // Allocate a new array and populate it with the store information.
382f22ef01cSRoman Divacky   MachineInstr::mmo_iterator Result = allocateMemRefsArray(Num);
383f22ef01cSRoman Divacky   unsigned Index = 0;
384f22ef01cSRoman Divacky   for (MachineInstr::mmo_iterator I = Begin; I != End; ++I) {
385f22ef01cSRoman Divacky     if ((*I)->isStore()) {
386f22ef01cSRoman Divacky       if (!(*I)->isLoad())
387f22ef01cSRoman Divacky         // Reuse the MMO.
388f22ef01cSRoman Divacky         Result[Index] = *I;
389f22ef01cSRoman Divacky       else {
390f22ef01cSRoman Divacky         // Clone the MMO and unset the load flag.
391f22ef01cSRoman Divacky         MachineMemOperand *JustStore =
3922754fe60SDimitry Andric           getMachineMemOperand((*I)->getPointerInfo(),
393f22ef01cSRoman Divacky                                (*I)->getFlags() & ~MachineMemOperand::MOLoad,
3942754fe60SDimitry Andric                                (*I)->getSize(), (*I)->getBaseAlignment(),
395d88c1a5aSDimitry Andric                                (*I)->getAAInfo(), nullptr,
396c4394386SDimitry Andric                                (*I)->getSyncScopeID(), (*I)->getOrdering(),
397d88c1a5aSDimitry Andric                                (*I)->getFailureOrdering());
398f22ef01cSRoman Divacky         Result[Index] = JustStore;
399f22ef01cSRoman Divacky       }
400f22ef01cSRoman Divacky       ++Index;
401f22ef01cSRoman Divacky     }
402f22ef01cSRoman Divacky   }
403f22ef01cSRoman Divacky   return std::make_pair(Result, Result + Num);
404f22ef01cSRoman Divacky }
405f22ef01cSRoman Divacky 
4067d523365SDimitry Andric const char *MachineFunction::createExternalSymbolName(StringRef Name) {
4077d523365SDimitry Andric   char *Dest = Allocator.Allocate<char>(Name.size() + 1);
4087d523365SDimitry Andric   std::copy(Name.begin(), Name.end(), Dest);
4097d523365SDimitry Andric   Dest[Name.size()] = 0;
4107d523365SDimitry Andric   return Dest;
4117d523365SDimitry Andric }
4127d523365SDimitry Andric 
4133861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
4143ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineFunction::dump() const {
415f22ef01cSRoman Divacky   print(dbgs());
416f22ef01cSRoman Divacky }
4173861d79fSDimitry Andric #endif
4183861d79fSDimitry Andric 
4193861d79fSDimitry Andric StringRef MachineFunction::getName() const {
4203861d79fSDimitry Andric   assert(getFunction() && "No function!");
4213861d79fSDimitry Andric   return getFunction()->getName();
4223861d79fSDimitry Andric }
423f22ef01cSRoman Divacky 
4243ca95b02SDimitry Andric void MachineFunction::print(raw_ostream &OS, const SlotIndexes *Indexes) const {
4253861d79fSDimitry Andric   OS << "# Machine code for function " << getName() << ": ";
4263ca95b02SDimitry Andric   getProperties().print(OS);
427d88c1a5aSDimitry Andric   OS << '\n';
428f22ef01cSRoman Divacky 
429f22ef01cSRoman Divacky   // Print Frame Information
430f22ef01cSRoman Divacky   FrameInfo->print(*this, OS);
431f22ef01cSRoman Divacky 
432f22ef01cSRoman Divacky   // Print JumpTable Information
433f22ef01cSRoman Divacky   if (JumpTableInfo)
434f22ef01cSRoman Divacky     JumpTableInfo->print(OS);
435f22ef01cSRoman Divacky 
436f22ef01cSRoman Divacky   // Print Constant Pool
437f22ef01cSRoman Divacky   ConstantPool->print(OS);
438f22ef01cSRoman Divacky 
43939d628a0SDimitry Andric   const TargetRegisterInfo *TRI = getSubtarget().getRegisterInfo();
440f22ef01cSRoman Divacky 
441f22ef01cSRoman Divacky   if (RegInfo && !RegInfo->livein_empty()) {
442f22ef01cSRoman Divacky     OS << "Function Live Ins: ";
443f22ef01cSRoman Divacky     for (MachineRegisterInfo::livein_iterator
444f22ef01cSRoman Divacky          I = RegInfo->livein_begin(), E = RegInfo->livein_end(); I != E; ++I) {
445bd5abe19SDimitry Andric       OS << PrintReg(I->first, TRI);
446f22ef01cSRoman Divacky       if (I->second)
447bd5abe19SDimitry Andric         OS << " in " << PrintReg(I->second, TRI);
44891bc56edSDimitry Andric       if (std::next(I) != E)
449f22ef01cSRoman Divacky         OS << ", ";
450f22ef01cSRoman Divacky     }
451f22ef01cSRoman Divacky     OS << '\n';
452f22ef01cSRoman Divacky   }
453f22ef01cSRoman Divacky 
4543dac3a9bSDimitry Andric   ModuleSlotTracker MST(getFunction()->getParent());
4553dac3a9bSDimitry Andric   MST.incorporateFunction(*getFunction());
45691bc56edSDimitry Andric   for (const auto &BB : *this) {
457f22ef01cSRoman Divacky     OS << '\n';
4583dac3a9bSDimitry Andric     BB.print(OS, MST, Indexes);
459f22ef01cSRoman Divacky   }
460f22ef01cSRoman Divacky 
4613861d79fSDimitry Andric   OS << "\n# End machine code for function " << getName() << ".\n\n";
462f22ef01cSRoman Divacky }
463f22ef01cSRoman Divacky 
464f22ef01cSRoman Divacky namespace llvm {
465f22ef01cSRoman Divacky   template<>
466f22ef01cSRoman Divacky   struct DOTGraphTraits<const MachineFunction*> : public DefaultDOTGraphTraits {
467f22ef01cSRoman Divacky 
468f22ef01cSRoman Divacky   DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {}
469f22ef01cSRoman Divacky 
470f22ef01cSRoman Divacky     static std::string getGraphName(const MachineFunction *F) {
471ff0cc061SDimitry Andric       return ("CFG for '" + F->getName() + "' function").str();
472f22ef01cSRoman Divacky     }
473f22ef01cSRoman Divacky 
474f22ef01cSRoman Divacky     std::string getNodeLabel(const MachineBasicBlock *Node,
475f22ef01cSRoman Divacky                              const MachineFunction *Graph) {
476f22ef01cSRoman Divacky       std::string OutStr;
477f22ef01cSRoman Divacky       {
478f22ef01cSRoman Divacky         raw_string_ostream OSS(OutStr);
479f22ef01cSRoman Divacky 
4802754fe60SDimitry Andric         if (isSimple()) {
4812754fe60SDimitry Andric           OSS << "BB#" << Node->getNumber();
4822754fe60SDimitry Andric           if (const BasicBlock *BB = Node->getBasicBlock())
4832754fe60SDimitry Andric             OSS << ": " << BB->getName();
4842754fe60SDimitry Andric         } else
485f22ef01cSRoman Divacky           Node->print(OSS);
486f22ef01cSRoman Divacky       }
487f22ef01cSRoman Divacky 
488f22ef01cSRoman Divacky       if (OutStr[0] == '\n') OutStr.erase(OutStr.begin());
489f22ef01cSRoman Divacky 
490f22ef01cSRoman Divacky       // Process string output to make it nicer...
491f22ef01cSRoman Divacky       for (unsigned i = 0; i != OutStr.length(); ++i)
492f22ef01cSRoman Divacky         if (OutStr[i] == '\n') {                            // Left justify
493f22ef01cSRoman Divacky           OutStr[i] = '\\';
494f22ef01cSRoman Divacky           OutStr.insert(OutStr.begin()+i+1, 'l');
495f22ef01cSRoman Divacky         }
496f22ef01cSRoman Divacky       return OutStr;
497f22ef01cSRoman Divacky     }
498f22ef01cSRoman Divacky   };
4993dac3a9bSDimitry Andric }
500f22ef01cSRoman Divacky 
501f22ef01cSRoman Divacky void MachineFunction::viewCFG() const
502f22ef01cSRoman Divacky {
503f22ef01cSRoman Divacky #ifndef NDEBUG
5043861d79fSDimitry Andric   ViewGraph(this, "mf" + getName());
505f22ef01cSRoman Divacky #else
506ffd1746dSEd Schouten   errs() << "MachineFunction::viewCFG is only available in debug builds on "
507f22ef01cSRoman Divacky          << "systems with Graphviz or gv!\n";
508f22ef01cSRoman Divacky #endif // NDEBUG
509f22ef01cSRoman Divacky }
510f22ef01cSRoman Divacky 
511f22ef01cSRoman Divacky void MachineFunction::viewCFGOnly() const
512f22ef01cSRoman Divacky {
513f22ef01cSRoman Divacky #ifndef NDEBUG
5143861d79fSDimitry Andric   ViewGraph(this, "mf" + getName(), true);
515f22ef01cSRoman Divacky #else
516ffd1746dSEd Schouten   errs() << "MachineFunction::viewCFGOnly is only available in debug builds on "
517f22ef01cSRoman Divacky          << "systems with Graphviz or gv!\n";
518f22ef01cSRoman Divacky #endif // NDEBUG
519f22ef01cSRoman Divacky }
520f22ef01cSRoman Divacky 
5218f0fd8f6SDimitry Andric /// Add the specified physical register as a live-in value and
522f22ef01cSRoman Divacky /// create a corresponding virtual register for it.
523f22ef01cSRoman Divacky unsigned MachineFunction::addLiveIn(unsigned PReg,
524dd6029ffSDimitry Andric                                     const TargetRegisterClass *RC) {
525f22ef01cSRoman Divacky   MachineRegisterInfo &MRI = getRegInfo();
526f22ef01cSRoman Divacky   unsigned VReg = MRI.getLiveInVirtReg(PReg);
527f22ef01cSRoman Divacky   if (VReg) {
52891bc56edSDimitry Andric     const TargetRegisterClass *VRegRC = MRI.getRegClass(VReg);
52991bc56edSDimitry Andric     (void)VRegRC;
53091bc56edSDimitry Andric     // A physical register can be added several times.
53191bc56edSDimitry Andric     // Between two calls, the register class of the related virtual register
53291bc56edSDimitry Andric     // may have been constrained to match some operation constraints.
53391bc56edSDimitry Andric     // In that case, check that the current register class includes the
53491bc56edSDimitry Andric     // physical register and is a sub class of the specified RC.
53591bc56edSDimitry Andric     assert((VRegRC == RC || (VRegRC->contains(PReg) &&
53691bc56edSDimitry Andric                              RC->hasSubClassEq(VRegRC))) &&
53791bc56edSDimitry Andric             "Register class mismatch!");
538f22ef01cSRoman Divacky     return VReg;
539f22ef01cSRoman Divacky   }
540f22ef01cSRoman Divacky   VReg = MRI.createVirtualRegister(RC);
541f22ef01cSRoman Divacky   MRI.addLiveIn(PReg, VReg);
542f22ef01cSRoman Divacky   return VReg;
543f22ef01cSRoman Divacky }
544f22ef01cSRoman Divacky 
5458f0fd8f6SDimitry Andric /// Return the MCSymbol for the specified non-empty jump table.
546f22ef01cSRoman Divacky /// If isLinkerPrivate is specified, an 'l' label is returned, otherwise a
547f22ef01cSRoman Divacky /// normal 'L' label is returned.
548f22ef01cSRoman Divacky MCSymbol *MachineFunction::getJTISymbol(unsigned JTI, MCContext &Ctx,
549f22ef01cSRoman Divacky                                         bool isLinkerPrivate) const {
550875ed548SDimitry Andric   const DataLayout &DL = getDataLayout();
551f22ef01cSRoman Divacky   assert(JumpTableInfo && "No jump tables");
552f22ef01cSRoman Divacky   assert(JTI < JumpTableInfo->getJumpTables().size() && "Invalid JTI!");
553f22ef01cSRoman Divacky 
554d88c1a5aSDimitry Andric   StringRef Prefix = isLinkerPrivate ? DL.getLinkerPrivateGlobalPrefix()
555875ed548SDimitry Andric                                      : DL.getPrivateGlobalPrefix();
556f22ef01cSRoman Divacky   SmallString<60> Name;
557f22ef01cSRoman Divacky   raw_svector_ostream(Name)
558f22ef01cSRoman Divacky     << Prefix << "JTI" << getFunctionNumber() << '_' << JTI;
559ff0cc061SDimitry Andric   return Ctx.getOrCreateSymbol(Name);
560f22ef01cSRoman Divacky }
561f22ef01cSRoman Divacky 
5628f0fd8f6SDimitry Andric /// Return a function-local symbol to represent the PIC base.
5632754fe60SDimitry Andric MCSymbol *MachineFunction::getPICBaseSymbol() const {
564875ed548SDimitry Andric   const DataLayout &DL = getDataLayout();
565875ed548SDimitry Andric   return Ctx.getOrCreateSymbol(Twine(DL.getPrivateGlobalPrefix()) +
5662754fe60SDimitry Andric                                Twine(getFunctionNumber()) + "$pb");
5672754fe60SDimitry Andric }
568f22ef01cSRoman Divacky 
569d88c1a5aSDimitry Andric /// \name Exception Handling
570d88c1a5aSDimitry Andric /// \{
571d88c1a5aSDimitry Andric 
572d88c1a5aSDimitry Andric LandingPadInfo &
573d88c1a5aSDimitry Andric MachineFunction::getOrCreateLandingPadInfo(MachineBasicBlock *LandingPad) {
574d88c1a5aSDimitry Andric   unsigned N = LandingPads.size();
575d88c1a5aSDimitry Andric   for (unsigned i = 0; i < N; ++i) {
576d88c1a5aSDimitry Andric     LandingPadInfo &LP = LandingPads[i];
577d88c1a5aSDimitry Andric     if (LP.LandingPadBlock == LandingPad)
578d88c1a5aSDimitry Andric       return LP;
579d88c1a5aSDimitry Andric   }
580d88c1a5aSDimitry Andric 
581d88c1a5aSDimitry Andric   LandingPads.push_back(LandingPadInfo(LandingPad));
582d88c1a5aSDimitry Andric   return LandingPads[N];
583d88c1a5aSDimitry Andric }
584d88c1a5aSDimitry Andric 
585d88c1a5aSDimitry Andric void MachineFunction::addInvoke(MachineBasicBlock *LandingPad,
586d88c1a5aSDimitry Andric                                 MCSymbol *BeginLabel, MCSymbol *EndLabel) {
587d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
588d88c1a5aSDimitry Andric   LP.BeginLabels.push_back(BeginLabel);
589d88c1a5aSDimitry Andric   LP.EndLabels.push_back(EndLabel);
590d88c1a5aSDimitry Andric }
591d88c1a5aSDimitry Andric 
592d88c1a5aSDimitry Andric MCSymbol *MachineFunction::addLandingPad(MachineBasicBlock *LandingPad) {
593d88c1a5aSDimitry Andric   MCSymbol *LandingPadLabel = Ctx.createTempSymbol();
594d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
595d88c1a5aSDimitry Andric   LP.LandingPadLabel = LandingPadLabel;
596d88c1a5aSDimitry Andric   return LandingPadLabel;
597d88c1a5aSDimitry Andric }
598d88c1a5aSDimitry Andric 
599d88c1a5aSDimitry Andric void MachineFunction::addCatchTypeInfo(MachineBasicBlock *LandingPad,
600d88c1a5aSDimitry Andric                                        ArrayRef<const GlobalValue *> TyInfo) {
601d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
602d88c1a5aSDimitry Andric   for (unsigned N = TyInfo.size(); N; --N)
603d88c1a5aSDimitry Andric     LP.TypeIds.push_back(getTypeIDFor(TyInfo[N - 1]));
604d88c1a5aSDimitry Andric }
605d88c1a5aSDimitry Andric 
606d88c1a5aSDimitry Andric void MachineFunction::addFilterTypeInfo(MachineBasicBlock *LandingPad,
607d88c1a5aSDimitry Andric                                         ArrayRef<const GlobalValue *> TyInfo) {
608d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
609d88c1a5aSDimitry Andric   std::vector<unsigned> IdsInFilter(TyInfo.size());
610d88c1a5aSDimitry Andric   for (unsigned I = 0, E = TyInfo.size(); I != E; ++I)
611d88c1a5aSDimitry Andric     IdsInFilter[I] = getTypeIDFor(TyInfo[I]);
612d88c1a5aSDimitry Andric   LP.TypeIds.push_back(getFilterIDFor(IdsInFilter));
613d88c1a5aSDimitry Andric }
614d88c1a5aSDimitry Andric 
615d88c1a5aSDimitry Andric void MachineFunction::tidyLandingPads(DenseMap<MCSymbol*, uintptr_t> *LPMap) {
616d88c1a5aSDimitry Andric   for (unsigned i = 0; i != LandingPads.size(); ) {
617d88c1a5aSDimitry Andric     LandingPadInfo &LandingPad = LandingPads[i];
618d88c1a5aSDimitry Andric     if (LandingPad.LandingPadLabel &&
619d88c1a5aSDimitry Andric         !LandingPad.LandingPadLabel->isDefined() &&
620d88c1a5aSDimitry Andric         (!LPMap || (*LPMap)[LandingPad.LandingPadLabel] == 0))
621d88c1a5aSDimitry Andric       LandingPad.LandingPadLabel = nullptr;
622d88c1a5aSDimitry Andric 
623d88c1a5aSDimitry Andric     // Special case: we *should* emit LPs with null LP MBB. This indicates
624d88c1a5aSDimitry Andric     // "nounwind" case.
625d88c1a5aSDimitry Andric     if (!LandingPad.LandingPadLabel && LandingPad.LandingPadBlock) {
626d88c1a5aSDimitry Andric       LandingPads.erase(LandingPads.begin() + i);
627d88c1a5aSDimitry Andric       continue;
628d88c1a5aSDimitry Andric     }
629d88c1a5aSDimitry Andric 
630d88c1a5aSDimitry Andric     for (unsigned j = 0, e = LandingPads[i].BeginLabels.size(); j != e; ++j) {
631d88c1a5aSDimitry Andric       MCSymbol *BeginLabel = LandingPad.BeginLabels[j];
632d88c1a5aSDimitry Andric       MCSymbol *EndLabel = LandingPad.EndLabels[j];
633d88c1a5aSDimitry Andric       if ((BeginLabel->isDefined() ||
634d88c1a5aSDimitry Andric            (LPMap && (*LPMap)[BeginLabel] != 0)) &&
635d88c1a5aSDimitry Andric           (EndLabel->isDefined() ||
636d88c1a5aSDimitry Andric            (LPMap && (*LPMap)[EndLabel] != 0))) continue;
637d88c1a5aSDimitry Andric 
638d88c1a5aSDimitry Andric       LandingPad.BeginLabels.erase(LandingPad.BeginLabels.begin() + j);
639d88c1a5aSDimitry Andric       LandingPad.EndLabels.erase(LandingPad.EndLabels.begin() + j);
640d88c1a5aSDimitry Andric       --j;
641d88c1a5aSDimitry Andric       --e;
642d88c1a5aSDimitry Andric     }
643d88c1a5aSDimitry Andric 
644d88c1a5aSDimitry Andric     // Remove landing pads with no try-ranges.
645d88c1a5aSDimitry Andric     if (LandingPads[i].BeginLabels.empty()) {
646d88c1a5aSDimitry Andric       LandingPads.erase(LandingPads.begin() + i);
647d88c1a5aSDimitry Andric       continue;
648d88c1a5aSDimitry Andric     }
649d88c1a5aSDimitry Andric 
650d88c1a5aSDimitry Andric     // If there is no landing pad, ensure that the list of typeids is empty.
651d88c1a5aSDimitry Andric     // If the only typeid is a cleanup, this is the same as having no typeids.
652d88c1a5aSDimitry Andric     if (!LandingPad.LandingPadBlock ||
653d88c1a5aSDimitry Andric         (LandingPad.TypeIds.size() == 1 && !LandingPad.TypeIds[0]))
654d88c1a5aSDimitry Andric       LandingPad.TypeIds.clear();
655d88c1a5aSDimitry Andric     ++i;
656d88c1a5aSDimitry Andric   }
657d88c1a5aSDimitry Andric }
658d88c1a5aSDimitry Andric 
659d88c1a5aSDimitry Andric void MachineFunction::addCleanup(MachineBasicBlock *LandingPad) {
660d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
661d88c1a5aSDimitry Andric   LP.TypeIds.push_back(0);
662d88c1a5aSDimitry Andric }
663d88c1a5aSDimitry Andric 
664d88c1a5aSDimitry Andric void MachineFunction::addSEHCatchHandler(MachineBasicBlock *LandingPad,
665d88c1a5aSDimitry Andric                                          const Function *Filter,
666d88c1a5aSDimitry Andric                                          const BlockAddress *RecoverBA) {
667d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
668d88c1a5aSDimitry Andric   SEHHandler Handler;
669d88c1a5aSDimitry Andric   Handler.FilterOrFinally = Filter;
670d88c1a5aSDimitry Andric   Handler.RecoverBA = RecoverBA;
671d88c1a5aSDimitry Andric   LP.SEHHandlers.push_back(Handler);
672d88c1a5aSDimitry Andric }
673d88c1a5aSDimitry Andric 
674d88c1a5aSDimitry Andric void MachineFunction::addSEHCleanupHandler(MachineBasicBlock *LandingPad,
675d88c1a5aSDimitry Andric                                            const Function *Cleanup) {
676d88c1a5aSDimitry Andric   LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad);
677d88c1a5aSDimitry Andric   SEHHandler Handler;
678d88c1a5aSDimitry Andric   Handler.FilterOrFinally = Cleanup;
679d88c1a5aSDimitry Andric   Handler.RecoverBA = nullptr;
680d88c1a5aSDimitry Andric   LP.SEHHandlers.push_back(Handler);
681d88c1a5aSDimitry Andric }
682d88c1a5aSDimitry Andric 
683d88c1a5aSDimitry Andric void MachineFunction::setCallSiteLandingPad(MCSymbol *Sym,
684d88c1a5aSDimitry Andric                                             ArrayRef<unsigned> Sites) {
685d88c1a5aSDimitry Andric   LPadToCallSiteMap[Sym].append(Sites.begin(), Sites.end());
686d88c1a5aSDimitry Andric }
687d88c1a5aSDimitry Andric 
688d88c1a5aSDimitry Andric unsigned MachineFunction::getTypeIDFor(const GlobalValue *TI) {
689d88c1a5aSDimitry Andric   for (unsigned i = 0, N = TypeInfos.size(); i != N; ++i)
690d88c1a5aSDimitry Andric     if (TypeInfos[i] == TI) return i + 1;
691d88c1a5aSDimitry Andric 
692d88c1a5aSDimitry Andric   TypeInfos.push_back(TI);
693d88c1a5aSDimitry Andric   return TypeInfos.size();
694d88c1a5aSDimitry Andric }
695d88c1a5aSDimitry Andric 
696d88c1a5aSDimitry Andric int MachineFunction::getFilterIDFor(std::vector<unsigned> &TyIds) {
697d88c1a5aSDimitry Andric   // If the new filter coincides with the tail of an existing filter, then
698d88c1a5aSDimitry Andric   // re-use the existing filter.  Folding filters more than this requires
699d88c1a5aSDimitry Andric   // re-ordering filters and/or their elements - probably not worth it.
700d88c1a5aSDimitry Andric   for (std::vector<unsigned>::iterator I = FilterEnds.begin(),
701d88c1a5aSDimitry Andric        E = FilterEnds.end(); I != E; ++I) {
702d88c1a5aSDimitry Andric     unsigned i = *I, j = TyIds.size();
703d88c1a5aSDimitry Andric 
704d88c1a5aSDimitry Andric     while (i && j)
705d88c1a5aSDimitry Andric       if (FilterIds[--i] != TyIds[--j])
706d88c1a5aSDimitry Andric         goto try_next;
707d88c1a5aSDimitry Andric 
708d88c1a5aSDimitry Andric     if (!j)
709d88c1a5aSDimitry Andric       // The new filter coincides with range [i, end) of the existing filter.
710d88c1a5aSDimitry Andric       return -(1 + i);
711d88c1a5aSDimitry Andric 
712d88c1a5aSDimitry Andric try_next:;
713d88c1a5aSDimitry Andric   }
714d88c1a5aSDimitry Andric 
715d88c1a5aSDimitry Andric   // Add the new filter.
716d88c1a5aSDimitry Andric   int FilterID = -(1 + FilterIds.size());
717d88c1a5aSDimitry Andric   FilterIds.reserve(FilterIds.size() + TyIds.size() + 1);
718d88c1a5aSDimitry Andric   FilterIds.insert(FilterIds.end(), TyIds.begin(), TyIds.end());
719d88c1a5aSDimitry Andric   FilterEnds.push_back(FilterIds.size());
720d88c1a5aSDimitry Andric   FilterIds.push_back(0); // terminator
721d88c1a5aSDimitry Andric   return FilterID;
722d88c1a5aSDimitry Andric }
723d88c1a5aSDimitry Andric 
724d88c1a5aSDimitry Andric void llvm::addLandingPadInfo(const LandingPadInst &I, MachineBasicBlock &MBB) {
725d88c1a5aSDimitry Andric   MachineFunction &MF = *MBB.getParent();
726d88c1a5aSDimitry Andric   if (const auto *PF = dyn_cast<Function>(
727d88c1a5aSDimitry Andric           I.getParent()->getParent()->getPersonalityFn()->stripPointerCasts()))
728d88c1a5aSDimitry Andric     MF.getMMI().addPersonality(PF);
729d88c1a5aSDimitry Andric 
730d88c1a5aSDimitry Andric   if (I.isCleanup())
731d88c1a5aSDimitry Andric     MF.addCleanup(&MBB);
732d88c1a5aSDimitry Andric 
733d88c1a5aSDimitry Andric   // FIXME: New EH - Add the clauses in reverse order. This isn't 100% correct,
734d88c1a5aSDimitry Andric   //        but we need to do it this way because of how the DWARF EH emitter
735d88c1a5aSDimitry Andric   //        processes the clauses.
736d88c1a5aSDimitry Andric   for (unsigned i = I.getNumClauses(); i != 0; --i) {
737d88c1a5aSDimitry Andric     Value *Val = I.getClause(i - 1);
738d88c1a5aSDimitry Andric     if (I.isCatch(i - 1)) {
739d88c1a5aSDimitry Andric       MF.addCatchTypeInfo(&MBB,
740d88c1a5aSDimitry Andric                           dyn_cast<GlobalValue>(Val->stripPointerCasts()));
741d88c1a5aSDimitry Andric     } else {
742d88c1a5aSDimitry Andric       // Add filters in a list.
743d88c1a5aSDimitry Andric       Constant *CVal = cast<Constant>(Val);
744d88c1a5aSDimitry Andric       SmallVector<const GlobalValue *, 4> FilterList;
745d88c1a5aSDimitry Andric       for (User::op_iterator II = CVal->op_begin(), IE = CVal->op_end();
746d88c1a5aSDimitry Andric            II != IE; ++II)
747d88c1a5aSDimitry Andric         FilterList.push_back(cast<GlobalValue>((*II)->stripPointerCasts()));
748d88c1a5aSDimitry Andric 
749d88c1a5aSDimitry Andric       MF.addFilterTypeInfo(&MBB, FilterList);
750d88c1a5aSDimitry Andric     }
751d88c1a5aSDimitry Andric   }
752d88c1a5aSDimitry Andric }
753d88c1a5aSDimitry Andric 
754d88c1a5aSDimitry Andric /// \}
755d88c1a5aSDimitry Andric 
756f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
757f22ef01cSRoman Divacky //  MachineJumpTableInfo implementation
758f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
759f22ef01cSRoman Divacky 
7608f0fd8f6SDimitry Andric /// Return the size of each entry in the jump table.
7613861d79fSDimitry Andric unsigned MachineJumpTableInfo::getEntrySize(const DataLayout &TD) const {
762f22ef01cSRoman Divacky   // The size of a jump table entry is 4 bytes unless the entry is just the
763f22ef01cSRoman Divacky   // address of a block, in which case it is the pointer size.
764f22ef01cSRoman Divacky   switch (getEntryKind()) {
765f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_BlockAddress:
766f22ef01cSRoman Divacky     return TD.getPointerSize();
767dff0c46cSDimitry Andric   case MachineJumpTableInfo::EK_GPRel64BlockAddress:
768dff0c46cSDimitry Andric     return 8;
769f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_GPRel32BlockAddress:
770f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_LabelDifference32:
771f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_Custom32:
772f22ef01cSRoman Divacky     return 4;
773f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_Inline:
774f22ef01cSRoman Divacky     return 0;
775f22ef01cSRoman Divacky   }
776dff0c46cSDimitry Andric   llvm_unreachable("Unknown jump table encoding!");
777f22ef01cSRoman Divacky }
778f22ef01cSRoman Divacky 
7798f0fd8f6SDimitry Andric /// Return the alignment of each entry in the jump table.
7803861d79fSDimitry Andric unsigned MachineJumpTableInfo::getEntryAlignment(const DataLayout &TD) const {
781f22ef01cSRoman Divacky   // The alignment of a jump table entry is the alignment of int32 unless the
782f22ef01cSRoman Divacky   // entry is just the address of a block, in which case it is the pointer
783f22ef01cSRoman Divacky   // alignment.
784f22ef01cSRoman Divacky   switch (getEntryKind()) {
785f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_BlockAddress:
786f22ef01cSRoman Divacky     return TD.getPointerABIAlignment();
787dff0c46cSDimitry Andric   case MachineJumpTableInfo::EK_GPRel64BlockAddress:
788dff0c46cSDimitry Andric     return TD.getABIIntegerTypeAlignment(64);
789f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_GPRel32BlockAddress:
790f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_LabelDifference32:
791f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_Custom32:
792f22ef01cSRoman Divacky     return TD.getABIIntegerTypeAlignment(32);
793f22ef01cSRoman Divacky   case MachineJumpTableInfo::EK_Inline:
794f22ef01cSRoman Divacky     return 1;
795f22ef01cSRoman Divacky   }
796dff0c46cSDimitry Andric   llvm_unreachable("Unknown jump table encoding!");
797f22ef01cSRoman Divacky }
798f22ef01cSRoman Divacky 
7998f0fd8f6SDimitry Andric /// Create a new jump table entry in the jump table info.
800f22ef01cSRoman Divacky unsigned MachineJumpTableInfo::createJumpTableIndex(
801f22ef01cSRoman Divacky                                const std::vector<MachineBasicBlock*> &DestBBs) {
802f22ef01cSRoman Divacky   assert(!DestBBs.empty() && "Cannot create an empty jump table!");
803f22ef01cSRoman Divacky   JumpTables.push_back(MachineJumpTableEntry(DestBBs));
804f22ef01cSRoman Divacky   return JumpTables.size()-1;
805f22ef01cSRoman Divacky }
806f22ef01cSRoman Divacky 
8078f0fd8f6SDimitry Andric /// If Old is the target of any jump tables, update the jump tables to branch
8088f0fd8f6SDimitry Andric /// to New instead.
809f22ef01cSRoman Divacky bool MachineJumpTableInfo::ReplaceMBBInJumpTables(MachineBasicBlock *Old,
810f22ef01cSRoman Divacky                                                   MachineBasicBlock *New) {
811f22ef01cSRoman Divacky   assert(Old != New && "Not making a change?");
812f22ef01cSRoman Divacky   bool MadeChange = false;
813f22ef01cSRoman Divacky   for (size_t i = 0, e = JumpTables.size(); i != e; ++i)
814f22ef01cSRoman Divacky     ReplaceMBBInJumpTable(i, Old, New);
815f22ef01cSRoman Divacky   return MadeChange;
816f22ef01cSRoman Divacky }
817f22ef01cSRoman Divacky 
8188f0fd8f6SDimitry Andric /// If Old is a target of the jump tables, update the jump table to branch to
8198f0fd8f6SDimitry Andric /// New instead.
820f22ef01cSRoman Divacky bool MachineJumpTableInfo::ReplaceMBBInJumpTable(unsigned Idx,
821f22ef01cSRoman Divacky                                                  MachineBasicBlock *Old,
822f22ef01cSRoman Divacky                                                  MachineBasicBlock *New) {
823f22ef01cSRoman Divacky   assert(Old != New && "Not making a change?");
824f22ef01cSRoman Divacky   bool MadeChange = false;
825f22ef01cSRoman Divacky   MachineJumpTableEntry &JTE = JumpTables[Idx];
826f22ef01cSRoman Divacky   for (size_t j = 0, e = JTE.MBBs.size(); j != e; ++j)
827f22ef01cSRoman Divacky     if (JTE.MBBs[j] == Old) {
828f22ef01cSRoman Divacky       JTE.MBBs[j] = New;
829f22ef01cSRoman Divacky       MadeChange = true;
830f22ef01cSRoman Divacky     }
831f22ef01cSRoman Divacky   return MadeChange;
832f22ef01cSRoman Divacky }
833f22ef01cSRoman Divacky 
834f22ef01cSRoman Divacky void MachineJumpTableInfo::print(raw_ostream &OS) const {
835f22ef01cSRoman Divacky   if (JumpTables.empty()) return;
836f22ef01cSRoman Divacky 
837f22ef01cSRoman Divacky   OS << "Jump Tables:\n";
838f22ef01cSRoman Divacky 
839f22ef01cSRoman Divacky   for (unsigned i = 0, e = JumpTables.size(); i != e; ++i) {
840f22ef01cSRoman Divacky     OS << "  jt#" << i << ": ";
841f22ef01cSRoman Divacky     for (unsigned j = 0, f = JumpTables[i].MBBs.size(); j != f; ++j)
842f22ef01cSRoman Divacky       OS << " BB#" << JumpTables[i].MBBs[j]->getNumber();
843f22ef01cSRoman Divacky   }
844f22ef01cSRoman Divacky 
845f22ef01cSRoman Divacky   OS << '\n';
846f22ef01cSRoman Divacky }
847f22ef01cSRoman Divacky 
8483861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
8493ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineJumpTableInfo::dump() const { print(dbgs()); }
8503861d79fSDimitry Andric #endif
851f22ef01cSRoman Divacky 
852f22ef01cSRoman Divacky 
853f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
854f22ef01cSRoman Divacky //  MachineConstantPool implementation
855f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
856f22ef01cSRoman Divacky 
857dff0c46cSDimitry Andric void MachineConstantPoolValue::anchor() { }
858dff0c46cSDimitry Andric 
8596122f3e6SDimitry Andric Type *MachineConstantPoolEntry::getType() const {
860f22ef01cSRoman Divacky   if (isMachineConstantPoolEntry())
861f22ef01cSRoman Divacky     return Val.MachineCPVal->getType();
862f22ef01cSRoman Divacky   return Val.ConstVal->getType();
863f22ef01cSRoman Divacky }
864f22ef01cSRoman Divacky 
8657d523365SDimitry Andric bool MachineConstantPoolEntry::needsRelocation() const {
866f22ef01cSRoman Divacky   if (isMachineConstantPoolEntry())
8677d523365SDimitry Andric     return true;
8687d523365SDimitry Andric   return Val.ConstVal->needsRelocation();
869f22ef01cSRoman Divacky }
870f22ef01cSRoman Divacky 
87191bc56edSDimitry Andric SectionKind
87291bc56edSDimitry Andric MachineConstantPoolEntry::getSectionKind(const DataLayout *DL) const {
8737d523365SDimitry Andric   if (needsRelocation())
8747d523365SDimitry Andric     return SectionKind::getReadOnlyWithRel();
87591bc56edSDimitry Andric   switch (DL->getTypeAllocSize(getType())) {
87691bc56edSDimitry Andric   case 4:
8777d523365SDimitry Andric     return SectionKind::getMergeableConst4();
87891bc56edSDimitry Andric   case 8:
8797d523365SDimitry Andric     return SectionKind::getMergeableConst8();
88091bc56edSDimitry Andric   case 16:
8817d523365SDimitry Andric     return SectionKind::getMergeableConst16();
8823ca95b02SDimitry Andric   case 32:
8833ca95b02SDimitry Andric     return SectionKind::getMergeableConst32();
88491bc56edSDimitry Andric   default:
8857d523365SDimitry Andric     return SectionKind::getReadOnly();
88691bc56edSDimitry Andric   }
88791bc56edSDimitry Andric }
88891bc56edSDimitry Andric 
889f22ef01cSRoman Divacky MachineConstantPool::~MachineConstantPool() {
890d88c1a5aSDimitry Andric   // A constant may be a member of both Constants and MachineCPVsSharingEntries,
891d88c1a5aSDimitry Andric   // so keep track of which we've deleted to avoid double deletions.
892d88c1a5aSDimitry Andric   DenseSet<MachineConstantPoolValue*> Deleted;
893f22ef01cSRoman Divacky   for (unsigned i = 0, e = Constants.size(); i != e; ++i)
894d88c1a5aSDimitry Andric     if (Constants[i].isMachineConstantPoolEntry()) {
895d88c1a5aSDimitry Andric       Deleted.insert(Constants[i].Val.MachineCPVal);
896f22ef01cSRoman Divacky       delete Constants[i].Val.MachineCPVal;
897d88c1a5aSDimitry Andric     }
898dd6029ffSDimitry Andric   for (DenseSet<MachineConstantPoolValue*>::iterator I =
899dd6029ffSDimitry Andric        MachineCPVsSharingEntries.begin(), E = MachineCPVsSharingEntries.end();
900d88c1a5aSDimitry Andric        I != E; ++I) {
901d88c1a5aSDimitry Andric     if (Deleted.count(*I) == 0)
902dd6029ffSDimitry Andric       delete *I;
903f22ef01cSRoman Divacky   }
904d88c1a5aSDimitry Andric }
905f22ef01cSRoman Divacky 
9068f0fd8f6SDimitry Andric /// Test whether the given two constants can be allocated the same constant pool
9078f0fd8f6SDimitry Andric /// entry.
908f22ef01cSRoman Divacky static bool CanShareConstantPoolEntry(const Constant *A, const Constant *B,
909875ed548SDimitry Andric                                       const DataLayout &DL) {
910f22ef01cSRoman Divacky   // Handle the trivial case quickly.
911f22ef01cSRoman Divacky   if (A == B) return true;
912f22ef01cSRoman Divacky 
913f22ef01cSRoman Divacky   // If they have the same type but weren't the same constant, quickly
914f22ef01cSRoman Divacky   // reject them.
915f22ef01cSRoman Divacky   if (A->getType() == B->getType()) return false;
916f22ef01cSRoman Divacky 
917dff0c46cSDimitry Andric   // We can't handle structs or arrays.
918dff0c46cSDimitry Andric   if (isa<StructType>(A->getType()) || isa<ArrayType>(A->getType()) ||
919dff0c46cSDimitry Andric       isa<StructType>(B->getType()) || isa<ArrayType>(B->getType()))
920dff0c46cSDimitry Andric     return false;
921dff0c46cSDimitry Andric 
922f22ef01cSRoman Divacky   // For now, only support constants with the same size.
923875ed548SDimitry Andric   uint64_t StoreSize = DL.getTypeStoreSize(A->getType());
924875ed548SDimitry Andric   if (StoreSize != DL.getTypeStoreSize(B->getType()) || StoreSize > 128)
925f22ef01cSRoman Divacky     return false;
926f22ef01cSRoman Divacky 
927dff0c46cSDimitry Andric   Type *IntTy = IntegerType::get(A->getContext(), StoreSize*8);
928f22ef01cSRoman Divacky 
929dff0c46cSDimitry Andric   // Try constant folding a bitcast of both instructions to an integer.  If we
930dff0c46cSDimitry Andric   // get two identical ConstantInt's, then we are good to share them.  We use
931dff0c46cSDimitry Andric   // the constant folding APIs to do this so that we get the benefit of
9323861d79fSDimitry Andric   // DataLayout.
933dff0c46cSDimitry Andric   if (isa<PointerType>(A->getType()))
9343ca95b02SDimitry Andric     A = ConstantFoldCastOperand(Instruction::PtrToInt,
9353ca95b02SDimitry Andric                                 const_cast<Constant *>(A), IntTy, DL);
936dff0c46cSDimitry Andric   else if (A->getType() != IntTy)
9373ca95b02SDimitry Andric     A = ConstantFoldCastOperand(Instruction::BitCast, const_cast<Constant *>(A),
9383ca95b02SDimitry Andric                                 IntTy, DL);
939dff0c46cSDimitry Andric   if (isa<PointerType>(B->getType()))
9403ca95b02SDimitry Andric     B = ConstantFoldCastOperand(Instruction::PtrToInt,
9413ca95b02SDimitry Andric                                 const_cast<Constant *>(B), IntTy, DL);
942dff0c46cSDimitry Andric   else if (B->getType() != IntTy)
9433ca95b02SDimitry Andric     B = ConstantFoldCastOperand(Instruction::BitCast, const_cast<Constant *>(B),
9443ca95b02SDimitry Andric                                 IntTy, DL);
945f22ef01cSRoman Divacky 
946dff0c46cSDimitry Andric   return A == B;
947f22ef01cSRoman Divacky }
948f22ef01cSRoman Divacky 
9498f0fd8f6SDimitry Andric /// Create a new entry in the constant pool or return an existing one.
9508f0fd8f6SDimitry Andric /// User must specify the log2 of the minimum required alignment for the object.
951f22ef01cSRoman Divacky unsigned MachineConstantPool::getConstantPoolIndex(const Constant *C,
952f22ef01cSRoman Divacky                                                    unsigned Alignment) {
953f22ef01cSRoman Divacky   assert(Alignment && "Alignment must be specified!");
954f22ef01cSRoman Divacky   if (Alignment > PoolAlignment) PoolAlignment = Alignment;
955f22ef01cSRoman Divacky 
956f22ef01cSRoman Divacky   // Check to see if we already have this constant.
957f22ef01cSRoman Divacky   //
958f22ef01cSRoman Divacky   // FIXME, this could be made much more efficient for large constant pools.
959f22ef01cSRoman Divacky   for (unsigned i = 0, e = Constants.size(); i != e; ++i)
960f22ef01cSRoman Divacky     if (!Constants[i].isMachineConstantPoolEntry() &&
961875ed548SDimitry Andric         CanShareConstantPoolEntry(Constants[i].Val.ConstVal, C, DL)) {
962f22ef01cSRoman Divacky       if ((unsigned)Constants[i].getAlignment() < Alignment)
963f22ef01cSRoman Divacky         Constants[i].Alignment = Alignment;
964f22ef01cSRoman Divacky       return i;
965f22ef01cSRoman Divacky     }
966f22ef01cSRoman Divacky 
967f22ef01cSRoman Divacky   Constants.push_back(MachineConstantPoolEntry(C, Alignment));
968f22ef01cSRoman Divacky   return Constants.size()-1;
969f22ef01cSRoman Divacky }
970f22ef01cSRoman Divacky 
971f22ef01cSRoman Divacky unsigned MachineConstantPool::getConstantPoolIndex(MachineConstantPoolValue *V,
972f22ef01cSRoman Divacky                                                    unsigned Alignment) {
973f22ef01cSRoman Divacky   assert(Alignment && "Alignment must be specified!");
974f22ef01cSRoman Divacky   if (Alignment > PoolAlignment) PoolAlignment = Alignment;
975f22ef01cSRoman Divacky 
976f22ef01cSRoman Divacky   // Check to see if we already have this constant.
977f22ef01cSRoman Divacky   //
978f22ef01cSRoman Divacky   // FIXME, this could be made much more efficient for large constant pools.
979f22ef01cSRoman Divacky   int Idx = V->getExistingMachineCPValue(this, Alignment);
980dd6029ffSDimitry Andric   if (Idx != -1) {
981dd6029ffSDimitry Andric     MachineCPVsSharingEntries.insert(V);
982f22ef01cSRoman Divacky     return (unsigned)Idx;
983dd6029ffSDimitry Andric   }
984f22ef01cSRoman Divacky 
985f22ef01cSRoman Divacky   Constants.push_back(MachineConstantPoolEntry(V, Alignment));
986f22ef01cSRoman Divacky   return Constants.size()-1;
987f22ef01cSRoman Divacky }
988f22ef01cSRoman Divacky 
989f22ef01cSRoman Divacky void MachineConstantPool::print(raw_ostream &OS) const {
990f22ef01cSRoman Divacky   if (Constants.empty()) return;
991f22ef01cSRoman Divacky 
992f22ef01cSRoman Divacky   OS << "Constant Pool:\n";
993f22ef01cSRoman Divacky   for (unsigned i = 0, e = Constants.size(); i != e; ++i) {
994f22ef01cSRoman Divacky     OS << "  cp#" << i << ": ";
995f22ef01cSRoman Divacky     if (Constants[i].isMachineConstantPoolEntry())
996f22ef01cSRoman Divacky       Constants[i].Val.MachineCPVal->print(OS);
997f22ef01cSRoman Divacky     else
99891bc56edSDimitry Andric       Constants[i].Val.ConstVal->printAsOperand(OS, /*PrintType=*/false);
999f22ef01cSRoman Divacky     OS << ", align=" << Constants[i].getAlignment();
1000f22ef01cSRoman Divacky     OS << "\n";
1001f22ef01cSRoman Divacky   }
1002f22ef01cSRoman Divacky }
1003f22ef01cSRoman Divacky 
10043861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
10053ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineConstantPool::dump() const { print(dbgs()); }
10063861d79fSDimitry Andric #endif
1007