15f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===--- IdentifierTable.h - Hash table for identifier lookup ---*- C++ -*-===//
25f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
35f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//                     The LLVM Compiler Infrastructure
45f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
50bc735ffcfb223c0186419547abaa5c84482663eChris Lattner// This file is distributed under the University of Illinois Open Source
60bc735ffcfb223c0186419547abaa5c84482663eChris Lattner// License. See LICENSE.TXT for details.
75f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
85f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===//
95f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
10c7229c338c21ef26b01ef3ecf9eec4fd373fa9ecChris Lattner// This file defines the IdentifierInfo, IdentifierTable, and Selector
11c7229c338c21ef26b01ef3ecf9eec4fd373fa9ecChris Lattner// interfaces.
125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===//
145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
15c7229c338c21ef26b01ef3ecf9eec4fd373fa9ecChris Lattner#ifndef LLVM_CLANG_BASIC_IDENTIFIERTABLE_H
16c7229c338c21ef26b01ef3ecf9eec4fd373fa9ecChris Lattner#define LLVM_CLANG_BASIC_IDENTIFIERTABLE_H
175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
181cd1b1e987f5e2f060d7972b13d83239b36d77d6Douglas Gregor#include "clang/Basic/OperatorKinds.h"
195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "clang/Basic/TokenKinds.h"
20686775deca8b8685eb90801495880e3abdd844c2Chris Lattner#include "clang/Basic/LLVM.h"
215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "llvm/ADT/StringMap.h"
22700030ebddb987936d4fee14d9412821d96e4840Kovarththanan Rajaratnam#include "llvm/ADT/StringRef.h"
2372b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek#include "llvm/ADT/OwningPtr.h"
241734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor#include "llvm/Support/PointerLikeTypeTraits.h"
251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump#include <cassert>
26403ba3522d1b1c97ae5fad81c1a2c4b3a754e1c1Nick Lewycky#include <string>
275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
2885994260c41a54cab061a434ed378fc448333a4eChris Lattnernamespace llvm {
2985994260c41a54cab061a434ed378fc448333a4eChris Lattner  template <typename T> struct DenseMapInfo;
3085994260c41a54cab061a434ed378fc448333a4eChris Lattner}
3129238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroff
325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencernamespace clang {
33ea684e699ea84e61711e279f5fa7a1b9f3d46bc2Cedric Venet  class LangOptions;
347caeabd868d46cf4e68478c6e9136dae4e735d21Chris Lattner  class IdentifierInfo;
35ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek  class IdentifierTable;
367caeabd868d46cf4e68478c6e9136dae4e735d21Chris Lattner  class SourceLocation;
372e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor  class MultiKeywordSelector; // private class used by Selector
382e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor  class DeclarationName;      // AST class that stores declaration names
392e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor
407caeabd868d46cf4e68478c6e9136dae4e735d21Chris Lattner  /// IdentifierLocPair - A simple pair of identifier info and location.
417caeabd868d46cf4e68478c6e9136dae4e735d21Chris Lattner  typedef std::pair<IdentifierInfo*, SourceLocation> IdentifierLocPair;
421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// IdentifierInfo - One of these records is kept for each identifier that
455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// is lexed.  This contains information about whether the token was #define'd,
465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// is a language keyword, or if it is a front-end token of some sort (e.g. a
475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// variable or function name).  The preprocessor keeps this information in a
481eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump/// set, and all tok::identifier tokens have a pointer to one of these.
495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass IdentifierInfo {
505cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  unsigned TokenID            : 9; // Front-end token ID or tok::identifier.
515142af38ed0dd2f592cbfa00fa6e2e14dd6cc516Douglas Gregor  // Objective-C keyword ('protocol' in '@protocol') or builtin (__builtin_inf).
525142af38ed0dd2f592cbfa00fa6e2e14dd6cc516Douglas Gregor  // First NUM_OBJC_KEYWORDS values are for Objective-C, the remaining values
535142af38ed0dd2f592cbfa00fa6e2e14dd6cc516Douglas Gregor  // are for builtins.
54c81ca9852806e75972bbb246fa916152e8fba541Ted Kremenek  unsigned ObjCOrBuiltinID    :11;
554365a7e46822700357a272d839ee2656d9603d5aChris Lattner  bool HasMacro               : 1; // True if there is a #define for this.
565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  bool IsExtension            : 1; // True if identifier is a lang extension.
5798d86b98b3fd0bd9c546123b16fd9995509aaae1Richard Smith  bool IsCXX11CompatKeyword   : 1; // True if identifier is a keyword in C++11.
585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  bool IsPoisoned             : 1; // True if identifier is poisoned.
595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  bool IsCPPOperatorKeyword   : 1; // True if ident is a C++ operator keyword.
606a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  bool NeedsHandleIdentifier  : 1; // See "RecomputeNeedsHandleIdentifier".
61eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  bool IsFromAST              : 1; // True if identifier was loaded (at least
62eee242ff426bf79149f221798966e58688383c1eDouglas Gregor                                   // partially) from an AST file.
63eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  bool ChangedAfterLoad       : 1; // True if identifier has changed from the
64eee242ff426bf79149f221798966e58688383c1eDouglas Gregor                                   // definition loaded from an AST file.
65646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis  bool RevertedTokenID        : 1; // True if RevertTokenIDToIdentifier was
66646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis                                   // called.
67eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  bool OutOfDate              : 1; // True if there may be additional
68eee242ff426bf79149f221798966e58688383c1eDouglas Gregor                                   // information about this identifier
69eee242ff426bf79149f221798966e58688383c1eDouglas Gregor                                   // stored externally.
7032ad2ee2618745ce3da51c2ae066ed5f21157c07Ted Kremenek  bool IsModulesImport               : 1; // True if this is the 'import' contextual
71c13a34b690d2dc2a03c2fea75a0a1438636c19ceDouglas Gregor                                   // keyword.
72c13a34b690d2dc2a03c2fea75a0a1438636c19ceDouglas Gregor  // 1 bit left in 32-bit word.
73eee242ff426bf79149f221798966e58688383c1eDouglas Gregor
745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  void *FETokenInfo;               // Managed by the language front-end.
75ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek  llvm::StringMapEntry<IdentifierInfo*> *Entry;
761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  IdentifierInfo(const IdentifierInfo&);  // NONCOPYABLE.
784365a7e46822700357a272d839ee2656d9603d5aChris Lattner  void operator=(const IdentifierInfo&);  // NONASSIGNABLE.
79ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek
801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  friend class IdentifierTable;
81295a2a617ac335f590e430ab7fcd98f8ce109251Douglas Gregor
825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
83ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek  IdentifierInfo();
845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8692e62b02226410bcad8584541b8f1ff4d35ebab9Chris Lattner  /// isStr - Return true if this is the identifier for the specified string.
8792e62b02226410bcad8584541b8f1ff4d35ebab9Chris Lattner  /// This is intended to be used for string literals only: II->isStr("foo").
8892e62b02226410bcad8584541b8f1ff4d35ebab9Chris Lattner  template <std::size_t StrLen>
8992e62b02226410bcad8584541b8f1ff4d35ebab9Chris Lattner  bool isStr(const char (&Str)[StrLen]) const {
907fe60650c1133ee74a3395cf1063690e274bb7acDaniel Dunbar    return getLength() == StrLen-1 && !memcmp(getNameStart(), Str, StrLen-1);
91845222ccd992282bf74b2fca53e7c3b84a81c098Chris Lattner  }
921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
937fe60650c1133ee74a3395cf1063690e274bb7acDaniel Dunbar  /// getNameStart - Return the beginning of the actual string for this
947fe60650c1133ee74a3395cf1063690e274bb7acDaniel Dunbar  /// identifier.  The returned string is properly null terminated.
955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ///
967fe60650c1133ee74a3395cf1063690e274bb7acDaniel Dunbar  const char *getNameStart() const {
97ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek    if (Entry) return Entry->getKeyData();
987e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek    // FIXME: This is gross. It would be best not to embed specific details
997e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek    // of the PTH file format here.
1001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    // The 'this' pointer really points to a
101ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek    // std::pair<IdentifierInfo, const char*>, where internal pointer
102ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek    // points to the external string data.
10340844a8b5a89676fb61898d61ea4a7fa98eb9b6bGabor Greif    typedef std::pair<IdentifierInfo, const char*> actualtype;
10440844a8b5a89676fb61898d61ea4a7fa98eb9b6bGabor Greif    return ((const actualtype*) this)->second;
1051f945f632ae993bbb9f9e69d908f7526924c1a88Chris Lattner  }
1061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1071f945f632ae993bbb9f9e69d908f7526924c1a88Chris Lattner  /// getLength - Efficiently return the length of this identifier info.
1081f945f632ae993bbb9f9e69d908f7526924c1a88Chris Lattner  ///
1091f945f632ae993bbb9f9e69d908f7526924c1a88Chris Lattner  unsigned getLength() const {
110ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek    if (Entry) return Entry->getKeyLength();
1117e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek    // FIXME: This is gross. It would be best not to embed specific details
1127e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek    // of the PTH file format here.
1131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    // The 'this' pointer really points to a
114ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek    // std::pair<IdentifierInfo, const char*>, where internal pointer
115ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek    // points to the external string data.
11640844a8b5a89676fb61898d61ea4a7fa98eb9b6bGabor Greif    typedef std::pair<IdentifierInfo, const char*> actualtype;
11740844a8b5a89676fb61898d61ea4a7fa98eb9b6bGabor Greif    const char* p = ((const actualtype*) this)->second - 2;
1187fe60650c1133ee74a3395cf1063690e274bb7acDaniel Dunbar    return (((unsigned) p[0]) | (((unsigned) p[1]) << 8)) - 1;
1197fe60650c1133ee74a3395cf1063690e274bb7acDaniel Dunbar  }
1207fe60650c1133ee74a3395cf1063690e274bb7acDaniel Dunbar
12101eb9b9683535d8a65c704ad2c545903409e2d36Daniel Dunbar  /// getName - Return the actual identifier string.
122686775deca8b8685eb90801495880e3abdd844c2Chris Lattner  StringRef getName() const {
123686775deca8b8685eb90801495880e3abdd844c2Chris Lattner    return StringRef(getNameStart(), getLength());
1245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
1251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1269c46de446d18f4a28446cb798d4131bd05515699Chris Lattner  /// hasMacroDefinition - Return true if this identifier is #defined to some
1279c46de446d18f4a28446cb798d4131bd05515699Chris Lattner  /// other value.
1289c46de446d18f4a28446cb798d4131bd05515699Chris Lattner  bool hasMacroDefinition() const {
1299c46de446d18f4a28446cb798d4131bd05515699Chris Lattner    return HasMacro;
1309c46de446d18f4a28446cb798d4131bd05515699Chris Lattner  }
1316a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  void setHasMacroDefinition(bool Val) {
1326a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner    if (HasMacro == Val) return;
1331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1346a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner    HasMacro = Val;
1356a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner    if (Val)
1366a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner      NeedsHandleIdentifier = 1;
1376a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner    else
1386a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner      RecomputeNeedsHandleIdentifier();
1396a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  }
1401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
141646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis  /// getTokenID - If this is a source-language token (e.g. 'for'), this API
1425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// can be used to cause the lexer to map identifiers to source-language
1435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// tokens.
1448e748ab52395328f2905855b295a22e33dc800b2Ted Kremenek  tok::TokenKind getTokenID() const { return (tok::TokenKind)TokenID; }
145646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis
146646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis  /// \brief True if RevertTokenIDToIdentifier() was called.
147646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis  bool hasRevertedTokenIDToIdentifier() const { return RevertedTokenID; }
148646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis
149646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis  /// \brief Revert TokenID to tok::identifier; used for GNU libstdc++ 4.2
150f573084db5807f6003282bdf53ca9d58bab1ddc4Argyrios Kyrtzidis  /// compatibility.
151f573084db5807f6003282bdf53ca9d58bab1ddc4Argyrios Kyrtzidis  ///
152f573084db5807f6003282bdf53ca9d58bab1ddc4Argyrios Kyrtzidis  /// TokenID is normally read-only but there are 2 instances where we revert it
153f573084db5807f6003282bdf53ca9d58bab1ddc4Argyrios Kyrtzidis  /// to tok::identifier for libstdc++ 4.2. Keep track of when this happens
1543c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl  /// using this method so we can inform serialization about it.
155646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis  void RevertTokenIDToIdentifier() {
156646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis    assert(TokenID != tok::identifier && "Already at tok::identifier");
157646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis    TokenID = tok::identifier;
158646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis    RevertedTokenID = true;
159646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis  }
1601eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// getPPKeywordID - Return the preprocessor keyword ID for this identifier.
1626d9a3e648d6bf6b347174152f191bd1377528f8cChris Lattner  /// For example, "define" will return tok::pp_define.
163387b98d37e08f7aa9ddf23e067e1d27e39ce16f3Chris Lattner  tok::PPKeywordKind getPPKeywordID() const;
1641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// getObjCKeywordID - Return the Objective-C keyword ID for the this
1665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// identifier.  For example, 'class' will return tok::objc_class if ObjC is
1675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// enabled.
168ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  tok::ObjCKeywordKind getObjCKeywordID() const {
1691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    if (ObjCOrBuiltinID < tok::NUM_OBJC_KEYWORDS)
1705142af38ed0dd2f592cbfa00fa6e2e14dd6cc516Douglas Gregor      return tok::ObjCKeywordKind(ObjCOrBuiltinID);
1715142af38ed0dd2f592cbfa00fa6e2e14dd6cc516Douglas Gregor    else
1725142af38ed0dd2f592cbfa00fa6e2e14dd6cc516Douglas Gregor      return tok::objc_not_keyword;
173ca63fa00786e51c207c829f4182f11a6c6b552beTed Kremenek  }
1745142af38ed0dd2f592cbfa00fa6e2e14dd6cc516Douglas Gregor  void setObjCKeywordID(tok::ObjCKeywordKind ID) { ObjCOrBuiltinID = ID; }
1753251ceb90b3fec68e86d6dcfa58836e20a7205c3Douglas Gregor
1765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// getBuiltinID - Return a value indicating whether this is a builtin
1775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// function.  0 is not-built-in.  1 is builtin-for-some-nonprimary-target.
1785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// 2+ are specific builtin functions.
1791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  unsigned getBuiltinID() const {
1805142af38ed0dd2f592cbfa00fa6e2e14dd6cc516Douglas Gregor    if (ObjCOrBuiltinID >= tok::NUM_OBJC_KEYWORDS)
1811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      return ObjCOrBuiltinID - tok::NUM_OBJC_KEYWORDS;
1825142af38ed0dd2f592cbfa00fa6e2e14dd6cc516Douglas Gregor    else
1835142af38ed0dd2f592cbfa00fa6e2e14dd6cc516Douglas Gregor      return 0;
1845142af38ed0dd2f592cbfa00fa6e2e14dd6cc516Douglas Gregor  }
1855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  void setBuiltinID(unsigned ID) {
1865142af38ed0dd2f592cbfa00fa6e2e14dd6cc516Douglas Gregor    ObjCOrBuiltinID = ID + tok::NUM_OBJC_KEYWORDS;
1871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    assert(ObjCOrBuiltinID - unsigned(tok::NUM_OBJC_KEYWORDS) == ID
1885142af38ed0dd2f592cbfa00fa6e2e14dd6cc516Douglas Gregor           && "ID too large for field!");
1895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
1903251ceb90b3fec68e86d6dcfa58836e20a7205c3Douglas Gregor
1913251ceb90b3fec68e86d6dcfa58836e20a7205c3Douglas Gregor  unsigned getObjCOrBuiltinID() const { return ObjCOrBuiltinID; }
1923251ceb90b3fec68e86d6dcfa58836e20a7205c3Douglas Gregor  void setObjCOrBuiltinID(unsigned ID) { ObjCOrBuiltinID = ID; }
1933251ceb90b3fec68e86d6dcfa58836e20a7205c3Douglas Gregor
1945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// get/setExtension - Initialize information about whether or not this
1955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// language token is an extension.  This controls extension warnings, and is
1965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// only valid if a custom token ID is set.
1975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  bool isExtensionToken() const { return IsExtension; }
1986a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  void setIsExtensionToken(bool Val) {
1996a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner    IsExtension = Val;
2006a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner    if (Val)
2016a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner      NeedsHandleIdentifier = 1;
2026a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner    else
2036a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner      RecomputeNeedsHandleIdentifier();
2046a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  }
2051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
20698d86b98b3fd0bd9c546123b16fd9995509aaae1Richard Smith  /// is/setIsCXX11CompatKeyword - Initialize information about whether or not
20798d86b98b3fd0bd9c546123b16fd9995509aaae1Richard Smith  /// this language token is a keyword in C++11. This controls compatibility
20898d86b98b3fd0bd9c546123b16fd9995509aaae1Richard Smith  /// warnings, and is only true when not parsing C++11. Once a compatibility
20998d86b98b3fd0bd9c546123b16fd9995509aaae1Richard Smith  /// problem has been diagnosed with this keyword, the flag will be cleared.
21098d86b98b3fd0bd9c546123b16fd9995509aaae1Richard Smith  bool isCXX11CompatKeyword() const { return IsCXX11CompatKeyword; }
21198d86b98b3fd0bd9c546123b16fd9995509aaae1Richard Smith  void setIsCXX11CompatKeyword(bool Val) {
21298d86b98b3fd0bd9c546123b16fd9995509aaae1Richard Smith    IsCXX11CompatKeyword = Val;
21398d86b98b3fd0bd9c546123b16fd9995509aaae1Richard Smith    if (Val)
21498d86b98b3fd0bd9c546123b16fd9995509aaae1Richard Smith      NeedsHandleIdentifier = 1;
21598d86b98b3fd0bd9c546123b16fd9995509aaae1Richard Smith    else
21698d86b98b3fd0bd9c546123b16fd9995509aaae1Richard Smith      RecomputeNeedsHandleIdentifier();
21798d86b98b3fd0bd9c546123b16fd9995509aaae1Richard Smith  }
21898d86b98b3fd0bd9c546123b16fd9995509aaae1Richard Smith
2195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// setIsPoisoned - Mark this identifier as poisoned.  After poisoning, the
2205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// Preprocessor will emit an error every time this token is used.
2216a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  void setIsPoisoned(bool Value = true) {
2226a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner    IsPoisoned = Value;
2236a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner    if (Value)
2246a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner      NeedsHandleIdentifier = 1;
2256a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner    else
2266a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner      RecomputeNeedsHandleIdentifier();
2276a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  }
2281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// isPoisoned - Return true if this token has been poisoned.
2305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  bool isPoisoned() const { return IsPoisoned; }
2311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// isCPlusPlusOperatorKeyword/setIsCPlusPlusOperatorKeyword controls whether
2335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// this identifier is a C++ alternate representation of an operator.
2346a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  void setIsCPlusPlusOperatorKeyword(bool Val = true) {
2356a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner    IsCPPOperatorKeyword = Val;
2366a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner    if (Val)
2376a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner      NeedsHandleIdentifier = 1;
2386a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner    else
2396a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner      RecomputeNeedsHandleIdentifier();
2406a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  }
2415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  bool isCPlusPlusOperatorKeyword() const { return IsCPPOperatorKeyword; }
2425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
2435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// getFETokenInfo/setFETokenInfo - The language front-end is allowed to
2445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// associate arbitrary metadata with this token.
2455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  template<typename T>
2465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  T *getFETokenInfo() const { return static_cast<T*>(FETokenInfo); }
2475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  void setFETokenInfo(void *T) { FETokenInfo = T; }
2486a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner
2496a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  /// isHandleIdentifierCase - Return true if the Preprocessor::HandleIdentifier
2506a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  /// must be called on a token of this identifier.  If this returns false, we
2516a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  /// know that HandleIdentifier will not affect the token.
2526a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  bool isHandleIdentifierCase() const { return NeedsHandleIdentifier; }
2531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2543c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl  /// isFromAST - Return true if the identifier in its current state was loaded
2553c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl  /// from an AST file.
2563c7f4134603d04b44f997b43c0a9def270f25386Sebastian Redl  bool isFromAST() const { return IsFromAST; }
257ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl
258eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  void setIsFromAST() { IsFromAST = true; }
259eee242ff426bf79149f221798966e58688383c1eDouglas Gregor
260eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  /// \brief Determine whether this identifier has changed since it was loaded
261eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  /// from an AST file.
262eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  bool hasChangedSinceDeserialization() const {
263eee242ff426bf79149f221798966e58688383c1eDouglas Gregor    return ChangedAfterLoad;
264eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  }
265eee242ff426bf79149f221798966e58688383c1eDouglas Gregor
266eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  /// \brief Note that this identifier has changed since it was loaded from
267eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  /// an AST file.
268eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  void setChangedSinceDeserialization() {
269eee242ff426bf79149f221798966e58688383c1eDouglas Gregor    ChangedAfterLoad = true;
270eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  }
271ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl
272eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  /// \brief Determine whether the information for this identifier is out of
273eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  /// date with respect to the external source.
274eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  bool isOutOfDate() const { return OutOfDate; }
275eee242ff426bf79149f221798966e58688383c1eDouglas Gregor
276eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  /// \brief Set whether the information for this identifier is out of
277eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  /// date with respect to the external source.
278eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  void setOutOfDate(bool OOD) {
279eee242ff426bf79149f221798966e58688383c1eDouglas Gregor    OutOfDate = OOD;
280eee242ff426bf79149f221798966e58688383c1eDouglas Gregor    if (OOD)
281eee242ff426bf79149f221798966e58688383c1eDouglas Gregor      NeedsHandleIdentifier = true;
282eee242ff426bf79149f221798966e58688383c1eDouglas Gregor    else
283eee242ff426bf79149f221798966e58688383c1eDouglas Gregor      RecomputeNeedsHandleIdentifier();
284eee242ff426bf79149f221798966e58688383c1eDouglas Gregor  }
285eee242ff426bf79149f221798966e58688383c1eDouglas Gregor
28632ad2ee2618745ce3da51c2ae066ed5f21157c07Ted Kremenek  /// \brief Determine whether this is the contextual keyword
28732ad2ee2618745ce3da51c2ae066ed5f21157c07Ted Kremenek  /// '__experimental_modules_import'.
28832ad2ee2618745ce3da51c2ae066ed5f21157c07Ted Kremenek  bool isModulesImport() const { return IsModulesImport; }
289c13a34b690d2dc2a03c2fea75a0a1438636c19ceDouglas Gregor
29032ad2ee2618745ce3da51c2ae066ed5f21157c07Ted Kremenek  /// \brief Set whether this identifier is the contextual keyword
29132ad2ee2618745ce3da51c2ae066ed5f21157c07Ted Kremenek  /// '__experimental_modules_import'.
29232ad2ee2618745ce3da51c2ae066ed5f21157c07Ted Kremenek  void setModulesImport(bool I) {
29332ad2ee2618745ce3da51c2ae066ed5f21157c07Ted Kremenek    IsModulesImport = I;
294c13a34b690d2dc2a03c2fea75a0a1438636c19ceDouglas Gregor    if (I)
295c13a34b690d2dc2a03c2fea75a0a1438636c19ceDouglas Gregor      NeedsHandleIdentifier = true;
296c13a34b690d2dc2a03c2fea75a0a1438636c19ceDouglas Gregor    else
297c13a34b690d2dc2a03c2fea75a0a1438636c19ceDouglas Gregor      RecomputeNeedsHandleIdentifier();
298c13a34b690d2dc2a03c2fea75a0a1438636c19ceDouglas Gregor  }
299c13a34b690d2dc2a03c2fea75a0a1438636c19ceDouglas Gregor
3006a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattnerprivate:
3016a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  /// RecomputeNeedsHandleIdentifier - The Preprocessor::HandleIdentifier does
3026a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  /// several special (but rare) things to identifiers of various sorts.  For
3036a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  /// example, it changes the "for" keyword token from tok::identifier to
3046a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  /// tok::for.
3056a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  ///
3066a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  /// This method is very tied to the definition of HandleIdentifier.  Any
3076a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  /// change to it should be reflected here.
3086a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  void RecomputeNeedsHandleIdentifier() {
3096a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner    NeedsHandleIdentifier =
3106a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner      (isPoisoned() | hasMacroDefinition() | isCPlusPlusOperatorKeyword() |
311eee242ff426bf79149f221798966e58688383c1eDouglas Gregor       isExtensionToken() | isCXX11CompatKeyword() || isOutOfDate() ||
31232ad2ee2618745ce3da51c2ae066ed5f21157c07Ted Kremenek       isModulesImport());
3136a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner  }
3145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
3155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
31628bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley/// \brief an RAII object for [un]poisoning an identifier
31728bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley/// within a certain scope. II is allowed to be null, in
31828bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley/// which case, objects of this type have no effect.
31928bbe4b8acc338476fe0825769b41fb32b423c72John Wiegleyclass PoisonIdentifierRAIIObject {
32028bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley  IdentifierInfo *const II;
32128bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley  const bool OldValue;
32228bbe4b8acc338476fe0825769b41fb32b423c72John Wiegleypublic:
32328bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley  PoisonIdentifierRAIIObject(IdentifierInfo *II, bool NewValue)
32428bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley    : II(II), OldValue(II ? II->isPoisoned() : false) {
32528bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley    if(II)
32628bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley      II->setIsPoisoned(NewValue);
32728bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley  }
32828bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley
32928bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley  ~PoisonIdentifierRAIIObject() {
33028bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley    if(II)
33128bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley      II->setIsPoisoned(OldValue);
33228bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley  }
33328bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley};
33428bbe4b8acc338476fe0825769b41fb32b423c72John Wiegley
33595f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor/// \brief An iterator that walks over all of the known identifiers
33695f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor/// in the lookup table.
33795f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor///
33895f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor/// Since this iterator uses an abstract interface via virtual
33995f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor/// functions, it uses an object-oriented interface rather than the
34095f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor/// more standard C++ STL iterator interface. In this OO-style
34195f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor/// iteration, the single function \c Next() provides dereference,
34295f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor/// advance, and end-of-sequence checking in a single
34395f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor/// operation. Subclasses of this iterator type will provide the
34495f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor/// actual functionality.
34595f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregorclass IdentifierIterator {
34695f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregorprivate:
34795f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  IdentifierIterator(const IdentifierIterator&); // Do not implement
34895f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  IdentifierIterator &operator=(const IdentifierIterator&); // Do not implement
34995f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor
35095f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregorprotected:
35195f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  IdentifierIterator() { }
35295f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor
35395f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregorpublic:
35495f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  virtual ~IdentifierIterator();
35595f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor
35695f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  /// \brief Retrieve the next string in the identifier table and
35795f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  /// advances the iterator for the following string.
35895f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  ///
35995f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  /// \returns The next string in the identifier table. If there is
360686775deca8b8685eb90801495880e3abdd844c2Chris Lattner  /// no such string, returns an empty \c StringRef.
361686775deca8b8685eb90801495880e3abdd844c2Chris Lattner  virtual StringRef Next() = 0;
36295f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor};
36395f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor
36472b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek/// IdentifierInfoLookup - An abstract class used by IdentifierTable that
365668c1a4fdcc56bdd050256b1688e116fe84b72dbDouglas Gregor///  provides an interface for performing lookups from strings
36672b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek/// (const char *) to IdentiferInfo objects.
36772b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenekclass IdentifierInfoLookup {
36872b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenekpublic:
36972b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  virtual ~IdentifierInfoLookup();
3701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
37172b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  /// get - Return the identifier token info for the specified named identifier.
37272b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  ///  Unlike the version in IdentifierTable, this returns a pointer instead
37372b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  ///  of a reference.  If the pointer is NULL then the IdentifierInfo cannot
37472b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  ///  be found.
375686775deca8b8685eb90801495880e3abdd844c2Chris Lattner  virtual IdentifierInfo* get(StringRef Name) = 0;
37695f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor
37795f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  /// \brief Retrieve an iterator into the set of all identifiers
37895f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  /// known to this identifier lookup source.
37995f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  ///
38095f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  /// This routine provides access to all of the identifiers known to
38195f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  /// the identifier lookup, allowing access to the contents of the
38295f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  /// identifiers without introducing the overhead of constructing
38395f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  /// IdentifierInfo objects for each.
38495f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  ///
38595f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  /// \returns A new iterator into the set of known identifiers. The
38695f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  /// caller is responsible for deleting this iterator.
38795f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  virtual IdentifierIterator *getIdentifiers() const;
3881eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump};
3898c5a760b82e73ed90b560090772db97e2ae27b09Douglas Gregor
3908c5a760b82e73ed90b560090772db97e2ae27b09Douglas Gregor/// \brief An abstract class used to resolve numerical identifier
3918c5a760b82e73ed90b560090772db97e2ae27b09Douglas Gregor/// references (meaningful only to some external source) into
3928c5a760b82e73ed90b560090772db97e2ae27b09Douglas Gregor/// IdentifierInfo pointers.
3938c5a760b82e73ed90b560090772db97e2ae27b09Douglas Gregorclass ExternalIdentifierLookup {
3948c5a760b82e73ed90b560090772db97e2ae27b09Douglas Gregorpublic:
3958c5a760b82e73ed90b560090772db97e2ae27b09Douglas Gregor  virtual ~ExternalIdentifierLookup();
3968c5a760b82e73ed90b560090772db97e2ae27b09Douglas Gregor
3978c5a760b82e73ed90b560090772db97e2ae27b09Douglas Gregor  /// \brief Return the identifier associated with the given ID number.
3988c5a760b82e73ed90b560090772db97e2ae27b09Douglas Gregor  ///
3998c5a760b82e73ed90b560090772db97e2ae27b09Douglas Gregor  /// The ID 0 is associated with the NULL identifier.
4008c5a760b82e73ed90b560090772db97e2ae27b09Douglas Gregor  virtual IdentifierInfo *GetIdentifier(unsigned ID) = 0;
4018c5a760b82e73ed90b560090772db97e2ae27b09Douglas Gregor};
4028c5a760b82e73ed90b560090772db97e2ae27b09Douglas Gregor
4035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// IdentifierTable - This table implements an efficient mapping from strings to
4045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// IdentifierInfo nodes.  It has no other purpose, but this is an
405fc8f0e14ad142ed811e90fbd9a30e419e301c717Chris Lattner/// extremely performance-critical piece of the code, as each occurrence of
4065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// every identifier goes through here when lexed.
4075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerclass IdentifierTable {
4085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Shark shows that using MallocAllocator is *much* slower than using this
4095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // BumpPtrAllocator!
410ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek  typedef llvm::StringMap<IdentifierInfo*, llvm::BumpPtrAllocator> HashTableTy;
4115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  HashTableTy HashTable;
4121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
41372b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  IdentifierInfoLookup* ExternalLookup;
4141cd1b1e987f5e2f060d7972b13d83239b36d77d6Douglas Gregor
4155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerpublic:
4165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// IdentifierTable ctor - Create the identifier table, populating it with
4175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// info about the language keywords for the language specified by LangOpts.
41872b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  IdentifierTable(const LangOptions &LangOpts,
41972b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek                  IdentifierInfoLookup* externalLookup = 0);
4201eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
421668c1a4fdcc56bdd050256b1688e116fe84b72dbDouglas Gregor  /// \brief Set the external identifier lookup mechanism.
422668c1a4fdcc56bdd050256b1688e116fe84b72dbDouglas Gregor  void setExternalIdentifierLookup(IdentifierInfoLookup *IILookup) {
423668c1a4fdcc56bdd050256b1688e116fe84b72dbDouglas Gregor    ExternalLookup = IILookup;
424668c1a4fdcc56bdd050256b1688e116fe84b72dbDouglas Gregor  }
425668c1a4fdcc56bdd050256b1688e116fe84b72dbDouglas Gregor
42695f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  /// \brief Retrieve the external identifier lookup object, if any.
42795f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  IdentifierInfoLookup *getExternalIdentifierLookup() const {
42895f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor    return ExternalLookup;
42995f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor  }
43095f4292cc526c629fead321c7fcfd4fe0f3bc66eDouglas Gregor
43172b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  llvm::BumpPtrAllocator& getAllocator() {
43272b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek    return HashTable.getAllocator();
43372b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  }
4341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// get - Return the identifier token info for the specified named identifier.
4365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ///
437686775deca8b8685eb90801495880e3abdd844c2Chris Lattner  IdentifierInfo &get(StringRef Name) {
438ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek    llvm::StringMapEntry<IdentifierInfo*> &Entry =
4396488292f5e204fed99bb43ab23b8342ddc03ce89Kovarththanan Rajaratnam      HashTable.GetOrCreateValue(Name);
4401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
441ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek    IdentifierInfo *II = Entry.getValue();
442d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar    if (II) return *II;
4431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
444d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar    // No entry; if we have an external lookup, look there first.
445d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar    if (ExternalLookup) {
4466488292f5e204fed99bb43ab23b8342ddc03ce89Kovarththanan Rajaratnam      II = ExternalLookup->get(Name);
447d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar      if (II) {
448d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar        // Cache in the StringMap for subsequent lookups.
449d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar        Entry.setValue(II);
450d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar        return *II;
451ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek      }
452d42ffbd22fc7eb61321f6a88173ee424991f01c6Ted Kremenek    }
453ccb9bac3adb35a2dc78c1737e7b2dc6537a16393Daniel Dunbar
454d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar    // Lookups failed, make a new IdentifierInfo.
455d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar    void *Mem = getAllocator().Allocate<IdentifierInfo>();
456d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar    II = new (Mem) IdentifierInfo();
457d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar    Entry.setValue(II);
458d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar
459d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar    // Make sure getName() knows how to find the IdentifierInfo
460d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar    // contents.
461d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar    II->Entry = &Entry;
462d43b333be82438102ff4c459b1fb5dfb764e5f0dDaniel Dunbar
463ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek    return *II;
4645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
4651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
466686775deca8b8685eb90801495880e3abdd844c2Chris Lattner  IdentifierInfo &get(StringRef Name, tok::TokenKind TokenCode) {
467646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis    IdentifierInfo &II = get(Name);
468646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis    II.TokenID = TokenCode;
469d648d373d6c14dccadd3bef7b560f6a7296f949ePeter Collingbourne    assert(II.TokenID == (unsigned) TokenCode && "TokenCode too large");
470646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis    return II;
471646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis  }
472646395bbcaa849c94bc2a3246c71d809ca719f01Argyrios Kyrtzidis
473ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl  /// \brief Gets an IdentifierInfo for the given name without consulting
474ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl  ///        external sources.
4755f8e3302242cf94de2f8e46d10167f57fcf747c3Douglas Gregor  ///
476ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl  /// This is a version of get() meant for external sources that want to
477ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl  /// introduce or modify an identifier. If they called get(), they would
478ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl  /// likely end up in a recursion.
479686775deca8b8685eb90801495880e3abdd844c2Chris Lattner  IdentifierInfo &getOwn(StringRef Name) {
4805f8e3302242cf94de2f8e46d10167f57fcf747c3Douglas Gregor    llvm::StringMapEntry<IdentifierInfo*> &Entry =
48165aa6885818d4b4eea2e5a9d12085b2398148662Jay Foad      HashTable.GetOrCreateValue(Name);
4821eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4835f8e3302242cf94de2f8e46d10167f57fcf747c3Douglas Gregor    IdentifierInfo *II = Entry.getValue();
484ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl    if (!II) {
4851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
486ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl      // Lookups failed, make a new IdentifierInfo.
487ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl      void *Mem = getAllocator().Allocate<IdentifierInfo>();
488ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl      II = new (Mem) IdentifierInfo();
489ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl      Entry.setValue(II);
4905f8e3302242cf94de2f8e46d10167f57fcf747c3Douglas Gregor
491ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl      // Make sure getName() knows how to find the IdentifierInfo
492ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl      // contents.
493ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl      II->Entry = &Entry;
494c13a34b690d2dc2a03c2fea75a0a1438636c19ceDouglas Gregor
495c13a34b690d2dc2a03c2fea75a0a1438636c19ceDouglas Gregor      // If this is the 'import' contextual keyword, mark it as such.
496c13a34b690d2dc2a03c2fea75a0a1438636c19ceDouglas Gregor      if (Name.equals("import"))
49732ad2ee2618745ce3da51c2ae066ed5f21157c07Ted Kremenek        II->setModulesImport(true);
498ffaab3e2bb13991bb3357e80f14bcae3745b2347Sebastian Redl    }
4995f8e3302242cf94de2f8e46d10167f57fcf747c3Douglas Gregor
5005f8e3302242cf94de2f8e46d10167f57fcf747c3Douglas Gregor    return *II;
5015f8e3302242cf94de2f8e46d10167f57fcf747c3Douglas Gregor  }
5025f8e3302242cf94de2f8e46d10167f57fcf747c3Douglas Gregor
5035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  typedef HashTableTy::const_iterator iterator;
5045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  typedef HashTableTy::const_iterator const_iterator;
5051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  iterator begin() const { return HashTable.begin(); }
5075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  iterator end() const   { return HashTable.end(); }
508c637e6b7afeebc6b4f751e4373715b6a8ea77272Ted Kremenek  unsigned size() const { return HashTable.size(); }
5091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// PrintStats - Print some statistics to stderr that indicate how well the
5115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// hashing is doing.
5125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  void PrintStats() const;
5131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  void AddKeywords(const LangOptions &LangOpts);
5155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer};
5165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
51785f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// ObjCMethodFamily - A family of Objective-C methods.  These
51885f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// families have no inherent meaning in the language, but are
51985f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// nonetheless central enough in the existing implementations to
52085f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// merit direct AST support.  While, in theory, arbitrary methods can
52185f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// be considered to form families, we focus here on the methods
52285f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// involving allocation and retain-count management, as these are the
52385f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// most "core" and the most likely to be useful to diverse clients
52485f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// without extra information.
52585f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall///
52685f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// Both selectors and actual method declarations may be classified
52785f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// into families.  Method families may impose additional restrictions
52885f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// beyond their selector name; for example, a method called '_init'
52985f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// that returns void is not considered to be in the 'init' family
53085f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// (but would be if it returned 'id').  It is also possible to
53185f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// explicitly change or remove a method's family.  Therefore the
53285f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// method's family should be considered the single source of truth.
53385f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCallenum ObjCMethodFamily {
53485f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  /// \brief No particular method family.
53585f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  OMF_None,
53685f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall
53785f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  // Selectors in these families may have arbitrary arity, may be
53885f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  // written with arbitrary leading underscores, and may have
53985f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  // additional CamelCase "words" in their first selector chunk
54085f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  // following the family name.
54185f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  OMF_alloc,
54285f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  OMF_copy,
54385f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  OMF_init,
54485f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  OMF_mutableCopy,
54585f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  OMF_new,
54685f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall
54785f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  // These families are singletons consisting only of the nullary
54885f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  // selector with the given name.
54985f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  OMF_autorelease,
55085f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  OMF_dealloc,
55180cb6e69d9e85231588ae604e4bc2bc9a07389afNico Weber  OMF_finalize,
55285f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  OMF_release,
55385f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  OMF_retain,
554926df6cfabf3eaa4afc990c097fa4619b76a9b57Douglas Gregor  OMF_retainCount,
5559670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian  OMF_self,
5569670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian
5579670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian  // performSelector families
5589670e179a67d868e171feac44fb8f9e2f108c5e8Fariborz Jahanian  OMF_performSelector
55985f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall};
56085f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall
56185f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// Enough bits to store any enumerator in ObjCMethodFamily or
56285f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// InvalidObjCMethodFamily.
56385f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCallenum { ObjCMethodFamilyBitWidth = 4 };
56485f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall
56585f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall/// An invalid value of ObjCMethodFamily.
56685f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCallenum { InvalidObjCMethodFamily = (1 << ObjCMethodFamilyBitWidth) - 1 };
56785f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall
56829238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroff/// Selector - This smart pointer class efficiently represents Objective-C
56929238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroff/// method names. This class will either point to an IdentifierInfo or a
57029238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroff/// MultiKeywordSelector (which is private). This enables us to optimize
5711eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump/// selectors that take no arguments and selectors that take 1 argument, which
57229238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroff/// accounts for 78% of all selectors in Cocoa.h.
573bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroffclass Selector {
57440847cfb58acc3cac7d68727df9455ac45f2e118David Blaikie  friend class Diagnostic;
5751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
576bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  enum IdentifierInfoFlag {
5770e2dfd3963465d238e791eb8996e4d49fa47b615Chris Lattner    // MultiKeywordSelector = 0.
578bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff    ZeroArg  = 0x1,
579bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff    OneArg   = 0x2,
580bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff    ArgFlags = ZeroArg|OneArg
581bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  };
582bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  uintptr_t InfoPtr; // a pointer to the MultiKeywordSelector or IdentifierInfo.
5831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
584bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  Selector(IdentifierInfo *II, unsigned nArgs) {
585bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff    InfoPtr = reinterpret_cast<uintptr_t>(II);
5860e2dfd3963465d238e791eb8996e4d49fa47b615Chris Lattner    assert((InfoPtr & ArgFlags) == 0 &&"Insufficiently aligned IdentifierInfo");
5870e2dfd3963465d238e791eb8996e4d49fa47b615Chris Lattner    assert(nArgs < 2 && "nArgs not equal to 0/1");
5880e2dfd3963465d238e791eb8996e4d49fa47b615Chris Lattner    InfoPtr |= nArgs+1;
589bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  }
590bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  Selector(MultiKeywordSelector *SI) {
591bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff    InfoPtr = reinterpret_cast<uintptr_t>(SI);
5920e2dfd3963465d238e791eb8996e4d49fa47b615Chris Lattner    assert((InfoPtr & ArgFlags) == 0 &&"Insufficiently aligned IdentifierInfo");
593bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  }
5941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
595bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  IdentifierInfo *getAsIdentifierInfo() const {
5960e2dfd3963465d238e791eb8996e4d49fa47b615Chris Lattner    if (getIdentifierInfoFlag())
597bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff      return reinterpret_cast<IdentifierInfo *>(InfoPtr & ~ArgFlags);
598bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff    return 0;
599bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  }
600bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  unsigned getIdentifierInfoFlag() const {
601bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff    return InfoPtr & ArgFlags;
602bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  }
6038af2c16571f3aade6d47ce81fa3857d01d375719Ted Kremenek
60485f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  static ObjCMethodFamily getMethodFamilyImpl(Selector sel);
60585f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall
6068af2c16571f3aade6d47ce81fa3857d01d375719Ted Kremenekpublic:
6078af2c16571f3aade6d47ce81fa3857d01d375719Ted Kremenek  friend class SelectorTable; // only the SelectorTable can create these
6088af2c16571f3aade6d47ce81fa3857d01d375719Ted Kremenek  friend class DeclarationName; // and the AST's DeclarationName.
6098af2c16571f3aade6d47ce81fa3857d01d375719Ted Kremenek
6108af2c16571f3aade6d47ce81fa3857d01d375719Ted Kremenek  /// The default ctor should only be used when creating data structures that
6118af2c16571f3aade6d47ce81fa3857d01d375719Ted Kremenek  ///  will contain selectors.
6128af2c16571f3aade6d47ce81fa3857d01d375719Ted Kremenek  Selector() : InfoPtr(0) {}
61390cd1bb1baac2a0221f3642de0cbea3244b116e5Steve Naroff  Selector(uintptr_t V) : InfoPtr(V) {}
6148af2c16571f3aade6d47ce81fa3857d01d375719Ted Kremenek
615bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  /// operator==/!= - Indicate whether the specified selectors are identical.
61697b7f26a92d87e514530a5b652460190ce48c974Ted Kremenek  bool operator==(Selector RHS) const {
617bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff    return InfoPtr == RHS.InfoPtr;
618bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  }
61997b7f26a92d87e514530a5b652460190ce48c974Ted Kremenek  bool operator!=(Selector RHS) const {
620bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff    return InfoPtr != RHS.InfoPtr;
621bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  }
622bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  void *getAsOpaquePtr() const {
623bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff    return reinterpret_cast<void*>(InfoPtr);
624bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  }
625405bad07391494d2eb025f8222c256c66b56e5f8Douglas Gregor
626405bad07391494d2eb025f8222c256c66b56e5f8Douglas Gregor  /// \brief Determine whether this is the empty selector.
627405bad07391494d2eb025f8222c256c66b56e5f8Douglas Gregor  bool isNull() const { return InfoPtr == 0; }
628405bad07391494d2eb025f8222c256c66b56e5f8Douglas Gregor
629bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  // Predicates to identify the selector type.
6301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  bool isKeywordSelector() const {
6311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return getIdentifierInfoFlag() != ZeroArg;
632bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  }
6331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  bool isUnarySelector() const {
634bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff    return getIdentifierInfoFlag() == ZeroArg;
635bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff  }
6365b6b72f53ad164497cf62484b60cdbb4361f1978Steve Naroff  unsigned getNumArgs() const;
637813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor
638813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor
639813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// \brief Retrieve the identifier at a given position in the selector.
640813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  ///
641813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// Note that the identifier pointer returned may be NULL. Clients that only
642813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// care about the text of the identifier string, and not the specific,
643813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// uniqued identifier pointer, should use \c getNameForSlot(), which returns
644813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// an empty string when the identifier pointer would be NULL.
645813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  ///
646813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// \param argIndex The index for which we want to retrieve the identifier.
647813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// This index shall be less than \c getNumArgs() unless this is a keyword
648813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// selector, in which case 0 is the only permissible value.
649813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  ///
650813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// \returns the uniqued identifier for this slot, or NULL if this slot has
651813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// no corresponding identifier.
652f836e3fea2c77fdbb18170fb313ee0d45551320bChris Lattner  IdentifierInfo *getIdentifierInfoForSlot(unsigned argIndex) const;
653813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor
654813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// \brief Retrieve the name at a given position in the selector.
655813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  ///
656813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// \param argIndex The index for which we want to retrieve the name.
657813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// This index shall be less than \c getNumArgs() unless this is a keyword
658813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// selector, in which case 0 is the only permissible value.
659813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  ///
660813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// \returns the name for this slot, which may be the empty string if no
661813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor  /// name was supplied.
662686775deca8b8685eb90801495880e3abdd844c2Chris Lattner  StringRef getNameForSlot(unsigned argIndex) const;
663813d8346529bf094eb2b249648906ba7fd226688Douglas Gregor
664077bf5e2f48acfa9e7d69429b6e4ba86ea14896dChris Lattner  /// getAsString - Derive the full selector name (e.g. "foo:bar:") and return
665077bf5e2f48acfa9e7d69429b6e4ba86ea14896dChris Lattner  /// it as an std::string.
666077bf5e2f48acfa9e7d69429b6e4ba86ea14896dChris Lattner  std::string getAsString() const;
6671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
66885f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  /// getMethodFamily - Derive the conventional family of this method.
66985f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  ObjCMethodFamily getMethodFamily() const {
67085f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall    return getMethodFamilyImpl(*this);
67185f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall  }
67285f3d76c0ecfdefcf83ea44a57b7a16119c8a045John McCall
67385994260c41a54cab061a434ed378fc448333a4eChris Lattner  static Selector getEmptyMarker() {
67485994260c41a54cab061a434ed378fc448333a4eChris Lattner    return Selector(uintptr_t(-1));
67585994260c41a54cab061a434ed378fc448333a4eChris Lattner  }
67685994260c41a54cab061a434ed378fc448333a4eChris Lattner  static Selector getTombstoneMarker() {
67785994260c41a54cab061a434ed378fc448333a4eChris Lattner    return Selector(uintptr_t(-2));
67885994260c41a54cab061a434ed378fc448333a4eChris Lattner  }
679bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff};
680bcfb06ac6da1aa3c74ac1ef7a49c2807522366e7Steve Naroff
68129238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroff/// SelectorTable - This table allows us to fully hide how we implement
68229238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroff/// multi-keyword caching.
68329238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroffclass SelectorTable {
6845f7d2284c4b2f08d155732454002e68dc40c33efChris Lattner  void *Impl;  // Actually a SelectorTableImpl
68529238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroff  SelectorTable(const SelectorTable&); // DISABLED: DO NOT IMPLEMENT
68629238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroff  void operator=(const SelectorTable&); // DISABLED: DO NOT IMPLEMENT
68729238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroffpublic:
68829238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroff  SelectorTable();
68929238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroff  ~SelectorTable();
69029238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroff
691ff38491c18b060526d754765b952f4a497a89416Chris Lattner  /// getSelector - This can create any sort of selector.  NumArgs indicates
692ff38491c18b060526d754765b952f4a497a89416Chris Lattner  /// whether this is a no argument selector "foo", a single argument selector
693ff38491c18b060526d754765b952f4a497a89416Chris Lattner  /// "foo:" or multi-argument "foo:bar:".
694ff38491c18b060526d754765b952f4a497a89416Chris Lattner  Selector getSelector(unsigned NumArgs, IdentifierInfo **IIV);
6951eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
696ff38491c18b060526d754765b952f4a497a89416Chris Lattner  Selector getUnarySelector(IdentifierInfo *ID) {
697ff38491c18b060526d754765b952f4a497a89416Chris Lattner    return Selector(ID, 1);
698ff38491c18b060526d754765b952f4a497a89416Chris Lattner  }
699ff38491c18b060526d754765b952f4a497a89416Chris Lattner  Selector getNullarySelector(IdentifierInfo *ID) {
700ff38491c18b060526d754765b952f4a497a89416Chris Lattner    return Selector(ID, 0);
701ff38491c18b060526d754765b952f4a497a89416Chris Lattner  }
70261f72cbd037e58f12cfe90cd442373f44092f030Steve Naroff
70397f55d6ffd548d1777d790c84b358030682f9de2Ted Kremenek  /// Return the total amount of memory allocated for managing selectors.
70497f55d6ffd548d1777d790c84b358030682f9de2Ted Kremenek  size_t getTotalMemory() const;
70597f55d6ffd548d1777d790c84b358030682f9de2Ted Kremenek
70661f72cbd037e58f12cfe90cd442373f44092f030Steve Naroff  /// constructSetterName - Return the setter name for the given
70761f72cbd037e58f12cfe90cd442373f44092f030Steve Naroff  /// identifier, i.e. "set" + Name where the initial character of Name
70861f72cbd037e58f12cfe90cd442373f44092f030Steve Naroff  /// has been capitalized.
709fdc92b7877535e6264fe43cfbdc8f01e9b224f81Steve Naroff  static Selector constructSetterName(IdentifierTable &Idents,
710fdc92b7877535e6264fe43cfbdc8f01e9b224f81Steve Naroff                                      SelectorTable &SelTable,
7118fe83e1df954d72c0f4ffc15d20a5222ec151c21Benjamin Kramer                                      const IdentifierInfo *Name);
71229238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroff};
71329238a0bf7cbf5b396efb451a0adb5fe4aa037caSteve Naroff
714e94ca9e4371c022329270436b3dd77adc4ddfa8fDouglas Gregor/// DeclarationNameExtra - Common base of the MultiKeywordSelector,
715e94ca9e4371c022329270436b3dd77adc4ddfa8fDouglas Gregor/// CXXSpecialName, and CXXOperatorIdName classes, all of which are
716e94ca9e4371c022329270436b3dd77adc4ddfa8fDouglas Gregor/// private classes that describe different kinds of names.
7172e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregorclass DeclarationNameExtra {
7182e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregorpublic:
7192e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor  /// ExtraKind - The kind of "extra" information stored in the
7202e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor  /// DeclarationName. See @c ExtraKindOrNumArgs for an explanation of
7212e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor  /// how these enumerator values are used.
7222e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor  enum ExtraKind {
7232e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor    CXXConstructor = 0,
7242e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor    CXXDestructor,
7252e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor    CXXConversionFunction,
726e94ca9e4371c022329270436b3dd77adc4ddfa8fDouglas Gregor#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
727e94ca9e4371c022329270436b3dd77adc4ddfa8fDouglas Gregor    CXXOperator##Name,
728e94ca9e4371c022329270436b3dd77adc4ddfa8fDouglas Gregor#include "clang/Basic/OperatorKinds.def"
7293e518bda00d710754ca077cf9be8dd821e16a854Sean Hunt    CXXLiteralOperator,
7302a3009a432bdcec59e6383d7b2b17494d6f91649Douglas Gregor    CXXUsingDirective,
7312e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor    NUM_EXTRA_KINDS
7322e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor  };
7335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
734e94ca9e4371c022329270436b3dd77adc4ddfa8fDouglas Gregor  /// ExtraKindOrNumArgs - Either the kind of C++ special name or
735e94ca9e4371c022329270436b3dd77adc4ddfa8fDouglas Gregor  /// operator-id (if the value is one of the CXX* enumerators of
736e94ca9e4371c022329270436b3dd77adc4ddfa8fDouglas Gregor  /// ExtraKind), in which case the DeclarationNameExtra is also a
7373e518bda00d710754ca077cf9be8dd821e16a854Sean Hunt  /// CXXSpecialName, (for CXXConstructor, CXXDestructor, or
7383e518bda00d710754ca077cf9be8dd821e16a854Sean Hunt  /// CXXConversionFunction) CXXOperatorIdName, or CXXLiteralOperatorName,
7393e518bda00d710754ca077cf9be8dd821e16a854Sean Hunt  /// it may be also name common to C++ using-directives (CXXUsingDirective),
7403e518bda00d710754ca077cf9be8dd821e16a854Sean Hunt  /// otherwise it is NUM_EXTRA_KINDS+NumArgs, where NumArgs is the number of
7412e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor  /// arguments in the Objective-C selector, in which case the
7422e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor  /// DeclarationNameExtra is also a MultiKeywordSelector.
7432e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor  unsigned ExtraKindOrNumArgs;
7442e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor};
7452e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor
7462e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor}  // end namespace clang
7474365a7e46822700357a272d839ee2656d9603d5aChris Lattner
7482e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregornamespace llvm {
7494365a7e46822700357a272d839ee2656d9603d5aChris Lattner/// Define DenseMapInfo so that Selectors can be used as keys in DenseMap and
7504365a7e46822700357a272d839ee2656d9603d5aChris Lattner/// DenseSets.
75185994260c41a54cab061a434ed378fc448333a4eChris Lattnertemplate <>
75285994260c41a54cab061a434ed378fc448333a4eChris Lattnerstruct DenseMapInfo<clang::Selector> {
75385994260c41a54cab061a434ed378fc448333a4eChris Lattner  static inline clang::Selector getEmptyKey() {
75485994260c41a54cab061a434ed378fc448333a4eChris Lattner    return clang::Selector::getEmptyMarker();
75585994260c41a54cab061a434ed378fc448333a4eChris Lattner  }
75685994260c41a54cab061a434ed378fc448333a4eChris Lattner  static inline clang::Selector getTombstoneKey() {
7571eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return clang::Selector::getTombstoneMarker();
75885994260c41a54cab061a434ed378fc448333a4eChris Lattner  }
7591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
76085994260c41a54cab061a434ed378fc448333a4eChris Lattner  static unsigned getHashValue(clang::Selector S);
7611eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
76285994260c41a54cab061a434ed378fc448333a4eChris Lattner  static bool isEqual(clang::Selector LHS, clang::Selector RHS) {
76385994260c41a54cab061a434ed378fc448333a4eChris Lattner    return LHS == RHS;
76485994260c41a54cab061a434ed378fc448333a4eChris Lattner  }
76585994260c41a54cab061a434ed378fc448333a4eChris Lattner};
766700030ebddb987936d4fee14d9412821d96e4840Kovarththanan Rajaratnam
76706159e878569e5f39bf0e8f11b84ac3ad0970597Chris Lattnertemplate <>
76806159e878569e5f39bf0e8f11b84ac3ad0970597Chris Lattnerstruct isPodLike<clang::Selector> { static const bool value = true; };
76906159e878569e5f39bf0e8f11b84ac3ad0970597Chris Lattner
770d36adf553cd71df96fe869204482e0f0d68c6bbfDouglas Gregortemplate<>
771d36adf553cd71df96fe869204482e0f0d68c6bbfDouglas Gregorclass PointerLikeTypeTraits<clang::Selector> {
772d36adf553cd71df96fe869204482e0f0d68c6bbfDouglas Gregorpublic:
773d36adf553cd71df96fe869204482e0f0d68c6bbfDouglas Gregor  static inline const void *getAsVoidPointer(clang::Selector P) {
774d36adf553cd71df96fe869204482e0f0d68c6bbfDouglas Gregor    return P.getAsOpaquePtr();
775d36adf553cd71df96fe869204482e0f0d68c6bbfDouglas Gregor  }
776d36adf553cd71df96fe869204482e0f0d68c6bbfDouglas Gregor  static inline clang::Selector getFromVoidPointer(const void *P) {
777d36adf553cd71df96fe869204482e0f0d68c6bbfDouglas Gregor    return clang::Selector(reinterpret_cast<uintptr_t>(P));
778d36adf553cd71df96fe869204482e0f0d68c6bbfDouglas Gregor  }
779d36adf553cd71df96fe869204482e0f0d68c6bbfDouglas Gregor  enum { NumLowBitsAvailable = 0 };
780d36adf553cd71df96fe869204482e0f0d68c6bbfDouglas Gregor};
78185994260c41a54cab061a434ed378fc448333a4eChris Lattner
7821734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor// Provide PointerLikeTypeTraits for IdentifierInfo pointers, which
7831734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor// are not guaranteed to be 8-byte aligned.
7841734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregortemplate<>
7851734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregorclass PointerLikeTypeTraits<clang::IdentifierInfo*> {
7861734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregorpublic:
7871734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor  static inline void *getAsVoidPointer(clang::IdentifierInfo* P) {
7881eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return P;
7891734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor  }
7901734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor  static inline clang::IdentifierInfo *getFromVoidPointer(void *P) {
7911734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor    return static_cast<clang::IdentifierInfo*>(P);
7921734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor  }
7931734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor  enum { NumLowBitsAvailable = 1 };
7941734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor};
7951734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor
7961734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregortemplate<>
7971734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregorclass PointerLikeTypeTraits<const clang::IdentifierInfo*> {
7981734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregorpublic:
7991734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor  static inline const void *getAsVoidPointer(const clang::IdentifierInfo* P) {
8001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return P;
8011734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor  }
8021734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor  static inline const clang::IdentifierInfo *getFromVoidPointer(const void *P) {
8031734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor    return static_cast<const clang::IdentifierInfo*>(P);
8041734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor  }
8051734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor  enum { NumLowBitsAvailable = 1 };
8061734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor};
8071734317845d60307d474b5da8a8d33adbaf5e723Douglas Gregor
8082e1cd4264d363ca869bf37ef160902f211d21b8cDouglas Gregor}  // end namespace llvm
8095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#endif
810