| Douglas Gregor | 81b747b | 2009-09-17 21:32:03 +0000 | [diff] [blame] | 1 | //===---------------- SemaCodeComplete.cpp - Code Completion ----*- 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 defines the code-completion semantic actions. | 
|  | 11 | // | 
|  | 12 | //===----------------------------------------------------------------------===// | 
| John McCall | 2d88708 | 2010-08-25 22:03:47 +0000 | [diff] [blame] | 13 | #include "clang/Sema/SemaInternal.h" | 
| Douglas Gregor | e737f50 | 2010-08-12 20:07:10 +0000 | [diff] [blame] | 14 | #include "clang/Sema/Lookup.h" | 
| John McCall | 120d63c | 2010-08-24 20:38:10 +0000 | [diff] [blame] | 15 | #include "clang/Sema/Overload.h" | 
| Douglas Gregor | 81b747b | 2009-09-17 21:32:03 +0000 | [diff] [blame] | 16 | #include "clang/Sema/CodeCompleteConsumer.h" | 
| Douglas Gregor | 719770d | 2010-04-06 17:30:22 +0000 | [diff] [blame] | 17 | #include "clang/Sema/ExternalSemaSource.h" | 
| John McCall | 5f1e094 | 2010-08-24 08:50:51 +0000 | [diff] [blame] | 18 | #include "clang/Sema/Scope.h" | 
| John McCall | 781472f | 2010-08-25 08:40:02 +0000 | [diff] [blame] | 19 | #include "clang/Sema/ScopeInfo.h" | 
| John McCall | 7cd088e | 2010-08-24 07:21:54 +0000 | [diff] [blame] | 20 | #include "clang/AST/DeclObjC.h" | 
| Douglas Gregor | b9d0ef7 | 2009-09-21 19:57:38 +0000 | [diff] [blame] | 21 | #include "clang/AST/ExprCXX.h" | 
| Douglas Gregor | 24a069f | 2009-11-17 17:59:40 +0000 | [diff] [blame] | 22 | #include "clang/AST/ExprObjC.h" | 
| Douglas Gregor | 3f7c7f4 | 2009-10-30 16:50:04 +0000 | [diff] [blame] | 23 | #include "clang/Lex/MacroInfo.h" | 
|  | 24 | #include "clang/Lex/Preprocessor.h" | 
| Douglas Gregor | d36adf5 | 2010-09-16 16:06:31 +0000 | [diff] [blame] | 25 | #include "llvm/ADT/DenseSet.h" | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 26 | #include "llvm/ADT/SmallPtrSet.h" | 
| Douglas Gregor | 6a68403 | 2009-09-28 03:51:44 +0000 | [diff] [blame] | 27 | #include "llvm/ADT/StringExtras.h" | 
| Douglas Gregor | 22f5699 | 2010-04-06 19:22:33 +0000 | [diff] [blame] | 28 | #include "llvm/ADT/StringSwitch.h" | 
| Douglas Gregor | 458433d | 2010-08-26 15:07:07 +0000 | [diff] [blame] | 29 | #include "llvm/ADT/Twine.h" | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 30 | #include <list> | 
|  | 31 | #include <map> | 
|  | 32 | #include <vector> | 
| Douglas Gregor | 81b747b | 2009-09-17 21:32:03 +0000 | [diff] [blame] | 33 |  | 
|  | 34 | using namespace clang; | 
| John McCall | 781472f | 2010-08-25 08:40:02 +0000 | [diff] [blame] | 35 | using namespace sema; | 
| Douglas Gregor | 81b747b | 2009-09-17 21:32:03 +0000 | [diff] [blame] | 36 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 37 | namespace { | 
|  | 38 | /// \brief A container of code-completion results. | 
|  | 39 | class ResultBuilder { | 
|  | 40 | public: | 
|  | 41 | /// \brief The type of a name-lookup filter, which can be provided to the | 
|  | 42 | /// name-lookup routines to specify which declarations should be included in | 
|  | 43 | /// the result set (when it returns true) and which declarations should be | 
|  | 44 | /// filtered out (returns false). | 
|  | 45 | typedef bool (ResultBuilder::*LookupFilter)(NamedDecl *) const; | 
|  | 46 |  | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 47 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 48 |  | 
|  | 49 | private: | 
|  | 50 | /// \brief The actual results we have found. | 
|  | 51 | std::vector<Result> Results; | 
|  | 52 |  | 
|  | 53 | /// \brief A record of all of the declarations we have found and placed | 
|  | 54 | /// into the result set, used to ensure that no declaration ever gets into | 
|  | 55 | /// the result set twice. | 
|  | 56 | llvm::SmallPtrSet<Decl*, 16> AllDeclsFound; | 
|  | 57 |  | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 58 | typedef std::pair<NamedDecl *, unsigned> DeclIndexPair; | 
|  | 59 |  | 
|  | 60 | /// \brief An entry in the shadow map, which is optimized to store | 
|  | 61 | /// a single (declaration, index) mapping (the common case) but | 
|  | 62 | /// can also store a list of (declaration, index) mappings. | 
|  | 63 | class ShadowMapEntry { | 
|  | 64 | typedef llvm::SmallVector<DeclIndexPair, 4> DeclIndexPairVector; | 
|  | 65 |  | 
|  | 66 | /// \brief Contains either the solitary NamedDecl * or a vector | 
|  | 67 | /// of (declaration, index) pairs. | 
|  | 68 | llvm::PointerUnion<NamedDecl *, DeclIndexPairVector*> DeclOrVector; | 
|  | 69 |  | 
|  | 70 | /// \brief When the entry contains a single declaration, this is | 
|  | 71 | /// the index associated with that entry. | 
|  | 72 | unsigned SingleDeclIndex; | 
|  | 73 |  | 
|  | 74 | public: | 
|  | 75 | ShadowMapEntry() : DeclOrVector(), SingleDeclIndex(0) { } | 
|  | 76 |  | 
|  | 77 | void Add(NamedDecl *ND, unsigned Index) { | 
|  | 78 | if (DeclOrVector.isNull()) { | 
|  | 79 | // 0 - > 1 elements: just set the single element information. | 
|  | 80 | DeclOrVector = ND; | 
|  | 81 | SingleDeclIndex = Index; | 
|  | 82 | return; | 
|  | 83 | } | 
|  | 84 |  | 
|  | 85 | if (NamedDecl *PrevND = DeclOrVector.dyn_cast<NamedDecl *>()) { | 
|  | 86 | // 1 -> 2 elements: create the vector of results and push in the | 
|  | 87 | // existing declaration. | 
|  | 88 | DeclIndexPairVector *Vec = new DeclIndexPairVector; | 
|  | 89 | Vec->push_back(DeclIndexPair(PrevND, SingleDeclIndex)); | 
|  | 90 | DeclOrVector = Vec; | 
|  | 91 | } | 
|  | 92 |  | 
|  | 93 | // Add the new element to the end of the vector. | 
|  | 94 | DeclOrVector.get<DeclIndexPairVector*>()->push_back( | 
|  | 95 | DeclIndexPair(ND, Index)); | 
|  | 96 | } | 
|  | 97 |  | 
|  | 98 | void Destroy() { | 
|  | 99 | if (DeclIndexPairVector *Vec | 
|  | 100 | = DeclOrVector.dyn_cast<DeclIndexPairVector *>()) { | 
|  | 101 | delete Vec; | 
|  | 102 | DeclOrVector = ((NamedDecl *)0); | 
|  | 103 | } | 
|  | 104 | } | 
|  | 105 |  | 
|  | 106 | // Iteration. | 
|  | 107 | class iterator; | 
|  | 108 | iterator begin() const; | 
|  | 109 | iterator end() const; | 
|  | 110 | }; | 
|  | 111 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 112 | /// \brief A mapping from declaration names to the declarations that have | 
|  | 113 | /// this name within a particular scope and their index within the list of | 
|  | 114 | /// results. | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 115 | typedef llvm::DenseMap<DeclarationName, ShadowMapEntry> ShadowMap; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 116 |  | 
|  | 117 | /// \brief The semantic analysis object for which results are being | 
|  | 118 | /// produced. | 
|  | 119 | Sema &SemaRef; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 120 |  | 
|  | 121 | /// \brief The allocator used to allocate new code-completion strings. | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 122 | CodeCompletionAllocator &Allocator; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 123 |  | 
|  | 124 | /// \brief If non-NULL, a filter function used to remove any code-completion | 
|  | 125 | /// results that are not desirable. | 
|  | 126 | LookupFilter Filter; | 
| Douglas Gregor | 45bcd43 | 2010-01-14 03:21:49 +0000 | [diff] [blame] | 127 |  | 
|  | 128 | /// \brief Whether we should allow declarations as | 
|  | 129 | /// nested-name-specifiers that would otherwise be filtered out. | 
|  | 130 | bool AllowNestedNameSpecifiers; | 
|  | 131 |  | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 132 | /// \brief If set, the type that we would prefer our resulting value | 
|  | 133 | /// declarations to have. | 
|  | 134 | /// | 
|  | 135 | /// Closely matching the preferred type gives a boost to a result's | 
|  | 136 | /// priority. | 
|  | 137 | CanQualType PreferredType; | 
|  | 138 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 139 | /// \brief A list of shadow maps, which is used to model name hiding at | 
|  | 140 | /// different levels of, e.g., the inheritance hierarchy. | 
|  | 141 | std::list<ShadowMap> ShadowMaps; | 
|  | 142 |  | 
| Douglas Gregor | 3cdee12 | 2010-08-26 16:36:48 +0000 | [diff] [blame] | 143 | /// \brief If we're potentially referring to a C++ member function, the set | 
|  | 144 | /// of qualifiers applied to the object type. | 
|  | 145 | Qualifiers ObjectTypeQualifiers; | 
|  | 146 |  | 
|  | 147 | /// \brief Whether the \p ObjectTypeQualifiers field is active. | 
|  | 148 | bool HasObjectTypeQualifiers; | 
|  | 149 |  | 
| Douglas Gregor | 265f749 | 2010-08-27 15:29:55 +0000 | [diff] [blame] | 150 | /// \brief The selector that we prefer. | 
|  | 151 | Selector PreferredSelector; | 
|  | 152 |  | 
| Douglas Gregor | ca45da0 | 2010-11-02 20:36:02 +0000 | [diff] [blame] | 153 | /// \brief The completion context in which we are gathering results. | 
| Douglas Gregor | cee9ff1 | 2010-09-20 22:39:41 +0000 | [diff] [blame] | 154 | CodeCompletionContext CompletionContext; | 
|  | 155 |  | 
| Douglas Gregor | ca45da0 | 2010-11-02 20:36:02 +0000 | [diff] [blame] | 156 | /// \brief If we are in an instance method definition, the @implementation | 
|  | 157 | /// object. | 
|  | 158 | ObjCImplementationDecl *ObjCImplementation; | 
|  | 159 |  | 
| Douglas Gregor | cee9ff1 | 2010-09-20 22:39:41 +0000 | [diff] [blame] | 160 | void AdjustResultPriorityForDecl(Result &R); | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 161 |  | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 162 | void MaybeAddConstructorResults(Result R); | 
|  | 163 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 164 | public: | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 165 | explicit ResultBuilder(Sema &SemaRef, CodeCompletionAllocator &Allocator, | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 166 | const CodeCompletionContext &CompletionContext, | 
|  | 167 | LookupFilter Filter = 0) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 168 | : SemaRef(SemaRef), Allocator(Allocator), Filter(Filter), | 
|  | 169 | AllowNestedNameSpecifiers(false), HasObjectTypeQualifiers(false), | 
| Douglas Gregor | ca45da0 | 2010-11-02 20:36:02 +0000 | [diff] [blame] | 170 | CompletionContext(CompletionContext), | 
|  | 171 | ObjCImplementation(0) | 
|  | 172 | { | 
|  | 173 | // If this is an Objective-C instance method definition, dig out the | 
|  | 174 | // corresponding implementation. | 
|  | 175 | switch (CompletionContext.getKind()) { | 
|  | 176 | case CodeCompletionContext::CCC_Expression: | 
|  | 177 | case CodeCompletionContext::CCC_ObjCMessageReceiver: | 
|  | 178 | case CodeCompletionContext::CCC_ParenthesizedExpression: | 
|  | 179 | case CodeCompletionContext::CCC_Statement: | 
|  | 180 | case CodeCompletionContext::CCC_Recovery: | 
|  | 181 | if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl()) | 
|  | 182 | if (Method->isInstanceMethod()) | 
|  | 183 | if (ObjCInterfaceDecl *Interface = Method->getClassInterface()) | 
|  | 184 | ObjCImplementation = Interface->getImplementation(); | 
|  | 185 | break; | 
|  | 186 |  | 
|  | 187 | default: | 
|  | 188 | break; | 
|  | 189 | } | 
|  | 190 | } | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 191 |  | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 192 | /// \brief Whether we should include code patterns in the completion | 
|  | 193 | /// results. | 
|  | 194 | bool includeCodePatterns() const { | 
|  | 195 | return SemaRef.CodeCompleter && | 
| Douglas Gregor | f696152 | 2010-08-27 21:18:54 +0000 | [diff] [blame] | 196 | SemaRef.CodeCompleter->includeCodePatterns(); | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 197 | } | 
|  | 198 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 199 | /// \brief Set the filter used for code-completion results. | 
|  | 200 | void setFilter(LookupFilter Filter) { | 
|  | 201 | this->Filter = Filter; | 
|  | 202 | } | 
|  | 203 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 204 | Result *data() { return Results.empty()? 0 : &Results.front(); } | 
|  | 205 | unsigned size() const { return Results.size(); } | 
|  | 206 | bool empty() const { return Results.empty(); } | 
|  | 207 |  | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 208 | /// \brief Specify the preferred type. | 
|  | 209 | void setPreferredType(QualType T) { | 
|  | 210 | PreferredType = SemaRef.Context.getCanonicalType(T); | 
|  | 211 | } | 
|  | 212 |  | 
| Douglas Gregor | 3cdee12 | 2010-08-26 16:36:48 +0000 | [diff] [blame] | 213 | /// \brief Set the cv-qualifiers on the object type, for us in filtering | 
|  | 214 | /// calls to member functions. | 
|  | 215 | /// | 
|  | 216 | /// When there are qualifiers in this set, they will be used to filter | 
|  | 217 | /// out member functions that aren't available (because there will be a | 
|  | 218 | /// cv-qualifier mismatch) or prefer functions with an exact qualifier | 
|  | 219 | /// match. | 
|  | 220 | void setObjectTypeQualifiers(Qualifiers Quals) { | 
|  | 221 | ObjectTypeQualifiers = Quals; | 
|  | 222 | HasObjectTypeQualifiers = true; | 
|  | 223 | } | 
|  | 224 |  | 
| Douglas Gregor | 265f749 | 2010-08-27 15:29:55 +0000 | [diff] [blame] | 225 | /// \brief Set the preferred selector. | 
|  | 226 | /// | 
|  | 227 | /// When an Objective-C method declaration result is added, and that | 
|  | 228 | /// method's selector matches this preferred selector, we give that method | 
|  | 229 | /// a slight priority boost. | 
|  | 230 | void setPreferredSelector(Selector Sel) { | 
|  | 231 | PreferredSelector = Sel; | 
|  | 232 | } | 
| Douglas Gregor | ca45da0 | 2010-11-02 20:36:02 +0000 | [diff] [blame] | 233 |  | 
| Douglas Gregor | cee9ff1 | 2010-09-20 22:39:41 +0000 | [diff] [blame] | 234 | /// \brief Retrieve the code-completion context for which results are | 
|  | 235 | /// being collected. | 
|  | 236 | const CodeCompletionContext &getCompletionContext() const { | 
|  | 237 | return CompletionContext; | 
|  | 238 | } | 
|  | 239 |  | 
| Douglas Gregor | 45bcd43 | 2010-01-14 03:21:49 +0000 | [diff] [blame] | 240 | /// \brief Specify whether nested-name-specifiers are allowed. | 
|  | 241 | void allowNestedNameSpecifiers(bool Allow = true) { | 
|  | 242 | AllowNestedNameSpecifiers = Allow; | 
|  | 243 | } | 
|  | 244 |  | 
| Douglas Gregor | b9d7757 | 2010-09-21 00:03:25 +0000 | [diff] [blame] | 245 | /// \brief Return the semantic analysis object for which we are collecting | 
|  | 246 | /// code completion results. | 
|  | 247 | Sema &getSema() const { return SemaRef; } | 
|  | 248 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 249 | /// \brief Retrieve the allocator used to allocate code completion strings. | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 250 | CodeCompletionAllocator &getAllocator() const { return Allocator; } | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 251 |  | 
| Douglas Gregor | e495b7f | 2010-01-14 00:20:49 +0000 | [diff] [blame] | 252 | /// \brief Determine whether the given declaration is at all interesting | 
|  | 253 | /// as a code-completion result. | 
| Douglas Gregor | 45bcd43 | 2010-01-14 03:21:49 +0000 | [diff] [blame] | 254 | /// | 
|  | 255 | /// \param ND the declaration that we are inspecting. | 
|  | 256 | /// | 
|  | 257 | /// \param AsNestedNameSpecifier will be set true if this declaration is | 
|  | 258 | /// only interesting when it is a nested-name-specifier. | 
|  | 259 | bool isInterestingDecl(NamedDecl *ND, bool &AsNestedNameSpecifier) const; | 
| Douglas Gregor | 6660d84 | 2010-01-14 00:41:07 +0000 | [diff] [blame] | 260 |  | 
|  | 261 | /// \brief Check whether the result is hidden by the Hiding declaration. | 
|  | 262 | /// | 
|  | 263 | /// \returns true if the result is hidden and cannot be found, false if | 
|  | 264 | /// the hidden result could still be found. When false, \p R may be | 
|  | 265 | /// modified to describe how the result can be found (e.g., via extra | 
|  | 266 | /// qualification). | 
|  | 267 | bool CheckHiddenResult(Result &R, DeclContext *CurContext, | 
|  | 268 | NamedDecl *Hiding); | 
|  | 269 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 270 | /// \brief Add a new result to this result set (if it isn't already in one | 
|  | 271 | /// of the shadow maps), or replace an existing result (for, e.g., a | 
|  | 272 | /// redeclaration). | 
| Douglas Gregor | 456c4a1 | 2009-09-21 20:12:40 +0000 | [diff] [blame] | 273 | /// | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 274 | /// \param CurContext the result to add (if it is unique). | 
| Douglas Gregor | 456c4a1 | 2009-09-21 20:12:40 +0000 | [diff] [blame] | 275 | /// | 
|  | 276 | /// \param R the context in which this result will be named. | 
|  | 277 | void MaybeAddResult(Result R, DeclContext *CurContext = 0); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 278 |  | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 279 | /// \brief Add a new result to this result set, where we already know | 
|  | 280 | /// the hiding declation (if any). | 
|  | 281 | /// | 
|  | 282 | /// \param R the result to add (if it is unique). | 
|  | 283 | /// | 
|  | 284 | /// \param CurContext the context in which this result will be named. | 
|  | 285 | /// | 
|  | 286 | /// \param Hiding the declaration that hides the result. | 
| Douglas Gregor | 0cc8404 | 2010-01-14 15:47:35 +0000 | [diff] [blame] | 287 | /// | 
|  | 288 | /// \param InBaseClass whether the result was found in a base | 
|  | 289 | /// class of the searched context. | 
|  | 290 | void AddResult(Result R, DeclContext *CurContext, NamedDecl *Hiding, | 
|  | 291 | bool InBaseClass); | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 292 |  | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 293 | /// \brief Add a new non-declaration result to this result set. | 
|  | 294 | void AddResult(Result R); | 
|  | 295 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 296 | /// \brief Enter into a new scope. | 
|  | 297 | void EnterNewScope(); | 
|  | 298 |  | 
|  | 299 | /// \brief Exit from the current scope. | 
|  | 300 | void ExitScope(); | 
|  | 301 |  | 
| Douglas Gregor | 55385fe | 2009-11-18 04:19:12 +0000 | [diff] [blame] | 302 | /// \brief Ignore this declaration, if it is seen again. | 
|  | 303 | void Ignore(Decl *D) { AllDeclsFound.insert(D->getCanonicalDecl()); } | 
|  | 304 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 305 | /// \name Name lookup predicates | 
|  | 306 | /// | 
|  | 307 | /// These predicates can be passed to the name lookup functions to filter the | 
|  | 308 | /// results of name lookup. All of the predicates have the same type, so that | 
|  | 309 | /// | 
|  | 310 | //@{ | 
| Douglas Gregor | 791215b | 2009-09-21 20:51:25 +0000 | [diff] [blame] | 311 | bool IsOrdinaryName(NamedDecl *ND) const; | 
| Douglas Gregor | 4710e5b | 2010-05-28 00:49:12 +0000 | [diff] [blame] | 312 | bool IsOrdinaryNonTypeName(NamedDecl *ND) const; | 
| Douglas Gregor | f957843 | 2010-07-28 21:50:18 +0000 | [diff] [blame] | 313 | bool IsIntegralConstantValue(NamedDecl *ND) const; | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 314 | bool IsOrdinaryNonValueName(NamedDecl *ND) const; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 315 | bool IsNestedNameSpecifier(NamedDecl *ND) const; | 
|  | 316 | bool IsEnum(NamedDecl *ND) const; | 
|  | 317 | bool IsClassOrStruct(NamedDecl *ND) const; | 
|  | 318 | bool IsUnion(NamedDecl *ND) const; | 
|  | 319 | bool IsNamespace(NamedDecl *ND) const; | 
|  | 320 | bool IsNamespaceOrAlias(NamedDecl *ND) const; | 
|  | 321 | bool IsType(NamedDecl *ND) const; | 
| Douglas Gregor | eb5758b | 2009-09-23 22:26:46 +0000 | [diff] [blame] | 322 | bool IsMember(NamedDecl *ND) const; | 
| Douglas Gregor | 80f4f4c | 2010-01-14 16:08:12 +0000 | [diff] [blame] | 323 | bool IsObjCIvar(NamedDecl *ND) const; | 
| Douglas Gregor | 8e254cf | 2010-05-27 23:06:34 +0000 | [diff] [blame] | 324 | bool IsObjCMessageReceiver(NamedDecl *ND) const; | 
| Douglas Gregor | fb62941 | 2010-08-23 21:17:50 +0000 | [diff] [blame] | 325 | bool IsObjCCollection(NamedDecl *ND) const; | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 326 | bool IsImpossibleToSatisfy(NamedDecl *ND) const; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 327 | //@} | 
|  | 328 | }; | 
|  | 329 | } | 
|  | 330 |  | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 331 | class ResultBuilder::ShadowMapEntry::iterator { | 
|  | 332 | llvm::PointerUnion<NamedDecl*, const DeclIndexPair*> DeclOrIterator; | 
|  | 333 | unsigned SingleDeclIndex; | 
|  | 334 |  | 
|  | 335 | public: | 
|  | 336 | typedef DeclIndexPair value_type; | 
|  | 337 | typedef value_type reference; | 
|  | 338 | typedef std::ptrdiff_t difference_type; | 
|  | 339 | typedef std::input_iterator_tag iterator_category; | 
|  | 340 |  | 
|  | 341 | class pointer { | 
|  | 342 | DeclIndexPair Value; | 
|  | 343 |  | 
|  | 344 | public: | 
|  | 345 | pointer(const DeclIndexPair &Value) : Value(Value) { } | 
|  | 346 |  | 
|  | 347 | const DeclIndexPair *operator->() const { | 
|  | 348 | return &Value; | 
|  | 349 | } | 
|  | 350 | }; | 
|  | 351 |  | 
|  | 352 | iterator() : DeclOrIterator((NamedDecl *)0), SingleDeclIndex(0) { } | 
|  | 353 |  | 
|  | 354 | iterator(NamedDecl *SingleDecl, unsigned Index) | 
|  | 355 | : DeclOrIterator(SingleDecl), SingleDeclIndex(Index) { } | 
|  | 356 |  | 
|  | 357 | iterator(const DeclIndexPair *Iterator) | 
|  | 358 | : DeclOrIterator(Iterator), SingleDeclIndex(0) { } | 
|  | 359 |  | 
|  | 360 | iterator &operator++() { | 
|  | 361 | if (DeclOrIterator.is<NamedDecl *>()) { | 
|  | 362 | DeclOrIterator = (NamedDecl *)0; | 
|  | 363 | SingleDeclIndex = 0; | 
|  | 364 | return *this; | 
|  | 365 | } | 
|  | 366 |  | 
|  | 367 | const DeclIndexPair *I = DeclOrIterator.get<const DeclIndexPair*>(); | 
|  | 368 | ++I; | 
|  | 369 | DeclOrIterator = I; | 
|  | 370 | return *this; | 
|  | 371 | } | 
|  | 372 |  | 
| Chris Lattner | 66392d4 | 2010-09-04 18:12:20 +0000 | [diff] [blame] | 373 | /*iterator operator++(int) { | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 374 | iterator tmp(*this); | 
|  | 375 | ++(*this); | 
|  | 376 | return tmp; | 
| Chris Lattner | 66392d4 | 2010-09-04 18:12:20 +0000 | [diff] [blame] | 377 | }*/ | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 378 |  | 
|  | 379 | reference operator*() const { | 
|  | 380 | if (NamedDecl *ND = DeclOrIterator.dyn_cast<NamedDecl *>()) | 
|  | 381 | return reference(ND, SingleDeclIndex); | 
|  | 382 |  | 
| Douglas Gregor | d490f95 | 2009-12-06 21:27:58 +0000 | [diff] [blame] | 383 | return *DeclOrIterator.get<const DeclIndexPair*>(); | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 384 | } | 
|  | 385 |  | 
|  | 386 | pointer operator->() const { | 
|  | 387 | return pointer(**this); | 
|  | 388 | } | 
|  | 389 |  | 
|  | 390 | friend bool operator==(const iterator &X, const iterator &Y) { | 
| Douglas Gregor | d490f95 | 2009-12-06 21:27:58 +0000 | [diff] [blame] | 391 | return X.DeclOrIterator.getOpaqueValue() | 
|  | 392 | == Y.DeclOrIterator.getOpaqueValue() && | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 393 | X.SingleDeclIndex == Y.SingleDeclIndex; | 
|  | 394 | } | 
|  | 395 |  | 
|  | 396 | friend bool operator!=(const iterator &X, const iterator &Y) { | 
| Douglas Gregor | d490f95 | 2009-12-06 21:27:58 +0000 | [diff] [blame] | 397 | return !(X == Y); | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 398 | } | 
|  | 399 | }; | 
|  | 400 |  | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 401 | ResultBuilder::ShadowMapEntry::iterator | 
|  | 402 | ResultBuilder::ShadowMapEntry::begin() const { | 
|  | 403 | if (DeclOrVector.isNull()) | 
|  | 404 | return iterator(); | 
|  | 405 |  | 
|  | 406 | if (NamedDecl *ND = DeclOrVector.dyn_cast<NamedDecl *>()) | 
|  | 407 | return iterator(ND, SingleDeclIndex); | 
|  | 408 |  | 
|  | 409 | return iterator(DeclOrVector.get<DeclIndexPairVector *>()->begin()); | 
|  | 410 | } | 
|  | 411 |  | 
|  | 412 | ResultBuilder::ShadowMapEntry::iterator | 
|  | 413 | ResultBuilder::ShadowMapEntry::end() const { | 
|  | 414 | if (DeclOrVector.is<NamedDecl *>() || DeclOrVector.isNull()) | 
|  | 415 | return iterator(); | 
|  | 416 |  | 
|  | 417 | return iterator(DeclOrVector.get<DeclIndexPairVector *>()->end()); | 
|  | 418 | } | 
|  | 419 |  | 
| Douglas Gregor | 456c4a1 | 2009-09-21 20:12:40 +0000 | [diff] [blame] | 420 | /// \brief Compute the qualification required to get from the current context | 
|  | 421 | /// (\p CurContext) to the target context (\p TargetContext). | 
|  | 422 | /// | 
|  | 423 | /// \param Context the AST context in which the qualification will be used. | 
|  | 424 | /// | 
|  | 425 | /// \param CurContext the context where an entity is being named, which is | 
|  | 426 | /// typically based on the current scope. | 
|  | 427 | /// | 
|  | 428 | /// \param TargetContext the context in which the named entity actually | 
|  | 429 | /// resides. | 
|  | 430 | /// | 
|  | 431 | /// \returns a nested name specifier that refers into the target context, or | 
|  | 432 | /// NULL if no qualification is needed. | 
|  | 433 | static NestedNameSpecifier * | 
|  | 434 | getRequiredQualification(ASTContext &Context, | 
|  | 435 | DeclContext *CurContext, | 
|  | 436 | DeclContext *TargetContext) { | 
|  | 437 | llvm::SmallVector<DeclContext *, 4> TargetParents; | 
|  | 438 |  | 
|  | 439 | for (DeclContext *CommonAncestor = TargetContext; | 
|  | 440 | CommonAncestor && !CommonAncestor->Encloses(CurContext); | 
|  | 441 | CommonAncestor = CommonAncestor->getLookupParent()) { | 
|  | 442 | if (CommonAncestor->isTransparentContext() || | 
|  | 443 | CommonAncestor->isFunctionOrMethod()) | 
|  | 444 | continue; | 
|  | 445 |  | 
|  | 446 | TargetParents.push_back(CommonAncestor); | 
|  | 447 | } | 
|  | 448 |  | 
|  | 449 | NestedNameSpecifier *Result = 0; | 
|  | 450 | while (!TargetParents.empty()) { | 
|  | 451 | DeclContext *Parent = TargetParents.back(); | 
|  | 452 | TargetParents.pop_back(); | 
|  | 453 |  | 
| Douglas Gregor | fb62941 | 2010-08-23 21:17:50 +0000 | [diff] [blame] | 454 | if (NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Parent)) { | 
|  | 455 | if (!Namespace->getIdentifier()) | 
|  | 456 | continue; | 
|  | 457 |  | 
| Douglas Gregor | 456c4a1 | 2009-09-21 20:12:40 +0000 | [diff] [blame] | 458 | Result = NestedNameSpecifier::Create(Context, Result, Namespace); | 
| Douglas Gregor | fb62941 | 2010-08-23 21:17:50 +0000 | [diff] [blame] | 459 | } | 
| Douglas Gregor | 456c4a1 | 2009-09-21 20:12:40 +0000 | [diff] [blame] | 460 | else if (TagDecl *TD = dyn_cast<TagDecl>(Parent)) | 
|  | 461 | Result = NestedNameSpecifier::Create(Context, Result, | 
|  | 462 | false, | 
|  | 463 | Context.getTypeDeclType(TD).getTypePtr()); | 
| Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 464 | } | 
| Douglas Gregor | 456c4a1 | 2009-09-21 20:12:40 +0000 | [diff] [blame] | 465 | return Result; | 
|  | 466 | } | 
|  | 467 |  | 
| Douglas Gregor | 45bcd43 | 2010-01-14 03:21:49 +0000 | [diff] [blame] | 468 | bool ResultBuilder::isInterestingDecl(NamedDecl *ND, | 
|  | 469 | bool &AsNestedNameSpecifier) const { | 
|  | 470 | AsNestedNameSpecifier = false; | 
|  | 471 |  | 
| Douglas Gregor | e495b7f | 2010-01-14 00:20:49 +0000 | [diff] [blame] | 472 | ND = ND->getUnderlyingDecl(); | 
|  | 473 | unsigned IDNS = ND->getIdentifierNamespace(); | 
| Douglas Gregor | f52cede | 2009-10-09 22:16:47 +0000 | [diff] [blame] | 474 |  | 
|  | 475 | // Skip unnamed entities. | 
| Douglas Gregor | e495b7f | 2010-01-14 00:20:49 +0000 | [diff] [blame] | 476 | if (!ND->getDeclName()) | 
|  | 477 | return false; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 478 |  | 
|  | 479 | // Friend declarations and declarations introduced due to friends are never | 
|  | 480 | // added as results. | 
| John McCall | 92b7f70 | 2010-03-11 07:50:04 +0000 | [diff] [blame] | 481 | if (IDNS & (Decl::IDNS_OrdinaryFriend | Decl::IDNS_TagFriend)) | 
| Douglas Gregor | e495b7f | 2010-01-14 00:20:49 +0000 | [diff] [blame] | 482 | return false; | 
|  | 483 |  | 
| Douglas Gregor | 7628294 | 2009-12-11 17:31:05 +0000 | [diff] [blame] | 484 | // Class template (partial) specializations are never added as results. | 
| Douglas Gregor | e495b7f | 2010-01-14 00:20:49 +0000 | [diff] [blame] | 485 | if (isa<ClassTemplateSpecializationDecl>(ND) || | 
|  | 486 | isa<ClassTemplatePartialSpecializationDecl>(ND)) | 
|  | 487 | return false; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 488 |  | 
| Douglas Gregor | 7628294 | 2009-12-11 17:31:05 +0000 | [diff] [blame] | 489 | // Using declarations themselves are never added as results. | 
| Douglas Gregor | e495b7f | 2010-01-14 00:20:49 +0000 | [diff] [blame] | 490 | if (isa<UsingDecl>(ND)) | 
|  | 491 | return false; | 
|  | 492 |  | 
|  | 493 | // Some declarations have reserved names that we don't want to ever show. | 
|  | 494 | if (const IdentifierInfo *Id = ND->getIdentifier()) { | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 495 | // __va_list_tag is a freak of nature. Find it and skip it. | 
|  | 496 | if (Id->isStr("__va_list_tag") || Id->isStr("__builtin_va_list")) | 
| Douglas Gregor | e495b7f | 2010-01-14 00:20:49 +0000 | [diff] [blame] | 497 | return false; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 498 |  | 
| Douglas Gregor | f52cede | 2009-10-09 22:16:47 +0000 | [diff] [blame] | 499 | // Filter out names reserved for the implementation (C99 7.1.3, | 
| Douglas Gregor | 797efb5 | 2010-07-14 17:44:04 +0000 | [diff] [blame] | 500 | // C++ [lib.global.names]) if they come from a system header. | 
| Daniel Dunbar | e013d68 | 2009-10-18 20:26:12 +0000 | [diff] [blame] | 501 | // | 
|  | 502 | // FIXME: Add predicate for this. | 
| Douglas Gregor | f52cede | 2009-10-09 22:16:47 +0000 | [diff] [blame] | 503 | if (Id->getLength() >= 2) { | 
| Daniel Dunbar | e013d68 | 2009-10-18 20:26:12 +0000 | [diff] [blame] | 504 | const char *Name = Id->getNameStart(); | 
| Douglas Gregor | f52cede | 2009-10-09 22:16:47 +0000 | [diff] [blame] | 505 | if (Name[0] == '_' && | 
| Douglas Gregor | 797efb5 | 2010-07-14 17:44:04 +0000 | [diff] [blame] | 506 | (Name[1] == '_' || (Name[1] >= 'A' && Name[1] <= 'Z')) && | 
|  | 507 | (ND->getLocation().isInvalid() || | 
|  | 508 | SemaRef.SourceMgr.isInSystemHeader( | 
|  | 509 | SemaRef.SourceMgr.getSpellingLoc(ND->getLocation())))) | 
| Douglas Gregor | e495b7f | 2010-01-14 00:20:49 +0000 | [diff] [blame] | 510 | return false; | 
| Douglas Gregor | f52cede | 2009-10-09 22:16:47 +0000 | [diff] [blame] | 511 | } | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 512 | } | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 513 |  | 
| Douglas Gregor | 9b0ba87 | 2010-11-09 03:59:40 +0000 | [diff] [blame] | 514 | // Skip out-of-line declarations and definitions. | 
|  | 515 | // NOTE: Unless it's an Objective-C property, method, or ivar, where | 
|  | 516 | // the contexts can be messy. | 
|  | 517 | if (!ND->getDeclContext()->Equals(ND->getLexicalDeclContext()) && | 
|  | 518 | !(isa<ObjCPropertyDecl>(ND) || isa<ObjCIvarDecl>(ND) || | 
|  | 519 | isa<ObjCMethodDecl>(ND))) | 
|  | 520 | return false; | 
|  | 521 |  | 
| Douglas Gregor | a5fb7c3 | 2010-08-16 23:05:20 +0000 | [diff] [blame] | 522 | if (Filter == &ResultBuilder::IsNestedNameSpecifier || | 
|  | 523 | ((isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND)) && | 
|  | 524 | Filter != &ResultBuilder::IsNamespace && | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 525 | Filter != &ResultBuilder::IsNamespaceOrAlias && | 
|  | 526 | Filter != 0)) | 
| Douglas Gregor | a5fb7c3 | 2010-08-16 23:05:20 +0000 | [diff] [blame] | 527 | AsNestedNameSpecifier = true; | 
|  | 528 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 529 | // Filter out any unwanted results. | 
| Douglas Gregor | 45bcd43 | 2010-01-14 03:21:49 +0000 | [diff] [blame] | 530 | if (Filter && !(this->*Filter)(ND)) { | 
|  | 531 | // Check whether it is interesting as a nested-name-specifier. | 
|  | 532 | if (AllowNestedNameSpecifiers && SemaRef.getLangOptions().CPlusPlus && | 
|  | 533 | IsNestedNameSpecifier(ND) && | 
|  | 534 | (Filter != &ResultBuilder::IsMember || | 
|  | 535 | (isa<CXXRecordDecl>(ND) && | 
|  | 536 | cast<CXXRecordDecl>(ND)->isInjectedClassName()))) { | 
|  | 537 | AsNestedNameSpecifier = true; | 
|  | 538 | return true; | 
|  | 539 | } | 
|  | 540 |  | 
| Douglas Gregor | e495b7f | 2010-01-14 00:20:49 +0000 | [diff] [blame] | 541 | return false; | 
| Douglas Gregor | a5fb7c3 | 2010-08-16 23:05:20 +0000 | [diff] [blame] | 542 | } | 
| Douglas Gregor | e495b7f | 2010-01-14 00:20:49 +0000 | [diff] [blame] | 543 | // ... then it must be interesting! | 
|  | 544 | return true; | 
|  | 545 | } | 
|  | 546 |  | 
| Douglas Gregor | 6660d84 | 2010-01-14 00:41:07 +0000 | [diff] [blame] | 547 | bool ResultBuilder::CheckHiddenResult(Result &R, DeclContext *CurContext, | 
|  | 548 | NamedDecl *Hiding) { | 
|  | 549 | // In C, there is no way to refer to a hidden name. | 
|  | 550 | // FIXME: This isn't true; we can find a tag name hidden by an ordinary | 
|  | 551 | // name if we introduce the tag type. | 
|  | 552 | if (!SemaRef.getLangOptions().CPlusPlus) | 
|  | 553 | return true; | 
|  | 554 |  | 
| Sebastian Redl | 7a126a4 | 2010-08-31 00:36:30 +0000 | [diff] [blame] | 555 | DeclContext *HiddenCtx = R.Declaration->getDeclContext()->getRedeclContext(); | 
| Douglas Gregor | 6660d84 | 2010-01-14 00:41:07 +0000 | [diff] [blame] | 556 |  | 
|  | 557 | // There is no way to qualify a name declared in a function or method. | 
|  | 558 | if (HiddenCtx->isFunctionOrMethod()) | 
|  | 559 | return true; | 
|  | 560 |  | 
| Sebastian Redl | 7a126a4 | 2010-08-31 00:36:30 +0000 | [diff] [blame] | 561 | if (HiddenCtx == Hiding->getDeclContext()->getRedeclContext()) | 
| Douglas Gregor | 6660d84 | 2010-01-14 00:41:07 +0000 | [diff] [blame] | 562 | return true; | 
|  | 563 |  | 
|  | 564 | // We can refer to the result with the appropriate qualification. Do it. | 
|  | 565 | R.Hidden = true; | 
|  | 566 | R.QualifierIsInformative = false; | 
|  | 567 |  | 
|  | 568 | if (!R.Qualifier) | 
|  | 569 | R.Qualifier = getRequiredQualification(SemaRef.Context, | 
|  | 570 | CurContext, | 
|  | 571 | R.Declaration->getDeclContext()); | 
|  | 572 | return false; | 
|  | 573 | } | 
|  | 574 |  | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 575 | /// \brief A simplified classification of types used to determine whether two | 
|  | 576 | /// types are "similar enough" when adjusting priorities. | 
| Douglas Gregor | 1827e10 | 2010-08-16 16:18:59 +0000 | [diff] [blame] | 577 | SimplifiedTypeClass clang::getSimplifiedTypeClass(CanQualType T) { | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 578 | switch (T->getTypeClass()) { | 
|  | 579 | case Type::Builtin: | 
|  | 580 | switch (cast<BuiltinType>(T)->getKind()) { | 
|  | 581 | case BuiltinType::Void: | 
|  | 582 | return STC_Void; | 
|  | 583 |  | 
|  | 584 | case BuiltinType::NullPtr: | 
|  | 585 | return STC_Pointer; | 
|  | 586 |  | 
|  | 587 | case BuiltinType::Overload: | 
|  | 588 | case BuiltinType::Dependent: | 
|  | 589 | case BuiltinType::UndeducedAuto: | 
|  | 590 | return STC_Other; | 
|  | 591 |  | 
|  | 592 | case BuiltinType::ObjCId: | 
|  | 593 | case BuiltinType::ObjCClass: | 
|  | 594 | case BuiltinType::ObjCSel: | 
|  | 595 | return STC_ObjectiveC; | 
|  | 596 |  | 
|  | 597 | default: | 
|  | 598 | return STC_Arithmetic; | 
|  | 599 | } | 
|  | 600 | return STC_Other; | 
|  | 601 |  | 
|  | 602 | case Type::Complex: | 
|  | 603 | return STC_Arithmetic; | 
|  | 604 |  | 
|  | 605 | case Type::Pointer: | 
|  | 606 | return STC_Pointer; | 
|  | 607 |  | 
|  | 608 | case Type::BlockPointer: | 
|  | 609 | return STC_Block; | 
|  | 610 |  | 
|  | 611 | case Type::LValueReference: | 
|  | 612 | case Type::RValueReference: | 
|  | 613 | return getSimplifiedTypeClass(T->getAs<ReferenceType>()->getPointeeType()); | 
|  | 614 |  | 
|  | 615 | case Type::ConstantArray: | 
|  | 616 | case Type::IncompleteArray: | 
|  | 617 | case Type::VariableArray: | 
|  | 618 | case Type::DependentSizedArray: | 
|  | 619 | return STC_Array; | 
|  | 620 |  | 
|  | 621 | case Type::DependentSizedExtVector: | 
|  | 622 | case Type::Vector: | 
|  | 623 | case Type::ExtVector: | 
|  | 624 | return STC_Arithmetic; | 
|  | 625 |  | 
|  | 626 | case Type::FunctionProto: | 
|  | 627 | case Type::FunctionNoProto: | 
|  | 628 | return STC_Function; | 
|  | 629 |  | 
|  | 630 | case Type::Record: | 
|  | 631 | return STC_Record; | 
|  | 632 |  | 
|  | 633 | case Type::Enum: | 
|  | 634 | return STC_Arithmetic; | 
|  | 635 |  | 
|  | 636 | case Type::ObjCObject: | 
|  | 637 | case Type::ObjCInterface: | 
|  | 638 | case Type::ObjCObjectPointer: | 
|  | 639 | return STC_ObjectiveC; | 
|  | 640 |  | 
|  | 641 | default: | 
|  | 642 | return STC_Other; | 
|  | 643 | } | 
|  | 644 | } | 
|  | 645 |  | 
|  | 646 | /// \brief Get the type that a given expression will have if this declaration | 
|  | 647 | /// is used as an expression in its "typical" code-completion form. | 
| Douglas Gregor | 1827e10 | 2010-08-16 16:18:59 +0000 | [diff] [blame] | 648 | QualType clang::getDeclUsageType(ASTContext &C, NamedDecl *ND) { | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 649 | ND = cast<NamedDecl>(ND->getUnderlyingDecl()); | 
|  | 650 |  | 
|  | 651 | if (TypeDecl *Type = dyn_cast<TypeDecl>(ND)) | 
|  | 652 | return C.getTypeDeclType(Type); | 
|  | 653 | if (ObjCInterfaceDecl *Iface = dyn_cast<ObjCInterfaceDecl>(ND)) | 
|  | 654 | return C.getObjCInterfaceType(Iface); | 
|  | 655 |  | 
|  | 656 | QualType T; | 
|  | 657 | if (FunctionDecl *Function = dyn_cast<FunctionDecl>(ND)) | 
| Douglas Gregor | 5291c3c | 2010-07-13 08:18:22 +0000 | [diff] [blame] | 658 | T = Function->getCallResultType(); | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 659 | else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND)) | 
| Douglas Gregor | 5291c3c | 2010-07-13 08:18:22 +0000 | [diff] [blame] | 660 | T = Method->getSendResultType(); | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 661 | else if (FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND)) | 
| Douglas Gregor | 5291c3c | 2010-07-13 08:18:22 +0000 | [diff] [blame] | 662 | T = FunTmpl->getTemplatedDecl()->getCallResultType(); | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 663 | else if (EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND)) | 
|  | 664 | T = C.getTypeDeclType(cast<EnumDecl>(Enumerator->getDeclContext())); | 
|  | 665 | else if (ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND)) | 
|  | 666 | T = Property->getType(); | 
|  | 667 | else if (ValueDecl *Value = dyn_cast<ValueDecl>(ND)) | 
|  | 668 | T = Value->getType(); | 
|  | 669 | else | 
|  | 670 | return QualType(); | 
|  | 671 |  | 
|  | 672 | return T.getNonReferenceType(); | 
|  | 673 | } | 
|  | 674 |  | 
| Douglas Gregor | cee9ff1 | 2010-09-20 22:39:41 +0000 | [diff] [blame] | 675 | void ResultBuilder::AdjustResultPriorityForDecl(Result &R) { | 
|  | 676 | // If this is an Objective-C method declaration whose selector matches our | 
|  | 677 | // preferred selector, give it a priority boost. | 
|  | 678 | if (!PreferredSelector.isNull()) | 
|  | 679 | if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(R.Declaration)) | 
|  | 680 | if (PreferredSelector == Method->getSelector()) | 
|  | 681 | R.Priority += CCD_SelectorMatch; | 
| Douglas Gregor | 08f43cd | 2010-09-20 23:11:55 +0000 | [diff] [blame] | 682 |  | 
| Douglas Gregor | cee9ff1 | 2010-09-20 22:39:41 +0000 | [diff] [blame] | 683 | // If we have a preferred type, adjust the priority for results with exactly- | 
|  | 684 | // matching or nearly-matching types. | 
|  | 685 | if (!PreferredType.isNull()) { | 
|  | 686 | QualType T = getDeclUsageType(SemaRef.Context, R.Declaration); | 
|  | 687 | if (!T.isNull()) { | 
|  | 688 | CanQualType TC = SemaRef.Context.getCanonicalType(T); | 
|  | 689 | // Check for exactly-matching types (modulo qualifiers). | 
|  | 690 | if (SemaRef.Context.hasSameUnqualifiedType(PreferredType, TC)) | 
|  | 691 | R.Priority /= CCF_ExactTypeMatch; | 
|  | 692 | // Check for nearly-matching types, based on classification of each. | 
|  | 693 | else if ((getSimplifiedTypeClass(PreferredType) | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 694 | == getSimplifiedTypeClass(TC)) && | 
| Douglas Gregor | cee9ff1 | 2010-09-20 22:39:41 +0000 | [diff] [blame] | 695 | !(PreferredType->isEnumeralType() && TC->isEnumeralType())) | 
|  | 696 | R.Priority /= CCF_SimilarTypeMatch; | 
|  | 697 | } | 
|  | 698 | } | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 699 | } | 
|  | 700 |  | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 701 | void ResultBuilder::MaybeAddConstructorResults(Result R) { | 
|  | 702 | if (!SemaRef.getLangOptions().CPlusPlus || !R.Declaration || | 
|  | 703 | !CompletionContext.wantConstructorResults()) | 
|  | 704 | return; | 
|  | 705 |  | 
|  | 706 | ASTContext &Context = SemaRef.Context; | 
|  | 707 | NamedDecl *D = R.Declaration; | 
|  | 708 | CXXRecordDecl *Record = 0; | 
|  | 709 | if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D)) | 
|  | 710 | Record = ClassTemplate->getTemplatedDecl(); | 
|  | 711 | else if ((Record = dyn_cast<CXXRecordDecl>(D))) { | 
|  | 712 | // Skip specializations and partial specializations. | 
|  | 713 | if (isa<ClassTemplateSpecializationDecl>(Record)) | 
|  | 714 | return; | 
|  | 715 | } else { | 
|  | 716 | // There are no constructors here. | 
|  | 717 | return; | 
|  | 718 | } | 
|  | 719 |  | 
|  | 720 | Record = Record->getDefinition(); | 
|  | 721 | if (!Record) | 
|  | 722 | return; | 
|  | 723 |  | 
|  | 724 |  | 
|  | 725 | QualType RecordTy = Context.getTypeDeclType(Record); | 
|  | 726 | DeclarationName ConstructorName | 
|  | 727 | = Context.DeclarationNames.getCXXConstructorName( | 
|  | 728 | Context.getCanonicalType(RecordTy)); | 
|  | 729 | for (DeclContext::lookup_result Ctors = Record->lookup(ConstructorName); | 
|  | 730 | Ctors.first != Ctors.second; ++Ctors.first) { | 
|  | 731 | R.Declaration = *Ctors.first; | 
|  | 732 | R.CursorKind = getCursorKindForDecl(R.Declaration); | 
|  | 733 | Results.push_back(R); | 
|  | 734 | } | 
|  | 735 | } | 
|  | 736 |  | 
| Douglas Gregor | e495b7f | 2010-01-14 00:20:49 +0000 | [diff] [blame] | 737 | void ResultBuilder::MaybeAddResult(Result R, DeclContext *CurContext) { | 
|  | 738 | assert(!ShadowMaps.empty() && "Must enter into a results scope"); | 
|  | 739 |  | 
|  | 740 | if (R.Kind != Result::RK_Declaration) { | 
|  | 741 | // For non-declaration results, just add the result. | 
|  | 742 | Results.push_back(R); | 
|  | 743 | return; | 
|  | 744 | } | 
|  | 745 |  | 
|  | 746 | // Look through using declarations. | 
|  | 747 | if (UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(R.Declaration)) { | 
|  | 748 | MaybeAddResult(Result(Using->getTargetDecl(), R.Qualifier), CurContext); | 
|  | 749 | return; | 
|  | 750 | } | 
|  | 751 |  | 
|  | 752 | Decl *CanonDecl = R.Declaration->getCanonicalDecl(); | 
|  | 753 | unsigned IDNS = CanonDecl->getIdentifierNamespace(); | 
|  | 754 |  | 
| Douglas Gregor | 45bcd43 | 2010-01-14 03:21:49 +0000 | [diff] [blame] | 755 | bool AsNestedNameSpecifier = false; | 
|  | 756 | if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier)) | 
| Douglas Gregor | e495b7f | 2010-01-14 00:20:49 +0000 | [diff] [blame] | 757 | return; | 
|  | 758 |  | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 759 | // C++ constructors are never found by name lookup. | 
|  | 760 | if (isa<CXXConstructorDecl>(R.Declaration)) | 
|  | 761 | return; | 
|  | 762 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 763 | ShadowMap &SMap = ShadowMaps.back(); | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 764 | ShadowMapEntry::iterator I, IEnd; | 
|  | 765 | ShadowMap::iterator NamePos = SMap.find(R.Declaration->getDeclName()); | 
|  | 766 | if (NamePos != SMap.end()) { | 
|  | 767 | I = NamePos->second.begin(); | 
|  | 768 | IEnd = NamePos->second.end(); | 
|  | 769 | } | 
|  | 770 |  | 
|  | 771 | for (; I != IEnd; ++I) { | 
|  | 772 | NamedDecl *ND = I->first; | 
|  | 773 | unsigned Index = I->second; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 774 | if (ND->getCanonicalDecl() == CanonDecl) { | 
|  | 775 | // This is a redeclaration. Always pick the newer declaration. | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 776 | Results[Index].Declaration = R.Declaration; | 
|  | 777 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 778 | // We're done. | 
|  | 779 | return; | 
|  | 780 | } | 
|  | 781 | } | 
|  | 782 |  | 
|  | 783 | // This is a new declaration in this scope. However, check whether this | 
|  | 784 | // declaration name is hidden by a similarly-named declaration in an outer | 
|  | 785 | // scope. | 
|  | 786 | std::list<ShadowMap>::iterator SM, SMEnd = ShadowMaps.end(); | 
|  | 787 | --SMEnd; | 
|  | 788 | for (SM = ShadowMaps.begin(); SM != SMEnd; ++SM) { | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 789 | ShadowMapEntry::iterator I, IEnd; | 
|  | 790 | ShadowMap::iterator NamePos = SM->find(R.Declaration->getDeclName()); | 
|  | 791 | if (NamePos != SM->end()) { | 
|  | 792 | I = NamePos->second.begin(); | 
|  | 793 | IEnd = NamePos->second.end(); | 
|  | 794 | } | 
|  | 795 | for (; I != IEnd; ++I) { | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 796 | // A tag declaration does not hide a non-tag declaration. | 
| John McCall | 0d6b164 | 2010-04-23 18:46:30 +0000 | [diff] [blame] | 797 | if (I->first->hasTagIdentifierNamespace() && | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 798 | (IDNS & (Decl::IDNS_Member | Decl::IDNS_Ordinary | | 
|  | 799 | Decl::IDNS_ObjCProtocol))) | 
|  | 800 | continue; | 
|  | 801 |  | 
|  | 802 | // Protocols are in distinct namespaces from everything else. | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 803 | if (((I->first->getIdentifierNamespace() & Decl::IDNS_ObjCProtocol) | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 804 | || (IDNS & Decl::IDNS_ObjCProtocol)) && | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 805 | I->first->getIdentifierNamespace() != IDNS) | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 806 | continue; | 
|  | 807 |  | 
|  | 808 | // The newly-added result is hidden by an entry in the shadow map. | 
| Douglas Gregor | 6660d84 | 2010-01-14 00:41:07 +0000 | [diff] [blame] | 809 | if (CheckHiddenResult(R, CurContext, I->first)) | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 810 | return; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 811 |  | 
|  | 812 | break; | 
|  | 813 | } | 
|  | 814 | } | 
|  | 815 |  | 
|  | 816 | // Make sure that any given declaration only shows up in the result set once. | 
|  | 817 | if (!AllDeclsFound.insert(CanonDecl)) | 
|  | 818 | return; | 
| Douglas Gregor | 265f749 | 2010-08-27 15:29:55 +0000 | [diff] [blame] | 819 |  | 
| Douglas Gregor | eb5758b | 2009-09-23 22:26:46 +0000 | [diff] [blame] | 820 | // If the filter is for nested-name-specifiers, then this result starts a | 
|  | 821 | // nested-name-specifier. | 
| Douglas Gregor | 12e1313 | 2010-05-26 22:00:08 +0000 | [diff] [blame] | 822 | if (AsNestedNameSpecifier) { | 
| Douglas Gregor | eb5758b | 2009-09-23 22:26:46 +0000 | [diff] [blame] | 823 | R.StartsNestedNameSpecifier = true; | 
| Douglas Gregor | 12e1313 | 2010-05-26 22:00:08 +0000 | [diff] [blame] | 824 | R.Priority = CCP_NestedNameSpecifier; | 
| Douglas Gregor | cee9ff1 | 2010-09-20 22:39:41 +0000 | [diff] [blame] | 825 | } else | 
|  | 826 | AdjustResultPriorityForDecl(R); | 
| Douglas Gregor | 265f749 | 2010-08-27 15:29:55 +0000 | [diff] [blame] | 827 |  | 
| Douglas Gregor | 0563c26 | 2009-09-22 23:15:58 +0000 | [diff] [blame] | 828 | // If this result is supposed to have an informative qualifier, add one. | 
| Douglas Gregor | eb5758b | 2009-09-23 22:26:46 +0000 | [diff] [blame] | 829 | if (R.QualifierIsInformative && !R.Qualifier && | 
|  | 830 | !R.StartsNestedNameSpecifier) { | 
| Douglas Gregor | 0563c26 | 2009-09-22 23:15:58 +0000 | [diff] [blame] | 831 | DeclContext *Ctx = R.Declaration->getDeclContext(); | 
|  | 832 | if (NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx)) | 
|  | 833 | R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, Namespace); | 
|  | 834 | else if (TagDecl *Tag = dyn_cast<TagDecl>(Ctx)) | 
|  | 835 | R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, false, | 
|  | 836 | SemaRef.Context.getTypeDeclType(Tag).getTypePtr()); | 
|  | 837 | else | 
|  | 838 | R.QualifierIsInformative = false; | 
|  | 839 | } | 
| Douglas Gregor | eb5758b | 2009-09-23 22:26:46 +0000 | [diff] [blame] | 840 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 841 | // Insert this result into the set of results and into the current shadow | 
|  | 842 | // map. | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 843 | SMap[R.Declaration->getDeclName()].Add(R.Declaration, Results.size()); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 844 | Results.push_back(R); | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 845 |  | 
|  | 846 | if (!AsNestedNameSpecifier) | 
|  | 847 | MaybeAddConstructorResults(R); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 848 | } | 
|  | 849 |  | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 850 | void ResultBuilder::AddResult(Result R, DeclContext *CurContext, | 
| Douglas Gregor | 0cc8404 | 2010-01-14 15:47:35 +0000 | [diff] [blame] | 851 | NamedDecl *Hiding, bool InBaseClass = false) { | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 852 | if (R.Kind != Result::RK_Declaration) { | 
|  | 853 | // For non-declaration results, just add the result. | 
|  | 854 | Results.push_back(R); | 
|  | 855 | return; | 
|  | 856 | } | 
|  | 857 |  | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 858 | // Look through using declarations. | 
|  | 859 | if (UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(R.Declaration)) { | 
|  | 860 | AddResult(Result(Using->getTargetDecl(), R.Qualifier), CurContext, Hiding); | 
|  | 861 | return; | 
|  | 862 | } | 
|  | 863 |  | 
| Douglas Gregor | 45bcd43 | 2010-01-14 03:21:49 +0000 | [diff] [blame] | 864 | bool AsNestedNameSpecifier = false; | 
|  | 865 | if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier)) | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 866 | return; | 
|  | 867 |  | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 868 | // C++ constructors are never found by name lookup. | 
|  | 869 | if (isa<CXXConstructorDecl>(R.Declaration)) | 
|  | 870 | return; | 
|  | 871 |  | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 872 | if (Hiding && CheckHiddenResult(R, CurContext, Hiding)) | 
|  | 873 | return; | 
|  | 874 |  | 
|  | 875 | // Make sure that any given declaration only shows up in the result set once. | 
|  | 876 | if (!AllDeclsFound.insert(R.Declaration->getCanonicalDecl())) | 
|  | 877 | return; | 
|  | 878 |  | 
|  | 879 | // If the filter is for nested-name-specifiers, then this result starts a | 
|  | 880 | // nested-name-specifier. | 
| Douglas Gregor | 12e1313 | 2010-05-26 22:00:08 +0000 | [diff] [blame] | 881 | if (AsNestedNameSpecifier) { | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 882 | R.StartsNestedNameSpecifier = true; | 
| Douglas Gregor | 12e1313 | 2010-05-26 22:00:08 +0000 | [diff] [blame] | 883 | R.Priority = CCP_NestedNameSpecifier; | 
|  | 884 | } | 
| Douglas Gregor | 0cc8404 | 2010-01-14 15:47:35 +0000 | [diff] [blame] | 885 | else if (Filter == &ResultBuilder::IsMember && !R.Qualifier && InBaseClass && | 
|  | 886 | isa<CXXRecordDecl>(R.Declaration->getDeclContext() | 
| Sebastian Redl | 7a126a4 | 2010-08-31 00:36:30 +0000 | [diff] [blame] | 887 | ->getRedeclContext())) | 
| Douglas Gregor | 0cc8404 | 2010-01-14 15:47:35 +0000 | [diff] [blame] | 888 | R.QualifierIsInformative = true; | 
|  | 889 |  | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 890 | // If this result is supposed to have an informative qualifier, add one. | 
|  | 891 | if (R.QualifierIsInformative && !R.Qualifier && | 
|  | 892 | !R.StartsNestedNameSpecifier) { | 
|  | 893 | DeclContext *Ctx = R.Declaration->getDeclContext(); | 
|  | 894 | if (NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx)) | 
|  | 895 | R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, Namespace); | 
|  | 896 | else if (TagDecl *Tag = dyn_cast<TagDecl>(Ctx)) | 
|  | 897 | R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, 0, false, | 
| Douglas Gregor | 45bcd43 | 2010-01-14 03:21:49 +0000 | [diff] [blame] | 898 | SemaRef.Context.getTypeDeclType(Tag).getTypePtr()); | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 899 | else | 
|  | 900 | R.QualifierIsInformative = false; | 
|  | 901 | } | 
|  | 902 |  | 
| Douglas Gregor | 12e1313 | 2010-05-26 22:00:08 +0000 | [diff] [blame] | 903 | // Adjust the priority if this result comes from a base class. | 
|  | 904 | if (InBaseClass) | 
|  | 905 | R.Priority += CCD_InBaseClass; | 
|  | 906 |  | 
| Douglas Gregor | cee9ff1 | 2010-09-20 22:39:41 +0000 | [diff] [blame] | 907 | AdjustResultPriorityForDecl(R); | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 908 |  | 
| Douglas Gregor | 3cdee12 | 2010-08-26 16:36:48 +0000 | [diff] [blame] | 909 | if (HasObjectTypeQualifiers) | 
|  | 910 | if (CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(R.Declaration)) | 
|  | 911 | if (Method->isInstance()) { | 
|  | 912 | Qualifiers MethodQuals | 
|  | 913 | = Qualifiers::fromCVRMask(Method->getTypeQualifiers()); | 
|  | 914 | if (ObjectTypeQualifiers == MethodQuals) | 
|  | 915 | R.Priority += CCD_ObjectQualifierMatch; | 
|  | 916 | else if (ObjectTypeQualifiers - MethodQuals) { | 
|  | 917 | // The method cannot be invoked, because doing so would drop | 
|  | 918 | // qualifiers. | 
|  | 919 | return; | 
|  | 920 | } | 
|  | 921 | } | 
|  | 922 |  | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 923 | // Insert this result into the set of results. | 
|  | 924 | Results.push_back(R); | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 925 |  | 
|  | 926 | if (!AsNestedNameSpecifier) | 
|  | 927 | MaybeAddConstructorResults(R); | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 928 | } | 
|  | 929 |  | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 930 | void ResultBuilder::AddResult(Result R) { | 
|  | 931 | assert(R.Kind != Result::RK_Declaration && | 
|  | 932 | "Declaration results need more context"); | 
|  | 933 | Results.push_back(R); | 
|  | 934 | } | 
|  | 935 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 936 | /// \brief Enter into a new scope. | 
|  | 937 | void ResultBuilder::EnterNewScope() { | 
|  | 938 | ShadowMaps.push_back(ShadowMap()); | 
|  | 939 | } | 
|  | 940 |  | 
|  | 941 | /// \brief Exit from the current scope. | 
|  | 942 | void ResultBuilder::ExitScope() { | 
| Douglas Gregor | fbcb5d6 | 2009-12-06 20:23:50 +0000 | [diff] [blame] | 943 | for (ShadowMap::iterator E = ShadowMaps.back().begin(), | 
|  | 944 | EEnd = ShadowMaps.back().end(); | 
|  | 945 | E != EEnd; | 
|  | 946 | ++E) | 
|  | 947 | E->second.Destroy(); | 
|  | 948 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 949 | ShadowMaps.pop_back(); | 
|  | 950 | } | 
|  | 951 |  | 
| Douglas Gregor | 791215b | 2009-09-21 20:51:25 +0000 | [diff] [blame] | 952 | /// \brief Determines whether this given declaration will be found by | 
|  | 953 | /// ordinary name lookup. | 
|  | 954 | bool ResultBuilder::IsOrdinaryName(NamedDecl *ND) const { | 
| Douglas Gregor | 4710e5b | 2010-05-28 00:49:12 +0000 | [diff] [blame] | 955 | ND = cast<NamedDecl>(ND->getUnderlyingDecl()); | 
|  | 956 |  | 
| Douglas Gregor | 791215b | 2009-09-21 20:51:25 +0000 | [diff] [blame] | 957 | unsigned IDNS = Decl::IDNS_Ordinary; | 
|  | 958 | if (SemaRef.getLangOptions().CPlusPlus) | 
| Douglas Gregor | 9b30b26 | 2010-06-15 20:26:51 +0000 | [diff] [blame] | 959 | IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member; | 
| Douglas Gregor | ca45da0 | 2010-11-02 20:36:02 +0000 | [diff] [blame] | 960 | else if (SemaRef.getLangOptions().ObjC1) { | 
|  | 961 | if (isa<ObjCIvarDecl>(ND)) | 
|  | 962 | return true; | 
|  | 963 | if (isa<ObjCPropertyDecl>(ND) && | 
|  | 964 | SemaRef.canSynthesizeProvisionalIvar(cast<ObjCPropertyDecl>(ND))) | 
|  | 965 | return true; | 
|  | 966 | } | 
|  | 967 |  | 
| Douglas Gregor | 791215b | 2009-09-21 20:51:25 +0000 | [diff] [blame] | 968 | return ND->getIdentifierNamespace() & IDNS; | 
|  | 969 | } | 
|  | 970 |  | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 971 | /// \brief Determines whether this given declaration will be found by | 
| Douglas Gregor | 4710e5b | 2010-05-28 00:49:12 +0000 | [diff] [blame] | 972 | /// ordinary name lookup but is not a type name. | 
|  | 973 | bool ResultBuilder::IsOrdinaryNonTypeName(NamedDecl *ND) const { | 
|  | 974 | ND = cast<NamedDecl>(ND->getUnderlyingDecl()); | 
|  | 975 | if (isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND)) | 
|  | 976 | return false; | 
|  | 977 |  | 
|  | 978 | unsigned IDNS = Decl::IDNS_Ordinary; | 
|  | 979 | if (SemaRef.getLangOptions().CPlusPlus) | 
| Douglas Gregor | 9b30b26 | 2010-06-15 20:26:51 +0000 | [diff] [blame] | 980 | IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member; | 
| Douglas Gregor | ca45da0 | 2010-11-02 20:36:02 +0000 | [diff] [blame] | 981 | else if (SemaRef.getLangOptions().ObjC1) { | 
|  | 982 | if (isa<ObjCIvarDecl>(ND)) | 
|  | 983 | return true; | 
|  | 984 | if (isa<ObjCPropertyDecl>(ND) && | 
|  | 985 | SemaRef.canSynthesizeProvisionalIvar(cast<ObjCPropertyDecl>(ND))) | 
|  | 986 | return true; | 
|  | 987 | } | 
|  | 988 |  | 
| Douglas Gregor | 4710e5b | 2010-05-28 00:49:12 +0000 | [diff] [blame] | 989 | return ND->getIdentifierNamespace() & IDNS; | 
|  | 990 | } | 
|  | 991 |  | 
| Douglas Gregor | f957843 | 2010-07-28 21:50:18 +0000 | [diff] [blame] | 992 | bool ResultBuilder::IsIntegralConstantValue(NamedDecl *ND) const { | 
|  | 993 | if (!IsOrdinaryNonTypeName(ND)) | 
|  | 994 | return 0; | 
|  | 995 |  | 
|  | 996 | if (ValueDecl *VD = dyn_cast<ValueDecl>(ND->getUnderlyingDecl())) | 
|  | 997 | if (VD->getType()->isIntegralOrEnumerationType()) | 
|  | 998 | return true; | 
|  | 999 |  | 
|  | 1000 | return false; | 
|  | 1001 | } | 
|  | 1002 |  | 
| Douglas Gregor | 4710e5b | 2010-05-28 00:49:12 +0000 | [diff] [blame] | 1003 | /// \brief Determines whether this given declaration will be found by | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1004 | /// ordinary name lookup. | 
|  | 1005 | bool ResultBuilder::IsOrdinaryNonValueName(NamedDecl *ND) const { | 
| Douglas Gregor | 4710e5b | 2010-05-28 00:49:12 +0000 | [diff] [blame] | 1006 | ND = cast<NamedDecl>(ND->getUnderlyingDecl()); | 
|  | 1007 |  | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1008 | unsigned IDNS = Decl::IDNS_Ordinary; | 
|  | 1009 | if (SemaRef.getLangOptions().CPlusPlus) | 
| John McCall | 0d6b164 | 2010-04-23 18:46:30 +0000 | [diff] [blame] | 1010 | IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace; | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1011 |  | 
|  | 1012 | return (ND->getIdentifierNamespace() & IDNS) && | 
| Douglas Gregor | 4710e5b | 2010-05-28 00:49:12 +0000 | [diff] [blame] | 1013 | !isa<ValueDecl>(ND) && !isa<FunctionTemplateDecl>(ND) && | 
|  | 1014 | !isa<ObjCPropertyDecl>(ND); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1015 | } | 
|  | 1016 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1017 | /// \brief Determines whether the given declaration is suitable as the | 
|  | 1018 | /// start of a C++ nested-name-specifier, e.g., a class or namespace. | 
|  | 1019 | bool ResultBuilder::IsNestedNameSpecifier(NamedDecl *ND) const { | 
|  | 1020 | // Allow us to find class templates, too. | 
|  | 1021 | if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND)) | 
|  | 1022 | ND = ClassTemplate->getTemplatedDecl(); | 
|  | 1023 |  | 
|  | 1024 | return SemaRef.isAcceptableNestedNameSpecifier(ND); | 
|  | 1025 | } | 
|  | 1026 |  | 
|  | 1027 | /// \brief Determines whether the given declaration is an enumeration. | 
|  | 1028 | bool ResultBuilder::IsEnum(NamedDecl *ND) const { | 
|  | 1029 | return isa<EnumDecl>(ND); | 
|  | 1030 | } | 
|  | 1031 |  | 
|  | 1032 | /// \brief Determines whether the given declaration is a class or struct. | 
|  | 1033 | bool ResultBuilder::IsClassOrStruct(NamedDecl *ND) const { | 
|  | 1034 | // Allow us to find class templates, too. | 
|  | 1035 | if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND)) | 
|  | 1036 | ND = ClassTemplate->getTemplatedDecl(); | 
|  | 1037 |  | 
|  | 1038 | if (RecordDecl *RD = dyn_cast<RecordDecl>(ND)) | 
| Abramo Bagnara | 465d41b | 2010-05-11 21:36:43 +0000 | [diff] [blame] | 1039 | return RD->getTagKind() == TTK_Class || | 
|  | 1040 | RD->getTagKind() == TTK_Struct; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1041 |  | 
|  | 1042 | return false; | 
|  | 1043 | } | 
|  | 1044 |  | 
|  | 1045 | /// \brief Determines whether the given declaration is a union. | 
|  | 1046 | bool ResultBuilder::IsUnion(NamedDecl *ND) const { | 
|  | 1047 | // Allow us to find class templates, too. | 
|  | 1048 | if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND)) | 
|  | 1049 | ND = ClassTemplate->getTemplatedDecl(); | 
|  | 1050 |  | 
|  | 1051 | if (RecordDecl *RD = dyn_cast<RecordDecl>(ND)) | 
| Abramo Bagnara | 465d41b | 2010-05-11 21:36:43 +0000 | [diff] [blame] | 1052 | return RD->getTagKind() == TTK_Union; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1053 |  | 
|  | 1054 | return false; | 
|  | 1055 | } | 
|  | 1056 |  | 
|  | 1057 | /// \brief Determines whether the given declaration is a namespace. | 
|  | 1058 | bool ResultBuilder::IsNamespace(NamedDecl *ND) const { | 
|  | 1059 | return isa<NamespaceDecl>(ND); | 
|  | 1060 | } | 
|  | 1061 |  | 
|  | 1062 | /// \brief Determines whether the given declaration is a namespace or | 
|  | 1063 | /// namespace alias. | 
|  | 1064 | bool ResultBuilder::IsNamespaceOrAlias(NamedDecl *ND) const { | 
|  | 1065 | return isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND); | 
|  | 1066 | } | 
|  | 1067 |  | 
| Douglas Gregor | 7628294 | 2009-12-11 17:31:05 +0000 | [diff] [blame] | 1068 | /// \brief Determines whether the given declaration is a type. | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1069 | bool ResultBuilder::IsType(NamedDecl *ND) const { | 
| Douglas Gregor | d32b022 | 2010-08-24 01:06:58 +0000 | [diff] [blame] | 1070 | if (UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND)) | 
|  | 1071 | ND = Using->getTargetDecl(); | 
|  | 1072 |  | 
|  | 1073 | return isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1074 | } | 
|  | 1075 |  | 
| Douglas Gregor | 7628294 | 2009-12-11 17:31:05 +0000 | [diff] [blame] | 1076 | /// \brief Determines which members of a class should be visible via | 
|  | 1077 | /// "." or "->".  Only value declarations, nested name specifiers, and | 
|  | 1078 | /// using declarations thereof should show up. | 
| Douglas Gregor | eb5758b | 2009-09-23 22:26:46 +0000 | [diff] [blame] | 1079 | bool ResultBuilder::IsMember(NamedDecl *ND) const { | 
| Douglas Gregor | 7628294 | 2009-12-11 17:31:05 +0000 | [diff] [blame] | 1080 | if (UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND)) | 
|  | 1081 | ND = Using->getTargetDecl(); | 
|  | 1082 |  | 
| Douglas Gregor | ce82196 | 2009-12-11 18:14:22 +0000 | [diff] [blame] | 1083 | return isa<ValueDecl>(ND) || isa<FunctionTemplateDecl>(ND) || | 
|  | 1084 | isa<ObjCPropertyDecl>(ND); | 
| Douglas Gregor | eb5758b | 2009-09-23 22:26:46 +0000 | [diff] [blame] | 1085 | } | 
|  | 1086 |  | 
| Douglas Gregor | 8e254cf | 2010-05-27 23:06:34 +0000 | [diff] [blame] | 1087 | static bool isObjCReceiverType(ASTContext &C, QualType T) { | 
|  | 1088 | T = C.getCanonicalType(T); | 
|  | 1089 | switch (T->getTypeClass()) { | 
|  | 1090 | case Type::ObjCObject: | 
|  | 1091 | case Type::ObjCInterface: | 
|  | 1092 | case Type::ObjCObjectPointer: | 
|  | 1093 | return true; | 
|  | 1094 |  | 
|  | 1095 | case Type::Builtin: | 
|  | 1096 | switch (cast<BuiltinType>(T)->getKind()) { | 
|  | 1097 | case BuiltinType::ObjCId: | 
|  | 1098 | case BuiltinType::ObjCClass: | 
|  | 1099 | case BuiltinType::ObjCSel: | 
|  | 1100 | return true; | 
|  | 1101 |  | 
|  | 1102 | default: | 
|  | 1103 | break; | 
|  | 1104 | } | 
|  | 1105 | return false; | 
|  | 1106 |  | 
|  | 1107 | default: | 
|  | 1108 | break; | 
|  | 1109 | } | 
|  | 1110 |  | 
|  | 1111 | if (!C.getLangOptions().CPlusPlus) | 
|  | 1112 | return false; | 
|  | 1113 |  | 
|  | 1114 | // FIXME: We could perform more analysis here to determine whether a | 
|  | 1115 | // particular class type has any conversions to Objective-C types. For now, | 
|  | 1116 | // just accept all class types. | 
|  | 1117 | return T->isDependentType() || T->isRecordType(); | 
|  | 1118 | } | 
|  | 1119 |  | 
|  | 1120 | bool ResultBuilder::IsObjCMessageReceiver(NamedDecl *ND) const { | 
|  | 1121 | QualType T = getDeclUsageType(SemaRef.Context, ND); | 
|  | 1122 | if (T.isNull()) | 
|  | 1123 | return false; | 
|  | 1124 |  | 
|  | 1125 | T = SemaRef.Context.getBaseElementType(T); | 
|  | 1126 | return isObjCReceiverType(SemaRef.Context, T); | 
|  | 1127 | } | 
|  | 1128 |  | 
| Douglas Gregor | fb62941 | 2010-08-23 21:17:50 +0000 | [diff] [blame] | 1129 | bool ResultBuilder::IsObjCCollection(NamedDecl *ND) const { | 
|  | 1130 | if ((SemaRef.getLangOptions().CPlusPlus && !IsOrdinaryName(ND)) || | 
|  | 1131 | (!SemaRef.getLangOptions().CPlusPlus && !IsOrdinaryNonTypeName(ND))) | 
|  | 1132 | return false; | 
|  | 1133 |  | 
|  | 1134 | QualType T = getDeclUsageType(SemaRef.Context, ND); | 
|  | 1135 | if (T.isNull()) | 
|  | 1136 | return false; | 
|  | 1137 |  | 
|  | 1138 | T = SemaRef.Context.getBaseElementType(T); | 
|  | 1139 | return T->isObjCObjectType() || T->isObjCObjectPointerType() || | 
|  | 1140 | T->isObjCIdType() || | 
|  | 1141 | (SemaRef.getLangOptions().CPlusPlus && T->isRecordType()); | 
|  | 1142 | } | 
| Douglas Gregor | 8e254cf | 2010-05-27 23:06:34 +0000 | [diff] [blame] | 1143 |  | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 1144 | bool ResultBuilder::IsImpossibleToSatisfy(NamedDecl *ND) const { | 
|  | 1145 | return false; | 
|  | 1146 | } | 
|  | 1147 |  | 
| Douglas Gregor | 80f4f4c | 2010-01-14 16:08:12 +0000 | [diff] [blame] | 1148 | /// \rief Determines whether the given declaration is an Objective-C | 
|  | 1149 | /// instance variable. | 
|  | 1150 | bool ResultBuilder::IsObjCIvar(NamedDecl *ND) const { | 
|  | 1151 | return isa<ObjCIvarDecl>(ND); | 
|  | 1152 | } | 
|  | 1153 |  | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 1154 | namespace { | 
|  | 1155 | /// \brief Visible declaration consumer that adds a code-completion result | 
|  | 1156 | /// for each visible declaration. | 
|  | 1157 | class CodeCompletionDeclConsumer : public VisibleDeclConsumer { | 
|  | 1158 | ResultBuilder &Results; | 
|  | 1159 | DeclContext *CurContext; | 
|  | 1160 |  | 
|  | 1161 | public: | 
|  | 1162 | CodeCompletionDeclConsumer(ResultBuilder &Results, DeclContext *CurContext) | 
|  | 1163 | : Results(Results), CurContext(CurContext) { } | 
|  | 1164 |  | 
| Douglas Gregor | 0cc8404 | 2010-01-14 15:47:35 +0000 | [diff] [blame] | 1165 | virtual void FoundDecl(NamedDecl *ND, NamedDecl *Hiding, bool InBaseClass) { | 
|  | 1166 | Results.AddResult(ND, CurContext, Hiding, InBaseClass); | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 1167 | } | 
|  | 1168 | }; | 
|  | 1169 | } | 
|  | 1170 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1171 | /// \brief Add type specifiers for the current language as keyword results. | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 1172 | static void AddTypeSpecifierResults(const LangOptions &LangOpts, | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1173 | ResultBuilder &Results) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 1174 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 12e1313 | 2010-05-26 22:00:08 +0000 | [diff] [blame] | 1175 | Results.AddResult(Result("short", CCP_Type)); | 
|  | 1176 | Results.AddResult(Result("long", CCP_Type)); | 
|  | 1177 | Results.AddResult(Result("signed", CCP_Type)); | 
|  | 1178 | Results.AddResult(Result("unsigned", CCP_Type)); | 
|  | 1179 | Results.AddResult(Result("void", CCP_Type)); | 
|  | 1180 | Results.AddResult(Result("char", CCP_Type)); | 
|  | 1181 | Results.AddResult(Result("int", CCP_Type)); | 
|  | 1182 | Results.AddResult(Result("float", CCP_Type)); | 
|  | 1183 | Results.AddResult(Result("double", CCP_Type)); | 
|  | 1184 | Results.AddResult(Result("enum", CCP_Type)); | 
|  | 1185 | Results.AddResult(Result("struct", CCP_Type)); | 
|  | 1186 | Results.AddResult(Result("union", CCP_Type)); | 
|  | 1187 | Results.AddResult(Result("const", CCP_Type)); | 
|  | 1188 | Results.AddResult(Result("volatile", CCP_Type)); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1189 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1190 | if (LangOpts.C99) { | 
|  | 1191 | // C99-specific | 
| Douglas Gregor | 12e1313 | 2010-05-26 22:00:08 +0000 | [diff] [blame] | 1192 | Results.AddResult(Result("_Complex", CCP_Type)); | 
|  | 1193 | Results.AddResult(Result("_Imaginary", CCP_Type)); | 
|  | 1194 | Results.AddResult(Result("_Bool", CCP_Type)); | 
|  | 1195 | Results.AddResult(Result("restrict", CCP_Type)); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1196 | } | 
|  | 1197 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1198 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1199 | if (LangOpts.CPlusPlus) { | 
|  | 1200 | // C++-specific | 
| Douglas Gregor | b05496d | 2010-09-20 21:11:48 +0000 | [diff] [blame] | 1201 | Results.AddResult(Result("bool", CCP_Type + | 
|  | 1202 | (LangOpts.ObjC1? CCD_bool_in_ObjC : 0))); | 
| Douglas Gregor | 12e1313 | 2010-05-26 22:00:08 +0000 | [diff] [blame] | 1203 | Results.AddResult(Result("class", CCP_Type)); | 
|  | 1204 | Results.AddResult(Result("wchar_t", CCP_Type)); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1205 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1206 | // typename qualified-id | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1207 | Builder.AddTypedTextChunk("typename"); | 
|  | 1208 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1209 | Builder.AddPlaceholderChunk("qualifier"); | 
|  | 1210 | Builder.AddTextChunk("::"); | 
|  | 1211 | Builder.AddPlaceholderChunk("name"); | 
|  | 1212 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1213 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1214 | if (LangOpts.CPlusPlus0x) { | 
| Douglas Gregor | 12e1313 | 2010-05-26 22:00:08 +0000 | [diff] [blame] | 1215 | Results.AddResult(Result("auto", CCP_Type)); | 
|  | 1216 | Results.AddResult(Result("char16_t", CCP_Type)); | 
|  | 1217 | Results.AddResult(Result("char32_t", CCP_Type)); | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1218 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1219 | Builder.AddTypedTextChunk("decltype"); | 
|  | 1220 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 1221 | Builder.AddPlaceholderChunk("expression"); | 
|  | 1222 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1223 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1224 | } | 
|  | 1225 | } | 
|  | 1226 |  | 
|  | 1227 | // GNU extensions | 
|  | 1228 | if (LangOpts.GNUMode) { | 
|  | 1229 | // FIXME: Enable when we actually support decimal floating point. | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 1230 | //    Results.AddResult(Result("_Decimal32")); | 
|  | 1231 | //    Results.AddResult(Result("_Decimal64")); | 
|  | 1232 | //    Results.AddResult(Result("_Decimal128")); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1233 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1234 | Builder.AddTypedTextChunk("typeof"); | 
|  | 1235 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1236 | Builder.AddPlaceholderChunk("expression"); | 
|  | 1237 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1238 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1239 | Builder.AddTypedTextChunk("typeof"); | 
|  | 1240 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 1241 | Builder.AddPlaceholderChunk("type"); | 
|  | 1242 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1243 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1244 | } | 
|  | 1245 | } | 
|  | 1246 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1247 | static void AddStorageSpecifiers(Sema::ParserCompletionContext CCC, | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1248 | const LangOptions &LangOpts, | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1249 | ResultBuilder &Results) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 1250 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1251 | // Note: we don't suggest either "auto" or "register", because both | 
|  | 1252 | // are pointless as storage specifiers. Elsewhere, we suggest "auto" | 
|  | 1253 | // in C++0x as a type specifier. | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 1254 | Results.AddResult(Result("extern")); | 
|  | 1255 | Results.AddResult(Result("static")); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1256 | } | 
|  | 1257 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1258 | static void AddFunctionSpecifiers(Sema::ParserCompletionContext CCC, | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1259 | const LangOptions &LangOpts, | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1260 | ResultBuilder &Results) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 1261 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1262 | switch (CCC) { | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1263 | case Sema::PCC_Class: | 
|  | 1264 | case Sema::PCC_MemberTemplate: | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1265 | if (LangOpts.CPlusPlus) { | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 1266 | Results.AddResult(Result("explicit")); | 
|  | 1267 | Results.AddResult(Result("friend")); | 
|  | 1268 | Results.AddResult(Result("mutable")); | 
|  | 1269 | Results.AddResult(Result("virtual")); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1270 | } | 
|  | 1271 | // Fall through | 
|  | 1272 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1273 | case Sema::PCC_ObjCInterface: | 
|  | 1274 | case Sema::PCC_ObjCImplementation: | 
|  | 1275 | case Sema::PCC_Namespace: | 
|  | 1276 | case Sema::PCC_Template: | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1277 | if (LangOpts.CPlusPlus || LangOpts.C99) | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 1278 | Results.AddResult(Result("inline")); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1279 | break; | 
|  | 1280 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1281 | case Sema::PCC_ObjCInstanceVariableList: | 
|  | 1282 | case Sema::PCC_Expression: | 
|  | 1283 | case Sema::PCC_Statement: | 
|  | 1284 | case Sema::PCC_ForInit: | 
|  | 1285 | case Sema::PCC_Condition: | 
|  | 1286 | case Sema::PCC_RecoveryInFunction: | 
|  | 1287 | case Sema::PCC_Type: | 
| Douglas Gregor | 0268810 | 2010-09-14 23:59:36 +0000 | [diff] [blame] | 1288 | case Sema::PCC_ParenthesizedExpression: | 
| Douglas Gregor | 68e3c2e | 2011-02-15 20:33:25 +0000 | [diff] [blame] | 1289 | case Sema::PCC_LocalDeclarationSpecifiers: | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1290 | break; | 
|  | 1291 | } | 
|  | 1292 | } | 
|  | 1293 |  | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 1294 | static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt); | 
|  | 1295 | static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt); | 
|  | 1296 | static void AddObjCVisibilityResults(const LangOptions &LangOpts, | 
| Douglas Gregor | c38c3e1 | 2010-01-13 21:54:15 +0000 | [diff] [blame] | 1297 | ResultBuilder &Results, | 
|  | 1298 | bool NeedAt); | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 1299 | static void AddObjCImplementationResults(const LangOptions &LangOpts, | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 1300 | ResultBuilder &Results, | 
|  | 1301 | bool NeedAt); | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 1302 | static void AddObjCInterfaceResults(const LangOptions &LangOpts, | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 1303 | ResultBuilder &Results, | 
|  | 1304 | bool NeedAt); | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 1305 | static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 1306 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1307 | static void AddTypedefResult(ResultBuilder &Results) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1308 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
|  | 1309 | Builder.AddTypedTextChunk("typedef"); | 
|  | 1310 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1311 | Builder.AddPlaceholderChunk("type"); | 
|  | 1312 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1313 | Builder.AddPlaceholderChunk("name"); | 
|  | 1314 | Results.AddResult(CodeCompletionResult(Builder.TakeString())); | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1315 | } | 
|  | 1316 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1317 | static bool WantTypesInContext(Sema::ParserCompletionContext CCC, | 
| Douglas Gregor | 4710e5b | 2010-05-28 00:49:12 +0000 | [diff] [blame] | 1318 | const LangOptions &LangOpts) { | 
| Douglas Gregor | 4710e5b | 2010-05-28 00:49:12 +0000 | [diff] [blame] | 1319 | switch (CCC) { | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1320 | case Sema::PCC_Namespace: | 
|  | 1321 | case Sema::PCC_Class: | 
|  | 1322 | case Sema::PCC_ObjCInstanceVariableList: | 
|  | 1323 | case Sema::PCC_Template: | 
|  | 1324 | case Sema::PCC_MemberTemplate: | 
|  | 1325 | case Sema::PCC_Statement: | 
|  | 1326 | case Sema::PCC_RecoveryInFunction: | 
|  | 1327 | case Sema::PCC_Type: | 
| Douglas Gregor | 0268810 | 2010-09-14 23:59:36 +0000 | [diff] [blame] | 1328 | case Sema::PCC_ParenthesizedExpression: | 
| Douglas Gregor | 68e3c2e | 2011-02-15 20:33:25 +0000 | [diff] [blame] | 1329 | case Sema::PCC_LocalDeclarationSpecifiers: | 
| Douglas Gregor | 4710e5b | 2010-05-28 00:49:12 +0000 | [diff] [blame] | 1330 | return true; | 
|  | 1331 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1332 | case Sema::PCC_Expression: | 
|  | 1333 | case Sema::PCC_Condition: | 
| Douglas Gregor | 0268810 | 2010-09-14 23:59:36 +0000 | [diff] [blame] | 1334 | return LangOpts.CPlusPlus; | 
|  | 1335 |  | 
|  | 1336 | case Sema::PCC_ObjCInterface: | 
|  | 1337 | case Sema::PCC_ObjCImplementation: | 
| Douglas Gregor | 4710e5b | 2010-05-28 00:49:12 +0000 | [diff] [blame] | 1338 | return false; | 
|  | 1339 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1340 | case Sema::PCC_ForInit: | 
| Douglas Gregor | 0268810 | 2010-09-14 23:59:36 +0000 | [diff] [blame] | 1341 | return LangOpts.CPlusPlus || LangOpts.ObjC1 || LangOpts.C99; | 
| Douglas Gregor | 4710e5b | 2010-05-28 00:49:12 +0000 | [diff] [blame] | 1342 | } | 
|  | 1343 |  | 
|  | 1344 | return false; | 
|  | 1345 | } | 
|  | 1346 |  | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1347 | /// \brief Add language constructs that show up for "ordinary" names. | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1348 | static void AddOrdinaryNameResults(Sema::ParserCompletionContext CCC, | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1349 | Scope *S, | 
|  | 1350 | Sema &SemaRef, | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1351 | ResultBuilder &Results) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1352 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
|  | 1353 |  | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 1354 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1355 | switch (CCC) { | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1356 | case Sema::PCC_Namespace: | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1357 | if (SemaRef.getLangOptions().CPlusPlus) { | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1358 | if (Results.includeCodePatterns()) { | 
|  | 1359 | // namespace <identifier> { declarations } | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1360 | Builder.AddTypedTextChunk("namespace"); | 
|  | 1361 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1362 | Builder.AddPlaceholderChunk("identifier"); | 
|  | 1363 | Builder.AddChunk(CodeCompletionString::CK_LeftBrace); | 
|  | 1364 | Builder.AddPlaceholderChunk("declarations"); | 
|  | 1365 | Builder.AddChunk(CodeCompletionString::CK_VerticalSpace); | 
|  | 1366 | Builder.AddChunk(CodeCompletionString::CK_RightBrace); | 
|  | 1367 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1368 | } | 
|  | 1369 |  | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1370 | // namespace identifier = identifier ; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1371 | Builder.AddTypedTextChunk("namespace"); | 
|  | 1372 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1373 | Builder.AddPlaceholderChunk("name"); | 
|  | 1374 | Builder.AddChunk(CodeCompletionString::CK_Equal); | 
|  | 1375 | Builder.AddPlaceholderChunk("namespace"); | 
|  | 1376 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1377 |  | 
|  | 1378 | // Using directives | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1379 | Builder.AddTypedTextChunk("using"); | 
|  | 1380 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1381 | Builder.AddTextChunk("namespace"); | 
|  | 1382 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1383 | Builder.AddPlaceholderChunk("identifier"); | 
|  | 1384 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1385 |  | 
|  | 1386 | // asm(string-literal) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1387 | Builder.AddTypedTextChunk("asm"); | 
|  | 1388 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 1389 | Builder.AddPlaceholderChunk("string-literal"); | 
|  | 1390 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1391 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1392 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1393 | if (Results.includeCodePatterns()) { | 
|  | 1394 | // Explicit template instantiation | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1395 | Builder.AddTypedTextChunk("template"); | 
|  | 1396 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1397 | Builder.AddPlaceholderChunk("declaration"); | 
|  | 1398 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1399 | } | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1400 | } | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 1401 |  | 
|  | 1402 | if (SemaRef.getLangOptions().ObjC1) | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 1403 | AddObjCTopLevelResults(Results, true); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 1404 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1405 | AddTypedefResult(Results); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1406 | // Fall through | 
|  | 1407 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1408 | case Sema::PCC_Class: | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1409 | if (SemaRef.getLangOptions().CPlusPlus) { | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1410 | // Using declaration | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1411 | Builder.AddTypedTextChunk("using"); | 
|  | 1412 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1413 | Builder.AddPlaceholderChunk("qualifier"); | 
|  | 1414 | Builder.AddTextChunk("::"); | 
|  | 1415 | Builder.AddPlaceholderChunk("name"); | 
|  | 1416 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1417 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1418 | // using typename qualifier::name (only in a dependent context) | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1419 | if (SemaRef.CurContext->isDependentContext()) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1420 | Builder.AddTypedTextChunk("using"); | 
|  | 1421 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1422 | Builder.AddTextChunk("typename"); | 
|  | 1423 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1424 | Builder.AddPlaceholderChunk("qualifier"); | 
|  | 1425 | Builder.AddTextChunk("::"); | 
|  | 1426 | Builder.AddPlaceholderChunk("name"); | 
|  | 1427 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1428 | } | 
|  | 1429 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1430 | if (CCC == Sema::PCC_Class) { | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1431 | AddTypedefResult(Results); | 
|  | 1432 |  | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1433 | // public: | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1434 | Builder.AddTypedTextChunk("public"); | 
|  | 1435 | Builder.AddChunk(CodeCompletionString::CK_Colon); | 
|  | 1436 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1437 |  | 
|  | 1438 | // protected: | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1439 | Builder.AddTypedTextChunk("protected"); | 
|  | 1440 | Builder.AddChunk(CodeCompletionString::CK_Colon); | 
|  | 1441 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1442 |  | 
|  | 1443 | // private: | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1444 | Builder.AddTypedTextChunk("private"); | 
|  | 1445 | Builder.AddChunk(CodeCompletionString::CK_Colon); | 
|  | 1446 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1447 | } | 
|  | 1448 | } | 
|  | 1449 | // Fall through | 
|  | 1450 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1451 | case Sema::PCC_Template: | 
|  | 1452 | case Sema::PCC_MemberTemplate: | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1453 | if (SemaRef.getLangOptions().CPlusPlus && Results.includeCodePatterns()) { | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1454 | // template < parameters > | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1455 | Builder.AddTypedTextChunk("template"); | 
|  | 1456 | Builder.AddChunk(CodeCompletionString::CK_LeftAngle); | 
|  | 1457 | Builder.AddPlaceholderChunk("parameters"); | 
|  | 1458 | Builder.AddChunk(CodeCompletionString::CK_RightAngle); | 
|  | 1459 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1460 | } | 
|  | 1461 |  | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 1462 | AddStorageSpecifiers(CCC, SemaRef.getLangOptions(), Results); | 
|  | 1463 | AddFunctionSpecifiers(CCC, SemaRef.getLangOptions(), Results); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1464 | break; | 
|  | 1465 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1466 | case Sema::PCC_ObjCInterface: | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 1467 | AddObjCInterfaceResults(SemaRef.getLangOptions(), Results, true); | 
|  | 1468 | AddStorageSpecifiers(CCC, SemaRef.getLangOptions(), Results); | 
|  | 1469 | AddFunctionSpecifiers(CCC, SemaRef.getLangOptions(), Results); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 1470 | break; | 
|  | 1471 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1472 | case Sema::PCC_ObjCImplementation: | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 1473 | AddObjCImplementationResults(SemaRef.getLangOptions(), Results, true); | 
|  | 1474 | AddStorageSpecifiers(CCC, SemaRef.getLangOptions(), Results); | 
|  | 1475 | AddFunctionSpecifiers(CCC, SemaRef.getLangOptions(), Results); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 1476 | break; | 
|  | 1477 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1478 | case Sema::PCC_ObjCInstanceVariableList: | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 1479 | AddObjCVisibilityResults(SemaRef.getLangOptions(), Results, true); | 
| Douglas Gregor | c38c3e1 | 2010-01-13 21:54:15 +0000 | [diff] [blame] | 1480 | break; | 
|  | 1481 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1482 | case Sema::PCC_RecoveryInFunction: | 
|  | 1483 | case Sema::PCC_Statement: { | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1484 | AddTypedefResult(Results); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1485 |  | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1486 | if (SemaRef.getLangOptions().CPlusPlus && Results.includeCodePatterns()) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1487 | Builder.AddTypedTextChunk("try"); | 
|  | 1488 | Builder.AddChunk(CodeCompletionString::CK_LeftBrace); | 
|  | 1489 | Builder.AddPlaceholderChunk("statements"); | 
|  | 1490 | Builder.AddChunk(CodeCompletionString::CK_VerticalSpace); | 
|  | 1491 | Builder.AddChunk(CodeCompletionString::CK_RightBrace); | 
|  | 1492 | Builder.AddTextChunk("catch"); | 
|  | 1493 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 1494 | Builder.AddPlaceholderChunk("declaration"); | 
|  | 1495 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1496 | Builder.AddChunk(CodeCompletionString::CK_LeftBrace); | 
|  | 1497 | Builder.AddPlaceholderChunk("statements"); | 
|  | 1498 | Builder.AddChunk(CodeCompletionString::CK_VerticalSpace); | 
|  | 1499 | Builder.AddChunk(CodeCompletionString::CK_RightBrace); | 
|  | 1500 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1501 | } | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 1502 | if (SemaRef.getLangOptions().ObjC1) | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 1503 | AddObjCStatementResults(Results, true); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 1504 |  | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1505 | if (Results.includeCodePatterns()) { | 
|  | 1506 | // if (condition) { statements } | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1507 | Builder.AddTypedTextChunk("if"); | 
|  | 1508 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1509 | if (SemaRef.getLangOptions().CPlusPlus) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1510 | Builder.AddPlaceholderChunk("condition"); | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1511 | else | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1512 | Builder.AddPlaceholderChunk("expression"); | 
|  | 1513 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1514 | Builder.AddChunk(CodeCompletionString::CK_LeftBrace); | 
|  | 1515 | Builder.AddPlaceholderChunk("statements"); | 
|  | 1516 | Builder.AddChunk(CodeCompletionString::CK_VerticalSpace); | 
|  | 1517 | Builder.AddChunk(CodeCompletionString::CK_RightBrace); | 
|  | 1518 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1519 |  | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1520 | // switch (condition) { } | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1521 | Builder.AddTypedTextChunk("switch"); | 
|  | 1522 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1523 | if (SemaRef.getLangOptions().CPlusPlus) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1524 | Builder.AddPlaceholderChunk("condition"); | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1525 | else | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1526 | Builder.AddPlaceholderChunk("expression"); | 
|  | 1527 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1528 | Builder.AddChunk(CodeCompletionString::CK_LeftBrace); | 
|  | 1529 | Builder.AddChunk(CodeCompletionString::CK_VerticalSpace); | 
|  | 1530 | Builder.AddChunk(CodeCompletionString::CK_RightBrace); | 
|  | 1531 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1532 | } | 
|  | 1533 |  | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1534 | // Switch-specific statements. | 
| John McCall | 781472f | 2010-08-25 08:40:02 +0000 | [diff] [blame] | 1535 | if (!SemaRef.getCurFunction()->SwitchStack.empty()) { | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1536 | // case expression: | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1537 | Builder.AddTypedTextChunk("case"); | 
|  | 1538 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1539 | Builder.AddPlaceholderChunk("expression"); | 
|  | 1540 | Builder.AddChunk(CodeCompletionString::CK_Colon); | 
|  | 1541 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1542 |  | 
|  | 1543 | // default: | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1544 | Builder.AddTypedTextChunk("default"); | 
|  | 1545 | Builder.AddChunk(CodeCompletionString::CK_Colon); | 
|  | 1546 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1547 | } | 
|  | 1548 |  | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1549 | if (Results.includeCodePatterns()) { | 
|  | 1550 | /// while (condition) { statements } | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1551 | Builder.AddTypedTextChunk("while"); | 
|  | 1552 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1553 | if (SemaRef.getLangOptions().CPlusPlus) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1554 | Builder.AddPlaceholderChunk("condition"); | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1555 | else | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1556 | Builder.AddPlaceholderChunk("expression"); | 
|  | 1557 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1558 | Builder.AddChunk(CodeCompletionString::CK_LeftBrace); | 
|  | 1559 | Builder.AddPlaceholderChunk("statements"); | 
|  | 1560 | Builder.AddChunk(CodeCompletionString::CK_VerticalSpace); | 
|  | 1561 | Builder.AddChunk(CodeCompletionString::CK_RightBrace); | 
|  | 1562 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1563 |  | 
|  | 1564 | // do { statements } while ( expression ); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1565 | Builder.AddTypedTextChunk("do"); | 
|  | 1566 | Builder.AddChunk(CodeCompletionString::CK_LeftBrace); | 
|  | 1567 | Builder.AddPlaceholderChunk("statements"); | 
|  | 1568 | Builder.AddChunk(CodeCompletionString::CK_VerticalSpace); | 
|  | 1569 | Builder.AddChunk(CodeCompletionString::CK_RightBrace); | 
|  | 1570 | Builder.AddTextChunk("while"); | 
|  | 1571 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 1572 | Builder.AddPlaceholderChunk("expression"); | 
|  | 1573 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1574 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1575 |  | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1576 | // for ( for-init-statement ; condition ; expression ) { statements } | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1577 | Builder.AddTypedTextChunk("for"); | 
|  | 1578 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1579 | if (SemaRef.getLangOptions().CPlusPlus || SemaRef.getLangOptions().C99) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1580 | Builder.AddPlaceholderChunk("init-statement"); | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1581 | else | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1582 | Builder.AddPlaceholderChunk("init-expression"); | 
|  | 1583 | Builder.AddChunk(CodeCompletionString::CK_SemiColon); | 
|  | 1584 | Builder.AddPlaceholderChunk("condition"); | 
|  | 1585 | Builder.AddChunk(CodeCompletionString::CK_SemiColon); | 
|  | 1586 | Builder.AddPlaceholderChunk("inc-expression"); | 
|  | 1587 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1588 | Builder.AddChunk(CodeCompletionString::CK_LeftBrace); | 
|  | 1589 | Builder.AddChunk(CodeCompletionString::CK_VerticalSpace); | 
|  | 1590 | Builder.AddPlaceholderChunk("statements"); | 
|  | 1591 | Builder.AddChunk(CodeCompletionString::CK_VerticalSpace); | 
|  | 1592 | Builder.AddChunk(CodeCompletionString::CK_RightBrace); | 
|  | 1593 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 1594 | } | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1595 |  | 
|  | 1596 | if (S->getContinueParent()) { | 
|  | 1597 | // continue ; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1598 | Builder.AddTypedTextChunk("continue"); | 
|  | 1599 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1600 | } | 
|  | 1601 |  | 
|  | 1602 | if (S->getBreakParent()) { | 
|  | 1603 | // break ; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1604 | Builder.AddTypedTextChunk("break"); | 
|  | 1605 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1606 | } | 
|  | 1607 |  | 
|  | 1608 | // "return expression ;" or "return ;", depending on whether we | 
|  | 1609 | // know the function is void or not. | 
|  | 1610 | bool isVoid = false; | 
|  | 1611 | if (FunctionDecl *Function = dyn_cast<FunctionDecl>(SemaRef.CurContext)) | 
|  | 1612 | isVoid = Function->getResultType()->isVoidType(); | 
|  | 1613 | else if (ObjCMethodDecl *Method | 
|  | 1614 | = dyn_cast<ObjCMethodDecl>(SemaRef.CurContext)) | 
|  | 1615 | isVoid = Method->getResultType()->isVoidType(); | 
| Douglas Gregor | 9ea9bdb | 2010-03-01 23:15:13 +0000 | [diff] [blame] | 1616 | else if (SemaRef.getCurBlock() && | 
|  | 1617 | !SemaRef.getCurBlock()->ReturnType.isNull()) | 
|  | 1618 | isVoid = SemaRef.getCurBlock()->ReturnType->isVoidType(); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1619 | Builder.AddTypedTextChunk("return"); | 
| Douglas Gregor | 9329800 | 2010-02-18 04:06:48 +0000 | [diff] [blame] | 1620 | if (!isVoid) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1621 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1622 | Builder.AddPlaceholderChunk("expression"); | 
| Douglas Gregor | 9329800 | 2010-02-18 04:06:48 +0000 | [diff] [blame] | 1623 | } | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1624 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1625 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1626 | // goto identifier ; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1627 | Builder.AddTypedTextChunk("goto"); | 
|  | 1628 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1629 | Builder.AddPlaceholderChunk("label"); | 
|  | 1630 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1631 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1632 | // Using directives | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1633 | Builder.AddTypedTextChunk("using"); | 
|  | 1634 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1635 | Builder.AddTextChunk("namespace"); | 
|  | 1636 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1637 | Builder.AddPlaceholderChunk("identifier"); | 
|  | 1638 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1639 | } | 
|  | 1640 |  | 
|  | 1641 | // Fall through (for statement expressions). | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1642 | case Sema::PCC_ForInit: | 
|  | 1643 | case Sema::PCC_Condition: | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 1644 | AddStorageSpecifiers(CCC, SemaRef.getLangOptions(), Results); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1645 | // Fall through: conditions and statements can have expressions. | 
|  | 1646 |  | 
| Douglas Gregor | 0268810 | 2010-09-14 23:59:36 +0000 | [diff] [blame] | 1647 | case Sema::PCC_ParenthesizedExpression: | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1648 | case Sema::PCC_Expression: { | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1649 | if (SemaRef.getLangOptions().CPlusPlus) { | 
|  | 1650 | // 'this', if we're in a non-static member function. | 
|  | 1651 | if (CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(SemaRef.CurContext)) | 
|  | 1652 | if (!Method->isStatic()) | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 1653 | Results.AddResult(Result("this")); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1654 |  | 
|  | 1655 | // true, false | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 1656 | Results.AddResult(Result("true")); | 
|  | 1657 | Results.AddResult(Result("false")); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1658 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1659 | // dynamic_cast < type-id > ( expression ) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1660 | Builder.AddTypedTextChunk("dynamic_cast"); | 
|  | 1661 | Builder.AddChunk(CodeCompletionString::CK_LeftAngle); | 
|  | 1662 | Builder.AddPlaceholderChunk("type"); | 
|  | 1663 | Builder.AddChunk(CodeCompletionString::CK_RightAngle); | 
|  | 1664 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 1665 | Builder.AddPlaceholderChunk("expression"); | 
|  | 1666 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1667 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1668 |  | 
|  | 1669 | // static_cast < type-id > ( expression ) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1670 | Builder.AddTypedTextChunk("static_cast"); | 
|  | 1671 | Builder.AddChunk(CodeCompletionString::CK_LeftAngle); | 
|  | 1672 | Builder.AddPlaceholderChunk("type"); | 
|  | 1673 | Builder.AddChunk(CodeCompletionString::CK_RightAngle); | 
|  | 1674 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 1675 | Builder.AddPlaceholderChunk("expression"); | 
|  | 1676 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1677 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1678 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1679 | // reinterpret_cast < type-id > ( expression ) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1680 | Builder.AddTypedTextChunk("reinterpret_cast"); | 
|  | 1681 | Builder.AddChunk(CodeCompletionString::CK_LeftAngle); | 
|  | 1682 | Builder.AddPlaceholderChunk("type"); | 
|  | 1683 | Builder.AddChunk(CodeCompletionString::CK_RightAngle); | 
|  | 1684 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 1685 | Builder.AddPlaceholderChunk("expression"); | 
|  | 1686 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1687 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1688 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1689 | // const_cast < type-id > ( expression ) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1690 | Builder.AddTypedTextChunk("const_cast"); | 
|  | 1691 | Builder.AddChunk(CodeCompletionString::CK_LeftAngle); | 
|  | 1692 | Builder.AddPlaceholderChunk("type"); | 
|  | 1693 | Builder.AddChunk(CodeCompletionString::CK_RightAngle); | 
|  | 1694 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 1695 | Builder.AddPlaceholderChunk("expression"); | 
|  | 1696 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1697 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1698 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1699 | // typeid ( expression-or-type ) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1700 | Builder.AddTypedTextChunk("typeid"); | 
|  | 1701 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 1702 | Builder.AddPlaceholderChunk("expression-or-type"); | 
|  | 1703 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1704 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1705 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1706 | // new T ( ... ) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1707 | Builder.AddTypedTextChunk("new"); | 
|  | 1708 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1709 | Builder.AddPlaceholderChunk("type"); | 
|  | 1710 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 1711 | Builder.AddPlaceholderChunk("expressions"); | 
|  | 1712 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1713 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1714 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1715 | // new T [ ] ( ... ) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1716 | Builder.AddTypedTextChunk("new"); | 
|  | 1717 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1718 | Builder.AddPlaceholderChunk("type"); | 
|  | 1719 | Builder.AddChunk(CodeCompletionString::CK_LeftBracket); | 
|  | 1720 | Builder.AddPlaceholderChunk("size"); | 
|  | 1721 | Builder.AddChunk(CodeCompletionString::CK_RightBracket); | 
|  | 1722 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 1723 | Builder.AddPlaceholderChunk("expressions"); | 
|  | 1724 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1725 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1726 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1727 | // delete expression | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1728 | Builder.AddTypedTextChunk("delete"); | 
|  | 1729 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1730 | Builder.AddPlaceholderChunk("expression"); | 
|  | 1731 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1732 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1733 | // delete [] expression | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1734 | Builder.AddTypedTextChunk("delete"); | 
|  | 1735 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1736 | Builder.AddChunk(CodeCompletionString::CK_LeftBracket); | 
|  | 1737 | Builder.AddChunk(CodeCompletionString::CK_RightBracket); | 
|  | 1738 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1739 | Builder.AddPlaceholderChunk("expression"); | 
|  | 1740 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1741 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1742 | // throw expression | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1743 | Builder.AddTypedTextChunk("throw"); | 
|  | 1744 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 1745 | Builder.AddPlaceholderChunk("expression"); | 
|  | 1746 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 12e1313 | 2010-05-26 22:00:08 +0000 | [diff] [blame] | 1747 |  | 
|  | 1748 | // FIXME: Rethrow? | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1749 | } | 
|  | 1750 |  | 
|  | 1751 | if (SemaRef.getLangOptions().ObjC1) { | 
|  | 1752 | // Add "super", if we're in an Objective-C class with a superclass. | 
| Ted Kremenek | 681e256 | 2010-05-31 21:43:10 +0000 | [diff] [blame] | 1753 | if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl()) { | 
|  | 1754 | // The interface can be NULL. | 
|  | 1755 | if (ObjCInterfaceDecl *ID = Method->getClassInterface()) | 
|  | 1756 | if (ID->getSuperClass()) | 
|  | 1757 | Results.AddResult(Result("super")); | 
|  | 1758 | } | 
|  | 1759 |  | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 1760 | AddObjCExpressionResults(Results, true); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1761 | } | 
|  | 1762 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 1763 | // sizeof expression | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1764 | Builder.AddTypedTextChunk("sizeof"); | 
|  | 1765 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 1766 | Builder.AddPlaceholderChunk("expression-or-type"); | 
|  | 1767 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 1768 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1769 | break; | 
|  | 1770 | } | 
| Douglas Gregor | d32b022 | 2010-08-24 01:06:58 +0000 | [diff] [blame] | 1771 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1772 | case Sema::PCC_Type: | 
| Douglas Gregor | 68e3c2e | 2011-02-15 20:33:25 +0000 | [diff] [blame] | 1773 | case Sema::PCC_LocalDeclarationSpecifiers: | 
| Douglas Gregor | d32b022 | 2010-08-24 01:06:58 +0000 | [diff] [blame] | 1774 | break; | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1775 | } | 
|  | 1776 |  | 
| Douglas Gregor | 4710e5b | 2010-05-28 00:49:12 +0000 | [diff] [blame] | 1777 | if (WantTypesInContext(CCC, SemaRef.getLangOptions())) | 
|  | 1778 | AddTypeSpecifierResults(SemaRef.getLangOptions(), Results); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1779 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 1780 | if (SemaRef.getLangOptions().CPlusPlus && CCC != Sema::PCC_Type) | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 1781 | Results.AddResult(Result("operator")); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 1782 | } | 
|  | 1783 |  | 
| Douglas Gregor | a63f6de | 2011-02-01 21:15:40 +0000 | [diff] [blame] | 1784 | /// \brief Retrieve the string representation of the given type as a string | 
|  | 1785 | /// that has the appropriate lifetime for code completion. | 
|  | 1786 | /// | 
|  | 1787 | /// This routine provides a fast path where we provide constant strings for | 
|  | 1788 | /// common type names. | 
|  | 1789 | const char *GetCompletionTypeString(QualType T, | 
|  | 1790 | ASTContext &Context, | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 1791 | CodeCompletionAllocator &Allocator) { | 
| Douglas Gregor | a63f6de | 2011-02-01 21:15:40 +0000 | [diff] [blame] | 1792 | PrintingPolicy Policy(Context.PrintingPolicy); | 
|  | 1793 | Policy.AnonymousTagLocations = false; | 
|  | 1794 |  | 
|  | 1795 | if (!T.getLocalQualifiers()) { | 
|  | 1796 | // Built-in type names are constant strings. | 
|  | 1797 | if (const BuiltinType *BT = dyn_cast<BuiltinType>(T)) | 
|  | 1798 | return BT->getName(Context.getLangOptions()); | 
|  | 1799 |  | 
|  | 1800 | // Anonymous tag types are constant strings. | 
|  | 1801 | if (const TagType *TagT = dyn_cast<TagType>(T)) | 
|  | 1802 | if (TagDecl *Tag = TagT->getDecl()) | 
|  | 1803 | if (!Tag->getIdentifier() && !Tag->getTypedefForAnonDecl()) { | 
|  | 1804 | switch (Tag->getTagKind()) { | 
|  | 1805 | case TTK_Struct: return "struct <anonymous>"; | 
|  | 1806 | case TTK_Class:  return "class <anonymous>"; | 
|  | 1807 | case TTK_Union:  return "union <anonymous>"; | 
|  | 1808 | case TTK_Enum:   return "enum <anonymous>"; | 
|  | 1809 | } | 
|  | 1810 | } | 
|  | 1811 | } | 
|  | 1812 |  | 
|  | 1813 | // Slow path: format the type as a string. | 
|  | 1814 | std::string Result; | 
|  | 1815 | T.getAsStringInternal(Result, Policy); | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 1816 | return Allocator.CopyString(Result); | 
| Douglas Gregor | a63f6de | 2011-02-01 21:15:40 +0000 | [diff] [blame] | 1817 | } | 
|  | 1818 |  | 
| Douglas Gregor | ff5ce6e | 2009-12-18 18:53:37 +0000 | [diff] [blame] | 1819 | /// \brief If the given declaration has an associated type, add it as a result | 
|  | 1820 | /// type chunk. | 
|  | 1821 | static void AddResultTypeChunk(ASTContext &Context, | 
|  | 1822 | NamedDecl *ND, | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1823 | CodeCompletionBuilder &Result) { | 
| Douglas Gregor | ff5ce6e | 2009-12-18 18:53:37 +0000 | [diff] [blame] | 1824 | if (!ND) | 
|  | 1825 | return; | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 1826 |  | 
|  | 1827 | // Skip constructors and conversion functions, which have their return types | 
|  | 1828 | // built into their names. | 
|  | 1829 | if (isa<CXXConstructorDecl>(ND) || isa<CXXConversionDecl>(ND)) | 
|  | 1830 | return; | 
|  | 1831 |  | 
| Douglas Gregor | ff5ce6e | 2009-12-18 18:53:37 +0000 | [diff] [blame] | 1832 | // Determine the type of the declaration (if it has a type). | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 1833 | QualType T; | 
| Douglas Gregor | ff5ce6e | 2009-12-18 18:53:37 +0000 | [diff] [blame] | 1834 | if (FunctionDecl *Function = dyn_cast<FunctionDecl>(ND)) | 
|  | 1835 | T = Function->getResultType(); | 
|  | 1836 | else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND)) | 
|  | 1837 | T = Method->getResultType(); | 
|  | 1838 | else if (FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND)) | 
|  | 1839 | T = FunTmpl->getTemplatedDecl()->getResultType(); | 
|  | 1840 | else if (EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND)) | 
|  | 1841 | T = Context.getTypeDeclType(cast<TypeDecl>(Enumerator->getDeclContext())); | 
|  | 1842 | else if (isa<UnresolvedUsingValueDecl>(ND)) { | 
|  | 1843 | /* Do nothing: ignore unresolved using declarations*/ | 
|  | 1844 | } else if (ValueDecl *Value = dyn_cast<ValueDecl>(ND)) | 
|  | 1845 | T = Value->getType(); | 
|  | 1846 | else if (ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND)) | 
|  | 1847 | T = Property->getType(); | 
|  | 1848 |  | 
|  | 1849 | if (T.isNull() || Context.hasSameType(T, Context.DependentTy)) | 
|  | 1850 | return; | 
|  | 1851 |  | 
| Douglas Gregor | a63f6de | 2011-02-01 21:15:40 +0000 | [diff] [blame] | 1852 | Result.AddResultTypeChunk(GetCompletionTypeString(T, Context, | 
|  | 1853 | Result.getAllocator())); | 
| Douglas Gregor | ff5ce6e | 2009-12-18 18:53:37 +0000 | [diff] [blame] | 1854 | } | 
|  | 1855 |  | 
| Douglas Gregor | aaa107a | 2010-08-23 23:51:41 +0000 | [diff] [blame] | 1856 | static void MaybeAddSentinel(ASTContext &Context, NamedDecl *FunctionOrMethod, | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1857 | CodeCompletionBuilder &Result) { | 
| Douglas Gregor | aaa107a | 2010-08-23 23:51:41 +0000 | [diff] [blame] | 1858 | if (SentinelAttr *Sentinel = FunctionOrMethod->getAttr<SentinelAttr>()) | 
|  | 1859 | if (Sentinel->getSentinel() == 0) { | 
|  | 1860 | if (Context.getLangOptions().ObjC1 && | 
|  | 1861 | Context.Idents.get("nil").hasMacroDefinition()) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1862 | Result.AddTextChunk(", nil"); | 
| Douglas Gregor | aaa107a | 2010-08-23 23:51:41 +0000 | [diff] [blame] | 1863 | else if (Context.Idents.get("NULL").hasMacroDefinition()) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1864 | Result.AddTextChunk(", NULL"); | 
| Douglas Gregor | aaa107a | 2010-08-23 23:51:41 +0000 | [diff] [blame] | 1865 | else | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1866 | Result.AddTextChunk(", (void*)0"); | 
| Douglas Gregor | aaa107a | 2010-08-23 23:51:41 +0000 | [diff] [blame] | 1867 | } | 
|  | 1868 | } | 
|  | 1869 |  | 
| Douglas Gregor | 83482d1 | 2010-08-24 16:15:59 +0000 | [diff] [blame] | 1870 | static std::string FormatFunctionParameter(ASTContext &Context, | 
| Douglas Gregor | aba4808 | 2010-08-29 19:47:46 +0000 | [diff] [blame] | 1871 | ParmVarDecl *Param, | 
|  | 1872 | bool SuppressName = false) { | 
| Douglas Gregor | 83482d1 | 2010-08-24 16:15:59 +0000 | [diff] [blame] | 1873 | bool ObjCMethodParam = isa<ObjCMethodDecl>(Param->getDeclContext()); | 
|  | 1874 | if (Param->getType()->isDependentType() || | 
|  | 1875 | !Param->getType()->isBlockPointerType()) { | 
|  | 1876 | // The argument for a dependent or non-block parameter is a placeholder | 
|  | 1877 | // containing that parameter's type. | 
|  | 1878 | std::string Result; | 
|  | 1879 |  | 
| Douglas Gregor | aba4808 | 2010-08-29 19:47:46 +0000 | [diff] [blame] | 1880 | if (Param->getIdentifier() && !ObjCMethodParam && !SuppressName) | 
| Douglas Gregor | 83482d1 | 2010-08-24 16:15:59 +0000 | [diff] [blame] | 1881 | Result = Param->getIdentifier()->getName(); | 
|  | 1882 |  | 
|  | 1883 | Param->getType().getAsStringInternal(Result, | 
|  | 1884 | Context.PrintingPolicy); | 
|  | 1885 |  | 
|  | 1886 | if (ObjCMethodParam) { | 
|  | 1887 | Result = "(" + Result; | 
|  | 1888 | Result += ")"; | 
| Douglas Gregor | aba4808 | 2010-08-29 19:47:46 +0000 | [diff] [blame] | 1889 | if (Param->getIdentifier() && !SuppressName) | 
| Douglas Gregor | 83482d1 | 2010-08-24 16:15:59 +0000 | [diff] [blame] | 1890 | Result += Param->getIdentifier()->getName(); | 
|  | 1891 | } | 
|  | 1892 | return Result; | 
|  | 1893 | } | 
|  | 1894 |  | 
|  | 1895 | // The argument for a block pointer parameter is a block literal with | 
|  | 1896 | // the appropriate type. | 
| Douglas Gregor | 830072c | 2011-02-15 22:37:09 +0000 | [diff] [blame] | 1897 | FunctionTypeLoc *Block = 0; | 
|  | 1898 | FunctionProtoTypeLoc *BlockProto = 0; | 
| Douglas Gregor | 83482d1 | 2010-08-24 16:15:59 +0000 | [diff] [blame] | 1899 | TypeLoc TL; | 
|  | 1900 | if (TypeSourceInfo *TSInfo = Param->getTypeSourceInfo()) { | 
|  | 1901 | TL = TSInfo->getTypeLoc().getUnqualifiedLoc(); | 
|  | 1902 | while (true) { | 
|  | 1903 | // Look through typedefs. | 
|  | 1904 | if (TypedefTypeLoc *TypedefTL = dyn_cast<TypedefTypeLoc>(&TL)) { | 
|  | 1905 | if (TypeSourceInfo *InnerTSInfo | 
|  | 1906 | = TypedefTL->getTypedefDecl()->getTypeSourceInfo()) { | 
|  | 1907 | TL = InnerTSInfo->getTypeLoc().getUnqualifiedLoc(); | 
|  | 1908 | continue; | 
|  | 1909 | } | 
|  | 1910 | } | 
|  | 1911 |  | 
|  | 1912 | // Look through qualified types | 
|  | 1913 | if (QualifiedTypeLoc *QualifiedTL = dyn_cast<QualifiedTypeLoc>(&TL)) { | 
|  | 1914 | TL = QualifiedTL->getUnqualifiedLoc(); | 
|  | 1915 | continue; | 
|  | 1916 | } | 
|  | 1917 |  | 
|  | 1918 | // Try to get the function prototype behind the block pointer type, | 
|  | 1919 | // then we're done. | 
|  | 1920 | if (BlockPointerTypeLoc *BlockPtr | 
|  | 1921 | = dyn_cast<BlockPointerTypeLoc>(&TL)) { | 
| Abramo Bagnara | 723df24 | 2010-12-14 22:11:44 +0000 | [diff] [blame] | 1922 | TL = BlockPtr->getPointeeLoc().IgnoreParens(); | 
| Douglas Gregor | 830072c | 2011-02-15 22:37:09 +0000 | [diff] [blame] | 1923 | Block = dyn_cast<FunctionTypeLoc>(&TL); | 
|  | 1924 | BlockProto = dyn_cast<FunctionProtoTypeLoc>(&TL); | 
| Douglas Gregor | 83482d1 | 2010-08-24 16:15:59 +0000 | [diff] [blame] | 1925 | } | 
|  | 1926 | break; | 
|  | 1927 | } | 
|  | 1928 | } | 
|  | 1929 |  | 
|  | 1930 | if (!Block) { | 
|  | 1931 | // We were unable to find a FunctionProtoTypeLoc with parameter names | 
|  | 1932 | // for the block; just use the parameter type as a placeholder. | 
|  | 1933 | std::string Result; | 
|  | 1934 | Param->getType().getUnqualifiedType(). | 
|  | 1935 | getAsStringInternal(Result, Context.PrintingPolicy); | 
|  | 1936 |  | 
|  | 1937 | if (ObjCMethodParam) { | 
|  | 1938 | Result = "(" + Result; | 
|  | 1939 | Result += ")"; | 
|  | 1940 | if (Param->getIdentifier()) | 
|  | 1941 | Result += Param->getIdentifier()->getName(); | 
|  | 1942 | } | 
|  | 1943 |  | 
|  | 1944 | return Result; | 
|  | 1945 | } | 
|  | 1946 |  | 
|  | 1947 | // We have the function prototype behind the block pointer type, as it was | 
|  | 1948 | // written in the source. | 
| Douglas Gregor | 3827625 | 2010-09-08 22:47:51 +0000 | [diff] [blame] | 1949 | std::string Result; | 
|  | 1950 | QualType ResultType = Block->getTypePtr()->getResultType(); | 
|  | 1951 | if (!ResultType->isVoidType()) | 
|  | 1952 | ResultType.getAsStringInternal(Result, Context.PrintingPolicy); | 
|  | 1953 |  | 
|  | 1954 | Result = '^' + Result; | 
| Douglas Gregor | 830072c | 2011-02-15 22:37:09 +0000 | [diff] [blame] | 1955 | if (!BlockProto || Block->getNumArgs() == 0) { | 
|  | 1956 | if (BlockProto && BlockProto->getTypePtr()->isVariadic()) | 
| Douglas Gregor | 3827625 | 2010-09-08 22:47:51 +0000 | [diff] [blame] | 1957 | Result += "(...)"; | 
| Douglas Gregor | c2760bc | 2010-10-02 23:49:58 +0000 | [diff] [blame] | 1958 | else | 
|  | 1959 | Result += "(void)"; | 
| Douglas Gregor | 3827625 | 2010-09-08 22:47:51 +0000 | [diff] [blame] | 1960 | } else { | 
|  | 1961 | Result += "("; | 
|  | 1962 | for (unsigned I = 0, N = Block->getNumArgs(); I != N; ++I) { | 
|  | 1963 | if (I) | 
|  | 1964 | Result += ", "; | 
|  | 1965 | Result += FormatFunctionParameter(Context, Block->getArg(I)); | 
|  | 1966 |  | 
| Douglas Gregor | 830072c | 2011-02-15 22:37:09 +0000 | [diff] [blame] | 1967 | if (I == N - 1 && BlockProto->getTypePtr()->isVariadic()) | 
| Douglas Gregor | 3827625 | 2010-09-08 22:47:51 +0000 | [diff] [blame] | 1968 | Result += ", ..."; | 
|  | 1969 | } | 
|  | 1970 | Result += ")"; | 
| Douglas Gregor | e17794f | 2010-08-31 05:13:43 +0000 | [diff] [blame] | 1971 | } | 
| Douglas Gregor | 3827625 | 2010-09-08 22:47:51 +0000 | [diff] [blame] | 1972 |  | 
| Douglas Gregor | c2760bc | 2010-10-02 23:49:58 +0000 | [diff] [blame] | 1973 | if (Param->getIdentifier()) | 
|  | 1974 | Result += Param->getIdentifier()->getName(); | 
|  | 1975 |  | 
| Douglas Gregor | 83482d1 | 2010-08-24 16:15:59 +0000 | [diff] [blame] | 1976 | return Result; | 
|  | 1977 | } | 
|  | 1978 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1979 | /// \brief Add function parameter chunks to the given code completion string. | 
|  | 1980 | static void AddFunctionParameterChunks(ASTContext &Context, | 
|  | 1981 | FunctionDecl *Function, | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1982 | CodeCompletionBuilder &Result, | 
|  | 1983 | unsigned Start = 0, | 
|  | 1984 | bool InOptional = false) { | 
| Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1985 | typedef CodeCompletionString::Chunk Chunk; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1986 | bool FirstParameter = true; | 
| Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 1987 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1988 | for (unsigned P = Start, N = Function->getNumParams(); P != N; ++P) { | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1989 | ParmVarDecl *Param = Function->getParamDecl(P); | 
|  | 1990 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1991 | if (Param->hasDefaultArg() && !InOptional) { | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 1992 | // When we see an optional default argument, put that argument and | 
|  | 1993 | // the remaining default arguments into a new, optional string. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 1994 | CodeCompletionBuilder Opt(Result.getAllocator()); | 
|  | 1995 | if (!FirstParameter) | 
|  | 1996 | Opt.AddChunk(Chunk(CodeCompletionString::CK_Comma)); | 
|  | 1997 | AddFunctionParameterChunks(Context, Function, Opt, P, true); | 
|  | 1998 | Result.AddOptionalChunk(Opt.TakeString()); | 
|  | 1999 | break; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2000 | } | 
|  | 2001 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2002 | if (FirstParameter) | 
|  | 2003 | FirstParameter = false; | 
|  | 2004 | else | 
|  | 2005 | Result.AddChunk(Chunk(CodeCompletionString::CK_Comma)); | 
|  | 2006 |  | 
|  | 2007 | InOptional = false; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2008 |  | 
|  | 2009 | // Format the placeholder string. | 
| Douglas Gregor | 83482d1 | 2010-08-24 16:15:59 +0000 | [diff] [blame] | 2010 | std::string PlaceholderStr = FormatFunctionParameter(Context, Param); | 
|  | 2011 |  | 
| Douglas Gregor | e17794f | 2010-08-31 05:13:43 +0000 | [diff] [blame] | 2012 | if (Function->isVariadic() && P == N - 1) | 
|  | 2013 | PlaceholderStr += ", ..."; | 
|  | 2014 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2015 | // Add the placeholder string. | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2016 | Result.AddPlaceholderChunk( | 
|  | 2017 | Result.getAllocator().CopyString(PlaceholderStr)); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2018 | } | 
| Douglas Gregor | b3d4525 | 2009-09-22 21:42:17 +0000 | [diff] [blame] | 2019 |  | 
|  | 2020 | if (const FunctionProtoType *Proto | 
|  | 2021 | = Function->getType()->getAs<FunctionProtoType>()) | 
| Douglas Gregor | aaa107a | 2010-08-23 23:51:41 +0000 | [diff] [blame] | 2022 | if (Proto->isVariadic()) { | 
| Douglas Gregor | e17794f | 2010-08-31 05:13:43 +0000 | [diff] [blame] | 2023 | if (Proto->getNumArgs() == 0) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2024 | Result.AddPlaceholderChunk("..."); | 
| Douglas Gregor | aaa107a | 2010-08-23 23:51:41 +0000 | [diff] [blame] | 2025 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2026 | MaybeAddSentinel(Context, Function, Result); | 
| Douglas Gregor | aaa107a | 2010-08-23 23:51:41 +0000 | [diff] [blame] | 2027 | } | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2028 | } | 
|  | 2029 |  | 
|  | 2030 | /// \brief Add template parameter chunks to the given code completion string. | 
|  | 2031 | static void AddTemplateParameterChunks(ASTContext &Context, | 
|  | 2032 | TemplateDecl *Template, | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2033 | CodeCompletionBuilder &Result, | 
|  | 2034 | unsigned MaxParameters = 0, | 
|  | 2035 | unsigned Start = 0, | 
|  | 2036 | bool InDefaultArg = false) { | 
| Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 2037 | typedef CodeCompletionString::Chunk Chunk; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2038 | bool FirstParameter = true; | 
|  | 2039 |  | 
|  | 2040 | TemplateParameterList *Params = Template->getTemplateParameters(); | 
|  | 2041 | TemplateParameterList::iterator PEnd = Params->end(); | 
|  | 2042 | if (MaxParameters) | 
|  | 2043 | PEnd = Params->begin() + MaxParameters; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2044 | for (TemplateParameterList::iterator P = Params->begin() + Start; | 
|  | 2045 | P != PEnd; ++P) { | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2046 | bool HasDefaultArg = false; | 
|  | 2047 | std::string PlaceholderStr; | 
|  | 2048 | if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(*P)) { | 
|  | 2049 | if (TTP->wasDeclaredWithTypename()) | 
|  | 2050 | PlaceholderStr = "typename"; | 
|  | 2051 | else | 
|  | 2052 | PlaceholderStr = "class"; | 
|  | 2053 |  | 
|  | 2054 | if (TTP->getIdentifier()) { | 
|  | 2055 | PlaceholderStr += ' '; | 
|  | 2056 | PlaceholderStr += TTP->getIdentifier()->getName(); | 
|  | 2057 | } | 
|  | 2058 |  | 
|  | 2059 | HasDefaultArg = TTP->hasDefaultArgument(); | 
|  | 2060 | } else if (NonTypeTemplateParmDecl *NTTP | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2061 | = dyn_cast<NonTypeTemplateParmDecl>(*P)) { | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2062 | if (NTTP->getIdentifier()) | 
|  | 2063 | PlaceholderStr = NTTP->getIdentifier()->getName(); | 
|  | 2064 | NTTP->getType().getAsStringInternal(PlaceholderStr, | 
|  | 2065 | Context.PrintingPolicy); | 
|  | 2066 | HasDefaultArg = NTTP->hasDefaultArgument(); | 
|  | 2067 | } else { | 
|  | 2068 | assert(isa<TemplateTemplateParmDecl>(*P)); | 
|  | 2069 | TemplateTemplateParmDecl *TTP = cast<TemplateTemplateParmDecl>(*P); | 
|  | 2070 |  | 
|  | 2071 | // Since putting the template argument list into the placeholder would | 
|  | 2072 | // be very, very long, we just use an abbreviation. | 
|  | 2073 | PlaceholderStr = "template<...> class"; | 
|  | 2074 | if (TTP->getIdentifier()) { | 
|  | 2075 | PlaceholderStr += ' '; | 
|  | 2076 | PlaceholderStr += TTP->getIdentifier()->getName(); | 
|  | 2077 | } | 
|  | 2078 |  | 
|  | 2079 | HasDefaultArg = TTP->hasDefaultArgument(); | 
|  | 2080 | } | 
|  | 2081 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2082 | if (HasDefaultArg && !InDefaultArg) { | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2083 | // When we see an optional default argument, put that argument and | 
|  | 2084 | // the remaining default arguments into a new, optional string. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2085 | CodeCompletionBuilder Opt(Result.getAllocator()); | 
|  | 2086 | if (!FirstParameter) | 
|  | 2087 | Opt.AddChunk(Chunk(CodeCompletionString::CK_Comma)); | 
|  | 2088 | AddTemplateParameterChunks(Context, Template, Opt, MaxParameters, | 
|  | 2089 | P - Params->begin(), true); | 
|  | 2090 | Result.AddOptionalChunk(Opt.TakeString()); | 
|  | 2091 | break; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2092 | } | 
|  | 2093 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2094 | InDefaultArg = false; | 
|  | 2095 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2096 | if (FirstParameter) | 
|  | 2097 | FirstParameter = false; | 
|  | 2098 | else | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2099 | Result.AddChunk(Chunk(CodeCompletionString::CK_Comma)); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2100 |  | 
|  | 2101 | // Add the placeholder string. | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2102 | Result.AddPlaceholderChunk( | 
|  | 2103 | Result.getAllocator().CopyString(PlaceholderStr)); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2104 | } | 
|  | 2105 | } | 
|  | 2106 |  | 
| Douglas Gregor | b9d0ef7 | 2009-09-21 19:57:38 +0000 | [diff] [blame] | 2107 | /// \brief Add a qualifier to the given code-completion string, if the | 
|  | 2108 | /// provided nested-name-specifier is non-NULL. | 
| Douglas Gregor | a61a879 | 2009-12-11 18:44:16 +0000 | [diff] [blame] | 2109 | static void | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2110 | AddQualifierToCompletionString(CodeCompletionBuilder &Result, | 
| Douglas Gregor | a61a879 | 2009-12-11 18:44:16 +0000 | [diff] [blame] | 2111 | NestedNameSpecifier *Qualifier, | 
|  | 2112 | bool QualifierIsInformative, | 
|  | 2113 | ASTContext &Context) { | 
| Douglas Gregor | b9d0ef7 | 2009-09-21 19:57:38 +0000 | [diff] [blame] | 2114 | if (!Qualifier) | 
|  | 2115 | return; | 
|  | 2116 |  | 
|  | 2117 | std::string PrintedNNS; | 
|  | 2118 | { | 
|  | 2119 | llvm::raw_string_ostream OS(PrintedNNS); | 
|  | 2120 | Qualifier->print(OS, Context.PrintingPolicy); | 
|  | 2121 | } | 
| Douglas Gregor | 0563c26 | 2009-09-22 23:15:58 +0000 | [diff] [blame] | 2122 | if (QualifierIsInformative) | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2123 | Result.AddInformativeChunk(Result.getAllocator().CopyString(PrintedNNS)); | 
| Douglas Gregor | 0563c26 | 2009-09-22 23:15:58 +0000 | [diff] [blame] | 2124 | else | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2125 | Result.AddTextChunk(Result.getAllocator().CopyString(PrintedNNS)); | 
| Douglas Gregor | b9d0ef7 | 2009-09-21 19:57:38 +0000 | [diff] [blame] | 2126 | } | 
|  | 2127 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2128 | static void | 
|  | 2129 | AddFunctionTypeQualsToCompletionString(CodeCompletionBuilder &Result, | 
|  | 2130 | FunctionDecl *Function) { | 
| Douglas Gregor | a61a879 | 2009-12-11 18:44:16 +0000 | [diff] [blame] | 2131 | const FunctionProtoType *Proto | 
|  | 2132 | = Function->getType()->getAs<FunctionProtoType>(); | 
|  | 2133 | if (!Proto || !Proto->getTypeQuals()) | 
|  | 2134 | return; | 
|  | 2135 |  | 
| Douglas Gregor | a63f6de | 2011-02-01 21:15:40 +0000 | [diff] [blame] | 2136 | // FIXME: Add ref-qualifier! | 
|  | 2137 |  | 
|  | 2138 | // Handle single qualifiers without copying | 
|  | 2139 | if (Proto->getTypeQuals() == Qualifiers::Const) { | 
|  | 2140 | Result.AddInformativeChunk(" const"); | 
|  | 2141 | return; | 
|  | 2142 | } | 
|  | 2143 |  | 
|  | 2144 | if (Proto->getTypeQuals() == Qualifiers::Volatile) { | 
|  | 2145 | Result.AddInformativeChunk(" volatile"); | 
|  | 2146 | return; | 
|  | 2147 | } | 
|  | 2148 |  | 
|  | 2149 | if (Proto->getTypeQuals() == Qualifiers::Restrict) { | 
|  | 2150 | Result.AddInformativeChunk(" restrict"); | 
|  | 2151 | return; | 
|  | 2152 | } | 
|  | 2153 |  | 
|  | 2154 | // Handle multiple qualifiers. | 
| Douglas Gregor | a61a879 | 2009-12-11 18:44:16 +0000 | [diff] [blame] | 2155 | std::string QualsStr; | 
|  | 2156 | if (Proto->getTypeQuals() & Qualifiers::Const) | 
|  | 2157 | QualsStr += " const"; | 
|  | 2158 | if (Proto->getTypeQuals() & Qualifiers::Volatile) | 
|  | 2159 | QualsStr += " volatile"; | 
|  | 2160 | if (Proto->getTypeQuals() & Qualifiers::Restrict) | 
|  | 2161 | QualsStr += " restrict"; | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2162 | Result.AddInformativeChunk(Result.getAllocator().CopyString(QualsStr)); | 
| Douglas Gregor | a61a879 | 2009-12-11 18:44:16 +0000 | [diff] [blame] | 2163 | } | 
|  | 2164 |  | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 2165 | /// \brief Add the name of the given declaration | 
|  | 2166 | static void AddTypedNameChunk(ASTContext &Context, NamedDecl *ND, | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2167 | CodeCompletionBuilder &Result) { | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 2168 | typedef CodeCompletionString::Chunk Chunk; | 
|  | 2169 |  | 
|  | 2170 | DeclarationName Name = ND->getDeclName(); | 
|  | 2171 | if (!Name) | 
|  | 2172 | return; | 
|  | 2173 |  | 
|  | 2174 | switch (Name.getNameKind()) { | 
| Douglas Gregor | a63f6de | 2011-02-01 21:15:40 +0000 | [diff] [blame] | 2175 | case DeclarationName::CXXOperatorName: { | 
|  | 2176 | const char *OperatorName = 0; | 
|  | 2177 | switch (Name.getCXXOverloadedOperator()) { | 
|  | 2178 | case OO_None: | 
|  | 2179 | case OO_Conditional: | 
|  | 2180 | case NUM_OVERLOADED_OPERATORS: | 
|  | 2181 | OperatorName = "operator"; | 
|  | 2182 | break; | 
|  | 2183 |  | 
|  | 2184 | #define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \ | 
|  | 2185 | case OO_##Name: OperatorName = "operator" Spelling; break; | 
|  | 2186 | #define OVERLOADED_OPERATOR_MULTI(Name,Spelling,Unary,Binary,MemberOnly) | 
|  | 2187 | #include "clang/Basic/OperatorKinds.def" | 
|  | 2188 |  | 
|  | 2189 | case OO_New:          OperatorName = "operator new"; break; | 
|  | 2190 | case OO_Delete:       OperatorName = "operator delete"; break; | 
|  | 2191 | case OO_Array_New:    OperatorName = "operator new[]"; break; | 
|  | 2192 | case OO_Array_Delete: OperatorName = "operator delete[]"; break; | 
|  | 2193 | case OO_Call:         OperatorName = "operator()"; break; | 
|  | 2194 | case OO_Subscript:    OperatorName = "operator[]"; break; | 
|  | 2195 | } | 
|  | 2196 | Result.AddTypedTextChunk(OperatorName); | 
|  | 2197 | break; | 
|  | 2198 | } | 
|  | 2199 |  | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 2200 | case DeclarationName::Identifier: | 
|  | 2201 | case DeclarationName::CXXConversionFunctionName: | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 2202 | case DeclarationName::CXXDestructorName: | 
|  | 2203 | case DeclarationName::CXXLiteralOperatorName: | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2204 | Result.AddTypedTextChunk( | 
|  | 2205 | Result.getAllocator().CopyString(ND->getNameAsString())); | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 2206 | break; | 
|  | 2207 |  | 
|  | 2208 | case DeclarationName::CXXUsingDirective: | 
|  | 2209 | case DeclarationName::ObjCZeroArgSelector: | 
|  | 2210 | case DeclarationName::ObjCOneArgSelector: | 
|  | 2211 | case DeclarationName::ObjCMultiArgSelector: | 
|  | 2212 | break; | 
|  | 2213 |  | 
|  | 2214 | case DeclarationName::CXXConstructorName: { | 
|  | 2215 | CXXRecordDecl *Record = 0; | 
|  | 2216 | QualType Ty = Name.getCXXNameType(); | 
|  | 2217 | if (const RecordType *RecordTy = Ty->getAs<RecordType>()) | 
|  | 2218 | Record = cast<CXXRecordDecl>(RecordTy->getDecl()); | 
|  | 2219 | else if (const InjectedClassNameType *InjectedTy | 
|  | 2220 | = Ty->getAs<InjectedClassNameType>()) | 
|  | 2221 | Record = InjectedTy->getDecl(); | 
|  | 2222 | else { | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2223 | Result.AddTypedTextChunk( | 
|  | 2224 | Result.getAllocator().CopyString(ND->getNameAsString())); | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 2225 | break; | 
|  | 2226 | } | 
|  | 2227 |  | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2228 | Result.AddTypedTextChunk( | 
|  | 2229 | Result.getAllocator().CopyString(Record->getNameAsString())); | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 2230 | if (ClassTemplateDecl *Template = Record->getDescribedClassTemplate()) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2231 | Result.AddChunk(Chunk(CodeCompletionString::CK_LeftAngle)); | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 2232 | AddTemplateParameterChunks(Context, Template, Result); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2233 | Result.AddChunk(Chunk(CodeCompletionString::CK_RightAngle)); | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 2234 | } | 
|  | 2235 | break; | 
|  | 2236 | } | 
|  | 2237 | } | 
|  | 2238 | } | 
|  | 2239 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2240 | /// \brief If possible, create a new code completion string for the given | 
|  | 2241 | /// result. | 
|  | 2242 | /// | 
|  | 2243 | /// \returns Either a new, heap-allocated code completion string describing | 
|  | 2244 | /// how to use this result, or NULL to indicate that the string or name of the | 
|  | 2245 | /// result is all that is needed. | 
|  | 2246 | CodeCompletionString * | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 2247 | CodeCompletionResult::CreateCodeCompletionString(Sema &S, | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2248 | CodeCompletionAllocator &Allocator) { | 
| Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 2249 | typedef CodeCompletionString::Chunk Chunk; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2250 | CodeCompletionBuilder Result(Allocator, Priority, Availability); | 
| Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 2251 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2252 | if (Kind == RK_Pattern) { | 
|  | 2253 | Pattern->Priority = Priority; | 
|  | 2254 | Pattern->Availability = Availability; | 
|  | 2255 | return Pattern; | 
|  | 2256 | } | 
| Douglas Gregor | 2b4074f | 2009-12-01 05:55:20 +0000 | [diff] [blame] | 2257 |  | 
| Douglas Gregor | 2b4074f | 2009-12-01 05:55:20 +0000 | [diff] [blame] | 2258 | if (Kind == RK_Keyword) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2259 | Result.AddTypedTextChunk(Keyword); | 
|  | 2260 | return Result.TakeString(); | 
| Douglas Gregor | 2b4074f | 2009-12-01 05:55:20 +0000 | [diff] [blame] | 2261 | } | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2262 |  | 
| Douglas Gregor | 3f7c7f4 | 2009-10-30 16:50:04 +0000 | [diff] [blame] | 2263 | if (Kind == RK_Macro) { | 
|  | 2264 | MacroInfo *MI = S.PP.getMacroInfo(Macro); | 
| Douglas Gregor | 2b4074f | 2009-12-01 05:55:20 +0000 | [diff] [blame] | 2265 | assert(MI && "Not a macro?"); | 
|  | 2266 |  | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2267 | Result.AddTypedTextChunk( | 
|  | 2268 | Result.getAllocator().CopyString(Macro->getName())); | 
| Douglas Gregor | 2b4074f | 2009-12-01 05:55:20 +0000 | [diff] [blame] | 2269 |  | 
|  | 2270 | if (!MI->isFunctionLike()) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2271 | return Result.TakeString(); | 
| Douglas Gregor | 3f7c7f4 | 2009-10-30 16:50:04 +0000 | [diff] [blame] | 2272 |  | 
|  | 2273 | // Format a function-like macro with placeholders for the arguments. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2274 | Result.AddChunk(Chunk(CodeCompletionString::CK_LeftParen)); | 
| Douglas Gregor | 3f7c7f4 | 2009-10-30 16:50:04 +0000 | [diff] [blame] | 2275 | for (MacroInfo::arg_iterator A = MI->arg_begin(), AEnd = MI->arg_end(); | 
|  | 2276 | A != AEnd; ++A) { | 
|  | 2277 | if (A != MI->arg_begin()) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2278 | Result.AddChunk(Chunk(CodeCompletionString::CK_Comma)); | 
| Douglas Gregor | 3f7c7f4 | 2009-10-30 16:50:04 +0000 | [diff] [blame] | 2279 |  | 
|  | 2280 | if (!MI->isVariadic() || A != AEnd - 1) { | 
|  | 2281 | // Non-variadic argument. | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2282 | Result.AddPlaceholderChunk( | 
|  | 2283 | Result.getAllocator().CopyString((*A)->getName())); | 
| Douglas Gregor | 3f7c7f4 | 2009-10-30 16:50:04 +0000 | [diff] [blame] | 2284 | continue; | 
|  | 2285 | } | 
|  | 2286 |  | 
|  | 2287 | // Variadic argument; cope with the different between GNU and C99 | 
|  | 2288 | // variadic macros, providing a single placeholder for the rest of the | 
|  | 2289 | // arguments. | 
|  | 2290 | if ((*A)->isStr("__VA_ARGS__")) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2291 | Result.AddPlaceholderChunk("..."); | 
| Douglas Gregor | 3f7c7f4 | 2009-10-30 16:50:04 +0000 | [diff] [blame] | 2292 | else { | 
|  | 2293 | std::string Arg = (*A)->getName(); | 
|  | 2294 | Arg += "..."; | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2295 | Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg)); | 
| Douglas Gregor | 3f7c7f4 | 2009-10-30 16:50:04 +0000 | [diff] [blame] | 2296 | } | 
|  | 2297 | } | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2298 | Result.AddChunk(Chunk(CodeCompletionString::CK_RightParen)); | 
|  | 2299 | return Result.TakeString(); | 
| Douglas Gregor | 3f7c7f4 | 2009-10-30 16:50:04 +0000 | [diff] [blame] | 2300 | } | 
|  | 2301 |  | 
| Douglas Gregor | d8e8a58 | 2010-05-25 21:41:55 +0000 | [diff] [blame] | 2302 | assert(Kind == RK_Declaration && "Missed a result kind?"); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2303 | NamedDecl *ND = Declaration; | 
|  | 2304 |  | 
| Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 2305 | if (StartsNestedNameSpecifier) { | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2306 | Result.AddTypedTextChunk( | 
|  | 2307 | Result.getAllocator().CopyString(ND->getNameAsString())); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2308 | Result.AddTextChunk("::"); | 
|  | 2309 | return Result.TakeString(); | 
| Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 2310 | } | 
|  | 2311 |  | 
| Douglas Gregor | ff5ce6e | 2009-12-18 18:53:37 +0000 | [diff] [blame] | 2312 | AddResultTypeChunk(S.Context, ND, Result); | 
|  | 2313 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2314 | if (FunctionDecl *Function = dyn_cast<FunctionDecl>(ND)) { | 
| Douglas Gregor | 0563c26 | 2009-09-22 23:15:58 +0000 | [diff] [blame] | 2315 | AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative, | 
|  | 2316 | S.Context); | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 2317 | AddTypedNameChunk(S.Context, ND, Result); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2318 | Result.AddChunk(Chunk(CodeCompletionString::CK_LeftParen)); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2319 | AddFunctionParameterChunks(S.Context, Function, Result); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2320 | Result.AddChunk(Chunk(CodeCompletionString::CK_RightParen)); | 
| Douglas Gregor | a61a879 | 2009-12-11 18:44:16 +0000 | [diff] [blame] | 2321 | AddFunctionTypeQualsToCompletionString(Result, Function); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2322 | return Result.TakeString(); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2323 | } | 
|  | 2324 |  | 
|  | 2325 | if (FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND)) { | 
| Douglas Gregor | 0563c26 | 2009-09-22 23:15:58 +0000 | [diff] [blame] | 2326 | AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative, | 
|  | 2327 | S.Context); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2328 | FunctionDecl *Function = FunTmpl->getTemplatedDecl(); | 
| Douglas Gregor | 6f942b2 | 2010-09-21 16:06:22 +0000 | [diff] [blame] | 2329 | AddTypedNameChunk(S.Context, Function, Result); | 
|  | 2330 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2331 | // Figure out which template parameters are deduced (or have default | 
|  | 2332 | // arguments). | 
|  | 2333 | llvm::SmallVector<bool, 16> Deduced; | 
|  | 2334 | S.MarkDeducedTemplateParameters(FunTmpl, Deduced); | 
|  | 2335 | unsigned LastDeducibleArgument; | 
|  | 2336 | for (LastDeducibleArgument = Deduced.size(); LastDeducibleArgument > 0; | 
|  | 2337 | --LastDeducibleArgument) { | 
|  | 2338 | if (!Deduced[LastDeducibleArgument - 1]) { | 
|  | 2339 | // C++0x: Figure out if the template argument has a default. If so, | 
|  | 2340 | // the user doesn't need to type this argument. | 
|  | 2341 | // FIXME: We need to abstract template parameters better! | 
|  | 2342 | bool HasDefaultArg = false; | 
|  | 2343 | NamedDecl *Param = FunTmpl->getTemplateParameters()->getParam( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2344 | LastDeducibleArgument - 1); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2345 | if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param)) | 
|  | 2346 | HasDefaultArg = TTP->hasDefaultArgument(); | 
|  | 2347 | else if (NonTypeTemplateParmDecl *NTTP | 
|  | 2348 | = dyn_cast<NonTypeTemplateParmDecl>(Param)) | 
|  | 2349 | HasDefaultArg = NTTP->hasDefaultArgument(); | 
|  | 2350 | else { | 
|  | 2351 | assert(isa<TemplateTemplateParmDecl>(Param)); | 
|  | 2352 | HasDefaultArg | 
| Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 2353 | = cast<TemplateTemplateParmDecl>(Param)->hasDefaultArgument(); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2354 | } | 
|  | 2355 |  | 
|  | 2356 | if (!HasDefaultArg) | 
|  | 2357 | break; | 
|  | 2358 | } | 
|  | 2359 | } | 
|  | 2360 |  | 
|  | 2361 | if (LastDeducibleArgument) { | 
|  | 2362 | // Some of the function template arguments cannot be deduced from a | 
|  | 2363 | // function call, so we introduce an explicit template argument list | 
|  | 2364 | // containing all of the arguments up to the first deducible argument. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2365 | Result.AddChunk(Chunk(CodeCompletionString::CK_LeftAngle)); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2366 | AddTemplateParameterChunks(S.Context, FunTmpl, Result, | 
|  | 2367 | LastDeducibleArgument); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2368 | Result.AddChunk(Chunk(CodeCompletionString::CK_RightAngle)); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2369 | } | 
|  | 2370 |  | 
|  | 2371 | // Add the function parameters | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2372 | Result.AddChunk(Chunk(CodeCompletionString::CK_LeftParen)); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2373 | AddFunctionParameterChunks(S.Context, Function, Result); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2374 | Result.AddChunk(Chunk(CodeCompletionString::CK_RightParen)); | 
| Douglas Gregor | a61a879 | 2009-12-11 18:44:16 +0000 | [diff] [blame] | 2375 | AddFunctionTypeQualsToCompletionString(Result, Function); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2376 | return Result.TakeString(); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2377 | } | 
|  | 2378 |  | 
|  | 2379 | if (TemplateDecl *Template = dyn_cast<TemplateDecl>(ND)) { | 
| Douglas Gregor | 0563c26 | 2009-09-22 23:15:58 +0000 | [diff] [blame] | 2380 | AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative, | 
|  | 2381 | S.Context); | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2382 | Result.AddTypedTextChunk( | 
|  | 2383 | Result.getAllocator().CopyString(Template->getNameAsString())); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2384 | Result.AddChunk(Chunk(CodeCompletionString::CK_LeftAngle)); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2385 | AddTemplateParameterChunks(S.Context, Template, Result); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2386 | Result.AddChunk(Chunk(CodeCompletionString::CK_RightAngle)); | 
|  | 2387 | return Result.TakeString(); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2388 | } | 
|  | 2389 |  | 
| Douglas Gregor | 9630eb6 | 2009-11-17 16:44:22 +0000 | [diff] [blame] | 2390 | if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND)) { | 
| Douglas Gregor | 9630eb6 | 2009-11-17 16:44:22 +0000 | [diff] [blame] | 2391 | Selector Sel = Method->getSelector(); | 
|  | 2392 | if (Sel.isUnarySelector()) { | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2393 | Result.AddTypedTextChunk(Result.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2394 | Sel.getIdentifierInfoForSlot(0)->getName())); | 
|  | 2395 | return Result.TakeString(); | 
| Douglas Gregor | 9630eb6 | 2009-11-17 16:44:22 +0000 | [diff] [blame] | 2396 | } | 
|  | 2397 |  | 
| Douglas Gregor | d3c6854 | 2009-11-19 01:08:35 +0000 | [diff] [blame] | 2398 | std::string SelName = Sel.getIdentifierInfoForSlot(0)->getName().str(); | 
|  | 2399 | SelName += ':'; | 
|  | 2400 | if (StartParameter == 0) | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2401 | Result.AddTypedTextChunk(Result.getAllocator().CopyString(SelName)); | 
| Douglas Gregor | d3c6854 | 2009-11-19 01:08:35 +0000 | [diff] [blame] | 2402 | else { | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2403 | Result.AddInformativeChunk(Result.getAllocator().CopyString(SelName)); | 
| Douglas Gregor | d3c6854 | 2009-11-19 01:08:35 +0000 | [diff] [blame] | 2404 |  | 
|  | 2405 | // If there is only one parameter, and we're past it, add an empty | 
|  | 2406 | // typed-text chunk since there is nothing to type. | 
|  | 2407 | if (Method->param_size() == 1) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2408 | Result.AddTypedTextChunk(""); | 
| Douglas Gregor | d3c6854 | 2009-11-19 01:08:35 +0000 | [diff] [blame] | 2409 | } | 
| Douglas Gregor | 9630eb6 | 2009-11-17 16:44:22 +0000 | [diff] [blame] | 2410 | unsigned Idx = 0; | 
|  | 2411 | for (ObjCMethodDecl::param_iterator P = Method->param_begin(), | 
|  | 2412 | PEnd = Method->param_end(); | 
|  | 2413 | P != PEnd; (void)++P, ++Idx) { | 
|  | 2414 | if (Idx > 0) { | 
| Douglas Gregor | d3c6854 | 2009-11-19 01:08:35 +0000 | [diff] [blame] | 2415 | std::string Keyword; | 
|  | 2416 | if (Idx > StartParameter) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2417 | Result.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
| Douglas Gregor | 9630eb6 | 2009-11-17 16:44:22 +0000 | [diff] [blame] | 2418 | if (IdentifierInfo *II = Sel.getIdentifierInfoForSlot(Idx)) | 
|  | 2419 | Keyword += II->getName().str(); | 
|  | 2420 | Keyword += ":"; | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 2421 | if (Idx < StartParameter || AllParametersAreInformative) | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2422 | Result.AddInformativeChunk(Result.getAllocator().CopyString(Keyword)); | 
| Douglas Gregor | c7b7b7a | 2010-10-18 21:05:04 +0000 | [diff] [blame] | 2423 | else | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2424 | Result.AddTypedTextChunk(Result.getAllocator().CopyString(Keyword)); | 
| Douglas Gregor | 9630eb6 | 2009-11-17 16:44:22 +0000 | [diff] [blame] | 2425 | } | 
| Douglas Gregor | d3c6854 | 2009-11-19 01:08:35 +0000 | [diff] [blame] | 2426 |  | 
|  | 2427 | // If we're before the starting parameter, skip the placeholder. | 
|  | 2428 | if (Idx < StartParameter) | 
|  | 2429 | continue; | 
| Douglas Gregor | 9630eb6 | 2009-11-17 16:44:22 +0000 | [diff] [blame] | 2430 |  | 
|  | 2431 | std::string Arg; | 
| Douglas Gregor | 83482d1 | 2010-08-24 16:15:59 +0000 | [diff] [blame] | 2432 |  | 
|  | 2433 | if ((*P)->getType()->isBlockPointerType() && !DeclaringEntity) | 
| Douglas Gregor | aba4808 | 2010-08-29 19:47:46 +0000 | [diff] [blame] | 2434 | Arg = FormatFunctionParameter(S.Context, *P, true); | 
| Douglas Gregor | 83482d1 | 2010-08-24 16:15:59 +0000 | [diff] [blame] | 2435 | else { | 
|  | 2436 | (*P)->getType().getAsStringInternal(Arg, S.Context.PrintingPolicy); | 
|  | 2437 | Arg = "(" + Arg + ")"; | 
|  | 2438 | if (IdentifierInfo *II = (*P)->getIdentifier()) | 
| Douglas Gregor | aba4808 | 2010-08-29 19:47:46 +0000 | [diff] [blame] | 2439 | if (DeclaringEntity || AllParametersAreInformative) | 
|  | 2440 | Arg += II->getName().str(); | 
| Douglas Gregor | 83482d1 | 2010-08-24 16:15:59 +0000 | [diff] [blame] | 2441 | } | 
|  | 2442 |  | 
| Douglas Gregor | e17794f | 2010-08-31 05:13:43 +0000 | [diff] [blame] | 2443 | if (Method->isVariadic() && (P + 1) == PEnd) | 
|  | 2444 | Arg += ", ..."; | 
|  | 2445 |  | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 2446 | if (DeclaringEntity) | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2447 | Result.AddTextChunk(Result.getAllocator().CopyString(Arg)); | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 2448 | else if (AllParametersAreInformative) | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2449 | Result.AddInformativeChunk(Result.getAllocator().CopyString(Arg)); | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 2450 | else | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2451 | Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg)); | 
| Douglas Gregor | 9630eb6 | 2009-11-17 16:44:22 +0000 | [diff] [blame] | 2452 | } | 
|  | 2453 |  | 
| Douglas Gregor | 2a17af0 | 2009-12-23 00:21:46 +0000 | [diff] [blame] | 2454 | if (Method->isVariadic()) { | 
| Douglas Gregor | e17794f | 2010-08-31 05:13:43 +0000 | [diff] [blame] | 2455 | if (Method->param_size() == 0) { | 
|  | 2456 | if (DeclaringEntity) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2457 | Result.AddTextChunk(", ..."); | 
| Douglas Gregor | e17794f | 2010-08-31 05:13:43 +0000 | [diff] [blame] | 2458 | else if (AllParametersAreInformative) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2459 | Result.AddInformativeChunk(", ..."); | 
| Douglas Gregor | e17794f | 2010-08-31 05:13:43 +0000 | [diff] [blame] | 2460 | else | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2461 | Result.AddPlaceholderChunk(", ..."); | 
| Douglas Gregor | e17794f | 2010-08-31 05:13:43 +0000 | [diff] [blame] | 2462 | } | 
| Douglas Gregor | aaa107a | 2010-08-23 23:51:41 +0000 | [diff] [blame] | 2463 |  | 
|  | 2464 | MaybeAddSentinel(S.Context, Method, Result); | 
| Douglas Gregor | 2a17af0 | 2009-12-23 00:21:46 +0000 | [diff] [blame] | 2465 | } | 
|  | 2466 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2467 | return Result.TakeString(); | 
| Douglas Gregor | 9630eb6 | 2009-11-17 16:44:22 +0000 | [diff] [blame] | 2468 | } | 
|  | 2469 |  | 
| Douglas Gregor | 2b4074f | 2009-12-01 05:55:20 +0000 | [diff] [blame] | 2470 | if (Qualifier) | 
| Douglas Gregor | 0563c26 | 2009-09-22 23:15:58 +0000 | [diff] [blame] | 2471 | AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative, | 
|  | 2472 | S.Context); | 
| Douglas Gregor | 2b4074f | 2009-12-01 05:55:20 +0000 | [diff] [blame] | 2473 |  | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2474 | Result.AddTypedTextChunk( | 
|  | 2475 | Result.getAllocator().CopyString(ND->getNameAsString())); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2476 | return Result.TakeString(); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2477 | } | 
|  | 2478 |  | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2479 | CodeCompletionString * | 
|  | 2480 | CodeCompleteConsumer::OverloadCandidate::CreateSignatureString( | 
|  | 2481 | unsigned CurrentArg, | 
| Douglas Gregor | 32be4a5 | 2010-10-11 21:37:58 +0000 | [diff] [blame] | 2482 | Sema &S, | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2483 | CodeCompletionAllocator &Allocator) const { | 
| Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 2484 | typedef CodeCompletionString::Chunk Chunk; | 
|  | 2485 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2486 | // FIXME: Set priority, availability appropriately. | 
|  | 2487 | CodeCompletionBuilder Result(Allocator, 1, CXAvailability_Available); | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2488 | FunctionDecl *FDecl = getFunction(); | 
| Douglas Gregor | ff5ce6e | 2009-12-18 18:53:37 +0000 | [diff] [blame] | 2489 | AddResultTypeChunk(S.Context, FDecl, Result); | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2490 | const FunctionProtoType *Proto | 
|  | 2491 | = dyn_cast<FunctionProtoType>(getFunctionType()); | 
|  | 2492 | if (!FDecl && !Proto) { | 
|  | 2493 | // Function without a prototype. Just give the return type and a | 
|  | 2494 | // highlighted ellipsis. | 
|  | 2495 | const FunctionType *FT = getFunctionType(); | 
| Douglas Gregor | a63f6de | 2011-02-01 21:15:40 +0000 | [diff] [blame] | 2496 | Result.AddTextChunk(GetCompletionTypeString(FT->getResultType(), | 
|  | 2497 | S.Context, | 
|  | 2498 | Result.getAllocator())); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2499 | Result.AddChunk(Chunk(CodeCompletionString::CK_LeftParen)); | 
|  | 2500 | Result.AddChunk(Chunk(CodeCompletionString::CK_CurrentParameter, "...")); | 
|  | 2501 | Result.AddChunk(Chunk(CodeCompletionString::CK_RightParen)); | 
|  | 2502 | return Result.TakeString(); | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2503 | } | 
|  | 2504 |  | 
|  | 2505 | if (FDecl) | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2506 | Result.AddTextChunk( | 
|  | 2507 | Result.getAllocator().CopyString(FDecl->getNameAsString())); | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2508 | else | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2509 | Result.AddTextChunk( | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2510 | Result.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2511 | Proto->getResultType().getAsString(S.Context.PrintingPolicy))); | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2512 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2513 | Result.AddChunk(Chunk(CodeCompletionString::CK_LeftParen)); | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2514 | unsigned NumParams = FDecl? FDecl->getNumParams() : Proto->getNumArgs(); | 
|  | 2515 | for (unsigned I = 0; I != NumParams; ++I) { | 
|  | 2516 | if (I) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2517 | Result.AddChunk(Chunk(CodeCompletionString::CK_Comma)); | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2518 |  | 
|  | 2519 | std::string ArgString; | 
|  | 2520 | QualType ArgType; | 
|  | 2521 |  | 
|  | 2522 | if (FDecl) { | 
|  | 2523 | ArgString = FDecl->getParamDecl(I)->getNameAsString(); | 
|  | 2524 | ArgType = FDecl->getParamDecl(I)->getOriginalType(); | 
|  | 2525 | } else { | 
|  | 2526 | ArgType = Proto->getArgType(I); | 
|  | 2527 | } | 
|  | 2528 |  | 
|  | 2529 | ArgType.getAsStringInternal(ArgString, S.Context.PrintingPolicy); | 
|  | 2530 |  | 
|  | 2531 | if (I == CurrentArg) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2532 | Result.AddChunk(Chunk(CodeCompletionString::CK_CurrentParameter, | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2533 | Result.getAllocator().CopyString(ArgString))); | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2534 | else | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2535 | Result.AddTextChunk(Result.getAllocator().CopyString(ArgString)); | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2536 | } | 
|  | 2537 |  | 
|  | 2538 | if (Proto && Proto->isVariadic()) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2539 | Result.AddChunk(Chunk(CodeCompletionString::CK_Comma)); | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2540 | if (CurrentArg < NumParams) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2541 | Result.AddTextChunk("..."); | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2542 | else | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2543 | Result.AddChunk(Chunk(CodeCompletionString::CK_CurrentParameter, "...")); | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2544 | } | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2545 | Result.AddChunk(Chunk(CodeCompletionString::CK_RightParen)); | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2546 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2547 | return Result.TakeString(); | 
| Douglas Gregor | 86d802e | 2009-09-23 00:34:09 +0000 | [diff] [blame] | 2548 | } | 
|  | 2549 |  | 
| Douglas Gregor | 1827e10 | 2010-08-16 16:18:59 +0000 | [diff] [blame] | 2550 | unsigned clang::getMacroUsagePriority(llvm::StringRef MacroName, | 
| Douglas Gregor | b05496d | 2010-09-20 21:11:48 +0000 | [diff] [blame] | 2551 | const LangOptions &LangOpts, | 
| Douglas Gregor | 1827e10 | 2010-08-16 16:18:59 +0000 | [diff] [blame] | 2552 | bool PreferredTypeIsPointer) { | 
|  | 2553 | unsigned Priority = CCP_Macro; | 
|  | 2554 |  | 
| Douglas Gregor | b05496d | 2010-09-20 21:11:48 +0000 | [diff] [blame] | 2555 | // Treat the "nil", "Nil" and "NULL" macros as null pointer constants. | 
|  | 2556 | if (MacroName.equals("nil") || MacroName.equals("NULL") || | 
|  | 2557 | MacroName.equals("Nil")) { | 
| Douglas Gregor | 1827e10 | 2010-08-16 16:18:59 +0000 | [diff] [blame] | 2558 | Priority = CCP_Constant; | 
|  | 2559 | if (PreferredTypeIsPointer) | 
|  | 2560 | Priority = Priority / CCF_SimilarTypeMatch; | 
| Douglas Gregor | b05496d | 2010-09-20 21:11:48 +0000 | [diff] [blame] | 2561 | } | 
|  | 2562 | // Treat "YES", "NO", "true", and "false" as constants. | 
|  | 2563 | else if (MacroName.equals("YES") || MacroName.equals("NO") || | 
|  | 2564 | MacroName.equals("true") || MacroName.equals("false")) | 
|  | 2565 | Priority = CCP_Constant; | 
|  | 2566 | // Treat "bool" as a type. | 
|  | 2567 | else if (MacroName.equals("bool")) | 
|  | 2568 | Priority = CCP_Type + (LangOpts.ObjC1? CCD_bool_in_ObjC : 0); | 
|  | 2569 |  | 
| Douglas Gregor | 1827e10 | 2010-08-16 16:18:59 +0000 | [diff] [blame] | 2570 |  | 
|  | 2571 | return Priority; | 
|  | 2572 | } | 
|  | 2573 |  | 
| Douglas Gregor | e8d7beb | 2010-09-03 23:30:36 +0000 | [diff] [blame] | 2574 | CXCursorKind clang::getCursorKindForDecl(Decl *D) { | 
|  | 2575 | if (!D) | 
|  | 2576 | return CXCursor_UnexposedDecl; | 
|  | 2577 |  | 
|  | 2578 | switch (D->getKind()) { | 
|  | 2579 | case Decl::Enum:               return CXCursor_EnumDecl; | 
|  | 2580 | case Decl::EnumConstant:       return CXCursor_EnumConstantDecl; | 
|  | 2581 | case Decl::Field:              return CXCursor_FieldDecl; | 
|  | 2582 | case Decl::Function: | 
|  | 2583 | return CXCursor_FunctionDecl; | 
|  | 2584 | case Decl::ObjCCategory:       return CXCursor_ObjCCategoryDecl; | 
|  | 2585 | case Decl::ObjCCategoryImpl:   return CXCursor_ObjCCategoryImplDecl; | 
|  | 2586 | case Decl::ObjCClass: | 
|  | 2587 | // FIXME | 
|  | 2588 | return CXCursor_UnexposedDecl; | 
|  | 2589 | case Decl::ObjCForwardProtocol: | 
|  | 2590 | // FIXME | 
|  | 2591 | return CXCursor_UnexposedDecl; | 
|  | 2592 | case Decl::ObjCImplementation: return CXCursor_ObjCImplementationDecl; | 
|  | 2593 | case Decl::ObjCInterface:      return CXCursor_ObjCInterfaceDecl; | 
|  | 2594 | case Decl::ObjCIvar:           return CXCursor_ObjCIvarDecl; | 
|  | 2595 | case Decl::ObjCMethod: | 
|  | 2596 | return cast<ObjCMethodDecl>(D)->isInstanceMethod() | 
|  | 2597 | ? CXCursor_ObjCInstanceMethodDecl : CXCursor_ObjCClassMethodDecl; | 
|  | 2598 | case Decl::CXXMethod:          return CXCursor_CXXMethod; | 
|  | 2599 | case Decl::CXXConstructor:     return CXCursor_Constructor; | 
|  | 2600 | case Decl::CXXDestructor:      return CXCursor_Destructor; | 
|  | 2601 | case Decl::CXXConversion:      return CXCursor_ConversionFunction; | 
|  | 2602 | case Decl::ObjCProperty:       return CXCursor_ObjCPropertyDecl; | 
|  | 2603 | case Decl::ObjCProtocol:       return CXCursor_ObjCProtocolDecl; | 
|  | 2604 | case Decl::ParmVar:            return CXCursor_ParmDecl; | 
|  | 2605 | case Decl::Typedef:            return CXCursor_TypedefDecl; | 
|  | 2606 | case Decl::Var:                return CXCursor_VarDecl; | 
|  | 2607 | case Decl::Namespace:          return CXCursor_Namespace; | 
|  | 2608 | case Decl::NamespaceAlias:     return CXCursor_NamespaceAlias; | 
|  | 2609 | case Decl::TemplateTypeParm:   return CXCursor_TemplateTypeParameter; | 
|  | 2610 | case Decl::NonTypeTemplateParm:return CXCursor_NonTypeTemplateParameter; | 
|  | 2611 | case Decl::TemplateTemplateParm:return CXCursor_TemplateTemplateParameter; | 
|  | 2612 | case Decl::FunctionTemplate:   return CXCursor_FunctionTemplate; | 
|  | 2613 | case Decl::ClassTemplate:      return CXCursor_ClassTemplate; | 
|  | 2614 | case Decl::ClassTemplatePartialSpecialization: | 
|  | 2615 | return CXCursor_ClassTemplatePartialSpecialization; | 
|  | 2616 | case Decl::UsingDirective:     return CXCursor_UsingDirective; | 
|  | 2617 |  | 
|  | 2618 | case Decl::Using: | 
|  | 2619 | case Decl::UnresolvedUsingValue: | 
|  | 2620 | case Decl::UnresolvedUsingTypename: | 
|  | 2621 | return CXCursor_UsingDeclaration; | 
|  | 2622 |  | 
|  | 2623 | default: | 
|  | 2624 | if (TagDecl *TD = dyn_cast<TagDecl>(D)) { | 
|  | 2625 | switch (TD->getTagKind()) { | 
|  | 2626 | case TTK_Struct: return CXCursor_StructDecl; | 
|  | 2627 | case TTK_Class:  return CXCursor_ClassDecl; | 
|  | 2628 | case TTK_Union:  return CXCursor_UnionDecl; | 
|  | 2629 | case TTK_Enum:   return CXCursor_EnumDecl; | 
|  | 2630 | } | 
|  | 2631 | } | 
|  | 2632 | } | 
|  | 2633 |  | 
|  | 2634 | return CXCursor_UnexposedDecl; | 
|  | 2635 | } | 
|  | 2636 |  | 
| Douglas Gregor | 590c7d5 | 2010-07-08 20:55:51 +0000 | [diff] [blame] | 2637 | static void AddMacroResults(Preprocessor &PP, ResultBuilder &Results, | 
|  | 2638 | bool TargetTypeIsPointer = false) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 2639 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 590c7d5 | 2010-07-08 20:55:51 +0000 | [diff] [blame] | 2640 |  | 
| Douglas Gregor | 3f7c7f4 | 2009-10-30 16:50:04 +0000 | [diff] [blame] | 2641 | Results.EnterNewScope(); | 
| Douglas Gregor | c7b7b7a | 2010-10-18 21:05:04 +0000 | [diff] [blame] | 2642 |  | 
| Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 2643 | for (Preprocessor::macro_iterator M = PP.macro_begin(), | 
|  | 2644 | MEnd = PP.macro_end(); | 
| Douglas Gregor | 590c7d5 | 2010-07-08 20:55:51 +0000 | [diff] [blame] | 2645 | M != MEnd; ++M) { | 
| Douglas Gregor | 1827e10 | 2010-08-16 16:18:59 +0000 | [diff] [blame] | 2646 | Results.AddResult(Result(M->first, | 
|  | 2647 | getMacroUsagePriority(M->first->getName(), | 
| Douglas Gregor | b05496d | 2010-09-20 21:11:48 +0000 | [diff] [blame] | 2648 | PP.getLangOptions(), | 
| Douglas Gregor | 1827e10 | 2010-08-16 16:18:59 +0000 | [diff] [blame] | 2649 | TargetTypeIsPointer))); | 
| Douglas Gregor | 590c7d5 | 2010-07-08 20:55:51 +0000 | [diff] [blame] | 2650 | } | 
| Douglas Gregor | c7b7b7a | 2010-10-18 21:05:04 +0000 | [diff] [blame] | 2651 |  | 
| Douglas Gregor | 3f7c7f4 | 2009-10-30 16:50:04 +0000 | [diff] [blame] | 2652 | Results.ExitScope(); | 
| Douglas Gregor | c7b7b7a | 2010-10-18 21:05:04 +0000 | [diff] [blame] | 2653 |  | 
| Douglas Gregor | 3f7c7f4 | 2009-10-30 16:50:04 +0000 | [diff] [blame] | 2654 | } | 
|  | 2655 |  | 
| Douglas Gregor | aa5f77b | 2010-08-23 21:54:33 +0000 | [diff] [blame] | 2656 | static void AddPrettyFunctionResults(const LangOptions &LangOpts, | 
|  | 2657 | ResultBuilder &Results) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 2658 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | aa5f77b | 2010-08-23 21:54:33 +0000 | [diff] [blame] | 2659 |  | 
|  | 2660 | Results.EnterNewScope(); | 
| Douglas Gregor | c7b7b7a | 2010-10-18 21:05:04 +0000 | [diff] [blame] | 2661 |  | 
| Douglas Gregor | aa5f77b | 2010-08-23 21:54:33 +0000 | [diff] [blame] | 2662 | Results.AddResult(Result("__PRETTY_FUNCTION__", CCP_Constant)); | 
|  | 2663 | Results.AddResult(Result("__FUNCTION__", CCP_Constant)); | 
|  | 2664 | if (LangOpts.C99 || LangOpts.CPlusPlus0x) | 
|  | 2665 | Results.AddResult(Result("__func__", CCP_Constant)); | 
|  | 2666 | Results.ExitScope(); | 
|  | 2667 | } | 
|  | 2668 |  | 
| Daniel Dunbar | 3a2838d | 2009-11-13 08:58:20 +0000 | [diff] [blame] | 2669 | static void HandleCodeCompleteResults(Sema *S, | 
|  | 2670 | CodeCompleteConsumer *CodeCompleter, | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2671 | CodeCompletionContext Context, | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 2672 | CodeCompletionResult *Results, | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2673 | unsigned NumResults) { | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2674 | if (CodeCompleter) | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2675 | CodeCompleter->ProcessCodeCompleteResults(*S, Context, Results, NumResults); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 2676 | } | 
|  | 2677 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2678 | static enum CodeCompletionContext::Kind mapCodeCompletionContext(Sema &S, | 
|  | 2679 | Sema::ParserCompletionContext PCC) { | 
|  | 2680 | switch (PCC) { | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 2681 | case Sema::PCC_Namespace: | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2682 | return CodeCompletionContext::CCC_TopLevel; | 
|  | 2683 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 2684 | case Sema::PCC_Class: | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2685 | return CodeCompletionContext::CCC_ClassStructUnion; | 
|  | 2686 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 2687 | case Sema::PCC_ObjCInterface: | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2688 | return CodeCompletionContext::CCC_ObjCInterface; | 
|  | 2689 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 2690 | case Sema::PCC_ObjCImplementation: | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2691 | return CodeCompletionContext::CCC_ObjCImplementation; | 
|  | 2692 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 2693 | case Sema::PCC_ObjCInstanceVariableList: | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2694 | return CodeCompletionContext::CCC_ObjCIvarList; | 
|  | 2695 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 2696 | case Sema::PCC_Template: | 
|  | 2697 | case Sema::PCC_MemberTemplate: | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 2698 | if (S.CurContext->isFileContext()) | 
|  | 2699 | return CodeCompletionContext::CCC_TopLevel; | 
|  | 2700 | else if (S.CurContext->isRecord()) | 
|  | 2701 | return CodeCompletionContext::CCC_ClassStructUnion; | 
|  | 2702 | else | 
|  | 2703 | return CodeCompletionContext::CCC_Other; | 
|  | 2704 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 2705 | case Sema::PCC_RecoveryInFunction: | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 2706 | return CodeCompletionContext::CCC_Recovery; | 
| Douglas Gregor | a5450a0 | 2010-10-18 22:01:46 +0000 | [diff] [blame] | 2707 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 2708 | case Sema::PCC_ForInit: | 
| Douglas Gregor | a5450a0 | 2010-10-18 22:01:46 +0000 | [diff] [blame] | 2709 | if (S.getLangOptions().CPlusPlus || S.getLangOptions().C99 || | 
|  | 2710 | S.getLangOptions().ObjC1) | 
|  | 2711 | return CodeCompletionContext::CCC_ParenthesizedExpression; | 
|  | 2712 | else | 
|  | 2713 | return CodeCompletionContext::CCC_Expression; | 
|  | 2714 |  | 
|  | 2715 | case Sema::PCC_Expression: | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 2716 | case Sema::PCC_Condition: | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2717 | return CodeCompletionContext::CCC_Expression; | 
|  | 2718 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 2719 | case Sema::PCC_Statement: | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2720 | return CodeCompletionContext::CCC_Statement; | 
| Douglas Gregor | 72db108 | 2010-08-24 01:11:00 +0000 | [diff] [blame] | 2721 |  | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 2722 | case Sema::PCC_Type: | 
| Douglas Gregor | 72db108 | 2010-08-24 01:11:00 +0000 | [diff] [blame] | 2723 | return CodeCompletionContext::CCC_Type; | 
| Douglas Gregor | 0268810 | 2010-09-14 23:59:36 +0000 | [diff] [blame] | 2724 |  | 
|  | 2725 | case Sema::PCC_ParenthesizedExpression: | 
|  | 2726 | return CodeCompletionContext::CCC_ParenthesizedExpression; | 
| Douglas Gregor | 68e3c2e | 2011-02-15 20:33:25 +0000 | [diff] [blame] | 2727 |  | 
|  | 2728 | case Sema::PCC_LocalDeclarationSpecifiers: | 
|  | 2729 | return CodeCompletionContext::CCC_Type; | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2730 | } | 
|  | 2731 |  | 
|  | 2732 | return CodeCompletionContext::CCC_Other; | 
|  | 2733 | } | 
|  | 2734 |  | 
| Douglas Gregor | f696152 | 2010-08-27 21:18:54 +0000 | [diff] [blame] | 2735 | /// \brief If we're in a C++ virtual member function, add completion results | 
|  | 2736 | /// that invoke the functions we override, since it's common to invoke the | 
|  | 2737 | /// overridden function as well as adding new functionality. | 
|  | 2738 | /// | 
|  | 2739 | /// \param S The semantic analysis object for which we are generating results. | 
|  | 2740 | /// | 
|  | 2741 | /// \param InContext This context in which the nested-name-specifier preceding | 
|  | 2742 | /// the code-completion point | 
|  | 2743 | static void MaybeAddOverrideCalls(Sema &S, DeclContext *InContext, | 
|  | 2744 | ResultBuilder &Results) { | 
|  | 2745 | // Look through blocks. | 
|  | 2746 | DeclContext *CurContext = S.CurContext; | 
|  | 2747 | while (isa<BlockDecl>(CurContext)) | 
|  | 2748 | CurContext = CurContext->getParent(); | 
|  | 2749 |  | 
|  | 2750 |  | 
|  | 2751 | CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(CurContext); | 
|  | 2752 | if (!Method || !Method->isVirtual()) | 
|  | 2753 | return; | 
|  | 2754 |  | 
|  | 2755 | // We need to have names for all of the parameters, if we're going to | 
|  | 2756 | // generate a forwarding call. | 
|  | 2757 | for (CXXMethodDecl::param_iterator P = Method->param_begin(), | 
|  | 2758 | PEnd = Method->param_end(); | 
|  | 2759 | P != PEnd; | 
|  | 2760 | ++P) { | 
|  | 2761 | if (!(*P)->getDeclName()) | 
|  | 2762 | return; | 
|  | 2763 | } | 
|  | 2764 |  | 
|  | 2765 | for (CXXMethodDecl::method_iterator M = Method->begin_overridden_methods(), | 
|  | 2766 | MEnd = Method->end_overridden_methods(); | 
|  | 2767 | M != MEnd; ++M) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2768 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
| Douglas Gregor | f696152 | 2010-08-27 21:18:54 +0000 | [diff] [blame] | 2769 | CXXMethodDecl *Overridden = const_cast<CXXMethodDecl *>(*M); | 
|  | 2770 | if (Overridden->getCanonicalDecl() == Method->getCanonicalDecl()) | 
|  | 2771 | continue; | 
|  | 2772 |  | 
|  | 2773 | // If we need a nested-name-specifier, add one now. | 
|  | 2774 | if (!InContext) { | 
|  | 2775 | NestedNameSpecifier *NNS | 
|  | 2776 | = getRequiredQualification(S.Context, CurContext, | 
|  | 2777 | Overridden->getDeclContext()); | 
|  | 2778 | if (NNS) { | 
|  | 2779 | std::string Str; | 
|  | 2780 | llvm::raw_string_ostream OS(Str); | 
|  | 2781 | NNS->print(OS, S.Context.PrintingPolicy); | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2782 | Builder.AddTextChunk(Results.getAllocator().CopyString(OS.str())); | 
| Douglas Gregor | f696152 | 2010-08-27 21:18:54 +0000 | [diff] [blame] | 2783 | } | 
|  | 2784 | } else if (!InContext->Equals(Overridden->getDeclContext())) | 
|  | 2785 | continue; | 
|  | 2786 |  | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2787 | Builder.AddTypedTextChunk(Results.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2788 | Overridden->getNameAsString())); | 
|  | 2789 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
| Douglas Gregor | f696152 | 2010-08-27 21:18:54 +0000 | [diff] [blame] | 2790 | bool FirstParam = true; | 
|  | 2791 | for (CXXMethodDecl::param_iterator P = Method->param_begin(), | 
|  | 2792 | PEnd = Method->param_end(); | 
|  | 2793 | P != PEnd; ++P) { | 
|  | 2794 | if (FirstParam) | 
|  | 2795 | FirstParam = false; | 
|  | 2796 | else | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2797 | Builder.AddChunk(CodeCompletionString::CK_Comma); | 
| Douglas Gregor | f696152 | 2010-08-27 21:18:54 +0000 | [diff] [blame] | 2798 |  | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 2799 | Builder.AddPlaceholderChunk(Results.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2800 | (*P)->getIdentifier()->getName())); | 
| Douglas Gregor | f696152 | 2010-08-27 21:18:54 +0000 | [diff] [blame] | 2801 | } | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2802 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 2803 | Results.AddResult(CodeCompletionResult(Builder.TakeString(), | 
| Douglas Gregor | f696152 | 2010-08-27 21:18:54 +0000 | [diff] [blame] | 2804 | CCP_SuperCompletion, | 
|  | 2805 | CXCursor_CXXMethod)); | 
|  | 2806 | Results.Ignore(Overridden); | 
|  | 2807 | } | 
|  | 2808 | } | 
|  | 2809 |  | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 2810 | void Sema::CodeCompleteOrdinaryName(Scope *S, | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2811 | ParserCompletionContext CompletionContext) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 2812 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2813 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 2814 | mapCodeCompletionContext(*this, CompletionContext)); | 
| Douglas Gregor | f696152 | 2010-08-27 21:18:54 +0000 | [diff] [blame] | 2815 | Results.EnterNewScope(); | 
| Douglas Gregor | cee9ff1 | 2010-09-20 22:39:41 +0000 | [diff] [blame] | 2816 |  | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 2817 | // Determine how to filter results, e.g., so that the names of | 
|  | 2818 | // values (functions, enumerators, function templates, etc.) are | 
|  | 2819 | // only allowed where we can have an expression. | 
|  | 2820 | switch (CompletionContext) { | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2821 | case PCC_Namespace: | 
|  | 2822 | case PCC_Class: | 
|  | 2823 | case PCC_ObjCInterface: | 
|  | 2824 | case PCC_ObjCImplementation: | 
|  | 2825 | case PCC_ObjCInstanceVariableList: | 
|  | 2826 | case PCC_Template: | 
|  | 2827 | case PCC_MemberTemplate: | 
| Douglas Gregor | 72db108 | 2010-08-24 01:11:00 +0000 | [diff] [blame] | 2828 | case PCC_Type: | 
| Douglas Gregor | 68e3c2e | 2011-02-15 20:33:25 +0000 | [diff] [blame] | 2829 | case PCC_LocalDeclarationSpecifiers: | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 2830 | Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName); | 
|  | 2831 | break; | 
|  | 2832 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2833 | case PCC_Statement: | 
| Douglas Gregor | 0268810 | 2010-09-14 23:59:36 +0000 | [diff] [blame] | 2834 | case PCC_ParenthesizedExpression: | 
| Douglas Gregor | eb0d014 | 2010-08-24 23:58:17 +0000 | [diff] [blame] | 2835 | case PCC_Expression: | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2836 | case PCC_ForInit: | 
|  | 2837 | case PCC_Condition: | 
| Douglas Gregor | 4710e5b | 2010-05-28 00:49:12 +0000 | [diff] [blame] | 2838 | if (WantTypesInContext(CompletionContext, getLangOptions())) | 
|  | 2839 | Results.setFilter(&ResultBuilder::IsOrdinaryName); | 
|  | 2840 | else | 
|  | 2841 | Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName); | 
| Douglas Gregor | f696152 | 2010-08-27 21:18:54 +0000 | [diff] [blame] | 2842 |  | 
|  | 2843 | if (getLangOptions().CPlusPlus) | 
|  | 2844 | MaybeAddOverrideCalls(*this, /*InContext=*/0, Results); | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 2845 | break; | 
| Douglas Gregor | dc84534 | 2010-05-25 05:58:43 +0000 | [diff] [blame] | 2846 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2847 | case PCC_RecoveryInFunction: | 
| Douglas Gregor | dc84534 | 2010-05-25 05:58:43 +0000 | [diff] [blame] | 2848 | // Unfiltered | 
|  | 2849 | break; | 
| Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 2850 | } | 
|  | 2851 |  | 
| Douglas Gregor | 3cdee12 | 2010-08-26 16:36:48 +0000 | [diff] [blame] | 2852 | // If we are in a C++ non-static member function, check the qualifiers on | 
|  | 2853 | // the member function to filter/prioritize the results list. | 
|  | 2854 | if (CXXMethodDecl *CurMethod = dyn_cast<CXXMethodDecl>(CurContext)) | 
|  | 2855 | if (CurMethod->isInstance()) | 
|  | 2856 | Results.setObjectTypeQualifiers( | 
|  | 2857 | Qualifiers::fromCVRMask(CurMethod->getTypeQualifiers())); | 
|  | 2858 |  | 
| Douglas Gregor | 1ca6ae8 | 2010-01-14 01:09:38 +0000 | [diff] [blame] | 2859 | CodeCompletionDeclConsumer Consumer(Results, CurContext); | 
| Douglas Gregor | 8071e42 | 2010-08-15 06:18:01 +0000 | [diff] [blame] | 2860 | LookupVisibleDecls(S, LookupOrdinaryName, Consumer, | 
|  | 2861 | CodeCompleter->includeGlobals()); | 
| Douglas Gregor | 2a7925c | 2009-12-07 09:54:55 +0000 | [diff] [blame] | 2862 |  | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 2863 | AddOrdinaryNameResults(CompletionContext, S, *this, Results); | 
| Douglas Gregor | 2a7925c | 2009-12-07 09:54:55 +0000 | [diff] [blame] | 2864 | Results.ExitScope(); | 
|  | 2865 |  | 
| Douglas Gregor | aa5f77b | 2010-08-23 21:54:33 +0000 | [diff] [blame] | 2866 | switch (CompletionContext) { | 
| Douglas Gregor | 0268810 | 2010-09-14 23:59:36 +0000 | [diff] [blame] | 2867 | case PCC_ParenthesizedExpression: | 
| Douglas Gregor | 72db108 | 2010-08-24 01:11:00 +0000 | [diff] [blame] | 2868 | case PCC_Expression: | 
|  | 2869 | case PCC_Statement: | 
|  | 2870 | case PCC_RecoveryInFunction: | 
|  | 2871 | if (S->getFnParent()) | 
|  | 2872 | AddPrettyFunctionResults(PP.getLangOptions(), Results); | 
|  | 2873 | break; | 
|  | 2874 |  | 
|  | 2875 | case PCC_Namespace: | 
|  | 2876 | case PCC_Class: | 
|  | 2877 | case PCC_ObjCInterface: | 
|  | 2878 | case PCC_ObjCImplementation: | 
|  | 2879 | case PCC_ObjCInstanceVariableList: | 
|  | 2880 | case PCC_Template: | 
|  | 2881 | case PCC_MemberTemplate: | 
|  | 2882 | case PCC_ForInit: | 
|  | 2883 | case PCC_Condition: | 
|  | 2884 | case PCC_Type: | 
| Douglas Gregor | 68e3c2e | 2011-02-15 20:33:25 +0000 | [diff] [blame] | 2885 | case PCC_LocalDeclarationSpecifiers: | 
| Douglas Gregor | 72db108 | 2010-08-24 01:11:00 +0000 | [diff] [blame] | 2886 | break; | 
| Douglas Gregor | aa5f77b | 2010-08-23 21:54:33 +0000 | [diff] [blame] | 2887 | } | 
|  | 2888 |  | 
| Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 2889 | if (CodeCompleter->includeMacros()) | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 2890 | AddMacroResults(PP, Results); | 
| Douglas Gregor | aa5f77b | 2010-08-23 21:54:33 +0000 | [diff] [blame] | 2891 |  | 
| Douglas Gregor | cee9ff1 | 2010-09-20 22:39:41 +0000 | [diff] [blame] | 2892 | HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(), | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2893 | Results.data(),Results.size()); | 
| Douglas Gregor | 791215b | 2009-09-21 20:51:25 +0000 | [diff] [blame] | 2894 | } | 
|  | 2895 |  | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 2896 | static void AddClassMessageCompletions(Sema &SemaRef, Scope *S, | 
|  | 2897 | ParsedType Receiver, | 
|  | 2898 | IdentifierInfo **SelIdents, | 
|  | 2899 | unsigned NumSelIdents, | 
| Douglas Gregor | 70c5ac7 | 2010-09-20 23:34:21 +0000 | [diff] [blame] | 2900 | bool AtArgumentExpression, | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 2901 | bool IsSuper, | 
|  | 2902 | ResultBuilder &Results); | 
|  | 2903 |  | 
|  | 2904 | void Sema::CodeCompleteDeclSpec(Scope *S, DeclSpec &DS, | 
|  | 2905 | bool AllowNonIdentifiers, | 
|  | 2906 | bool AllowNestedNameSpecifiers) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 2907 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2908 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 2909 | AllowNestedNameSpecifiers | 
|  | 2910 | ? CodeCompletionContext::CCC_PotentiallyQualifiedName | 
|  | 2911 | : CodeCompletionContext::CCC_Name); | 
| Douglas Gregor | 2ccccb3 | 2010-08-23 18:23:48 +0000 | [diff] [blame] | 2912 | Results.EnterNewScope(); | 
|  | 2913 |  | 
|  | 2914 | // Type qualifiers can come after names. | 
|  | 2915 | Results.AddResult(Result("const")); | 
|  | 2916 | Results.AddResult(Result("volatile")); | 
|  | 2917 | if (getLangOptions().C99) | 
|  | 2918 | Results.AddResult(Result("restrict")); | 
|  | 2919 |  | 
|  | 2920 | if (getLangOptions().CPlusPlus) { | 
|  | 2921 | if (AllowNonIdentifiers) { | 
|  | 2922 | Results.AddResult(Result("operator")); | 
|  | 2923 | } | 
|  | 2924 |  | 
|  | 2925 | // Add nested-name-specifiers. | 
|  | 2926 | if (AllowNestedNameSpecifiers) { | 
|  | 2927 | Results.allowNestedNameSpecifiers(); | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 2928 | Results.setFilter(&ResultBuilder::IsImpossibleToSatisfy); | 
| Douglas Gregor | 2ccccb3 | 2010-08-23 18:23:48 +0000 | [diff] [blame] | 2929 | CodeCompletionDeclConsumer Consumer(Results, CurContext); | 
|  | 2930 | LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer, | 
|  | 2931 | CodeCompleter->includeGlobals()); | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 2932 | Results.setFilter(0); | 
| Douglas Gregor | 2ccccb3 | 2010-08-23 18:23:48 +0000 | [diff] [blame] | 2933 | } | 
|  | 2934 | } | 
|  | 2935 | Results.ExitScope(); | 
|  | 2936 |  | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 2937 | // If we're in a context where we might have an expression (rather than a | 
|  | 2938 | // declaration), and what we've seen so far is an Objective-C type that could | 
|  | 2939 | // be a receiver of a class message, this may be a class message send with | 
|  | 2940 | // the initial opening bracket '[' missing. Add appropriate completions. | 
|  | 2941 | if (AllowNonIdentifiers && !AllowNestedNameSpecifiers && | 
|  | 2942 | DS.getTypeSpecType() == DeclSpec::TST_typename && | 
|  | 2943 | DS.getStorageClassSpecAsWritten() == DeclSpec::SCS_unspecified && | 
|  | 2944 | !DS.isThreadSpecified() && !DS.isExternInLinkageSpec() && | 
|  | 2945 | DS.getTypeSpecComplex() == DeclSpec::TSC_unspecified && | 
|  | 2946 | DS.getTypeSpecSign() == DeclSpec::TSS_unspecified && | 
|  | 2947 | DS.getTypeQualifiers() == 0 && | 
|  | 2948 | S && | 
|  | 2949 | (S->getFlags() & Scope::DeclScope) != 0 && | 
|  | 2950 | (S->getFlags() & (Scope::ClassScope | Scope::TemplateParamScope | | 
|  | 2951 | Scope::FunctionPrototypeScope | | 
|  | 2952 | Scope::AtCatchScope)) == 0) { | 
|  | 2953 | ParsedType T = DS.getRepAsType(); | 
|  | 2954 | if (!T.get().isNull() && T.get()->isObjCObjectOrInterfaceType()) | 
| Douglas Gregor | 70c5ac7 | 2010-09-20 23:34:21 +0000 | [diff] [blame] | 2955 | AddClassMessageCompletions(*this, S, T, 0, 0, false, false, Results); | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 2956 | } | 
|  | 2957 |  | 
| Douglas Gregor | 4497dd4 | 2010-08-24 04:59:56 +0000 | [diff] [blame] | 2958 | // Note that we intentionally suppress macro results here, since we do not | 
|  | 2959 | // encourage using macros to produce the names of entities. | 
|  | 2960 |  | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 2961 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 2962 | Results.getCompletionContext(), | 
| Douglas Gregor | 2ccccb3 | 2010-08-23 18:23:48 +0000 | [diff] [blame] | 2963 | Results.data(), Results.size()); | 
|  | 2964 | } | 
|  | 2965 |  | 
| Douglas Gregor | fb62941 | 2010-08-23 21:17:50 +0000 | [diff] [blame] | 2966 | struct Sema::CodeCompleteExpressionData { | 
|  | 2967 | CodeCompleteExpressionData(QualType PreferredType = QualType()) | 
|  | 2968 | : PreferredType(PreferredType), IntegralConstantExpression(false), | 
|  | 2969 | ObjCCollection(false) { } | 
|  | 2970 |  | 
|  | 2971 | QualType PreferredType; | 
|  | 2972 | bool IntegralConstantExpression; | 
|  | 2973 | bool ObjCCollection; | 
|  | 2974 | llvm::SmallVector<Decl *, 4> IgnoreDecls; | 
|  | 2975 | }; | 
|  | 2976 |  | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 2977 | /// \brief Perform code-completion in an expression context when we know what | 
|  | 2978 | /// type we're looking for. | 
| Douglas Gregor | f957843 | 2010-07-28 21:50:18 +0000 | [diff] [blame] | 2979 | /// | 
|  | 2980 | /// \param IntegralConstantExpression Only permit integral constant | 
|  | 2981 | /// expressions. | 
| Douglas Gregor | fb62941 | 2010-08-23 21:17:50 +0000 | [diff] [blame] | 2982 | void Sema::CodeCompleteExpression(Scope *S, | 
|  | 2983 | const CodeCompleteExpressionData &Data) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 2984 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 2985 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 2986 | CodeCompletionContext::CCC_Expression); | 
| Douglas Gregor | fb62941 | 2010-08-23 21:17:50 +0000 | [diff] [blame] | 2987 | if (Data.ObjCCollection) | 
|  | 2988 | Results.setFilter(&ResultBuilder::IsObjCCollection); | 
|  | 2989 | else if (Data.IntegralConstantExpression) | 
| Douglas Gregor | f957843 | 2010-07-28 21:50:18 +0000 | [diff] [blame] | 2990 | Results.setFilter(&ResultBuilder::IsIntegralConstantValue); | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 2991 | else if (WantTypesInContext(PCC_Expression, getLangOptions())) | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 2992 | Results.setFilter(&ResultBuilder::IsOrdinaryName); | 
|  | 2993 | else | 
|  | 2994 | Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName); | 
| Douglas Gregor | fb62941 | 2010-08-23 21:17:50 +0000 | [diff] [blame] | 2995 |  | 
|  | 2996 | if (!Data.PreferredType.isNull()) | 
|  | 2997 | Results.setPreferredType(Data.PreferredType.getNonReferenceType()); | 
|  | 2998 |  | 
|  | 2999 | // Ignore any declarations that we were told that we don't care about. | 
|  | 3000 | for (unsigned I = 0, N = Data.IgnoreDecls.size(); I != N; ++I) | 
|  | 3001 | Results.Ignore(Data.IgnoreDecls[I]); | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 3002 |  | 
|  | 3003 | CodeCompletionDeclConsumer Consumer(Results, CurContext); | 
| Douglas Gregor | 8071e42 | 2010-08-15 06:18:01 +0000 | [diff] [blame] | 3004 | LookupVisibleDecls(S, LookupOrdinaryName, Consumer, | 
|  | 3005 | CodeCompleter->includeGlobals()); | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 3006 |  | 
|  | 3007 | Results.EnterNewScope(); | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3008 | AddOrdinaryNameResults(PCC_Expression, S, *this, Results); | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 3009 | Results.ExitScope(); | 
|  | 3010 |  | 
| Douglas Gregor | 590c7d5 | 2010-07-08 20:55:51 +0000 | [diff] [blame] | 3011 | bool PreferredTypeIsPointer = false; | 
| Douglas Gregor | fb62941 | 2010-08-23 21:17:50 +0000 | [diff] [blame] | 3012 | if (!Data.PreferredType.isNull()) | 
|  | 3013 | PreferredTypeIsPointer = Data.PreferredType->isAnyPointerType() | 
|  | 3014 | || Data.PreferredType->isMemberPointerType() | 
|  | 3015 | || Data.PreferredType->isBlockPointerType(); | 
| Douglas Gregor | 590c7d5 | 2010-07-08 20:55:51 +0000 | [diff] [blame] | 3016 |  | 
| Douglas Gregor | aa5f77b | 2010-08-23 21:54:33 +0000 | [diff] [blame] | 3017 | if (S->getFnParent() && | 
|  | 3018 | !Data.ObjCCollection && | 
|  | 3019 | !Data.IntegralConstantExpression) | 
|  | 3020 | AddPrettyFunctionResults(PP.getLangOptions(), Results); | 
|  | 3021 |  | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 3022 | if (CodeCompleter->includeMacros()) | 
| Douglas Gregor | 590c7d5 | 2010-07-08 20:55:51 +0000 | [diff] [blame] | 3023 | AddMacroResults(PP, Results, PreferredTypeIsPointer); | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3024 | HandleCodeCompleteResults(this, CodeCompleter, | 
| Douglas Gregor | fb62941 | 2010-08-23 21:17:50 +0000 | [diff] [blame] | 3025 | CodeCompletionContext(CodeCompletionContext::CCC_Expression, | 
|  | 3026 | Data.PreferredType), | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3027 | Results.data(),Results.size()); | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 3028 | } | 
|  | 3029 |  | 
| Douglas Gregor | ac5fd84 | 2010-09-18 01:28:11 +0000 | [diff] [blame] | 3030 | void Sema::CodeCompletePostfixExpression(Scope *S, ExprResult E) { | 
|  | 3031 | if (E.isInvalid()) | 
|  | 3032 | CodeCompleteOrdinaryName(S, PCC_RecoveryInFunction); | 
|  | 3033 | else if (getLangOptions().ObjC1) | 
|  | 3034 | CodeCompleteObjCInstanceMessage(S, E.take(), 0, 0, false); | 
| Douglas Gregor | 78edf51 | 2010-09-15 16:23:04 +0000 | [diff] [blame] | 3035 | } | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 3036 |  | 
| Douglas Gregor | 7344921 | 2010-12-09 23:01:55 +0000 | [diff] [blame] | 3037 | /// \brief The set of properties that have already been added, referenced by | 
|  | 3038 | /// property name. | 
|  | 3039 | typedef llvm::SmallPtrSet<IdentifierInfo*, 16> AddedPropertiesSet; | 
|  | 3040 |  | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3041 | static void AddObjCProperties(ObjCContainerDecl *Container, | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 3042 | bool AllowCategories, | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3043 | DeclContext *CurContext, | 
| Douglas Gregor | 7344921 | 2010-12-09 23:01:55 +0000 | [diff] [blame] | 3044 | AddedPropertiesSet &AddedProperties, | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3045 | ResultBuilder &Results) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 3046 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3047 |  | 
|  | 3048 | // Add properties in this container. | 
|  | 3049 | for (ObjCContainerDecl::prop_iterator P = Container->prop_begin(), | 
|  | 3050 | PEnd = Container->prop_end(); | 
|  | 3051 | P != PEnd; | 
| Douglas Gregor | 7344921 | 2010-12-09 23:01:55 +0000 | [diff] [blame] | 3052 | ++P) { | 
|  | 3053 | if (AddedProperties.insert(P->getIdentifier())) | 
|  | 3054 | Results.MaybeAddResult(Result(*P, 0), CurContext); | 
|  | 3055 | } | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3056 |  | 
|  | 3057 | // Add properties in referenced protocols. | 
|  | 3058 | if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) { | 
|  | 3059 | for (ObjCProtocolDecl::protocol_iterator P = Protocol->protocol_begin(), | 
|  | 3060 | PEnd = Protocol->protocol_end(); | 
|  | 3061 | P != PEnd; ++P) | 
| Douglas Gregor | 7344921 | 2010-12-09 23:01:55 +0000 | [diff] [blame] | 3062 | AddObjCProperties(*P, AllowCategories, CurContext, AddedProperties, | 
|  | 3063 | Results); | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3064 | } else if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)){ | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 3065 | if (AllowCategories) { | 
|  | 3066 | // Look through categories. | 
|  | 3067 | for (ObjCCategoryDecl *Category = IFace->getCategoryList(); | 
|  | 3068 | Category; Category = Category->getNextClassCategory()) | 
| Douglas Gregor | 7344921 | 2010-12-09 23:01:55 +0000 | [diff] [blame] | 3069 | AddObjCProperties(Category, AllowCategories, CurContext, | 
|  | 3070 | AddedProperties, Results); | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 3071 | } | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3072 |  | 
|  | 3073 | // Look through protocols. | 
| Ted Kremenek | 53b9441 | 2010-09-01 01:21:15 +0000 | [diff] [blame] | 3074 | for (ObjCInterfaceDecl::all_protocol_iterator | 
|  | 3075 | I = IFace->all_referenced_protocol_begin(), | 
|  | 3076 | E = IFace->all_referenced_protocol_end(); I != E; ++I) | 
| Douglas Gregor | 7344921 | 2010-12-09 23:01:55 +0000 | [diff] [blame] | 3077 | AddObjCProperties(*I, AllowCategories, CurContext, AddedProperties, | 
|  | 3078 | Results); | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3079 |  | 
|  | 3080 | // Look in the superclass. | 
|  | 3081 | if (IFace->getSuperClass()) | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 3082 | AddObjCProperties(IFace->getSuperClass(), AllowCategories, CurContext, | 
| Douglas Gregor | 7344921 | 2010-12-09 23:01:55 +0000 | [diff] [blame] | 3083 | AddedProperties, Results); | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3084 | } else if (const ObjCCategoryDecl *Category | 
|  | 3085 | = dyn_cast<ObjCCategoryDecl>(Container)) { | 
|  | 3086 | // Look through protocols. | 
| Ted Kremenek | 53b9441 | 2010-09-01 01:21:15 +0000 | [diff] [blame] | 3087 | for (ObjCCategoryDecl::protocol_iterator P = Category->protocol_begin(), | 
|  | 3088 | PEnd = Category->protocol_end(); | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3089 | P != PEnd; ++P) | 
| Douglas Gregor | 7344921 | 2010-12-09 23:01:55 +0000 | [diff] [blame] | 3090 | AddObjCProperties(*P, AllowCategories, CurContext, AddedProperties, | 
|  | 3091 | Results); | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3092 | } | 
|  | 3093 | } | 
|  | 3094 |  | 
| Douglas Gregor | 81b747b | 2009-09-17 21:32:03 +0000 | [diff] [blame] | 3095 | void Sema::CodeCompleteMemberReferenceExpr(Scope *S, ExprTy *BaseE, | 
|  | 3096 | SourceLocation OpLoc, | 
|  | 3097 | bool IsArrow) { | 
|  | 3098 | if (!BaseE || !CodeCompleter) | 
|  | 3099 | return; | 
|  | 3100 |  | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 3101 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3102 |  | 
| Douglas Gregor | 81b747b | 2009-09-17 21:32:03 +0000 | [diff] [blame] | 3103 | Expr *Base = static_cast<Expr *>(BaseE); | 
|  | 3104 | QualType BaseType = Base->getType(); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3105 |  | 
|  | 3106 | if (IsArrow) { | 
|  | 3107 | if (const PointerType *Ptr = BaseType->getAs<PointerType>()) | 
|  | 3108 | BaseType = Ptr->getPointeeType(); | 
|  | 3109 | else if (BaseType->isObjCObjectPointerType()) | 
| Douglas Gregor | 3cdee12 | 2010-08-26 16:36:48 +0000 | [diff] [blame] | 3110 | /*Do nothing*/ ; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3111 | else | 
|  | 3112 | return; | 
|  | 3113 | } | 
|  | 3114 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3115 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3116 | CodeCompletionContext(CodeCompletionContext::CCC_MemberAccess, | 
|  | 3117 | BaseType), | 
|  | 3118 | &ResultBuilder::IsMember); | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3119 | Results.EnterNewScope(); | 
|  | 3120 | if (const RecordType *Record = BaseType->getAs<RecordType>()) { | 
| Douglas Gregor | 3cdee12 | 2010-08-26 16:36:48 +0000 | [diff] [blame] | 3121 | // Indicate that we are performing a member access, and the cv-qualifiers | 
|  | 3122 | // for the base object type. | 
|  | 3123 | Results.setObjectTypeQualifiers(BaseType.getQualifiers()); | 
|  | 3124 |  | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3125 | // Access to a C/C++ class, struct, or union. | 
| Douglas Gregor | 45bcd43 | 2010-01-14 03:21:49 +0000 | [diff] [blame] | 3126 | Results.allowNestedNameSpecifiers(); | 
| Douglas Gregor | 0cc8404 | 2010-01-14 15:47:35 +0000 | [diff] [blame] | 3127 | CodeCompletionDeclConsumer Consumer(Results, CurContext); | 
| Douglas Gregor | 8071e42 | 2010-08-15 06:18:01 +0000 | [diff] [blame] | 3128 | LookupVisibleDecls(Record->getDecl(), LookupMemberName, Consumer, | 
|  | 3129 | CodeCompleter->includeGlobals()); | 
| Daniel Dunbar | 3a2838d | 2009-11-13 08:58:20 +0000 | [diff] [blame] | 3130 |  | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3131 | if (getLangOptions().CPlusPlus) { | 
|  | 3132 | if (!Results.empty()) { | 
|  | 3133 | // The "template" keyword can follow "->" or "." in the grammar. | 
|  | 3134 | // However, we only want to suggest the template keyword if something | 
|  | 3135 | // is dependent. | 
|  | 3136 | bool IsDependent = BaseType->isDependentType(); | 
|  | 3137 | if (!IsDependent) { | 
|  | 3138 | for (Scope *DepScope = S; DepScope; DepScope = DepScope->getParent()) | 
|  | 3139 | if (DeclContext *Ctx = (DeclContext *)DepScope->getEntity()) { | 
|  | 3140 | IsDependent = Ctx->isDependentContext(); | 
|  | 3141 | break; | 
|  | 3142 | } | 
|  | 3143 | } | 
| Daniel Dunbar | 3a2838d | 2009-11-13 08:58:20 +0000 | [diff] [blame] | 3144 |  | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3145 | if (IsDependent) | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 3146 | Results.AddResult(Result("template")); | 
| Daniel Dunbar | 3a2838d | 2009-11-13 08:58:20 +0000 | [diff] [blame] | 3147 | } | 
| Daniel Dunbar | 3a2838d | 2009-11-13 08:58:20 +0000 | [diff] [blame] | 3148 | } | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3149 | } else if (!IsArrow && BaseType->getAsObjCInterfacePointerType()) { | 
|  | 3150 | // Objective-C property reference. | 
| Douglas Gregor | 7344921 | 2010-12-09 23:01:55 +0000 | [diff] [blame] | 3151 | AddedPropertiesSet AddedProperties; | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3152 |  | 
|  | 3153 | // Add property results based on our interface. | 
|  | 3154 | const ObjCObjectPointerType *ObjCPtr | 
|  | 3155 | = BaseType->getAsObjCInterfacePointerType(); | 
|  | 3156 | assert(ObjCPtr && "Non-NULL pointer guaranteed above!"); | 
| Douglas Gregor | 7344921 | 2010-12-09 23:01:55 +0000 | [diff] [blame] | 3157 | AddObjCProperties(ObjCPtr->getInterfaceDecl(), true, CurContext, | 
|  | 3158 | AddedProperties, Results); | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3159 |  | 
|  | 3160 | // Add properties from the protocols in a qualified interface. | 
|  | 3161 | for (ObjCObjectPointerType::qual_iterator I = ObjCPtr->qual_begin(), | 
|  | 3162 | E = ObjCPtr->qual_end(); | 
|  | 3163 | I != E; ++I) | 
| Douglas Gregor | 7344921 | 2010-12-09 23:01:55 +0000 | [diff] [blame] | 3164 | AddObjCProperties(*I, true, CurContext, AddedProperties, Results); | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3165 | } else if ((IsArrow && BaseType->isObjCObjectPointerType()) || | 
| John McCall | c12c5bb | 2010-05-15 11:32:37 +0000 | [diff] [blame] | 3166 | (!IsArrow && BaseType->isObjCObjectType())) { | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3167 | // Objective-C instance variable access. | 
|  | 3168 | ObjCInterfaceDecl *Class = 0; | 
|  | 3169 | if (const ObjCObjectPointerType *ObjCPtr | 
|  | 3170 | = BaseType->getAs<ObjCObjectPointerType>()) | 
|  | 3171 | Class = ObjCPtr->getInterfaceDecl(); | 
|  | 3172 | else | 
| John McCall | c12c5bb | 2010-05-15 11:32:37 +0000 | [diff] [blame] | 3173 | Class = BaseType->getAs<ObjCObjectType>()->getInterface(); | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3174 |  | 
|  | 3175 | // Add all ivars from this class and its superclasses. | 
| Douglas Gregor | 80f4f4c | 2010-01-14 16:08:12 +0000 | [diff] [blame] | 3176 | if (Class) { | 
|  | 3177 | CodeCompletionDeclConsumer Consumer(Results, CurContext); | 
|  | 3178 | Results.setFilter(&ResultBuilder::IsObjCIvar); | 
| Douglas Gregor | 8071e42 | 2010-08-15 06:18:01 +0000 | [diff] [blame] | 3179 | LookupVisibleDecls(Class, LookupMemberName, Consumer, | 
|  | 3180 | CodeCompleter->includeGlobals()); | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3181 | } | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3182 | } | 
| Douglas Gregor | 95ac655 | 2009-11-18 01:29:26 +0000 | [diff] [blame] | 3183 |  | 
|  | 3184 | // FIXME: How do we cope with isa? | 
|  | 3185 |  | 
|  | 3186 | Results.ExitScope(); | 
| Daniel Dunbar | 3a2838d | 2009-11-13 08:58:20 +0000 | [diff] [blame] | 3187 |  | 
| Daniel Dunbar | 3a2838d | 2009-11-13 08:58:20 +0000 | [diff] [blame] | 3188 | // Hand off the results found for code completion. | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3189 | HandleCodeCompleteResults(this, CodeCompleter, | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3190 | Results.getCompletionContext(), | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3191 | Results.data(),Results.size()); | 
| Douglas Gregor | 81b747b | 2009-09-17 21:32:03 +0000 | [diff] [blame] | 3192 | } | 
|  | 3193 |  | 
| Douglas Gregor | 374929f | 2009-09-18 15:37:17 +0000 | [diff] [blame] | 3194 | void Sema::CodeCompleteTag(Scope *S, unsigned TagSpec) { | 
|  | 3195 | if (!CodeCompleter) | 
|  | 3196 | return; | 
|  | 3197 |  | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 3198 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3199 | ResultBuilder::LookupFilter Filter = 0; | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3200 | enum CodeCompletionContext::Kind ContextKind | 
|  | 3201 | = CodeCompletionContext::CCC_Other; | 
| Douglas Gregor | 374929f | 2009-09-18 15:37:17 +0000 | [diff] [blame] | 3202 | switch ((DeclSpec::TST)TagSpec) { | 
|  | 3203 | case DeclSpec::TST_enum: | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3204 | Filter = &ResultBuilder::IsEnum; | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3205 | ContextKind = CodeCompletionContext::CCC_EnumTag; | 
| Douglas Gregor | 374929f | 2009-09-18 15:37:17 +0000 | [diff] [blame] | 3206 | break; | 
|  | 3207 |  | 
|  | 3208 | case DeclSpec::TST_union: | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3209 | Filter = &ResultBuilder::IsUnion; | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3210 | ContextKind = CodeCompletionContext::CCC_UnionTag; | 
| Douglas Gregor | 374929f | 2009-09-18 15:37:17 +0000 | [diff] [blame] | 3211 | break; | 
|  | 3212 |  | 
|  | 3213 | case DeclSpec::TST_struct: | 
| Douglas Gregor | 374929f | 2009-09-18 15:37:17 +0000 | [diff] [blame] | 3214 | case DeclSpec::TST_class: | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3215 | Filter = &ResultBuilder::IsClassOrStruct; | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3216 | ContextKind = CodeCompletionContext::CCC_ClassOrStructTag; | 
| Douglas Gregor | 374929f | 2009-09-18 15:37:17 +0000 | [diff] [blame] | 3217 | break; | 
|  | 3218 |  | 
|  | 3219 | default: | 
|  | 3220 | assert(false && "Unknown type specifier kind in CodeCompleteTag"); | 
|  | 3221 | return; | 
|  | 3222 | } | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3223 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3224 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), ContextKind); | 
| Douglas Gregor | 5d2fc40 | 2010-01-14 03:27:13 +0000 | [diff] [blame] | 3225 | CodeCompletionDeclConsumer Consumer(Results, CurContext); | 
| John McCall | 0d6b164 | 2010-04-23 18:46:30 +0000 | [diff] [blame] | 3226 |  | 
|  | 3227 | // First pass: look for tags. | 
|  | 3228 | Results.setFilter(Filter); | 
| Douglas Gregor | 8071e42 | 2010-08-15 06:18:01 +0000 | [diff] [blame] | 3229 | LookupVisibleDecls(S, LookupTagName, Consumer, | 
|  | 3230 | CodeCompleter->includeGlobals()); | 
| John McCall | 0d6b164 | 2010-04-23 18:46:30 +0000 | [diff] [blame] | 3231 |  | 
| Douglas Gregor | 8071e42 | 2010-08-15 06:18:01 +0000 | [diff] [blame] | 3232 | if (CodeCompleter->includeGlobals()) { | 
|  | 3233 | // Second pass: look for nested name specifiers. | 
|  | 3234 | Results.setFilter(&ResultBuilder::IsNestedNameSpecifier); | 
|  | 3235 | LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer); | 
|  | 3236 | } | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3237 |  | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3238 | HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(), | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3239 | Results.data(),Results.size()); | 
| Douglas Gregor | 374929f | 2009-09-18 15:37:17 +0000 | [diff] [blame] | 3240 | } | 
|  | 3241 |  | 
| Douglas Gregor | 1a480c4 | 2010-08-27 17:35:51 +0000 | [diff] [blame] | 3242 | void Sema::CodeCompleteTypeQualifiers(DeclSpec &DS) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3243 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 3244 | CodeCompletionContext::CCC_TypeQualifiers); | 
| Douglas Gregor | 1a480c4 | 2010-08-27 17:35:51 +0000 | [diff] [blame] | 3245 | Results.EnterNewScope(); | 
|  | 3246 | if (!(DS.getTypeQualifiers() & DeclSpec::TQ_const)) | 
|  | 3247 | Results.AddResult("const"); | 
|  | 3248 | if (!(DS.getTypeQualifiers() & DeclSpec::TQ_volatile)) | 
|  | 3249 | Results.AddResult("volatile"); | 
|  | 3250 | if (getLangOptions().C99 && | 
|  | 3251 | !(DS.getTypeQualifiers() & DeclSpec::TQ_restrict)) | 
|  | 3252 | Results.AddResult("restrict"); | 
|  | 3253 | Results.ExitScope(); | 
|  | 3254 | HandleCodeCompleteResults(this, CodeCompleter, | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3255 | Results.getCompletionContext(), | 
| Douglas Gregor | 1a480c4 | 2010-08-27 17:35:51 +0000 | [diff] [blame] | 3256 | Results.data(), Results.size()); | 
|  | 3257 | } | 
|  | 3258 |  | 
| Douglas Gregor | 3e1005f | 2009-09-21 18:10:23 +0000 | [diff] [blame] | 3259 | void Sema::CodeCompleteCase(Scope *S) { | 
| John McCall | 781472f | 2010-08-25 08:40:02 +0000 | [diff] [blame] | 3260 | if (getCurFunction()->SwitchStack.empty() || !CodeCompleter) | 
| Douglas Gregor | 3e1005f | 2009-09-21 18:10:23 +0000 | [diff] [blame] | 3261 | return; | 
|  | 3262 |  | 
| John McCall | 781472f | 2010-08-25 08:40:02 +0000 | [diff] [blame] | 3263 | SwitchStmt *Switch = getCurFunction()->SwitchStack.back(); | 
| Douglas Gregor | f957843 | 2010-07-28 21:50:18 +0000 | [diff] [blame] | 3264 | if (!Switch->getCond()->getType()->isEnumeralType()) { | 
| Douglas Gregor | fb62941 | 2010-08-23 21:17:50 +0000 | [diff] [blame] | 3265 | CodeCompleteExpressionData Data(Switch->getCond()->getType()); | 
|  | 3266 | Data.IntegralConstantExpression = true; | 
|  | 3267 | CodeCompleteExpression(S, Data); | 
| Douglas Gregor | 3e1005f | 2009-09-21 18:10:23 +0000 | [diff] [blame] | 3268 | return; | 
| Douglas Gregor | f957843 | 2010-07-28 21:50:18 +0000 | [diff] [blame] | 3269 | } | 
| Douglas Gregor | 3e1005f | 2009-09-21 18:10:23 +0000 | [diff] [blame] | 3270 |  | 
|  | 3271 | // Code-complete the cases of a switch statement over an enumeration type | 
|  | 3272 | // by providing the list of | 
|  | 3273 | EnumDecl *Enum = Switch->getCond()->getType()->getAs<EnumType>()->getDecl(); | 
|  | 3274 |  | 
|  | 3275 | // Determine which enumerators we have already seen in the switch statement. | 
|  | 3276 | // FIXME: Ideally, we would also be able to look *past* the code-completion | 
|  | 3277 | // token, in case we are code-completing in the middle of the switch and not | 
|  | 3278 | // at the end. However, we aren't able to do so at the moment. | 
|  | 3279 | llvm::SmallPtrSet<EnumConstantDecl *, 8> EnumeratorsSeen; | 
| Douglas Gregor | b9d0ef7 | 2009-09-21 19:57:38 +0000 | [diff] [blame] | 3280 | NestedNameSpecifier *Qualifier = 0; | 
| Douglas Gregor | 3e1005f | 2009-09-21 18:10:23 +0000 | [diff] [blame] | 3281 | for (SwitchCase *SC = Switch->getSwitchCaseList(); SC; | 
|  | 3282 | SC = SC->getNextSwitchCase()) { | 
|  | 3283 | CaseStmt *Case = dyn_cast<CaseStmt>(SC); | 
|  | 3284 | if (!Case) | 
|  | 3285 | continue; | 
|  | 3286 |  | 
|  | 3287 | Expr *CaseVal = Case->getLHS()->IgnoreParenCasts(); | 
|  | 3288 | if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CaseVal)) | 
|  | 3289 | if (EnumConstantDecl *Enumerator | 
|  | 3290 | = dyn_cast<EnumConstantDecl>(DRE->getDecl())) { | 
|  | 3291 | // We look into the AST of the case statement to determine which | 
|  | 3292 | // enumerator was named. Alternatively, we could compute the value of | 
|  | 3293 | // the integral constant expression, then compare it against the | 
|  | 3294 | // values of each enumerator. However, value-based approach would not | 
|  | 3295 | // work as well with C++ templates where enumerators declared within a | 
|  | 3296 | // template are type- and value-dependent. | 
|  | 3297 | EnumeratorsSeen.insert(Enumerator); | 
|  | 3298 |  | 
| Douglas Gregor | b9d0ef7 | 2009-09-21 19:57:38 +0000 | [diff] [blame] | 3299 | // If this is a qualified-id, keep track of the nested-name-specifier | 
|  | 3300 | // so that we can reproduce it as part of code completion, e.g., | 
| Douglas Gregor | 3e1005f | 2009-09-21 18:10:23 +0000 | [diff] [blame] | 3301 | // | 
|  | 3302 | //   switch (TagD.getKind()) { | 
|  | 3303 | //     case TagDecl::TK_enum: | 
|  | 3304 | //       break; | 
|  | 3305 | //     case XXX | 
|  | 3306 | // | 
| Douglas Gregor | b9d0ef7 | 2009-09-21 19:57:38 +0000 | [diff] [blame] | 3307 | // At the XXX, our completions are TagDecl::TK_union, | 
| Douglas Gregor | 3e1005f | 2009-09-21 18:10:23 +0000 | [diff] [blame] | 3308 | // TagDecl::TK_struct, and TagDecl::TK_class, rather than TK_union, | 
|  | 3309 | // TK_struct, and TK_class. | 
| Douglas Gregor | a2813ce | 2009-10-23 18:54:35 +0000 | [diff] [blame] | 3310 | Qualifier = DRE->getQualifier(); | 
| Douglas Gregor | 3e1005f | 2009-09-21 18:10:23 +0000 | [diff] [blame] | 3311 | } | 
|  | 3312 | } | 
|  | 3313 |  | 
| Douglas Gregor | b9d0ef7 | 2009-09-21 19:57:38 +0000 | [diff] [blame] | 3314 | if (getLangOptions().CPlusPlus && !Qualifier && EnumeratorsSeen.empty()) { | 
|  | 3315 | // If there are no prior enumerators in C++, check whether we have to | 
|  | 3316 | // qualify the names of the enumerators that we suggest, because they | 
|  | 3317 | // may not be visible in this scope. | 
|  | 3318 | Qualifier = getRequiredQualification(Context, CurContext, | 
|  | 3319 | Enum->getDeclContext()); | 
|  | 3320 |  | 
|  | 3321 | // FIXME: Scoped enums need to start with "EnumDecl" as the context! | 
|  | 3322 | } | 
|  | 3323 |  | 
| Douglas Gregor | 3e1005f | 2009-09-21 18:10:23 +0000 | [diff] [blame] | 3324 | // Add any enumerators that have not yet been mentioned. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3325 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 3326 | CodeCompletionContext::CCC_Expression); | 
| Douglas Gregor | 3e1005f | 2009-09-21 18:10:23 +0000 | [diff] [blame] | 3327 | Results.EnterNewScope(); | 
|  | 3328 | for (EnumDecl::enumerator_iterator E = Enum->enumerator_begin(), | 
|  | 3329 | EEnd = Enum->enumerator_end(); | 
|  | 3330 | E != EEnd; ++E) { | 
|  | 3331 | if (EnumeratorsSeen.count(*E)) | 
|  | 3332 | continue; | 
|  | 3333 |  | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 3334 | Results.AddResult(CodeCompletionResult(*E, Qualifier), | 
| Douglas Gregor | 608300b | 2010-01-14 16:14:35 +0000 | [diff] [blame] | 3335 | CurContext, 0, false); | 
| Douglas Gregor | 3e1005f | 2009-09-21 18:10:23 +0000 | [diff] [blame] | 3336 | } | 
|  | 3337 | Results.ExitScope(); | 
| Douglas Gregor | 2f880e4 | 2010-04-06 20:02:15 +0000 | [diff] [blame] | 3338 |  | 
| Douglas Gregor | 0c8296d | 2009-11-07 00:00:49 +0000 | [diff] [blame] | 3339 | if (CodeCompleter->includeMacros()) | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 3340 | AddMacroResults(PP, Results); | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3341 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 3342 | CodeCompletionContext::CCC_Expression, | 
|  | 3343 | Results.data(),Results.size()); | 
| Douglas Gregor | 3e1005f | 2009-09-21 18:10:23 +0000 | [diff] [blame] | 3344 | } | 
|  | 3345 |  | 
| Douglas Gregor | 9c6a0e9 | 2009-09-22 15:41:20 +0000 | [diff] [blame] | 3346 | namespace { | 
|  | 3347 | struct IsBetterOverloadCandidate { | 
|  | 3348 | Sema &S; | 
| John McCall | 5769d61 | 2010-02-08 23:07:23 +0000 | [diff] [blame] | 3349 | SourceLocation Loc; | 
| Douglas Gregor | 9c6a0e9 | 2009-09-22 15:41:20 +0000 | [diff] [blame] | 3350 |  | 
|  | 3351 | public: | 
| John McCall | 5769d61 | 2010-02-08 23:07:23 +0000 | [diff] [blame] | 3352 | explicit IsBetterOverloadCandidate(Sema &S, SourceLocation Loc) | 
|  | 3353 | : S(S), Loc(Loc) { } | 
| Douglas Gregor | 9c6a0e9 | 2009-09-22 15:41:20 +0000 | [diff] [blame] | 3354 |  | 
|  | 3355 | bool | 
|  | 3356 | operator()(const OverloadCandidate &X, const OverloadCandidate &Y) const { | 
| John McCall | 120d63c | 2010-08-24 20:38:10 +0000 | [diff] [blame] | 3357 | return isBetterOverloadCandidate(S, X, Y, Loc); | 
| Douglas Gregor | 9c6a0e9 | 2009-09-22 15:41:20 +0000 | [diff] [blame] | 3358 | } | 
|  | 3359 | }; | 
|  | 3360 | } | 
|  | 3361 |  | 
| Douglas Gregor | d28dcd7 | 2010-05-30 06:10:08 +0000 | [diff] [blame] | 3362 | static bool anyNullArguments(Expr **Args, unsigned NumArgs) { | 
|  | 3363 | if (NumArgs && !Args) | 
|  | 3364 | return true; | 
|  | 3365 |  | 
|  | 3366 | for (unsigned I = 0; I != NumArgs; ++I) | 
|  | 3367 | if (!Args[I]) | 
|  | 3368 | return true; | 
|  | 3369 |  | 
|  | 3370 | return false; | 
|  | 3371 | } | 
|  | 3372 |  | 
| Douglas Gregor | 9c6a0e9 | 2009-09-22 15:41:20 +0000 | [diff] [blame] | 3373 | void Sema::CodeCompleteCall(Scope *S, ExprTy *FnIn, | 
|  | 3374 | ExprTy **ArgsIn, unsigned NumArgs) { | 
|  | 3375 | if (!CodeCompleter) | 
|  | 3376 | return; | 
| Douglas Gregor | ef96eac | 2009-12-11 19:06:04 +0000 | [diff] [blame] | 3377 |  | 
|  | 3378 | // When we're code-completing for a call, we fall back to ordinary | 
|  | 3379 | // name code-completion whenever we can't produce specific | 
|  | 3380 | // results. We may want to revisit this strategy in the future, | 
|  | 3381 | // e.g., by merging the two kinds of results. | 
|  | 3382 |  | 
| Douglas Gregor | 9c6a0e9 | 2009-09-22 15:41:20 +0000 | [diff] [blame] | 3383 | Expr *Fn = (Expr *)FnIn; | 
|  | 3384 | Expr **Args = (Expr **)ArgsIn; | 
| Douglas Gregor | ef96eac | 2009-12-11 19:06:04 +0000 | [diff] [blame] | 3385 |  | 
| Douglas Gregor | 9c6a0e9 | 2009-09-22 15:41:20 +0000 | [diff] [blame] | 3386 | // Ignore type-dependent call expressions entirely. | 
| Douglas Gregor | d28dcd7 | 2010-05-30 06:10:08 +0000 | [diff] [blame] | 3387 | if (!Fn || Fn->isTypeDependent() || anyNullArguments(Args, NumArgs) || | 
| Douglas Gregor | ef96eac | 2009-12-11 19:06:04 +0000 | [diff] [blame] | 3388 | Expr::hasAnyTypeDependentArguments(Args, NumArgs)) { | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3389 | CodeCompleteOrdinaryName(S, PCC_Expression); | 
| Douglas Gregor | 9c6a0e9 | 2009-09-22 15:41:20 +0000 | [diff] [blame] | 3390 | return; | 
| Douglas Gregor | ef96eac | 2009-12-11 19:06:04 +0000 | [diff] [blame] | 3391 | } | 
| Douglas Gregor | 9c6a0e9 | 2009-09-22 15:41:20 +0000 | [diff] [blame] | 3392 |  | 
| John McCall | 3b4294e | 2009-12-16 12:17:52 +0000 | [diff] [blame] | 3393 | // Build an overload candidate set based on the functions we find. | 
| John McCall | 5769d61 | 2010-02-08 23:07:23 +0000 | [diff] [blame] | 3394 | SourceLocation Loc = Fn->getExprLoc(); | 
|  | 3395 | OverloadCandidateSet CandidateSet(Loc); | 
| John McCall | 3b4294e | 2009-12-16 12:17:52 +0000 | [diff] [blame] | 3396 |  | 
| Douglas Gregor | 9c6a0e9 | 2009-09-22 15:41:20 +0000 | [diff] [blame] | 3397 | // FIXME: What if we're calling something that isn't a function declaration? | 
|  | 3398 | // FIXME: What if we're calling a pseudo-destructor? | 
|  | 3399 | // FIXME: What if we're calling a member function? | 
|  | 3400 |  | 
| Douglas Gregor | c026540 | 2010-01-21 15:46:19 +0000 | [diff] [blame] | 3401 | typedef CodeCompleteConsumer::OverloadCandidate ResultCandidate; | 
|  | 3402 | llvm::SmallVector<ResultCandidate, 8> Results; | 
|  | 3403 |  | 
| John McCall | 3b4294e | 2009-12-16 12:17:52 +0000 | [diff] [blame] | 3404 | Expr *NakedFn = Fn->IgnoreParenCasts(); | 
|  | 3405 | if (UnresolvedLookupExpr *ULE = dyn_cast<UnresolvedLookupExpr>(NakedFn)) | 
|  | 3406 | AddOverloadedCallCandidates(ULE, Args, NumArgs, CandidateSet, | 
|  | 3407 | /*PartialOverloading=*/ true); | 
|  | 3408 | else if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(NakedFn)) { | 
|  | 3409 | FunctionDecl *FDecl = dyn_cast<FunctionDecl>(DRE->getDecl()); | 
| Douglas Gregor | c026540 | 2010-01-21 15:46:19 +0000 | [diff] [blame] | 3410 | if (FDecl) { | 
| Douglas Gregor | d28dcd7 | 2010-05-30 06:10:08 +0000 | [diff] [blame] | 3411 | if (!getLangOptions().CPlusPlus || | 
|  | 3412 | !FDecl->getType()->getAs<FunctionProtoType>()) | 
| Douglas Gregor | c026540 | 2010-01-21 15:46:19 +0000 | [diff] [blame] | 3413 | Results.push_back(ResultCandidate(FDecl)); | 
|  | 3414 | else | 
| John McCall | 86820f5 | 2010-01-26 01:37:31 +0000 | [diff] [blame] | 3415 | // FIXME: access? | 
| John McCall | 9aa472c | 2010-03-19 07:35:19 +0000 | [diff] [blame] | 3416 | AddOverloadCandidate(FDecl, DeclAccessPair::make(FDecl, AS_none), | 
|  | 3417 | Args, NumArgs, CandidateSet, | 
| Douglas Gregor | c27d6c5 | 2010-04-16 17:41:49 +0000 | [diff] [blame] | 3418 | false, /*PartialOverloading*/true); | 
| Douglas Gregor | c026540 | 2010-01-21 15:46:19 +0000 | [diff] [blame] | 3419 | } | 
| John McCall | 3b4294e | 2009-12-16 12:17:52 +0000 | [diff] [blame] | 3420 | } | 
| Douglas Gregor | 9c6a0e9 | 2009-09-22 15:41:20 +0000 | [diff] [blame] | 3421 |  | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 3422 | QualType ParamType; | 
|  | 3423 |  | 
| Douglas Gregor | c026540 | 2010-01-21 15:46:19 +0000 | [diff] [blame] | 3424 | if (!CandidateSet.empty()) { | 
|  | 3425 | // Sort the overload candidate set by placing the best overloads first. | 
|  | 3426 | std::stable_sort(CandidateSet.begin(), CandidateSet.end(), | 
| John McCall | 5769d61 | 2010-02-08 23:07:23 +0000 | [diff] [blame] | 3427 | IsBetterOverloadCandidate(*this, Loc)); | 
| Douglas Gregor | 9c6a0e9 | 2009-09-22 15:41:20 +0000 | [diff] [blame] | 3428 |  | 
| Douglas Gregor | c026540 | 2010-01-21 15:46:19 +0000 | [diff] [blame] | 3429 | // Add the remaining viable overload candidates as code-completion reslults. | 
|  | 3430 | for (OverloadCandidateSet::iterator Cand = CandidateSet.begin(), | 
|  | 3431 | CandEnd = CandidateSet.end(); | 
|  | 3432 | Cand != CandEnd; ++Cand) { | 
|  | 3433 | if (Cand->Viable) | 
|  | 3434 | Results.push_back(ResultCandidate(Cand->Function)); | 
|  | 3435 | } | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 3436 |  | 
|  | 3437 | // From the viable candidates, try to determine the type of this parameter. | 
|  | 3438 | for (unsigned I = 0, N = Results.size(); I != N; ++I) { | 
|  | 3439 | if (const FunctionType *FType = Results[I].getFunctionType()) | 
|  | 3440 | if (const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(FType)) | 
|  | 3441 | if (NumArgs < Proto->getNumArgs()) { | 
|  | 3442 | if (ParamType.isNull()) | 
|  | 3443 | ParamType = Proto->getArgType(NumArgs); | 
|  | 3444 | else if (!Context.hasSameUnqualifiedType( | 
|  | 3445 | ParamType.getNonReferenceType(), | 
|  | 3446 | Proto->getArgType(NumArgs).getNonReferenceType())) { | 
|  | 3447 | ParamType = QualType(); | 
|  | 3448 | break; | 
|  | 3449 | } | 
|  | 3450 | } | 
|  | 3451 | } | 
|  | 3452 | } else { | 
|  | 3453 | // Try to determine the parameter type from the type of the expression | 
|  | 3454 | // being called. | 
|  | 3455 | QualType FunctionType = Fn->getType(); | 
|  | 3456 | if (const PointerType *Ptr = FunctionType->getAs<PointerType>()) | 
|  | 3457 | FunctionType = Ptr->getPointeeType(); | 
|  | 3458 | else if (const BlockPointerType *BlockPtr | 
|  | 3459 | = FunctionType->getAs<BlockPointerType>()) | 
|  | 3460 | FunctionType = BlockPtr->getPointeeType(); | 
|  | 3461 | else if (const MemberPointerType *MemPtr | 
|  | 3462 | = FunctionType->getAs<MemberPointerType>()) | 
|  | 3463 | FunctionType = MemPtr->getPointeeType(); | 
|  | 3464 |  | 
|  | 3465 | if (const FunctionProtoType *Proto | 
|  | 3466 | = FunctionType->getAs<FunctionProtoType>()) { | 
|  | 3467 | if (NumArgs < Proto->getNumArgs()) | 
|  | 3468 | ParamType = Proto->getArgType(NumArgs); | 
|  | 3469 | } | 
| Douglas Gregor | 9c6a0e9 | 2009-09-22 15:41:20 +0000 | [diff] [blame] | 3470 | } | 
| Douglas Gregor | ef96eac | 2009-12-11 19:06:04 +0000 | [diff] [blame] | 3471 |  | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 3472 | if (ParamType.isNull()) | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3473 | CodeCompleteOrdinaryName(S, PCC_Expression); | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 3474 | else | 
|  | 3475 | CodeCompleteExpression(S, ParamType); | 
|  | 3476 |  | 
| Douglas Gregor | 2e4c7a5 | 2010-04-06 20:19:47 +0000 | [diff] [blame] | 3477 | if (!Results.empty()) | 
| Douglas Gregor | ef96eac | 2009-12-11 19:06:04 +0000 | [diff] [blame] | 3478 | CodeCompleter->ProcessOverloadCandidates(*this, NumArgs, Results.data(), | 
|  | 3479 | Results.size()); | 
| Douglas Gregor | 9c6a0e9 | 2009-09-22 15:41:20 +0000 | [diff] [blame] | 3480 | } | 
|  | 3481 |  | 
| John McCall | d226f65 | 2010-08-21 09:40:31 +0000 | [diff] [blame] | 3482 | void Sema::CodeCompleteInitializer(Scope *S, Decl *D) { | 
|  | 3483 | ValueDecl *VD = dyn_cast_or_null<ValueDecl>(D); | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 3484 | if (!VD) { | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3485 | CodeCompleteOrdinaryName(S, PCC_Expression); | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 3486 | return; | 
|  | 3487 | } | 
|  | 3488 |  | 
|  | 3489 | CodeCompleteExpression(S, VD->getType()); | 
|  | 3490 | } | 
|  | 3491 |  | 
|  | 3492 | void Sema::CodeCompleteReturn(Scope *S) { | 
|  | 3493 | QualType ResultType; | 
|  | 3494 | if (isa<BlockDecl>(CurContext)) { | 
|  | 3495 | if (BlockScopeInfo *BSI = getCurBlock()) | 
|  | 3496 | ResultType = BSI->ReturnType; | 
|  | 3497 | } else if (FunctionDecl *Function = dyn_cast<FunctionDecl>(CurContext)) | 
|  | 3498 | ResultType = Function->getResultType(); | 
|  | 3499 | else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(CurContext)) | 
|  | 3500 | ResultType = Method->getResultType(); | 
|  | 3501 |  | 
|  | 3502 | if (ResultType.isNull()) | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3503 | CodeCompleteOrdinaryName(S, PCC_Expression); | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 3504 | else | 
|  | 3505 | CodeCompleteExpression(S, ResultType); | 
|  | 3506 | } | 
|  | 3507 |  | 
|  | 3508 | void Sema::CodeCompleteAssignmentRHS(Scope *S, ExprTy *LHS) { | 
|  | 3509 | if (LHS) | 
|  | 3510 | CodeCompleteExpression(S, static_cast<Expr *>(LHS)->getType()); | 
|  | 3511 | else | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3512 | CodeCompleteOrdinaryName(S, PCC_Expression); | 
| Douglas Gregor | 5ac3bdb | 2010-05-30 01:49:25 +0000 | [diff] [blame] | 3513 | } | 
|  | 3514 |  | 
| Jeffrey Yasskin | 9ab1454 | 2010-04-08 16:38:48 +0000 | [diff] [blame] | 3515 | void Sema::CodeCompleteQualifiedId(Scope *S, CXXScopeSpec &SS, | 
| Douglas Gregor | 81b747b | 2009-09-17 21:32:03 +0000 | [diff] [blame] | 3516 | bool EnteringContext) { | 
|  | 3517 | if (!SS.getScopeRep() || !CodeCompleter) | 
|  | 3518 | return; | 
|  | 3519 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3520 | DeclContext *Ctx = computeDeclContext(SS, EnteringContext); | 
|  | 3521 | if (!Ctx) | 
|  | 3522 | return; | 
| Douglas Gregor | d1cd31a | 2009-12-11 18:28:39 +0000 | [diff] [blame] | 3523 |  | 
|  | 3524 | // Try to instantiate any non-dependent declaration contexts before | 
|  | 3525 | // we look in them. | 
| John McCall | 77bb1aa | 2010-05-01 00:40:08 +0000 | [diff] [blame] | 3526 | if (!isDependentScopeSpecifier(SS) && RequireCompleteDeclContext(SS, Ctx)) | 
| Douglas Gregor | d1cd31a | 2009-12-11 18:28:39 +0000 | [diff] [blame] | 3527 | return; | 
|  | 3528 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3529 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 3530 | CodeCompletionContext::CCC_Name); | 
| Douglas Gregor | f696152 | 2010-08-27 21:18:54 +0000 | [diff] [blame] | 3531 | Results.EnterNewScope(); | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3532 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3533 | // The "template" keyword can follow "::" in the grammar, but only | 
|  | 3534 | // put it into the grammar if the nested-name-specifier is dependent. | 
|  | 3535 | NestedNameSpecifier *NNS = (NestedNameSpecifier *)SS.getScopeRep(); | 
|  | 3536 | if (!Results.empty() && NNS->isDependent()) | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 3537 | Results.AddResult("template"); | 
| Douglas Gregor | f696152 | 2010-08-27 21:18:54 +0000 | [diff] [blame] | 3538 |  | 
|  | 3539 | // Add calls to overridden virtual functions, if there are any. | 
|  | 3540 | // | 
|  | 3541 | // FIXME: This isn't wonderful, because we don't know whether we're actually | 
|  | 3542 | // in a context that permits expressions. This is a general issue with | 
|  | 3543 | // qualified-id completions. | 
|  | 3544 | if (!EnteringContext) | 
|  | 3545 | MaybeAddOverrideCalls(*this, Ctx, Results); | 
|  | 3546 | Results.ExitScope(); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3547 |  | 
| Douglas Gregor | f696152 | 2010-08-27 21:18:54 +0000 | [diff] [blame] | 3548 | CodeCompletionDeclConsumer Consumer(Results, CurContext); | 
|  | 3549 | LookupVisibleDecls(Ctx, LookupOrdinaryName, Consumer); | 
|  | 3550 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3551 | HandleCodeCompleteResults(this, CodeCompleter, | 
| Douglas Gregor | f696152 | 2010-08-27 21:18:54 +0000 | [diff] [blame] | 3552 | CodeCompletionContext::CCC_Name, | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3553 | Results.data(),Results.size()); | 
| Douglas Gregor | 81b747b | 2009-09-17 21:32:03 +0000 | [diff] [blame] | 3554 | } | 
| Douglas Gregor | 49f40bd | 2009-09-18 19:03:04 +0000 | [diff] [blame] | 3555 |  | 
|  | 3556 | void Sema::CodeCompleteUsing(Scope *S) { | 
|  | 3557 | if (!CodeCompleter) | 
|  | 3558 | return; | 
|  | 3559 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3560 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3561 | CodeCompletionContext::CCC_PotentiallyQualifiedName, | 
|  | 3562 | &ResultBuilder::IsNestedNameSpecifier); | 
| Douglas Gregor | 8e0a0e4 | 2009-09-22 23:31:26 +0000 | [diff] [blame] | 3563 | Results.EnterNewScope(); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3564 |  | 
|  | 3565 | // If we aren't in class scope, we could see the "namespace" keyword. | 
|  | 3566 | if (!S->isClassScope()) | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 3567 | Results.AddResult(CodeCompletionResult("namespace")); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3568 |  | 
|  | 3569 | // After "using", we can see anything that would start a | 
|  | 3570 | // nested-name-specifier. | 
| Douglas Gregor | 5d2fc40 | 2010-01-14 03:27:13 +0000 | [diff] [blame] | 3571 | CodeCompletionDeclConsumer Consumer(Results, CurContext); | 
| Douglas Gregor | 8071e42 | 2010-08-15 06:18:01 +0000 | [diff] [blame] | 3572 | LookupVisibleDecls(S, LookupOrdinaryName, Consumer, | 
|  | 3573 | CodeCompleter->includeGlobals()); | 
| Douglas Gregor | 8e0a0e4 | 2009-09-22 23:31:26 +0000 | [diff] [blame] | 3574 | Results.ExitScope(); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3575 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3576 | HandleCodeCompleteResults(this, CodeCompleter, | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3577 | CodeCompletionContext::CCC_PotentiallyQualifiedName, | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3578 | Results.data(),Results.size()); | 
| Douglas Gregor | 49f40bd | 2009-09-18 19:03:04 +0000 | [diff] [blame] | 3579 | } | 
|  | 3580 |  | 
|  | 3581 | void Sema::CodeCompleteUsingDirective(Scope *S) { | 
|  | 3582 | if (!CodeCompleter) | 
|  | 3583 | return; | 
|  | 3584 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3585 | // After "using namespace", we expect to see a namespace name or namespace | 
|  | 3586 | // alias. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3587 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 3588 | CodeCompletionContext::CCC_Namespace, | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3589 | &ResultBuilder::IsNamespaceOrAlias); | 
| Douglas Gregor | 8e0a0e4 | 2009-09-22 23:31:26 +0000 | [diff] [blame] | 3590 | Results.EnterNewScope(); | 
| Douglas Gregor | 5d2fc40 | 2010-01-14 03:27:13 +0000 | [diff] [blame] | 3591 | CodeCompletionDeclConsumer Consumer(Results, CurContext); | 
| Douglas Gregor | 8071e42 | 2010-08-15 06:18:01 +0000 | [diff] [blame] | 3592 | LookupVisibleDecls(S, LookupOrdinaryName, Consumer, | 
|  | 3593 | CodeCompleter->includeGlobals()); | 
| Douglas Gregor | 8e0a0e4 | 2009-09-22 23:31:26 +0000 | [diff] [blame] | 3594 | Results.ExitScope(); | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3595 | HandleCodeCompleteResults(this, CodeCompleter, | 
| Douglas Gregor | 8071e42 | 2010-08-15 06:18:01 +0000 | [diff] [blame] | 3596 | CodeCompletionContext::CCC_Namespace, | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3597 | Results.data(),Results.size()); | 
| Douglas Gregor | 49f40bd | 2009-09-18 19:03:04 +0000 | [diff] [blame] | 3598 | } | 
|  | 3599 |  | 
|  | 3600 | void Sema::CodeCompleteNamespaceDecl(Scope *S)  { | 
|  | 3601 | if (!CodeCompleter) | 
|  | 3602 | return; | 
|  | 3603 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3604 | DeclContext *Ctx = (DeclContext *)S->getEntity(); | 
|  | 3605 | if (!S->getParent()) | 
|  | 3606 | Ctx = Context.getTranslationUnitDecl(); | 
|  | 3607 |  | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3608 | bool SuppressedGlobalResults | 
|  | 3609 | = Ctx && !CodeCompleter->includeGlobals() && isa<TranslationUnitDecl>(Ctx); | 
|  | 3610 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3611 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3612 | SuppressedGlobalResults | 
|  | 3613 | ? CodeCompletionContext::CCC_Namespace | 
|  | 3614 | : CodeCompletionContext::CCC_Other, | 
|  | 3615 | &ResultBuilder::IsNamespace); | 
|  | 3616 |  | 
|  | 3617 | if (Ctx && Ctx->isFileContext() && !SuppressedGlobalResults) { | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3618 | // We only want to see those namespaces that have already been defined | 
|  | 3619 | // within this scope, because its likely that the user is creating an | 
|  | 3620 | // extended namespace declaration. Keep track of the most recent | 
|  | 3621 | // definition of each namespace. | 
|  | 3622 | std::map<NamespaceDecl *, NamespaceDecl *> OrigToLatest; | 
|  | 3623 | for (DeclContext::specific_decl_iterator<NamespaceDecl> | 
|  | 3624 | NS(Ctx->decls_begin()), NSEnd(Ctx->decls_end()); | 
|  | 3625 | NS != NSEnd; ++NS) | 
|  | 3626 | OrigToLatest[NS->getOriginalNamespace()] = *NS; | 
|  | 3627 |  | 
|  | 3628 | // Add the most recent definition (or extended definition) of each | 
|  | 3629 | // namespace to the list of results. | 
| Douglas Gregor | 8e0a0e4 | 2009-09-22 23:31:26 +0000 | [diff] [blame] | 3630 | Results.EnterNewScope(); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3631 | for (std::map<NamespaceDecl *, NamespaceDecl *>::iterator | 
|  | 3632 | NS = OrigToLatest.begin(), NSEnd = OrigToLatest.end(); | 
|  | 3633 | NS != NSEnd; ++NS) | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 3634 | Results.AddResult(CodeCompletionResult(NS->second, 0), | 
| Douglas Gregor | 608300b | 2010-01-14 16:14:35 +0000 | [diff] [blame] | 3635 | CurContext, 0, false); | 
| Douglas Gregor | 8e0a0e4 | 2009-09-22 23:31:26 +0000 | [diff] [blame] | 3636 | Results.ExitScope(); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3637 | } | 
|  | 3638 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3639 | HandleCodeCompleteResults(this, CodeCompleter, | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3640 | Results.getCompletionContext(), | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3641 | Results.data(),Results.size()); | 
| Douglas Gregor | 49f40bd | 2009-09-18 19:03:04 +0000 | [diff] [blame] | 3642 | } | 
|  | 3643 |  | 
|  | 3644 | void Sema::CodeCompleteNamespaceAliasDecl(Scope *S)  { | 
|  | 3645 | if (!CodeCompleter) | 
|  | 3646 | return; | 
|  | 3647 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3648 | // After "namespace", we expect to see a namespace or alias. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3649 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 3650 | CodeCompletionContext::CCC_Namespace, | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3651 | &ResultBuilder::IsNamespaceOrAlias); | 
| Douglas Gregor | 5d2fc40 | 2010-01-14 03:27:13 +0000 | [diff] [blame] | 3652 | CodeCompletionDeclConsumer Consumer(Results, CurContext); | 
| Douglas Gregor | 8071e42 | 2010-08-15 06:18:01 +0000 | [diff] [blame] | 3653 | LookupVisibleDecls(S, LookupOrdinaryName, Consumer, | 
|  | 3654 | CodeCompleter->includeGlobals()); | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3655 | HandleCodeCompleteResults(this, CodeCompleter, | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3656 | Results.getCompletionContext(), | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3657 | Results.data(),Results.size()); | 
| Douglas Gregor | 49f40bd | 2009-09-18 19:03:04 +0000 | [diff] [blame] | 3658 | } | 
|  | 3659 |  | 
| Douglas Gregor | ed8d322 | 2009-09-18 20:05:18 +0000 | [diff] [blame] | 3660 | void Sema::CodeCompleteOperatorName(Scope *S) { | 
|  | 3661 | if (!CodeCompleter) | 
|  | 3662 | return; | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3663 |  | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 3664 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3665 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 3666 | CodeCompletionContext::CCC_Type, | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3667 | &ResultBuilder::IsType); | 
| Douglas Gregor | 8e0a0e4 | 2009-09-22 23:31:26 +0000 | [diff] [blame] | 3668 | Results.EnterNewScope(); | 
| Douglas Gregor | ed8d322 | 2009-09-18 20:05:18 +0000 | [diff] [blame] | 3669 |  | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3670 | // Add the names of overloadable operators. | 
|  | 3671 | #define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly)      \ | 
|  | 3672 | if (std::strcmp(Spelling, "?"))                                                  \ | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 3673 | Results.AddResult(Result(Spelling)); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3674 | #include "clang/Basic/OperatorKinds.def" | 
|  | 3675 |  | 
|  | 3676 | // Add any type names visible from the current scope | 
| Douglas Gregor | 45bcd43 | 2010-01-14 03:21:49 +0000 | [diff] [blame] | 3677 | Results.allowNestedNameSpecifiers(); | 
| Douglas Gregor | 5d2fc40 | 2010-01-14 03:27:13 +0000 | [diff] [blame] | 3678 | CodeCompletionDeclConsumer Consumer(Results, CurContext); | 
| Douglas Gregor | 8071e42 | 2010-08-15 06:18:01 +0000 | [diff] [blame] | 3679 | LookupVisibleDecls(S, LookupOrdinaryName, Consumer, | 
|  | 3680 | CodeCompleter->includeGlobals()); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3681 |  | 
|  | 3682 | // Add any type specifiers | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 3683 | AddTypeSpecifierResults(getLangOptions(), Results); | 
| Douglas Gregor | 8e0a0e4 | 2009-09-22 23:31:26 +0000 | [diff] [blame] | 3684 | Results.ExitScope(); | 
| Douglas Gregor | 86d9a52 | 2009-09-21 16:56:56 +0000 | [diff] [blame] | 3685 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3686 | HandleCodeCompleteResults(this, CodeCompleter, | 
| Douglas Gregor | 8071e42 | 2010-08-15 06:18:01 +0000 | [diff] [blame] | 3687 | CodeCompletionContext::CCC_Type, | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3688 | Results.data(),Results.size()); | 
| Douglas Gregor | ed8d322 | 2009-09-18 20:05:18 +0000 | [diff] [blame] | 3689 | } | 
| Douglas Gregor | 49f40bd | 2009-09-18 19:03:04 +0000 | [diff] [blame] | 3690 |  | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3691 | void Sema::CodeCompleteConstructorInitializer(Decl *ConstructorD, | 
| Sean Hunt | cbb6748 | 2011-01-08 20:30:50 +0000 | [diff] [blame] | 3692 | CXXCtorInitializer** Initializers, | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3693 | unsigned NumInitializers) { | 
|  | 3694 | CXXConstructorDecl *Constructor | 
|  | 3695 | = static_cast<CXXConstructorDecl *>(ConstructorD); | 
|  | 3696 | if (!Constructor) | 
|  | 3697 | return; | 
|  | 3698 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3699 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3700 | CodeCompletionContext::CCC_PotentiallyQualifiedName); | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3701 | Results.EnterNewScope(); | 
|  | 3702 |  | 
|  | 3703 | // Fill in any already-initialized fields or base classes. | 
|  | 3704 | llvm::SmallPtrSet<FieldDecl *, 4> InitializedFields; | 
|  | 3705 | llvm::SmallPtrSet<CanQualType, 4> InitializedBases; | 
|  | 3706 | for (unsigned I = 0; I != NumInitializers; ++I) { | 
|  | 3707 | if (Initializers[I]->isBaseInitializer()) | 
|  | 3708 | InitializedBases.insert( | 
|  | 3709 | Context.getCanonicalType(QualType(Initializers[I]->getBaseClass(), 0))); | 
|  | 3710 | else | 
| Francois Pichet | 00eb3f9 | 2010-12-04 09:14:42 +0000 | [diff] [blame] | 3711 | InitializedFields.insert(cast<FieldDecl>( | 
|  | 3712 | Initializers[I]->getAnyMember())); | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3713 | } | 
|  | 3714 |  | 
|  | 3715 | // Add completions for base classes. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3716 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
| Douglas Gregor | 0c431c8 | 2010-08-29 19:27:27 +0000 | [diff] [blame] | 3717 | bool SawLastInitializer = (NumInitializers == 0); | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3718 | CXXRecordDecl *ClassDecl = Constructor->getParent(); | 
|  | 3719 | for (CXXRecordDecl::base_class_iterator Base = ClassDecl->bases_begin(), | 
|  | 3720 | BaseEnd = ClassDecl->bases_end(); | 
|  | 3721 | Base != BaseEnd; ++Base) { | 
| Douglas Gregor | 0c431c8 | 2010-08-29 19:27:27 +0000 | [diff] [blame] | 3722 | if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) { | 
|  | 3723 | SawLastInitializer | 
|  | 3724 | = NumInitializers > 0 && | 
|  | 3725 | Initializers[NumInitializers - 1]->isBaseInitializer() && | 
|  | 3726 | Context.hasSameUnqualifiedType(Base->getType(), | 
|  | 3727 | QualType(Initializers[NumInitializers - 1]->getBaseClass(), 0)); | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3728 | continue; | 
| Douglas Gregor | 0c431c8 | 2010-08-29 19:27:27 +0000 | [diff] [blame] | 3729 | } | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3730 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3731 | Builder.AddTypedTextChunk( | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 3732 | Results.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3733 | Base->getType().getAsString(Context.PrintingPolicy))); | 
|  | 3734 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 3735 | Builder.AddPlaceholderChunk("args"); | 
|  | 3736 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 3737 | Results.AddResult(CodeCompletionResult(Builder.TakeString(), | 
| Douglas Gregor | 0c431c8 | 2010-08-29 19:27:27 +0000 | [diff] [blame] | 3738 | SawLastInitializer? CCP_NextInitializer | 
|  | 3739 | : CCP_MemberDeclaration)); | 
|  | 3740 | SawLastInitializer = false; | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3741 | } | 
|  | 3742 |  | 
|  | 3743 | // Add completions for virtual base classes. | 
|  | 3744 | for (CXXRecordDecl::base_class_iterator Base = ClassDecl->vbases_begin(), | 
|  | 3745 | BaseEnd = ClassDecl->vbases_end(); | 
|  | 3746 | Base != BaseEnd; ++Base) { | 
| Douglas Gregor | 0c431c8 | 2010-08-29 19:27:27 +0000 | [diff] [blame] | 3747 | if (!InitializedBases.insert(Context.getCanonicalType(Base->getType()))) { | 
|  | 3748 | SawLastInitializer | 
|  | 3749 | = NumInitializers > 0 && | 
|  | 3750 | Initializers[NumInitializers - 1]->isBaseInitializer() && | 
|  | 3751 | Context.hasSameUnqualifiedType(Base->getType(), | 
|  | 3752 | QualType(Initializers[NumInitializers - 1]->getBaseClass(), 0)); | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3753 | continue; | 
| Douglas Gregor | 0c431c8 | 2010-08-29 19:27:27 +0000 | [diff] [blame] | 3754 | } | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3755 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3756 | Builder.AddTypedTextChunk( | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 3757 | Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3758 | Base->getType().getAsString(Context.PrintingPolicy))); | 
|  | 3759 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 3760 | Builder.AddPlaceholderChunk("args"); | 
|  | 3761 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 3762 | Results.AddResult(CodeCompletionResult(Builder.TakeString(), | 
| Douglas Gregor | 0c431c8 | 2010-08-29 19:27:27 +0000 | [diff] [blame] | 3763 | SawLastInitializer? CCP_NextInitializer | 
|  | 3764 | : CCP_MemberDeclaration)); | 
|  | 3765 | SawLastInitializer = false; | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3766 | } | 
|  | 3767 |  | 
|  | 3768 | // Add completions for members. | 
|  | 3769 | for (CXXRecordDecl::field_iterator Field = ClassDecl->field_begin(), | 
|  | 3770 | FieldEnd = ClassDecl->field_end(); | 
|  | 3771 | Field != FieldEnd; ++Field) { | 
| Douglas Gregor | 0c431c8 | 2010-08-29 19:27:27 +0000 | [diff] [blame] | 3772 | if (!InitializedFields.insert(cast<FieldDecl>(Field->getCanonicalDecl()))) { | 
|  | 3773 | SawLastInitializer | 
|  | 3774 | = NumInitializers > 0 && | 
| Francois Pichet | 00eb3f9 | 2010-12-04 09:14:42 +0000 | [diff] [blame] | 3775 | Initializers[NumInitializers - 1]->isAnyMemberInitializer() && | 
|  | 3776 | Initializers[NumInitializers - 1]->getAnyMember() == *Field; | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3777 | continue; | 
| Douglas Gregor | 0c431c8 | 2010-08-29 19:27:27 +0000 | [diff] [blame] | 3778 | } | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3779 |  | 
|  | 3780 | if (!Field->getDeclName()) | 
|  | 3781 | continue; | 
|  | 3782 |  | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 3783 | Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3784 | Field->getIdentifier()->getName())); | 
|  | 3785 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 3786 | Builder.AddPlaceholderChunk("args"); | 
|  | 3787 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 3788 | Results.AddResult(CodeCompletionResult(Builder.TakeString(), | 
| Douglas Gregor | 0c431c8 | 2010-08-29 19:27:27 +0000 | [diff] [blame] | 3789 | SawLastInitializer? CCP_NextInitializer | 
| Douglas Gregor | a67e03f | 2010-09-09 21:42:20 +0000 | [diff] [blame] | 3790 | : CCP_MemberDeclaration, | 
|  | 3791 | CXCursor_MemberRef)); | 
| Douglas Gregor | 0c431c8 | 2010-08-29 19:27:27 +0000 | [diff] [blame] | 3792 | SawLastInitializer = false; | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3793 | } | 
|  | 3794 | Results.ExitScope(); | 
|  | 3795 |  | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 3796 | HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(), | 
| Douglas Gregor | 0133f52 | 2010-08-28 00:00:50 +0000 | [diff] [blame] | 3797 | Results.data(), Results.size()); | 
|  | 3798 | } | 
|  | 3799 |  | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3800 | // Macro that expands to @Keyword or Keyword, depending on whether NeedAt is | 
|  | 3801 | // true or false. | 
|  | 3802 | #define OBJC_AT_KEYWORD_NAME(NeedAt,Keyword) NeedAt? "@" #Keyword : #Keyword | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 3803 | static void AddObjCImplementationResults(const LangOptions &LangOpts, | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3804 | ResultBuilder &Results, | 
|  | 3805 | bool NeedAt) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 3806 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3807 | // Since we have an implementation, we can end it. | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 3808 | Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,end))); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3809 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3810 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3811 | if (LangOpts.ObjC2) { | 
|  | 3812 | // @dynamic | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3813 | Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,dynamic)); | 
|  | 3814 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 3815 | Builder.AddPlaceholderChunk("property"); | 
|  | 3816 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3817 |  | 
|  | 3818 | // @synthesize | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3819 | Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,synthesize)); | 
|  | 3820 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 3821 | Builder.AddPlaceholderChunk("property"); | 
|  | 3822 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3823 | } | 
|  | 3824 | } | 
|  | 3825 |  | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 3826 | static void AddObjCInterfaceResults(const LangOptions &LangOpts, | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3827 | ResultBuilder &Results, | 
|  | 3828 | bool NeedAt) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 3829 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3830 |  | 
|  | 3831 | // Since we have an interface or protocol, we can end it. | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 3832 | Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,end))); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3833 |  | 
|  | 3834 | if (LangOpts.ObjC2) { | 
|  | 3835 | // @property | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 3836 | Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,property))); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3837 |  | 
|  | 3838 | // @required | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 3839 | Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,required))); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3840 |  | 
|  | 3841 | // @optional | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 3842 | Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,optional))); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3843 | } | 
|  | 3844 | } | 
|  | 3845 |  | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 3846 | static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 3847 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3848 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3849 |  | 
|  | 3850 | // @class name ; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3851 | Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,class)); | 
|  | 3852 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 3853 | Builder.AddPlaceholderChunk("name"); | 
|  | 3854 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3855 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 3856 | if (Results.includeCodePatterns()) { | 
|  | 3857 | // @interface name | 
|  | 3858 | // FIXME: Could introduce the whole pattern, including superclasses and | 
|  | 3859 | // such. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3860 | Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,interface)); | 
|  | 3861 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 3862 | Builder.AddPlaceholderChunk("class"); | 
|  | 3863 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3864 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 3865 | // @protocol name | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3866 | Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,protocol)); | 
|  | 3867 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 3868 | Builder.AddPlaceholderChunk("protocol"); | 
|  | 3869 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 3870 |  | 
|  | 3871 | // @implementation name | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3872 | Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,implementation)); | 
|  | 3873 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 3874 | Builder.AddPlaceholderChunk("class"); | 
|  | 3875 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 3876 | } | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3877 |  | 
|  | 3878 | // @compatibility_alias name | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3879 | Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,compatibility_alias)); | 
|  | 3880 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 3881 | Builder.AddPlaceholderChunk("alias"); | 
|  | 3882 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 3883 | Builder.AddPlaceholderChunk("class"); | 
|  | 3884 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3885 | } | 
|  | 3886 |  | 
| John McCall | d226f65 | 2010-08-21 09:40:31 +0000 | [diff] [blame] | 3887 | void Sema::CodeCompleteObjCAtDirective(Scope *S, Decl *ObjCImpDecl, | 
| Douglas Gregor | c464ae8 | 2009-12-07 09:27:33 +0000 | [diff] [blame] | 3888 | bool InInterface) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 3889 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3890 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 3891 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | c464ae8 | 2009-12-07 09:27:33 +0000 | [diff] [blame] | 3892 | Results.EnterNewScope(); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3893 | if (ObjCImpDecl) | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 3894 | AddObjCImplementationResults(getLangOptions(), Results, false); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3895 | else if (InInterface) | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 3896 | AddObjCInterfaceResults(getLangOptions(), Results, false); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3897 | else | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 3898 | AddObjCTopLevelResults(Results, false); | 
| Douglas Gregor | c464ae8 | 2009-12-07 09:27:33 +0000 | [diff] [blame] | 3899 | Results.ExitScope(); | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3900 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 3901 | CodeCompletionContext::CCC_Other, | 
|  | 3902 | Results.data(),Results.size()); | 
| Douglas Gregor | c464ae8 | 2009-12-07 09:27:33 +0000 | [diff] [blame] | 3903 | } | 
|  | 3904 |  | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 3905 | static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 3906 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3907 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
| Douglas Gregor | 9a0c85e | 2009-12-07 09:51:25 +0000 | [diff] [blame] | 3908 |  | 
|  | 3909 | // @encode ( type-name ) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3910 | Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,encode)); | 
|  | 3911 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 3912 | Builder.AddPlaceholderChunk("type-name"); | 
|  | 3913 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 3914 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 9a0c85e | 2009-12-07 09:51:25 +0000 | [diff] [blame] | 3915 |  | 
|  | 3916 | // @protocol ( protocol-name ) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3917 | Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,protocol)); | 
|  | 3918 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 3919 | Builder.AddPlaceholderChunk("protocol-name"); | 
|  | 3920 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 3921 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 9a0c85e | 2009-12-07 09:51:25 +0000 | [diff] [blame] | 3922 |  | 
|  | 3923 | // @selector ( selector ) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3924 | Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,selector)); | 
|  | 3925 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 3926 | Builder.AddPlaceholderChunk("selector"); | 
|  | 3927 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 3928 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | 9a0c85e | 2009-12-07 09:51:25 +0000 | [diff] [blame] | 3929 | } | 
|  | 3930 |  | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 3931 | static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 3932 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3933 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3934 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 3935 | if (Results.includeCodePatterns()) { | 
|  | 3936 | // @try { statements } @catch ( declaration ) { statements } @finally | 
|  | 3937 | //   { statements } | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3938 | Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,try)); | 
|  | 3939 | Builder.AddChunk(CodeCompletionString::CK_LeftBrace); | 
|  | 3940 | Builder.AddPlaceholderChunk("statements"); | 
|  | 3941 | Builder.AddChunk(CodeCompletionString::CK_RightBrace); | 
|  | 3942 | Builder.AddTextChunk("@catch"); | 
|  | 3943 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 3944 | Builder.AddPlaceholderChunk("parameter"); | 
|  | 3945 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 3946 | Builder.AddChunk(CodeCompletionString::CK_LeftBrace); | 
|  | 3947 | Builder.AddPlaceholderChunk("statements"); | 
|  | 3948 | Builder.AddChunk(CodeCompletionString::CK_RightBrace); | 
|  | 3949 | Builder.AddTextChunk("@finally"); | 
|  | 3950 | Builder.AddChunk(CodeCompletionString::CK_LeftBrace); | 
|  | 3951 | Builder.AddPlaceholderChunk("statements"); | 
|  | 3952 | Builder.AddChunk(CodeCompletionString::CK_RightBrace); | 
|  | 3953 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 3954 | } | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3955 |  | 
| Douglas Gregor | 9a0c85e | 2009-12-07 09:51:25 +0000 | [diff] [blame] | 3956 | // @throw | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3957 | Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,throw)); | 
|  | 3958 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 3959 | Builder.AddPlaceholderChunk("expression"); | 
|  | 3960 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3961 |  | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 3962 | if (Results.includeCodePatterns()) { | 
|  | 3963 | // @synchronized ( expression ) { statements } | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3964 | Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,synchronized)); | 
|  | 3965 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 3966 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 3967 | Builder.AddPlaceholderChunk("expression"); | 
|  | 3968 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 3969 | Builder.AddChunk(CodeCompletionString::CK_LeftBrace); | 
|  | 3970 | Builder.AddPlaceholderChunk("statements"); | 
|  | 3971 | Builder.AddChunk(CodeCompletionString::CK_RightBrace); | 
|  | 3972 | Results.AddResult(Result(Builder.TakeString())); | 
| Douglas Gregor | c8bddde | 2010-05-28 00:22:41 +0000 | [diff] [blame] | 3973 | } | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 3974 | } | 
| Douglas Gregor | 9a0c85e | 2009-12-07 09:51:25 +0000 | [diff] [blame] | 3975 |  | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 3976 | static void AddObjCVisibilityResults(const LangOptions &LangOpts, | 
| Douglas Gregor | c38c3e1 | 2010-01-13 21:54:15 +0000 | [diff] [blame] | 3977 | ResultBuilder &Results, | 
|  | 3978 | bool NeedAt) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 3979 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 3980 | Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,private))); | 
|  | 3981 | Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,protected))); | 
|  | 3982 | Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,public))); | 
| Douglas Gregor | c38c3e1 | 2010-01-13 21:54:15 +0000 | [diff] [blame] | 3983 | if (LangOpts.ObjC2) | 
| Douglas Gregor | a447781 | 2010-01-14 16:01:26 +0000 | [diff] [blame] | 3984 | Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,package))); | 
| Douglas Gregor | c38c3e1 | 2010-01-13 21:54:15 +0000 | [diff] [blame] | 3985 | } | 
|  | 3986 |  | 
|  | 3987 | void Sema::CodeCompleteObjCAtVisibility(Scope *S) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3988 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 3989 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | c38c3e1 | 2010-01-13 21:54:15 +0000 | [diff] [blame] | 3990 | Results.EnterNewScope(); | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 3991 | AddObjCVisibilityResults(getLangOptions(), Results, false); | 
| Douglas Gregor | c38c3e1 | 2010-01-13 21:54:15 +0000 | [diff] [blame] | 3992 | Results.ExitScope(); | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 3993 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 3994 | CodeCompletionContext::CCC_Other, | 
|  | 3995 | Results.data(),Results.size()); | 
| Douglas Gregor | c38c3e1 | 2010-01-13 21:54:15 +0000 | [diff] [blame] | 3996 | } | 
|  | 3997 |  | 
|  | 3998 | void Sema::CodeCompleteObjCAtStatement(Scope *S) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 3999 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 4000 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | b6ac245 | 2010-01-13 21:24:21 +0000 | [diff] [blame] | 4001 | Results.EnterNewScope(); | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 4002 | AddObjCStatementResults(Results, false); | 
|  | 4003 | AddObjCExpressionResults(Results, false); | 
| Douglas Gregor | 9a0c85e | 2009-12-07 09:51:25 +0000 | [diff] [blame] | 4004 | Results.ExitScope(); | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 4005 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 4006 | CodeCompletionContext::CCC_Other, | 
|  | 4007 | Results.data(),Results.size()); | 
| Douglas Gregor | 9a0c85e | 2009-12-07 09:51:25 +0000 | [diff] [blame] | 4008 | } | 
|  | 4009 |  | 
|  | 4010 | void Sema::CodeCompleteObjCAtExpression(Scope *S) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4011 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 4012 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | 9a0c85e | 2009-12-07 09:51:25 +0000 | [diff] [blame] | 4013 | Results.EnterNewScope(); | 
| Douglas Gregor | bca403c | 2010-01-13 23:51:12 +0000 | [diff] [blame] | 4014 | AddObjCExpressionResults(Results, false); | 
| Douglas Gregor | 9a0c85e | 2009-12-07 09:51:25 +0000 | [diff] [blame] | 4015 | Results.ExitScope(); | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 4016 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 4017 | CodeCompletionContext::CCC_Other, | 
|  | 4018 | Results.data(),Results.size()); | 
| Douglas Gregor | 9a0c85e | 2009-12-07 09:51:25 +0000 | [diff] [blame] | 4019 | } | 
|  | 4020 |  | 
| Douglas Gregor | 988358f | 2009-11-19 00:14:45 +0000 | [diff] [blame] | 4021 | /// \brief Determine whether the addition of the given flag to an Objective-C | 
|  | 4022 | /// property's attributes will cause a conflict. | 
|  | 4023 | static bool ObjCPropertyFlagConflicts(unsigned Attributes, unsigned NewFlag) { | 
|  | 4024 | // Check if we've already added this flag. | 
|  | 4025 | if (Attributes & NewFlag) | 
|  | 4026 | return true; | 
|  | 4027 |  | 
|  | 4028 | Attributes |= NewFlag; | 
|  | 4029 |  | 
|  | 4030 | // Check for collisions with "readonly". | 
|  | 4031 | if ((Attributes & ObjCDeclSpec::DQ_PR_readonly) && | 
|  | 4032 | (Attributes & (ObjCDeclSpec::DQ_PR_readwrite | | 
|  | 4033 | ObjCDeclSpec::DQ_PR_assign | | 
|  | 4034 | ObjCDeclSpec::DQ_PR_copy | | 
|  | 4035 | ObjCDeclSpec::DQ_PR_retain))) | 
|  | 4036 | return true; | 
|  | 4037 |  | 
|  | 4038 | // Check for more than one of { assign, copy, retain }. | 
|  | 4039 | unsigned AssignCopyRetMask = Attributes & (ObjCDeclSpec::DQ_PR_assign | | 
|  | 4040 | ObjCDeclSpec::DQ_PR_copy | | 
|  | 4041 | ObjCDeclSpec::DQ_PR_retain); | 
|  | 4042 | if (AssignCopyRetMask && | 
|  | 4043 | AssignCopyRetMask != ObjCDeclSpec::DQ_PR_assign && | 
|  | 4044 | AssignCopyRetMask != ObjCDeclSpec::DQ_PR_copy && | 
|  | 4045 | AssignCopyRetMask != ObjCDeclSpec::DQ_PR_retain) | 
|  | 4046 | return true; | 
|  | 4047 |  | 
|  | 4048 | return false; | 
|  | 4049 | } | 
|  | 4050 |  | 
| Douglas Gregor | a93b108 | 2009-11-18 23:08:07 +0000 | [diff] [blame] | 4051 | void Sema::CodeCompleteObjCPropertyFlags(Scope *S, ObjCDeclSpec &ODS) { | 
| Steve Naroff | ece8e71 | 2009-10-08 21:55:05 +0000 | [diff] [blame] | 4052 | if (!CodeCompleter) | 
|  | 4053 | return; | 
| Douglas Gregor | d3c6854 | 2009-11-19 01:08:35 +0000 | [diff] [blame] | 4054 |  | 
| Steve Naroff | ece8e71 | 2009-10-08 21:55:05 +0000 | [diff] [blame] | 4055 | unsigned Attributes = ODS.getPropertyAttributes(); | 
|  | 4056 |  | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 4057 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4058 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 4059 | CodeCompletionContext::CCC_Other); | 
| Steve Naroff | ece8e71 | 2009-10-08 21:55:05 +0000 | [diff] [blame] | 4060 | Results.EnterNewScope(); | 
| Douglas Gregor | 988358f | 2009-11-19 00:14:45 +0000 | [diff] [blame] | 4061 | if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readonly)) | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 4062 | Results.AddResult(CodeCompletionResult("readonly")); | 
| Douglas Gregor | 988358f | 2009-11-19 00:14:45 +0000 | [diff] [blame] | 4063 | if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_assign)) | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 4064 | Results.AddResult(CodeCompletionResult("assign")); | 
| Douglas Gregor | 988358f | 2009-11-19 00:14:45 +0000 | [diff] [blame] | 4065 | if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readwrite)) | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 4066 | Results.AddResult(CodeCompletionResult("readwrite")); | 
| Douglas Gregor | 988358f | 2009-11-19 00:14:45 +0000 | [diff] [blame] | 4067 | if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_retain)) | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 4068 | Results.AddResult(CodeCompletionResult("retain")); | 
| Douglas Gregor | 988358f | 2009-11-19 00:14:45 +0000 | [diff] [blame] | 4069 | if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_copy)) | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 4070 | Results.AddResult(CodeCompletionResult("copy")); | 
| Douglas Gregor | 988358f | 2009-11-19 00:14:45 +0000 | [diff] [blame] | 4071 | if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_nonatomic)) | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 4072 | Results.AddResult(CodeCompletionResult("nonatomic")); | 
| Douglas Gregor | 988358f | 2009-11-19 00:14:45 +0000 | [diff] [blame] | 4073 | if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_setter)) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4074 | CodeCompletionBuilder Setter(Results.getAllocator()); | 
|  | 4075 | Setter.AddTypedTextChunk("setter"); | 
|  | 4076 | Setter.AddTextChunk(" = "); | 
|  | 4077 | Setter.AddPlaceholderChunk("method"); | 
|  | 4078 | Results.AddResult(CodeCompletionResult(Setter.TakeString())); | 
| Douglas Gregor | 54f0161 | 2009-11-19 00:01:57 +0000 | [diff] [blame] | 4079 | } | 
| Douglas Gregor | 988358f | 2009-11-19 00:14:45 +0000 | [diff] [blame] | 4080 | if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_getter)) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4081 | CodeCompletionBuilder Getter(Results.getAllocator()); | 
|  | 4082 | Getter.AddTypedTextChunk("getter"); | 
|  | 4083 | Getter.AddTextChunk(" = "); | 
|  | 4084 | Getter.AddPlaceholderChunk("method"); | 
|  | 4085 | Results.AddResult(CodeCompletionResult(Getter.TakeString())); | 
| Douglas Gregor | 54f0161 | 2009-11-19 00:01:57 +0000 | [diff] [blame] | 4086 | } | 
| Steve Naroff | ece8e71 | 2009-10-08 21:55:05 +0000 | [diff] [blame] | 4087 | Results.ExitScope(); | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 4088 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 4089 | CodeCompletionContext::CCC_Other, | 
|  | 4090 | Results.data(),Results.size()); | 
| Steve Naroff | ece8e71 | 2009-10-08 21:55:05 +0000 | [diff] [blame] | 4091 | } | 
| Steve Naroff | c4df6d2 | 2009-11-07 02:08:14 +0000 | [diff] [blame] | 4092 |  | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4093 | /// \brief Descripts the kind of Objective-C method that we want to find | 
|  | 4094 | /// via code completion. | 
|  | 4095 | enum ObjCMethodKind { | 
|  | 4096 | MK_Any, //< Any kind of method, provided it means other specified criteria. | 
|  | 4097 | MK_ZeroArgSelector, //< Zero-argument (unary) selector. | 
|  | 4098 | MK_OneArgSelector //< One-argument selector. | 
|  | 4099 | }; | 
|  | 4100 |  | 
| Douglas Gregor | 458433d | 2010-08-26 15:07:07 +0000 | [diff] [blame] | 4101 | static bool isAcceptableObjCSelector(Selector Sel, | 
|  | 4102 | ObjCMethodKind WantKind, | 
|  | 4103 | IdentifierInfo **SelIdents, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4104 | unsigned NumSelIdents, | 
|  | 4105 | bool AllowSameLength = true) { | 
| Douglas Gregor | 458433d | 2010-08-26 15:07:07 +0000 | [diff] [blame] | 4106 | if (NumSelIdents > Sel.getNumArgs()) | 
|  | 4107 | return false; | 
|  | 4108 |  | 
|  | 4109 | switch (WantKind) { | 
|  | 4110 | case MK_Any:             break; | 
|  | 4111 | case MK_ZeroArgSelector: return Sel.isUnarySelector(); | 
|  | 4112 | case MK_OneArgSelector:  return Sel.getNumArgs() == 1; | 
|  | 4113 | } | 
|  | 4114 |  | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4115 | if (!AllowSameLength && NumSelIdents && NumSelIdents == Sel.getNumArgs()) | 
|  | 4116 | return false; | 
|  | 4117 |  | 
| Douglas Gregor | 458433d | 2010-08-26 15:07:07 +0000 | [diff] [blame] | 4118 | for (unsigned I = 0; I != NumSelIdents; ++I) | 
|  | 4119 | if (SelIdents[I] != Sel.getIdentifierInfoForSlot(I)) | 
|  | 4120 | return false; | 
|  | 4121 |  | 
|  | 4122 | return true; | 
|  | 4123 | } | 
|  | 4124 |  | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4125 | static bool isAcceptableObjCMethod(ObjCMethodDecl *Method, | 
|  | 4126 | ObjCMethodKind WantKind, | 
|  | 4127 | IdentifierInfo **SelIdents, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4128 | unsigned NumSelIdents, | 
|  | 4129 | bool AllowSameLength = true) { | 
| Douglas Gregor | 458433d | 2010-08-26 15:07:07 +0000 | [diff] [blame] | 4130 | return isAcceptableObjCSelector(Method->getSelector(), WantKind, SelIdents, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4131 | NumSelIdents, AllowSameLength); | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4132 | } | 
| Douglas Gregor | d36adf5 | 2010-09-16 16:06:31 +0000 | [diff] [blame] | 4133 |  | 
|  | 4134 | namespace { | 
|  | 4135 | /// \brief A set of selectors, which is used to avoid introducing multiple | 
|  | 4136 | /// completions with the same selector into the result set. | 
|  | 4137 | typedef llvm::SmallPtrSet<Selector, 16> VisitedSelectorSet; | 
|  | 4138 | } | 
|  | 4139 |  | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4140 | /// \brief Add all of the Objective-C methods in the given Objective-C | 
|  | 4141 | /// container to the set of results. | 
|  | 4142 | /// | 
|  | 4143 | /// The container will be a class, protocol, category, or implementation of | 
|  | 4144 | /// any of the above. This mether will recurse to include methods from | 
|  | 4145 | /// the superclasses of classes along with their categories, protocols, and | 
|  | 4146 | /// implementations. | 
|  | 4147 | /// | 
|  | 4148 | /// \param Container the container in which we'll look to find methods. | 
|  | 4149 | /// | 
|  | 4150 | /// \param WantInstance whether to add instance methods (only); if false, this | 
|  | 4151 | /// routine will add factory methods (only). | 
|  | 4152 | /// | 
|  | 4153 | /// \param CurContext the context in which we're performing the lookup that | 
|  | 4154 | /// finds methods. | 
|  | 4155 | /// | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4156 | /// \param AllowSameLength Whether we allow a method to be added to the list | 
|  | 4157 | /// when it has the same number of parameters as we have selector identifiers. | 
|  | 4158 | /// | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4159 | /// \param Results the structure into which we'll add results. | 
|  | 4160 | static void AddObjCMethods(ObjCContainerDecl *Container, | 
|  | 4161 | bool WantInstanceMethods, | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4162 | ObjCMethodKind WantKind, | 
| Douglas Gregor | d3c6854 | 2009-11-19 01:08:35 +0000 | [diff] [blame] | 4163 | IdentifierInfo **SelIdents, | 
|  | 4164 | unsigned NumSelIdents, | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4165 | DeclContext *CurContext, | 
| Douglas Gregor | d36adf5 | 2010-09-16 16:06:31 +0000 | [diff] [blame] | 4166 | VisitedSelectorSet &Selectors, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4167 | bool AllowSameLength, | 
| Douglas Gregor | 408be5a | 2010-08-25 01:08:01 +0000 | [diff] [blame] | 4168 | ResultBuilder &Results, | 
|  | 4169 | bool InOriginalClass = true) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 4170 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4171 | for (ObjCContainerDecl::method_iterator M = Container->meth_begin(), | 
|  | 4172 | MEnd = Container->meth_end(); | 
|  | 4173 | M != MEnd; ++M) { | 
| Douglas Gregor | d3c6854 | 2009-11-19 01:08:35 +0000 | [diff] [blame] | 4174 | if ((*M)->isInstanceMethod() == WantInstanceMethods) { | 
|  | 4175 | // Check whether the selector identifiers we've been given are a | 
|  | 4176 | // subset of the identifiers for this particular method. | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4177 | if (!isAcceptableObjCMethod(*M, WantKind, SelIdents, NumSelIdents, | 
|  | 4178 | AllowSameLength)) | 
| Douglas Gregor | d3c6854 | 2009-11-19 01:08:35 +0000 | [diff] [blame] | 4179 | continue; | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4180 |  | 
| Douglas Gregor | d36adf5 | 2010-09-16 16:06:31 +0000 | [diff] [blame] | 4181 | if (!Selectors.insert((*M)->getSelector())) | 
|  | 4182 | continue; | 
|  | 4183 |  | 
| Douglas Gregor | d3c6854 | 2009-11-19 01:08:35 +0000 | [diff] [blame] | 4184 | Result R = Result(*M, 0); | 
|  | 4185 | R.StartParameter = NumSelIdents; | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4186 | R.AllParametersAreInformative = (WantKind != MK_Any); | 
| Douglas Gregor | 408be5a | 2010-08-25 01:08:01 +0000 | [diff] [blame] | 4187 | if (!InOriginalClass) | 
|  | 4188 | R.Priority += CCD_InBaseClass; | 
| Douglas Gregor | d3c6854 | 2009-11-19 01:08:35 +0000 | [diff] [blame] | 4189 | Results.MaybeAddResult(R, CurContext); | 
|  | 4190 | } | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4191 | } | 
|  | 4192 |  | 
| Douglas Gregor | e396c7b | 2010-09-16 15:34:59 +0000 | [diff] [blame] | 4193 | // Visit the protocols of protocols. | 
|  | 4194 | if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) { | 
|  | 4195 | const ObjCList<ObjCProtocolDecl> &Protocols | 
|  | 4196 | = Protocol->getReferencedProtocols(); | 
|  | 4197 | for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(), | 
|  | 4198 | E = Protocols.end(); | 
|  | 4199 | I != E; ++I) | 
|  | 4200 | AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents, NumSelIdents, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4201 | CurContext, Selectors, AllowSameLength, Results, false); | 
| Douglas Gregor | e396c7b | 2010-09-16 15:34:59 +0000 | [diff] [blame] | 4202 | } | 
|  | 4203 |  | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4204 | ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container); | 
|  | 4205 | if (!IFace) | 
|  | 4206 | return; | 
|  | 4207 |  | 
|  | 4208 | // Add methods in protocols. | 
|  | 4209 | const ObjCList<ObjCProtocolDecl> &Protocols= IFace->getReferencedProtocols(); | 
|  | 4210 | for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(), | 
|  | 4211 | E = Protocols.end(); | 
|  | 4212 | I != E; ++I) | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4213 | AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents, NumSelIdents, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4214 | CurContext, Selectors, AllowSameLength, Results, false); | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4215 |  | 
|  | 4216 | // Add methods in categories. | 
|  | 4217 | for (ObjCCategoryDecl *CatDecl = IFace->getCategoryList(); CatDecl; | 
|  | 4218 | CatDecl = CatDecl->getNextClassCategory()) { | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4219 | AddObjCMethods(CatDecl, WantInstanceMethods, WantKind, SelIdents, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4220 | NumSelIdents, CurContext, Selectors, AllowSameLength, | 
|  | 4221 | Results, InOriginalClass); | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4222 |  | 
|  | 4223 | // Add a categories protocol methods. | 
|  | 4224 | const ObjCList<ObjCProtocolDecl> &Protocols | 
|  | 4225 | = CatDecl->getReferencedProtocols(); | 
|  | 4226 | for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(), | 
|  | 4227 | E = Protocols.end(); | 
|  | 4228 | I != E; ++I) | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4229 | AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4230 | NumSelIdents, CurContext, Selectors, AllowSameLength, | 
|  | 4231 | Results, false); | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4232 |  | 
|  | 4233 | // Add methods in category implementations. | 
|  | 4234 | if (ObjCCategoryImplDecl *Impl = CatDecl->getImplementation()) | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4235 | AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4236 | NumSelIdents, CurContext, Selectors, AllowSameLength, | 
|  | 4237 | Results, InOriginalClass); | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4238 | } | 
|  | 4239 |  | 
|  | 4240 | // Add methods in superclass. | 
|  | 4241 | if (IFace->getSuperClass()) | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4242 | AddObjCMethods(IFace->getSuperClass(), WantInstanceMethods, WantKind, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4243 | SelIdents, NumSelIdents, CurContext, Selectors, | 
|  | 4244 | AllowSameLength, Results, false); | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4245 |  | 
|  | 4246 | // Add methods in our implementation, if any. | 
|  | 4247 | if (ObjCImplementationDecl *Impl = IFace->getImplementation()) | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4248 | AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4249 | NumSelIdents, CurContext, Selectors, AllowSameLength, | 
|  | 4250 | Results, InOriginalClass); | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4251 | } | 
|  | 4252 |  | 
|  | 4253 |  | 
| Douglas Gregor | bdb2d50 | 2010-12-21 17:34:17 +0000 | [diff] [blame] | 4254 | void Sema::CodeCompleteObjCPropertyGetter(Scope *S, Decl *ClassDecl) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 4255 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4256 |  | 
|  | 4257 | // Try to find the interface where getters might live. | 
| John McCall | d226f65 | 2010-08-21 09:40:31 +0000 | [diff] [blame] | 4258 | ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(ClassDecl); | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4259 | if (!Class) { | 
|  | 4260 | if (ObjCCategoryDecl *Category | 
| John McCall | d226f65 | 2010-08-21 09:40:31 +0000 | [diff] [blame] | 4261 | = dyn_cast_or_null<ObjCCategoryDecl>(ClassDecl)) | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4262 | Class = Category->getClassInterface(); | 
|  | 4263 |  | 
|  | 4264 | if (!Class) | 
|  | 4265 | return; | 
|  | 4266 | } | 
|  | 4267 |  | 
|  | 4268 | // Find all of the potential getters. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4269 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 4270 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4271 | Results.EnterNewScope(); | 
|  | 4272 |  | 
| Douglas Gregor | d36adf5 | 2010-09-16 16:06:31 +0000 | [diff] [blame] | 4273 | VisitedSelectorSet Selectors; | 
|  | 4274 | AddObjCMethods(Class, true, MK_ZeroArgSelector, 0, 0, CurContext, Selectors, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4275 | /*AllowSameLength=*/true, Results); | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4276 | Results.ExitScope(); | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 4277 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 4278 | CodeCompletionContext::CCC_Other, | 
|  | 4279 | Results.data(),Results.size()); | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4280 | } | 
|  | 4281 |  | 
| Douglas Gregor | bdb2d50 | 2010-12-21 17:34:17 +0000 | [diff] [blame] | 4282 | void Sema::CodeCompleteObjCPropertySetter(Scope *S, Decl *ObjCImplDecl) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 4283 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4284 |  | 
|  | 4285 | // Try to find the interface where setters might live. | 
|  | 4286 | ObjCInterfaceDecl *Class | 
| John McCall | d226f65 | 2010-08-21 09:40:31 +0000 | [diff] [blame] | 4287 | = dyn_cast_or_null<ObjCInterfaceDecl>(ObjCImplDecl); | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4288 | if (!Class) { | 
|  | 4289 | if (ObjCCategoryDecl *Category | 
| John McCall | d226f65 | 2010-08-21 09:40:31 +0000 | [diff] [blame] | 4290 | = dyn_cast_or_null<ObjCCategoryDecl>(ObjCImplDecl)) | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4291 | Class = Category->getClassInterface(); | 
|  | 4292 |  | 
|  | 4293 | if (!Class) | 
|  | 4294 | return; | 
|  | 4295 | } | 
|  | 4296 |  | 
|  | 4297 | // Find all of the potential getters. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4298 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 4299 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4300 | Results.EnterNewScope(); | 
|  | 4301 |  | 
| Douglas Gregor | d36adf5 | 2010-09-16 16:06:31 +0000 | [diff] [blame] | 4302 | VisitedSelectorSet Selectors; | 
|  | 4303 | AddObjCMethods(Class, true, MK_OneArgSelector, 0, 0, CurContext, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4304 | Selectors, /*AllowSameLength=*/true, Results); | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4305 |  | 
|  | 4306 | Results.ExitScope(); | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 4307 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 4308 | CodeCompletionContext::CCC_Other, | 
|  | 4309 | Results.data(),Results.size()); | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4310 | } | 
|  | 4311 |  | 
| Douglas Gregor | afc4578 | 2011-02-15 22:19:42 +0000 | [diff] [blame] | 4312 | void Sema::CodeCompleteObjCPassingType(Scope *S, ObjCDeclSpec &DS, | 
|  | 4313 | bool IsParameter) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 4314 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4315 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 4316 | CodeCompletionContext::CCC_Type); | 
| Douglas Gregor | d32b022 | 2010-08-24 01:06:58 +0000 | [diff] [blame] | 4317 | Results.EnterNewScope(); | 
|  | 4318 |  | 
|  | 4319 | // Add context-sensitive, Objective-C parameter-passing keywords. | 
|  | 4320 | bool AddedInOut = false; | 
|  | 4321 | if ((DS.getObjCDeclQualifier() & | 
|  | 4322 | (ObjCDeclSpec::DQ_In | ObjCDeclSpec::DQ_Inout)) == 0) { | 
|  | 4323 | Results.AddResult("in"); | 
|  | 4324 | Results.AddResult("inout"); | 
|  | 4325 | AddedInOut = true; | 
|  | 4326 | } | 
|  | 4327 | if ((DS.getObjCDeclQualifier() & | 
|  | 4328 | (ObjCDeclSpec::DQ_Out | ObjCDeclSpec::DQ_Inout)) == 0) { | 
|  | 4329 | Results.AddResult("out"); | 
|  | 4330 | if (!AddedInOut) | 
|  | 4331 | Results.AddResult("inout"); | 
|  | 4332 | } | 
|  | 4333 | if ((DS.getObjCDeclQualifier() & | 
|  | 4334 | (ObjCDeclSpec::DQ_Bycopy | ObjCDeclSpec::DQ_Byref | | 
|  | 4335 | ObjCDeclSpec::DQ_Oneway)) == 0) { | 
|  | 4336 | Results.AddResult("bycopy"); | 
|  | 4337 | Results.AddResult("byref"); | 
|  | 4338 | Results.AddResult("oneway"); | 
|  | 4339 | } | 
|  | 4340 |  | 
| Douglas Gregor | afc4578 | 2011-02-15 22:19:42 +0000 | [diff] [blame] | 4341 | // If we're completing the return type of an Objective-C method and the | 
|  | 4342 | // identifier IBAction refers to a macro, provide a completion item for | 
|  | 4343 | // an action, e.g., | 
|  | 4344 | //   IBAction)<#selector#>:(id)sender | 
|  | 4345 | if (DS.getObjCDeclQualifier() == 0 && !IsParameter && | 
|  | 4346 | Context.Idents.get("IBAction").hasMacroDefinition()) { | 
|  | 4347 | typedef CodeCompletionString::Chunk Chunk; | 
|  | 4348 | CodeCompletionBuilder Builder(Results.getAllocator(), CCP_CodePattern, | 
|  | 4349 | CXAvailability_Available); | 
|  | 4350 | Builder.AddTypedTextChunk("IBAction"); | 
|  | 4351 | Builder.AddChunk(Chunk(CodeCompletionString::CK_RightParen)); | 
|  | 4352 | Builder.AddPlaceholderChunk("selector"); | 
|  | 4353 | Builder.AddChunk(Chunk(CodeCompletionString::CK_Colon)); | 
|  | 4354 | Builder.AddChunk(Chunk(CodeCompletionString::CK_LeftParen)); | 
|  | 4355 | Builder.AddTextChunk("id"); | 
|  | 4356 | Builder.AddChunk(Chunk(CodeCompletionString::CK_RightParen)); | 
|  | 4357 | Builder.AddTextChunk("sender"); | 
|  | 4358 | Results.AddResult(CodeCompletionResult(Builder.TakeString())); | 
|  | 4359 | } | 
|  | 4360 |  | 
| Douglas Gregor | d32b022 | 2010-08-24 01:06:58 +0000 | [diff] [blame] | 4361 | // Add various builtin type names and specifiers. | 
|  | 4362 | AddOrdinaryNameResults(PCC_Type, S, *this, Results); | 
|  | 4363 | Results.ExitScope(); | 
|  | 4364 |  | 
|  | 4365 | // Add the various type names | 
|  | 4366 | Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName); | 
|  | 4367 | CodeCompletionDeclConsumer Consumer(Results, CurContext); | 
|  | 4368 | LookupVisibleDecls(S, LookupOrdinaryName, Consumer, | 
|  | 4369 | CodeCompleter->includeGlobals()); | 
|  | 4370 |  | 
|  | 4371 | if (CodeCompleter->includeMacros()) | 
|  | 4372 | AddMacroResults(PP, Results); | 
|  | 4373 |  | 
|  | 4374 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 4375 | CodeCompletionContext::CCC_Type, | 
|  | 4376 | Results.data(), Results.size()); | 
|  | 4377 | } | 
|  | 4378 |  | 
| Douglas Gregor | 22f5699 | 2010-04-06 19:22:33 +0000 | [diff] [blame] | 4379 | /// \brief When we have an expression with type "id", we may assume | 
|  | 4380 | /// that it has some more-specific class type based on knowledge of | 
|  | 4381 | /// common uses of Objective-C. This routine returns that class type, | 
|  | 4382 | /// or NULL if no better result could be determined. | 
|  | 4383 | static ObjCInterfaceDecl *GetAssumedMessageSendExprType(Expr *E) { | 
| Douglas Gregor | 78edf51 | 2010-09-15 16:23:04 +0000 | [diff] [blame] | 4384 | ObjCMessageExpr *Msg = dyn_cast_or_null<ObjCMessageExpr>(E); | 
| Douglas Gregor | 22f5699 | 2010-04-06 19:22:33 +0000 | [diff] [blame] | 4385 | if (!Msg) | 
|  | 4386 | return 0; | 
|  | 4387 |  | 
|  | 4388 | Selector Sel = Msg->getSelector(); | 
|  | 4389 | if (Sel.isNull()) | 
|  | 4390 | return 0; | 
|  | 4391 |  | 
|  | 4392 | IdentifierInfo *Id = Sel.getIdentifierInfoForSlot(0); | 
|  | 4393 | if (!Id) | 
|  | 4394 | return 0; | 
|  | 4395 |  | 
|  | 4396 | ObjCMethodDecl *Method = Msg->getMethodDecl(); | 
|  | 4397 | if (!Method) | 
|  | 4398 | return 0; | 
|  | 4399 |  | 
|  | 4400 | // Determine the class that we're sending the message to. | 
| Douglas Gregor | 04badcf | 2010-04-21 00:45:42 +0000 | [diff] [blame] | 4401 | ObjCInterfaceDecl *IFace = 0; | 
|  | 4402 | switch (Msg->getReceiverKind()) { | 
|  | 4403 | case ObjCMessageExpr::Class: | 
| John McCall | c12c5bb | 2010-05-15 11:32:37 +0000 | [diff] [blame] | 4404 | if (const ObjCObjectType *ObjType | 
|  | 4405 | = Msg->getClassReceiver()->getAs<ObjCObjectType>()) | 
|  | 4406 | IFace = ObjType->getInterface(); | 
| Douglas Gregor | 04badcf | 2010-04-21 00:45:42 +0000 | [diff] [blame] | 4407 | break; | 
|  | 4408 |  | 
|  | 4409 | case ObjCMessageExpr::Instance: { | 
|  | 4410 | QualType T = Msg->getInstanceReceiver()->getType(); | 
|  | 4411 | if (const ObjCObjectPointerType *Ptr = T->getAs<ObjCObjectPointerType>()) | 
|  | 4412 | IFace = Ptr->getInterfaceDecl(); | 
|  | 4413 | break; | 
|  | 4414 | } | 
|  | 4415 |  | 
|  | 4416 | case ObjCMessageExpr::SuperInstance: | 
|  | 4417 | case ObjCMessageExpr::SuperClass: | 
|  | 4418 | break; | 
| Douglas Gregor | 22f5699 | 2010-04-06 19:22:33 +0000 | [diff] [blame] | 4419 | } | 
|  | 4420 |  | 
|  | 4421 | if (!IFace) | 
|  | 4422 | return 0; | 
|  | 4423 |  | 
|  | 4424 | ObjCInterfaceDecl *Super = IFace->getSuperClass(); | 
|  | 4425 | if (Method->isInstanceMethod()) | 
|  | 4426 | return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName()) | 
|  | 4427 | .Case("retain", IFace) | 
|  | 4428 | .Case("autorelease", IFace) | 
|  | 4429 | .Case("copy", IFace) | 
|  | 4430 | .Case("copyWithZone", IFace) | 
|  | 4431 | .Case("mutableCopy", IFace) | 
|  | 4432 | .Case("mutableCopyWithZone", IFace) | 
|  | 4433 | .Case("awakeFromCoder", IFace) | 
|  | 4434 | .Case("replacementObjectFromCoder", IFace) | 
|  | 4435 | .Case("class", IFace) | 
|  | 4436 | .Case("classForCoder", IFace) | 
|  | 4437 | .Case("superclass", Super) | 
|  | 4438 | .Default(0); | 
|  | 4439 |  | 
|  | 4440 | return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName()) | 
|  | 4441 | .Case("new", IFace) | 
|  | 4442 | .Case("alloc", IFace) | 
|  | 4443 | .Case("allocWithZone", IFace) | 
|  | 4444 | .Case("class", IFace) | 
|  | 4445 | .Case("superclass", Super) | 
|  | 4446 | .Default(0); | 
|  | 4447 | } | 
|  | 4448 |  | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4449 | // Add a special completion for a message send to "super", which fills in the | 
|  | 4450 | // most likely case of forwarding all of our arguments to the superclass | 
|  | 4451 | // function. | 
|  | 4452 | /// | 
|  | 4453 | /// \param S The semantic analysis object. | 
|  | 4454 | /// | 
|  | 4455 | /// \param S NeedSuperKeyword Whether we need to prefix this completion with | 
|  | 4456 | /// the "super" keyword. Otherwise, we just need to provide the arguments. | 
|  | 4457 | /// | 
|  | 4458 | /// \param SelIdents The identifiers in the selector that have already been | 
|  | 4459 | /// provided as arguments for a send to "super". | 
|  | 4460 | /// | 
|  | 4461 | /// \param NumSelIdents The number of identifiers in \p SelIdents. | 
|  | 4462 | /// | 
|  | 4463 | /// \param Results The set of results to augment. | 
|  | 4464 | /// | 
|  | 4465 | /// \returns the Objective-C method declaration that would be invoked by | 
|  | 4466 | /// this "super" completion. If NULL, no completion was added. | 
|  | 4467 | static ObjCMethodDecl *AddSuperSendCompletion(Sema &S, bool NeedSuperKeyword, | 
|  | 4468 | IdentifierInfo **SelIdents, | 
|  | 4469 | unsigned NumSelIdents, | 
|  | 4470 | ResultBuilder &Results) { | 
|  | 4471 | ObjCMethodDecl *CurMethod = S.getCurMethodDecl(); | 
|  | 4472 | if (!CurMethod) | 
|  | 4473 | return 0; | 
|  | 4474 |  | 
|  | 4475 | ObjCInterfaceDecl *Class = CurMethod->getClassInterface(); | 
|  | 4476 | if (!Class) | 
|  | 4477 | return 0; | 
|  | 4478 |  | 
|  | 4479 | // Try to find a superclass method with the same selector. | 
|  | 4480 | ObjCMethodDecl *SuperMethod = 0; | 
|  | 4481 | while ((Class = Class->getSuperClass()) && !SuperMethod) | 
|  | 4482 | SuperMethod = Class->getMethod(CurMethod->getSelector(), | 
|  | 4483 | CurMethod->isInstanceMethod()); | 
|  | 4484 |  | 
|  | 4485 | if (!SuperMethod) | 
|  | 4486 | return 0; | 
|  | 4487 |  | 
|  | 4488 | // Check whether the superclass method has the same signature. | 
|  | 4489 | if (CurMethod->param_size() != SuperMethod->param_size() || | 
|  | 4490 | CurMethod->isVariadic() != SuperMethod->isVariadic()) | 
|  | 4491 | return 0; | 
|  | 4492 |  | 
|  | 4493 | for (ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin(), | 
|  | 4494 | CurPEnd = CurMethod->param_end(), | 
|  | 4495 | SuperP = SuperMethod->param_begin(); | 
|  | 4496 | CurP != CurPEnd; ++CurP, ++SuperP) { | 
|  | 4497 | // Make sure the parameter types are compatible. | 
|  | 4498 | if (!S.Context.hasSameUnqualifiedType((*CurP)->getType(), | 
|  | 4499 | (*SuperP)->getType())) | 
|  | 4500 | return 0; | 
|  | 4501 |  | 
|  | 4502 | // Make sure we have a parameter name to forward! | 
|  | 4503 | if (!(*CurP)->getIdentifier()) | 
|  | 4504 | return 0; | 
|  | 4505 | } | 
|  | 4506 |  | 
|  | 4507 | // We have a superclass method. Now, form the send-to-super completion. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4508 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4509 |  | 
|  | 4510 | // Give this completion a return type. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4511 | AddResultTypeChunk(S.Context, SuperMethod, Builder); | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4512 |  | 
|  | 4513 | // If we need the "super" keyword, add it (plus some spacing). | 
|  | 4514 | if (NeedSuperKeyword) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4515 | Builder.AddTypedTextChunk("super"); | 
|  | 4516 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4517 | } | 
|  | 4518 |  | 
|  | 4519 | Selector Sel = CurMethod->getSelector(); | 
|  | 4520 | if (Sel.isUnarySelector()) { | 
|  | 4521 | if (NeedSuperKeyword) | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 4522 | Builder.AddTextChunk(Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4523 | Sel.getIdentifierInfoForSlot(0)->getName())); | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4524 | else | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 4525 | Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4526 | Sel.getIdentifierInfoForSlot(0)->getName())); | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4527 | } else { | 
|  | 4528 | ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin(); | 
|  | 4529 | for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I, ++CurP) { | 
|  | 4530 | if (I > NumSelIdents) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4531 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4532 |  | 
|  | 4533 | if (I < NumSelIdents) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4534 | Builder.AddInformativeChunk( | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 4535 | Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4536 | Sel.getIdentifierInfoForSlot(I)->getName().str() + ":")); | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4537 | else if (NeedSuperKeyword || I > NumSelIdents) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4538 | Builder.AddTextChunk( | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 4539 | Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4540 | Sel.getIdentifierInfoForSlot(I)->getName().str() + ":")); | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 4541 | Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4542 | (*CurP)->getIdentifier()->getName())); | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4543 | } else { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4544 | Builder.AddTypedTextChunk( | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 4545 | Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4546 | Sel.getIdentifierInfoForSlot(I)->getName().str() + ":")); | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 4547 | Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4548 | (*CurP)->getIdentifier()->getName())); | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4549 | } | 
|  | 4550 | } | 
|  | 4551 | } | 
|  | 4552 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4553 | Results.AddResult(CodeCompletionResult(Builder.TakeString(), CCP_SuperCompletion, | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4554 | SuperMethod->isInstanceMethod() | 
|  | 4555 | ? CXCursor_ObjCInstanceMethodDecl | 
|  | 4556 | : CXCursor_ObjCClassMethodDecl)); | 
|  | 4557 | return SuperMethod; | 
|  | 4558 | } | 
|  | 4559 |  | 
| Douglas Gregor | 8e254cf | 2010-05-27 23:06:34 +0000 | [diff] [blame] | 4560 | void Sema::CodeCompleteObjCMessageReceiver(Scope *S) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 4561 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4562 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 4563 | CodeCompletionContext::CCC_ObjCMessageReceiver, | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 4564 | &ResultBuilder::IsObjCMessageReceiver); | 
| Douglas Gregor | 8e254cf | 2010-05-27 23:06:34 +0000 | [diff] [blame] | 4565 |  | 
| Douglas Gregor | 8e254cf | 2010-05-27 23:06:34 +0000 | [diff] [blame] | 4566 | CodeCompletionDeclConsumer Consumer(Results, CurContext); | 
|  | 4567 | Results.EnterNewScope(); | 
| Douglas Gregor | 8071e42 | 2010-08-15 06:18:01 +0000 | [diff] [blame] | 4568 | LookupVisibleDecls(S, LookupOrdinaryName, Consumer, | 
|  | 4569 | CodeCompleter->includeGlobals()); | 
| Douglas Gregor | 8e254cf | 2010-05-27 23:06:34 +0000 | [diff] [blame] | 4570 |  | 
|  | 4571 | // If we are in an Objective-C method inside a class that has a superclass, | 
|  | 4572 | // add "super" as an option. | 
|  | 4573 | if (ObjCMethodDecl *Method = getCurMethodDecl()) | 
|  | 4574 | if (ObjCInterfaceDecl *Iface = Method->getClassInterface()) | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4575 | if (Iface->getSuperClass()) { | 
| Douglas Gregor | 8e254cf | 2010-05-27 23:06:34 +0000 | [diff] [blame] | 4576 | Results.AddResult(Result("super")); | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4577 |  | 
|  | 4578 | AddSuperSendCompletion(*this, /*NeedSuperKeyword=*/true, 0, 0, Results); | 
|  | 4579 | } | 
| Douglas Gregor | 8e254cf | 2010-05-27 23:06:34 +0000 | [diff] [blame] | 4580 |  | 
|  | 4581 | Results.ExitScope(); | 
|  | 4582 |  | 
|  | 4583 | if (CodeCompleter->includeMacros()) | 
|  | 4584 | AddMacroResults(PP, Results); | 
| Douglas Gregor | cee9ff1 | 2010-09-20 22:39:41 +0000 | [diff] [blame] | 4585 | HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(), | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 4586 | Results.data(), Results.size()); | 
| Douglas Gregor | 8e254cf | 2010-05-27 23:06:34 +0000 | [diff] [blame] | 4587 |  | 
|  | 4588 | } | 
|  | 4589 |  | 
| Douglas Gregor | 2725ca8 | 2010-04-21 19:57:20 +0000 | [diff] [blame] | 4590 | void Sema::CodeCompleteObjCSuperMessage(Scope *S, SourceLocation SuperLoc, | 
|  | 4591 | IdentifierInfo **SelIdents, | 
| Douglas Gregor | 70c5ac7 | 2010-09-20 23:34:21 +0000 | [diff] [blame] | 4592 | unsigned NumSelIdents, | 
|  | 4593 | bool AtArgumentExpression) { | 
| Douglas Gregor | 2725ca8 | 2010-04-21 19:57:20 +0000 | [diff] [blame] | 4594 | ObjCInterfaceDecl *CDecl = 0; | 
|  | 4595 | if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) { | 
|  | 4596 | // Figure out which interface we're in. | 
|  | 4597 | CDecl = CurMethod->getClassInterface(); | 
|  | 4598 | if (!CDecl) | 
|  | 4599 | return; | 
|  | 4600 |  | 
|  | 4601 | // Find the superclass of this class. | 
|  | 4602 | CDecl = CDecl->getSuperClass(); | 
|  | 4603 | if (!CDecl) | 
|  | 4604 | return; | 
|  | 4605 |  | 
|  | 4606 | if (CurMethod->isInstanceMethod()) { | 
|  | 4607 | // We are inside an instance method, which means that the message | 
|  | 4608 | // send [super ...] is actually calling an instance method on the | 
| Douglas Gregor | 6b0656a | 2010-10-13 21:24:53 +0000 | [diff] [blame] | 4609 | // current object. | 
|  | 4610 | return CodeCompleteObjCInstanceMessage(S, 0, | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4611 | SelIdents, NumSelIdents, | 
| Douglas Gregor | 70c5ac7 | 2010-09-20 23:34:21 +0000 | [diff] [blame] | 4612 | AtArgumentExpression, | 
| Douglas Gregor | 6b0656a | 2010-10-13 21:24:53 +0000 | [diff] [blame] | 4613 | CDecl); | 
| Douglas Gregor | 2725ca8 | 2010-04-21 19:57:20 +0000 | [diff] [blame] | 4614 | } | 
|  | 4615 |  | 
|  | 4616 | // Fall through to send to the superclass in CDecl. | 
|  | 4617 | } else { | 
|  | 4618 | // "super" may be the name of a type or variable. Figure out which | 
|  | 4619 | // it is. | 
|  | 4620 | IdentifierInfo *Super = &Context.Idents.get("super"); | 
|  | 4621 | NamedDecl *ND = LookupSingleName(S, Super, SuperLoc, | 
|  | 4622 | LookupOrdinaryName); | 
|  | 4623 | if ((CDecl = dyn_cast_or_null<ObjCInterfaceDecl>(ND))) { | 
|  | 4624 | // "super" names an interface. Use it. | 
|  | 4625 | } else if (TypeDecl *TD = dyn_cast_or_null<TypeDecl>(ND)) { | 
| John McCall | c12c5bb | 2010-05-15 11:32:37 +0000 | [diff] [blame] | 4626 | if (const ObjCObjectType *Iface | 
|  | 4627 | = Context.getTypeDeclType(TD)->getAs<ObjCObjectType>()) | 
|  | 4628 | CDecl = Iface->getInterface(); | 
| Douglas Gregor | 2725ca8 | 2010-04-21 19:57:20 +0000 | [diff] [blame] | 4629 | } else if (ND && isa<UnresolvedUsingTypenameDecl>(ND)) { | 
|  | 4630 | // "super" names an unresolved type; we can't be more specific. | 
|  | 4631 | } else { | 
|  | 4632 | // Assume that "super" names some kind of value and parse that way. | 
|  | 4633 | CXXScopeSpec SS; | 
|  | 4634 | UnqualifiedId id; | 
|  | 4635 | id.setIdentifier(Super, SuperLoc); | 
| John McCall | 60d7b3a | 2010-08-24 06:29:42 +0000 | [diff] [blame] | 4636 | ExprResult SuperExpr = ActOnIdExpression(S, SS, id, false, false); | 
| Douglas Gregor | 2725ca8 | 2010-04-21 19:57:20 +0000 | [diff] [blame] | 4637 | return CodeCompleteObjCInstanceMessage(S, (Expr *)SuperExpr.get(), | 
| Douglas Gregor | 70c5ac7 | 2010-09-20 23:34:21 +0000 | [diff] [blame] | 4638 | SelIdents, NumSelIdents, | 
|  | 4639 | AtArgumentExpression); | 
| Douglas Gregor | 2725ca8 | 2010-04-21 19:57:20 +0000 | [diff] [blame] | 4640 | } | 
|  | 4641 |  | 
|  | 4642 | // Fall through | 
|  | 4643 | } | 
|  | 4644 |  | 
| John McCall | b3d8748 | 2010-08-24 05:47:05 +0000 | [diff] [blame] | 4645 | ParsedType Receiver; | 
| Douglas Gregor | 2725ca8 | 2010-04-21 19:57:20 +0000 | [diff] [blame] | 4646 | if (CDecl) | 
| John McCall | b3d8748 | 2010-08-24 05:47:05 +0000 | [diff] [blame] | 4647 | Receiver = ParsedType::make(Context.getObjCInterfaceType(CDecl)); | 
| Douglas Gregor | 2725ca8 | 2010-04-21 19:57:20 +0000 | [diff] [blame] | 4648 | return CodeCompleteObjCClassMessage(S, Receiver, SelIdents, | 
| Douglas Gregor | 70c5ac7 | 2010-09-20 23:34:21 +0000 | [diff] [blame] | 4649 | NumSelIdents, AtArgumentExpression, | 
|  | 4650 | /*IsSuper=*/true); | 
| Douglas Gregor | 2725ca8 | 2010-04-21 19:57:20 +0000 | [diff] [blame] | 4651 | } | 
|  | 4652 |  | 
| Douglas Gregor | b9d7757 | 2010-09-21 00:03:25 +0000 | [diff] [blame] | 4653 | /// \brief Given a set of code-completion results for the argument of a message | 
|  | 4654 | /// send, determine the preferred type (if any) for that argument expression. | 
|  | 4655 | static QualType getPreferredArgumentTypeForMessageSend(ResultBuilder &Results, | 
|  | 4656 | unsigned NumSelIdents) { | 
|  | 4657 | typedef CodeCompletionResult Result; | 
|  | 4658 | ASTContext &Context = Results.getSema().Context; | 
|  | 4659 |  | 
|  | 4660 | QualType PreferredType; | 
|  | 4661 | unsigned BestPriority = CCP_Unlikely * 2; | 
|  | 4662 | Result *ResultsData = Results.data(); | 
|  | 4663 | for (unsigned I = 0, N = Results.size(); I != N; ++I) { | 
|  | 4664 | Result &R = ResultsData[I]; | 
|  | 4665 | if (R.Kind == Result::RK_Declaration && | 
|  | 4666 | isa<ObjCMethodDecl>(R.Declaration)) { | 
|  | 4667 | if (R.Priority <= BestPriority) { | 
|  | 4668 | ObjCMethodDecl *Method = cast<ObjCMethodDecl>(R.Declaration); | 
|  | 4669 | if (NumSelIdents <= Method->param_size()) { | 
|  | 4670 | QualType MyPreferredType = Method->param_begin()[NumSelIdents - 1] | 
|  | 4671 | ->getType(); | 
|  | 4672 | if (R.Priority < BestPriority || PreferredType.isNull()) { | 
|  | 4673 | BestPriority = R.Priority; | 
|  | 4674 | PreferredType = MyPreferredType; | 
|  | 4675 | } else if (!Context.hasSameUnqualifiedType(PreferredType, | 
|  | 4676 | MyPreferredType)) { | 
|  | 4677 | PreferredType = QualType(); | 
|  | 4678 | } | 
|  | 4679 | } | 
|  | 4680 | } | 
|  | 4681 | } | 
|  | 4682 | } | 
|  | 4683 |  | 
|  | 4684 | return PreferredType; | 
|  | 4685 | } | 
|  | 4686 |  | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4687 | static void AddClassMessageCompletions(Sema &SemaRef, Scope *S, | 
|  | 4688 | ParsedType Receiver, | 
|  | 4689 | IdentifierInfo **SelIdents, | 
|  | 4690 | unsigned NumSelIdents, | 
| Douglas Gregor | 70c5ac7 | 2010-09-20 23:34:21 +0000 | [diff] [blame] | 4691 | bool AtArgumentExpression, | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4692 | bool IsSuper, | 
|  | 4693 | ResultBuilder &Results) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 4694 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 24a069f | 2009-11-17 17:59:40 +0000 | [diff] [blame] | 4695 | ObjCInterfaceDecl *CDecl = 0; | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4696 |  | 
| Douglas Gregor | 24a069f | 2009-11-17 17:59:40 +0000 | [diff] [blame] | 4697 | // If the given name refers to an interface type, retrieve the | 
|  | 4698 | // corresponding declaration. | 
| Douglas Gregor | 2725ca8 | 2010-04-21 19:57:20 +0000 | [diff] [blame] | 4699 | if (Receiver) { | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4700 | QualType T = SemaRef.GetTypeFromParser(Receiver, 0); | 
| Douglas Gregor | 2725ca8 | 2010-04-21 19:57:20 +0000 | [diff] [blame] | 4701 | if (!T.isNull()) | 
| John McCall | c12c5bb | 2010-05-15 11:32:37 +0000 | [diff] [blame] | 4702 | if (const ObjCObjectType *Interface = T->getAs<ObjCObjectType>()) | 
|  | 4703 | CDecl = Interface->getInterface(); | 
| Douglas Gregor | 24a069f | 2009-11-17 17:59:40 +0000 | [diff] [blame] | 4704 | } | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4705 |  | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4706 | // Add all of the factory methods in this Objective-C class, its protocols, | 
|  | 4707 | // superclasses, categories, implementation, etc. | 
| Steve Naroff | c4df6d2 | 2009-11-07 02:08:14 +0000 | [diff] [blame] | 4708 | Results.EnterNewScope(); | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4709 |  | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4710 | // If this is a send-to-super, try to add the special "super" send | 
|  | 4711 | // completion. | 
|  | 4712 | if (IsSuper) { | 
|  | 4713 | if (ObjCMethodDecl *SuperMethod | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4714 | = AddSuperSendCompletion(SemaRef, false, SelIdents, NumSelIdents, | 
|  | 4715 | Results)) | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4716 | Results.Ignore(SuperMethod); | 
|  | 4717 | } | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4718 |  | 
| Douglas Gregor | 265f749 | 2010-08-27 15:29:55 +0000 | [diff] [blame] | 4719 | // If we're inside an Objective-C method definition, prefer its selector to | 
|  | 4720 | // others. | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4721 | if (ObjCMethodDecl *CurMethod = SemaRef.getCurMethodDecl()) | 
| Douglas Gregor | 265f749 | 2010-08-27 15:29:55 +0000 | [diff] [blame] | 4722 | Results.setPreferredSelector(CurMethod->getSelector()); | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4723 |  | 
| Douglas Gregor | d36adf5 | 2010-09-16 16:06:31 +0000 | [diff] [blame] | 4724 | VisitedSelectorSet Selectors; | 
| Douglas Gregor | 13438f9 | 2010-04-06 16:40:00 +0000 | [diff] [blame] | 4725 | if (CDecl) | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4726 | AddObjCMethods(CDecl, false, MK_Any, SelIdents, NumSelIdents, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4727 | SemaRef.CurContext, Selectors, AtArgumentExpression, | 
|  | 4728 | Results); | 
| Douglas Gregor | 2725ca8 | 2010-04-21 19:57:20 +0000 | [diff] [blame] | 4729 | else { | 
| Douglas Gregor | 13438f9 | 2010-04-06 16:40:00 +0000 | [diff] [blame] | 4730 | // We're messaging "id" as a type; provide all class/factory methods. | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4731 |  | 
| Douglas Gregor | 719770d | 2010-04-06 17:30:22 +0000 | [diff] [blame] | 4732 | // If we have an external source, load the entire class method | 
| Sebastian Redl | 3c7f413 | 2010-08-18 23:57:06 +0000 | [diff] [blame] | 4733 | // pool from the AST file. | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4734 | if (SemaRef.ExternalSource) { | 
|  | 4735 | for (uint32_t I = 0, | 
|  | 4736 | N = SemaRef.ExternalSource->GetNumExternalSelectors(); | 
| John McCall | 76bd1f3 | 2010-06-01 09:23:16 +0000 | [diff] [blame] | 4737 | I != N; ++I) { | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4738 | Selector Sel = SemaRef.ExternalSource->GetExternalSelector(I); | 
|  | 4739 | if (Sel.isNull() || SemaRef.MethodPool.count(Sel)) | 
| Douglas Gregor | 719770d | 2010-04-06 17:30:22 +0000 | [diff] [blame] | 4740 | continue; | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4741 |  | 
|  | 4742 | SemaRef.ReadMethodPool(Sel); | 
| Douglas Gregor | 719770d | 2010-04-06 17:30:22 +0000 | [diff] [blame] | 4743 | } | 
|  | 4744 | } | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4745 |  | 
|  | 4746 | for (Sema::GlobalMethodPool::iterator M = SemaRef.MethodPool.begin(), | 
|  | 4747 | MEnd = SemaRef.MethodPool.end(); | 
| Sebastian Redl | db9d214 | 2010-08-02 23:18:59 +0000 | [diff] [blame] | 4748 | M != MEnd; ++M) { | 
|  | 4749 | for (ObjCMethodList *MethList = &M->second.second; | 
|  | 4750 | MethList && MethList->Method; | 
| Douglas Gregor | 13438f9 | 2010-04-06 16:40:00 +0000 | [diff] [blame] | 4751 | MethList = MethList->Next) { | 
|  | 4752 | if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents, | 
|  | 4753 | NumSelIdents)) | 
|  | 4754 | continue; | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4755 |  | 
| Douglas Gregor | 13438f9 | 2010-04-06 16:40:00 +0000 | [diff] [blame] | 4756 | Result R(MethList->Method, 0); | 
|  | 4757 | R.StartParameter = NumSelIdents; | 
|  | 4758 | R.AllParametersAreInformative = false; | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4759 | Results.MaybeAddResult(R, SemaRef.CurContext); | 
| Douglas Gregor | 13438f9 | 2010-04-06 16:40:00 +0000 | [diff] [blame] | 4760 | } | 
|  | 4761 | } | 
|  | 4762 | } | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4763 |  | 
|  | 4764 | Results.ExitScope(); | 
|  | 4765 | } | 
| Douglas Gregor | 13438f9 | 2010-04-06 16:40:00 +0000 | [diff] [blame] | 4766 |  | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4767 | void Sema::CodeCompleteObjCClassMessage(Scope *S, ParsedType Receiver, | 
|  | 4768 | IdentifierInfo **SelIdents, | 
|  | 4769 | unsigned NumSelIdents, | 
| Douglas Gregor | 70c5ac7 | 2010-09-20 23:34:21 +0000 | [diff] [blame] | 4770 | bool AtArgumentExpression, | 
| Douglas Gregor | c7b6d88 | 2010-09-16 15:14:18 +0000 | [diff] [blame] | 4771 | bool IsSuper) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4772 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 4773 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | 70c5ac7 | 2010-09-20 23:34:21 +0000 | [diff] [blame] | 4774 | AddClassMessageCompletions(*this, S, Receiver, SelIdents, NumSelIdents, | 
|  | 4775 | AtArgumentExpression, IsSuper, Results); | 
| Douglas Gregor | b9d7757 | 2010-09-21 00:03:25 +0000 | [diff] [blame] | 4776 |  | 
|  | 4777 | // If we're actually at the argument expression (rather than prior to the | 
|  | 4778 | // selector), we're actually performing code completion for an expression. | 
|  | 4779 | // Determine whether we have a single, best method. If so, we can | 
|  | 4780 | // code-complete the expression using the corresponding parameter type as | 
|  | 4781 | // our preferred type, improving completion results. | 
|  | 4782 | if (AtArgumentExpression) { | 
|  | 4783 | QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results, | 
|  | 4784 | NumSelIdents); | 
|  | 4785 | if (PreferredType.isNull()) | 
|  | 4786 | CodeCompleteOrdinaryName(S, PCC_Expression); | 
|  | 4787 | else | 
|  | 4788 | CodeCompleteExpression(S, PreferredType); | 
|  | 4789 | return; | 
|  | 4790 | } | 
|  | 4791 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 4792 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 4793 | CodeCompletionContext::CCC_Other, | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4794 | Results.data(), Results.size()); | 
| Steve Naroff | c4df6d2 | 2009-11-07 02:08:14 +0000 | [diff] [blame] | 4795 | } | 
|  | 4796 |  | 
| Douglas Gregor | d3c6854 | 2009-11-19 01:08:35 +0000 | [diff] [blame] | 4797 | void Sema::CodeCompleteObjCInstanceMessage(Scope *S, ExprTy *Receiver, | 
|  | 4798 | IdentifierInfo **SelIdents, | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4799 | unsigned NumSelIdents, | 
| Douglas Gregor | 70c5ac7 | 2010-09-20 23:34:21 +0000 | [diff] [blame] | 4800 | bool AtArgumentExpression, | 
| Douglas Gregor | 6b0656a | 2010-10-13 21:24:53 +0000 | [diff] [blame] | 4801 | ObjCInterfaceDecl *Super) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 4802 | typedef CodeCompletionResult Result; | 
| Steve Naroff | c4df6d2 | 2009-11-07 02:08:14 +0000 | [diff] [blame] | 4803 |  | 
|  | 4804 | Expr *RecExpr = static_cast<Expr *>(Receiver); | 
| Steve Naroff | c4df6d2 | 2009-11-07 02:08:14 +0000 | [diff] [blame] | 4805 |  | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4806 | // If necessary, apply function/array conversion to the receiver. | 
|  | 4807 | // C99 6.7.5.3p[7,8]. | 
| Douglas Gregor | 78edf51 | 2010-09-15 16:23:04 +0000 | [diff] [blame] | 4808 | if (RecExpr) | 
|  | 4809 | DefaultFunctionArrayLvalueConversion(RecExpr); | 
| Douglas Gregor | 6b0656a | 2010-10-13 21:24:53 +0000 | [diff] [blame] | 4810 | QualType ReceiverType = RecExpr? RecExpr->getType() | 
|  | 4811 | : Super? Context.getObjCObjectPointerType( | 
|  | 4812 | Context.getObjCInterfaceType(Super)) | 
|  | 4813 | : Context.getObjCIdType(); | 
| Steve Naroff | c4df6d2 | 2009-11-07 02:08:14 +0000 | [diff] [blame] | 4814 |  | 
| Douglas Gregor | da89264 | 2010-11-08 21:12:30 +0000 | [diff] [blame] | 4815 | // If we're messaging an expression with type "id" or "Class", check | 
|  | 4816 | // whether we know something special about the receiver that allows | 
|  | 4817 | // us to assume a more-specific receiver type. | 
|  | 4818 | if (ReceiverType->isObjCIdType() || ReceiverType->isObjCClassType()) | 
|  | 4819 | if (ObjCInterfaceDecl *IFace = GetAssumedMessageSendExprType(RecExpr)) { | 
|  | 4820 | if (ReceiverType->isObjCClassType()) | 
|  | 4821 | return CodeCompleteObjCClassMessage(S, | 
|  | 4822 | ParsedType::make(Context.getObjCInterfaceType(IFace)), | 
|  | 4823 | SelIdents, NumSelIdents, | 
|  | 4824 | AtArgumentExpression, Super); | 
|  | 4825 |  | 
|  | 4826 | ReceiverType = Context.getObjCObjectPointerType( | 
|  | 4827 | Context.getObjCInterfaceType(IFace)); | 
|  | 4828 | } | 
|  | 4829 |  | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4830 | // Build the set of methods we can see. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4831 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 4832 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4833 | Results.EnterNewScope(); | 
| Douglas Gregor | 22f5699 | 2010-04-06 19:22:33 +0000 | [diff] [blame] | 4834 |  | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4835 | // If this is a send-to-super, try to add the special "super" send | 
|  | 4836 | // completion. | 
| Douglas Gregor | 6b0656a | 2010-10-13 21:24:53 +0000 | [diff] [blame] | 4837 | if (Super) { | 
| Douglas Gregor | 03d8aec | 2010-08-27 15:10:57 +0000 | [diff] [blame] | 4838 | if (ObjCMethodDecl *SuperMethod | 
|  | 4839 | = AddSuperSendCompletion(*this, false, SelIdents, NumSelIdents, | 
|  | 4840 | Results)) | 
|  | 4841 | Results.Ignore(SuperMethod); | 
|  | 4842 | } | 
|  | 4843 |  | 
| Douglas Gregor | 265f749 | 2010-08-27 15:29:55 +0000 | [diff] [blame] | 4844 | // If we're inside an Objective-C method definition, prefer its selector to | 
|  | 4845 | // others. | 
|  | 4846 | if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) | 
|  | 4847 | Results.setPreferredSelector(CurMethod->getSelector()); | 
| Douglas Gregor | 36ecb04 | 2009-11-17 23:22:23 +0000 | [diff] [blame] | 4848 |  | 
| Douglas Gregor | d36adf5 | 2010-09-16 16:06:31 +0000 | [diff] [blame] | 4849 | // Keep track of the selectors we've already added. | 
|  | 4850 | VisitedSelectorSet Selectors; | 
|  | 4851 |  | 
| Douglas Gregor | f74a419 | 2009-11-18 00:06:18 +0000 | [diff] [blame] | 4852 | // Handle messages to Class. This really isn't a message to an instance | 
|  | 4853 | // method, so we treat it the same way we would treat a message send to a | 
|  | 4854 | // class method. | 
|  | 4855 | if (ReceiverType->isObjCClassType() || | 
|  | 4856 | ReceiverType->isObjCQualifiedClassType()) { | 
|  | 4857 | if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) { | 
|  | 4858 | if (ObjCInterfaceDecl *ClassDecl = CurMethod->getClassInterface()) | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4859 | AddObjCMethods(ClassDecl, false, MK_Any, SelIdents, NumSelIdents, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4860 | CurContext, Selectors, AtArgumentExpression, Results); | 
| Douglas Gregor | f74a419 | 2009-11-18 00:06:18 +0000 | [diff] [blame] | 4861 | } | 
|  | 4862 | } | 
|  | 4863 | // Handle messages to a qualified ID ("id<foo>"). | 
|  | 4864 | else if (const ObjCObjectPointerType *QualID | 
|  | 4865 | = ReceiverType->getAsObjCQualifiedIdType()) { | 
|  | 4866 | // Search protocols for instance methods. | 
|  | 4867 | for (ObjCObjectPointerType::qual_iterator I = QualID->qual_begin(), | 
|  | 4868 | E = QualID->qual_end(); | 
|  | 4869 | I != E; ++I) | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4870 | AddObjCMethods(*I, true, MK_Any, SelIdents, NumSelIdents, CurContext, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4871 | Selectors, AtArgumentExpression, Results); | 
| Douglas Gregor | f74a419 | 2009-11-18 00:06:18 +0000 | [diff] [blame] | 4872 | } | 
|  | 4873 | // Handle messages to a pointer to interface type. | 
|  | 4874 | else if (const ObjCObjectPointerType *IFacePtr | 
|  | 4875 | = ReceiverType->getAsObjCInterfacePointerType()) { | 
|  | 4876 | // Search the class, its superclasses, etc., for instance methods. | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4877 | AddObjCMethods(IFacePtr->getInterfaceDecl(), true, MK_Any, SelIdents, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4878 | NumSelIdents, CurContext, Selectors, AtArgumentExpression, | 
|  | 4879 | Results); | 
| Douglas Gregor | f74a419 | 2009-11-18 00:06:18 +0000 | [diff] [blame] | 4880 |  | 
|  | 4881 | // Search protocols for instance methods. | 
|  | 4882 | for (ObjCObjectPointerType::qual_iterator I = IFacePtr->qual_begin(), | 
|  | 4883 | E = IFacePtr->qual_end(); | 
|  | 4884 | I != E; ++I) | 
| Douglas Gregor | 4ad9685 | 2009-11-19 07:41:15 +0000 | [diff] [blame] | 4885 | AddObjCMethods(*I, true, MK_Any, SelIdents, NumSelIdents, CurContext, | 
| Douglas Gregor | cf54426 | 2010-11-17 21:36:08 +0000 | [diff] [blame] | 4886 | Selectors, AtArgumentExpression, Results); | 
| Douglas Gregor | f74a419 | 2009-11-18 00:06:18 +0000 | [diff] [blame] | 4887 | } | 
| Douglas Gregor | 13438f9 | 2010-04-06 16:40:00 +0000 | [diff] [blame] | 4888 | // Handle messages to "id". | 
|  | 4889 | else if (ReceiverType->isObjCIdType()) { | 
| Douglas Gregor | 719770d | 2010-04-06 17:30:22 +0000 | [diff] [blame] | 4890 | // We're messaging "id", so provide all instance methods we know | 
|  | 4891 | // about as code-completion results. | 
|  | 4892 |  | 
|  | 4893 | // If we have an external source, load the entire class method | 
| Sebastian Redl | 3c7f413 | 2010-08-18 23:57:06 +0000 | [diff] [blame] | 4894 | // pool from the AST file. | 
| Douglas Gregor | 719770d | 2010-04-06 17:30:22 +0000 | [diff] [blame] | 4895 | if (ExternalSource) { | 
| John McCall | 76bd1f3 | 2010-06-01 09:23:16 +0000 | [diff] [blame] | 4896 | for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors(); | 
|  | 4897 | I != N; ++I) { | 
|  | 4898 | Selector Sel = ExternalSource->GetExternalSelector(I); | 
| Sebastian Redl | db9d214 | 2010-08-02 23:18:59 +0000 | [diff] [blame] | 4899 | if (Sel.isNull() || MethodPool.count(Sel)) | 
| Douglas Gregor | 719770d | 2010-04-06 17:30:22 +0000 | [diff] [blame] | 4900 | continue; | 
|  | 4901 |  | 
| Sebastian Redl | db9d214 | 2010-08-02 23:18:59 +0000 | [diff] [blame] | 4902 | ReadMethodPool(Sel); | 
| Douglas Gregor | 719770d | 2010-04-06 17:30:22 +0000 | [diff] [blame] | 4903 | } | 
|  | 4904 | } | 
|  | 4905 |  | 
| Sebastian Redl | db9d214 | 2010-08-02 23:18:59 +0000 | [diff] [blame] | 4906 | for (GlobalMethodPool::iterator M = MethodPool.begin(), | 
|  | 4907 | MEnd = MethodPool.end(); | 
|  | 4908 | M != MEnd; ++M) { | 
|  | 4909 | for (ObjCMethodList *MethList = &M->second.first; | 
|  | 4910 | MethList && MethList->Method; | 
| Douglas Gregor | 13438f9 | 2010-04-06 16:40:00 +0000 | [diff] [blame] | 4911 | MethList = MethList->Next) { | 
|  | 4912 | if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents, | 
|  | 4913 | NumSelIdents)) | 
|  | 4914 | continue; | 
| Douglas Gregor | d36adf5 | 2010-09-16 16:06:31 +0000 | [diff] [blame] | 4915 |  | 
|  | 4916 | if (!Selectors.insert(MethList->Method->getSelector())) | 
|  | 4917 | continue; | 
|  | 4918 |  | 
| Douglas Gregor | 13438f9 | 2010-04-06 16:40:00 +0000 | [diff] [blame] | 4919 | Result R(MethList->Method, 0); | 
|  | 4920 | R.StartParameter = NumSelIdents; | 
|  | 4921 | R.AllParametersAreInformative = false; | 
|  | 4922 | Results.MaybeAddResult(R, CurContext); | 
|  | 4923 | } | 
|  | 4924 | } | 
|  | 4925 | } | 
| Steve Naroff | c4df6d2 | 2009-11-07 02:08:14 +0000 | [diff] [blame] | 4926 | Results.ExitScope(); | 
| Douglas Gregor | b9d7757 | 2010-09-21 00:03:25 +0000 | [diff] [blame] | 4927 |  | 
|  | 4928 |  | 
|  | 4929 | // If we're actually at the argument expression (rather than prior to the | 
|  | 4930 | // selector), we're actually performing code completion for an expression. | 
|  | 4931 | // Determine whether we have a single, best method. If so, we can | 
|  | 4932 | // code-complete the expression using the corresponding parameter type as | 
|  | 4933 | // our preferred type, improving completion results. | 
|  | 4934 | if (AtArgumentExpression) { | 
|  | 4935 | QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results, | 
|  | 4936 | NumSelIdents); | 
|  | 4937 | if (PreferredType.isNull()) | 
|  | 4938 | CodeCompleteOrdinaryName(S, PCC_Expression); | 
|  | 4939 | else | 
|  | 4940 | CodeCompleteExpression(S, PreferredType); | 
|  | 4941 | return; | 
|  | 4942 | } | 
|  | 4943 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 4944 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 4945 | CodeCompletionContext::CCC_Other, | 
|  | 4946 | Results.data(),Results.size()); | 
| Steve Naroff | c4df6d2 | 2009-11-07 02:08:14 +0000 | [diff] [blame] | 4947 | } | 
| Douglas Gregor | 55385fe | 2009-11-18 04:19:12 +0000 | [diff] [blame] | 4948 |  | 
| Douglas Gregor | fb62941 | 2010-08-23 21:17:50 +0000 | [diff] [blame] | 4949 | void Sema::CodeCompleteObjCForCollection(Scope *S, | 
|  | 4950 | DeclGroupPtrTy IterationVar) { | 
|  | 4951 | CodeCompleteExpressionData Data; | 
|  | 4952 | Data.ObjCCollection = true; | 
|  | 4953 |  | 
|  | 4954 | if (IterationVar.getAsOpaquePtr()) { | 
|  | 4955 | DeclGroupRef DG = IterationVar.getAsVal<DeclGroupRef>(); | 
|  | 4956 | for (DeclGroupRef::iterator I = DG.begin(), End = DG.end(); I != End; ++I) { | 
|  | 4957 | if (*I) | 
|  | 4958 | Data.IgnoreDecls.push_back(*I); | 
|  | 4959 | } | 
|  | 4960 | } | 
|  | 4961 |  | 
|  | 4962 | CodeCompleteExpression(S, Data); | 
|  | 4963 | } | 
|  | 4964 |  | 
| Douglas Gregor | 458433d | 2010-08-26 15:07:07 +0000 | [diff] [blame] | 4965 | void Sema::CodeCompleteObjCSelector(Scope *S, IdentifierInfo **SelIdents, | 
|  | 4966 | unsigned NumSelIdents) { | 
|  | 4967 | // If we have an external source, load the entire class method | 
|  | 4968 | // pool from the AST file. | 
|  | 4969 | if (ExternalSource) { | 
|  | 4970 | for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors(); | 
|  | 4971 | I != N; ++I) { | 
|  | 4972 | Selector Sel = ExternalSource->GetExternalSelector(I); | 
|  | 4973 | if (Sel.isNull() || MethodPool.count(Sel)) | 
|  | 4974 | continue; | 
|  | 4975 |  | 
|  | 4976 | ReadMethodPool(Sel); | 
|  | 4977 | } | 
|  | 4978 | } | 
|  | 4979 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4980 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 4981 | CodeCompletionContext::CCC_SelectorName); | 
| Douglas Gregor | 458433d | 2010-08-26 15:07:07 +0000 | [diff] [blame] | 4982 | Results.EnterNewScope(); | 
|  | 4983 | for (GlobalMethodPool::iterator M = MethodPool.begin(), | 
|  | 4984 | MEnd = MethodPool.end(); | 
|  | 4985 | M != MEnd; ++M) { | 
|  | 4986 |  | 
|  | 4987 | Selector Sel = M->first; | 
|  | 4988 | if (!isAcceptableObjCSelector(Sel, MK_Any, SelIdents, NumSelIdents)) | 
|  | 4989 | continue; | 
|  | 4990 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4991 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
| Douglas Gregor | 458433d | 2010-08-26 15:07:07 +0000 | [diff] [blame] | 4992 | if (Sel.isUnarySelector()) { | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 4993 | Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 4994 | Sel.getIdentifierInfoForSlot(0)->getName())); | 
|  | 4995 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | 458433d | 2010-08-26 15:07:07 +0000 | [diff] [blame] | 4996 | continue; | 
|  | 4997 | } | 
|  | 4998 |  | 
| Douglas Gregor | 2d9e21f | 2010-08-26 16:46:39 +0000 | [diff] [blame] | 4999 | std::string Accumulator; | 
| Douglas Gregor | 458433d | 2010-08-26 15:07:07 +0000 | [diff] [blame] | 5000 | for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I) { | 
| Douglas Gregor | 2d9e21f | 2010-08-26 16:46:39 +0000 | [diff] [blame] | 5001 | if (I == NumSelIdents) { | 
|  | 5002 | if (!Accumulator.empty()) { | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 5003 | Builder.AddInformativeChunk(Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5004 | Accumulator)); | 
| Douglas Gregor | 2d9e21f | 2010-08-26 16:46:39 +0000 | [diff] [blame] | 5005 | Accumulator.clear(); | 
|  | 5006 | } | 
|  | 5007 | } | 
|  | 5008 |  | 
|  | 5009 | Accumulator += Sel.getIdentifierInfoForSlot(I)->getName().str(); | 
|  | 5010 | Accumulator += ':'; | 
| Douglas Gregor | 458433d | 2010-08-26 15:07:07 +0000 | [diff] [blame] | 5011 | } | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 5012 | Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( Accumulator)); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5013 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | 458433d | 2010-08-26 15:07:07 +0000 | [diff] [blame] | 5014 | } | 
|  | 5015 | Results.ExitScope(); | 
|  | 5016 |  | 
|  | 5017 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 5018 | CodeCompletionContext::CCC_SelectorName, | 
|  | 5019 | Results.data(), Results.size()); | 
|  | 5020 | } | 
|  | 5021 |  | 
| Douglas Gregor | 55385fe | 2009-11-18 04:19:12 +0000 | [diff] [blame] | 5022 | /// \brief Add all of the protocol declarations that we find in the given | 
|  | 5023 | /// (translation unit) context. | 
|  | 5024 | static void AddProtocolResults(DeclContext *Ctx, DeclContext *CurContext, | 
| Douglas Gregor | 083128f | 2009-11-18 04:49:41 +0000 | [diff] [blame] | 5025 | bool OnlyForwardDeclarations, | 
| Douglas Gregor | 55385fe | 2009-11-18 04:19:12 +0000 | [diff] [blame] | 5026 | ResultBuilder &Results) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 5027 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 55385fe | 2009-11-18 04:19:12 +0000 | [diff] [blame] | 5028 |  | 
|  | 5029 | for (DeclContext::decl_iterator D = Ctx->decls_begin(), | 
|  | 5030 | DEnd = Ctx->decls_end(); | 
|  | 5031 | D != DEnd; ++D) { | 
|  | 5032 | // Record any protocols we find. | 
|  | 5033 | if (ObjCProtocolDecl *Proto = dyn_cast<ObjCProtocolDecl>(*D)) | 
| Douglas Gregor | 083128f | 2009-11-18 04:49:41 +0000 | [diff] [blame] | 5034 | if (!OnlyForwardDeclarations || Proto->isForwardDecl()) | 
| Douglas Gregor | 608300b | 2010-01-14 16:14:35 +0000 | [diff] [blame] | 5035 | Results.AddResult(Result(Proto, 0), CurContext, 0, false); | 
| Douglas Gregor | 55385fe | 2009-11-18 04:19:12 +0000 | [diff] [blame] | 5036 |  | 
|  | 5037 | // Record any forward-declared protocols we find. | 
|  | 5038 | if (ObjCForwardProtocolDecl *Forward | 
|  | 5039 | = dyn_cast<ObjCForwardProtocolDecl>(*D)) { | 
|  | 5040 | for (ObjCForwardProtocolDecl::protocol_iterator | 
|  | 5041 | P = Forward->protocol_begin(), | 
|  | 5042 | PEnd = Forward->protocol_end(); | 
|  | 5043 | P != PEnd; ++P) | 
| Douglas Gregor | 083128f | 2009-11-18 04:49:41 +0000 | [diff] [blame] | 5044 | if (!OnlyForwardDeclarations || (*P)->isForwardDecl()) | 
| Douglas Gregor | 608300b | 2010-01-14 16:14:35 +0000 | [diff] [blame] | 5045 | Results.AddResult(Result(*P, 0), CurContext, 0, false); | 
| Douglas Gregor | 55385fe | 2009-11-18 04:19:12 +0000 | [diff] [blame] | 5046 | } | 
|  | 5047 | } | 
|  | 5048 | } | 
|  | 5049 |  | 
|  | 5050 | void Sema::CodeCompleteObjCProtocolReferences(IdentifierLocPair *Protocols, | 
|  | 5051 | unsigned NumProtocols) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5052 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 5053 | CodeCompletionContext::CCC_ObjCProtocolName); | 
| Douglas Gregor | 55385fe | 2009-11-18 04:19:12 +0000 | [diff] [blame] | 5054 |  | 
| Douglas Gregor | 70c2335 | 2010-12-09 21:44:02 +0000 | [diff] [blame] | 5055 | if (CodeCompleter && CodeCompleter->includeGlobals()) { | 
|  | 5056 | Results.EnterNewScope(); | 
|  | 5057 |  | 
|  | 5058 | // Tell the result set to ignore all of the protocols we have | 
|  | 5059 | // already seen. | 
|  | 5060 | // FIXME: This doesn't work when caching code-completion results. | 
|  | 5061 | for (unsigned I = 0; I != NumProtocols; ++I) | 
|  | 5062 | if (ObjCProtocolDecl *Protocol = LookupProtocol(Protocols[I].first, | 
|  | 5063 | Protocols[I].second)) | 
|  | 5064 | Results.Ignore(Protocol); | 
| Douglas Gregor | 55385fe | 2009-11-18 04:19:12 +0000 | [diff] [blame] | 5065 |  | 
| Douglas Gregor | 70c2335 | 2010-12-09 21:44:02 +0000 | [diff] [blame] | 5066 | // Add all protocols. | 
|  | 5067 | AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, false, | 
|  | 5068 | Results); | 
| Douglas Gregor | 083128f | 2009-11-18 04:49:41 +0000 | [diff] [blame] | 5069 |  | 
| Douglas Gregor | 70c2335 | 2010-12-09 21:44:02 +0000 | [diff] [blame] | 5070 | Results.ExitScope(); | 
|  | 5071 | } | 
|  | 5072 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 5073 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 5074 | CodeCompletionContext::CCC_ObjCProtocolName, | 
|  | 5075 | Results.data(),Results.size()); | 
| Douglas Gregor | 083128f | 2009-11-18 04:49:41 +0000 | [diff] [blame] | 5076 | } | 
|  | 5077 |  | 
|  | 5078 | void Sema::CodeCompleteObjCProtocolDecl(Scope *) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5079 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 5080 | CodeCompletionContext::CCC_ObjCProtocolName); | 
| Douglas Gregor | 083128f | 2009-11-18 04:49:41 +0000 | [diff] [blame] | 5081 |  | 
| Douglas Gregor | 70c2335 | 2010-12-09 21:44:02 +0000 | [diff] [blame] | 5082 | if (CodeCompleter && CodeCompleter->includeGlobals()) { | 
|  | 5083 | Results.EnterNewScope(); | 
|  | 5084 |  | 
|  | 5085 | // Add all protocols. | 
|  | 5086 | AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, true, | 
|  | 5087 | Results); | 
| Douglas Gregor | 55385fe | 2009-11-18 04:19:12 +0000 | [diff] [blame] | 5088 |  | 
| Douglas Gregor | 70c2335 | 2010-12-09 21:44:02 +0000 | [diff] [blame] | 5089 | Results.ExitScope(); | 
|  | 5090 | } | 
|  | 5091 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 5092 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 5093 | CodeCompletionContext::CCC_ObjCProtocolName, | 
|  | 5094 | Results.data(),Results.size()); | 
| Douglas Gregor | 55385fe | 2009-11-18 04:19:12 +0000 | [diff] [blame] | 5095 | } | 
| Douglas Gregor | 3b49aca | 2009-11-18 16:26:39 +0000 | [diff] [blame] | 5096 |  | 
|  | 5097 | /// \brief Add all of the Objective-C interface declarations that we find in | 
|  | 5098 | /// the given (translation unit) context. | 
|  | 5099 | static void AddInterfaceResults(DeclContext *Ctx, DeclContext *CurContext, | 
|  | 5100 | bool OnlyForwardDeclarations, | 
|  | 5101 | bool OnlyUnimplemented, | 
|  | 5102 | ResultBuilder &Results) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 5103 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 3b49aca | 2009-11-18 16:26:39 +0000 | [diff] [blame] | 5104 |  | 
|  | 5105 | for (DeclContext::decl_iterator D = Ctx->decls_begin(), | 
|  | 5106 | DEnd = Ctx->decls_end(); | 
|  | 5107 | D != DEnd; ++D) { | 
| Douglas Gregor | deacbdc | 2010-08-11 12:19:30 +0000 | [diff] [blame] | 5108 | // Record any interfaces we find. | 
|  | 5109 | if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(*D)) | 
|  | 5110 | if ((!OnlyForwardDeclarations || Class->isForwardDecl()) && | 
|  | 5111 | (!OnlyUnimplemented || !Class->getImplementation())) | 
|  | 5112 | Results.AddResult(Result(Class, 0), CurContext, 0, false); | 
| Douglas Gregor | 3b49aca | 2009-11-18 16:26:39 +0000 | [diff] [blame] | 5113 |  | 
|  | 5114 | // Record any forward-declared interfaces we find. | 
|  | 5115 | if (ObjCClassDecl *Forward = dyn_cast<ObjCClassDecl>(*D)) { | 
|  | 5116 | for (ObjCClassDecl::iterator C = Forward->begin(), CEnd = Forward->end(); | 
| Douglas Gregor | deacbdc | 2010-08-11 12:19:30 +0000 | [diff] [blame] | 5117 | C != CEnd; ++C) | 
|  | 5118 | if ((!OnlyForwardDeclarations || C->getInterface()->isForwardDecl()) && | 
|  | 5119 | (!OnlyUnimplemented || !C->getInterface()->getImplementation())) | 
|  | 5120 | Results.AddResult(Result(C->getInterface(), 0), CurContext, | 
| Douglas Gregor | 608300b | 2010-01-14 16:14:35 +0000 | [diff] [blame] | 5121 | 0, false); | 
| Douglas Gregor | 3b49aca | 2009-11-18 16:26:39 +0000 | [diff] [blame] | 5122 | } | 
|  | 5123 | } | 
|  | 5124 | } | 
|  | 5125 |  | 
|  | 5126 | void Sema::CodeCompleteObjCInterfaceDecl(Scope *S) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5127 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 5128 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | 3b49aca | 2009-11-18 16:26:39 +0000 | [diff] [blame] | 5129 | Results.EnterNewScope(); | 
|  | 5130 |  | 
|  | 5131 | // Add all classes. | 
|  | 5132 | AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, true, | 
|  | 5133 | false, Results); | 
|  | 5134 |  | 
|  | 5135 | Results.ExitScope(); | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 5136 | // FIXME: Add a special context for this, use cached global completion | 
|  | 5137 | // results. | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 5138 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 5139 | CodeCompletionContext::CCC_Other, | 
|  | 5140 | Results.data(),Results.size()); | 
| Douglas Gregor | 3b49aca | 2009-11-18 16:26:39 +0000 | [diff] [blame] | 5141 | } | 
|  | 5142 |  | 
| Douglas Gregor | c83c687 | 2010-04-15 22:33:43 +0000 | [diff] [blame] | 5143 | void Sema::CodeCompleteObjCSuperclass(Scope *S, IdentifierInfo *ClassName, | 
|  | 5144 | SourceLocation ClassNameLoc) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5145 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 5146 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | 3b49aca | 2009-11-18 16:26:39 +0000 | [diff] [blame] | 5147 | Results.EnterNewScope(); | 
|  | 5148 |  | 
|  | 5149 | // Make sure that we ignore the class we're currently defining. | 
|  | 5150 | NamedDecl *CurClass | 
| Douglas Gregor | c83c687 | 2010-04-15 22:33:43 +0000 | [diff] [blame] | 5151 | = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName); | 
| Douglas Gregor | 33ced0b | 2009-11-18 19:08:43 +0000 | [diff] [blame] | 5152 | if (CurClass && isa<ObjCInterfaceDecl>(CurClass)) | 
| Douglas Gregor | 3b49aca | 2009-11-18 16:26:39 +0000 | [diff] [blame] | 5153 | Results.Ignore(CurClass); | 
|  | 5154 |  | 
|  | 5155 | // Add all classes. | 
|  | 5156 | AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false, | 
|  | 5157 | false, Results); | 
|  | 5158 |  | 
|  | 5159 | Results.ExitScope(); | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 5160 | // FIXME: Add a special context for this, use cached global completion | 
|  | 5161 | // results. | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 5162 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 5163 | CodeCompletionContext::CCC_Other, | 
|  | 5164 | Results.data(),Results.size()); | 
| Douglas Gregor | 3b49aca | 2009-11-18 16:26:39 +0000 | [diff] [blame] | 5165 | } | 
|  | 5166 |  | 
|  | 5167 | void Sema::CodeCompleteObjCImplementationDecl(Scope *S) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5168 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 5169 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | 3b49aca | 2009-11-18 16:26:39 +0000 | [diff] [blame] | 5170 | Results.EnterNewScope(); | 
|  | 5171 |  | 
|  | 5172 | // Add all unimplemented classes. | 
|  | 5173 | AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false, | 
|  | 5174 | true, Results); | 
|  | 5175 |  | 
|  | 5176 | Results.ExitScope(); | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 5177 | // FIXME: Add a special context for this, use cached global completion | 
|  | 5178 | // results. | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 5179 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 5180 | CodeCompletionContext::CCC_Other, | 
|  | 5181 | Results.data(),Results.size()); | 
| Douglas Gregor | 3b49aca | 2009-11-18 16:26:39 +0000 | [diff] [blame] | 5182 | } | 
| Douglas Gregor | 33ced0b | 2009-11-18 19:08:43 +0000 | [diff] [blame] | 5183 |  | 
|  | 5184 | void Sema::CodeCompleteObjCInterfaceCategory(Scope *S, | 
| Douglas Gregor | c83c687 | 2010-04-15 22:33:43 +0000 | [diff] [blame] | 5185 | IdentifierInfo *ClassName, | 
|  | 5186 | SourceLocation ClassNameLoc) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 5187 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 33ced0b | 2009-11-18 19:08:43 +0000 | [diff] [blame] | 5188 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5189 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 5190 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | 33ced0b | 2009-11-18 19:08:43 +0000 | [diff] [blame] | 5191 |  | 
|  | 5192 | // Ignore any categories we find that have already been implemented by this | 
|  | 5193 | // interface. | 
|  | 5194 | llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames; | 
|  | 5195 | NamedDecl *CurClass | 
| Douglas Gregor | c83c687 | 2010-04-15 22:33:43 +0000 | [diff] [blame] | 5196 | = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName); | 
| Douglas Gregor | 33ced0b | 2009-11-18 19:08:43 +0000 | [diff] [blame] | 5197 | if (ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass)) | 
|  | 5198 | for (ObjCCategoryDecl *Category = Class->getCategoryList(); Category; | 
|  | 5199 | Category = Category->getNextClassCategory()) | 
|  | 5200 | CategoryNames.insert(Category->getIdentifier()); | 
|  | 5201 |  | 
|  | 5202 | // Add all of the categories we know about. | 
|  | 5203 | Results.EnterNewScope(); | 
|  | 5204 | TranslationUnitDecl *TU = Context.getTranslationUnitDecl(); | 
|  | 5205 | for (DeclContext::decl_iterator D = TU->decls_begin(), | 
|  | 5206 | DEnd = TU->decls_end(); | 
|  | 5207 | D != DEnd; ++D) | 
|  | 5208 | if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(*D)) | 
|  | 5209 | if (CategoryNames.insert(Category->getIdentifier())) | 
| Douglas Gregor | 608300b | 2010-01-14 16:14:35 +0000 | [diff] [blame] | 5210 | Results.AddResult(Result(Category, 0), CurContext, 0, false); | 
| Douglas Gregor | 33ced0b | 2009-11-18 19:08:43 +0000 | [diff] [blame] | 5211 | Results.ExitScope(); | 
|  | 5212 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 5213 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 5214 | CodeCompletionContext::CCC_Other, | 
|  | 5215 | Results.data(),Results.size()); | 
| Douglas Gregor | 33ced0b | 2009-11-18 19:08:43 +0000 | [diff] [blame] | 5216 | } | 
|  | 5217 |  | 
|  | 5218 | void Sema::CodeCompleteObjCImplementationCategory(Scope *S, | 
| Douglas Gregor | c83c687 | 2010-04-15 22:33:43 +0000 | [diff] [blame] | 5219 | IdentifierInfo *ClassName, | 
|  | 5220 | SourceLocation ClassNameLoc) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 5221 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 33ced0b | 2009-11-18 19:08:43 +0000 | [diff] [blame] | 5222 |  | 
|  | 5223 | // Find the corresponding interface. If we couldn't find the interface, the | 
|  | 5224 | // program itself is ill-formed. However, we'll try to be helpful still by | 
|  | 5225 | // providing the list of all of the categories we know about. | 
|  | 5226 | NamedDecl *CurClass | 
| Douglas Gregor | c83c687 | 2010-04-15 22:33:43 +0000 | [diff] [blame] | 5227 | = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName); | 
| Douglas Gregor | 33ced0b | 2009-11-18 19:08:43 +0000 | [diff] [blame] | 5228 | ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass); | 
|  | 5229 | if (!Class) | 
| Douglas Gregor | c83c687 | 2010-04-15 22:33:43 +0000 | [diff] [blame] | 5230 | return CodeCompleteObjCInterfaceCategory(S, ClassName, ClassNameLoc); | 
| Douglas Gregor | 33ced0b | 2009-11-18 19:08:43 +0000 | [diff] [blame] | 5231 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5232 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 5233 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | 33ced0b | 2009-11-18 19:08:43 +0000 | [diff] [blame] | 5234 |  | 
|  | 5235 | // Add all of the categories that have have corresponding interface | 
|  | 5236 | // declarations in this class and any of its superclasses, except for | 
|  | 5237 | // already-implemented categories in the class itself. | 
|  | 5238 | llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames; | 
|  | 5239 | Results.EnterNewScope(); | 
|  | 5240 | bool IgnoreImplemented = true; | 
|  | 5241 | while (Class) { | 
|  | 5242 | for (ObjCCategoryDecl *Category = Class->getCategoryList(); Category; | 
|  | 5243 | Category = Category->getNextClassCategory()) | 
|  | 5244 | if ((!IgnoreImplemented || !Category->getImplementation()) && | 
|  | 5245 | CategoryNames.insert(Category->getIdentifier())) | 
| Douglas Gregor | 608300b | 2010-01-14 16:14:35 +0000 | [diff] [blame] | 5246 | Results.AddResult(Result(Category, 0), CurContext, 0, false); | 
| Douglas Gregor | 33ced0b | 2009-11-18 19:08:43 +0000 | [diff] [blame] | 5247 |  | 
|  | 5248 | Class = Class->getSuperClass(); | 
|  | 5249 | IgnoreImplemented = false; | 
|  | 5250 | } | 
|  | 5251 | Results.ExitScope(); | 
|  | 5252 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 5253 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 5254 | CodeCompletionContext::CCC_Other, | 
|  | 5255 | Results.data(),Results.size()); | 
| Douglas Gregor | 33ced0b | 2009-11-18 19:08:43 +0000 | [diff] [blame] | 5256 | } | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 5257 |  | 
| John McCall | d226f65 | 2010-08-21 09:40:31 +0000 | [diff] [blame] | 5258 | void Sema::CodeCompleteObjCPropertyDefinition(Scope *S, Decl *ObjCImpDecl) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 5259 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5260 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 5261 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 5262 |  | 
|  | 5263 | // Figure out where this @synthesize lives. | 
|  | 5264 | ObjCContainerDecl *Container | 
| John McCall | d226f65 | 2010-08-21 09:40:31 +0000 | [diff] [blame] | 5265 | = dyn_cast_or_null<ObjCContainerDecl>(ObjCImpDecl); | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 5266 | if (!Container || | 
|  | 5267 | (!isa<ObjCImplementationDecl>(Container) && | 
|  | 5268 | !isa<ObjCCategoryImplDecl>(Container))) | 
|  | 5269 | return; | 
|  | 5270 |  | 
|  | 5271 | // Ignore any properties that have already been implemented. | 
|  | 5272 | for (DeclContext::decl_iterator D = Container->decls_begin(), | 
|  | 5273 | DEnd = Container->decls_end(); | 
|  | 5274 | D != DEnd; ++D) | 
|  | 5275 | if (ObjCPropertyImplDecl *PropertyImpl = dyn_cast<ObjCPropertyImplDecl>(*D)) | 
|  | 5276 | Results.Ignore(PropertyImpl->getPropertyDecl()); | 
|  | 5277 |  | 
|  | 5278 | // Add any properties that we find. | 
| Douglas Gregor | 7344921 | 2010-12-09 23:01:55 +0000 | [diff] [blame] | 5279 | AddedPropertiesSet AddedProperties; | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 5280 | Results.EnterNewScope(); | 
|  | 5281 | if (ObjCImplementationDecl *ClassImpl | 
|  | 5282 | = dyn_cast<ObjCImplementationDecl>(Container)) | 
|  | 5283 | AddObjCProperties(ClassImpl->getClassInterface(), false, CurContext, | 
| Douglas Gregor | 7344921 | 2010-12-09 23:01:55 +0000 | [diff] [blame] | 5284 | AddedProperties, Results); | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 5285 | else | 
|  | 5286 | AddObjCProperties(cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl(), | 
| Douglas Gregor | 7344921 | 2010-12-09 23:01:55 +0000 | [diff] [blame] | 5287 | false, CurContext, AddedProperties, Results); | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 5288 | Results.ExitScope(); | 
|  | 5289 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 5290 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 5291 | CodeCompletionContext::CCC_Other, | 
|  | 5292 | Results.data(),Results.size()); | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 5293 | } | 
|  | 5294 |  | 
|  | 5295 | void Sema::CodeCompleteObjCPropertySynthesizeIvar(Scope *S, | 
|  | 5296 | IdentifierInfo *PropertyName, | 
| John McCall | d226f65 | 2010-08-21 09:40:31 +0000 | [diff] [blame] | 5297 | Decl *ObjCImpDecl) { | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 5298 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5299 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 5300 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 5301 |  | 
|  | 5302 | // Figure out where this @synthesize lives. | 
|  | 5303 | ObjCContainerDecl *Container | 
| John McCall | d226f65 | 2010-08-21 09:40:31 +0000 | [diff] [blame] | 5304 | = dyn_cast_or_null<ObjCContainerDecl>(ObjCImpDecl); | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 5305 | if (!Container || | 
|  | 5306 | (!isa<ObjCImplementationDecl>(Container) && | 
|  | 5307 | !isa<ObjCCategoryImplDecl>(Container))) | 
|  | 5308 | return; | 
|  | 5309 |  | 
|  | 5310 | // Figure out which interface we're looking into. | 
|  | 5311 | ObjCInterfaceDecl *Class = 0; | 
|  | 5312 | if (ObjCImplementationDecl *ClassImpl | 
|  | 5313 | = dyn_cast<ObjCImplementationDecl>(Container)) | 
|  | 5314 | Class = ClassImpl->getClassInterface(); | 
|  | 5315 | else | 
|  | 5316 | Class = cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl() | 
|  | 5317 | ->getClassInterface(); | 
|  | 5318 |  | 
|  | 5319 | // Add all of the instance variables in this class and its superclasses. | 
|  | 5320 | Results.EnterNewScope(); | 
|  | 5321 | for(; Class; Class = Class->getSuperClass()) { | 
|  | 5322 | // FIXME: We could screen the type of each ivar for compatibility with | 
|  | 5323 | // the property, but is that being too paternal? | 
|  | 5324 | for (ObjCInterfaceDecl::ivar_iterator IVar = Class->ivar_begin(), | 
|  | 5325 | IVarEnd = Class->ivar_end(); | 
|  | 5326 | IVar != IVarEnd; ++IVar) | 
| Douglas Gregor | 608300b | 2010-01-14 16:14:35 +0000 | [diff] [blame] | 5327 | Results.AddResult(Result(*IVar, 0), CurContext, 0, false); | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 5328 | } | 
|  | 5329 | Results.ExitScope(); | 
|  | 5330 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 5331 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 5332 | CodeCompletionContext::CCC_Other, | 
|  | 5333 | Results.data(),Results.size()); | 
| Douglas Gregor | 322328b | 2009-11-18 22:32:06 +0000 | [diff] [blame] | 5334 | } | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5335 |  | 
| Douglas Gregor | 408be5a | 2010-08-25 01:08:01 +0000 | [diff] [blame] | 5336 | // Mapping from selectors to the methods that implement that selector, along | 
|  | 5337 | // with the "in original class" flag. | 
|  | 5338 | typedef llvm::DenseMap<Selector, std::pair<ObjCMethodDecl *, bool> > | 
|  | 5339 | KnownMethodsMap; | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5340 |  | 
|  | 5341 | /// \brief Find all of the methods that reside in the given container | 
|  | 5342 | /// (and its superclasses, protocols, etc.) that meet the given | 
|  | 5343 | /// criteria. Insert those methods into the map of known methods, | 
|  | 5344 | /// indexed by selector so they can be easily found. | 
|  | 5345 | static void FindImplementableMethods(ASTContext &Context, | 
|  | 5346 | ObjCContainerDecl *Container, | 
|  | 5347 | bool WantInstanceMethods, | 
|  | 5348 | QualType ReturnType, | 
| Douglas Gregor | 408be5a | 2010-08-25 01:08:01 +0000 | [diff] [blame] | 5349 | KnownMethodsMap &KnownMethods, | 
|  | 5350 | bool InOriginalClass = true) { | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5351 | if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)) { | 
|  | 5352 | // Recurse into protocols. | 
|  | 5353 | const ObjCList<ObjCProtocolDecl> &Protocols | 
|  | 5354 | = IFace->getReferencedProtocols(); | 
|  | 5355 | for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(), | 
| Douglas Gregor | ea76618 | 2010-10-18 18:21:28 +0000 | [diff] [blame] | 5356 | E = Protocols.end(); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5357 | I != E; ++I) | 
|  | 5358 | FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType, | 
| Douglas Gregor | ea76618 | 2010-10-18 18:21:28 +0000 | [diff] [blame] | 5359 | KnownMethods, InOriginalClass); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5360 |  | 
| Douglas Gregor | ea76618 | 2010-10-18 18:21:28 +0000 | [diff] [blame] | 5361 | // Add methods from any class extensions and categories. | 
|  | 5362 | for (const ObjCCategoryDecl *Cat = IFace->getCategoryList(); Cat; | 
|  | 5363 | Cat = Cat->getNextClassCategory()) | 
| Fariborz Jahanian | 80aa1cd | 2010-06-22 23:20:40 +0000 | [diff] [blame] | 5364 | FindImplementableMethods(Context, const_cast<ObjCCategoryDecl*>(Cat), | 
|  | 5365 | WantInstanceMethods, ReturnType, | 
| Douglas Gregor | ea76618 | 2010-10-18 18:21:28 +0000 | [diff] [blame] | 5366 | KnownMethods, false); | 
|  | 5367 |  | 
|  | 5368 | // Visit the superclass. | 
|  | 5369 | if (IFace->getSuperClass()) | 
|  | 5370 | FindImplementableMethods(Context, IFace->getSuperClass(), | 
|  | 5371 | WantInstanceMethods, ReturnType, | 
|  | 5372 | KnownMethods, false); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5373 | } | 
|  | 5374 |  | 
|  | 5375 | if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) { | 
|  | 5376 | // Recurse into protocols. | 
|  | 5377 | const ObjCList<ObjCProtocolDecl> &Protocols | 
|  | 5378 | = Category->getReferencedProtocols(); | 
|  | 5379 | for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(), | 
| Douglas Gregor | ea76618 | 2010-10-18 18:21:28 +0000 | [diff] [blame] | 5380 | E = Protocols.end(); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5381 | I != E; ++I) | 
|  | 5382 | FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType, | 
| Douglas Gregor | ea76618 | 2010-10-18 18:21:28 +0000 | [diff] [blame] | 5383 | KnownMethods, InOriginalClass); | 
|  | 5384 |  | 
|  | 5385 | // If this category is the original class, jump to the interface. | 
|  | 5386 | if (InOriginalClass && Category->getClassInterface()) | 
|  | 5387 | FindImplementableMethods(Context, Category->getClassInterface(), | 
|  | 5388 | WantInstanceMethods, ReturnType, KnownMethods, | 
|  | 5389 | false); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5390 | } | 
|  | 5391 |  | 
|  | 5392 | if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) { | 
|  | 5393 | // Recurse into protocols. | 
|  | 5394 | const ObjCList<ObjCProtocolDecl> &Protocols | 
|  | 5395 | = Protocol->getReferencedProtocols(); | 
|  | 5396 | for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(), | 
|  | 5397 | E = Protocols.end(); | 
|  | 5398 | I != E; ++I) | 
|  | 5399 | FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType, | 
| Douglas Gregor | ea76618 | 2010-10-18 18:21:28 +0000 | [diff] [blame] | 5400 | KnownMethods, false); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5401 | } | 
|  | 5402 |  | 
|  | 5403 | // Add methods in this container. This operation occurs last because | 
|  | 5404 | // we want the methods from this container to override any methods | 
|  | 5405 | // we've previously seen with the same selector. | 
|  | 5406 | for (ObjCContainerDecl::method_iterator M = Container->meth_begin(), | 
|  | 5407 | MEnd = Container->meth_end(); | 
|  | 5408 | M != MEnd; ++M) { | 
|  | 5409 | if ((*M)->isInstanceMethod() == WantInstanceMethods) { | 
|  | 5410 | if (!ReturnType.isNull() && | 
|  | 5411 | !Context.hasSameUnqualifiedType(ReturnType, (*M)->getResultType())) | 
|  | 5412 | continue; | 
|  | 5413 |  | 
| Douglas Gregor | 408be5a | 2010-08-25 01:08:01 +0000 | [diff] [blame] | 5414 | KnownMethods[(*M)->getSelector()] = std::make_pair(*M, InOriginalClass); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5415 | } | 
|  | 5416 | } | 
|  | 5417 | } | 
|  | 5418 |  | 
|  | 5419 | void Sema::CodeCompleteObjCMethodDecl(Scope *S, | 
|  | 5420 | bool IsInstanceMethod, | 
| John McCall | b3d8748 | 2010-08-24 05:47:05 +0000 | [diff] [blame] | 5421 | ParsedType ReturnTy, | 
| John McCall | d226f65 | 2010-08-21 09:40:31 +0000 | [diff] [blame] | 5422 | Decl *IDecl) { | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5423 | // Determine the return type of the method we're declaring, if | 
|  | 5424 | // provided. | 
|  | 5425 | QualType ReturnType = GetTypeFromParser(ReturnTy); | 
|  | 5426 |  | 
| Douglas Gregor | ea76618 | 2010-10-18 18:21:28 +0000 | [diff] [blame] | 5427 | // Determine where we should start searching for methods. | 
|  | 5428 | ObjCContainerDecl *SearchDecl = 0; | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5429 | bool IsInImplementation = false; | 
| John McCall | d226f65 | 2010-08-21 09:40:31 +0000 | [diff] [blame] | 5430 | if (Decl *D = IDecl) { | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5431 | if (ObjCImplementationDecl *Impl = dyn_cast<ObjCImplementationDecl>(D)) { | 
|  | 5432 | SearchDecl = Impl->getClassInterface(); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5433 | IsInImplementation = true; | 
|  | 5434 | } else if (ObjCCategoryImplDecl *CatImpl | 
| Douglas Gregor | ea76618 | 2010-10-18 18:21:28 +0000 | [diff] [blame] | 5435 | = dyn_cast<ObjCCategoryImplDecl>(D)) { | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5436 | SearchDecl = CatImpl->getCategoryDecl(); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5437 | IsInImplementation = true; | 
| Douglas Gregor | ea76618 | 2010-10-18 18:21:28 +0000 | [diff] [blame] | 5438 | } else | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5439 | SearchDecl = dyn_cast<ObjCContainerDecl>(D); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5440 | } | 
|  | 5441 |  | 
|  | 5442 | if (!SearchDecl && S) { | 
| Douglas Gregor | ea76618 | 2010-10-18 18:21:28 +0000 | [diff] [blame] | 5443 | if (DeclContext *DC = static_cast<DeclContext *>(S->getEntity())) | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5444 | SearchDecl = dyn_cast<ObjCContainerDecl>(DC); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5445 | } | 
|  | 5446 |  | 
| Douglas Gregor | ea76618 | 2010-10-18 18:21:28 +0000 | [diff] [blame] | 5447 | if (!SearchDecl) { | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 5448 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 5449 | CodeCompletionContext::CCC_Other, | 
|  | 5450 | 0, 0); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5451 | return; | 
|  | 5452 | } | 
|  | 5453 |  | 
|  | 5454 | // Find all of the methods that we could declare/implement here. | 
|  | 5455 | KnownMethodsMap KnownMethods; | 
|  | 5456 | FindImplementableMethods(Context, SearchDecl, IsInstanceMethod, | 
| Douglas Gregor | ea76618 | 2010-10-18 18:21:28 +0000 | [diff] [blame] | 5457 | ReturnType, KnownMethods); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5458 |  | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5459 | // Add declarations or definitions for each of the known methods. | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 5460 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5461 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 5462 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5463 | Results.EnterNewScope(); | 
|  | 5464 | PrintingPolicy Policy(Context.PrintingPolicy); | 
|  | 5465 | Policy.AnonymousTagLocations = false; | 
|  | 5466 | for (KnownMethodsMap::iterator M = KnownMethods.begin(), | 
|  | 5467 | MEnd = KnownMethods.end(); | 
|  | 5468 | M != MEnd; ++M) { | 
| Douglas Gregor | 408be5a | 2010-08-25 01:08:01 +0000 | [diff] [blame] | 5469 | ObjCMethodDecl *Method = M->second.first; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5470 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5471 |  | 
|  | 5472 | // If the result type was not already provided, add it to the | 
|  | 5473 | // pattern as (type). | 
|  | 5474 | if (ReturnType.isNull()) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5475 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
| Douglas Gregor | a63f6de | 2011-02-01 21:15:40 +0000 | [diff] [blame] | 5476 | Builder.AddTextChunk(GetCompletionTypeString(Method->getResultType(), | 
|  | 5477 | Context, | 
|  | 5478 | Builder.getAllocator())); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5479 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5480 | } | 
|  | 5481 |  | 
|  | 5482 | Selector Sel = Method->getSelector(); | 
|  | 5483 |  | 
|  | 5484 | // Add the first part of the selector to the pattern. | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 5485 | Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5486 | Sel.getIdentifierInfoForSlot(0)->getName())); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5487 |  | 
|  | 5488 | // Add parameters to the pattern. | 
|  | 5489 | unsigned I = 0; | 
|  | 5490 | for (ObjCMethodDecl::param_iterator P = Method->param_begin(), | 
|  | 5491 | PEnd = Method->param_end(); | 
|  | 5492 | P != PEnd; (void)++P, ++I) { | 
|  | 5493 | // Add the part of the selector name. | 
|  | 5494 | if (I == 0) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5495 | Builder.AddTypedTextChunk(":"); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5496 | else if (I < Sel.getNumArgs()) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5497 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5498 | Builder.AddTypedTextChunk( | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 5499 | Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5500 | (Sel.getIdentifierInfoForSlot(I)->getName() | 
|  | 5501 | + ":").str())); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5502 | } else | 
|  | 5503 | break; | 
|  | 5504 |  | 
|  | 5505 | // Add the parameter type. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5506 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
| Douglas Gregor | a63f6de | 2011-02-01 21:15:40 +0000 | [diff] [blame] | 5507 | Builder.AddTextChunk(GetCompletionTypeString((*P)->getOriginalType(), | 
|  | 5508 | Context, | 
|  | 5509 | Builder.getAllocator())); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5510 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5511 |  | 
|  | 5512 | if (IdentifierInfo *Id = (*P)->getIdentifier()) | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 5513 | Builder.AddTextChunk(Builder.getAllocator().CopyString( Id->getName())); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5514 | } | 
|  | 5515 |  | 
|  | 5516 | if (Method->isVariadic()) { | 
|  | 5517 | if (Method->param_size() > 0) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5518 | Builder.AddChunk(CodeCompletionString::CK_Comma); | 
|  | 5519 | Builder.AddTextChunk("..."); | 
| Douglas Gregor | e17794f | 2010-08-31 05:13:43 +0000 | [diff] [blame] | 5520 | } | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5521 |  | 
| Douglas Gregor | 447107d | 2010-05-28 00:57:46 +0000 | [diff] [blame] | 5522 | if (IsInImplementation && Results.includeCodePatterns()) { | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5523 | // We will be defining the method here, so add a compound statement. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5524 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5525 | Builder.AddChunk(CodeCompletionString::CK_LeftBrace); | 
|  | 5526 | Builder.AddChunk(CodeCompletionString::CK_VerticalSpace); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5527 | if (!Method->getResultType()->isVoidType()) { | 
|  | 5528 | // If the result type is not void, add a return clause. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5529 | Builder.AddTextChunk("return"); | 
|  | 5530 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5531 | Builder.AddPlaceholderChunk("expression"); | 
|  | 5532 | Builder.AddChunk(CodeCompletionString::CK_SemiColon); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5533 | } else | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5534 | Builder.AddPlaceholderChunk("statements"); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5535 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5536 | Builder.AddChunk(CodeCompletionString::CK_VerticalSpace); | 
|  | 5537 | Builder.AddChunk(CodeCompletionString::CK_RightBrace); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5538 | } | 
|  | 5539 |  | 
| Douglas Gregor | 408be5a | 2010-08-25 01:08:01 +0000 | [diff] [blame] | 5540 | unsigned Priority = CCP_CodePattern; | 
|  | 5541 | if (!M->second.second) | 
|  | 5542 | Priority += CCD_InBaseClass; | 
|  | 5543 |  | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5544 | Results.AddResult(Result(Builder.TakeString(), Priority, | 
| Douglas Gregor | 16ed9ad | 2010-08-17 16:06:07 +0000 | [diff] [blame] | 5545 | Method->isInstanceMethod() | 
|  | 5546 | ? CXCursor_ObjCInstanceMethodDecl | 
|  | 5547 | : CXCursor_ObjCClassMethodDecl)); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5548 | } | 
|  | 5549 |  | 
|  | 5550 | Results.ExitScope(); | 
|  | 5551 |  | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 5552 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 5553 | CodeCompletionContext::CCC_Other, | 
|  | 5554 | Results.data(),Results.size()); | 
| Douglas Gregor | e8f5a17 | 2010-04-07 00:21:17 +0000 | [diff] [blame] | 5555 | } | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 5556 |  | 
|  | 5557 | void Sema::CodeCompleteObjCMethodDeclSelector(Scope *S, | 
|  | 5558 | bool IsInstanceMethod, | 
| Douglas Gregor | 40ed9a1 | 2010-07-08 23:37:41 +0000 | [diff] [blame] | 5559 | bool AtParameterName, | 
| John McCall | b3d8748 | 2010-08-24 05:47:05 +0000 | [diff] [blame] | 5560 | ParsedType ReturnTy, | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 5561 | IdentifierInfo **SelIdents, | 
|  | 5562 | unsigned NumSelIdents) { | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 5563 | // If we have an external source, load the entire class method | 
| Sebastian Redl | 3c7f413 | 2010-08-18 23:57:06 +0000 | [diff] [blame] | 5564 | // pool from the AST file. | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 5565 | if (ExternalSource) { | 
|  | 5566 | for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors(); | 
|  | 5567 | I != N; ++I) { | 
|  | 5568 | Selector Sel = ExternalSource->GetExternalSelector(I); | 
| Sebastian Redl | db9d214 | 2010-08-02 23:18:59 +0000 | [diff] [blame] | 5569 | if (Sel.isNull() || MethodPool.count(Sel)) | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 5570 | continue; | 
| Sebastian Redl | db9d214 | 2010-08-02 23:18:59 +0000 | [diff] [blame] | 5571 |  | 
|  | 5572 | ReadMethodPool(Sel); | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 5573 | } | 
|  | 5574 | } | 
|  | 5575 |  | 
|  | 5576 | // Build the set of methods we can see. | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 5577 | typedef CodeCompletionResult Result; | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5578 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
|  | 5579 | CodeCompletionContext::CCC_Other); | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 5580 |  | 
|  | 5581 | if (ReturnTy) | 
|  | 5582 | Results.setPreferredType(GetTypeFromParser(ReturnTy).getNonReferenceType()); | 
| Sebastian Redl | db9d214 | 2010-08-02 23:18:59 +0000 | [diff] [blame] | 5583 |  | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 5584 | Results.EnterNewScope(); | 
| Sebastian Redl | db9d214 | 2010-08-02 23:18:59 +0000 | [diff] [blame] | 5585 | for (GlobalMethodPool::iterator M = MethodPool.begin(), | 
|  | 5586 | MEnd = MethodPool.end(); | 
|  | 5587 | M != MEnd; ++M) { | 
|  | 5588 | for (ObjCMethodList *MethList = IsInstanceMethod ? &M->second.first : | 
|  | 5589 | &M->second.second; | 
|  | 5590 | MethList && MethList->Method; | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 5591 | MethList = MethList->Next) { | 
|  | 5592 | if (!isAcceptableObjCMethod(MethList->Method, MK_Any, SelIdents, | 
|  | 5593 | NumSelIdents)) | 
|  | 5594 | continue; | 
|  | 5595 |  | 
| Douglas Gregor | 40ed9a1 | 2010-07-08 23:37:41 +0000 | [diff] [blame] | 5596 | if (AtParameterName) { | 
|  | 5597 | // Suggest parameter names we've seen before. | 
|  | 5598 | if (NumSelIdents && NumSelIdents <= MethList->Method->param_size()) { | 
|  | 5599 | ParmVarDecl *Param = MethList->Method->param_begin()[NumSelIdents-1]; | 
|  | 5600 | if (Param->getIdentifier()) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5601 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 5602 | Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5603 | Param->getIdentifier()->getName())); | 
|  | 5604 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | 40ed9a1 | 2010-07-08 23:37:41 +0000 | [diff] [blame] | 5605 | } | 
|  | 5606 | } | 
|  | 5607 |  | 
|  | 5608 | continue; | 
|  | 5609 | } | 
|  | 5610 |  | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 5611 | Result R(MethList->Method, 0); | 
|  | 5612 | R.StartParameter = NumSelIdents; | 
|  | 5613 | R.AllParametersAreInformative = false; | 
|  | 5614 | R.DeclaringEntity = true; | 
|  | 5615 | Results.MaybeAddResult(R, CurContext); | 
|  | 5616 | } | 
|  | 5617 | } | 
|  | 5618 |  | 
|  | 5619 | Results.ExitScope(); | 
| Douglas Gregor | e6b1bb6 | 2010-08-11 21:23:17 +0000 | [diff] [blame] | 5620 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 5621 | CodeCompletionContext::CCC_Other, | 
|  | 5622 | Results.data(),Results.size()); | 
| Douglas Gregor | 1f5537a | 2010-07-08 23:20:03 +0000 | [diff] [blame] | 5623 | } | 
| Douglas Gregor | 87c08a5 | 2010-08-13 22:48:40 +0000 | [diff] [blame] | 5624 |  | 
| Douglas Gregor | f29c523 | 2010-08-24 22:20:20 +0000 | [diff] [blame] | 5625 | void Sema::CodeCompletePreprocessorDirective(bool InConditional) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5626 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 5627 | CodeCompletionContext::CCC_PreprocessorDirective); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5628 | Results.EnterNewScope(); | 
|  | 5629 |  | 
|  | 5630 | // #if <condition> | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5631 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
|  | 5632 | Builder.AddTypedTextChunk("if"); | 
|  | 5633 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5634 | Builder.AddPlaceholderChunk("condition"); | 
|  | 5635 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5636 |  | 
|  | 5637 | // #ifdef <macro> | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5638 | Builder.AddTypedTextChunk("ifdef"); | 
|  | 5639 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5640 | Builder.AddPlaceholderChunk("macro"); | 
|  | 5641 | Results.AddResult(Builder.TakeString()); | 
|  | 5642 |  | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5643 | // #ifndef <macro> | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5644 | Builder.AddTypedTextChunk("ifndef"); | 
|  | 5645 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5646 | Builder.AddPlaceholderChunk("macro"); | 
|  | 5647 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5648 |  | 
|  | 5649 | if (InConditional) { | 
|  | 5650 | // #elif <condition> | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5651 | Builder.AddTypedTextChunk("elif"); | 
|  | 5652 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5653 | Builder.AddPlaceholderChunk("condition"); | 
|  | 5654 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5655 |  | 
|  | 5656 | // #else | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5657 | Builder.AddTypedTextChunk("else"); | 
|  | 5658 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5659 |  | 
|  | 5660 | // #endif | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5661 | Builder.AddTypedTextChunk("endif"); | 
|  | 5662 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5663 | } | 
|  | 5664 |  | 
|  | 5665 | // #include "header" | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5666 | Builder.AddTypedTextChunk("include"); | 
|  | 5667 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5668 | Builder.AddTextChunk("\""); | 
|  | 5669 | Builder.AddPlaceholderChunk("header"); | 
|  | 5670 | Builder.AddTextChunk("\""); | 
|  | 5671 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5672 |  | 
|  | 5673 | // #include <header> | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5674 | Builder.AddTypedTextChunk("include"); | 
|  | 5675 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5676 | Builder.AddTextChunk("<"); | 
|  | 5677 | Builder.AddPlaceholderChunk("header"); | 
|  | 5678 | Builder.AddTextChunk(">"); | 
|  | 5679 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5680 |  | 
|  | 5681 | // #define <macro> | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5682 | Builder.AddTypedTextChunk("define"); | 
|  | 5683 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5684 | Builder.AddPlaceholderChunk("macro"); | 
|  | 5685 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5686 |  | 
|  | 5687 | // #define <macro>(<args>) | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5688 | Builder.AddTypedTextChunk("define"); | 
|  | 5689 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5690 | Builder.AddPlaceholderChunk("macro"); | 
|  | 5691 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 5692 | Builder.AddPlaceholderChunk("args"); | 
|  | 5693 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 5694 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5695 |  | 
|  | 5696 | // #undef <macro> | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5697 | Builder.AddTypedTextChunk("undef"); | 
|  | 5698 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5699 | Builder.AddPlaceholderChunk("macro"); | 
|  | 5700 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5701 |  | 
|  | 5702 | // #line <number> | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5703 | Builder.AddTypedTextChunk("line"); | 
|  | 5704 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5705 | Builder.AddPlaceholderChunk("number"); | 
|  | 5706 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5707 |  | 
|  | 5708 | // #line <number> "filename" | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5709 | Builder.AddTypedTextChunk("line"); | 
|  | 5710 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5711 | Builder.AddPlaceholderChunk("number"); | 
|  | 5712 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5713 | Builder.AddTextChunk("\""); | 
|  | 5714 | Builder.AddPlaceholderChunk("filename"); | 
|  | 5715 | Builder.AddTextChunk("\""); | 
|  | 5716 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5717 |  | 
|  | 5718 | // #error <message> | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5719 | Builder.AddTypedTextChunk("error"); | 
|  | 5720 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5721 | Builder.AddPlaceholderChunk("message"); | 
|  | 5722 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5723 |  | 
|  | 5724 | // #pragma <arguments> | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5725 | Builder.AddTypedTextChunk("pragma"); | 
|  | 5726 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5727 | Builder.AddPlaceholderChunk("arguments"); | 
|  | 5728 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5729 |  | 
|  | 5730 | if (getLangOptions().ObjC1) { | 
|  | 5731 | // #import "header" | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5732 | Builder.AddTypedTextChunk("import"); | 
|  | 5733 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5734 | Builder.AddTextChunk("\""); | 
|  | 5735 | Builder.AddPlaceholderChunk("header"); | 
|  | 5736 | Builder.AddTextChunk("\""); | 
|  | 5737 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5738 |  | 
|  | 5739 | // #import <header> | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5740 | Builder.AddTypedTextChunk("import"); | 
|  | 5741 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5742 | Builder.AddTextChunk("<"); | 
|  | 5743 | Builder.AddPlaceholderChunk("header"); | 
|  | 5744 | Builder.AddTextChunk(">"); | 
|  | 5745 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5746 | } | 
|  | 5747 |  | 
|  | 5748 | // #include_next "header" | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5749 | Builder.AddTypedTextChunk("include_next"); | 
|  | 5750 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5751 | Builder.AddTextChunk("\""); | 
|  | 5752 | Builder.AddPlaceholderChunk("header"); | 
|  | 5753 | Builder.AddTextChunk("\""); | 
|  | 5754 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5755 |  | 
|  | 5756 | // #include_next <header> | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5757 | Builder.AddTypedTextChunk("include_next"); | 
|  | 5758 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5759 | Builder.AddTextChunk("<"); | 
|  | 5760 | Builder.AddPlaceholderChunk("header"); | 
|  | 5761 | Builder.AddTextChunk(">"); | 
|  | 5762 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5763 |  | 
|  | 5764 | // #warning <message> | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5765 | Builder.AddTypedTextChunk("warning"); | 
|  | 5766 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5767 | Builder.AddPlaceholderChunk("message"); | 
|  | 5768 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5769 |  | 
|  | 5770 | // Note: #ident and #sccs are such crazy anachronisms that we don't provide | 
|  | 5771 | // completions for them. And __include_macros is a Clang-internal extension | 
|  | 5772 | // that we don't want to encourage anyone to use. | 
|  | 5773 |  | 
|  | 5774 | // FIXME: we don't support #assert or #unassert, so don't suggest them. | 
|  | 5775 | Results.ExitScope(); | 
|  | 5776 |  | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5777 | HandleCodeCompleteResults(this, CodeCompleter, | 
| Douglas Gregor | 721f359 | 2010-08-25 18:41:16 +0000 | [diff] [blame] | 5778 | CodeCompletionContext::CCC_PreprocessorDirective, | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5779 | Results.data(), Results.size()); | 
|  | 5780 | } | 
|  | 5781 |  | 
|  | 5782 | void Sema::CodeCompleteInPreprocessorConditionalExclusion(Scope *S) { | 
| Douglas Gregor | f29c523 | 2010-08-24 22:20:20 +0000 | [diff] [blame] | 5783 | CodeCompleteOrdinaryName(S, | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 5784 | S->getFnParent()? Sema::PCC_RecoveryInFunction | 
|  | 5785 | : Sema::PCC_Namespace); | 
| Douglas Gregor | f44e854 | 2010-08-24 19:08:16 +0000 | [diff] [blame] | 5786 | } | 
|  | 5787 |  | 
| Douglas Gregor | f29c523 | 2010-08-24 22:20:20 +0000 | [diff] [blame] | 5788 | void Sema::CodeCompletePreprocessorMacroName(bool IsDefinition) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5789 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 5790 | IsDefinition? CodeCompletionContext::CCC_MacroName | 
|  | 5791 | : CodeCompletionContext::CCC_MacroNameUse); | 
| Douglas Gregor | 1fbb447 | 2010-08-24 20:21:13 +0000 | [diff] [blame] | 5792 | if (!IsDefinition && (!CodeCompleter || CodeCompleter->includeMacros())) { | 
|  | 5793 | // Add just the names of macros, not their arguments. | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5794 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
| Douglas Gregor | 1fbb447 | 2010-08-24 20:21:13 +0000 | [diff] [blame] | 5795 | Results.EnterNewScope(); | 
|  | 5796 | for (Preprocessor::macro_iterator M = PP.macro_begin(), | 
|  | 5797 | MEnd = PP.macro_end(); | 
|  | 5798 | M != MEnd; ++M) { | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 5799 | Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5800 | M->first->getName())); | 
|  | 5801 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | 1fbb447 | 2010-08-24 20:21:13 +0000 | [diff] [blame] | 5802 | } | 
|  | 5803 | Results.ExitScope(); | 
|  | 5804 | } else if (IsDefinition) { | 
|  | 5805 | // FIXME: Can we detect when the user just wrote an include guard above? | 
|  | 5806 | } | 
|  | 5807 |  | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 5808 | HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(), | 
| Douglas Gregor | 1fbb447 | 2010-08-24 20:21:13 +0000 | [diff] [blame] | 5809 | Results.data(), Results.size()); | 
|  | 5810 | } | 
|  | 5811 |  | 
| Douglas Gregor | f29c523 | 2010-08-24 22:20:20 +0000 | [diff] [blame] | 5812 | void Sema::CodeCompletePreprocessorExpression() { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5813 | ResultBuilder Results(*this, CodeCompleter->getAllocator(), | 
| Douglas Gregor | 52779fb | 2010-09-23 23:01:17 +0000 | [diff] [blame] | 5814 | CodeCompletionContext::CCC_PreprocessorExpression); | 
| Douglas Gregor | f29c523 | 2010-08-24 22:20:20 +0000 | [diff] [blame] | 5815 |  | 
|  | 5816 | if (!CodeCompleter || CodeCompleter->includeMacros()) | 
|  | 5817 | AddMacroResults(PP, Results); | 
|  | 5818 |  | 
|  | 5819 | // defined (<macro>) | 
|  | 5820 | Results.EnterNewScope(); | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5821 | CodeCompletionBuilder Builder(Results.getAllocator()); | 
|  | 5822 | Builder.AddTypedTextChunk("defined"); | 
|  | 5823 | Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace); | 
|  | 5824 | Builder.AddChunk(CodeCompletionString::CK_LeftParen); | 
|  | 5825 | Builder.AddPlaceholderChunk("macro"); | 
|  | 5826 | Builder.AddChunk(CodeCompletionString::CK_RightParen); | 
|  | 5827 | Results.AddResult(Builder.TakeString()); | 
| Douglas Gregor | f29c523 | 2010-08-24 22:20:20 +0000 | [diff] [blame] | 5828 | Results.ExitScope(); | 
|  | 5829 |  | 
|  | 5830 | HandleCodeCompleteResults(this, CodeCompleter, | 
|  | 5831 | CodeCompletionContext::CCC_PreprocessorExpression, | 
|  | 5832 | Results.data(), Results.size()); | 
|  | 5833 | } | 
|  | 5834 |  | 
|  | 5835 | void Sema::CodeCompletePreprocessorMacroArgument(Scope *S, | 
|  | 5836 | IdentifierInfo *Macro, | 
|  | 5837 | MacroInfo *MacroInfo, | 
|  | 5838 | unsigned Argument) { | 
|  | 5839 | // FIXME: In the future, we could provide "overload" results, much like we | 
|  | 5840 | // do for function calls. | 
|  | 5841 |  | 
|  | 5842 | CodeCompleteOrdinaryName(S, | 
| John McCall | f312b1e | 2010-08-26 23:41:50 +0000 | [diff] [blame] | 5843 | S->getFnParent()? Sema::PCC_RecoveryInFunction | 
|  | 5844 | : Sema::PCC_Namespace); | 
| Douglas Gregor | f29c523 | 2010-08-24 22:20:20 +0000 | [diff] [blame] | 5845 | } | 
|  | 5846 |  | 
| Douglas Gregor | 55817af | 2010-08-25 17:04:25 +0000 | [diff] [blame] | 5847 | void Sema::CodeCompleteNaturalLanguage() { | 
| Douglas Gregor | 55817af | 2010-08-25 17:04:25 +0000 | [diff] [blame] | 5848 | HandleCodeCompleteResults(this, CodeCompleter, | 
| Douglas Gregor | af1c6b5 | 2010-08-25 17:10:00 +0000 | [diff] [blame] | 5849 | CodeCompletionContext::CCC_NaturalLanguage, | 
| Douglas Gregor | 55817af | 2010-08-25 17:04:25 +0000 | [diff] [blame] | 5850 | 0, 0); | 
|  | 5851 | } | 
|  | 5852 |  | 
| Douglas Gregor | dae6875 | 2011-02-01 22:57:45 +0000 | [diff] [blame] | 5853 | void Sema::GatherGlobalCodeCompletions(CodeCompletionAllocator &Allocator, | 
| John McCall | 0a2c5e2 | 2010-08-25 06:19:51 +0000 | [diff] [blame] | 5854 | llvm::SmallVectorImpl<CodeCompletionResult> &Results) { | 
| Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 5855 | ResultBuilder Builder(*this, Allocator, CodeCompletionContext::CCC_Recovery); | 
| Douglas Gregor | 8071e42 | 2010-08-15 06:18:01 +0000 | [diff] [blame] | 5856 | if (!CodeCompleter || CodeCompleter->includeGlobals()) { | 
|  | 5857 | CodeCompletionDeclConsumer Consumer(Builder, | 
|  | 5858 | Context.getTranslationUnitDecl()); | 
|  | 5859 | LookupVisibleDecls(Context.getTranslationUnitDecl(), LookupAnyName, | 
|  | 5860 | Consumer); | 
|  | 5861 | } | 
| Douglas Gregor | 87c08a5 | 2010-08-13 22:48:40 +0000 | [diff] [blame] | 5862 |  | 
|  | 5863 | if (!CodeCompleter || CodeCompleter->includeMacros()) | 
|  | 5864 | AddMacroResults(PP, Builder); | 
|  | 5865 |  | 
|  | 5866 | Results.clear(); | 
|  | 5867 | Results.insert(Results.end(), | 
|  | 5868 | Builder.data(), Builder.data() + Builder.size()); | 
|  | 5869 | } |