From 2a2457158d6ed364066bdeb8edbb9f7a3b9c37a4 Mon Sep 17 00:00:00 2001 From: Reid Spencer Date: Tue, 30 May 2006 21:18:23 +0000 Subject: [PATCH] Fix more bugs. This version now passes all of the Feature test except for a floating point conversion problem with NAN in intrinsics.ll. llvm2cpp makes an attempt to provide the correct conversion, based on the implementation in the CBackend writer, but it doesn't get NAN correct. llvm-svn: 28568 --- tools/llvm2cpp/CppWriter.cpp | 379 ++++++++++++++++++++++------------- 1 file changed, 244 insertions(+), 135 deletions(-) diff --git a/tools/llvm2cpp/CppWriter.cpp b/tools/llvm2cpp/CppWriter.cpp index a30635eb2fc..2783467aa73 100644 --- a/tools/llvm2cpp/CppWriter.cpp +++ b/tools/llvm2cpp/CppWriter.cpp @@ -24,17 +24,32 @@ #include "llvm/ADT/StringExtras.h" #include "llvm/ADT/STLExtras.h" #include "llvm/Support/MathExtras.h" +#include "llvm/Support/CommandLine.h" #include #include #include using namespace llvm; +static cl::opt +ModName("modname", cl::desc("Specify the module name to use"), + cl::value_desc("module name")); + +static cl::opt +FuncName("funcname", cl::desc("Specify the name of the generated function"), + cl::value_desc("function name")); + +static cl::opt +Fragment("fragment", cl::desc("Don't generate a complete program")); + namespace { typedef std::vector TypeList; typedef std::map TypeMap; typedef std::map ValueMap; typedef std::set NameSet; +typedef std::set TypeSet; +typedef std::set ValueSet; +typedef std::map ForwardRefMap; class CppWriter { std::ostream &Out; @@ -45,6 +60,9 @@ class CppWriter { TypeMap UnresolvedTypes; TypeList TypeStack; NameSet UsedNames; + TypeSet DefinedTypes; + ValueSet DefinedValues; + ForwardRefMap ForwardRefs; public: inline CppWriter(std::ostream &o, const Module *M) @@ -76,8 +94,75 @@ private: inline void printTypeDef(const Type* Ty); bool printTypeDefInternal(const Type* Ty); void printEscapedString(const std::string& str); + + std::string getOpName(Value*); + + void printCFP(const ConstantFP* CFP); }; +// printCFP - Print a floating point constant .. very carefully :) +// This makes sure that conversion to/from floating yields the same binary +// result so that we don't lose precision. +void +CppWriter::printCFP(const ConstantFP *CFP) { +#if HAVE_PRINTF_A + char Buffer[100]; + sprintf(Buffer, "%A", CFP->getValue()); + if ((!strncmp(Buffer, "0x", 2) || + !strncmp(Buffer, "-0x", 3) || + !strncmp(Buffer, "+0x", 3)) && + (atof(Buffer) == CFP->getValue())) + Out << Buffer; + else { +#else + std::string StrVal = ftostr(CFP->getValue()); + + while (StrVal[0] == ' ') + StrVal.erase(StrVal.begin()); + + // Check to make sure that the stringized number is not some string like "Inf" + // or NaN. Check that the string matches the "[-+]?[0-9]" regex. + if (((StrVal[0] >= '0' && StrVal[0] <= '9') || + ((StrVal[0] == '-' || StrVal[0] == '+') && + (StrVal[1] >= '0' && StrVal[1] <= '9'))) && + (atof(StrVal.c_str()) == CFP->getValue())) + Out << StrVal; + else if (CFP->getType() == Type::DoubleTy) { + Out << "0x" << std::hex << DoubleToBits(CFP->getValue()) << std::dec + << "ULL /* " << StrVal << " */"; + } else { + Out << "0x" << std::hex << FloatToBits(CFP->getValue()) << std::dec + << "U /* " << StrVal << " */"; + } +#endif +#if HAVE_PRINTF_A + } +#endif +} + +std::string +CppWriter::getOpName(Value* V) { + if (!isa(V) || DefinedValues.find(V) != DefinedValues.end()) + return getCppName(V); + + // See if its alread in the map of forward references, if so just return the + // name we already set up for it + ForwardRefMap::const_iterator I = ForwardRefs.find(V); + if (I != ForwardRefs.end()) + return I->second; + + // This is a new forward reference. Generate a unique name for it + std::string result(std::string("fwdref_") + utostr(uniqueNum++)); + + // Yes, this is a hack. An Argument is the smallest instantiable value that + // we can make as a placeholder for the real value. We'll replace these + // Argument instances later. + Out << " Argument* " << result << " = new Argument(" + << getCppName(V->getType()) << ");\n"; + ForwardRefs[V] = result; + return result; +} + // printEscapedString - Print each character of the specified string, escaping // it if it is not printable or if it is an escape char. void @@ -239,7 +324,9 @@ CppWriter::getCppName(const Type* Ty) void CppWriter::printModule(const Module *M) { Out << "\n// Module Construction\n"; Out << "Module* mod = new Module(\""; - if (M->getModuleIdentifier() == "-") + if (!ModName.empty()) + printEscapedString(ModName); + else if (M->getModuleIdentifier() == "-") printEscapedString(""); else printEscapedString(M->getModuleIdentifier()); @@ -406,27 +493,6 @@ CppWriter::printTypeDef(const Type* Ty) { TypeStack.clear(); printTypeDefInternal(Ty); assert(TypeStack.empty()); - // early resolve as many unresolved types as possible. Search the unresolved - // types map for the type we just printed. Now that its definition is complete - // we can resolve any preview references to it. This prevents a cascade of - // unresolved types. - TypeMap::iterator I = UnresolvedTypes.find(Ty); - if (I != UnresolvedTypes.end()) { - Out << "cast(" << I->second - << "_fwd.get())->refineAbstractTypeTo(" << I->second << ");\n"; - Out << I->second << " = cast<"; - switch (Ty->getTypeID()) { - case Type::FunctionTyID: Out << "FunctionType"; break; - case Type::ArrayTyID: Out << "ArrayType"; break; - case Type::StructTyID: Out << "StructType"; break; - case Type::PackedTyID: Out << "PackedType"; break; - case Type::PointerTyID: Out << "PointerType"; break; - case Type::OpaqueTyID: Out << "OpaqueType"; break; - default: Out << "NoSuchDerivedType"; break; - } - Out << ">(" << I->second << "_fwd.get());\n\n"; - UnresolvedTypes.erase(I); - } } bool @@ -435,16 +501,17 @@ CppWriter::printTypeDefInternal(const Type* Ty) { if (Ty->isPrimitiveType()) return false; - // Determine if the name is in the name list before we modify that list. - TypeMap::const_iterator TNI = TypeNames.find(Ty); + // If we already defined this type, we don't need to define it again. + if (DefinedTypes.find(Ty) != DefinedTypes.end()) + return false; - // Everything below needs the name for the type so get it now + // Everything below needs the name for the type so get it now. std::string typeName(getCppName(Ty)); // Search the type stack for recursion. If we find it, then generate this // as an OpaqueType, but make sure not to do this multiple times because // the type could appear in multiple places on the stack. Once the opaque - // definition is issues, it must not be re-issued. Consequently we have to + // definition is issued, it must not be re-issued. Consequently we have to // check the UnresolvedTypes list as well. if (isOnStack(Ty)) { TypeMap::const_iterator I = UnresolvedTypes.find(Ty); @@ -455,18 +522,10 @@ CppWriter::printTypeDefInternal(const Type* Ty) { return true; } - // Avoid printing things we have already printed. Since TNI was obtained - // before the name was inserted with getCppName and because we know the name - // is not on the stack (currently being defined), we can surmise here that if - // we got the name we've also already emitted its definition. - if (TNI != TypeNames.end()) - return false; - // We're going to print a derived type which, by definition, contains other // types. So, push this one we're printing onto the type stack to assist with // recursive definitions. - TypeStack.push_back(Ty); // push on type stack - bool didRecurse = false; + TypeStack.push_back(Ty); // Print the type definition switch (Ty->getTypeID()) { @@ -559,6 +618,33 @@ CppWriter::printTypeDefInternal(const Type* Ty) { // Pop us off the type stack TypeStack.pop_back(); + + // Indicate that this type is now defined. + DefinedTypes.insert(Ty); + + // Early resolve as many unresolved types as possible. Search the unresolved + // types map for the type we just printed. Now that its definition is complete + // we can resolve any previous references to it. This prevents a cascade of + // unresolved types. + TypeMap::iterator I = UnresolvedTypes.find(Ty); + if (I != UnresolvedTypes.end()) { + Out << "cast(" << I->second + << "_fwd.get())->refineAbstractTypeTo(" << I->second << ");\n"; + Out << I->second << " = cast<"; + switch (Ty->getTypeID()) { + case Type::FunctionTyID: Out << "FunctionType"; break; + case Type::ArrayTyID: Out << "ArrayType"; break; + case Type::StructTyID: Out << "StructType"; break; + case Type::PackedTyID: Out << "PackedType"; break; + case Type::PointerTyID: Out << "PointerType"; break; + case Type::OpaqueTyID: Out << "OpaqueType"; break; + default: Out << "NoSuchDerivedType"; break; + } + Out << ">(" << I->second << "_fwd.get());\n\n"; + UnresolvedTypes.erase(I); + } + + // Finally, separate the type definition from other with a newline. Out << "\n"; // We weren't a recursive type @@ -684,6 +770,8 @@ void CppWriter::printConstant(const Constant *CV) { } else if (const ConstantFP *CFP = dyn_cast(CV)) { Out << "ConstantFP* " << constName << " = ConstantFP::get(" << typeName << ", "; + char buffer[64]; + sprintf(buffer,"%A",CFP->getValue()); // 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 @@ -698,25 +786,28 @@ void CppWriter::printConstant(const Constant *CV) { if (((StrVal[0] >= '0' && StrVal[0] <= '9') || ((StrVal[0] == '-' || StrVal[0] == '+') && (StrVal[1] >= '0' && StrVal[1] <= '9'))) && - (atof(StrVal.c_str()) == CFP->getValue())) + (atof(StrVal.c_str()) == CFP->getValue())) { - Out << StrVal << ");"; + Out << StrVal; } else { + // Otherwise we could not reparse it to exactly the same value, so we must // output the string in hexadecimal format! - assert(sizeof(double) == sizeof(uint64_t) && - "assuming double is 64 bits!"); - Out << "0x" << utohexstr(DoubleToBits(CFP->getValue())) << ");"; + assert(sizeof(double) == sizeof(uint64_t) && + "assuming that double is 64 bits!"); + Out << "0x" << std::hex << DoubleToBits(CFP->getValue()) << std::dec + << "ULL /* " << StrVal << " */"; } + Out << ");"; } else if (const ConstantArray *CA = dyn_cast(CV)) { if (CA->isString() && CA->getType()->getElementType() == Type::SByteTy) { Out << "Constant* " << constName << " = ConstantArray::get(\""; printEscapedString(CA->getAsString()); // Determine if we want null termination or not. if (CA->getType()->getNumElements() <= CA->getAsString().length()) - Out << "\", " << CA->getType()->getNumElements(); + Out << "\", false";// No null terminator else - Out << "\", 0"; // Indicate that the null terminator should be added. + Out << "\", true"; // Indicate that the null terminator should be added. Out << ");"; } else { Out << "std::vector " << constName << "_elems;\n"; @@ -840,6 +931,11 @@ void CppWriter::printFunctionBody(const Function *F) { if (F->isExternal()) return; // external functions have no bodies. + // Clear the DefinedValues and ForwardRefs maps because we can't have + // cross-function forward refs + ForwardRefs.clear(); + DefinedValues.clear(); + // Create all the argument values if (!F->arg_empty()) { Out << " Function::arg_iterator args = " << getCppName(F) @@ -876,6 +972,17 @@ void CppWriter::printFunctionBody(const Function *F) { printInstruction(I,bbname); } } + + // Loop over the ForwardRefs and resolve them now that all instructions + // are generated. + if (!ForwardRefs.empty()) + Out << "\n // Resolve Forward References\n"; + while (!ForwardRefs.empty()) { + ForwardRefMap::iterator I = ForwardRefs.begin(); + Out << " " << I->second << "->replaceAllUsesWith(" + << getCppName(I->first) << "); delete " << I->second << ";\n"; + ForwardRefs.erase(I); + } } // printInstruction - This member is called for each Instruction in a function. @@ -884,25 +991,30 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) { std::string iName(getCppName(I)); + // Before we emit this instruction, we need to take care of generating any + // forward references. So, we get the names of all the operands in advance + std::string* opNames = new std::string[I->getNumOperands()]; + for (unsigned i = 0; i < I->getNumOperands(); i++) { + opNames[i] = getOpName(I->getOperand(i)); + } + switch (I->getOpcode()) { case Instruction::Ret: { const ReturnInst* ret = cast(I); - Out << " ReturnInst* " << iName << " = new ReturnInst("; - if (ret->getReturnValue()) - Out << getCppName(ret->getReturnValue()) << ", "; - Out << bbname << ");"; + Out << " ReturnInst* " << iName << " = new ReturnInst(" + << (ret->getReturnValue() ? opNames[0] + ", " : "") << bbname << ");"; break; } case Instruction::Br: { const BranchInst* br = cast(I); Out << " BranchInst* " << iName << " = new BranchInst(" ; if (br->getNumOperands() == 3 ) { - Out << getCppName(br->getOperand(0)) << ", " - << getCppName(br->getOperand(1)) << ", " - << getCppName(br->getOperand(2)) << ", "; + Out << opNames[0] << ", " + << opNames[1] << ", " + << opNames[2] << ", "; } else if (br->getNumOperands() == 1) { - Out << getCppName(br->getOperand(0)) << ", "; + Out << opNames[0] << ", "; } else { assert(!"branch with 2 operands?"); } @@ -912,13 +1024,13 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) case Instruction::Switch: { const SwitchInst* sw = cast(I); Out << " SwitchInst* " << iName << " = new SwitchInst(" - << getCppName(sw->getOperand(0)) << ", " - << getCppName(sw->getOperand(1)) << ", " + << opNames[0] << ", " + << opNames[1] << ", " << sw->getNumCases() << ", " << bbname << ");\n"; - for (unsigned i = 1; i < sw->getNumCases(); i++ ) { + for (unsigned i = 2; i < sw->getNumOperands(); i += 2 ) { Out << " " << iName << "->addCase(" - << getCppName(sw->getCaseValue(i)) << ", " - << getCppName(sw->getSuccessor(i)) << ");\n"; + << opNames[i] << ", " + << opNames[i+1] << ");\n"; } break; } @@ -927,11 +1039,11 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) Out << " std::vector " << iName << "_params;\n"; for (unsigned i = 3; i < inv->getNumOperands(); ++i) Out << " " << iName << "_params.push_back(" - << getCppName(inv->getOperand(i)) << ");\n"; + << opNames[i] << ");\n"; Out << " InvokeInst* " << iName << " = new InvokeInst(" - << getCppName(inv->getCalledFunction()) << ", " - << getCppName(inv->getNormalDest()) << ", " - << getCppName(inv->getUnwindDest()) << ", " + << opNames[0] << ", " + << opNames[1] << ", " + << opNames[2] << ", " << iName << "_params, \""; printEscapedString(inv->getName()); Out << "\", " << bbname << ");\n"; @@ -974,8 +1086,7 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) case Instruction::Shr: Out << "Instruction::Shr"; break; default: Out << "Instruction::BadOpCode"; break; } - Out << ", " << getCppName(I->getOperand(0)); - Out << ", " << getCppName(I->getOperand(1)) << ", \""; + Out << ", " << opNames[0] << ", " << opNames[1] << ", \""; printEscapedString(I->getName()); Out << "\", " << bbname << ");"; break; @@ -996,8 +1107,7 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) case Instruction::SetGT: Out << "Instruction::SetGT"; break; default: Out << "Instruction::BadOpCode"; break; } - Out << ", " << getCppName(I->getOperand(0)); - Out << ", " << getCppName(I->getOperand(1)) << ", \""; + Out << ", " << opNames[0] << ", " << opNames[1] << ", \""; printEscapedString(I->getName()); Out << "\", " << bbname << ");"; break; @@ -1007,7 +1117,7 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) Out << " MallocInst* " << iName << " = new MallocInst(" << getCppName(mallocI->getAllocatedType()) << ", "; if (mallocI->isArrayAllocation()) - Out << getCppName(mallocI->getArraySize()) << ", "; + Out << opNames[0] << ", " ; Out << "\""; printEscapedString(mallocI->getName()); Out << "\", " << bbname << ");"; @@ -1026,7 +1136,7 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) Out << " AllocaInst* " << iName << " = new AllocaInst(" << getCppName(allocaI->getAllocatedType()) << ", "; if (allocaI->isArrayAllocation()) - Out << getCppName(allocaI->getArraySize()) << ", "; + Out << opNames[0] << ", "; Out << "\""; printEscapedString(allocaI->getName()); Out << "\", " << bbname << ");"; @@ -1038,7 +1148,7 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) case Instruction::Load:{ const LoadInst* load = cast(I); Out << " LoadInst* " << iName << " = new LoadInst(" - << getCppName(load->getOperand(0)) << ", \""; + << opNames[0] << ", \""; printEscapedString(load->getName()); Out << "\", " << (load->isVolatile() ? "true" : "false" ) << ", " << bbname << ");\n"; @@ -1047,8 +1157,8 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) case Instruction::Store: { const StoreInst* store = cast(I); Out << " StoreInst* " << iName << " = new StoreInst(" - << getCppName(store->getOperand(0)) << ", " - << getCppName(store->getOperand(1)) << ", " + << opNames[0] << ", " + << opNames[1] << ", " << (store->isVolatile() ? "true" : "false") << ", " << bbname << ");\n"; break; @@ -1057,18 +1167,17 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) const GetElementPtrInst* gep = cast(I); if (gep->getNumOperands() <= 2) { Out << " GetElementPtrInst* " << iName << " = new GetElementPtrInst(" - << getCppName(gep->getOperand(0)); + << opNames[0]; if (gep->getNumOperands() == 2) - Out << ", " << getCppName(gep->getOperand(1)); - Out << ", " << bbname; + Out << ", " << opNames[1]; } else { Out << " std::vector " << iName << "_indices;\n"; for (unsigned i = 1; i < gep->getNumOperands(); ++i ) { Out << " " << iName << "_indices.push_back(" - << getCppName(gep->getOperand(i)) << ");\n"; + << opNames[i] << ");\n"; } Out << " Instruction* " << iName << " = new GetElementPtrInst(" - << getCppName(gep->getOperand(0)) << ", " << iName << "_indices"; + << opNames[0] << ", " << iName << "_indices"; } Out << ", \""; printEscapedString(gep->getName()); @@ -1085,17 +1194,16 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) Out << " " << iName << "->reserveOperandSpace(" << phi->getNumIncomingValues() << ");\n"; - for (unsigned i = 0; i < phi->getNumIncomingValues(); ++i) { + for (unsigned i = 0; i < phi->getNumOperands(); i+=2) { Out << " " << iName << "->addIncoming(" - << getCppName(phi->getIncomingValue(i)) << ", " - << getCppName(phi->getIncomingBlock(i)) << ");\n"; + << opNames[i] << ", " << opNames[i+1] << ");\n"; } break; } case Instruction::Cast: { const CastInst* cst = cast(I); Out << " CastInst* " << iName << " = new CastInst(" - << getCppName(cst->getOperand(0)) << ", " + << opNames[0] << ", " << getCppName(cst->getType()) << ", \""; printEscapedString(cst->getName()); Out << "\", " << bbname << ");\n"; @@ -1112,25 +1220,19 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) } if (call->getNumOperands() > 3) { Out << " std::vector " << iName << "_params;\n"; - for (unsigned i = 1; i < call->getNumOperands(); ++i) { - Out << " " << iName << "_params.push_back(" - << getCppName(call->getOperand(i)) << ");\n"; - } + for (unsigned i = 1; i < call->getNumOperands(); ++i) + Out << " " << iName << "_params.push_back(" << opNames[i] << ");\n"; Out << " CallInst* " << iName << " = new CallInst(" - << getCppName(call->getOperand(0)) << ", " - << iName << "_params, \""; + << opNames[0] << ", " << iName << "_params, \""; } else if (call->getNumOperands() == 3) { Out << " CallInst* " << iName << " = new CallInst(" - << getCppName(call->getOperand(0)) << ", " - << getCppName(call->getOperand(1)) << ", " - << getCppName(call->getOperand(2)) << ", \""; + << opNames[0] << ", " << opNames[1] << ", " << opNames[2] << ", \""; } else if (call->getNumOperands() == 2) { Out << " CallInst* " << iName << " = new CallInst(" - << getCppName(call->getOperand(0)) << ", " - << getCppName(call->getOperand(1)) << ", \""; + << opNames[0] << ", " << opNames[1] << ", \""; } else { - Out << " CallInst* " << iName << " = new CallInst(" - << getCppName(call->getOperand(0)) << ", \""; + Out << " CallInst* " << iName << " = new CallInst(" << opNames[0] + << ", \""; } printEscapedString(call->getName()); Out << "\", " << bbname << ");\n"; @@ -1145,9 +1247,7 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) case Instruction::Select: { const SelectInst* sel = cast(I); Out << " SelectInst* " << getCppName(sel) << " = new SelectInst("; - Out << getCppName(sel->getCondition()) << ", "; - Out << getCppName(sel->getTrueValue()) << ", "; - Out << getCppName(sel->getFalseValue()) << ", \""; + Out << opNames[0] << ", " << opNames[1] << ", " << opNames[2] << ", \""; printEscapedString(sel->getName()); Out << "\", " << bbname << ");\n"; break; @@ -1161,8 +1261,7 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) case Instruction::VAArg: { const VAArgInst* va = cast(I); Out << " VAArgInst* " << getCppName(va) << " = new VAArgInst(" - << getCppName(va->getOperand(0)) << ", " - << getCppName(va->getType()) << ", \""; + << opNames[0] << ", " << getCppName(va->getType()) << ", \""; printEscapedString(va->getName()); Out << "\", " << bbname << ");\n"; break; @@ -1170,8 +1269,8 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) case Instruction::ExtractElement: { const ExtractElementInst* eei = cast(I); Out << " ExtractElementInst* " << getCppName(eei) - << " = new ExtractElementInst(" << getCppName(eei->getOperand(0)) - << ", " << getCppName(eei->getOperand(1)) << ", \""; + << " = new ExtractElementInst(" << opNames[0] + << ", " << opNames[1] << ", \""; printEscapedString(eei->getName()); Out << "\", " << bbname << ");\n"; break; @@ -1179,9 +1278,8 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) case Instruction::InsertElement: { const InsertElementInst* iei = cast(I); Out << " InsertElementInst* " << getCppName(iei) - << " = new InsertElementInst(" << getCppName(iei->getOperand(0)) - << ", " << getCppName(iei->getOperand(1)) << ", " - << getCppName(iei->getOperand(2)) << ", \""; + << " = new InsertElementInst(" << opNames[0] + << ", " << opNames[1] << ", " << opNames[2] << ", \""; printEscapedString(iei->getName()); Out << "\", " << bbname << ");\n"; break; @@ -1189,15 +1287,15 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) case Instruction::ShuffleVector: { const ShuffleVectorInst* svi = cast(I); Out << " ShuffleVectorInst* " << getCppName(svi) - << " = new ShuffleVectorInst(" << getCppName(svi->getOperand(0)) - << ", " << getCppName(svi->getOperand(1)) << ", " - << getCppName(svi->getOperand(2)) << ", \""; + << " = new ShuffleVectorInst(" << opNames[0] + << ", " << opNames[1] << ", " << opNames[2] << ", \""; printEscapedString(svi->getName()); Out << "\", " << bbname << ");\n"; break; } } Out << "\n"; + delete [] opNames; } } // end anonymous llvm @@ -1205,38 +1303,49 @@ CppWriter::printInstruction(const Instruction *I, const std::string& bbname) namespace llvm { void WriteModuleToCppFile(Module* mod, std::ostream& o) { - o << "#include \n"; - o << "#include \n"; - o << "#include \n"; - o << "#include \n"; - o << "#include \n"; - o << "#include \n"; - o << "#include \n"; - o << "#include \n"; - o << "#include \n"; - o << "#include \n"; - o << "#include \n"; - o << "#include \n"; - o << "#include \n"; - o << "#include \n"; - o << "#include \n\n"; - o << "using namespace llvm;\n\n"; - o << "Module* makeLLVMModule();\n\n"; - o << "int main(int argc, char**argv) {\n"; - o << " Module* Mod = makeLLVMModule();\n"; - o << " verifyModule(*Mod, PrintMessageAction);\n"; - o << " std::cerr.flush();\n"; - o << " std::cout.flush();\n"; - o << " PassManager PM;\n"; - o << " PM.add(new PrintModulePass(&std::cout));\n"; - o << " PM.run(*Mod);\n"; - o << " return 0;\n"; - o << "}\n\n"; - o << "Module* makeLLVMModule() {\n"; - CppWriter W(o, mod); - W.printModule(mod); - o << "return mod;\n"; - o << "}\n"; + std::string fname = FuncName.getValue(); + if (fname.empty()) + fname = "makeLLVMModule"; + if (Fragment) { + o << "Module* " << fname << "() {\n"; + CppWriter W(o, mod); + W.printModule(mod); + o << "return mod;\n"; + o << "}\n"; + } else { + o << "#include \n"; + o << "#include \n"; + o << "#include \n"; + o << "#include \n"; + o << "#include \n"; + o << "#include \n"; + o << "#include \n"; + o << "#include \n"; + o << "#include \n"; + o << "#include \n"; + o << "#include \n"; + o << "#include \n"; + o << "#include \n"; + o << "#include \n"; + o << "#include \n\n"; + o << "using namespace llvm;\n\n"; + o << "Module* " << fname << "();\n\n"; + o << "int main(int argc, char**argv) {\n"; + o << " Module* Mod = makeLLVMModule();\n"; + o << " verifyModule(*Mod, PrintMessageAction);\n"; + o << " std::cerr.flush();\n"; + o << " std::cout.flush();\n"; + o << " PassManager PM;\n"; + o << " PM.add(new PrintModulePass(&std::cout));\n"; + o << " PM.run(*Mod);\n"; + o << " return 0;\n"; + o << "}\n\n"; + o << "Module* " << fname << "() {\n"; + CppWriter W(o, mod); + W.printModule(mod); + o << "return mod;\n"; + o << "}\n"; + } } }