X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=lib%2FVMCore%2FAsmWriter.cpp;h=d8a1394497b7c4b4f30b75719e990de3b453d510;hb=ce8a14915d2971039b576e03a32e0ba7c421dba7;hp=aa11d81f27e0804636a29f28dbedd2b258b9ec8d;hpb=94dc1f29cd1dc669680b7d3caa1f59862a656b83;p=oota-llvm.git diff --git a/lib/VMCore/AsmWriter.cpp b/lib/VMCore/AsmWriter.cpp index aa11d81f27e..d8a1394497b 100644 --- a/lib/VMCore/AsmWriter.cpp +++ b/lib/VMCore/AsmWriter.cpp @@ -6,34 +6,38 @@ // LLVM code, because of of the primary uses of it is for debugging // transformations. // -// TODO: print out the type name instead of the full type if a particular type -// is in the symbol table... -// //===----------------------------------------------------------------------===// #include "llvm/Assembly/CachedWriter.h" #include "llvm/Assembly/Writer.h" +#include "llvm/Assembly/PrintModulePass.h" #include "llvm/SlotCalculator.h" +#include "llvm/DerivedTypes.h" +#include "llvm/Instruction.h" #include "llvm/Module.h" -#include "llvm/Function.h" -#include "llvm/GlobalVariable.h" -#include "llvm/BasicBlock.h" -#include "llvm/ConstantVals.h" +#include "llvm/Constants.h" #include "llvm/iMemory.h" #include "llvm/iTerminators.h" #include "llvm/iPHINode.h" #include "llvm/iOther.h" #include "llvm/SymbolTable.h" -#include "llvm/Argument.h" #include "Support/StringExtras.h" #include "Support/STLExtras.h" #include -#include using std::string; using std::map; using std::vector; using std::ostream; +static RegisterPass +X("printm", "Print module to stderr",PassInfo::Analysis|PassInfo::Optimization); +static RegisterPass +Y("print","Print function to stderr",PassInfo::Analysis|PassInfo::Optimization); + +static void WriteAsOperandInternal(ostream &Out, const Value *V, bool PrintName, + map &TypeTable, + SlotCalculator *Table); + static const Module *getModuleFromVal(const Value *V) { if (const Argument *MA = dyn_cast(V)) return MA->getParent() ? MA->getParent()->getParent() : 0; @@ -44,8 +48,6 @@ static const Module *getModuleFromVal(const Value *V) { return M ? M->getParent() : 0; } else if (const GlobalValue *GV = dyn_cast(V)) return GV->getParent(); - else if (const Module *Mod = dyn_cast(V)) - return Mod; return 0; } @@ -61,46 +63,10 @@ static SlotCalculator *createSlotCalculator(const Value *V) { return new SlotCalculator(GV->getParent(), true); } else if (const Function *Func = dyn_cast(V)) { return new SlotCalculator(Func, true); - } else if (const Module *Mod = dyn_cast(V)) { - return new SlotCalculator(Mod, true); } return 0; } -// WriteAsOperand - Write the name of the specified value out to the specified -// ostream. This can be useful when you just want to print int %reg126, not the -// whole instruction that generated it. -// -static void WriteAsOperandInternal(ostream &Out, const Value *V, bool PrintName, - SlotCalculator *Table) { - if (PrintName && V->hasName()) { - Out << " %" << V->getName(); - } else { - if (const Constant *CPV = dyn_cast(V)) { - Out << " " << CPV->getStrValue(); - } else { - int Slot; - if (Table) { - Slot = Table->getValSlot(V); - } else { - if (const Type *Ty = dyn_cast(V)) { - Out << " " << Ty->getDescription(); - return; - } - - Table = createSlotCalculator(V); - if (Table == 0) { Out << "BAD VALUE TYPE!"; return; } - - Slot = Table->getValSlot(V); - delete Table; - } - if (Slot >= 0) Out << " %" << Slot; - else if (PrintName) - Out << ""; // Not embeded into a location? - } - } -} - // If the module has a symbol table, take all global types and stuff their // names into the TypeNames map. @@ -151,17 +117,17 @@ static string calcTypeName(const Type *Ty, vector &TypeStack, string Result; switch (Ty->getPrimitiveID()) { case Type::FunctionTyID: { - const FunctionType *MTy = cast(Ty); - Result = calcTypeName(MTy->getReturnType(), TypeStack, TypeNames) + " ("; + const FunctionType *FTy = cast(Ty); + Result = calcTypeName(FTy->getReturnType(), TypeStack, TypeNames) + " ("; for (FunctionType::ParamTypes::const_iterator - I = MTy->getParamTypes().begin(), - E = MTy->getParamTypes().end(); I != E; ++I) { - if (I != MTy->getParamTypes().begin()) + I = FTy->getParamTypes().begin(), + E = FTy->getParamTypes().end(); I != E; ++I) { + if (I != FTy->getParamTypes().begin()) Result += ", "; Result += calcTypeName(*I, TypeStack, TypeNames); } - if (MTy->isVarArg()) { - if (!MTy->getParamTypes().empty()) Result += ", "; + if (FTy->isVarArg()) { + if (!FTy->getParamTypes().empty()) Result += ", "; Result += "..."; } Result += ")"; @@ -186,15 +152,12 @@ static string calcTypeName(const Type *Ty, vector &TypeStack, break; case Type::ArrayTyID: { const ArrayType *ATy = cast(Ty); - int NumElements = ATy->getNumElements(); - Result = "["; - if (NumElements != -1) Result += itostr(NumElements) + " x "; + Result = "[" + utostr(ATy->getNumElements()) + " x "; Result += calcTypeName(ATy->getElementType(), TypeStack, TypeNames) + "]"; break; } default: - assert(0 && "Unhandled case in getTypeProps!"); - Result = ""; + Result = ""; } TypeStack.pop_back(); // Remove self from stack... @@ -246,17 +209,209 @@ ostream &WriteTypeSymbolic(ostream &Out, const Type *Ty, const Module *M) { } } +static void WriteConstantInt(ostream &Out, const Constant *CV, bool PrintName, + map &TypeTable, + SlotCalculator *Table) { + if (const ConstantBool *CB = dyn_cast(CV)) { + Out << (CB == ConstantBool::True ? "true" : "false"); + } else if (const ConstantSInt *CI = dyn_cast(CV)) { + Out << CI->getValue(); + } else if (const ConstantUInt *CI = dyn_cast(CV)) { + Out << CI->getValue(); + } else if (const ConstantFP *CFP = dyn_cast(CV)) { + // We would like to output the FP constant value in exponential notation, + // but we cannot do this if doing so will lose precision. Check here to + // make sure that we only output it in exponential format if we can parse + // the value back and get the same value. + // + std::string StrVal = ftostr(CFP->getValue()); + + // Check to make sure that the stringized number is not some string like + // "Inf" or NaN, that atof will accept, but the lexer will not. Check that + // the string matches the "[-+]?[0-9]" regex. + // + if ((StrVal[0] >= '0' && StrVal[0] <= '9') || + ((StrVal[0] == '-' || StrVal[0] == '+') && + (StrVal[0] >= '0' && StrVal[0] <= '9'))) + // Reparse stringized version! + if (atof(StrVal.c_str()) == CFP->getValue()) { + Out << StrVal; return; + } + + // Otherwise we could not reparse it to exactly the same value, so we must + // output the string in hexadecimal format! + // + // Behave nicely in the face of C TBAA rules... see: + // http://www.nullstone.com/htmls/category/aliastyp.htm + // + double Val = CFP->getValue(); + char *Ptr = (char*)&Val; + assert(sizeof(double) == sizeof(uint64_t) && sizeof(double) == 8 && + "assuming that double is 64 bits!"); + Out << "0x" << utohexstr(*(uint64_t*)Ptr); + + } else if (const ConstantArray *CA = dyn_cast(CV)) { + // As a special case, print the array as a string if it is an array of + // ubytes or an array of sbytes with positive values. + // + const Type *ETy = CA->getType()->getElementType(); + bool isString = (ETy == Type::SByteTy || ETy == Type::UByteTy); + + if (ETy == Type::SByteTy) + for (unsigned i = 0; i < CA->getNumOperands(); ++i) + if (cast(CA->getOperand(i))->getValue() < 0) { + isString = false; + break; + } + + if (isString) { + Out << "c\""; + for (unsigned i = 0; i < CA->getNumOperands(); ++i) { + unsigned char C = (ETy == Type::SByteTy) ? + (unsigned char)cast(CA->getOperand(i))->getValue() : + (unsigned char)cast(CA->getOperand(i))->getValue(); + + if (isprint(C) && C != '"' && C != '\\') { + Out << C; + } else { + Out << '\\' + << (char) ((C/16 < 10) ? ( C/16 +'0') : ( C/16 -10+'A')) + << (char)(((C&15) < 10) ? ((C&15)+'0') : ((C&15)-10+'A')); + } + } + Out << "\""; + + } else { // Cannot output in string format... + Out << "["; + if (CA->getNumOperands()) { + Out << " "; + printTypeInt(Out, ETy, TypeTable); + WriteAsOperandInternal(Out, CA->getOperand(0), + PrintName, TypeTable, Table); + for (unsigned i = 1, e = CA->getNumOperands(); i != e; ++i) { + Out << ", "; + printTypeInt(Out, ETy, TypeTable); + WriteAsOperandInternal(Out, CA->getOperand(i), PrintName, + TypeTable, Table); + } + } + Out << " ]"; + } + } else if (const ConstantStruct *CS = dyn_cast(CV)) { + Out << "{"; + if (CS->getNumOperands()) { + Out << " "; + printTypeInt(Out, CS->getOperand(0)->getType(), TypeTable); + + WriteAsOperandInternal(Out, CS->getOperand(0), + PrintName, TypeTable, Table); + + for (unsigned i = 1; i < CS->getNumOperands(); i++) { + Out << ", "; + printTypeInt(Out, CS->getOperand(i)->getType(), TypeTable); + + WriteAsOperandInternal(Out, CS->getOperand(i), + PrintName, TypeTable, Table); + } + } + + Out << " }"; + } else if (isa(CV)) { + Out << "null"; + + } else if (const ConstantPointerRef *PR = dyn_cast(CV)) { + const GlobalValue *V = PR->getValue(); + if (V->hasName()) { + Out << "%" << V->getName(); + } else if (Table) { + int Slot = Table->getValSlot(V); + if (Slot >= 0) + Out << "%" << Slot; + else + Out << ""; + } else { + Out << ""; + } + + } else if (const ConstantExpr *CE = dyn_cast(CV)) { + Out << CE->getOpcodeName(); + + bool isGEP = CE->getOpcode() == Instruction::GetElementPtr; + Out << " ("; + + for (User::const_op_iterator OI=CE->op_begin(); OI != CE->op_end(); ++OI) { + printTypeInt(Out, (*OI)->getType(), TypeTable); + WriteAsOperandInternal(Out, *OI, PrintName, TypeTable, Table); + if (OI+1 != CE->op_end()) + Out << ", "; + } + + if (CE->getOpcode() == Instruction::Cast) { + Out << " to "; + printTypeInt(Out, CE->getType(), TypeTable); + } + Out << ")"; + + } else { + Out << ""; + } +} + + +// WriteAsOperand - Write the name of the specified value out to the specified +// ostream. This can be useful when you just want to print int %reg126, not the +// whole instruction that generated it. +// +static void WriteAsOperandInternal(ostream &Out, const Value *V, bool PrintName, + map &TypeTable, + SlotCalculator *Table) { + Out << " "; + if (PrintName && V->hasName()) { + Out << "%" << V->getName(); + } else { + if (const Constant *CV = dyn_cast(V)) { + WriteConstantInt(Out, CV, PrintName, TypeTable, Table); + } else { + int Slot; + if (Table) { + Slot = Table->getValSlot(V); + } else { + if (const Type *Ty = dyn_cast(V)) { + Out << Ty->getDescription(); + return; + } + + Table = createSlotCalculator(V); + if (Table == 0) { Out << "BAD VALUE TYPE!"; return; } + + Slot = Table->getValSlot(V); + delete Table; + } + if (Slot >= 0) Out << "%" << Slot; + else if (PrintName) + Out << ""; // Not embeded into a location? + } + } +} + + // WriteAsOperand - Write the name of the specified value out to the specified // ostream. This can be useful when you just want to print int %reg126, not the // whole instruction that generated it. // ostream &WriteAsOperand(ostream &Out, const Value *V, bool PrintType, - bool PrintName, SlotCalculator *Table) { - if (PrintType) - WriteTypeSymbolic(Out, V->getType(), getModuleFromVal(V)); + bool PrintName, const Module *Context) { + map TypeNames; + if (Context == 0) Context = getModuleFromVal(V); + + if (Context && Context->hasSymbolTable()) + fillTypeNameTable(Context, TypeNames); - WriteAsOperandInternal(Out, V, PrintName, Table); + if (PrintType) + printTypeInt(Out, V->getType(), TypeNames); + + WriteAsOperandInternal(Out, V, PrintName, TypeNames, 0); return Out; } @@ -281,10 +436,12 @@ public: inline void write(const GlobalVariable *G) { printGlobal(G); } inline void write(const Function *F) { printFunction(F); } inline void write(const BasicBlock *BB) { printBasicBlock(BB); } - inline void write(const Instruction *I) { printInstruction(I); } + inline void write(const Instruction *I) { printInstruction(*I); } inline void write(const Constant *CPV) { printConstant(CPV); } inline void write(const Type *Ty) { printType(Ty); } + void writeOperand(const Value *Op, bool PrintType, bool PrintName = true); + private : void printModule(const Module *M); void printSymbolTable(const SymbolTable &ST); @@ -293,21 +450,74 @@ private : void printFunction(const Function *F); void printArgument(const Argument *FA); void printBasicBlock(const BasicBlock *BB); - void printInstruction(const Instruction *I); - ostream &printType(const Type *Ty); + void printInstruction(const Instruction &I); - void writeOperand(const Value *Op, bool PrintType, bool PrintName = true); + // printType - Go to extreme measures to attempt to print out a short, + // symbolic version of a type name. + // + ostream &printType(const Type *Ty) { + return printTypeInt(Out, Ty, TypeNames); + } + + // printTypeAtLeastOneLevel - Print out one level of the possibly complex type + // without considering any symbolic types that we may have equal to it. + // + ostream &printTypeAtLeastOneLevel(const Type *Ty); // printInfoComment - Print a little comment after the instruction indicating // which slot it occupies. - void printInfoComment(const Value *V); + void printInfoComment(const Value &V); }; +// printTypeAtLeastOneLevel - Print out one level of the possibly complex type +// without considering any symbolic types that we may have equal to it. +// +ostream &AssemblyWriter::printTypeAtLeastOneLevel(const Type *Ty) { + if (const FunctionType *FTy = dyn_cast(Ty)) { + printType(FTy->getReturnType()) << " ("; + for (FunctionType::ParamTypes::const_iterator + I = FTy->getParamTypes().begin(), + E = FTy->getParamTypes().end(); I != E; ++I) { + if (I != FTy->getParamTypes().begin()) + Out << ", "; + printType(*I); + } + if (FTy->isVarArg()) { + if (!FTy->getParamTypes().empty()) Out << ", "; + Out << "..."; + } + Out << ")"; + } else if (const StructType *STy = dyn_cast(Ty)) { + Out << "{ "; + for (StructType::ElementTypes::const_iterator + I = STy->getElementTypes().begin(), + E = STy->getElementTypes().end(); I != E; ++I) { + if (I != STy->getElementTypes().begin()) + Out << ", "; + printType(*I); + } + Out << " }"; + } else if (const PointerType *PTy = dyn_cast(Ty)) { + printType(PTy->getElementType()) << "*"; + } else if (const ArrayType *ATy = dyn_cast(Ty)) { + Out << "[" << ATy->getNumElements() << " x "; + printType(ATy->getElementType()) << "]"; + } else if (const OpaqueType *OTy = dyn_cast(Ty)) { + Out << OTy->getDescription(); + } else { + if (!Ty->isPrimitiveType()) + Out << ""; + printType(Ty); + } + return Out; +} + + void AssemblyWriter::writeOperand(const Value *Operand, bool PrintType, bool PrintName) { if (PrintType) { Out << " "; printType(Operand->getType()); } - WriteAsOperandInternal(Out, Operand, PrintName, &Table); + WriteAsOperandInternal(Out, Operand, PrintName, TypeNames, &Table); } @@ -316,13 +526,14 @@ void AssemblyWriter::printModule(const Module *M) { if (M->hasSymbolTable()) printSymbolTable(*M->getSymbolTable()); - for_each(M->gbegin(), M->gend(), - bind_obj(this, &AssemblyWriter::printGlobal)); + for (Module::const_giterator I = M->gbegin(), E = M->gend(); I != E; ++I) + printGlobal(I); - Out << "implementation\n"; + Out << "\nimplementation ; Functions:\n"; // Output all of the functions... - for_each(M->begin(), M->end(), bind_obj(this,&AssemblyWriter::printFunction)); + for (Module::const_iterator I = M->begin(), E = M->end(); I != E; ++I) + printFunction(I); } void AssemblyWriter::printGlobal(const GlobalVariable *GV) { @@ -337,7 +548,7 @@ void AssemblyWriter::printGlobal(const GlobalVariable *GV) { if (GV->hasInitializer()) writeOperand(GV->getInitializer(), false, false); - printInfoComment(GV); + printInfoComment(*GV); Out << "\n"; } @@ -355,7 +566,12 @@ void AssemblyWriter::printSymbolTable(const SymbolTable &ST) { if (const Constant *CPV = dyn_cast(V)) { printConstant(CPV); } else if (const Type *Ty = dyn_cast(V)) { - Out << "\t%" << I->first << " = type " << Ty->getDescription() << "\n"; + Out << "\t%" << I->first << " = type "; + + // Make sure we print out at least one level of the type structure, so + // that we do not get %FILE = type %FILE + // + printTypeAtLeastOneLevel(Ty) << "\n"; } } } @@ -369,69 +585,56 @@ void AssemblyWriter::printConstant(const Constant *CPV) { if (!CPV->hasName()) return; // Print out name... - Out << "\t%" << CPV->getName() << " = "; - - // Print out the constant type... - printType(CPV->getType()); + Out << "\t%" << CPV->getName() << " ="; // Write the value out now... - writeOperand(CPV, false, false); - - if (!CPV->hasName() && CPV->getType() != Type::VoidTy) { - int Slot = Table.getValSlot(CPV); // Print out the def slot taken... - Out << "\t\t; <"; - printType(CPV->getType()) << ">:"; - if (Slot >= 0) Out << Slot; - else Out << ""; - } + writeOperand(CPV, true, false); + printInfoComment(*CPV); Out << "\n"; } // printFunction - Print all aspects of a function. // -void AssemblyWriter::printFunction(const Function *M) { +void AssemblyWriter::printFunction(const Function *F) { // Print out the return type and name... - Out << "\n" << (M->isExternal() ? "declare " : "") - << (M->hasInternalLinkage() ? "internal " : ""); - printType(M->getReturnType()) << " \"" << M->getName() << "\"("; - Table.incorporateFunction(M); + Out << "\n" << (F->isExternal() ? "declare " : "") + << (F->hasInternalLinkage() ? "internal " : ""); + printType(F->getReturnType()) << " %" << F->getName() << "("; + Table.incorporateFunction(F); // Loop over the arguments, printing them... - const FunctionType *MT = M->getFunctionType(); + const FunctionType *FT = F->getFunctionType(); - if (!M->isExternal()) { - for_each(M->getArgumentList().begin(), M->getArgumentList().end(), - bind_obj(this, &AssemblyWriter::printArgument)); + if (!F->isExternal()) { + for(Function::const_aiterator I = F->abegin(), E = F->aend(); I != E; ++I) + printArgument(I); } else { // Loop over the arguments, printing them... - const FunctionType *MT = M->getFunctionType(); - for (FunctionType::ParamTypes::const_iterator I = MT->getParamTypes().begin(), - E = MT->getParamTypes().end(); I != E; ++I) { - if (I != MT->getParamTypes().begin()) Out << ", "; + for (FunctionType::ParamTypes::const_iterator I = FT->getParamTypes().begin(), + E = FT->getParamTypes().end(); I != E; ++I) { + if (I != FT->getParamTypes().begin()) Out << ", "; printType(*I); } } // Finish printing arguments... - if (MT->isVarArg()) { - if (MT->getParamTypes().size()) Out << ", "; + if (FT->isVarArg()) { + if (FT->getParamTypes().size()) Out << ", "; Out << "..."; // Output varargs portion of signature! } - Out << ")\n"; + Out << ")"; - if (!M->isExternal()) { - // Loop over the symbol table, emitting all named constants... - if (M->hasSymbolTable()) - printSymbolTable(*M->getSymbolTable()); - - Out << "begin"; + if (F->isExternal()) { + Out << "\n"; + } else { + Out << " {"; // Output all of its basic blocks... for the function - for_each(M->begin(), M->end(), - bind_obj(this, &AssemblyWriter::printBasicBlock)); + for (Function::const_iterator I = F->begin(), E = F->end(); I != E; ++I) + printBasicBlock(I); - Out << "end\n"; + Out << "}\n"; } Table.purgeFunction(); @@ -442,7 +645,7 @@ void AssemblyWriter::printFunction(const Function *M) { // void AssemblyWriter::printArgument(const Argument *Arg) { // Insert commas as we go... the first arg doesn't get a comma - if (Arg != Arg->getParent()->getArgumentList().front()) Out << ", "; + if (Arg != &Arg->getParent()->afront()) Out << ", "; // Output type... printType(Arg->getType()); @@ -458,84 +661,87 @@ void AssemblyWriter::printArgument(const Argument *Arg) { // void AssemblyWriter::printBasicBlock(const BasicBlock *BB) { if (BB->hasName()) { // Print out the label if it exists... - Out << "\n" << BB->getName() << ":"; - } else { + Out << "\n" << BB->getName() << ":\t\t\t\t\t;[#uses=" + << BB->use_size() << "]"; // Output # uses + } else if (!BB->use_empty()) { // Don't print block # of no uses... int Slot = Table.getValSlot(BB); Out << "\n;