blob: 268d3f5cfc27ab7a8f9f61eab398cdede399be47 [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 }
Chris Lattner379a8d22003-04-16 20:28:45 +0000924 else
925 switch (F->getLinkage()) {
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000926 case GlobalValue::InternalLinkage: Out << "internal "; break;
927 case GlobalValue::LinkOnceLinkage: Out << "linkonce "; break;
928 case GlobalValue::WeakLinkage: Out << "weak "; break;
929 case GlobalValue::AppendingLinkage: Out << "appending "; break;
930 case GlobalValue::DLLImportLinkage: Out << "dllimport "; break;
931 case GlobalValue::DLLExportLinkage: Out << "dllexport "; break;
932 case GlobalValue::ExternalWeakLinkage: Out << "extern_weak "; break;
Chris Lattner379a8d22003-04-16 20:28:45 +0000933 case GlobalValue::ExternalLinkage: break;
Misha Brukman70457632004-11-14 21:04:34 +0000934 case GlobalValue::GhostLinkage:
Bill Wendlingf3baad32006-12-07 01:30:32 +0000935 cerr << "GhostLinkage not allowed in AsmWriter!\n";
Misha Brukman70457632004-11-14 21:04:34 +0000936 abort();
Chris Lattner379a8d22003-04-16 20:28:45 +0000937 }
938
Chris Lattnerf7b6d312005-05-06 20:26:43 +0000939 // Print the calling convention.
940 switch (F->getCallingConv()) {
941 case CallingConv::C: break; // default
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +0000942 case CallingConv::CSRet: Out << "csretcc "; break;
943 case CallingConv::Fast: Out << "fastcc "; break;
944 case CallingConv::Cold: Out << "coldcc "; break;
945 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
946 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +0000947 default: Out << "cc" << F->getCallingConv() << " "; break;
948 }
949
Misha Brukman21bbdb92004-06-04 21:11:51 +0000950 printType(F->getReturnType()) << ' ';
Chris Lattner5b337482003-10-18 05:57:43 +0000951 if (!F->getName().empty())
Misha Brukmana6619a92004-06-21 21:53:56 +0000952 Out << getLLVMName(F->getName());
Chris Lattner5b337482003-10-18 05:57:43 +0000953 else
Misha Brukmana6619a92004-06-21 21:53:56 +0000954 Out << "\"\"";
955 Out << '(';
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000956 Machine.incorporateFunction(F);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000957
Chris Lattner7bfee412001-10-29 16:05:51 +0000958 // Loop over the arguments, printing them...
Chris Lattner113f4f42002-06-25 16:13:24 +0000959 const FunctionType *FT = F->getFunctionType();
Chris Lattnerfee714f2001-09-07 16:36:04 +0000960
Chris Lattner54932b02006-12-06 04:41:52 +0000961 for (Function::const_arg_iterator I = F->arg_begin(), E = F->arg_end();
962 I != E; ++I)
Chris Lattner149376d2002-10-13 20:57:00 +0000963 printArgument(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000964
965 // Finish printing arguments...
Chris Lattner113f4f42002-06-25 16:13:24 +0000966 if (FT->isVarArg()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000967 if (FT->getNumParams()) Out << ", ";
968 Out << "..."; // Output varargs portion of signature!
Chris Lattnerfee714f2001-09-07 16:36:04 +0000969 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000970 Out << ')';
Chris Lattnerfee714f2001-09-07 16:36:04 +0000971
Chris Lattner4b96c542005-11-12 00:10:19 +0000972 if (F->hasSection())
973 Out << " section \"" << F->getSection() << '"';
Chris Lattnerf8a974d2005-11-06 06:48:53 +0000974 if (F->getAlignment())
975 Out << " align " << F->getAlignment();
Chris Lattner4b96c542005-11-12 00:10:19 +0000976
Chris Lattner113f4f42002-06-25 16:13:24 +0000977 if (F->isExternal()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000978 Out << "\n";
Chris Lattnerb2f02e52002-05-06 03:00:40 +0000979 } else {
Misha Brukmana6619a92004-06-21 21:53:56 +0000980 Out << " {";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000981
Chris Lattner6915f8f2002-04-07 22:49:37 +0000982 // Output all of its basic blocks... for the function
Chris Lattner113f4f42002-06-25 16:13:24 +0000983 for (Function::const_iterator I = F->begin(), E = F->end(); I != E; ++I)
984 printBasicBlock(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000985
Misha Brukmana6619a92004-06-21 21:53:56 +0000986 Out << "}\n";
Chris Lattnerfee714f2001-09-07 16:36:04 +0000987 }
988
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000989 Machine.purgeFunction();
Chris Lattner2f7c9632001-06-06 20:29:01 +0000990}
991
Misha Brukmanc566ca362004-03-02 00:22:19 +0000992/// printArgument - This member is called for every argument that is passed into
993/// the function. Simply print it out
994///
Chris Lattner2e9fa6d2002-04-09 19:48:49 +0000995void AssemblyWriter::printArgument(const Argument *Arg) {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000996 // Insert commas as we go... the first arg doesn't get a comma
Chris Lattnerfb972ad2005-03-15 05:03:36 +0000997 if (Arg != Arg->getParent()->arg_begin()) Out << ", ";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000998
999 // Output type...
Chris Lattner7bfee412001-10-29 16:05:51 +00001000 printType(Arg->getType());
Misha Brukmanb1c93172005-04-21 23:48:37 +00001001
Chris Lattner2f7c9632001-06-06 20:29:01 +00001002 // Output name, if available...
1003 if (Arg->hasName())
Misha Brukmana6619a92004-06-21 21:53:56 +00001004 Out << ' ' << getLLVMName(Arg->getName());
Chris Lattner2f7c9632001-06-06 20:29:01 +00001005}
1006
Misha Brukmanc566ca362004-03-02 00:22:19 +00001007/// printBasicBlock - This member is called for each basic block in a method.
1008///
Chris Lattner7bfee412001-10-29 16:05:51 +00001009void AssemblyWriter::printBasicBlock(const BasicBlock *BB) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001010 if (BB->hasName()) { // Print out the label if it exists...
Alkis Evlogimenos991d6ad2004-12-10 05:41:10 +00001011 Out << "\n" << getLLVMName(BB->getName(), false) << ':';
Chris Lattner408dbdb2002-05-14 16:02:05 +00001012 } else if (!BB->use_empty()) { // Don't print block # of no uses...
Misha Brukmana6619a92004-06-21 21:53:56 +00001013 Out << "\n; <label>:";
Chris Lattner757ee0b2004-06-09 19:41:19 +00001014 int Slot = Machine.getSlot(BB);
1015 if (Slot != -1)
Misha Brukmana6619a92004-06-21 21:53:56 +00001016 Out << Slot;
Chris Lattner757ee0b2004-06-09 19:41:19 +00001017 else
Misha Brukmana6619a92004-06-21 21:53:56 +00001018 Out << "<badref>";
Chris Lattner58185f22002-10-02 19:38:55 +00001019 }
Chris Lattner2447ef52003-11-20 00:09:43 +00001020
1021 if (BB->getParent() == 0)
Misha Brukmana6619a92004-06-21 21:53:56 +00001022 Out << "\t\t; Error: Block without parent!";
Chris Lattner2447ef52003-11-20 00:09:43 +00001023 else {
1024 if (BB != &BB->getParent()->front()) { // Not the entry block?
1025 // Output predecessors for the block...
Misha Brukmana6619a92004-06-21 21:53:56 +00001026 Out << "\t\t;";
Chris Lattner2447ef52003-11-20 00:09:43 +00001027 pred_const_iterator PI = pred_begin(BB), PE = pred_end(BB);
Misha Brukmanb1c93172005-04-21 23:48:37 +00001028
Chris Lattner2447ef52003-11-20 00:09:43 +00001029 if (PI == PE) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001030 Out << " No predecessors!";
Chris Lattner2447ef52003-11-20 00:09:43 +00001031 } else {
Misha Brukmana6619a92004-06-21 21:53:56 +00001032 Out << " preds =";
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001033 writeOperand(*PI, false);
Chris Lattner2447ef52003-11-20 00:09:43 +00001034 for (++PI; PI != PE; ++PI) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001035 Out << ',';
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001036 writeOperand(*PI, false);
Chris Lattner2447ef52003-11-20 00:09:43 +00001037 }
Chris Lattner00211f12003-11-16 22:59:57 +00001038 }
Chris Lattner58185f22002-10-02 19:38:55 +00001039 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001040 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001041
Misha Brukmana6619a92004-06-21 21:53:56 +00001042 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001043
Misha Brukmana6619a92004-06-21 21:53:56 +00001044 if (AnnotationWriter) AnnotationWriter->emitBasicBlockStartAnnot(BB, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001045
Chris Lattnerfee714f2001-09-07 16:36:04 +00001046 // Output all of the instructions in the basic block...
Chris Lattner113f4f42002-06-25 16:13:24 +00001047 for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I != E; ++I)
1048 printInstruction(*I);
Chris Lattner96cdd272004-03-08 18:51:45 +00001049
Misha Brukmana6619a92004-06-21 21:53:56 +00001050 if (AnnotationWriter) AnnotationWriter->emitBasicBlockEndAnnot(BB, Out);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001051}
1052
Chris Lattner862e3382001-10-13 06:42:36 +00001053
Misha Brukmanc566ca362004-03-02 00:22:19 +00001054/// printInfoComment - Print a little comment after the instruction indicating
1055/// which slot it occupies.
1056///
Chris Lattner113f4f42002-06-25 16:13:24 +00001057void AssemblyWriter::printInfoComment(const Value &V) {
1058 if (V.getType() != Type::VoidTy) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001059 Out << "\t\t; <";
Misha Brukman21bbdb92004-06-04 21:11:51 +00001060 printType(V.getType()) << '>';
Chris Lattner862e3382001-10-13 06:42:36 +00001061
Chris Lattner113f4f42002-06-25 16:13:24 +00001062 if (!V.hasName()) {
Chris Lattner757ee0b2004-06-09 19:41:19 +00001063 int SlotNum = Machine.getSlot(&V);
1064 if (SlotNum == -1)
Misha Brukmana6619a92004-06-21 21:53:56 +00001065 Out << ":<badref>";
Reid Spencer8beac692004-06-09 15:26:53 +00001066 else
Misha Brukmana6619a92004-06-21 21:53:56 +00001067 Out << ':' << SlotNum; // Print out the def slot taken.
Chris Lattner862e3382001-10-13 06:42:36 +00001068 }
Chris Lattnerb6c21db2005-02-01 01:24:01 +00001069 Out << " [#uses=" << V.getNumUses() << ']'; // Output # uses
Chris Lattner862e3382001-10-13 06:42:36 +00001070 }
1071}
1072
Reid Spencere7141c82006-08-28 01:02:49 +00001073// This member is called for each Instruction in a function..
Chris Lattner113f4f42002-06-25 16:13:24 +00001074void AssemblyWriter::printInstruction(const Instruction &I) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001075 if (AnnotationWriter) AnnotationWriter->emitInstructionAnnot(&I, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001076
Misha Brukmana6619a92004-06-21 21:53:56 +00001077 Out << "\t";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001078
1079 // Print out name if it exists...
Chris Lattner113f4f42002-06-25 16:13:24 +00001080 if (I.hasName())
Misha Brukmana6619a92004-06-21 21:53:56 +00001081 Out << getLLVMName(I.getName()) << " = ";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001082
Chris Lattner06038452005-05-06 05:51:46 +00001083 // If this is a volatile load or store, print out the volatile marker.
Chris Lattner504f9242003-09-08 17:45:59 +00001084 if ((isa<LoadInst>(I) && cast<LoadInst>(I).isVolatile()) ||
Chris Lattner06038452005-05-06 05:51:46 +00001085 (isa<StoreInst>(I) && cast<StoreInst>(I).isVolatile())) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001086 Out << "volatile ";
Chris Lattner06038452005-05-06 05:51:46 +00001087 } else if (isa<CallInst>(I) && cast<CallInst>(I).isTailCall()) {
1088 // If this is a call, check if it's a tail call.
1089 Out << "tail ";
1090 }
Chris Lattner504f9242003-09-08 17:45:59 +00001091
Chris Lattner2f7c9632001-06-06 20:29:01 +00001092 // Print out the opcode...
Misha Brukmana6619a92004-06-21 21:53:56 +00001093 Out << I.getOpcodeName();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001094
Reid Spencer45e52392006-12-03 06:27:29 +00001095 // Print out the compare instruction predicates
1096 if (const FCmpInst *FCI = dyn_cast<FCmpInst>(&I)) {
Reid Spencer812a1be2006-12-04 05:19:18 +00001097 Out << " " << getPredicateText(FCI->getPredicate());
Reid Spencer45e52392006-12-03 06:27:29 +00001098 } else if (const ICmpInst *ICI = dyn_cast<ICmpInst>(&I)) {
Reid Spencer812a1be2006-12-04 05:19:18 +00001099 Out << " " << getPredicateText(ICI->getPredicate());
Reid Spencer45e52392006-12-03 06:27:29 +00001100 }
1101
Chris Lattner2f7c9632001-06-06 20:29:01 +00001102 // Print out the type of the operands...
Chris Lattner113f4f42002-06-25 16:13:24 +00001103 const Value *Operand = I.getNumOperands() ? I.getOperand(0) : 0;
Chris Lattner2f7c9632001-06-06 20:29:01 +00001104
1105 // Special case conditional branches to swizzle the condition out to the front
Chris Lattner113f4f42002-06-25 16:13:24 +00001106 if (isa<BranchInst>(I) && I.getNumOperands() > 1) {
1107 writeOperand(I.getOperand(2), true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001108 Out << ',';
Chris Lattner2f7c9632001-06-06 20:29:01 +00001109 writeOperand(Operand, true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001110 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001111 writeOperand(I.getOperand(1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001112
Chris Lattner8d48df22002-04-13 18:34:38 +00001113 } else if (isa<SwitchInst>(I)) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001114 // Special case switch statement to get formatting nice and correct...
Misha Brukmana6619a92004-06-21 21:53:56 +00001115 writeOperand(Operand , true); Out << ',';
1116 writeOperand(I.getOperand(1), true); Out << " [";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001117
Chris Lattner113f4f42002-06-25 16:13:24 +00001118 for (unsigned op = 2, Eop = I.getNumOperands(); op < Eop; op += 2) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001119 Out << "\n\t\t";
1120 writeOperand(I.getOperand(op ), true); Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001121 writeOperand(I.getOperand(op+1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001122 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001123 Out << "\n\t]";
Chris Lattnerda558102001-10-02 03:41:24 +00001124 } else if (isa<PHINode>(I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001125 Out << ' ';
Chris Lattner113f4f42002-06-25 16:13:24 +00001126 printType(I.getType());
Misha Brukmana6619a92004-06-21 21:53:56 +00001127 Out << ' ';
Chris Lattner2f7c9632001-06-06 20:29:01 +00001128
Chris Lattner113f4f42002-06-25 16:13:24 +00001129 for (unsigned op = 0, Eop = I.getNumOperands(); op < Eop; op += 2) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001130 if (op) Out << ", ";
Misha Brukmanb1c93172005-04-21 23:48:37 +00001131 Out << '[';
Misha Brukmana6619a92004-06-21 21:53:56 +00001132 writeOperand(I.getOperand(op ), false); Out << ',';
1133 writeOperand(I.getOperand(op+1), false); Out << " ]";
Chris Lattner931ef3b2001-06-11 15:04:20 +00001134 }
Chris Lattner862e3382001-10-13 06:42:36 +00001135 } else if (isa<ReturnInst>(I) && !Operand) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001136 Out << " void";
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001137 } else if (const CallInst *CI = dyn_cast<CallInst>(&I)) {
1138 // Print the calling convention being used.
1139 switch (CI->getCallingConv()) {
1140 case CallingConv::C: break; // default
Chris Lattner29d20852006-05-19 21:58:52 +00001141 case CallingConv::CSRet: Out << " csretcc"; break;
1142 case CallingConv::Fast: Out << " fastcc"; break;
1143 case CallingConv::Cold: Out << " coldcc"; break;
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +00001144 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
1145 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001146 default: Out << " cc" << CI->getCallingConv(); break;
1147 }
1148
Chris Lattner463d6a52003-08-05 15:34:45 +00001149 const PointerType *PTy = cast<PointerType>(Operand->getType());
1150 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1151 const Type *RetTy = FTy->getReturnType();
Chris Lattner2f2d9472001-11-06 21:28:12 +00001152
Chris Lattner463d6a52003-08-05 15:34:45 +00001153 // If possible, print out the short form of the call instruction. We can
Chris Lattner6915f8f2002-04-07 22:49:37 +00001154 // only do this if the first argument is a pointer to a nonvararg function,
Chris Lattner463d6a52003-08-05 15:34:45 +00001155 // and if the return type is not a pointer to a function.
Chris Lattner2f2d9472001-11-06 21:28:12 +00001156 //
Chris Lattner463d6a52003-08-05 15:34:45 +00001157 if (!FTy->isVarArg() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +00001158 (!isa<PointerType>(RetTy) ||
Chris Lattnerd9a36a62002-07-25 20:58:51 +00001159 !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001160 Out << ' '; printType(RetTy);
Chris Lattner2f2d9472001-11-06 21:28:12 +00001161 writeOperand(Operand, false);
1162 } else {
1163 writeOperand(Operand, true);
1164 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001165 Out << '(';
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001166 if (CI->getNumOperands() > 1) writeOperand(CI->getOperand(1), true);
Chris Lattner113f4f42002-06-25 16:13:24 +00001167 for (unsigned op = 2, Eop = I.getNumOperands(); op < Eop; ++op) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001168 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001169 writeOperand(I.getOperand(op), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001170 }
1171
Misha Brukmana6619a92004-06-21 21:53:56 +00001172 Out << " )";
Chris Lattner113f4f42002-06-25 16:13:24 +00001173 } else if (const InvokeInst *II = dyn_cast<InvokeInst>(&I)) {
Chris Lattner463d6a52003-08-05 15:34:45 +00001174 const PointerType *PTy = cast<PointerType>(Operand->getType());
1175 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1176 const Type *RetTy = FTy->getReturnType();
1177
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001178 // Print the calling convention being used.
1179 switch (II->getCallingConv()) {
1180 case CallingConv::C: break; // default
Chris Lattner29d20852006-05-19 21:58:52 +00001181 case CallingConv::CSRet: Out << " csretcc"; break;
1182 case CallingConv::Fast: Out << " fastcc"; break;
1183 case CallingConv::Cold: Out << " coldcc"; break;
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +00001184 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
1185 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001186 default: Out << " cc" << II->getCallingConv(); break;
1187 }
1188
Chris Lattner463d6a52003-08-05 15:34:45 +00001189 // If possible, print out the short form of the invoke instruction. We can
1190 // only do this if the first argument is a pointer to a nonvararg function,
1191 // and if the return type is not a pointer to a function.
1192 //
1193 if (!FTy->isVarArg() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +00001194 (!isa<PointerType>(RetTy) ||
Chris Lattner463d6a52003-08-05 15:34:45 +00001195 !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001196 Out << ' '; printType(RetTy);
Chris Lattner463d6a52003-08-05 15:34:45 +00001197 writeOperand(Operand, false);
1198 } else {
1199 writeOperand(Operand, true);
1200 }
1201
Misha Brukmana6619a92004-06-21 21:53:56 +00001202 Out << '(';
Chris Lattner113f4f42002-06-25 16:13:24 +00001203 if (I.getNumOperands() > 3) writeOperand(I.getOperand(3), true);
1204 for (unsigned op = 4, Eop = I.getNumOperands(); op < Eop; ++op) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001205 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001206 writeOperand(I.getOperand(op), true);
Chris Lattner862e3382001-10-13 06:42:36 +00001207 }
1208
Misha Brukmana6619a92004-06-21 21:53:56 +00001209 Out << " )\n\t\t\tto";
Chris Lattner862e3382001-10-13 06:42:36 +00001210 writeOperand(II->getNormalDest(), true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001211 Out << " unwind";
Chris Lattnerfae8ab32004-02-08 21:44:31 +00001212 writeOperand(II->getUnwindDest(), true);
Chris Lattner862e3382001-10-13 06:42:36 +00001213
Chris Lattner113f4f42002-06-25 16:13:24 +00001214 } else if (const AllocationInst *AI = dyn_cast<AllocationInst>(&I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001215 Out << ' ';
Chris Lattner8d48df22002-04-13 18:34:38 +00001216 printType(AI->getType()->getElementType());
1217 if (AI->isArrayAllocation()) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001218 Out << ',';
Chris Lattner8d48df22002-04-13 18:34:38 +00001219 writeOperand(AI->getArraySize(), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001220 }
Nate Begeman848622f2005-11-05 09:21:28 +00001221 if (AI->getAlignment()) {
Chris Lattner7aeee3a2005-11-05 21:20:34 +00001222 Out << ", align " << AI->getAlignment();
Nate Begeman848622f2005-11-05 09:21:28 +00001223 }
Chris Lattner862e3382001-10-13 06:42:36 +00001224 } else if (isa<CastInst>(I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001225 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001226 Out << " to ";
Chris Lattner113f4f42002-06-25 16:13:24 +00001227 printType(I.getType());
Chris Lattner5b337482003-10-18 05:57:43 +00001228 } else if (isa<VAArgInst>(I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001229 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001230 Out << ", ";
Chris Lattnerf70da102003-05-08 02:44:12 +00001231 printType(I.getType());
Chris Lattner2f7c9632001-06-06 20:29:01 +00001232 } else if (Operand) { // Print the normal way...
1233
Misha Brukmanb1c93172005-04-21 23:48:37 +00001234 // PrintAllTypes - Instructions who have operands of all the same type
Chris Lattner2f7c9632001-06-06 20:29:01 +00001235 // omit the type from all but the first operand. If the instruction has
1236 // different type operands (for example br), then they are all printed.
1237 bool PrintAllTypes = false;
1238 const Type *TheType = Operand->getType();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001239
Chris Lattner52bd5cb92004-03-12 05:53:14 +00001240 // Shift Left & Right print both types even for Ubyte LHS, and select prints
1241 // types even if all operands are bools.
Chris Lattnerbbe0a422006-04-08 01:18:18 +00001242 if (isa<ShiftInst>(I) || isa<SelectInst>(I) || isa<StoreInst>(I) ||
1243 isa<ShuffleVectorInst>(I)) {
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001244 PrintAllTypes = true;
1245 } else {
1246 for (unsigned i = 1, E = I.getNumOperands(); i != E; ++i) {
1247 Operand = I.getOperand(i);
1248 if (Operand->getType() != TheType) {
1249 PrintAllTypes = true; // We have differing types! Print them all!
1250 break;
1251 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001252 }
1253 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001254
Chris Lattner7bfee412001-10-29 16:05:51 +00001255 if (!PrintAllTypes) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001256 Out << ' ';
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001257 printType(TheType);
Chris Lattner7bfee412001-10-29 16:05:51 +00001258 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001259
Chris Lattner113f4f42002-06-25 16:13:24 +00001260 for (unsigned i = 0, E = I.getNumOperands(); i != E; ++i) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001261 if (i) Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001262 writeOperand(I.getOperand(i), PrintAllTypes);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001263 }
1264 }
1265
Chris Lattner862e3382001-10-13 06:42:36 +00001266 printInfoComment(I);
Misha Brukmana6619a92004-06-21 21:53:56 +00001267 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001268}
1269
1270
1271//===----------------------------------------------------------------------===//
1272// External Interface declarations
1273//===----------------------------------------------------------------------===//
1274
Chris Lattner8339f7d2003-10-30 23:41:03 +00001275void Module::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001276 SlotMachine SlotTable(this);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001277 AssemblyWriter W(o, SlotTable, this, AAW);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001278 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001279}
1280
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001281void GlobalVariable::print(std::ostream &o) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001282 SlotMachine SlotTable(getParent());
Chris Lattner8339f7d2003-10-30 23:41:03 +00001283 AssemblyWriter W(o, SlotTable, getParent(), 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001284 W.write(this);
Chris Lattneradfe0d12001-09-10 20:08:19 +00001285}
1286
Chris Lattner8339f7d2003-10-30 23:41:03 +00001287void Function::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001288 SlotMachine SlotTable(getParent());
Chris Lattner8339f7d2003-10-30 23:41:03 +00001289 AssemblyWriter W(o, SlotTable, getParent(), AAW);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001290
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001291 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001292}
1293
Chris Lattnereef2fe72006-01-24 04:13:11 +00001294void InlineAsm::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001295 WriteAsOperand(o, this, true, 0);
Chris Lattnereef2fe72006-01-24 04:13:11 +00001296}
1297
Chris Lattner8339f7d2003-10-30 23:41:03 +00001298void BasicBlock::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001299 SlotMachine SlotTable(getParent());
Misha Brukmanb1c93172005-04-21 23:48:37 +00001300 AssemblyWriter W(o, SlotTable,
Chris Lattner8339f7d2003-10-30 23:41:03 +00001301 getParent() ? getParent()->getParent() : 0, AAW);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001302 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001303}
1304
Chris Lattner8339f7d2003-10-30 23:41:03 +00001305void Instruction::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001306 const Function *F = getParent() ? getParent()->getParent() : 0;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001307 SlotMachine SlotTable(F);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001308 AssemblyWriter W(o, SlotTable, F ? F->getParent() : 0, AAW);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001309
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001310 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001311}
Chris Lattner7db79582001-11-07 04:21:57 +00001312
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001313void Constant::print(std::ostream &o) const {
1314 if (this == 0) { o << "<null> constant value\n"; return; }
Chris Lattner7d734802002-09-10 15:53:49 +00001315
Misha Brukman21bbdb92004-06-04 21:11:51 +00001316 o << ' ' << getType()->getDescription() << ' ';
Evan Cheng5b19a802006-03-01 22:17:00 +00001317
1318 std::map<const Type *, std::string> TypeTable;
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001319 WriteConstantInt(o, this, TypeTable, 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001320}
1321
Misha Brukmanb1c93172005-04-21 23:48:37 +00001322void Type::print(std::ostream &o) const {
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001323 if (this == 0)
1324 o << "<null Type>";
1325 else
1326 o << getDescription();
1327}
1328
Chris Lattner2e9fa6d2002-04-09 19:48:49 +00001329void Argument::print(std::ostream &o) const {
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001330 WriteAsOperand(o, this, true, getParent() ? getParent()->getParent() : 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001331}
1332
Reid Spencer52641832004-05-25 18:14:38 +00001333// Value::dump - allow easy printing of Values from the debugger.
1334// Located here because so much of the needed functionality is here.
Bill Wendling22e978a2006-12-07 20:04:42 +00001335void Value::dump() const { print(*cerr.stream()); cerr << '\n'; }
Reid Spencer52641832004-05-25 18:14:38 +00001336
1337// Type::dump - allow easy printing of Values from the debugger.
1338// Located here because so much of the needed functionality is here.
Bill Wendling22e978a2006-12-07 20:04:42 +00001339void Type::dump() const { print(*cerr.stream()); cerr << '\n'; }
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001340
1341//===----------------------------------------------------------------------===//
Chris Lattnerfc9f1c92006-12-06 06:40:49 +00001342// SlotMachine Implementation
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001343//===----------------------------------------------------------------------===//
1344
1345#if 0
Bill Wendlingf3baad32006-12-07 01:30:32 +00001346#define SC_DEBUG(X) cerr << X
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001347#else
1348#define SC_DEBUG(X)
1349#endif
1350
1351// Module level constructor. Causes the contents of the Module (sans functions)
1352// to be added to the slot table.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001353SlotMachine::SlotMachine(const Module *M)
Reid Spencer56010e42004-05-26 21:56:09 +00001354 : TheModule(M) ///< Saved for lazy initialization.
1355 , TheFunction(0)
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001356 , FunctionProcessed(false)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001357{
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001358}
1359
1360// Function level constructor. Causes the contents of the Module and the one
1361// function provided to be added to the slot table.
Chris Lattner23e829a2006-12-06 05:12:21 +00001362SlotMachine::SlotMachine(const Function *F)
1363 : TheModule(F ? F->getParent() : 0) ///< Saved for lazy initialization
Reid Spencer56010e42004-05-26 21:56:09 +00001364 , TheFunction(F) ///< Saved for lazy initialization
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001365 , FunctionProcessed(false)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001366{
Reid Spencer56010e42004-05-26 21:56:09 +00001367}
1368
1369inline void SlotMachine::initialize(void) {
Chris Lattner23e829a2006-12-06 05:12:21 +00001370 if (TheModule) {
Misha Brukmanb1c93172005-04-21 23:48:37 +00001371 processModule();
Reid Spencer56010e42004-05-26 21:56:09 +00001372 TheModule = 0; ///< Prevent re-processing next time we're called.
1373 }
Chris Lattner193de902006-12-06 05:27:40 +00001374 if (TheFunction && !FunctionProcessed)
Misha Brukmanb1c93172005-04-21 23:48:37 +00001375 processFunction();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001376}
1377
1378// Iterate through all the global variables, functions, and global
Misha Brukmanb1c93172005-04-21 23:48:37 +00001379// variable initializers and create slots for them.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001380void SlotMachine::processModule() {
1381 SC_DEBUG("begin processModule!\n");
1382
Chris Lattner0dda8612006-12-06 06:15:43 +00001383 // Add all of the unnamed global variables to the value table.
Chris Lattner54932b02006-12-06 04:41:52 +00001384 for (Module::const_global_iterator I = TheModule->global_begin(),
1385 E = TheModule->global_end(); I != E; ++I)
Chris Lattner0dda8612006-12-06 06:15:43 +00001386 if (!I->hasName())
1387 getOrCreateSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001388
Chris Lattner0dda8612006-12-06 06:15:43 +00001389 // Add all the unnamed functions to the table.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001390 for (Module::const_iterator I = TheModule->begin(), E = TheModule->end();
1391 I != E; ++I)
Chris Lattner0dda8612006-12-06 06:15:43 +00001392 if (!I->hasName())
1393 getOrCreateSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001394
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001395 SC_DEBUG("end processModule!\n");
1396}
1397
1398
Reid Spencer56010e42004-05-26 21:56:09 +00001399// Process the arguments, basic blocks, and instructions of a function.
1400void SlotMachine::processFunction() {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001401 SC_DEBUG("begin processFunction!\n");
1402
Chris Lattner0dda8612006-12-06 06:15:43 +00001403 // Add all the function arguments with no names.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001404 for(Function::const_arg_iterator AI = TheFunction->arg_begin(),
Chris Lattner531f9e92005-03-15 04:54:21 +00001405 AE = TheFunction->arg_end(); AI != AE; ++AI)
Chris Lattner0dda8612006-12-06 06:15:43 +00001406 if (!AI->hasName())
1407 getOrCreateSlot(AI);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001408
1409 SC_DEBUG("Inserting Instructions:\n");
1410
Chris Lattner0dda8612006-12-06 06:15:43 +00001411 // Add all of the basic blocks and instructions with no names.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001412 for (Function::const_iterator BB = TheFunction->begin(),
Reid Spencer56010e42004-05-26 21:56:09 +00001413 E = TheFunction->end(); BB != E; ++BB) {
Chris Lattner0dda8612006-12-06 06:15:43 +00001414 if (!BB->hasName())
1415 getOrCreateSlot(BB);
1416 for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I != E; ++I)
1417 if (I->getType() != Type::VoidTy && !I->hasName())
1418 getOrCreateSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001419 }
1420
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001421 FunctionProcessed = true;
1422
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001423 SC_DEBUG("end processFunction!\n");
1424}
1425
Chris Lattner193de902006-12-06 05:27:40 +00001426/// Clean up after incorporating a function. This is the only way to get out of
1427/// the function incorporation state that affects the
1428/// getSlot/getOrCreateSlot lock. Function incorporation state is indicated
1429/// by TheFunction != 0.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001430void SlotMachine::purgeFunction() {
1431 SC_DEBUG("begin purgeFunction!\n");
1432 fMap.clear(); // Simply discard the function level map
Reid Spencer56010e42004-05-26 21:56:09 +00001433 TheFunction = 0;
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001434 FunctionProcessed = false;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001435 SC_DEBUG("end purgeFunction!\n");
1436}
1437
1438/// Get the slot number for a value. This function will assert if you
Chris Lattner193de902006-12-06 05:27:40 +00001439/// ask for a Value that hasn't previously been inserted with getOrCreateSlot.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001440/// Types are forbidden because Type does not inherit from Value (any more).
Chris Lattner757ee0b2004-06-09 19:41:19 +00001441int SlotMachine::getSlot(const Value *V) {
Chris Lattner23e829a2006-12-06 05:12:21 +00001442 assert(V && "Can't get slot for null Value");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001443 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1444 "Can't insert a non-GlobalValue Constant into SlotMachine");
Reid Spencer56010e42004-05-26 21:56:09 +00001445
1446 // Check for uninitialized state and do lazy initialization
1447 this->initialize();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001448
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001449 // Get the type of the value
1450 const Type* VTy = V->getType();
1451
1452 // Find the type plane in the module map
1453 TypedPlanes::const_iterator MI = mMap.find(VTy);
1454
Chris Lattner23e829a2006-12-06 05:12:21 +00001455 if (TheFunction) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001456 // Lookup the type in the function map too
1457 TypedPlanes::const_iterator FI = fMap.find(VTy);
1458 // If there is a corresponding type plane in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001459 if (FI != fMap.end()) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001460 // Lookup the Value in the function map
1461 ValueMap::const_iterator FVI = FI->second.map.find(V);
1462 // If the value doesn't exist in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001463 if (FVI == FI->second.map.end()) {
Chris Lattner68a038e2004-06-09 22:22:10 +00001464 // Look up the value in the module map.
1465 if (MI == mMap.end()) return -1;
Reid Spencer56010e42004-05-26 21:56:09 +00001466 ValueMap::const_iterator MVI = MI->second.map.find(V);
1467 // If we didn't find it, it wasn't inserted
Chris Lattner757ee0b2004-06-09 19:41:19 +00001468 if (MVI == MI->second.map.end()) return -1;
Chris Lattner23e829a2006-12-06 05:12:21 +00001469 assert(MVI != MI->second.map.end() && "Value not found");
Reid Spencer56010e42004-05-26 21:56:09 +00001470 // We found it only at the module level
Misha Brukmanb1c93172005-04-21 23:48:37 +00001471 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001472
1473 // else the value exists in the function map
1474 } else {
Reid Spencer56010e42004-05-26 21:56:09 +00001475 // Return the slot number as the module's contribution to
1476 // the type plane plus the index in the function's contribution
1477 // to the type plane.
Chris Lattnerb1f04782004-06-15 21:07:32 +00001478 if (MI != mMap.end())
1479 return MI->second.next_slot + FVI->second;
1480 else
1481 return FVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001482 }
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001483 }
1484 }
1485
Reid Spencer8beac692004-06-09 15:26:53 +00001486 // N.B. Can get here only if either !TheFunction or the function doesn't
1487 // have a corresponding type plane for the Value
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001488
1489 // Make sure the type plane exists
Chris Lattner757ee0b2004-06-09 19:41:19 +00001490 if (MI == mMap.end()) return -1;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001491 // Lookup the value in the module's map
1492 ValueMap::const_iterator MVI = MI->second.map.find(V);
1493 // Make sure we found it.
Chris Lattner757ee0b2004-06-09 19:41:19 +00001494 if (MVI == MI->second.map.end()) return -1;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001495 // Return it.
1496 return MVI->second;
1497}
1498
Reid Spencer58d30f22004-07-04 11:50:43 +00001499
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001500// Create a new slot, or return the existing slot if it is already
1501// inserted. Note that the logic here parallels getSlot but instead
1502// of asserting when the Value* isn't found, it inserts the value.
Chris Lattner193de902006-12-06 05:27:40 +00001503unsigned SlotMachine::getOrCreateSlot(const Value *V) {
Chris Lattner0dda8612006-12-06 06:15:43 +00001504 const Type* VTy = V->getType();
1505 assert(VTy != Type::VoidTy && !V->hasName() && "Doesn't need a slot!");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001506 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1507 "Can't insert a non-GlobalValue Constant into SlotMachine");
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001508
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001509 // Look up the type plane for the Value's type from the module map
1510 TypedPlanes::const_iterator MI = mMap.find(VTy);
1511
Chris Lattner23e829a2006-12-06 05:12:21 +00001512 if (TheFunction) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001513 // Get the type plane for the Value's type from the function map
1514 TypedPlanes::const_iterator FI = fMap.find(VTy);
1515 // If there is a corresponding type plane in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001516 if (FI != fMap.end()) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001517 // Lookup the Value in the function map
1518 ValueMap::const_iterator FVI = FI->second.map.find(V);
1519 // If the value doesn't exist in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001520 if (FVI == FI->second.map.end()) {
Reid Spencer56010e42004-05-26 21:56:09 +00001521 // If there is no corresponding type plane in the module map
Chris Lattner23e829a2006-12-06 05:12:21 +00001522 if (MI == mMap.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001523 return insertValue(V);
1524 // Look up the value in the module map
1525 ValueMap::const_iterator MVI = MI->second.map.find(V);
1526 // If we didn't find it, it wasn't inserted
Chris Lattner23e829a2006-12-06 05:12:21 +00001527 if (MVI == MI->second.map.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001528 return insertValue(V);
1529 else
1530 // We found it only at the module level
1531 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001532
1533 // else the value exists in the function map
1534 } else {
Chris Lattner23e829a2006-12-06 05:12:21 +00001535 if (MI == mMap.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001536 return FVI->second;
1537 else
1538 // Return the slot number as the module's contribution to
1539 // the type plane plus the index in the function's contribution
1540 // to the type plane.
1541 return MI->second.next_slot + FVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001542 }
1543
1544 // else there is not a corresponding type plane in the function map
1545 } else {
1546 // If the type plane doesn't exists at the module level
Chris Lattner23e829a2006-12-06 05:12:21 +00001547 if (MI == mMap.end()) {
Reid Spencer56010e42004-05-26 21:56:09 +00001548 return insertValue(V);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001549 // else type plane exists at the module level, examine it
1550 } else {
Reid Spencer56010e42004-05-26 21:56:09 +00001551 // Look up the value in the module's map
1552 ValueMap::const_iterator MVI = MI->second.map.find(V);
1553 // If we didn't find it there either
Chris Lattner23e829a2006-12-06 05:12:21 +00001554 if (MVI == MI->second.map.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001555 // Return the slot number as the module's contribution to
1556 // the type plane plus the index of the function map insertion.
1557 return MI->second.next_slot + insertValue(V);
1558 else
1559 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001560 }
1561 }
1562 }
1563
Chris Lattner0dda8612006-12-06 06:15:43 +00001564 // N.B. Can only get here if TheFunction == 0
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001565
1566 // If the module map's type plane is not for the Value's type
Chris Lattner23e829a2006-12-06 05:12:21 +00001567 if (MI != mMap.end()) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001568 // Lookup the value in the module's map
1569 ValueMap::const_iterator MVI = MI->second.map.find(V);
Chris Lattner23e829a2006-12-06 05:12:21 +00001570 if (MVI != MI->second.map.end())
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001571 return MVI->second;
1572 }
1573
1574 return insertValue(V);
1575}
1576
1577
1578// Low level insert function. Minimal checking is done. This
Chris Lattner193de902006-12-06 05:27:40 +00001579// function is just for the convenience of getOrCreateSlot (above).
Chris Lattner23e829a2006-12-06 05:12:21 +00001580unsigned SlotMachine::insertValue(const Value *V) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001581 assert(V && "Can't insert a null Value into SlotMachine!");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001582 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
Chris Lattner7e4e7662006-12-06 05:42:32 +00001583 "Can't insert a non-GlobalValue Constant into SlotMachine");
1584 assert(V->getType() != Type::VoidTy && !V->hasName());
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001585
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001586 const Type *VTy = V->getType();
1587 unsigned DestSlot = 0;
1588
Chris Lattner23e829a2006-12-06 05:12:21 +00001589 if (TheFunction) {
1590 TypedPlanes::iterator I = fMap.find(VTy);
1591 if (I == fMap.end())
Reid Spencer58d30f22004-07-04 11:50:43 +00001592 I = fMap.insert(std::make_pair(VTy,ValuePlane())).first;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001593 DestSlot = I->second.map[V] = I->second.next_slot++;
1594 } else {
Chris Lattner23e829a2006-12-06 05:12:21 +00001595 TypedPlanes::iterator I = mMap.find(VTy);
1596 if (I == mMap.end())
Reid Spencer58d30f22004-07-04 11:50:43 +00001597 I = mMap.insert(std::make_pair(VTy,ValuePlane())).first;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001598 DestSlot = I->second.map[V] = I->second.next_slot++;
1599 }
1600
Misha Brukmanb1c93172005-04-21 23:48:37 +00001601 SC_DEBUG(" Inserting value [" << VTy << "] = " << V << " slot=" <<
Reid Spencer56010e42004-05-26 21:56:09 +00001602 DestSlot << " [");
Chris Lattnerf8090cb2006-12-06 05:55:41 +00001603 // G = Global, F = Function, o = other
1604 SC_DEBUG((isa<GlobalVariable>(V) ? 'G' : (isa<Function>(V) ? 'F' : 'o')));
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001605 SC_DEBUG("]\n");
1606 return DestSlot;
1607}