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