include/llvm/SlotCalculator.h

   1 //===-- llvm/SlotCalculator.h - Calculate value slots -----------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the LLVM research group and is distributed under
   6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This class calculates the slots that values will land in.  This is useful for
  11 // when writing bytecode or assembly out, because you have to know these things.
  12 //
  13 // Specifically, this class calculates the "type plane numbering" that you see
  14 // for a function if you strip out all of the symbols in it.  For assembly
  15 // writing, this is used when a symbol does not have a name.  For bytecode
  16 // writing, this is always used, and the symbol table is added on later.
  17 //
  18 //===----------------------------------------------------------------------===//
  19
  20 #ifndef LLVM_SLOTCALCULATOR_H
  21 #define LLVM_SLOTCALCULATOR_H
  22
  23 #include <vector>
  24 #include <map>
  25
  26 namespace llvm {
  27
  28 class Value;
  29 class Module;
  30 class Function;
  31 class SymbolTable;
  32
  33 class SlotCalculator {
  34   const Module *TheModule;
  35   bool IgnoreNamedNodes;     // Shall we not count named nodes?
  36
  37   typedef std::vector<const Value*> TypePlane;
  38   std::vector<TypePlane> Table;
  39   std::map<const Value *, unsigned> NodeMap;
  40
  41   // ModuleLevel - Used to keep track of which values belong to the module,
  42   // and which values belong to the currently incorporated function.
  43   //
  44   std::vector<unsigned> ModuleLevel;
  45
  46 public:
  47   SlotCalculator(const Module *M, bool IgnoreNamed);
  48   // Start out in incorp state
  49   SlotCalculator(const Function *M, bool IgnoreNamed);
  50   inline ~SlotCalculator() {}
  51
  52   // getSlot returns < 0 on error!
  53   int getSlot(const Value *D) const;
  54
  55   inline unsigned getNumPlanes() const { return Table.size(); }
  56   inline unsigned getModuleLevel(unsigned Plane) const {
  57     return Plane < ModuleLevel.size() ? ModuleLevel[Plane] : 0;
  58   }
  59
  60   inline const TypePlane &getPlane(unsigned Plane) const {
  61     return Table[Plane];
  62   }
  63
  64   // If you'd like to deal with a function, use these two methods to get its
  65   // data into the SlotCalculator!
  66   //
  67   void incorporateFunction(const Function *F);
  68   void purgeFunction();
  69
  70 protected:
  71   // getOrCreateSlot - Values can be crammed into here at will... if
  72   // they haven't been inserted already, they get inserted, otherwise
  73   // they are ignored.
  74   //
  75   int getOrCreateSlot(const Value *D);
  76
  77   // insertValue - Insert a value into the value table... Return the
  78   // slot that it occupies, or -1 if the declaration is to be ignored
  79   // because of the IgnoreNamedNodes flag.
  80   //
  81   int insertValue(const Value *D, bool dontIgnore = false);
  82
  83   // doInsertValue - Small helper function to be called only be insertVal.
  84   int doInsertValue(const Value *D);
  85
  86   // processModule - Process all of the module level function declarations and
  87   // types that are available.
  88   //
  89   void processModule();
  90
  91   // processSymbolTable - Insert all of the values in the specified symbol table
  92   // into the values table...
  93   //
  94   void processSymbolTable(const SymbolTable *ST);
  95   void processSymbolTableConstants(const SymbolTable *ST);
  96 };
  97
  98 } // End llvm namespace
  99
 100 #endif