AsmWriter.cpp revision f7a551fd929a4a849883cde04cfe54e0590a38b7
1009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner//===-- Writer.cpp - Library for Printing VM assembly files ------*- C++ -*--=// 2009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// 3009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// This library implements the functionality defined in llvm/Assembly/Writer.h 4009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// 5009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// This library uses the Analysis library to figure out offsets for 6009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// variables in the method tables... 7009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// 8009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// TODO: print out the type name instead of the full type if a particular type 9009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// is in the symbol table... 10009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// 11009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner//===----------------------------------------------------------------------===// 12009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 13da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner#include "llvm/Assembly/CachedWriter.h" 14009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner#include "llvm/Analysis/SlotCalculator.h" 15009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner#include "llvm/Module.h" 16009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner#include "llvm/Method.h" 1770cc3397f84c2e1fd69c059a0ef89e398e847b00Chris Lattner#include "llvm/GlobalVariable.h" 18009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner#include "llvm/BasicBlock.h" 19e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner#include "llvm/ConstantVals.h" 20009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner#include "llvm/iMemory.h" 21e02fa8551d20081534afa46e0976811687e5183aChris Lattner#include "llvm/iTerminators.h" 227061dc50b2513731d7b346ab16183cda4a44619fChris Lattner#include "llvm/iPHINode.h" 237061dc50b2513731d7b346ab16183cda4a44619fChris Lattner#include "llvm/iOther.h" 24007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner#include "llvm/SymbolTable.h" 25cee8f9ae67104576b2028125b56e9ba4856a1d66Chris Lattner#include "Support/StringExtras.h" 26cee8f9ae67104576b2028125b56e9ba4856a1d66Chris Lattner#include "Support/STLExtras.h" 27007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner#include <algorithm> 28c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner#include <map> 29c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner 30207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattnerstatic const Module *getModuleFromVal(const Value *V) { 31207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (const MethodArgument *MA =dyn_cast<const MethodArgument>(V)) 32207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner return MA->getParent() ? MA->getParent()->getParent() : 0; 33207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner else if (const BasicBlock *BB = dyn_cast<const BasicBlock>(V)) 34207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner return BB->getParent() ? BB->getParent()->getParent() : 0; 35207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner else if (const Instruction *I = dyn_cast<const Instruction>(V)) { 36207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner const Method *M = I->getParent() ? I->getParent()->getParent() : 0; 37207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner return M ? M->getParent() : 0; 38207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner } else if (const GlobalValue *GV =dyn_cast<const GlobalValue>(V)) 39207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner return GV->getParent(); 40207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner else if (const Module *Mod = dyn_cast<const Module>(V)) 41207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner return Mod; 42207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner return 0; 43207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner} 44207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 45c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattnerstatic SlotCalculator *createSlotCalculator(const Value *V) { 46c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner assert(!isa<Type>(V) && "Can't create an SC for a type!"); 47c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner if (const MethodArgument *MA =dyn_cast<const MethodArgument>(V)){ 48c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner return new SlotCalculator(MA->getParent(), true); 49c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner } else if (const Instruction *I = dyn_cast<const Instruction>(V)) { 50c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner return new SlotCalculator(I->getParent()->getParent(), true); 51c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner } else if (const BasicBlock *BB = dyn_cast<const BasicBlock>(V)) { 52c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner return new SlotCalculator(BB->getParent(), true); 53c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner } else if (const GlobalVariable *GV =dyn_cast<const GlobalVariable>(V)){ 54c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner return new SlotCalculator(GV->getParent(), true); 55c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner } else if (const Method *Meth = dyn_cast<const Method>(V)) { 56c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner return new SlotCalculator(Meth, true); 57c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner } else if (const Module *Mod = dyn_cast<const Module>(V)) { 58c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner return new SlotCalculator(Mod, true); 59c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner } 60c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner return 0; 61c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner} 62009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 63622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner// WriteAsOperand - Write the name of the specified value out to the specified 64622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner// ostream. This can be useful when you just want to print int %reg126, not the 65622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner// whole instruction that generated it. 66622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner// 67207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattnerstatic void WriteAsOperandInternal(ostream &Out, const Value *V, bool PrintName, 68207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner SlotCalculator *Table) { 69007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner if (PrintName && V->hasName()) { 70622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner Out << " %" << V->getName(); 71622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner } else { 72e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner if (const Constant *CPV = dyn_cast<const Constant>(V)) { 73622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner Out << " " << CPV->getStrValue(); 74622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner } else { 75622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner int Slot; 76622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner if (Table) { 77622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner Slot = Table->getValSlot(V); 78622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner } else { 79207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (const Type *Ty = dyn_cast<const Type>(V)) { 80207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner Out << " " << Ty->getDescription(); 81207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner return; 82207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner } 83c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner 84c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner Table = createSlotCalculator(V); 85207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (Table == 0) { Out << "BAD VALUE TYPE!"; return; } 86c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner 87622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner Slot = Table->getValSlot(V); 88622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner delete Table; 89622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner } 90622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner if (Slot >= 0) Out << " %" << Slot; 91622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner else if (PrintName) 92622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner Out << "<badref>"; // Not embeded into a location? 93622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner } 94622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner } 95207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner} 96207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 97207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 98207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner// If the module has a symbol table, take all global types and stuff their 99207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner// names into the TypeNames map. 100207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner// 101207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattnerstatic void fillTypeNameTable(const Module *M, 102207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner map<const Type *, string> &TypeNames) { 103207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (M && M->hasSymbolTable()) { 104207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner const SymbolTable *ST = M->getSymbolTable(); 105207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner SymbolTable::const_iterator PI = ST->find(Type::TypeTy); 106207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (PI != ST->end()) { 107207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner SymbolTable::type_const_iterator I = PI->second.begin(); 108207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner for (; I != PI->second.end(); ++I) { 109207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // As a heuristic, don't insert pointer to primitive types, because 110207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // they are used too often to have a single useful name. 111207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // 112207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner const Type *Ty = cast<const Type>(I->second); 113207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (!isa<PointerType>(Ty) || 1147a1767520611d9ff6face702068de858e1cadf2cChris Lattner !cast<PointerType>(Ty)->getElementType()->isPrimitiveType()) 115207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner TypeNames.insert(make_pair(Ty, "%"+I->first)); 116207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner } 117207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner } 118207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner } 119207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner} 120207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 121207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 122207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 123207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattnerstatic string calcTypeName(const Type *Ty, vector<const Type *> &TypeStack, 124207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner map<const Type *, string> &TypeNames) { 125207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (Ty->isPrimitiveType()) return Ty->getDescription(); // Base case 126207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 127207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // Check to see if the type is named. 128207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner map<const Type *, string>::iterator I = TypeNames.find(Ty); 129207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (I != TypeNames.end()) return I->second; 130207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 131207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // Check to see if the Type is already on the stack... 132207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner unsigned Slot = 0, CurSize = TypeStack.size(); 133207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner while (Slot < CurSize && TypeStack[Slot] != Ty) ++Slot; // Scan for type 134207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 135207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // This is another base case for the recursion. In this case, we know 136207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // that we have looped back to a type that we have previously visited. 137207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // Generate the appropriate upreference to handle this. 138207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // 139207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (Slot < CurSize) 140207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner return "\\" + utostr(CurSize-Slot); // Here's the upreference 141207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 142207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner TypeStack.push_back(Ty); // Recursive case: Add us to the stack.. 143207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 144207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner string Result; 145207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner switch (Ty->getPrimitiveID()) { 146207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner case Type::MethodTyID: { 147207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner const MethodType *MTy = cast<const MethodType>(Ty); 148207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner Result = calcTypeName(MTy->getReturnType(), TypeStack, TypeNames) + " ("; 149207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner for (MethodType::ParamTypes::const_iterator 150207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner I = MTy->getParamTypes().begin(), 151207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner E = MTy->getParamTypes().end(); I != E; ++I) { 152207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (I != MTy->getParamTypes().begin()) 153207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner Result += ", "; 154207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner Result += calcTypeName(*I, TypeStack, TypeNames); 155207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner } 156207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (MTy->isVarArg()) { 157207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (!MTy->getParamTypes().empty()) Result += ", "; 158207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner Result += "..."; 159207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner } 160207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner Result += ")"; 161207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner break; 162207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner } 163207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner case Type::StructTyID: { 164207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner const StructType *STy = cast<const StructType>(Ty); 165207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner Result = "{ "; 166207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner for (StructType::ElementTypes::const_iterator 167207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner I = STy->getElementTypes().begin(), 168207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner E = STy->getElementTypes().end(); I != E; ++I) { 169207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (I != STy->getElementTypes().begin()) 170207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner Result += ", "; 171207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner Result += calcTypeName(*I, TypeStack, TypeNames); 172207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner } 173207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner Result += " }"; 174207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner break; 175207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner } 176207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner case Type::PointerTyID: 1777a1767520611d9ff6face702068de858e1cadf2cChris Lattner Result = calcTypeName(cast<const PointerType>(Ty)->getElementType(), 178207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner TypeStack, TypeNames) + " *"; 179207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner break; 180207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner case Type::ArrayTyID: { 181207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner const ArrayType *ATy = cast<const ArrayType>(Ty); 182207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner int NumElements = ATy->getNumElements(); 183207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner Result = "["; 184207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (NumElements != -1) Result += itostr(NumElements) + " x "; 185207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner Result += calcTypeName(ATy->getElementType(), TypeStack, TypeNames) + "]"; 186207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner break; 187207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner } 188207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner default: 189207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner assert(0 && "Unhandled case in getTypeProps!"); 190207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner Result = "<error>"; 191207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner } 192207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 193207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner TypeStack.pop_back(); // Remove self from stack... 194207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner return Result; 195207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner} 196207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 197207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 198207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner// printTypeInt - The internal guts of printing out a type that has a 199207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner// potentially named portion. 200207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner// 201207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattnerstatic ostream &printTypeInt(ostream &Out, const Type *Ty, 202207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner map<const Type *, string> &TypeNames) { 203207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // Primitive types always print out their description, regardless of whether 204207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // they have been named or not. 205207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // 206207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (Ty->isPrimitiveType()) return Out << Ty->getDescription(); 207207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 208207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // Check to see if the type is named. 209207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner map<const Type *, string>::iterator I = TypeNames.find(Ty); 210207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (I != TypeNames.end()) return Out << I->second; 211207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 212207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // Otherwise we have a type that has not been named but is a derived type. 213207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // Carefully recurse the type hierarchy to print out any contained symbolic 214207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // names. 215207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // 216207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner vector<const Type *> TypeStack; 217207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner string TypeName = calcTypeName(Ty, TypeStack, TypeNames); 218207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner TypeNames.insert(make_pair(Ty, TypeName)); // Cache type name for later use 219207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner return Out << TypeName; 220207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner} 221207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 222e51e03b3c649ed9419bd0e920c03ef9023ccee48Chris Lattner 223207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner// WriteTypeSymbolic - This attempts to write the specified type as a symbolic 224207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner// type, iff there is an entry in the modules symbol table for the specified 225207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner// type or one of it's component types. This is slower than a simple x << Type; 226207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner// 227207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattnerostream &WriteTypeSymbolic(ostream &Out, const Type *Ty, const Module *M) { 228207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner Out << " "; 229207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 230207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // If they want us to print out a type, attempt to make it symbolic if there 231207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner // is a symbol table in the module... 232207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner if (M && M->hasSymbolTable()) { 233207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner map<const Type *, string> TypeNames; 234207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner fillTypeNameTable(M, TypeNames); 235207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 2367b8660d72f35f5ddea0c81eb71f2bdd60fd62832Chris Lattner return printTypeInt(Out, Ty, TypeNames); 237207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner } else { 2387b8660d72f35f5ddea0c81eb71f2bdd60fd62832Chris Lattner return Out << Ty->getDescription(); 239207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner } 240207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner} 241207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 242207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 243207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner// WriteAsOperand - Write the name of the specified value out to the specified 244207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner// ostream. This can be useful when you just want to print int %reg126, not the 245207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner// whole instruction that generated it. 246207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner// 247207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattnerostream &WriteAsOperand(ostream &Out, const Value *V, bool PrintType, 248207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner bool PrintName, SlotCalculator *Table) { 2497b8660d72f35f5ddea0c81eb71f2bdd60fd62832Chris Lattner if (PrintType) 2507b8660d72f35f5ddea0c81eb71f2bdd60fd62832Chris Lattner WriteTypeSymbolic(Out, V->getType(), getModuleFromVal(V)); 251207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner 252207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner WriteAsOperandInternal(Out, V, PrintName, Table); 253622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner return Out; 254622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner} 255622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner 256622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner 257d8c2e42aeff8bdb3ac905b4721b3d3ca1f904cfaChris Lattner 258007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattnerclass AssemblyWriter { 259009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner ostream &Out; 260009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner SlotCalculator &Table; 261c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner const Module *TheModule; 262c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner map<const Type *, string> TypeNames; 263009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattnerpublic: 264c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner inline AssemblyWriter(ostream &o, SlotCalculator &Tab, const Module *M) 265c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner : Out(o), Table(Tab), TheModule(M) { 266c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner 267c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner // If the module has a symbol table, take all global types and stuff their 268c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner // names into the TypeNames map. 269c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner // 270207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner fillTypeNameTable(M, TypeNames); 271009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner } 272009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 273c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner inline void write(const Module *M) { printModule(M); } 274c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner inline void write(const GlobalVariable *G) { printGlobal(G); } 275c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner inline void write(const Method *M) { printMethod(M); } 276c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner inline void write(const BasicBlock *BB) { printBasicBlock(BB); } 277c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner inline void write(const Instruction *I) { printInstruction(I); } 278e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner inline void write(const Constant *CPV) { printConstant(CPV); } 279da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner inline void write(const Type *Ty) { printType(Ty); } 280009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 281009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattnerprivate : 282c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner void printModule(const Module *M); 283c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner void printSymbolTable(const SymbolTable &ST); 284e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner void printConstant(const Constant *CPV); 285c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner void printGlobal(const GlobalVariable *GV); 286c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner void printMethod(const Method *M); 287c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner void printMethodArgument(const MethodArgument *MA); 288c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner void printBasicBlock(const BasicBlock *BB); 289c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner void printInstruction(const Instruction *I); 290c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner ostream &printType(const Type *Ty); 291c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner 292009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner void writeOperand(const Value *Op, bool PrintType, bool PrintName = true); 293e02fa8551d20081534afa46e0976811687e5183aChris Lattner 294e02fa8551d20081534afa46e0976811687e5183aChris Lattner // printInfoComment - Print a little comment after the instruction indicating 295e02fa8551d20081534afa46e0976811687e5183aChris Lattner // which slot it occupies. 296e02fa8551d20081534afa46e0976811687e5183aChris Lattner void printInfoComment(const Value *V); 297009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner}; 298009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 299009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 300007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattnervoid AssemblyWriter::writeOperand(const Value *Operand, bool PrintType, 301007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner bool PrintName) { 302c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner if (PrintType) { Out << " "; printType(Operand->getType()); } 303207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner WriteAsOperandInternal(Out, Operand, PrintName, &Table); 304007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner} 305007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner 306007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner 307c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattnervoid AssemblyWriter::printModule(const Module *M) { 308007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner // Loop over the symbol table, emitting all named constants... 309007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner if (M->hasSymbolTable()) 310c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner printSymbolTable(*M->getSymbolTable()); 31170cc3397f84c2e1fd69c059a0ef89e398e847b00Chris Lattner 31270cc3397f84c2e1fd69c059a0ef89e398e847b00Chris Lattner for_each(M->gbegin(), M->gend(), 313c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner bind_obj(this, &AssemblyWriter::printGlobal)); 314007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner 3155efa3ccbd17e81358f87474e53492871b1c83e41Vikram S. Adve Out << "implementation\n"; 3165efa3ccbd17e81358f87474e53492871b1c83e41Vikram S. Adve 317007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner // Output all of the methods... 318c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner for_each(M->begin(), M->end(), bind_obj(this,&AssemblyWriter::printMethod)); 319009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner} 320009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 321c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattnervoid AssemblyWriter::printGlobal(const GlobalVariable *GV) { 32270cc3397f84c2e1fd69c059a0ef89e398e847b00Chris Lattner if (GV->hasName()) Out << "%" << GV->getName() << " = "; 323d70684f7585a85c4248c1c224059478108741c70Chris Lattner 324dda719665ba2ffd2eb1c32a0d2daa3921448db7cChris Lattner if (GV->hasInternalLinkage()) Out << "internal "; 325d70684f7585a85c4248c1c224059478108741c70Chris Lattner if (!GV->hasInitializer()) Out << "uninitialized "; 326d70684f7585a85c4248c1c224059478108741c70Chris Lattner 327c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner Out << (GV->isConstant() ? "constant " : "global "); 3287a1767520611d9ff6face702068de858e1cadf2cChris Lattner printType(GV->getType()->getElementType()); 329d70684f7585a85c4248c1c224059478108741c70Chris Lattner 330d70684f7585a85c4248c1c224059478108741c70Chris Lattner if (GV->hasInitializer()) 331d70684f7585a85c4248c1c224059478108741c70Chris Lattner writeOperand(GV->getInitializer(), false, false); 332d70684f7585a85c4248c1c224059478108741c70Chris Lattner 333e02fa8551d20081534afa46e0976811687e5183aChris Lattner printInfoComment(GV); 334d70684f7585a85c4248c1c224059478108741c70Chris Lattner Out << endl; 33570cc3397f84c2e1fd69c059a0ef89e398e847b00Chris Lattner} 33670cc3397f84c2e1fd69c059a0ef89e398e847b00Chris Lattner 337009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 338c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner// printSymbolTable - Run through symbol table looking for named constants 339007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner// if a named constant is found, emit it's declaration... 340007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner// 341c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattnervoid AssemblyWriter::printSymbolTable(const SymbolTable &ST) { 342007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner for (SymbolTable::const_iterator TI = ST.begin(); TI != ST.end(); ++TI) { 343007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner SymbolTable::type_const_iterator I = ST.type_begin(TI->first); 344007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner SymbolTable::type_const_iterator End = ST.type_end(TI->first); 345007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner 346007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner for (; I != End; ++I) { 347007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner const Value *V = I->second; 348e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner if (const Constant *CPV = dyn_cast<const Constant>(V)) { 349c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner printConstant(CPV); 3501d87bcf4909b06dcd86320722653341f08b8b396Chris Lattner } else if (const Type *Ty = dyn_cast<const Type>(V)) { 351007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner Out << "\t%" << I->first << " = type " << Ty->getDescription() << endl; 352007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner } 353007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner } 354739a56d26ddd76f7d073745d8be25c53cf39dce5Chris Lattner } 355009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner} 356009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 357009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 358c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner// printConstant - Print out a constant pool entry... 359009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// 360e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattnervoid AssemblyWriter::printConstant(const Constant *CPV) { 361007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner // Don't print out unnamed constants, they will be inlined 362007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner if (!CPV->hasName()) return; 363009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 3641333ed5b4fd3d30ba9ef56741af4d345a9e43953Chris Lattner // Print out name... 3651333ed5b4fd3d30ba9ef56741af4d345a9e43953Chris Lattner Out << "\t%" << CPV->getName() << " = "; 366009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 3671333ed5b4fd3d30ba9ef56741af4d345a9e43953Chris Lattner // Print out the constant type... 368c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner printType(CPV->getType()); 369009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 370009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner // Write the value out now... 371009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner writeOperand(CPV, false, false); 372009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 373009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner if (!CPV->hasName() && CPV->getType() != Type::VoidTy) { 374009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner int Slot = Table.getValSlot(CPV); // Print out the def slot taken... 375c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner Out << "\t\t; <"; 376c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner printType(CPV->getType()) << ">:"; 377009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner if (Slot >= 0) Out << Slot; 378009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner else Out << "<badref>"; 379009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner } 380009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 381009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Out << endl; 382009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner} 383009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 384c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner// printMethod - Print all aspects of a method. 385009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// 386c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattnervoid AssemblyWriter::printMethod(const Method *M) { 387009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner // Print out the return type and name... 388dda719665ba2ffd2eb1c32a0d2daa3921448db7cChris Lattner Out << "\n" << (M->isExternal() ? "declare " : "") 389dda719665ba2ffd2eb1c32a0d2daa3921448db7cChris Lattner << (M->hasInternalLinkage() ? "internal " : ""); 390c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner printType(M->getReturnType()) << " \"" << M->getName() << "\"("; 391009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Table.incorporateMethod(M); 392007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner 393c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner // Loop over the arguments, printing them... 394e02fa8551d20081534afa46e0976811687e5183aChris Lattner const MethodType *MT = cast<const MethodType>(M->getMethodType()); 395007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner 396e02fa8551d20081534afa46e0976811687e5183aChris Lattner if (!M->isExternal()) { 397e02fa8551d20081534afa46e0976811687e5183aChris Lattner for_each(M->getArgumentList().begin(), M->getArgumentList().end(), 398c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner bind_obj(this, &AssemblyWriter::printMethodArgument)); 399e02fa8551d20081534afa46e0976811687e5183aChris Lattner } else { 400c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner // Loop over the arguments, printing them... 401e02fa8551d20081534afa46e0976811687e5183aChris Lattner const MethodType *MT = cast<const MethodType>(M->getMethodType()); 402e02fa8551d20081534afa46e0976811687e5183aChris Lattner for (MethodType::ParamTypes::const_iterator I = MT->getParamTypes().begin(), 403e02fa8551d20081534afa46e0976811687e5183aChris Lattner E = MT->getParamTypes().end(); I != E; ++I) { 404e02fa8551d20081534afa46e0976811687e5183aChris Lattner if (I != MT->getParamTypes().begin()) Out << ", "; 405c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner printType(*I); 406e02fa8551d20081534afa46e0976811687e5183aChris Lattner } 407e02fa8551d20081534afa46e0976811687e5183aChris Lattner } 408007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner 409007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner // Finish printing arguments... 410007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner if (MT->isVarArg()) { 411007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner if (MT->getParamTypes().size()) Out << ", "; 412007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner Out << "..."; // Output varargs portion of signature! 413007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner } 414007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner Out << ")\n"; 415007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner 416007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner if (!M->isExternal()) { 417007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner // Loop over the symbol table, emitting all named constants... 418007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner if (M->hasSymbolTable()) 419c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner printSymbolTable(*M->getSymbolTable()); 420007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner 421007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner Out << "begin"; 422007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner 423007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner // Output all of its basic blocks... for the method 424007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner for_each(M->begin(), M->end(), 425c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner bind_obj(this, &AssemblyWriter::printBasicBlock)); 426007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner 427739a56d26ddd76f7d073745d8be25c53cf39dce5Chris Lattner Out << "end\n"; 428007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner } 429007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner 430007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner Table.purgeMethod(); 431009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner} 432009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 433c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner// printMethodArgument - This member is called for every argument that 434009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// is passed into the method. Simply print it out 435009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// 436c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattnervoid AssemblyWriter::printMethodArgument(const MethodArgument *Arg) { 437009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner // Insert commas as we go... the first arg doesn't get a comma 438009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner if (Arg != Arg->getParent()->getArgumentList().front()) Out << ", "; 439009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 440009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner // Output type... 441c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner printType(Arg->getType()); 442009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 443009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner // Output name, if available... 444009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner if (Arg->hasName()) 445009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Out << " %" << Arg->getName(); 446009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner else if (Table.getValSlot(Arg) < 0) 447009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Out << "<badref>"; 448009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner} 449009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 450c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner// printBasicBlock - This member is called for each basic block in a methd. 451009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// 452c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattnervoid AssemblyWriter::printBasicBlock(const BasicBlock *BB) { 453009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner if (BB->hasName()) { // Print out the label if it exists... 454b9a4578df5e240d2e2e5d3f1f538188a054d70b4Chris Lattner Out << "\n" << BB->getName() << ":"; 455009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner } else { 456009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner int Slot = Table.getValSlot(BB); 457b9a4578df5e240d2e2e5d3f1f538188a054d70b4Chris Lattner Out << "\n; <label>:"; 458009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner if (Slot >= 0) 459b9a4578df5e240d2e2e5d3f1f538188a054d70b4Chris Lattner Out << Slot; // Extra newline seperates out label's 460009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner else 461b9a4578df5e240d2e2e5d3f1f538188a054d70b4Chris Lattner Out << "<badref>"; 462009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner } 463b9a4578df5e240d2e2e5d3f1f538188a054d70b4Chris Lattner Out << "\t\t\t\t\t;[#uses=" << BB->use_size() << "]\n"; // Output # uses 464009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 465007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner // Output all of the instructions in the basic block... 466007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner for_each(BB->begin(), BB->end(), 467c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner bind_obj(this, &AssemblyWriter::printInstruction)); 468009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner} 469009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 470e02fa8551d20081534afa46e0976811687e5183aChris Lattner 471e02fa8551d20081534afa46e0976811687e5183aChris Lattner// printInfoComment - Print a little comment after the instruction indicating 472e02fa8551d20081534afa46e0976811687e5183aChris Lattner// which slot it occupies. 473e02fa8551d20081534afa46e0976811687e5183aChris Lattner// 474e02fa8551d20081534afa46e0976811687e5183aChris Lattnervoid AssemblyWriter::printInfoComment(const Value *V) { 475e02fa8551d20081534afa46e0976811687e5183aChris Lattner if (V->getType() != Type::VoidTy) { 476c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner Out << "\t\t; <"; 477c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner printType(V->getType()) << ">"; 478e02fa8551d20081534afa46e0976811687e5183aChris Lattner 479e02fa8551d20081534afa46e0976811687e5183aChris Lattner if (!V->hasName()) { 480e02fa8551d20081534afa46e0976811687e5183aChris Lattner int Slot = Table.getValSlot(V); // Print out the def slot taken... 481e02fa8551d20081534afa46e0976811687e5183aChris Lattner if (Slot >= 0) Out << ":" << Slot; 482e02fa8551d20081534afa46e0976811687e5183aChris Lattner else Out << ":<badref>"; 483e02fa8551d20081534afa46e0976811687e5183aChris Lattner } 484f7a551fd929a4a849883cde04cfe54e0590a38b7Chris Lattner Out << " [#uses=" << V->use_size() << "]"; // Output # uses 485e02fa8551d20081534afa46e0976811687e5183aChris Lattner } 486e02fa8551d20081534afa46e0976811687e5183aChris Lattner} 487e02fa8551d20081534afa46e0976811687e5183aChris Lattner 488c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner// printInstruction - This member is called for each Instruction in a methd. 489009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// 490c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattnervoid AssemblyWriter::printInstruction(const Instruction *I) { 491009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Out << "\t"; 492009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 493009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner // Print out name if it exists... 494009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner if (I && I->hasName()) 495009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Out << "%" << I->getName() << " = "; 496009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 497009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner // Print out the opcode... 498a41f50dea8573e4a610b5aa5e45b5c368559b084Chris Lattner Out << I->getOpcodeName(); 499009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 500009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner // Print out the type of the operands... 501c8b25d40cbec063b1ca99cc1adf794399c6d05c0Chris Lattner const Value *Operand = I->getNumOperands() ? I->getOperand(0) : 0; 502009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 503009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner // Special case conditional branches to swizzle the condition out to the front 504a41f50dea8573e4a610b5aa5e45b5c368559b084Chris Lattner if (I->getOpcode() == Instruction::Br && I->getNumOperands() > 1) { 505009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner writeOperand(I->getOperand(2), true); 506009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Out << ","; 507009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner writeOperand(Operand, true); 508009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Out << ","; 509009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner writeOperand(I->getOperand(1), true); 510009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 511a41f50dea8573e4a610b5aa5e45b5c368559b084Chris Lattner } else if (I->getOpcode() == Instruction::Switch) { 512009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner // Special case switch statement to get formatting nice and correct... 513009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner writeOperand(Operand , true); Out << ","; 514009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner writeOperand(I->getOperand(1), true); Out << " ["; 515009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 516c8b25d40cbec063b1ca99cc1adf794399c6d05c0Chris Lattner for (unsigned op = 2, Eop = I->getNumOperands(); op < Eop; op += 2) { 517009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Out << "\n\t\t"; 518c8b25d40cbec063b1ca99cc1adf794399c6d05c0Chris Lattner writeOperand(I->getOperand(op ), true); Out << ","; 519009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner writeOperand(I->getOperand(op+1), true); 520009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner } 521009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Out << "\n\t]"; 522b00c582b6d40e6b9ff2d1ed4f5eaf7930e792aceChris Lattner } else if (isa<PHINode>(I)) { 523c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner Out << " "; 5248f410cac044a21a94afece41345ccd9b72047675Chris Lattner printType(I->getType()); 525eed1fc79735c3ade3dddc0d45eec2c9811969e21Chris Lattner Out << " "; 526009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 5278f410cac044a21a94afece41345ccd9b72047675Chris Lattner for (unsigned op = 0, Eop = I->getNumOperands(); op < Eop; op += 2) { 5288f410cac044a21a94afece41345ccd9b72047675Chris Lattner if (op) Out << ", "; 5298f410cac044a21a94afece41345ccd9b72047675Chris Lattner Out << "["; 530c8b25d40cbec063b1ca99cc1adf794399c6d05c0Chris Lattner writeOperand(I->getOperand(op ), false); Out << ","; 53128d480b31623371e9d738d17a62dd0bd6cdce1cdChris Lattner writeOperand(I->getOperand(op+1), false); Out << " ]"; 532c24d2088dc3d79e3b7e38a358b4a71f156c06836Chris Lattner } 533e02fa8551d20081534afa46e0976811687e5183aChris Lattner } else if (isa<ReturnInst>(I) && !Operand) { 534009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Out << " void"; 535e02fa8551d20081534afa46e0976811687e5183aChris Lattner } else if (isa<CallInst>(I)) { 536268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner const PointerType *PTy = dyn_cast<PointerType>(Operand->getType()); 5377a1767520611d9ff6face702068de858e1cadf2cChris Lattner const MethodType *MTy = PTy ?dyn_cast<MethodType>(PTy->getElementType()):0; 538268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner const Type *RetTy = MTy ? MTy->getReturnType() : 0; 539268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner 540268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner // If possible, print out the short form of the call instruction, but we can 541268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner // only do this if the first argument is a pointer to a nonvararg method, 542268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner // and if the value returned is not a pointer to a method. 543268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner // 544268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner if (RetTy && !MTy->isVarArg() && 545268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner (!isa<PointerType>(RetTy)||!isa<MethodType>(cast<PointerType>(RetTy)))){ 546268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner Out << " "; printType(RetTy); 547268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner writeOperand(Operand, false); 548268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner } else { 549268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner writeOperand(Operand, true); 550268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner } 551009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Out << "("; 552c8b25d40cbec063b1ca99cc1adf794399c6d05c0Chris Lattner if (I->getNumOperands() > 1) writeOperand(I->getOperand(1), true); 553c8b25d40cbec063b1ca99cc1adf794399c6d05c0Chris Lattner for (unsigned op = 2, Eop = I->getNumOperands(); op < Eop; ++op) { 554009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Out << ","; 555c8b25d40cbec063b1ca99cc1adf794399c6d05c0Chris Lattner writeOperand(I->getOperand(op), true); 556009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner } 557009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 558009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Out << " )"; 559e02fa8551d20081534afa46e0976811687e5183aChris Lattner } else if (const InvokeInst *II = dyn_cast<InvokeInst>(I)) { 560e02fa8551d20081534afa46e0976811687e5183aChris Lattner // TODO: Should try to print out short form of the Invoke instruction 561e02fa8551d20081534afa46e0976811687e5183aChris Lattner writeOperand(Operand, true); 562e02fa8551d20081534afa46e0976811687e5183aChris Lattner Out << "("; 563e02fa8551d20081534afa46e0976811687e5183aChris Lattner if (I->getNumOperands() > 3) writeOperand(I->getOperand(3), true); 564e02fa8551d20081534afa46e0976811687e5183aChris Lattner for (unsigned op = 4, Eop = I->getNumOperands(); op < Eop; ++op) { 565e02fa8551d20081534afa46e0976811687e5183aChris Lattner Out << ","; 566e02fa8551d20081534afa46e0976811687e5183aChris Lattner writeOperand(I->getOperand(op), true); 567e02fa8551d20081534afa46e0976811687e5183aChris Lattner } 568e02fa8551d20081534afa46e0976811687e5183aChris Lattner 569e02fa8551d20081534afa46e0976811687e5183aChris Lattner Out << " )\n\t\t\tto"; 570e02fa8551d20081534afa46e0976811687e5183aChris Lattner writeOperand(II->getNormalDest(), true); 571e02fa8551d20081534afa46e0976811687e5183aChris Lattner Out << " except"; 572e02fa8551d20081534afa46e0976811687e5183aChris Lattner writeOperand(II->getExceptionalDest(), true); 573e02fa8551d20081534afa46e0976811687e5183aChris Lattner 574a41f50dea8573e4a610b5aa5e45b5c368559b084Chris Lattner } else if (I->getOpcode() == Instruction::Malloc || 575a41f50dea8573e4a610b5aa5e45b5c368559b084Chris Lattner I->getOpcode() == Instruction::Alloca) { 576c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner Out << " "; 5777a1767520611d9ff6face702068de858e1cadf2cChris Lattner printType(cast<const PointerType>(I->getType())->getElementType()); 578c8b25d40cbec063b1ca99cc1adf794399c6d05c0Chris Lattner if (I->getNumOperands()) { 579c8b25d40cbec063b1ca99cc1adf794399c6d05c0Chris Lattner Out << ","; 580c8b25d40cbec063b1ca99cc1adf794399c6d05c0Chris Lattner writeOperand(I->getOperand(0), true); 581009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner } 582e02fa8551d20081534afa46e0976811687e5183aChris Lattner } else if (isa<CastInst>(I)) { 5830908309e3c4b4f423e88d8d8fe8060cb10eaa1c9Chris Lattner writeOperand(Operand, true); 584c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner Out << " to "; 585c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner printType(I->getType()); 586009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner } else if (Operand) { // Print the normal way... 587009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 588009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner // PrintAllTypes - Instructions who have operands of all the same type 589009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner // omit the type from all but the first operand. If the instruction has 590009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner // different type operands (for example br), then they are all printed. 591009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner bool PrintAllTypes = false; 592009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner const Type *TheType = Operand->getType(); 593009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 594c8b25d40cbec063b1ca99cc1adf794399c6d05c0Chris Lattner for (unsigned i = 1, E = I->getNumOperands(); i != E; ++i) { 595c8b25d40cbec063b1ca99cc1adf794399c6d05c0Chris Lattner Operand = I->getOperand(i); 596009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner if (Operand->getType() != TheType) { 597009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner PrintAllTypes = true; // We have differing types! Print them all! 598009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner break; 599009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner } 600009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner } 601009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 602f434cd1b30a77afa26de46a3e7afb8a07b536c6fChris Lattner // Shift Left & Right print both types even for Ubyte LHS 603f434cd1b30a77afa26de46a3e7afb8a07b536c6fChris Lattner if (isa<ShiftInst>(I)) PrintAllTypes = true; 604f434cd1b30a77afa26de46a3e7afb8a07b536c6fChris Lattner 605c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner if (!PrintAllTypes) { 606c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner Out << " "; 607c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner printType(I->getOperand(0)->getType()); 608c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner } 609009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 610c8b25d40cbec063b1ca99cc1adf794399c6d05c0Chris Lattner for (unsigned i = 0, E = I->getNumOperands(); i != E; ++i) { 611009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner if (i) Out << ","; 612c8b25d40cbec063b1ca99cc1adf794399c6d05c0Chris Lattner writeOperand(I->getOperand(i), PrintAllTypes); 613009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner } 614009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner } 615009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 616e02fa8551d20081534afa46e0976811687e5183aChris Lattner printInfoComment(I); 617009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner Out << endl; 618009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner} 619009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 620009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 621c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner// printType - Go to extreme measures to attempt to print out a short, symbolic 622c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner// version of a type name. 623c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner// 624c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattnerostream &AssemblyWriter::printType(const Type *Ty) { 625207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner return printTypeInt(Out, Ty, TypeNames); 626c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner} 627c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner 628c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner 629009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner//===----------------------------------------------------------------------===// 630009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// External Interface declarations 631009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner//===----------------------------------------------------------------------===// 632009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 633009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 634009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 635009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattnervoid WriteToAssembly(const Module *M, ostream &o) { 636009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner if (M == 0) { o << "<null> module\n"; return; } 637009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner SlotCalculator SlotTable(M, true); 638c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner AssemblyWriter W(o, SlotTable, M); 639009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 640009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner W.write(M); 641009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner} 642009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 643b0e4523624fbe493a945ea73a8f3c6d6526f3f27Chris Lattnervoid WriteToAssembly(const GlobalVariable *G, ostream &o) { 644b0e4523624fbe493a945ea73a8f3c6d6526f3f27Chris Lattner if (G == 0) { o << "<null> global variable\n"; return; } 645b0e4523624fbe493a945ea73a8f3c6d6526f3f27Chris Lattner SlotCalculator SlotTable(G->getParent(), true); 646c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner AssemblyWriter W(o, SlotTable, G->getParent()); 647b0e4523624fbe493a945ea73a8f3c6d6526f3f27Chris Lattner W.write(G); 648b0e4523624fbe493a945ea73a8f3c6d6526f3f27Chris Lattner} 649b0e4523624fbe493a945ea73a8f3c6d6526f3f27Chris Lattner 650009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattnervoid WriteToAssembly(const Method *M, ostream &o) { 651009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner if (M == 0) { o << "<null> method\n"; return; } 652009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner SlotCalculator SlotTable(M->getParent(), true); 653c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner AssemblyWriter W(o, SlotTable, M->getParent()); 654009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 655009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner W.write(M); 656009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner} 657009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 658009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 659009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattnervoid WriteToAssembly(const BasicBlock *BB, ostream &o) { 660009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner if (BB == 0) { o << "<null> basic block\n"; return; } 661009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 662009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner SlotCalculator SlotTable(BB->getParent(), true); 663c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner AssemblyWriter W(o, SlotTable, 664c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner BB->getParent() ? BB->getParent()->getParent() : 0); 665009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 666009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner W.write(BB); 667009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner} 668009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 669e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattnervoid WriteToAssembly(const Constant *CPV, ostream &o) { 670009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner if (CPV == 0) { o << "<null> constant pool value\n"; return; } 671c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner o << " " << CPV->getType()->getDescription() << " " << CPV->getStrValue(); 672009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner} 673009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 674009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattnervoid WriteToAssembly(const Instruction *I, ostream &o) { 675009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner if (I == 0) { o << "<null> instruction\n"; return; } 676009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 677c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner const Method *M = I->getParent() ? I->getParent()->getParent() : 0; 678c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner SlotCalculator SlotTable(M, true); 679c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner AssemblyWriter W(o, SlotTable, M ? M->getParent() : 0); 680009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner 681009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner W.write(I); 682009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner} 683da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner 684da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattnervoid CachedWriter::setModule(const Module *M) { 685da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner delete SC; delete AW; 686da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner if (M) { 687da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner SC = new SlotCalculator(M, true); 688da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner AW = new AssemblyWriter(Out, *SC, M); 689da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner } else { 690da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner SC = 0; AW = 0; 691da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner } 692da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner} 693da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner 694da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris LattnerCachedWriter::~CachedWriter() { 695da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner delete AW; 696da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner delete SC; 697da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner} 698da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner 699da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris LattnerCachedWriter &CachedWriter::operator<<(const Value *V) { 700da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner assert(AW && SC && "CachedWriter does not have a current module!"); 701da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner switch (V->getValueType()) { 702da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner case Value::ConstantVal: 703da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner Out << " "; AW->write(V->getType()); 704e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner Out << " " << cast<Constant>(V)->getStrValue(); break; 705da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner case Value::MethodArgumentVal: 706da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner AW->write(V->getType()); Out << " " << V->getName(); break; 707da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner case Value::TypeVal: AW->write(cast<const Type>(V)); break; 708da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner case Value::InstructionVal: AW->write(cast<Instruction>(V)); break; 709da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner case Value::BasicBlockVal: AW->write(cast<BasicBlock>(V)); break; 710da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner case Value::MethodVal: AW->write(cast<Method>(V)); break; 711da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner case Value::GlobalVariableVal: AW->write(cast<GlobalVariable>(V)); break; 712da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner case Value::ModuleVal: AW->write(cast<Module>(V)); break; 713da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner default: Out << "<unknown value type: " << V->getValueType() << ">"; break; 714da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner } 715da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner return *this; 716da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner} 717