blob: b96fbff88cdb67996a5f357d8f714f9f868d8798 [file] [log] [blame]
Chris Lattnerf7e79482002-04-07 22:31:46 +00001//===-- AsmWriter.cpp - Printing LLVM as an assembly file -----------------===//
Misha Brukmanb1c93172005-04-21 23:48:37 +00002//
John Criswell482202a2003-10-20 19:43:21 +00003// The LLVM Compiler Infrastructure
4//
5// This file was developed by the LLVM research group and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
Misha Brukmanb1c93172005-04-21 23:48:37 +00007//
John Criswell482202a2003-10-20 19:43:21 +00008//===----------------------------------------------------------------------===//
Chris Lattner2f7c9632001-06-06 20:29:01 +00009//
10// This library implements the functionality defined in llvm/Assembly/Writer.h
11//
Chris Lattner189088e2002-04-12 18:21:53 +000012// Note that these routines must be extremely tolerant of various errors in the
Chris Lattnerf70da102003-05-08 02:44:12 +000013// LLVM code, because it can be used for debugging transformations.
Chris Lattner189088e2002-04-12 18:21:53 +000014//
Chris Lattner2f7c9632001-06-06 20:29:01 +000015//===----------------------------------------------------------------------===//
16
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +000017#include "llvm/Assembly/Writer.h"
Chris Lattner7f8845a2002-07-23 18:07:49 +000018#include "llvm/Assembly/PrintModulePass.h"
Chris Lattner8339f7d2003-10-30 23:41:03 +000019#include "llvm/Assembly/AsmAnnotationWriter.h"
Chris Lattnerf7b6d312005-05-06 20:26:43 +000020#include "llvm/CallingConv.h"
Chris Lattnerc70b3f62004-01-20 19:50:34 +000021#include "llvm/Constants.h"
Chris Lattner913d18f2002-04-29 18:46:50 +000022#include "llvm/DerivedTypes.h"
Reid Spencer1517de32007-04-09 06:10:42 +000023#include "llvm/ParameterAttributes.h"
Chris Lattner8bbcda22006-01-25 18:57:27 +000024#include "llvm/InlineAsm.h"
Vikram S. Adveb952b542002-07-14 23:14:45 +000025#include "llvm/Instruction.h"
Misha Brukman2d3fa9e2004-07-29 16:53:53 +000026#include "llvm/Instructions.h"
Chris Lattnerc70b3f62004-01-20 19:50:34 +000027#include "llvm/Module.h"
Reid Spencer3aaaa0b2007-02-05 20:47:22 +000028#include "llvm/ValueSymbolTable.h"
Reid Spencer32af9e82007-01-06 07:24:44 +000029#include "llvm/TypeSymbolTable.h"
Reid Spencer7c16caa2004-09-01 22:55:40 +000030#include "llvm/ADT/StringExtras.h"
31#include "llvm/ADT/STLExtras.h"
Bill Wendlingdfc91892006-11-28 02:09:03 +000032#include "llvm/Support/CFG.h"
Jim Laskeyb74c6662005-08-17 19:34:49 +000033#include "llvm/Support/MathExtras.h"
Bill Wendlingdfc91892006-11-28 02:09:03 +000034#include "llvm/Support/Streams.h"
Chris Lattnerfee714f2001-09-07 16:36:04 +000035#include <algorithm>
Reid Spencerbdf03b42007-05-22 19:27:35 +000036#include <cctype>
Chris Lattner189d19f2003-11-21 20:23:48 +000037using namespace llvm;
Brian Gaeke960707c2003-11-11 22:41:34 +000038
Chris Lattner60a7dd12004-07-15 02:51:31 +000039namespace llvm {
Reid Spencer16f2f7f2004-05-26 07:18:52 +000040
Reid Spencer294715b2005-05-15 16:13:11 +000041// Make virtual table appear in this compilation unit.
42AssemblyAnnotationWriter::~AssemblyAnnotationWriter() {}
43
Reid Spencer16f2f7f2004-05-26 07:18:52 +000044/// This class provides computation of slot numbers for LLVM Assembly writing.
45/// @brief LLVM Assembly Writing Slot Computation.
46class SlotMachine {
47
48/// @name Types
49/// @{
50public:
51
52 /// @brief A mapping of Values to slot numbers
Reid Spencer50816782007-03-19 18:32:53 +000053 typedef std::map<const Value*,unsigned> ValueMap;
Reid Spencer16f2f7f2004-05-26 07:18:52 +000054
55/// @}
56/// @name Constructors
57/// @{
58public:
59 /// @brief Construct from a module
Chris Lattner23e829a2006-12-06 05:12:21 +000060 SlotMachine(const Module *M);
Reid Spencer16f2f7f2004-05-26 07:18:52 +000061
62 /// @brief Construct from a function, starting out in incorp state.
Chris Lattner23e829a2006-12-06 05:12:21 +000063 SlotMachine(const Function *F);
Reid Spencer16f2f7f2004-05-26 07:18:52 +000064
65/// @}
66/// @name Accessors
67/// @{
68public:
69 /// Return the slot number of the specified value in it's type
Chris Lattner5e043322007-01-11 03:54:27 +000070 /// plane. If something is not in the SlotMachine, return -1.
71 int getLocalSlot(const Value *V);
72 int getGlobalSlot(const GlobalValue *V);
Reid Spencer8beac692004-06-09 15:26:53 +000073
Reid Spencer16f2f7f2004-05-26 07:18:52 +000074/// @}
75/// @name Mutators
76/// @{
77public:
Misha Brukmanb1c93172005-04-21 23:48:37 +000078 /// If you'd like to deal with a function instead of just a module, use
Reid Spencer16f2f7f2004-05-26 07:18:52 +000079 /// this method to get its data into the SlotMachine.
Misha Brukmanb1c93172005-04-21 23:48:37 +000080 void incorporateFunction(const Function *F) {
81 TheFunction = F;
Reid Spencerb0ac8c42004-08-16 07:46:33 +000082 FunctionProcessed = false;
83 }
Reid Spencer16f2f7f2004-05-26 07:18:52 +000084
Misha Brukmanb1c93172005-04-21 23:48:37 +000085 /// After calling incorporateFunction, use this method to remove the
86 /// most recently incorporated function from the SlotMachine. This
Reid Spencer16f2f7f2004-05-26 07:18:52 +000087 /// will reset the state of the machine back to just the module contents.
88 void purgeFunction();
89
90/// @}
91/// @name Implementation Details
92/// @{
93private:
Reid Spencer56010e42004-05-26 21:56:09 +000094 /// This function does the actual initialization.
95 inline void initialize();
96
Chris Lattnerea862a32007-01-09 07:55:49 +000097 /// CreateModuleSlot - Insert the specified GlobalValue* into the slot table.
98 void CreateModuleSlot(const GlobalValue *V);
99
100 /// CreateFunctionSlot - Insert the specified Value* into the slot table.
101 void CreateFunctionSlot(const Value *V);
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000102
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000103 /// Add all of the module level global variables (and their initializers)
104 /// and function declarations, but not the contents of those functions.
105 void processModule();
106
Reid Spencer56010e42004-05-26 21:56:09 +0000107 /// Add all of the functions arguments, basic blocks, and instructions
108 void processFunction();
109
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000110 SlotMachine(const SlotMachine &); // DO NOT IMPLEMENT
111 void operator=(const SlotMachine &); // DO NOT IMPLEMENT
112
113/// @}
114/// @name Data
115/// @{
116public:
117
118 /// @brief The module for which we are holding slot numbers
Reid Spencer56010e42004-05-26 21:56:09 +0000119 const Module* TheModule;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000120
Reid Spencer56010e42004-05-26 21:56:09 +0000121 /// @brief The function for which we are holding slot numbers
122 const Function* TheFunction;
Reid Spencerb0ac8c42004-08-16 07:46:33 +0000123 bool FunctionProcessed;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000124
125 /// @brief The TypePlanes map for the module level data
Reid Spencer50816782007-03-19 18:32:53 +0000126 ValueMap mMap;
127 unsigned mNext;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000128
129 /// @brief The TypePlanes map for the function level data
Reid Spencer50816782007-03-19 18:32:53 +0000130 ValueMap fMap;
131 unsigned fNext;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000132
133/// @}
134
135};
136
Chris Lattner60a7dd12004-07-15 02:51:31 +0000137} // end namespace llvm
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000138
Devang Patel8c78a0b2007-05-03 01:11:54 +0000139char PrintModulePass::ID = 0;
Chris Lattnerc2d3d312006-08-27 22:42:52 +0000140static RegisterPass<PrintModulePass>
Chris Lattnere7134c52006-08-21 17:20:01 +0000141X("printm", "Print module to stderr");
Devang Patel8c78a0b2007-05-03 01:11:54 +0000142char PrintFunctionPass::ID = 0;
Chris Lattnerc2d3d312006-08-27 22:42:52 +0000143static RegisterPass<PrintFunctionPass>
Chris Lattnere7134c52006-08-21 17:20:01 +0000144Y("print","Print function to stderr");
Chris Lattner7f8845a2002-07-23 18:07:49 +0000145
Misha Brukmanb1c93172005-04-21 23:48:37 +0000146static void WriteAsOperandInternal(std::ostream &Out, const Value *V,
Chris Lattnera9f0a112006-12-06 05:50:41 +0000147 std::map<const Type *, std::string> &TypeTable,
Reid Spencer58d30f22004-07-04 11:50:43 +0000148 SlotMachine *Machine);
149
Chris Lattnerb86620e2001-10-29 16:37:48 +0000150static const Module *getModuleFromVal(const Value *V) {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000151 if (const Argument *MA = dyn_cast<Argument>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000152 return MA->getParent() ? MA->getParent()->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000153 else if (const BasicBlock *BB = dyn_cast<BasicBlock>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000154 return BB->getParent() ? BB->getParent()->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000155 else if (const Instruction *I = dyn_cast<Instruction>(V)) {
Chris Lattner57698e22002-03-26 18:01:55 +0000156 const Function *M = I->getParent() ? I->getParent()->getParent() : 0;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000157 return M ? M->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000158 } else if (const GlobalValue *GV = dyn_cast<GlobalValue>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000159 return GV->getParent();
Chris Lattnerb86620e2001-10-29 16:37:48 +0000160 return 0;
161}
162
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000163static SlotMachine *createSlotMachine(const Value *V) {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000164 if (const Argument *FA = dyn_cast<Argument>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000165 return new SlotMachine(FA->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000166 } else if (const Instruction *I = dyn_cast<Instruction>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000167 return new SlotMachine(I->getParent()->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000168 } else if (const BasicBlock *BB = dyn_cast<BasicBlock>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000169 return new SlotMachine(BB->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000170 } else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(V)){
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000171 return new SlotMachine(GV->getParent());
Anton Korobeynikova97b6942007-04-25 14:27:10 +0000172 } else if (const GlobalAlias *GA = dyn_cast<GlobalAlias>(V)){
173 return new SlotMachine(GA->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000174 } else if (const Function *Func = dyn_cast<Function>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000175 return new SlotMachine(Func);
Chris Lattner7bfee412001-10-29 16:05:51 +0000176 }
177 return 0;
178}
Chris Lattner2f7c9632001-06-06 20:29:01 +0000179
Reid Spencer788e3172007-01-26 08:02:52 +0000180/// NameNeedsQuotes - Return true if the specified llvm name should be wrapped
181/// with ""'s.
Reid Spencerbdf03b42007-05-22 19:27:35 +0000182static std::string QuoteNameIfNeeded(const std::string &Name) {
183 std::string result;
184 bool needsQuotes = Name[0] >= '0' && Name[0] <= '9';
185 // Scan the name to see if it needs quotes and to replace funky chars with
186 // their octal equivalent.
Chris Lattner1c343042003-08-22 05:40:38 +0000187 for (unsigned i = 0, e = Name.size(); i != e; ++i) {
188 char C = Name[i];
189 assert(C != '"' && "Illegal character in LLVM value name!");
Reid Spencerbdf03b42007-05-22 19:27:35 +0000190 if (isalnum(C) || C == '-' || C == '.' || C == '_')
191 result += C;
192 else if (C == '\\') {
193 needsQuotes = true;
194 result += "\\\\";
195 } else if (isprint(C)) {
196 needsQuotes = true;
197 result += C;
198 } else {
199 needsQuotes = true;
200 result += "\\";
201 char hex1 = (C >> 4) & 0x0F;
202 if (hex1 < 10)
203 result += hex1 + '0';
204 else
205 result += hex1 - 10 + 'A';
206 char hex2 = C & 0x0F;
207 if (hex2 < 10)
208 result += hex2 + '0';
209 else
210 result += hex2 - 10 + 'A';
211 }
Reid Spencer788e3172007-01-26 08:02:52 +0000212 }
Reid Spencerbdf03b42007-05-22 19:27:35 +0000213 if (needsQuotes) {
214 result.insert(0,"\"");
215 result += '"';
216 }
217 return result;
Reid Spencer788e3172007-01-26 08:02:52 +0000218}
219
220enum PrefixType {
221 GlobalPrefix,
222 LabelPrefix,
223 LocalPrefix
224};
225
226/// getLLVMName - Turn the specified string into an 'LLVM name', which is either
227/// prefixed with % (if the string only contains simple characters) or is
228/// surrounded with ""'s (if it has special chars in it).
229static std::string getLLVMName(const std::string &Name, PrefixType Prefix) {
230 assert(!Name.empty() && "Cannot get empty name!");
Reid Spencer788e3172007-01-26 08:02:52 +0000231 switch (Prefix) {
232 default: assert(0 && "Bad prefix!");
Reid Spencerbdf03b42007-05-22 19:27:35 +0000233 case GlobalPrefix: return '@' + QuoteNameIfNeeded(Name);
234 case LabelPrefix: return QuoteNameIfNeeded(Name);
235 case LocalPrefix: return '%' + QuoteNameIfNeeded(Name);
Reid Spencer788e3172007-01-26 08:02:52 +0000236 }
Chris Lattner1c343042003-08-22 05:40:38 +0000237}
238
Chris Lattnerb86620e2001-10-29 16:37:48 +0000239
Misha Brukmanc566ca362004-03-02 00:22:19 +0000240/// fillTypeNameTable - If the module has a symbol table, take all global types
241/// and stuff their names into the TypeNames map.
242///
Chris Lattnerb86620e2001-10-29 16:37:48 +0000243static void fillTypeNameTable(const Module *M,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000244 std::map<const Type *, std::string> &TypeNames) {
Chris Lattner98cf1f52002-11-20 18:36:02 +0000245 if (!M) return;
Reid Spencer32af9e82007-01-06 07:24:44 +0000246 const TypeSymbolTable &ST = M->getTypeSymbolTable();
247 TypeSymbolTable::const_iterator TI = ST.begin();
248 for (; TI != ST.end(); ++TI) {
Reid Spencere7e96712004-05-25 08:53:40 +0000249 // As a heuristic, don't insert pointer to primitive types, because
250 // they are used too often to have a single useful name.
251 //
252 const Type *Ty = cast<Type>(TI->second);
253 if (!isa<PointerType>(Ty) ||
Reid Spencer56010e42004-05-26 21:56:09 +0000254 !cast<PointerType>(Ty)->getElementType()->isPrimitiveType() ||
Chris Lattner03c49532007-01-15 02:27:26 +0000255 !cast<PointerType>(Ty)->getElementType()->isInteger() ||
Reid Spencer56010e42004-05-26 21:56:09 +0000256 isa<OpaqueType>(cast<PointerType>(Ty)->getElementType()))
Reid Spencer788e3172007-01-26 08:02:52 +0000257 TypeNames.insert(std::make_pair(Ty, getLLVMName(TI->first, LocalPrefix)));
Chris Lattnerb86620e2001-10-29 16:37:48 +0000258 }
259}
260
261
262
Misha Brukmanb1c93172005-04-21 23:48:37 +0000263static void calcTypeName(const Type *Ty,
John Criswellcd116ba2004-06-01 14:54:08 +0000264 std::vector<const Type *> &TypeStack,
265 std::map<const Type *, std::string> &TypeNames,
266 std::string & Result){
Chris Lattner03c49532007-01-15 02:27:26 +0000267 if (Ty->isInteger() || (Ty->isPrimitiveType() && !isa<OpaqueType>(Ty))) {
John Criswellcd116ba2004-06-01 14:54:08 +0000268 Result += Ty->getDescription(); // Base case
269 return;
270 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000271
272 // Check to see if the type is named.
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000273 std::map<const Type *, std::string>::iterator I = TypeNames.find(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000274 if (I != TypeNames.end()) {
275 Result += I->second;
276 return;
277 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000278
John Criswellcd116ba2004-06-01 14:54:08 +0000279 if (isa<OpaqueType>(Ty)) {
280 Result += "opaque";
281 return;
282 }
Chris Lattnerf14ead92003-10-30 00:22:33 +0000283
Chris Lattnerb86620e2001-10-29 16:37:48 +0000284 // Check to see if the Type is already on the stack...
285 unsigned Slot = 0, CurSize = TypeStack.size();
286 while (Slot < CurSize && TypeStack[Slot] != Ty) ++Slot; // Scan for type
287
Misha Brukmanb1c93172005-04-21 23:48:37 +0000288 // This is another base case for the recursion. In this case, we know
Chris Lattnerb86620e2001-10-29 16:37:48 +0000289 // that we have looped back to a type that we have previously visited.
290 // Generate the appropriate upreference to handle this.
John Criswellcd116ba2004-06-01 14:54:08 +0000291 if (Slot < CurSize) {
292 Result += "\\" + utostr(CurSize-Slot); // Here's the upreference
293 return;
294 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000295
296 TypeStack.push_back(Ty); // Recursive case: Add us to the stack..
Misha Brukmanb1c93172005-04-21 23:48:37 +0000297
Chris Lattner6b727592004-06-17 18:19:28 +0000298 switch (Ty->getTypeID()) {
Reid Spencer7a9c62b2007-01-12 07:05:14 +0000299 case Type::IntegerTyID: {
300 unsigned BitWidth = cast<IntegerType>(Ty)->getBitWidth();
Reid Spencerc8721592007-01-12 07:25:20 +0000301 Result += "i" + utostr(BitWidth);
Reid Spencer7a9c62b2007-01-12 07:05:14 +0000302 break;
303 }
Chris Lattner91db5822002-03-29 03:44:36 +0000304 case Type::FunctionTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000305 const FunctionType *FTy = cast<FunctionType>(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000306 calcTypeName(FTy->getReturnType(), TypeStack, TypeNames, Result);
307 Result += " (";
Reid Spencer8c4914c2006-12-31 05:24:50 +0000308 unsigned Idx = 1;
Reid Spencer1517de32007-04-09 06:10:42 +0000309 const ParamAttrsList *Attrs = FTy->getParamAttrs();
Chris Lattnerfa829be2004-02-09 04:14:01 +0000310 for (FunctionType::param_iterator I = FTy->param_begin(),
311 E = FTy->param_end(); I != E; ++I) {
312 if (I != FTy->param_begin())
Chris Lattnerb86620e2001-10-29 16:37:48 +0000313 Result += ", ";
John Criswellcd116ba2004-06-01 14:54:08 +0000314 calcTypeName(*I, TypeStack, TypeNames, Result);
Reid Spencera472f662007-04-11 02:44:20 +0000315 if (Attrs && Attrs->getParamAttrs(Idx) != ParamAttr::None) {
Reid Spencer8c4914c2006-12-31 05:24:50 +0000316 Result += + " ";
Reid Spencer1517de32007-04-09 06:10:42 +0000317 Result += Attrs->getParamAttrsTextByIndex(Idx);
Reid Spencer8c4914c2006-12-31 05:24:50 +0000318 }
319 Idx++;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000320 }
Chris Lattnerd816b532002-04-13 20:53:41 +0000321 if (FTy->isVarArg()) {
Chris Lattnerfa829be2004-02-09 04:14:01 +0000322 if (FTy->getNumParams()) Result += ", ";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000323 Result += "...";
324 }
325 Result += ")";
Reid Spencera472f662007-04-11 02:44:20 +0000326 if (Attrs && Attrs->getParamAttrs(0) != ParamAttr::None) {
Reid Spencer136a91c2007-01-05 17:06:19 +0000327 Result += " ";
Reid Spencer1517de32007-04-09 06:10:42 +0000328 Result += Attrs->getParamAttrsTextByIndex(0);
Reid Spencer136a91c2007-01-05 17:06:19 +0000329 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000330 break;
331 }
332 case Type::StructTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000333 const StructType *STy = cast<StructType>(Ty);
Andrew Lenharthdcb3c972006-12-08 18:06:16 +0000334 if (STy->isPacked())
335 Result += '<';
John Criswellcd116ba2004-06-01 14:54:08 +0000336 Result += "{ ";
Chris Lattnerac6db752004-02-09 04:37:31 +0000337 for (StructType::element_iterator I = STy->element_begin(),
338 E = STy->element_end(); I != E; ++I) {
339 if (I != STy->element_begin())
Chris Lattnerb86620e2001-10-29 16:37:48 +0000340 Result += ", ";
John Criswellcd116ba2004-06-01 14:54:08 +0000341 calcTypeName(*I, TypeStack, TypeNames, Result);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000342 }
343 Result += " }";
Andrew Lenharthdcb3c972006-12-08 18:06:16 +0000344 if (STy->isPacked())
345 Result += '>';
Chris Lattnerb86620e2001-10-29 16:37:48 +0000346 break;
347 }
348 case Type::PointerTyID:
Misha Brukmanb1c93172005-04-21 23:48:37 +0000349 calcTypeName(cast<PointerType>(Ty)->getElementType(),
John Criswellcd116ba2004-06-01 14:54:08 +0000350 TypeStack, TypeNames, Result);
351 Result += "*";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000352 break;
353 case Type::ArrayTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000354 const ArrayType *ATy = cast<ArrayType>(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000355 Result += "[" + utostr(ATy->getNumElements()) + " x ";
356 calcTypeName(ATy->getElementType(), TypeStack, TypeNames, Result);
357 Result += "]";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000358 break;
359 }
Reid Spencerd84d35b2007-02-15 02:26:10 +0000360 case Type::VectorTyID: {
361 const VectorType *PTy = cast<VectorType>(Ty);
Brian Gaeke02209042004-08-20 06:00:58 +0000362 Result += "<" + utostr(PTy->getNumElements()) + " x ";
363 calcTypeName(PTy->getElementType(), TypeStack, TypeNames, Result);
364 Result += ">";
365 break;
366 }
Chris Lattner15285ab2003-05-14 17:50:47 +0000367 case Type::OpaqueTyID:
John Criswellcd116ba2004-06-01 14:54:08 +0000368 Result += "opaque";
Chris Lattner15285ab2003-05-14 17:50:47 +0000369 break;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000370 default:
John Criswellcd116ba2004-06-01 14:54:08 +0000371 Result += "<unrecognized-type>";
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000372 break;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000373 }
374
375 TypeStack.pop_back(); // Remove self from stack...
Chris Lattnerb86620e2001-10-29 16:37:48 +0000376}
377
378
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000379/// printTypeInt - The internal guts of printing out a type that has a
380/// potentially named portion.
381///
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000382static std::ostream &printTypeInt(std::ostream &Out, const Type *Ty,
383 std::map<const Type *, std::string> &TypeNames) {
Chris Lattnerb86620e2001-10-29 16:37:48 +0000384 // Primitive types always print out their description, regardless of whether
385 // they have been named or not.
386 //
Chris Lattner03c49532007-01-15 02:27:26 +0000387 if (Ty->isInteger() || (Ty->isPrimitiveType() && !isa<OpaqueType>(Ty)))
Chris Lattner92d60532003-10-30 00:12:51 +0000388 return Out << Ty->getDescription();
Chris Lattnerb86620e2001-10-29 16:37:48 +0000389
390 // Check to see if the type is named.
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000391 std::map<const Type *, std::string>::iterator I = TypeNames.find(Ty);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000392 if (I != TypeNames.end()) return Out << I->second;
393
394 // Otherwise we have a type that has not been named but is a derived type.
395 // Carefully recurse the type hierarchy to print out any contained symbolic
396 // names.
397 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000398 std::vector<const Type *> TypeStack;
John Criswellcd116ba2004-06-01 14:54:08 +0000399 std::string TypeName;
400 calcTypeName(Ty, TypeStack, TypeNames, TypeName);
Chris Lattner7f74a562002-01-20 22:54:45 +0000401 TypeNames.insert(std::make_pair(Ty, TypeName));//Cache type name for later use
John Criswellcd116ba2004-06-01 14:54:08 +0000402 return (Out << TypeName);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000403}
404
Chris Lattner34b95182001-10-31 04:33:19 +0000405
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000406/// WriteTypeSymbolic - This attempts to write the specified type as a symbolic
407/// type, iff there is an entry in the modules symbol table for the specified
408/// type or one of it's component types. This is slower than a simple x << Type
409///
Chris Lattner189d19f2003-11-21 20:23:48 +0000410std::ostream &llvm::WriteTypeSymbolic(std::ostream &Out, const Type *Ty,
411 const Module *M) {
Misha Brukmanb1c93172005-04-21 23:48:37 +0000412 Out << ' ';
Chris Lattnerb86620e2001-10-29 16:37:48 +0000413
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000414 // If they want us to print out a type, but there is no context, we can't
415 // print it symbolically.
416 if (!M)
Chris Lattner72f866e2001-10-29 16:40:32 +0000417 return Out << Ty->getDescription();
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000418
419 std::map<const Type *, std::string> TypeNames;
420 fillTypeNameTable(M, TypeNames);
421 return printTypeInt(Out, Ty, TypeNames);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000422}
423
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000424// PrintEscapedString - Print each character of the specified string, escaping
425// it if it is not printable or if it is an escape char.
426static void PrintEscapedString(const std::string &Str, std::ostream &Out) {
427 for (unsigned i = 0, e = Str.size(); i != e; ++i) {
428 unsigned char C = Str[i];
429 if (isprint(C) && C != '"' && C != '\\') {
430 Out << C;
431 } else {
432 Out << '\\'
433 << (char) ((C/16 < 10) ? ( C/16 +'0') : ( C/16 -10+'A'))
434 << (char)(((C&15) < 10) ? ((C&15)+'0') : ((C&15)-10+'A'));
435 }
436 }
437}
438
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000439static const char *getPredicateText(unsigned predicate) {
Reid Spencer812a1be2006-12-04 05:19:18 +0000440 const char * pred = "unknown";
441 switch (predicate) {
442 case FCmpInst::FCMP_FALSE: pred = "false"; break;
443 case FCmpInst::FCMP_OEQ: pred = "oeq"; break;
444 case FCmpInst::FCMP_OGT: pred = "ogt"; break;
445 case FCmpInst::FCMP_OGE: pred = "oge"; break;
446 case FCmpInst::FCMP_OLT: pred = "olt"; break;
447 case FCmpInst::FCMP_OLE: pred = "ole"; break;
448 case FCmpInst::FCMP_ONE: pred = "one"; break;
449 case FCmpInst::FCMP_ORD: pred = "ord"; break;
450 case FCmpInst::FCMP_UNO: pred = "uno"; break;
451 case FCmpInst::FCMP_UEQ: pred = "ueq"; break;
452 case FCmpInst::FCMP_UGT: pred = "ugt"; break;
453 case FCmpInst::FCMP_UGE: pred = "uge"; break;
454 case FCmpInst::FCMP_ULT: pred = "ult"; break;
455 case FCmpInst::FCMP_ULE: pred = "ule"; break;
456 case FCmpInst::FCMP_UNE: pred = "une"; break;
457 case FCmpInst::FCMP_TRUE: pred = "true"; break;
458 case ICmpInst::ICMP_EQ: pred = "eq"; break;
459 case ICmpInst::ICMP_NE: pred = "ne"; break;
460 case ICmpInst::ICMP_SGT: pred = "sgt"; break;
461 case ICmpInst::ICMP_SGE: pred = "sge"; break;
462 case ICmpInst::ICMP_SLT: pred = "slt"; break;
463 case ICmpInst::ICMP_SLE: pred = "sle"; break;
464 case ICmpInst::ICMP_UGT: pred = "ugt"; break;
465 case ICmpInst::ICMP_UGE: pred = "uge"; break;
466 case ICmpInst::ICMP_ULT: pred = "ult"; break;
467 case ICmpInst::ICMP_ULE: pred = "ule"; break;
468 }
469 return pred;
470}
471
Misha Brukmanb1c93172005-04-21 23:48:37 +0000472/// @brief Internal constant writer.
473static void WriteConstantInt(std::ostream &Out, const Constant *CV,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000474 std::map<const Type *, std::string> &TypeTable,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000475 SlotMachine *Machine) {
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000476 const int IndentSize = 4;
477 static std::string Indent = "\n";
Zhou Sheng75b871f2007-01-11 12:24:14 +0000478 if (const ConstantInt *CI = dyn_cast<ConstantInt>(CV)) {
Reid Spencer542964f2007-01-11 18:21:29 +0000479 if (CI->getType() == Type::Int1Ty)
Reid Spencercddc9df2007-01-12 04:24:46 +0000480 Out << (CI->getZExtValue() ? "true" : "false");
481 else
Reid Spencere2eb1fa2007-02-27 20:25:25 +0000482 Out << CI->getValue().toStringSigned(10);
Chris Lattner1e194682002-04-18 18:53:13 +0000483 } else if (const ConstantFP *CFP = dyn_cast<ConstantFP>(CV)) {
Dale Johannesen028084e2007-09-12 03:30:33 +0000484 if (&CFP->getValueAPF().getSemantics() == &APFloat::IEEEdouble ||
485 &CFP->getValueAPF().getSemantics() == &APFloat::IEEEsingle) {
486 // We would like to output the FP constant value in exponential notation,
487 // but we cannot do this if doing so will lose precision. Check here to
488 // make sure that we only output it in exponential format if we can parse
489 // the value back and get the same value.
490 //
491 bool isDouble = &CFP->getValueAPF().getSemantics()==&APFloat::IEEEdouble;
492 double Val = (isDouble) ? CFP->getValueAPF().convertToDouble() :
493 CFP->getValueAPF().convertToFloat();
494 std::string StrVal = ftostr(CFP->getValueAPF());
Chris Lattner1e194682002-04-18 18:53:13 +0000495
Dale Johannesen028084e2007-09-12 03:30:33 +0000496 // Check to make sure that the stringized number is not some string like
497 // "Inf" or NaN, that atof will accept, but the lexer will not. Check
498 // that the string matches the "[-+]?[0-9]" regex.
499 //
500 if ((StrVal[0] >= '0' && StrVal[0] <= '9') ||
501 ((StrVal[0] == '-' || StrVal[0] == '+') &&
502 (StrVal[1] >= '0' && StrVal[1] <= '9'))) {
503 // Reparse stringized version!
504 if (atof(StrVal.c_str()) == Val) {
505 Out << StrVal;
506 return;
507 }
Chris Lattner1e194682002-04-18 18:53:13 +0000508 }
Dale Johannesen028084e2007-09-12 03:30:33 +0000509 // Otherwise we could not reparse it to exactly the same value, so we must
510 // output the string in hexadecimal format!
511 assert(sizeof(double) == sizeof(uint64_t) &&
512 "assuming that double is 64 bits!");
513 Out << "0x" << utohexstr(DoubleToBits(Val));
514 } else {
515 // Some form of long double. These appear as a magic letter identifying
516 // the type, then a fixed number of hex digits.
517 Out << "0x";
518 if (&CFP->getValueAPF().getSemantics() == &APFloat::x87DoubleExtended)
519 Out << 'K';
520 else if (&CFP->getValueAPF().getSemantics() == &APFloat::IEEEquad)
521 Out << 'L';
522 else
523 assert(0 && "Unsupported floating point type");
524 const uint64_t* p = CFP->getValueAPF().convertToAPInt().getRawData();
525 uint64_t word = *p;
526 int shiftcount=60;
527 int width = CFP->getValueAPF().convertToAPInt().getBitWidth();
528 for (int j=0; j<width; j+=4, shiftcount-=4) {
529 unsigned int nibble = (word>>shiftcount) & 15;
530 if (nibble < 10)
531 Out << (unsigned char)(nibble + '0');
532 else
533 Out << (unsigned char)(nibble - 10 + 'A');
534 if (shiftcount == 0) {
535 word = *(++p);
536 shiftcount = 60;
537 if (width-j-4 < 64)
538 shiftcount = width-j-4;
539 }
540 }
541 }
Chris Lattner76b2ff42004-02-15 05:55:15 +0000542 } else if (isa<ConstantAggregateZero>(CV)) {
543 Out << "zeroinitializer";
Chris Lattner1e194682002-04-18 18:53:13 +0000544 } else if (const ConstantArray *CA = dyn_cast<ConstantArray>(CV)) {
545 // As a special case, print the array as a string if it is an array of
546 // ubytes or an array of sbytes with positive values.
Misha Brukmanb1c93172005-04-21 23:48:37 +0000547 //
Chris Lattner1e194682002-04-18 18:53:13 +0000548 const Type *ETy = CA->getType()->getElementType();
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000549 if (CA->isString()) {
Chris Lattner1e194682002-04-18 18:53:13 +0000550 Out << "c\"";
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000551 PrintEscapedString(CA->getAsString(), Out);
Chris Lattner1e194682002-04-18 18:53:13 +0000552 Out << "\"";
553
554 } else { // Cannot output in string format...
Misha Brukman21bbdb92004-06-04 21:11:51 +0000555 Out << '[';
Chris Lattnerd84bb632002-04-16 21:36:08 +0000556 if (CA->getNumOperands()) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000557 Out << ' ';
Chris Lattner1e194682002-04-18 18:53:13 +0000558 printTypeInt(Out, ETy, TypeTable);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000559 WriteAsOperandInternal(Out, CA->getOperand(0),
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000560 TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000561 for (unsigned i = 1, e = CA->getNumOperands(); i != e; ++i) {
562 Out << ", ";
Chris Lattner1e194682002-04-18 18:53:13 +0000563 printTypeInt(Out, ETy, TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000564 WriteAsOperandInternal(Out, CA->getOperand(i), TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000565 }
566 }
567 Out << " ]";
568 }
569 } else if (const ConstantStruct *CS = dyn_cast<ConstantStruct>(CV)) {
Andrew Lenharth0d124b82007-01-08 18:21:30 +0000570 if (CS->getType()->isPacked())
571 Out << '<';
Misha Brukman21bbdb92004-06-04 21:11:51 +0000572 Out << '{';
Jim Laskey3bb78742006-02-25 12:27:03 +0000573 unsigned N = CS->getNumOperands();
574 if (N) {
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000575 if (N > 2) {
576 Indent += std::string(IndentSize, ' ');
577 Out << Indent;
578 } else {
579 Out << ' ';
580 }
Chris Lattnerd84bb632002-04-16 21:36:08 +0000581 printTypeInt(Out, CS->getOperand(0)->getType(), TypeTable);
582
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000583 WriteAsOperandInternal(Out, CS->getOperand(0), TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000584
Jim Laskey3bb78742006-02-25 12:27:03 +0000585 for (unsigned i = 1; i < N; i++) {
Chris Lattnerd84bb632002-04-16 21:36:08 +0000586 Out << ", ";
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000587 if (N > 2) Out << Indent;
Chris Lattnerd84bb632002-04-16 21:36:08 +0000588 printTypeInt(Out, CS->getOperand(i)->getType(), TypeTable);
589
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000590 WriteAsOperandInternal(Out, CS->getOperand(i), TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000591 }
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000592 if (N > 2) Indent.resize(Indent.size() - IndentSize);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000593 }
Jim Laskey3bb78742006-02-25 12:27:03 +0000594
Chris Lattnerd84bb632002-04-16 21:36:08 +0000595 Out << " }";
Andrew Lenharth0d124b82007-01-08 18:21:30 +0000596 if (CS->getType()->isPacked())
597 Out << '>';
Reid Spencerd84d35b2007-02-15 02:26:10 +0000598 } else if (const ConstantVector *CP = dyn_cast<ConstantVector>(CV)) {
Brian Gaeke02209042004-08-20 06:00:58 +0000599 const Type *ETy = CP->getType()->getElementType();
Misha Brukmanb1c93172005-04-21 23:48:37 +0000600 assert(CP->getNumOperands() > 0 &&
Brian Gaeke02209042004-08-20 06:00:58 +0000601 "Number of operands for a PackedConst must be > 0");
602 Out << '<';
603 Out << ' ';
604 printTypeInt(Out, ETy, TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000605 WriteAsOperandInternal(Out, CP->getOperand(0), TypeTable, Machine);
Brian Gaeke02209042004-08-20 06:00:58 +0000606 for (unsigned i = 1, e = CP->getNumOperands(); i != e; ++i) {
607 Out << ", ";
608 printTypeInt(Out, ETy, TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000609 WriteAsOperandInternal(Out, CP->getOperand(i), TypeTable, Machine);
Brian Gaeke02209042004-08-20 06:00:58 +0000610 }
611 Out << " >";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000612 } else if (isa<ConstantPointerNull>(CV)) {
613 Out << "null";
614
Chris Lattner5e0b9f22004-10-16 18:08:06 +0000615 } else if (isa<UndefValue>(CV)) {
616 Out << "undef";
617
Chris Lattner3cd8c562002-07-30 18:54:25 +0000618 } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CV)) {
Reid Spencer812a1be2006-12-04 05:19:18 +0000619 Out << CE->getOpcodeName();
620 if (CE->isCompare())
621 Out << " " << getPredicateText(CE->getPredicate());
622 Out << " (";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000623
Vikram S. Adveb952b542002-07-14 23:14:45 +0000624 for (User::const_op_iterator OI=CE->op_begin(); OI != CE->op_end(); ++OI) {
625 printTypeInt(Out, (*OI)->getType(), TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000626 WriteAsOperandInternal(Out, *OI, TypeTable, Machine);
Vikram S. Adveb952b542002-07-14 23:14:45 +0000627 if (OI+1 != CE->op_end())
Chris Lattner3cd8c562002-07-30 18:54:25 +0000628 Out << ", ";
Vikram S. Adveb952b542002-07-14 23:14:45 +0000629 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000630
Reid Spencer6c38f0b2006-11-27 01:05:10 +0000631 if (CE->isCast()) {
Chris Lattner83b396b2002-08-15 19:37:43 +0000632 Out << " to ";
633 printTypeInt(Out, CE->getType(), TypeTable);
634 }
Reid Spencer6c38f0b2006-11-27 01:05:10 +0000635
Misha Brukman21bbdb92004-06-04 21:11:51 +0000636 Out << ')';
Chris Lattner83b396b2002-08-15 19:37:43 +0000637
Chris Lattnerd84bb632002-04-16 21:36:08 +0000638 } else {
Vikram S. Adveb952b542002-07-14 23:14:45 +0000639 Out << "<placeholder or erroneous Constant>";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000640 }
641}
642
643
Misha Brukmanc566ca362004-03-02 00:22:19 +0000644/// WriteAsOperand - Write the name of the specified value out to the specified
645/// ostream. This can be useful when you just want to print int %reg126, not
646/// the whole instruction that generated it.
647///
Misha Brukmanb1c93172005-04-21 23:48:37 +0000648static void WriteAsOperandInternal(std::ostream &Out, const Value *V,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000649 std::map<const Type*, std::string> &TypeTable,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000650 SlotMachine *Machine) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000651 Out << ' ';
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000652 if (V->hasName())
Reid Spencer788e3172007-01-26 08:02:52 +0000653 Out << getLLVMName(V->getName(),
654 isa<GlobalValue>(V) ? GlobalPrefix : LocalPrefix);
Reid Spenceraccd7c72004-07-17 23:47:01 +0000655 else {
656 const Constant *CV = dyn_cast<Constant>(V);
Chris Lattnera2d810d2006-01-25 22:26:05 +0000657 if (CV && !isa<GlobalValue>(CV)) {
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000658 WriteConstantInt(Out, CV, TypeTable, Machine);
Chris Lattnera2d810d2006-01-25 22:26:05 +0000659 } else if (const InlineAsm *IA = dyn_cast<InlineAsm>(V)) {
660 Out << "asm ";
661 if (IA->hasSideEffects())
662 Out << "sideeffect ";
663 Out << '"';
664 PrintEscapedString(IA->getAsmString(), Out);
665 Out << "\", \"";
666 PrintEscapedString(IA->getConstraintString(), Out);
667 Out << '"';
668 } else {
Reid Spencer788e3172007-01-26 08:02:52 +0000669 char Prefix = '%';
Chris Lattnerd84bb632002-04-16 21:36:08 +0000670 int Slot;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000671 if (Machine) {
Reid Spencer788e3172007-01-26 08:02:52 +0000672 if (const GlobalValue *GV = dyn_cast<GlobalValue>(V)) {
Chris Lattner5e043322007-01-11 03:54:27 +0000673 Slot = Machine->getGlobalSlot(GV);
Reid Spencer788e3172007-01-26 08:02:52 +0000674 Prefix = '@';
675 } else {
Chris Lattner5e043322007-01-11 03:54:27 +0000676 Slot = Machine->getLocalSlot(V);
Reid Spencer788e3172007-01-26 08:02:52 +0000677 }
Chris Lattnerd84bb632002-04-16 21:36:08 +0000678 } else {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000679 Machine = createSlotMachine(V);
Chris Lattner5e043322007-01-11 03:54:27 +0000680 if (Machine) {
Reid Spencer788e3172007-01-26 08:02:52 +0000681 if (const GlobalValue *GV = dyn_cast<GlobalValue>(V)) {
Chris Lattner5e043322007-01-11 03:54:27 +0000682 Slot = Machine->getGlobalSlot(GV);
Reid Spencer788e3172007-01-26 08:02:52 +0000683 Prefix = '@';
684 } else {
Chris Lattner5e043322007-01-11 03:54:27 +0000685 Slot = Machine->getLocalSlot(V);
Reid Spencer788e3172007-01-26 08:02:52 +0000686 }
Chris Lattner5e043322007-01-11 03:54:27 +0000687 } else {
Chris Lattner757ee0b2004-06-09 19:41:19 +0000688 Slot = -1;
Chris Lattner5e043322007-01-11 03:54:27 +0000689 }
Reid Spencer56010e42004-05-26 21:56:09 +0000690 delete Machine;
Chris Lattnerd84bb632002-04-16 21:36:08 +0000691 }
Chris Lattner757ee0b2004-06-09 19:41:19 +0000692 if (Slot != -1)
Reid Spencer788e3172007-01-26 08:02:52 +0000693 Out << Prefix << Slot;
Chris Lattner757ee0b2004-06-09 19:41:19 +0000694 else
695 Out << "<badref>";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000696 }
697 }
698}
699
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000700/// WriteAsOperand - Write the name of the specified value out to the specified
701/// ostream. This can be useful when you just want to print int %reg126, not
702/// the whole instruction that generated it.
703///
Chris Lattner189d19f2003-11-21 20:23:48 +0000704std::ostream &llvm::WriteAsOperand(std::ostream &Out, const Value *V,
Chris Lattner0dda8612006-12-06 06:15:43 +0000705 bool PrintType, const Module *Context) {
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000706 std::map<const Type *, std::string> TypeNames;
Chris Lattner5a9f63e2002-07-10 16:48:17 +0000707 if (Context == 0) Context = getModuleFromVal(V);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000708
Chris Lattner98cf1f52002-11-20 18:36:02 +0000709 if (Context)
Chris Lattner5a9f63e2002-07-10 16:48:17 +0000710 fillTypeNameTable(Context, TypeNames);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000711
712 if (PrintType)
713 printTypeInt(Out, V->getType(), TypeNames);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000714
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000715 WriteAsOperandInternal(Out, V, TypeNames, 0);
Chris Lattner5e5abe32001-07-20 19:15:21 +0000716 return Out;
717}
718
Reid Spencer58d30f22004-07-04 11:50:43 +0000719
Chris Lattner189d19f2003-11-21 20:23:48 +0000720namespace llvm {
Chris Lattner2e9fee42001-07-12 23:35:26 +0000721
Chris Lattnerfee714f2001-09-07 16:36:04 +0000722class AssemblyWriter {
Misha Brukmana6619a92004-06-21 21:53:56 +0000723 std::ostream &Out;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000724 SlotMachine &Machine;
Chris Lattner7bfee412001-10-29 16:05:51 +0000725 const Module *TheModule;
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000726 std::map<const Type *, std::string> TypeNames;
Chris Lattner8339f7d2003-10-30 23:41:03 +0000727 AssemblyAnnotationWriter *AnnotationWriter;
Chris Lattner2f7c9632001-06-06 20:29:01 +0000728public:
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000729 inline AssemblyWriter(std::ostream &o, SlotMachine &Mac, const Module *M,
Chris Lattner8339f7d2003-10-30 23:41:03 +0000730 AssemblyAnnotationWriter *AAW)
Misha Brukmana6619a92004-06-21 21:53:56 +0000731 : Out(o), Machine(Mac), TheModule(M), AnnotationWriter(AAW) {
Chris Lattner7bfee412001-10-29 16:05:51 +0000732
733 // If the module has a symbol table, take all global types and stuff their
734 // names into the TypeNames map.
735 //
Chris Lattnerb86620e2001-10-29 16:37:48 +0000736 fillTypeNameTable(M, TypeNames);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000737 }
738
Anton Korobeynikova97b6942007-04-25 14:27:10 +0000739 inline void write(const Module *M) { printModule(M); }
740 inline void write(const GlobalVariable *G) { printGlobal(G); }
741 inline void write(const GlobalAlias *G) { printAlias(G); }
742 inline void write(const Function *F) { printFunction(F); }
743 inline void write(const BasicBlock *BB) { printBasicBlock(BB); }
Chris Lattner113f4f42002-06-25 16:13:24 +0000744 inline void write(const Instruction *I) { printInstruction(*I); }
Anton Korobeynikova97b6942007-04-25 14:27:10 +0000745 inline void write(const Type *Ty) { printType(Ty); }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000746
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000747 void writeOperand(const Value *Op, bool PrintType);
Chris Lattner1e194682002-04-18 18:53:13 +0000748
Misha Brukman4685e262004-04-28 15:31:21 +0000749 const Module* getModule() { return TheModule; }
750
Misha Brukmand92f54a2004-11-15 19:30:05 +0000751private:
Chris Lattner7bfee412001-10-29 16:05:51 +0000752 void printModule(const Module *M);
Reid Spencer32af9e82007-01-06 07:24:44 +0000753 void printTypeSymbolTable(const TypeSymbolTable &ST);
Chris Lattner7bfee412001-10-29 16:05:51 +0000754 void printGlobal(const GlobalVariable *GV);
Anton Korobeynikova97b6942007-04-25 14:27:10 +0000755 void printAlias(const GlobalAlias *GV);
Chris Lattner57698e22002-03-26 18:01:55 +0000756 void printFunction(const Function *F);
Reid Spencer1517de32007-04-09 06:10:42 +0000757 void printArgument(const Argument *FA, uint16_t ParamAttrs);
Chris Lattner7bfee412001-10-29 16:05:51 +0000758 void printBasicBlock(const BasicBlock *BB);
Chris Lattner113f4f42002-06-25 16:13:24 +0000759 void printInstruction(const Instruction &I);
Chris Lattnerd816b532002-04-13 20:53:41 +0000760
761 // printType - Go to extreme measures to attempt to print out a short,
762 // symbolic version of a type name.
763 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000764 std::ostream &printType(const Type *Ty) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000765 return printTypeInt(Out, Ty, TypeNames);
Chris Lattnerd816b532002-04-13 20:53:41 +0000766 }
767
768 // printTypeAtLeastOneLevel - Print out one level of the possibly complex type
769 // without considering any symbolic types that we may have equal to it.
770 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000771 std::ostream &printTypeAtLeastOneLevel(const Type *Ty);
Chris Lattner7bfee412001-10-29 16:05:51 +0000772
Chris Lattner862e3382001-10-13 06:42:36 +0000773 // printInfoComment - Print a little comment after the instruction indicating
774 // which slot it occupies.
Chris Lattner113f4f42002-06-25 16:13:24 +0000775 void printInfoComment(const Value &V);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000776};
Reid Spencerf43ac622004-05-27 22:04:46 +0000777} // end of llvm namespace
Chris Lattner2f7c9632001-06-06 20:29:01 +0000778
Misha Brukmanc566ca362004-03-02 00:22:19 +0000779/// printTypeAtLeastOneLevel - Print out one level of the possibly complex type
780/// without considering any symbolic types that we may have equal to it.
781///
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000782std::ostream &AssemblyWriter::printTypeAtLeastOneLevel(const Type *Ty) {
Reid Spencer7a9c62b2007-01-12 07:05:14 +0000783 if (const IntegerType *ITy = dyn_cast<IntegerType>(Ty))
784 Out << "i" << utostr(ITy->getBitWidth());
785 else if (const FunctionType *FTy = dyn_cast<FunctionType>(Ty)) {
Reid Spencer8c4914c2006-12-31 05:24:50 +0000786 printType(FTy->getReturnType());
Reid Spencer8c4914c2006-12-31 05:24:50 +0000787 Out << " (";
788 unsigned Idx = 1;
Reid Spencer1517de32007-04-09 06:10:42 +0000789 const ParamAttrsList *Attrs = FTy->getParamAttrs();
Chris Lattnerfa829be2004-02-09 04:14:01 +0000790 for (FunctionType::param_iterator I = FTy->param_begin(),
791 E = FTy->param_end(); I != E; ++I) {
792 if (I != FTy->param_begin())
Misha Brukmana6619a92004-06-21 21:53:56 +0000793 Out << ", ";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000794 printType(*I);
Reid Spencera472f662007-04-11 02:44:20 +0000795 if (Attrs && Attrs->getParamAttrs(Idx) != ParamAttr::None) {
Reid Spencer1517de32007-04-09 06:10:42 +0000796 Out << " " << Attrs->getParamAttrsTextByIndex(Idx);
Reid Spencer8c4914c2006-12-31 05:24:50 +0000797 }
798 Idx++;
Chris Lattnerd816b532002-04-13 20:53:41 +0000799 }
800 if (FTy->isVarArg()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000801 if (FTy->getNumParams()) Out << ", ";
802 Out << "...";
Chris Lattnerd816b532002-04-13 20:53:41 +0000803 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000804 Out << ')';
Reid Spencera472f662007-04-11 02:44:20 +0000805 if (Attrs && Attrs->getParamAttrs(0) != ParamAttr::None)
Reid Spencer1517de32007-04-09 06:10:42 +0000806 Out << ' ' << Attrs->getParamAttrsTextByIndex(0);
Chris Lattner113f4f42002-06-25 16:13:24 +0000807 } else if (const StructType *STy = dyn_cast<StructType>(Ty)) {
Andrew Lenharthdcb3c972006-12-08 18:06:16 +0000808 if (STy->isPacked())
809 Out << '<';
Misha Brukmana6619a92004-06-21 21:53:56 +0000810 Out << "{ ";
Chris Lattnerac6db752004-02-09 04:37:31 +0000811 for (StructType::element_iterator I = STy->element_begin(),
812 E = STy->element_end(); I != E; ++I) {
813 if (I != STy->element_begin())
Misha Brukmana6619a92004-06-21 21:53:56 +0000814 Out << ", ";
Chris Lattnerd816b532002-04-13 20:53:41 +0000815 printType(*I);
816 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000817 Out << " }";
Andrew Lenharthdcb3c972006-12-08 18:06:16 +0000818 if (STy->isPacked())
819 Out << '>';
Chris Lattner113f4f42002-06-25 16:13:24 +0000820 } else if (const PointerType *PTy = dyn_cast<PointerType>(Ty)) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000821 printType(PTy->getElementType()) << '*';
Chris Lattner113f4f42002-06-25 16:13:24 +0000822 } else if (const ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000823 Out << '[' << ATy->getNumElements() << " x ";
Misha Brukman21bbdb92004-06-04 21:11:51 +0000824 printType(ATy->getElementType()) << ']';
Reid Spencerd84d35b2007-02-15 02:26:10 +0000825 } else if (const VectorType *PTy = dyn_cast<VectorType>(Ty)) {
Reid Spencere203d352004-08-20 15:37:30 +0000826 Out << '<' << PTy->getNumElements() << " x ";
827 printType(PTy->getElementType()) << '>';
828 }
Reid Spencerde46e482006-11-02 20:25:50 +0000829 else if (isa<OpaqueType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000830 Out << "opaque";
Chris Lattnerd816b532002-04-13 20:53:41 +0000831 } else {
Vikram S. Adveb952b542002-07-14 23:14:45 +0000832 if (!Ty->isPrimitiveType())
Misha Brukmana6619a92004-06-21 21:53:56 +0000833 Out << "<unknown derived type>";
Chris Lattnerd816b532002-04-13 20:53:41 +0000834 printType(Ty);
835 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000836 return Out;
Chris Lattnerd816b532002-04-13 20:53:41 +0000837}
838
839
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000840void AssemblyWriter::writeOperand(const Value *Operand, bool PrintType) {
841 if (Operand == 0) {
Chris Lattner08f7d0c2005-02-24 16:58:29 +0000842 Out << "<null operand!>";
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000843 } else {
844 if (PrintType) { Out << ' '; printType(Operand->getType()); }
845 WriteAsOperandInternal(Out, Operand, TypeNames, &Machine);
Chris Lattner08f7d0c2005-02-24 16:58:29 +0000846 }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000847}
848
Chris Lattner2f7c9632001-06-06 20:29:01 +0000849
Chris Lattner7bfee412001-10-29 16:05:51 +0000850void AssemblyWriter::printModule(const Module *M) {
Chris Lattner4d8689e2005-03-02 23:12:40 +0000851 if (!M->getModuleIdentifier().empty() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +0000852 // Don't print the ID if it will start a new line (which would
Chris Lattner4d8689e2005-03-02 23:12:40 +0000853 // require a comment char before it).
854 M->getModuleIdentifier().find('\n') == std::string::npos)
855 Out << "; ModuleID = '" << M->getModuleIdentifier() << "'\n";
856
Owen Andersone2237542006-10-18 02:21:12 +0000857 if (!M->getDataLayout().empty())
Chris Lattner04897162006-10-22 06:06:56 +0000858 Out << "target datalayout = \"" << M->getDataLayout() << "\"\n";
Reid Spencer48f98c82004-07-25 21:44:54 +0000859 if (!M->getTargetTriple().empty())
Reid Spencerffec7df2004-07-25 21:29:43 +0000860 Out << "target triple = \"" << M->getTargetTriple() << "\"\n";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000861
Chris Lattnereef2fe72006-01-24 04:13:11 +0000862 if (!M->getModuleInlineAsm().empty()) {
Chris Lattnerefaf35d2006-01-24 00:45:30 +0000863 // Split the string into lines, to make it easier to read the .ll file.
Chris Lattnereef2fe72006-01-24 04:13:11 +0000864 std::string Asm = M->getModuleInlineAsm();
Chris Lattnerefaf35d2006-01-24 00:45:30 +0000865 size_t CurPos = 0;
866 size_t NewLine = Asm.find_first_of('\n', CurPos);
867 while (NewLine != std::string::npos) {
868 // We found a newline, print the portion of the asm string from the
869 // last newline up to this newline.
870 Out << "module asm \"";
871 PrintEscapedString(std::string(Asm.begin()+CurPos, Asm.begin()+NewLine),
872 Out);
873 Out << "\"\n";
874 CurPos = NewLine+1;
875 NewLine = Asm.find_first_of('\n', CurPos);
876 }
Chris Lattner3acaf5c2006-01-24 00:40:17 +0000877 Out << "module asm \"";
Chris Lattnerefaf35d2006-01-24 00:45:30 +0000878 PrintEscapedString(std::string(Asm.begin()+CurPos, Asm.end()), Out);
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000879 Out << "\"\n";
880 }
881
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000882 // Loop over the dependent libraries and emit them.
Chris Lattner730cfe42004-09-14 04:51:44 +0000883 Module::lib_iterator LI = M->lib_begin();
884 Module::lib_iterator LE = M->lib_end();
Reid Spencer48f98c82004-07-25 21:44:54 +0000885 if (LI != LE) {
Chris Lattner730cfe42004-09-14 04:51:44 +0000886 Out << "deplibs = [ ";
887 while (LI != LE) {
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000888 Out << '"' << *LI << '"';
Reid Spencerffec7df2004-07-25 21:29:43 +0000889 ++LI;
Chris Lattner730cfe42004-09-14 04:51:44 +0000890 if (LI != LE)
891 Out << ", ";
Reid Spencerffec7df2004-07-25 21:29:43 +0000892 }
893 Out << " ]\n";
Reid Spencercc5ff642004-07-25 18:08:18 +0000894 }
Reid Spencerb9e08772004-09-13 23:44:23 +0000895
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000896 // Loop over the symbol table, emitting all named constants.
Reid Spencer32af9e82007-01-06 07:24:44 +0000897 printTypeSymbolTable(M->getTypeSymbolTable());
Misha Brukmanb1c93172005-04-21 23:48:37 +0000898
Chris Lattner54932b02006-12-06 04:41:52 +0000899 for (Module::const_global_iterator I = M->global_begin(), E = M->global_end();
900 I != E; ++I)
Chris Lattner113f4f42002-06-25 16:13:24 +0000901 printGlobal(I);
Chris Lattnerd2747052007-04-26 02:24:10 +0000902
903 // Output all aliases.
904 if (!M->alias_empty()) Out << "\n";
905 for (Module::const_alias_iterator I = M->alias_begin(), E = M->alias_end();
906 I != E; ++I)
907 printAlias(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000908
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000909 // Output all of the functions.
Chris Lattner113f4f42002-06-25 16:13:24 +0000910 for (Module::const_iterator I = M->begin(), E = M->end(); I != E; ++I)
911 printFunction(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000912}
913
Chris Lattner7bfee412001-10-29 16:05:51 +0000914void AssemblyWriter::printGlobal(const GlobalVariable *GV) {
Reid Spencer788e3172007-01-26 08:02:52 +0000915 if (GV->hasName()) Out << getLLVMName(GV->getName(), GlobalPrefix) << " = ";
Chris Lattner37798642001-09-18 04:01:05 +0000916
Misha Brukmanb1c93172005-04-21 23:48:37 +0000917 if (!GV->hasInitializer())
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000918 switch (GV->getLinkage()) {
919 case GlobalValue::DLLImportLinkage: Out << "dllimport "; break;
920 case GlobalValue::ExternalWeakLinkage: Out << "extern_weak "; break;
921 default: Out << "external "; break;
Anton Korobeynikova0554d92007-01-12 19:20:47 +0000922 } else {
Chris Lattner379a8d22003-04-16 20:28:45 +0000923 switch (GV->getLinkage()) {
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000924 case GlobalValue::InternalLinkage: Out << "internal "; break;
925 case GlobalValue::LinkOnceLinkage: Out << "linkonce "; break;
926 case GlobalValue::WeakLinkage: Out << "weak "; break;
927 case GlobalValue::AppendingLinkage: Out << "appending "; break;
928 case GlobalValue::DLLImportLinkage: Out << "dllimport "; break;
929 case GlobalValue::DLLExportLinkage: Out << "dllexport "; break;
930 case GlobalValue::ExternalWeakLinkage: Out << "extern_weak "; break;
931 case GlobalValue::ExternalLinkage: break;
Misha Brukman70457632004-11-14 21:04:34 +0000932 case GlobalValue::GhostLinkage:
Bill Wendlingf3baad32006-12-07 01:30:32 +0000933 cerr << "GhostLinkage not allowed in AsmWriter!\n";
Misha Brukman70457632004-11-14 21:04:34 +0000934 abort();
Chris Lattner379a8d22003-04-16 20:28:45 +0000935 }
Anton Korobeynikova0554d92007-01-12 19:20:47 +0000936 switch (GV->getVisibility()) {
Chris Lattner90d2e422007-01-15 18:28:18 +0000937 default: assert(0 && "Invalid visibility style!");
Anton Korobeynikova0554d92007-01-12 19:20:47 +0000938 case GlobalValue::DefaultVisibility: break;
939 case GlobalValue::HiddenVisibility: Out << "hidden "; break;
Anton Korobeynikov39f3cff2007-04-29 18:35:00 +0000940 case GlobalValue::ProtectedVisibility: Out << "protected "; break;
Anton Korobeynikova0554d92007-01-12 19:20:47 +0000941 }
942 }
Lauro Ramos Venancio749e4662007-04-12 18:32:50 +0000943
944 if (GV->isThreadLocal()) Out << "thread_local ";
Misha Brukmana6619a92004-06-21 21:53:56 +0000945 Out << (GV->isConstant() ? "constant " : "global ");
Chris Lattner2413b162001-12-04 00:03:30 +0000946 printType(GV->getType()->getElementType());
Chris Lattner37798642001-09-18 04:01:05 +0000947
Reid Spenceraccd7c72004-07-17 23:47:01 +0000948 if (GV->hasInitializer()) {
949 Constant* C = cast<Constant>(GV->getInitializer());
950 assert(C && "GlobalVar initializer isn't constant?");
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000951 writeOperand(GV->getInitializer(), false);
Reid Spenceraccd7c72004-07-17 23:47:01 +0000952 }
Anton Korobeynikova97b6942007-04-25 14:27:10 +0000953
Chris Lattner4b96c542005-11-12 00:10:19 +0000954 if (GV->hasSection())
955 Out << ", section \"" << GV->getSection() << '"';
956 if (GV->getAlignment())
Chris Lattnerf8a974d2005-11-06 06:48:53 +0000957 Out << ", align " << GV->getAlignment();
Anton Korobeynikova97b6942007-04-25 14:27:10 +0000958
Chris Lattner113f4f42002-06-25 16:13:24 +0000959 printInfoComment(*GV);
Misha Brukmana6619a92004-06-21 21:53:56 +0000960 Out << "\n";
Chris Lattnerda975502001-09-10 07:58:01 +0000961}
962
Anton Korobeynikova97b6942007-04-25 14:27:10 +0000963void AssemblyWriter::printAlias(const GlobalAlias *GA) {
964 Out << getLLVMName(GA->getName(), GlobalPrefix) << " = ";
965 switch (GA->getVisibility()) {
966 default: assert(0 && "Invalid visibility style!");
967 case GlobalValue::DefaultVisibility: break;
968 case GlobalValue::HiddenVisibility: Out << "hidden "; break;
Anton Korobeynikov39f3cff2007-04-29 18:35:00 +0000969 case GlobalValue::ProtectedVisibility: Out << "protected "; break;
Anton Korobeynikova97b6942007-04-25 14:27:10 +0000970 }
971
972 Out << "alias ";
973
974 switch (GA->getLinkage()) {
975 case GlobalValue::WeakLinkage: Out << "weak "; break;
976 case GlobalValue::InternalLinkage: Out << "internal "; break;
977 case GlobalValue::ExternalLinkage: break;
978 default:
979 assert(0 && "Invalid alias linkage");
980 }
981
Anton Korobeynikov546ea7e2007-04-29 18:02:48 +0000982 const Constant *Aliasee = GA->getAliasee();
Anton Korobeynikova97b6942007-04-25 14:27:10 +0000983
984 if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(Aliasee)) {
985 printType(GV->getType());
986 Out << " " << getLLVMName(GV->getName(), GlobalPrefix);
987 } else if (const Function *F = dyn_cast<Function>(Aliasee)) {
988 printType(F->getFunctionType());
989 Out << "* ";
990
991 if (!F->getName().empty())
992 Out << getLLVMName(F->getName(), GlobalPrefix);
993 else
994 Out << "@\"\"";
Anton Korobeynikovb18f8f82007-04-28 13:45:00 +0000995 } else {
996 const ConstantExpr *CE = 0;
997 if ((CE = dyn_cast<ConstantExpr>(Aliasee)) &&
998 (CE->getOpcode() == Instruction::BitCast)) {
999 writeOperand(CE, false);
1000 } else
1001 assert(0 && "Unsupported aliasee");
1002 }
1003
Anton Korobeynikova97b6942007-04-25 14:27:10 +00001004 printInfoComment(*GA);
1005 Out << "\n";
1006}
1007
Reid Spencer32af9e82007-01-06 07:24:44 +00001008void AssemblyWriter::printTypeSymbolTable(const TypeSymbolTable &ST) {
Reid Spencere7e96712004-05-25 08:53:40 +00001009 // Print the types.
Reid Spencer32af9e82007-01-06 07:24:44 +00001010 for (TypeSymbolTable::const_iterator TI = ST.begin(), TE = ST.end();
1011 TI != TE; ++TI) {
Reid Spencer788e3172007-01-26 08:02:52 +00001012 Out << "\t" << getLLVMName(TI->first, LocalPrefix) << " = type ";
Reid Spencere7e96712004-05-25 08:53:40 +00001013
1014 // Make sure we print out at least one level of the type structure, so
1015 // that we do not get %FILE = type %FILE
1016 //
1017 printTypeAtLeastOneLevel(TI->second) << "\n";
1018 }
Reid Spencer32af9e82007-01-06 07:24:44 +00001019}
1020
Misha Brukmanc566ca362004-03-02 00:22:19 +00001021/// printFunction - Print all aspects of a function.
1022///
Chris Lattner113f4f42002-06-25 16:13:24 +00001023void AssemblyWriter::printFunction(const Function *F) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001024 // Print out the return type and name...
Misha Brukmana6619a92004-06-21 21:53:56 +00001025 Out << "\n";
Chris Lattner379a8d22003-04-16 20:28:45 +00001026
Misha Brukmana6619a92004-06-21 21:53:56 +00001027 if (AnnotationWriter) AnnotationWriter->emitFunctionAnnot(F, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001028
Reid Spencer5301e7c2007-01-30 20:08:39 +00001029 if (F->isDeclaration())
Chris Lattner10f03a62007-08-19 22:15:26 +00001030 Out << "declare ";
1031 else
Reid Spencer7ce2d2a2006-12-29 20:29:48 +00001032 Out << "define ";
Chris Lattner10f03a62007-08-19 22:15:26 +00001033
1034 switch (F->getLinkage()) {
1035 case GlobalValue::InternalLinkage: Out << "internal "; break;
1036 case GlobalValue::LinkOnceLinkage: Out << "linkonce "; break;
1037 case GlobalValue::WeakLinkage: Out << "weak "; break;
1038 case GlobalValue::AppendingLinkage: Out << "appending "; break;
1039 case GlobalValue::DLLImportLinkage: Out << "dllimport "; break;
1040 case GlobalValue::DLLExportLinkage: Out << "dllexport "; break;
1041 case GlobalValue::ExternalWeakLinkage: Out << "extern_weak "; break;
1042 case GlobalValue::ExternalLinkage: break;
1043 case GlobalValue::GhostLinkage:
1044 cerr << "GhostLinkage not allowed in AsmWriter!\n";
1045 abort();
1046 }
1047 switch (F->getVisibility()) {
1048 default: assert(0 && "Invalid visibility style!");
1049 case GlobalValue::DefaultVisibility: break;
1050 case GlobalValue::HiddenVisibility: Out << "hidden "; break;
1051 case GlobalValue::ProtectedVisibility: Out << "protected "; break;
Reid Spencer7ce2d2a2006-12-29 20:29:48 +00001052 }
Chris Lattner379a8d22003-04-16 20:28:45 +00001053
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001054 // Print the calling convention.
1055 switch (F->getCallingConv()) {
1056 case CallingConv::C: break; // default
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +00001057 case CallingConv::Fast: Out << "fastcc "; break;
1058 case CallingConv::Cold: Out << "coldcc "; break;
1059 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
1060 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001061 default: Out << "cc" << F->getCallingConv() << " "; break;
1062 }
1063
Reid Spencer8c4914c2006-12-31 05:24:50 +00001064 const FunctionType *FT = F->getFunctionType();
Reid Spencer1517de32007-04-09 06:10:42 +00001065 const ParamAttrsList *Attrs = FT->getParamAttrs();
Misha Brukman21bbdb92004-06-04 21:11:51 +00001066 printType(F->getReturnType()) << ' ';
Chris Lattner5b337482003-10-18 05:57:43 +00001067 if (!F->getName().empty())
Reid Spencer788e3172007-01-26 08:02:52 +00001068 Out << getLLVMName(F->getName(), GlobalPrefix);
Chris Lattner5b337482003-10-18 05:57:43 +00001069 else
Reid Spencer788e3172007-01-26 08:02:52 +00001070 Out << "@\"\"";
Misha Brukmana6619a92004-06-21 21:53:56 +00001071 Out << '(';
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001072 Machine.incorporateFunction(F);
Chris Lattnerfee714f2001-09-07 16:36:04 +00001073
Chris Lattner7bfee412001-10-29 16:05:51 +00001074 // Loop over the arguments, printing them...
Chris Lattnerfee714f2001-09-07 16:36:04 +00001075
Reid Spencer8c4914c2006-12-31 05:24:50 +00001076 unsigned Idx = 1;
Chris Lattner82738fe2007-04-18 00:57:22 +00001077 if (!F->isDeclaration()) {
1078 // If this isn't a declaration, print the argument names as well.
1079 for (Function::const_arg_iterator I = F->arg_begin(), E = F->arg_end();
1080 I != E; ++I) {
1081 // Insert commas as we go... the first arg doesn't get a comma
1082 if (I != F->arg_begin()) Out << ", ";
1083 printArgument(I, (Attrs ? Attrs->getParamAttrs(Idx)
1084 : uint16_t(ParamAttr::None)));
1085 Idx++;
1086 }
1087 } else {
1088 // Otherwise, print the types from the function type.
1089 for (unsigned i = 0, e = FT->getNumParams(); i != e; ++i) {
1090 // Insert commas as we go... the first arg doesn't get a comma
1091 if (i) Out << ", ";
1092
1093 // Output type...
1094 printType(FT->getParamType(i));
1095
1096 unsigned ArgAttrs = ParamAttr::None;
1097 if (Attrs) ArgAttrs = Attrs->getParamAttrs(i+1);
1098 if (ArgAttrs != ParamAttr::None)
1099 Out << ' ' << ParamAttrsList::getParamAttrsText(ArgAttrs);
1100 }
Reid Spencer8c4914c2006-12-31 05:24:50 +00001101 }
Chris Lattnerfee714f2001-09-07 16:36:04 +00001102
1103 // Finish printing arguments...
Chris Lattner113f4f42002-06-25 16:13:24 +00001104 if (FT->isVarArg()) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001105 if (FT->getNumParams()) Out << ", ";
1106 Out << "..."; // Output varargs portion of signature!
Chris Lattnerfee714f2001-09-07 16:36:04 +00001107 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001108 Out << ')';
Reid Spencera472f662007-04-11 02:44:20 +00001109 if (Attrs && Attrs->getParamAttrs(0) != ParamAttr::None)
Reid Spencer1517de32007-04-09 06:10:42 +00001110 Out << ' ' << Attrs->getParamAttrsTextByIndex(0);
Chris Lattner4b96c542005-11-12 00:10:19 +00001111 if (F->hasSection())
1112 Out << " section \"" << F->getSection() << '"';
Chris Lattnerf8a974d2005-11-06 06:48:53 +00001113 if (F->getAlignment())
1114 Out << " align " << F->getAlignment();
Chris Lattner4b96c542005-11-12 00:10:19 +00001115
Reid Spencer5301e7c2007-01-30 20:08:39 +00001116 if (F->isDeclaration()) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001117 Out << "\n";
Chris Lattnerb2f02e52002-05-06 03:00:40 +00001118 } else {
Misha Brukmana6619a92004-06-21 21:53:56 +00001119 Out << " {";
Misha Brukmanb1c93172005-04-21 23:48:37 +00001120
Chris Lattner6915f8f2002-04-07 22:49:37 +00001121 // Output all of its basic blocks... for the function
Chris Lattner113f4f42002-06-25 16:13:24 +00001122 for (Function::const_iterator I = F->begin(), E = F->end(); I != E; ++I)
1123 printBasicBlock(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +00001124
Misha Brukmana6619a92004-06-21 21:53:56 +00001125 Out << "}\n";
Chris Lattnerfee714f2001-09-07 16:36:04 +00001126 }
1127
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001128 Machine.purgeFunction();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001129}
1130
Misha Brukmanc566ca362004-03-02 00:22:19 +00001131/// printArgument - This member is called for every argument that is passed into
1132/// the function. Simply print it out
1133///
Reid Spencer1517de32007-04-09 06:10:42 +00001134void AssemblyWriter::printArgument(const Argument *Arg, uint16_t Attrs) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001135 // Output type...
Chris Lattner7bfee412001-10-29 16:05:51 +00001136 printType(Arg->getType());
Misha Brukmanb1c93172005-04-21 23:48:37 +00001137
Reid Spencera472f662007-04-11 02:44:20 +00001138 if (Attrs != ParamAttr::None)
Reid Spencer1517de32007-04-09 06:10:42 +00001139 Out << ' ' << ParamAttrsList::getParamAttrsText(Attrs);
Reid Spencer8c4914c2006-12-31 05:24:50 +00001140
Chris Lattner2f7c9632001-06-06 20:29:01 +00001141 // Output name, if available...
1142 if (Arg->hasName())
Reid Spencer788e3172007-01-26 08:02:52 +00001143 Out << ' ' << getLLVMName(Arg->getName(), LocalPrefix);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001144}
1145
Misha Brukmanc566ca362004-03-02 00:22:19 +00001146/// printBasicBlock - This member is called for each basic block in a method.
1147///
Chris Lattner7bfee412001-10-29 16:05:51 +00001148void AssemblyWriter::printBasicBlock(const BasicBlock *BB) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001149 if (BB->hasName()) { // Print out the label if it exists...
Reid Spencer788e3172007-01-26 08:02:52 +00001150 Out << "\n" << getLLVMName(BB->getName(), LabelPrefix) << ':';
Chris Lattner408dbdb2002-05-14 16:02:05 +00001151 } else if (!BB->use_empty()) { // Don't print block # of no uses...
Misha Brukmana6619a92004-06-21 21:53:56 +00001152 Out << "\n; <label>:";
Chris Lattner5e043322007-01-11 03:54:27 +00001153 int Slot = Machine.getLocalSlot(BB);
Chris Lattner757ee0b2004-06-09 19:41:19 +00001154 if (Slot != -1)
Misha Brukmana6619a92004-06-21 21:53:56 +00001155 Out << Slot;
Chris Lattner757ee0b2004-06-09 19:41:19 +00001156 else
Misha Brukmana6619a92004-06-21 21:53:56 +00001157 Out << "<badref>";
Chris Lattner58185f22002-10-02 19:38:55 +00001158 }
Chris Lattner2447ef52003-11-20 00:09:43 +00001159
1160 if (BB->getParent() == 0)
Misha Brukmana6619a92004-06-21 21:53:56 +00001161 Out << "\t\t; Error: Block without parent!";
Chris Lattner2447ef52003-11-20 00:09:43 +00001162 else {
Dan Gohmandcb291f2007-03-22 16:38:57 +00001163 if (BB != &BB->getParent()->getEntryBlock()) { // Not the entry block?
Chris Lattner2447ef52003-11-20 00:09:43 +00001164 // Output predecessors for the block...
Misha Brukmana6619a92004-06-21 21:53:56 +00001165 Out << "\t\t;";
Chris Lattner2447ef52003-11-20 00:09:43 +00001166 pred_const_iterator PI = pred_begin(BB), PE = pred_end(BB);
Misha Brukmanb1c93172005-04-21 23:48:37 +00001167
Chris Lattner2447ef52003-11-20 00:09:43 +00001168 if (PI == PE) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001169 Out << " No predecessors!";
Chris Lattner2447ef52003-11-20 00:09:43 +00001170 } else {
Misha Brukmana6619a92004-06-21 21:53:56 +00001171 Out << " preds =";
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001172 writeOperand(*PI, false);
Chris Lattner2447ef52003-11-20 00:09:43 +00001173 for (++PI; PI != PE; ++PI) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001174 Out << ',';
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001175 writeOperand(*PI, false);
Chris Lattner2447ef52003-11-20 00:09:43 +00001176 }
Chris Lattner00211f12003-11-16 22:59:57 +00001177 }
Chris Lattner58185f22002-10-02 19:38:55 +00001178 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001179 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001180
Misha Brukmana6619a92004-06-21 21:53:56 +00001181 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001182
Misha Brukmana6619a92004-06-21 21:53:56 +00001183 if (AnnotationWriter) AnnotationWriter->emitBasicBlockStartAnnot(BB, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001184
Chris Lattnerfee714f2001-09-07 16:36:04 +00001185 // Output all of the instructions in the basic block...
Chris Lattner113f4f42002-06-25 16:13:24 +00001186 for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I != E; ++I)
1187 printInstruction(*I);
Chris Lattner96cdd272004-03-08 18:51:45 +00001188
Misha Brukmana6619a92004-06-21 21:53:56 +00001189 if (AnnotationWriter) AnnotationWriter->emitBasicBlockEndAnnot(BB, Out);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001190}
1191
Chris Lattner862e3382001-10-13 06:42:36 +00001192
Misha Brukmanc566ca362004-03-02 00:22:19 +00001193/// printInfoComment - Print a little comment after the instruction indicating
1194/// which slot it occupies.
1195///
Chris Lattner113f4f42002-06-25 16:13:24 +00001196void AssemblyWriter::printInfoComment(const Value &V) {
1197 if (V.getType() != Type::VoidTy) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001198 Out << "\t\t; <";
Misha Brukman21bbdb92004-06-04 21:11:51 +00001199 printType(V.getType()) << '>';
Chris Lattner862e3382001-10-13 06:42:36 +00001200
Chris Lattner113f4f42002-06-25 16:13:24 +00001201 if (!V.hasName()) {
Chris Lattner5e043322007-01-11 03:54:27 +00001202 int SlotNum;
1203 if (const GlobalValue *GV = dyn_cast<GlobalValue>(&V))
1204 SlotNum = Machine.getGlobalSlot(GV);
1205 else
1206 SlotNum = Machine.getLocalSlot(&V);
Chris Lattner757ee0b2004-06-09 19:41:19 +00001207 if (SlotNum == -1)
Misha Brukmana6619a92004-06-21 21:53:56 +00001208 Out << ":<badref>";
Reid Spencer8beac692004-06-09 15:26:53 +00001209 else
Misha Brukmana6619a92004-06-21 21:53:56 +00001210 Out << ':' << SlotNum; // Print out the def slot taken.
Chris Lattner862e3382001-10-13 06:42:36 +00001211 }
Chris Lattnerb6c21db2005-02-01 01:24:01 +00001212 Out << " [#uses=" << V.getNumUses() << ']'; // Output # uses
Chris Lattner862e3382001-10-13 06:42:36 +00001213 }
1214}
1215
Reid Spencere7141c82006-08-28 01:02:49 +00001216// This member is called for each Instruction in a function..
Chris Lattner113f4f42002-06-25 16:13:24 +00001217void AssemblyWriter::printInstruction(const Instruction &I) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001218 if (AnnotationWriter) AnnotationWriter->emitInstructionAnnot(&I, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001219
Misha Brukmana6619a92004-06-21 21:53:56 +00001220 Out << "\t";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001221
1222 // Print out name if it exists...
Chris Lattner113f4f42002-06-25 16:13:24 +00001223 if (I.hasName())
Reid Spencer788e3172007-01-26 08:02:52 +00001224 Out << getLLVMName(I.getName(), LocalPrefix) << " = ";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001225
Chris Lattner06038452005-05-06 05:51:46 +00001226 // If this is a volatile load or store, print out the volatile marker.
Chris Lattner504f9242003-09-08 17:45:59 +00001227 if ((isa<LoadInst>(I) && cast<LoadInst>(I).isVolatile()) ||
Chris Lattner06038452005-05-06 05:51:46 +00001228 (isa<StoreInst>(I) && cast<StoreInst>(I).isVolatile())) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001229 Out << "volatile ";
Chris Lattner06038452005-05-06 05:51:46 +00001230 } else if (isa<CallInst>(I) && cast<CallInst>(I).isTailCall()) {
1231 // If this is a call, check if it's a tail call.
1232 Out << "tail ";
1233 }
Chris Lattner504f9242003-09-08 17:45:59 +00001234
Chris Lattner2f7c9632001-06-06 20:29:01 +00001235 // Print out the opcode...
Misha Brukmana6619a92004-06-21 21:53:56 +00001236 Out << I.getOpcodeName();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001237
Reid Spencer45e52392006-12-03 06:27:29 +00001238 // Print out the compare instruction predicates
1239 if (const FCmpInst *FCI = dyn_cast<FCmpInst>(&I)) {
Reid Spencer812a1be2006-12-04 05:19:18 +00001240 Out << " " << getPredicateText(FCI->getPredicate());
Reid Spencer45e52392006-12-03 06:27:29 +00001241 } else if (const ICmpInst *ICI = dyn_cast<ICmpInst>(&I)) {
Reid Spencer812a1be2006-12-04 05:19:18 +00001242 Out << " " << getPredicateText(ICI->getPredicate());
Reid Spencer45e52392006-12-03 06:27:29 +00001243 }
1244
Chris Lattner2f7c9632001-06-06 20:29:01 +00001245 // Print out the type of the operands...
Chris Lattner113f4f42002-06-25 16:13:24 +00001246 const Value *Operand = I.getNumOperands() ? I.getOperand(0) : 0;
Chris Lattner2f7c9632001-06-06 20:29:01 +00001247
1248 // Special case conditional branches to swizzle the condition out to the front
Chris Lattner113f4f42002-06-25 16:13:24 +00001249 if (isa<BranchInst>(I) && I.getNumOperands() > 1) {
1250 writeOperand(I.getOperand(2), true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001251 Out << ',';
Chris Lattner2f7c9632001-06-06 20:29:01 +00001252 writeOperand(Operand, true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001253 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001254 writeOperand(I.getOperand(1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001255
Chris Lattner8d48df22002-04-13 18:34:38 +00001256 } else if (isa<SwitchInst>(I)) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001257 // Special case switch statement to get formatting nice and correct...
Misha Brukmana6619a92004-06-21 21:53:56 +00001258 writeOperand(Operand , true); Out << ',';
1259 writeOperand(I.getOperand(1), true); Out << " [";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001260
Chris Lattner113f4f42002-06-25 16:13:24 +00001261 for (unsigned op = 2, Eop = I.getNumOperands(); op < Eop; op += 2) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001262 Out << "\n\t\t";
1263 writeOperand(I.getOperand(op ), true); Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001264 writeOperand(I.getOperand(op+1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001265 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001266 Out << "\n\t]";
Chris Lattnerda558102001-10-02 03:41:24 +00001267 } else if (isa<PHINode>(I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001268 Out << ' ';
Chris Lattner113f4f42002-06-25 16:13:24 +00001269 printType(I.getType());
Misha Brukmana6619a92004-06-21 21:53:56 +00001270 Out << ' ';
Chris Lattner2f7c9632001-06-06 20:29:01 +00001271
Chris Lattner113f4f42002-06-25 16:13:24 +00001272 for (unsigned op = 0, Eop = I.getNumOperands(); op < Eop; op += 2) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001273 if (op) Out << ", ";
Misha Brukmanb1c93172005-04-21 23:48:37 +00001274 Out << '[';
Misha Brukmana6619a92004-06-21 21:53:56 +00001275 writeOperand(I.getOperand(op ), false); Out << ',';
1276 writeOperand(I.getOperand(op+1), false); Out << " ]";
Chris Lattner931ef3b2001-06-11 15:04:20 +00001277 }
Chris Lattner862e3382001-10-13 06:42:36 +00001278 } else if (isa<ReturnInst>(I) && !Operand) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001279 Out << " void";
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001280 } else if (const CallInst *CI = dyn_cast<CallInst>(&I)) {
1281 // Print the calling convention being used.
1282 switch (CI->getCallingConv()) {
1283 case CallingConv::C: break; // default
Chris Lattner29d20852006-05-19 21:58:52 +00001284 case CallingConv::Fast: Out << " fastcc"; break;
1285 case CallingConv::Cold: Out << " coldcc"; break;
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +00001286 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
1287 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001288 default: Out << " cc" << CI->getCallingConv(); break;
1289 }
1290
Reid Spencer1517de32007-04-09 06:10:42 +00001291 const PointerType *PTy = cast<PointerType>(Operand->getType());
1292 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1293 const Type *RetTy = FTy->getReturnType();
1294 const ParamAttrsList *PAL = FTy->getParamAttrs();
Chris Lattner2f2d9472001-11-06 21:28:12 +00001295
Chris Lattner463d6a52003-08-05 15:34:45 +00001296 // If possible, print out the short form of the call instruction. We can
Chris Lattner6915f8f2002-04-07 22:49:37 +00001297 // only do this if the first argument is a pointer to a nonvararg function,
Chris Lattner463d6a52003-08-05 15:34:45 +00001298 // and if the return type is not a pointer to a function.
Chris Lattner2f2d9472001-11-06 21:28:12 +00001299 //
Chris Lattner463d6a52003-08-05 15:34:45 +00001300 if (!FTy->isVarArg() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +00001301 (!isa<PointerType>(RetTy) ||
Chris Lattnerd9a36a62002-07-25 20:58:51 +00001302 !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001303 Out << ' '; printType(RetTy);
Chris Lattner2f2d9472001-11-06 21:28:12 +00001304 writeOperand(Operand, false);
1305 } else {
1306 writeOperand(Operand, true);
1307 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001308 Out << '(';
Reid Spencer8c4914c2006-12-31 05:24:50 +00001309 for (unsigned op = 1, Eop = I.getNumOperands(); op < Eop; ++op) {
1310 if (op > 1)
1311 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001312 writeOperand(I.getOperand(op), true);
Reid Spencera472f662007-04-11 02:44:20 +00001313 if (PAL && PAL->getParamAttrs(op) != ParamAttr::None)
Reid Spencer1517de32007-04-09 06:10:42 +00001314 Out << " " << PAL->getParamAttrsTextByIndex(op);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001315 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001316 Out << " )";
Reid Spencera472f662007-04-11 02:44:20 +00001317 if (PAL && PAL->getParamAttrs(0) != ParamAttr::None)
Reid Spencer1517de32007-04-09 06:10:42 +00001318 Out << ' ' << PAL->getParamAttrsTextByIndex(0);
Chris Lattner113f4f42002-06-25 16:13:24 +00001319 } else if (const InvokeInst *II = dyn_cast<InvokeInst>(&I)) {
Reid Spencer1517de32007-04-09 06:10:42 +00001320 const PointerType *PTy = cast<PointerType>(Operand->getType());
1321 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1322 const Type *RetTy = FTy->getReturnType();
1323 const ParamAttrsList *PAL = FTy->getParamAttrs();
Chris Lattner463d6a52003-08-05 15:34:45 +00001324
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001325 // Print the calling convention being used.
1326 switch (II->getCallingConv()) {
1327 case CallingConv::C: break; // default
Chris Lattner29d20852006-05-19 21:58:52 +00001328 case CallingConv::Fast: Out << " fastcc"; break;
1329 case CallingConv::Cold: Out << " coldcc"; break;
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +00001330 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
1331 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001332 default: Out << " cc" << II->getCallingConv(); break;
1333 }
1334
Chris Lattner463d6a52003-08-05 15:34:45 +00001335 // If possible, print out the short form of the invoke instruction. We can
1336 // only do this if the first argument is a pointer to a nonvararg function,
1337 // and if the return type is not a pointer to a function.
1338 //
1339 if (!FTy->isVarArg() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +00001340 (!isa<PointerType>(RetTy) ||
Chris Lattner463d6a52003-08-05 15:34:45 +00001341 !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001342 Out << ' '; printType(RetTy);
Chris Lattner463d6a52003-08-05 15:34:45 +00001343 writeOperand(Operand, false);
1344 } else {
1345 writeOperand(Operand, true);
1346 }
1347
Misha Brukmana6619a92004-06-21 21:53:56 +00001348 Out << '(';
Reid Spencer8c4914c2006-12-31 05:24:50 +00001349 for (unsigned op = 3, Eop = I.getNumOperands(); op < Eop; ++op) {
1350 if (op > 3)
1351 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001352 writeOperand(I.getOperand(op), true);
Reid Spencera472f662007-04-11 02:44:20 +00001353 if (PAL && PAL->getParamAttrs(op-2) != ParamAttr::None)
Reid Spencer1517de32007-04-09 06:10:42 +00001354 Out << " " << PAL->getParamAttrsTextByIndex(op-2);
Chris Lattner862e3382001-10-13 06:42:36 +00001355 }
1356
Reid Spencer136a91c2007-01-05 17:06:19 +00001357 Out << " )";
Reid Spencera472f662007-04-11 02:44:20 +00001358 if (PAL && PAL->getParamAttrs(0) != ParamAttr::None)
Reid Spencer1517de32007-04-09 06:10:42 +00001359 Out << " " << PAL->getParamAttrsTextByIndex(0);
Reid Spencer136a91c2007-01-05 17:06:19 +00001360 Out << "\n\t\t\tto";
Chris Lattner862e3382001-10-13 06:42:36 +00001361 writeOperand(II->getNormalDest(), true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001362 Out << " unwind";
Chris Lattnerfae8ab32004-02-08 21:44:31 +00001363 writeOperand(II->getUnwindDest(), true);
Chris Lattner862e3382001-10-13 06:42:36 +00001364
Chris Lattner113f4f42002-06-25 16:13:24 +00001365 } else if (const AllocationInst *AI = dyn_cast<AllocationInst>(&I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001366 Out << ' ';
Chris Lattner8d48df22002-04-13 18:34:38 +00001367 printType(AI->getType()->getElementType());
1368 if (AI->isArrayAllocation()) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001369 Out << ',';
Chris Lattner8d48df22002-04-13 18:34:38 +00001370 writeOperand(AI->getArraySize(), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001371 }
Nate Begeman848622f2005-11-05 09:21:28 +00001372 if (AI->getAlignment()) {
Chris Lattner7aeee3a2005-11-05 21:20:34 +00001373 Out << ", align " << AI->getAlignment();
Nate Begeman848622f2005-11-05 09:21:28 +00001374 }
Chris Lattner862e3382001-10-13 06:42:36 +00001375 } else if (isa<CastInst>(I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001376 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001377 Out << " to ";
Chris Lattner113f4f42002-06-25 16:13:24 +00001378 printType(I.getType());
Chris Lattner5b337482003-10-18 05:57:43 +00001379 } else if (isa<VAArgInst>(I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001380 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001381 Out << ", ";
Chris Lattnerf70da102003-05-08 02:44:12 +00001382 printType(I.getType());
Chris Lattner2f7c9632001-06-06 20:29:01 +00001383 } else if (Operand) { // Print the normal way...
1384
Misha Brukmanb1c93172005-04-21 23:48:37 +00001385 // PrintAllTypes - Instructions who have operands of all the same type
Chris Lattner2f7c9632001-06-06 20:29:01 +00001386 // omit the type from all but the first operand. If the instruction has
1387 // different type operands (for example br), then they are all printed.
1388 bool PrintAllTypes = false;
1389 const Type *TheType = Operand->getType();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001390
Reid Spencer0cdd04f2007-02-02 13:54:55 +00001391 // Select, Store and ShuffleVector always print all types.
Reid Spencer2341c222007-02-02 02:16:23 +00001392 if (isa<SelectInst>(I) || isa<StoreInst>(I) || isa<ShuffleVectorInst>(I)) {
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001393 PrintAllTypes = true;
1394 } else {
1395 for (unsigned i = 1, E = I.getNumOperands(); i != E; ++i) {
1396 Operand = I.getOperand(i);
1397 if (Operand->getType() != TheType) {
1398 PrintAllTypes = true; // We have differing types! Print them all!
1399 break;
1400 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001401 }
1402 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001403
Chris Lattner7bfee412001-10-29 16:05:51 +00001404 if (!PrintAllTypes) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001405 Out << ' ';
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001406 printType(TheType);
Chris Lattner7bfee412001-10-29 16:05:51 +00001407 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001408
Chris Lattner113f4f42002-06-25 16:13:24 +00001409 for (unsigned i = 0, E = I.getNumOperands(); i != E; ++i) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001410 if (i) Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001411 writeOperand(I.getOperand(i), PrintAllTypes);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001412 }
1413 }
Christopher Lamb84485702007-04-22 19:24:39 +00001414
1415 // Print post operand alignment for load/store
1416 if (isa<LoadInst>(I) && cast<LoadInst>(I).getAlignment()) {
1417 Out << ", align " << cast<LoadInst>(I).getAlignment();
1418 } else if (isa<StoreInst>(I) && cast<StoreInst>(I).getAlignment()) {
1419 Out << ", align " << cast<StoreInst>(I).getAlignment();
1420 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001421
Chris Lattner862e3382001-10-13 06:42:36 +00001422 printInfoComment(I);
Misha Brukmana6619a92004-06-21 21:53:56 +00001423 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001424}
1425
1426
1427//===----------------------------------------------------------------------===//
1428// External Interface declarations
1429//===----------------------------------------------------------------------===//
1430
Chris Lattner8339f7d2003-10-30 23:41:03 +00001431void Module::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001432 SlotMachine SlotTable(this);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001433 AssemblyWriter W(o, SlotTable, this, AAW);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001434 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001435}
1436
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001437void GlobalVariable::print(std::ostream &o) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001438 SlotMachine SlotTable(getParent());
Chris Lattner8339f7d2003-10-30 23:41:03 +00001439 AssemblyWriter W(o, SlotTable, getParent(), 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001440 W.write(this);
Chris Lattneradfe0d12001-09-10 20:08:19 +00001441}
1442
Anton Korobeynikova97b6942007-04-25 14:27:10 +00001443void GlobalAlias::print(std::ostream &o) const {
1444 SlotMachine SlotTable(getParent());
1445 AssemblyWriter W(o, SlotTable, getParent(), 0);
1446 W.write(this);
1447}
1448
Chris Lattner8339f7d2003-10-30 23:41:03 +00001449void Function::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001450 SlotMachine SlotTable(getParent());
Chris Lattner8339f7d2003-10-30 23:41:03 +00001451 AssemblyWriter W(o, SlotTable, getParent(), AAW);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001452
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001453 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001454}
1455
Chris Lattnereef2fe72006-01-24 04:13:11 +00001456void InlineAsm::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001457 WriteAsOperand(o, this, true, 0);
Chris Lattnereef2fe72006-01-24 04:13:11 +00001458}
1459
Chris Lattner8339f7d2003-10-30 23:41:03 +00001460void BasicBlock::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001461 SlotMachine SlotTable(getParent());
Misha Brukmanb1c93172005-04-21 23:48:37 +00001462 AssemblyWriter W(o, SlotTable,
Chris Lattner8339f7d2003-10-30 23:41:03 +00001463 getParent() ? getParent()->getParent() : 0, AAW);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001464 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001465}
1466
Chris Lattner8339f7d2003-10-30 23:41:03 +00001467void Instruction::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001468 const Function *F = getParent() ? getParent()->getParent() : 0;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001469 SlotMachine SlotTable(F);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001470 AssemblyWriter W(o, SlotTable, F ? F->getParent() : 0, AAW);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001471
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001472 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001473}
Chris Lattner7db79582001-11-07 04:21:57 +00001474
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001475void Constant::print(std::ostream &o) const {
1476 if (this == 0) { o << "<null> constant value\n"; return; }
Chris Lattner7d734802002-09-10 15:53:49 +00001477
Misha Brukman21bbdb92004-06-04 21:11:51 +00001478 o << ' ' << getType()->getDescription() << ' ';
Evan Cheng5b19a802006-03-01 22:17:00 +00001479
1480 std::map<const Type *, std::string> TypeTable;
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001481 WriteConstantInt(o, this, TypeTable, 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001482}
1483
Misha Brukmanb1c93172005-04-21 23:48:37 +00001484void Type::print(std::ostream &o) const {
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001485 if (this == 0)
1486 o << "<null Type>";
1487 else
1488 o << getDescription();
1489}
1490
Chris Lattner2e9fa6d2002-04-09 19:48:49 +00001491void Argument::print(std::ostream &o) const {
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001492 WriteAsOperand(o, this, true, getParent() ? getParent()->getParent() : 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001493}
1494
Reid Spencer52641832004-05-25 18:14:38 +00001495// Value::dump - allow easy printing of Values from the debugger.
1496// Located here because so much of the needed functionality is here.
Bill Wendling22e978a2006-12-07 20:04:42 +00001497void Value::dump() const { print(*cerr.stream()); cerr << '\n'; }
Reid Spencer52641832004-05-25 18:14:38 +00001498
1499// Type::dump - allow easy printing of Values from the debugger.
1500// Located here because so much of the needed functionality is here.
Bill Wendling22e978a2006-12-07 20:04:42 +00001501void Type::dump() const { print(*cerr.stream()); cerr << '\n'; }
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001502
Reid Spencer4388f0b2007-04-22 05:46:44 +00001503void
1504ParamAttrsList::dump() const {
1505 cerr << "PAL[ ";
1506 for (unsigned i = 0; i < attrs.size(); ++i) {
1507 uint16_t index = getParamIndex(i);
1508 uint16_t attrs = getParamAttrs(index);
1509 cerr << "{" << index << "," << attrs << "} ";
1510 }
1511 cerr << "]\n";
1512}
1513
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001514//===----------------------------------------------------------------------===//
Chris Lattnerfc9f1c92006-12-06 06:40:49 +00001515// SlotMachine Implementation
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001516//===----------------------------------------------------------------------===//
1517
1518#if 0
Bill Wendlingf3baad32006-12-07 01:30:32 +00001519#define SC_DEBUG(X) cerr << X
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001520#else
1521#define SC_DEBUG(X)
1522#endif
1523
1524// Module level constructor. Causes the contents of the Module (sans functions)
1525// to be added to the slot table.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001526SlotMachine::SlotMachine(const Module *M)
Reid Spencer56010e42004-05-26 21:56:09 +00001527 : TheModule(M) ///< Saved for lazy initialization.
1528 , TheFunction(0)
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001529 , FunctionProcessed(false)
Reid Spencer50816782007-03-19 18:32:53 +00001530 , mMap(), mNext(0), fMap(), fNext(0)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001531{
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001532}
1533
1534// Function level constructor. Causes the contents of the Module and the one
1535// function provided to be added to the slot table.
Chris Lattner23e829a2006-12-06 05:12:21 +00001536SlotMachine::SlotMachine(const Function *F)
1537 : TheModule(F ? F->getParent() : 0) ///< Saved for lazy initialization
Reid Spencer56010e42004-05-26 21:56:09 +00001538 , TheFunction(F) ///< Saved for lazy initialization
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001539 , FunctionProcessed(false)
Reid Spencer50816782007-03-19 18:32:53 +00001540 , mMap(), mNext(0), fMap(), fNext(0)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001541{
Reid Spencer56010e42004-05-26 21:56:09 +00001542}
1543
Chris Lattner5e043322007-01-11 03:54:27 +00001544inline void SlotMachine::initialize() {
Chris Lattner23e829a2006-12-06 05:12:21 +00001545 if (TheModule) {
Misha Brukmanb1c93172005-04-21 23:48:37 +00001546 processModule();
Reid Spencer56010e42004-05-26 21:56:09 +00001547 TheModule = 0; ///< Prevent re-processing next time we're called.
1548 }
Chris Lattner193de902006-12-06 05:27:40 +00001549 if (TheFunction && !FunctionProcessed)
Misha Brukmanb1c93172005-04-21 23:48:37 +00001550 processFunction();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001551}
1552
1553// Iterate through all the global variables, functions, and global
Misha Brukmanb1c93172005-04-21 23:48:37 +00001554// variable initializers and create slots for them.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001555void SlotMachine::processModule() {
1556 SC_DEBUG("begin processModule!\n");
1557
Chris Lattner0dda8612006-12-06 06:15:43 +00001558 // Add all of the unnamed global variables to the value table.
Chris Lattner54932b02006-12-06 04:41:52 +00001559 for (Module::const_global_iterator I = TheModule->global_begin(),
1560 E = TheModule->global_end(); I != E; ++I)
Chris Lattner0dda8612006-12-06 06:15:43 +00001561 if (!I->hasName())
Chris Lattnerea862a32007-01-09 07:55:49 +00001562 CreateModuleSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001563
Chris Lattner0dda8612006-12-06 06:15:43 +00001564 // Add all the unnamed functions to the table.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001565 for (Module::const_iterator I = TheModule->begin(), E = TheModule->end();
1566 I != E; ++I)
Chris Lattner0dda8612006-12-06 06:15:43 +00001567 if (!I->hasName())
Chris Lattnerea862a32007-01-09 07:55:49 +00001568 CreateModuleSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001569
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001570 SC_DEBUG("end processModule!\n");
1571}
1572
1573
Reid Spencer56010e42004-05-26 21:56:09 +00001574// Process the arguments, basic blocks, and instructions of a function.
1575void SlotMachine::processFunction() {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001576 SC_DEBUG("begin processFunction!\n");
Reid Spencer50816782007-03-19 18:32:53 +00001577 fNext = 0;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001578
Chris Lattner0dda8612006-12-06 06:15:43 +00001579 // Add all the function arguments with no names.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001580 for(Function::const_arg_iterator AI = TheFunction->arg_begin(),
Chris Lattner531f9e92005-03-15 04:54:21 +00001581 AE = TheFunction->arg_end(); AI != AE; ++AI)
Chris Lattner0dda8612006-12-06 06:15:43 +00001582 if (!AI->hasName())
Chris Lattnerea862a32007-01-09 07:55:49 +00001583 CreateFunctionSlot(AI);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001584
1585 SC_DEBUG("Inserting Instructions:\n");
1586
Chris Lattner0dda8612006-12-06 06:15:43 +00001587 // Add all of the basic blocks and instructions with no names.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001588 for (Function::const_iterator BB = TheFunction->begin(),
Reid Spencer56010e42004-05-26 21:56:09 +00001589 E = TheFunction->end(); BB != E; ++BB) {
Chris Lattner0dda8612006-12-06 06:15:43 +00001590 if (!BB->hasName())
Chris Lattnerea862a32007-01-09 07:55:49 +00001591 CreateFunctionSlot(BB);
Chris Lattner0dda8612006-12-06 06:15:43 +00001592 for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I != E; ++I)
1593 if (I->getType() != Type::VoidTy && !I->hasName())
Chris Lattnerea862a32007-01-09 07:55:49 +00001594 CreateFunctionSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001595 }
1596
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001597 FunctionProcessed = true;
1598
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001599 SC_DEBUG("end processFunction!\n");
1600}
1601
Chris Lattner193de902006-12-06 05:27:40 +00001602/// Clean up after incorporating a function. This is the only way to get out of
Chris Lattner5e043322007-01-11 03:54:27 +00001603/// the function incorporation state that affects get*Slot/Create*Slot. Function
Chris Lattner89e774e2007-01-09 07:46:21 +00001604/// incorporation state is indicated by TheFunction != 0.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001605void SlotMachine::purgeFunction() {
1606 SC_DEBUG("begin purgeFunction!\n");
1607 fMap.clear(); // Simply discard the function level map
Reid Spencer56010e42004-05-26 21:56:09 +00001608 TheFunction = 0;
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001609 FunctionProcessed = false;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001610 SC_DEBUG("end purgeFunction!\n");
1611}
1612
Chris Lattner5e043322007-01-11 03:54:27 +00001613/// getGlobalSlot - Get the slot number of a global value.
1614int SlotMachine::getGlobalSlot(const GlobalValue *V) {
1615 // Check for uninitialized state and do lazy initialization.
1616 initialize();
1617
1618 // Find the type plane in the module map
Reid Spencer50816782007-03-19 18:32:53 +00001619 ValueMap::const_iterator MI = mMap.find(V);
Chris Lattner5e043322007-01-11 03:54:27 +00001620 if (MI == mMap.end()) return -1;
Reid Spencer50816782007-03-19 18:32:53 +00001621
1622 return MI->second;
Chris Lattner5e043322007-01-11 03:54:27 +00001623}
Reid Spencer56010e42004-05-26 21:56:09 +00001624
Chris Lattner5e043322007-01-11 03:54:27 +00001625
1626/// getLocalSlot - Get the slot number for a value that is local to a function.
1627int SlotMachine::getLocalSlot(const Value *V) {
1628 assert(!isa<Constant>(V) && "Can't get a constant or global slot with this!");
1629
1630 // Check for uninitialized state and do lazy initialization.
1631 initialize();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001632
Reid Spencer50816782007-03-19 18:32:53 +00001633 ValueMap::const_iterator FI = fMap.find(V);
Chris Lattner5e043322007-01-11 03:54:27 +00001634 if (FI == fMap.end()) return -1;
1635
Reid Spencer50816782007-03-19 18:32:53 +00001636 return FI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001637}
1638
Reid Spencer58d30f22004-07-04 11:50:43 +00001639
Chris Lattnerea862a32007-01-09 07:55:49 +00001640/// CreateModuleSlot - Insert the specified GlobalValue* into the slot table.
1641void SlotMachine::CreateModuleSlot(const GlobalValue *V) {
Chris Lattner04398e82007-01-09 08:04:59 +00001642 assert(V && "Can't insert a null Value into SlotMachine!");
Reid Spencer50816782007-03-19 18:32:53 +00001643 assert(V->getType() != Type::VoidTy && "Doesn't need a slot!");
1644 assert(!V->hasName() && "Doesn't need a slot!");
Chris Lattner04398e82007-01-09 08:04:59 +00001645
Reid Spencer50816782007-03-19 18:32:53 +00001646 unsigned DestSlot = mNext++;
1647 mMap[V] = DestSlot;
Chris Lattner04398e82007-01-09 08:04:59 +00001648
Reid Spencer50816782007-03-19 18:32:53 +00001649 SC_DEBUG(" Inserting value [" << V->getType() << "] = " << V << " slot=" <<
Chris Lattner04398e82007-01-09 08:04:59 +00001650 DestSlot << " [");
Anton Korobeynikova97b6942007-04-25 14:27:10 +00001651 // G = Global, F = Function, A = Alias, o = other
1652 SC_DEBUG((isa<GlobalVariable>(V) ? 'G' :
1653 (isa<Function> ? 'F' :
1654 (isa<GlobalAlias> ? 'A' : 'o'))) << "]\n");
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001655}
1656
1657
Chris Lattnerea862a32007-01-09 07:55:49 +00001658/// CreateSlot - Create a new slot for the specified value if it has no name.
1659void SlotMachine::CreateFunctionSlot(const Value *V) {
1660 const Type *VTy = V->getType();
1661 assert(VTy != Type::VoidTy && !V->hasName() && "Doesn't need a slot!");
Chris Lattner04398e82007-01-09 08:04:59 +00001662
Reid Spencer50816782007-03-19 18:32:53 +00001663 unsigned DestSlot = fNext++;
1664 fMap[V] = DestSlot;
Chris Lattner04398e82007-01-09 08:04:59 +00001665
Chris Lattnerf8090cb2006-12-06 05:55:41 +00001666 // G = Global, F = Function, o = other
Chris Lattner04398e82007-01-09 08:04:59 +00001667 SC_DEBUG(" Inserting value [" << VTy << "] = " << V << " slot=" <<
1668 DestSlot << " [o]\n");
1669}