blob: 349e1bd15fdc56e50e7805a2057cd302cce74966 [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 Lattner7db79582001-11-07 04:21:57 +000017#include "llvm/Assembly/CachedWriter.h"
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +000018#include "llvm/Assembly/Writer.h"
Chris Lattner7f8845a2002-07-23 18:07:49 +000019#include "llvm/Assembly/PrintModulePass.h"
Chris Lattner8339f7d2003-10-30 23:41:03 +000020#include "llvm/Assembly/AsmAnnotationWriter.h"
Chris Lattnerf7b6d312005-05-06 20:26:43 +000021#include "llvm/CallingConv.h"
Chris Lattnerc70b3f62004-01-20 19:50:34 +000022#include "llvm/Constants.h"
Chris Lattner913d18f2002-04-29 18:46:50 +000023#include "llvm/DerivedTypes.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"
Misha Brukman4685e262004-04-28 15:31:21 +000028#include "llvm/Assembly/Writer.h"
Chris Lattner58185f22002-10-02 19:38:55 +000029#include "llvm/Support/CFG.h"
Reid Spencer7c16caa2004-09-01 22:55:40 +000030#include "llvm/ADT/StringExtras.h"
31#include "llvm/ADT/STLExtras.h"
Jim Laskeyb74c6662005-08-17 19:34:49 +000032#include "llvm/Support/MathExtras.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;
Reid Spencer58d30f22004-07-04 11:50:43 +000051 typedef std::map<const Type*, unsigned> TypeMap;
Reid Spencer16f2f7f2004-05-26 07:18:52 +000052
53 /// @brief A plane with next slot number and ValueMap
Misha Brukmanb1c93172005-04-21 23:48:37 +000054 struct ValuePlane {
Reid Spencer16f2f7f2004-05-26 07:18:52 +000055 unsigned next_slot; ///< The next slot number to use
56 ValueMap map; ///< The map of Value* -> unsigned
Reid Spencer58d30f22004-07-04 11:50:43 +000057 ValuePlane() { next_slot = 0; } ///< Make sure we start at 0
58 };
59
60 struct TypePlane {
61 unsigned next_slot;
62 TypeMap map;
63 TypePlane() { next_slot = 0; }
64 void clear() { map.clear(); next_slot = 0; }
Reid Spencer16f2f7f2004-05-26 07:18:52 +000065 };
66
67 /// @brief The map of planes by Type
Reid Spencer58d30f22004-07-04 11:50:43 +000068 typedef std::map<const Type*, ValuePlane> TypedPlanes;
Reid Spencer16f2f7f2004-05-26 07:18:52 +000069
70/// @}
71/// @name Constructors
72/// @{
73public:
74 /// @brief Construct from a module
75 SlotMachine(const Module *M );
76
77 /// @brief Construct from a function, starting out in incorp state.
78 SlotMachine(const Function *F );
79
80/// @}
81/// @name Accessors
82/// @{
83public:
84 /// Return the slot number of the specified value in it's type
85 /// plane. Its an error to ask for something not in the SlotMachine.
86 /// Its an error to ask for a Type*
Chris Lattner757ee0b2004-06-09 19:41:19 +000087 int getSlot(const Value *V);
Reid Spencer58d30f22004-07-04 11:50:43 +000088 int getSlot(const Type*Ty);
Reid Spencer8beac692004-06-09 15:26:53 +000089
90 /// Determine if a Value has a slot or not
91 bool hasSlot(const Value* V);
Reid Spencer58d30f22004-07-04 11:50:43 +000092 bool hasSlot(const Type* Ty);
Reid Spencer16f2f7f2004-05-26 07:18:52 +000093
94/// @}
95/// @name Mutators
96/// @{
97public:
Misha Brukmanb1c93172005-04-21 23:48:37 +000098 /// If you'd like to deal with a function instead of just a module, use
Reid Spencer16f2f7f2004-05-26 07:18:52 +000099 /// this method to get its data into the SlotMachine.
Misha Brukmanb1c93172005-04-21 23:48:37 +0000100 void incorporateFunction(const Function *F) {
101 TheFunction = F;
Reid Spencerb0ac8c42004-08-16 07:46:33 +0000102 FunctionProcessed = false;
103 }
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000104
Misha Brukmanb1c93172005-04-21 23:48:37 +0000105 /// After calling incorporateFunction, use this method to remove the
106 /// most recently incorporated function from the SlotMachine. This
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000107 /// will reset the state of the machine back to just the module contents.
108 void purgeFunction();
109
110/// @}
111/// @name Implementation Details
112/// @{
113private:
Reid Spencer56010e42004-05-26 21:56:09 +0000114 /// This function does the actual initialization.
115 inline void initialize();
116
Misha Brukmanb1c93172005-04-21 23:48:37 +0000117 /// Values can be crammed into here at will. If they haven't
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000118 /// been inserted already, they get inserted, otherwise they are ignored.
119 /// Either way, the slot number for the Value* is returned.
120 unsigned createSlot(const Value *V);
Reid Spencer58d30f22004-07-04 11:50:43 +0000121 unsigned createSlot(const Type* Ty);
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000122
123 /// Insert a value into the value table. Return the slot number
124 /// that it now occupies. BadThings(TM) will happen if you insert a
Misha Brukmanb1c93172005-04-21 23:48:37 +0000125 /// Value that's already been inserted.
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000126 unsigned insertValue( const Value *V );
Reid Spencer58d30f22004-07-04 11:50:43 +0000127 unsigned insertValue( const Type* Ty);
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000128
129 /// Add all of the module level global variables (and their initializers)
130 /// and function declarations, but not the contents of those functions.
131 void processModule();
132
Reid Spencer56010e42004-05-26 21:56:09 +0000133 /// Add all of the functions arguments, basic blocks, and instructions
134 void processFunction();
135
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000136 SlotMachine(const SlotMachine &); // DO NOT IMPLEMENT
137 void operator=(const SlotMachine &); // DO NOT IMPLEMENT
138
139/// @}
140/// @name Data
141/// @{
142public:
143
144 /// @brief The module for which we are holding slot numbers
Reid Spencer56010e42004-05-26 21:56:09 +0000145 const Module* TheModule;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000146
Reid Spencer56010e42004-05-26 21:56:09 +0000147 /// @brief The function for which we are holding slot numbers
148 const Function* TheFunction;
Reid Spencerb0ac8c42004-08-16 07:46:33 +0000149 bool FunctionProcessed;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000150
151 /// @brief The TypePlanes map for the module level data
152 TypedPlanes mMap;
Reid Spencer58d30f22004-07-04 11:50:43 +0000153 TypePlane mTypes;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000154
155 /// @brief The TypePlanes map for the function level data
156 TypedPlanes fMap;
Reid Spencer58d30f22004-07-04 11:50:43 +0000157 TypePlane fTypes;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000158
159/// @}
160
161};
162
Chris Lattner60a7dd12004-07-15 02:51:31 +0000163} // end namespace llvm
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000164
Chris Lattnerc8b70922002-07-26 21:12:46 +0000165static RegisterPass<PrintModulePass>
166X("printm", "Print module to stderr",PassInfo::Analysis|PassInfo::Optimization);
167static RegisterPass<PrintFunctionPass>
168Y("print","Print function to stderr",PassInfo::Analysis|PassInfo::Optimization);
Chris Lattner7f8845a2002-07-23 18:07:49 +0000169
Misha Brukmanb1c93172005-04-21 23:48:37 +0000170static void WriteAsOperandInternal(std::ostream &Out, const Value *V,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000171 bool PrintName,
172 std::map<const Type *, std::string> &TypeTable,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000173 SlotMachine *Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000174
Misha Brukmanb1c93172005-04-21 23:48:37 +0000175static void WriteAsOperandInternal(std::ostream &Out, const Type *T,
Reid Spencer58d30f22004-07-04 11:50:43 +0000176 bool PrintName,
177 std::map<const Type *, std::string> &TypeTable,
178 SlotMachine *Machine);
179
Chris Lattnerb86620e2001-10-29 16:37:48 +0000180static const Module *getModuleFromVal(const Value *V) {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000181 if (const Argument *MA = dyn_cast<Argument>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000182 return MA->getParent() ? MA->getParent()->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000183 else if (const BasicBlock *BB = dyn_cast<BasicBlock>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000184 return BB->getParent() ? BB->getParent()->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000185 else if (const Instruction *I = dyn_cast<Instruction>(V)) {
Chris Lattner57698e22002-03-26 18:01:55 +0000186 const Function *M = I->getParent() ? I->getParent()->getParent() : 0;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000187 return M ? M->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000188 } else if (const GlobalValue *GV = dyn_cast<GlobalValue>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000189 return GV->getParent();
Chris Lattnerb86620e2001-10-29 16:37:48 +0000190 return 0;
191}
192
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000193static SlotMachine *createSlotMachine(const Value *V) {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000194 if (const Argument *FA = dyn_cast<Argument>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000195 return new SlotMachine(FA->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000196 } else if (const Instruction *I = dyn_cast<Instruction>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000197 return new SlotMachine(I->getParent()->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000198 } else if (const BasicBlock *BB = dyn_cast<BasicBlock>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000199 return new SlotMachine(BB->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000200 } else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(V)){
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000201 return new SlotMachine(GV->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000202 } else if (const Function *Func = dyn_cast<Function>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000203 return new SlotMachine(Func);
Chris Lattner7bfee412001-10-29 16:05:51 +0000204 }
205 return 0;
206}
Chris Lattner2f7c9632001-06-06 20:29:01 +0000207
Chris Lattner1c343042003-08-22 05:40:38 +0000208// getLLVMName - Turn the specified string into an 'LLVM name', which is either
209// prefixed with % (if the string only contains simple characters) or is
210// surrounded with ""'s (if it has special chars in it).
Alkis Evlogimenos991d6ad2004-12-10 05:41:10 +0000211static std::string getLLVMName(const std::string &Name,
212 bool prefixName = true) {
Chris Lattner1c343042003-08-22 05:40:38 +0000213 assert(!Name.empty() && "Cannot get empty name!");
214
215 // First character cannot start with a number...
216 if (Name[0] >= '0' && Name[0] <= '9')
217 return "\"" + Name + "\"";
218
219 // Scan to see if we have any characters that are not on the "white list"
220 for (unsigned i = 0, e = Name.size(); i != e; ++i) {
221 char C = Name[i];
222 assert(C != '"' && "Illegal character in LLVM value name!");
223 if ((C < 'a' || C > 'z') && (C < 'A' || C > 'Z') && (C < '0' || C > '9') &&
224 C != '-' && C != '.' && C != '_')
225 return "\"" + Name + "\"";
226 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000227
Chris Lattner1c343042003-08-22 05:40:38 +0000228 // If we get here, then the identifier is legal to use as a "VarID".
Alkis Evlogimenos991d6ad2004-12-10 05:41:10 +0000229 if (prefixName)
230 return "%"+Name;
231 else
232 return Name;
Chris Lattner1c343042003-08-22 05:40:38 +0000233}
234
Chris Lattnerb86620e2001-10-29 16:37:48 +0000235
Misha Brukmanc566ca362004-03-02 00:22:19 +0000236/// fillTypeNameTable - If the module has a symbol table, take all global types
237/// and stuff their names into the TypeNames map.
238///
Chris Lattnerb86620e2001-10-29 16:37:48 +0000239static void fillTypeNameTable(const Module *M,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000240 std::map<const Type *, std::string> &TypeNames) {
Chris Lattner98cf1f52002-11-20 18:36:02 +0000241 if (!M) return;
242 const SymbolTable &ST = M->getSymbolTable();
Reid Spencere7e96712004-05-25 08:53:40 +0000243 SymbolTable::type_const_iterator TI = ST.type_begin();
244 for (; TI != ST.type_end(); ++TI ) {
245 // As a heuristic, don't insert pointer to primitive types, because
246 // they are used too often to have a single useful name.
247 //
248 const Type *Ty = cast<Type>(TI->second);
249 if (!isa<PointerType>(Ty) ||
Reid Spencer56010e42004-05-26 21:56:09 +0000250 !cast<PointerType>(Ty)->getElementType()->isPrimitiveType() ||
251 isa<OpaqueType>(cast<PointerType>(Ty)->getElementType()))
Reid Spencere7e96712004-05-25 08:53:40 +0000252 TypeNames.insert(std::make_pair(Ty, getLLVMName(TI->first)));
Chris Lattnerb86620e2001-10-29 16:37:48 +0000253 }
254}
255
256
257
Misha Brukmanb1c93172005-04-21 23:48:37 +0000258static void calcTypeName(const Type *Ty,
John Criswellcd116ba2004-06-01 14:54:08 +0000259 std::vector<const Type *> &TypeStack,
260 std::map<const Type *, std::string> &TypeNames,
261 std::string & Result){
262 if (Ty->isPrimitiveType() && !isa<OpaqueType>(Ty)) {
263 Result += Ty->getDescription(); // Base case
264 return;
265 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000266
267 // Check to see if the type is named.
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000268 std::map<const Type *, std::string>::iterator I = TypeNames.find(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000269 if (I != TypeNames.end()) {
270 Result += I->second;
271 return;
272 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000273
John Criswellcd116ba2004-06-01 14:54:08 +0000274 if (isa<OpaqueType>(Ty)) {
275 Result += "opaque";
276 return;
277 }
Chris Lattnerf14ead92003-10-30 00:22:33 +0000278
Chris Lattnerb86620e2001-10-29 16:37:48 +0000279 // Check to see if the Type is already on the stack...
280 unsigned Slot = 0, CurSize = TypeStack.size();
281 while (Slot < CurSize && TypeStack[Slot] != Ty) ++Slot; // Scan for type
282
Misha Brukmanb1c93172005-04-21 23:48:37 +0000283 // This is another base case for the recursion. In this case, we know
Chris Lattnerb86620e2001-10-29 16:37:48 +0000284 // that we have looped back to a type that we have previously visited.
285 // Generate the appropriate upreference to handle this.
John Criswellcd116ba2004-06-01 14:54:08 +0000286 if (Slot < CurSize) {
287 Result += "\\" + utostr(CurSize-Slot); // Here's the upreference
288 return;
289 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000290
291 TypeStack.push_back(Ty); // Recursive case: Add us to the stack..
Misha Brukmanb1c93172005-04-21 23:48:37 +0000292
Chris Lattner6b727592004-06-17 18:19:28 +0000293 switch (Ty->getTypeID()) {
Chris Lattner91db5822002-03-29 03:44:36 +0000294 case Type::FunctionTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000295 const FunctionType *FTy = cast<FunctionType>(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000296 calcTypeName(FTy->getReturnType(), TypeStack, TypeNames, Result);
297 Result += " (";
Chris Lattnerfa829be2004-02-09 04:14:01 +0000298 for (FunctionType::param_iterator I = FTy->param_begin(),
299 E = FTy->param_end(); I != E; ++I) {
300 if (I != FTy->param_begin())
Chris Lattnerb86620e2001-10-29 16:37:48 +0000301 Result += ", ";
John Criswellcd116ba2004-06-01 14:54:08 +0000302 calcTypeName(*I, TypeStack, TypeNames, Result);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000303 }
Chris Lattnerd816b532002-04-13 20:53:41 +0000304 if (FTy->isVarArg()) {
Chris Lattnerfa829be2004-02-09 04:14:01 +0000305 if (FTy->getNumParams()) Result += ", ";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000306 Result += "...";
307 }
308 Result += ")";
309 break;
310 }
311 case Type::StructTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000312 const StructType *STy = cast<StructType>(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000313 Result += "{ ";
Chris Lattnerac6db752004-02-09 04:37:31 +0000314 for (StructType::element_iterator I = STy->element_begin(),
315 E = STy->element_end(); I != E; ++I) {
316 if (I != STy->element_begin())
Chris Lattnerb86620e2001-10-29 16:37:48 +0000317 Result += ", ";
John Criswellcd116ba2004-06-01 14:54:08 +0000318 calcTypeName(*I, TypeStack, TypeNames, Result);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000319 }
320 Result += " }";
321 break;
322 }
323 case Type::PointerTyID:
Misha Brukmanb1c93172005-04-21 23:48:37 +0000324 calcTypeName(cast<PointerType>(Ty)->getElementType(),
John Criswellcd116ba2004-06-01 14:54:08 +0000325 TypeStack, TypeNames, Result);
326 Result += "*";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000327 break;
328 case Type::ArrayTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000329 const ArrayType *ATy = cast<ArrayType>(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000330 Result += "[" + utostr(ATy->getNumElements()) + " x ";
331 calcTypeName(ATy->getElementType(), TypeStack, TypeNames, Result);
332 Result += "]";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000333 break;
334 }
Brian Gaeke02209042004-08-20 06:00:58 +0000335 case Type::PackedTyID: {
336 const PackedType *PTy = cast<PackedType>(Ty);
337 Result += "<" + utostr(PTy->getNumElements()) + " x ";
338 calcTypeName(PTy->getElementType(), TypeStack, TypeNames, Result);
339 Result += ">";
340 break;
341 }
Chris Lattner15285ab2003-05-14 17:50:47 +0000342 case Type::OpaqueTyID:
John Criswellcd116ba2004-06-01 14:54:08 +0000343 Result += "opaque";
Chris Lattner15285ab2003-05-14 17:50:47 +0000344 break;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000345 default:
John Criswellcd116ba2004-06-01 14:54:08 +0000346 Result += "<unrecognized-type>";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000347 }
348
349 TypeStack.pop_back(); // Remove self from stack...
John Criswellcd116ba2004-06-01 14:54:08 +0000350 return;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000351}
352
353
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000354/// printTypeInt - The internal guts of printing out a type that has a
355/// potentially named portion.
356///
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000357static std::ostream &printTypeInt(std::ostream &Out, const Type *Ty,
358 std::map<const Type *, std::string> &TypeNames) {
Chris Lattnerb86620e2001-10-29 16:37:48 +0000359 // Primitive types always print out their description, regardless of whether
360 // they have been named or not.
361 //
Chris Lattner92d60532003-10-30 00:12:51 +0000362 if (Ty->isPrimitiveType() && !isa<OpaqueType>(Ty))
363 return Out << Ty->getDescription();
Chris Lattnerb86620e2001-10-29 16:37:48 +0000364
365 // Check to see if the type is named.
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000366 std::map<const Type *, std::string>::iterator I = TypeNames.find(Ty);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000367 if (I != TypeNames.end()) return Out << I->second;
368
369 // Otherwise we have a type that has not been named but is a derived type.
370 // Carefully recurse the type hierarchy to print out any contained symbolic
371 // names.
372 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000373 std::vector<const Type *> TypeStack;
John Criswellcd116ba2004-06-01 14:54:08 +0000374 std::string TypeName;
375 calcTypeName(Ty, TypeStack, TypeNames, TypeName);
Chris Lattner7f74a562002-01-20 22:54:45 +0000376 TypeNames.insert(std::make_pair(Ty, TypeName));//Cache type name for later use
John Criswellcd116ba2004-06-01 14:54:08 +0000377 return (Out << TypeName);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000378}
379
Chris Lattner34b95182001-10-31 04:33:19 +0000380
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000381/// WriteTypeSymbolic - This attempts to write the specified type as a symbolic
382/// type, iff there is an entry in the modules symbol table for the specified
383/// type or one of it's component types. This is slower than a simple x << Type
384///
Chris Lattner189d19f2003-11-21 20:23:48 +0000385std::ostream &llvm::WriteTypeSymbolic(std::ostream &Out, const Type *Ty,
386 const Module *M) {
Misha Brukmanb1c93172005-04-21 23:48:37 +0000387 Out << ' ';
Chris Lattnerb86620e2001-10-29 16:37:48 +0000388
389 // If they want us to print out a type, attempt to make it symbolic if there
390 // is a symbol table in the module...
Chris Lattner98cf1f52002-11-20 18:36:02 +0000391 if (M) {
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000392 std::map<const Type *, std::string> TypeNames;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000393 fillTypeNameTable(M, TypeNames);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000394
Chris Lattner72f866e2001-10-29 16:40:32 +0000395 return printTypeInt(Out, Ty, TypeNames);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000396 } else {
Chris Lattner72f866e2001-10-29 16:40:32 +0000397 return Out << Ty->getDescription();
Chris Lattnerb86620e2001-10-29 16:37:48 +0000398 }
399}
400
Misha Brukmanb1c93172005-04-21 23:48:37 +0000401/// @brief Internal constant writer.
402static void WriteConstantInt(std::ostream &Out, const Constant *CV,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000403 bool PrintName,
404 std::map<const Type *, std::string> &TypeTable,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000405 SlotMachine *Machine) {
Chris Lattner1e194682002-04-18 18:53:13 +0000406 if (const ConstantBool *CB = dyn_cast<ConstantBool>(CV)) {
407 Out << (CB == ConstantBool::True ? "true" : "false");
408 } else if (const ConstantSInt *CI = dyn_cast<ConstantSInt>(CV)) {
409 Out << CI->getValue();
410 } else if (const ConstantUInt *CI = dyn_cast<ConstantUInt>(CV)) {
411 Out << CI->getValue();
412 } else if (const ConstantFP *CFP = dyn_cast<ConstantFP>(CV)) {
413 // We would like to output the FP constant value in exponential notation,
414 // but we cannot do this if doing so will lose precision. Check here to
415 // make sure that we only output it in exponential format if we can parse
416 // the value back and get the same value.
417 //
418 std::string StrVal = ftostr(CFP->getValue());
419
420 // Check to make sure that the stringized number is not some string like
421 // "Inf" or NaN, that atof will accept, but the lexer will not. Check that
422 // the string matches the "[-+]?[0-9]" regex.
423 //
424 if ((StrVal[0] >= '0' && StrVal[0] <= '9') ||
425 ((StrVal[0] == '-' || StrVal[0] == '+') &&
Brian Gaeke87b4f072003-06-17 23:55:35 +0000426 (StrVal[1] >= '0' && StrVal[1] <= '9')))
Chris Lattner1e194682002-04-18 18:53:13 +0000427 // Reparse stringized version!
428 if (atof(StrVal.c_str()) == CFP->getValue()) {
Chris Lattner472cc102005-01-04 01:56:57 +0000429 Out << StrVal;
430 return;
Chris Lattner1e194682002-04-18 18:53:13 +0000431 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000432
Chris Lattner1e194682002-04-18 18:53:13 +0000433 // Otherwise we could not reparse it to exactly the same value, so we must
434 // output the string in hexadecimal format!
Chris Lattner472cc102005-01-04 01:56:57 +0000435 assert(sizeof(double) == sizeof(uint64_t) &&
Chris Lattner1e194682002-04-18 18:53:13 +0000436 "assuming that double is 64 bits!");
Jim Laskeyb74c6662005-08-17 19:34:49 +0000437 Out << "0x" << utohexstr(DoubleToBits(CFP->getValue()));
Chris Lattner1e194682002-04-18 18:53:13 +0000438
Chris Lattner76b2ff42004-02-15 05:55:15 +0000439 } else if (isa<ConstantAggregateZero>(CV)) {
440 Out << "zeroinitializer";
Chris Lattner1e194682002-04-18 18:53:13 +0000441 } else if (const ConstantArray *CA = dyn_cast<ConstantArray>(CV)) {
442 // As a special case, print the array as a string if it is an array of
443 // ubytes or an array of sbytes with positive values.
Misha Brukmanb1c93172005-04-21 23:48:37 +0000444 //
Chris Lattner1e194682002-04-18 18:53:13 +0000445 const Type *ETy = CA->getType()->getElementType();
446 bool isString = (ETy == Type::SByteTy || ETy == Type::UByteTy);
447
448 if (ETy == Type::SByteTy)
449 for (unsigned i = 0; i < CA->getNumOperands(); ++i)
450 if (cast<ConstantSInt>(CA->getOperand(i))->getValue() < 0) {
451 isString = false;
452 break;
453 }
454
455 if (isString) {
456 Out << "c\"";
457 for (unsigned i = 0; i < CA->getNumOperands(); ++i) {
Misha Brukmanb1c93172005-04-21 23:48:37 +0000458 unsigned char C =
Chris Lattneraa6ff272004-06-04 23:53:20 +0000459 (unsigned char)cast<ConstantInt>(CA->getOperand(i))->getRawValue();
Misha Brukmanb1c93172005-04-21 23:48:37 +0000460
Chris Lattnere5fd3862002-07-31 23:56:44 +0000461 if (isprint(C) && C != '"' && C != '\\') {
Chris Lattner1e194682002-04-18 18:53:13 +0000462 Out << C;
463 } else {
464 Out << '\\'
465 << (char) ((C/16 < 10) ? ( C/16 +'0') : ( C/16 -10+'A'))
466 << (char)(((C&15) < 10) ? ((C&15)+'0') : ((C&15)-10+'A'));
467 }
468 }
469 Out << "\"";
470
471 } else { // Cannot output in string format...
Misha Brukman21bbdb92004-06-04 21:11:51 +0000472 Out << '[';
Chris Lattnerd84bb632002-04-16 21:36:08 +0000473 if (CA->getNumOperands()) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000474 Out << ' ';
Chris Lattner1e194682002-04-18 18:53:13 +0000475 printTypeInt(Out, ETy, TypeTable);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000476 WriteAsOperandInternal(Out, CA->getOperand(0),
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000477 PrintName, TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000478 for (unsigned i = 1, e = CA->getNumOperands(); i != e; ++i) {
479 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(i), PrintName,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000482 TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000483 }
484 }
485 Out << " ]";
486 }
487 } else if (const ConstantStruct *CS = dyn_cast<ConstantStruct>(CV)) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000488 Out << '{';
Chris Lattnerd84bb632002-04-16 21:36:08 +0000489 if (CS->getNumOperands()) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000490 Out << ' ';
Chris Lattnerd84bb632002-04-16 21:36:08 +0000491 printTypeInt(Out, CS->getOperand(0)->getType(), TypeTable);
492
493 WriteAsOperandInternal(Out, CS->getOperand(0),
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000494 PrintName, TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000495
496 for (unsigned i = 1; i < CS->getNumOperands(); i++) {
497 Out << ", ";
498 printTypeInt(Out, CS->getOperand(i)->getType(), TypeTable);
499
500 WriteAsOperandInternal(Out, CS->getOperand(i),
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000501 PrintName, TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000502 }
503 }
504
505 Out << " }";
Brian Gaeke02209042004-08-20 06:00:58 +0000506 } else if (const ConstantPacked *CP = dyn_cast<ConstantPacked>(CV)) {
507 const Type *ETy = CP->getType()->getElementType();
Misha Brukmanb1c93172005-04-21 23:48:37 +0000508 assert(CP->getNumOperands() > 0 &&
Brian Gaeke02209042004-08-20 06:00:58 +0000509 "Number of operands for a PackedConst must be > 0");
510 Out << '<';
511 Out << ' ';
512 printTypeInt(Out, ETy, TypeTable);
513 WriteAsOperandInternal(Out, CP->getOperand(0),
514 PrintName, TypeTable, Machine);
515 for (unsigned i = 1, e = CP->getNumOperands(); i != e; ++i) {
516 Out << ", ";
517 printTypeInt(Out, ETy, TypeTable);
518 WriteAsOperandInternal(Out, CP->getOperand(i), PrintName,
519 TypeTable, Machine);
520 }
521 Out << " >";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000522 } else if (isa<ConstantPointerNull>(CV)) {
523 Out << "null";
524
Chris Lattner5e0b9f22004-10-16 18:08:06 +0000525 } else if (isa<UndefValue>(CV)) {
526 Out << "undef";
527
Chris Lattner3cd8c562002-07-30 18:54:25 +0000528 } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CV)) {
Chris Lattnerb8257222003-03-06 23:23:32 +0000529 Out << CE->getOpcodeName() << " (";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000530
Vikram S. Adveb952b542002-07-14 23:14:45 +0000531 for (User::const_op_iterator OI=CE->op_begin(); OI != CE->op_end(); ++OI) {
532 printTypeInt(Out, (*OI)->getType(), TypeTable);
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000533 WriteAsOperandInternal(Out, *OI, PrintName, TypeTable, Machine);
Vikram S. Adveb952b542002-07-14 23:14:45 +0000534 if (OI+1 != CE->op_end())
Chris Lattner3cd8c562002-07-30 18:54:25 +0000535 Out << ", ";
Vikram S. Adveb952b542002-07-14 23:14:45 +0000536 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000537
Chris Lattnercfe8f532002-08-16 21:17:11 +0000538 if (CE->getOpcode() == Instruction::Cast) {
Chris Lattner83b396b2002-08-15 19:37:43 +0000539 Out << " to ";
540 printTypeInt(Out, CE->getType(), TypeTable);
541 }
Misha Brukman21bbdb92004-06-04 21:11:51 +0000542 Out << ')';
Chris Lattner83b396b2002-08-15 19:37:43 +0000543
Chris Lattnerd84bb632002-04-16 21:36:08 +0000544 } else {
Vikram S. Adveb952b542002-07-14 23:14:45 +0000545 Out << "<placeholder or erroneous Constant>";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000546 }
547}
548
549
Misha Brukmanc566ca362004-03-02 00:22:19 +0000550/// WriteAsOperand - Write the name of the specified value out to the specified
551/// ostream. This can be useful when you just want to print int %reg126, not
552/// the whole instruction that generated it.
553///
Misha Brukmanb1c93172005-04-21 23:48:37 +0000554static void WriteAsOperandInternal(std::ostream &Out, const Value *V,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000555 bool PrintName,
556 std::map<const Type*, std::string> &TypeTable,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000557 SlotMachine *Machine) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000558 Out << ' ';
Reid Spenceraccd7c72004-07-17 23:47:01 +0000559 if ((PrintName || isa<GlobalValue>(V)) && V->hasName())
Chris Lattner1c343042003-08-22 05:40:38 +0000560 Out << getLLVMName(V->getName());
Reid Spenceraccd7c72004-07-17 23:47:01 +0000561 else {
562 const Constant *CV = dyn_cast<Constant>(V);
563 if (CV && !isa<GlobalValue>(CV))
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000564 WriteConstantInt(Out, CV, PrintName, TypeTable, Machine);
Reid Spenceraccd7c72004-07-17 23:47:01 +0000565 else {
Chris Lattnerd84bb632002-04-16 21:36:08 +0000566 int Slot;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000567 if (Machine) {
Reid Spencer56010e42004-05-26 21:56:09 +0000568 Slot = Machine->getSlot(V);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000569 } else {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000570 Machine = createSlotMachine(V);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000571 if (Machine == 0)
Chris Lattner757ee0b2004-06-09 19:41:19 +0000572 Slot = Machine->getSlot(V);
573 else
574 Slot = -1;
Reid Spencer56010e42004-05-26 21:56:09 +0000575 delete Machine;
Chris Lattnerd84bb632002-04-16 21:36:08 +0000576 }
Chris Lattner757ee0b2004-06-09 19:41:19 +0000577 if (Slot != -1)
578 Out << '%' << Slot;
579 else
580 Out << "<badref>";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000581 }
582 }
583}
584
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000585/// WriteAsOperand - Write the name of the specified value out to the specified
586/// ostream. This can be useful when you just want to print int %reg126, not
587/// the whole instruction that generated it.
588///
Chris Lattner189d19f2003-11-21 20:23:48 +0000589std::ostream &llvm::WriteAsOperand(std::ostream &Out, const Value *V,
Misha Brukmanb1c93172005-04-21 23:48:37 +0000590 bool PrintType, bool PrintName,
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000591 const Module *Context) {
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000592 std::map<const Type *, std::string> TypeNames;
Chris Lattner5a9f63e2002-07-10 16:48:17 +0000593 if (Context == 0) Context = getModuleFromVal(V);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000594
Chris Lattner98cf1f52002-11-20 18:36:02 +0000595 if (Context)
Chris Lattner5a9f63e2002-07-10 16:48:17 +0000596 fillTypeNameTable(Context, TypeNames);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000597
598 if (PrintType)
599 printTypeInt(Out, V->getType(), TypeNames);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000600
Chris Lattner5a9f63e2002-07-10 16:48:17 +0000601 WriteAsOperandInternal(Out, V, PrintName, TypeNames, 0);
Chris Lattner5e5abe32001-07-20 19:15:21 +0000602 return Out;
603}
604
Misha Brukmanb1c93172005-04-21 23:48:37 +0000605/// WriteAsOperandInternal - Write the name of the specified value out to
606/// the specified ostream. This can be useful when you just want to print
Reid Spencer58d30f22004-07-04 11:50:43 +0000607/// int %reg126, not the whole instruction that generated it.
608///
Misha Brukmanb1c93172005-04-21 23:48:37 +0000609static void WriteAsOperandInternal(std::ostream &Out, const Type *T,
Reid Spencer58d30f22004-07-04 11:50:43 +0000610 bool PrintName,
611 std::map<const Type*, std::string> &TypeTable,
612 SlotMachine *Machine) {
613 Out << ' ';
614 int Slot;
615 if (Machine) {
616 Slot = Machine->getSlot(T);
617 if (Slot != -1)
618 Out << '%' << Slot;
619 else
620 Out << "<badref>";
621 } else {
622 Out << T->getDescription();
623 }
624}
625
626/// WriteAsOperand - Write the name of the specified value out to the specified
627/// ostream. This can be useful when you just want to print int %reg126, not
628/// the whole instruction that generated it.
629///
630std::ostream &llvm::WriteAsOperand(std::ostream &Out, const Type *Ty,
Misha Brukmanb1c93172005-04-21 23:48:37 +0000631 bool PrintType, bool PrintName,
Reid Spencer58d30f22004-07-04 11:50:43 +0000632 const Module *Context) {
633 std::map<const Type *, std::string> TypeNames;
634 assert(Context != 0 && "Can't write types as operand without module context");
635
636 fillTypeNameTable(Context, TypeNames);
637
638 // if (PrintType)
639 // printTypeInt(Out, V->getType(), TypeNames);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000640
Reid Spencer58d30f22004-07-04 11:50:43 +0000641 printTypeInt(Out, Ty, TypeNames);
642
643 WriteAsOperandInternal(Out, Ty, PrintName, TypeNames, 0);
644 return Out;
645}
646
Chris Lattner189d19f2003-11-21 20:23:48 +0000647namespace llvm {
Chris Lattner2e9fee42001-07-12 23:35:26 +0000648
Chris Lattnerfee714f2001-09-07 16:36:04 +0000649class AssemblyWriter {
Misha Brukmana6619a92004-06-21 21:53:56 +0000650 std::ostream &Out;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000651 SlotMachine &Machine;
Chris Lattner7bfee412001-10-29 16:05:51 +0000652 const Module *TheModule;
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000653 std::map<const Type *, std::string> TypeNames;
Chris Lattner8339f7d2003-10-30 23:41:03 +0000654 AssemblyAnnotationWriter *AnnotationWriter;
Chris Lattner2f7c9632001-06-06 20:29:01 +0000655public:
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000656 inline AssemblyWriter(std::ostream &o, SlotMachine &Mac, const Module *M,
Chris Lattner8339f7d2003-10-30 23:41:03 +0000657 AssemblyAnnotationWriter *AAW)
Misha Brukmana6619a92004-06-21 21:53:56 +0000658 : Out(o), Machine(Mac), TheModule(M), AnnotationWriter(AAW) {
Chris Lattner7bfee412001-10-29 16:05:51 +0000659
660 // If the module has a symbol table, take all global types and stuff their
661 // names into the TypeNames map.
662 //
Chris Lattnerb86620e2001-10-29 16:37:48 +0000663 fillTypeNameTable(M, TypeNames);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000664 }
665
Chris Lattner7bfee412001-10-29 16:05:51 +0000666 inline void write(const Module *M) { printModule(M); }
667 inline void write(const GlobalVariable *G) { printGlobal(G); }
Chris Lattner57698e22002-03-26 18:01:55 +0000668 inline void write(const Function *F) { printFunction(F); }
Chris Lattner7bfee412001-10-29 16:05:51 +0000669 inline void write(const BasicBlock *BB) { printBasicBlock(BB); }
Chris Lattner113f4f42002-06-25 16:13:24 +0000670 inline void write(const Instruction *I) { printInstruction(*I); }
Chris Lattner3462ae32001-12-03 22:26:30 +0000671 inline void write(const Constant *CPV) { printConstant(CPV); }
Chris Lattner7db79582001-11-07 04:21:57 +0000672 inline void write(const Type *Ty) { printType(Ty); }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000673
Chris Lattner1e194682002-04-18 18:53:13 +0000674 void writeOperand(const Value *Op, bool PrintType, bool PrintName = true);
675
Misha Brukman4685e262004-04-28 15:31:21 +0000676 const Module* getModule() { return TheModule; }
677
Misha Brukmand92f54a2004-11-15 19:30:05 +0000678private:
Chris Lattner7bfee412001-10-29 16:05:51 +0000679 void printModule(const Module *M);
680 void printSymbolTable(const SymbolTable &ST);
Chris Lattner3462ae32001-12-03 22:26:30 +0000681 void printConstant(const Constant *CPV);
Chris Lattner7bfee412001-10-29 16:05:51 +0000682 void printGlobal(const GlobalVariable *GV);
Chris Lattner57698e22002-03-26 18:01:55 +0000683 void printFunction(const Function *F);
Chris Lattner2e9fa6d2002-04-09 19:48:49 +0000684 void printArgument(const Argument *FA);
Chris Lattner7bfee412001-10-29 16:05:51 +0000685 void printBasicBlock(const BasicBlock *BB);
Chris Lattner113f4f42002-06-25 16:13:24 +0000686 void printInstruction(const Instruction &I);
Chris Lattnerd816b532002-04-13 20:53:41 +0000687
688 // printType - Go to extreme measures to attempt to print out a short,
689 // symbolic version of a type name.
690 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000691 std::ostream &printType(const Type *Ty) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000692 return printTypeInt(Out, Ty, TypeNames);
Chris Lattnerd816b532002-04-13 20:53:41 +0000693 }
694
695 // printTypeAtLeastOneLevel - Print out one level of the possibly complex type
696 // without considering any symbolic types that we may have equal to it.
697 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000698 std::ostream &printTypeAtLeastOneLevel(const Type *Ty);
Chris Lattner7bfee412001-10-29 16:05:51 +0000699
Chris Lattner862e3382001-10-13 06:42:36 +0000700 // printInfoComment - Print a little comment after the instruction indicating
701 // which slot it occupies.
Chris Lattner113f4f42002-06-25 16:13:24 +0000702 void printInfoComment(const Value &V);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000703};
Reid Spencerf43ac622004-05-27 22:04:46 +0000704} // end of llvm namespace
Chris Lattner2f7c9632001-06-06 20:29:01 +0000705
Misha Brukmanc566ca362004-03-02 00:22:19 +0000706/// printTypeAtLeastOneLevel - Print out one level of the possibly complex type
707/// without considering any symbolic types that we may have equal to it.
708///
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000709std::ostream &AssemblyWriter::printTypeAtLeastOneLevel(const Type *Ty) {
Chris Lattner113f4f42002-06-25 16:13:24 +0000710 if (const FunctionType *FTy = dyn_cast<FunctionType>(Ty)) {
Chris Lattnerd816b532002-04-13 20:53:41 +0000711 printType(FTy->getReturnType()) << " (";
Chris Lattnerfa829be2004-02-09 04:14:01 +0000712 for (FunctionType::param_iterator I = FTy->param_begin(),
713 E = FTy->param_end(); I != E; ++I) {
714 if (I != FTy->param_begin())
Misha Brukmana6619a92004-06-21 21:53:56 +0000715 Out << ", ";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000716 printType(*I);
Chris Lattnerd816b532002-04-13 20:53:41 +0000717 }
718 if (FTy->isVarArg()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000719 if (FTy->getNumParams()) Out << ", ";
720 Out << "...";
Chris Lattnerd816b532002-04-13 20:53:41 +0000721 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000722 Out << ')';
Chris Lattner113f4f42002-06-25 16:13:24 +0000723 } else if (const StructType *STy = dyn_cast<StructType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000724 Out << "{ ";
Chris Lattnerac6db752004-02-09 04:37:31 +0000725 for (StructType::element_iterator I = STy->element_begin(),
726 E = STy->element_end(); I != E; ++I) {
727 if (I != STy->element_begin())
Misha Brukmana6619a92004-06-21 21:53:56 +0000728 Out << ", ";
Chris Lattnerd816b532002-04-13 20:53:41 +0000729 printType(*I);
730 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000731 Out << " }";
Chris Lattner113f4f42002-06-25 16:13:24 +0000732 } else if (const PointerType *PTy = dyn_cast<PointerType>(Ty)) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000733 printType(PTy->getElementType()) << '*';
Chris Lattner113f4f42002-06-25 16:13:24 +0000734 } else if (const ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000735 Out << '[' << ATy->getNumElements() << " x ";
Misha Brukman21bbdb92004-06-04 21:11:51 +0000736 printType(ATy->getElementType()) << ']';
Reid Spencere203d352004-08-20 15:37:30 +0000737 } else if (const PackedType *PTy = dyn_cast<PackedType>(Ty)) {
738 Out << '<' << PTy->getNumElements() << " x ";
739 printType(PTy->getElementType()) << '>';
740 }
741 else if (const OpaqueType *OTy = dyn_cast<OpaqueType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000742 Out << "opaque";
Chris Lattnerd816b532002-04-13 20:53:41 +0000743 } else {
Vikram S. Adveb952b542002-07-14 23:14:45 +0000744 if (!Ty->isPrimitiveType())
Misha Brukmana6619a92004-06-21 21:53:56 +0000745 Out << "<unknown derived type>";
Chris Lattnerd816b532002-04-13 20:53:41 +0000746 printType(Ty);
747 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000748 return Out;
Chris Lattnerd816b532002-04-13 20:53:41 +0000749}
750
751
Misha Brukmanb1c93172005-04-21 23:48:37 +0000752void AssemblyWriter::writeOperand(const Value *Operand, bool PrintType,
Reid Spencer56010e42004-05-26 21:56:09 +0000753 bool PrintName) {
Chris Lattner08f7d0c2005-02-24 16:58:29 +0000754 if (Operand != 0) {
755 if (PrintType) { Out << ' '; printType(Operand->getType()); }
756 WriteAsOperandInternal(Out, Operand, PrintName, TypeNames, &Machine);
757 } else {
758 Out << "<null operand!>";
759 }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000760}
761
Chris Lattner2f7c9632001-06-06 20:29:01 +0000762
Chris Lattner7bfee412001-10-29 16:05:51 +0000763void AssemblyWriter::printModule(const Module *M) {
Chris Lattner4d8689e2005-03-02 23:12:40 +0000764 if (!M->getModuleIdentifier().empty() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +0000765 // Don't print the ID if it will start a new line (which would
Chris Lattner4d8689e2005-03-02 23:12:40 +0000766 // require a comment char before it).
767 M->getModuleIdentifier().find('\n') == std::string::npos)
768 Out << "; ModuleID = '" << M->getModuleIdentifier() << "'\n";
769
Chris Lattner8068e0c2003-08-24 13:48:48 +0000770 switch (M->getEndianness()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000771 case Module::LittleEndian: Out << "target endian = little\n"; break;
772 case Module::BigEndian: Out << "target endian = big\n"; break;
Chris Lattner8068e0c2003-08-24 13:48:48 +0000773 case Module::AnyEndianness: break;
774 }
775 switch (M->getPointerSize()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000776 case Module::Pointer32: Out << "target pointersize = 32\n"; break;
777 case Module::Pointer64: Out << "target pointersize = 64\n"; break;
Chris Lattner8068e0c2003-08-24 13:48:48 +0000778 case Module::AnyPointerSize: break;
779 }
Reid Spencer48f98c82004-07-25 21:44:54 +0000780 if (!M->getTargetTriple().empty())
Reid Spencerffec7df2004-07-25 21:29:43 +0000781 Out << "target triple = \"" << M->getTargetTriple() << "\"\n";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000782
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 Lattner531f9e92005-03-15 04:54:21 +0000800 for (Module::const_global_iterator I = M->global_begin(), E = M->global_end(); I != E; ++I)
Chris Lattner113f4f42002-06-25 16:13:24 +0000801 printGlobal(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000802
Misha Brukmana6619a92004-06-21 21:53:56 +0000803 Out << "\nimplementation ; Functions:\n";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000804
Chris Lattner2cdd49d2004-09-14 05:06:58 +0000805 // Output all of the functions.
Chris Lattner113f4f42002-06-25 16:13:24 +0000806 for (Module::const_iterator I = M->begin(), E = M->end(); I != E; ++I)
807 printFunction(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000808}
809
Chris Lattner7bfee412001-10-29 16:05:51 +0000810void AssemblyWriter::printGlobal(const GlobalVariable *GV) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000811 if (GV->hasName()) Out << getLLVMName(GV->getName()) << " = ";
Chris Lattner37798642001-09-18 04:01:05 +0000812
Misha Brukmanb1c93172005-04-21 23:48:37 +0000813 if (!GV->hasInitializer())
Misha Brukmana6619a92004-06-21 21:53:56 +0000814 Out << "external ";
Chris Lattner379a8d22003-04-16 20:28:45 +0000815 else
816 switch (GV->getLinkage()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000817 case GlobalValue::InternalLinkage: Out << "internal "; break;
818 case GlobalValue::LinkOnceLinkage: Out << "linkonce "; break;
819 case GlobalValue::WeakLinkage: Out << "weak "; break;
820 case GlobalValue::AppendingLinkage: Out << "appending "; break;
Chris Lattner379a8d22003-04-16 20:28:45 +0000821 case GlobalValue::ExternalLinkage: break;
Misha Brukman70457632004-11-14 21:04:34 +0000822 case GlobalValue::GhostLinkage:
823 std::cerr << "GhostLinkage not allowed in AsmWriter!\n";
824 abort();
Chris Lattner379a8d22003-04-16 20:28:45 +0000825 }
Chris Lattner37798642001-09-18 04:01:05 +0000826
Misha Brukmana6619a92004-06-21 21:53:56 +0000827 Out << (GV->isConstant() ? "constant " : "global ");
Chris Lattner2413b162001-12-04 00:03:30 +0000828 printType(GV->getType()->getElementType());
Chris Lattner37798642001-09-18 04:01:05 +0000829
Reid Spenceraccd7c72004-07-17 23:47:01 +0000830 if (GV->hasInitializer()) {
831 Constant* C = cast<Constant>(GV->getInitializer());
832 assert(C && "GlobalVar initializer isn't constant?");
Reid Spencerc9c90cf2004-07-18 01:04:19 +0000833 writeOperand(GV->getInitializer(), false, isa<GlobalValue>(C));
Reid Spenceraccd7c72004-07-17 23:47:01 +0000834 }
Chris Lattner37798642001-09-18 04:01:05 +0000835
Chris Lattner113f4f42002-06-25 16:13:24 +0000836 printInfoComment(*GV);
Misha Brukmana6619a92004-06-21 21:53:56 +0000837 Out << "\n";
Chris Lattnerda975502001-09-10 07:58:01 +0000838}
839
Chris Lattnerfee714f2001-09-07 16:36:04 +0000840
Reid Spencere7e96712004-05-25 08:53:40 +0000841// printSymbolTable - Run through symbol table looking for constants
842// and types. Emit their declarations.
Chris Lattner7bfee412001-10-29 16:05:51 +0000843void AssemblyWriter::printSymbolTable(const SymbolTable &ST) {
Reid Spencere7e96712004-05-25 08:53:40 +0000844
845 // Print the types.
846 for (SymbolTable::type_const_iterator TI = ST.type_begin();
847 TI != ST.type_end(); ++TI ) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000848 Out << "\t" << getLLVMName(TI->first) << " = type ";
Reid Spencere7e96712004-05-25 08:53:40 +0000849
850 // Make sure we print out at least one level of the type structure, so
851 // that we do not get %FILE = type %FILE
852 //
853 printTypeAtLeastOneLevel(TI->second) << "\n";
854 }
Misha Brukmanb1c93172005-04-21 23:48:37 +0000855
Reid Spencere7e96712004-05-25 08:53:40 +0000856 // Print the constants, in type plane order.
857 for (SymbolTable::plane_const_iterator PI = ST.plane_begin();
858 PI != ST.plane_end(); ++PI ) {
859 SymbolTable::value_const_iterator VI = ST.value_begin(PI->first);
860 SymbolTable::value_const_iterator VE = ST.value_end(PI->first);
861
862 for (; VI != VE; ++VI) {
Reid Spenceraccd7c72004-07-17 23:47:01 +0000863 const Value* V = VI->second;
864 const Constant *CPV = dyn_cast<Constant>(V) ;
865 if (CPV && !isa<GlobalValue>(V)) {
Reid Spencer56010e42004-05-26 21:56:09 +0000866 printConstant(CPV);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000867 }
868 }
Chris Lattnera7620d92001-07-15 06:35:59 +0000869 }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000870}
871
872
Misha Brukmanc566ca362004-03-02 00:22:19 +0000873/// printConstant - Print out a constant pool entry...
874///
Chris Lattner3462ae32001-12-03 22:26:30 +0000875void AssemblyWriter::printConstant(const Constant *CPV) {
Chris Lattnerfee714f2001-09-07 16:36:04 +0000876 // Don't print out unnamed constants, they will be inlined
877 if (!CPV->hasName()) return;
Chris Lattner2f7c9632001-06-06 20:29:01 +0000878
Chris Lattneree998be2001-07-26 16:29:38 +0000879 // Print out name...
Misha Brukmana6619a92004-06-21 21:53:56 +0000880 Out << "\t" << getLLVMName(CPV->getName()) << " =";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000881
882 // Write the value out now...
Chris Lattnerd84bb632002-04-16 21:36:08 +0000883 writeOperand(CPV, true, false);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000884
Chris Lattner113f4f42002-06-25 16:13:24 +0000885 printInfoComment(*CPV);
Misha Brukmana6619a92004-06-21 21:53:56 +0000886 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000887}
888
Misha Brukmanc566ca362004-03-02 00:22:19 +0000889/// printFunction - Print all aspects of a function.
890///
Chris Lattner113f4f42002-06-25 16:13:24 +0000891void AssemblyWriter::printFunction(const Function *F) {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000892 // Print out the return type and name...
Misha Brukmana6619a92004-06-21 21:53:56 +0000893 Out << "\n";
Chris Lattner379a8d22003-04-16 20:28:45 +0000894
Chris Lattner97e36f22004-12-05 06:44:09 +0000895 // Ensure that no local symbols conflict with global symbols.
896 const_cast<Function*>(F)->renameLocalSymbols();
897
Misha Brukmana6619a92004-06-21 21:53:56 +0000898 if (AnnotationWriter) AnnotationWriter->emitFunctionAnnot(F, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +0000899
Chris Lattner379a8d22003-04-16 20:28:45 +0000900 if (F->isExternal())
Misha Brukmana6619a92004-06-21 21:53:56 +0000901 Out << "declare ";
Chris Lattner379a8d22003-04-16 20:28:45 +0000902 else
903 switch (F->getLinkage()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000904 case GlobalValue::InternalLinkage: Out << "internal "; break;
905 case GlobalValue::LinkOnceLinkage: Out << "linkonce "; break;
906 case GlobalValue::WeakLinkage: Out << "weak "; break;
907 case GlobalValue::AppendingLinkage: Out << "appending "; break;
Chris Lattner379a8d22003-04-16 20:28:45 +0000908 case GlobalValue::ExternalLinkage: break;
Misha Brukman70457632004-11-14 21:04:34 +0000909 case GlobalValue::GhostLinkage:
910 std::cerr << "GhostLinkage not allowed in AsmWriter!\n";
911 abort();
Chris Lattner379a8d22003-04-16 20:28:45 +0000912 }
913
Chris Lattnerf7b6d312005-05-06 20:26:43 +0000914 // Print the calling convention.
915 switch (F->getCallingConv()) {
916 case CallingConv::C: break; // default
917 case CallingConv::Fast: Out << "fastcc "; break;
918 case CallingConv::Cold: Out << "coldcc "; break;
919 default: Out << "cc" << F->getCallingConv() << " "; break;
920 }
921
Misha Brukman21bbdb92004-06-04 21:11:51 +0000922 printType(F->getReturnType()) << ' ';
Chris Lattner5b337482003-10-18 05:57:43 +0000923 if (!F->getName().empty())
Misha Brukmana6619a92004-06-21 21:53:56 +0000924 Out << getLLVMName(F->getName());
Chris Lattner5b337482003-10-18 05:57:43 +0000925 else
Misha Brukmana6619a92004-06-21 21:53:56 +0000926 Out << "\"\"";
927 Out << '(';
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000928 Machine.incorporateFunction(F);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000929
Chris Lattner7bfee412001-10-29 16:05:51 +0000930 // Loop over the arguments, printing them...
Chris Lattner113f4f42002-06-25 16:13:24 +0000931 const FunctionType *FT = F->getFunctionType();
Chris Lattnerfee714f2001-09-07 16:36:04 +0000932
Chris Lattner531f9e92005-03-15 04:54:21 +0000933 for(Function::const_arg_iterator I = F->arg_begin(), E = F->arg_end(); I != E; ++I)
Chris Lattner149376d2002-10-13 20:57:00 +0000934 printArgument(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000935
936 // Finish printing arguments...
Chris Lattner113f4f42002-06-25 16:13:24 +0000937 if (FT->isVarArg()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000938 if (FT->getNumParams()) Out << ", ";
939 Out << "..."; // Output varargs portion of signature!
Chris Lattnerfee714f2001-09-07 16:36:04 +0000940 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000941 Out << ')';
Chris Lattnerfee714f2001-09-07 16:36:04 +0000942
Chris Lattner113f4f42002-06-25 16:13:24 +0000943 if (F->isExternal()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000944 Out << "\n";
Chris Lattnerb2f02e52002-05-06 03:00:40 +0000945 } else {
Misha Brukmana6619a92004-06-21 21:53:56 +0000946 Out << " {";
Misha Brukmanb1c93172005-04-21 23:48:37 +0000947
Chris Lattner6915f8f2002-04-07 22:49:37 +0000948 // Output all of its basic blocks... for the function
Chris Lattner113f4f42002-06-25 16:13:24 +0000949 for (Function::const_iterator I = F->begin(), E = F->end(); I != E; ++I)
950 printBasicBlock(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000951
Misha Brukmana6619a92004-06-21 21:53:56 +0000952 Out << "}\n";
Chris Lattnerfee714f2001-09-07 16:36:04 +0000953 }
954
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000955 Machine.purgeFunction();
Chris Lattner2f7c9632001-06-06 20:29:01 +0000956}
957
Misha Brukmanc566ca362004-03-02 00:22:19 +0000958/// printArgument - This member is called for every argument that is passed into
959/// the function. Simply print it out
960///
Chris Lattner2e9fa6d2002-04-09 19:48:49 +0000961void AssemblyWriter::printArgument(const Argument *Arg) {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000962 // Insert commas as we go... the first arg doesn't get a comma
Chris Lattnerfb972ad2005-03-15 05:03:36 +0000963 if (Arg != Arg->getParent()->arg_begin()) Out << ", ";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000964
965 // Output type...
Chris Lattner7bfee412001-10-29 16:05:51 +0000966 printType(Arg->getType());
Misha Brukmanb1c93172005-04-21 23:48:37 +0000967
Chris Lattner2f7c9632001-06-06 20:29:01 +0000968 // Output name, if available...
969 if (Arg->hasName())
Misha Brukmana6619a92004-06-21 21:53:56 +0000970 Out << ' ' << getLLVMName(Arg->getName());
Chris Lattner2f7c9632001-06-06 20:29:01 +0000971}
972
Misha Brukmanc566ca362004-03-02 00:22:19 +0000973/// printBasicBlock - This member is called for each basic block in a method.
974///
Chris Lattner7bfee412001-10-29 16:05:51 +0000975void AssemblyWriter::printBasicBlock(const BasicBlock *BB) {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000976 if (BB->hasName()) { // Print out the label if it exists...
Alkis Evlogimenos991d6ad2004-12-10 05:41:10 +0000977 Out << "\n" << getLLVMName(BB->getName(), false) << ':';
Chris Lattner408dbdb2002-05-14 16:02:05 +0000978 } else if (!BB->use_empty()) { // Don't print block # of no uses...
Misha Brukmana6619a92004-06-21 21:53:56 +0000979 Out << "\n; <label>:";
Chris Lattner757ee0b2004-06-09 19:41:19 +0000980 int Slot = Machine.getSlot(BB);
981 if (Slot != -1)
Misha Brukmana6619a92004-06-21 21:53:56 +0000982 Out << Slot;
Chris Lattner757ee0b2004-06-09 19:41:19 +0000983 else
Misha Brukmana6619a92004-06-21 21:53:56 +0000984 Out << "<badref>";
Chris Lattner58185f22002-10-02 19:38:55 +0000985 }
Chris Lattner2447ef52003-11-20 00:09:43 +0000986
987 if (BB->getParent() == 0)
Misha Brukmana6619a92004-06-21 21:53:56 +0000988 Out << "\t\t; Error: Block without parent!";
Chris Lattner2447ef52003-11-20 00:09:43 +0000989 else {
990 if (BB != &BB->getParent()->front()) { // Not the entry block?
991 // Output predecessors for the block...
Misha Brukmana6619a92004-06-21 21:53:56 +0000992 Out << "\t\t;";
Chris Lattner2447ef52003-11-20 00:09:43 +0000993 pred_const_iterator PI = pred_begin(BB), PE = pred_end(BB);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000994
Chris Lattner2447ef52003-11-20 00:09:43 +0000995 if (PI == PE) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000996 Out << " No predecessors!";
Chris Lattner2447ef52003-11-20 00:09:43 +0000997 } else {
Misha Brukmana6619a92004-06-21 21:53:56 +0000998 Out << " preds =";
Chris Lattner00211f12003-11-16 22:59:57 +0000999 writeOperand(*PI, false, true);
Chris Lattner2447ef52003-11-20 00:09:43 +00001000 for (++PI; PI != PE; ++PI) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001001 Out << ',';
Chris Lattner2447ef52003-11-20 00:09:43 +00001002 writeOperand(*PI, false, true);
1003 }
Chris Lattner00211f12003-11-16 22:59:57 +00001004 }
Chris Lattner58185f22002-10-02 19:38:55 +00001005 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001006 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001007
Misha Brukmana6619a92004-06-21 21:53:56 +00001008 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001009
Misha Brukmana6619a92004-06-21 21:53:56 +00001010 if (AnnotationWriter) AnnotationWriter->emitBasicBlockStartAnnot(BB, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001011
Chris Lattnerfee714f2001-09-07 16:36:04 +00001012 // Output all of the instructions in the basic block...
Chris Lattner113f4f42002-06-25 16:13:24 +00001013 for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I != E; ++I)
1014 printInstruction(*I);
Chris Lattner96cdd272004-03-08 18:51:45 +00001015
Misha Brukmana6619a92004-06-21 21:53:56 +00001016 if (AnnotationWriter) AnnotationWriter->emitBasicBlockEndAnnot(BB, Out);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001017}
1018
Chris Lattner862e3382001-10-13 06:42:36 +00001019
Misha Brukmanc566ca362004-03-02 00:22:19 +00001020/// printInfoComment - Print a little comment after the instruction indicating
1021/// which slot it occupies.
1022///
Chris Lattner113f4f42002-06-25 16:13:24 +00001023void AssemblyWriter::printInfoComment(const Value &V) {
1024 if (V.getType() != Type::VoidTy) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001025 Out << "\t\t; <";
Misha Brukman21bbdb92004-06-04 21:11:51 +00001026 printType(V.getType()) << '>';
Chris Lattner862e3382001-10-13 06:42:36 +00001027
Chris Lattner113f4f42002-06-25 16:13:24 +00001028 if (!V.hasName()) {
Chris Lattner757ee0b2004-06-09 19:41:19 +00001029 int SlotNum = Machine.getSlot(&V);
1030 if (SlotNum == -1)
Misha Brukmana6619a92004-06-21 21:53:56 +00001031 Out << ":<badref>";
Reid Spencer8beac692004-06-09 15:26:53 +00001032 else
Misha Brukmana6619a92004-06-21 21:53:56 +00001033 Out << ':' << SlotNum; // Print out the def slot taken.
Chris Lattner862e3382001-10-13 06:42:36 +00001034 }
Chris Lattnerb6c21db2005-02-01 01:24:01 +00001035 Out << " [#uses=" << V.getNumUses() << ']'; // Output # uses
Chris Lattner862e3382001-10-13 06:42:36 +00001036 }
1037}
1038
Reid Spencer8beac692004-06-09 15:26:53 +00001039/// printInstruction - This member is called for each Instruction in a function..
Misha Brukmanc566ca362004-03-02 00:22:19 +00001040///
Chris Lattner113f4f42002-06-25 16:13:24 +00001041void AssemblyWriter::printInstruction(const Instruction &I) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001042 if (AnnotationWriter) AnnotationWriter->emitInstructionAnnot(&I, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001043
Misha Brukmana6619a92004-06-21 21:53:56 +00001044 Out << "\t";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001045
1046 // Print out name if it exists...
Chris Lattner113f4f42002-06-25 16:13:24 +00001047 if (I.hasName())
Misha Brukmana6619a92004-06-21 21:53:56 +00001048 Out << getLLVMName(I.getName()) << " = ";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001049
Chris Lattner06038452005-05-06 05:51:46 +00001050 // If this is a volatile load or store, print out the volatile marker.
Chris Lattner504f9242003-09-08 17:45:59 +00001051 if ((isa<LoadInst>(I) && cast<LoadInst>(I).isVolatile()) ||
Chris Lattner06038452005-05-06 05:51:46 +00001052 (isa<StoreInst>(I) && cast<StoreInst>(I).isVolatile())) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001053 Out << "volatile ";
Chris Lattner06038452005-05-06 05:51:46 +00001054 } else if (isa<CallInst>(I) && cast<CallInst>(I).isTailCall()) {
1055 // If this is a call, check if it's a tail call.
1056 Out << "tail ";
1057 }
Chris Lattner504f9242003-09-08 17:45:59 +00001058
Chris Lattner2f7c9632001-06-06 20:29:01 +00001059 // Print out the opcode...
Misha Brukmana6619a92004-06-21 21:53:56 +00001060 Out << I.getOpcodeName();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001061
1062 // Print out the type of the operands...
Chris Lattner113f4f42002-06-25 16:13:24 +00001063 const Value *Operand = I.getNumOperands() ? I.getOperand(0) : 0;
Chris Lattner2f7c9632001-06-06 20:29:01 +00001064
1065 // Special case conditional branches to swizzle the condition out to the front
Chris Lattner113f4f42002-06-25 16:13:24 +00001066 if (isa<BranchInst>(I) && I.getNumOperands() > 1) {
1067 writeOperand(I.getOperand(2), true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001068 Out << ',';
Chris Lattner2f7c9632001-06-06 20:29:01 +00001069 writeOperand(Operand, true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001070 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001071 writeOperand(I.getOperand(1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001072
Chris Lattner8d48df22002-04-13 18:34:38 +00001073 } else if (isa<SwitchInst>(I)) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001074 // Special case switch statement to get formatting nice and correct...
Misha Brukmana6619a92004-06-21 21:53:56 +00001075 writeOperand(Operand , true); Out << ',';
1076 writeOperand(I.getOperand(1), true); Out << " [";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001077
Chris Lattner113f4f42002-06-25 16:13:24 +00001078 for (unsigned op = 2, Eop = I.getNumOperands(); op < Eop; op += 2) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001079 Out << "\n\t\t";
1080 writeOperand(I.getOperand(op ), true); Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001081 writeOperand(I.getOperand(op+1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001082 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001083 Out << "\n\t]";
Chris Lattnerda558102001-10-02 03:41:24 +00001084 } else if (isa<PHINode>(I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001085 Out << ' ';
Chris Lattner113f4f42002-06-25 16:13:24 +00001086 printType(I.getType());
Misha Brukmana6619a92004-06-21 21:53:56 +00001087 Out << ' ';
Chris Lattner2f7c9632001-06-06 20:29:01 +00001088
Chris Lattner113f4f42002-06-25 16:13:24 +00001089 for (unsigned op = 0, Eop = I.getNumOperands(); op < Eop; op += 2) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001090 if (op) Out << ", ";
Misha Brukmanb1c93172005-04-21 23:48:37 +00001091 Out << '[';
Misha Brukmana6619a92004-06-21 21:53:56 +00001092 writeOperand(I.getOperand(op ), false); Out << ',';
1093 writeOperand(I.getOperand(op+1), false); Out << " ]";
Chris Lattner931ef3b2001-06-11 15:04:20 +00001094 }
Chris Lattner862e3382001-10-13 06:42:36 +00001095 } else if (isa<ReturnInst>(I) && !Operand) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001096 Out << " void";
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001097 } else if (const CallInst *CI = dyn_cast<CallInst>(&I)) {
1098 // Print the calling convention being used.
1099 switch (CI->getCallingConv()) {
1100 case CallingConv::C: break; // default
1101 case CallingConv::Fast: Out << " fastcc"; break;
1102 case CallingConv::Cold: Out << " coldcc"; break;
1103 default: Out << " cc" << CI->getCallingConv(); break;
1104 }
1105
Chris Lattner463d6a52003-08-05 15:34:45 +00001106 const PointerType *PTy = cast<PointerType>(Operand->getType());
1107 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1108 const Type *RetTy = FTy->getReturnType();
Chris Lattner2f2d9472001-11-06 21:28:12 +00001109
Chris Lattner463d6a52003-08-05 15:34:45 +00001110 // If possible, print out the short form of the call instruction. We can
Chris Lattner6915f8f2002-04-07 22:49:37 +00001111 // only do this if the first argument is a pointer to a nonvararg function,
Chris Lattner463d6a52003-08-05 15:34:45 +00001112 // and if the return type is not a pointer to a function.
Chris Lattner2f2d9472001-11-06 21:28:12 +00001113 //
Chris Lattner463d6a52003-08-05 15:34:45 +00001114 if (!FTy->isVarArg() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +00001115 (!isa<PointerType>(RetTy) ||
Chris Lattnerd9a36a62002-07-25 20:58:51 +00001116 !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001117 Out << ' '; printType(RetTy);
Chris Lattner2f2d9472001-11-06 21:28:12 +00001118 writeOperand(Operand, false);
1119 } else {
1120 writeOperand(Operand, true);
1121 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001122 Out << '(';
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001123 if (CI->getNumOperands() > 1) writeOperand(CI->getOperand(1), true);
Chris Lattner113f4f42002-06-25 16:13:24 +00001124 for (unsigned op = 2, Eop = I.getNumOperands(); op < Eop; ++op) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001125 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001126 writeOperand(I.getOperand(op), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001127 }
1128
Misha Brukmana6619a92004-06-21 21:53:56 +00001129 Out << " )";
Chris Lattner113f4f42002-06-25 16:13:24 +00001130 } else if (const InvokeInst *II = dyn_cast<InvokeInst>(&I)) {
Chris Lattner463d6a52003-08-05 15:34:45 +00001131 const PointerType *PTy = cast<PointerType>(Operand->getType());
1132 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1133 const Type *RetTy = FTy->getReturnType();
1134
Chris Lattnerf7b6d312005-05-06 20:26:43 +00001135 // Print the calling convention being used.
1136 switch (II->getCallingConv()) {
1137 case CallingConv::C: break; // default
1138 case CallingConv::Fast: Out << " fastcc"; break;
1139 case CallingConv::Cold: Out << " coldcc"; break;
1140 default: Out << " cc" << II->getCallingConv(); break;
1141 }
1142
Chris Lattner463d6a52003-08-05 15:34:45 +00001143 // If possible, print out the short form of the invoke instruction. We can
1144 // only do this if the first argument is a pointer to a nonvararg function,
1145 // and if the return type is not a pointer to a function.
1146 //
1147 if (!FTy->isVarArg() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +00001148 (!isa<PointerType>(RetTy) ||
Chris Lattner463d6a52003-08-05 15:34:45 +00001149 !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001150 Out << ' '; printType(RetTy);
Chris Lattner463d6a52003-08-05 15:34:45 +00001151 writeOperand(Operand, false);
1152 } else {
1153 writeOperand(Operand, true);
1154 }
1155
Misha Brukmana6619a92004-06-21 21:53:56 +00001156 Out << '(';
Chris Lattner113f4f42002-06-25 16:13:24 +00001157 if (I.getNumOperands() > 3) writeOperand(I.getOperand(3), true);
1158 for (unsigned op = 4, Eop = I.getNumOperands(); op < Eop; ++op) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001159 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001160 writeOperand(I.getOperand(op), true);
Chris Lattner862e3382001-10-13 06:42:36 +00001161 }
1162
Misha Brukmana6619a92004-06-21 21:53:56 +00001163 Out << " )\n\t\t\tto";
Chris Lattner862e3382001-10-13 06:42:36 +00001164 writeOperand(II->getNormalDest(), true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001165 Out << " unwind";
Chris Lattnerfae8ab32004-02-08 21:44:31 +00001166 writeOperand(II->getUnwindDest(), true);
Chris Lattner862e3382001-10-13 06:42:36 +00001167
Chris Lattner113f4f42002-06-25 16:13:24 +00001168 } else if (const AllocationInst *AI = dyn_cast<AllocationInst>(&I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001169 Out << ' ';
Chris Lattner8d48df22002-04-13 18:34:38 +00001170 printType(AI->getType()->getElementType());
1171 if (AI->isArrayAllocation()) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001172 Out << ',';
Chris Lattner8d48df22002-04-13 18:34:38 +00001173 writeOperand(AI->getArraySize(), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001174 }
Nate Begeman848622f2005-11-05 09:21:28 +00001175 if (AI->getAlignment()) {
Chris Lattner7aeee3a2005-11-05 21:20:34 +00001176 Out << ", align " << AI->getAlignment();
Nate Begeman848622f2005-11-05 09:21:28 +00001177 }
Chris Lattner862e3382001-10-13 06:42:36 +00001178 } else if (isa<CastInst>(I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001179 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001180 Out << " to ";
Chris Lattner113f4f42002-06-25 16:13:24 +00001181 printType(I.getType());
Chris Lattner5b337482003-10-18 05:57:43 +00001182 } else if (isa<VAArgInst>(I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001183 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001184 Out << ", ";
Chris Lattnerf70da102003-05-08 02:44:12 +00001185 printType(I.getType());
Chris Lattner2f7c9632001-06-06 20:29:01 +00001186 } else if (Operand) { // Print the normal way...
1187
Misha Brukmanb1c93172005-04-21 23:48:37 +00001188 // PrintAllTypes - Instructions who have operands of all the same type
Chris Lattner2f7c9632001-06-06 20:29:01 +00001189 // omit the type from all but the first operand. If the instruction has
1190 // different type operands (for example br), then they are all printed.
1191 bool PrintAllTypes = false;
1192 const Type *TheType = Operand->getType();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001193
Chris Lattner52bd5cb92004-03-12 05:53:14 +00001194 // Shift Left & Right print both types even for Ubyte LHS, and select prints
1195 // types even if all operands are bools.
Chris Lattner159485f2005-02-09 17:45:03 +00001196 if (isa<ShiftInst>(I) || isa<SelectInst>(I) || isa<StoreInst>(I)) {
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001197 PrintAllTypes = true;
1198 } else {
1199 for (unsigned i = 1, E = I.getNumOperands(); i != E; ++i) {
1200 Operand = I.getOperand(i);
1201 if (Operand->getType() != TheType) {
1202 PrintAllTypes = true; // We have differing types! Print them all!
1203 break;
1204 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001205 }
1206 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001207
Chris Lattner7bfee412001-10-29 16:05:51 +00001208 if (!PrintAllTypes) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001209 Out << ' ';
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001210 printType(TheType);
Chris Lattner7bfee412001-10-29 16:05:51 +00001211 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001212
Chris Lattner113f4f42002-06-25 16:13:24 +00001213 for (unsigned i = 0, E = I.getNumOperands(); i != E; ++i) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001214 if (i) Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001215 writeOperand(I.getOperand(i), PrintAllTypes);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001216 }
1217 }
1218
Chris Lattner862e3382001-10-13 06:42:36 +00001219 printInfoComment(I);
Misha Brukmana6619a92004-06-21 21:53:56 +00001220 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001221}
1222
1223
1224//===----------------------------------------------------------------------===//
1225// External Interface declarations
1226//===----------------------------------------------------------------------===//
1227
Chris Lattner8339f7d2003-10-30 23:41:03 +00001228void Module::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001229 SlotMachine SlotTable(this);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001230 AssemblyWriter W(o, SlotTable, this, AAW);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001231 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001232}
1233
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001234void GlobalVariable::print(std::ostream &o) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001235 SlotMachine SlotTable(getParent());
Chris Lattner8339f7d2003-10-30 23:41:03 +00001236 AssemblyWriter W(o, SlotTable, getParent(), 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001237 W.write(this);
Chris Lattneradfe0d12001-09-10 20:08:19 +00001238}
1239
Chris Lattner8339f7d2003-10-30 23:41:03 +00001240void Function::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001241 SlotMachine SlotTable(getParent());
Chris Lattner8339f7d2003-10-30 23:41:03 +00001242 AssemblyWriter W(o, SlotTable, getParent(), AAW);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001243
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001244 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001245}
1246
Chris Lattner8339f7d2003-10-30 23:41:03 +00001247void BasicBlock::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001248 SlotMachine SlotTable(getParent());
Misha Brukmanb1c93172005-04-21 23:48:37 +00001249 AssemblyWriter W(o, SlotTable,
Chris Lattner8339f7d2003-10-30 23:41:03 +00001250 getParent() ? getParent()->getParent() : 0, AAW);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001251 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001252}
1253
Chris Lattner8339f7d2003-10-30 23:41:03 +00001254void Instruction::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001255 const Function *F = getParent() ? getParent()->getParent() : 0;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001256 SlotMachine SlotTable(F);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001257 AssemblyWriter W(o, SlotTable, F ? F->getParent() : 0, AAW);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001258
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001259 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001260}
Chris Lattner7db79582001-11-07 04:21:57 +00001261
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001262void Constant::print(std::ostream &o) const {
1263 if (this == 0) { o << "<null> constant value\n"; return; }
Chris Lattner7d734802002-09-10 15:53:49 +00001264
Misha Brukman21bbdb92004-06-04 21:11:51 +00001265 o << ' ' << getType()->getDescription() << ' ';
Chris Lattner1e194682002-04-18 18:53:13 +00001266
Chris Lattnerab7d1ab2003-05-08 02:08:14 +00001267 std::map<const Type *, std::string> TypeTable;
Chris Lattner1e194682002-04-18 18:53:13 +00001268 WriteConstantInt(o, this, false, TypeTable, 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001269}
1270
Misha Brukmanb1c93172005-04-21 23:48:37 +00001271void Type::print(std::ostream &o) const {
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001272 if (this == 0)
1273 o << "<null Type>";
1274 else
1275 o << getDescription();
1276}
1277
Chris Lattner2e9fa6d2002-04-09 19:48:49 +00001278void Argument::print(std::ostream &o) const {
Chris Lattnere52ed452004-07-13 23:14:34 +00001279 WriteAsOperand(o, this, true, true,
1280 getParent() ? getParent()->getParent() : 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001281}
1282
Reid Spencer52641832004-05-25 18:14:38 +00001283// Value::dump - allow easy printing of Values from the debugger.
1284// Located here because so much of the needed functionality is here.
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001285void Value::dump() const { print(std::cerr); }
Reid Spencer52641832004-05-25 18:14:38 +00001286
1287// Type::dump - allow easy printing of Values from the debugger.
1288// Located here because so much of the needed functionality is here.
Reid Spencere7e96712004-05-25 08:53:40 +00001289void Type::dump() const { print(std::cerr); }
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001290
1291//===----------------------------------------------------------------------===//
1292// CachedWriter Class Implementation
1293//===----------------------------------------------------------------------===//
1294
Chris Lattner7db79582001-11-07 04:21:57 +00001295void CachedWriter::setModule(const Module *M) {
1296 delete SC; delete AW;
1297 if (M) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001298 SC = new SlotMachine(M );
Misha Brukman21bbdb92004-06-04 21:11:51 +00001299 AW = new AssemblyWriter(Out, *SC, M, 0);
Chris Lattner7db79582001-11-07 04:21:57 +00001300 } else {
1301 SC = 0; AW = 0;
1302 }
1303}
1304
1305CachedWriter::~CachedWriter() {
1306 delete AW;
1307 delete SC;
1308}
1309
Chris Lattner60a7dd12004-07-15 02:51:31 +00001310CachedWriter &CachedWriter::operator<<(const Value &V) {
Chris Lattner7db79582001-11-07 04:21:57 +00001311 assert(AW && SC && "CachedWriter does not have a current module!");
Chris Lattner60a7dd12004-07-15 02:51:31 +00001312 if (const Instruction *I = dyn_cast<Instruction>(&V))
Chris Lattner80d2d532004-06-26 19:40:40 +00001313 AW->write(I);
Chris Lattner60a7dd12004-07-15 02:51:31 +00001314 else if (const BasicBlock *BB = dyn_cast<BasicBlock>(&V))
Chris Lattner80d2d532004-06-26 19:40:40 +00001315 AW->write(BB);
Chris Lattner60a7dd12004-07-15 02:51:31 +00001316 else if (const Function *F = dyn_cast<Function>(&V))
Chris Lattner80d2d532004-06-26 19:40:40 +00001317 AW->write(F);
Chris Lattner60a7dd12004-07-15 02:51:31 +00001318 else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(&V))
Chris Lattner80d2d532004-06-26 19:40:40 +00001319 AW->write(GV);
Misha Brukmanb1c93172005-04-21 23:48:37 +00001320 else
Chris Lattner60a7dd12004-07-15 02:51:31 +00001321 AW->writeOperand(&V, true, true);
Chris Lattner7db79582001-11-07 04:21:57 +00001322 return *this;
1323}
Misha Brukman4685e262004-04-28 15:31:21 +00001324
Chris Lattner60a7dd12004-07-15 02:51:31 +00001325CachedWriter& CachedWriter::operator<<(const Type &Ty) {
Misha Brukman4685e262004-04-28 15:31:21 +00001326 if (SymbolicTypes) {
1327 const Module *M = AW->getModule();
Chris Lattner60a7dd12004-07-15 02:51:31 +00001328 if (M) WriteTypeSymbolic(Out, &Ty, M);
Reid Spencer58d30f22004-07-04 11:50:43 +00001329 } else {
Chris Lattner60a7dd12004-07-15 02:51:31 +00001330 AW->write(&Ty);
Reid Spencer58d30f22004-07-04 11:50:43 +00001331 }
1332 return *this;
Misha Brukman4685e262004-04-28 15:31:21 +00001333}
Misha Brukmanda546ea2004-04-28 19:24:28 +00001334
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001335//===----------------------------------------------------------------------===//
1336//===-- SlotMachine Implementation
1337//===----------------------------------------------------------------------===//
1338
1339#if 0
1340#define SC_DEBUG(X) std::cerr << X
1341#else
1342#define SC_DEBUG(X)
1343#endif
1344
1345// Module level constructor. Causes the contents of the Module (sans functions)
1346// to be added to the slot table.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001347SlotMachine::SlotMachine(const Module *M)
Reid Spencer56010e42004-05-26 21:56:09 +00001348 : TheModule(M) ///< Saved for lazy initialization.
1349 , TheFunction(0)
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001350 , FunctionProcessed(false)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001351 , mMap()
Reid Spencer58d30f22004-07-04 11:50:43 +00001352 , mTypes()
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001353 , fMap()
Reid Spencer58d30f22004-07-04 11:50:43 +00001354 , fTypes()
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001355{
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001356}
1357
1358// Function level constructor. Causes the contents of the Module and the one
1359// function provided to be added to the slot table.
Misha Brukmanb1c93172005-04-21 23:48:37 +00001360SlotMachine::SlotMachine(const Function *F )
Reid Spencer56010e42004-05-26 21:56:09 +00001361 : TheModule( F ? F->getParent() : 0 ) ///< Saved for lazy initialization
1362 , TheFunction(F) ///< Saved for lazy initialization
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001363 , FunctionProcessed(false)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001364 , mMap()
Reid Spencer58d30f22004-07-04 11:50:43 +00001365 , mTypes()
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001366 , fMap()
Reid Spencer58d30f22004-07-04 11:50:43 +00001367 , fTypes()
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001368{
Reid Spencer56010e42004-05-26 21:56:09 +00001369}
1370
1371inline void SlotMachine::initialize(void) {
Misha Brukmanb1c93172005-04-21 23:48:37 +00001372 if ( TheModule) {
1373 processModule();
Reid Spencer56010e42004-05-26 21:56:09 +00001374 TheModule = 0; ///< Prevent re-processing next time we're called.
1375 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001376 if ( TheFunction && ! FunctionProcessed) {
1377 processFunction();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001378 }
1379}
1380
1381// Iterate through all the global variables, functions, and global
Misha Brukmanb1c93172005-04-21 23:48:37 +00001382// variable initializers and create slots for them.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001383void SlotMachine::processModule() {
1384 SC_DEBUG("begin processModule!\n");
1385
1386 // Add all of the global variables to the value table...
Chris Lattner531f9e92005-03-15 04:54:21 +00001387 for (Module::const_global_iterator I = TheModule->global_begin(), E = TheModule->global_end();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001388 I != E; ++I)
1389 createSlot(I);
1390
1391 // Add all the functions to the table
1392 for (Module::const_iterator I = TheModule->begin(), E = TheModule->end();
1393 I != E; ++I)
1394 createSlot(I);
1395
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001396 SC_DEBUG("end processModule!\n");
1397}
1398
1399
Reid Spencer56010e42004-05-26 21:56:09 +00001400// Process the arguments, basic blocks, and instructions of a function.
1401void SlotMachine::processFunction() {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001402 SC_DEBUG("begin processFunction!\n");
1403
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001404 // Add all the function arguments
Misha Brukmanb1c93172005-04-21 23:48:37 +00001405 for(Function::const_arg_iterator AI = TheFunction->arg_begin(),
Chris Lattner531f9e92005-03-15 04:54:21 +00001406 AE = TheFunction->arg_end(); AI != AE; ++AI)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001407 createSlot(AI);
1408
1409 SC_DEBUG("Inserting Instructions:\n");
1410
1411 // Add all of the basic blocks and instructions
Misha Brukmanb1c93172005-04-21 23:48:37 +00001412 for (Function::const_iterator BB = TheFunction->begin(),
Reid Spencer56010e42004-05-26 21:56:09 +00001413 E = TheFunction->end(); BB != E; ++BB) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001414 createSlot(BB);
1415 for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I!=E; ++I) {
1416 createSlot(I);
1417 }
1418 }
1419
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001420 FunctionProcessed = true;
1421
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001422 SC_DEBUG("end processFunction!\n");
1423}
1424
1425// Clean up after incorporating a function. This is the only way
Reid Spencer56010e42004-05-26 21:56:09 +00001426// to get out of the function incorporation state that affects the
1427// getSlot/createSlot lock. Function incorporation state is indicated
1428// by TheFunction != 0.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001429void SlotMachine::purgeFunction() {
1430 SC_DEBUG("begin purgeFunction!\n");
1431 fMap.clear(); // Simply discard the function level map
Reid Spencer58d30f22004-07-04 11:50:43 +00001432 fTypes.clear();
Reid Spencer56010e42004-05-26 21:56:09 +00001433 TheFunction = 0;
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001434 FunctionProcessed = false;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001435 SC_DEBUG("end purgeFunction!\n");
1436}
1437
1438/// Get the slot number for a value. This function will assert if you
1439/// ask for a Value that hasn't previously been inserted with createSlot.
1440/// Types are forbidden because Type does not inherit from Value (any more).
Chris Lattner757ee0b2004-06-09 19:41:19 +00001441int SlotMachine::getSlot(const Value *V) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001442 assert( V && "Can't get slot for null Value" );
Misha Brukmanb1c93172005-04-21 23:48:37 +00001443 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1444 "Can't insert a non-GlobalValue Constant into SlotMachine");
Reid Spencer56010e42004-05-26 21:56:09 +00001445
1446 // Check for uninitialized state and do lazy initialization
1447 this->initialize();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001448
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001449 // Get the type of the value
1450 const Type* VTy = V->getType();
1451
1452 // Find the type plane in the module map
1453 TypedPlanes::const_iterator MI = mMap.find(VTy);
1454
Reid Spencer56010e42004-05-26 21:56:09 +00001455 if ( TheFunction ) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001456 // Lookup the type in the function map too
1457 TypedPlanes::const_iterator FI = fMap.find(VTy);
1458 // If there is a corresponding type plane in the function map
1459 if ( FI != fMap.end() ) {
1460 // Lookup the Value in the function map
1461 ValueMap::const_iterator FVI = FI->second.map.find(V);
1462 // If the value doesn't exist in the function map
1463 if ( FVI == FI->second.map.end() ) {
Chris Lattner68a038e2004-06-09 22:22:10 +00001464 // Look up the value in the module map.
1465 if (MI == mMap.end()) return -1;
Reid Spencer56010e42004-05-26 21:56:09 +00001466 ValueMap::const_iterator MVI = MI->second.map.find(V);
1467 // If we didn't find it, it wasn't inserted
Chris Lattner757ee0b2004-06-09 19:41:19 +00001468 if (MVI == MI->second.map.end()) return -1;
Reid Spencer56010e42004-05-26 21:56:09 +00001469 assert( MVI != MI->second.map.end() && "Value not found");
1470 // We found it only at the module level
Misha Brukmanb1c93172005-04-21 23:48:37 +00001471 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001472
1473 // else the value exists in the function map
1474 } else {
Reid Spencer56010e42004-05-26 21:56:09 +00001475 // Return the slot number as the module's contribution to
1476 // the type plane plus the index in the function's contribution
1477 // to the type plane.
Chris Lattnerb1f04782004-06-15 21:07:32 +00001478 if (MI != mMap.end())
1479 return MI->second.next_slot + FVI->second;
1480 else
1481 return FVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001482 }
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001483 }
1484 }
1485
Reid Spencer8beac692004-06-09 15:26:53 +00001486 // N.B. Can get here only if either !TheFunction or the function doesn't
1487 // have a corresponding type plane for the Value
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001488
1489 // Make sure the type plane exists
Chris Lattner757ee0b2004-06-09 19:41:19 +00001490 if (MI == mMap.end()) return -1;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001491 // Lookup the value in the module's map
1492 ValueMap::const_iterator MVI = MI->second.map.find(V);
1493 // Make sure we found it.
Chris Lattner757ee0b2004-06-09 19:41:19 +00001494 if (MVI == MI->second.map.end()) return -1;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001495 // Return it.
1496 return MVI->second;
1497}
1498
Reid Spencer58d30f22004-07-04 11:50:43 +00001499/// Get the slot number for a value. This function will assert if you
1500/// ask for a Value that hasn't previously been inserted with createSlot.
1501/// Types are forbidden because Type does not inherit from Value (any more).
1502int SlotMachine::getSlot(const Type *Ty) {
1503 assert( Ty && "Can't get slot for null Type" );
1504
1505 // Check for uninitialized state and do lazy initialization
1506 this->initialize();
1507
1508 if ( TheFunction ) {
1509 // Lookup the Type in the function map
1510 TypeMap::const_iterator FTI = fTypes.map.find(Ty);
1511 // If the Type doesn't exist in the function map
1512 if ( FTI == fTypes.map.end() ) {
1513 TypeMap::const_iterator MTI = mTypes.map.find(Ty);
1514 // If we didn't find it, it wasn't inserted
Misha Brukmanb1c93172005-04-21 23:48:37 +00001515 if (MTI == mTypes.map.end())
Reid Spencer58d30f22004-07-04 11:50:43 +00001516 return -1;
1517 // We found it only at the module level
Misha Brukmanb1c93172005-04-21 23:48:37 +00001518 return MTI->second;
Reid Spencer58d30f22004-07-04 11:50:43 +00001519
1520 // else the value exists in the function map
1521 } else {
1522 // Return the slot number as the module's contribution to
1523 // the type plane plus the index in the function's contribution
1524 // to the type plane.
1525 return mTypes.next_slot + FTI->second;
1526 }
1527 }
1528
1529 // N.B. Can get here only if either !TheFunction
1530
1531 // Lookup the value in the module's map
1532 TypeMap::const_iterator MTI = mTypes.map.find(Ty);
1533 // Make sure we found it.
1534 if (MTI == mTypes.map.end()) return -1;
1535 // Return it.
1536 return MTI->second;
1537}
1538
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001539// Create a new slot, or return the existing slot if it is already
1540// inserted. Note that the logic here parallels getSlot but instead
1541// of asserting when the Value* isn't found, it inserts the value.
1542unsigned SlotMachine::createSlot(const Value *V) {
1543 assert( V && "Can't insert a null Value to SlotMachine");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001544 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1545 "Can't insert a non-GlobalValue Constant into SlotMachine");
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001546
1547 const Type* VTy = V->getType();
1548
1549 // Just ignore void typed things
1550 if (VTy == Type::VoidTy) return 0; // FIXME: Wrong return value!
1551
1552 // Look up the type plane for the Value's type from the module map
1553 TypedPlanes::const_iterator MI = mMap.find(VTy);
1554
Reid Spencer56010e42004-05-26 21:56:09 +00001555 if ( TheFunction ) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001556 // Get the type plane for the Value's type from the function map
1557 TypedPlanes::const_iterator FI = fMap.find(VTy);
1558 // If there is a corresponding type plane in the function map
1559 if ( FI != fMap.end() ) {
1560 // Lookup the Value in the function map
1561 ValueMap::const_iterator FVI = FI->second.map.find(V);
1562 // If the value doesn't exist in the function map
1563 if ( FVI == FI->second.map.end() ) {
Reid Spencer56010e42004-05-26 21:56:09 +00001564 // If there is no corresponding type plane in the module map
1565 if ( MI == mMap.end() )
1566 return insertValue(V);
1567 // Look up the value in the module map
1568 ValueMap::const_iterator MVI = MI->second.map.find(V);
1569 // If we didn't find it, it wasn't inserted
1570 if ( MVI == MI->second.map.end() )
1571 return insertValue(V);
1572 else
1573 // We found it only at the module level
1574 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001575
1576 // else the value exists in the function map
1577 } else {
Reid Spencer56010e42004-05-26 21:56:09 +00001578 if ( MI == mMap.end() )
1579 return FVI->second;
1580 else
1581 // Return the slot number as the module's contribution to
1582 // the type plane plus the index in the function's contribution
1583 // to the type plane.
1584 return MI->second.next_slot + FVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001585 }
1586
1587 // else there is not a corresponding type plane in the function map
1588 } else {
1589 // If the type plane doesn't exists at the module level
1590 if ( MI == mMap.end() ) {
Reid Spencer56010e42004-05-26 21:56:09 +00001591 return insertValue(V);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001592 // else type plane exists at the module level, examine it
1593 } else {
Reid Spencer56010e42004-05-26 21:56:09 +00001594 // Look up the value in the module's map
1595 ValueMap::const_iterator MVI = MI->second.map.find(V);
1596 // If we didn't find it there either
1597 if ( MVI == MI->second.map.end() )
1598 // Return the slot number as the module's contribution to
1599 // the type plane plus the index of the function map insertion.
1600 return MI->second.next_slot + insertValue(V);
1601 else
1602 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001603 }
1604 }
1605 }
1606
Reid Spencer56010e42004-05-26 21:56:09 +00001607 // N.B. Can only get here if !TheFunction
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001608
1609 // If the module map's type plane is not for the Value's type
1610 if ( MI != mMap.end() ) {
1611 // Lookup the value in the module's map
1612 ValueMap::const_iterator MVI = MI->second.map.find(V);
Misha Brukmanb1c93172005-04-21 23:48:37 +00001613 if ( MVI != MI->second.map.end() )
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001614 return MVI->second;
1615 }
1616
1617 return insertValue(V);
1618}
1619
Reid Spencer58d30f22004-07-04 11:50:43 +00001620// Create a new slot, or return the existing slot if it is already
1621// inserted. Note that the logic here parallels getSlot but instead
1622// of asserting when the Value* isn't found, it inserts the value.
1623unsigned SlotMachine::createSlot(const Type *Ty) {
1624 assert( Ty && "Can't insert a null Type to SlotMachine");
1625
1626 if ( TheFunction ) {
1627 // Lookup the Type in the function map
1628 TypeMap::const_iterator FTI = fTypes.map.find(Ty);
1629 // If the type doesn't exist in the function map
1630 if ( FTI == fTypes.map.end() ) {
1631 // Look up the type in the module map
1632 TypeMap::const_iterator MTI = mTypes.map.find(Ty);
1633 // If we didn't find it, it wasn't inserted
1634 if ( MTI == mTypes.map.end() )
1635 return insertValue(Ty);
1636 else
1637 // We found it only at the module level
1638 return MTI->second;
1639
1640 // else the value exists in the function map
1641 } else {
1642 // Return the slot number as the module's contribution to
1643 // the type plane plus the index in the function's contribution
1644 // to the type plane.
1645 return mTypes.next_slot + FTI->second;
1646 }
1647 }
1648
1649 // N.B. Can only get here if !TheFunction
1650
1651 // Lookup the type in the module's map
1652 TypeMap::const_iterator MTI = mTypes.map.find(Ty);
Misha Brukmanb1c93172005-04-21 23:48:37 +00001653 if ( MTI != mTypes.map.end() )
Reid Spencer58d30f22004-07-04 11:50:43 +00001654 return MTI->second;
1655
1656 return insertValue(Ty);
1657}
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001658
1659// Low level insert function. Minimal checking is done. This
1660// function is just for the convenience of createSlot (above).
1661unsigned SlotMachine::insertValue(const Value *V ) {
1662 assert(V && "Can't insert a null Value into SlotMachine!");
Misha Brukmanb1c93172005-04-21 23:48:37 +00001663 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1664 "Can't insert a non-GlobalValue Constant into SlotMachine");
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001665
Reid Spencer56010e42004-05-26 21:56:09 +00001666 // If this value does not contribute to a plane (is void)
Misha Brukmanb1c93172005-04-21 23:48:37 +00001667 // or if the value already has a name then ignore it.
Reid Spencer56010e42004-05-26 21:56:09 +00001668 if (V->getType() == Type::VoidTy || V->hasName() ) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001669 SC_DEBUG("ignored value " << *V << "\n");
1670 return 0; // FIXME: Wrong return value
1671 }
1672
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001673 const Type *VTy = V->getType();
1674 unsigned DestSlot = 0;
1675
Reid Spencer56010e42004-05-26 21:56:09 +00001676 if ( TheFunction ) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001677 TypedPlanes::iterator I = fMap.find( VTy );
Misha Brukmanb1c93172005-04-21 23:48:37 +00001678 if ( I == fMap.end() )
Reid Spencer58d30f22004-07-04 11:50:43 +00001679 I = fMap.insert(std::make_pair(VTy,ValuePlane())).first;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001680 DestSlot = I->second.map[V] = I->second.next_slot++;
1681 } else {
1682 TypedPlanes::iterator I = mMap.find( VTy );
1683 if ( I == mMap.end() )
Reid Spencer58d30f22004-07-04 11:50:43 +00001684 I = mMap.insert(std::make_pair(VTy,ValuePlane())).first;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001685 DestSlot = I->second.map[V] = I->second.next_slot++;
1686 }
1687
Misha Brukmanb1c93172005-04-21 23:48:37 +00001688 SC_DEBUG(" Inserting value [" << VTy << "] = " << V << " slot=" <<
Reid Spencer56010e42004-05-26 21:56:09 +00001689 DestSlot << " [");
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001690 // G = Global, C = Constant, T = Type, F = Function, o = other
Misha Brukmanb1c93172005-04-21 23:48:37 +00001691 SC_DEBUG((isa<GlobalVariable>(V) ? 'G' : (isa<Function>(V) ? 'F' :
Reid Spenceraccd7c72004-07-17 23:47:01 +00001692 (isa<Constant>(V) ? 'C' : 'o'))));
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001693 SC_DEBUG("]\n");
1694 return DestSlot;
1695}
1696
Reid Spencer58d30f22004-07-04 11:50:43 +00001697// Low level insert function. Minimal checking is done. This
1698// function is just for the convenience of createSlot (above).
1699unsigned SlotMachine::insertValue(const Type *Ty ) {
1700 assert(Ty && "Can't insert a null Type into SlotMachine!");
1701
1702 unsigned DestSlot = 0;
1703
1704 if ( TheFunction ) {
1705 DestSlot = fTypes.map[Ty] = fTypes.next_slot++;
1706 } else {
1707 DestSlot = fTypes.map[Ty] = fTypes.next_slot++;
1708 }
1709 SC_DEBUG(" Inserting type [" << DestSlot << "] = " << Ty << "\n");
1710 return DestSlot;
1711}
1712
Reid Spencere7e96712004-05-25 08:53:40 +00001713// vim: sw=2