AsmWriter.cpp revision 42a162ed8023088e7a4a7d81325381bd8d437bb7
18da78afce3609f8ac31bef9d1310744a47bbd0ccChris Lattner//===-- AsmWriter.cpp - Printing LLVM as an assembly file -----------------===//
2fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman//
3b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell//                     The LLVM Compiler Infrastructure
4b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell//
5b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell// This file was developed by the LLVM research group and is distributed under
6b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell// the University of Illinois Open Source License. See LICENSE.TXT for details.
7fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman//
8b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell//===----------------------------------------------------------------------===//
9009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner//
10009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner// This library implements the functionality defined in llvm/Assembly/Writer.h
11009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner//
1202b9399baef2afc1a0c8c83152d2f28145658bbcChris Lattner// Note that these routines must be extremely tolerant of various errors in the
138f77daef04355c00b78b645f5aae5694e7a8b631Chris Lattner// LLVM code, because it can be used for debugging transformations.
1402b9399baef2afc1a0c8c83152d2f28145658bbcChris Lattner//
15009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner//===----------------------------------------------------------------------===//
16009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
17da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner#include "llvm/Assembly/CachedWriter.h"
1875cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner#include "llvm/Assembly/Writer.h"
19f082b80828c13dcb7fb29ad5167ed161c1031534Chris Lattner#include "llvm/Assembly/PrintModulePass.h"
2095e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattner#include "llvm/Assembly/AsmAnnotationWriter.h"
21d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner#include "llvm/CallingConv.h"
22f2d577b27a56a928fa3e77ab0cd83e7597751313Chris Lattner#include "llvm/Constants.h"
233eb59c0074978cb4687779365dbe664e2e18e0b2Chris Lattner#include "llvm/DerivedTypes.h"
24b4dbb4445c67fc46e56765035c2787028a66a614Vikram S. Adve#include "llvm/Instruction.h"
2544336292fcd9f3f99cbfc2c3366bea0cf95bb675Misha Brukman#include "llvm/Instructions.h"
26f2d577b27a56a928fa3e77ab0cd83e7597751313Chris Lattner#include "llvm/Module.h"
27007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner#include "llvm/SymbolTable.h"
285cf1acff3c32e52d79ea8b0b47e341d4e975ed8cMisha Brukman#include "llvm/Assembly/Writer.h"
29061269be9723f842931f18ad207acf9a66e0cf76Chris Lattner#include "llvm/Support/CFG.h"
30551ccae044b0ff658fe629dd67edd5ffe75d10e8Reid Spencer#include "llvm/ADT/StringExtras.h"
31551ccae044b0ff658fe629dd67edd5ffe75d10e8Reid Spencer#include "llvm/ADT/STLExtras.h"
32cb6682fa44e13262bdef7dd22b4ba90f8c2e7b97Jim Laskey#include "llvm/Support/MathExtras.h"
33007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner#include <algorithm>
3431f8499e83dc4dccbb57ea7e76d1fd49b7010d0cChris Lattnerusing namespace llvm;
35d0fde30ce850b78371fd1386338350591f9ff494Brian Gaeke
360a8e8e1a4ea46fa5da067369ac43d8a459d0cac0Chris Lattnernamespace llvm {
370d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
38edd5d9ece15f73ec1a31423a4ae39774aa6c521cReid Spencer// Make virtual table appear in this compilation unit.
39edd5d9ece15f73ec1a31423a4ae39774aa6c521cReid SpencerAssemblyAnnotationWriter::~AssemblyAnnotationWriter() {}
40edd5d9ece15f73ec1a31423a4ae39774aa6c521cReid Spencer
410d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// This class provides computation of slot numbers for LLVM Assembly writing.
420d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @brief LLVM Assembly Writing Slot Computation.
430d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencerclass SlotMachine {
440d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
450d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @name Types
460d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @{
470d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencerpublic:
480d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
490d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// @brief A mapping of Values to slot numbers
500d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  typedef std::map<const Value*, unsigned> ValueMap;
510e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  typedef std::map<const Type*, unsigned> TypeMap;
520d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
530d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// @brief A plane with next slot number and ValueMap
54fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  struct ValuePlane {
550d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    unsigned next_slot;        ///< The next slot number to use
560d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    ValueMap map;              ///< The map of Value* -> unsigned
570e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    ValuePlane() { next_slot = 0; } ///< Make sure we start at 0
580e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  };
590e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
600e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  struct TypePlane {
610e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    unsigned next_slot;
620e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    TypeMap map;
630e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    TypePlane() { next_slot = 0; }
640e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    void clear() { map.clear(); next_slot = 0; }
650d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  };
660d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
670d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// @brief The map of planes by Type
680e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  typedef std::map<const Type*, ValuePlane> TypedPlanes;
690d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
700d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @}
710d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @name Constructors
720d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @{
730d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencerpublic:
740d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// @brief Construct from a module
750d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SlotMachine(const Module *M );
760d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
770d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// @brief Construct from a function, starting out in incorp state.
780d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SlotMachine(const Function *F );
790d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
800d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @}
810d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @name Accessors
820d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @{
830d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencerpublic:
840d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// Return the slot number of the specified value in it's type
850d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// plane.  Its an error to ask for something not in the SlotMachine.
860d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// Its an error to ask for a Type*
8769566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner  int getSlot(const Value *V);
880e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  int getSlot(const Type*Ty);
89fc621e28781e3cf22f1850e07f285581334e2402Reid Spencer
90fc621e28781e3cf22f1850e07f285581334e2402Reid Spencer  /// Determine if a Value has a slot or not
91fc621e28781e3cf22f1850e07f285581334e2402Reid Spencer  bool hasSlot(const Value* V);
920e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  bool hasSlot(const Type* Ty);
930d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
940d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @}
950d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @name Mutators
960d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @{
970d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencerpublic:
98fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  /// If you'd like to deal with a function instead of just a module, use
990d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// this method to get its data into the SlotMachine.
100fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  void incorporateFunction(const Function *F) {
101fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman    TheFunction = F;
10228531c71cd142f623d56a7ea13df2e014eebff54Reid Spencer    FunctionProcessed = false;
10328531c71cd142f623d56a7ea13df2e014eebff54Reid Spencer  }
1040d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
105fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  /// After calling incorporateFunction, use this method to remove the
106fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  /// most recently incorporated function from the SlotMachine. This
1070d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// will reset the state of the machine back to just the module contents.
1080d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  void purgeFunction();
1090d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1100d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @}
1110d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @name Implementation Details
1120d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @{
1130d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencerprivate:
114b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  /// This function does the actual initialization.
115b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  inline void initialize();
116b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer
117fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  /// Values can be crammed into here at will. If they haven't
1180d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// been inserted already, they get inserted, otherwise they are ignored.
1190d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// Either way, the slot number for the Value* is returned.
1200d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  unsigned createSlot(const Value *V);
1210e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  unsigned createSlot(const Type* Ty);
1220d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1230d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// Insert a value into the value table. Return the slot number
1240d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// that it now occupies.  BadThings(TM) will happen if you insert a
125fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  /// Value that's already been inserted.
1260d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  unsigned insertValue( const Value *V );
1270e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  unsigned insertValue( const Type* Ty);
1280d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1290d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// Add all of the module level global variables (and their initializers)
1300d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// and function declarations, but not the contents of those functions.
1310d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  void processModule();
1320d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
133b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  /// Add all of the functions arguments, basic blocks, and instructions
134b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  void processFunction();
135b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer
1360d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SlotMachine(const SlotMachine &);  // DO NOT IMPLEMENT
1370d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  void operator=(const SlotMachine &);  // DO NOT IMPLEMENT
1380d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1390d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @}
1400d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @name Data
1410d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @{
1420d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencerpublic:
1430d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1440d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// @brief The module for which we are holding slot numbers
145b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  const Module* TheModule;
1460d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
147b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  /// @brief The function for which we are holding slot numbers
148b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  const Function* TheFunction;
14928531c71cd142f623d56a7ea13df2e014eebff54Reid Spencer  bool FunctionProcessed;
1500d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1510d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// @brief The TypePlanes map for the module level data
1520d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  TypedPlanes mMap;
1530e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  TypePlane mTypes;
1540d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1550d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  /// @brief The TypePlanes map for the function level data
1560d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  TypedPlanes fMap;
1570e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  TypePlane fTypes;
1580d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1590d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// @}
1600d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1610d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer};
1620d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1630a8e8e1a4ea46fa5da067369ac43d8a459d0cac0Chris Lattner}  // end namespace llvm
1640d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
165a6275ccdf5e1aa208afde56c498e2b13e16442f0Chris Lattnerstatic RegisterPass<PrintModulePass>
166a6275ccdf5e1aa208afde56c498e2b13e16442f0Chris LattnerX("printm", "Print module to stderr",PassInfo::Analysis|PassInfo::Optimization);
167a6275ccdf5e1aa208afde56c498e2b13e16442f0Chris Lattnerstatic RegisterPass<PrintFunctionPass>
168a6275ccdf5e1aa208afde56c498e2b13e16442f0Chris LattnerY("print","Print function to stderr",PassInfo::Analysis|PassInfo::Optimization);
169f082b80828c13dcb7fb29ad5167ed161c1031534Chris Lattner
170fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukmanstatic void WriteAsOperandInternal(std::ostream &Out, const Value *V,
1717b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner                                   bool PrintName,
1727b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner                                 std::map<const Type *, std::string> &TypeTable,
1730d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer                                   SlotMachine *Machine);
1747a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner
175fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukmanstatic void WriteAsOperandInternal(std::ostream &Out, const Type *T,
1760e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer                                   bool PrintName,
1770e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer                                 std::map<const Type *, std::string> &TypeTable,
1780e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer                                   SlotMachine *Machine);
1790e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
180207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattnerstatic const Module *getModuleFromVal(const Value *V) {
181949a3628024248db01d5b13e03c415e0c88e90e4Chris Lattner  if (const Argument *MA = dyn_cast<Argument>(V))
182207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    return MA->getParent() ? MA->getParent()->getParent() : 0;
183949a3628024248db01d5b13e03c415e0c88e90e4Chris Lattner  else if (const BasicBlock *BB = dyn_cast<BasicBlock>(V))
184207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    return BB->getParent() ? BB->getParent()->getParent() : 0;
185949a3628024248db01d5b13e03c415e0c88e90e4Chris Lattner  else if (const Instruction *I = dyn_cast<Instruction>(V)) {
18679df7c0aaa18129e55968c8783ef8346807bd4afChris Lattner    const Function *M = I->getParent() ? I->getParent()->getParent() : 0;
187207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    return M ? M->getParent() : 0;
188949a3628024248db01d5b13e03c415e0c88e90e4Chris Lattner  } else if (const GlobalValue *GV = dyn_cast<GlobalValue>(V))
189207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    return GV->getParent();
190207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  return 0;
191207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner}
192207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
1930d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencerstatic SlotMachine *createSlotMachine(const Value *V) {
194949a3628024248db01d5b13e03c415e0c88e90e4Chris Lattner  if (const Argument *FA = dyn_cast<Argument>(V)) {
1950d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    return new SlotMachine(FA->getParent());
196949a3628024248db01d5b13e03c415e0c88e90e4Chris Lattner  } else if (const Instruction *I = dyn_cast<Instruction>(V)) {
1970d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    return new SlotMachine(I->getParent()->getParent());
198949a3628024248db01d5b13e03c415e0c88e90e4Chris Lattner  } else if (const BasicBlock *BB = dyn_cast<BasicBlock>(V)) {
1990d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    return new SlotMachine(BB->getParent());
200949a3628024248db01d5b13e03c415e0c88e90e4Chris Lattner  } else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(V)){
2010d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    return new SlotMachine(GV->getParent());
202949a3628024248db01d5b13e03c415e0c88e90e4Chris Lattner  } else if (const Function *Func = dyn_cast<Function>(V)) {
2030d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    return new SlotMachine(Func);
204c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner  }
205c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner  return 0;
206c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner}
207009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
20824b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner// getLLVMName - Turn the specified string into an 'LLVM name', which is either
20924b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner// prefixed with % (if the string only contains simple characters) or is
21024b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner// surrounded with ""'s (if it has special chars in it).
2119913e596782bf9c0f1f31bf397ccc0ddfa967debAlkis Evlogimenosstatic std::string getLLVMName(const std::string &Name,
2129913e596782bf9c0f1f31bf397ccc0ddfa967debAlkis Evlogimenos                               bool prefixName = true) {
21324b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner  assert(!Name.empty() && "Cannot get empty name!");
21424b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner
21524b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner  // First character cannot start with a number...
21624b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner  if (Name[0] >= '0' && Name[0] <= '9')
21724b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner    return "\"" + Name + "\"";
21824b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner
21924b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner  // Scan to see if we have any characters that are not on the "white list"
22024b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner  for (unsigned i = 0, e = Name.size(); i != e; ++i) {
22124b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner    char C = Name[i];
22224b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner    assert(C != '"' && "Illegal character in LLVM value name!");
22324b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner    if ((C < 'a' || C > 'z') && (C < 'A' || C > 'Z') && (C < '0' || C > '9') &&
22424b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner        C != '-' && C != '.' && C != '_')
22524b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner      return "\"" + Name + "\"";
22624b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner  }
227fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
22824b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner  // If we get here, then the identifier is legal to use as a "VarID".
2299913e596782bf9c0f1f31bf397ccc0ddfa967debAlkis Evlogimenos  if (prefixName)
2309913e596782bf9c0f1f31bf397ccc0ddfa967debAlkis Evlogimenos    return "%"+Name;
2319913e596782bf9c0f1f31bf397ccc0ddfa967debAlkis Evlogimenos  else
2329913e596782bf9c0f1f31bf397ccc0ddfa967debAlkis Evlogimenos    return Name;
23324b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner}
23424b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner
235207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
236ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman/// fillTypeNameTable - If the module has a symbol table, take all global types
237ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman/// and stuff their names into the TypeNames map.
238ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman///
239207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattnerstatic void fillTypeNameTable(const Module *M,
2407b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner                              std::map<const Type *, std::string> &TypeNames) {
2416e6026b46569b01f8f6d4dcdb6c899c3a9c76b3eChris Lattner  if (!M) return;
2426e6026b46569b01f8f6d4dcdb6c899c3a9c76b3eChris Lattner  const SymbolTable &ST = M->getSymbolTable();
2439231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer  SymbolTable::type_const_iterator TI = ST.type_begin();
2449231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer  for (; TI != ST.type_end(); ++TI ) {
2459231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer    // As a heuristic, don't insert pointer to primitive types, because
2469231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer    // they are used too often to have a single useful name.
2479231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer    //
2489231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer    const Type *Ty = cast<Type>(TI->second);
2499231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer    if (!isa<PointerType>(Ty) ||
250b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        !cast<PointerType>(Ty)->getElementType()->isPrimitiveType() ||
251b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        isa<OpaqueType>(cast<PointerType>(Ty)->getElementType()))
2529231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer      TypeNames.insert(std::make_pair(Ty, getLLVMName(TI->first)));
253207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  }
254207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner}
255207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
256207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
257207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
258fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukmanstatic void calcTypeName(const Type *Ty,
2594ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell                         std::vector<const Type *> &TypeStack,
2604ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell                         std::map<const Type *, std::string> &TypeNames,
2614ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell                         std::string & Result){
2624ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell  if (Ty->isPrimitiveType() && !isa<OpaqueType>(Ty)) {
2634ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    Result += Ty->getDescription();  // Base case
2644ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    return;
2654ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell  }
266207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
267207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  // Check to see if the type is named.
2687b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner  std::map<const Type *, std::string>::iterator I = TypeNames.find(Ty);
2694ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell  if (I != TypeNames.end()) {
2704ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    Result += I->second;
2714ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    return;
2724ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell  }
273207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
2744ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell  if (isa<OpaqueType>(Ty)) {
2754ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    Result += "opaque";
2764ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    return;
2774ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell  }
27888c17380646dbb7c0c5054a392de5a46d17620baChris Lattner
279207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  // Check to see if the Type is already on the stack...
280207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  unsigned Slot = 0, CurSize = TypeStack.size();
281207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  while (Slot < CurSize && TypeStack[Slot] != Ty) ++Slot; // Scan for type
282207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
283fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  // This is another base case for the recursion.  In this case, we know
284207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  // that we have looped back to a type that we have previously visited.
285207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  // Generate the appropriate upreference to handle this.
2864ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell  if (Slot < CurSize) {
2874ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    Result += "\\" + utostr(CurSize-Slot);     // Here's the upreference
2884ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    return;
2894ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell  }
290207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
291207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  TypeStack.push_back(Ty);    // Recursive case: Add us to the stack..
292fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
293f70c22b019494723d0e706f93d6542dfaa6e73a5Chris Lattner  switch (Ty->getTypeID()) {
2946bfd6a578a3a4fa95c585c988ee712ba880f9923Chris Lattner  case Type::FunctionTyID: {
295949a3628024248db01d5b13e03c415e0c88e90e4Chris Lattner    const FunctionType *FTy = cast<FunctionType>(Ty);
2964ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    calcTypeName(FTy->getReturnType(), TypeStack, TypeNames, Result);
2974ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    Result += " (";
298d5d89967206e1153d24abdb7b22002f7533f55c7Chris Lattner    for (FunctionType::param_iterator I = FTy->param_begin(),
299d5d89967206e1153d24abdb7b22002f7533f55c7Chris Lattner           E = FTy->param_end(); I != E; ++I) {
300d5d89967206e1153d24abdb7b22002f7533f55c7Chris Lattner      if (I != FTy->param_begin())
301207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner        Result += ", ";
3024ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell      calcTypeName(*I, TypeStack, TypeNames, Result);
303207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    }
3042761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner    if (FTy->isVarArg()) {
305d5d89967206e1153d24abdb7b22002f7533f55c7Chris Lattner      if (FTy->getNumParams()) Result += ", ";
306207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner      Result += "...";
307207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    }
308207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    Result += ")";
309207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    break;
310207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  }
311207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  case Type::StructTyID: {
312949a3628024248db01d5b13e03c415e0c88e90e4Chris Lattner    const StructType *STy = cast<StructType>(Ty);
3134ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    Result += "{ ";
314d21cd809b656d3011ec089536857e048e037159cChris Lattner    for (StructType::element_iterator I = STy->element_begin(),
315d21cd809b656d3011ec089536857e048e037159cChris Lattner           E = STy->element_end(); I != E; ++I) {
316d21cd809b656d3011ec089536857e048e037159cChris Lattner      if (I != STy->element_begin())
317207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner        Result += ", ";
3184ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell      calcTypeName(*I, TypeStack, TypeNames, Result);
319207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    }
320207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    Result += " }";
321207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    break;
322207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  }
323207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  case Type::PointerTyID:
324fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman    calcTypeName(cast<PointerType>(Ty)->getElementType(),
3254ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell                          TypeStack, TypeNames, Result);
3264ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    Result += "*";
327207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    break;
328207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  case Type::ArrayTyID: {
329949a3628024248db01d5b13e03c415e0c88e90e4Chris Lattner    const ArrayType *ATy = cast<ArrayType>(Ty);
3304ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    Result += "[" + utostr(ATy->getNumElements()) + " x ";
3314ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    calcTypeName(ATy->getElementType(), TypeStack, TypeNames, Result);
3324ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    Result += "]";
333207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    break;
334207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  }
335715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke  case Type::PackedTyID: {
336715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke    const PackedType *PTy = cast<PackedType>(Ty);
337715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke    Result += "<" + utostr(PTy->getNumElements()) + " x ";
338715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke    calcTypeName(PTy->getElementType(), TypeStack, TypeNames, Result);
339715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke    Result += ">";
340715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke    break;
341715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke  }
3429e094c40df10d928a1242f8652dc56ebd41d38fdChris Lattner  case Type::OpaqueTyID:
3434ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    Result += "opaque";
3449e094c40df10d928a1242f8652dc56ebd41d38fdChris Lattner    break;
345207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  default:
3464ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell    Result += "<unrecognized-type>";
347207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  }
348207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
349207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  TypeStack.pop_back();       // Remove self from stack...
3504ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell  return;
351207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner}
352207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
353207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
3549d0802e7dd165dd9b7372f12799f248e7f133287Misha Brukman/// printTypeInt - The internal guts of printing out a type that has a
3559d0802e7dd165dd9b7372f12799f248e7f133287Misha Brukman/// potentially named portion.
3569d0802e7dd165dd9b7372f12799f248e7f133287Misha Brukman///
3577b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattnerstatic std::ostream &printTypeInt(std::ostream &Out, const Type *Ty,
3587b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner                              std::map<const Type *, std::string> &TypeNames) {
359207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  // Primitive types always print out their description, regardless of whether
360207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  // they have been named or not.
361207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  //
362daf2a492a7eef1d4f68b1ed04b401ca6140c191fChris Lattner  if (Ty->isPrimitiveType() && !isa<OpaqueType>(Ty))
363daf2a492a7eef1d4f68b1ed04b401ca6140c191fChris Lattner    return Out << Ty->getDescription();
364207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
365207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  // Check to see if the type is named.
3667b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner  std::map<const Type *, std::string>::iterator I = TypeNames.find(Ty);
367207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  if (I != TypeNames.end()) return Out << I->second;
368207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
369207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  // Otherwise we have a type that has not been named but is a derived type.
370207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  // Carefully recurse the type hierarchy to print out any contained symbolic
371207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  // names.
372207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  //
3737b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner  std::vector<const Type *> TypeStack;
3744ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell  std::string TypeName;
3754ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell  calcTypeName(Ty, TypeStack, TypeNames, TypeName);
376697954c15da58bd8b186dbafdedd8b06db770201Chris Lattner  TypeNames.insert(std::make_pair(Ty, TypeName));//Cache type name for later use
3774ff620a867b7fcd13fb641c4ea872bd9be4b7b71John Criswell  return (Out << TypeName);
378207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner}
379207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
380e51e03b3c649ed9419bd0e920c03ef9023ccee48Chris Lattner
3819d0802e7dd165dd9b7372f12799f248e7f133287Misha Brukman/// WriteTypeSymbolic - This attempts to write the specified type as a symbolic
3829d0802e7dd165dd9b7372f12799f248e7f133287Misha Brukman/// type, iff there is an entry in the modules symbol table for the specified
3839d0802e7dd165dd9b7372f12799f248e7f133287Misha Brukman/// type or one of it's component types. This is slower than a simple x << Type
3849d0802e7dd165dd9b7372f12799f248e7f133287Misha Brukman///
38531f8499e83dc4dccbb57ea7e76d1fd49b7010d0cChris Lattnerstd::ostream &llvm::WriteTypeSymbolic(std::ostream &Out, const Type *Ty,
38631f8499e83dc4dccbb57ea7e76d1fd49b7010d0cChris Lattner                                      const Module *M) {
387fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  Out << ' ';
388207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
389207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  // If they want us to print out a type, attempt to make it symbolic if there
390207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  // is a symbol table in the module...
3916e6026b46569b01f8f6d4dcdb6c899c3a9c76b3eChris Lattner  if (M) {
3927b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner    std::map<const Type *, std::string> TypeNames;
393207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    fillTypeNameTable(M, TypeNames);
394fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
3957b8660d72f35f5ddea0c81eb71f2bdd60fd62832Chris Lattner    return printTypeInt(Out, Ty, TypeNames);
396207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  } else {
3977b8660d72f35f5ddea0c81eb71f2bdd60fd62832Chris Lattner    return Out << Ty->getDescription();
398207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner  }
399207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner}
400207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
40118365506ebe3e28593cb36951f17f08e75712a29Chris Lattner// PrintEscapedString - Print each character of the specified string, escaping
40218365506ebe3e28593cb36951f17f08e75712a29Chris Lattner// it if it is not printable or if it is an escape char.
40318365506ebe3e28593cb36951f17f08e75712a29Chris Lattnerstatic void PrintEscapedString(const std::string &Str, std::ostream &Out) {
40418365506ebe3e28593cb36951f17f08e75712a29Chris Lattner  for (unsigned i = 0, e = Str.size(); i != e; ++i) {
40518365506ebe3e28593cb36951f17f08e75712a29Chris Lattner    unsigned char C = Str[i];
40618365506ebe3e28593cb36951f17f08e75712a29Chris Lattner    if (isprint(C) && C != '"' && C != '\\') {
40718365506ebe3e28593cb36951f17f08e75712a29Chris Lattner      Out << C;
40818365506ebe3e28593cb36951f17f08e75712a29Chris Lattner    } else {
40918365506ebe3e28593cb36951f17f08e75712a29Chris Lattner      Out << '\\'
41018365506ebe3e28593cb36951f17f08e75712a29Chris Lattner          << (char) ((C/16  < 10) ? ( C/16 +'0') : ( C/16 -10+'A'))
41118365506ebe3e28593cb36951f17f08e75712a29Chris Lattner          << (char)(((C&15) < 10) ? ((C&15)+'0') : ((C&15)-10+'A'));
41218365506ebe3e28593cb36951f17f08e75712a29Chris Lattner    }
41318365506ebe3e28593cb36951f17f08e75712a29Chris Lattner  }
41418365506ebe3e28593cb36951f17f08e75712a29Chris Lattner}
41518365506ebe3e28593cb36951f17f08e75712a29Chris Lattner
416fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman/// @brief Internal constant writer.
417fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukmanstatic void WriteConstantInt(std::ostream &Out, const Constant *CV,
4187b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner                             bool PrintName,
4197b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner                             std::map<const Type *, std::string> &TypeTable,
4200d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer                             SlotMachine *Machine) {
42166e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner  if (const ConstantBool *CB = dyn_cast<ConstantBool>(CV)) {
42266e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    Out << (CB == ConstantBool::True ? "true" : "false");
42366e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner  } else if (const ConstantSInt *CI = dyn_cast<ConstantSInt>(CV)) {
42466e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    Out << CI->getValue();
42566e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner  } else if (const ConstantUInt *CI = dyn_cast<ConstantUInt>(CV)) {
42666e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    Out << CI->getValue();
42766e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner  } else if (const ConstantFP *CFP = dyn_cast<ConstantFP>(CV)) {
42866e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    // We would like to output the FP constant value in exponential notation,
42966e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    // but we cannot do this if doing so will lose precision.  Check here to
43066e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    // make sure that we only output it in exponential format if we can parse
43166e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    // the value back and get the same value.
43266e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    //
43366e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    std::string StrVal = ftostr(CFP->getValue());
43466e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner
43566e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    // Check to make sure that the stringized number is not some string like
43666e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    // "Inf" or NaN, that atof will accept, but the lexer will not.  Check that
43766e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    // the string matches the "[-+]?[0-9]" regex.
43866e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    //
43966e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    if ((StrVal[0] >= '0' && StrVal[0] <= '9') ||
44066e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner        ((StrVal[0] == '-' || StrVal[0] == '+') &&
441b471a23a8506b7476890ee34a25ffbd1b553f3ffBrian Gaeke         (StrVal[1] >= '0' && StrVal[1] <= '9')))
44266e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner      // Reparse stringized version!
44366e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner      if (atof(StrVal.c_str()) == CFP->getValue()) {
44471d94d16b6011b4e9be69678addb9265a6c65266Chris Lattner        Out << StrVal;
44571d94d16b6011b4e9be69678addb9265a6c65266Chris Lattner        return;
44666e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner      }
447fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
44866e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    // Otherwise we could not reparse it to exactly the same value, so we must
44966e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    // output the string in hexadecimal format!
45071d94d16b6011b4e9be69678addb9265a6c65266Chris Lattner    assert(sizeof(double) == sizeof(uint64_t) &&
45166e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner           "assuming that double is 64 bits!");
452cb6682fa44e13262bdef7dd22b4ba90f8c2e7b97Jim Laskey    Out << "0x" << utohexstr(DoubleToBits(CFP->getValue()));
45366e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner
454de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner  } else if (isa<ConstantAggregateZero>(CV)) {
455de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner    Out << "zeroinitializer";
45666e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner  } else if (const ConstantArray *CA = dyn_cast<ConstantArray>(CV)) {
45766e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    // As a special case, print the array as a string if it is an array of
45866e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    // ubytes or an array of sbytes with positive values.
459fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman    //
46066e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    const Type *ETy = CA->getType()->getElementType();
46118365506ebe3e28593cb36951f17f08e75712a29Chris Lattner    if (CA->isString()) {
46266e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner      Out << "c\"";
46318365506ebe3e28593cb36951f17f08e75712a29Chris Lattner      PrintEscapedString(CA->getAsString(), Out);
46466e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner      Out << "\"";
46566e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner
46666e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner    } else {                // Cannot output in string format...
46740c732cc52a5c71cead461646667ef5b13801923Misha Brukman      Out << '[';
4687a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner      if (CA->getNumOperands()) {
46940c732cc52a5c71cead461646667ef5b13801923Misha Brukman        Out << ' ';
47066e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner        printTypeInt(Out, ETy, TypeTable);
4717a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner        WriteAsOperandInternal(Out, CA->getOperand(0),
4720d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer                               PrintName, TypeTable, Machine);
4737a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner        for (unsigned i = 1, e = CA->getNumOperands(); i != e; ++i) {
4747a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner          Out << ", ";
47566e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner          printTypeInt(Out, ETy, TypeTable);
4767a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner          WriteAsOperandInternal(Out, CA->getOperand(i), PrintName,
4770d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer                                 TypeTable, Machine);
4787a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner        }
4797a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner      }
4807a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner      Out << " ]";
4817a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner    }
4827a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner  } else if (const ConstantStruct *CS = dyn_cast<ConstantStruct>(CV)) {
48340c732cc52a5c71cead461646667ef5b13801923Misha Brukman    Out << '{';
4847a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner    if (CS->getNumOperands()) {
48540c732cc52a5c71cead461646667ef5b13801923Misha Brukman      Out << ' ';
4867a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner      printTypeInt(Out, CS->getOperand(0)->getType(), TypeTable);
4877a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner
4887a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner      WriteAsOperandInternal(Out, CS->getOperand(0),
4890d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer                             PrintName, TypeTable, Machine);
4907a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner
4917a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner      for (unsigned i = 1; i < CS->getNumOperands(); i++) {
4927a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner        Out << ", ";
4937a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner        printTypeInt(Out, CS->getOperand(i)->getType(), TypeTable);
4947a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner
4957a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner        WriteAsOperandInternal(Out, CS->getOperand(i),
4960d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer                               PrintName, TypeTable, Machine);
4977a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner      }
4987a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner    }
4997a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner
5007a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner    Out << " }";
501715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke  } else if (const ConstantPacked *CP = dyn_cast<ConstantPacked>(CV)) {
502715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke      const Type *ETy = CP->getType()->getElementType();
503fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman      assert(CP->getNumOperands() > 0 &&
504715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke             "Number of operands for a PackedConst must be > 0");
505715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke      Out << '<';
506715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke      Out << ' ';
507715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke      printTypeInt(Out, ETy, TypeTable);
508715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke      WriteAsOperandInternal(Out, CP->getOperand(0),
509715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke                             PrintName, TypeTable, Machine);
510715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke      for (unsigned i = 1, e = CP->getNumOperands(); i != e; ++i) {
511715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke          Out << ", ";
512715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke          printTypeInt(Out, ETy, TypeTable);
513715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke          WriteAsOperandInternal(Out, CP->getOperand(i), PrintName,
514715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke                                 TypeTable, Machine);
515715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke      }
516715c90ba524e736190a6380695ab337eeb5148beBrian Gaeke      Out << " >";
5177a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner  } else if (isa<ConstantPointerNull>(CV)) {
5187a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner    Out << "null";
5197a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner
520b976e668165e1875a8f1eb7af800e33bb1e4393dChris Lattner  } else if (isa<UndefValue>(CV)) {
521b976e668165e1875a8f1eb7af800e33bb1e4393dChris Lattner    Out << "undef";
522b976e668165e1875a8f1eb7af800e33bb1e4393dChris Lattner
523c188eeb08c873da142a47398be6c405ce3f34f51Chris Lattner  } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CV)) {
52438d877365030f90b099d53ab6a66bd62271e330fChris Lattner    Out << CE->getOpcodeName() << " (";
525fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
526b4dbb4445c67fc46e56765035c2787028a66a614Vikram S. Adve    for (User::const_op_iterator OI=CE->op_begin(); OI != CE->op_end(); ++OI) {
527b4dbb4445c67fc46e56765035c2787028a66a614Vikram S. Adve      printTypeInt(Out, (*OI)->getType(), TypeTable);
5280d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      WriteAsOperandInternal(Out, *OI, PrintName, TypeTable, Machine);
529b4dbb4445c67fc46e56765035c2787028a66a614Vikram S. Adve      if (OI+1 != CE->op_end())
530c188eeb08c873da142a47398be6c405ce3f34f51Chris Lattner        Out << ", ";
531b4dbb4445c67fc46e56765035c2787028a66a614Vikram S. Adve    }
532fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
5331c93e5bd26cf1b41ed7bdf5561b8f20607488b0fChris Lattner    if (CE->getOpcode() == Instruction::Cast) {
53495586b8c833aeca112907e69f545a6ea6e2103ffChris Lattner      Out << " to ";
53595586b8c833aeca112907e69f545a6ea6e2103ffChris Lattner      printTypeInt(Out, CE->getType(), TypeTable);
53695586b8c833aeca112907e69f545a6ea6e2103ffChris Lattner    }
53740c732cc52a5c71cead461646667ef5b13801923Misha Brukman    Out << ')';
53895586b8c833aeca112907e69f545a6ea6e2103ffChris Lattner
5397a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner  } else {
540b4dbb4445c67fc46e56765035c2787028a66a614Vikram S. Adve    Out << "<placeholder or erroneous Constant>";
5417a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner  }
5427a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner}
5437a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner
5447a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner
545ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman/// WriteAsOperand - Write the name of the specified value out to the specified
546ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman/// ostream.  This can be useful when you just want to print int %reg126, not
547ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman/// the whole instruction that generated it.
548ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman///
549fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukmanstatic void WriteAsOperandInternal(std::ostream &Out, const Value *V,
5507b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner                                   bool PrintName,
5517b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner                                  std::map<const Type*, std::string> &TypeTable,
5520d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer                                   SlotMachine *Machine) {
55340c732cc52a5c71cead461646667ef5b13801923Misha Brukman  Out << ' ';
5547970396014eacbe719eb171448ddc546c1ad2289Reid Spencer  if ((PrintName || isa<GlobalValue>(V)) && V->hasName())
55524b8a5d6e7c67b0fdfe98a5b6c3791b313dc904bChris Lattner    Out << getLLVMName(V->getName());
5567970396014eacbe719eb171448ddc546c1ad2289Reid Spencer  else {
5577970396014eacbe719eb171448ddc546c1ad2289Reid Spencer    const Constant *CV = dyn_cast<Constant>(V);
5587970396014eacbe719eb171448ddc546c1ad2289Reid Spencer    if (CV && !isa<GlobalValue>(CV))
5590d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      WriteConstantInt(Out, CV, PrintName, TypeTable, Machine);
5607970396014eacbe719eb171448ddc546c1ad2289Reid Spencer    else {
5617a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner      int Slot;
5620d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      if (Machine) {
563b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        Slot = Machine->getSlot(V);
5647a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner      } else {
5650d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer        Machine = createSlotMachine(V);
566fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman        if (Machine == 0)
56769566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner          Slot = Machine->getSlot(V);
56869566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner        else
56969566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner          Slot = -1;
570b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        delete Machine;
5717a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner      }
57269566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner      if (Slot != -1)
57369566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner        Out << '%' << Slot;
57469566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner      else
57569566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner        Out << "<badref>";
5767a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner    }
5777a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner  }
5787a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner}
5797a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner
5809d0802e7dd165dd9b7372f12799f248e7f133287Misha Brukman/// WriteAsOperand - Write the name of the specified value out to the specified
5819d0802e7dd165dd9b7372f12799f248e7f133287Misha Brukman/// ostream.  This can be useful when you just want to print int %reg126, not
5829d0802e7dd165dd9b7372f12799f248e7f133287Misha Brukman/// the whole instruction that generated it.
5839d0802e7dd165dd9b7372f12799f248e7f133287Misha Brukman///
58431f8499e83dc4dccbb57ea7e76d1fd49b7010d0cChris Lattnerstd::ostream &llvm::WriteAsOperand(std::ostream &Out, const Value *V,
585fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman                                   bool PrintType, bool PrintName,
5869d0802e7dd165dd9b7372f12799f248e7f133287Misha Brukman                                   const Module *Context) {
5877b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner  std::map<const Type *, std::string> TypeNames;
588607dc6880ecfd4807de2163d37f2fa8877b7f62dChris Lattner  if (Context == 0) Context = getModuleFromVal(V);
5897a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner
5906e6026b46569b01f8f6d4dcdb6c899c3a9c76b3eChris Lattner  if (Context)
591607dc6880ecfd4807de2163d37f2fa8877b7f62dChris Lattner    fillTypeNameTable(Context, TypeNames);
592207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner
5937a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner  if (PrintType)
5947a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner    printTypeInt(Out, V->getType(), TypeNames);
595fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
596607dc6880ecfd4807de2163d37f2fa8877b7f62dChris Lattner  WriteAsOperandInternal(Out, V, PrintName, TypeNames, 0);
597622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner  return Out;
598622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner}
599622f740a7dcf0b3520244e58b2233898fd4a46e4Chris Lattner
600fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman/// WriteAsOperandInternal - Write the name of the specified value out to
601fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman/// the specified ostream.  This can be useful when you just want to print
6020e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer/// int %reg126, not the whole instruction that generated it.
6030e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer///
604fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukmanstatic void WriteAsOperandInternal(std::ostream &Out, const Type *T,
6050e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer                                   bool PrintName,
6060e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer                                  std::map<const Type*, std::string> &TypeTable,
6070e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer                                   SlotMachine *Machine) {
6080e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  Out << ' ';
6090e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  int Slot;
6100e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  if (Machine) {
6110e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    Slot = Machine->getSlot(T);
6120e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    if (Slot != -1)
6130e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      Out << '%' << Slot;
6140e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    else
6150e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      Out << "<badref>";
6160e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  } else {
6170e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    Out << T->getDescription();
6180e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  }
6190e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer}
6200e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
6210e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer/// WriteAsOperand - Write the name of the specified value out to the specified
6220e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer/// ostream.  This can be useful when you just want to print int %reg126, not
6230e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer/// the whole instruction that generated it.
6240e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer///
6250e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencerstd::ostream &llvm::WriteAsOperand(std::ostream &Out, const Type *Ty,
626fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman                                   bool PrintType, bool PrintName,
6270e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer                                   const Module *Context) {
6280e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  std::map<const Type *, std::string> TypeNames;
6290e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  assert(Context != 0 && "Can't write types as operand without module context");
6300e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
6310e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  fillTypeNameTable(Context, TypeNames);
6320e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
6330e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  // if (PrintType)
6340e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    // printTypeInt(Out, V->getType(), TypeNames);
635fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
6360e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  printTypeInt(Out, Ty, TypeNames);
6370e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
6380e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  WriteAsOperandInternal(Out, Ty, PrintName, TypeNames, 0);
6390e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  return Out;
6400e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer}
6410e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
64231f8499e83dc4dccbb57ea7e76d1fd49b7010d0cChris Lattnernamespace llvm {
643d8c2e42aeff8bdb3ac905b4721b3d3ca1f904cfaChris Lattner
644007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattnerclass AssemblyWriter {
6450313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  std::ostream &Out;
6460d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SlotMachine &Machine;
647c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner  const Module *TheModule;
6487b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner  std::map<const Type *, std::string> TypeNames;
64995e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattner  AssemblyAnnotationWriter *AnnotationWriter;
650009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattnerpublic:
6510d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  inline AssemblyWriter(std::ostream &o, SlotMachine &Mac, const Module *M,
65295e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattner                        AssemblyAnnotationWriter *AAW)
6530313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    : Out(o), Machine(Mac), TheModule(M), AnnotationWriter(AAW) {
654c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner
655c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner    // If the module has a symbol table, take all global types and stuff their
656c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner    // names into the TypeNames map.
657c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner    //
658207b5bc6a15e12a87c3c861da680b8b23559a34cChris Lattner    fillTypeNameTable(M, TypeNames);
659009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  }
660009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
661c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner  inline void write(const Module *M)         { printModule(M);      }
662c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner  inline void write(const GlobalVariable *G) { printGlobal(G);      }
66379df7c0aaa18129e55968c8783ef8346807bd4afChris Lattner  inline void write(const Function *F)       { printFunction(F);    }
664c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner  inline void write(const BasicBlock *BB)    { printBasicBlock(BB); }
6657e70829632f82de15db187845666aaca6e04b792Chris Lattner  inline void write(const Instruction *I)    { printInstruction(*I); }
666e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner  inline void write(const Constant *CPV)     { printConstant(CPV);  }
667da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner  inline void write(const Type *Ty)          { printType(Ty);       }
668009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
66966e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner  void writeOperand(const Value *Op, bool PrintType, bool PrintName = true);
67066e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner
6715cf1acff3c32e52d79ea8b0b47e341d4e975ed8cMisha Brukman  const Module* getModule() { return TheModule; }
6725cf1acff3c32e52d79ea8b0b47e341d4e975ed8cMisha Brukman
673f771bea7402f05f59b2857586de9934ff1c7bbf5Misha Brukmanprivate:
674c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner  void printModule(const Module *M);
675c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner  void printSymbolTable(const SymbolTable &ST);
676e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner  void printConstant(const Constant *CPV);
677c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner  void printGlobal(const GlobalVariable *GV);
67879df7c0aaa18129e55968c8783ef8346807bd4afChris Lattner  void printFunction(const Function *F);
67973e214244f2403b5ba0ef81b8839600f3c8ffebcChris Lattner  void printArgument(const Argument *FA);
680c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner  void printBasicBlock(const BasicBlock *BB);
6817e70829632f82de15db187845666aaca6e04b792Chris Lattner  void printInstruction(const Instruction &I);
6822761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner
6832761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner  // printType - Go to extreme measures to attempt to print out a short,
6842761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner  // symbolic version of a type name.
6852761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner  //
6867b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner  std::ostream &printType(const Type *Ty) {
6870313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    return printTypeInt(Out, Ty, TypeNames);
6882761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner  }
6892761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner
6902761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner  // printTypeAtLeastOneLevel - Print out one level of the possibly complex type
6912761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner  // without considering any symbolic types that we may have equal to it.
6922761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner  //
6937b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner  std::ostream &printTypeAtLeastOneLevel(const Type *Ty);
694c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner
695e02fa8551d20081534afa46e0976811687e5183aChris Lattner  // printInfoComment - Print a little comment after the instruction indicating
696e02fa8551d20081534afa46e0976811687e5183aChris Lattner  // which slot it occupies.
6977e70829632f82de15db187845666aaca6e04b792Chris Lattner  void printInfoComment(const Value &V);
698009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner};
69973b7495a96e21e1515df86414e0636dbf1395fc5Reid Spencer}  // end of llvm namespace
700009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
701ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman/// printTypeAtLeastOneLevel - Print out one level of the possibly complex type
702ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman/// without considering any symbolic types that we may have equal to it.
703ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman///
7047b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattnerstd::ostream &AssemblyWriter::printTypeAtLeastOneLevel(const Type *Ty) {
7057e70829632f82de15db187845666aaca6e04b792Chris Lattner  if (const FunctionType *FTy = dyn_cast<FunctionType>(Ty)) {
7062761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner    printType(FTy->getReturnType()) << " (";
707d5d89967206e1153d24abdb7b22002f7533f55c7Chris Lattner    for (FunctionType::param_iterator I = FTy->param_begin(),
708d5d89967206e1153d24abdb7b22002f7533f55c7Chris Lattner           E = FTy->param_end(); I != E; ++I) {
709d5d89967206e1153d24abdb7b22002f7533f55c7Chris Lattner      if (I != FTy->param_begin())
7100313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman        Out << ", ";
7117a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner      printType(*I);
7122761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner    }
7132761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner    if (FTy->isVarArg()) {
7140313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      if (FTy->getNumParams()) Out << ", ";
7150313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      Out << "...";
7162761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner    }
7170313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << ')';
7187e70829632f82de15db187845666aaca6e04b792Chris Lattner  } else if (const StructType *STy = dyn_cast<StructType>(Ty)) {
7190313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << "{ ";
720d21cd809b656d3011ec089536857e048e037159cChris Lattner    for (StructType::element_iterator I = STy->element_begin(),
721d21cd809b656d3011ec089536857e048e037159cChris Lattner           E = STy->element_end(); I != E; ++I) {
722d21cd809b656d3011ec089536857e048e037159cChris Lattner      if (I != STy->element_begin())
7230313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman        Out << ", ";
7242761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner      printType(*I);
7252761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner    }
7260313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << " }";
7277e70829632f82de15db187845666aaca6e04b792Chris Lattner  } else if (const PointerType *PTy = dyn_cast<PointerType>(Ty)) {
72840c732cc52a5c71cead461646667ef5b13801923Misha Brukman    printType(PTy->getElementType()) << '*';
7297e70829632f82de15db187845666aaca6e04b792Chris Lattner  } else if (const ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
7300313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << '[' << ATy->getNumElements() << " x ";
73140c732cc52a5c71cead461646667ef5b13801923Misha Brukman    printType(ATy->getElementType()) << ']';
7325527c0b6d1bda2c63212f91837792663469fd764Reid Spencer  } else if (const PackedType *PTy = dyn_cast<PackedType>(Ty)) {
7335527c0b6d1bda2c63212f91837792663469fd764Reid Spencer    Out << '<' << PTy->getNumElements() << " x ";
7345527c0b6d1bda2c63212f91837792663469fd764Reid Spencer    printType(PTy->getElementType()) << '>';
7355527c0b6d1bda2c63212f91837792663469fd764Reid Spencer  }
7365527c0b6d1bda2c63212f91837792663469fd764Reid Spencer  else if (const OpaqueType *OTy = dyn_cast<OpaqueType>(Ty)) {
7370313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << "opaque";
7382761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner  } else {
739b4dbb4445c67fc46e56765035c2787028a66a614Vikram S. Adve    if (!Ty->isPrimitiveType())
7400313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      Out << "<unknown derived type>";
7412761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner    printType(Ty);
7422761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner  }
7430313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  return Out;
7442761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner}
7452761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner
7462761e9f076bbe6c961629aece62db4b836a41ef8Chris Lattner
747fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukmanvoid AssemblyWriter::writeOperand(const Value *Operand, bool PrintType,
748b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer                                  bool PrintName) {
749aab1820816facf83d7bda8f01e3d85ced99601f3Chris Lattner  if (Operand != 0) {
750aab1820816facf83d7bda8f01e3d85ced99601f3Chris Lattner    if (PrintType) { Out << ' '; printType(Operand->getType()); }
751aab1820816facf83d7bda8f01e3d85ced99601f3Chris Lattner    WriteAsOperandInternal(Out, Operand, PrintName, TypeNames, &Machine);
752aab1820816facf83d7bda8f01e3d85ced99601f3Chris Lattner  } else {
753aab1820816facf83d7bda8f01e3d85ced99601f3Chris Lattner    Out << "<null operand!>";
754aab1820816facf83d7bda8f01e3d85ced99601f3Chris Lattner  }
755007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner}
756007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner
757007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner
758c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattnervoid AssemblyWriter::printModule(const Module *M) {
75931ab1b3d74440908889fc8ea0615c1d397478360Chris Lattner  if (!M->getModuleIdentifier().empty() &&
760fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman      // Don't print the ID if it will start a new line (which would
76131ab1b3d74440908889fc8ea0615c1d397478360Chris Lattner      // require a comment char before it).
76231ab1b3d74440908889fc8ea0615c1d397478360Chris Lattner      M->getModuleIdentifier().find('\n') == std::string::npos)
76331ab1b3d74440908889fc8ea0615c1d397478360Chris Lattner    Out << "; ModuleID = '" << M->getModuleIdentifier() << "'\n";
76431ab1b3d74440908889fc8ea0615c1d397478360Chris Lattner
765eb5d3a1526b2604feb07fe874907e30c8d2fcd6dChris Lattner  switch (M->getEndianness()) {
7660313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  case Module::LittleEndian: Out << "target endian = little\n"; break;
7670313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  case Module::BigEndian:    Out << "target endian = big\n";    break;
768eb5d3a1526b2604feb07fe874907e30c8d2fcd6dChris Lattner  case Module::AnyEndianness: break;
769eb5d3a1526b2604feb07fe874907e30c8d2fcd6dChris Lattner  }
770eb5d3a1526b2604feb07fe874907e30c8d2fcd6dChris Lattner  switch (M->getPointerSize()) {
7710313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  case Module::Pointer32:    Out << "target pointersize = 32\n"; break;
7720313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  case Module::Pointer64:    Out << "target pointersize = 64\n"; break;
773eb5d3a1526b2604feb07fe874907e30c8d2fcd6dChris Lattner  case Module::AnyPointerSize: break;
774eb5d3a1526b2604feb07fe874907e30c8d2fcd6dChris Lattner  }
775cddc86f27c822af960fc8d344a86a63fce6eab76Reid Spencer  if (!M->getTargetTriple().empty())
776c9a1f0dcdde056a2cdbc446a9d6853684402263fReid Spencer    Out << "target triple = \"" << M->getTargetTriple() << "\"\n";
777fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
77818365506ebe3e28593cb36951f17f08e75712a29Chris Lattner  if (!M->getInlineAsm().empty()) {
77942a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner    // Split the string into lines, to make it easier to read the .ll file.
78042a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner    std::string Asm = M->getInlineAsm();
78142a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner    size_t CurPos = 0;
78242a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner    size_t NewLine = Asm.find_first_of('\n', CurPos);
78342a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner    while (NewLine != std::string::npos) {
78442a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner      // We found a newline, print the portion of the asm string from the
78542a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner      // last newline up to this newline.
78642a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner      Out << "module asm \"";
78742a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner      PrintEscapedString(std::string(Asm.begin()+CurPos, Asm.begin()+NewLine),
78842a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner                         Out);
78942a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner      Out << "\"\n";
79042a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner      CurPos = NewLine+1;
79142a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner      NewLine = Asm.find_first_of('\n', CurPos);
79242a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner    }
79371cdba31774ce18ed32e89f1c6f7716ab923aa46Chris Lattner    Out << "module asm \"";
79442a162ed8023088e7a4a7d81325381bd8d437bb7Chris Lattner    PrintEscapedString(std::string(Asm.begin()+CurPos, Asm.end()), Out);
79518365506ebe3e28593cb36951f17f08e75712a29Chris Lattner    Out << "\"\n";
79618365506ebe3e28593cb36951f17f08e75712a29Chris Lattner  }
79718365506ebe3e28593cb36951f17f08e75712a29Chris Lattner
79844da7d79760998f9a70340b49a3811229838224dChris Lattner  // Loop over the dependent libraries and emit them.
799cfe97b78e787d232f4a82f5ff0afadf75ba95f6cChris Lattner  Module::lib_iterator LI = M->lib_begin();
800cfe97b78e787d232f4a82f5ff0afadf75ba95f6cChris Lattner  Module::lib_iterator LE = M->lib_end();
801cddc86f27c822af960fc8d344a86a63fce6eab76Reid Spencer  if (LI != LE) {
802cfe97b78e787d232f4a82f5ff0afadf75ba95f6cChris Lattner    Out << "deplibs = [ ";
803cfe97b78e787d232f4a82f5ff0afadf75ba95f6cChris Lattner    while (LI != LE) {
80444da7d79760998f9a70340b49a3811229838224dChris Lattner      Out << '"' << *LI << '"';
805c9a1f0dcdde056a2cdbc446a9d6853684402263fReid Spencer      ++LI;
806cfe97b78e787d232f4a82f5ff0afadf75ba95f6cChris Lattner      if (LI != LE)
807cfe97b78e787d232f4a82f5ff0afadf75ba95f6cChris Lattner        Out << ", ";
808c9a1f0dcdde056a2cdbc446a9d6853684402263fReid Spencer    }
809c9a1f0dcdde056a2cdbc446a9d6853684402263fReid Spencer    Out << " ]\n";
81083f6a77c8fc27eba21bca05be37663f3a6e123a1Reid Spencer  }
811e59eaf407a675845e278c2769e3dbed9f1daae8cReid Spencer
81244da7d79760998f9a70340b49a3811229838224dChris Lattner  // Loop over the symbol table, emitting all named constants.
8136e6026b46569b01f8f6d4dcdb6c899c3a9c76b3eChris Lattner  printSymbolTable(M->getSymbolTable());
814fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
815e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  for (Module::const_global_iterator I = M->global_begin(), E = M->global_end(); I != E; ++I)
8167e70829632f82de15db187845666aaca6e04b792Chris Lattner    printGlobal(I);
817007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner
8180313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  Out << "\nimplementation   ; Functions:\n";
819fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
82044da7d79760998f9a70340b49a3811229838224dChris Lattner  // Output all of the functions.
8217e70829632f82de15db187845666aaca6e04b792Chris Lattner  for (Module::const_iterator I = M->begin(), E = M->end(); I != E; ++I)
8227e70829632f82de15db187845666aaca6e04b792Chris Lattner    printFunction(I);
823009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner}
824009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
825c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattnervoid AssemblyWriter::printGlobal(const GlobalVariable *GV) {
8260313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  if (GV->hasName()) Out << getLLVMName(GV->getName()) << " = ";
827d70684f7585a85c4248c1c224059478108741c70Chris Lattner
828fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  if (!GV->hasInitializer())
8290313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << "external ";
8304ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner  else
8314ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner    switch (GV->getLinkage()) {
8320313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    case GlobalValue::InternalLinkage:  Out << "internal "; break;
8330313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    case GlobalValue::LinkOnceLinkage:  Out << "linkonce "; break;
8340313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    case GlobalValue::WeakLinkage:      Out << "weak "; break;
8350313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    case GlobalValue::AppendingLinkage: Out << "appending "; break;
8364ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner    case GlobalValue::ExternalLinkage: break;
8373c8f2c66e585f946569fa62cbf26509009d2df93Misha Brukman    case GlobalValue::GhostLinkage:
8383c8f2c66e585f946569fa62cbf26509009d2df93Misha Brukman      std::cerr << "GhostLinkage not allowed in AsmWriter!\n";
8393c8f2c66e585f946569fa62cbf26509009d2df93Misha Brukman      abort();
8404ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner    }
841d70684f7585a85c4248c1c224059478108741c70Chris Lattner
8420313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  Out << (GV->isConstant() ? "constant " : "global ");
8437a1767520611d9ff6face702068de858e1cadf2cChris Lattner  printType(GV->getType()->getElementType());
844d70684f7585a85c4248c1c224059478108741c70Chris Lattner
8457970396014eacbe719eb171448ddc546c1ad2289Reid Spencer  if (GV->hasInitializer()) {
8467970396014eacbe719eb171448ddc546c1ad2289Reid Spencer    Constant* C = cast<Constant>(GV->getInitializer());
8477970396014eacbe719eb171448ddc546c1ad2289Reid Spencer    assert(C &&  "GlobalVar initializer isn't constant?");
848acc928042daf6912267544a623bdaba3633fec36Reid Spencer    writeOperand(GV->getInitializer(), false, isa<GlobalValue>(C));
8497970396014eacbe719eb171448ddc546c1ad2289Reid Spencer  }
85030caa2876cd500b49e27b6810ab5dec96398c6faChris Lattner
85160962dbfbaa4a9c3ca6fced08fb90b748aa99203Chris Lattner  if (GV->hasSection())
85260962dbfbaa4a9c3ca6fced08fb90b748aa99203Chris Lattner    Out << ", section \"" << GV->getSection() << '"';
85360962dbfbaa4a9c3ca6fced08fb90b748aa99203Chris Lattner  if (GV->getAlignment())
85430caa2876cd500b49e27b6810ab5dec96398c6faChris Lattner    Out << ", align " << GV->getAlignment();
85560962dbfbaa4a9c3ca6fced08fb90b748aa99203Chris Lattner
8567e70829632f82de15db187845666aaca6e04b792Chris Lattner  printInfoComment(*GV);
8570313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  Out << "\n";
85870cc3397f84c2e1fd69c059a0ef89e398e847b00Chris Lattner}
85970cc3397f84c2e1fd69c059a0ef89e398e847b00Chris Lattner
860009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
8619231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer// printSymbolTable - Run through symbol table looking for constants
8629231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer// and types. Emit their declarations.
863c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattnervoid AssemblyWriter::printSymbolTable(const SymbolTable &ST) {
8649231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer
8659231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer  // Print the types.
8669231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer  for (SymbolTable::type_const_iterator TI = ST.type_begin();
8679231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer       TI != ST.type_end(); ++TI ) {
8680313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << "\t" << getLLVMName(TI->first) << " = type ";
8699231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer
8709231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer    // Make sure we print out at least one level of the type structure, so
8719231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer    // that we do not get %FILE = type %FILE
8729231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer    //
8739231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer    printTypeAtLeastOneLevel(TI->second) << "\n";
8749231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer  }
875fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
8769231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer  // Print the constants, in type plane order.
8779231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer  for (SymbolTable::plane_const_iterator PI = ST.plane_begin();
8789231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer       PI != ST.plane_end(); ++PI ) {
8799231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer    SymbolTable::value_const_iterator VI = ST.value_begin(PI->first);
8809231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer    SymbolTable::value_const_iterator VE = ST.value_end(PI->first);
8819231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer
8829231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer    for (; VI != VE; ++VI) {
8837970396014eacbe719eb171448ddc546c1ad2289Reid Spencer      const Value* V = VI->second;
8847970396014eacbe719eb171448ddc546c1ad2289Reid Spencer      const Constant *CPV = dyn_cast<Constant>(V) ;
8857970396014eacbe719eb171448ddc546c1ad2289Reid Spencer      if (CPV && !isa<GlobalValue>(V)) {
886b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        printConstant(CPV);
887007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner      }
888007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner    }
889739a56d26ddd76f7d073745d8be25c53cf39dce5Chris Lattner  }
890009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner}
891009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
892009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
893ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman/// printConstant - Print out a constant pool entry...
894ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman///
895e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattnervoid AssemblyWriter::printConstant(const Constant *CPV) {
896007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner  // Don't print out unnamed constants, they will be inlined
897007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner  if (!CPV->hasName()) return;
898009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
8991333ed5b4fd3d30ba9ef56741af4d345a9e43953Chris Lattner  // Print out name...
9000313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  Out << "\t" << getLLVMName(CPV->getName()) << " =";
901009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
902009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  // Write the value out now...
9037a716addedc4938fa0ec6b77e5eeaced6eafc5d0Chris Lattner  writeOperand(CPV, true, false);
904009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
9057e70829632f82de15db187845666aaca6e04b792Chris Lattner  printInfoComment(*CPV);
9060313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  Out << "\n";
907009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner}
908009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
909ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman/// printFunction - Print all aspects of a function.
910ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman///
9117e70829632f82de15db187845666aaca6e04b792Chris Lattnervoid AssemblyWriter::printFunction(const Function *F) {
912009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  // Print out the return type and name...
9130313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  Out << "\n";
9144ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner
915333529e6be764aa1e4f83eb25817407a758f37c8Chris Lattner  // Ensure that no local symbols conflict with global symbols.
916333529e6be764aa1e4f83eb25817407a758f37c8Chris Lattner  const_cast<Function*>(F)->renameLocalSymbols();
917333529e6be764aa1e4f83eb25817407a758f37c8Chris Lattner
9180313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  if (AnnotationWriter) AnnotationWriter->emitFunctionAnnot(F, Out);
91995e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattner
9204ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner  if (F->isExternal())
9210313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << "declare ";
9224ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner  else
9234ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner    switch (F->getLinkage()) {
9240313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    case GlobalValue::InternalLinkage:  Out << "internal "; break;
9250313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    case GlobalValue::LinkOnceLinkage:  Out << "linkonce "; break;
9260313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    case GlobalValue::WeakLinkage:      Out << "weak "; break;
9270313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    case GlobalValue::AppendingLinkage: Out << "appending "; break;
9284ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner    case GlobalValue::ExternalLinkage: break;
9293c8f2c66e585f946569fa62cbf26509009d2df93Misha Brukman    case GlobalValue::GhostLinkage:
9303c8f2c66e585f946569fa62cbf26509009d2df93Misha Brukman      std::cerr << "GhostLinkage not allowed in AsmWriter!\n";
9313c8f2c66e585f946569fa62cbf26509009d2df93Misha Brukman      abort();
9324ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner    }
9334ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner
934d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner  // Print the calling convention.
935d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner  switch (F->getCallingConv()) {
936d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner  case CallingConv::C: break;   // default
937d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner  case CallingConv::Fast: Out << "fastcc "; break;
938d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner  case CallingConv::Cold: Out << "coldcc "; break;
939d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner  default: Out << "cc" << F->getCallingConv() << " "; break;
940d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner  }
941d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner
94240c732cc52a5c71cead461646667ef5b13801923Misha Brukman  printType(F->getReturnType()) << ' ';
9434d45bd007d0c3a3a6e5b3876b67fd495acb0db2bChris Lattner  if (!F->getName().empty())
9440313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << getLLVMName(F->getName());
9454d45bd007d0c3a3a6e5b3876b67fd495acb0db2bChris Lattner  else
9460313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << "\"\"";
9470313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  Out << '(';
9480d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  Machine.incorporateFunction(F);
949007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner
950c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner  // Loop over the arguments, printing them...
9517e70829632f82de15db187845666aaca6e04b792Chris Lattner  const FunctionType *FT = F->getFunctionType();
952007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner
953e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  for(Function::const_arg_iterator I = F->arg_begin(), E = F->arg_end(); I != E; ++I)
95469da5cf26143e4542d4bf8c78ffac6d079efe5c9Chris Lattner    printArgument(I);
955007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner
956007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner  // Finish printing arguments...
9577e70829632f82de15db187845666aaca6e04b792Chris Lattner  if (FT->isVarArg()) {
9580313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    if (FT->getNumParams()) Out << ", ";
9590313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << "...";  // Output varargs portion of signature!
960007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner  }
9610313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  Out << ')';
962007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner
96360962dbfbaa4a9c3ca6fced08fb90b748aa99203Chris Lattner  if (F->hasSection())
96460962dbfbaa4a9c3ca6fced08fb90b748aa99203Chris Lattner    Out << " section \"" << F->getSection() << '"';
96530caa2876cd500b49e27b6810ab5dec96398c6faChris Lattner  if (F->getAlignment())
96630caa2876cd500b49e27b6810ab5dec96398c6faChris Lattner    Out << " align " << F->getAlignment();
96760962dbfbaa4a9c3ca6fced08fb90b748aa99203Chris Lattner
9687e70829632f82de15db187845666aaca6e04b792Chris Lattner  if (F->isExternal()) {
9690313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << "\n";
97003e2acb37f675b62c66a8cc78965e8b2623972ecChris Lattner  } else {
9710313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << " {";
972fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
973b579400cd72f274607f4964a9649ea4d38e04c46Chris Lattner    // Output all of its basic blocks... for the function
9747e70829632f82de15db187845666aaca6e04b792Chris Lattner    for (Function::const_iterator I = F->begin(), E = F->end(); I != E; ++I)
9757e70829632f82de15db187845666aaca6e04b792Chris Lattner      printBasicBlock(I);
976007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner
9770313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << "}\n";
978007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner  }
979007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner
9800d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  Machine.purgeFunction();
981009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner}
982009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
983ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman/// printArgument - This member is called for every argument that is passed into
984ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman/// the function.  Simply print it out
985ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman///
98673e214244f2403b5ba0ef81b8839600f3c8ffebcChris Lattnervoid AssemblyWriter::printArgument(const Argument *Arg) {
987009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  // Insert commas as we go... the first arg doesn't get a comma
98839220ded940aa408d5dce9b8ab33e87e1d7a23d7Chris Lattner  if (Arg != Arg->getParent()->arg_begin()) Out << ", ";
989009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
990009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  // Output type...
991c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner  printType(Arg->getType());
992fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
993009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  // Output name, if available...
994009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  if (Arg->hasName())
9950313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << ' ' << getLLVMName(Arg->getName());
996009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner}
997009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
998ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman/// printBasicBlock - This member is called for each basic block in a method.
999ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman///
1000c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattnervoid AssemblyWriter::printBasicBlock(const BasicBlock *BB) {
1001009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  if (BB->hasName()) {              // Print out the label if it exists...
10029913e596782bf9c0f1f31bf397ccc0ddfa967debAlkis Evlogimenos    Out << "\n" << getLLVMName(BB->getName(), false) << ':';
1003afc38686b426645ad10562c7eddfd6785663f1bbChris Lattner  } else if (!BB->use_empty()) {      // Don't print block # of no uses...
10040313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << "\n; <label>:";
100569566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner    int Slot = Machine.getSlot(BB);
100669566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner    if (Slot != -1)
10070313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      Out << Slot;
100869566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner    else
10090313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      Out << "<badref>";
1010061269be9723f842931f18ad207acf9a66e0cf76Chris Lattner  }
10114e4d8624c5c44578d64fe1344670520cceec03f2Chris Lattner
10124e4d8624c5c44578d64fe1344670520cceec03f2Chris Lattner  if (BB->getParent() == 0)
10130313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << "\t\t; Error: Block without parent!";
10144e4d8624c5c44578d64fe1344670520cceec03f2Chris Lattner  else {
10154e4d8624c5c44578d64fe1344670520cceec03f2Chris Lattner    if (BB != &BB->getParent()->front()) {  // Not the entry block?
10164e4d8624c5c44578d64fe1344670520cceec03f2Chris Lattner      // Output predecessors for the block...
10170313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      Out << "\t\t;";
10184e4d8624c5c44578d64fe1344670520cceec03f2Chris Lattner      pred_const_iterator PI = pred_begin(BB), PE = pred_end(BB);
1019fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
10204e4d8624c5c44578d64fe1344670520cceec03f2Chris Lattner      if (PI == PE) {
10210313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman        Out << " No predecessors!";
10224e4d8624c5c44578d64fe1344670520cceec03f2Chris Lattner      } else {
10230313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman        Out << " preds =";
102440efcec8e879b3470b3c0ad9d8124abce2cd8915Chris Lattner        writeOperand(*PI, false, true);
10254e4d8624c5c44578d64fe1344670520cceec03f2Chris Lattner        for (++PI; PI != PE; ++PI) {
10260313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman          Out << ',';
10274e4d8624c5c44578d64fe1344670520cceec03f2Chris Lattner          writeOperand(*PI, false, true);
10284e4d8624c5c44578d64fe1344670520cceec03f2Chris Lattner        }
102940efcec8e879b3470b3c0ad9d8124abce2cd8915Chris Lattner      }
1030061269be9723f842931f18ad207acf9a66e0cf76Chris Lattner    }
1031009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  }
1032fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
10330313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  Out << "\n";
1034009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
10350313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  if (AnnotationWriter) AnnotationWriter->emitBasicBlockStartAnnot(BB, Out);
103695e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattner
1037007377f381e253cc559db8d3c94fa89b0eb55fadChris Lattner  // Output all of the instructions in the basic block...
10387e70829632f82de15db187845666aaca6e04b792Chris Lattner  for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I != E; ++I)
10397e70829632f82de15db187845666aaca6e04b792Chris Lattner    printInstruction(*I);
10409f717ef279f4b82e28c341c98a9aa602f01f9b27Chris Lattner
10410313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  if (AnnotationWriter) AnnotationWriter->emitBasicBlockEndAnnot(BB, Out);
1042009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner}
1043009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
1044e02fa8551d20081534afa46e0976811687e5183aChris Lattner
1045ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman/// printInfoComment - Print a little comment after the instruction indicating
1046ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman/// which slot it occupies.
1047ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman///
10487e70829632f82de15db187845666aaca6e04b792Chris Lattnervoid AssemblyWriter::printInfoComment(const Value &V) {
10497e70829632f82de15db187845666aaca6e04b792Chris Lattner  if (V.getType() != Type::VoidTy) {
10500313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << "\t\t; <";
105140c732cc52a5c71cead461646667ef5b13801923Misha Brukman    printType(V.getType()) << '>';
1052e02fa8551d20081534afa46e0976811687e5183aChris Lattner
10537e70829632f82de15db187845666aaca6e04b792Chris Lattner    if (!V.hasName()) {
105469566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner      int SlotNum = Machine.getSlot(&V);
105569566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner      if (SlotNum == -1)
10560313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman        Out << ":<badref>";
1057fc621e28781e3cf22f1850e07f285581334e2402Reid Spencer      else
10580313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman        Out << ':' << SlotNum; // Print out the def slot taken.
1059e02fa8551d20081534afa46e0976811687e5183aChris Lattner    }
10605c461404fed8d1def8eae02e6574bc5b16502575Chris Lattner    Out << " [#uses=" << V.getNumUses() << ']';  // Output # uses
1061e02fa8551d20081534afa46e0976811687e5183aChris Lattner  }
1062e02fa8551d20081534afa46e0976811687e5183aChris Lattner}
1063e02fa8551d20081534afa46e0976811687e5183aChris Lattner
1064fc621e28781e3cf22f1850e07f285581334e2402Reid Spencer/// printInstruction - This member is called for each Instruction in a function..
1065ab5c6003d23c19d23bbff1054900d208c55e0188Misha Brukman///
10667e70829632f82de15db187845666aaca6e04b792Chris Lattnervoid AssemblyWriter::printInstruction(const Instruction &I) {
10670313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  if (AnnotationWriter) AnnotationWriter->emitInstructionAnnot(&I, Out);
106895e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattner
10690313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  Out << "\t";
1070009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
1071009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  // Print out name if it exists...
10727e70829632f82de15db187845666aaca6e04b792Chris Lattner  if (I.hasName())
10730313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << getLLVMName(I.getName()) << " = ";
1074009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
1075ddb6db4fa11d06217d01d8431596131abdfb7ef0Chris Lattner  // If this is a volatile load or store, print out the volatile marker.
1076e5e475e09d1e8a9ea1bda58b536867ff16600399Chris Lattner  if ((isa<LoadInst>(I)  && cast<LoadInst>(I).isVolatile()) ||
1077ddb6db4fa11d06217d01d8431596131abdfb7ef0Chris Lattner      (isa<StoreInst>(I) && cast<StoreInst>(I).isVolatile())) {
10780313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      Out << "volatile ";
1079ddb6db4fa11d06217d01d8431596131abdfb7ef0Chris Lattner  } else if (isa<CallInst>(I) && cast<CallInst>(I).isTailCall()) {
1080ddb6db4fa11d06217d01d8431596131abdfb7ef0Chris Lattner    // If this is a call, check if it's a tail call.
1081ddb6db4fa11d06217d01d8431596131abdfb7ef0Chris Lattner    Out << "tail ";
1082ddb6db4fa11d06217d01d8431596131abdfb7ef0Chris Lattner  }
1083e5e475e09d1e8a9ea1bda58b536867ff16600399Chris Lattner
1084009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  // Print out the opcode...
10850313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  Out << I.getOpcodeName();
1086009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
1087009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  // Print out the type of the operands...
10887e70829632f82de15db187845666aaca6e04b792Chris Lattner  const Value *Operand = I.getNumOperands() ? I.getOperand(0) : 0;
1089009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
1090009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  // Special case conditional branches to swizzle the condition out to the front
10917e70829632f82de15db187845666aaca6e04b792Chris Lattner  if (isa<BranchInst>(I) && I.getNumOperands() > 1) {
10927e70829632f82de15db187845666aaca6e04b792Chris Lattner    writeOperand(I.getOperand(2), true);
10930313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << ',';
1094009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner    writeOperand(Operand, true);
10950313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << ',';
10967e70829632f82de15db187845666aaca6e04b792Chris Lattner    writeOperand(I.getOperand(1), true);
1097009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
109894dc1f29cd1dc669680b7d3caa1f59862a656b83Chris Lattner  } else if (isa<SwitchInst>(I)) {
1099009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner    // Special case switch statement to get formatting nice and correct...
11000313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    writeOperand(Operand        , true); Out << ',';
11010313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    writeOperand(I.getOperand(1), true); Out << " [";
1102009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
11037e70829632f82de15db187845666aaca6e04b792Chris Lattner    for (unsigned op = 2, Eop = I.getNumOperands(); op < Eop; op += 2) {
11040313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      Out << "\n\t\t";
11050313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      writeOperand(I.getOperand(op  ), true); Out << ',';
11067e70829632f82de15db187845666aaca6e04b792Chris Lattner      writeOperand(I.getOperand(op+1), true);
1107009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner    }
11080313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << "\n\t]";
1109b00c582b6d40e6b9ff2d1ed4f5eaf7930e792aceChris Lattner  } else if (isa<PHINode>(I)) {
11100313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << ' ';
11117e70829632f82de15db187845666aaca6e04b792Chris Lattner    printType(I.getType());
11120313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << ' ';
1113009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
11147e70829632f82de15db187845666aaca6e04b792Chris Lattner    for (unsigned op = 0, Eop = I.getNumOperands(); op < Eop; op += 2) {
11150313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      if (op) Out << ", ";
1116fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman      Out << '[';
11170313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      writeOperand(I.getOperand(op  ), false); Out << ',';
11180313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      writeOperand(I.getOperand(op+1), false); Out << " ]";
1119c24d2088dc3d79e3b7e38a358b4a71f156c06836Chris Lattner    }
1120e02fa8551d20081534afa46e0976811687e5183aChris Lattner  } else if (isa<ReturnInst>(I) && !Operand) {
11210313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << " void";
1122d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner  } else if (const CallInst *CI = dyn_cast<CallInst>(&I)) {
1123d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    // Print the calling convention being used.
1124d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    switch (CI->getCallingConv()) {
1125d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    case CallingConv::C: break;   // default
1126d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    case CallingConv::Fast: Out << " fastcc"; break;
1127d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    case CallingConv::Cold: Out << " coldcc"; break;
1128d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    default: Out << " cc" << CI->getCallingConv(); break;
1129d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    }
1130d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner
11317a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    const PointerType  *PTy = cast<PointerType>(Operand->getType());
11327a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
11337a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    const Type       *RetTy = FTy->getReturnType();
1134268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner
11357a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    // If possible, print out the short form of the call instruction.  We can
1136b579400cd72f274607f4964a9649ea4d38e04c46Chris Lattner    // only do this if the first argument is a pointer to a nonvararg function,
11377a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    // and if the return type is not a pointer to a function.
1138268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner    //
11397a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    if (!FTy->isVarArg() &&
1140fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman        (!isa<PointerType>(RetTy) ||
1141c1b2718acf9d566c26188f2968dece0bf3f187e3Chris Lattner         !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
11420313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      Out << ' '; printType(RetTy);
1143268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner      writeOperand(Operand, false);
1144268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner    } else {
1145268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner      writeOperand(Operand, true);
1146268de0464ee9f0938bfa145fdd5e7f1a46b21cf7Chris Lattner    }
11470313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << '(';
1148d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    if (CI->getNumOperands() > 1) writeOperand(CI->getOperand(1), true);
11497e70829632f82de15db187845666aaca6e04b792Chris Lattner    for (unsigned op = 2, Eop = I.getNumOperands(); op < Eop; ++op) {
11500313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      Out << ',';
11517e70829632f82de15db187845666aaca6e04b792Chris Lattner      writeOperand(I.getOperand(op), true);
1152009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner    }
1153009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
11540313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << " )";
11557e70829632f82de15db187845666aaca6e04b792Chris Lattner  } else if (const InvokeInst *II = dyn_cast<InvokeInst>(&I)) {
11567a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    const PointerType  *PTy = cast<PointerType>(Operand->getType());
11577a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
11587a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    const Type       *RetTy = FTy->getReturnType();
11597a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner
1160d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    // Print the calling convention being used.
1161d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    switch (II->getCallingConv()) {
1162d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    case CallingConv::C: break;   // default
1163d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    case CallingConv::Fast: Out << " fastcc"; break;
1164d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    case CallingConv::Cold: Out << " coldcc"; break;
1165d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    default: Out << " cc" << II->getCallingConv(); break;
1166d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner    }
1167d511898b581826694e6c651e23343b1e1ba5ba64Chris Lattner
11687a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    // If possible, print out the short form of the invoke instruction. We can
11697a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    // only do this if the first argument is a pointer to a nonvararg function,
11707a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    // and if the return type is not a pointer to a function.
11717a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    //
11727a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    if (!FTy->isVarArg() &&
1173fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman        (!isa<PointerType>(RetTy) ||
11747a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner         !isa<FunctionType>(cast<PointerType>(RetTy)->getElementType()))) {
11750313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      Out << ' '; printType(RetTy);
11767a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner      writeOperand(Operand, false);
11777a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    } else {
11787a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner      writeOperand(Operand, true);
11797a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner    }
11807a012299ced5cff02cec47055a63d3b2a78bb36fChris Lattner
11810313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << '(';
11827e70829632f82de15db187845666aaca6e04b792Chris Lattner    if (I.getNumOperands() > 3) writeOperand(I.getOperand(3), true);
11837e70829632f82de15db187845666aaca6e04b792Chris Lattner    for (unsigned op = 4, Eop = I.getNumOperands(); op < Eop; ++op) {
11840313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      Out << ',';
11857e70829632f82de15db187845666aaca6e04b792Chris Lattner      writeOperand(I.getOperand(op), true);
1186e02fa8551d20081534afa46e0976811687e5183aChris Lattner    }
1187e02fa8551d20081534afa46e0976811687e5183aChris Lattner
11880313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << " )\n\t\t\tto";
1189e02fa8551d20081534afa46e0976811687e5183aChris Lattner    writeOperand(II->getNormalDest(), true);
11900313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << " unwind";
1191aeb2a1d70807aa626f335fb23d47bc604ffeaa15Chris Lattner    writeOperand(II->getUnwindDest(), true);
1192e02fa8551d20081534afa46e0976811687e5183aChris Lattner
11937e70829632f82de15db187845666aaca6e04b792Chris Lattner  } else if (const AllocationInst *AI = dyn_cast<AllocationInst>(&I)) {
11940313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << ' ';
119594dc1f29cd1dc669680b7d3caa1f59862a656b83Chris Lattner    printType(AI->getType()->getElementType());
119694dc1f29cd1dc669680b7d3caa1f59862a656b83Chris Lattner    if (AI->isArrayAllocation()) {
11970313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      Out << ',';
119894dc1f29cd1dc669680b7d3caa1f59862a656b83Chris Lattner      writeOperand(AI->getArraySize(), true);
1199009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner    }
120014b0529532904b9e5a1e34526b4a3209f3e5bc62Nate Begeman    if (AI->getAlignment()) {
12019fad0b9974beb76bef763b8fed71d0c8644403d4Chris Lattner      Out << ", align " << AI->getAlignment();
120214b0529532904b9e5a1e34526b4a3209f3e5bc62Nate Begeman    }
1203e02fa8551d20081534afa46e0976811687e5183aChris Lattner  } else if (isa<CastInst>(I)) {
120441495a21f9e881a2bea017caad9eed731f8f37e8Chris Lattner    if (Operand) writeOperand(Operand, true);   // Work with broken code
12050313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << " to ";
12067e70829632f82de15db187845666aaca6e04b792Chris Lattner    printType(I.getType());
12074d45bd007d0c3a3a6e5b3876b67fd495acb0db2bChris Lattner  } else if (isa<VAArgInst>(I)) {
120841495a21f9e881a2bea017caad9eed731f8f37e8Chris Lattner    if (Operand) writeOperand(Operand, true);   // Work with broken code
12090313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman    Out << ", ";
12108f77daef04355c00b78b645f5aae5694e7a8b631Chris Lattner    printType(I.getType());
1211009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  } else if (Operand) {   // Print the normal way...
1212009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
1213fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman    // PrintAllTypes - Instructions who have operands of all the same type
1214009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner    // omit the type from all but the first operand.  If the instruction has
1215009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner    // different type operands (for example br), then they are all printed.
1216009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner    bool PrintAllTypes = false;
1217009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner    const Type *TheType = Operand->getType();
1218009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
1219cfdd148972c049291ee6f24f3201a573d9ac5809Chris Lattner    // Shift Left & Right print both types even for Ubyte LHS, and select prints
1220cfdd148972c049291ee6f24f3201a573d9ac5809Chris Lattner    // types even if all operands are bools.
1221fe0343a1cdb6f77efa5df24b22626b92d3871d39Chris Lattner    if (isa<ShiftInst>(I) || isa<SelectInst>(I) || isa<StoreInst>(I)) {
1222ffd9bf404cd36c93d82b0080113ccc6d230915b3Chris Lattner      PrintAllTypes = true;
1223ffd9bf404cd36c93d82b0080113ccc6d230915b3Chris Lattner    } else {
1224ffd9bf404cd36c93d82b0080113ccc6d230915b3Chris Lattner      for (unsigned i = 1, E = I.getNumOperands(); i != E; ++i) {
1225ffd9bf404cd36c93d82b0080113ccc6d230915b3Chris Lattner        Operand = I.getOperand(i);
1226ffd9bf404cd36c93d82b0080113ccc6d230915b3Chris Lattner        if (Operand->getType() != TheType) {
1227ffd9bf404cd36c93d82b0080113ccc6d230915b3Chris Lattner          PrintAllTypes = true;    // We have differing types!  Print them all!
1228ffd9bf404cd36c93d82b0080113ccc6d230915b3Chris Lattner          break;
1229ffd9bf404cd36c93d82b0080113ccc6d230915b3Chris Lattner        }
1230009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner      }
1231009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner    }
1232fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman
1233c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner    if (!PrintAllTypes) {
12340313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      Out << ' ';
1235ffd9bf404cd36c93d82b0080113ccc6d230915b3Chris Lattner      printType(TheType);
1236c1824996451d5ea93864fcd34ba8ea5e5f03e368Chris Lattner    }
1237009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
12387e70829632f82de15db187845666aaca6e04b792Chris Lattner    for (unsigned i = 0, E = I.getNumOperands(); i != E; ++i) {
12390313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman      if (i) Out << ',';
12407e70829632f82de15db187845666aaca6e04b792Chris Lattner      writeOperand(I.getOperand(i), PrintAllTypes);
1241009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner    }
1242009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner  }
1243009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
1244e02fa8551d20081534afa46e0976811687e5183aChris Lattner  printInfoComment(I);
12450313e0b44dcd77acd334d3f324c7da527179a690Misha Brukman  Out << "\n";
1246009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner}
1247009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
1248009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
1249009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner//===----------------------------------------------------------------------===//
1250009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner//                       External Interface declarations
1251009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner//===----------------------------------------------------------------------===//
1252009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
125395e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattnervoid Module::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
12540d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SlotMachine SlotTable(this);
125595e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattner  AssemblyWriter W(o, SlotTable, this, AAW);
125675cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner  W.write(this);
1257009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner}
1258009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
125975cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattnervoid GlobalVariable::print(std::ostream &o) const {
12600d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SlotMachine SlotTable(getParent());
126195e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattner  AssemblyWriter W(o, SlotTable, getParent(), 0);
126275cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner  W.write(this);
1263b0e4523624fbe493a945ea73a8f3c6d6526f3f27Chris Lattner}
1264b0e4523624fbe493a945ea73a8f3c6d6526f3f27Chris Lattner
126595e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattnervoid Function::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
12660d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SlotMachine SlotTable(getParent());
126795e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattner  AssemblyWriter W(o, SlotTable, getParent(), AAW);
1268009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
126975cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner  W.write(this);
1270009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner}
1271009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
127295e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattnervoid BasicBlock::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
12730d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SlotMachine SlotTable(getParent());
1274fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  AssemblyWriter W(o, SlotTable,
127595e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattner                   getParent() ? getParent()->getParent() : 0, AAW);
127675cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner  W.write(this);
127775cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner}
1278009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
127995e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattnervoid Instruction::print(std::ostream &o, AssemblyAnnotationWriter *AAW) const {
128075cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner  const Function *F = getParent() ? getParent()->getParent() : 0;
12810d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SlotMachine SlotTable(F);
128295e5a2cc3fc55b262d2baa1cc417115dfa4a18eaChris Lattner  AssemblyWriter W(o, SlotTable, F ? F->getParent() : 0, AAW);
1283009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
128475cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner  W.write(this);
128575cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner}
1286009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
128775cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattnervoid Constant::print(std::ostream &o) const {
128875cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner  if (this == 0) { o << "<null> constant value\n"; return; }
12893bc06b33dac11e9372b06484139b4d4c11788273Chris Lattner
129040c732cc52a5c71cead461646667ef5b13801923Misha Brukman  o << ' ' << getType()->getDescription() << ' ';
129166e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner
12927b13f56c36eb7572e1053974e5487a5f1c3080aaChris Lattner  std::map<const Type *, std::string> TypeTable;
129366e810be9f97fa744ca949d54b5d54358add84b8Chris Lattner  WriteConstantInt(o, this, false, TypeTable, 0);
1294009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner}
1295009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
1296fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukmanvoid Type::print(std::ostream &o) const {
129775cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner  if (this == 0)
129875cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner    o << "<null Type>";
129975cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner  else
130075cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner    o << getDescription();
1301009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner}
1302009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
130373e214244f2403b5ba0ef81b8839600f3c8ffebcChris Lattnervoid Argument::print(std::ostream &o) const {
1304144d9baf5e5665ab1b0e815402d92ee615034b01Chris Lattner  WriteAsOperand(o, this, true, true,
1305144d9baf5e5665ab1b0e815402d92ee615034b01Chris Lattner                 getParent() ? getParent()->getParent() : 0);
130675cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner}
1307009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
1308fa452c05504ebee698e00050d04e1a43b0f21045Reid Spencer// Value::dump - allow easy printing of  Values from the debugger.
1309fa452c05504ebee698e00050d04e1a43b0f21045Reid Spencer// Located here because so much of the needed functionality is here.
131075cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattnervoid Value::dump() const { print(std::cerr); }
1311fa452c05504ebee698e00050d04e1a43b0f21045Reid Spencer
1312fa452c05504ebee698e00050d04e1a43b0f21045Reid Spencer// Type::dump - allow easy printing of  Values from the debugger.
1313fa452c05504ebee698e00050d04e1a43b0f21045Reid Spencer// Located here because so much of the needed functionality is here.
13149231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencervoid Type::dump() const { print(std::cerr); }
1315009505452b713ed2e3a8e99c5545a6e721c65495Chris Lattner
131675cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner//===----------------------------------------------------------------------===//
131775cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner//  CachedWriter Class Implementation
131875cf7cf00d391ac6ca22d6240fa9d99ed427d1d5Chris Lattner//===----------------------------------------------------------------------===//
1319da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner
1320da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattnervoid CachedWriter::setModule(const Module *M) {
1321da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner  delete SC; delete AW;
1322da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner  if (M) {
13230d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    SC = new SlotMachine(M );
132440c732cc52a5c71cead461646667ef5b13801923Misha Brukman    AW = new AssemblyWriter(Out, *SC, M, 0);
1325da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner  } else {
1326da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner    SC = 0; AW = 0;
1327da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner  }
1328da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner}
1329da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner
1330da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris LattnerCachedWriter::~CachedWriter() {
1331da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner  delete AW;
1332da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner  delete SC;
1333da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner}
1334da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner
13350a8e8e1a4ea46fa5da067369ac43d8a459d0cac0Chris LattnerCachedWriter &CachedWriter::operator<<(const Value &V) {
1336da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner  assert(AW && SC && "CachedWriter does not have a current module!");
13370a8e8e1a4ea46fa5da067369ac43d8a459d0cac0Chris Lattner  if (const Instruction *I = dyn_cast<Instruction>(&V))
1338fae098a56b403e91affcb44de7e981fc9f34ea12Chris Lattner    AW->write(I);
13390a8e8e1a4ea46fa5da067369ac43d8a459d0cac0Chris Lattner  else if (const BasicBlock *BB = dyn_cast<BasicBlock>(&V))
1340fae098a56b403e91affcb44de7e981fc9f34ea12Chris Lattner    AW->write(BB);
13410a8e8e1a4ea46fa5da067369ac43d8a459d0cac0Chris Lattner  else if (const Function *F = dyn_cast<Function>(&V))
1342fae098a56b403e91affcb44de7e981fc9f34ea12Chris Lattner    AW->write(F);
13430a8e8e1a4ea46fa5da067369ac43d8a459d0cac0Chris Lattner  else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(&V))
1344fae098a56b403e91affcb44de7e981fc9f34ea12Chris Lattner    AW->write(GV);
1345fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  else
13460a8e8e1a4ea46fa5da067369ac43d8a459d0cac0Chris Lattner    AW->writeOperand(&V, true, true);
1347da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner  return *this;
1348da1fbcc5c389e86840f6131afc22b45ce7e9ad08Chris Lattner}
13495cf1acff3c32e52d79ea8b0b47e341d4e975ed8cMisha Brukman
13500a8e8e1a4ea46fa5da067369ac43d8a459d0cac0Chris LattnerCachedWriter& CachedWriter::operator<<(const Type &Ty) {
13515cf1acff3c32e52d79ea8b0b47e341d4e975ed8cMisha Brukman  if (SymbolicTypes) {
13525cf1acff3c32e52d79ea8b0b47e341d4e975ed8cMisha Brukman    const Module *M = AW->getModule();
13530a8e8e1a4ea46fa5da067369ac43d8a459d0cac0Chris Lattner    if (M) WriteTypeSymbolic(Out, &Ty, M);
13540e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  } else {
13550a8e8e1a4ea46fa5da067369ac43d8a459d0cac0Chris Lattner    AW->write(&Ty);
13560e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  }
13570e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  return *this;
13585cf1acff3c32e52d79ea8b0b47e341d4e975ed8cMisha Brukman}
1359e5242def97c52077b4e0a77a501f324867f7bce5Misha Brukman
13600d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer//===----------------------------------------------------------------------===//
13610d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer//===--                    SlotMachine Implementation
13620d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer//===----------------------------------------------------------------------===//
13630d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
13640d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer#if 0
13650d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer#define SC_DEBUG(X) std::cerr << X
13660d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer#else
13670d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer#define SC_DEBUG(X)
13680d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer#endif
13690d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
13700d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer// Module level constructor. Causes the contents of the Module (sans functions)
13710d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer// to be added to the slot table.
1372fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha BrukmanSlotMachine::SlotMachine(const Module *M)
1373b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  : TheModule(M)    ///< Saved for lazy initialization.
1374b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  , TheFunction(0)
137528531c71cd142f623d56a7ea13df2e014eebff54Reid Spencer  , FunctionProcessed(false)
13760d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  , mMap()
13770e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  , mTypes()
13780d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  , fMap()
13790e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  , fTypes()
13800d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer{
13810d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer}
13820d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
13830d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer// Function level constructor. Causes the contents of the Module and the one
13840d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer// function provided to be added to the slot table.
1385fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha BrukmanSlotMachine::SlotMachine(const Function *F )
1386b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  : TheModule( F ? F->getParent() : 0 ) ///< Saved for lazy initialization
1387b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  , TheFunction(F) ///< Saved for lazy initialization
138828531c71cd142f623d56a7ea13df2e014eebff54Reid Spencer  , FunctionProcessed(false)
13890d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  , mMap()
13900e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  , mTypes()
13910d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  , fMap()
13920e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  , fTypes()
13930d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer{
1394b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer}
1395b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer
1396b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencerinline void SlotMachine::initialize(void) {
1397fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  if ( TheModule) {
1398fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman    processModule();
1399b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer    TheModule = 0; ///< Prevent re-processing next time we're called.
1400b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  }
1401fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  if ( TheFunction && ! FunctionProcessed) {
1402fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman    processFunction();
14030d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  }
14040d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer}
14050d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
14060d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer// Iterate through all the global variables, functions, and global
1407fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman// variable initializers and create slots for them.
14080d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencervoid SlotMachine::processModule() {
14090d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SC_DEBUG("begin processModule!\n");
14100d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
14110d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  // Add all of the global variables to the value table...
1412e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  for (Module::const_global_iterator I = TheModule->global_begin(), E = TheModule->global_end();
14130d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer       I != E; ++I)
14140d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    createSlot(I);
14150d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
14160d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  // Add all the functions to the table
14170d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  for (Module::const_iterator I = TheModule->begin(), E = TheModule->end();
14180d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer       I != E; ++I)
14190d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    createSlot(I);
14200d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
14210d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SC_DEBUG("end processModule!\n");
14220d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer}
14230d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
14240d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1425b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer// Process the arguments, basic blocks, and instructions  of a function.
1426b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencervoid SlotMachine::processFunction() {
14270d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SC_DEBUG("begin processFunction!\n");
14280d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
14290d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  // Add all the function arguments
1430fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  for(Function::const_arg_iterator AI = TheFunction->arg_begin(),
1431e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner      AE = TheFunction->arg_end(); AI != AE; ++AI)
14320d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    createSlot(AI);
14330d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
14340d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SC_DEBUG("Inserting Instructions:\n");
14350d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
14360d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  // Add all of the basic blocks and instructions
1437fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  for (Function::const_iterator BB = TheFunction->begin(),
1438b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer       E = TheFunction->end(); BB != E; ++BB) {
14390d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    createSlot(BB);
14400d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); I!=E; ++I) {
14410d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      createSlot(I);
14420d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    }
14430d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  }
14440d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
144528531c71cd142f623d56a7ea13df2e014eebff54Reid Spencer  FunctionProcessed = true;
144628531c71cd142f623d56a7ea13df2e014eebff54Reid Spencer
14470d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SC_DEBUG("end processFunction!\n");
14480d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer}
14490d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
14500d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer// Clean up after incorporating a function. This is the only way
1451b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer// to get out of the function incorporation state that affects the
1452b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer// getSlot/createSlot lock. Function incorporation state is indicated
1453b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer// by TheFunction != 0.
14540d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencervoid SlotMachine::purgeFunction() {
14550d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SC_DEBUG("begin purgeFunction!\n");
14560d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  fMap.clear(); // Simply discard the function level map
14570e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  fTypes.clear();
1458b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  TheFunction = 0;
145928531c71cd142f623d56a7ea13df2e014eebff54Reid Spencer  FunctionProcessed = false;
14600d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SC_DEBUG("end purgeFunction!\n");
14610d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer}
14620d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
14630d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// Get the slot number for a value. This function will assert if you
14640d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// ask for a Value that hasn't previously been inserted with createSlot.
14650d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer/// Types are forbidden because Type does not inherit from Value (any more).
146669566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattnerint SlotMachine::getSlot(const Value *V) {
14670d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  assert( V && "Can't get slot for null Value" );
1468fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1469fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman    "Can't insert a non-GlobalValue Constant into SlotMachine");
1470b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer
1471b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  // Check for uninitialized state and do lazy initialization
1472b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  this->initialize();
14730d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
14740d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  // Get the type of the value
14750d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  const Type* VTy = V->getType();
14760d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
14770d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  // Find the type plane in the module map
14780d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  TypedPlanes::const_iterator MI = mMap.find(VTy);
14790d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1480b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  if ( TheFunction ) {
14810d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    // Lookup the type in the function map too
14820d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    TypedPlanes::const_iterator FI = fMap.find(VTy);
14830d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    // If there is a corresponding type plane in the function map
14840d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    if ( FI != fMap.end() ) {
14850d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      // Lookup the Value in the function map
14860d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      ValueMap::const_iterator FVI = FI->second.map.find(V);
14870d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      // If the value doesn't exist in the function map
14880d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      if ( FVI == FI->second.map.end() ) {
1489e9e326e2eb223ea4bc6762a099ce3c314e4daedbChris Lattner        // Look up the value in the module map.
1490e9e326e2eb223ea4bc6762a099ce3c314e4daedbChris Lattner        if (MI == mMap.end()) return -1;
1491b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        ValueMap::const_iterator MVI = MI->second.map.find(V);
1492b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        // If we didn't find it, it wasn't inserted
149369566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner        if (MVI == MI->second.map.end()) return -1;
1494b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        assert( MVI != MI->second.map.end() && "Value not found");
1495b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        // We found it only at the module level
1496fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman        return MVI->second;
14970d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
14980d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      // else the value exists in the function map
14990d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      } else {
1500b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        // Return the slot number as the module's contribution to
1501b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        // the type plane plus the index in the function's contribution
1502b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        // to the type plane.
1503d1cd3285601d4accf370779ac520c1e494a08b87Chris Lattner        if (MI != mMap.end())
1504d1cd3285601d4accf370779ac520c1e494a08b87Chris Lattner          return MI->second.next_slot + FVI->second;
1505d1cd3285601d4accf370779ac520c1e494a08b87Chris Lattner        else
1506d1cd3285601d4accf370779ac520c1e494a08b87Chris Lattner          return FVI->second;
15070d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      }
15080d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    }
15090d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  }
15100d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1511fc621e28781e3cf22f1850e07f285581334e2402Reid Spencer  // N.B. Can get here only if either !TheFunction or the function doesn't
1512fc621e28781e3cf22f1850e07f285581334e2402Reid Spencer  // have a corresponding type plane for the Value
15130d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
15140d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  // Make sure the type plane exists
151569566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner  if (MI == mMap.end()) return -1;
15160d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  // Lookup the value in the module's map
15170d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  ValueMap::const_iterator MVI = MI->second.map.find(V);
15180d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  // Make sure we found it.
151969566459cc27c5210472a87ae2fe7b2ca64f9fc8Chris Lattner  if (MVI == MI->second.map.end()) return -1;
15200d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  // Return it.
15210d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  return MVI->second;
15220d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer}
15230d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
15240e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer/// Get the slot number for a value. This function will assert if you
15250e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer/// ask for a Value that hasn't previously been inserted with createSlot.
15260e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer/// Types are forbidden because Type does not inherit from Value (any more).
15270e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencerint SlotMachine::getSlot(const Type *Ty) {
15280e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  assert( Ty && "Can't get slot for null Type" );
15290e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
15300e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  // Check for uninitialized state and do lazy initialization
15310e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  this->initialize();
15320e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
15330e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  if ( TheFunction ) {
15340e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    // Lookup the Type in the function map
15350e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    TypeMap::const_iterator FTI = fTypes.map.find(Ty);
15360e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    // If the Type doesn't exist in the function map
15370e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    if ( FTI == fTypes.map.end() ) {
15380e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      TypeMap::const_iterator MTI = mTypes.map.find(Ty);
15390e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      // If we didn't find it, it wasn't inserted
1540fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman      if (MTI == mTypes.map.end())
15410e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer        return -1;
15420e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      // We found it only at the module level
1543fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman      return MTI->second;
15440e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
15450e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    // else the value exists in the function map
15460e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    } else {
15470e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      // Return the slot number as the module's contribution to
15480e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      // the type plane plus the index in the function's contribution
15490e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      // to the type plane.
15500e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      return mTypes.next_slot + FTI->second;
15510e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    }
15520e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  }
15530e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
15540e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  // N.B. Can get here only if either !TheFunction
15550e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
15560e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  // Lookup the value in the module's map
15570e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  TypeMap::const_iterator MTI = mTypes.map.find(Ty);
15580e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  // Make sure we found it.
15590e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  if (MTI == mTypes.map.end()) return -1;
15600e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  // Return it.
15610e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  return MTI->second;
15620e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer}
15630e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
15640d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer// Create a new slot, or return the existing slot if it is already
15650d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer// inserted. Note that the logic here parallels getSlot but instead
15660d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer// of asserting when the Value* isn't found, it inserts the value.
15670d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencerunsigned SlotMachine::createSlot(const Value *V) {
15680d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  assert( V && "Can't insert a null Value to SlotMachine");
1569fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1570fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman    "Can't insert a non-GlobalValue Constant into SlotMachine");
15710d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
15720d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  const Type* VTy = V->getType();
15730d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
15740d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  // Just ignore void typed things
15750d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  if (VTy == Type::VoidTy) return 0; // FIXME: Wrong return value!
15760d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
15770d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  // Look up the type plane for the Value's type from the module map
15780d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  TypedPlanes::const_iterator MI = mMap.find(VTy);
15790d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1580b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  if ( TheFunction ) {
15810d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    // Get the type plane for the Value's type from the function map
15820d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    TypedPlanes::const_iterator FI = fMap.find(VTy);
15830d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    // If there is a corresponding type plane in the function map
15840d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    if ( FI != fMap.end() ) {
15850d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      // Lookup the Value in the function map
15860d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      ValueMap::const_iterator FVI = FI->second.map.find(V);
15870d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      // If the value doesn't exist in the function map
15880d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      if ( FVI == FI->second.map.end() ) {
1589b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        // If there is no corresponding type plane in the module map
1590b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        if ( MI == mMap.end() )
1591b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer          return insertValue(V);
1592b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        // Look up the value in the module map
1593b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        ValueMap::const_iterator MVI = MI->second.map.find(V);
1594b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        // If we didn't find it, it wasn't inserted
1595b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        if ( MVI == MI->second.map.end() )
1596b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer          return insertValue(V);
1597b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        else
1598b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer          // We found it only at the module level
1599b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer          return MVI->second;
16000d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
16010d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      // else the value exists in the function map
16020d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      } else {
1603b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        if ( MI == mMap.end() )
1604b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer          return FVI->second;
1605b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        else
1606b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer          // Return the slot number as the module's contribution to
1607b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer          // the type plane plus the index in the function's contribution
1608b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer          // to the type plane.
1609b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer          return MI->second.next_slot + FVI->second;
16100d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      }
16110d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
16120d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    // else there is not a corresponding type plane in the function map
16130d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    } else {
16140d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      // If the type plane doesn't exists at the module level
16150d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      if ( MI == mMap.end() ) {
1616b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        return insertValue(V);
16170d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      // else type plane exists at the module level, examine it
16180d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      } else {
1619b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        // Look up the value in the module's map
1620b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        ValueMap::const_iterator MVI = MI->second.map.find(V);
1621b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        // If we didn't find it there either
1622b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        if ( MVI == MI->second.map.end() )
1623b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer          // Return the slot number as the module's contribution to
1624b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer          // the type plane plus the index of the function map insertion.
1625b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer          return MI->second.next_slot + insertValue(V);
1626b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer        else
1627b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer          return MVI->second;
16280d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      }
16290d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    }
16300d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  }
16310d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1632b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  // N.B. Can only get here if !TheFunction
16330d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
16340d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  // If the module map's type plane is not for the Value's type
16350d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  if ( MI != mMap.end() ) {
16360d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    // Lookup the value in the module's map
16370d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    ValueMap::const_iterator MVI = MI->second.map.find(V);
1638fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman    if ( MVI != MI->second.map.end() )
16390d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      return MVI->second;
16400d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  }
16410d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
16420d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  return insertValue(V);
16430d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer}
16440d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
16450e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer// Create a new slot, or return the existing slot if it is already
16460e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer// inserted. Note that the logic here parallels getSlot but instead
16470e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer// of asserting when the Value* isn't found, it inserts the value.
16480e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencerunsigned SlotMachine::createSlot(const Type *Ty) {
16490e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  assert( Ty && "Can't insert a null Type to SlotMachine");
16500e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
16510e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  if ( TheFunction ) {
16520e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    // Lookup the Type in the function map
16530e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    TypeMap::const_iterator FTI = fTypes.map.find(Ty);
16540e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    // If the type doesn't exist in the function map
16550e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    if ( FTI == fTypes.map.end() ) {
16560e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      // Look up the type in the module map
16570e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      TypeMap::const_iterator MTI = mTypes.map.find(Ty);
16580e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      // If we didn't find it, it wasn't inserted
16590e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      if ( MTI == mTypes.map.end() )
16600e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer        return insertValue(Ty);
16610e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      else
16620e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer        // We found it only at the module level
16630e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer        return MTI->second;
16640e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
16650e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    // else the value exists in the function map
16660e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    } else {
16670e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      // Return the slot number as the module's contribution to
16680e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      // the type plane plus the index in the function's contribution
16690e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      // to the type plane.
16700e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      return mTypes.next_slot + FTI->second;
16710e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    }
16720e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  }
16730e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
16740e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  // N.B. Can only get here if !TheFunction
16750e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
16760e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  // Lookup the type in the module's map
16770e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  TypeMap::const_iterator MTI = mTypes.map.find(Ty);
1678fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  if ( MTI != mTypes.map.end() )
16790e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    return MTI->second;
16800e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
16810e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  return insertValue(Ty);
16820e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer}
16830d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
16840d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer// Low level insert function. Minimal checking is done. This
16850d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer// function is just for the convenience of createSlot (above).
16860d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencerunsigned SlotMachine::insertValue(const Value *V ) {
16870d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  assert(V && "Can't insert a null Value into SlotMachine!");
1688fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  assert(!isa<Constant>(V) || isa<GlobalValue>(V) &&
1689fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman    "Can't insert a non-GlobalValue Constant into SlotMachine");
16900d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1691b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  // If this value does not contribute to a plane (is void)
1692fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  // or if the value already has a name then ignore it.
1693b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  if (V->getType() == Type::VoidTy || V->hasName() ) {
16940d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      SC_DEBUG("ignored value " << *V << "\n");
16950d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer      return 0;   // FIXME: Wrong return value
16960d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  }
16970d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
16980d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  const Type *VTy = V->getType();
16990d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  unsigned DestSlot = 0;
17000d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1701b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer  if ( TheFunction ) {
17020d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    TypedPlanes::iterator I = fMap.find( VTy );
1703fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman    if ( I == fMap.end() )
17040e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      I = fMap.insert(std::make_pair(VTy,ValuePlane())).first;
17050d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    DestSlot = I->second.map[V] = I->second.next_slot++;
17060d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  } else {
17070d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    TypedPlanes::iterator I = mMap.find( VTy );
17080d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    if ( I == mMap.end() )
17090e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer      I = mMap.insert(std::make_pair(VTy,ValuePlane())).first;
17100d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer    DestSlot = I->second.map[V] = I->second.next_slot++;
17110d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  }
17120d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
1713fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  SC_DEBUG("  Inserting value [" << VTy << "] = " << V << " slot=" <<
1714b03de0c780a74d63f6b331eb8f615b46c4cddd7bReid Spencer           DestSlot << " [");
17150d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  // G = Global, C = Constant, T = Type, F = Function, o = other
1716fd93908ae8b9684fe71c239e3c6cfe13ff6a2663Misha Brukman  SC_DEBUG((isa<GlobalVariable>(V) ? 'G' : (isa<Function>(V) ? 'F' :
17177970396014eacbe719eb171448ddc546c1ad2289Reid Spencer           (isa<Constant>(V) ? 'C' : 'o'))));
17180d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  SC_DEBUG("]\n");
17190d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer  return DestSlot;
17200d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer}
17210d1b77e2824115d97f8392c3f4ec257b27ebbd7aReid Spencer
17220e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer// Low level insert function. Minimal checking is done. This
17230e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer// function is just for the convenience of createSlot (above).
17240e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencerunsigned SlotMachine::insertValue(const Type *Ty ) {
17250e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  assert(Ty && "Can't insert a null Type into SlotMachine!");
17260e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
17270e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  unsigned DestSlot = 0;
17280e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
17290e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  if ( TheFunction ) {
17300e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    DestSlot = fTypes.map[Ty] = fTypes.next_slot++;
17310e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  } else {
17320e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer    DestSlot = fTypes.map[Ty] = fTypes.next_slot++;
17330e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  }
17340e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  SC_DEBUG("  Inserting type [" << DestSlot << "] = " << Ty << "\n");
17350e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer  return DestSlot;
17360e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer}
17370e25e1cdcf1f73bb9f56b91155d5c0ef42ef0a44Reid Spencer
17389231ac8b6f2c3f9877bdb7a223f7392061258ab6Reid Spencer// vim: sw=2
1739