blob: 4cfba7668f5a9a2d7966287bff1eec6c033ffded [file] [log] [blame]
Chris Lattnerf7e79482002-04-07 22:31:46 +00001//===-- AsmWriter.cpp - Printing LLVM as an assembly file -----------------===//
John Criswell482202a2003-10-20 19:43:21 +00002//
3// 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.
7//
8//===----------------------------------------------------------------------===//
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 Lattnerc70b3f62004-01-20 19:50:34 +000021#include "llvm/Constants.h"
Chris Lattner913d18f2002-04-29 18:46:50 +000022#include "llvm/DerivedTypes.h"
Vikram S. Adveb952b542002-07-14 23:14:45 +000023#include "llvm/Instruction.h"
Misha Brukman2d3fa9e2004-07-29 16:53:53 +000024#include "llvm/Instructions.h"
Chris Lattnerc70b3f62004-01-20 19:50:34 +000025#include "llvm/Module.h"
Chris Lattnerfee714f2001-09-07 16:36:04 +000026#include "llvm/SymbolTable.h"
Misha Brukman4685e262004-04-28 15:31:21 +000027#include "llvm/Assembly/Writer.h"
Chris Lattner58185f22002-10-02 19:38:55 +000028#include "llvm/Support/CFG.h"
Chris Lattner5de22042001-11-27 00:03:19 +000029#include "Support/StringExtras.h"
30#include "Support/STLExtras.h"
Chris Lattnerfee714f2001-09-07 16:36:04 +000031#include <algorithm>
Chris Lattner189d19f2003-11-21 20:23:48 +000032using namespace llvm;
Brian Gaeke960707c2003-11-11 22:41:34 +000033
Chris Lattner60a7dd12004-07-15 02:51:31 +000034namespace llvm {
Reid Spencer16f2f7f2004-05-26 07:18:52 +000035
36/// This class provides computation of slot numbers for LLVM Assembly writing.
37/// @brief LLVM Assembly Writing Slot Computation.
38class SlotMachine {
39
40/// @name Types
41/// @{
42public:
43
44 /// @brief A mapping of Values to slot numbers
45 typedef std::map<const Value*, unsigned> ValueMap;
Reid Spencer58d30f22004-07-04 11:50:43 +000046 typedef std::map<const Type*, unsigned> TypeMap;
Reid Spencer16f2f7f2004-05-26 07:18:52 +000047
48 /// @brief A plane with next slot number and ValueMap
Reid Spencer58d30f22004-07-04 11:50:43 +000049 struct ValuePlane {
Reid Spencer16f2f7f2004-05-26 07:18:52 +000050 unsigned next_slot; ///< The next slot number to use
51 ValueMap map; ///< The map of Value* -> unsigned
Reid Spencer58d30f22004-07-04 11:50:43 +000052 ValuePlane() { next_slot = 0; } ///< Make sure we start at 0
53 };
54
55 struct TypePlane {
56 unsigned next_slot;
57 TypeMap map;
58 TypePlane() { next_slot = 0; }
59 void clear() { map.clear(); next_slot = 0; }
Reid Spencer16f2f7f2004-05-26 07:18:52 +000060 };
61
62 /// @brief The map of planes by Type
Reid Spencer58d30f22004-07-04 11:50:43 +000063 typedef std::map<const Type*, ValuePlane> TypedPlanes;
Reid Spencer16f2f7f2004-05-26 07:18:52 +000064
65/// @}
66/// @name Constructors
67/// @{
68public:
69 /// @brief Construct from a module
70 SlotMachine(const Module *M );
71
72 /// @brief Construct from a function, starting out in incorp state.
73 SlotMachine(const Function *F );
74
75/// @}
76/// @name Accessors
77/// @{
78public:
79 /// Return the slot number of the specified value in it's type
80 /// plane. Its an error to ask for something not in the SlotMachine.
81 /// Its an error to ask for a Type*
Chris Lattner757ee0b2004-06-09 19:41:19 +000082 int getSlot(const Value *V);
Reid Spencer58d30f22004-07-04 11:50:43 +000083 int getSlot(const Type*Ty);
Reid Spencer8beac692004-06-09 15:26:53 +000084
85 /// Determine if a Value has a slot or not
86 bool hasSlot(const Value* V);
Reid Spencer58d30f22004-07-04 11:50:43 +000087 bool hasSlot(const Type* Ty);
Reid Spencer16f2f7f2004-05-26 07:18:52 +000088
89/// @}
90/// @name Mutators
91/// @{
92public:
93 /// If you'd like to deal with a function instead of just a module, use
94 /// this method to get its data into the SlotMachine.
Reid Spencerb0ac8c42004-08-16 07:46:33 +000095 void incorporateFunction(const Function *F) {
96 TheFunction = F;
97 FunctionProcessed = false;
98 }
Reid Spencer16f2f7f2004-05-26 07:18:52 +000099
100 /// After calling incorporateFunction, use this method to remove the
101 /// most recently incorporated function from the SlotMachine. This
102 /// will reset the state of the machine back to just the module contents.
103 void purgeFunction();
104
105/// @}
106/// @name Implementation Details
107/// @{
108private:
Reid Spencer56010e42004-05-26 21:56:09 +0000109 /// This function does the actual initialization.
110 inline void initialize();
111
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000112 /// Values can be crammed into here at will. If they haven't
113 /// been inserted already, they get inserted, otherwise they are ignored.
114 /// Either way, the slot number for the Value* is returned.
115 unsigned createSlot(const Value *V);
Reid Spencer58d30f22004-07-04 11:50:43 +0000116 unsigned createSlot(const Type* Ty);
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000117
118 /// Insert a value into the value table. Return the slot number
119 /// that it now occupies. BadThings(TM) will happen if you insert a
120 /// Value that's already been inserted.
121 unsigned insertValue( const Value *V );
Reid Spencer58d30f22004-07-04 11:50:43 +0000122 unsigned insertValue( const Type* Ty);
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000123
124 /// Add all of the module level global variables (and their initializers)
125 /// and function declarations, but not the contents of those functions.
126 void processModule();
127
Reid Spencer56010e42004-05-26 21:56:09 +0000128 /// Add all of the functions arguments, basic blocks, and instructions
129 void processFunction();
130
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000131 SlotMachine(const SlotMachine &); // DO NOT IMPLEMENT
132 void operator=(const SlotMachine &); // DO NOT IMPLEMENT
133
134/// @}
135/// @name Data
136/// @{
137public:
138
139 /// @brief The module for which we are holding slot numbers
Reid Spencer56010e42004-05-26 21:56:09 +0000140 const Module* TheModule;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000141
Reid Spencer56010e42004-05-26 21:56:09 +0000142 /// @brief The function for which we are holding slot numbers
143 const Function* TheFunction;
Reid Spencerb0ac8c42004-08-16 07:46:33 +0000144 bool FunctionProcessed;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000145
146 /// @brief The TypePlanes map for the module level data
147 TypedPlanes mMap;
Reid Spencer58d30f22004-07-04 11:50:43 +0000148 TypePlane mTypes;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000149
150 /// @brief The TypePlanes map for the function level data
151 TypedPlanes fMap;
Reid Spencer58d30f22004-07-04 11:50:43 +0000152 TypePlane fTypes;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000153
154/// @}
155
156};
157
Chris Lattner60a7dd12004-07-15 02:51:31 +0000158} // end namespace llvm
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000159
Chris Lattnerc8b70922002-07-26 21:12:46 +0000160static RegisterPass<PrintModulePass>
161X("printm", "Print module to stderr",PassInfo::Analysis|PassInfo::Optimization);
162static RegisterPass<PrintFunctionPass>
163Y("print","Print function to stderr",PassInfo::Analysis|PassInfo::Optimization);
Chris Lattner7f8845a2002-07-23 18:07:49 +0000164
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000165static void WriteAsOperandInternal(std::ostream &Out, const Value *V,
166 bool PrintName,
167 std::map<const Type *, std::string> &TypeTable,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000168 SlotMachine *Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000169
Reid Spencer58d30f22004-07-04 11:50:43 +0000170static void WriteAsOperandInternal(std::ostream &Out, const Type *T,
171 bool PrintName,
172 std::map<const Type *, std::string> &TypeTable,
173 SlotMachine *Machine);
174
Chris Lattnerb86620e2001-10-29 16:37:48 +0000175static const Module *getModuleFromVal(const Value *V) {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000176 if (const Argument *MA = dyn_cast<Argument>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000177 return MA->getParent() ? MA->getParent()->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000178 else if (const BasicBlock *BB = dyn_cast<BasicBlock>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000179 return BB->getParent() ? BB->getParent()->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000180 else if (const Instruction *I = dyn_cast<Instruction>(V)) {
Chris Lattner57698e22002-03-26 18:01:55 +0000181 const Function *M = I->getParent() ? I->getParent()->getParent() : 0;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000182 return M ? M->getParent() : 0;
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000183 } else if (const GlobalValue *GV = dyn_cast<GlobalValue>(V))
Chris Lattnerb86620e2001-10-29 16:37:48 +0000184 return GV->getParent();
Chris Lattnerb86620e2001-10-29 16:37:48 +0000185 return 0;
186}
187
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000188static SlotMachine *createSlotMachine(const Value *V) {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000189 if (const Argument *FA = dyn_cast<Argument>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000190 return new SlotMachine(FA->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000191 } else if (const Instruction *I = dyn_cast<Instruction>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000192 return new SlotMachine(I->getParent()->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000193 } else if (const BasicBlock *BB = dyn_cast<BasicBlock>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000194 return new SlotMachine(BB->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000195 } else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(V)){
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000196 return new SlotMachine(GV->getParent());
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000197 } else if (const Function *Func = dyn_cast<Function>(V)) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000198 return new SlotMachine(Func);
Chris Lattner7bfee412001-10-29 16:05:51 +0000199 }
200 return 0;
201}
Chris Lattner2f7c9632001-06-06 20:29:01 +0000202
Chris Lattner1c343042003-08-22 05:40:38 +0000203// getLLVMName - Turn the specified string into an 'LLVM name', which is either
204// prefixed with % (if the string only contains simple characters) or is
205// surrounded with ""'s (if it has special chars in it).
206static std::string getLLVMName(const std::string &Name) {
207 assert(!Name.empty() && "Cannot get empty name!");
208
209 // First character cannot start with a number...
210 if (Name[0] >= '0' && Name[0] <= '9')
211 return "\"" + Name + "\"";
212
213 // Scan to see if we have any characters that are not on the "white list"
214 for (unsigned i = 0, e = Name.size(); i != e; ++i) {
215 char C = Name[i];
216 assert(C != '"' && "Illegal character in LLVM value name!");
217 if ((C < 'a' || C > 'z') && (C < 'A' || C > 'Z') && (C < '0' || C > '9') &&
218 C != '-' && C != '.' && C != '_')
219 return "\"" + Name + "\"";
220 }
221
222 // If we get here, then the identifier is legal to use as a "VarID".
223 return "%"+Name;
224}
225
Chris Lattnerb86620e2001-10-29 16:37:48 +0000226
Misha Brukmanc566ca362004-03-02 00:22:19 +0000227/// fillTypeNameTable - If the module has a symbol table, take all global types
228/// and stuff their names into the TypeNames map.
229///
Chris Lattnerb86620e2001-10-29 16:37:48 +0000230static void fillTypeNameTable(const Module *M,
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000231 std::map<const Type *, std::string> &TypeNames) {
Chris Lattner98cf1f52002-11-20 18:36:02 +0000232 if (!M) return;
233 const SymbolTable &ST = M->getSymbolTable();
Reid Spencere7e96712004-05-25 08:53:40 +0000234 SymbolTable::type_const_iterator TI = ST.type_begin();
235 for (; TI != ST.type_end(); ++TI ) {
236 // As a heuristic, don't insert pointer to primitive types, because
237 // they are used too often to have a single useful name.
238 //
239 const Type *Ty = cast<Type>(TI->second);
240 if (!isa<PointerType>(Ty) ||
Reid Spencer56010e42004-05-26 21:56:09 +0000241 !cast<PointerType>(Ty)->getElementType()->isPrimitiveType() ||
242 isa<OpaqueType>(cast<PointerType>(Ty)->getElementType()))
Reid Spencere7e96712004-05-25 08:53:40 +0000243 TypeNames.insert(std::make_pair(Ty, getLLVMName(TI->first)));
Chris Lattnerb86620e2001-10-29 16:37:48 +0000244 }
245}
246
247
248
John Criswellcd116ba2004-06-01 14:54:08 +0000249static void calcTypeName(const Type *Ty,
250 std::vector<const Type *> &TypeStack,
251 std::map<const Type *, std::string> &TypeNames,
252 std::string & Result){
253 if (Ty->isPrimitiveType() && !isa<OpaqueType>(Ty)) {
254 Result += Ty->getDescription(); // Base case
255 return;
256 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000257
258 // Check to see if the type is named.
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000259 std::map<const Type *, std::string>::iterator I = TypeNames.find(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000260 if (I != TypeNames.end()) {
261 Result += I->second;
262 return;
263 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000264
John Criswellcd116ba2004-06-01 14:54:08 +0000265 if (isa<OpaqueType>(Ty)) {
266 Result += "opaque";
267 return;
268 }
Chris Lattnerf14ead92003-10-30 00:22:33 +0000269
Chris Lattnerb86620e2001-10-29 16:37:48 +0000270 // Check to see if the Type is already on the stack...
271 unsigned Slot = 0, CurSize = TypeStack.size();
272 while (Slot < CurSize && TypeStack[Slot] != Ty) ++Slot; // Scan for type
273
274 // This is another base case for the recursion. In this case, we know
275 // that we have looped back to a type that we have previously visited.
276 // Generate the appropriate upreference to handle this.
John Criswellcd116ba2004-06-01 14:54:08 +0000277 if (Slot < CurSize) {
278 Result += "\\" + utostr(CurSize-Slot); // Here's the upreference
279 return;
280 }
Chris Lattnerb86620e2001-10-29 16:37:48 +0000281
282 TypeStack.push_back(Ty); // Recursive case: Add us to the stack..
283
Chris Lattner6b727592004-06-17 18:19:28 +0000284 switch (Ty->getTypeID()) {
Chris Lattner91db5822002-03-29 03:44:36 +0000285 case Type::FunctionTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000286 const FunctionType *FTy = cast<FunctionType>(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000287 calcTypeName(FTy->getReturnType(), TypeStack, TypeNames, Result);
288 Result += " (";
Chris Lattnerfa829be2004-02-09 04:14:01 +0000289 for (FunctionType::param_iterator I = FTy->param_begin(),
290 E = FTy->param_end(); I != E; ++I) {
291 if (I != FTy->param_begin())
Chris Lattnerb86620e2001-10-29 16:37:48 +0000292 Result += ", ";
John Criswellcd116ba2004-06-01 14:54:08 +0000293 calcTypeName(*I, TypeStack, TypeNames, Result);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000294 }
Chris Lattnerd816b532002-04-13 20:53:41 +0000295 if (FTy->isVarArg()) {
Chris Lattnerfa829be2004-02-09 04:14:01 +0000296 if (FTy->getNumParams()) Result += ", ";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000297 Result += "...";
298 }
299 Result += ")";
300 break;
301 }
302 case Type::StructTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000303 const StructType *STy = cast<StructType>(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000304 Result += "{ ";
Chris Lattnerac6db752004-02-09 04:37:31 +0000305 for (StructType::element_iterator I = STy->element_begin(),
306 E = STy->element_end(); I != E; ++I) {
307 if (I != STy->element_begin())
Chris Lattnerb86620e2001-10-29 16:37:48 +0000308 Result += ", ";
John Criswellcd116ba2004-06-01 14:54:08 +0000309 calcTypeName(*I, TypeStack, TypeNames, Result);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000310 }
311 Result += " }";
312 break;
313 }
314 case Type::PointerTyID:
John Criswellcd116ba2004-06-01 14:54:08 +0000315 calcTypeName(cast<PointerType>(Ty)->getElementType(),
316 TypeStack, TypeNames, Result);
317 Result += "*";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000318 break;
319 case Type::ArrayTyID: {
Chris Lattnerf26a8ee2003-07-23 15:30:06 +0000320 const ArrayType *ATy = cast<ArrayType>(Ty);
John Criswellcd116ba2004-06-01 14:54:08 +0000321 Result += "[" + utostr(ATy->getNumElements()) + " x ";
322 calcTypeName(ATy->getElementType(), TypeStack, TypeNames, Result);
323 Result += "]";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000324 break;
325 }
Chris Lattner15285ab2003-05-14 17:50:47 +0000326 case Type::OpaqueTyID:
John Criswellcd116ba2004-06-01 14:54:08 +0000327 Result += "opaque";
Chris Lattner15285ab2003-05-14 17:50:47 +0000328 break;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000329 default:
John Criswellcd116ba2004-06-01 14:54:08 +0000330 Result += "<unrecognized-type>";
Chris Lattnerb86620e2001-10-29 16:37:48 +0000331 }
332
333 TypeStack.pop_back(); // Remove self from stack...
John Criswellcd116ba2004-06-01 14:54:08 +0000334 return;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000335}
336
337
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000338/// printTypeInt - The internal guts of printing out a type that has a
339/// potentially named portion.
340///
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000341static std::ostream &printTypeInt(std::ostream &Out, const Type *Ty,
342 std::map<const Type *, std::string> &TypeNames) {
Chris Lattnerb86620e2001-10-29 16:37:48 +0000343 // Primitive types always print out their description, regardless of whether
344 // they have been named or not.
345 //
Chris Lattner92d60532003-10-30 00:12:51 +0000346 if (Ty->isPrimitiveType() && !isa<OpaqueType>(Ty))
347 return Out << Ty->getDescription();
Chris Lattnerb86620e2001-10-29 16:37:48 +0000348
349 // Check to see if the type is named.
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000350 std::map<const Type *, std::string>::iterator I = TypeNames.find(Ty);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000351 if (I != TypeNames.end()) return Out << I->second;
352
353 // Otherwise we have a type that has not been named but is a derived type.
354 // Carefully recurse the type hierarchy to print out any contained symbolic
355 // names.
356 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000357 std::vector<const Type *> TypeStack;
John Criswellcd116ba2004-06-01 14:54:08 +0000358 std::string TypeName;
359 calcTypeName(Ty, TypeStack, TypeNames, TypeName);
Chris Lattner7f74a562002-01-20 22:54:45 +0000360 TypeNames.insert(std::make_pair(Ty, TypeName));//Cache type name for later use
John Criswellcd116ba2004-06-01 14:54:08 +0000361 return (Out << TypeName);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000362}
363
Chris Lattner34b95182001-10-31 04:33:19 +0000364
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000365/// WriteTypeSymbolic - This attempts to write the specified type as a symbolic
366/// type, iff there is an entry in the modules symbol table for the specified
367/// type or one of it's component types. This is slower than a simple x << Type
368///
Chris Lattner189d19f2003-11-21 20:23:48 +0000369std::ostream &llvm::WriteTypeSymbolic(std::ostream &Out, const Type *Ty,
370 const Module *M) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000371 Out << ' ';
Chris Lattnerb86620e2001-10-29 16:37:48 +0000372
373 // If they want us to print out a type, attempt to make it symbolic if there
374 // is a symbol table in the module...
Chris Lattner98cf1f52002-11-20 18:36:02 +0000375 if (M) {
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000376 std::map<const Type *, std::string> TypeNames;
Chris Lattnerb86620e2001-10-29 16:37:48 +0000377 fillTypeNameTable(M, TypeNames);
378
Chris Lattner72f866e2001-10-29 16:40:32 +0000379 return printTypeInt(Out, Ty, TypeNames);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000380 } else {
Chris Lattner72f866e2001-10-29 16:40:32 +0000381 return Out << Ty->getDescription();
Chris Lattnerb86620e2001-10-29 16:37:48 +0000382 }
383}
384
Reid Spenceraccd7c72004-07-17 23:47:01 +0000385/// @brief Internal constant writer.
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000386static void WriteConstantInt(std::ostream &Out, const Constant *CV,
387 bool PrintName,
388 std::map<const Type *, std::string> &TypeTable,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000389 SlotMachine *Machine) {
Chris Lattner1e194682002-04-18 18:53:13 +0000390 if (const ConstantBool *CB = dyn_cast<ConstantBool>(CV)) {
391 Out << (CB == ConstantBool::True ? "true" : "false");
392 } else if (const ConstantSInt *CI = dyn_cast<ConstantSInt>(CV)) {
393 Out << CI->getValue();
394 } else if (const ConstantUInt *CI = dyn_cast<ConstantUInt>(CV)) {
395 Out << CI->getValue();
396 } else if (const ConstantFP *CFP = dyn_cast<ConstantFP>(CV)) {
397 // We would like to output the FP constant value in exponential notation,
398 // but we cannot do this if doing so will lose precision. Check here to
399 // make sure that we only output it in exponential format if we can parse
400 // the value back and get the same value.
401 //
402 std::string StrVal = ftostr(CFP->getValue());
403
404 // Check to make sure that the stringized number is not some string like
405 // "Inf" or NaN, that atof will accept, but the lexer will not. Check that
406 // the string matches the "[-+]?[0-9]" regex.
407 //
408 if ((StrVal[0] >= '0' && StrVal[0] <= '9') ||
409 ((StrVal[0] == '-' || StrVal[0] == '+') &&
Brian Gaeke87b4f072003-06-17 23:55:35 +0000410 (StrVal[1] >= '0' && StrVal[1] <= '9')))
Chris Lattner1e194682002-04-18 18:53:13 +0000411 // Reparse stringized version!
412 if (atof(StrVal.c_str()) == CFP->getValue()) {
413 Out << StrVal; return;
414 }
415
416 // Otherwise we could not reparse it to exactly the same value, so we must
417 // output the string in hexadecimal format!
418 //
419 // Behave nicely in the face of C TBAA rules... see:
420 // http://www.nullstone.com/htmls/category/aliastyp.htm
421 //
422 double Val = CFP->getValue();
423 char *Ptr = (char*)&Val;
424 assert(sizeof(double) == sizeof(uint64_t) && sizeof(double) == 8 &&
425 "assuming that double is 64 bits!");
426 Out << "0x" << utohexstr(*(uint64_t*)Ptr);
427
Chris Lattner76b2ff42004-02-15 05:55:15 +0000428 } else if (isa<ConstantAggregateZero>(CV)) {
429 Out << "zeroinitializer";
Chris Lattner1e194682002-04-18 18:53:13 +0000430 } else if (const ConstantArray *CA = dyn_cast<ConstantArray>(CV)) {
431 // As a special case, print the array as a string if it is an array of
432 // ubytes or an array of sbytes with positive values.
433 //
434 const Type *ETy = CA->getType()->getElementType();
435 bool isString = (ETy == Type::SByteTy || ETy == Type::UByteTy);
436
437 if (ETy == Type::SByteTy)
438 for (unsigned i = 0; i < CA->getNumOperands(); ++i)
439 if (cast<ConstantSInt>(CA->getOperand(i))->getValue() < 0) {
440 isString = false;
441 break;
442 }
443
444 if (isString) {
445 Out << "c\"";
446 for (unsigned i = 0; i < CA->getNumOperands(); ++i) {
Chris Lattneraa6ff272004-06-04 23:53:20 +0000447 unsigned char C =
448 (unsigned char)cast<ConstantInt>(CA->getOperand(i))->getRawValue();
Chris Lattner1e194682002-04-18 18:53:13 +0000449
Chris Lattnere5fd3862002-07-31 23:56:44 +0000450 if (isprint(C) && C != '"' && C != '\\') {
Chris Lattner1e194682002-04-18 18:53:13 +0000451 Out << C;
452 } else {
453 Out << '\\'
454 << (char) ((C/16 < 10) ? ( C/16 +'0') : ( C/16 -10+'A'))
455 << (char)(((C&15) < 10) ? ((C&15)+'0') : ((C&15)-10+'A'));
456 }
457 }
458 Out << "\"";
459
460 } else { // Cannot output in string format...
Misha Brukman21bbdb92004-06-04 21:11:51 +0000461 Out << '[';
Chris Lattnerd84bb632002-04-16 21:36:08 +0000462 if (CA->getNumOperands()) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000463 Out << ' ';
Chris Lattner1e194682002-04-18 18:53:13 +0000464 printTypeInt(Out, ETy, TypeTable);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000465 WriteAsOperandInternal(Out, CA->getOperand(0),
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000466 PrintName, TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000467 for (unsigned i = 1, e = CA->getNumOperands(); i != e; ++i) {
468 Out << ", ";
Chris Lattner1e194682002-04-18 18:53:13 +0000469 printTypeInt(Out, ETy, TypeTable);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000470 WriteAsOperandInternal(Out, CA->getOperand(i), PrintName,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000471 TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000472 }
473 }
474 Out << " ]";
475 }
476 } else if (const ConstantStruct *CS = dyn_cast<ConstantStruct>(CV)) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000477 Out << '{';
Chris Lattnerd84bb632002-04-16 21:36:08 +0000478 if (CS->getNumOperands()) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000479 Out << ' ';
Chris Lattnerd84bb632002-04-16 21:36:08 +0000480 printTypeInt(Out, CS->getOperand(0)->getType(), TypeTable);
481
482 WriteAsOperandInternal(Out, CS->getOperand(0),
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000483 PrintName, TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000484
485 for (unsigned i = 1; i < CS->getNumOperands(); i++) {
486 Out << ", ";
487 printTypeInt(Out, CS->getOperand(i)->getType(), TypeTable);
488
489 WriteAsOperandInternal(Out, CS->getOperand(i),
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000490 PrintName, TypeTable, Machine);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000491 }
492 }
493
494 Out << " }";
495 } else if (isa<ConstantPointerNull>(CV)) {
496 Out << "null";
497
Chris Lattner3cd8c562002-07-30 18:54:25 +0000498 } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CV)) {
Chris Lattnerb8257222003-03-06 23:23:32 +0000499 Out << CE->getOpcodeName() << " (";
Vikram S. Adveb952b542002-07-14 23:14:45 +0000500
501 for (User::const_op_iterator OI=CE->op_begin(); OI != CE->op_end(); ++OI) {
502 printTypeInt(Out, (*OI)->getType(), TypeTable);
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000503 WriteAsOperandInternal(Out, *OI, PrintName, TypeTable, Machine);
Vikram S. Adveb952b542002-07-14 23:14:45 +0000504 if (OI+1 != CE->op_end())
Chris Lattner3cd8c562002-07-30 18:54:25 +0000505 Out << ", ";
Vikram S. Adveb952b542002-07-14 23:14:45 +0000506 }
507
Chris Lattnercfe8f532002-08-16 21:17:11 +0000508 if (CE->getOpcode() == Instruction::Cast) {
Chris Lattner83b396b2002-08-15 19:37:43 +0000509 Out << " to ";
510 printTypeInt(Out, CE->getType(), TypeTable);
511 }
Misha Brukman21bbdb92004-06-04 21:11:51 +0000512 Out << ')';
Chris Lattner83b396b2002-08-15 19:37:43 +0000513
Chris Lattnerd84bb632002-04-16 21:36:08 +0000514 } else {
Vikram S. Adveb952b542002-07-14 23:14:45 +0000515 Out << "<placeholder or erroneous Constant>";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000516 }
517}
518
519
Misha Brukmanc566ca362004-03-02 00:22:19 +0000520/// WriteAsOperand - Write the name of the specified value out to the specified
521/// ostream. This can be useful when you just want to print int %reg126, not
522/// the whole instruction that generated it.
523///
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000524static void WriteAsOperandInternal(std::ostream &Out, const Value *V,
525 bool PrintName,
526 std::map<const Type*, std::string> &TypeTable,
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000527 SlotMachine *Machine) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000528 Out << ' ';
Reid Spenceraccd7c72004-07-17 23:47:01 +0000529 if ((PrintName || isa<GlobalValue>(V)) && V->hasName())
Chris Lattner1c343042003-08-22 05:40:38 +0000530 Out << getLLVMName(V->getName());
Reid Spenceraccd7c72004-07-17 23:47:01 +0000531 else {
532 const Constant *CV = dyn_cast<Constant>(V);
533 if (CV && !isa<GlobalValue>(CV))
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000534 WriteConstantInt(Out, CV, PrintName, TypeTable, Machine);
Reid Spenceraccd7c72004-07-17 23:47:01 +0000535 else {
Chris Lattnerd84bb632002-04-16 21:36:08 +0000536 int Slot;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000537 if (Machine) {
Reid Spencer56010e42004-05-26 21:56:09 +0000538 Slot = Machine->getSlot(V);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000539 } else {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000540 Machine = createSlotMachine(V);
Chris Lattner757ee0b2004-06-09 19:41:19 +0000541 if (Machine == 0)
542 Slot = Machine->getSlot(V);
543 else
544 Slot = -1;
Reid Spencer56010e42004-05-26 21:56:09 +0000545 delete Machine;
Chris Lattnerd84bb632002-04-16 21:36:08 +0000546 }
Chris Lattner757ee0b2004-06-09 19:41:19 +0000547 if (Slot != -1)
548 Out << '%' << Slot;
549 else
550 Out << "<badref>";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000551 }
552 }
553}
554
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000555/// WriteAsOperand - Write the name of the specified value out to the specified
556/// ostream. This can be useful when you just want to print int %reg126, not
557/// the whole instruction that generated it.
558///
Chris Lattner189d19f2003-11-21 20:23:48 +0000559std::ostream &llvm::WriteAsOperand(std::ostream &Out, const Value *V,
Misha Brukmanb22d09c2004-03-01 19:48:13 +0000560 bool PrintType, bool PrintName,
561 const Module *Context) {
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000562 std::map<const Type *, std::string> TypeNames;
Chris Lattner5a9f63e2002-07-10 16:48:17 +0000563 if (Context == 0) Context = getModuleFromVal(V);
Chris Lattnerb86620e2001-10-29 16:37:48 +0000564
Chris Lattner98cf1f52002-11-20 18:36:02 +0000565 if (Context)
Chris Lattner5a9f63e2002-07-10 16:48:17 +0000566 fillTypeNameTable(Context, TypeNames);
Chris Lattnerd84bb632002-04-16 21:36:08 +0000567
568 if (PrintType)
569 printTypeInt(Out, V->getType(), TypeNames);
570
Chris Lattner5a9f63e2002-07-10 16:48:17 +0000571 WriteAsOperandInternal(Out, V, PrintName, TypeNames, 0);
Chris Lattner5e5abe32001-07-20 19:15:21 +0000572 return Out;
573}
574
Reid Spencer58d30f22004-07-04 11:50:43 +0000575/// WriteAsOperandInternal - Write the name of the specified value out to
576/// the specified ostream. This can be useful when you just want to print
577/// int %reg126, not the whole instruction that generated it.
578///
579static void WriteAsOperandInternal(std::ostream &Out, const Type *T,
580 bool PrintName,
581 std::map<const Type*, std::string> &TypeTable,
582 SlotMachine *Machine) {
583 Out << ' ';
584 int Slot;
585 if (Machine) {
586 Slot = Machine->getSlot(T);
587 if (Slot != -1)
588 Out << '%' << Slot;
589 else
590 Out << "<badref>";
591 } else {
592 Out << T->getDescription();
593 }
594}
595
596/// WriteAsOperand - Write the name of the specified value out to the specified
597/// ostream. This can be useful when you just want to print int %reg126, not
598/// the whole instruction that generated it.
599///
600std::ostream &llvm::WriteAsOperand(std::ostream &Out, const Type *Ty,
601 bool PrintType, bool PrintName,
602 const Module *Context) {
603 std::map<const Type *, std::string> TypeNames;
604 assert(Context != 0 && "Can't write types as operand without module context");
605
606 fillTypeNameTable(Context, TypeNames);
607
608 // if (PrintType)
609 // printTypeInt(Out, V->getType(), TypeNames);
610
611 printTypeInt(Out, Ty, TypeNames);
612
613 WriteAsOperandInternal(Out, Ty, PrintName, TypeNames, 0);
614 return Out;
615}
616
Chris Lattner189d19f2003-11-21 20:23:48 +0000617namespace llvm {
Chris Lattner2e9fee42001-07-12 23:35:26 +0000618
Chris Lattnerfee714f2001-09-07 16:36:04 +0000619class AssemblyWriter {
Misha Brukmana6619a92004-06-21 21:53:56 +0000620 std::ostream &Out;
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000621 SlotMachine &Machine;
Chris Lattner7bfee412001-10-29 16:05:51 +0000622 const Module *TheModule;
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000623 std::map<const Type *, std::string> TypeNames;
Chris Lattner8339f7d2003-10-30 23:41:03 +0000624 AssemblyAnnotationWriter *AnnotationWriter;
Chris Lattner2f7c9632001-06-06 20:29:01 +0000625public:
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000626 inline AssemblyWriter(std::ostream &o, SlotMachine &Mac, const Module *M,
Chris Lattner8339f7d2003-10-30 23:41:03 +0000627 AssemblyAnnotationWriter *AAW)
Misha Brukmana6619a92004-06-21 21:53:56 +0000628 : Out(o), Machine(Mac), TheModule(M), AnnotationWriter(AAW) {
Chris Lattner7bfee412001-10-29 16:05:51 +0000629
630 // If the module has a symbol table, take all global types and stuff their
631 // names into the TypeNames map.
632 //
Chris Lattnerb86620e2001-10-29 16:37:48 +0000633 fillTypeNameTable(M, TypeNames);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000634 }
635
Chris Lattner7bfee412001-10-29 16:05:51 +0000636 inline void write(const Module *M) { printModule(M); }
637 inline void write(const GlobalVariable *G) { printGlobal(G); }
Chris Lattner57698e22002-03-26 18:01:55 +0000638 inline void write(const Function *F) { printFunction(F); }
Chris Lattner7bfee412001-10-29 16:05:51 +0000639 inline void write(const BasicBlock *BB) { printBasicBlock(BB); }
Chris Lattner113f4f42002-06-25 16:13:24 +0000640 inline void write(const Instruction *I) { printInstruction(*I); }
Chris Lattner3462ae32001-12-03 22:26:30 +0000641 inline void write(const Constant *CPV) { printConstant(CPV); }
Chris Lattner7db79582001-11-07 04:21:57 +0000642 inline void write(const Type *Ty) { printType(Ty); }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000643
Chris Lattner1e194682002-04-18 18:53:13 +0000644 void writeOperand(const Value *Op, bool PrintType, bool PrintName = true);
645
Misha Brukman4685e262004-04-28 15:31:21 +0000646 const Module* getModule() { return TheModule; }
647
Chris Lattner2f7c9632001-06-06 20:29:01 +0000648private :
Chris Lattner7bfee412001-10-29 16:05:51 +0000649 void printModule(const Module *M);
650 void printSymbolTable(const SymbolTable &ST);
Chris Lattner3462ae32001-12-03 22:26:30 +0000651 void printConstant(const Constant *CPV);
Chris Lattner7bfee412001-10-29 16:05:51 +0000652 void printGlobal(const GlobalVariable *GV);
Chris Lattner57698e22002-03-26 18:01:55 +0000653 void printFunction(const Function *F);
Chris Lattner2e9fa6d2002-04-09 19:48:49 +0000654 void printArgument(const Argument *FA);
Chris Lattner7bfee412001-10-29 16:05:51 +0000655 void printBasicBlock(const BasicBlock *BB);
Chris Lattner113f4f42002-06-25 16:13:24 +0000656 void printInstruction(const Instruction &I);
Chris Lattnerd816b532002-04-13 20:53:41 +0000657
658 // printType - Go to extreme measures to attempt to print out a short,
659 // symbolic version of a type name.
660 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000661 std::ostream &printType(const Type *Ty) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000662 return printTypeInt(Out, Ty, TypeNames);
Chris Lattnerd816b532002-04-13 20:53:41 +0000663 }
664
665 // printTypeAtLeastOneLevel - Print out one level of the possibly complex type
666 // without considering any symbolic types that we may have equal to it.
667 //
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000668 std::ostream &printTypeAtLeastOneLevel(const Type *Ty);
Chris Lattner7bfee412001-10-29 16:05:51 +0000669
Chris Lattner862e3382001-10-13 06:42:36 +0000670 // printInfoComment - Print a little comment after the instruction indicating
671 // which slot it occupies.
Chris Lattner113f4f42002-06-25 16:13:24 +0000672 void printInfoComment(const Value &V);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000673};
Reid Spencerf43ac622004-05-27 22:04:46 +0000674} // end of llvm namespace
Chris Lattner2f7c9632001-06-06 20:29:01 +0000675
Misha Brukmanc566ca362004-03-02 00:22:19 +0000676/// printTypeAtLeastOneLevel - Print out one level of the possibly complex type
677/// without considering any symbolic types that we may have equal to it.
678///
Chris Lattnerab7d1ab2003-05-08 02:08:14 +0000679std::ostream &AssemblyWriter::printTypeAtLeastOneLevel(const Type *Ty) {
Chris Lattner113f4f42002-06-25 16:13:24 +0000680 if (const FunctionType *FTy = dyn_cast<FunctionType>(Ty)) {
Chris Lattnerd816b532002-04-13 20:53:41 +0000681 printType(FTy->getReturnType()) << " (";
Chris Lattnerfa829be2004-02-09 04:14:01 +0000682 for (FunctionType::param_iterator I = FTy->param_begin(),
683 E = FTy->param_end(); I != E; ++I) {
684 if (I != FTy->param_begin())
Misha Brukmana6619a92004-06-21 21:53:56 +0000685 Out << ", ";
Chris Lattnerd84bb632002-04-16 21:36:08 +0000686 printType(*I);
Chris Lattnerd816b532002-04-13 20:53:41 +0000687 }
688 if (FTy->isVarArg()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000689 if (FTy->getNumParams()) Out << ", ";
690 Out << "...";
Chris Lattnerd816b532002-04-13 20:53:41 +0000691 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000692 Out << ')';
Chris Lattner113f4f42002-06-25 16:13:24 +0000693 } else if (const StructType *STy = dyn_cast<StructType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000694 Out << "{ ";
Chris Lattnerac6db752004-02-09 04:37:31 +0000695 for (StructType::element_iterator I = STy->element_begin(),
696 E = STy->element_end(); I != E; ++I) {
697 if (I != STy->element_begin())
Misha Brukmana6619a92004-06-21 21:53:56 +0000698 Out << ", ";
Chris Lattnerd816b532002-04-13 20:53:41 +0000699 printType(*I);
700 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000701 Out << " }";
Chris Lattner113f4f42002-06-25 16:13:24 +0000702 } else if (const PointerType *PTy = dyn_cast<PointerType>(Ty)) {
Misha Brukman21bbdb92004-06-04 21:11:51 +0000703 printType(PTy->getElementType()) << '*';
Chris Lattner113f4f42002-06-25 16:13:24 +0000704 } else if (const ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000705 Out << '[' << ATy->getNumElements() << " x ";
Misha Brukman21bbdb92004-06-04 21:11:51 +0000706 printType(ATy->getElementType()) << ']';
Chris Lattner113f4f42002-06-25 16:13:24 +0000707 } else if (const OpaqueType *OTy = dyn_cast<OpaqueType>(Ty)) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000708 Out << "opaque";
Chris Lattnerd816b532002-04-13 20:53:41 +0000709 } else {
Vikram S. Adveb952b542002-07-14 23:14:45 +0000710 if (!Ty->isPrimitiveType())
Misha Brukmana6619a92004-06-21 21:53:56 +0000711 Out << "<unknown derived type>";
Chris Lattnerd816b532002-04-13 20:53:41 +0000712 printType(Ty);
713 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000714 return Out;
Chris Lattnerd816b532002-04-13 20:53:41 +0000715}
716
717
Chris Lattnerfee714f2001-09-07 16:36:04 +0000718void AssemblyWriter::writeOperand(const Value *Operand, bool PrintType,
Reid Spencer56010e42004-05-26 21:56:09 +0000719 bool PrintName) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000720 if (PrintType) { Out << ' '; printType(Operand->getType()); }
721 WriteAsOperandInternal(Out, Operand, PrintName, TypeNames, &Machine);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000722}
723
Chris Lattner2f7c9632001-06-06 20:29:01 +0000724
Chris Lattner7bfee412001-10-29 16:05:51 +0000725void AssemblyWriter::printModule(const Module *M) {
Chris Lattner8068e0c2003-08-24 13:48:48 +0000726 switch (M->getEndianness()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000727 case Module::LittleEndian: Out << "target endian = little\n"; break;
728 case Module::BigEndian: Out << "target endian = big\n"; break;
Chris Lattner8068e0c2003-08-24 13:48:48 +0000729 case Module::AnyEndianness: break;
730 }
731 switch (M->getPointerSize()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000732 case Module::Pointer32: Out << "target pointersize = 32\n"; break;
733 case Module::Pointer64: Out << "target pointersize = 64\n"; break;
Chris Lattner8068e0c2003-08-24 13:48:48 +0000734 case Module::AnyPointerSize: break;
735 }
Reid Spencer48f98c82004-07-25 21:44:54 +0000736 if (!M->getTargetTriple().empty())
Reid Spencerffec7df2004-07-25 21:29:43 +0000737 Out << "target triple = \"" << M->getTargetTriple() << "\"\n";
Reid Spencercc5ff642004-07-25 18:08:18 +0000738
739 // Loop over the dependent libraries and emit them
Reid Spencer48f98c82004-07-25 21:44:54 +0000740 Module::lib_iterator LI= M->lib_begin();
741 Module::lib_iterator LE= M->lib_end();
742 if (LI != LE) {
Reid Spencerffec7df2004-07-25 21:29:43 +0000743 Out << "deplibs = [\n";
Reid Spencer48f98c82004-07-25 21:44:54 +0000744 while ( LI != LE ) {
Reid Spencerffec7df2004-07-25 21:29:43 +0000745 Out << "\"" << *LI << "\"";
746 ++LI;
747 if ( LI != LE )
748 Out << ",\n";
749 }
750 Out << " ]\n";
Reid Spencercc5ff642004-07-25 18:08:18 +0000751 }
Chris Lattner8068e0c2003-08-24 13:48:48 +0000752
Chris Lattnerfee714f2001-09-07 16:36:04 +0000753 // Loop over the symbol table, emitting all named constants...
Chris Lattner98cf1f52002-11-20 18:36:02 +0000754 printSymbolTable(M->getSymbolTable());
Chris Lattnerda975502001-09-10 07:58:01 +0000755
Chris Lattner113f4f42002-06-25 16:13:24 +0000756 for (Module::const_giterator I = M->gbegin(), E = M->gend(); I != E; ++I)
757 printGlobal(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000758
Misha Brukmana6619a92004-06-21 21:53:56 +0000759 Out << "\nimplementation ; Functions:\n";
Vikram S. Adve13ba19a2001-09-18 12:48:16 +0000760
Chris Lattner6915f8f2002-04-07 22:49:37 +0000761 // Output all of the functions...
Chris Lattner113f4f42002-06-25 16:13:24 +0000762 for (Module::const_iterator I = M->begin(), E = M->end(); I != E; ++I)
763 printFunction(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000764}
765
Chris Lattner7bfee412001-10-29 16:05:51 +0000766void AssemblyWriter::printGlobal(const GlobalVariable *GV) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000767 if (GV->hasName()) Out << getLLVMName(GV->getName()) << " = ";
Chris Lattner37798642001-09-18 04:01:05 +0000768
Chris Lattner379a8d22003-04-16 20:28:45 +0000769 if (!GV->hasInitializer())
Misha Brukmana6619a92004-06-21 21:53:56 +0000770 Out << "external ";
Chris Lattner379a8d22003-04-16 20:28:45 +0000771 else
772 switch (GV->getLinkage()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000773 case GlobalValue::InternalLinkage: Out << "internal "; break;
774 case GlobalValue::LinkOnceLinkage: Out << "linkonce "; break;
775 case GlobalValue::WeakLinkage: Out << "weak "; break;
776 case GlobalValue::AppendingLinkage: Out << "appending "; break;
Chris Lattner379a8d22003-04-16 20:28:45 +0000777 case GlobalValue::ExternalLinkage: break;
778 }
Chris Lattner37798642001-09-18 04:01:05 +0000779
Misha Brukmana6619a92004-06-21 21:53:56 +0000780 Out << (GV->isConstant() ? "constant " : "global ");
Chris Lattner2413b162001-12-04 00:03:30 +0000781 printType(GV->getType()->getElementType());
Chris Lattner37798642001-09-18 04:01:05 +0000782
Reid Spenceraccd7c72004-07-17 23:47:01 +0000783 if (GV->hasInitializer()) {
784 Constant* C = cast<Constant>(GV->getInitializer());
785 assert(C && "GlobalVar initializer isn't constant?");
Reid Spencerc9c90cf2004-07-18 01:04:19 +0000786 writeOperand(GV->getInitializer(), false, isa<GlobalValue>(C));
Reid Spenceraccd7c72004-07-17 23:47:01 +0000787 }
Chris Lattner37798642001-09-18 04:01:05 +0000788
Chris Lattner113f4f42002-06-25 16:13:24 +0000789 printInfoComment(*GV);
Misha Brukmana6619a92004-06-21 21:53:56 +0000790 Out << "\n";
Chris Lattnerda975502001-09-10 07:58:01 +0000791}
792
Chris Lattnerfee714f2001-09-07 16:36:04 +0000793
Reid Spencere7e96712004-05-25 08:53:40 +0000794// printSymbolTable - Run through symbol table looking for constants
795// and types. Emit their declarations.
Chris Lattner7bfee412001-10-29 16:05:51 +0000796void AssemblyWriter::printSymbolTable(const SymbolTable &ST) {
Reid Spencere7e96712004-05-25 08:53:40 +0000797
798 // Print the types.
799 for (SymbolTable::type_const_iterator TI = ST.type_begin();
800 TI != ST.type_end(); ++TI ) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000801 Out << "\t" << getLLVMName(TI->first) << " = type ";
Reid Spencere7e96712004-05-25 08:53:40 +0000802
803 // Make sure we print out at least one level of the type structure, so
804 // that we do not get %FILE = type %FILE
805 //
806 printTypeAtLeastOneLevel(TI->second) << "\n";
807 }
Chris Lattnerfee714f2001-09-07 16:36:04 +0000808
Reid Spencere7e96712004-05-25 08:53:40 +0000809 // Print the constants, in type plane order.
810 for (SymbolTable::plane_const_iterator PI = ST.plane_begin();
811 PI != ST.plane_end(); ++PI ) {
812 SymbolTable::value_const_iterator VI = ST.value_begin(PI->first);
813 SymbolTable::value_const_iterator VE = ST.value_end(PI->first);
814
815 for (; VI != VE; ++VI) {
Reid Spenceraccd7c72004-07-17 23:47:01 +0000816 const Value* V = VI->second;
817 const Constant *CPV = dyn_cast<Constant>(V) ;
818 if (CPV && !isa<GlobalValue>(V)) {
Reid Spencer56010e42004-05-26 21:56:09 +0000819 printConstant(CPV);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000820 }
821 }
Chris Lattnera7620d92001-07-15 06:35:59 +0000822 }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000823}
824
825
Misha Brukmanc566ca362004-03-02 00:22:19 +0000826/// printConstant - Print out a constant pool entry...
827///
Chris Lattner3462ae32001-12-03 22:26:30 +0000828void AssemblyWriter::printConstant(const Constant *CPV) {
Chris Lattnerfee714f2001-09-07 16:36:04 +0000829 // Don't print out unnamed constants, they will be inlined
830 if (!CPV->hasName()) return;
Chris Lattner2f7c9632001-06-06 20:29:01 +0000831
Chris Lattneree998be2001-07-26 16:29:38 +0000832 // Print out name...
Misha Brukmana6619a92004-06-21 21:53:56 +0000833 Out << "\t" << getLLVMName(CPV->getName()) << " =";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000834
835 // Write the value out now...
Chris Lattnerd84bb632002-04-16 21:36:08 +0000836 writeOperand(CPV, true, false);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000837
Chris Lattner113f4f42002-06-25 16:13:24 +0000838 printInfoComment(*CPV);
Misha Brukmana6619a92004-06-21 21:53:56 +0000839 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000840}
841
Misha Brukmanc566ca362004-03-02 00:22:19 +0000842/// printFunction - Print all aspects of a function.
843///
Chris Lattner113f4f42002-06-25 16:13:24 +0000844void AssemblyWriter::printFunction(const Function *F) {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000845 // Print out the return type and name...
Misha Brukmana6619a92004-06-21 21:53:56 +0000846 Out << "\n";
Chris Lattner379a8d22003-04-16 20:28:45 +0000847
Misha Brukmana6619a92004-06-21 21:53:56 +0000848 if (AnnotationWriter) AnnotationWriter->emitFunctionAnnot(F, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +0000849
Chris Lattner379a8d22003-04-16 20:28:45 +0000850 if (F->isExternal())
Misha Brukmana6619a92004-06-21 21:53:56 +0000851 Out << "declare ";
Chris Lattner379a8d22003-04-16 20:28:45 +0000852 else
853 switch (F->getLinkage()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000854 case GlobalValue::InternalLinkage: Out << "internal "; break;
855 case GlobalValue::LinkOnceLinkage: Out << "linkonce "; break;
856 case GlobalValue::WeakLinkage: Out << "weak "; break;
857 case GlobalValue::AppendingLinkage: Out << "appending "; break;
Chris Lattner379a8d22003-04-16 20:28:45 +0000858 case GlobalValue::ExternalLinkage: break;
859 }
860
Misha Brukman21bbdb92004-06-04 21:11:51 +0000861 printType(F->getReturnType()) << ' ';
Chris Lattner5b337482003-10-18 05:57:43 +0000862 if (!F->getName().empty())
Misha Brukmana6619a92004-06-21 21:53:56 +0000863 Out << getLLVMName(F->getName());
Chris Lattner5b337482003-10-18 05:57:43 +0000864 else
Misha Brukmana6619a92004-06-21 21:53:56 +0000865 Out << "\"\"";
866 Out << '(';
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000867 Machine.incorporateFunction(F);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000868
Chris Lattner7bfee412001-10-29 16:05:51 +0000869 // Loop over the arguments, printing them...
Chris Lattner113f4f42002-06-25 16:13:24 +0000870 const FunctionType *FT = F->getFunctionType();
Chris Lattnerfee714f2001-09-07 16:36:04 +0000871
Chris Lattner149376d2002-10-13 20:57:00 +0000872 for(Function::const_aiterator I = F->abegin(), E = F->aend(); I != E; ++I)
873 printArgument(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000874
875 // Finish printing arguments...
Chris Lattner113f4f42002-06-25 16:13:24 +0000876 if (FT->isVarArg()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000877 if (FT->getNumParams()) Out << ", ";
878 Out << "..."; // Output varargs portion of signature!
Chris Lattnerfee714f2001-09-07 16:36:04 +0000879 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000880 Out << ')';
Chris Lattnerfee714f2001-09-07 16:36:04 +0000881
Chris Lattner113f4f42002-06-25 16:13:24 +0000882 if (F->isExternal()) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000883 Out << "\n";
Chris Lattnerb2f02e52002-05-06 03:00:40 +0000884 } else {
Misha Brukmana6619a92004-06-21 21:53:56 +0000885 Out << " {";
Chris Lattnerfee714f2001-09-07 16:36:04 +0000886
Chris Lattner6915f8f2002-04-07 22:49:37 +0000887 // Output all of its basic blocks... for the function
Chris Lattner113f4f42002-06-25 16:13:24 +0000888 for (Function::const_iterator I = F->begin(), E = F->end(); I != E; ++I)
889 printBasicBlock(I);
Chris Lattnerfee714f2001-09-07 16:36:04 +0000890
Misha Brukmana6619a92004-06-21 21:53:56 +0000891 Out << "}\n";
Chris Lattnerfee714f2001-09-07 16:36:04 +0000892 }
893
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000894 Machine.purgeFunction();
Chris Lattner2f7c9632001-06-06 20:29:01 +0000895}
896
Misha Brukmanc566ca362004-03-02 00:22:19 +0000897/// printArgument - This member is called for every argument that is passed into
898/// the function. Simply print it out
899///
Chris Lattner2e9fa6d2002-04-09 19:48:49 +0000900void AssemblyWriter::printArgument(const Argument *Arg) {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000901 // Insert commas as we go... the first arg doesn't get a comma
Misha Brukmana6619a92004-06-21 21:53:56 +0000902 if (Arg != &Arg->getParent()->afront()) Out << ", ";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000903
904 // Output type...
Chris Lattner7bfee412001-10-29 16:05:51 +0000905 printType(Arg->getType());
Chris Lattner2f7c9632001-06-06 20:29:01 +0000906
907 // Output name, if available...
908 if (Arg->hasName())
Misha Brukmana6619a92004-06-21 21:53:56 +0000909 Out << ' ' << getLLVMName(Arg->getName());
Chris Lattner2f7c9632001-06-06 20:29:01 +0000910}
911
Misha Brukmanc566ca362004-03-02 00:22:19 +0000912/// printBasicBlock - This member is called for each basic block in a method.
913///
Chris Lattner7bfee412001-10-29 16:05:51 +0000914void AssemblyWriter::printBasicBlock(const BasicBlock *BB) {
Chris Lattner2f7c9632001-06-06 20:29:01 +0000915 if (BB->hasName()) { // Print out the label if it exists...
Misha Brukmana6619a92004-06-21 21:53:56 +0000916 Out << "\n" << BB->getName() << ':';
Chris Lattner408dbdb2002-05-14 16:02:05 +0000917 } else if (!BB->use_empty()) { // Don't print block # of no uses...
Misha Brukmana6619a92004-06-21 21:53:56 +0000918 Out << "\n; <label>:";
Chris Lattner757ee0b2004-06-09 19:41:19 +0000919 int Slot = Machine.getSlot(BB);
920 if (Slot != -1)
Misha Brukmana6619a92004-06-21 21:53:56 +0000921 Out << Slot;
Chris Lattner757ee0b2004-06-09 19:41:19 +0000922 else
Misha Brukmana6619a92004-06-21 21:53:56 +0000923 Out << "<badref>";
Chris Lattner58185f22002-10-02 19:38:55 +0000924 }
Chris Lattner2447ef52003-11-20 00:09:43 +0000925
926 if (BB->getParent() == 0)
Misha Brukmana6619a92004-06-21 21:53:56 +0000927 Out << "\t\t; Error: Block without parent!";
Chris Lattner2447ef52003-11-20 00:09:43 +0000928 else {
929 if (BB != &BB->getParent()->front()) { // Not the entry block?
930 // Output predecessors for the block...
Misha Brukmana6619a92004-06-21 21:53:56 +0000931 Out << "\t\t;";
Chris Lattner2447ef52003-11-20 00:09:43 +0000932 pred_const_iterator PI = pred_begin(BB), PE = pred_end(BB);
933
934 if (PI == PE) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000935 Out << " No predecessors!";
Chris Lattner2447ef52003-11-20 00:09:43 +0000936 } else {
Misha Brukmana6619a92004-06-21 21:53:56 +0000937 Out << " preds =";
Chris Lattner00211f12003-11-16 22:59:57 +0000938 writeOperand(*PI, false, true);
Chris Lattner2447ef52003-11-20 00:09:43 +0000939 for (++PI; PI != PE; ++PI) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000940 Out << ',';
Chris Lattner2447ef52003-11-20 00:09:43 +0000941 writeOperand(*PI, false, true);
942 }
Chris Lattner00211f12003-11-16 22:59:57 +0000943 }
Chris Lattner58185f22002-10-02 19:38:55 +0000944 }
Chris Lattner2f7c9632001-06-06 20:29:01 +0000945 }
Chris Lattner408dbdb2002-05-14 16:02:05 +0000946
Misha Brukmana6619a92004-06-21 21:53:56 +0000947 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000948
Misha Brukmana6619a92004-06-21 21:53:56 +0000949 if (AnnotationWriter) AnnotationWriter->emitBasicBlockStartAnnot(BB, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +0000950
Chris Lattnerfee714f2001-09-07 16:36:04 +0000951 // Output all of the instructions in the basic block...
Chris Lattner113f4f42002-06-25 16:13:24 +0000952 for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I != E; ++I)
953 printInstruction(*I);
Chris Lattner96cdd272004-03-08 18:51:45 +0000954
Misha Brukmana6619a92004-06-21 21:53:56 +0000955 if (AnnotationWriter) AnnotationWriter->emitBasicBlockEndAnnot(BB, Out);
Chris Lattner2f7c9632001-06-06 20:29:01 +0000956}
957
Chris Lattner862e3382001-10-13 06:42:36 +0000958
Misha Brukmanc566ca362004-03-02 00:22:19 +0000959/// printInfoComment - Print a little comment after the instruction indicating
960/// which slot it occupies.
961///
Chris Lattner113f4f42002-06-25 16:13:24 +0000962void AssemblyWriter::printInfoComment(const Value &V) {
963 if (V.getType() != Type::VoidTy) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000964 Out << "\t\t; <";
Misha Brukman21bbdb92004-06-04 21:11:51 +0000965 printType(V.getType()) << '>';
Chris Lattner862e3382001-10-13 06:42:36 +0000966
Chris Lattner113f4f42002-06-25 16:13:24 +0000967 if (!V.hasName()) {
Chris Lattner757ee0b2004-06-09 19:41:19 +0000968 int SlotNum = Machine.getSlot(&V);
969 if (SlotNum == -1)
Misha Brukmana6619a92004-06-21 21:53:56 +0000970 Out << ":<badref>";
Reid Spencer8beac692004-06-09 15:26:53 +0000971 else
Misha Brukmana6619a92004-06-21 21:53:56 +0000972 Out << ':' << SlotNum; // Print out the def slot taken.
Chris Lattner862e3382001-10-13 06:42:36 +0000973 }
Misha Brukmana6619a92004-06-21 21:53:56 +0000974 Out << " [#uses=" << V.use_size() << ']'; // Output # uses
Chris Lattner862e3382001-10-13 06:42:36 +0000975 }
976}
977
Reid Spencer8beac692004-06-09 15:26:53 +0000978/// printInstruction - This member is called for each Instruction in a function..
Misha Brukmanc566ca362004-03-02 00:22:19 +0000979///
Chris Lattner113f4f42002-06-25 16:13:24 +0000980void AssemblyWriter::printInstruction(const Instruction &I) {
Misha Brukmana6619a92004-06-21 21:53:56 +0000981 if (AnnotationWriter) AnnotationWriter->emitInstructionAnnot(&I, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +0000982
Misha Brukmana6619a92004-06-21 21:53:56 +0000983 Out << "\t";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000984
985 // Print out name if it exists...
Chris Lattner113f4f42002-06-25 16:13:24 +0000986 if (I.hasName())
Misha Brukmana6619a92004-06-21 21:53:56 +0000987 Out << getLLVMName(I.getName()) << " = ";
Chris Lattner2f7c9632001-06-06 20:29:01 +0000988
Chris Lattner504f9242003-09-08 17:45:59 +0000989 // If this is a volatile load or store, print out the volatile marker
990 if ((isa<LoadInst>(I) && cast<LoadInst>(I).isVolatile()) ||
991 (isa<StoreInst>(I) && cast<StoreInst>(I).isVolatile()))
Misha Brukmana6619a92004-06-21 21:53:56 +0000992 Out << "volatile ";
Chris Lattner504f9242003-09-08 17:45:59 +0000993
Chris Lattner2f7c9632001-06-06 20:29:01 +0000994 // Print out the opcode...
Misha Brukmana6619a92004-06-21 21:53:56 +0000995 Out << I.getOpcodeName();
Chris Lattner2f7c9632001-06-06 20:29:01 +0000996
997 // Print out the type of the operands...
Chris Lattner113f4f42002-06-25 16:13:24 +0000998 const Value *Operand = I.getNumOperands() ? I.getOperand(0) : 0;
Chris Lattner2f7c9632001-06-06 20:29:01 +0000999
1000 // Special case conditional branches to swizzle the condition out to the front
Chris Lattner113f4f42002-06-25 16:13:24 +00001001 if (isa<BranchInst>(I) && I.getNumOperands() > 1) {
1002 writeOperand(I.getOperand(2), true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001003 Out << ',';
Chris Lattner2f7c9632001-06-06 20:29:01 +00001004 writeOperand(Operand, true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001005 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001006 writeOperand(I.getOperand(1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001007
Chris Lattner8d48df22002-04-13 18:34:38 +00001008 } else if (isa<SwitchInst>(I)) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00001009 // Special case switch statement to get formatting nice and correct...
Misha Brukmana6619a92004-06-21 21:53:56 +00001010 writeOperand(Operand , true); Out << ',';
1011 writeOperand(I.getOperand(1), true); Out << " [";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001012
Chris Lattner113f4f42002-06-25 16:13:24 +00001013 for (unsigned op = 2, Eop = I.getNumOperands(); op < Eop; op += 2) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001014 Out << "\n\t\t";
1015 writeOperand(I.getOperand(op ), true); Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001016 writeOperand(I.getOperand(op+1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001017 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001018 Out << "\n\t]";
Chris Lattnerda558102001-10-02 03:41:24 +00001019 } else if (isa<PHINode>(I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001020 Out << ' ';
Chris Lattner113f4f42002-06-25 16:13:24 +00001021 printType(I.getType());
Misha Brukmana6619a92004-06-21 21:53:56 +00001022 Out << ' ';
Chris Lattner2f7c9632001-06-06 20:29:01 +00001023
Chris Lattner113f4f42002-06-25 16:13:24 +00001024 for (unsigned op = 0, Eop = I.getNumOperands(); op < Eop; op += 2) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001025 if (op) Out << ", ";
1026 Out << '[';
1027 writeOperand(I.getOperand(op ), false); Out << ',';
1028 writeOperand(I.getOperand(op+1), false); Out << " ]";
Chris Lattner931ef3b2001-06-11 15:04:20 +00001029 }
Chris Lattner862e3382001-10-13 06:42:36 +00001030 } else if (isa<ReturnInst>(I) && !Operand) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001031 Out << " void";
Chris Lattner862e3382001-10-13 06:42:36 +00001032 } else if (isa<CallInst>(I)) {
Chris Lattner463d6a52003-08-05 15:34:45 +00001033 const PointerType *PTy = cast<PointerType>(Operand->getType());
1034 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1035 const Type *RetTy = FTy->getReturnType();
Chris Lattner2f2d9472001-11-06 21:28:12 +00001036
Chris Lattner463d6a52003-08-05 15:34:45 +00001037 // If possible, print out the short form of the call instruction. We can
Chris Lattner6915f8f2002-04-07 22:49:37 +00001038 // only do this if the first argument is a pointer to a nonvararg function,
Chris Lattner463d6a52003-08-05 15:34:45 +00001039 // and if the return type is not a pointer to a function.
Chris Lattner2f2d9472001-11-06 21:28:12 +00001040 //
Chris Lattner463d6a52003-08-05 15:34:45 +00001041 if (!FTy->isVarArg() &&
Chris Lattner8d48df22002-04-13 18:34:38 +00001042 (!isa<PointerType>(RetTy) ||
Chris Lattnerd9a36a62002-07-25 20:58:51 +00001043 !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001044 Out << ' '; printType(RetTy);
Chris Lattner2f2d9472001-11-06 21:28:12 +00001045 writeOperand(Operand, false);
1046 } else {
1047 writeOperand(Operand, true);
1048 }
Misha Brukmana6619a92004-06-21 21:53:56 +00001049 Out << '(';
Chris Lattner113f4f42002-06-25 16:13:24 +00001050 if (I.getNumOperands() > 1) writeOperand(I.getOperand(1), true);
1051 for (unsigned op = 2, Eop = I.getNumOperands(); op < Eop; ++op) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001052 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001053 writeOperand(I.getOperand(op), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001054 }
1055
Misha Brukmana6619a92004-06-21 21:53:56 +00001056 Out << " )";
Chris Lattner113f4f42002-06-25 16:13:24 +00001057 } else if (const InvokeInst *II = dyn_cast<InvokeInst>(&I)) {
Chris Lattner463d6a52003-08-05 15:34:45 +00001058 const PointerType *PTy = cast<PointerType>(Operand->getType());
1059 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
1060 const Type *RetTy = FTy->getReturnType();
1061
1062 // If possible, print out the short form of the invoke instruction. We can
1063 // only do this if the first argument is a pointer to a nonvararg function,
1064 // and if the return type is not a pointer to a function.
1065 //
1066 if (!FTy->isVarArg() &&
1067 (!isa<PointerType>(RetTy) ||
1068 !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001069 Out << ' '; printType(RetTy);
Chris Lattner463d6a52003-08-05 15:34:45 +00001070 writeOperand(Operand, false);
1071 } else {
1072 writeOperand(Operand, true);
1073 }
1074
Misha Brukmana6619a92004-06-21 21:53:56 +00001075 Out << '(';
Chris Lattner113f4f42002-06-25 16:13:24 +00001076 if (I.getNumOperands() > 3) writeOperand(I.getOperand(3), true);
1077 for (unsigned op = 4, Eop = I.getNumOperands(); op < Eop; ++op) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001078 Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001079 writeOperand(I.getOperand(op), true);
Chris Lattner862e3382001-10-13 06:42:36 +00001080 }
1081
Misha Brukmana6619a92004-06-21 21:53:56 +00001082 Out << " )\n\t\t\tto";
Chris Lattner862e3382001-10-13 06:42:36 +00001083 writeOperand(II->getNormalDest(), true);
Misha Brukmana6619a92004-06-21 21:53:56 +00001084 Out << " unwind";
Chris Lattnerfae8ab32004-02-08 21:44:31 +00001085 writeOperand(II->getUnwindDest(), true);
Chris Lattner862e3382001-10-13 06:42:36 +00001086
Chris Lattner113f4f42002-06-25 16:13:24 +00001087 } else if (const AllocationInst *AI = dyn_cast<AllocationInst>(&I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001088 Out << ' ';
Chris Lattner8d48df22002-04-13 18:34:38 +00001089 printType(AI->getType()->getElementType());
1090 if (AI->isArrayAllocation()) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001091 Out << ',';
Chris Lattner8d48df22002-04-13 18:34:38 +00001092 writeOperand(AI->getArraySize(), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001093 }
Chris Lattner862e3382001-10-13 06:42:36 +00001094 } else if (isa<CastInst>(I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001095 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001096 Out << " to ";
Chris Lattner113f4f42002-06-25 16:13:24 +00001097 printType(I.getType());
Chris Lattner5b337482003-10-18 05:57:43 +00001098 } else if (isa<VAArgInst>(I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001099 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001100 Out << ", ";
Chris Lattnerf70da102003-05-08 02:44:12 +00001101 printType(I.getType());
Chris Lattner5b337482003-10-18 05:57:43 +00001102 } else if (const VANextInst *VAN = dyn_cast<VANextInst>(&I)) {
Chris Lattnerc96e96b2003-11-17 01:17:04 +00001103 if (Operand) writeOperand(Operand, true); // Work with broken code
Misha Brukmana6619a92004-06-21 21:53:56 +00001104 Out << ", ";
Chris Lattner5b337482003-10-18 05:57:43 +00001105 printType(VAN->getArgType());
Chris Lattner2f7c9632001-06-06 20:29:01 +00001106 } else if (Operand) { // Print the normal way...
1107
1108 // PrintAllTypes - Instructions who have operands of all the same type
1109 // omit the type from all but the first operand. If the instruction has
1110 // different type operands (for example br), then they are all printed.
1111 bool PrintAllTypes = false;
1112 const Type *TheType = Operand->getType();
Chris Lattner2f7c9632001-06-06 20:29:01 +00001113
Chris Lattner52bd5cb92004-03-12 05:53:14 +00001114 // Shift Left & Right print both types even for Ubyte LHS, and select prints
1115 // types even if all operands are bools.
1116 if (isa<ShiftInst>(I) || isa<SelectInst>(I)) {
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001117 PrintAllTypes = true;
1118 } else {
1119 for (unsigned i = 1, E = I.getNumOperands(); i != E; ++i) {
1120 Operand = I.getOperand(i);
1121 if (Operand->getType() != TheType) {
1122 PrintAllTypes = true; // We have differing types! Print them all!
1123 break;
1124 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001125 }
1126 }
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001127
Chris Lattner7bfee412001-10-29 16:05:51 +00001128 if (!PrintAllTypes) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001129 Out << ' ';
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00001130 printType(TheType);
Chris Lattner7bfee412001-10-29 16:05:51 +00001131 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00001132
Chris Lattner113f4f42002-06-25 16:13:24 +00001133 for (unsigned i = 0, E = I.getNumOperands(); i != E; ++i) {
Misha Brukmana6619a92004-06-21 21:53:56 +00001134 if (i) Out << ',';
Chris Lattner113f4f42002-06-25 16:13:24 +00001135 writeOperand(I.getOperand(i), PrintAllTypes);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001136 }
1137 }
1138
Chris Lattner862e3382001-10-13 06:42:36 +00001139 printInfoComment(I);
Misha Brukmana6619a92004-06-21 21:53:56 +00001140 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +00001141}
1142
1143
1144//===----------------------------------------------------------------------===//
1145// External Interface declarations
1146//===----------------------------------------------------------------------===//
1147
Chris Lattner8339f7d2003-10-30 23:41:03 +00001148void Module::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001149 SlotMachine SlotTable(this);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001150 AssemblyWriter W(o, SlotTable, this, AAW);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001151 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001152}
1153
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001154void GlobalVariable::print(std::ostream &o) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001155 SlotMachine SlotTable(getParent());
Chris Lattner8339f7d2003-10-30 23:41:03 +00001156 AssemblyWriter W(o, SlotTable, getParent(), 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001157 W.write(this);
Chris Lattneradfe0d12001-09-10 20:08:19 +00001158}
1159
Chris Lattner8339f7d2003-10-30 23:41:03 +00001160void Function::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001161 SlotMachine SlotTable(getParent());
Chris Lattner8339f7d2003-10-30 23:41:03 +00001162 AssemblyWriter W(o, SlotTable, getParent(), AAW);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001163
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001164 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001165}
1166
Chris Lattner8339f7d2003-10-30 23:41:03 +00001167void BasicBlock::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001168 SlotMachine SlotTable(getParent());
Chris Lattner7bfee412001-10-29 16:05:51 +00001169 AssemblyWriter W(o, SlotTable,
Chris Lattner8339f7d2003-10-30 23:41:03 +00001170 getParent() ? getParent()->getParent() : 0, AAW);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001171 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001172}
1173
Chris Lattner8339f7d2003-10-30 23:41:03 +00001174void Instruction::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001175 const Function *F = getParent() ? getParent()->getParent() : 0;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001176 SlotMachine SlotTable(F);
Chris Lattner8339f7d2003-10-30 23:41:03 +00001177 AssemblyWriter W(o, SlotTable, F ? F->getParent() : 0, AAW);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001178
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001179 W.write(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00001180}
Chris Lattner7db79582001-11-07 04:21:57 +00001181
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001182void Constant::print(std::ostream &o) const {
1183 if (this == 0) { o << "<null> constant value\n"; return; }
Chris Lattner7d734802002-09-10 15:53:49 +00001184
Misha Brukman21bbdb92004-06-04 21:11:51 +00001185 o << ' ' << getType()->getDescription() << ' ';
Chris Lattner1e194682002-04-18 18:53:13 +00001186
Chris Lattnerab7d1ab2003-05-08 02:08:14 +00001187 std::map<const Type *, std::string> TypeTable;
Chris Lattner1e194682002-04-18 18:53:13 +00001188 WriteConstantInt(o, this, false, TypeTable, 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001189}
1190
1191void Type::print(std::ostream &o) const {
1192 if (this == 0)
1193 o << "<null Type>";
1194 else
1195 o << getDescription();
1196}
1197
Chris Lattner2e9fa6d2002-04-09 19:48:49 +00001198void Argument::print(std::ostream &o) const {
Chris Lattnere52ed452004-07-13 23:14:34 +00001199 WriteAsOperand(o, this, true, true,
1200 getParent() ? getParent()->getParent() : 0);
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001201}
1202
Reid Spencer52641832004-05-25 18:14:38 +00001203// Value::dump - allow easy printing of Values from the debugger.
1204// Located here because so much of the needed functionality is here.
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001205void Value::dump() const { print(std::cerr); }
Reid Spencer52641832004-05-25 18:14:38 +00001206
1207// Type::dump - allow easy printing of Values from the debugger.
1208// Located here because so much of the needed functionality is here.
Reid Spencere7e96712004-05-25 08:53:40 +00001209void Type::dump() const { print(std::cerr); }
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00001210
1211//===----------------------------------------------------------------------===//
1212// CachedWriter Class Implementation
1213//===----------------------------------------------------------------------===//
1214
Chris Lattner7db79582001-11-07 04:21:57 +00001215void CachedWriter::setModule(const Module *M) {
1216 delete SC; delete AW;
1217 if (M) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001218 SC = new SlotMachine(M );
Misha Brukman21bbdb92004-06-04 21:11:51 +00001219 AW = new AssemblyWriter(Out, *SC, M, 0);
Chris Lattner7db79582001-11-07 04:21:57 +00001220 } else {
1221 SC = 0; AW = 0;
1222 }
1223}
1224
1225CachedWriter::~CachedWriter() {
1226 delete AW;
1227 delete SC;
1228}
1229
Chris Lattner60a7dd12004-07-15 02:51:31 +00001230CachedWriter &CachedWriter::operator<<(const Value &V) {
Chris Lattner7db79582001-11-07 04:21:57 +00001231 assert(AW && SC && "CachedWriter does not have a current module!");
Chris Lattner60a7dd12004-07-15 02:51:31 +00001232 if (const Instruction *I = dyn_cast<Instruction>(&V))
Chris Lattner80d2d532004-06-26 19:40:40 +00001233 AW->write(I);
Chris Lattner60a7dd12004-07-15 02:51:31 +00001234 else if (const BasicBlock *BB = dyn_cast<BasicBlock>(&V))
Chris Lattner80d2d532004-06-26 19:40:40 +00001235 AW->write(BB);
Chris Lattner60a7dd12004-07-15 02:51:31 +00001236 else if (const Function *F = dyn_cast<Function>(&V))
Chris Lattner80d2d532004-06-26 19:40:40 +00001237 AW->write(F);
Chris Lattner60a7dd12004-07-15 02:51:31 +00001238 else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(&V))
Chris Lattner80d2d532004-06-26 19:40:40 +00001239 AW->write(GV);
Chris Lattner80d2d532004-06-26 19:40:40 +00001240 else
Chris Lattner60a7dd12004-07-15 02:51:31 +00001241 AW->writeOperand(&V, true, true);
Chris Lattner7db79582001-11-07 04:21:57 +00001242 return *this;
1243}
Misha Brukman4685e262004-04-28 15:31:21 +00001244
Chris Lattner60a7dd12004-07-15 02:51:31 +00001245CachedWriter& CachedWriter::operator<<(const Type &Ty) {
Misha Brukman4685e262004-04-28 15:31:21 +00001246 if (SymbolicTypes) {
1247 const Module *M = AW->getModule();
Chris Lattner60a7dd12004-07-15 02:51:31 +00001248 if (M) WriteTypeSymbolic(Out, &Ty, M);
Reid Spencer58d30f22004-07-04 11:50:43 +00001249 } else {
Chris Lattner60a7dd12004-07-15 02:51:31 +00001250 AW->write(&Ty);
Reid Spencer58d30f22004-07-04 11:50:43 +00001251 }
1252 return *this;
Misha Brukman4685e262004-04-28 15:31:21 +00001253}
Misha Brukmanda546ea2004-04-28 19:24:28 +00001254
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001255//===----------------------------------------------------------------------===//
1256//===-- SlotMachine Implementation
1257//===----------------------------------------------------------------------===//
1258
1259#if 0
1260#define SC_DEBUG(X) std::cerr << X
1261#else
1262#define SC_DEBUG(X)
1263#endif
1264
1265// Module level constructor. Causes the contents of the Module (sans functions)
1266// to be added to the slot table.
1267SlotMachine::SlotMachine(const Module *M)
Reid Spencer56010e42004-05-26 21:56:09 +00001268 : TheModule(M) ///< Saved for lazy initialization.
1269 , TheFunction(0)
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001270 , FunctionProcessed(false)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001271 , mMap()
Reid Spencer58d30f22004-07-04 11:50:43 +00001272 , mTypes()
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001273 , fMap()
Reid Spencer58d30f22004-07-04 11:50:43 +00001274 , fTypes()
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001275{
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001276}
1277
1278// Function level constructor. Causes the contents of the Module and the one
1279// function provided to be added to the slot table.
1280SlotMachine::SlotMachine(const Function *F )
Reid Spencer56010e42004-05-26 21:56:09 +00001281 : TheModule( F ? F->getParent() : 0 ) ///< Saved for lazy initialization
1282 , TheFunction(F) ///< Saved for lazy initialization
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001283 , FunctionProcessed(false)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001284 , mMap()
Reid Spencer58d30f22004-07-04 11:50:43 +00001285 , mTypes()
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001286 , fMap()
Reid Spencer58d30f22004-07-04 11:50:43 +00001287 , fTypes()
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001288{
Reid Spencer56010e42004-05-26 21:56:09 +00001289}
1290
1291inline void SlotMachine::initialize(void) {
1292 if ( TheModule) {
1293 processModule();
1294 TheModule = 0; ///< Prevent re-processing next time we're called.
1295 }
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001296 if ( TheFunction && ! FunctionProcessed) {
Reid Spencer56010e42004-05-26 21:56:09 +00001297 processFunction();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001298 }
1299}
1300
1301// Iterate through all the global variables, functions, and global
1302// variable initializers and create slots for them.
1303void SlotMachine::processModule() {
1304 SC_DEBUG("begin processModule!\n");
1305
1306 // Add all of the global variables to the value table...
1307 for (Module::const_giterator I = TheModule->gbegin(), E = TheModule->gend();
1308 I != E; ++I)
1309 createSlot(I);
1310
1311 // Add all the functions to the table
1312 for (Module::const_iterator I = TheModule->begin(), E = TheModule->end();
1313 I != E; ++I)
1314 createSlot(I);
1315
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001316 SC_DEBUG("end processModule!\n");
1317}
1318
1319
Reid Spencer56010e42004-05-26 21:56:09 +00001320// Process the arguments, basic blocks, and instructions of a function.
1321void SlotMachine::processFunction() {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001322 SC_DEBUG("begin processFunction!\n");
1323
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001324 // Add all the function arguments
Reid Spencer56010e42004-05-26 21:56:09 +00001325 for(Function::const_aiterator AI = TheFunction->abegin(),
1326 AE = TheFunction->aend(); AI != AE; ++AI)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001327 createSlot(AI);
1328
1329 SC_DEBUG("Inserting Instructions:\n");
1330
1331 // Add all of the basic blocks and instructions
Reid Spencer56010e42004-05-26 21:56:09 +00001332 for (Function::const_iterator BB = TheFunction->begin(),
1333 E = TheFunction->end(); BB != E; ++BB) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001334 createSlot(BB);
1335 for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I!=E; ++I) {
1336 createSlot(I);
1337 }
1338 }
1339
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001340 FunctionProcessed = true;
1341
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001342 SC_DEBUG("end processFunction!\n");
1343}
1344
1345// Clean up after incorporating a function. This is the only way
Reid Spencer56010e42004-05-26 21:56:09 +00001346// to get out of the function incorporation state that affects the
1347// getSlot/createSlot lock. Function incorporation state is indicated
1348// by TheFunction != 0.
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001349void SlotMachine::purgeFunction() {
1350 SC_DEBUG("begin purgeFunction!\n");
1351 fMap.clear(); // Simply discard the function level map
Reid Spencer58d30f22004-07-04 11:50:43 +00001352 fTypes.clear();
Reid Spencer56010e42004-05-26 21:56:09 +00001353 TheFunction = 0;
Reid Spencerb0ac8c42004-08-16 07:46:33 +00001354 FunctionProcessed = false;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001355 SC_DEBUG("end purgeFunction!\n");
1356}
1357
1358/// Get the slot number for a value. This function will assert if you
1359/// ask for a Value that hasn't previously been inserted with createSlot.
1360/// Types are forbidden because Type does not inherit from Value (any more).
Chris Lattner757ee0b2004-06-09 19:41:19 +00001361int SlotMachine::getSlot(const Value *V) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001362 assert( V && "Can't get slot for null Value" );
Reid Spencer56010e42004-05-26 21:56:09 +00001363 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1364 "Can't insert a non-GlobalValue Constant into SlotMachine");
1365
1366 // Check for uninitialized state and do lazy initialization
1367 this->initialize();
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001368
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001369 // Get the type of the value
1370 const Type* VTy = V->getType();
1371
1372 // Find the type plane in the module map
1373 TypedPlanes::const_iterator MI = mMap.find(VTy);
1374
Reid Spencer56010e42004-05-26 21:56:09 +00001375 if ( TheFunction ) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001376 // Lookup the type in the function map too
1377 TypedPlanes::const_iterator FI = fMap.find(VTy);
1378 // If there is a corresponding type plane in the function map
1379 if ( FI != fMap.end() ) {
1380 // Lookup the Value in the function map
1381 ValueMap::const_iterator FVI = FI->second.map.find(V);
1382 // If the value doesn't exist in the function map
1383 if ( FVI == FI->second.map.end() ) {
Chris Lattner68a038e2004-06-09 22:22:10 +00001384 // Look up the value in the module map.
1385 if (MI == mMap.end()) return -1;
Reid Spencer56010e42004-05-26 21:56:09 +00001386 ValueMap::const_iterator MVI = MI->second.map.find(V);
1387 // If we didn't find it, it wasn't inserted
Chris Lattner757ee0b2004-06-09 19:41:19 +00001388 if (MVI == MI->second.map.end()) return -1;
Reid Spencer56010e42004-05-26 21:56:09 +00001389 assert( MVI != MI->second.map.end() && "Value not found");
1390 // We found it only at the module level
1391 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001392
1393 // else the value exists in the function map
1394 } else {
Reid Spencer56010e42004-05-26 21:56:09 +00001395 // Return the slot number as the module's contribution to
1396 // the type plane plus the index in the function's contribution
1397 // to the type plane.
Chris Lattnerb1f04782004-06-15 21:07:32 +00001398 if (MI != mMap.end())
1399 return MI->second.next_slot + FVI->second;
1400 else
1401 return FVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001402 }
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001403 }
1404 }
1405
Reid Spencer8beac692004-06-09 15:26:53 +00001406 // N.B. Can get here only if either !TheFunction or the function doesn't
1407 // have a corresponding type plane for the Value
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001408
1409 // Make sure the type plane exists
Chris Lattner757ee0b2004-06-09 19:41:19 +00001410 if (MI == mMap.end()) return -1;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001411 // Lookup the value in the module's map
1412 ValueMap::const_iterator MVI = MI->second.map.find(V);
1413 // Make sure we found it.
Chris Lattner757ee0b2004-06-09 19:41:19 +00001414 if (MVI == MI->second.map.end()) return -1;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001415 // Return it.
1416 return MVI->second;
1417}
1418
Reid Spencer58d30f22004-07-04 11:50:43 +00001419/// Get the slot number for a value. This function will assert if you
1420/// ask for a Value that hasn't previously been inserted with createSlot.
1421/// Types are forbidden because Type does not inherit from Value (any more).
1422int SlotMachine::getSlot(const Type *Ty) {
1423 assert( Ty && "Can't get slot for null Type" );
1424
1425 // Check for uninitialized state and do lazy initialization
1426 this->initialize();
1427
1428 if ( TheFunction ) {
1429 // Lookup the Type in the function map
1430 TypeMap::const_iterator FTI = fTypes.map.find(Ty);
1431 // If the Type doesn't exist in the function map
1432 if ( FTI == fTypes.map.end() ) {
1433 TypeMap::const_iterator MTI = mTypes.map.find(Ty);
1434 // If we didn't find it, it wasn't inserted
1435 if (MTI == mTypes.map.end())
1436 return -1;
1437 // We found it only at the module level
1438 return MTI->second;
1439
1440 // else the value exists in the function map
1441 } else {
1442 // Return the slot number as the module's contribution to
1443 // the type plane plus the index in the function's contribution
1444 // to the type plane.
1445 return mTypes.next_slot + FTI->second;
1446 }
1447 }
1448
1449 // N.B. Can get here only if either !TheFunction
1450
1451 // Lookup the value in the module's map
1452 TypeMap::const_iterator MTI = mTypes.map.find(Ty);
1453 // Make sure we found it.
1454 if (MTI == mTypes.map.end()) return -1;
1455 // Return it.
1456 return MTI->second;
1457}
1458
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001459// Create a new slot, or return the existing slot if it is already
1460// inserted. Note that the logic here parallels getSlot but instead
1461// of asserting when the Value* isn't found, it inserts the value.
1462unsigned SlotMachine::createSlot(const Value *V) {
1463 assert( V && "Can't insert a null Value to SlotMachine");
Reid Spencer56010e42004-05-26 21:56:09 +00001464 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1465 "Can't insert a non-GlobalValue Constant into SlotMachine");
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001466
1467 const Type* VTy = V->getType();
1468
1469 // Just ignore void typed things
1470 if (VTy == Type::VoidTy) return 0; // FIXME: Wrong return value!
1471
1472 // Look up the type plane for the Value's type from the module map
1473 TypedPlanes::const_iterator MI = mMap.find(VTy);
1474
Reid Spencer56010e42004-05-26 21:56:09 +00001475 if ( TheFunction ) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001476 // Get the type plane for the Value's type from the function map
1477 TypedPlanes::const_iterator FI = fMap.find(VTy);
1478 // If there is a corresponding type plane in the function map
1479 if ( FI != fMap.end() ) {
1480 // Lookup the Value in the function map
1481 ValueMap::const_iterator FVI = FI->second.map.find(V);
1482 // If the value doesn't exist in the function map
1483 if ( FVI == FI->second.map.end() ) {
Reid Spencer56010e42004-05-26 21:56:09 +00001484 // If there is no corresponding type plane in the module map
1485 if ( MI == mMap.end() )
1486 return insertValue(V);
1487 // Look up the value in the module map
1488 ValueMap::const_iterator MVI = MI->second.map.find(V);
1489 // If we didn't find it, it wasn't inserted
1490 if ( MVI == MI->second.map.end() )
1491 return insertValue(V);
1492 else
1493 // We found it only at the module level
1494 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001495
1496 // else the value exists in the function map
1497 } else {
Reid Spencer56010e42004-05-26 21:56:09 +00001498 if ( MI == mMap.end() )
1499 return FVI->second;
1500 else
1501 // Return the slot number as the module's contribution to
1502 // the type plane plus the index in the function's contribution
1503 // to the type plane.
1504 return MI->second.next_slot + FVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001505 }
1506
1507 // else there is not a corresponding type plane in the function map
1508 } else {
1509 // If the type plane doesn't exists at the module level
1510 if ( MI == mMap.end() ) {
Reid Spencer56010e42004-05-26 21:56:09 +00001511 return insertValue(V);
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001512 // else type plane exists at the module level, examine it
1513 } else {
Reid Spencer56010e42004-05-26 21:56:09 +00001514 // Look up the value in the module's map
1515 ValueMap::const_iterator MVI = MI->second.map.find(V);
1516 // If we didn't find it there either
1517 if ( MVI == MI->second.map.end() )
1518 // Return the slot number as the module's contribution to
1519 // the type plane plus the index of the function map insertion.
1520 return MI->second.next_slot + insertValue(V);
1521 else
1522 return MVI->second;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001523 }
1524 }
1525 }
1526
Reid Spencer56010e42004-05-26 21:56:09 +00001527 // N.B. Can only get here if !TheFunction
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001528
1529 // If the module map's type plane is not for the Value's type
1530 if ( MI != mMap.end() ) {
1531 // Lookup the value in the module's map
1532 ValueMap::const_iterator MVI = MI->second.map.find(V);
1533 if ( MVI != MI->second.map.end() )
1534 return MVI->second;
1535 }
1536
1537 return insertValue(V);
1538}
1539
Reid Spencer58d30f22004-07-04 11:50:43 +00001540// Create a new slot, or return the existing slot if it is already
1541// inserted. Note that the logic here parallels getSlot but instead
1542// of asserting when the Value* isn't found, it inserts the value.
1543unsigned SlotMachine::createSlot(const Type *Ty) {
1544 assert( Ty && "Can't insert a null Type to SlotMachine");
1545
1546 if ( TheFunction ) {
1547 // Lookup the Type in the function map
1548 TypeMap::const_iterator FTI = fTypes.map.find(Ty);
1549 // If the type doesn't exist in the function map
1550 if ( FTI == fTypes.map.end() ) {
1551 // Look up the type in the module map
1552 TypeMap::const_iterator MTI = mTypes.map.find(Ty);
1553 // If we didn't find it, it wasn't inserted
1554 if ( MTI == mTypes.map.end() )
1555 return insertValue(Ty);
1556 else
1557 // We found it only at the module level
1558 return MTI->second;
1559
1560 // else the value exists in the function map
1561 } else {
1562 // Return the slot number as the module's contribution to
1563 // the type plane plus the index in the function's contribution
1564 // to the type plane.
1565 return mTypes.next_slot + FTI->second;
1566 }
1567 }
1568
1569 // N.B. Can only get here if !TheFunction
1570
1571 // Lookup the type in the module's map
1572 TypeMap::const_iterator MTI = mTypes.map.find(Ty);
1573 if ( MTI != mTypes.map.end() )
1574 return MTI->second;
1575
1576 return insertValue(Ty);
1577}
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001578
1579// Low level insert function. Minimal checking is done. This
1580// function is just for the convenience of createSlot (above).
1581unsigned SlotMachine::insertValue(const Value *V ) {
1582 assert(V && "Can't insert a null Value into SlotMachine!");
Reid Spencer56010e42004-05-26 21:56:09 +00001583 assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1584 "Can't insert a non-GlobalValue Constant into SlotMachine");
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001585
Reid Spencer56010e42004-05-26 21:56:09 +00001586 // If this value does not contribute to a plane (is void)
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001587 // or if the value already has a name then ignore it.
Reid Spencer56010e42004-05-26 21:56:09 +00001588 if (V->getType() == Type::VoidTy || V->hasName() ) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001589 SC_DEBUG("ignored value " << *V << "\n");
1590 return 0; // FIXME: Wrong return value
1591 }
1592
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001593 const Type *VTy = V->getType();
1594 unsigned DestSlot = 0;
1595
Reid Spencer56010e42004-05-26 21:56:09 +00001596 if ( TheFunction ) {
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001597 TypedPlanes::iterator I = fMap.find( VTy );
1598 if ( I == fMap.end() )
Reid Spencer58d30f22004-07-04 11:50:43 +00001599 I = fMap.insert(std::make_pair(VTy,ValuePlane())).first;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001600 DestSlot = I->second.map[V] = I->second.next_slot++;
1601 } else {
1602 TypedPlanes::iterator I = mMap.find( VTy );
1603 if ( I == mMap.end() )
Reid Spencer58d30f22004-07-04 11:50:43 +00001604 I = mMap.insert(std::make_pair(VTy,ValuePlane())).first;
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001605 DestSlot = I->second.map[V] = I->second.next_slot++;
1606 }
1607
1608 SC_DEBUG(" Inserting value [" << VTy << "] = " << V << " slot=" <<
Reid Spencer56010e42004-05-26 21:56:09 +00001609 DestSlot << " [");
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001610 // G = Global, C = Constant, T = Type, F = Function, o = other
Reid Spenceraccd7c72004-07-17 23:47:01 +00001611 SC_DEBUG((isa<GlobalVariable>(V) ? 'G' : (isa<Function>(V) ? 'F' :
1612 (isa<Constant>(V) ? 'C' : 'o'))));
Reid Spencer16f2f7f2004-05-26 07:18:52 +00001613 SC_DEBUG("]\n");
1614 return DestSlot;
1615}
1616
Reid Spencer58d30f22004-07-04 11:50:43 +00001617// Low level insert function. Minimal checking is done. This
1618// function is just for the convenience of createSlot (above).
1619unsigned SlotMachine::insertValue(const Type *Ty ) {
1620 assert(Ty && "Can't insert a null Type into SlotMachine!");
1621
1622 unsigned DestSlot = 0;
1623
1624 if ( TheFunction ) {
1625 DestSlot = fTypes.map[Ty] = fTypes.next_slot++;
1626 } else {
1627 DestSlot = fTypes.map[Ty] = fTypes.next_slot++;
1628 }
1629 SC_DEBUG(" Inserting type [" << DestSlot << "] = " << Ty << "\n");
1630 return DestSlot;
1631}
1632
Reid Spencere7e96712004-05-25 08:53:40 +00001633// vim: sw=2