blob: eb2f3fdf57ae68c2b7b3e51815d1eedd2d3b1308 [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 Spencer7c16caa2004-09-01 22:55:40 +000028#include "llvm/ADT/StringExtras.h"
29#include "llvm/ADT/STLExtras.h"
Bill Wendlingdfc91892006-11-28 02:09:03 +000030#include "llvm/Support/CFG.h"
Jim Laskeyb74c6662005-08-17 19:34:49 +000031#include "llvm/Support/MathExtras.h"
Bill Wendlingdfc91892006-11-28 02:09:03 +000032#include "llvm/Support/Streams.h"
Chris Lattnerfee714f2001-09-07 16:36:04 +000033#include <algorithm>
Chris Lattner189d19f2003-11-21 20:23:48 +000034using namespace llvm;
Brian Gaeke960707c2003-11-11 22:41:34 +000035
Chris Lattner60a7dd12004-07-15 02:51:31 +000036namespace llvm {
Reid Spencer16f2f7f2004-05-26 07:18:52 +000037
Reid Spencer294715b2005-05-15 16:13:11 +000038// Make virtual table appear in this compilation unit.
39AssemblyAnnotationWriter::~AssemblyAnnotationWriter() {}
40
Reid Spencer16f2f7f2004-05-26 07:18:52 +000041/// This class provides computation of slot numbers for LLVM Assembly writing.
42/// @brief LLVM Assembly Writing Slot Computation.
43class SlotMachine {
44
45/// @name Types
46/// @{
47public:
48
49 /// @brief A mapping of Values to slot numbers
50 typedef std::map<const Value*, unsigned> ValueMap;
51
52 /// @brief A plane with next slot number and ValueMap
Misha Brukmanb1c93172005-04-21 23:48:37 +000053 struct ValuePlane {
Reid Spencer16f2f7f2004-05-26 07:18:52 +000054 unsigned next_slot; ///< The next slot number to use
55 ValueMap map; ///< The map of Value* -> unsigned
Reid Spencer58d30f22004-07-04 11:50:43 +000056 ValuePlane() { next_slot = 0; } ///< Make sure we start at 0
57 };
58
Reid Spencer16f2f7f2004-05-26 07:18:52 +000059 /// @brief The map of planes by Type
Reid Spencer58d30f22004-07-04 11:50:43 +000060 typedef std::map<const Type*, ValuePlane> TypedPlanes;
Reid Spencer16f2f7f2004-05-26 07:18:52 +000061
62/// @}
63/// @name Constructors
64/// @{
65public:
66 /// @brief Construct from a module
Chris Lattner23e829a2006-12-06 05:12:21 +000067 SlotMachine(const Module *M);
Reid Spencer16f2f7f2004-05-26 07:18:52 +000068
69 /// @brief Construct from a function, starting out in incorp state.
Chris Lattner23e829a2006-12-06 05:12:21 +000070 SlotMachine(const Function *F);
Reid Spencer16f2f7f2004-05-26 07:18:52 +000071
72/// @}
73/// @name Accessors
74/// @{
75public:
76 /// Return the slot number of the specified value in it's type
77 /// plane. Its an error to ask for something not in the SlotMachine.
78 /// Its an error to ask for a Type*
Chris Lattner757ee0b2004-06-09 19:41:19 +000079 int getSlot(const Value *V);
Reid Spencer8beac692004-06-09 15:26:53 +000080
Reid Spencer16f2f7f2004-05-26 07:18:52 +000081/// @}
82/// @name Mutators
83/// @{
84public:
Misha Brukmanb1c93172005-04-21 23:48:37 +000085 /// If you'd like to deal with a function instead of just a module, use
Reid Spencer16f2f7f2004-05-26 07:18:52 +000086 /// this method to get its data into the SlotMachine.
Misha Brukmanb1c93172005-04-21 23:48:37 +000087 void incorporateFunction(const Function *F) {
88 TheFunction = F;
Reid Spencerb0ac8c42004-08-16 07:46:33 +000089 FunctionProcessed = false;
90 }
Reid Spencer16f2f7f2004-05-26 07:18:52 +000091
Misha Brukmanb1c93172005-04-21 23:48:37 +000092 /// After calling incorporateFunction, use this method to remove the
93 /// most recently incorporated function from the SlotMachine. This
Reid Spencer16f2f7f2004-05-26 07:18:52 +000094 /// will reset the state of the machine back to just the module contents.
95 void purgeFunction();
96
97/// @}
98/// @name Implementation Details
99/// @{
100private:
Reid Spencer56010e42004-05-26 21:56:09 +0000101 /// This function does the actual initialization.
102 inline void initialize();
103
Misha Brukmanb1c93172005-04-21 23:48:37 +0000104 /// Values can be crammed into here at will. If they haven't
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000105 /// been inserted already, they get inserted, otherwise they are ignored.
106 /// Either way, the slot number for the Value* is returned.
Chris Lattner193de902006-12-06 05:27:40 +0000107 unsigned getOrCreateSlot(const Value *V);
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000108
109 /// Insert a value into the value table. Return the slot number
110 /// that it now occupies. BadThings(TM) will happen if you insert a
Misha Brukmanb1c93172005-04-21 23:48:37 +0000111 /// Value that's already been inserted.
Chris Lattner23e829a2006-12-06 05:12:21 +0000112 unsigned insertValue(const Value *V);
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000113
114 /// Add all of the module level global variables (and their initializers)
115 /// and function declarations, but not the contents of those functions.
116 void processModule();
117
Reid Spencer56010e42004-05-26 21:56:09 +0000118 /// Add all of the functions arguments, basic blocks, and instructions
119 void processFunction();
120
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000121 SlotMachine(const SlotMachine &); // DO NOT IMPLEMENT
122 void operator=(const SlotMachine &); // DO NOT IMPLEMENT
123
124/// @}
125/// @name Data
126/// @{
127public:
128
129 /// @brief The module for which we are holding slot numbers
Reid Spencer56010e42004-05-26 21:56:09 +0000130 const Module* TheModule;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000131
Reid Spencer56010e42004-05-26 21:56:09 +0000132 /// @brief The function for which we are holding slot numbers
133 const Function* TheFunction;
Reid Spencerb0ac8c42004-08-16 07:46:33 +0000134 bool FunctionProcessed;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000135
136 /// @brief The TypePlanes map for the module level data
137 TypedPlanes mMap;
138
139 /// @brief The TypePlanes map for the function level data
140 TypedPlanes fMap;
141
142/// @}
143
144};
145
Chris Lattner60a7dd12004-07-15 02:51:31 +0000146} // end namespace llvm
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000147
Chris Lattnerc2d3d312006-08-27 22:42:52 +0000148static RegisterPass<PrintModulePass>
Chris Lattnere7134c52006-08-21 17:20:01 +0000149X("printm", "Print module to stderr");
Chris Lattnerc2d3d312006-08-27 22:42:52 +0000150static RegisterPass<PrintFunctionPass>
Chris Lattnere7134c52006-08-21 17:20:01 +0000151Y("print","Print function to stderr");
Chris Lattner7f8845a2002-07-23 18:07:49 +0000152
Misha Brukmanb1c93172005-04-21 23:48:37 +0000153static void WriteAsOperandInternal(std::ostream &Out, const Value *V,
Chris Lattnera9f0a112006-12-06 05:50:41 +0000154 std::map<const Type *, std::string> &TypeTable,
Reid Spencer58d30f22004-07-04 11:50:43 +0000155 SlotMachine *Machine);
156
Chris Lattnerb86620e2001-10-29 16:37:48 +0000157static const Module *getModuleFromVal(const Value *V) {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000158 if (const Argument *MA = dyn_cast<Argument>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000159 return MA->getParent() ? MA->getParent()->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000160 else if (const BasicBlock *BB = dyn_cast<BasicBlock>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000161 return BB->getParent() ? BB->getParent()->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000162 else if (const Instruction *I = dyn_cast<Instruction>(V)) {
Chris Lattner57698e22002-03-26 18:01:55 +0000163 const Function *M = I->getParent() ? I->getParent()->getParent() : 0;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000164 return M ? M->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000165 } else if (const GlobalValue *GV = dyn_cast<GlobalValue>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000166 return GV->getParent();
Chris Lattnerb86620e2001-10-29 16:37:48 +0000167 return 0;
168}
169
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000170static SlotMachine *createSlotMachine(const Value *V) {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000171 if (const Argument *FA = dyn_cast<Argument>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000172 return new SlotMachine(FA->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000173 } else if (const Instruction *I = dyn_cast<Instruction>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000174 return new SlotMachine(I->getParent()->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000175 } else if (const BasicBlock *BB = dyn_cast<BasicBlock>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000176 return new SlotMachine(BB->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000177 } else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(V)){
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000178 return new SlotMachine(GV->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000179 } else if (const Function *Func = dyn_cast<Function>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000180 return new SlotMachine(Func);
Chris Lattner7bfee412001-10-29 16:05:51 +0000181 }
182 return 0;
183}
Chris Lattner2f7c9632001-06-06 20:29:01 +0000184
Chris Lattner1c343042003-08-22 05:40:38 +0000185// getLLVMName - Turn the specified string into an 'LLVM name', which is either
186// prefixed with % (if the string only contains simple characters) or is
187// surrounded with ""'s (if it has special chars in it).
Alkis Evlogimenos991d6ad2004-12-10 05:41:10 +0000188static std::string getLLVMName(const std::string &Name,
189 bool prefixName = true) {
Chris Lattner1c343042003-08-22 05:40:38 +0000190 assert(!Name.empty() && "Cannot get empty name!");
191
192 // First character cannot start with a number...
193 if (Name[0] >= '0' && Name[0] <= '9')
194 return "\"" + Name + "\"";
195
196 // Scan to see if we have any characters that are not on the "white list"
197 for (unsigned i = 0, e = Name.size(); i != e; ++i) {
198 char C = Name[i];
199 assert(C != '"' && "Illegal character in LLVM value name!");
200 if ((C < 'a' || C > 'z') && (C < 'A' || C > 'Z') && (C < '0' || C > '9') &&
201 C != '-' && C != '.' && C != '_')
202 return "\"" + Name + "\"";
203 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000204
Chris Lattner1c343042003-08-22 05:40:38 +0000205 // If we get here, then the identifier is legal to use as a "VarID".
Alkis Evlogimenos991d6ad2004-12-10 05:41:10 +0000206 if (prefixName)
207 return "%"+Name;
208 else
209 return Name;
Chris Lattner1c343042003-08-22 05:40:38 +0000210}
211
Chris Lattnerb86620e2001-10-29 16:37:48 +0000212
Misha Brukmanc566ca362004-03-02 00:22:19 +0000213/// fillTypeNameTable - If the module has a symbol table, take all global types
214/// and stuff their names into the TypeNames map.
215///
Chris Lattnerb86620e2001-10-29 16:37:48 +0000216static void fillTypeNameTable(const Module *M,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000217 std::map<const Type *, std::string> &TypeNames) {
Chris Lattner98cf1f52002-11-20 18:36:02 +0000218 if (!M) return;
219 const SymbolTable &ST = M->getSymbolTable();
Reid Spencere7e96712004-05-25 08:53:40 +0000220 SymbolTable::type_const_iterator TI = ST.type_begin();
Chris Lattner23e829a2006-12-06 05:12:21 +0000221 for (; TI != ST.type_end(); ++TI) {
Reid Spencere7e96712004-05-25 08:53:40 +0000222 // As a heuristic, don't insert pointer to primitive types, because
223 // they are used too often to have a single useful name.
224 //
225 const Type *Ty = cast<Type>(TI->second);
226 if (!isa<PointerType>(Ty) ||
Reid Spencer56010e42004-05-26 21:56:09 +0000227 !cast<PointerType>(Ty)->getElementType()->isPrimitiveType() ||
228 isa<OpaqueType>(cast<PointerType>(Ty)->getElementType()))
Reid Spencere7e96712004-05-25 08:53:40 +0000229 TypeNames.insert(std::make_pair(Ty, getLLVMName(TI->first)));
Chris Lattnerb86620e2001-10-29 16:37:48 +0000230 }
231}
232
233
234
Misha Brukmanb1c93172005-04-21 23:48:37 +0000235static void calcTypeName(const Type *Ty,
John Criswellcd116ba2004-06-01 14:54:08 +0000236 std::vector<const Type *> &TypeStack,
237 std::map<const Type *, std::string> &TypeNames,
238 std::string & Result){
239 if (Ty->isPrimitiveType() && !isa<OpaqueType>(Ty)) {
240 Result += Ty->getDescription(); // Base case
241 return;
242 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000243
244 // Check to see if the type is named.
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000245 std::map<const Type *, std::string>::iterator I = TypeNames.find(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000246 if (I != TypeNames.end()) {
247 Result += I->second;
248 return;
249 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000250
John Criswellcd116ba2004-06-01 14:54:08 +0000251 if (isa<OpaqueType>(Ty)) {
252 Result += "opaque";
253 return;
254 }
Chris Lattnerf14ead92003-10-30 00:22:33 +0000255
Chris Lattnerb86620e2001-10-29 16:37:48 +0000256 // Check to see if the Type is already on the stack...
257 unsigned Slot = 0, CurSize = TypeStack.size();
258 while (Slot < CurSize && TypeStack[Slot] != Ty) ++Slot; // Scan for type
259
Misha Brukmanb1c93172005-04-21 23:48:37 +0000260 // This is another base case for the recursion. In this case, we know
Chris Lattnerb86620e2001-10-29 16:37:48 +0000261 // that we have looped back to a type that we have previously visited.
262 // Generate the appropriate upreference to handle this.
John Criswellcd116ba2004-06-01 14:54:08 +0000263 if (Slot < CurSize) {
264 Result += "\\" + utostr(CurSize-Slot); // Here's the upreference
265 return;
266 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000267
268 TypeStack.push_back(Ty); // Recursive case: Add us to the stack..
Misha Brukmanb1c93172005-04-21 23:48:37 +0000269
Chris Lattner6b727592004-06-17 18:19:28 +0000270 switch (Ty->getTypeID()) {
Chris Lattner91db5822002-03-29 03:44:36 +0000271 case Type::FunctionTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000272 const FunctionType *FTy = cast<FunctionType>(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000273 calcTypeName(FTy->getReturnType(), TypeStack, TypeNames, Result);
274 Result += " (";
Chris Lattnerfa829be2004-02-09 04:14:01 +0000275 for (FunctionType::param_iterator I = FTy->param_begin(),
276 E = FTy->param_end(); I != E; ++I) {
277 if (I != FTy->param_begin())
Chris Lattnerb86620e2001-10-29 16:37:48 +0000278 Result += ", ";
John Criswellcd116ba2004-06-01 14:54:08 +0000279 calcTypeName(*I, TypeStack, TypeNames, Result);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000280 }
Chris Lattnerd816b532002-04-13 20:53:41 +0000281 if (FTy->isVarArg()) {
Chris Lattnerfa829be2004-02-09 04:14:01 +0000282 if (FTy->getNumParams()) Result += ", ";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000283 Result += "...";
284 }
285 Result += ")";
286 break;
287 }
288 case Type::StructTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000289 const StructType *STy = cast<StructType>(Ty);
Andrew Lenharthdcb3c972006-12-08 18:06:16 +0000290 if (STy->isPacked())
291 Result += '<';
John Criswellcd116ba2004-06-01 14:54:08 +0000292 Result += "{ ";
Chris Lattnerac6db752004-02-09 04:37:31 +0000293 for (StructType::element_iterator I = STy->element_begin(),
294 E = STy->element_end(); I != E; ++I) {
295 if (I != STy->element_begin())
Chris Lattnerb86620e2001-10-29 16:37:48 +0000296 Result += ", ";
John Criswellcd116ba2004-06-01 14:54:08 +0000297 calcTypeName(*I, TypeStack, TypeNames, Result);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000298 }
299 Result += " }";
Andrew Lenharthdcb3c972006-12-08 18:06:16 +0000300 if (STy->isPacked())
301 Result += '>';
Chris Lattnerb86620e2001-10-29 16:37:48 +0000302 break;
303 }
304 case Type::PointerTyID:
Misha Brukmanb1c93172005-04-21 23:48:37 +0000305 calcTypeName(cast<PointerType>(Ty)->getElementType(),
John Criswellcd116ba2004-06-01 14:54:08 +0000306 TypeStack, TypeNames, Result);
307 Result += "*";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000308 break;
309 case Type::ArrayTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000310 const ArrayType *ATy = cast<ArrayType>(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000311 Result += "[" + utostr(ATy->getNumElements()) + " x ";
312 calcTypeName(ATy->getElementType(), TypeStack, TypeNames, Result);
313 Result += "]";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000314 break;
315 }
Brian Gaeke02209042004-08-20 06:00:58 +0000316 case Type::PackedTyID: {
317 const PackedType *PTy = cast<PackedType>(Ty);
318 Result += "<" + utostr(PTy->getNumElements()) + " x ";
319 calcTypeName(PTy->getElementType(), TypeStack, TypeNames, Result);
320 Result += ">";
321 break;
322 }
Chris Lattner15285ab2003-05-14 17:50:47 +0000323 case Type::OpaqueTyID:
John Criswellcd116ba2004-06-01 14:54:08 +0000324 Result += "opaque";
Chris Lattner15285ab2003-05-14 17:50:47 +0000325 break;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000326 default:
John Criswellcd116ba2004-06-01 14:54:08 +0000327 Result += "<unrecognized-type>";
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000328 break;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000329 }
330
331 TypeStack.pop_back(); // Remove self from stack...
Chris Lattnerb86620e2001-10-29 16:37:48 +0000332}
333
334
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000335/// printTypeInt - The internal guts of printing out a type that has a
336/// potentially named portion.
337///
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000338static std::ostream &printTypeInt(std::ostream &Out, const Type *Ty,
339 std::map<const Type *, std::string> &TypeNames) {
Chris Lattnerb86620e2001-10-29 16:37:48 +0000340 // Primitive types always print out their description, regardless of whether
341 // they have been named or not.
342 //
Chris Lattner92d60532003-10-30 00:12:51 +0000343 if (Ty->isPrimitiveType() && !isa<OpaqueType>(Ty))
344 return Out << Ty->getDescription();
Chris Lattnerb86620e2001-10-29 16:37:48 +0000345
346 // Check to see if the type is named.
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000347 std::map<const Type *, std::string>::iterator I = TypeNames.find(Ty);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000348 if (I != TypeNames.end()) return Out << I->second;
349
350 // Otherwise we have a type that has not been named but is a derived type.
351 // Carefully recurse the type hierarchy to print out any contained symbolic
352 // names.
353 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000354 std::vector<const Type *> TypeStack;
John Criswellcd116ba2004-06-01 14:54:08 +0000355 std::string TypeName;
356 calcTypeName(Ty, TypeStack, TypeNames, TypeName);
Chris Lattner7f74a562002-01-20 22:54:45 +0000357 TypeNames.insert(std::make_pair(Ty, TypeName));//Cache type name for later use
John Criswellcd116ba2004-06-01 14:54:08 +0000358 return (Out << TypeName);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000359}
360
Chris Lattner34b95182001-10-31 04:33:19 +0000361
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000362/// WriteTypeSymbolic - This attempts to write the specified type as a symbolic
363/// type, iff there is an entry in the modules symbol table for the specified
364/// type or one of it's component types. This is slower than a simple x << Type
365///
Chris Lattner189d19f2003-11-21 20:23:48 +0000366std::ostream &llvm::WriteTypeSymbolic(std::ostream &Out, const Type *Ty,
367 const Module *M) {
Misha Brukmanb1c93172005-04-21 23:48:37 +0000368 Out << ' ';
Chris Lattnerb86620e2001-10-29 16:37:48 +0000369
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000370 // If they want us to print out a type, but there is no context, we can't
371 // print it symbolically.
372 if (!M)
Chris Lattner72f866e2001-10-29 16:40:32 +0000373 return Out << Ty->getDescription();
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000374
375 std::map<const Type *, std::string> TypeNames;
376 fillTypeNameTable(M, TypeNames);
377 return printTypeInt(Out, Ty, TypeNames);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000378}
379
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000380// PrintEscapedString - Print each character of the specified string, escaping
381// it if it is not printable or if it is an escape char.
382static void PrintEscapedString(const std::string &Str, std::ostream &Out) {
383 for (unsigned i = 0, e = Str.size(); i != e; ++i) {
384 unsigned char C = Str[i];
385 if (isprint(C) && C != '"' && C != '\\') {
386 Out << C;
387 } else {
388 Out << '\\'
389 << (char) ((C/16 < 10) ? ( C/16 +'0') : ( C/16 -10+'A'))
390 << (char)(((C&15) < 10) ? ((C&15)+'0') : ((C&15)-10+'A'));
391 }
392 }
393}
394
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000395static const char *getPredicateText(unsigned predicate) {
Reid Spencer812a1be2006-12-04 05:19:18 +0000396 const char * pred = "unknown";
397 switch (predicate) {
398 case FCmpInst::FCMP_FALSE: pred = "false"; break;
399 case FCmpInst::FCMP_OEQ: pred = "oeq"; break;
400 case FCmpInst::FCMP_OGT: pred = "ogt"; break;
401 case FCmpInst::FCMP_OGE: pred = "oge"; break;
402 case FCmpInst::FCMP_OLT: pred = "olt"; break;
403 case FCmpInst::FCMP_OLE: pred = "ole"; break;
404 case FCmpInst::FCMP_ONE: pred = "one"; break;
405 case FCmpInst::FCMP_ORD: pred = "ord"; break;
406 case FCmpInst::FCMP_UNO: pred = "uno"; break;
407 case FCmpInst::FCMP_UEQ: pred = "ueq"; break;
408 case FCmpInst::FCMP_UGT: pred = "ugt"; break;
409 case FCmpInst::FCMP_UGE: pred = "uge"; break;
410 case FCmpInst::FCMP_ULT: pred = "ult"; break;
411 case FCmpInst::FCMP_ULE: pred = "ule"; break;
412 case FCmpInst::FCMP_UNE: pred = "une"; break;
413 case FCmpInst::FCMP_TRUE: pred = "true"; break;
414 case ICmpInst::ICMP_EQ: pred = "eq"; break;
415 case ICmpInst::ICMP_NE: pred = "ne"; break;
416 case ICmpInst::ICMP_SGT: pred = "sgt"; break;
417 case ICmpInst::ICMP_SGE: pred = "sge"; break;
418 case ICmpInst::ICMP_SLT: pred = "slt"; break;
419 case ICmpInst::ICMP_SLE: pred = "sle"; break;
420 case ICmpInst::ICMP_UGT: pred = "ugt"; break;
421 case ICmpInst::ICMP_UGE: pred = "uge"; break;
422 case ICmpInst::ICMP_ULT: pred = "ult"; break;
423 case ICmpInst::ICMP_ULE: pred = "ule"; break;
424 }
425 return pred;
426}
427
Misha Brukmanb1c93172005-04-21 23:48:37 +0000428/// @brief Internal constant writer.
429static void WriteConstantInt(std::ostream &Out, const Constant *CV,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000430 std::map<const Type *, std::string> &TypeTable,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000431 SlotMachine *Machine) {
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000432 const int IndentSize = 4;
433 static std::string Indent = "\n";
Chris Lattner1e194682002-04-18 18:53:13 +0000434 if (const ConstantBool *CB = dyn_cast<ConstantBool>(CV)) {
Chris Lattner94e39cb2006-09-28 22:50:29 +0000435 Out << (CB->getValue() ? "true" : "false");
Reid Spencere0fc4df2006-10-20 07:07:24 +0000436 } else if (const ConstantInt *CI = dyn_cast<ConstantInt>(CV)) {
Reid Spencer2acb8e12006-12-19 21:16:35 +0000437 Out << CI->getSExtValue();
Chris Lattner1e194682002-04-18 18:53:13 +0000438 } else if (const ConstantFP *CFP = dyn_cast<ConstantFP>(CV)) {
439 // We would like to output the FP constant value in exponential notation,
440 // but we cannot do this if doing so will lose precision. Check here to
441 // make sure that we only output it in exponential format if we can parse
442 // the value back and get the same value.
443 //
444 std::string StrVal = ftostr(CFP->getValue());
445
446 // Check to make sure that the stringized number is not some string like
447 // "Inf" or NaN, that atof will accept, but the lexer will not. Check that
448 // the string matches the "[-+]?[0-9]" regex.
449 //
450 if ((StrVal[0] >= '0' && StrVal[0] <= '9') ||
451 ((StrVal[0] == '-' || StrVal[0] == '+') &&
Brian Gaeke87b4f072003-06-17 23:55:35 +0000452 (StrVal[1] >= '0' && StrVal[1] <= '9')))
Chris Lattner1e194682002-04-18 18:53:13 +0000453 // Reparse stringized version!
454 if (atof(StrVal.c_str()) == CFP->getValue()) {
Chris Lattner472cc102005-01-04 01:56:57 +0000455 Out << StrVal;
456 return;
Chris Lattner1e194682002-04-18 18:53:13 +0000457 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000458
Chris Lattner1e194682002-04-18 18:53:13 +0000459 // Otherwise we could not reparse it to exactly the same value, so we must
460 // output the string in hexadecimal format!
Chris Lattner472cc102005-01-04 01:56:57 +0000461 assert(sizeof(double) == sizeof(uint64_t) &&
Chris Lattner1e194682002-04-18 18:53:13 +0000462 "assuming that double is 64 bits!");
Jim Laskeyb74c6662005-08-17 19:34:49 +0000463 Out << "0x" << utohexstr(DoubleToBits(CFP->getValue()));
Chris Lattner1e194682002-04-18 18:53:13 +0000464
Chris Lattner76b2ff42004-02-15 05:55:15 +0000465 } else if (isa<ConstantAggregateZero>(CV)) {
466 Out << "zeroinitializer";
Chris Lattner1e194682002-04-18 18:53:13 +0000467 } else if (const ConstantArray *CA = dyn_cast<ConstantArray>(CV)) {
468 // As a special case, print the array as a string if it is an array of
469 // ubytes or an array of sbytes with positive values.
Misha Brukmanb1c93172005-04-21 23:48:37 +0000470 //
Chris Lattner1e194682002-04-18 18:53:13 +0000471 const Type *ETy = CA->getType()->getElementType();
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000472 if (CA->isString()) {
Chris Lattner1e194682002-04-18 18:53:13 +0000473 Out << "c\"";
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000474 PrintEscapedString(CA->getAsString(), Out);
Chris Lattner1e194682002-04-18 18:53:13 +0000475 Out << "\"";
476
477 } else { // Cannot output in string format...
Misha Brukman21bbdb92004-06-04 21:11:51 +0000478 Out << '[';
Chris Lattnerd84bb632002-04-16 21:36:08 +0000479 if (CA->getNumOperands()) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000480 Out << ' ';
Chris Lattner1e194682002-04-18 18:53:13 +0000481 printTypeInt(Out, ETy, TypeTable);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000482 WriteAsOperandInternal(Out, CA->getOperand(0),
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000483 TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000484 for (unsigned i = 1, e = CA->getNumOperands(); i != e; ++i) {
485 Out << ", ";
Chris Lattner1e194682002-04-18 18:53:13 +0000486 printTypeInt(Out, ETy, TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000487 WriteAsOperandInternal(Out, CA->getOperand(i), TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000488 }
489 }
490 Out << " ]";
491 }
492 } else if (const ConstantStruct *CS = dyn_cast<ConstantStruct>(CV)) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000493 Out << '{';
Jim Laskey3bb78742006-02-25 12:27:03 +0000494 unsigned N = CS->getNumOperands();
495 if (N) {
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000496 if (N > 2) {
497 Indent += std::string(IndentSize, ' ');
498 Out << Indent;
499 } else {
500 Out << ' ';
501 }
Chris Lattnerd84bb632002-04-16 21:36:08 +0000502 printTypeInt(Out, CS->getOperand(0)->getType(), TypeTable);
503
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000504 WriteAsOperandInternal(Out, CS->getOperand(0), TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000505
Jim Laskey3bb78742006-02-25 12:27:03 +0000506 for (unsigned i = 1; i < N; i++) {
Chris Lattnerd84bb632002-04-16 21:36:08 +0000507 Out << ", ";
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000508 if (N > 2) Out << Indent;
Chris Lattnerd84bb632002-04-16 21:36:08 +0000509 printTypeInt(Out, CS->getOperand(i)->getType(), TypeTable);
510
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000511 WriteAsOperandInternal(Out, CS->getOperand(i), TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000512 }
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000513 if (N > 2) Indent.resize(Indent.size() - IndentSize);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000514 }
Jim Laskey3bb78742006-02-25 12:27:03 +0000515
Chris Lattnerd84bb632002-04-16 21:36:08 +0000516 Out << " }";
Brian Gaeke02209042004-08-20 06:00:58 +0000517 } else if (const ConstantPacked *CP = dyn_cast<ConstantPacked>(CV)) {
518 const Type *ETy = CP->getType()->getElementType();
Misha Brukmanb1c93172005-04-21 23:48:37 +0000519 assert(CP->getNumOperands() > 0 &&
Brian Gaeke02209042004-08-20 06:00:58 +0000520 "Number of operands for a PackedConst must be > 0");
521 Out << '<';
522 Out << ' ';
523 printTypeInt(Out, ETy, TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000524 WriteAsOperandInternal(Out, CP->getOperand(0), TypeTable, Machine);
Brian Gaeke02209042004-08-20 06:00:58 +0000525 for (unsigned i = 1, e = CP->getNumOperands(); i != e; ++i) {
526 Out << ", ";
527 printTypeInt(Out, ETy, TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000528 WriteAsOperandInternal(Out, CP->getOperand(i), TypeTable, Machine);
Brian Gaeke02209042004-08-20 06:00:58 +0000529 }
530 Out << " >";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000531 } else if (isa<ConstantPointerNull>(CV)) {
532 Out << "null";
533
Chris Lattner5e0b9f22004-10-16 18:08:06 +0000534 } else if (isa<UndefValue>(CV)) {
535 Out << "undef";
536
Chris Lattner3cd8c562002-07-30 18:54:25 +0000537 } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CV)) {
Reid Spencer812a1be2006-12-04 05:19:18 +0000538 Out << CE->getOpcodeName();
539 if (CE->isCompare())
540 Out << " " << getPredicateText(CE->getPredicate());
541 Out << " (";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000542
Vikram S. Adveb952b542002-07-14 23:14:45 +0000543 for (User::const_op_iterator OI=CE->op_begin(); OI != CE->op_end(); ++OI) {
544 printTypeInt(Out, (*OI)->getType(), TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000545 WriteAsOperandInternal(Out, *OI, TypeTable, Machine);
Vikram S. Adveb952b542002-07-14 23:14:45 +0000546 if (OI+1 != CE->op_end())
Chris Lattner3cd8c562002-07-30 18:54:25 +0000547 Out << ", ";
Vikram S. Adveb952b542002-07-14 23:14:45 +0000548 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000549
Reid Spencer6c38f0b2006-11-27 01:05:10 +0000550 if (CE->isCast()) {
Chris Lattner83b396b2002-08-15 19:37:43 +0000551 Out << " to ";
552 printTypeInt(Out, CE->getType(), TypeTable);
553 }
Reid Spencer6c38f0b2006-11-27 01:05:10 +0000554
Misha Brukman21bbdb92004-06-04 21:11:51 +0000555 Out << ')';
Chris Lattner83b396b2002-08-15 19:37:43 +0000556
Chris Lattnerd84bb632002-04-16 21:36:08 +0000557 } else {
Vikram S. Adveb952b542002-07-14 23:14:45 +0000558 Out << "<placeholder or erroneous Constant>";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000559 }
560}
561
562
Misha Brukmanc566ca362004-03-02 00:22:19 +0000563/// WriteAsOperand - Write the name of the specified value out to the specified
564/// ostream. This can be useful when you just want to print int %reg126, not
565/// the whole instruction that generated it.
566///
Misha Brukmanb1c93172005-04-21 23:48:37 +0000567static void WriteAsOperandInternal(std::ostream &Out, const Value *V,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000568 std::map<const Type*, std::string> &TypeTable,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000569 SlotMachine *Machine) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000570 Out << ' ';
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000571 if (V->hasName())
Chris Lattner1c343042003-08-22 05:40:38 +0000572 Out << getLLVMName(V->getName());
Reid Spenceraccd7c72004-07-17 23:47:01 +0000573 else {
574 const Constant *CV = dyn_cast<Constant>(V);
Chris Lattnera2d810d2006-01-25 22:26:05 +0000575 if (CV && !isa<GlobalValue>(CV)) {
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000576 WriteConstantInt(Out, CV, TypeTable, Machine);
Chris Lattnera2d810d2006-01-25 22:26:05 +0000577 } else if (const InlineAsm *IA = dyn_cast<InlineAsm>(V)) {
578 Out << "asm ";
579 if (IA->hasSideEffects())
580 Out << "sideeffect ";
581 Out << '"';
582 PrintEscapedString(IA->getAsmString(), Out);
583 Out << "\", \"";
584 PrintEscapedString(IA->getConstraintString(), Out);
585 Out << '"';
586 } else {
Chris Lattnerd84bb632002-04-16 21:36:08 +0000587 int Slot;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000588 if (Machine) {
Reid Spencer56010e42004-05-26 21:56:09 +0000589 Slot = Machine->getSlot(V);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000590 } else {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000591 Machine = createSlotMachine(V);
Chris Lattner96749c42006-05-14 18:46:52 +0000592 if (Machine)
Chris Lattner757ee0b2004-06-09 19:41:19 +0000593 Slot = Machine->getSlot(V);
594 else
595 Slot = -1;
Reid Spencer56010e42004-05-26 21:56:09 +0000596 delete Machine;
Chris Lattnerd84bb632002-04-16 21:36:08 +0000597 }
Chris Lattner757ee0b2004-06-09 19:41:19 +0000598 if (Slot != -1)
599 Out << '%' << Slot;
600 else
601 Out << "<badref>";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000602 }
603 }
604}
605
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000606/// WriteAsOperand - Write the name of the specified value out to the specified
607/// ostream. This can be useful when you just want to print int %reg126, not
608/// the whole instruction that generated it.
609///
Chris Lattner189d19f2003-11-21 20:23:48 +0000610std::ostream &llvm::WriteAsOperand(std::ostream &Out, const Value *V,
Chris Lattner0dda8612006-12-06 06:15:43 +0000611 bool PrintType, const Module *Context) {
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000612 std::map<const Type *, std::string> TypeNames;
Chris Lattner5a9f63e2002-07-10 16:48:17 +0000613 if (Context == 0) Context = getModuleFromVal(V);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000614
Chris Lattner98cf1f52002-11-20 18:36:02 +0000615 if (Context)
Chris Lattner5a9f63e2002-07-10 16:48:17 +0000616 fillTypeNameTable(Context, TypeNames);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000617
618 if (PrintType)
619 printTypeInt(Out, V->getType(), TypeNames);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000620
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000621 WriteAsOperandInternal(Out, V, TypeNames, 0);
Chris Lattner5e5abe32001-07-20 19:15:21 +0000622 return Out;
623}
624
Reid Spencer58d30f22004-07-04 11:50:43 +0000625
Chris Lattner189d19f2003-11-21 20:23:48 +0000626namespace llvm {
Chris Lattner2e9fee42001-07-12 23:35:26 +0000627
Chris Lattnerfee714f2001-09-07 16:36:04 +0000628class AssemblyWriter {
Misha Brukmana6619a92004-06-21 21:53:56 +0000629 std::ostream &Out;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000630 SlotMachine &Machine;
Chris Lattner7bfee412001-10-29 16:05:51 +0000631 const Module *TheModule;
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000632 std::map<const Type *, std::string> TypeNames;
Chris Lattner8339f7d2003-10-30 23:41:03 +0000633 AssemblyAnnotationWriter *AnnotationWriter;
Chris Lattner2f7c9632001-06-06 20:29:01 +0000634public:
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000635 inline AssemblyWriter(std::ostream &o, SlotMachine &Mac, const Module *M,
Chris Lattner8339f7d2003-10-30 23:41:03 +0000636 AssemblyAnnotationWriter *AAW)
Misha Brukmana6619a92004-06-21 21:53:56 +0000637 : Out(o), Machine(Mac), TheModule(M), AnnotationWriter(AAW) {
Chris Lattner7bfee412001-10-29 16:05:51 +0000638
639 // If the module has a symbol table, take all global types and stuff their
640 // names into the TypeNames map.
641 //
Chris Lattnerb86620e2001-10-29 16:37:48 +0000642 fillTypeNameTable(M, TypeNames);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000643 }
644
Chris Lattner7bfee412001-10-29 16:05:51 +0000645 inline void write(const Module *M) { printModule(M); }
646 inline void write(const GlobalVariable *G) { printGlobal(G); }
Chris Lattner57698e22002-03-26 18:01:55 +0000647 inline void write(const Function *F) { printFunction(F); }
Chris Lattner7bfee412001-10-29 16:05:51 +0000648 inline void write(const BasicBlock *BB) { printBasicBlock(BB); }
Chris Lattner113f4f42002-06-25 16:13:24 +0000649 inline void write(const Instruction *I) { printInstruction(*I); }
Chris Lattner3462ae32001-12-03 22:26:30 +0000650 inline void write(const Constant *CPV) { printConstant(CPV); }
Chris Lattner7db79582001-11-07 04:21:57 +0000651 inline void write(const Type *Ty) { printType(Ty); }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000652
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000653 void writeOperand(const Value *Op, bool PrintType);
Chris Lattner1e194682002-04-18 18:53:13 +0000654
Misha Brukman4685e262004-04-28 15:31:21 +0000655 const Module* getModule() { return TheModule; }
656
Misha Brukmand92f54a2004-11-15 19:30:05 +0000657private:
Chris Lattner7bfee412001-10-29 16:05:51 +0000658 void printModule(const Module *M);
659 void printSymbolTable(const SymbolTable &ST);
Chris Lattner3462ae32001-12-03 22:26:30 +0000660 void printConstant(const Constant *CPV);
Chris Lattner7bfee412001-10-29 16:05:51 +0000661 void printGlobal(const GlobalVariable *GV);
Chris Lattner57698e22002-03-26 18:01:55 +0000662 void printFunction(const Function *F);
Chris Lattner2e9fa6d2002-04-09 19:48:49 +0000663 void printArgument(const Argument *FA);
Chris Lattner7bfee412001-10-29 16:05:51 +0000664 void printBasicBlock(const BasicBlock *BB);
Chris Lattner113f4f42002-06-25 16:13:24 +0000665 void printInstruction(const Instruction &I);
Chris Lattnerd816b532002-04-13 20:53:41 +0000666
667 // printType - Go to extreme measures to attempt to print out a short,
668 // symbolic version of a type name.
669 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000670 std::ostream &printType(const Type *Ty) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000671 return printTypeInt(Out, Ty, TypeNames);
Chris Lattnerd816b532002-04-13 20:53:41 +0000672 }
673
674 // printTypeAtLeastOneLevel - Print out one level of the possibly complex type
675 // without considering any symbolic types that we may have equal to it.
676 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000677 std::ostream &printTypeAtLeastOneLevel(const Type *Ty);
Chris Lattner7bfee412001-10-29 16:05:51 +0000678
Chris Lattner862e3382001-10-13 06:42:36 +0000679 // printInfoComment - Print a little comment after the instruction indicating
680 // which slot it occupies.
Chris Lattner113f4f42002-06-25 16:13:24 +0000681 void printInfoComment(const Value &V);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000682};
Reid Spencerf43ac622004-05-27 22:04:46 +0000683} // end of llvm namespace
Chris Lattner2f7c9632001-06-06 20:29:01 +0000684
Misha Brukmanc566ca362004-03-02 00:22:19 +0000685/// printTypeAtLeastOneLevel - Print out one level of the possibly complex type
686/// without considering any symbolic types that we may have equal to it.
687///
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000688std::ostream &AssemblyWriter::printTypeAtLeastOneLevel(const Type *Ty) {
Chris Lattner113f4f42002-06-25 16:13:24 +0000689 if (const FunctionType *FTy = dyn_cast<FunctionType>(Ty)) {
Chris Lattnerd816b532002-04-13 20:53:41 +0000690 printType(FTy->getReturnType()) << " (";
Chris Lattnerfa829be2004-02-09 04:14:01 +0000691 for (FunctionType::param_iterator I = FTy->param_begin(),
692 E = FTy->param_end(); I != E; ++I) {
693 if (I != FTy->param_begin())
Misha Brukmana6619a92004-06-21 21:53:56 +0000694 Out << ", ";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000695 printType(*I);
Chris Lattnerd816b532002-04-13 20:53:41 +0000696 }
697 if (FTy->isVarArg()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000698 if (FTy->getNumParams()) Out << ", ";
699 Out << "...";
Chris Lattnerd816b532002-04-13 20:53:41 +0000700 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000701 Out << ')';
Chris Lattner113f4f42002-06-25 16:13:24 +0000702 } else if (const StructType *STy = dyn_cast<StructType>(Ty)) {
Andrew Lenharthdcb3c972006-12-08 18:06:16 +0000703 if (STy->isPacked())
704 Out << '<';
Misha Brukmana6619a92004-06-21 21:53:56 +0000705 Out << "{ ";
Chris Lattnerac6db752004-02-09 04:37:31 +0000706 for (StructType::element_iterator I = STy->element_begin(),
707 E = STy->element_end(); I != E; ++I) {
708 if (I != STy->element_begin())
Misha Brukmana6619a92004-06-21 21:53:56 +0000709 Out << ", ";
Chris Lattnerd816b532002-04-13 20:53:41 +0000710 printType(*I);
711 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000712 Out << " }";
Andrew Lenharthdcb3c972006-12-08 18:06:16 +0000713 if (STy->isPacked())
714 Out << '>';
Chris Lattner113f4f42002-06-25 16:13:24 +0000715 } else if (const PointerType *PTy = dyn_cast<PointerType>(Ty)) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000716 printType(PTy->getElementType()) << '*';
Chris Lattner113f4f42002-06-25 16:13:24 +0000717 } else if (const ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000718 Out << '[' << ATy->getNumElements() << " x ";
Misha Brukman21bbdb92004-06-04 21:11:51 +0000719 printType(ATy->getElementType()) << ']';
Reid Spencere203d352004-08-20 15:37:30 +0000720 } else if (const PackedType *PTy = dyn_cast<PackedType>(Ty)) {
721 Out << '<' << PTy->getNumElements() << " x ";
722 printType(PTy->getElementType()) << '>';
723 }
Reid Spencerde46e482006-11-02 20:25:50 +0000724 else if (isa<OpaqueType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000725 Out << "opaque";
Chris Lattnerd816b532002-04-13 20:53:41 +0000726 } else {
Vikram S. Adveb952b542002-07-14 23:14:45 +0000727 if (!Ty->isPrimitiveType())
Misha Brukmana6619a92004-06-21 21:53:56 +0000728 Out << "<unknown derived type>";
Chris Lattnerd816b532002-04-13 20:53:41 +0000729 printType(Ty);
730 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000731 return Out;
Chris Lattnerd816b532002-04-13 20:53:41 +0000732}
733
734
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000735void AssemblyWriter::writeOperand(const Value *Operand, bool PrintType) {
736 if (Operand == 0) {
Chris Lattner08f7d0c2005-02-24 16:58:29 +0000737 Out << "<null operand!>";
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000738 } else {
739 if (PrintType) { Out << ' '; printType(Operand->getType()); }
740 WriteAsOperandInternal(Out, Operand, TypeNames, &Machine);
Chris Lattner08f7d0c2005-02-24 16:58:29 +0000741 }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000742}
743
Chris Lattner2f7c9632001-06-06 20:29:01 +0000744
Chris Lattner7bfee412001-10-29 16:05:51 +0000745void AssemblyWriter::printModule(const Module *M) {
Chris Lattner4d8689e2005-03-02 23:12:40 +0000746 if (!M->getModuleIdentifier().empty() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +0000747 // Don't print the ID if it will start a new line (which would
Chris Lattner4d8689e2005-03-02 23:12:40 +0000748 // require a comment char before it).
749 M->getModuleIdentifier().find('\n') == std::string::npos)
750 Out << "; ModuleID = '" << M->getModuleIdentifier() << "'\n";
751
Owen Andersone2237542006-10-18 02:21:12 +0000752 if (!M->getDataLayout().empty())
Chris Lattner04897162006-10-22 06:06:56 +0000753 Out << "target datalayout = \"" << M->getDataLayout() << "\"\n";
Owen Andersone2237542006-10-18 02:21:12 +0000754
Chris Lattner8068e0c2003-08-24 13:48:48 +0000755 switch (M->getEndianness()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000756 case Module::LittleEndian: Out << "target endian = little\n"; break;
757 case Module::BigEndian: Out << "target endian = big\n"; break;
Chris Lattner8068e0c2003-08-24 13:48:48 +0000758 case Module::AnyEndianness: break;
759 }
760 switch (M->getPointerSize()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000761 case Module::Pointer32: Out << "target pointersize = 32\n"; break;
762 case Module::Pointer64: Out << "target pointersize = 64\n"; break;
Chris Lattner8068e0c2003-08-24 13:48:48 +0000763 case Module::AnyPointerSize: break;
764 }
Reid Spencer48f98c82004-07-25 21:44:54 +0000765 if (!M->getTargetTriple().empty())
Reid Spencerffec7df2004-07-25 21:29:43 +0000766 Out << "target triple = \"" << M->getTargetTriple() << "\"\n";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000767
Chris Lattnereef2fe72006-01-24 04:13:11 +0000768 if (!M->getModuleInlineAsm().empty()) {
Chris Lattnerefaf35d2006-01-24 00:45:30 +0000769 // Split the string into lines, to make it easier to read the .ll file.
Chris Lattnereef2fe72006-01-24 04:13:11 +0000770 std::string Asm = M->getModuleInlineAsm();
Chris Lattnerefaf35d2006-01-24 00:45:30 +0000771 size_t CurPos = 0;
772 size_t NewLine = Asm.find_first_of('\n', CurPos);
773 while (NewLine != std::string::npos) {
774 // We found a newline, print the portion of the asm string from the
775 // last newline up to this newline.
776 Out << "module asm \"";
777 PrintEscapedString(std::string(Asm.begin()+CurPos, Asm.begin()+NewLine),
778 Out);
779 Out << "\"\n";
780 CurPos = NewLine+1;
781 NewLine = Asm.find_first_of('\n', CurPos);
782 }
Chris Lattner3acaf5c2006-01-24 00:40:17 +0000783 Out << "module asm \"";
Chris Lattnerefaf35d2006-01-24 00:45:30 +0000784 PrintEscapedString(std::string(Asm.begin()+CurPos, Asm.end()), Out);
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000785 Out << "\"\n";
786 }
787
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000788 // Loop over the dependent libraries and emit them.
Chris Lattner730cfe42004-09-14 04:51:44 +0000789 Module::lib_iterator LI = M->lib_begin();
790 Module::lib_iterator LE = M->lib_end();
Reid Spencer48f98c82004-07-25 21:44:54 +0000791 if (LI != LE) {
Chris Lattner730cfe42004-09-14 04:51:44 +0000792 Out << "deplibs = [ ";
793 while (LI != LE) {
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000794 Out << '"' << *LI << '"';
Reid Spencerffec7df2004-07-25 21:29:43 +0000795 ++LI;
Chris Lattner730cfe42004-09-14 04:51:44 +0000796 if (LI != LE)
797 Out << ", ";
Reid Spencerffec7df2004-07-25 21:29:43 +0000798 }
799 Out << " ]\n";
Reid Spencercc5ff642004-07-25 18:08:18 +0000800 }
Reid Spencerb9e08772004-09-13 23:44:23 +0000801
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000802 // Loop over the symbol table, emitting all named constants.
Chris Lattner98cf1f52002-11-20 18:36:02 +0000803 printSymbolTable(M->getSymbolTable());
Misha Brukmanb1c93172005-04-21 23:48:37 +0000804
Chris Lattner54932b02006-12-06 04:41:52 +0000805 for (Module::const_global_iterator I = M->global_begin(), E = M->global_end();
806 I != E; ++I)
Chris Lattner113f4f42002-06-25 16:13:24 +0000807 printGlobal(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000808
Misha Brukmana6619a92004-06-21 21:53:56 +0000809 Out << "\nimplementation ; Functions:\n";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000810
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000811 // Output all of the functions.
Chris Lattner113f4f42002-06-25 16:13:24 +0000812 for (Module::const_iterator I = M->begin(), E = M->end(); I != E; ++I)
813 printFunction(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000814}
815
Chris Lattner7bfee412001-10-29 16:05:51 +0000816void AssemblyWriter::printGlobal(const GlobalVariable *GV) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000817 if (GV->hasName()) Out << getLLVMName(GV->getName()) << " = ";
Chris Lattner37798642001-09-18 04:01:05 +0000818
Misha Brukmanb1c93172005-04-21 23:48:37 +0000819 if (!GV->hasInitializer())
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000820 switch (GV->getLinkage()) {
821 case GlobalValue::DLLImportLinkage: Out << "dllimport "; break;
822 case GlobalValue::ExternalWeakLinkage: Out << "extern_weak "; break;
823 default: Out << "external "; break;
824 }
Chris Lattner379a8d22003-04-16 20:28:45 +0000825 else
826 switch (GV->getLinkage()) {
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000827 case GlobalValue::InternalLinkage: Out << "internal "; break;
828 case GlobalValue::LinkOnceLinkage: Out << "linkonce "; break;
829 case GlobalValue::WeakLinkage: Out << "weak "; break;
830 case GlobalValue::AppendingLinkage: Out << "appending "; break;
831 case GlobalValue::DLLImportLinkage: Out << "dllimport "; break;
832 case GlobalValue::DLLExportLinkage: Out << "dllexport "; break;
833 case GlobalValue::ExternalWeakLinkage: Out << "extern_weak "; break;
834 case GlobalValue::ExternalLinkage: break;
Misha Brukman70457632004-11-14 21:04:34 +0000835 case GlobalValue::GhostLinkage:
Bill Wendlingf3baad32006-12-07 01:30:32 +0000836 cerr << "GhostLinkage not allowed in AsmWriter!\n";
Misha Brukman70457632004-11-14 21:04:34 +0000837 abort();
Chris Lattner379a8d22003-04-16 20:28:45 +0000838 }
Chris Lattner37798642001-09-18 04:01:05 +0000839
Misha Brukmana6619a92004-06-21 21:53:56 +0000840 Out << (GV->isConstant() ? "constant " : "global ");
Chris Lattner2413b162001-12-04 00:03:30 +0000841 printType(GV->getType()->getElementType());
Chris Lattner37798642001-09-18 04:01:05 +0000842
Reid Spenceraccd7c72004-07-17 23:47:01 +0000843 if (GV->hasInitializer()) {
844 Constant* C = cast<Constant>(GV->getInitializer());
845 assert(C && "GlobalVar initializer isn't constant?");
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000846 writeOperand(GV->getInitializer(), false);
Reid Spenceraccd7c72004-07-17 23:47:01 +0000847 }
Chris Lattnerf8a974d2005-11-06 06:48:53 +0000848
Chris Lattner4b96c542005-11-12 00:10:19 +0000849 if (GV->hasSection())
850 Out << ", section \"" << GV->getSection() << '"';
851 if (GV->getAlignment())
Chris Lattnerf8a974d2005-11-06 06:48:53 +0000852 Out << ", align " << GV->getAlignment();
Chris Lattner4b96c542005-11-12 00:10:19 +0000853
Chris Lattner113f4f42002-06-25 16:13:24 +0000854 printInfoComment(*GV);
Misha Brukmana6619a92004-06-21 21:53:56 +0000855 Out << "\n";
Chris Lattnerda975502001-09-10 07:58:01 +0000856}
857
Chris Lattnerfee714f2001-09-07 16:36:04 +0000858
Reid Spencere7e96712004-05-25 08:53:40 +0000859// printSymbolTable - Run through symbol table looking for constants
860// and types. Emit their declarations.
Chris Lattner7bfee412001-10-29 16:05:51 +0000861void AssemblyWriter::printSymbolTable(const SymbolTable &ST) {
Reid Spencere7e96712004-05-25 08:53:40 +0000862
863 // Print the types.
864 for (SymbolTable::type_const_iterator TI = ST.type_begin();
Chris Lattner23e829a2006-12-06 05:12:21 +0000865 TI != ST.type_end(); ++TI) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000866 Out << "\t" << getLLVMName(TI->first) << " = type ";
Reid Spencere7e96712004-05-25 08:53:40 +0000867
868 // Make sure we print out at least one level of the type structure, so
869 // that we do not get %FILE = type %FILE
870 //
871 printTypeAtLeastOneLevel(TI->second) << "\n";
872 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000873
Reid Spencere7e96712004-05-25 08:53:40 +0000874 // Print the constants, in type plane order.
875 for (SymbolTable::plane_const_iterator PI = ST.plane_begin();
Chris Lattner23e829a2006-12-06 05:12:21 +0000876 PI != ST.plane_end(); ++PI) {
Reid Spencere7e96712004-05-25 08:53:40 +0000877 SymbolTable::value_const_iterator VI = ST.value_begin(PI->first);
878 SymbolTable::value_const_iterator VE = ST.value_end(PI->first);
879
880 for (; VI != VE; ++VI) {
Reid Spenceraccd7c72004-07-17 23:47:01 +0000881 const Value* V = VI->second;
882 const Constant *CPV = dyn_cast<Constant>(V) ;
883 if (CPV && !isa<GlobalValue>(V)) {
Reid Spencer56010e42004-05-26 21:56:09 +0000884 printConstant(CPV);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000885 }
886 }
Chris Lattnera7620d92001-07-15 06:35:59 +0000887 }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000888}
889
890
Misha Brukmanc566ca362004-03-02 00:22:19 +0000891/// printConstant - Print out a constant pool entry...
892///
Chris Lattner3462ae32001-12-03 22:26:30 +0000893void AssemblyWriter::printConstant(const Constant *CPV) {
Chris Lattnerfee714f2001-09-07 16:36:04 +0000894 // Don't print out unnamed constants, they will be inlined
895 if (!CPV->hasName()) return;
Chris Lattner2f7c9632001-06-06 20:29:01 +0000896
Chris Lattneree998be2001-07-26 16:29:38 +0000897 // Print out name...
Misha Brukmana6619a92004-06-21 21:53:56 +0000898 Out << "\t" << getLLVMName(CPV->getName()) << " =";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000899
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000900 // Write the value out now.
901 writeOperand(CPV, true);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000902
Chris Lattner113f4f42002-06-25 16:13:24 +0000903 printInfoComment(*CPV);
Misha Brukmana6619a92004-06-21 21:53:56 +0000904 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000905}
906
Misha Brukmanc566ca362004-03-02 00:22:19 +0000907/// printFunction - Print all aspects of a function.
908///
Chris Lattner113f4f42002-06-25 16:13:24 +0000909void AssemblyWriter::printFunction(const Function *F) {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000910 // Print out the return type and name...
Misha Brukmana6619a92004-06-21 21:53:56 +0000911 Out << "\n";
Chris Lattner379a8d22003-04-16 20:28:45 +0000912
Chris Lattner97e36f22004-12-05 06:44:09 +0000913 // Ensure that no local symbols conflict with global symbols.
914 const_cast<Function*>(F)->renameLocalSymbols();
915
Misha Brukmana6619a92004-06-21 21:53:56 +0000916 if (AnnotationWriter) AnnotationWriter->emitFunctionAnnot(F, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +0000917
Chris Lattner379a8d22003-04-16 20:28:45 +0000918 if (F->isExternal())
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000919 switch (F->getLinkage()) {
920 case GlobalValue::DLLImportLinkage: Out << "declare dllimport "; break;
921 case GlobalValue::ExternalWeakLinkage: Out << "declare extern_weak "; break;
922 default: Out << "declare ";
923 }
Reid Spencer7ce2d2a2006-12-29 20:29:48 +0000924 else {
925 Out << "define ";
Chris Lattner379a8d22003-04-16 20:28:45 +0000926 switch (F->getLinkage()) {
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000927 case GlobalValue::InternalLinkage: Out << "internal "; break;
928 case GlobalValue::LinkOnceLinkage: Out << "linkonce "; break;
929 case GlobalValue::WeakLinkage: Out << "weak "; break;
930 case GlobalValue::AppendingLinkage: Out << "appending "; break;
931 case GlobalValue::DLLImportLinkage: Out << "dllimport "; break;
932 case GlobalValue::DLLExportLinkage: Out << "dllexport "; break;
933 case GlobalValue::ExternalWeakLinkage: Out << "extern_weak "; break;
Chris Lattner379a8d22003-04-16 20:28:45 +0000934 case GlobalValue::ExternalLinkage: break;
Misha Brukman70457632004-11-14 21:04:34 +0000935 case GlobalValue::GhostLinkage:
Bill Wendlingf3baad32006-12-07 01:30:32 +0000936 cerr << "GhostLinkage not allowed in AsmWriter!\n";
Misha Brukman70457632004-11-14 21:04:34 +0000937 abort();
Chris Lattner379a8d22003-04-16 20:28:45 +0000938 }
Reid Spencer7ce2d2a2006-12-29 20:29:48 +0000939 }
Chris Lattner379a8d22003-04-16 20:28:45 +0000940
Chris Lattnerf7b6d312005-05-06 20:26:43 +0000941 // Print the calling convention.
942 switch (F->getCallingConv()) {
943 case CallingConv::C: break; // default
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +0000944 case CallingConv::CSRet: Out << "csretcc "; break;
945 case CallingConv::Fast: Out << "fastcc "; break;
946 case CallingConv::Cold: Out << "coldcc "; break;
947 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
948 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +0000949 default: Out << "cc" << F->getCallingConv() << " "; break;
950 }
951
Misha Brukman21bbdb92004-06-04 21:11:51 +0000952 printType(F->getReturnType()) << ' ';
Chris Lattner5b337482003-10-18 05:57:43 +0000953 if (!F->getName().empty())
Misha Brukmana6619a92004-06-21 21:53:56 +0000954 Out << getLLVMName(F->getName());
Chris Lattner5b337482003-10-18 05:57:43 +0000955 else
Misha Brukmana6619a92004-06-21 21:53:56 +0000956 Out << "\"\"";
957 Out << '(';
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000958 Machine.incorporateFunction(F);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000959
Chris Lattner7bfee412001-10-29 16:05:51 +0000960 // Loop over the arguments, printing them...
Chris Lattner113f4f42002-06-25 16:13:24 +0000961 const FunctionType *FT = F->getFunctionType();
Chris Lattnerfee714f2001-09-07 16:36:04 +0000962
Chris Lattner54932b02006-12-06 04:41:52 +0000963 for (Function::const_arg_iterator I = F->arg_begin(), E = F->arg_end();
964 I != E; ++I)
Chris Lattner149376d2002-10-13 20:57:00 +0000965 printArgument(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000966
967 // Finish printing arguments...
Chris Lattner113f4f42002-06-25 16:13:24 +0000968 if (FT->isVarArg()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000969 if (FT->getNumParams()) Out << ", ";
970 Out << "..."; // Output varargs portion of signature!
Chris Lattnerfee714f2001-09-07 16:36:04 +0000971 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000972 Out << ')';
Chris Lattnerfee714f2001-09-07 16:36:04 +0000973
Chris Lattner4b96c542005-11-12 00:10:19 +0000974 if (F->hasSection())
975 Out << " section \"" << F->getSection() << '"';
Chris Lattnerf8a974d2005-11-06 06:48:53 +0000976 if (F->getAlignment())
977 Out << " align " << F->getAlignment();
Chris Lattner4b96c542005-11-12 00:10:19 +0000978
Chris Lattner113f4f42002-06-25 16:13:24 +0000979 if (F->isExternal()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000980 Out << "\n";
Chris Lattnerb2f02e52002-05-06 03:00:40 +0000981 } else {
Misha Brukmana6619a92004-06-21 21:53:56 +0000982 Out << " {";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000983
Chris Lattner6915f8f2002-04-07 22:49:37 +0000984 // Output all of its basic blocks... for the function
Chris Lattner113f4f42002-06-25 16:13:24 +0000985 for (Function::const_iterator I = F->begin(), E = F->end(); I != E; ++I)
986 printBasicBlock(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000987
Misha Brukmana6619a92004-06-21 21:53:56 +0000988 Out << "}\n";
Chris Lattnerfee714f2001-09-07 16:36:04 +0000989 }
990
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000991 Machine.purgeFunction();
Chris Lattner2f7c9632001-06-06 20:29:01 +0000992}
993
Misha Brukmanc566ca362004-03-02 00:22:19 +0000994/// printArgument - This member is called for every argument that is passed into
995/// the function. Simply print it out
996///
Chris Lattner2e9fa6d2002-04-09 19:48:49 +0000997void AssemblyWriter::printArgument(const Argument *Arg) {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000998 // Insert commas as we go... the first arg doesn't get a comma
Chris Lattnerfb972ad2005-03-15 05:03:36 +0000999 if (Arg != Arg->getParent()->arg_begin()) Out << ", ";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001000
1001 // Output type...
Chris Lattner7bfee412001-10-29 16:05:51 +00001002 printType(Arg->getType());
Misha Brukmanb1c93172005-04-21 23:48:37 +00001003
Chris Lattner2f7c9632001-06-06 20:29:01 +00001004 // Output name, if available...
1005 if (Arg->hasName())
Misha Brukmana6619a92004-06-21 21:53:56 +00001006 Out << ' ' << getLLVMName(Arg->getName());
Chris Lattner2f7c9632001-06-06 20:29:01 +00001007}
1008
Misha Brukmanc566ca362004-03-02 00:22:19 +00001009/// printBasicBlock - This member is called for each basic block in a method.
1010///
Chris Lattner7bfee412001-10-29 16:05:51 +00001011void AssemblyWriter::printBasicBlock(const BasicBlock *BB) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001012 if (BB->hasName()) { // Print out the label if it exists...
Alkis Evlogimenos991d6ad2004-12-10 05:41:10 +00001013 Out << "\n" << getLLVMName(BB->getName(), false) << ':';
Chris Lattner408dbdb2002-05-14 16:02:05 +00001014 } else if (!BB->use_empty()) { // Don't print block # of no uses...
Misha Brukmana6619a92004-06-21 21:53:56 +00001015 Out << "\n; <label>:";
Chris Lattner757ee0b2004-06-09 19:41:19 +00001016 int Slot = Machine.getSlot(BB);
1017 if (Slot != -1)
Misha Brukmana6619a92004-06-21 21:53:56 +00001018 Out << Slot;
Chris Lattner757ee0b2004-06-09 19:41:19 +00001019 else
Misha Brukmana6619a92004-06-21 21:53:56 +00001020 Out << "<badref>";
Chris Lattner58185f22002-10-02 19:38:55 +00001021 }
Chris Lattner2447ef52003-11-20 00:09:43 +00001022
1023 if (BB->getParent() == 0)
Misha Brukmana6619a92004-06-21 21:53:56 +00001024 Out << "\t\t; Error: Block without parent!";
Chris Lattner2447ef52003-11-20 00:09:43 +00001025 else {
1026 if (BB != &BB->getParent()->front()) { // Not the entry block?
1027 // Output predecessors for the block...
Misha Brukmana6619a92004-06-21 21:53:56 +00001028 Out << "\t\t;";
Chris Lattner2447ef52003-11-20 00:09:43 +00001029 pred_const_iterator PI = pred_begin(BB), PE = pred_end(BB);
Misha Brukmanb1c93172005-04-21 23:48:37 +00001030
Chris Lattner2447ef52003-11-20 00:09:43 +00001031 if (PI == PE) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001032 Out << " No predecessors!";
Chris Lattner2447ef52003-11-20 00:09:43 +00001033 } else {
Misha Brukmana6619a92004-06-21 21:53:56 +00001034 Out << " preds =";
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001035 writeOperand(*PI, false);
Chris Lattner2447ef52003-11-20 00:09:43 +00001036 for (++PI; PI != PE; ++PI) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001037 Out << ',';
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001038 writeOperand(*PI, false);
Chris Lattner2447ef52003-11-20 00:09:43 +00001039 }
Chris Lattner00211f12003-11-16 22:59:57 +00001040 }
Chris Lattner58185f22002-10-02 19:38:55 +00001041 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001042 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001043
Misha Brukmana6619a92004-06-21 21:53:56 +00001044 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001045
Misha Brukmana6619a92004-06-21 21:53:56 +00001046 if (AnnotationWriter) AnnotationWriter->emitBasicBlockStartAnnot(BB, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001047
Chris Lattnerfee714f2001-09-07 16:36:04 +00001048 // Output all of the instructions in the basic block...
Chris Lattner113f4f42002-06-25 16:13:24 +00001049 for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I != E; ++I)
1050 printInstruction(*I);
Chris Lattner96cdd272004-03-08 18:51:45 +00001051
Misha Brukmana6619a92004-06-21 21:53:56 +00001052 if (AnnotationWriter) AnnotationWriter->emitBasicBlockEndAnnot(BB, Out);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001053}
1054
Chris Lattner862e3382001-10-13 06:42:36 +00001055
Misha Brukmanc566ca362004-03-02 00:22:19 +00001056/// printInfoComment - Print a little comment after the instruction indicating
1057/// which slot it occupies.
1058///
Chris Lattner113f4f42002-06-25 16:13:24 +00001059void AssemblyWriter::printInfoComment(const Value &V) {
1060 if (V.getType() != Type::VoidTy) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001061 Out << "\t\t; <";
Misha Brukman21bbdb92004-06-04 21:11:51 +00001062 printType(V.getType()) << '>';
Chris Lattner862e3382001-10-13 06:42:36 +00001063
Chris Lattner113f4f42002-06-25 16:13:24 +00001064 if (!V.hasName()) {
Chris Lattner757ee0b2004-06-09 19:41:19 +00001065 int SlotNum = Machine.getSlot(&V);
1066 if (SlotNum == -1)
Misha Brukmana6619a92004-06-21 21:53:56 +00001067 Out << ":<badref>";
Reid Spencer8beac692004-06-09 15:26:53 +00001068 else
Misha Brukmana6619a92004-06-21 21:53:56 +00001069 Out << ':' << SlotNum; // Print out the def slot taken.
Chris Lattner862e3382001-10-13 06:42:36 +00001070 }
Chris Lattnerb6c21db2005-02-01 01:24:01 +00001071 Out << " [#uses=" << V.getNumUses() << ']'; // Output # uses
Chris Lattner862e3382001-10-13 06:42:36 +00001072 }
1073}
1074
Reid Spencere7141c82006-08-28 01:02:49 +00001075// This member is called for each Instruction in a function..
Chris Lattner113f4f42002-06-25 16:13:24 +00001076void AssemblyWriter::printInstruction(const Instruction &I) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001077 if (AnnotationWriter) AnnotationWriter->emitInstructionAnnot(&I, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001078
Misha Brukmana6619a92004-06-21 21:53:56 +00001079 Out << "\t";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001080
1081 // Print out name if it exists...
Chris Lattner113f4f42002-06-25 16:13:24 +00001082 if (I.hasName())
Misha Brukmana6619a92004-06-21 21:53:56 +00001083 Out << getLLVMName(I.getName()) << " = ";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001084
Chris Lattner06038452005-05-06 05:51:46 +00001085 // If this is a volatile load or store, print out the volatile marker.
Chris Lattner504f9242003-09-08 17:45:59 +00001086 if ((isa<LoadInst>(I) && cast<LoadInst>(I).isVolatile()) ||
Chris Lattner06038452005-05-06 05:51:46 +00001087 (isa<StoreInst>(I) && cast<StoreInst>(I).isVolatile())) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001088 Out << "volatile ";
Chris Lattner06038452005-05-06 05:51:46 +00001089 } else if (isa<CallInst>(I) && cast<CallInst>(I).isTailCall()) {
1090 // If this is a call, check if it's a tail call.
1091 Out << "tail ";
1092 }
Chris Lattner504f9242003-09-08 17:45:59 +00001093
Chris Lattner2f7c9632001-06-06 20:29:01 +00001094 // Print out the opcode...
Misha Brukmana6619a92004-06-21 21:53:56 +00001095 Out << I.getOpcodeName();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001096
Reid Spencer45e52392006-12-03 06:27:29 +00001097 // Print out the compare instruction predicates
1098 if (const FCmpInst *FCI = dyn_cast<FCmpInst>(&I)) {
Reid Spencer812a1be2006-12-04 05:19:18 +00001099 Out << " " << getPredicateText(FCI->getPredicate());
Reid Spencer45e52392006-12-03 06:27:29 +00001100 } else if (const ICmpInst *ICI = dyn_cast<ICmpInst>(&I)) {
Reid Spencer812a1be2006-12-04 05:19:18 +00001101 Out << " " << getPredicateText(ICI->getPredicate());
Reid Spencer45e52392006-12-03 06:27:29 +00001102 }
1103
Chris Lattner2f7c9632001-06-06 20:29:01 +00001104 // Print out the type of the operands...
Chris Lattner113f4f42002-06-25 16:13:24 +00001105 const Value *Operand = I.getNumOperands() ? I.getOperand(0) : 0;
Chris Lattner2f7c9632001-06-06 20:29:01 +00001106
1107 // Special case conditional branches to swizzle the condition out to the front
Chris Lattner113f4f42002-06-25 16:13:24 +00001108 if (isa<BranchInst>(I) && I.getNumOperands() > 1) {
1109 writeOperand(I.getOperand(2), true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001110 Out << ',';
Chris Lattner2f7c9632001-06-06 20:29:01 +00001111 writeOperand(Operand, true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001112 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001113 writeOperand(I.getOperand(1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001114
Chris Lattner8d48df22002-04-13 18:34:38 +00001115 } else if (isa<SwitchInst>(I)) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001116 // Special case switch statement to get formatting nice and correct...
Misha Brukmana6619a92004-06-21 21:53:56 +00001117 writeOperand(Operand , true); Out << ',';
1118 writeOperand(I.getOperand(1), true); Out << " [";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001119
Chris Lattner113f4f42002-06-25 16:13:24 +00001120 for (unsigned op = 2, Eop = I.getNumOperands(); op < Eop; op += 2) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001121 Out << "\n\t\t";
1122 writeOperand(I.getOperand(op ), true); Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001123 writeOperand(I.getOperand(op+1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001124 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001125 Out << "\n\t]";
Chris Lattnerda558102001-10-02 03:41:24 +00001126 } else if (isa<PHINode>(I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001127 Out << ' ';
Chris Lattner113f4f42002-06-25 16:13:24 +00001128 printType(I.getType());
Misha Brukmana6619a92004-06-21 21:53:56 +00001129 Out << ' ';
Chris Lattner2f7c9632001-06-06 20:29:01 +00001130
Chris Lattner113f4f42002-06-25 16:13:24 +00001131 for (unsigned op = 0, Eop = I.getNumOperands(); op < Eop; op += 2) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001132 if (op) Out << ", ";
Misha Brukmanb1c93172005-04-21 23:48:37 +00001133 Out << '[';
Misha Brukmana6619a92004-06-21 21:53:56 +00001134 writeOperand(I.getOperand(op ), false); Out << ',';
1135 writeOperand(I.getOperand(op+1), false); Out << " ]";
Chris Lattner931ef3b2001-06-11 15:04:20 +00001136 }
Chris Lattner862e3382001-10-13 06:42:36 +00001137 } else if (isa<ReturnInst>(I) && !Operand) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001138 Out << " void";
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001139 } else if (const CallInst *CI = dyn_cast<CallInst>(&I)) {
1140 // Print the calling convention being used.
1141 switch (CI->getCallingConv()) {
1142 case CallingConv::C: break; // default
Chris Lattner29d20852006-05-19 21:58:52 +00001143 case CallingConv::CSRet: Out << " csretcc"; break;
1144 case CallingConv::Fast: Out << " fastcc"; break;
1145 case CallingConv::Cold: Out << " coldcc"; break;
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +00001146 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
1147 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001148 default: Out << " cc" << CI->getCallingConv(); break;
1149 }
1150
Chris Lattner463d6a52003-08-05 15:34:45 +00001151 const PointerType *PTy = cast<PointerType>(Operand->getType());
1152 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1153 const Type *RetTy = FTy->getReturnType();
Chris Lattner2f2d9472001-11-06 21:28:12 +00001154
Chris Lattner463d6a52003-08-05 15:34:45 +00001155 // If possible, print out the short form of the call instruction. We can
Chris Lattner6915f8f2002-04-07 22:49:37 +00001156 // only do this if the first argument is a pointer to a nonvararg function,
Chris Lattner463d6a52003-08-05 15:34:45 +00001157 // and if the return type is not a pointer to a function.
Chris Lattner2f2d9472001-11-06 21:28:12 +00001158 //
Chris Lattner463d6a52003-08-05 15:34:45 +00001159 if (!FTy->isVarArg() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +00001160 (!isa<PointerType>(RetTy) ||
Chris Lattnerd9a36a62002-07-25 20:58:51 +00001161 !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001162 Out << ' '; printType(RetTy);
Chris Lattner2f2d9472001-11-06 21:28:12 +00001163 writeOperand(Operand, false);
1164 } else {
1165 writeOperand(Operand, true);
1166 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001167 Out << '(';
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001168 if (CI->getNumOperands() > 1) writeOperand(CI->getOperand(1), true);
Chris Lattner113f4f42002-06-25 16:13:24 +00001169 for (unsigned op = 2, Eop = I.getNumOperands(); op < Eop; ++op) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001170 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001171 writeOperand(I.getOperand(op), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001172 }
1173
Misha Brukmana6619a92004-06-21 21:53:56 +00001174 Out << " )";
Chris Lattner113f4f42002-06-25 16:13:24 +00001175 } else if (const InvokeInst *II = dyn_cast<InvokeInst>(&I)) {
Chris Lattner463d6a52003-08-05 15:34:45 +00001176 const PointerType *PTy = cast<PointerType>(Operand->getType());
1177 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1178 const Type *RetTy = FTy->getReturnType();
1179
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001180 // Print the calling convention being used.
1181 switch (II->getCallingConv()) {
1182 case CallingConv::C: break; // default
Chris Lattner29d20852006-05-19 21:58:52 +00001183 case CallingConv::CSRet: Out << " csretcc"; break;
1184 case CallingConv::Fast: Out << " fastcc"; break;
1185 case CallingConv::Cold: Out << " coldcc"; break;
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +00001186 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
1187 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001188 default: Out << " cc" << II->getCallingConv(); break;
1189 }
1190
Chris Lattner463d6a52003-08-05 15:34:45 +00001191 // If possible, print out the short form of the invoke instruction. We can
1192 // only do this if the first argument is a pointer to a nonvararg function,
1193 // and if the return type is not a pointer to a function.
1194 //
1195 if (!FTy->isVarArg() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +00001196 (!isa<PointerType>(RetTy) ||
Chris Lattner463d6a52003-08-05 15:34:45 +00001197 !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001198 Out << ' '; printType(RetTy);
Chris Lattner463d6a52003-08-05 15:34:45 +00001199 writeOperand(Operand, false);
1200 } else {
1201 writeOperand(Operand, true);
1202 }
1203
Misha Brukmana6619a92004-06-21 21:53:56 +00001204 Out << '(';
Chris Lattner113f4f42002-06-25 16:13:24 +00001205 if (I.getNumOperands() > 3) writeOperand(I.getOperand(3), true);
1206 for (unsigned op = 4, Eop = I.getNumOperands(); op < Eop; ++op) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001207 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001208 writeOperand(I.getOperand(op), true);
Chris Lattner862e3382001-10-13 06:42:36 +00001209 }
1210
Misha Brukmana6619a92004-06-21 21:53:56 +00001211 Out << " )\n\t\t\tto";
Chris Lattner862e3382001-10-13 06:42:36 +00001212 writeOperand(II->getNormalDest(), true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001213 Out << " unwind";
Chris Lattnerfae8ab32004-02-08 21:44:31 +00001214 writeOperand(II->getUnwindDest(), true);
Chris Lattner862e3382001-10-13 06:42:36 +00001215
Chris Lattner113f4f42002-06-25 16:13:24 +00001216 } else if (const AllocationInst *AI = dyn_cast<AllocationInst>(&I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001217 Out << ' ';
Chris Lattner8d48df22002-04-13 18:34:38 +00001218 printType(AI->getType()->getElementType());
1219 if (AI->isArrayAllocation()) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001220 Out << ',';
Chris Lattner8d48df22002-04-13 18:34:38 +00001221 writeOperand(AI->getArraySize(), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001222 }
Nate Begeman848622f2005-11-05 09:21:28 +00001223 if (AI->getAlignment()) {
Chris Lattner7aeee3a2005-11-05 21:20:34 +00001224 Out << ", align " << AI->getAlignment();
Nate Begeman848622f2005-11-05 09:21:28 +00001225 }
Chris Lattner862e3382001-10-13 06:42:36 +00001226 } else if (isa<CastInst>(I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001227 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001228 Out << " to ";
Chris Lattner113f4f42002-06-25 16:13:24 +00001229 printType(I.getType());
Chris Lattner5b337482003-10-18 05:57:43 +00001230 } else if (isa<VAArgInst>(I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001231 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001232 Out << ", ";
Chris Lattnerf70da102003-05-08 02:44:12 +00001233 printType(I.getType());
Chris Lattner2f7c9632001-06-06 20:29:01 +00001234 } else if (Operand) { // Print the normal way...
1235
Misha Brukmanb1c93172005-04-21 23:48:37 +00001236 // PrintAllTypes - Instructions who have operands of all the same type
Chris Lattner2f7c9632001-06-06 20:29:01 +00001237 // omit the type from all but the first operand. If the instruction has
1238 // different type operands (for example br), then they are all printed.
1239 bool PrintAllTypes = false;
1240 const Type *TheType = Operand->getType();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001241
Chris Lattner52bd5cb92004-03-12 05:53:14 +00001242 // Shift Left & Right print both types even for Ubyte LHS, and select prints
1243 // types even if all operands are bools.
Chris Lattnerbbe0a422006-04-08 01:18:18 +00001244 if (isa<ShiftInst>(I) || isa<SelectInst>(I) || isa<StoreInst>(I) ||
1245 isa<ShuffleVectorInst>(I)) {
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001246 PrintAllTypes = true;
1247 } else {
1248 for (unsigned i = 1, E = I.getNumOperands(); i != E; ++i) {
1249 Operand = I.getOperand(i);
1250 if (Operand->getType() != TheType) {
1251 PrintAllTypes = true; // We have differing types! Print them all!
1252 break;
1253 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001254 }
1255 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001256
Chris Lattner7bfee412001-10-29 16:05:51 +00001257 if (!PrintAllTypes) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001258 Out << ' ';
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001259 printType(TheType);
Chris Lattner7bfee412001-10-29 16:05:51 +00001260 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001261
Chris Lattner113f4f42002-06-25 16:13:24 +00001262 for (unsigned i = 0, E = I.getNumOperands(); i != E; ++i) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001263 if (i) Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001264 writeOperand(I.getOperand(i), PrintAllTypes);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001265 }
1266 }
1267
Chris Lattner862e3382001-10-13 06:42:36 +00001268 printInfoComment(I);
Misha Brukmana6619a92004-06-21 21:53:56 +00001269 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001270}
1271
1272
1273//===----------------------------------------------------------------------===//
1274// External Interface declarations
1275//===----------------------------------------------------------------------===//
1276
Chris Lattner8339f7d2003-10-30 23:41:03 +00001277void Module::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001278 SlotMachine SlotTable(this);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001279 AssemblyWriter W(o, SlotTable, this, AAW);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001280 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001281}
1282
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001283void GlobalVariable::print(std::ostream &o) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001284 SlotMachine SlotTable(getParent());
Chris Lattner8339f7d2003-10-30 23:41:03 +00001285 AssemblyWriter W(o, SlotTable, getParent(), 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001286 W.write(this);
Chris Lattneradfe0d12001-09-10 20:08:19 +00001287}
1288
Chris Lattner8339f7d2003-10-30 23:41:03 +00001289void Function::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001290 SlotMachine SlotTable(getParent());
Chris Lattner8339f7d2003-10-30 23:41:03 +00001291 AssemblyWriter W(o, SlotTable, getParent(), AAW);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001292
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001293 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001294}
1295
Chris Lattnereef2fe72006-01-24 04:13:11 +00001296void InlineAsm::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001297 WriteAsOperand(o, this, true, 0);
Chris Lattnereef2fe72006-01-24 04:13:11 +00001298}
1299
Chris Lattner8339f7d2003-10-30 23:41:03 +00001300void BasicBlock::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001301 SlotMachine SlotTable(getParent());
Misha Brukmanb1c93172005-04-21 23:48:37 +00001302 AssemblyWriter W(o, SlotTable,
Chris Lattner8339f7d2003-10-30 23:41:03 +00001303 getParent() ? getParent()->getParent() : 0, AAW);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001304 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001305}
1306
Chris Lattner8339f7d2003-10-30 23:41:03 +00001307void Instruction::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001308 const Function *F = getParent() ? getParent()->getParent() : 0;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001309 SlotMachine SlotTable(F);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001310 AssemblyWriter W(o, SlotTable, F ? F->getParent() : 0, AAW);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001311
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001312 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001313}
Chris Lattner7db79582001-11-07 04:21:57 +00001314
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001315void Constant::print(std::ostream &o) const {
1316 if (this == 0) { o << "<null> constant value\n"; return; }
Chris Lattner7d734802002-09-10 15:53:49 +00001317
Misha Brukman21bbdb92004-06-04 21:11:51 +00001318 o << ' ' << getType()->getDescription() << ' ';
Evan Cheng5b19a802006-03-01 22:17:00 +00001319
1320 std::map<const Type *, std::string> TypeTable;
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001321 WriteConstantInt(o, this, TypeTable, 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001322}
1323
Misha Brukmanb1c93172005-04-21 23:48:37 +00001324void Type::print(std::ostream &o) const {
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001325 if (this == 0)
1326 o << "<null Type>";
1327 else
1328 o << getDescription();
1329}
1330
Chris Lattner2e9fa6d2002-04-09 19:48:49 +00001331void Argument::print(std::ostream &o) const {
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001332 WriteAsOperand(o, this, true, getParent() ? getParent()->getParent() : 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001333}
1334
Reid Spencer52641832004-05-25 18:14:38 +00001335// Value::dump - allow easy printing of Values from the debugger.
1336// Located here because so much of the needed functionality is here.
Bill Wendling22e978a2006-12-07 20:04:42 +00001337void Value::dump() const { print(*cerr.stream()); cerr << '\n'; }
Reid Spencer52641832004-05-25 18:14:38 +00001338
1339// Type::dump - allow easy printing of Values from the debugger.
1340// Located here because so much of the needed functionality is here.
Bill Wendling22e978a2006-12-07 20:04:42 +00001341void Type::dump() const { print(*cerr.stream()); cerr << '\n'; }
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001342
1343//===----------------------------------------------------------------------===//
Chris Lattnerfc9f1c92006-12-06 06:40:49 +00001344// SlotMachine Implementation
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001345//===----------------------------------------------------------------------===//
1346
1347#if 0
Bill Wendlingf3baad32006-12-07 01:30:32 +00001348#define SC_DEBUG(X) cerr << X
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001349#else
1350#define SC_DEBUG(X)
1351#endif
1352
1353// Module level constructor. Causes the contents of the Module (sans functions)
1354// to be added to the slot table.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001355SlotMachine::SlotMachine(const Module *M)
Reid Spencer56010e42004-05-26 21:56:09 +00001356 : TheModule(M) ///< Saved for lazy initialization.
1357 , TheFunction(0)
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001358 , FunctionProcessed(false)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001359{
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001360}
1361
1362// Function level constructor. Causes the contents of the Module and the one
1363// function provided to be added to the slot table.
Chris Lattner23e829a2006-12-06 05:12:21 +00001364SlotMachine::SlotMachine(const Function *F)
1365 : TheModule(F ? F->getParent() : 0) ///< Saved for lazy initialization
Reid Spencer56010e42004-05-26 21:56:09 +00001366 , TheFunction(F) ///< Saved for lazy initialization
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001367 , FunctionProcessed(false)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001368{
Reid Spencer56010e42004-05-26 21:56:09 +00001369}
1370
1371inline void SlotMachine::initialize(void) {
Chris Lattner23e829a2006-12-06 05:12:21 +00001372 if (TheModule) {
Misha Brukmanb1c93172005-04-21 23:48:37 +00001373 processModule();
Reid Spencer56010e42004-05-26 21:56:09 +00001374 TheModule = 0; ///< Prevent re-processing next time we're called.
1375 }
Chris Lattner193de902006-12-06 05:27:40 +00001376 if (TheFunction && !FunctionProcessed)
Misha Brukmanb1c93172005-04-21 23:48:37 +00001377 processFunction();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001378}
1379
1380// Iterate through all the global variables, functions, and global
Misha Brukmanb1c93172005-04-21 23:48:37 +00001381// variable initializers and create slots for them.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001382void SlotMachine::processModule() {
1383 SC_DEBUG("begin processModule!\n");
1384
Chris Lattner0dda8612006-12-06 06:15:43 +00001385 // Add all of the unnamed global variables to the value table.
Chris Lattner54932b02006-12-06 04:41:52 +00001386 for (Module::const_global_iterator I = TheModule->global_begin(),
1387 E = TheModule->global_end(); I != E; ++I)
Chris Lattner0dda8612006-12-06 06:15:43 +00001388 if (!I->hasName())
1389 getOrCreateSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001390
Chris Lattner0dda8612006-12-06 06:15:43 +00001391 // Add all the unnamed functions to the table.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001392 for (Module::const_iterator I = TheModule->begin(), E = TheModule->end();
1393 I != E; ++I)
Chris Lattner0dda8612006-12-06 06:15:43 +00001394 if (!I->hasName())
1395 getOrCreateSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001396
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001397 SC_DEBUG("end processModule!\n");
1398}
1399
1400
Reid Spencer56010e42004-05-26 21:56:09 +00001401// Process the arguments, basic blocks, and instructions of a function.
1402void SlotMachine::processFunction() {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001403 SC_DEBUG("begin processFunction!\n");
1404
Chris Lattner0dda8612006-12-06 06:15:43 +00001405 // Add all the function arguments with no names.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001406 for(Function::const_arg_iterator AI = TheFunction->arg_begin(),
Chris Lattner531f9e92005-03-15 04:54:21 +00001407 AE = TheFunction->arg_end(); AI != AE; ++AI)
Chris Lattner0dda8612006-12-06 06:15:43 +00001408 if (!AI->hasName())
1409 getOrCreateSlot(AI);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001410
1411 SC_DEBUG("Inserting Instructions:\n");
1412
Chris Lattner0dda8612006-12-06 06:15:43 +00001413 // Add all of the basic blocks and instructions with no names.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001414 for (Function::const_iterator BB = TheFunction->begin(),
Reid Spencer56010e42004-05-26 21:56:09 +00001415 E = TheFunction->end(); BB != E; ++BB) {
Chris Lattner0dda8612006-12-06 06:15:43 +00001416 if (!BB->hasName())
1417 getOrCreateSlot(BB);
1418 for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I != E; ++I)
1419 if (I->getType() != Type::VoidTy && !I->hasName())
1420 getOrCreateSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001421 }
1422
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001423 FunctionProcessed = true;
1424
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001425 SC_DEBUG("end processFunction!\n");
1426}
1427
Chris Lattner193de902006-12-06 05:27:40 +00001428/// Clean up after incorporating a function. This is the only way to get out of
1429/// the function incorporation state that affects the
1430/// getSlot/getOrCreateSlot lock. Function incorporation state is indicated
1431/// by TheFunction != 0.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001432void SlotMachine::purgeFunction() {
1433 SC_DEBUG("begin purgeFunction!\n");
1434 fMap.clear(); // Simply discard the function level map
Reid Spencer56010e42004-05-26 21:56:09 +00001435 TheFunction = 0;
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001436 FunctionProcessed = false;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001437 SC_DEBUG("end purgeFunction!\n");
1438}
1439
1440/// Get the slot number for a value. This function will assert if you
Chris Lattner193de902006-12-06 05:27:40 +00001441/// ask for a Value that hasn't previously been inserted with getOrCreateSlot.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001442/// Types are forbidden because Type does not inherit from Value (any more).
Chris Lattner757ee0b2004-06-09 19:41:19 +00001443int SlotMachine::getSlot(const Value *V) {
Chris Lattner23e829a2006-12-06 05:12:21 +00001444 assert(V && "Can't get slot for null Value");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001445 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1446 "Can't insert a non-GlobalValue Constant into SlotMachine");
Reid Spencer56010e42004-05-26 21:56:09 +00001447
1448 // Check for uninitialized state and do lazy initialization
1449 this->initialize();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001450
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001451 // Get the type of the value
1452 const Type* VTy = V->getType();
1453
1454 // Find the type plane in the module map
1455 TypedPlanes::const_iterator MI = mMap.find(VTy);
1456
Chris Lattner23e829a2006-12-06 05:12:21 +00001457 if (TheFunction) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001458 // Lookup the type in the function map too
1459 TypedPlanes::const_iterator FI = fMap.find(VTy);
1460 // If there is a corresponding type plane in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001461 if (FI != fMap.end()) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001462 // Lookup the Value in the function map
1463 ValueMap::const_iterator FVI = FI->second.map.find(V);
1464 // If the value doesn't exist in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001465 if (FVI == FI->second.map.end()) {
Chris Lattner68a038e2004-06-09 22:22:10 +00001466 // Look up the value in the module map.
1467 if (MI == mMap.end()) return -1;
Reid Spencer56010e42004-05-26 21:56:09 +00001468 ValueMap::const_iterator MVI = MI->second.map.find(V);
1469 // If we didn't find it, it wasn't inserted
Chris Lattner757ee0b2004-06-09 19:41:19 +00001470 if (MVI == MI->second.map.end()) return -1;
Chris Lattner23e829a2006-12-06 05:12:21 +00001471 assert(MVI != MI->second.map.end() && "Value not found");
Reid Spencer56010e42004-05-26 21:56:09 +00001472 // We found it only at the module level
Misha Brukmanb1c93172005-04-21 23:48:37 +00001473 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001474
1475 // else the value exists in the function map
1476 } else {
Reid Spencer56010e42004-05-26 21:56:09 +00001477 // Return the slot number as the module's contribution to
1478 // the type plane plus the index in the function's contribution
1479 // to the type plane.
Chris Lattnerb1f04782004-06-15 21:07:32 +00001480 if (MI != mMap.end())
1481 return MI->second.next_slot + FVI->second;
1482 else
1483 return FVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001484 }
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001485 }
1486 }
1487
Reid Spencer8beac692004-06-09 15:26:53 +00001488 // N.B. Can get here only if either !TheFunction or the function doesn't
1489 // have a corresponding type plane for the Value
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001490
1491 // Make sure the type plane exists
Chris Lattner757ee0b2004-06-09 19:41:19 +00001492 if (MI == mMap.end()) return -1;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001493 // Lookup the value in the module's map
1494 ValueMap::const_iterator MVI = MI->second.map.find(V);
1495 // Make sure we found it.
Chris Lattner757ee0b2004-06-09 19:41:19 +00001496 if (MVI == MI->second.map.end()) return -1;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001497 // Return it.
1498 return MVI->second;
1499}
1500
Reid Spencer58d30f22004-07-04 11:50:43 +00001501
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001502// Create a new slot, or return the existing slot if it is already
1503// inserted. Note that the logic here parallels getSlot but instead
1504// of asserting when the Value* isn't found, it inserts the value.
Chris Lattner193de902006-12-06 05:27:40 +00001505unsigned SlotMachine::getOrCreateSlot(const Value *V) {
Chris Lattner0dda8612006-12-06 06:15:43 +00001506 const Type* VTy = V->getType();
1507 assert(VTy != Type::VoidTy && !V->hasName() && "Doesn't need a slot!");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001508 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1509 "Can't insert a non-GlobalValue Constant into SlotMachine");
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001510
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001511 // Look up the type plane for the Value's type from the module map
1512 TypedPlanes::const_iterator MI = mMap.find(VTy);
1513
Chris Lattner23e829a2006-12-06 05:12:21 +00001514 if (TheFunction) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001515 // Get the type plane for the Value's type from the function map
1516 TypedPlanes::const_iterator FI = fMap.find(VTy);
1517 // If there is a corresponding type plane in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001518 if (FI != fMap.end()) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001519 // Lookup the Value in the function map
1520 ValueMap::const_iterator FVI = FI->second.map.find(V);
1521 // If the value doesn't exist in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001522 if (FVI == FI->second.map.end()) {
Reid Spencer56010e42004-05-26 21:56:09 +00001523 // If there is no corresponding type plane in the module map
Chris Lattner23e829a2006-12-06 05:12:21 +00001524 if (MI == mMap.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001525 return insertValue(V);
1526 // Look up the value in the module map
1527 ValueMap::const_iterator MVI = MI->second.map.find(V);
1528 // If we didn't find it, it wasn't inserted
Chris Lattner23e829a2006-12-06 05:12:21 +00001529 if (MVI == MI->second.map.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001530 return insertValue(V);
1531 else
1532 // We found it only at the module level
1533 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001534
1535 // else the value exists in the function map
1536 } else {
Chris Lattner23e829a2006-12-06 05:12:21 +00001537 if (MI == mMap.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001538 return FVI->second;
1539 else
1540 // Return the slot number as the module's contribution to
1541 // the type plane plus the index in the function's contribution
1542 // to the type plane.
1543 return MI->second.next_slot + FVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001544 }
1545
1546 // else there is not a corresponding type plane in the function map
1547 } else {
1548 // If the type plane doesn't exists at the module level
Chris Lattner23e829a2006-12-06 05:12:21 +00001549 if (MI == mMap.end()) {
Reid Spencer56010e42004-05-26 21:56:09 +00001550 return insertValue(V);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001551 // else type plane exists at the module level, examine it
1552 } else {
Reid Spencer56010e42004-05-26 21:56:09 +00001553 // Look up the value in the module's map
1554 ValueMap::const_iterator MVI = MI->second.map.find(V);
1555 // If we didn't find it there either
Chris Lattner23e829a2006-12-06 05:12:21 +00001556 if (MVI == MI->second.map.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001557 // Return the slot number as the module's contribution to
1558 // the type plane plus the index of the function map insertion.
1559 return MI->second.next_slot + insertValue(V);
1560 else
1561 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001562 }
1563 }
1564 }
1565
Chris Lattner0dda8612006-12-06 06:15:43 +00001566 // N.B. Can only get here if TheFunction == 0
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001567
1568 // If the module map's type plane is not for the Value's type
Chris Lattner23e829a2006-12-06 05:12:21 +00001569 if (MI != mMap.end()) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001570 // Lookup the value in the module's map
1571 ValueMap::const_iterator MVI = MI->second.map.find(V);
Chris Lattner23e829a2006-12-06 05:12:21 +00001572 if (MVI != MI->second.map.end())
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001573 return MVI->second;
1574 }
1575
1576 return insertValue(V);
1577}
1578
1579
1580// Low level insert function. Minimal checking is done. This
Chris Lattner193de902006-12-06 05:27:40 +00001581// function is just for the convenience of getOrCreateSlot (above).
Chris Lattner23e829a2006-12-06 05:12:21 +00001582unsigned SlotMachine::insertValue(const Value *V) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001583 assert(V && "Can't insert a null Value into SlotMachine!");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001584 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
Chris Lattner7e4e7662006-12-06 05:42:32 +00001585 "Can't insert a non-GlobalValue Constant into SlotMachine");
1586 assert(V->getType() != Type::VoidTy && !V->hasName());
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001587
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001588 const Type *VTy = V->getType();
1589 unsigned DestSlot = 0;
1590
Chris Lattner23e829a2006-12-06 05:12:21 +00001591 if (TheFunction) {
1592 TypedPlanes::iterator I = fMap.find(VTy);
1593 if (I == fMap.end())
Reid Spencer58d30f22004-07-04 11:50:43 +00001594 I = fMap.insert(std::make_pair(VTy,ValuePlane())).first;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001595 DestSlot = I->second.map[V] = I->second.next_slot++;
1596 } else {
Chris Lattner23e829a2006-12-06 05:12:21 +00001597 TypedPlanes::iterator I = mMap.find(VTy);
1598 if (I == mMap.end())
Reid Spencer58d30f22004-07-04 11:50:43 +00001599 I = mMap.insert(std::make_pair(VTy,ValuePlane())).first;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001600 DestSlot = I->second.map[V] = I->second.next_slot++;
1601 }
1602
Misha Brukmanb1c93172005-04-21 23:48:37 +00001603 SC_DEBUG(" Inserting value [" << VTy << "] = " << V << " slot=" <<
Reid Spencer56010e42004-05-26 21:56:09 +00001604 DestSlot << " [");
Chris Lattnerf8090cb2006-12-06 05:55:41 +00001605 // G = Global, F = Function, o = other
1606 SC_DEBUG((isa<GlobalVariable>(V) ? 'G' : (isa<Function>(V) ? 'F' : 'o')));
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001607 SC_DEBUG("]\n");
1608 return DestSlot;
1609}