blob: a0d82bc402c5dcc8aec535031e8aa93da4f6885f [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);
John Criswellcd116ba2004-06-01 14:54:08 +0000290 Result += "{ ";
Chris Lattnerac6db752004-02-09 04:37:31 +0000291 for (StructType::element_iterator I = STy->element_begin(),
292 E = STy->element_end(); I != E; ++I) {
293 if (I != STy->element_begin())
Chris Lattnerb86620e2001-10-29 16:37:48 +0000294 Result += ", ";
John Criswellcd116ba2004-06-01 14:54:08 +0000295 calcTypeName(*I, TypeStack, TypeNames, Result);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000296 }
297 Result += " }";
298 break;
299 }
300 case Type::PointerTyID:
Misha Brukmanb1c93172005-04-21 23:48:37 +0000301 calcTypeName(cast<PointerType>(Ty)->getElementType(),
John Criswellcd116ba2004-06-01 14:54:08 +0000302 TypeStack, TypeNames, Result);
303 Result += "*";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000304 break;
305 case Type::ArrayTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000306 const ArrayType *ATy = cast<ArrayType>(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000307 Result += "[" + utostr(ATy->getNumElements()) + " x ";
308 calcTypeName(ATy->getElementType(), TypeStack, TypeNames, Result);
309 Result += "]";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000310 break;
311 }
Brian Gaeke02209042004-08-20 06:00:58 +0000312 case Type::PackedTyID: {
313 const PackedType *PTy = cast<PackedType>(Ty);
314 Result += "<" + utostr(PTy->getNumElements()) + " x ";
315 calcTypeName(PTy->getElementType(), TypeStack, TypeNames, Result);
316 Result += ">";
317 break;
318 }
Chris Lattner15285ab2003-05-14 17:50:47 +0000319 case Type::OpaqueTyID:
John Criswellcd116ba2004-06-01 14:54:08 +0000320 Result += "opaque";
Chris Lattner15285ab2003-05-14 17:50:47 +0000321 break;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000322 default:
John Criswellcd116ba2004-06-01 14:54:08 +0000323 Result += "<unrecognized-type>";
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000324 break;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000325 }
326
327 TypeStack.pop_back(); // Remove self from stack...
Chris Lattnerb86620e2001-10-29 16:37:48 +0000328}
329
330
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000331/// printTypeInt - The internal guts of printing out a type that has a
332/// potentially named portion.
333///
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000334static std::ostream &printTypeInt(std::ostream &Out, const Type *Ty,
335 std::map<const Type *, std::string> &TypeNames) {
Chris Lattnerb86620e2001-10-29 16:37:48 +0000336 // Primitive types always print out their description, regardless of whether
337 // they have been named or not.
338 //
Chris Lattner92d60532003-10-30 00:12:51 +0000339 if (Ty->isPrimitiveType() && !isa<OpaqueType>(Ty))
340 return Out << Ty->getDescription();
Chris Lattnerb86620e2001-10-29 16:37:48 +0000341
342 // Check to see if the type is named.
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000343 std::map<const Type *, std::string>::iterator I = TypeNames.find(Ty);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000344 if (I != TypeNames.end()) return Out << I->second;
345
346 // Otherwise we have a type that has not been named but is a derived type.
347 // Carefully recurse the type hierarchy to print out any contained symbolic
348 // names.
349 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000350 std::vector<const Type *> TypeStack;
John Criswellcd116ba2004-06-01 14:54:08 +0000351 std::string TypeName;
352 calcTypeName(Ty, TypeStack, TypeNames, TypeName);
Chris Lattner7f74a562002-01-20 22:54:45 +0000353 TypeNames.insert(std::make_pair(Ty, TypeName));//Cache type name for later use
John Criswellcd116ba2004-06-01 14:54:08 +0000354 return (Out << TypeName);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000355}
356
Chris Lattner34b95182001-10-31 04:33:19 +0000357
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000358/// WriteTypeSymbolic - This attempts to write the specified type as a symbolic
359/// type, iff there is an entry in the modules symbol table for the specified
360/// type or one of it's component types. This is slower than a simple x << Type
361///
Chris Lattner189d19f2003-11-21 20:23:48 +0000362std::ostream &llvm::WriteTypeSymbolic(std::ostream &Out, const Type *Ty,
363 const Module *M) {
Misha Brukmanb1c93172005-04-21 23:48:37 +0000364 Out << ' ';
Chris Lattnerb86620e2001-10-29 16:37:48 +0000365
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000366 // If they want us to print out a type, but there is no context, we can't
367 // print it symbolically.
368 if (!M)
Chris Lattner72f866e2001-10-29 16:40:32 +0000369 return Out << Ty->getDescription();
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000370
371 std::map<const Type *, std::string> TypeNames;
372 fillTypeNameTable(M, TypeNames);
373 return printTypeInt(Out, Ty, TypeNames);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000374}
375
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000376// PrintEscapedString - Print each character of the specified string, escaping
377// it if it is not printable or if it is an escape char.
378static void PrintEscapedString(const std::string &Str, std::ostream &Out) {
379 for (unsigned i = 0, e = Str.size(); i != e; ++i) {
380 unsigned char C = Str[i];
381 if (isprint(C) && C != '"' && C != '\\') {
382 Out << C;
383 } else {
384 Out << '\\'
385 << (char) ((C/16 < 10) ? ( C/16 +'0') : ( C/16 -10+'A'))
386 << (char)(((C&15) < 10) ? ((C&15)+'0') : ((C&15)-10+'A'));
387 }
388 }
389}
390
Chris Lattnerfc9f1c92006-12-06 06:40:49 +0000391static const char *getPredicateText(unsigned predicate) {
Reid Spencer812a1be2006-12-04 05:19:18 +0000392 const char * pred = "unknown";
393 switch (predicate) {
394 case FCmpInst::FCMP_FALSE: pred = "false"; break;
395 case FCmpInst::FCMP_OEQ: pred = "oeq"; break;
396 case FCmpInst::FCMP_OGT: pred = "ogt"; break;
397 case FCmpInst::FCMP_OGE: pred = "oge"; break;
398 case FCmpInst::FCMP_OLT: pred = "olt"; break;
399 case FCmpInst::FCMP_OLE: pred = "ole"; break;
400 case FCmpInst::FCMP_ONE: pred = "one"; break;
401 case FCmpInst::FCMP_ORD: pred = "ord"; break;
402 case FCmpInst::FCMP_UNO: pred = "uno"; break;
403 case FCmpInst::FCMP_UEQ: pred = "ueq"; break;
404 case FCmpInst::FCMP_UGT: pred = "ugt"; break;
405 case FCmpInst::FCMP_UGE: pred = "uge"; break;
406 case FCmpInst::FCMP_ULT: pred = "ult"; break;
407 case FCmpInst::FCMP_ULE: pred = "ule"; break;
408 case FCmpInst::FCMP_UNE: pred = "une"; break;
409 case FCmpInst::FCMP_TRUE: pred = "true"; break;
410 case ICmpInst::ICMP_EQ: pred = "eq"; break;
411 case ICmpInst::ICMP_NE: pred = "ne"; break;
412 case ICmpInst::ICMP_SGT: pred = "sgt"; break;
413 case ICmpInst::ICMP_SGE: pred = "sge"; break;
414 case ICmpInst::ICMP_SLT: pred = "slt"; break;
415 case ICmpInst::ICMP_SLE: pred = "sle"; break;
416 case ICmpInst::ICMP_UGT: pred = "ugt"; break;
417 case ICmpInst::ICMP_UGE: pred = "uge"; break;
418 case ICmpInst::ICMP_ULT: pred = "ult"; break;
419 case ICmpInst::ICMP_ULE: pred = "ule"; break;
420 }
421 return pred;
422}
423
Misha Brukmanb1c93172005-04-21 23:48:37 +0000424/// @brief Internal constant writer.
425static void WriteConstantInt(std::ostream &Out, const Constant *CV,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000426 std::map<const Type *, std::string> &TypeTable,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000427 SlotMachine *Machine) {
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000428 const int IndentSize = 4;
429 static std::string Indent = "\n";
Chris Lattner1e194682002-04-18 18:53:13 +0000430 if (const ConstantBool *CB = dyn_cast<ConstantBool>(CV)) {
Chris Lattner94e39cb2006-09-28 22:50:29 +0000431 Out << (CB->getValue() ? "true" : "false");
Reid Spencere0fc4df2006-10-20 07:07:24 +0000432 } else if (const ConstantInt *CI = dyn_cast<ConstantInt>(CV)) {
433 if (CI->getType()->isSigned())
434 Out << CI->getSExtValue();
435 else
436 Out << CI->getZExtValue();
Chris Lattner1e194682002-04-18 18:53:13 +0000437 } else if (const ConstantFP *CFP = dyn_cast<ConstantFP>(CV)) {
438 // We would like to output the FP constant value in exponential notation,
439 // but we cannot do this if doing so will lose precision. Check here to
440 // make sure that we only output it in exponential format if we can parse
441 // the value back and get the same value.
442 //
443 std::string StrVal = ftostr(CFP->getValue());
444
445 // Check to make sure that the stringized number is not some string like
446 // "Inf" or NaN, that atof will accept, but the lexer will not. Check that
447 // the string matches the "[-+]?[0-9]" regex.
448 //
449 if ((StrVal[0] >= '0' && StrVal[0] <= '9') ||
450 ((StrVal[0] == '-' || StrVal[0] == '+') &&
Brian Gaeke87b4f072003-06-17 23:55:35 +0000451 (StrVal[1] >= '0' && StrVal[1] <= '9')))
Chris Lattner1e194682002-04-18 18:53:13 +0000452 // Reparse stringized version!
453 if (atof(StrVal.c_str()) == CFP->getValue()) {
Chris Lattner472cc102005-01-04 01:56:57 +0000454 Out << StrVal;
455 return;
Chris Lattner1e194682002-04-18 18:53:13 +0000456 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000457
Chris Lattner1e194682002-04-18 18:53:13 +0000458 // Otherwise we could not reparse it to exactly the same value, so we must
459 // output the string in hexadecimal format!
Chris Lattner472cc102005-01-04 01:56:57 +0000460 assert(sizeof(double) == sizeof(uint64_t) &&
Chris Lattner1e194682002-04-18 18:53:13 +0000461 "assuming that double is 64 bits!");
Jim Laskeyb74c6662005-08-17 19:34:49 +0000462 Out << "0x" << utohexstr(DoubleToBits(CFP->getValue()));
Chris Lattner1e194682002-04-18 18:53:13 +0000463
Chris Lattner76b2ff42004-02-15 05:55:15 +0000464 } else if (isa<ConstantAggregateZero>(CV)) {
465 Out << "zeroinitializer";
Chris Lattner1e194682002-04-18 18:53:13 +0000466 } else if (const ConstantArray *CA = dyn_cast<ConstantArray>(CV)) {
467 // As a special case, print the array as a string if it is an array of
468 // ubytes or an array of sbytes with positive values.
Misha Brukmanb1c93172005-04-21 23:48:37 +0000469 //
Chris Lattner1e194682002-04-18 18:53:13 +0000470 const Type *ETy = CA->getType()->getElementType();
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000471 if (CA->isString()) {
Chris Lattner1e194682002-04-18 18:53:13 +0000472 Out << "c\"";
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000473 PrintEscapedString(CA->getAsString(), Out);
Chris Lattner1e194682002-04-18 18:53:13 +0000474 Out << "\"";
475
476 } else { // Cannot output in string format...
Misha Brukman21bbdb92004-06-04 21:11:51 +0000477 Out << '[';
Chris Lattnerd84bb632002-04-16 21:36:08 +0000478 if (CA->getNumOperands()) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000479 Out << ' ';
Chris Lattner1e194682002-04-18 18:53:13 +0000480 printTypeInt(Out, ETy, TypeTable);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000481 WriteAsOperandInternal(Out, CA->getOperand(0),
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000482 TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000483 for (unsigned i = 1, e = CA->getNumOperands(); i != e; ++i) {
484 Out << ", ";
Chris Lattner1e194682002-04-18 18:53:13 +0000485 printTypeInt(Out, ETy, TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000486 WriteAsOperandInternal(Out, CA->getOperand(i), TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000487 }
488 }
489 Out << " ]";
490 }
491 } else if (const ConstantStruct *CS = dyn_cast<ConstantStruct>(CV)) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000492 Out << '{';
Jim Laskey3bb78742006-02-25 12:27:03 +0000493 unsigned N = CS->getNumOperands();
494 if (N) {
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000495 if (N > 2) {
496 Indent += std::string(IndentSize, ' ');
497 Out << Indent;
498 } else {
499 Out << ' ';
500 }
Chris Lattnerd84bb632002-04-16 21:36:08 +0000501 printTypeInt(Out, CS->getOperand(0)->getType(), TypeTable);
502
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000503 WriteAsOperandInternal(Out, CS->getOperand(0), TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000504
Jim Laskey3bb78742006-02-25 12:27:03 +0000505 for (unsigned i = 1; i < N; i++) {
Chris Lattnerd84bb632002-04-16 21:36:08 +0000506 Out << ", ";
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000507 if (N > 2) Out << Indent;
Chris Lattnerd84bb632002-04-16 21:36:08 +0000508 printTypeInt(Out, CS->getOperand(i)->getType(), TypeTable);
509
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000510 WriteAsOperandInternal(Out, CS->getOperand(i), TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000511 }
Jim Laskey6be3d8e2006-02-27 10:33:53 +0000512 if (N > 2) Indent.resize(Indent.size() - IndentSize);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000513 }
Jim Laskey3bb78742006-02-25 12:27:03 +0000514
Chris Lattnerd84bb632002-04-16 21:36:08 +0000515 Out << " }";
Brian Gaeke02209042004-08-20 06:00:58 +0000516 } else if (const ConstantPacked *CP = dyn_cast<ConstantPacked>(CV)) {
517 const Type *ETy = CP->getType()->getElementType();
Misha Brukmanb1c93172005-04-21 23:48:37 +0000518 assert(CP->getNumOperands() > 0 &&
Brian Gaeke02209042004-08-20 06:00:58 +0000519 "Number of operands for a PackedConst must be > 0");
520 Out << '<';
521 Out << ' ';
522 printTypeInt(Out, ETy, TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000523 WriteAsOperandInternal(Out, CP->getOperand(0), TypeTable, Machine);
Brian Gaeke02209042004-08-20 06:00:58 +0000524 for (unsigned i = 1, e = CP->getNumOperands(); i != e; ++i) {
525 Out << ", ";
526 printTypeInt(Out, ETy, TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000527 WriteAsOperandInternal(Out, CP->getOperand(i), TypeTable, Machine);
Brian Gaeke02209042004-08-20 06:00:58 +0000528 }
529 Out << " >";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000530 } else if (isa<ConstantPointerNull>(CV)) {
531 Out << "null";
532
Chris Lattner5e0b9f22004-10-16 18:08:06 +0000533 } else if (isa<UndefValue>(CV)) {
534 Out << "undef";
535
Chris Lattner3cd8c562002-07-30 18:54:25 +0000536 } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CV)) {
Reid Spencer812a1be2006-12-04 05:19:18 +0000537 Out << CE->getOpcodeName();
538 if (CE->isCompare())
539 Out << " " << getPredicateText(CE->getPredicate());
540 Out << " (";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000541
Vikram S. Adveb952b542002-07-14 23:14:45 +0000542 for (User::const_op_iterator OI=CE->op_begin(); OI != CE->op_end(); ++OI) {
543 printTypeInt(Out, (*OI)->getType(), TypeTable);
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000544 WriteAsOperandInternal(Out, *OI, TypeTable, Machine);
Vikram S. Adveb952b542002-07-14 23:14:45 +0000545 if (OI+1 != CE->op_end())
Chris Lattner3cd8c562002-07-30 18:54:25 +0000546 Out << ", ";
Vikram S. Adveb952b542002-07-14 23:14:45 +0000547 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000548
Reid Spencer6c38f0b2006-11-27 01:05:10 +0000549 if (CE->isCast()) {
Chris Lattner83b396b2002-08-15 19:37:43 +0000550 Out << " to ";
551 printTypeInt(Out, CE->getType(), TypeTable);
552 }
Reid Spencer6c38f0b2006-11-27 01:05:10 +0000553
Misha Brukman21bbdb92004-06-04 21:11:51 +0000554 Out << ')';
Chris Lattner83b396b2002-08-15 19:37:43 +0000555
Chris Lattnerd84bb632002-04-16 21:36:08 +0000556 } else {
Vikram S. Adveb952b542002-07-14 23:14:45 +0000557 Out << "<placeholder or erroneous Constant>";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000558 }
559}
560
561
Misha Brukmanc566ca362004-03-02 00:22:19 +0000562/// WriteAsOperand - Write the name of the specified value out to the specified
563/// ostream. This can be useful when you just want to print int %reg126, not
564/// the whole instruction that generated it.
565///
Misha Brukmanb1c93172005-04-21 23:48:37 +0000566static void WriteAsOperandInternal(std::ostream &Out, const Value *V,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000567 std::map<const Type*, std::string> &TypeTable,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000568 SlotMachine *Machine) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000569 Out << ' ';
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000570 if (V->hasName())
Chris Lattner1c343042003-08-22 05:40:38 +0000571 Out << getLLVMName(V->getName());
Reid Spenceraccd7c72004-07-17 23:47:01 +0000572 else {
573 const Constant *CV = dyn_cast<Constant>(V);
Chris Lattnera2d810d2006-01-25 22:26:05 +0000574 if (CV && !isa<GlobalValue>(CV)) {
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000575 WriteConstantInt(Out, CV, TypeTable, Machine);
Chris Lattnera2d810d2006-01-25 22:26:05 +0000576 } else if (const InlineAsm *IA = dyn_cast<InlineAsm>(V)) {
577 Out << "asm ";
578 if (IA->hasSideEffects())
579 Out << "sideeffect ";
580 Out << '"';
581 PrintEscapedString(IA->getAsmString(), Out);
582 Out << "\", \"";
583 PrintEscapedString(IA->getConstraintString(), Out);
584 Out << '"';
585 } else {
Chris Lattnerd84bb632002-04-16 21:36:08 +0000586 int Slot;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000587 if (Machine) {
Reid Spencer56010e42004-05-26 21:56:09 +0000588 Slot = Machine->getSlot(V);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000589 } else {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000590 Machine = createSlotMachine(V);
Chris Lattner96749c42006-05-14 18:46:52 +0000591 if (Machine)
Chris Lattner757ee0b2004-06-09 19:41:19 +0000592 Slot = Machine->getSlot(V);
593 else
594 Slot = -1;
Reid Spencer56010e42004-05-26 21:56:09 +0000595 delete Machine;
Chris Lattnerd84bb632002-04-16 21:36:08 +0000596 }
Chris Lattner757ee0b2004-06-09 19:41:19 +0000597 if (Slot != -1)
598 Out << '%' << Slot;
599 else
600 Out << "<badref>";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000601 }
602 }
603}
604
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000605/// WriteAsOperand - Write the name of the specified value out to the specified
606/// ostream. This can be useful when you just want to print int %reg126, not
607/// the whole instruction that generated it.
608///
Chris Lattner189d19f2003-11-21 20:23:48 +0000609std::ostream &llvm::WriteAsOperand(std::ostream &Out, const Value *V,
Chris Lattner0dda8612006-12-06 06:15:43 +0000610 bool PrintType, const Module *Context) {
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000611 std::map<const Type *, std::string> TypeNames;
Chris Lattner5a9f63e2002-07-10 16:48:17 +0000612 if (Context == 0) Context = getModuleFromVal(V);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000613
Chris Lattner98cf1f52002-11-20 18:36:02 +0000614 if (Context)
Chris Lattner5a9f63e2002-07-10 16:48:17 +0000615 fillTypeNameTable(Context, TypeNames);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000616
617 if (PrintType)
618 printTypeInt(Out, V->getType(), TypeNames);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000619
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000620 WriteAsOperandInternal(Out, V, TypeNames, 0);
Chris Lattner5e5abe32001-07-20 19:15:21 +0000621 return Out;
622}
623
Reid Spencer58d30f22004-07-04 11:50:43 +0000624
Chris Lattner189d19f2003-11-21 20:23:48 +0000625namespace llvm {
Chris Lattner2e9fee42001-07-12 23:35:26 +0000626
Chris Lattnerfee714f2001-09-07 16:36:04 +0000627class AssemblyWriter {
Misha Brukmana6619a92004-06-21 21:53:56 +0000628 std::ostream &Out;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000629 SlotMachine &Machine;
Chris Lattner7bfee412001-10-29 16:05:51 +0000630 const Module *TheModule;
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000631 std::map<const Type *, std::string> TypeNames;
Chris Lattner8339f7d2003-10-30 23:41:03 +0000632 AssemblyAnnotationWriter *AnnotationWriter;
Chris Lattner2f7c9632001-06-06 20:29:01 +0000633public:
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000634 inline AssemblyWriter(std::ostream &o, SlotMachine &Mac, const Module *M,
Chris Lattner8339f7d2003-10-30 23:41:03 +0000635 AssemblyAnnotationWriter *AAW)
Misha Brukmana6619a92004-06-21 21:53:56 +0000636 : Out(o), Machine(Mac), TheModule(M), AnnotationWriter(AAW) {
Chris Lattner7bfee412001-10-29 16:05:51 +0000637
638 // If the module has a symbol table, take all global types and stuff their
639 // names into the TypeNames map.
640 //
Chris Lattnerb86620e2001-10-29 16:37:48 +0000641 fillTypeNameTable(M, TypeNames);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000642 }
643
Chris Lattner7bfee412001-10-29 16:05:51 +0000644 inline void write(const Module *M) { printModule(M); }
645 inline void write(const GlobalVariable *G) { printGlobal(G); }
Chris Lattner57698e22002-03-26 18:01:55 +0000646 inline void write(const Function *F) { printFunction(F); }
Chris Lattner7bfee412001-10-29 16:05:51 +0000647 inline void write(const BasicBlock *BB) { printBasicBlock(BB); }
Chris Lattner113f4f42002-06-25 16:13:24 +0000648 inline void write(const Instruction *I) { printInstruction(*I); }
Chris Lattner3462ae32001-12-03 22:26:30 +0000649 inline void write(const Constant *CPV) { printConstant(CPV); }
Chris Lattner7db79582001-11-07 04:21:57 +0000650 inline void write(const Type *Ty) { printType(Ty); }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000651
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000652 void writeOperand(const Value *Op, bool PrintType);
Chris Lattner1e194682002-04-18 18:53:13 +0000653
Misha Brukman4685e262004-04-28 15:31:21 +0000654 const Module* getModule() { return TheModule; }
655
Misha Brukmand92f54a2004-11-15 19:30:05 +0000656private:
Chris Lattner7bfee412001-10-29 16:05:51 +0000657 void printModule(const Module *M);
658 void printSymbolTable(const SymbolTable &ST);
Chris Lattner3462ae32001-12-03 22:26:30 +0000659 void printConstant(const Constant *CPV);
Chris Lattner7bfee412001-10-29 16:05:51 +0000660 void printGlobal(const GlobalVariable *GV);
Chris Lattner57698e22002-03-26 18:01:55 +0000661 void printFunction(const Function *F);
Chris Lattner2e9fa6d2002-04-09 19:48:49 +0000662 void printArgument(const Argument *FA);
Chris Lattner7bfee412001-10-29 16:05:51 +0000663 void printBasicBlock(const BasicBlock *BB);
Chris Lattner113f4f42002-06-25 16:13:24 +0000664 void printInstruction(const Instruction &I);
Chris Lattnerd816b532002-04-13 20:53:41 +0000665
666 // printType - Go to extreme measures to attempt to print out a short,
667 // symbolic version of a type name.
668 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000669 std::ostream &printType(const Type *Ty) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000670 return printTypeInt(Out, Ty, TypeNames);
Chris Lattnerd816b532002-04-13 20:53:41 +0000671 }
672
673 // printTypeAtLeastOneLevel - Print out one level of the possibly complex type
674 // without considering any symbolic types that we may have equal to it.
675 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000676 std::ostream &printTypeAtLeastOneLevel(const Type *Ty);
Chris Lattner7bfee412001-10-29 16:05:51 +0000677
Chris Lattner862e3382001-10-13 06:42:36 +0000678 // printInfoComment - Print a little comment after the instruction indicating
679 // which slot it occupies.
Chris Lattner113f4f42002-06-25 16:13:24 +0000680 void printInfoComment(const Value &V);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000681};
Reid Spencerf43ac622004-05-27 22:04:46 +0000682} // end of llvm namespace
Chris Lattner2f7c9632001-06-06 20:29:01 +0000683
Misha Brukmanc566ca362004-03-02 00:22:19 +0000684/// printTypeAtLeastOneLevel - Print out one level of the possibly complex type
685/// without considering any symbolic types that we may have equal to it.
686///
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000687std::ostream &AssemblyWriter::printTypeAtLeastOneLevel(const Type *Ty) {
Chris Lattner113f4f42002-06-25 16:13:24 +0000688 if (const FunctionType *FTy = dyn_cast<FunctionType>(Ty)) {
Chris Lattnerd816b532002-04-13 20:53:41 +0000689 printType(FTy->getReturnType()) << " (";
Chris Lattnerfa829be2004-02-09 04:14:01 +0000690 for (FunctionType::param_iterator I = FTy->param_begin(),
691 E = FTy->param_end(); I != E; ++I) {
692 if (I != FTy->param_begin())
Misha Brukmana6619a92004-06-21 21:53:56 +0000693 Out << ", ";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000694 printType(*I);
Chris Lattnerd816b532002-04-13 20:53:41 +0000695 }
696 if (FTy->isVarArg()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000697 if (FTy->getNumParams()) Out << ", ";
698 Out << "...";
Chris Lattnerd816b532002-04-13 20:53:41 +0000699 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000700 Out << ')';
Chris Lattner113f4f42002-06-25 16:13:24 +0000701 } else if (const StructType *STy = dyn_cast<StructType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000702 Out << "{ ";
Chris Lattnerac6db752004-02-09 04:37:31 +0000703 for (StructType::element_iterator I = STy->element_begin(),
704 E = STy->element_end(); I != E; ++I) {
705 if (I != STy->element_begin())
Misha Brukmana6619a92004-06-21 21:53:56 +0000706 Out << ", ";
Chris Lattnerd816b532002-04-13 20:53:41 +0000707 printType(*I);
708 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000709 Out << " }";
Chris Lattner113f4f42002-06-25 16:13:24 +0000710 } else if (const PointerType *PTy = dyn_cast<PointerType>(Ty)) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000711 printType(PTy->getElementType()) << '*';
Chris Lattner113f4f42002-06-25 16:13:24 +0000712 } else if (const ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000713 Out << '[' << ATy->getNumElements() << " x ";
Misha Brukman21bbdb92004-06-04 21:11:51 +0000714 printType(ATy->getElementType()) << ']';
Reid Spencere203d352004-08-20 15:37:30 +0000715 } else if (const PackedType *PTy = dyn_cast<PackedType>(Ty)) {
716 Out << '<' << PTy->getNumElements() << " x ";
717 printType(PTy->getElementType()) << '>';
718 }
Reid Spencerde46e482006-11-02 20:25:50 +0000719 else if (isa<OpaqueType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000720 Out << "opaque";
Chris Lattnerd816b532002-04-13 20:53:41 +0000721 } else {
Vikram S. Adveb952b542002-07-14 23:14:45 +0000722 if (!Ty->isPrimitiveType())
Misha Brukmana6619a92004-06-21 21:53:56 +0000723 Out << "<unknown derived type>";
Chris Lattnerd816b532002-04-13 20:53:41 +0000724 printType(Ty);
725 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000726 return Out;
Chris Lattnerd816b532002-04-13 20:53:41 +0000727}
728
729
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000730void AssemblyWriter::writeOperand(const Value *Operand, bool PrintType) {
731 if (Operand == 0) {
Chris Lattner08f7d0c2005-02-24 16:58:29 +0000732 Out << "<null operand!>";
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000733 } else {
734 if (PrintType) { Out << ' '; printType(Operand->getType()); }
735 WriteAsOperandInternal(Out, Operand, TypeNames, &Machine);
Chris Lattner08f7d0c2005-02-24 16:58:29 +0000736 }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000737}
738
Chris Lattner2f7c9632001-06-06 20:29:01 +0000739
Chris Lattner7bfee412001-10-29 16:05:51 +0000740void AssemblyWriter::printModule(const Module *M) {
Chris Lattner4d8689e2005-03-02 23:12:40 +0000741 if (!M->getModuleIdentifier().empty() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +0000742 // Don't print the ID if it will start a new line (which would
Chris Lattner4d8689e2005-03-02 23:12:40 +0000743 // require a comment char before it).
744 M->getModuleIdentifier().find('\n') == std::string::npos)
745 Out << "; ModuleID = '" << M->getModuleIdentifier() << "'\n";
746
Owen Andersone2237542006-10-18 02:21:12 +0000747 if (!M->getDataLayout().empty())
Chris Lattner04897162006-10-22 06:06:56 +0000748 Out << "target datalayout = \"" << M->getDataLayout() << "\"\n";
Owen Andersone2237542006-10-18 02:21:12 +0000749
Chris Lattner8068e0c2003-08-24 13:48:48 +0000750 switch (M->getEndianness()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000751 case Module::LittleEndian: Out << "target endian = little\n"; break;
752 case Module::BigEndian: Out << "target endian = big\n"; break;
Chris Lattner8068e0c2003-08-24 13:48:48 +0000753 case Module::AnyEndianness: break;
754 }
755 switch (M->getPointerSize()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000756 case Module::Pointer32: Out << "target pointersize = 32\n"; break;
757 case Module::Pointer64: Out << "target pointersize = 64\n"; break;
Chris Lattner8068e0c2003-08-24 13:48:48 +0000758 case Module::AnyPointerSize: break;
759 }
Reid Spencer48f98c82004-07-25 21:44:54 +0000760 if (!M->getTargetTriple().empty())
Reid Spencerffec7df2004-07-25 21:29:43 +0000761 Out << "target triple = \"" << M->getTargetTriple() << "\"\n";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000762
Chris Lattnereef2fe72006-01-24 04:13:11 +0000763 if (!M->getModuleInlineAsm().empty()) {
Chris Lattnerefaf35d2006-01-24 00:45:30 +0000764 // Split the string into lines, to make it easier to read the .ll file.
Chris Lattnereef2fe72006-01-24 04:13:11 +0000765 std::string Asm = M->getModuleInlineAsm();
Chris Lattnerefaf35d2006-01-24 00:45:30 +0000766 size_t CurPos = 0;
767 size_t NewLine = Asm.find_first_of('\n', CurPos);
768 while (NewLine != std::string::npos) {
769 // We found a newline, print the portion of the asm string from the
770 // last newline up to this newline.
771 Out << "module asm \"";
772 PrintEscapedString(std::string(Asm.begin()+CurPos, Asm.begin()+NewLine),
773 Out);
774 Out << "\"\n";
775 CurPos = NewLine+1;
776 NewLine = Asm.find_first_of('\n', CurPos);
777 }
Chris Lattner3acaf5c2006-01-24 00:40:17 +0000778 Out << "module asm \"";
Chris Lattnerefaf35d2006-01-24 00:45:30 +0000779 PrintEscapedString(std::string(Asm.begin()+CurPos, Asm.end()), Out);
Chris Lattner6ed87bd2006-01-23 23:03:36 +0000780 Out << "\"\n";
781 }
782
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000783 // Loop over the dependent libraries and emit them.
Chris Lattner730cfe42004-09-14 04:51:44 +0000784 Module::lib_iterator LI = M->lib_begin();
785 Module::lib_iterator LE = M->lib_end();
Reid Spencer48f98c82004-07-25 21:44:54 +0000786 if (LI != LE) {
Chris Lattner730cfe42004-09-14 04:51:44 +0000787 Out << "deplibs = [ ";
788 while (LI != LE) {
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000789 Out << '"' << *LI << '"';
Reid Spencerffec7df2004-07-25 21:29:43 +0000790 ++LI;
Chris Lattner730cfe42004-09-14 04:51:44 +0000791 if (LI != LE)
792 Out << ", ";
Reid Spencerffec7df2004-07-25 21:29:43 +0000793 }
794 Out << " ]\n";
Reid Spencercc5ff642004-07-25 18:08:18 +0000795 }
Reid Spencerb9e08772004-09-13 23:44:23 +0000796
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000797 // Loop over the symbol table, emitting all named constants.
Chris Lattner98cf1f52002-11-20 18:36:02 +0000798 printSymbolTable(M->getSymbolTable());
Misha Brukmanb1c93172005-04-21 23:48:37 +0000799
Chris Lattner54932b02006-12-06 04:41:52 +0000800 for (Module::const_global_iterator I = M->global_begin(), E = M->global_end();
801 I != E; ++I)
Chris Lattner113f4f42002-06-25 16:13:24 +0000802 printGlobal(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000803
Misha Brukmana6619a92004-06-21 21:53:56 +0000804 Out << "\nimplementation ; Functions:\n";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000805
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000806 // Output all of the functions.
Chris Lattner113f4f42002-06-25 16:13:24 +0000807 for (Module::const_iterator I = M->begin(), E = M->end(); I != E; ++I)
808 printFunction(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000809}
810
Chris Lattner7bfee412001-10-29 16:05:51 +0000811void AssemblyWriter::printGlobal(const GlobalVariable *GV) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000812 if (GV->hasName()) Out << getLLVMName(GV->getName()) << " = ";
Chris Lattner37798642001-09-18 04:01:05 +0000813
Misha Brukmanb1c93172005-04-21 23:48:37 +0000814 if (!GV->hasInitializer())
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000815 switch (GV->getLinkage()) {
816 case GlobalValue::DLLImportLinkage: Out << "dllimport "; break;
817 case GlobalValue::ExternalWeakLinkage: Out << "extern_weak "; break;
818 default: Out << "external "; break;
819 }
Chris Lattner379a8d22003-04-16 20:28:45 +0000820 else
821 switch (GV->getLinkage()) {
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000822 case GlobalValue::InternalLinkage: Out << "internal "; break;
823 case GlobalValue::LinkOnceLinkage: Out << "linkonce "; break;
824 case GlobalValue::WeakLinkage: Out << "weak "; break;
825 case GlobalValue::AppendingLinkage: Out << "appending "; break;
826 case GlobalValue::DLLImportLinkage: Out << "dllimport "; break;
827 case GlobalValue::DLLExportLinkage: Out << "dllexport "; break;
828 case GlobalValue::ExternalWeakLinkage: Out << "extern_weak "; break;
829 case GlobalValue::ExternalLinkage: break;
Misha Brukman70457632004-11-14 21:04:34 +0000830 case GlobalValue::GhostLinkage:
Bill Wendlingf3baad32006-12-07 01:30:32 +0000831 cerr << "GhostLinkage not allowed in AsmWriter!\n";
Misha Brukman70457632004-11-14 21:04:34 +0000832 abort();
Chris Lattner379a8d22003-04-16 20:28:45 +0000833 }
Chris Lattner37798642001-09-18 04:01:05 +0000834
Misha Brukmana6619a92004-06-21 21:53:56 +0000835 Out << (GV->isConstant() ? "constant " : "global ");
Chris Lattner2413b162001-12-04 00:03:30 +0000836 printType(GV->getType()->getElementType());
Chris Lattner37798642001-09-18 04:01:05 +0000837
Reid Spenceraccd7c72004-07-17 23:47:01 +0000838 if (GV->hasInitializer()) {
839 Constant* C = cast<Constant>(GV->getInitializer());
840 assert(C && "GlobalVar initializer isn't constant?");
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000841 writeOperand(GV->getInitializer(), false);
Reid Spenceraccd7c72004-07-17 23:47:01 +0000842 }
Chris Lattnerf8a974d2005-11-06 06:48:53 +0000843
Chris Lattner4b96c542005-11-12 00:10:19 +0000844 if (GV->hasSection())
845 Out << ", section \"" << GV->getSection() << '"';
846 if (GV->getAlignment())
Chris Lattnerf8a974d2005-11-06 06:48:53 +0000847 Out << ", align " << GV->getAlignment();
Chris Lattner4b96c542005-11-12 00:10:19 +0000848
Chris Lattner113f4f42002-06-25 16:13:24 +0000849 printInfoComment(*GV);
Misha Brukmana6619a92004-06-21 21:53:56 +0000850 Out << "\n";
Chris Lattnerda975502001-09-10 07:58:01 +0000851}
852
Chris Lattnerfee714f2001-09-07 16:36:04 +0000853
Reid Spencere7e96712004-05-25 08:53:40 +0000854// printSymbolTable - Run through symbol table looking for constants
855// and types. Emit their declarations.
Chris Lattner7bfee412001-10-29 16:05:51 +0000856void AssemblyWriter::printSymbolTable(const SymbolTable &ST) {
Reid Spencere7e96712004-05-25 08:53:40 +0000857
858 // Print the types.
859 for (SymbolTable::type_const_iterator TI = ST.type_begin();
Chris Lattner23e829a2006-12-06 05:12:21 +0000860 TI != ST.type_end(); ++TI) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000861 Out << "\t" << getLLVMName(TI->first) << " = type ";
Reid Spencere7e96712004-05-25 08:53:40 +0000862
863 // Make sure we print out at least one level of the type structure, so
864 // that we do not get %FILE = type %FILE
865 //
866 printTypeAtLeastOneLevel(TI->second) << "\n";
867 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000868
Reid Spencere7e96712004-05-25 08:53:40 +0000869 // Print the constants, in type plane order.
870 for (SymbolTable::plane_const_iterator PI = ST.plane_begin();
Chris Lattner23e829a2006-12-06 05:12:21 +0000871 PI != ST.plane_end(); ++PI) {
Reid Spencere7e96712004-05-25 08:53:40 +0000872 SymbolTable::value_const_iterator VI = ST.value_begin(PI->first);
873 SymbolTable::value_const_iterator VE = ST.value_end(PI->first);
874
875 for (; VI != VE; ++VI) {
Reid Spenceraccd7c72004-07-17 23:47:01 +0000876 const Value* V = VI->second;
877 const Constant *CPV = dyn_cast<Constant>(V) ;
878 if (CPV && !isa<GlobalValue>(V)) {
Reid Spencer56010e42004-05-26 21:56:09 +0000879 printConstant(CPV);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000880 }
881 }
Chris Lattnera7620d92001-07-15 06:35:59 +0000882 }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000883}
884
885
Misha Brukmanc566ca362004-03-02 00:22:19 +0000886/// printConstant - Print out a constant pool entry...
887///
Chris Lattner3462ae32001-12-03 22:26:30 +0000888void AssemblyWriter::printConstant(const Constant *CPV) {
Chris Lattnerfee714f2001-09-07 16:36:04 +0000889 // Don't print out unnamed constants, they will be inlined
890 if (!CPV->hasName()) return;
Chris Lattner2f7c9632001-06-06 20:29:01 +0000891
Chris Lattneree998be2001-07-26 16:29:38 +0000892 // Print out name...
Misha Brukmana6619a92004-06-21 21:53:56 +0000893 Out << "\t" << getLLVMName(CPV->getName()) << " =";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000894
Chris Lattner78e2e8b2006-12-06 06:24:27 +0000895 // Write the value out now.
896 writeOperand(CPV, true);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000897
Chris Lattner113f4f42002-06-25 16:13:24 +0000898 printInfoComment(*CPV);
Misha Brukmana6619a92004-06-21 21:53:56 +0000899 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000900}
901
Misha Brukmanc566ca362004-03-02 00:22:19 +0000902/// printFunction - Print all aspects of a function.
903///
Chris Lattner113f4f42002-06-25 16:13:24 +0000904void AssemblyWriter::printFunction(const Function *F) {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000905 // Print out the return type and name...
Misha Brukmana6619a92004-06-21 21:53:56 +0000906 Out << "\n";
Chris Lattner379a8d22003-04-16 20:28:45 +0000907
Chris Lattner97e36f22004-12-05 06:44:09 +0000908 // Ensure that no local symbols conflict with global symbols.
909 const_cast<Function*>(F)->renameLocalSymbols();
910
Misha Brukmana6619a92004-06-21 21:53:56 +0000911 if (AnnotationWriter) AnnotationWriter->emitFunctionAnnot(F, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +0000912
Chris Lattner379a8d22003-04-16 20:28:45 +0000913 if (F->isExternal())
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000914 switch (F->getLinkage()) {
915 case GlobalValue::DLLImportLinkage: Out << "declare dllimport "; break;
916 case GlobalValue::ExternalWeakLinkage: Out << "declare extern_weak "; break;
917 default: Out << "declare ";
918 }
Chris Lattner379a8d22003-04-16 20:28:45 +0000919 else
920 switch (F->getLinkage()) {
Anton Korobeynikovd61d39e2006-09-14 18:23:27 +0000921 case GlobalValue::InternalLinkage: Out << "internal "; break;
922 case GlobalValue::LinkOnceLinkage: Out << "linkonce "; break;
923 case GlobalValue::WeakLinkage: Out << "weak "; break;
924 case GlobalValue::AppendingLinkage: Out << "appending "; break;
925 case GlobalValue::DLLImportLinkage: Out << "dllimport "; break;
926 case GlobalValue::DLLExportLinkage: Out << "dllexport "; break;
927 case GlobalValue::ExternalWeakLinkage: Out << "extern_weak "; break;
Chris Lattner379a8d22003-04-16 20:28:45 +0000928 case GlobalValue::ExternalLinkage: break;
Misha Brukman70457632004-11-14 21:04:34 +0000929 case GlobalValue::GhostLinkage:
Bill Wendlingf3baad32006-12-07 01:30:32 +0000930 cerr << "GhostLinkage not allowed in AsmWriter!\n";
Misha Brukman70457632004-11-14 21:04:34 +0000931 abort();
Chris Lattner379a8d22003-04-16 20:28:45 +0000932 }
933
Chris Lattnerf7b6d312005-05-06 20:26:43 +0000934 // Print the calling convention.
935 switch (F->getCallingConv()) {
936 case CallingConv::C: break; // default
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +0000937 case CallingConv::CSRet: Out << "csretcc "; break;
938 case CallingConv::Fast: Out << "fastcc "; break;
939 case CallingConv::Cold: Out << "coldcc "; break;
940 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
941 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +0000942 default: Out << "cc" << F->getCallingConv() << " "; break;
943 }
944
Misha Brukman21bbdb92004-06-04 21:11:51 +0000945 printType(F->getReturnType()) << ' ';
Chris Lattner5b337482003-10-18 05:57:43 +0000946 if (!F->getName().empty())
Misha Brukmana6619a92004-06-21 21:53:56 +0000947 Out << getLLVMName(F->getName());
Chris Lattner5b337482003-10-18 05:57:43 +0000948 else
Misha Brukmana6619a92004-06-21 21:53:56 +0000949 Out << "\"\"";
950 Out << '(';
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000951 Machine.incorporateFunction(F);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000952
Chris Lattner7bfee412001-10-29 16:05:51 +0000953 // Loop over the arguments, printing them...
Chris Lattner113f4f42002-06-25 16:13:24 +0000954 const FunctionType *FT = F->getFunctionType();
Chris Lattnerfee714f2001-09-07 16:36:04 +0000955
Chris Lattner54932b02006-12-06 04:41:52 +0000956 for (Function::const_arg_iterator I = F->arg_begin(), E = F->arg_end();
957 I != E; ++I)
Chris Lattner149376d2002-10-13 20:57:00 +0000958 printArgument(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000959
960 // Finish printing arguments...
Chris Lattner113f4f42002-06-25 16:13:24 +0000961 if (FT->isVarArg()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000962 if (FT->getNumParams()) Out << ", ";
963 Out << "..."; // Output varargs portion of signature!
Chris Lattnerfee714f2001-09-07 16:36:04 +0000964 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000965 Out << ')';
Chris Lattnerfee714f2001-09-07 16:36:04 +0000966
Chris Lattner4b96c542005-11-12 00:10:19 +0000967 if (F->hasSection())
968 Out << " section \"" << F->getSection() << '"';
Chris Lattnerf8a974d2005-11-06 06:48:53 +0000969 if (F->getAlignment())
970 Out << " align " << F->getAlignment();
Chris Lattner4b96c542005-11-12 00:10:19 +0000971
Chris Lattner113f4f42002-06-25 16:13:24 +0000972 if (F->isExternal()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000973 Out << "\n";
Chris Lattnerb2f02e52002-05-06 03:00:40 +0000974 } else {
Misha Brukmana6619a92004-06-21 21:53:56 +0000975 Out << " {";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000976
Chris Lattner6915f8f2002-04-07 22:49:37 +0000977 // Output all of its basic blocks... for the function
Chris Lattner113f4f42002-06-25 16:13:24 +0000978 for (Function::const_iterator I = F->begin(), E = F->end(); I != E; ++I)
979 printBasicBlock(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000980
Misha Brukmana6619a92004-06-21 21:53:56 +0000981 Out << "}\n";
Chris Lattnerfee714f2001-09-07 16:36:04 +0000982 }
983
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000984 Machine.purgeFunction();
Chris Lattner2f7c9632001-06-06 20:29:01 +0000985}
986
Misha Brukmanc566ca362004-03-02 00:22:19 +0000987/// printArgument - This member is called for every argument that is passed into
988/// the function. Simply print it out
989///
Chris Lattner2e9fa6d2002-04-09 19:48:49 +0000990void AssemblyWriter::printArgument(const Argument *Arg) {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000991 // Insert commas as we go... the first arg doesn't get a comma
Chris Lattnerfb972ad2005-03-15 05:03:36 +0000992 if (Arg != Arg->getParent()->arg_begin()) Out << ", ";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000993
994 // Output type...
Chris Lattner7bfee412001-10-29 16:05:51 +0000995 printType(Arg->getType());
Misha Brukmanb1c93172005-04-21 23:48:37 +0000996
Chris Lattner2f7c9632001-06-06 20:29:01 +0000997 // Output name, if available...
998 if (Arg->hasName())
Misha Brukmana6619a92004-06-21 21:53:56 +0000999 Out << ' ' << getLLVMName(Arg->getName());
Chris Lattner2f7c9632001-06-06 20:29:01 +00001000}
1001
Misha Brukmanc566ca362004-03-02 00:22:19 +00001002/// printBasicBlock - This member is called for each basic block in a method.
1003///
Chris Lattner7bfee412001-10-29 16:05:51 +00001004void AssemblyWriter::printBasicBlock(const BasicBlock *BB) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001005 if (BB->hasName()) { // Print out the label if it exists...
Alkis Evlogimenos991d6ad2004-12-10 05:41:10 +00001006 Out << "\n" << getLLVMName(BB->getName(), false) << ':';
Chris Lattner408dbdb2002-05-14 16:02:05 +00001007 } else if (!BB->use_empty()) { // Don't print block # of no uses...
Misha Brukmana6619a92004-06-21 21:53:56 +00001008 Out << "\n; <label>:";
Chris Lattner757ee0b2004-06-09 19:41:19 +00001009 int Slot = Machine.getSlot(BB);
1010 if (Slot != -1)
Misha Brukmana6619a92004-06-21 21:53:56 +00001011 Out << Slot;
Chris Lattner757ee0b2004-06-09 19:41:19 +00001012 else
Misha Brukmana6619a92004-06-21 21:53:56 +00001013 Out << "<badref>";
Chris Lattner58185f22002-10-02 19:38:55 +00001014 }
Chris Lattner2447ef52003-11-20 00:09:43 +00001015
1016 if (BB->getParent() == 0)
Misha Brukmana6619a92004-06-21 21:53:56 +00001017 Out << "\t\t; Error: Block without parent!";
Chris Lattner2447ef52003-11-20 00:09:43 +00001018 else {
1019 if (BB != &BB->getParent()->front()) { // Not the entry block?
1020 // Output predecessors for the block...
Misha Brukmana6619a92004-06-21 21:53:56 +00001021 Out << "\t\t;";
Chris Lattner2447ef52003-11-20 00:09:43 +00001022 pred_const_iterator PI = pred_begin(BB), PE = pred_end(BB);
Misha Brukmanb1c93172005-04-21 23:48:37 +00001023
Chris Lattner2447ef52003-11-20 00:09:43 +00001024 if (PI == PE) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001025 Out << " No predecessors!";
Chris Lattner2447ef52003-11-20 00:09:43 +00001026 } else {
Misha Brukmana6619a92004-06-21 21:53:56 +00001027 Out << " preds =";
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001028 writeOperand(*PI, false);
Chris Lattner2447ef52003-11-20 00:09:43 +00001029 for (++PI; PI != PE; ++PI) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001030 Out << ',';
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001031 writeOperand(*PI, false);
Chris Lattner2447ef52003-11-20 00:09:43 +00001032 }
Chris Lattner00211f12003-11-16 22:59:57 +00001033 }
Chris Lattner58185f22002-10-02 19:38:55 +00001034 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001035 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001036
Misha Brukmana6619a92004-06-21 21:53:56 +00001037 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001038
Misha Brukmana6619a92004-06-21 21:53:56 +00001039 if (AnnotationWriter) AnnotationWriter->emitBasicBlockStartAnnot(BB, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001040
Chris Lattnerfee714f2001-09-07 16:36:04 +00001041 // Output all of the instructions in the basic block...
Chris Lattner113f4f42002-06-25 16:13:24 +00001042 for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I != E; ++I)
1043 printInstruction(*I);
Chris Lattner96cdd272004-03-08 18:51:45 +00001044
Misha Brukmana6619a92004-06-21 21:53:56 +00001045 if (AnnotationWriter) AnnotationWriter->emitBasicBlockEndAnnot(BB, Out);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001046}
1047
Chris Lattner862e3382001-10-13 06:42:36 +00001048
Misha Brukmanc566ca362004-03-02 00:22:19 +00001049/// printInfoComment - Print a little comment after the instruction indicating
1050/// which slot it occupies.
1051///
Chris Lattner113f4f42002-06-25 16:13:24 +00001052void AssemblyWriter::printInfoComment(const Value &V) {
1053 if (V.getType() != Type::VoidTy) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001054 Out << "\t\t; <";
Misha Brukman21bbdb92004-06-04 21:11:51 +00001055 printType(V.getType()) << '>';
Chris Lattner862e3382001-10-13 06:42:36 +00001056
Chris Lattner113f4f42002-06-25 16:13:24 +00001057 if (!V.hasName()) {
Chris Lattner757ee0b2004-06-09 19:41:19 +00001058 int SlotNum = Machine.getSlot(&V);
1059 if (SlotNum == -1)
Misha Brukmana6619a92004-06-21 21:53:56 +00001060 Out << ":<badref>";
Reid Spencer8beac692004-06-09 15:26:53 +00001061 else
Misha Brukmana6619a92004-06-21 21:53:56 +00001062 Out << ':' << SlotNum; // Print out the def slot taken.
Chris Lattner862e3382001-10-13 06:42:36 +00001063 }
Chris Lattnerb6c21db2005-02-01 01:24:01 +00001064 Out << " [#uses=" << V.getNumUses() << ']'; // Output # uses
Chris Lattner862e3382001-10-13 06:42:36 +00001065 }
1066}
1067
Reid Spencere7141c82006-08-28 01:02:49 +00001068// This member is called for each Instruction in a function..
Chris Lattner113f4f42002-06-25 16:13:24 +00001069void AssemblyWriter::printInstruction(const Instruction &I) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001070 if (AnnotationWriter) AnnotationWriter->emitInstructionAnnot(&I, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001071
Misha Brukmana6619a92004-06-21 21:53:56 +00001072 Out << "\t";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001073
1074 // Print out name if it exists...
Chris Lattner113f4f42002-06-25 16:13:24 +00001075 if (I.hasName())
Misha Brukmana6619a92004-06-21 21:53:56 +00001076 Out << getLLVMName(I.getName()) << " = ";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001077
Chris Lattner06038452005-05-06 05:51:46 +00001078 // If this is a volatile load or store, print out the volatile marker.
Chris Lattner504f9242003-09-08 17:45:59 +00001079 if ((isa<LoadInst>(I) && cast<LoadInst>(I).isVolatile()) ||
Chris Lattner06038452005-05-06 05:51:46 +00001080 (isa<StoreInst>(I) && cast<StoreInst>(I).isVolatile())) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001081 Out << "volatile ";
Chris Lattner06038452005-05-06 05:51:46 +00001082 } else if (isa<CallInst>(I) && cast<CallInst>(I).isTailCall()) {
1083 // If this is a call, check if it's a tail call.
1084 Out << "tail ";
1085 }
Chris Lattner504f9242003-09-08 17:45:59 +00001086
Chris Lattner2f7c9632001-06-06 20:29:01 +00001087 // Print out the opcode...
Misha Brukmana6619a92004-06-21 21:53:56 +00001088 Out << I.getOpcodeName();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001089
Reid Spencer45e52392006-12-03 06:27:29 +00001090 // Print out the compare instruction predicates
1091 if (const FCmpInst *FCI = dyn_cast<FCmpInst>(&I)) {
Reid Spencer812a1be2006-12-04 05:19:18 +00001092 Out << " " << getPredicateText(FCI->getPredicate());
Reid Spencer45e52392006-12-03 06:27:29 +00001093 } else if (const ICmpInst *ICI = dyn_cast<ICmpInst>(&I)) {
Reid Spencer812a1be2006-12-04 05:19:18 +00001094 Out << " " << getPredicateText(ICI->getPredicate());
Reid Spencer45e52392006-12-03 06:27:29 +00001095 }
1096
Chris Lattner2f7c9632001-06-06 20:29:01 +00001097 // Print out the type of the operands...
Chris Lattner113f4f42002-06-25 16:13:24 +00001098 const Value *Operand = I.getNumOperands() ? I.getOperand(0) : 0;
Chris Lattner2f7c9632001-06-06 20:29:01 +00001099
1100 // Special case conditional branches to swizzle the condition out to the front
Chris Lattner113f4f42002-06-25 16:13:24 +00001101 if (isa<BranchInst>(I) && I.getNumOperands() > 1) {
1102 writeOperand(I.getOperand(2), true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001103 Out << ',';
Chris Lattner2f7c9632001-06-06 20:29:01 +00001104 writeOperand(Operand, true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001105 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001106 writeOperand(I.getOperand(1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001107
Chris Lattner8d48df22002-04-13 18:34:38 +00001108 } else if (isa<SwitchInst>(I)) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001109 // Special case switch statement to get formatting nice and correct...
Misha Brukmana6619a92004-06-21 21:53:56 +00001110 writeOperand(Operand , true); Out << ',';
1111 writeOperand(I.getOperand(1), true); Out << " [";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001112
Chris Lattner113f4f42002-06-25 16:13:24 +00001113 for (unsigned op = 2, Eop = I.getNumOperands(); op < Eop; op += 2) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001114 Out << "\n\t\t";
1115 writeOperand(I.getOperand(op ), true); Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001116 writeOperand(I.getOperand(op+1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001117 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001118 Out << "\n\t]";
Chris Lattnerda558102001-10-02 03:41:24 +00001119 } else if (isa<PHINode>(I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001120 Out << ' ';
Chris Lattner113f4f42002-06-25 16:13:24 +00001121 printType(I.getType());
Misha Brukmana6619a92004-06-21 21:53:56 +00001122 Out << ' ';
Chris Lattner2f7c9632001-06-06 20:29:01 +00001123
Chris Lattner113f4f42002-06-25 16:13:24 +00001124 for (unsigned op = 0, Eop = I.getNumOperands(); op < Eop; op += 2) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001125 if (op) Out << ", ";
Misha Brukmanb1c93172005-04-21 23:48:37 +00001126 Out << '[';
Misha Brukmana6619a92004-06-21 21:53:56 +00001127 writeOperand(I.getOperand(op ), false); Out << ',';
1128 writeOperand(I.getOperand(op+1), false); Out << " ]";
Chris Lattner931ef3b2001-06-11 15:04:20 +00001129 }
Chris Lattner862e3382001-10-13 06:42:36 +00001130 } else if (isa<ReturnInst>(I) && !Operand) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001131 Out << " void";
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001132 } else if (const CallInst *CI = dyn_cast<CallInst>(&I)) {
1133 // Print the calling convention being used.
1134 switch (CI->getCallingConv()) {
1135 case CallingConv::C: break; // default
Chris Lattner29d20852006-05-19 21:58:52 +00001136 case CallingConv::CSRet: Out << " csretcc"; break;
1137 case CallingConv::Fast: Out << " fastcc"; break;
1138 case CallingConv::Cold: Out << " coldcc"; break;
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +00001139 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
1140 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001141 default: Out << " cc" << CI->getCallingConv(); break;
1142 }
1143
Chris Lattner463d6a52003-08-05 15:34:45 +00001144 const PointerType *PTy = cast<PointerType>(Operand->getType());
1145 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1146 const Type *RetTy = FTy->getReturnType();
Chris Lattner2f2d9472001-11-06 21:28:12 +00001147
Chris Lattner463d6a52003-08-05 15:34:45 +00001148 // If possible, print out the short form of the call instruction. We can
Chris Lattner6915f8f2002-04-07 22:49:37 +00001149 // only do this if the first argument is a pointer to a nonvararg function,
Chris Lattner463d6a52003-08-05 15:34:45 +00001150 // and if the return type is not a pointer to a function.
Chris Lattner2f2d9472001-11-06 21:28:12 +00001151 //
Chris Lattner463d6a52003-08-05 15:34:45 +00001152 if (!FTy->isVarArg() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +00001153 (!isa<PointerType>(RetTy) ||
Chris Lattnerd9a36a62002-07-25 20:58:51 +00001154 !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001155 Out << ' '; printType(RetTy);
Chris Lattner2f2d9472001-11-06 21:28:12 +00001156 writeOperand(Operand, false);
1157 } else {
1158 writeOperand(Operand, true);
1159 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001160 Out << '(';
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001161 if (CI->getNumOperands() > 1) writeOperand(CI->getOperand(1), true);
Chris Lattner113f4f42002-06-25 16:13:24 +00001162 for (unsigned op = 2, Eop = I.getNumOperands(); op < Eop; ++op) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001163 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001164 writeOperand(I.getOperand(op), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001165 }
1166
Misha Brukmana6619a92004-06-21 21:53:56 +00001167 Out << " )";
Chris Lattner113f4f42002-06-25 16:13:24 +00001168 } else if (const InvokeInst *II = dyn_cast<InvokeInst>(&I)) {
Chris Lattner463d6a52003-08-05 15:34:45 +00001169 const PointerType *PTy = cast<PointerType>(Operand->getType());
1170 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1171 const Type *RetTy = FTy->getReturnType();
1172
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001173 // Print the calling convention being used.
1174 switch (II->getCallingConv()) {
1175 case CallingConv::C: break; // default
Chris Lattner29d20852006-05-19 21:58:52 +00001176 case CallingConv::CSRet: Out << " csretcc"; break;
1177 case CallingConv::Fast: Out << " fastcc"; break;
1178 case CallingConv::Cold: Out << " coldcc"; break;
Anton Korobeynikov3c5b3df2006-09-20 22:03:51 +00001179 case CallingConv::X86_StdCall: Out << "x86_stdcallcc "; break;
1180 case CallingConv::X86_FastCall: Out << "x86_fastcallcc "; break;
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001181 default: Out << " cc" << II->getCallingConv(); break;
1182 }
1183
Chris Lattner463d6a52003-08-05 15:34:45 +00001184 // If possible, print out the short form of the invoke instruction. We can
1185 // only do this if the first argument is a pointer to a nonvararg function,
1186 // and if the return type is not a pointer to a function.
1187 //
1188 if (!FTy->isVarArg() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +00001189 (!isa<PointerType>(RetTy) ||
Chris Lattner463d6a52003-08-05 15:34:45 +00001190 !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001191 Out << ' '; printType(RetTy);
Chris Lattner463d6a52003-08-05 15:34:45 +00001192 writeOperand(Operand, false);
1193 } else {
1194 writeOperand(Operand, true);
1195 }
1196
Misha Brukmana6619a92004-06-21 21:53:56 +00001197 Out << '(';
Chris Lattner113f4f42002-06-25 16:13:24 +00001198 if (I.getNumOperands() > 3) writeOperand(I.getOperand(3), true);
1199 for (unsigned op = 4, Eop = I.getNumOperands(); op < Eop; ++op) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001200 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001201 writeOperand(I.getOperand(op), true);
Chris Lattner862e3382001-10-13 06:42:36 +00001202 }
1203
Misha Brukmana6619a92004-06-21 21:53:56 +00001204 Out << " )\n\t\t\tto";
Chris Lattner862e3382001-10-13 06:42:36 +00001205 writeOperand(II->getNormalDest(), true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001206 Out << " unwind";
Chris Lattnerfae8ab32004-02-08 21:44:31 +00001207 writeOperand(II->getUnwindDest(), true);
Chris Lattner862e3382001-10-13 06:42:36 +00001208
Chris Lattner113f4f42002-06-25 16:13:24 +00001209 } else if (const AllocationInst *AI = dyn_cast<AllocationInst>(&I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001210 Out << ' ';
Chris Lattner8d48df22002-04-13 18:34:38 +00001211 printType(AI->getType()->getElementType());
1212 if (AI->isArrayAllocation()) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001213 Out << ',';
Chris Lattner8d48df22002-04-13 18:34:38 +00001214 writeOperand(AI->getArraySize(), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001215 }
Nate Begeman848622f2005-11-05 09:21:28 +00001216 if (AI->getAlignment()) {
Chris Lattner7aeee3a2005-11-05 21:20:34 +00001217 Out << ", align " << AI->getAlignment();
Nate Begeman848622f2005-11-05 09:21:28 +00001218 }
Chris Lattner862e3382001-10-13 06:42:36 +00001219 } else if (isa<CastInst>(I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001220 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001221 Out << " to ";
Chris Lattner113f4f42002-06-25 16:13:24 +00001222 printType(I.getType());
Chris Lattner5b337482003-10-18 05:57:43 +00001223 } else if (isa<VAArgInst>(I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001224 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001225 Out << ", ";
Chris Lattnerf70da102003-05-08 02:44:12 +00001226 printType(I.getType());
Chris Lattner2f7c9632001-06-06 20:29:01 +00001227 } else if (Operand) { // Print the normal way...
1228
Misha Brukmanb1c93172005-04-21 23:48:37 +00001229 // PrintAllTypes - Instructions who have operands of all the same type
Chris Lattner2f7c9632001-06-06 20:29:01 +00001230 // omit the type from all but the first operand. If the instruction has
1231 // different type operands (for example br), then they are all printed.
1232 bool PrintAllTypes = false;
1233 const Type *TheType = Operand->getType();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001234
Chris Lattner52bd5cb92004-03-12 05:53:14 +00001235 // Shift Left & Right print both types even for Ubyte LHS, and select prints
1236 // types even if all operands are bools.
Chris Lattnerbbe0a422006-04-08 01:18:18 +00001237 if (isa<ShiftInst>(I) || isa<SelectInst>(I) || isa<StoreInst>(I) ||
1238 isa<ShuffleVectorInst>(I)) {
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001239 PrintAllTypes = true;
1240 } else {
1241 for (unsigned i = 1, E = I.getNumOperands(); i != E; ++i) {
1242 Operand = I.getOperand(i);
1243 if (Operand->getType() != TheType) {
1244 PrintAllTypes = true; // We have differing types! Print them all!
1245 break;
1246 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001247 }
1248 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001249
Chris Lattner7bfee412001-10-29 16:05:51 +00001250 if (!PrintAllTypes) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001251 Out << ' ';
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001252 printType(TheType);
Chris Lattner7bfee412001-10-29 16:05:51 +00001253 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001254
Chris Lattner113f4f42002-06-25 16:13:24 +00001255 for (unsigned i = 0, E = I.getNumOperands(); i != E; ++i) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001256 if (i) Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001257 writeOperand(I.getOperand(i), PrintAllTypes);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001258 }
1259 }
1260
Chris Lattner862e3382001-10-13 06:42:36 +00001261 printInfoComment(I);
Misha Brukmana6619a92004-06-21 21:53:56 +00001262 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001263}
1264
1265
1266//===----------------------------------------------------------------------===//
1267// External Interface declarations
1268//===----------------------------------------------------------------------===//
1269
Chris Lattner8339f7d2003-10-30 23:41:03 +00001270void Module::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001271 SlotMachine SlotTable(this);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001272 AssemblyWriter W(o, SlotTable, this, AAW);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001273 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001274}
1275
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001276void GlobalVariable::print(std::ostream &o) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001277 SlotMachine SlotTable(getParent());
Chris Lattner8339f7d2003-10-30 23:41:03 +00001278 AssemblyWriter W(o, SlotTable, getParent(), 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001279 W.write(this);
Chris Lattneradfe0d12001-09-10 20:08:19 +00001280}
1281
Chris Lattner8339f7d2003-10-30 23:41:03 +00001282void Function::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001283 SlotMachine SlotTable(getParent());
Chris Lattner8339f7d2003-10-30 23:41:03 +00001284 AssemblyWriter W(o, SlotTable, getParent(), AAW);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001285
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001286 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001287}
1288
Chris Lattnereef2fe72006-01-24 04:13:11 +00001289void InlineAsm::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001290 WriteAsOperand(o, this, true, 0);
Chris Lattnereef2fe72006-01-24 04:13:11 +00001291}
1292
Chris Lattner8339f7d2003-10-30 23:41:03 +00001293void BasicBlock::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001294 SlotMachine SlotTable(getParent());
Misha Brukmanb1c93172005-04-21 23:48:37 +00001295 AssemblyWriter W(o, SlotTable,
Chris Lattner8339f7d2003-10-30 23:41:03 +00001296 getParent() ? getParent()->getParent() : 0, AAW);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001297 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001298}
1299
Chris Lattner8339f7d2003-10-30 23:41:03 +00001300void Instruction::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001301 const Function *F = getParent() ? getParent()->getParent() : 0;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001302 SlotMachine SlotTable(F);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001303 AssemblyWriter W(o, SlotTable, F ? F->getParent() : 0, AAW);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001304
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001305 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001306}
Chris Lattner7db79582001-11-07 04:21:57 +00001307
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001308void Constant::print(std::ostream &o) const {
1309 if (this == 0) { o << "<null> constant value\n"; return; }
Chris Lattner7d734802002-09-10 15:53:49 +00001310
Misha Brukman21bbdb92004-06-04 21:11:51 +00001311 o << ' ' << getType()->getDescription() << ' ';
Evan Cheng5b19a802006-03-01 22:17:00 +00001312
1313 std::map<const Type *, std::string> TypeTable;
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001314 WriteConstantInt(o, this, TypeTable, 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001315}
1316
Misha Brukmanb1c93172005-04-21 23:48:37 +00001317void Type::print(std::ostream &o) const {
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001318 if (this == 0)
1319 o << "<null Type>";
1320 else
1321 o << getDescription();
1322}
1323
Chris Lattner2e9fa6d2002-04-09 19:48:49 +00001324void Argument::print(std::ostream &o) const {
Chris Lattner78e2e8b2006-12-06 06:24:27 +00001325 WriteAsOperand(o, this, true, getParent() ? getParent()->getParent() : 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001326}
1327
Reid Spencer52641832004-05-25 18:14:38 +00001328// Value::dump - allow easy printing of Values from the debugger.
1329// Located here because so much of the needed functionality is here.
Bill Wendling22e978a2006-12-07 20:04:42 +00001330void Value::dump() const { print(*cerr.stream()); cerr << '\n'; }
Reid Spencer52641832004-05-25 18:14:38 +00001331
1332// Type::dump - allow easy printing of Values from the debugger.
1333// Located here because so much of the needed functionality is here.
Bill Wendling22e978a2006-12-07 20:04:42 +00001334void Type::dump() const { print(*cerr.stream()); cerr << '\n'; }
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001335
1336//===----------------------------------------------------------------------===//
Chris Lattnerfc9f1c92006-12-06 06:40:49 +00001337// SlotMachine Implementation
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001338//===----------------------------------------------------------------------===//
1339
1340#if 0
Bill Wendlingf3baad32006-12-07 01:30:32 +00001341#define SC_DEBUG(X) cerr << X
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001342#else
1343#define SC_DEBUG(X)
1344#endif
1345
1346// Module level constructor. Causes the contents of the Module (sans functions)
1347// to be added to the slot table.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001348SlotMachine::SlotMachine(const Module *M)
Reid Spencer56010e42004-05-26 21:56:09 +00001349 : TheModule(M) ///< Saved for lazy initialization.
1350 , TheFunction(0)
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001351 , FunctionProcessed(false)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001352{
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001353}
1354
1355// Function level constructor. Causes the contents of the Module and the one
1356// function provided to be added to the slot table.
Chris Lattner23e829a2006-12-06 05:12:21 +00001357SlotMachine::SlotMachine(const Function *F)
1358 : TheModule(F ? F->getParent() : 0) ///< Saved for lazy initialization
Reid Spencer56010e42004-05-26 21:56:09 +00001359 , TheFunction(F) ///< Saved for lazy initialization
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001360 , FunctionProcessed(false)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001361{
Reid Spencer56010e42004-05-26 21:56:09 +00001362}
1363
1364inline void SlotMachine::initialize(void) {
Chris Lattner23e829a2006-12-06 05:12:21 +00001365 if (TheModule) {
Misha Brukmanb1c93172005-04-21 23:48:37 +00001366 processModule();
Reid Spencer56010e42004-05-26 21:56:09 +00001367 TheModule = 0; ///< Prevent re-processing next time we're called.
1368 }
Chris Lattner193de902006-12-06 05:27:40 +00001369 if (TheFunction && !FunctionProcessed)
Misha Brukmanb1c93172005-04-21 23:48:37 +00001370 processFunction();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001371}
1372
1373// Iterate through all the global variables, functions, and global
Misha Brukmanb1c93172005-04-21 23:48:37 +00001374// variable initializers and create slots for them.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001375void SlotMachine::processModule() {
1376 SC_DEBUG("begin processModule!\n");
1377
Chris Lattner0dda8612006-12-06 06:15:43 +00001378 // Add all of the unnamed global variables to the value table.
Chris Lattner54932b02006-12-06 04:41:52 +00001379 for (Module::const_global_iterator I = TheModule->global_begin(),
1380 E = TheModule->global_end(); I != E; ++I)
Chris Lattner0dda8612006-12-06 06:15:43 +00001381 if (!I->hasName())
1382 getOrCreateSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001383
Chris Lattner0dda8612006-12-06 06:15:43 +00001384 // Add all the unnamed functions to the table.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001385 for (Module::const_iterator I = TheModule->begin(), E = TheModule->end();
1386 I != E; ++I)
Chris Lattner0dda8612006-12-06 06:15:43 +00001387 if (!I->hasName())
1388 getOrCreateSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001389
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001390 SC_DEBUG("end processModule!\n");
1391}
1392
1393
Reid Spencer56010e42004-05-26 21:56:09 +00001394// Process the arguments, basic blocks, and instructions of a function.
1395void SlotMachine::processFunction() {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001396 SC_DEBUG("begin processFunction!\n");
1397
Chris Lattner0dda8612006-12-06 06:15:43 +00001398 // Add all the function arguments with no names.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001399 for(Function::const_arg_iterator AI = TheFunction->arg_begin(),
Chris Lattner531f9e92005-03-15 04:54:21 +00001400 AE = TheFunction->arg_end(); AI != AE; ++AI)
Chris Lattner0dda8612006-12-06 06:15:43 +00001401 if (!AI->hasName())
1402 getOrCreateSlot(AI);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001403
1404 SC_DEBUG("Inserting Instructions:\n");
1405
Chris Lattner0dda8612006-12-06 06:15:43 +00001406 // Add all of the basic blocks and instructions with no names.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001407 for (Function::const_iterator BB = TheFunction->begin(),
Reid Spencer56010e42004-05-26 21:56:09 +00001408 E = TheFunction->end(); BB != E; ++BB) {
Chris Lattner0dda8612006-12-06 06:15:43 +00001409 if (!BB->hasName())
1410 getOrCreateSlot(BB);
1411 for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I != E; ++I)
1412 if (I->getType() != Type::VoidTy && !I->hasName())
1413 getOrCreateSlot(I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001414 }
1415
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001416 FunctionProcessed = true;
1417
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001418 SC_DEBUG("end processFunction!\n");
1419}
1420
Chris Lattner193de902006-12-06 05:27:40 +00001421/// Clean up after incorporating a function. This is the only way to get out of
1422/// the function incorporation state that affects the
1423/// getSlot/getOrCreateSlot lock. Function incorporation state is indicated
1424/// by TheFunction != 0.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001425void SlotMachine::purgeFunction() {
1426 SC_DEBUG("begin purgeFunction!\n");
1427 fMap.clear(); // Simply discard the function level map
Reid Spencer56010e42004-05-26 21:56:09 +00001428 TheFunction = 0;
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001429 FunctionProcessed = false;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001430 SC_DEBUG("end purgeFunction!\n");
1431}
1432
1433/// Get the slot number for a value. This function will assert if you
Chris Lattner193de902006-12-06 05:27:40 +00001434/// ask for a Value that hasn't previously been inserted with getOrCreateSlot.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001435/// Types are forbidden because Type does not inherit from Value (any more).
Chris Lattner757ee0b2004-06-09 19:41:19 +00001436int SlotMachine::getSlot(const Value *V) {
Chris Lattner23e829a2006-12-06 05:12:21 +00001437 assert(V && "Can't get slot for null Value");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001438 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1439 "Can't insert a non-GlobalValue Constant into SlotMachine");
Reid Spencer56010e42004-05-26 21:56:09 +00001440
1441 // Check for uninitialized state and do lazy initialization
1442 this->initialize();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001443
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001444 // Get the type of the value
1445 const Type* VTy = V->getType();
1446
1447 // Find the type plane in the module map
1448 TypedPlanes::const_iterator MI = mMap.find(VTy);
1449
Chris Lattner23e829a2006-12-06 05:12:21 +00001450 if (TheFunction) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001451 // Lookup the type in the function map too
1452 TypedPlanes::const_iterator FI = fMap.find(VTy);
1453 // If there is a corresponding type plane in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001454 if (FI != fMap.end()) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001455 // Lookup the Value in the function map
1456 ValueMap::const_iterator FVI = FI->second.map.find(V);
1457 // If the value doesn't exist in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001458 if (FVI == FI->second.map.end()) {
Chris Lattner68a038e2004-06-09 22:22:10 +00001459 // Look up the value in the module map.
1460 if (MI == mMap.end()) return -1;
Reid Spencer56010e42004-05-26 21:56:09 +00001461 ValueMap::const_iterator MVI = MI->second.map.find(V);
1462 // If we didn't find it, it wasn't inserted
Chris Lattner757ee0b2004-06-09 19:41:19 +00001463 if (MVI == MI->second.map.end()) return -1;
Chris Lattner23e829a2006-12-06 05:12:21 +00001464 assert(MVI != MI->second.map.end() && "Value not found");
Reid Spencer56010e42004-05-26 21:56:09 +00001465 // We found it only at the module level
Misha Brukmanb1c93172005-04-21 23:48:37 +00001466 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001467
1468 // else the value exists in the function map
1469 } else {
Reid Spencer56010e42004-05-26 21:56:09 +00001470 // Return the slot number as the module's contribution to
1471 // the type plane plus the index in the function's contribution
1472 // to the type plane.
Chris Lattnerb1f04782004-06-15 21:07:32 +00001473 if (MI != mMap.end())
1474 return MI->second.next_slot + FVI->second;
1475 else
1476 return FVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001477 }
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001478 }
1479 }
1480
Reid Spencer8beac692004-06-09 15:26:53 +00001481 // N.B. Can get here only if either !TheFunction or the function doesn't
1482 // have a corresponding type plane for the Value
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001483
1484 // Make sure the type plane exists
Chris Lattner757ee0b2004-06-09 19:41:19 +00001485 if (MI == mMap.end()) return -1;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001486 // Lookup the value in the module's map
1487 ValueMap::const_iterator MVI = MI->second.map.find(V);
1488 // Make sure we found it.
Chris Lattner757ee0b2004-06-09 19:41:19 +00001489 if (MVI == MI->second.map.end()) return -1;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001490 // Return it.
1491 return MVI->second;
1492}
1493
Reid Spencer58d30f22004-07-04 11:50:43 +00001494
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001495// Create a new slot, or return the existing slot if it is already
1496// inserted. Note that the logic here parallels getSlot but instead
1497// of asserting when the Value* isn't found, it inserts the value.
Chris Lattner193de902006-12-06 05:27:40 +00001498unsigned SlotMachine::getOrCreateSlot(const Value *V) {
Chris Lattner0dda8612006-12-06 06:15:43 +00001499 const Type* VTy = V->getType();
1500 assert(VTy != Type::VoidTy && !V->hasName() && "Doesn't need a slot!");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001501 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1502 "Can't insert a non-GlobalValue Constant into SlotMachine");
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001503
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001504 // Look up the type plane for the Value's type from the module map
1505 TypedPlanes::const_iterator MI = mMap.find(VTy);
1506
Chris Lattner23e829a2006-12-06 05:12:21 +00001507 if (TheFunction) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001508 // Get the type plane for the Value's type from the function map
1509 TypedPlanes::const_iterator FI = fMap.find(VTy);
1510 // If there is a corresponding type plane in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001511 if (FI != fMap.end()) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001512 // Lookup the Value in the function map
1513 ValueMap::const_iterator FVI = FI->second.map.find(V);
1514 // If the value doesn't exist in the function map
Chris Lattner23e829a2006-12-06 05:12:21 +00001515 if (FVI == FI->second.map.end()) {
Reid Spencer56010e42004-05-26 21:56:09 +00001516 // If there is no corresponding type plane in the module map
Chris Lattner23e829a2006-12-06 05:12:21 +00001517 if (MI == mMap.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001518 return insertValue(V);
1519 // Look up the value in the module map
1520 ValueMap::const_iterator MVI = MI->second.map.find(V);
1521 // If we didn't find it, it wasn't inserted
Chris Lattner23e829a2006-12-06 05:12:21 +00001522 if (MVI == MI->second.map.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001523 return insertValue(V);
1524 else
1525 // We found it only at the module level
1526 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001527
1528 // else the value exists in the function map
1529 } else {
Chris Lattner23e829a2006-12-06 05:12:21 +00001530 if (MI == mMap.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001531 return FVI->second;
1532 else
1533 // Return the slot number as the module's contribution to
1534 // the type plane plus the index in the function's contribution
1535 // to the type plane.
1536 return MI->second.next_slot + FVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001537 }
1538
1539 // else there is not a corresponding type plane in the function map
1540 } else {
1541 // If the type plane doesn't exists at the module level
Chris Lattner23e829a2006-12-06 05:12:21 +00001542 if (MI == mMap.end()) {
Reid Spencer56010e42004-05-26 21:56:09 +00001543 return insertValue(V);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001544 // else type plane exists at the module level, examine it
1545 } else {
Reid Spencer56010e42004-05-26 21:56:09 +00001546 // Look up the value in the module's map
1547 ValueMap::const_iterator MVI = MI->second.map.find(V);
1548 // If we didn't find it there either
Chris Lattner23e829a2006-12-06 05:12:21 +00001549 if (MVI == MI->second.map.end())
Reid Spencer56010e42004-05-26 21:56:09 +00001550 // Return the slot number as the module's contribution to
1551 // the type plane plus the index of the function map insertion.
1552 return MI->second.next_slot + insertValue(V);
1553 else
1554 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001555 }
1556 }
1557 }
1558
Chris Lattner0dda8612006-12-06 06:15:43 +00001559 // N.B. Can only get here if TheFunction == 0
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001560
1561 // If the module map's type plane is not for the Value's type
Chris Lattner23e829a2006-12-06 05:12:21 +00001562 if (MI != mMap.end()) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001563 // Lookup the value in the module's map
1564 ValueMap::const_iterator MVI = MI->second.map.find(V);
Chris Lattner23e829a2006-12-06 05:12:21 +00001565 if (MVI != MI->second.map.end())
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001566 return MVI->second;
1567 }
1568
1569 return insertValue(V);
1570}
1571
1572
1573// Low level insert function. Minimal checking is done. This
Chris Lattner193de902006-12-06 05:27:40 +00001574// function is just for the convenience of getOrCreateSlot (above).
Chris Lattner23e829a2006-12-06 05:12:21 +00001575unsigned SlotMachine::insertValue(const Value *V) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001576 assert(V && "Can't insert a null Value into SlotMachine!");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001577 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
Chris Lattner7e4e7662006-12-06 05:42:32 +00001578 "Can't insert a non-GlobalValue Constant into SlotMachine");
1579 assert(V->getType() != Type::VoidTy && !V->hasName());
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001580
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001581 const Type *VTy = V->getType();
1582 unsigned DestSlot = 0;
1583
Chris Lattner23e829a2006-12-06 05:12:21 +00001584 if (TheFunction) {
1585 TypedPlanes::iterator I = fMap.find(VTy);
1586 if (I == fMap.end())
Reid Spencer58d30f22004-07-04 11:50:43 +00001587 I = fMap.insert(std::make_pair(VTy,ValuePlane())).first;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001588 DestSlot = I->second.map[V] = I->second.next_slot++;
1589 } else {
Chris Lattner23e829a2006-12-06 05:12:21 +00001590 TypedPlanes::iterator I = mMap.find(VTy);
1591 if (I == mMap.end())
Reid Spencer58d30f22004-07-04 11:50:43 +00001592 I = mMap.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 }
1595
Misha Brukmanb1c93172005-04-21 23:48:37 +00001596 SC_DEBUG(" Inserting value [" << VTy << "] = " << V << " slot=" <<
Reid Spencer56010e42004-05-26 21:56:09 +00001597 DestSlot << " [");
Chris Lattnerf8090cb2006-12-06 05:55:41 +00001598 // G = Global, F = Function, o = other
1599 SC_DEBUG((isa<GlobalVariable>(V) ? 'G' : (isa<Function>(V) ? 'F' : 'o')));
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001600 SC_DEBUG("]\n");
1601 return DestSlot;
1602}