Chris Lattner | f7e7948 | 2002-04-07 22:31:46 +0000 | [diff] [blame] | 1 | //===-- AsmWriter.cpp - Printing LLVM as an assembly file -----------------===// |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 2 | // |
| 3 | // This library implements the functionality defined in llvm/Assembly/Writer.h |
| 4 | // |
Chris Lattner | 189088e | 2002-04-12 18:21:53 +0000 | [diff] [blame] | 5 | // Note that these routines must be extremely tolerant of various errors in the |
| 6 | // LLVM code, because of of the primary uses of it is for debugging |
| 7 | // transformations. |
| 8 | // |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 9 | // TODO: print out the type name instead of the full type if a particular type |
Chris Lattner | 189088e | 2002-04-12 18:21:53 +0000 | [diff] [blame] | 10 | // is in the symbol table... |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
Chris Lattner | 7db7958 | 2001-11-07 04:21:57 +0000 | [diff] [blame] | 14 | #include "llvm/Assembly/CachedWriter.h" |
Chris Lattner | c0b4c7b | 2002-04-08 22:03:40 +0000 | [diff] [blame] | 15 | #include "llvm/Assembly/Writer.h" |
Chris Lattner | 6915f8f | 2002-04-07 22:49:37 +0000 | [diff] [blame] | 16 | #include "llvm/SlotCalculator.h" |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 17 | #include "llvm/Module.h" |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 18 | #include "llvm/Function.h" |
Chris Lattner | da97550 | 2001-09-10 07:58:01 +0000 | [diff] [blame] | 19 | #include "llvm/GlobalVariable.h" |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 20 | #include "llvm/BasicBlock.h" |
Chris Lattner | 3462ae3 | 2001-12-03 22:26:30 +0000 | [diff] [blame] | 21 | #include "llvm/ConstantVals.h" |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 22 | #include "llvm/iMemory.h" |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 23 | #include "llvm/iTerminators.h" |
Chris Lattner | fb5ae02 | 2001-12-03 18:02:31 +0000 | [diff] [blame] | 24 | #include "llvm/iPHINode.h" |
| 25 | #include "llvm/iOther.h" |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 26 | #include "llvm/SymbolTable.h" |
Chris Lattner | 2e9fa6d | 2002-04-09 19:48:49 +0000 | [diff] [blame] | 27 | #include "llvm/Argument.h" |
Chris Lattner | 5de2204 | 2001-11-27 00:03:19 +0000 | [diff] [blame] | 28 | #include "Support/StringExtras.h" |
| 29 | #include "Support/STLExtras.h" |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 30 | #include <algorithm> |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 31 | #include <map> |
Chris Lattner | 7f74a56 | 2002-01-20 22:54:45 +0000 | [diff] [blame] | 32 | using std::string; |
| 33 | using std::map; |
| 34 | using std::vector; |
| 35 | using std::ostream; |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 36 | |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 37 | static const Module *getModuleFromVal(const Value *V) { |
Chris Lattner | 2e9fa6d | 2002-04-09 19:48:49 +0000 | [diff] [blame] | 38 | if (const Argument *MA = dyn_cast<const Argument>(V)) |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 39 | return MA->getParent() ? MA->getParent()->getParent() : 0; |
| 40 | else if (const BasicBlock *BB = dyn_cast<const BasicBlock>(V)) |
| 41 | return BB->getParent() ? BB->getParent()->getParent() : 0; |
| 42 | else if (const Instruction *I = dyn_cast<const Instruction>(V)) { |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 43 | const Function *M = I->getParent() ? I->getParent()->getParent() : 0; |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 44 | return M ? M->getParent() : 0; |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 45 | } else if (const GlobalValue *GV = dyn_cast<const GlobalValue>(V)) |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 46 | return GV->getParent(); |
| 47 | else if (const Module *Mod = dyn_cast<const Module>(V)) |
| 48 | return Mod; |
| 49 | return 0; |
| 50 | } |
| 51 | |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 52 | static SlotCalculator *createSlotCalculator(const Value *V) { |
| 53 | assert(!isa<Type>(V) && "Can't create an SC for a type!"); |
Chris Lattner | 2e9fa6d | 2002-04-09 19:48:49 +0000 | [diff] [blame] | 54 | if (const Argument *FA = dyn_cast<const Argument>(V)) { |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 55 | return new SlotCalculator(FA->getParent(), true); |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 56 | } else if (const Instruction *I = dyn_cast<const Instruction>(V)) { |
| 57 | return new SlotCalculator(I->getParent()->getParent(), true); |
| 58 | } else if (const BasicBlock *BB = dyn_cast<const BasicBlock>(V)) { |
| 59 | return new SlotCalculator(BB->getParent(), true); |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 60 | } else if (const GlobalVariable *GV = dyn_cast<const GlobalVariable>(V)){ |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 61 | return new SlotCalculator(GV->getParent(), true); |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 62 | } else if (const Function *Func = dyn_cast<const Function>(V)) { |
| 63 | return new SlotCalculator(Func, true); |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 64 | } else if (const Module *Mod = dyn_cast<const Module>(V)) { |
| 65 | return new SlotCalculator(Mod, true); |
| 66 | } |
| 67 | return 0; |
| 68 | } |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 69 | |
Chris Lattner | 5e5abe3 | 2001-07-20 19:15:21 +0000 | [diff] [blame] | 70 | // WriteAsOperand - Write the name of the specified value out to the specified |
| 71 | // ostream. This can be useful when you just want to print int %reg126, not the |
| 72 | // whole instruction that generated it. |
| 73 | // |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 74 | static void WriteAsOperandInternal(ostream &Out, const Value *V, bool PrintName, |
| 75 | SlotCalculator *Table) { |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 76 | if (PrintName && V->hasName()) { |
Chris Lattner | 5e5abe3 | 2001-07-20 19:15:21 +0000 | [diff] [blame] | 77 | Out << " %" << V->getName(); |
| 78 | } else { |
Chris Lattner | 3462ae3 | 2001-12-03 22:26:30 +0000 | [diff] [blame] | 79 | if (const Constant *CPV = dyn_cast<const Constant>(V)) { |
Chris Lattner | 5e5abe3 | 2001-07-20 19:15:21 +0000 | [diff] [blame] | 80 | Out << " " << CPV->getStrValue(); |
| 81 | } else { |
| 82 | int Slot; |
| 83 | if (Table) { |
| 84 | Slot = Table->getValSlot(V); |
| 85 | } else { |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 86 | if (const Type *Ty = dyn_cast<const Type>(V)) { |
| 87 | Out << " " << Ty->getDescription(); |
| 88 | return; |
| 89 | } |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 90 | |
| 91 | Table = createSlotCalculator(V); |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 92 | if (Table == 0) { Out << "BAD VALUE TYPE!"; return; } |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 93 | |
Chris Lattner | 5e5abe3 | 2001-07-20 19:15:21 +0000 | [diff] [blame] | 94 | Slot = Table->getValSlot(V); |
| 95 | delete Table; |
| 96 | } |
| 97 | if (Slot >= 0) Out << " %" << Slot; |
| 98 | else if (PrintName) |
| 99 | Out << "<badref>"; // Not embeded into a location? |
| 100 | } |
| 101 | } |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 102 | } |
| 103 | |
| 104 | |
| 105 | // If the module has a symbol table, take all global types and stuff their |
| 106 | // names into the TypeNames map. |
| 107 | // |
| 108 | static void fillTypeNameTable(const Module *M, |
| 109 | map<const Type *, string> &TypeNames) { |
| 110 | if (M && M->hasSymbolTable()) { |
| 111 | const SymbolTable *ST = M->getSymbolTable(); |
| 112 | SymbolTable::const_iterator PI = ST->find(Type::TypeTy); |
| 113 | if (PI != ST->end()) { |
| 114 | SymbolTable::type_const_iterator I = PI->second.begin(); |
| 115 | for (; I != PI->second.end(); ++I) { |
| 116 | // As a heuristic, don't insert pointer to primitive types, because |
| 117 | // they are used too often to have a single useful name. |
| 118 | // |
| 119 | const Type *Ty = cast<const Type>(I->second); |
| 120 | if (!isa<PointerType>(Ty) || |
Chris Lattner | 2413b16 | 2001-12-04 00:03:30 +0000 | [diff] [blame] | 121 | !cast<PointerType>(Ty)->getElementType()->isPrimitiveType()) |
Chris Lattner | 7f74a56 | 2002-01-20 22:54:45 +0000 | [diff] [blame] | 122 | TypeNames.insert(std::make_pair(Ty, "%"+I->first)); |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 123 | } |
| 124 | } |
| 125 | } |
| 126 | } |
| 127 | |
| 128 | |
| 129 | |
| 130 | static string calcTypeName(const Type *Ty, vector<const Type *> &TypeStack, |
| 131 | map<const Type *, string> &TypeNames) { |
| 132 | if (Ty->isPrimitiveType()) return Ty->getDescription(); // Base case |
| 133 | |
| 134 | // Check to see if the type is named. |
| 135 | map<const Type *, string>::iterator I = TypeNames.find(Ty); |
| 136 | if (I != TypeNames.end()) return I->second; |
| 137 | |
| 138 | // Check to see if the Type is already on the stack... |
| 139 | unsigned Slot = 0, CurSize = TypeStack.size(); |
| 140 | while (Slot < CurSize && TypeStack[Slot] != Ty) ++Slot; // Scan for type |
| 141 | |
| 142 | // This is another base case for the recursion. In this case, we know |
| 143 | // that we have looped back to a type that we have previously visited. |
| 144 | // Generate the appropriate upreference to handle this. |
| 145 | // |
| 146 | if (Slot < CurSize) |
| 147 | return "\\" + utostr(CurSize-Slot); // Here's the upreference |
| 148 | |
| 149 | TypeStack.push_back(Ty); // Recursive case: Add us to the stack.. |
| 150 | |
| 151 | string Result; |
| 152 | switch (Ty->getPrimitiveID()) { |
Chris Lattner | 91db582 | 2002-03-29 03:44:36 +0000 | [diff] [blame] | 153 | case Type::FunctionTyID: { |
| 154 | const FunctionType *MTy = cast<const FunctionType>(Ty); |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 155 | Result = calcTypeName(MTy->getReturnType(), TypeStack, TypeNames) + " ("; |
Chris Lattner | 91db582 | 2002-03-29 03:44:36 +0000 | [diff] [blame] | 156 | for (FunctionType::ParamTypes::const_iterator |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 157 | I = MTy->getParamTypes().begin(), |
| 158 | E = MTy->getParamTypes().end(); I != E; ++I) { |
| 159 | if (I != MTy->getParamTypes().begin()) |
| 160 | Result += ", "; |
| 161 | Result += calcTypeName(*I, TypeStack, TypeNames); |
| 162 | } |
| 163 | if (MTy->isVarArg()) { |
| 164 | if (!MTy->getParamTypes().empty()) Result += ", "; |
| 165 | Result += "..."; |
| 166 | } |
| 167 | Result += ")"; |
| 168 | break; |
| 169 | } |
| 170 | case Type::StructTyID: { |
| 171 | const StructType *STy = cast<const StructType>(Ty); |
| 172 | Result = "{ "; |
| 173 | for (StructType::ElementTypes::const_iterator |
| 174 | I = STy->getElementTypes().begin(), |
| 175 | E = STy->getElementTypes().end(); I != E; ++I) { |
| 176 | if (I != STy->getElementTypes().begin()) |
| 177 | Result += ", "; |
| 178 | Result += calcTypeName(*I, TypeStack, TypeNames); |
| 179 | } |
| 180 | Result += " }"; |
| 181 | break; |
| 182 | } |
| 183 | case Type::PointerTyID: |
Chris Lattner | 2413b16 | 2001-12-04 00:03:30 +0000 | [diff] [blame] | 184 | Result = calcTypeName(cast<const PointerType>(Ty)->getElementType(), |
Chris Lattner | 189088e | 2002-04-12 18:21:53 +0000 | [diff] [blame] | 185 | TypeStack, TypeNames) + "*"; |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 186 | break; |
| 187 | case Type::ArrayTyID: { |
| 188 | const ArrayType *ATy = cast<const ArrayType>(Ty); |
| 189 | int NumElements = ATy->getNumElements(); |
| 190 | Result = "["; |
| 191 | if (NumElements != -1) Result += itostr(NumElements) + " x "; |
| 192 | Result += calcTypeName(ATy->getElementType(), TypeStack, TypeNames) + "]"; |
| 193 | break; |
| 194 | } |
| 195 | default: |
| 196 | assert(0 && "Unhandled case in getTypeProps!"); |
| 197 | Result = "<error>"; |
| 198 | } |
| 199 | |
| 200 | TypeStack.pop_back(); // Remove self from stack... |
| 201 | return Result; |
| 202 | } |
| 203 | |
| 204 | |
| 205 | // printTypeInt - The internal guts of printing out a type that has a |
| 206 | // potentially named portion. |
| 207 | // |
| 208 | static ostream &printTypeInt(ostream &Out, const Type *Ty, |
| 209 | map<const Type *, string> &TypeNames) { |
| 210 | // Primitive types always print out their description, regardless of whether |
| 211 | // they have been named or not. |
| 212 | // |
| 213 | if (Ty->isPrimitiveType()) return Out << Ty->getDescription(); |
| 214 | |
| 215 | // Check to see if the type is named. |
| 216 | map<const Type *, string>::iterator I = TypeNames.find(Ty); |
| 217 | if (I != TypeNames.end()) return Out << I->second; |
| 218 | |
| 219 | // Otherwise we have a type that has not been named but is a derived type. |
| 220 | // Carefully recurse the type hierarchy to print out any contained symbolic |
| 221 | // names. |
| 222 | // |
| 223 | vector<const Type *> TypeStack; |
| 224 | string TypeName = calcTypeName(Ty, TypeStack, TypeNames); |
Chris Lattner | 7f74a56 | 2002-01-20 22:54:45 +0000 | [diff] [blame] | 225 | TypeNames.insert(std::make_pair(Ty, TypeName));//Cache type name for later use |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 226 | return Out << TypeName; |
| 227 | } |
| 228 | |
Chris Lattner | 34b9518 | 2001-10-31 04:33:19 +0000 | [diff] [blame] | 229 | |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 230 | // WriteTypeSymbolic - This attempts to write the specified type as a symbolic |
| 231 | // type, iff there is an entry in the modules symbol table for the specified |
| 232 | // type or one of it's component types. This is slower than a simple x << Type; |
| 233 | // |
| 234 | ostream &WriteTypeSymbolic(ostream &Out, const Type *Ty, const Module *M) { |
| 235 | Out << " "; |
| 236 | |
| 237 | // If they want us to print out a type, attempt to make it symbolic if there |
| 238 | // is a symbol table in the module... |
| 239 | if (M && M->hasSymbolTable()) { |
| 240 | map<const Type *, string> TypeNames; |
| 241 | fillTypeNameTable(M, TypeNames); |
| 242 | |
Chris Lattner | 72f866e | 2001-10-29 16:40:32 +0000 | [diff] [blame] | 243 | return printTypeInt(Out, Ty, TypeNames); |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 244 | } else { |
Chris Lattner | 72f866e | 2001-10-29 16:40:32 +0000 | [diff] [blame] | 245 | return Out << Ty->getDescription(); |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 246 | } |
| 247 | } |
| 248 | |
| 249 | |
| 250 | // WriteAsOperand - Write the name of the specified value out to the specified |
| 251 | // ostream. This can be useful when you just want to print int %reg126, not the |
| 252 | // whole instruction that generated it. |
| 253 | // |
| 254 | ostream &WriteAsOperand(ostream &Out, const Value *V, bool PrintType, |
| 255 | bool PrintName, SlotCalculator *Table) { |
Chris Lattner | 72f866e | 2001-10-29 16:40:32 +0000 | [diff] [blame] | 256 | if (PrintType) |
| 257 | WriteTypeSymbolic(Out, V->getType(), getModuleFromVal(V)); |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 258 | |
| 259 | WriteAsOperandInternal(Out, V, PrintName, Table); |
Chris Lattner | 5e5abe3 | 2001-07-20 19:15:21 +0000 | [diff] [blame] | 260 | return Out; |
| 261 | } |
| 262 | |
| 263 | |
Chris Lattner | 2e9fee4 | 2001-07-12 23:35:26 +0000 | [diff] [blame] | 264 | |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 265 | class AssemblyWriter { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 266 | ostream &Out; |
| 267 | SlotCalculator &Table; |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 268 | const Module *TheModule; |
| 269 | map<const Type *, string> TypeNames; |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 270 | public: |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 271 | inline AssemblyWriter(ostream &o, SlotCalculator &Tab, const Module *M) |
| 272 | : Out(o), Table(Tab), TheModule(M) { |
| 273 | |
| 274 | // If the module has a symbol table, take all global types and stuff their |
| 275 | // names into the TypeNames map. |
| 276 | // |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 277 | fillTypeNameTable(M, TypeNames); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 278 | } |
| 279 | |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 280 | inline void write(const Module *M) { printModule(M); } |
| 281 | inline void write(const GlobalVariable *G) { printGlobal(G); } |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 282 | inline void write(const Function *F) { printFunction(F); } |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 283 | inline void write(const BasicBlock *BB) { printBasicBlock(BB); } |
| 284 | inline void write(const Instruction *I) { printInstruction(I); } |
Chris Lattner | 3462ae3 | 2001-12-03 22:26:30 +0000 | [diff] [blame] | 285 | inline void write(const Constant *CPV) { printConstant(CPV); } |
Chris Lattner | 7db7958 | 2001-11-07 04:21:57 +0000 | [diff] [blame] | 286 | inline void write(const Type *Ty) { printType(Ty); } |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 287 | |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 288 | private : |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 289 | void printModule(const Module *M); |
| 290 | void printSymbolTable(const SymbolTable &ST); |
Chris Lattner | 3462ae3 | 2001-12-03 22:26:30 +0000 | [diff] [blame] | 291 | void printConstant(const Constant *CPV); |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 292 | void printGlobal(const GlobalVariable *GV); |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 293 | void printFunction(const Function *F); |
Chris Lattner | 2e9fa6d | 2002-04-09 19:48:49 +0000 | [diff] [blame] | 294 | void printArgument(const Argument *FA); |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 295 | void printBasicBlock(const BasicBlock *BB); |
| 296 | void printInstruction(const Instruction *I); |
| 297 | ostream &printType(const Type *Ty); |
| 298 | |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 299 | void writeOperand(const Value *Op, bool PrintType, bool PrintName = true); |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 300 | |
| 301 | // printInfoComment - Print a little comment after the instruction indicating |
| 302 | // which slot it occupies. |
| 303 | void printInfoComment(const Value *V); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 304 | }; |
| 305 | |
| 306 | |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 307 | void AssemblyWriter::writeOperand(const Value *Operand, bool PrintType, |
| 308 | bool PrintName) { |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 309 | if (PrintType) { Out << " "; printType(Operand->getType()); } |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 310 | WriteAsOperandInternal(Out, Operand, PrintName, &Table); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 311 | } |
| 312 | |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 313 | |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 314 | void AssemblyWriter::printModule(const Module *M) { |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 315 | // Loop over the symbol table, emitting all named constants... |
| 316 | if (M->hasSymbolTable()) |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 317 | printSymbolTable(*M->getSymbolTable()); |
Chris Lattner | da97550 | 2001-09-10 07:58:01 +0000 | [diff] [blame] | 318 | |
| 319 | for_each(M->gbegin(), M->gend(), |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 320 | bind_obj(this, &AssemblyWriter::printGlobal)); |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 321 | |
Vikram S. Adve | 13ba19a | 2001-09-18 12:48:16 +0000 | [diff] [blame] | 322 | Out << "implementation\n"; |
| 323 | |
Chris Lattner | 6915f8f | 2002-04-07 22:49:37 +0000 | [diff] [blame] | 324 | // Output all of the functions... |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 325 | for_each(M->begin(), M->end(), bind_obj(this,&AssemblyWriter::printFunction)); |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 326 | } |
| 327 | |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 328 | void AssemblyWriter::printGlobal(const GlobalVariable *GV) { |
Chris Lattner | da97550 | 2001-09-10 07:58:01 +0000 | [diff] [blame] | 329 | if (GV->hasName()) Out << "%" << GV->getName() << " = "; |
Chris Lattner | 3779864 | 2001-09-18 04:01:05 +0000 | [diff] [blame] | 330 | |
Chris Lattner | 841d8b9 | 2001-11-26 18:54:16 +0000 | [diff] [blame] | 331 | if (GV->hasInternalLinkage()) Out << "internal "; |
Chris Lattner | 3779864 | 2001-09-18 04:01:05 +0000 | [diff] [blame] | 332 | if (!GV->hasInitializer()) Out << "uninitialized "; |
| 333 | |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 334 | Out << (GV->isConstant() ? "constant " : "global "); |
Chris Lattner | 2413b16 | 2001-12-04 00:03:30 +0000 | [diff] [blame] | 335 | printType(GV->getType()->getElementType()); |
Chris Lattner | 3779864 | 2001-09-18 04:01:05 +0000 | [diff] [blame] | 336 | |
| 337 | if (GV->hasInitializer()) |
| 338 | writeOperand(GV->getInitializer(), false, false); |
| 339 | |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 340 | printInfoComment(GV); |
Chris Lattner | 7f74a56 | 2002-01-20 22:54:45 +0000 | [diff] [blame] | 341 | Out << "\n"; |
Chris Lattner | da97550 | 2001-09-10 07:58:01 +0000 | [diff] [blame] | 342 | } |
| 343 | |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 344 | |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 345 | // printSymbolTable - Run through symbol table looking for named constants |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 346 | // if a named constant is found, emit it's declaration... |
| 347 | // |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 348 | void AssemblyWriter::printSymbolTable(const SymbolTable &ST) { |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 349 | for (SymbolTable::const_iterator TI = ST.begin(); TI != ST.end(); ++TI) { |
| 350 | SymbolTable::type_const_iterator I = ST.type_begin(TI->first); |
| 351 | SymbolTable::type_const_iterator End = ST.type_end(TI->first); |
| 352 | |
| 353 | for (; I != End; ++I) { |
| 354 | const Value *V = I->second; |
Chris Lattner | 3462ae3 | 2001-12-03 22:26:30 +0000 | [diff] [blame] | 355 | if (const Constant *CPV = dyn_cast<const Constant>(V)) { |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 356 | printConstant(CPV); |
Chris Lattner | 3856934 | 2001-10-01 20:11:19 +0000 | [diff] [blame] | 357 | } else if (const Type *Ty = dyn_cast<const Type>(V)) { |
Chris Lattner | 7f74a56 | 2002-01-20 22:54:45 +0000 | [diff] [blame] | 358 | Out << "\t%" << I->first << " = type " << Ty->getDescription() << "\n"; |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 359 | } |
| 360 | } |
Chris Lattner | a7620d9 | 2001-07-15 06:35:59 +0000 | [diff] [blame] | 361 | } |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 362 | } |
| 363 | |
| 364 | |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 365 | // printConstant - Print out a constant pool entry... |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 366 | // |
Chris Lattner | 3462ae3 | 2001-12-03 22:26:30 +0000 | [diff] [blame] | 367 | void AssemblyWriter::printConstant(const Constant *CPV) { |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 368 | // Don't print out unnamed constants, they will be inlined |
| 369 | if (!CPV->hasName()) return; |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 370 | |
Chris Lattner | ee998be | 2001-07-26 16:29:38 +0000 | [diff] [blame] | 371 | // Print out name... |
| 372 | Out << "\t%" << CPV->getName() << " = "; |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 373 | |
Chris Lattner | ee998be | 2001-07-26 16:29:38 +0000 | [diff] [blame] | 374 | // Print out the constant type... |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 375 | printType(CPV->getType()); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 376 | |
| 377 | // Write the value out now... |
| 378 | writeOperand(CPV, false, false); |
| 379 | |
| 380 | if (!CPV->hasName() && CPV->getType() != Type::VoidTy) { |
| 381 | int Slot = Table.getValSlot(CPV); // Print out the def slot taken... |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 382 | Out << "\t\t; <"; |
| 383 | printType(CPV->getType()) << ">:"; |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 384 | if (Slot >= 0) Out << Slot; |
| 385 | else Out << "<badref>"; |
| 386 | } |
| 387 | |
Chris Lattner | 7f74a56 | 2002-01-20 22:54:45 +0000 | [diff] [blame] | 388 | Out << "\n"; |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 389 | } |
| 390 | |
Chris Lattner | 6915f8f | 2002-04-07 22:49:37 +0000 | [diff] [blame] | 391 | // printFunction - Print all aspects of a function. |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 392 | // |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 393 | void AssemblyWriter::printFunction(const Function *M) { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 394 | // Print out the return type and name... |
Chris Lattner | 841d8b9 | 2001-11-26 18:54:16 +0000 | [diff] [blame] | 395 | Out << "\n" << (M->isExternal() ? "declare " : "") |
| 396 | << (M->hasInternalLinkage() ? "internal " : ""); |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 397 | printType(M->getReturnType()) << " \"" << M->getName() << "\"("; |
Chris Lattner | 6915f8f | 2002-04-07 22:49:37 +0000 | [diff] [blame] | 398 | Table.incorporateFunction(M); |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 399 | |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 400 | // Loop over the arguments, printing them... |
Chris Lattner | 91db582 | 2002-03-29 03:44:36 +0000 | [diff] [blame] | 401 | const FunctionType *MT = M->getFunctionType(); |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 402 | |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 403 | if (!M->isExternal()) { |
| 404 | for_each(M->getArgumentList().begin(), M->getArgumentList().end(), |
Chris Lattner | 2e9fa6d | 2002-04-09 19:48:49 +0000 | [diff] [blame] | 405 | bind_obj(this, &AssemblyWriter::printArgument)); |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 406 | } else { |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 407 | // Loop over the arguments, printing them... |
Chris Lattner | 91db582 | 2002-03-29 03:44:36 +0000 | [diff] [blame] | 408 | const FunctionType *MT = M->getFunctionType(); |
| 409 | for (FunctionType::ParamTypes::const_iterator I = MT->getParamTypes().begin(), |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 410 | E = MT->getParamTypes().end(); I != E; ++I) { |
| 411 | if (I != MT->getParamTypes().begin()) Out << ", "; |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 412 | printType(*I); |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 413 | } |
| 414 | } |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 415 | |
| 416 | // Finish printing arguments... |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 417 | if (MT->isVarArg()) { |
| 418 | if (MT->getParamTypes().size()) Out << ", "; |
| 419 | Out << "..."; // Output varargs portion of signature! |
| 420 | } |
| 421 | Out << ")\n"; |
| 422 | |
| 423 | if (!M->isExternal()) { |
| 424 | // Loop over the symbol table, emitting all named constants... |
| 425 | if (M->hasSymbolTable()) |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 426 | printSymbolTable(*M->getSymbolTable()); |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 427 | |
| 428 | Out << "begin"; |
| 429 | |
Chris Lattner | 6915f8f | 2002-04-07 22:49:37 +0000 | [diff] [blame] | 430 | // Output all of its basic blocks... for the function |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 431 | for_each(M->begin(), M->end(), |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 432 | bind_obj(this, &AssemblyWriter::printBasicBlock)); |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 433 | |
Chris Lattner | a7620d9 | 2001-07-15 06:35:59 +0000 | [diff] [blame] | 434 | Out << "end\n"; |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 435 | } |
| 436 | |
Chris Lattner | 6915f8f | 2002-04-07 22:49:37 +0000 | [diff] [blame] | 437 | Table.purgeFunction(); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 438 | } |
| 439 | |
Chris Lattner | 2e9fa6d | 2002-04-09 19:48:49 +0000 | [diff] [blame] | 440 | // printArgument - This member is called for every argument that |
Chris Lattner | 6915f8f | 2002-04-07 22:49:37 +0000 | [diff] [blame] | 441 | // is passed into the function. Simply print it out |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 442 | // |
Chris Lattner | 2e9fa6d | 2002-04-09 19:48:49 +0000 | [diff] [blame] | 443 | void AssemblyWriter::printArgument(const Argument *Arg) { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 444 | // Insert commas as we go... the first arg doesn't get a comma |
| 445 | if (Arg != Arg->getParent()->getArgumentList().front()) Out << ", "; |
| 446 | |
| 447 | // Output type... |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 448 | printType(Arg->getType()); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 449 | |
| 450 | // Output name, if available... |
| 451 | if (Arg->hasName()) |
| 452 | Out << " %" << Arg->getName(); |
| 453 | else if (Table.getValSlot(Arg) < 0) |
| 454 | Out << "<badref>"; |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 455 | } |
| 456 | |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 457 | // printBasicBlock - This member is called for each basic block in a methd. |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 458 | // |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 459 | void AssemblyWriter::printBasicBlock(const BasicBlock *BB) { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 460 | if (BB->hasName()) { // Print out the label if it exists... |
Chris Lattner | a2f0187 | 2001-06-07 16:58:55 +0000 | [diff] [blame] | 461 | Out << "\n" << BB->getName() << ":"; |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 462 | } else { |
| 463 | int Slot = Table.getValSlot(BB); |
Chris Lattner | a2f0187 | 2001-06-07 16:58:55 +0000 | [diff] [blame] | 464 | Out << "\n; <label>:"; |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 465 | if (Slot >= 0) |
Chris Lattner | a2f0187 | 2001-06-07 16:58:55 +0000 | [diff] [blame] | 466 | Out << Slot; // Extra newline seperates out label's |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 467 | else |
Chris Lattner | a2f0187 | 2001-06-07 16:58:55 +0000 | [diff] [blame] | 468 | Out << "<badref>"; |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 469 | } |
Chris Lattner | a2f0187 | 2001-06-07 16:58:55 +0000 | [diff] [blame] | 470 | Out << "\t\t\t\t\t;[#uses=" << BB->use_size() << "]\n"; // Output # uses |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 471 | |
Chris Lattner | fee714f | 2001-09-07 16:36:04 +0000 | [diff] [blame] | 472 | // Output all of the instructions in the basic block... |
| 473 | for_each(BB->begin(), BB->end(), |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 474 | bind_obj(this, &AssemblyWriter::printInstruction)); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 475 | } |
| 476 | |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 477 | |
| 478 | // printInfoComment - Print a little comment after the instruction indicating |
| 479 | // which slot it occupies. |
| 480 | // |
| 481 | void AssemblyWriter::printInfoComment(const Value *V) { |
| 482 | if (V->getType() != Type::VoidTy) { |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 483 | Out << "\t\t; <"; |
| 484 | printType(V->getType()) << ">"; |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 485 | |
| 486 | if (!V->hasName()) { |
| 487 | int Slot = Table.getValSlot(V); // Print out the def slot taken... |
| 488 | if (Slot >= 0) Out << ":" << Slot; |
| 489 | else Out << ":<badref>"; |
| 490 | } |
Chris Lattner | 3cb3a1f | 2001-12-14 16:29:12 +0000 | [diff] [blame] | 491 | Out << " [#uses=" << V->use_size() << "]"; // Output # uses |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 492 | } |
| 493 | } |
| 494 | |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 495 | // printInstruction - This member is called for each Instruction in a methd. |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 496 | // |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 497 | void AssemblyWriter::printInstruction(const Instruction *I) { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 498 | Out << "\t"; |
| 499 | |
| 500 | // Print out name if it exists... |
| 501 | if (I && I->hasName()) |
| 502 | Out << "%" << I->getName() << " = "; |
| 503 | |
| 504 | // Print out the opcode... |
Chris Lattner | b1ca9cb | 2001-07-07 19:24:15 +0000 | [diff] [blame] | 505 | Out << I->getOpcodeName(); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 506 | |
| 507 | // Print out the type of the operands... |
Chris Lattner | a073acb | 2001-07-07 08:36:50 +0000 | [diff] [blame] | 508 | const Value *Operand = I->getNumOperands() ? I->getOperand(0) : 0; |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 509 | |
| 510 | // Special case conditional branches to swizzle the condition out to the front |
Chris Lattner | 8d48df2 | 2002-04-13 18:34:38 +0000 | [diff] [blame] | 511 | if (isa<BranchInst>(I) && I->getNumOperands() > 1) { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 512 | writeOperand(I->getOperand(2), true); |
| 513 | Out << ","; |
| 514 | writeOperand(Operand, true); |
| 515 | Out << ","; |
| 516 | writeOperand(I->getOperand(1), true); |
| 517 | |
Chris Lattner | 8d48df2 | 2002-04-13 18:34:38 +0000 | [diff] [blame] | 518 | } else if (isa<SwitchInst>(I)) { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 519 | // Special case switch statement to get formatting nice and correct... |
| 520 | writeOperand(Operand , true); Out << ","; |
| 521 | writeOperand(I->getOperand(1), true); Out << " ["; |
| 522 | |
Chris Lattner | a073acb | 2001-07-07 08:36:50 +0000 | [diff] [blame] | 523 | for (unsigned op = 2, Eop = I->getNumOperands(); op < Eop; op += 2) { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 524 | Out << "\n\t\t"; |
Chris Lattner | a073acb | 2001-07-07 08:36:50 +0000 | [diff] [blame] | 525 | writeOperand(I->getOperand(op ), true); Out << ","; |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 526 | writeOperand(I->getOperand(op+1), true); |
| 527 | } |
| 528 | Out << "\n\t]"; |
Chris Lattner | da55810 | 2001-10-02 03:41:24 +0000 | [diff] [blame] | 529 | } else if (isa<PHINode>(I)) { |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 530 | Out << " "; |
Chris Lattner | 29644b3 | 2001-11-06 01:48:45 +0000 | [diff] [blame] | 531 | printType(I->getType()); |
Chris Lattner | 9d3292d | 2001-11-06 08:33:46 +0000 | [diff] [blame] | 532 | Out << " "; |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 533 | |
Chris Lattner | 29644b3 | 2001-11-06 01:48:45 +0000 | [diff] [blame] | 534 | for (unsigned op = 0, Eop = I->getNumOperands(); op < Eop; op += 2) { |
| 535 | if (op) Out << ", "; |
| 536 | Out << "["; |
Chris Lattner | a073acb | 2001-07-07 08:36:50 +0000 | [diff] [blame] | 537 | writeOperand(I->getOperand(op ), false); Out << ","; |
Chris Lattner | 4b94e23 | 2001-06-21 05:29:56 +0000 | [diff] [blame] | 538 | writeOperand(I->getOperand(op+1), false); Out << " ]"; |
Chris Lattner | 931ef3b | 2001-06-11 15:04:20 +0000 | [diff] [blame] | 539 | } |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 540 | } else if (isa<ReturnInst>(I) && !Operand) { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 541 | Out << " void"; |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 542 | } else if (isa<CallInst>(I)) { |
Chris Lattner | 2f2d947 | 2001-11-06 21:28:12 +0000 | [diff] [blame] | 543 | const PointerType *PTy = dyn_cast<PointerType>(Operand->getType()); |
Chris Lattner | 91db582 | 2002-03-29 03:44:36 +0000 | [diff] [blame] | 544 | const FunctionType*MTy = PTy ? dyn_cast<FunctionType>(PTy->getElementType()):0; |
Chris Lattner | 2f2d947 | 2001-11-06 21:28:12 +0000 | [diff] [blame] | 545 | const Type *RetTy = MTy ? MTy->getReturnType() : 0; |
| 546 | |
| 547 | // If possible, print out the short form of the call instruction, but we can |
Chris Lattner | 6915f8f | 2002-04-07 22:49:37 +0000 | [diff] [blame] | 548 | // only do this if the first argument is a pointer to a nonvararg function, |
| 549 | // and if the value returned is not a pointer to a function. |
Chris Lattner | 2f2d947 | 2001-11-06 21:28:12 +0000 | [diff] [blame] | 550 | // |
Chris Lattner | 8d48df2 | 2002-04-13 18:34:38 +0000 | [diff] [blame] | 551 | if (RetTy && MTy && !MTy->isVarArg() && |
| 552 | (!isa<PointerType>(RetTy) || |
| 553 | !isa<FunctionType>(cast<PointerType>(RetTy)))) { |
Chris Lattner | 2f2d947 | 2001-11-06 21:28:12 +0000 | [diff] [blame] | 554 | Out << " "; printType(RetTy); |
| 555 | writeOperand(Operand, false); |
| 556 | } else { |
| 557 | writeOperand(Operand, true); |
| 558 | } |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 559 | Out << "("; |
Chris Lattner | a073acb | 2001-07-07 08:36:50 +0000 | [diff] [blame] | 560 | if (I->getNumOperands() > 1) writeOperand(I->getOperand(1), true); |
| 561 | for (unsigned op = 2, Eop = I->getNumOperands(); op < Eop; ++op) { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 562 | Out << ","; |
Chris Lattner | a073acb | 2001-07-07 08:36:50 +0000 | [diff] [blame] | 563 | writeOperand(I->getOperand(op), true); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 564 | } |
| 565 | |
| 566 | Out << " )"; |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 567 | } else if (const InvokeInst *II = dyn_cast<InvokeInst>(I)) { |
| 568 | // TODO: Should try to print out short form of the Invoke instruction |
| 569 | writeOperand(Operand, true); |
| 570 | Out << "("; |
| 571 | if (I->getNumOperands() > 3) writeOperand(I->getOperand(3), true); |
| 572 | for (unsigned op = 4, Eop = I->getNumOperands(); op < Eop; ++op) { |
| 573 | Out << ","; |
| 574 | writeOperand(I->getOperand(op), true); |
| 575 | } |
| 576 | |
| 577 | Out << " )\n\t\t\tto"; |
| 578 | writeOperand(II->getNormalDest(), true); |
| 579 | Out << " except"; |
| 580 | writeOperand(II->getExceptionalDest(), true); |
| 581 | |
Chris Lattner | 8d48df2 | 2002-04-13 18:34:38 +0000 | [diff] [blame] | 582 | } else if (const AllocationInst *AI = dyn_cast<AllocationInst>(I)) { |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 583 | Out << " "; |
Chris Lattner | 8d48df2 | 2002-04-13 18:34:38 +0000 | [diff] [blame] | 584 | printType(AI->getType()->getElementType()); |
| 585 | if (AI->isArrayAllocation()) { |
Chris Lattner | a073acb | 2001-07-07 08:36:50 +0000 | [diff] [blame] | 586 | Out << ","; |
Chris Lattner | 8d48df2 | 2002-04-13 18:34:38 +0000 | [diff] [blame] | 587 | writeOperand(AI->getArraySize(), true); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 588 | } |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 589 | } else if (isa<CastInst>(I)) { |
Chris Lattner | a682182 | 2001-07-08 04:57:15 +0000 | [diff] [blame] | 590 | writeOperand(Operand, true); |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 591 | Out << " to "; |
| 592 | printType(I->getType()); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 593 | } else if (Operand) { // Print the normal way... |
| 594 | |
| 595 | // PrintAllTypes - Instructions who have operands of all the same type |
| 596 | // omit the type from all but the first operand. If the instruction has |
| 597 | // different type operands (for example br), then they are all printed. |
| 598 | bool PrintAllTypes = false; |
| 599 | const Type *TheType = Operand->getType(); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 600 | |
Chris Lattner | a073acb | 2001-07-07 08:36:50 +0000 | [diff] [blame] | 601 | for (unsigned i = 1, E = I->getNumOperands(); i != E; ++i) { |
| 602 | Operand = I->getOperand(i); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 603 | if (Operand->getType() != TheType) { |
| 604 | PrintAllTypes = true; // We have differing types! Print them all! |
| 605 | break; |
| 606 | } |
| 607 | } |
| 608 | |
Chris Lattner | b6fe234 | 2001-10-20 09:33:10 +0000 | [diff] [blame] | 609 | // Shift Left & Right print both types even for Ubyte LHS |
| 610 | if (isa<ShiftInst>(I)) PrintAllTypes = true; |
| 611 | |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 612 | if (!PrintAllTypes) { |
| 613 | Out << " "; |
| 614 | printType(I->getOperand(0)->getType()); |
| 615 | } |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 616 | |
Chris Lattner | a073acb | 2001-07-07 08:36:50 +0000 | [diff] [blame] | 617 | for (unsigned i = 0, E = I->getNumOperands(); i != E; ++i) { |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 618 | if (i) Out << ","; |
Chris Lattner | a073acb | 2001-07-07 08:36:50 +0000 | [diff] [blame] | 619 | writeOperand(I->getOperand(i), PrintAllTypes); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 620 | } |
| 621 | } |
| 622 | |
Chris Lattner | 862e338 | 2001-10-13 06:42:36 +0000 | [diff] [blame] | 623 | printInfoComment(I); |
Chris Lattner | 7f74a56 | 2002-01-20 22:54:45 +0000 | [diff] [blame] | 624 | Out << "\n"; |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 625 | } |
| 626 | |
| 627 | |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 628 | // printType - Go to extreme measures to attempt to print out a short, symbolic |
| 629 | // version of a type name. |
| 630 | // |
| 631 | ostream &AssemblyWriter::printType(const Type *Ty) { |
Chris Lattner | b86620e | 2001-10-29 16:37:48 +0000 | [diff] [blame] | 632 | return printTypeInt(Out, Ty, TypeNames); |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 633 | } |
| 634 | |
| 635 | |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 636 | //===----------------------------------------------------------------------===// |
| 637 | // External Interface declarations |
| 638 | //===----------------------------------------------------------------------===// |
| 639 | |
| 640 | |
Chris Lattner | c0b4c7b | 2002-04-08 22:03:40 +0000 | [diff] [blame] | 641 | void Module::print(std::ostream &o) const { |
| 642 | SlotCalculator SlotTable(this, true); |
| 643 | AssemblyWriter W(o, SlotTable, this); |
| 644 | W.write(this); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 645 | } |
| 646 | |
Chris Lattner | c0b4c7b | 2002-04-08 22:03:40 +0000 | [diff] [blame] | 647 | void GlobalVariable::print(std::ostream &o) const { |
| 648 | SlotCalculator SlotTable(getParent(), true); |
| 649 | AssemblyWriter W(o, SlotTable, getParent()); |
| 650 | W.write(this); |
Chris Lattner | adfe0d1 | 2001-09-10 20:08:19 +0000 | [diff] [blame] | 651 | } |
| 652 | |
Chris Lattner | c0b4c7b | 2002-04-08 22:03:40 +0000 | [diff] [blame] | 653 | void Function::print(std::ostream &o) const { |
| 654 | SlotCalculator SlotTable(getParent(), true); |
| 655 | AssemblyWriter W(o, SlotTable, getParent()); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 656 | |
Chris Lattner | c0b4c7b | 2002-04-08 22:03:40 +0000 | [diff] [blame] | 657 | W.write(this); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 658 | } |
| 659 | |
Chris Lattner | c0b4c7b | 2002-04-08 22:03:40 +0000 | [diff] [blame] | 660 | void BasicBlock::print(std::ostream &o) const { |
| 661 | SlotCalculator SlotTable(getParent(), true); |
Chris Lattner | 7bfee41 | 2001-10-29 16:05:51 +0000 | [diff] [blame] | 662 | AssemblyWriter W(o, SlotTable, |
Chris Lattner | c0b4c7b | 2002-04-08 22:03:40 +0000 | [diff] [blame] | 663 | getParent() ? getParent()->getParent() : 0); |
| 664 | W.write(this); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 665 | } |
| 666 | |
Chris Lattner | c0b4c7b | 2002-04-08 22:03:40 +0000 | [diff] [blame] | 667 | void Instruction::print(std::ostream &o) const { |
| 668 | const Function *F = getParent() ? getParent()->getParent() : 0; |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 669 | SlotCalculator SlotTable(F, true); |
| 670 | AssemblyWriter W(o, SlotTable, F ? F->getParent() : 0); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 671 | |
Chris Lattner | c0b4c7b | 2002-04-08 22:03:40 +0000 | [diff] [blame] | 672 | W.write(this); |
Chris Lattner | 2f7c963 | 2001-06-06 20:29:01 +0000 | [diff] [blame] | 673 | } |
Chris Lattner | 7db7958 | 2001-11-07 04:21:57 +0000 | [diff] [blame] | 674 | |
Chris Lattner | c0b4c7b | 2002-04-08 22:03:40 +0000 | [diff] [blame] | 675 | void Constant::print(std::ostream &o) const { |
| 676 | if (this == 0) { o << "<null> constant value\n"; return; } |
| 677 | o << " " << getType()->getDescription() << " " << getStrValue(); |
| 678 | } |
| 679 | |
| 680 | void Type::print(std::ostream &o) const { |
| 681 | if (this == 0) |
| 682 | o << "<null Type>"; |
| 683 | else |
| 684 | o << getDescription(); |
| 685 | } |
| 686 | |
Chris Lattner | 2e9fa6d | 2002-04-09 19:48:49 +0000 | [diff] [blame] | 687 | void Argument::print(std::ostream &o) const { |
Chris Lattner | c0b4c7b | 2002-04-08 22:03:40 +0000 | [diff] [blame] | 688 | o << getType() << " " << getName(); |
| 689 | } |
| 690 | |
| 691 | void Value::dump() const { print(std::cerr); } |
| 692 | |
| 693 | //===----------------------------------------------------------------------===// |
| 694 | // CachedWriter Class Implementation |
| 695 | //===----------------------------------------------------------------------===// |
| 696 | |
Chris Lattner | 7db7958 | 2001-11-07 04:21:57 +0000 | [diff] [blame] | 697 | void CachedWriter::setModule(const Module *M) { |
| 698 | delete SC; delete AW; |
| 699 | if (M) { |
| 700 | SC = new SlotCalculator(M, true); |
| 701 | AW = new AssemblyWriter(Out, *SC, M); |
| 702 | } else { |
| 703 | SC = 0; AW = 0; |
| 704 | } |
| 705 | } |
| 706 | |
| 707 | CachedWriter::~CachedWriter() { |
| 708 | delete AW; |
| 709 | delete SC; |
| 710 | } |
| 711 | |
| 712 | CachedWriter &CachedWriter::operator<<(const Value *V) { |
| 713 | assert(AW && SC && "CachedWriter does not have a current module!"); |
| 714 | switch (V->getValueType()) { |
| 715 | case Value::ConstantVal: |
| 716 | Out << " "; AW->write(V->getType()); |
Chris Lattner | 3462ae3 | 2001-12-03 22:26:30 +0000 | [diff] [blame] | 717 | Out << " " << cast<Constant>(V)->getStrValue(); break; |
Chris Lattner | 2e9fa6d | 2002-04-09 19:48:49 +0000 | [diff] [blame] | 718 | case Value::ArgumentVal: |
Chris Lattner | 7db7958 | 2001-11-07 04:21:57 +0000 | [diff] [blame] | 719 | AW->write(V->getType()); Out << " " << V->getName(); break; |
| 720 | case Value::TypeVal: AW->write(cast<const Type>(V)); break; |
| 721 | case Value::InstructionVal: AW->write(cast<Instruction>(V)); break; |
| 722 | case Value::BasicBlockVal: AW->write(cast<BasicBlock>(V)); break; |
Chris Lattner | 57698e2 | 2002-03-26 18:01:55 +0000 | [diff] [blame] | 723 | case Value::FunctionVal: AW->write(cast<Function>(V)); break; |
Chris Lattner | 7db7958 | 2001-11-07 04:21:57 +0000 | [diff] [blame] | 724 | case Value::GlobalVariableVal: AW->write(cast<GlobalVariable>(V)); break; |
| 725 | case Value::ModuleVal: AW->write(cast<Module>(V)); break; |
| 726 | default: Out << "<unknown value type: " << V->getValueType() << ">"; break; |
| 727 | } |
| 728 | return *this; |
| 729 | } |