include/llvm/TypeSymbolTable.h

   1 //===-- llvm/TypeSymbolTable.h - Implement a Type Symtab --------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file implements the name/type symbol table for LLVM.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #ifndef LLVM_TYPE_SYMBOL_TABLE_H
  15 #define LLVM_TYPE_SYMBOL_TABLE_H
  16
  17 #include "llvm/Type.h"
  18 #include <map>
  19
  20 namespace llvm {
  21
  22 class StringRef;
  23
  24 /// This class provides a symbol table of name/type pairs with operations to
  25 /// support constructing, searching and iterating over the symbol table. The
  26 /// class derives from AbstractTypeUser so that the contents of the symbol
  27 /// table can be updated when abstract types become concrete.
  28 class TypeSymbolTable : public AbstractTypeUser {
  29
  30 /// @name Types
  31 /// @{
  32 public:
  33
  34   /// @brief A mapping of names to types.
  35   typedef std::map<const std::string, const Type*> TypeMap;
  36
  37   /// @brief An iterator over the TypeMap.
  38   typedef TypeMap::iterator iterator;
  39
  40   /// @brief A const_iterator over the TypeMap.
  41   typedef TypeMap::const_iterator const_iterator;
  42
  43 /// @}
  44 /// @name Constructors
  45 /// @{
  46 public:
  47
  48   TypeSymbolTable():LastUnique(0) {}
  49   ~TypeSymbolTable();
  50
  51 /// @}
  52 /// @name Accessors
  53 /// @{
  54 public:
  55
  56   /// Generates a unique name for a type based on the \p BaseName by
  57   /// incrementing an integer and appending it to the name, if necessary
  58   /// @returns the unique name
  59   /// @brief Get a unique name for a type
  60   std::string getUniqueName(const StringRef &BaseName) const;
  61
  62   /// This method finds the type with the given \p name in the type map
  63   /// and returns it.
  64   /// @returns null if the name is not found, otherwise the Type
  65   /// associated with the \p name.
  66   /// @brief Lookup a type by name.
  67   Type *lookup(const StringRef &name) const;
  68
  69   /// @returns true iff the symbol table is empty.
  70   /// @brief Determine if the symbol table is empty
  71   inline bool empty() const { return tmap.empty(); }
  72
  73   /// @returns the size of the symbol table
  74   /// @brief The number of name/type pairs is returned.
  75   inline unsigned size() const { return unsigned(tmap.size()); }
  76
  77   /// This function can be used from the debugger to display the
  78   /// content of the symbol table while debugging.
  79   /// @brief Print out symbol table on stderr
  80   void dump() const;
  81
  82 /// @}
  83 /// @name Iteration
  84 /// @{
  85 public:
  86   /// Get an iterator to the start of the symbol table
  87   inline iterator begin() { return tmap.begin(); }
  88
  89   /// @brief Get a const_iterator to the start of the symbol table
  90   inline const_iterator begin() const { return tmap.begin(); }
  91
  92   /// Get an iterator to the end of the symbol table.
  93   inline iterator end() { return tmap.end(); }
  94
  95   /// Get a const_iterator to the end of the symbol table.
  96   inline const_iterator end() const { return tmap.end(); }
  97
  98 /// @}
  99 /// @name Mutators
 100 /// @{
 101 public:
 102
 103   /// Inserts a type into the symbol table with the specified name. There can be
 104   /// a many-to-one mapping between names and types. This method allows a type
 105   /// with an existing entry in the symbol table to get a new name.
 106   /// @brief Insert a type under a new name.
 107   void insert(const StringRef &Name, const Type *Typ);
 108
 109   /// Remove a type at the specified position in the symbol table.
 110   /// @returns the removed Type.
 111   /// @returns the Type that was erased from the symbol table.
 112   Type* remove(iterator TI);
 113
 114 /// @}
 115 /// @name AbstractTypeUser Methods
 116 /// @{
 117 private:
 118   /// This function is called when one of the types in the type plane
 119   /// is refined.
 120   virtual void refineAbstractType(const DerivedType *OldTy, const Type *NewTy);
 121
 122   /// This function markes a type as being concrete (defined).
 123   virtual void typeBecameConcrete(const DerivedType *AbsTy);
 124
 125 /// @}
 126 /// @name Internal Data
 127 /// @{
 128 private:
 129   TypeMap tmap; ///< This is the mapping of names to types.
 130   mutable uint32_t LastUnique; ///< Counter for tracking unique names
 131
 132 /// @}
 133
 134 };
 135
 136 } // End llvm namespace
 137
 138 #endif