blob: 704334b78ac2d655ccf2baa112b876ce82b95cff [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"
Chris Lattner8bbcda22006-01-25 18:57:27 +000023#include "llvm/InlineAsm.h"
Vikram S. Adveb952b542002-07-14 23:14:45 +000024#include "llvm/Instruction.h"
Misha Brukman2d3fa9e2004-07-29 16:53:53 +000025#include "llvm/Instructions.h"
Chris Lattnerc70b3f62004-01-20 19:50:34 +000026#include "llvm/Module.h"
Chris Lattnerfee714f2001-09-07 16:36:04 +000027#include "llvm/SymbolTable.h"
Reid Spencer32af9e82007-01-06 07:24:44 +000028#include "llvm/TypeSymbolTable.h"
Reid Spencer7c16caa2004-09-01 22:55:40 +000029#include "llvm/ADT/StringExtras.h"
30#include "llvm/ADT/STLExtras.h"
Bill Wendlingdfc91892006-11-28 02:09:03 +000031#include "llvm/Support/CFG.h"
Jim Laskeyb74c6662005-08-17 19:34:49 +000032#include "llvm/Support/MathExtras.h"
Bill Wendlingdfc91892006-11-28 02:09:03 +000033#include "llvm/Support/Streams.h"
Chris Lattnerfee714f2001-09-07 16:36:04 +000034#include <algorithm>
Chris Lattner189d19f2003-11-21 20:23:48 +000035using namespace llvm;
Brian Gaeke960707c2003-11-11 22:41:34 +000036
Chris Lattner60a7dd12004-07-15 02:51:31 +000037namespace llvm {
Reid Spencer16f2f7f2004-05-26 07:18:52 +000038
Reid Spencer294715b2005-05-15 16:13:11 +000039// Make virtual table appear in this compilation unit.
40AssemblyAnnotationWriter::~AssemblyAnnotationWriter() {}
41
Reid Spencer16f2f7f2004-05-26 07:18:52 +000042/// This class provides computation of slot numbers for LLVM Assembly writing.
43/// @brief LLVM Assembly Writing Slot Computation.
44class SlotMachine {
45
46/// @name Types
47/// @{
48public:
49
50 /// @brief A mapping of Values to slot numbers
51 typedef std::map<const Value*, unsigned> ValueMap;
52
53 /// @brief A plane with next slot number and ValueMap
Misha Brukmanb1c93172005-04-21 23:48:37 +000054 struct ValuePlane {
Reid Spencer16f2f7f2004-05-26 07:18:52 +000055 unsigned next_slot; ///< The next slot number to use
56 ValueMap map; ///< The map of Value* -> unsigned
Reid Spencer58d30f22004-07-04 11:50:43 +000057 ValuePlane() { next_slot = 0; } ///< Make sure we start at 0
58 };
59
Reid Spencer16f2f7f2004-05-26 07:18:52 +000060 /// @brief The map of planes by Type
Reid Spencer58d30f22004-07-04 11:50:43 +000061 typedef std::map<const Type*, ValuePlane> TypedPlanes;
Reid Spencer16f2f7f2004-05-26 07:18:52 +000062
63/// @}
64/// @name Constructors
65/// @{
66public:
67 /// @brief Construct from a module
Chris Lattner23e829a2006-12-06 05:12:21 +000068 SlotMachine(const Module *M);
Reid Spencer16f2f7f2004-05-26 07:18:52 +000069
70 /// @brief Construct from a function, starting out in incorp state.
Chris Lattner23e829a2006-12-06 05:12:21 +000071 SlotMachine(const Function *F);
Reid Spencer16f2f7f2004-05-26 07:18:52 +000072
73/// @}
74/// @name Accessors
75/// @{
76public:
77 /// Return the slot number of the specified value in it's type
78 /// plane. Its an error to ask for something not in the SlotMachine.
79 /// Its an error to ask for a Type*
Chris Lattner757ee0b2004-06-09 19:41:19 +000080 int getSlot(const Value *V);
Reid Spencer8beac692004-06-09 15:26:53 +000081
Reid Spencer16f2f7f2004-05-26 07:18:52 +000082/// @}
83/// @name Mutators
84/// @{
85public:
Misha Brukmanb1c93172005-04-21 23:48:37 +000086 /// If you'd like to deal with a function instead of just a module, use
Reid Spencer16f2f7f2004-05-26 07:18:52 +000087 /// this method to get its data into the SlotMachine.
Misha Brukmanb1c93172005-04-21 23:48:37 +000088 void incorporateFunction(const Function *F) {
89 TheFunction = F;
Reid Spencerb0ac8c42004-08-16 07:46:33 +000090 FunctionProcessed = false;
91 }
Reid Spencer16f2f7f2004-05-26 07:18:52 +000092
Misha Brukmanb1c93172005-04-21 23:48:37 +000093 /// After calling incorporateFunction, use this method to remove the
94 /// most recently incorporated function from the SlotMachine. This
Reid Spencer16f2f7f2004-05-26 07:18:52 +000095 /// will reset the state of the machine back to just the module contents.
96 void purgeFunction();
97
98/// @}
99/// @name Implementation Details
100/// @{
101private:
Reid Spencer56010e42004-05-26 21:56:09 +0000102 /// This function does the actual initialization.
103 inline void initialize();
104
Misha Brukmanb1c93172005-04-21 23:48:37 +0000105 /// Values can be crammed into here at will. If they haven't
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000106 /// been inserted already, they get inserted, otherwise they are ignored.
107 /// Either way, the slot number for the Value* is returned.
Chris Lattner193de902006-12-06 05:27:40 +0000108 unsigned getOrCreateSlot(const Value *V);
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000109
110 /// Insert a value into the value table. Return the slot number
111 /// that it now occupies. BadThings(TM) will happen if you insert a
Misha Brukmanb1c93172005-04-21 23:48:37 +0000112 /// Value that's already been inserted.
Chris Lattner23e829a2006-12-06 05:12:21 +0000113 unsigned insertValue(const Value *V);
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000114
115 /// Add all of the module level global variables (and their initializers)
116 /// and function declarations, but not the contents of those functions.
117 void processModule();
118
Reid Spencer56010e42004-05-26 21:56:09 +0000119 /// Add all of the functions arguments, basic blocks, and instructions
120 void processFunction();
121
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000122 SlotMachine(const SlotMachine &); // DO NOT IMPLEMENT
123 void operator=(const SlotMachine &); // DO NOT IMPLEMENT
124
125/// @}
126/// @name Data
127/// @{
128public:
129
130 /// @brief The module for which we are holding slot numbers
Reid Spencer56010e42004-05-26 21:56:09 +0000131 const Module* TheModule;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000132
Reid Spencer56010e42004-05-26 21:56:09 +0000133 /// @brief The function for which we are holding slot numbers
134 const Function* TheFunction;
Reid Spencerb0ac8c42004-08-16 07:46:33 +0000135 bool FunctionProcessed;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000136
137 /// @brief The TypePlanes map for the module level data
138 TypedPlanes mMap;
139
140 /// @brief The TypePlanes map for the function level data
141 TypedPlanes fMap;
142
143/// @}
144
145};
146
Chris Lattner60a7dd12004-07-15 02:51:31 +0000147} // end namespace llvm
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000148
Chris Lattnerc2d3d312006-08-27 22:42:52 +0000149static RegisterPass<PrintModulePass>
Chris Lattnere7134c52006-08-21 17:20:01 +0000150X("printm", "Print module to stderr");
Chris Lattnerc2d3d312006-08-27 22:42:52 +0000151static RegisterPass<PrintFunctionPass>
Chris Lattnere7134c52006-08-21 17:20:01 +0000152Y("print","Print function to stderr");
Chris Lattner7f8845a2002-07-23 18:07:49 +0000153
Misha Brukmanb1c93172005-04-21 23:48:37 +0000154static void WriteAsOperandInternal(std::ostream &Out, const Value *V,
Chris Lattnera9f0a112006-12-06 05:50:41 +0000155 std::map<const Type *, std::string> &TypeTable,
Reid Spencer58d30f22004-07-04 11:50:43 +0000156 SlotMachine *Machine);
157
Chris Lattnerb86620e2001-10-29 16:37:48 +0000158static const Module *getModuleFromVal(const Value *V) {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000159 if (const Argument *MA = dyn_cast<Argument>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000160 return MA->getParent() ? MA->getParent()->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000161 else if (const BasicBlock *BB = dyn_cast<BasicBlock>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000162 return BB->getParent() ? BB->getParent()->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000163 else if (const Instruction *I = dyn_cast<Instruction>(V)) {
Chris Lattner57698e22002-03-26 18:01:55 +0000164 const Function *M = I->getParent() ? I->getParent()->getParent() : 0;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000165 return M ? M->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000166 } else if (const GlobalValue *GV = dyn_cast<GlobalValue>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000167 return GV->getParent();
Chris Lattnerb86620e2001-10-29 16:37:48 +0000168 return 0;
169}
170
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000171static SlotMachine *createSlotMachine(const Value *V) {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000172 if (const Argument *FA = dyn_cast<Argument>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000173 return new SlotMachine(FA->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000174 } else if (const Instruction *I = dyn_cast<Instruction>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000175 return new SlotMachine(I->getParent()->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000176 } else if (const BasicBlock *BB = dyn_cast<BasicBlock>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000177 return new SlotMachine(BB->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000178 } else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(V)){
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000179 return new SlotMachine(GV->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000180 } else if (const Function *Func = dyn_cast<Function>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000181 return new SlotMachine(Func);
Chris Lattner7bfee412001-10-29 16:05:51 +0000182 }
183 return 0;
184}
Chris Lattner2f7c9632001-06-06 20:29:01 +0000185
Chris Lattner1c343042003-08-22 05:40:38 +0000186// getLLVMName - Turn the specified string into an 'LLVM name', which is either
187// prefixed with % (if the string only contains simple characters) or is
188// surrounded with ""'s (if it has special chars in it).
Alkis Evlogimenos991d6ad2004-12-10 05:41:10 +0000189static std::string getLLVMName(const std::string &Name,
190 bool prefixName = true) {
Chris Lattner1c343042003-08-22 05:40:38 +0000191 assert(!Name.empty() && "Cannot get empty name!");
192
193 // First character cannot start with a number...
194 if (Name[0] >= '0' && Name[0] <= '9')
195 return "\"" + Name + "\"";
196
197 // Scan to see if we have any characters that are not on the "white list"
198 for (unsigned i = 0, e = Name.size(); i != e; ++i) {
199 char C = Name[i];
200 assert(C != '"' && "Illegal character in LLVM value name!");
201 if ((C < 'a' || C > 'z') && (C < 'A' || C > 'Z') && (C < '0' || C > '9') &&
202 C != '-' && C != '.' && C != '_')
203 return "\"" + Name + "\"";
204 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000205
Chris Lattner1c343042003-08-22 05:40:38 +0000206 // If we get here, then the identifier is legal to use as a "VarID".
Alkis Evlogimenos991d6ad2004-12-10 05:41:10 +0000207 if (prefixName)
208 return "%"+Name;
209 else
210 return Name;
Chris Lattner1c343042003-08-22 05:40:38 +0000211}
212
Chris Lattnerb86620e2001-10-29 16:37:48 +0000213
Misha Brukmanc566ca362004-03-02 00:22:19 +0000214/// fillTypeNameTable - If the module has a symbol table, take all global types
215/// and stuff their names into the TypeNames map.
216///
Chris Lattnerb86620e2001-10-29 16:37:48 +0000217static void fillTypeNameTable(const Module *M,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000218 std::map<const Type *, std::string> &TypeNames) {
Chris Lattner98cf1f52002-11-20 18:36:02 +0000219 if (!M) return;
Reid Spencer32af9e82007-01-06 07:24:44 +0000220 const TypeSymbolTable &ST = M->getTypeSymbolTable();
221 TypeSymbolTable::const_iterator TI = ST.begin();
222 for (; TI != ST.end(); ++TI) {
Reid Spencere7e96712004-05-25 08:53:40 +0000223 // As a heuristic, don't insert pointer to primitive types, because
224 // they are used too often to have a single useful name.
225 //
226 const Type *Ty = cast<Type>(TI->second);
227 if (!isa<PointerType>(Ty) ||
Reid Spencer56010e42004-05-26 21:56:09 +0000228 !cast<PointerType>(Ty)->getElementType()->isPrimitiveType() ||
229 isa<OpaqueType>(cast<PointerType>(Ty)->getElementType()))
Reid Spencere7e96712004-05-25 08:53:40 +0000230 TypeNames.insert(std::make_pair(Ty, getLLVMName(TI->first)));
Chris Lattnerb86620e2001-10-29 16:37:48 +0000231 }
232}
233
234
235
Misha Brukmanb1c93172005-04-21 23:48:37 +0000236static void calcTypeName(const Type *Ty,
John Criswellcd116ba2004-06-01 14:54:08 +0000237 std::vector<const Type *> &TypeStack,
238 std::map<const Type *, std::string> &TypeNames,
239 std::string & Result){
240 if (Ty->isPrimitiveType() && !isa<OpaqueType>(Ty)) {
241 Result += Ty->getDescription(); // Base case
242 return;
243 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000244
245 // Check to see if the type is named.
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000246 std::map<const Type *, std::string>::iterator I = TypeNames.find(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000247 if (I != TypeNames.end()) {
248 Result += I->second;
249 return;
250 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000251
John Criswellcd116ba2004-06-01 14:54:08 +0000252 if (isa<OpaqueType>(Ty)) {
253 Result += "opaque";
254 return;
255 }
Chris Lattnerf14ead92003-10-30 00:22:33 +0000256
Chris Lattnerb86620e2001-10-29 16:37:48 +0000257 // Check to see if the Type is already on the stack...
258 unsigned Slot = 0, CurSize = TypeStack.size();
259 while (Slot < CurSize && TypeStack[Slot] != Ty) ++Slot; // Scan for type
260
Misha Brukmanb1c93172005-04-21 23:48:37 +0000261 // This is another base case for the recursion. In this case, we know
Chris Lattnerb86620e2001-10-29 16:37:48 +0000262 // that we have looped back to a type that we have previously visited.
263 // Generate the appropriate upreference to handle this.
John Criswellcd116ba2004-06-01 14:54:08 +0000264 if (Slot < CurSize) {
265 Result += "\\" + utostr(CurSize-Slot); // Here's the upreference
266 return;
267 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000268
269 TypeStack.push_back(Ty); // Recursive case: Add us to the stack..
Misha Brukmanb1c93172005-04-21 23:48:37 +0000270
Chris Lattner6b727592004-06-17 18:19:28 +0000271 switch (Ty->getTypeID()) {
Chris Lattner91db5822002-03-29 03:44:36 +0000272 case Type::FunctionTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000273 const FunctionType *FTy = cast<FunctionType>(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000274 calcTypeName(FTy->getReturnType(), TypeStack, TypeNames, Result);
275 Result += " (";
Reid Spencer8c4914c2006-12-31 05:24:50 +0000276 unsigned Idx = 1;
Chris Lattnerfa829be2004-02-09 04:14:01 +0000277 for (FunctionType::param_iterator I = FTy->param_begin(),
278 E = FTy->param_end(); I != E; ++I) {
279 if (I != FTy->param_begin())
Chris Lattnerb86620e2001-10-29 16:37:48 +0000280 Result += ", ";
John Criswellcd116ba2004-06-01 14:54:08 +0000281 calcTypeName(*I, TypeStack, TypeNames, Result);
Reid Spencer8c4914c2006-12-31 05:24:50 +0000282 if (FTy->getParamAttrs(Idx)) {
283 Result += + " ";
284 Result += FunctionType::getParamAttrsText(FTy->getParamAttrs(Idx));
285 }
286 Idx++;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000287 }
Chris Lattnerd816b532002-04-13 20:53:41 +0000288 if (FTy->isVarArg()) {
Chris Lattnerfa829be2004-02-09 04:14:01 +0000289 if (FTy->getNumParams()) Result += ", ";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000290 Result += "...";
291 }
292 Result += ")";
Reid Spencer136a91c2007-01-05 17:06:19 +0000293 if (FTy->getParamAttrs(0)) {
294 Result += " ";
295 Result += FunctionType::getParamAttrsText(FTy->getParamAttrs(0));
296 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000297 break;
298 }
299 case Type::StructTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000300 const StructType *STy = cast<StructType>(Ty);
Andrew Lenharthdcb3c972006-12-08 18:06:16 +0000301 if (STy->isPacked())
302 Result += '<';
John Criswellcd116ba2004-06-01 14:54:08 +0000303 Result += "{ ";
Chris Lattnerac6db752004-02-09 04:37:31 +0000304 for (StructType::element_iterator I = STy->element_begin(),
305 E = STy->element_end(); I != E; ++I) {
306 if (I != STy->element_begin())
Chris Lattnerb86620e2001-10-29 16:37:48 +0000307 Result += ", ";
John Criswellcd116ba2004-06-01 14:54:08 +0000308 calcTypeName(*I, TypeStack, TypeNames, Result);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000309 }
310 Result += " }";
Andrew Lenharthdcb3c972006-12-08 18:06:16 +0000311 if (STy->isPacked())
312 Result += '>';
Chris Lattnerb86620e2001-10-29 16:37:48 +0000313 break;
314 }
315 case Type::PointerTyID:
Misha Brukmanb1c93172005-04-21 23:48:37 +0000316 calcTypeName(cast<PointerType>(Ty)->getElementType(),
John Criswellcd116ba2004-06-01 14:54:08 +0000317 TypeStack, TypeNames, Result);
318 Result += "*";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000319 break;
320 case Type::ArrayTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000321 const ArrayType *ATy = cast<ArrayType>(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000322 Result += "[" + utostr(ATy->getNumElements()) + " x ";
323 calcTypeName(ATy->getElementType(), TypeStack, TypeNames, Result);
324 Result += "]";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000325 break;
326 }
Brian Gaeke02209042004-08-20 06:00:58 +0000327 case Type::PackedTyID: {
328 const PackedType *PTy = cast<PackedType>(Ty);
329 Result += "<" + utostr(PTy->getNumElements()) + " x ";
330 calcTypeName(PTy->getElementType(), TypeStack, TypeNames, Result);
331 Result += ">";
332 break;
333 }
Chris Lattner15285ab2003-05-14 17:50:47 +0000334 case Type::OpaqueTyID:
John Criswellcd116ba2004-06-01 14:54:08 +0000335 Result += "opaque";
Chris Lattner15285ab2003-05-14 17:50:47 +0000336 break;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000337 default:
John Criswellcd116ba2004-06-01 14:54:08 +0000338 Result += "<unrecognized-type>";
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000339 break;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000340 }
341
342 TypeStack.pop_back(); // Remove self from stack...
Chris Lattnerb86620e2001-10-29 16:37:48 +0000343}
344
345
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000346/// printTypeInt - The internal guts of printing out a type that has a
347/// potentially named portion.
348///
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000349static std::ostream &printTypeInt(std::ostream &Out, const Type *Ty,
350 std::map<const Type *, std::string> &TypeNames) {
Chris Lattnerb86620e2001-10-29 16:37:48 +0000351 // Primitive types always print out their description, regardless of whether
352 // they have been named or not.
353 //
Chris Lattner92d60532003-10-30 00:12:51 +0000354 if (Ty->isPrimitiveType() && !isa<OpaqueType>(Ty))
355 return Out << Ty->getDescription();
Chris Lattnerb86620e2001-10-29 16:37:48 +0000356
357 // Check to see if the type is named.
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000358 std::map<const Type *, std::string>::iterator I = TypeNames.find(Ty);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000359 if (I != TypeNames.end()) return Out << I->second;
360
361 // Otherwise we have a type that has not been named but is a derived type.
362 // Carefully recurse the type hierarchy to print out any contained symbolic
363 // names.
364 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000365 std::vector<const Type *> TypeStack;
John Criswellcd116ba2004-06-01 14:54:08 +0000366 std::string TypeName;
367 calcTypeName(Ty, TypeStack, TypeNames, TypeName);
Chris Lattner7f74a562002-01-20 22:54:45 +0000368 TypeNames.insert(std::make_pair(Ty, TypeName));//Cache type name for later use
John Criswellcd116ba2004-06-01 14:54:08 +0000369 return (Out << TypeName);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000370}
371
Chris Lattner34b95182001-10-31 04:33:19 +0000372
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000373/// WriteTypeSymbolic - This attempts to write the specified type as a symbolic
374/// type, iff there is an entry in the modules symbol table for the specified
375/// type or one of it's component types. This is slower than a simple x << Type
376///
Chris Lattner189d19f2003-11-21 20:23:48 +0000377std::ostream &llvm::WriteTypeSymbolic(std::ostream &Out, const Type *Ty,
378 const Module *M) {
Misha Brukmanb1c93172005-04-21 23:48:37 +0000379 Out << ' ';
Chris Lattnerb86620e2001-10-29 16:37:48 +0000380
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000381 // If they want us to print out a type, but there is no context, we can't
382 // print it symbolically.
383 if (!M)
Chris Lattner72f866e2001-10-29 16:40:32 +0000384 return Out << Ty->getDescription();
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000385
386 std::map<const Type *, std::string> TypeNames;
387 fillTypeNameTable(M, TypeNames);
388 return printTypeInt(Out, Ty, TypeNames);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000389}
390
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000391// PrintEscapedString - Print each character of the specified string, escaping
392// it if it is not printable or if it is an escape char.
393static void PrintEscapedString(const std::string &Str, std::ostream &Out) {
394 for (unsigned i = 0, e = Str.size(); i != e; ++i) {
395 unsigned char C = Str[i];
396 if (isprint(C) && C != '"' && C != '\\') {
397 Out << C;
398 } else {
399 Out << '\\'
400 << (char) ((C/16 < 10) ? ( C/16 +'0') : ( C/16 -10+'A'))
401 << (char)(((C&15) < 10) ? ((C&15)+'0') : ((C&15)-10+'A'));
402 }
403 }
404}
405
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000406static const char *getPredicateText(unsigned predicate) {
Reid Spencer812a1be2006-12-04 05:19:18 +0000407 const char * pred = "unknown";
408 switch (predicate) {
409 case FCmpInst::FCMP_FALSE: pred = "false"; break;
410 case FCmpInst::FCMP_OEQ: pred = "oeq"; break;
411 case FCmpInst::FCMP_OGT: pred = "ogt"; break;
412 case FCmpInst::FCMP_OGE: pred = "oge"; break;
413 case FCmpInst::FCMP_OLT: pred = "olt"; break;
414 case FCmpInst::FCMP_OLE: pred = "ole"; break;
415 case FCmpInst::FCMP_ONE: pred = "one"; break;
416 case FCmpInst::FCMP_ORD: pred = "ord"; break;
417 case FCmpInst::FCMP_UNO: pred = "uno"; break;
418 case FCmpInst::FCMP_UEQ: pred = "ueq"; break;
419 case FCmpInst::FCMP_UGT: pred = "ugt"; break;
420 case FCmpInst::FCMP_UGE: pred = "uge"; break;
421 case FCmpInst::FCMP_ULT: pred = "ult"; break;
422 case FCmpInst::FCMP_ULE: pred = "ule"; break;
423 case FCmpInst::FCMP_UNE: pred = "une"; break;
424 case FCmpInst::FCMP_TRUE: pred = "true"; break;
425 case ICmpInst::ICMP_EQ: pred = "eq"; break;
426 case ICmpInst::ICMP_NE: pred = "ne"; break;
427 case ICmpInst::ICMP_SGT: pred = "sgt"; break;
428 case ICmpInst::ICMP_SGE: pred = "sge"; break;
429 case ICmpInst::ICMP_SLT: pred = "slt"; break;
430 case ICmpInst::ICMP_SLE: pred = "sle"; break;
431 case ICmpInst::ICMP_UGT: pred = "ugt"; break;
432 case ICmpInst::ICMP_UGE: pred = "uge"; break;
433 case ICmpInst::ICMP_ULT: pred = "ult"; break;
434 case ICmpInst::ICMP_ULE: pred = "ule"; break;
435 }
436 return pred;
437}
438
Misha Brukmanb1c93172005-04-21 23:48:37 +0000439/// @brief Internal constant writer.
440static void WriteConstantInt(std::ostream &Out, const Constant *CV,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000441 std::map<const Type *, std::string> &TypeTable,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000442 SlotMachine *Machine) {
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000443 const int IndentSize = 4;
444 static std::string Indent = "\n";
Chris Lattner1e194682002-04-18 18:53:13 +0000445 if (const ConstantBool *CB = dyn_cast<ConstantBool>(CV)) {
Chris Lattner94e39cb2006-09-28 22:50:29 +0000446 Out << (CB->getValue() ? "true" : "false");
Reid Spencere0fc4df2006-10-20 07:07:24 +0000447 } else if (const ConstantInt *CI = dyn_cast<ConstantInt>(CV)) {
Reid Spencer2acb8e12006-12-19 21:16:35 +0000448 Out << CI->getSExtValue();
Chris Lattner1e194682002-04-18 18:53:13 +0000449 } else if (const ConstantFP *CFP = dyn_cast<ConstantFP>(CV)) {
450 // We would like to output the FP constant value in exponential notation,
451 // but we cannot do this if doing so will lose precision. Check here to
452 // make sure that we only output it in exponential format if we can parse
453 // the value back and get the same value.
454 //
455 std::string StrVal = ftostr(CFP->getValue());
456
457 // Check to make sure that the stringized number is not some string like
458 // "Inf" or NaN, that atof will accept, but the lexer will not. Check that
459 // the string matches the "[-+]?[0-9]" regex.
460 //
461 if ((StrVal[0] >= '0' && StrVal[0] <= '9') ||
462 ((StrVal[0] == '-' || StrVal[0] == '+') &&
Brian Gaeke87b4f072003-06-17 23:55:35 +0000463 (StrVal[1] >= '0' && StrVal[1] <= '9')))
Chris Lattner1e194682002-04-18 18:53:13 +0000464 // Reparse stringized version!
465 if (atof(StrVal.c_str()) == CFP->getValue()) {
Chris Lattner472cc102005-01-04 01:56:57 +0000466 Out << StrVal;
467 return;
Chris Lattner1e194682002-04-18 18:53:13 +0000468 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000469
Chris Lattner1e194682002-04-18 18:53:13 +0000470 // Otherwise we could not reparse it to exactly the same value, so we must
471 // output the string in hexadecimal format!
Chris Lattner472cc102005-01-04 01:56:57 +0000472 assert(sizeof(double) == sizeof(uint64_t) &&
Chris Lattner1e194682002-04-18 18:53:13 +0000473 "assuming that double is 64 bits!");
Jim Laskeyb74c6662005-08-17 19:34:49 +0000474 Out << "0x" << utohexstr(DoubleToBits(CFP->getValue()));
Chris Lattner1e194682002-04-18 18:53:13 +0000475
Chris Lattner76b2ff42004-02-15 05:55:15 +0000476 } else if (isa<ConstantAggregateZero>(CV)) {
477 Out << "zeroinitializer";
Chris Lattner1e194682002-04-18 18:53:13 +0000478 } else if (const ConstantArray *CA = dyn_cast<ConstantArray>(CV)) {
479 // As a special case, print the array as a string if it is an array of
480 // ubytes or an array of sbytes with positive values.
Misha Brukmanb1c93172005-04-21 23:48:37 +0000481 //
Chris Lattner1e194682002-04-18 18:53:13 +0000482 const Type *ETy = CA->getType()->getElementType();
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000483 if (CA->isString()) {
Chris Lattner1e194682002-04-18 18:53:13 +0000484 Out << "c\"";
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000485 PrintEscapedString(CA->getAsString(), Out);
Chris Lattner1e194682002-04-18 18:53:13 +0000486 Out << "\"";
487
488 } else { // Cannot output in string format...
Misha Brukman21bbdb92004-06-04 21:11:51 +0000489 Out << '[';
Chris Lattnerd84bb632002-04-16 21:36:08 +0000490 if (CA->getNumOperands()) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000491 Out << ' ';
Chris Lattner1e194682002-04-18 18:53:13 +0000492 printTypeInt(Out, ETy, TypeTable);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000493 WriteAsOperandInternal(Out, CA->getOperand(0),
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000494 TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000495 for (unsigned i = 1, e = CA->getNumOperands(); i != e; ++i) {
496 Out << ", ";
Chris Lattner1e194682002-04-18 18:53:13 +0000497 printTypeInt(Out, ETy, TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000498 WriteAsOperandInternal(Out, CA->getOperand(i), TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000499 }
500 }
501 Out << " ]";
502 }
503 } else if (const ConstantStruct *CS = dyn_cast<ConstantStruct>(CV)) {
Andrew Lenharth0d124b82007-01-08 18:21:30 +0000504 if (CS->getType()->isPacked())
505 Out << '<';
Misha Brukman21bbdb92004-06-04 21:11:51 +0000506 Out << '{';
Jim Laskey3bb78742006-02-25 12:27:03 +0000507 unsigned N = CS->getNumOperands();
508 if (N) {
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000509 if (N > 2) {
510 Indent += std::string(IndentSize, ' ');
511 Out << Indent;
512 } else {
513 Out << ' ';
514 }
Chris Lattnerd84bb632002-04-16 21:36:08 +0000515 printTypeInt(Out, CS->getOperand(0)->getType(), TypeTable);
516
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000517 WriteAsOperandInternal(Out, CS->getOperand(0), TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000518
Jim Laskey3bb78742006-02-25 12:27:03 +0000519 for (unsigned i = 1; i < N; i++) {
Chris Lattnerd84bb632002-04-16 21:36:08 +0000520 Out << ", ";
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000521 if (N > 2) Out << Indent;
Chris Lattnerd84bb632002-04-16 21:36:08 +0000522 printTypeInt(Out, CS->getOperand(i)->getType(), TypeTable);
523
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000524 WriteAsOperandInternal(Out, CS->getOperand(i), TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000525 }
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000526 if (N > 2) Indent.resize(Indent.size() - IndentSize);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000527 }
Jim Laskey3bb78742006-02-25 12:27:03 +0000528
Chris Lattnerd84bb632002-04-16 21:36:08 +0000529 Out << " }";
Andrew Lenharth0d124b82007-01-08 18:21:30 +0000530 if (CS->getType()->isPacked())
531 Out << '>';
Brian Gaeke02209042004-08-20 06:00:58 +0000532 } else if (const ConstantPacked *CP = dyn_cast<ConstantPacked>(CV)) {
533 const Type *ETy = CP->getType()->getElementType();
Misha Brukmanb1c93172005-04-21 23:48:37 +0000534 assert(CP->getNumOperands() > 0 &&
Brian Gaeke02209042004-08-20 06:00:58 +0000535 "Number of operands for a PackedConst must be > 0");
536 Out << '<';
537 Out << ' ';
538 printTypeInt(Out, ETy, TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000539 WriteAsOperandInternal(Out, CP->getOperand(0), TypeTable, Machine);
Brian Gaeke02209042004-08-20 06:00:58 +0000540 for (unsigned i = 1, e = CP->getNumOperands(); i != e; ++i) {
541 Out << ", ";
542 printTypeInt(Out, ETy, TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000543 WriteAsOperandInternal(Out, CP->getOperand(i), TypeTable, Machine);
Brian Gaeke02209042004-08-20 06:00:58 +0000544 }
545 Out << " >";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000546 } else if (isa<ConstantPointerNull>(CV)) {
547 Out << "null";
548
Chris Lattner5e0b9f22004-10-16 18:08:06 +0000549 } else if (isa<UndefValue>(CV)) {
550 Out << "undef";
551
Chris Lattner3cd8c562002-07-30 18:54:25 +0000552 } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CV)) {
Reid Spencer812a1be2006-12-04 05:19:18 +0000553 Out << CE->getOpcodeName();
554 if (CE->isCompare())
555 Out << " " << getPredicateText(CE->getPredicate());
556 Out << " (";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000557
Vikram S. Adveb952b542002-07-14 23:14:45 +0000558 for (User::const_op_iterator OI=CE->op_begin(); OI != CE->op_end(); ++OI) {
559 printTypeInt(Out, (*OI)->getType(), TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000560 WriteAsOperandInternal(Out, *OI, TypeTable, Machine);
Vikram S. Adveb952b542002-07-14 23:14:45 +0000561 if (OI+1 != CE->op_end())
Chris Lattner3cd8c562002-07-30 18:54:25 +0000562 Out << ", ";
Vikram S. Adveb952b542002-07-14 23:14:45 +0000563 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000564
Reid Spencer6c38f0b2006-11-27 01:05:10 +0000565 if (CE->isCast()) {
Chris Lattner83b396b2002-08-15 19:37:43 +0000566 Out << " to ";
567 printTypeInt(Out, CE->getType(), TypeTable);
568 }
Reid Spencer6c38f0b2006-11-27 01:05:10 +0000569
Misha Brukman21bbdb92004-06-04 21:11:51 +0000570 Out << ')';
Chris Lattner83b396b2002-08-15 19:37:43 +0000571
Chris Lattnerd84bb632002-04-16 21:36:08 +0000572 } else {
Vikram S. Adveb952b542002-07-14 23:14:45 +0000573 Out << "<placeholder or erroneous Constant>";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000574 }
575}
576
577
Misha Brukmanc566ca362004-03-02 00:22:19 +0000578/// WriteAsOperand - Write the name of the specified value out to the specified
579/// ostream. This can be useful when you just want to print int %reg126, not
580/// the whole instruction that generated it.
581///
Misha Brukmanb1c93172005-04-21 23:48:37 +0000582static void WriteAsOperandInternal(std::ostream &Out, const Value *V,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000583 std::map<const Type*, std::string> &TypeTable,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000584 SlotMachine *Machine) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000585 Out << ' ';
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000586 if (V->hasName())
Chris Lattner1c343042003-08-22 05:40:38 +0000587 Out << getLLVMName(V->getName());
Reid Spenceraccd7c72004-07-17 23:47:01 +0000588 else {
589 const Constant *CV = dyn_cast<Constant>(V);
Chris Lattnera2d810d2006-01-25 22:26:05 +0000590 if (CV && !isa<GlobalValue>(CV)) {
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000591 WriteConstantInt(Out, CV, TypeTable, Machine);
Chris Lattnera2d810d2006-01-25 22:26:05 +0000592 } else if (const InlineAsm *IA = dyn_cast<InlineAsm>(V)) {
593 Out << "asm ";
594 if (IA->hasSideEffects())
595 Out << "sideeffect ";
596 Out << '"';
597 PrintEscapedString(IA->getAsmString(), Out);
598 Out << "\", \"";
599 PrintEscapedString(IA->getConstraintString(), Out);
600 Out << '"';
601 } else {
Chris Lattnerd84bb632002-04-16 21:36:08 +0000602 int Slot;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000603 if (Machine) {
Reid Spencer56010e42004-05-26 21:56:09 +0000604 Slot = Machine->getSlot(V);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000605 } else {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000606 Machine = createSlotMachine(V);
Chris Lattner96749c42006-05-14 18:46:52 +0000607 if (Machine)
Chris Lattner757ee0b2004-06-09 19:41:19 +0000608 Slot = Machine->getSlot(V);
609 else
610 Slot = -1;
Reid Spencer56010e42004-05-26 21:56:09 +0000611 delete Machine;
Chris Lattnerd84bb632002-04-16 21:36:08 +0000612 }
Chris Lattner757ee0b2004-06-09 19:41:19 +0000613 if (Slot != -1)
614 Out << '%' << Slot;
615 else
616 Out << "<badref>";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000617 }
618 }
619}
620
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000621/// WriteAsOperand - Write the name of the specified value out to the specified
622/// ostream. This can be useful when you just want to print int %reg126, not
623/// the whole instruction that generated it.
624///
Chris Lattner189d19f2003-11-21 20:23:48 +0000625std::ostream &llvm::WriteAsOperand(std::ostream &Out, const Value *V,
Chris Lattner0dda8612006-12-06 06:15:43 +0000626 bool PrintType, const Module *Context) {
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000627 std::map<const Type *, std::string> TypeNames;
Chris Lattner5a9f63e2002-07-10 16:48:17 +0000628 if (Context == 0) Context = getModuleFromVal(V);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000629
Chris Lattner98cf1f52002-11-20 18:36:02 +0000630 if (Context)
Chris Lattner5a9f63e2002-07-10 16:48:17 +0000631 fillTypeNameTable(Context, TypeNames);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000632
633 if (PrintType)
634 printTypeInt(Out, V->getType(), TypeNames);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000635
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000636 WriteAsOperandInternal(Out, V, TypeNames, 0);
Chris Lattner5e5abe32001-07-20 19:15:21 +0000637 return Out;
638}
639
Reid Spencer58d30f22004-07-04 11:50:43 +0000640
Chris Lattner189d19f2003-11-21 20:23:48 +0000641namespace llvm {
Chris Lattner2e9fee42001-07-12 23:35:26 +0000642
Chris Lattnerfee714f2001-09-07 16:36:04 +0000643class AssemblyWriter {
Misha Brukmana6619a92004-06-21 21:53:56 +0000644 std::ostream &Out;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000645 SlotMachine &Machine;
Chris Lattner7bfee412001-10-29 16:05:51 +0000646 const Module *TheModule;
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000647 std::map<const Type *, std::string> TypeNames;
Chris Lattner8339f7d2003-10-30 23:41:03 +0000648 AssemblyAnnotationWriter *AnnotationWriter;
Chris Lattner2f7c9632001-06-06 20:29:01 +0000649public:
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000650 inline AssemblyWriter(std::ostream &o, SlotMachine &Mac, const Module *M,
Chris Lattner8339f7d2003-10-30 23:41:03 +0000651 AssemblyAnnotationWriter *AAW)
Misha Brukmana6619a92004-06-21 21:53:56 +0000652 : Out(o), Machine(Mac), TheModule(M), AnnotationWriter(AAW) {
Chris Lattner7bfee412001-10-29 16:05:51 +0000653
654 // If the module has a symbol table, take all global types and stuff their
655 // names into the TypeNames map.
656 //
Chris Lattnerb86620e2001-10-29 16:37:48 +0000657 fillTypeNameTable(M, TypeNames);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000658 }
659
Chris Lattner7bfee412001-10-29 16:05:51 +0000660 inline void write(const Module *M) { printModule(M); }
661 inline void write(const GlobalVariable *G) { printGlobal(G); }
Chris Lattner57698e22002-03-26 18:01:55 +0000662 inline void write(const Function *F) { printFunction(F); }
Chris Lattner7bfee412001-10-29 16:05:51 +0000663 inline void write(const BasicBlock *BB) { printBasicBlock(BB); }
Chris Lattner113f4f42002-06-25 16:13:24 +0000664 inline void write(const Instruction *I) { printInstruction(*I); }
Chris Lattner3462ae32001-12-03 22:26:30 +0000665 inline void write(const Constant *CPV) { printConstant(CPV); }
Chris Lattner7db79582001-11-07 04:21:57 +0000666 inline void write(const Type *Ty) { printType(Ty); }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000667
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000668 void writeOperand(const Value *Op, bool PrintType);
Chris Lattner1e194682002-04-18 18:53:13 +0000669
Misha Brukman4685e262004-04-28 15:31:21 +0000670 const Module* getModule() { return TheModule; }
671
Misha Brukmand92f54a2004-11-15 19:30:05 +0000672private:
Chris Lattner7bfee412001-10-29 16:05:51 +0000673 void printModule(const Module *M);
Reid Spencer32af9e82007-01-06 07:24:44 +0000674 void printTypeSymbolTable(const TypeSymbolTable &ST);
675 void printValueSymbolTable(const SymbolTable &ST);
Chris Lattner3462ae32001-12-03 22:26:30 +0000676 void printConstant(const Constant *CPV);
Chris Lattner7bfee412001-10-29 16:05:51 +0000677 void printGlobal(const GlobalVariable *GV);
Chris Lattner57698e22002-03-26 18:01:55 +0000678 void printFunction(const Function *F);
Reid Spencer8c4914c2006-12-31 05:24:50 +0000679 void printArgument(const Argument *FA, FunctionType::ParameterAttributes A);
Chris Lattner7bfee412001-10-29 16:05:51 +0000680 void printBasicBlock(const BasicBlock *BB);
Chris Lattner113f4f42002-06-25 16:13:24 +0000681 void printInstruction(const Instruction &I);
Chris Lattnerd816b532002-04-13 20:53:41 +0000682
683 // printType - Go to extreme measures to attempt to print out a short,
684 // symbolic version of a type name.
685 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000686 std::ostream &printType(const Type *Ty) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000687 return printTypeInt(Out, Ty, TypeNames);
Chris Lattnerd816b532002-04-13 20:53:41 +0000688 }
689
690 // printTypeAtLeastOneLevel - Print out one level of the possibly complex type
691 // without considering any symbolic types that we may have equal to it.
692 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000693 std::ostream &printTypeAtLeastOneLevel(const Type *Ty);
Chris Lattner7bfee412001-10-29 16:05:51 +0000694
Chris Lattner862e3382001-10-13 06:42:36 +0000695 // printInfoComment - Print a little comment after the instruction indicating
696 // which slot it occupies.
Chris Lattner113f4f42002-06-25 16:13:24 +0000697 void printInfoComment(const Value &V);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000698};
Reid Spencerf43ac622004-05-27 22:04:46 +0000699} // end of llvm namespace
Chris Lattner2f7c9632001-06-06 20:29:01 +0000700
Misha Brukmanc566ca362004-03-02 00:22:19 +0000701/// printTypeAtLeastOneLevel - Print out one level of the possibly complex type
702/// without considering any symbolic types that we may have equal to it.
703///
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000704std::ostream &AssemblyWriter::printTypeAtLeastOneLevel(const Type *Ty) {
Chris Lattner113f4f42002-06-25 16:13:24 +0000705 if (const FunctionType *FTy = dyn_cast<FunctionType>(Ty)) {
Reid Spencer8c4914c2006-12-31 05:24:50 +0000706 printType(FTy->getReturnType());
Reid Spencer8c4914c2006-12-31 05:24:50 +0000707 Out << " (";
708 unsigned Idx = 1;
Chris Lattnerfa829be2004-02-09 04:14:01 +0000709 for (FunctionType::param_iterator I = FTy->param_begin(),
710 E = FTy->param_end(); I != E; ++I) {
711 if (I != FTy->param_begin())
Misha Brukmana6619a92004-06-21 21:53:56 +0000712 Out << ", ";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000713 printType(*I);
Reid Spencer8c4914c2006-12-31 05:24:50 +0000714 if (FTy->getParamAttrs(Idx)) {
715 Out << " " << FunctionType::getParamAttrsText(FTy->getParamAttrs(Idx));
716 }
717 Idx++;
Chris Lattnerd816b532002-04-13 20:53:41 +0000718 }
719 if (FTy->isVarArg()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000720 if (FTy->getNumParams()) Out << ", ";
721 Out << "...";
Chris Lattnerd816b532002-04-13 20:53:41 +0000722 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000723 Out << ')';
Reid Spencer136a91c2007-01-05 17:06:19 +0000724 if (FTy->getParamAttrs(0))
725 Out << ' ' << FunctionType::getParamAttrsText(FTy->getParamAttrs(0));
Chris Lattner113f4f42002-06-25 16:13:24 +0000726 } else if (const StructType *STy = dyn_cast<StructType>(Ty)) {
Andrew Lenharthdcb3c972006-12-08 18:06:16 +0000727 if (STy->isPacked())
728 Out << '<';
Misha Brukmana6619a92004-06-21 21:53:56 +0000729 Out << "{ ";
Chris Lattnerac6db752004-02-09 04:37:31 +0000730 for (StructType::element_iterator I = STy->element_begin(),
731 E = STy->element_end(); I != E; ++I) {
732 if (I != STy->element_begin())
Misha Brukmana6619a92004-06-21 21:53:56 +0000733 Out << ", ";
Chris Lattnerd816b532002-04-13 20:53:41 +0000734 printType(*I);
735 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000736 Out << " }";
Andrew Lenharthdcb3c972006-12-08 18:06:16 +0000737 if (STy->isPacked())
738 Out << '>';
Chris Lattner113f4f42002-06-25 16:13:24 +0000739 } else if (const PointerType *PTy = dyn_cast<PointerType>(Ty)) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000740 printType(PTy->getElementType()) << '*';
Chris Lattner113f4f42002-06-25 16:13:24 +0000741 } else if (const ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000742 Out << '[' << ATy->getNumElements() << " x ";
Misha Brukman21bbdb92004-06-04 21:11:51 +0000743 printType(ATy->getElementType()) << ']';
Reid Spencere203d352004-08-20 15:37:30 +0000744 } else if (const PackedType *PTy = dyn_cast<PackedType>(Ty)) {
745 Out << '<' << PTy->getNumElements() << " x ";
746 printType(PTy->getElementType()) << '>';
747 }
Reid Spencerde46e482006-11-02 20:25:50 +0000748 else if (isa<OpaqueType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000749 Out << "opaque";
Chris Lattnerd816b532002-04-13 20:53:41 +0000750 } else {
Vikram S. Adveb952b542002-07-14 23:14:45 +0000751 if (!Ty->isPrimitiveType())
Misha Brukmana6619a92004-06-21 21:53:56 +0000752 Out << "<unknown derived type>";
Chris Lattnerd816b532002-04-13 20:53:41 +0000753 printType(Ty);
754 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000755 return Out;
Chris Lattnerd816b532002-04-13 20:53:41 +0000756}
757
758
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000759void AssemblyWriter::writeOperand(const Value *Operand, bool PrintType) {
760 if (Operand == 0) {
Chris Lattner08f7d0c2005-02-24 16:58:29 +0000761 Out << "<null operand!>";
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000762 } else {
763 if (PrintType) { Out << ' '; printType(Operand->getType()); }
764 WriteAsOperandInternal(Out, Operand, TypeNames, &Machine);
Chris Lattner08f7d0c2005-02-24 16:58:29 +0000765 }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000766}
767
Chris Lattner2f7c9632001-06-06 20:29:01 +0000768
Chris Lattner7bfee412001-10-29 16:05:51 +0000769void AssemblyWriter::printModule(const Module *M) {
Chris Lattner4d8689e2005-03-02 23:12:40 +0000770 if (!M->getModuleIdentifier().empty() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +0000771 // Don't print the ID if it will start a new line (which would
Chris Lattner4d8689e2005-03-02 23:12:40 +0000772 // require a comment char before it).
773 M->getModuleIdentifier().find('\n') == std::string::npos)
774 Out << "; ModuleID = '" << M->getModuleIdentifier() << "'\n";
775
Owen Andersone2237542006-10-18 02:21:12 +0000776 if (!M->getDataLayout().empty())
Chris Lattner04897162006-10-22 06:06:56 +0000777 Out << "target datalayout = \"" << M->getDataLayout() << "\"\n";
Owen Andersone2237542006-10-18 02:21:12 +0000778
Chris Lattner8068e0c2003-08-24 13:48:48 +0000779 switch (M->getEndianness()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000780 case Module::LittleEndian: Out << "target endian = little\n"; break;
781 case Module::BigEndian: Out << "target endian = big\n"; break;
Chris Lattner8068e0c2003-08-24 13:48:48 +0000782 case Module::AnyEndianness: break;
783 }
784 switch (M->getPointerSize()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000785 case Module::Pointer32: Out << "target pointersize = 32\n"; break;
786 case Module::Pointer64: Out << "target pointersize = 64\n"; break;
Chris Lattner8068e0c2003-08-24 13:48:48 +0000787 case Module::AnyPointerSize: break;
788 }
Reid Spencer48f98c82004-07-25 21:44:54 +0000789 if (!M->getTargetTriple().empty())
Reid Spencerffec7df2004-07-25 21:29:43 +0000790 Out << "target triple = \"" << M->getTargetTriple() << "\"\n";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000791
Chris Lattnereef2fe72006-01-24 04:13:11 +0000792 if (!M->getModuleInlineAsm().empty()) {
Chris Lattnerefaf35d2006-01-24 00:45:30 +0000793 // Split the string into lines, to make it easier to read the .ll file.
Chris Lattnereef2fe72006-01-24 04:13:11 +0000794 std::string Asm = M->getModuleInlineAsm();
Chris Lattnerefaf35d2006-01-24 00:45:30 +0000795 size_t CurPos = 0;
796 size_t NewLine = Asm.find_first_of('\n', CurPos);
797 while (NewLine != std::string::npos) {
798 // We found a newline, print the portion of the asm string from the
799 // last newline up to this newline.
800 Out << "module asm \"";
801 PrintEscapedString(std::string(Asm.begin()+CurPos, Asm.begin()+NewLine),
802 Out);
803 Out << "\"\n";
804 CurPos = NewLine+1;
805 NewLine = Asm.find_first_of('\n', CurPos);
806 }
Chris Lattner3acaf5c2006-01-24 00:40:17 +0000807 Out << "module asm \"";
Chris Lattnerefaf35d2006-01-24 00:45:30 +0000808 PrintEscapedString(std::string(Asm.begin()+CurPos, Asm.end()), Out);
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000809 Out << "\"\n";
810 }
811
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000812 // Loop over the dependent libraries and emit them.
Chris Lattner730cfe42004-09-14 04:51:44 +0000813 Module::lib_iterator LI = M->lib_begin();
814 Module::lib_iterator LE = M->lib_end();
Reid Spencer48f98c82004-07-25 21:44:54 +0000815 if (LI != LE) {
Chris Lattner730cfe42004-09-14 04:51:44 +0000816 Out << "deplibs = [ ";
817 while (LI != LE) {
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000818 Out << '"' << *LI << '"';
Reid Spencerffec7df2004-07-25 21:29:43 +0000819 ++LI;
Chris Lattner730cfe42004-09-14 04:51:44 +0000820 if (LI != LE)
821 Out << ", ";
Reid Spencerffec7df2004-07-25 21:29:43 +0000822 }
823 Out << " ]\n";
Reid Spencercc5ff642004-07-25 18:08:18 +0000824 }
Reid Spencerb9e08772004-09-13 23:44:23 +0000825
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000826 // Loop over the symbol table, emitting all named constants.
Reid Spencer32af9e82007-01-06 07:24:44 +0000827 printTypeSymbolTable(M->getTypeSymbolTable());
828 printValueSymbolTable(M->getValueSymbolTable());
Misha Brukmanb1c93172005-04-21 23:48:37 +0000829
Chris Lattner54932b02006-12-06 04:41:52 +0000830 for (Module::const_global_iterator I = M->global_begin(), E = M->global_end();
831 I != E; ++I)
Chris Lattner113f4f42002-06-25 16:13:24 +0000832 printGlobal(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000833
Misha Brukmana6619a92004-06-21 21:53:56 +0000834 Out << "\nimplementation ; Functions:\n";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000835
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000836 // Output all of the functions.
Chris Lattner113f4f42002-06-25 16:13:24 +0000837 for (Module::const_iterator I = M->begin(), E = M->end(); I != E; ++I)
838 printFunction(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000839}
840
Chris Lattner7bfee412001-10-29 16:05:51 +0000841void AssemblyWriter::printGlobal(const GlobalVariable *GV) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000842 if (GV->hasName()) Out << getLLVMName(GV->getName()) << " = ";
Chris Lattner37798642001-09-18 04:01:05 +0000843
Misha Brukmanb1c93172005-04-21 23:48:37 +0000844 if (!GV->hasInitializer())
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000845 switch (GV->getLinkage()) {
846 case GlobalValue::DLLImportLinkage: Out << "dllimport "; break;
847 case GlobalValue::ExternalWeakLinkage: Out << "extern_weak "; break;
848 default: Out << "external "; break;
849 }
Chris Lattner379a8d22003-04-16 20:28:45 +0000850 else
851 switch (GV->getLinkage()) {
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000852 case GlobalValue::InternalLinkage: Out << "internal "; break;
853 case GlobalValue::LinkOnceLinkage: Out << "linkonce "; break;
854 case GlobalValue::WeakLinkage: Out << "weak "; break;
855 case GlobalValue::AppendingLinkage: Out << "appending "; break;
856 case GlobalValue::DLLImportLinkage: Out << "dllimport "; break;
857 case GlobalValue::DLLExportLinkage: Out << "dllexport "; break;
858 case GlobalValue::ExternalWeakLinkage: Out << "extern_weak "; break;
859 case GlobalValue::ExternalLinkage: break;
Misha Brukman70457632004-11-14 21:04:34 +0000860 case GlobalValue::GhostLinkage:
Bill Wendlingf3baad32006-12-07 01:30:32 +0000861 cerr << "GhostLinkage not allowed in AsmWriter!\n";
Misha Brukman70457632004-11-14 21:04:34 +0000862 abort();
Chris Lattner379a8d22003-04-16 20:28:45 +0000863 }
Chris Lattner37798642001-09-18 04:01:05 +0000864
Misha Brukmana6619a92004-06-21 21:53:56 +0000865 Out << (GV->isConstant() ? "constant " : "global ");
Chris Lattner2413b162001-12-04 00:03:30 +0000866 printType(GV->getType()->getElementType());
Chris Lattner37798642001-09-18 04:01:05 +0000867
Reid Spenceraccd7c72004-07-17 23:47:01 +0000868 if (GV->hasInitializer()) {
869 Constant* C = cast<Constant>(GV->getInitializer());
870 assert(C && "GlobalVar initializer isn't constant?");
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000871 writeOperand(GV->getInitializer(), false);
Reid Spenceraccd7c72004-07-17 23:47:01 +0000872 }
Chris Lattnerf8a974d2005-11-06 06:48:53 +0000873
Chris Lattner4b96c542005-11-12 00:10:19 +0000874 if (GV->hasSection())
875 Out << ", section \"" << GV->getSection() << '"';
876 if (GV->getAlignment())
Chris Lattnerf8a974d2005-11-06 06:48:53 +0000877 Out << ", align " << GV->getAlignment();
Chris Lattner4b96c542005-11-12 00:10:19 +0000878
Chris Lattner113f4f42002-06-25 16:13:24 +0000879 printInfoComment(*GV);
Misha Brukmana6619a92004-06-21 21:53:56 +0000880 Out << "\n";
Chris Lattnerda975502001-09-10 07:58:01 +0000881}
882
Reid Spencer32af9e82007-01-06 07:24:44 +0000883void AssemblyWriter::printTypeSymbolTable(const TypeSymbolTable &ST) {
Reid Spencere7e96712004-05-25 08:53:40 +0000884 // Print the types.
Reid Spencer32af9e82007-01-06 07:24:44 +0000885 for (TypeSymbolTable::const_iterator TI = ST.begin(), TE = ST.end();
886 TI != TE; ++TI) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000887 Out << "\t" << getLLVMName(TI->first) << " = type ";
Reid Spencere7e96712004-05-25 08:53:40 +0000888
889 // Make sure we print out at least one level of the type structure, so
890 // that we do not get %FILE = type %FILE
891 //
892 printTypeAtLeastOneLevel(TI->second) << "\n";
893 }
Reid Spencer32af9e82007-01-06 07:24:44 +0000894}
895
896// printSymbolTable - Run through symbol table looking for constants
897// and types. Emit their declarations.
898void AssemblyWriter::printValueSymbolTable(const SymbolTable &ST) {
Misha Brukmanb1c93172005-04-21 23:48:37 +0000899
Reid Spencere7e96712004-05-25 08:53:40 +0000900 // Print the constants, in type plane order.
901 for (SymbolTable::plane_const_iterator PI = ST.plane_begin();
Chris Lattner23e829a2006-12-06 05:12:21 +0000902 PI != ST.plane_end(); ++PI) {
Reid Spencere7e96712004-05-25 08:53:40 +0000903 SymbolTable::value_const_iterator VI = ST.value_begin(PI->first);
904 SymbolTable::value_const_iterator VE = ST.value_end(PI->first);
905
906 for (; VI != VE; ++VI) {
Reid Spenceraccd7c72004-07-17 23:47:01 +0000907 const Value* V = VI->second;
908 const Constant *CPV = dyn_cast<Constant>(V) ;
909 if (CPV && !isa<GlobalValue>(V)) {
Reid Spencer56010e42004-05-26 21:56:09 +0000910 printConstant(CPV);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000911 }
912 }
Chris Lattnera7620d92001-07-15 06:35:59 +0000913 }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000914}
915
916
Misha Brukmanc566ca362004-03-02 00:22:19 +0000917/// printConstant - Print out a constant pool entry...
918///
Chris Lattner3462ae32001-12-03 22:26:30 +0000919void AssemblyWriter::printConstant(const Constant *CPV) {
Chris Lattnerfee714f2001-09-07 16:36:04 +0000920 // Don't print out unnamed constants, they will be inlined
921 if (!CPV->hasName()) return;
Chris Lattner2f7c9632001-06-06 20:29:01 +0000922
Chris Lattneree998be2001-07-26 16:29:38 +0000923 // Print out name...
Misha Brukmana6619a92004-06-21 21:53:56 +0000924 Out << "\t" << getLLVMName(CPV->getName()) << " =";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000925
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000926 // Write the value out now.
927 writeOperand(CPV, true);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000928
Chris Lattner113f4f42002-06-25 16:13:24 +0000929 printInfoComment(*CPV);
Misha Brukmana6619a92004-06-21 21:53:56 +0000930 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000931}
932
Misha Brukmanc566ca362004-03-02 00:22:19 +0000933/// printFunction - Print all aspects of a function.
934///
Chris Lattner113f4f42002-06-25 16:13:24 +0000935void AssemblyWriter::printFunction(const Function *F) {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000936 // Print out the return type and name...
Misha Brukmana6619a92004-06-21 21:53:56 +0000937 Out << "\n";
Chris Lattner379a8d22003-04-16 20:28:45 +0000938
Chris Lattner97e36f22004-12-05 06:44:09 +0000939 // Ensure that no local symbols conflict with global symbols.
940 const_cast<Function*>(F)->renameLocalSymbols();
941
Misha Brukmana6619a92004-06-21 21:53:56 +0000942 if (AnnotationWriter) AnnotationWriter->emitFunctionAnnot(F, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +0000943
Chris Lattner379a8d22003-04-16 20:28:45 +0000944 if (F->isExternal())
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000945 switch (F->getLinkage()) {
946 case GlobalValue::DLLImportLinkage: Out << "declare dllimport "; break;
947 case GlobalValue::ExternalWeakLinkage: Out << "declare extern_weak "; break;
948 default: Out << "declare ";
949 }
Reid Spencer7ce2d2a2006-12-29 20:29:48 +0000950 else {
951 Out << "define ";
Chris Lattner379a8d22003-04-16 20:28:45 +0000952 switch (F->getLinkage()) {
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000953 case GlobalValue::InternalLinkage: Out << "internal "; break;
954 case GlobalValue::LinkOnceLinkage: Out << "linkonce "; break;
955 case GlobalValue::WeakLinkage: Out << "weak "; break;
956 case GlobalValue::AppendingLinkage: Out << "appending "; break;
957 case GlobalValue::DLLImportLinkage: Out << "dllimport "; break;
958 case GlobalValue::DLLExportLinkage: Out << "dllexport "; break;
959 case GlobalValue::ExternalWeakLinkage: Out << "extern_weak "; break;
Chris Lattner379a8d22003-04-16 20:28:45 +0000960 case GlobalValue::ExternalLinkage: break;
Misha Brukman70457632004-11-14 21:04:34 +0000961 case GlobalValue::GhostLinkage:
Bill Wendlingf3baad32006-12-07 01:30:32 +0000962 cerr << "GhostLinkage not allowed in AsmWriter!\n";
Misha Brukman70457632004-11-14 21:04:34 +0000963 abort();
Chris Lattner379a8d22003-04-16 20:28:45 +0000964 }
Reid Spencer7ce2d2a2006-12-29 20:29:48 +0000965 }
Chris Lattner379a8d22003-04-16 20:28:45 +0000966
Chris Lattnerf7b6d312005-05-06 20:26:43 +0000967 // Print the calling convention.
968 switch (F->getCallingConv()) {
969 case CallingConv::C: break; // default
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +0000970 case CallingConv::CSRet: Out << "csretcc "; break;
971 case CallingConv::Fast: Out << "fastcc "; break;
972 case CallingConv::Cold: Out << "coldcc "; break;
973 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
974 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +0000975 default: Out << "cc" << F->getCallingConv() << " "; break;
976 }
977
Reid Spencer8c4914c2006-12-31 05:24:50 +0000978 const FunctionType *FT = F->getFunctionType();
Misha Brukman21bbdb92004-06-04 21:11:51 +0000979 printType(F->getReturnType()) << ' ';
Chris Lattner5b337482003-10-18 05:57:43 +0000980 if (!F->getName().empty())
Misha Brukmana6619a92004-06-21 21:53:56 +0000981 Out << getLLVMName(F->getName());
Chris Lattner5b337482003-10-18 05:57:43 +0000982 else
Misha Brukmana6619a92004-06-21 21:53:56 +0000983 Out << "\"\"";
984 Out << '(';
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000985 Machine.incorporateFunction(F);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000986
Chris Lattner7bfee412001-10-29 16:05:51 +0000987 // Loop over the arguments, printing them...
Chris Lattnerfee714f2001-09-07 16:36:04 +0000988
Reid Spencer8c4914c2006-12-31 05:24:50 +0000989 unsigned Idx = 1;
Chris Lattner54932b02006-12-06 04:41:52 +0000990 for (Function::const_arg_iterator I = F->arg_begin(), E = F->arg_end();
Reid Spencer8c4914c2006-12-31 05:24:50 +0000991 I != E; ++I) {
992 // Insert commas as we go... the first arg doesn't get a comma
993 if (I != F->arg_begin()) Out << ", ";
994 printArgument(I, FT->getParamAttrs(Idx));
995 Idx++;
996 }
Chris Lattnerfee714f2001-09-07 16:36:04 +0000997
998 // Finish printing arguments...
Chris Lattner113f4f42002-06-25 16:13:24 +0000999 if (FT->isVarArg()) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001000 if (FT->getNumParams()) Out << ", ";
1001 Out << "..."; // Output varargs portion of signature!
Chris Lattnerfee714f2001-09-07 16:36:04 +00001002 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001003 Out << ')';
Reid Spencer136a91c2007-01-05 17:06:19 +00001004 if (FT->getParamAttrs(0))
1005 Out << ' ' << FunctionType::getParamAttrsText(FT->getParamAttrs(0));
Chris Lattner4b96c542005-11-12 00:10:19 +00001006 if (F->hasSection())
1007 Out << " section \"" << F->getSection() << '"';
Chris Lattnerf8a974d2005-11-06 06:48:53 +00001008 if (F->getAlignment())
1009 Out << " align " << F->getAlignment();
Chris Lattner4b96c542005-11-12 00:10:19 +00001010
Chris Lattner113f4f42002-06-25 16:13:24 +00001011 if (F->isExternal()) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001012 Out << "\n";
Chris Lattnerb2f02e52002-05-06 03:00:40 +00001013 } else {
Misha Brukmana6619a92004-06-21 21:53:56 +00001014 Out << " {";
Misha Brukmanb1c93172005-04-21 23:48:37 +00001015
Chris Lattner6915f8f2002-04-07 22:49:37 +00001016 // Output all of its basic blocks... for the function
Chris Lattner113f4f42002-06-25 16:13:24 +00001017 for (Function::const_iterator I = F->begin(), E = F->end(); I != E; ++I)
1018 printBasicBlock(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +00001019
Misha Brukmana6619a92004-06-21 21:53:56 +00001020 Out << "}\n";
Chris Lattnerfee714f2001-09-07 16:36:04 +00001021 }
1022
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001023 Machine.purgeFunction();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001024}
1025
Misha Brukmanc566ca362004-03-02 00:22:19 +00001026/// printArgument - This member is called for every argument that is passed into
1027/// the function. Simply print it out
1028///
Reid Spencer8c4914c2006-12-31 05:24:50 +00001029void AssemblyWriter::printArgument(const Argument *Arg,
1030 FunctionType::ParameterAttributes attrs) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001031 // Output type...
Chris Lattner7bfee412001-10-29 16:05:51 +00001032 printType(Arg->getType());
Misha Brukmanb1c93172005-04-21 23:48:37 +00001033
Reid Spencer8c4914c2006-12-31 05:24:50 +00001034 if (attrs != FunctionType::NoAttributeSet)
1035 Out << ' ' << FunctionType::getParamAttrsText(attrs);
1036
Chris Lattner2f7c9632001-06-06 20:29:01 +00001037 // Output name, if available...
1038 if (Arg->hasName())
Misha Brukmana6619a92004-06-21 21:53:56 +00001039 Out << ' ' << getLLVMName(Arg->getName());
Chris Lattner2f7c9632001-06-06 20:29:01 +00001040}
1041
Misha Brukmanc566ca362004-03-02 00:22:19 +00001042/// printBasicBlock - This member is called for each basic block in a method.
1043///
Chris Lattner7bfee412001-10-29 16:05:51 +00001044void AssemblyWriter::printBasicBlock(const BasicBlock *BB) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001045 if (BB->hasName()) { // Print out the label if it exists...
Alkis Evlogimenos991d6ad2004-12-10 05:41:10 +00001046 Out << "\n" << getLLVMName(BB->getName(), false) << ':';
Chris Lattner408dbdb2002-05-14 16:02:05 +00001047 } else if (!BB->use_empty()) { // Don't print block # of no uses...
Misha Brukmana6619a92004-06-21 21:53:56 +00001048 Out << "\n; <label>:";
Chris Lattner757ee0b2004-06-09 19:41:19 +00001049 int Slot = Machine.getSlot(BB);
1050 if (Slot != -1)
Misha Brukmana6619a92004-06-21 21:53:56 +00001051 Out << Slot;
Chris Lattner757ee0b2004-06-09 19:41:19 +00001052 else
Misha Brukmana6619a92004-06-21 21:53:56 +00001053 Out << "<badref>";
Chris Lattner58185f22002-10-02 19:38:55 +00001054 }
Chris Lattner2447ef52003-11-20 00:09:43 +00001055
1056 if (BB->getParent() == 0)
Misha Brukmana6619a92004-06-21 21:53:56 +00001057 Out << "\t\t; Error: Block without parent!";
Chris Lattner2447ef52003-11-20 00:09:43 +00001058 else {
1059 if (BB != &BB->getParent()->front()) { // Not the entry block?
1060 // Output predecessors for the block...
Misha Brukmana6619a92004-06-21 21:53:56 +00001061 Out << "\t\t;";
Chris Lattner2447ef52003-11-20 00:09:43 +00001062 pred_const_iterator PI = pred_begin(BB), PE = pred_end(BB);
Misha Brukmanb1c93172005-04-21 23:48:37 +00001063
Chris Lattner2447ef52003-11-20 00:09:43 +00001064 if (PI == PE) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001065 Out << " No predecessors!";
Chris Lattner2447ef52003-11-20 00:09:43 +00001066 } else {
Misha Brukmana6619a92004-06-21 21:53:56 +00001067 Out << " preds =";
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001068 writeOperand(*PI, false);
Chris Lattner2447ef52003-11-20 00:09:43 +00001069 for (++PI; PI != PE; ++PI) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001070 Out << ',';
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001071 writeOperand(*PI, false);
Chris Lattner2447ef52003-11-20 00:09:43 +00001072 }
Chris Lattner00211f12003-11-16 22:59:57 +00001073 }
Chris Lattner58185f22002-10-02 19:38:55 +00001074 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001075 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001076
Misha Brukmana6619a92004-06-21 21:53:56 +00001077 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001078
Misha Brukmana6619a92004-06-21 21:53:56 +00001079 if (AnnotationWriter) AnnotationWriter->emitBasicBlockStartAnnot(BB, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001080
Chris Lattnerfee714f2001-09-07 16:36:04 +00001081 // Output all of the instructions in the basic block...
Chris Lattner113f4f42002-06-25 16:13:24 +00001082 for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I != E; ++I)
1083 printInstruction(*I);
Chris Lattner96cdd272004-03-08 18:51:45 +00001084
Misha Brukmana6619a92004-06-21 21:53:56 +00001085 if (AnnotationWriter) AnnotationWriter->emitBasicBlockEndAnnot(BB, Out);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001086}
1087
Chris Lattner862e3382001-10-13 06:42:36 +00001088
Misha Brukmanc566ca362004-03-02 00:22:19 +00001089/// printInfoComment - Print a little comment after the instruction indicating
1090/// which slot it occupies.
1091///
Chris Lattner113f4f42002-06-25 16:13:24 +00001092void AssemblyWriter::printInfoComment(const Value &V) {
1093 if (V.getType() != Type::VoidTy) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001094 Out << "\t\t; <";
Misha Brukman21bbdb92004-06-04 21:11:51 +00001095 printType(V.getType()) << '>';
Chris Lattner862e3382001-10-13 06:42:36 +00001096
Chris Lattner113f4f42002-06-25 16:13:24 +00001097 if (!V.hasName()) {
Chris Lattner757ee0b2004-06-09 19:41:19 +00001098 int SlotNum = Machine.getSlot(&V);
1099 if (SlotNum == -1)
Misha Brukmana6619a92004-06-21 21:53:56 +00001100 Out << ":<badref>";
Reid Spencer8beac692004-06-09 15:26:53 +00001101 else
Misha Brukmana6619a92004-06-21 21:53:56 +00001102 Out << ':' << SlotNum; // Print out the def slot taken.
Chris Lattner862e3382001-10-13 06:42:36 +00001103 }
Chris Lattnerb6c21db2005-02-01 01:24:01 +00001104 Out << " [#uses=" << V.getNumUses() << ']'; // Output # uses
Chris Lattner862e3382001-10-13 06:42:36 +00001105 }
1106}
1107
Reid Spencere7141c82006-08-28 01:02:49 +00001108// This member is called for each Instruction in a function..
Chris Lattner113f4f42002-06-25 16:13:24 +00001109void AssemblyWriter::printInstruction(const Instruction &I) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001110 if (AnnotationWriter) AnnotationWriter->emitInstructionAnnot(&I, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001111
Misha Brukmana6619a92004-06-21 21:53:56 +00001112 Out << "\t";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001113
1114 // Print out name if it exists...
Chris Lattner113f4f42002-06-25 16:13:24 +00001115 if (I.hasName())
Misha Brukmana6619a92004-06-21 21:53:56 +00001116 Out << getLLVMName(I.getName()) << " = ";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001117
Chris Lattner06038452005-05-06 05:51:46 +00001118 // If this is a volatile load or store, print out the volatile marker.
Chris Lattner504f9242003-09-08 17:45:59 +00001119 if ((isa<LoadInst>(I) && cast<LoadInst>(I).isVolatile()) ||
Chris Lattner06038452005-05-06 05:51:46 +00001120 (isa<StoreInst>(I) && cast<StoreInst>(I).isVolatile())) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001121 Out << "volatile ";
Chris Lattner06038452005-05-06 05:51:46 +00001122 } else if (isa<CallInst>(I) && cast<CallInst>(I).isTailCall()) {
1123 // If this is a call, check if it's a tail call.
1124 Out << "tail ";
1125 }
Chris Lattner504f9242003-09-08 17:45:59 +00001126
Chris Lattner2f7c9632001-06-06 20:29:01 +00001127 // Print out the opcode...
Misha Brukmana6619a92004-06-21 21:53:56 +00001128 Out << I.getOpcodeName();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001129
Reid Spencer45e52392006-12-03 06:27:29 +00001130 // Print out the compare instruction predicates
1131 if (const FCmpInst *FCI = dyn_cast<FCmpInst>(&I)) {
Reid Spencer812a1be2006-12-04 05:19:18 +00001132 Out << " " << getPredicateText(FCI->getPredicate());
Reid Spencer45e52392006-12-03 06:27:29 +00001133 } else if (const ICmpInst *ICI = dyn_cast<ICmpInst>(&I)) {
Reid Spencer812a1be2006-12-04 05:19:18 +00001134 Out << " " << getPredicateText(ICI->getPredicate());
Reid Spencer45e52392006-12-03 06:27:29 +00001135 }
1136
Chris Lattner2f7c9632001-06-06 20:29:01 +00001137 // Print out the type of the operands...
Chris Lattner113f4f42002-06-25 16:13:24 +00001138 const Value *Operand = I.getNumOperands() ? I.getOperand(0) : 0;
Chris Lattner2f7c9632001-06-06 20:29:01 +00001139
1140 // Special case conditional branches to swizzle the condition out to the front
Chris Lattner113f4f42002-06-25 16:13:24 +00001141 if (isa<BranchInst>(I) && I.getNumOperands() > 1) {
1142 writeOperand(I.getOperand(2), true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001143 Out << ',';
Chris Lattner2f7c9632001-06-06 20:29:01 +00001144 writeOperand(Operand, true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001145 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001146 writeOperand(I.getOperand(1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001147
Chris Lattner8d48df22002-04-13 18:34:38 +00001148 } else if (isa<SwitchInst>(I)) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001149 // Special case switch statement to get formatting nice and correct...
Misha Brukmana6619a92004-06-21 21:53:56 +00001150 writeOperand(Operand , true); Out << ',';
1151 writeOperand(I.getOperand(1), true); Out << " [";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001152
Chris Lattner113f4f42002-06-25 16:13:24 +00001153 for (unsigned op = 2, Eop = I.getNumOperands(); op < Eop; op += 2) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001154 Out << "\n\t\t";
1155 writeOperand(I.getOperand(op ), true); Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001156 writeOperand(I.getOperand(op+1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001157 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001158 Out << "\n\t]";
Chris Lattnerda558102001-10-02 03:41:24 +00001159 } else if (isa<PHINode>(I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001160 Out << ' ';
Chris Lattner113f4f42002-06-25 16:13:24 +00001161 printType(I.getType());
Misha Brukmana6619a92004-06-21 21:53:56 +00001162 Out << ' ';
Chris Lattner2f7c9632001-06-06 20:29:01 +00001163
Chris Lattner113f4f42002-06-25 16:13:24 +00001164 for (unsigned op = 0, Eop = I.getNumOperands(); op < Eop; op += 2) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001165 if (op) Out << ", ";
Misha Brukmanb1c93172005-04-21 23:48:37 +00001166 Out << '[';
Misha Brukmana6619a92004-06-21 21:53:56 +00001167 writeOperand(I.getOperand(op ), false); Out << ',';
1168 writeOperand(I.getOperand(op+1), false); Out << " ]";
Chris Lattner931ef3b2001-06-11 15:04:20 +00001169 }
Chris Lattner862e3382001-10-13 06:42:36 +00001170 } else if (isa<ReturnInst>(I) && !Operand) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001171 Out << " void";
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001172 } else if (const CallInst *CI = dyn_cast<CallInst>(&I)) {
1173 // Print the calling convention being used.
1174 switch (CI->getCallingConv()) {
1175 case CallingConv::C: break; // default
Chris Lattner29d20852006-05-19 21:58:52 +00001176 case CallingConv::CSRet: Out << " csretcc"; break;
1177 case CallingConv::Fast: Out << " fastcc"; break;
1178 case CallingConv::Cold: Out << " coldcc"; break;
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +00001179 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
1180 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001181 default: Out << " cc" << CI->getCallingConv(); break;
1182 }
1183
Chris Lattner463d6a52003-08-05 15:34:45 +00001184 const PointerType *PTy = cast<PointerType>(Operand->getType());
1185 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1186 const Type *RetTy = FTy->getReturnType();
Chris Lattner2f2d9472001-11-06 21:28:12 +00001187
Chris Lattner463d6a52003-08-05 15:34:45 +00001188 // If possible, print out the short form of the call instruction. We can
Chris Lattner6915f8f2002-04-07 22:49:37 +00001189 // only do this if the first argument is a pointer to a nonvararg function,
Chris Lattner463d6a52003-08-05 15:34:45 +00001190 // and if the return type is not a pointer to a function.
Chris Lattner2f2d9472001-11-06 21:28:12 +00001191 //
Chris Lattner463d6a52003-08-05 15:34:45 +00001192 if (!FTy->isVarArg() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +00001193 (!isa<PointerType>(RetTy) ||
Chris Lattnerd9a36a62002-07-25 20:58:51 +00001194 !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001195 Out << ' '; printType(RetTy);
Chris Lattner2f2d9472001-11-06 21:28:12 +00001196 writeOperand(Operand, false);
1197 } else {
1198 writeOperand(Operand, true);
1199 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001200 Out << '(';
Reid Spencer8c4914c2006-12-31 05:24:50 +00001201 for (unsigned op = 1, Eop = I.getNumOperands(); op < Eop; ++op) {
1202 if (op > 1)
1203 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001204 writeOperand(I.getOperand(op), true);
Reid Spencer8c4914c2006-12-31 05:24:50 +00001205 if (FTy->getParamAttrs(op) != FunctionType::NoAttributeSet)
1206 Out << " " << FTy->getParamAttrsText(FTy->getParamAttrs(op));
Chris Lattner2f7c9632001-06-06 20:29:01 +00001207 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001208 Out << " )";
Reid Spencer136a91c2007-01-05 17:06:19 +00001209 if (FTy->getParamAttrs(0) != FunctionType::NoAttributeSet)
1210 Out << ' ' << FTy->getParamAttrsText(FTy->getParamAttrs(0));
Chris Lattner113f4f42002-06-25 16:13:24 +00001211 } else if (const InvokeInst *II = dyn_cast<InvokeInst>(&I)) {
Chris Lattner463d6a52003-08-05 15:34:45 +00001212 const PointerType *PTy = cast<PointerType>(Operand->getType());
1213 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1214 const Type *RetTy = FTy->getReturnType();
1215
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001216 // Print the calling convention being used.
1217 switch (II->getCallingConv()) {
1218 case CallingConv::C: break; // default
Chris Lattner29d20852006-05-19 21:58:52 +00001219 case CallingConv::CSRet: Out << " csretcc"; break;
1220 case CallingConv::Fast: Out << " fastcc"; break;
1221 case CallingConv::Cold: Out << " coldcc"; break;
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +00001222 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
1223 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001224 default: Out << " cc" << II->getCallingConv(); break;
1225 }
1226
Chris Lattner463d6a52003-08-05 15:34:45 +00001227 // If possible, print out the short form of the invoke instruction. We can
1228 // only do this if the first argument is a pointer to a nonvararg function,
1229 // and if the return type is not a pointer to a function.
1230 //
1231 if (!FTy->isVarArg() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +00001232 (!isa<PointerType>(RetTy) ||
Chris Lattner463d6a52003-08-05 15:34:45 +00001233 !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001234 Out << ' '; printType(RetTy);
Chris Lattner463d6a52003-08-05 15:34:45 +00001235 writeOperand(Operand, false);
1236 } else {
1237 writeOperand(Operand, true);
1238 }
1239
Misha Brukmana6619a92004-06-21 21:53:56 +00001240 Out << '(';
Reid Spencer8c4914c2006-12-31 05:24:50 +00001241 for (unsigned op = 3, Eop = I.getNumOperands(); op < Eop; ++op) {
1242 if (op > 3)
1243 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001244 writeOperand(I.getOperand(op), true);
Reid Spencer9cfa4e82006-12-31 22:17:01 +00001245 if (FTy->getParamAttrs(op-2) != FunctionType::NoAttributeSet)
1246 Out << " " << FTy->getParamAttrsText(FTy->getParamAttrs(op-2));
Chris Lattner862e3382001-10-13 06:42:36 +00001247 }
1248
Reid Spencer136a91c2007-01-05 17:06:19 +00001249 Out << " )";
1250 if (FTy->getParamAttrs(0) != FunctionType::NoAttributeSet)
1251 Out << " " << FTy->getParamAttrsText(FTy->getParamAttrs(0));
1252 Out << "\n\t\t\tto";
Chris Lattner862e3382001-10-13 06:42:36 +00001253 writeOperand(II->getNormalDest(), true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001254 Out << " unwind";
Chris Lattnerfae8ab32004-02-08 21:44:31 +00001255 writeOperand(II->getUnwindDest(), true);
Chris Lattner862e3382001-10-13 06:42:36 +00001256
Chris Lattner113f4f42002-06-25 16:13:24 +00001257 } else if (const AllocationInst *AI = dyn_cast<AllocationInst>(&I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001258 Out << ' ';
Chris Lattner8d48df22002-04-13 18:34:38 +00001259 printType(AI->getType()->getElementType());
1260 if (AI->isArrayAllocation()) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001261 Out << ',';
Chris Lattner8d48df22002-04-13 18:34:38 +00001262 writeOperand(AI->getArraySize(), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001263 }
Nate Begeman848622f2005-11-05 09:21:28 +00001264 if (AI->getAlignment()) {
Chris Lattner7aeee3a2005-11-05 21:20:34 +00001265 Out << ", align " << AI->getAlignment();
Nate Begeman848622f2005-11-05 09:21:28 +00001266 }
Chris Lattner862e3382001-10-13 06:42:36 +00001267 } else if (isa<CastInst>(I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001268 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001269 Out << " to ";
Chris Lattner113f4f42002-06-25 16:13:24 +00001270 printType(I.getType());
Chris Lattner5b337482003-10-18 05:57:43 +00001271 } else if (isa<VAArgInst>(I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001272 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001273 Out << ", ";
Chris Lattnerf70da102003-05-08 02:44:12 +00001274 printType(I.getType());
Chris Lattner2f7c9632001-06-06 20:29:01 +00001275 } else if (Operand) { // Print the normal way...
1276
Misha Brukmanb1c93172005-04-21 23:48:37 +00001277 // PrintAllTypes - Instructions who have operands of all the same type
Chris Lattner2f7c9632001-06-06 20:29:01 +00001278 // omit the type from all but the first operand. If the instruction has
1279 // different type operands (for example br), then they are all printed.
1280 bool PrintAllTypes = false;
1281 const Type *TheType = Operand->getType();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001282
Chris Lattner52bd5cb92004-03-12 05:53:14 +00001283 // Shift Left & Right print both types even for Ubyte LHS, and select prints
1284 // types even if all operands are bools.
Chris Lattnerbbe0a422006-04-08 01:18:18 +00001285 if (isa<ShiftInst>(I) || isa<SelectInst>(I) || isa<StoreInst>(I) ||
1286 isa<ShuffleVectorInst>(I)) {
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001287 PrintAllTypes = true;
1288 } else {
1289 for (unsigned i = 1, E = I.getNumOperands(); i != E; ++i) {
1290 Operand = I.getOperand(i);
1291 if (Operand->getType() != TheType) {
1292 PrintAllTypes = true; // We have differing types! Print them all!
1293 break;
1294 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001295 }
1296 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001297
Chris Lattner7bfee412001-10-29 16:05:51 +00001298 if (!PrintAllTypes) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001299 Out << ' ';
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001300 printType(TheType);
Chris Lattner7bfee412001-10-29 16:05:51 +00001301 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001302
Chris Lattner113f4f42002-06-25 16:13:24 +00001303 for (unsigned i = 0, E = I.getNumOperands(); i != E; ++i) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001304 if (i) Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001305 writeOperand(I.getOperand(i), PrintAllTypes);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001306 }
1307 }
1308
Chris Lattner862e3382001-10-13 06:42:36 +00001309 printInfoComment(I);
Misha Brukmana6619a92004-06-21 21:53:56 +00001310 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001311}
1312
1313
1314//===----------------------------------------------------------------------===//
1315// External Interface declarations
1316//===----------------------------------------------------------------------===//
1317
Chris Lattner8339f7d2003-10-30 23:41:03 +00001318void Module::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001319 SlotMachine SlotTable(this);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001320 AssemblyWriter W(o, SlotTable, this, AAW);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001321 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001322}
1323
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001324void GlobalVariable::print(std::ostream &o) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001325 SlotMachine SlotTable(getParent());
Chris Lattner8339f7d2003-10-30 23:41:03 +00001326 AssemblyWriter W(o, SlotTable, getParent(), 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001327 W.write(this);
Chris Lattneradfe0d12001-09-10 20:08:19 +00001328}
1329
Chris Lattner8339f7d2003-10-30 23:41:03 +00001330void Function::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001331 SlotMachine SlotTable(getParent());
Chris Lattner8339f7d2003-10-30 23:41:03 +00001332 AssemblyWriter W(o, SlotTable, getParent(), AAW);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001333
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001334 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001335}
1336
Chris Lattnereef2fe72006-01-24 04:13:11 +00001337void InlineAsm::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001338 WriteAsOperand(o, this, true, 0);
Chris Lattnereef2fe72006-01-24 04:13:11 +00001339}
1340
Chris Lattner8339f7d2003-10-30 23:41:03 +00001341void BasicBlock::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001342 SlotMachine SlotTable(getParent());
Misha Brukmanb1c93172005-04-21 23:48:37 +00001343 AssemblyWriter W(o, SlotTable,
Chris Lattner8339f7d2003-10-30 23:41:03 +00001344 getParent() ? getParent()->getParent() : 0, AAW);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001345 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001346}
1347
Chris Lattner8339f7d2003-10-30 23:41:03 +00001348void Instruction::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001349 const Function *F = getParent() ? getParent()->getParent() : 0;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001350 SlotMachine SlotTable(F);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001351 AssemblyWriter W(o, SlotTable, F ? F->getParent() : 0, AAW);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001352
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001353 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001354}
Chris Lattner7db79582001-11-07 04:21:57 +00001355
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001356void Constant::print(std::ostream &o) const {
1357 if (this == 0) { o << "<null> constant value\n"; return; }
Chris Lattner7d734802002-09-10 15:53:49 +00001358
Misha Brukman21bbdb92004-06-04 21:11:51 +00001359 o << ' ' << getType()->getDescription() << ' ';
Evan Cheng5b19a802006-03-01 22:17:00 +00001360
1361 std::map<const Type *, std::string> TypeTable;
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001362 WriteConstantInt(o, this, TypeTable, 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001363}
1364
Misha Brukmanb1c93172005-04-21 23:48:37 +00001365void Type::print(std::ostream &o) const {
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001366 if (this == 0)
1367 o << "<null Type>";
1368 else
1369 o << getDescription();
1370}
1371
Chris Lattner2e9fa6d2002-04-09 19:48:49 +00001372void Argument::print(std::ostream &o) const {
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001373 WriteAsOperand(o, this, true, getParent() ? getParent()->getParent() : 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001374}
1375
Reid Spencer52641832004-05-25 18:14:38 +00001376// Value::dump - allow easy printing of Values from the debugger.
1377// Located here because so much of the needed functionality is here.
Bill Wendling22e978a2006-12-07 20:04:42 +00001378void Value::dump() const { print(*cerr.stream()); cerr << '\n'; }
Reid Spencer52641832004-05-25 18:14:38 +00001379
1380// Type::dump - allow easy printing of Values from the debugger.
1381// Located here because so much of the needed functionality is here.
Bill Wendling22e978a2006-12-07 20:04:42 +00001382void Type::dump() const { print(*cerr.stream()); cerr << '\n'; }
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001383
1384//===----------------------------------------------------------------------===//
Chris Lattnerfc9f1c92006-12-06 06:40:49 +00001385// SlotMachine Implementation
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001386//===----------------------------------------------------------------------===//
1387
1388#if 0
Bill Wendlingf3baad32006-12-07 01:30:32 +00001389#define SC_DEBUG(X) cerr << X
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001390#else
1391#define SC_DEBUG(X)
1392#endif
1393
1394// Module level constructor. Causes the contents of the Module (sans functions)
1395// to be added to the slot table.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001396SlotMachine::SlotMachine(const Module *M)
Reid Spencer56010e42004-05-26 21:56:09 +00001397 : TheModule(M) ///< Saved for lazy initialization.
1398 , TheFunction(0)
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001399 , FunctionProcessed(false)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001400{
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001401}
1402
1403// Function level constructor. Causes the contents of the Module and the one
1404// function provided to be added to the slot table.
Chris Lattner23e829a2006-12-06 05:12:21 +00001405SlotMachine::SlotMachine(const Function *F)
1406 : TheModule(F ? F->getParent() : 0) ///< Saved for lazy initialization
Reid Spencer56010e42004-05-26 21:56:09 +00001407 , TheFunction(F) ///< Saved for lazy initialization
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001408 , FunctionProcessed(false)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001409{
Reid Spencer56010e42004-05-26 21:56:09 +00001410}
1411
1412inline void SlotMachine::initialize(void) {
Chris Lattner23e829a2006-12-06 05:12:21 +00001413 if (TheModule) {
Misha Brukmanb1c93172005-04-21 23:48:37 +00001414 processModule();
Reid Spencer56010e42004-05-26 21:56:09 +00001415 TheModule = 0; ///< Prevent re-processing next time we're called.
1416 }
Chris Lattner193de902006-12-06 05:27:40 +00001417 if (TheFunction && !FunctionProcessed)
Misha Brukmanb1c93172005-04-21 23:48:37 +00001418 processFunction();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001419}
1420
1421// Iterate through all the global variables, functions, and global
Misha Brukmanb1c93172005-04-21 23:48:37 +00001422// variable initializers and create slots for them.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001423void SlotMachine::processModule() {
1424 SC_DEBUG("begin processModule!\n");
1425
Chris Lattner0dda8612006-12-06 06:15:43 +00001426 // Add all of the unnamed global variables to the value table.
Chris Lattner54932b02006-12-06 04:41:52 +00001427 for (Module::const_global_iterator I = TheModule->global_begin(),
1428 E = TheModule->global_end(); I != E; ++I)
Chris Lattner0dda8612006-12-06 06:15:43 +00001429 if (!I->hasName())
1430 getOrCreateSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001431
Chris Lattner0dda8612006-12-06 06:15:43 +00001432 // Add all the unnamed functions to the table.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001433 for (Module::const_iterator I = TheModule->begin(), E = TheModule->end();
1434 I != E; ++I)
Chris Lattner0dda8612006-12-06 06:15:43 +00001435 if (!I->hasName())
1436 getOrCreateSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001437
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001438 SC_DEBUG("end processModule!\n");
1439}
1440
1441
Reid Spencer56010e42004-05-26 21:56:09 +00001442// Process the arguments, basic blocks, and instructions of a function.
1443void SlotMachine::processFunction() {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001444 SC_DEBUG("begin processFunction!\n");
1445
Chris Lattner0dda8612006-12-06 06:15:43 +00001446 // Add all the function arguments with no names.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001447 for(Function::const_arg_iterator AI = TheFunction->arg_begin(),
Chris Lattner531f9e92005-03-15 04:54:21 +00001448 AE = TheFunction->arg_end(); AI != AE; ++AI)
Chris Lattner0dda8612006-12-06 06:15:43 +00001449 if (!AI->hasName())
1450 getOrCreateSlot(AI);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001451
1452 SC_DEBUG("Inserting Instructions:\n");
1453
Chris Lattner0dda8612006-12-06 06:15:43 +00001454 // Add all of the basic blocks and instructions with no names.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001455 for (Function::const_iterator BB = TheFunction->begin(),
Reid Spencer56010e42004-05-26 21:56:09 +00001456 E = TheFunction->end(); BB != E; ++BB) {
Chris Lattner0dda8612006-12-06 06:15:43 +00001457 if (!BB->hasName())
1458 getOrCreateSlot(BB);
1459 for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I != E; ++I)
1460 if (I->getType() != Type::VoidTy && !I->hasName())
1461 getOrCreateSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001462 }
1463
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001464 FunctionProcessed = true;
1465
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001466 SC_DEBUG("end processFunction!\n");
1467}
1468
Chris Lattner193de902006-12-06 05:27:40 +00001469/// Clean up after incorporating a function. This is the only way to get out of
1470/// the function incorporation state that affects the
1471/// getSlot/getOrCreateSlot lock. Function incorporation state is indicated
1472/// by TheFunction != 0.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001473void SlotMachine::purgeFunction() {
1474 SC_DEBUG("begin purgeFunction!\n");
1475 fMap.clear(); // Simply discard the function level map
Reid Spencer56010e42004-05-26 21:56:09 +00001476 TheFunction = 0;
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001477 FunctionProcessed = false;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001478 SC_DEBUG("end purgeFunction!\n");
1479}
1480
1481/// Get the slot number for a value. This function will assert if you
Chris Lattner193de902006-12-06 05:27:40 +00001482/// ask for a Value that hasn't previously been inserted with getOrCreateSlot.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001483/// Types are forbidden because Type does not inherit from Value (any more).
Chris Lattner757ee0b2004-06-09 19:41:19 +00001484int SlotMachine::getSlot(const Value *V) {
Chris Lattner23e829a2006-12-06 05:12:21 +00001485 assert(V && "Can't get slot for null Value");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001486 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1487 "Can't insert a non-GlobalValue Constant into SlotMachine");
Reid Spencer56010e42004-05-26 21:56:09 +00001488
1489 // Check for uninitialized state and do lazy initialization
1490 this->initialize();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001491
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001492 // Get the type of the value
1493 const Type* VTy = V->getType();
1494
1495 // Find the type plane in the module map
1496 TypedPlanes::const_iterator MI = mMap.find(VTy);
1497
Chris Lattner23e829a2006-12-06 05:12:21 +00001498 if (TheFunction) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001499 // Lookup the type in the function map too
1500 TypedPlanes::const_iterator FI = fMap.find(VTy);
1501 // If there is a corresponding type plane in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001502 if (FI != fMap.end()) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001503 // Lookup the Value in the function map
1504 ValueMap::const_iterator FVI = FI->second.map.find(V);
1505 // If the value doesn't exist in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001506 if (FVI == FI->second.map.end()) {
Chris Lattner68a038e2004-06-09 22:22:10 +00001507 // Look up the value in the module map.
1508 if (MI == mMap.end()) return -1;
Reid Spencer56010e42004-05-26 21:56:09 +00001509 ValueMap::const_iterator MVI = MI->second.map.find(V);
1510 // If we didn't find it, it wasn't inserted
Chris Lattner757ee0b2004-06-09 19:41:19 +00001511 if (MVI == MI->second.map.end()) return -1;
Chris Lattner23e829a2006-12-06 05:12:21 +00001512 assert(MVI != MI->second.map.end() && "Value not found");
Reid Spencer56010e42004-05-26 21:56:09 +00001513 // We found it only at the module level
Misha Brukmanb1c93172005-04-21 23:48:37 +00001514 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001515
1516 // else the value exists in the function map
1517 } else {
Reid Spencer56010e42004-05-26 21:56:09 +00001518 // Return the slot number as the module's contribution to
1519 // the type plane plus the index in the function's contribution
1520 // to the type plane.
Chris Lattnerb1f04782004-06-15 21:07:32 +00001521 if (MI != mMap.end())
1522 return MI->second.next_slot + FVI->second;
1523 else
1524 return FVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001525 }
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001526 }
1527 }
1528
Reid Spencer8beac692004-06-09 15:26:53 +00001529 // N.B. Can get here only if either !TheFunction or the function doesn't
1530 // have a corresponding type plane for the Value
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001531
1532 // Make sure the type plane exists
Chris Lattner757ee0b2004-06-09 19:41:19 +00001533 if (MI == mMap.end()) return -1;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001534 // Lookup the value in the module's map
1535 ValueMap::const_iterator MVI = MI->second.map.find(V);
1536 // Make sure we found it.
Chris Lattner757ee0b2004-06-09 19:41:19 +00001537 if (MVI == MI->second.map.end()) return -1;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001538 // Return it.
1539 return MVI->second;
1540}
1541
Reid Spencer58d30f22004-07-04 11:50:43 +00001542
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001543// Create a new slot, or return the existing slot if it is already
1544// inserted. Note that the logic here parallels getSlot but instead
1545// of asserting when the Value* isn't found, it inserts the value.
Chris Lattner193de902006-12-06 05:27:40 +00001546unsigned SlotMachine::getOrCreateSlot(const Value *V) {
Chris Lattner0dda8612006-12-06 06:15:43 +00001547 const Type* VTy = V->getType();
1548 assert(VTy != Type::VoidTy && !V->hasName() && "Doesn't need a slot!");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001549 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1550 "Can't insert a non-GlobalValue Constant into SlotMachine");
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001551
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001552 // Look up the type plane for the Value's type from the module map
1553 TypedPlanes::const_iterator MI = mMap.find(VTy);
1554
Chris Lattner23e829a2006-12-06 05:12:21 +00001555 if (TheFunction) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001556 // Get the type plane for the Value's type from the function map
1557 TypedPlanes::const_iterator FI = fMap.find(VTy);
1558 // If there is a corresponding type plane in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001559 if (FI != fMap.end()) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001560 // Lookup the Value in the function map
1561 ValueMap::const_iterator FVI = FI->second.map.find(V);
1562 // If the value doesn't exist in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001563 if (FVI == FI->second.map.end()) {
Reid Spencer56010e42004-05-26 21:56:09 +00001564 // If there is no corresponding type plane in the module map
Chris Lattner23e829a2006-12-06 05:12:21 +00001565 if (MI == mMap.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001566 return insertValue(V);
1567 // Look up the value in the module map
1568 ValueMap::const_iterator MVI = MI->second.map.find(V);
1569 // If we didn't find it, it wasn't inserted
Chris Lattner23e829a2006-12-06 05:12:21 +00001570 if (MVI == MI->second.map.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001571 return insertValue(V);
1572 else
1573 // We found it only at the module level
1574 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001575
1576 // else the value exists in the function map
1577 } else {
Chris Lattner23e829a2006-12-06 05:12:21 +00001578 if (MI == mMap.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001579 return FVI->second;
1580 else
1581 // Return the slot number as the module's contribution to
1582 // the type plane plus the index in the function's contribution
1583 // to the type plane.
1584 return MI->second.next_slot + FVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001585 }
1586
1587 // else there is not a corresponding type plane in the function map
1588 } else {
1589 // If the type plane doesn't exists at the module level
Chris Lattner23e829a2006-12-06 05:12:21 +00001590 if (MI == mMap.end()) {
Reid Spencer56010e42004-05-26 21:56:09 +00001591 return insertValue(V);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001592 // else type plane exists at the module level, examine it
1593 } else {
Reid Spencer56010e42004-05-26 21:56:09 +00001594 // Look up the value in the module's map
1595 ValueMap::const_iterator MVI = MI->second.map.find(V);
1596 // If we didn't find it there either
Chris Lattner23e829a2006-12-06 05:12:21 +00001597 if (MVI == MI->second.map.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001598 // Return the slot number as the module's contribution to
1599 // the type plane plus the index of the function map insertion.
1600 return MI->second.next_slot + insertValue(V);
1601 else
1602 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001603 }
1604 }
1605 }
1606
Chris Lattner0dda8612006-12-06 06:15:43 +00001607 // N.B. Can only get here if TheFunction == 0
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001608
1609 // If the module map's type plane is not for the Value's type
Chris Lattner23e829a2006-12-06 05:12:21 +00001610 if (MI != mMap.end()) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001611 // Lookup the value in the module's map
1612 ValueMap::const_iterator MVI = MI->second.map.find(V);
Chris Lattner23e829a2006-12-06 05:12:21 +00001613 if (MVI != MI->second.map.end())
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001614 return MVI->second;
1615 }
1616
1617 return insertValue(V);
1618}
1619
1620
1621// Low level insert function. Minimal checking is done. This
Chris Lattner193de902006-12-06 05:27:40 +00001622// function is just for the convenience of getOrCreateSlot (above).
Chris Lattner23e829a2006-12-06 05:12:21 +00001623unsigned SlotMachine::insertValue(const Value *V) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001624 assert(V && "Can't insert a null Value into SlotMachine!");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001625 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
Chris Lattner7e4e7662006-12-06 05:42:32 +00001626 "Can't insert a non-GlobalValue Constant into SlotMachine");
1627 assert(V->getType() != Type::VoidTy && !V->hasName());
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001628
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001629 const Type *VTy = V->getType();
1630 unsigned DestSlot = 0;
1631
Chris Lattner23e829a2006-12-06 05:12:21 +00001632 if (TheFunction) {
1633 TypedPlanes::iterator I = fMap.find(VTy);
1634 if (I == fMap.end())
Reid Spencer58d30f22004-07-04 11:50:43 +00001635 I = fMap.insert(std::make_pair(VTy,ValuePlane())).first;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001636 DestSlot = I->second.map[V] = I->second.next_slot++;
1637 } else {
Chris Lattner23e829a2006-12-06 05:12:21 +00001638 TypedPlanes::iterator I = mMap.find(VTy);
1639 if (I == mMap.end())
Reid Spencer58d30f22004-07-04 11:50:43 +00001640 I = mMap.insert(std::make_pair(VTy,ValuePlane())).first;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001641 DestSlot = I->second.map[V] = I->second.next_slot++;
1642 }
1643
Misha Brukmanb1c93172005-04-21 23:48:37 +00001644 SC_DEBUG(" Inserting value [" << VTy << "] = " << V << " slot=" <<
Reid Spencer56010e42004-05-26 21:56:09 +00001645 DestSlot << " [");
Chris Lattnerf8090cb2006-12-06 05:55:41 +00001646 // G = Global, F = Function, o = other
1647 SC_DEBUG((isa<GlobalVariable>(V) ? 'G' : (isa<Function>(V) ? 'F' : 'o')));
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001648 SC_DEBUG("]\n");
1649 return DestSlot;
1650}