1//===--- DeclObjC.h - Classes for representing declarations -----*- 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 DeclObjC interface and subclasses.
11//
12//===----------------------------------------------------------------------===//
13
14#ifndef LLVM_CLANG_AST_DECLOBJC_H
15#define LLVM_CLANG_AST_DECLOBJC_H
16
17#include "clang/AST/Decl.h"
18#include "clang/AST/SelectorLocationsKind.h"
19#include "llvm/ADT/STLExtras.h"
20#include "llvm/Support/Compiler.h"
21
22namespace clang {
23class Expr;
24class Stmt;
25class FunctionDecl;
26class RecordDecl;
27class ObjCIvarDecl;
28class ObjCMethodDecl;
29class ObjCProtocolDecl;
30class ObjCCategoryDecl;
31class ObjCPropertyDecl;
32class ObjCPropertyImplDecl;
33class CXXCtorInitializer;
34
35class ObjCListBase {
36  ObjCListBase(const ObjCListBase &) LLVM_DELETED_FUNCTION;
37  void operator=(const ObjCListBase &) LLVM_DELETED_FUNCTION;
38protected:
39  /// List is an array of pointers to objects that are not owned by this object.
40  void **List;
41  unsigned NumElts;
42
43public:
44  ObjCListBase() : List(0), NumElts(0) {}
45  unsigned size() const { return NumElts; }
46  bool empty() const { return NumElts == 0; }
47
48protected:
49  void set(void *const* InList, unsigned Elts, ASTContext &Ctx);
50};
51
52
53/// ObjCList - This is a simple template class used to hold various lists of
54/// decls etc, which is heavily used by the ObjC front-end.  This only use case
55/// this supports is setting the list all at once and then reading elements out
56/// of it.
57template <typename T>
58class ObjCList : public ObjCListBase {
59public:
60  void set(T* const* InList, unsigned Elts, ASTContext &Ctx) {
61    ObjCListBase::set(reinterpret_cast<void*const*>(InList), Elts, Ctx);
62  }
63
64  typedef T* const * iterator;
65  iterator begin() const { return (iterator)List; }
66  iterator end() const { return (iterator)List+NumElts; }
67
68  T* operator[](unsigned Idx) const {
69    assert(Idx < NumElts && "Invalid access");
70    return (T*)List[Idx];
71  }
72};
73
74/// \brief A list of Objective-C protocols, along with the source
75/// locations at which they were referenced.
76class ObjCProtocolList : public ObjCList<ObjCProtocolDecl> {
77  SourceLocation *Locations;
78
79  using ObjCList<ObjCProtocolDecl>::set;
80
81public:
82  ObjCProtocolList() : ObjCList<ObjCProtocolDecl>(), Locations(0) { }
83
84  typedef const SourceLocation *loc_iterator;
85  loc_iterator loc_begin() const { return Locations; }
86  loc_iterator loc_end() const { return Locations + size(); }
87
88  void set(ObjCProtocolDecl* const* InList, unsigned Elts,
89           const SourceLocation *Locs, ASTContext &Ctx);
90};
91
92
93/// ObjCMethodDecl - Represents an instance or class method declaration.
94/// ObjC methods can be declared within 4 contexts: class interfaces,
95/// categories, protocols, and class implementations. While C++ member
96/// functions leverage C syntax, Objective-C method syntax is modeled after
97/// Smalltalk (using colons to specify argument types/expressions).
98/// Here are some brief examples:
99///
100/// Setter/getter instance methods:
101/// - (void)setMenu:(NSMenu *)menu;
102/// - (NSMenu *)menu;
103///
104/// Instance method that takes 2 NSView arguments:
105/// - (void)replaceSubview:(NSView *)oldView with:(NSView *)newView;
106///
107/// Getter class method:
108/// + (NSMenu *)defaultMenu;
109///
110/// A selector represents a unique name for a method. The selector names for
111/// the above methods are setMenu:, menu, replaceSubview:with:, and defaultMenu.
112///
113class ObjCMethodDecl : public NamedDecl, public DeclContext {
114public:
115  enum ImplementationControl { None, Required, Optional };
116private:
117  // The conventional meaning of this method; an ObjCMethodFamily.
118  // This is not serialized; instead, it is computed on demand and
119  // cached.
120  mutable unsigned Family : ObjCMethodFamilyBitWidth;
121
122  /// instance (true) or class (false) method.
123  unsigned IsInstance : 1;
124  unsigned IsVariadic : 1;
125
126  /// True if this method is the getter or setter for an explicit property.
127  unsigned IsPropertyAccessor : 1;
128
129  // Method has a definition.
130  unsigned IsDefined : 1;
131
132  /// \brief Method redeclaration in the same interface.
133  unsigned IsRedeclaration : 1;
134
135  /// \brief Is redeclared in the same interface.
136  mutable unsigned HasRedeclaration : 1;
137
138  // NOTE: VC++ treats enums as signed, avoid using ImplementationControl enum
139  /// \@required/\@optional
140  unsigned DeclImplementation : 2;
141
142  // NOTE: VC++ treats enums as signed, avoid using the ObjCDeclQualifier enum
143  /// in, inout, etc.
144  unsigned objcDeclQualifier : 6;
145
146  /// \brief Indicates whether this method has a related result type.
147  unsigned RelatedResultType : 1;
148
149  /// \brief Whether the locations of the selector identifiers are in a
150  /// "standard" position, a enum SelectorLocationsKind.
151  unsigned SelLocsKind : 2;
152
153  /// \brief Whether this method overrides any other in the class hierarchy.
154  ///
155  /// A method is said to override any method in the class's
156  /// base classes, its protocols, or its categories' protocols, that has
157  /// the same selector and is of the same kind (class or instance).
158  /// A method in an implementation is not considered as overriding the same
159  /// method in the interface or its categories.
160  unsigned IsOverriding : 1;
161
162  /// \brief Indicates if the method was a definition but its body was skipped.
163  unsigned HasSkippedBody : 1;
164
165  // Result type of this method.
166  QualType MethodDeclType;
167
168  // Type source information for the result type.
169  TypeSourceInfo *ResultTInfo;
170
171  /// \brief Array of ParmVarDecls for the formal parameters of this method
172  /// and optionally followed by selector locations.
173  void *ParamsAndSelLocs;
174  unsigned NumParams;
175
176  /// List of attributes for this method declaration.
177  SourceLocation DeclEndLoc; // the location of the ';' or '{'.
178
179  // The following are only used for method definitions, null otherwise.
180  LazyDeclStmtPtr Body;
181
182  /// SelfDecl - Decl for the implicit self parameter. This is lazily
183  /// constructed by createImplicitParams.
184  ImplicitParamDecl *SelfDecl;
185  /// CmdDecl - Decl for the implicit _cmd parameter. This is lazily
186  /// constructed by createImplicitParams.
187  ImplicitParamDecl *CmdDecl;
188
189  SelectorLocationsKind getSelLocsKind() const {
190    return (SelectorLocationsKind)SelLocsKind;
191  }
192  bool hasStandardSelLocs() const {
193    return getSelLocsKind() != SelLoc_NonStandard;
194  }
195
196  /// \brief Get a pointer to the stored selector identifiers locations array.
197  /// No locations will be stored if HasStandardSelLocs is true.
198  SourceLocation *getStoredSelLocs() {
199    return reinterpret_cast<SourceLocation*>(getParams() + NumParams);
200  }
201  const SourceLocation *getStoredSelLocs() const {
202    return reinterpret_cast<const SourceLocation*>(getParams() + NumParams);
203  }
204
205  /// \brief Get a pointer to the stored selector identifiers locations array.
206  /// No locations will be stored if HasStandardSelLocs is true.
207  ParmVarDecl **getParams() {
208    return reinterpret_cast<ParmVarDecl **>(ParamsAndSelLocs);
209  }
210  const ParmVarDecl *const *getParams() const {
211    return reinterpret_cast<const ParmVarDecl *const *>(ParamsAndSelLocs);
212  }
213
214  /// \brief Get the number of stored selector identifiers locations.
215  /// No locations will be stored if HasStandardSelLocs is true.
216  unsigned getNumStoredSelLocs() const {
217    if (hasStandardSelLocs())
218      return 0;
219    return getNumSelectorLocs();
220  }
221
222  void setParamsAndSelLocs(ASTContext &C,
223                           ArrayRef<ParmVarDecl*> Params,
224                           ArrayRef<SourceLocation> SelLocs);
225
226  ObjCMethodDecl(SourceLocation beginLoc, SourceLocation endLoc,
227                 Selector SelInfo, QualType T,
228                 TypeSourceInfo *ResultTInfo,
229                 DeclContext *contextDecl,
230                 bool isInstance = true,
231                 bool isVariadic = false,
232                 bool isPropertyAccessor = false,
233                 bool isImplicitlyDeclared = false,
234                 bool isDefined = false,
235                 ImplementationControl impControl = None,
236                 bool HasRelatedResultType = false)
237  : NamedDecl(ObjCMethod, contextDecl, beginLoc, SelInfo),
238    DeclContext(ObjCMethod), Family(InvalidObjCMethodFamily),
239    IsInstance(isInstance), IsVariadic(isVariadic),
240    IsPropertyAccessor(isPropertyAccessor),
241    IsDefined(isDefined), IsRedeclaration(0), HasRedeclaration(0),
242    DeclImplementation(impControl), objcDeclQualifier(OBJC_TQ_None),
243    RelatedResultType(HasRelatedResultType),
244    SelLocsKind(SelLoc_StandardNoSpace), IsOverriding(0), HasSkippedBody(0),
245    MethodDeclType(T), ResultTInfo(ResultTInfo),
246    ParamsAndSelLocs(0), NumParams(0),
247    DeclEndLoc(endLoc), Body(), SelfDecl(0), CmdDecl(0) {
248    setImplicit(isImplicitlyDeclared);
249  }
250
251  /// \brief A definition will return its interface declaration.
252  /// An interface declaration will return its definition.
253  /// Otherwise it will return itself.
254  virtual ObjCMethodDecl *getNextRedeclaration();
255
256public:
257  static ObjCMethodDecl *Create(ASTContext &C,
258                                SourceLocation beginLoc,
259                                SourceLocation endLoc,
260                                Selector SelInfo,
261                                QualType T,
262                                TypeSourceInfo *ResultTInfo,
263                                DeclContext *contextDecl,
264                                bool isInstance = true,
265                                bool isVariadic = false,
266                                bool isPropertyAccessor = false,
267                                bool isImplicitlyDeclared = false,
268                                bool isDefined = false,
269                                ImplementationControl impControl = None,
270                                bool HasRelatedResultType = false);
271
272  static ObjCMethodDecl *CreateDeserialized(ASTContext &C, unsigned ID);
273
274  virtual ObjCMethodDecl *getCanonicalDecl();
275  const ObjCMethodDecl *getCanonicalDecl() const {
276    return const_cast<ObjCMethodDecl*>(this)->getCanonicalDecl();
277  }
278
279  ObjCDeclQualifier getObjCDeclQualifier() const {
280    return ObjCDeclQualifier(objcDeclQualifier);
281  }
282  void setObjCDeclQualifier(ObjCDeclQualifier QV) { objcDeclQualifier = QV; }
283
284  /// \brief Determine whether this method has a result type that is related
285  /// to the message receiver's type.
286  bool hasRelatedResultType() const { return RelatedResultType; }
287
288  /// \brief Note whether this method has a related result type.
289  void SetRelatedResultType(bool RRT = true) { RelatedResultType = RRT; }
290
291  /// \brief True if this is a method redeclaration in the same interface.
292  bool isRedeclaration() const { return IsRedeclaration; }
293  void setAsRedeclaration(const ObjCMethodDecl *PrevMethod);
294
295  /// \brief Returns the location where the declarator ends. It will be
296  /// the location of ';' for a method declaration and the location of '{'
297  /// for a method definition.
298  SourceLocation getDeclaratorEndLoc() const { return DeclEndLoc; }
299
300  // Location information, modeled after the Stmt API.
301  SourceLocation getLocStart() const LLVM_READONLY { return getLocation(); }
302  SourceLocation getLocEnd() const LLVM_READONLY;
303  virtual SourceRange getSourceRange() const LLVM_READONLY {
304    return SourceRange(getLocation(), getLocEnd());
305  }
306
307  SourceLocation getSelectorStartLoc() const {
308    if (isImplicit())
309      return getLocStart();
310    return getSelectorLoc(0);
311  }
312  SourceLocation getSelectorLoc(unsigned Index) const {
313    assert(Index < getNumSelectorLocs() && "Index out of range!");
314    if (hasStandardSelLocs())
315      return getStandardSelectorLoc(Index, getSelector(),
316                                   getSelLocsKind() == SelLoc_StandardWithSpace,
317                      llvm::makeArrayRef(const_cast<ParmVarDecl**>(getParams()),
318                                         NumParams),
319                                   DeclEndLoc);
320    return getStoredSelLocs()[Index];
321  }
322
323  void getSelectorLocs(SmallVectorImpl<SourceLocation> &SelLocs) const;
324
325  unsigned getNumSelectorLocs() const {
326    if (isImplicit())
327      return 0;
328    Selector Sel = getSelector();
329    if (Sel.isUnarySelector())
330      return 1;
331    return Sel.getNumArgs();
332  }
333
334  ObjCInterfaceDecl *getClassInterface();
335  const ObjCInterfaceDecl *getClassInterface() const {
336    return const_cast<ObjCMethodDecl*>(this)->getClassInterface();
337  }
338
339  Selector getSelector() const { return getDeclName().getObjCSelector(); }
340
341  QualType getResultType() const { return MethodDeclType; }
342  void setResultType(QualType T) { MethodDeclType = T; }
343
344  /// \brief Determine the type of an expression that sends a message to this
345  /// function.
346  QualType getSendResultType() const {
347    return getResultType().getNonLValueExprType(getASTContext());
348  }
349
350  TypeSourceInfo *getResultTypeSourceInfo() const { return ResultTInfo; }
351  void setResultTypeSourceInfo(TypeSourceInfo *TInfo) { ResultTInfo = TInfo; }
352
353  // Iterator access to formal parameters.
354  unsigned param_size() const { return NumParams; }
355  typedef const ParmVarDecl *const *param_const_iterator;
356  typedef ParmVarDecl *const *param_iterator;
357  param_const_iterator param_begin() const { return getParams(); }
358  param_const_iterator param_end() const { return getParams() + NumParams; }
359  param_iterator param_begin() { return getParams(); }
360  param_iterator param_end() { return getParams() + NumParams; }
361  // This method returns and of the parameters which are part of the selector
362  // name mangling requirements.
363  param_const_iterator sel_param_end() const {
364    return param_begin() + getSelector().getNumArgs();
365  }
366
367  /// \brief Sets the method's parameters and selector source locations.
368  /// If the method is implicit (not coming from source) \p SelLocs is
369  /// ignored.
370  void setMethodParams(ASTContext &C,
371                       ArrayRef<ParmVarDecl*> Params,
372                       ArrayRef<SourceLocation> SelLocs =
373                           ArrayRef<SourceLocation>());
374
375  // Iterator access to parameter types.
376  typedef std::const_mem_fun_t<QualType, ParmVarDecl> deref_fun;
377  typedef llvm::mapped_iterator<param_const_iterator, deref_fun>
378      arg_type_iterator;
379
380  arg_type_iterator arg_type_begin() const {
381    return llvm::map_iterator(param_begin(), deref_fun(&ParmVarDecl::getType));
382  }
383  arg_type_iterator arg_type_end() const {
384    return llvm::map_iterator(param_end(), deref_fun(&ParmVarDecl::getType));
385  }
386
387  /// createImplicitParams - Used to lazily create the self and cmd
388  /// implict parameters. This must be called prior to using getSelfDecl()
389  /// or getCmdDecl(). The call is ignored if the implicit paramters
390  /// have already been created.
391  void createImplicitParams(ASTContext &Context, const ObjCInterfaceDecl *ID);
392
393  ImplicitParamDecl * getSelfDecl() const { return SelfDecl; }
394  void setSelfDecl(ImplicitParamDecl *SD) { SelfDecl = SD; }
395  ImplicitParamDecl * getCmdDecl() const { return CmdDecl; }
396  void setCmdDecl(ImplicitParamDecl *CD) { CmdDecl = CD; }
397
398  /// Determines the family of this method.
399  ObjCMethodFamily getMethodFamily() const;
400
401  bool isInstanceMethod() const { return IsInstance; }
402  void setInstanceMethod(bool isInst) { IsInstance = isInst; }
403  bool isVariadic() const { return IsVariadic; }
404  void setVariadic(bool isVar) { IsVariadic = isVar; }
405
406  bool isClassMethod() const { return !IsInstance; }
407
408  bool isPropertyAccessor() const { return IsPropertyAccessor; }
409  void setPropertyAccessor(bool isAccessor) { IsPropertyAccessor = isAccessor; }
410
411  bool isDefined() const { return IsDefined; }
412  void setDefined(bool isDefined) { IsDefined = isDefined; }
413
414  /// \brief Whether this method overrides any other in the class hierarchy.
415  ///
416  /// A method is said to override any method in the class's
417  /// base classes, its protocols, or its categories' protocols, that has
418  /// the same selector and is of the same kind (class or instance).
419  /// A method in an implementation is not considered as overriding the same
420  /// method in the interface or its categories.
421  bool isOverriding() const { return IsOverriding; }
422  void setOverriding(bool isOverriding) { IsOverriding = isOverriding; }
423
424  /// \brief Return overridden methods for the given \p Method.
425  ///
426  /// An ObjC method is considered to override any method in the class's
427  /// base classes (and base's categories), its protocols, or its categories'
428  /// protocols, that has
429  /// the same selector and is of the same kind (class or instance).
430  /// A method in an implementation is not considered as overriding the same
431  /// method in the interface or its categories.
432  void getOverriddenMethods(
433                     SmallVectorImpl<const ObjCMethodDecl *> &Overridden) const;
434
435  /// \brief True if the method was a definition but its body was skipped.
436  bool hasSkippedBody() const { return HasSkippedBody; }
437  void setHasSkippedBody(bool Skipped = true) { HasSkippedBody = Skipped; }
438
439  /// \brief Returns the property associated with this method's selector.
440  ///
441  /// Note that even if this particular method is not marked as a property
442  /// accessor, it is still possible for it to match a property declared in a
443  /// superclass. Pass \c false if you only want to check the current class.
444  const ObjCPropertyDecl *findPropertyDecl(bool CheckOverrides = true) const;
445
446  // Related to protocols declared in  \@protocol
447  void setDeclImplementation(ImplementationControl ic) {
448    DeclImplementation = ic;
449  }
450  ImplementationControl getImplementationControl() const {
451    return ImplementationControl(DeclImplementation);
452  }
453
454  /// \brief Determine whether this method has a body.
455  virtual bool hasBody() const { return Body.isValid(); }
456
457  /// \brief Retrieve the body of this method, if it has one.
458  virtual Stmt *getBody() const;
459
460  void setLazyBody(uint64_t Offset) { Body = Offset; }
461
462  CompoundStmt *getCompoundBody() { return (CompoundStmt*)getBody(); }
463  void setBody(Stmt *B) { Body = B; }
464
465  /// \brief Returns whether this specific method is a definition.
466  bool isThisDeclarationADefinition() const { return hasBody(); }
467
468  // Implement isa/cast/dyncast/etc.
469  static bool classof(const Decl *D) { return classofKind(D->getKind()); }
470  static bool classofKind(Kind K) { return K == ObjCMethod; }
471  static DeclContext *castToDeclContext(const ObjCMethodDecl *D) {
472    return static_cast<DeclContext *>(const_cast<ObjCMethodDecl*>(D));
473  }
474  static ObjCMethodDecl *castFromDeclContext(const DeclContext *DC) {
475    return static_cast<ObjCMethodDecl *>(const_cast<DeclContext*>(DC));
476  }
477
478  friend class ASTDeclReader;
479  friend class ASTDeclWriter;
480};
481
482/// ObjCContainerDecl - Represents a container for method declarations.
483/// Current sub-classes are ObjCInterfaceDecl, ObjCCategoryDecl,
484/// ObjCProtocolDecl, and ObjCImplDecl.
485///
486class ObjCContainerDecl : public NamedDecl, public DeclContext {
487  virtual void anchor();
488
489  SourceLocation AtStart;
490
491  // These two locations in the range mark the end of the method container.
492  // The first points to the '@' token, and the second to the 'end' token.
493  SourceRange AtEnd;
494public:
495
496  ObjCContainerDecl(Kind DK, DeclContext *DC,
497                    IdentifierInfo *Id, SourceLocation nameLoc,
498                    SourceLocation atStartLoc)
499    : NamedDecl(DK, DC, nameLoc, Id), DeclContext(DK), AtStart(atStartLoc) {}
500
501  // Iterator access to properties.
502  typedef specific_decl_iterator<ObjCPropertyDecl> prop_iterator;
503  prop_iterator prop_begin() const {
504    return prop_iterator(decls_begin());
505  }
506  prop_iterator prop_end() const {
507    return prop_iterator(decls_end());
508  }
509
510  // Iterator access to instance/class methods.
511  typedef specific_decl_iterator<ObjCMethodDecl> method_iterator;
512  method_iterator meth_begin() const {
513    return method_iterator(decls_begin());
514  }
515  method_iterator meth_end() const {
516    return method_iterator(decls_end());
517  }
518
519  typedef filtered_decl_iterator<ObjCMethodDecl,
520                                 &ObjCMethodDecl::isInstanceMethod>
521    instmeth_iterator;
522  instmeth_iterator instmeth_begin() const {
523    return instmeth_iterator(decls_begin());
524  }
525  instmeth_iterator instmeth_end() const {
526    return instmeth_iterator(decls_end());
527  }
528
529  typedef filtered_decl_iterator<ObjCMethodDecl,
530                                 &ObjCMethodDecl::isClassMethod>
531    classmeth_iterator;
532  classmeth_iterator classmeth_begin() const {
533    return classmeth_iterator(decls_begin());
534  }
535  classmeth_iterator classmeth_end() const {
536    return classmeth_iterator(decls_end());
537  }
538
539  // Get the local instance/class method declared in this interface.
540  ObjCMethodDecl *getMethod(Selector Sel, bool isInstance,
541                            bool AllowHidden = false) const;
542  ObjCMethodDecl *getInstanceMethod(Selector Sel,
543                                    bool AllowHidden = false) const {
544    return getMethod(Sel, true/*isInstance*/, AllowHidden);
545  }
546  ObjCMethodDecl *getClassMethod(Selector Sel, bool AllowHidden = false) const {
547    return getMethod(Sel, false/*isInstance*/, AllowHidden);
548  }
549  bool HasUserDeclaredSetterMethod(const ObjCPropertyDecl *P) const;
550  ObjCIvarDecl *getIvarDecl(IdentifierInfo *Id) const;
551
552  ObjCPropertyDecl *FindPropertyDeclaration(IdentifierInfo *PropertyId) const;
553
554  typedef llvm::DenseMap<IdentifierInfo*, ObjCPropertyDecl*> PropertyMap;
555
556  typedef llvm::DenseMap<const ObjCProtocolDecl *, ObjCPropertyDecl*>
557            ProtocolPropertyMap;
558
559  typedef llvm::SmallVector<ObjCPropertyDecl*, 8> PropertyDeclOrder;
560
561  /// This routine collects list of properties to be implemented in the class.
562  /// This includes, class's and its conforming protocols' properties.
563  /// Note, the superclass's properties are not included in the list.
564  virtual void collectPropertiesToImplement(PropertyMap &PM,
565                                            PropertyDeclOrder &PO) const {}
566
567  SourceLocation getAtStartLoc() const { return AtStart; }
568  void setAtStartLoc(SourceLocation Loc) { AtStart = Loc; }
569
570  // Marks the end of the container.
571  SourceRange getAtEndRange() const {
572    return AtEnd;
573  }
574  void setAtEndRange(SourceRange atEnd) {
575    AtEnd = atEnd;
576  }
577
578  virtual SourceRange getSourceRange() const LLVM_READONLY {
579    return SourceRange(AtStart, getAtEndRange().getEnd());
580  }
581
582  // Implement isa/cast/dyncast/etc.
583  static bool classof(const Decl *D) { return classofKind(D->getKind()); }
584  static bool classofKind(Kind K) {
585    return K >= firstObjCContainer &&
586           K <= lastObjCContainer;
587  }
588
589  static DeclContext *castToDeclContext(const ObjCContainerDecl *D) {
590    return static_cast<DeclContext *>(const_cast<ObjCContainerDecl*>(D));
591  }
592  static ObjCContainerDecl *castFromDeclContext(const DeclContext *DC) {
593    return static_cast<ObjCContainerDecl *>(const_cast<DeclContext*>(DC));
594  }
595};
596
597/// \brief Represents an ObjC class declaration.
598///
599/// For example:
600///
601/// \code
602///   // MostPrimitive declares no super class (not particularly useful).
603///   \@interface MostPrimitive
604///     // no instance variables or methods.
605///   \@end
606///
607///   // NSResponder inherits from NSObject & implements NSCoding (a protocol).
608///   \@interface NSResponder : NSObject \<NSCoding>
609///   { // instance variables are represented by ObjCIvarDecl.
610///     id nextResponder; // nextResponder instance variable.
611///   }
612///   - (NSResponder *)nextResponder; // return a pointer to NSResponder.
613///   - (void)mouseMoved:(NSEvent *)theEvent; // return void, takes a pointer
614///   \@end                                    // to an NSEvent.
615/// \endcode
616///
617///   Unlike C/C++, forward class declarations are accomplished with \@class.
618///   Unlike C/C++, \@class allows for a list of classes to be forward declared.
619///   Unlike C++, ObjC is a single-rooted class model. In Cocoa, classes
620///   typically inherit from NSObject (an exception is NSProxy).
621///
622class ObjCInterfaceDecl : public ObjCContainerDecl
623                        , public Redeclarable<ObjCInterfaceDecl> {
624  virtual void anchor();
625
626  /// TypeForDecl - This indicates the Type object that represents this
627  /// TypeDecl.  It is a cache maintained by ASTContext::getObjCInterfaceType
628  mutable const Type *TypeForDecl;
629  friend class ASTContext;
630
631  struct DefinitionData {
632    /// \brief The definition of this class, for quick access from any
633    /// declaration.
634    ObjCInterfaceDecl *Definition;
635
636    /// Class's super class.
637    ObjCInterfaceDecl *SuperClass;
638
639    /// Protocols referenced in the \@interface  declaration
640    ObjCProtocolList ReferencedProtocols;
641
642    /// Protocols reference in both the \@interface and class extensions.
643    ObjCList<ObjCProtocolDecl> AllReferencedProtocols;
644
645    /// \brief List of categories and class extensions defined for this class.
646    ///
647    /// Categories are stored as a linked list in the AST, since the categories
648    /// and class extensions come long after the initial interface declaration,
649    /// and we avoid dynamically-resized arrays in the AST wherever possible.
650    ObjCCategoryDecl *CategoryList;
651
652    /// IvarList - List of all ivars defined by this class; including class
653    /// extensions and implementation. This list is built lazily.
654    ObjCIvarDecl *IvarList;
655
656    /// \brief Indicates that the contents of this Objective-C class will be
657    /// completed by the external AST source when required.
658    mutable bool ExternallyCompleted : 1;
659
660    /// \brief Indicates that the ivar cache does not yet include ivars
661    /// declared in the implementation.
662    mutable bool IvarListMissingImplementation : 1;
663
664    /// \brief The location of the superclass, if any.
665    SourceLocation SuperClassLoc;
666
667    /// \brief The location of the last location in this declaration, before
668    /// the properties/methods. For example, this will be the '>', '}', or
669    /// identifier,
670    SourceLocation EndLoc;
671
672    DefinitionData() : Definition(), SuperClass(), CategoryList(), IvarList(),
673                       ExternallyCompleted(),
674                       IvarListMissingImplementation(true) { }
675  };
676
677  ObjCInterfaceDecl(DeclContext *DC, SourceLocation atLoc, IdentifierInfo *Id,
678                    SourceLocation CLoc, ObjCInterfaceDecl *PrevDecl,
679                    bool isInternal);
680
681  void LoadExternalDefinition() const;
682
683  /// \brief Contains a pointer to the data associated with this class,
684  /// which will be NULL if this class has not yet been defined.
685  ///
686  /// The bit indicates when we don't need to check for out-of-date
687  /// declarations. It will be set unless modules are enabled.
688  llvm::PointerIntPair<DefinitionData *, 1, bool> Data;
689
690  DefinitionData &data() const {
691    assert(Data.getPointer() && "Declaration has no definition!");
692    return *Data.getPointer();
693  }
694
695  /// \brief Allocate the definition data for this class.
696  void allocateDefinitionData();
697
698  typedef Redeclarable<ObjCInterfaceDecl> redeclarable_base;
699  virtual ObjCInterfaceDecl *getNextRedeclaration() {
700    return RedeclLink.getNext();
701  }
702  virtual ObjCInterfaceDecl *getPreviousDeclImpl() {
703    return getPreviousDecl();
704  }
705  virtual ObjCInterfaceDecl *getMostRecentDeclImpl() {
706    return getMostRecentDecl();
707  }
708
709public:
710  static ObjCInterfaceDecl *Create(const ASTContext &C, DeclContext *DC,
711                                   SourceLocation atLoc,
712                                   IdentifierInfo *Id,
713                                   ObjCInterfaceDecl *PrevDecl,
714                                   SourceLocation ClassLoc = SourceLocation(),
715                                   bool isInternal = false);
716
717  static ObjCInterfaceDecl *CreateDeserialized(ASTContext &C, unsigned ID);
718
719  virtual SourceRange getSourceRange() const LLVM_READONLY {
720    if (isThisDeclarationADefinition())
721      return ObjCContainerDecl::getSourceRange();
722
723    return SourceRange(getAtStartLoc(), getLocation());
724  }
725
726  /// \brief Indicate that this Objective-C class is complete, but that
727  /// the external AST source will be responsible for filling in its contents
728  /// when a complete class is required.
729  void setExternallyCompleted();
730
731  const ObjCProtocolList &getReferencedProtocols() const {
732    assert(hasDefinition() && "Caller did not check for forward reference!");
733    if (data().ExternallyCompleted)
734      LoadExternalDefinition();
735
736    return data().ReferencedProtocols;
737  }
738
739  ObjCImplementationDecl *getImplementation() const;
740  void setImplementation(ObjCImplementationDecl *ImplD);
741
742  ObjCCategoryDecl *FindCategoryDeclaration(IdentifierInfo *CategoryId) const;
743
744  // Get the local instance/class method declared in a category.
745  ObjCMethodDecl *getCategoryInstanceMethod(Selector Sel) const;
746  ObjCMethodDecl *getCategoryClassMethod(Selector Sel) const;
747  ObjCMethodDecl *getCategoryMethod(Selector Sel, bool isInstance) const {
748    return isInstance ? getInstanceMethod(Sel)
749                      : getClassMethod(Sel);
750  }
751
752  typedef ObjCProtocolList::iterator protocol_iterator;
753
754  protocol_iterator protocol_begin() const {
755    // FIXME: Should make sure no callers ever do this.
756    if (!hasDefinition())
757      return protocol_iterator();
758
759    if (data().ExternallyCompleted)
760      LoadExternalDefinition();
761
762    return data().ReferencedProtocols.begin();
763  }
764  protocol_iterator protocol_end() const {
765    // FIXME: Should make sure no callers ever do this.
766    if (!hasDefinition())
767      return protocol_iterator();
768
769    if (data().ExternallyCompleted)
770      LoadExternalDefinition();
771
772    return data().ReferencedProtocols.end();
773  }
774
775  typedef ObjCProtocolList::loc_iterator protocol_loc_iterator;
776
777  protocol_loc_iterator protocol_loc_begin() const {
778    // FIXME: Should make sure no callers ever do this.
779    if (!hasDefinition())
780      return protocol_loc_iterator();
781
782    if (data().ExternallyCompleted)
783      LoadExternalDefinition();
784
785    return data().ReferencedProtocols.loc_begin();
786  }
787
788  protocol_loc_iterator protocol_loc_end() const {
789    // FIXME: Should make sure no callers ever do this.
790    if (!hasDefinition())
791      return protocol_loc_iterator();
792
793    if (data().ExternallyCompleted)
794      LoadExternalDefinition();
795
796    return data().ReferencedProtocols.loc_end();
797  }
798
799  typedef ObjCList<ObjCProtocolDecl>::iterator all_protocol_iterator;
800
801  all_protocol_iterator all_referenced_protocol_begin() const {
802    // FIXME: Should make sure no callers ever do this.
803    if (!hasDefinition())
804      return all_protocol_iterator();
805
806    if (data().ExternallyCompleted)
807      LoadExternalDefinition();
808
809    return data().AllReferencedProtocols.empty()
810             ? protocol_begin()
811             : data().AllReferencedProtocols.begin();
812  }
813  all_protocol_iterator all_referenced_protocol_end() const {
814    // FIXME: Should make sure no callers ever do this.
815    if (!hasDefinition())
816      return all_protocol_iterator();
817
818    if (data().ExternallyCompleted)
819      LoadExternalDefinition();
820
821    return data().AllReferencedProtocols.empty()
822             ? protocol_end()
823             : data().AllReferencedProtocols.end();
824  }
825
826  typedef specific_decl_iterator<ObjCIvarDecl> ivar_iterator;
827
828  ivar_iterator ivar_begin() const {
829    if (const ObjCInterfaceDecl *Def = getDefinition())
830      return ivar_iterator(Def->decls_begin());
831
832    // FIXME: Should make sure no callers ever do this.
833    return ivar_iterator();
834  }
835  ivar_iterator ivar_end() const {
836    if (const ObjCInterfaceDecl *Def = getDefinition())
837      return ivar_iterator(Def->decls_end());
838
839    // FIXME: Should make sure no callers ever do this.
840    return ivar_iterator();
841  }
842
843  unsigned ivar_size() const {
844    return std::distance(ivar_begin(), ivar_end());
845  }
846
847  bool ivar_empty() const { return ivar_begin() == ivar_end(); }
848
849  ObjCIvarDecl *all_declared_ivar_begin();
850  const ObjCIvarDecl *all_declared_ivar_begin() const {
851    // Even though this modifies IvarList, it's conceptually const:
852    // the ivar chain is essentially a cached property of ObjCInterfaceDecl.
853    return const_cast<ObjCInterfaceDecl *>(this)->all_declared_ivar_begin();
854  }
855  void setIvarList(ObjCIvarDecl *ivar) { data().IvarList = ivar; }
856
857  /// setProtocolList - Set the list of protocols that this interface
858  /// implements.
859  void setProtocolList(ObjCProtocolDecl *const* List, unsigned Num,
860                       const SourceLocation *Locs, ASTContext &C) {
861    data().ReferencedProtocols.set(List, Num, Locs, C);
862  }
863
864  /// mergeClassExtensionProtocolList - Merge class extension's protocol list
865  /// into the protocol list for this class.
866  void mergeClassExtensionProtocolList(ObjCProtocolDecl *const* List,
867                                       unsigned Num,
868                                       ASTContext &C);
869
870  /// \brief Determine whether this particular declaration of this class is
871  /// actually also a definition.
872  bool isThisDeclarationADefinition() const {
873    return getDefinition() == this;
874  }
875
876  /// \brief Determine whether this class has been defined.
877  bool hasDefinition() const {
878    // If the name of this class is out-of-date, bring it up-to-date, which
879    // might bring in a definition.
880    // Note: a null value indicates that we don't have a definition and that
881    // modules are enabled.
882    if (!Data.getOpaqueValue()) {
883      if (IdentifierInfo *II = getIdentifier()) {
884        if (II->isOutOfDate()) {
885          updateOutOfDate(*II);
886        }
887      }
888    }
889
890    return Data.getPointer();
891  }
892
893  /// \brief Retrieve the definition of this class, or NULL if this class
894  /// has been forward-declared (with \@class) but not yet defined (with
895  /// \@interface).
896  ObjCInterfaceDecl *getDefinition() {
897    return hasDefinition()? Data.getPointer()->Definition : 0;
898  }
899
900  /// \brief Retrieve the definition of this class, or NULL if this class
901  /// has been forward-declared (with \@class) but not yet defined (with
902  /// \@interface).
903  const ObjCInterfaceDecl *getDefinition() const {
904    return hasDefinition()? Data.getPointer()->Definition : 0;
905  }
906
907  /// \brief Starts the definition of this Objective-C class, taking it from
908  /// a forward declaration (\@class) to a definition (\@interface).
909  void startDefinition();
910
911  ObjCInterfaceDecl *getSuperClass() const {
912    // FIXME: Should make sure no callers ever do this.
913    if (!hasDefinition())
914      return 0;
915
916    if (data().ExternallyCompleted)
917      LoadExternalDefinition();
918
919    return data().SuperClass;
920  }
921
922  void setSuperClass(ObjCInterfaceDecl * superCls) {
923    data().SuperClass =
924      (superCls && superCls->hasDefinition()) ? superCls->getDefinition()
925                                              : superCls;
926  }
927
928  /// \brief Iterator that walks over the list of categories, filtering out
929  /// those that do not meet specific criteria.
930  ///
931  /// This class template is used for the various permutations of category
932  /// and extension iterators.
933  template<bool (*Filter)(ObjCCategoryDecl *)>
934  class filtered_category_iterator {
935    ObjCCategoryDecl *Current;
936
937    void findAcceptableCategory();
938
939  public:
940    typedef ObjCCategoryDecl *      value_type;
941    typedef value_type              reference;
942    typedef value_type              pointer;
943    typedef std::ptrdiff_t          difference_type;
944    typedef std::input_iterator_tag iterator_category;
945
946    filtered_category_iterator() : Current(0) { }
947    explicit filtered_category_iterator(ObjCCategoryDecl *Current)
948      : Current(Current)
949    {
950      findAcceptableCategory();
951    }
952
953    reference operator*() const { return Current; }
954    pointer operator->() const { return Current; }
955
956    filtered_category_iterator &operator++();
957
958    filtered_category_iterator operator++(int) {
959      filtered_category_iterator Tmp = *this;
960      ++(*this);
961      return Tmp;
962    }
963
964    friend bool operator==(filtered_category_iterator X,
965                           filtered_category_iterator Y) {
966      return X.Current == Y.Current;
967    }
968
969    friend bool operator!=(filtered_category_iterator X,
970                           filtered_category_iterator Y) {
971      return X.Current != Y.Current;
972    }
973  };
974
975private:
976  /// \brief Test whether the given category is visible.
977  ///
978  /// Used in the \c visible_categories_iterator.
979  static bool isVisibleCategory(ObjCCategoryDecl *Cat);
980
981public:
982  /// \brief Iterator that walks over the list of categories and extensions
983  /// that are visible, i.e., not hidden in a non-imported submodule.
984  typedef filtered_category_iterator<isVisibleCategory>
985    visible_categories_iterator;
986
987  /// \brief Retrieve an iterator to the beginning of the visible-categories
988  /// list.
989  visible_categories_iterator visible_categories_begin() const {
990    return visible_categories_iterator(getCategoryListRaw());
991  }
992
993  /// \brief Retrieve an iterator to the end of the visible-categories list.
994  visible_categories_iterator visible_categories_end() const {
995    return visible_categories_iterator();
996  }
997
998  /// \brief Determine whether the visible-categories list is empty.
999  bool visible_categories_empty() const {
1000    return visible_categories_begin() == visible_categories_end();
1001  }
1002
1003private:
1004  /// \brief Test whether the given category... is a category.
1005  ///
1006  /// Used in the \c known_categories_iterator.
1007  static bool isKnownCategory(ObjCCategoryDecl *) { return true; }
1008
1009public:
1010  /// \brief Iterator that walks over all of the known categories and
1011  /// extensions, including those that are hidden.
1012  typedef filtered_category_iterator<isKnownCategory> known_categories_iterator;
1013
1014  /// \brief Retrieve an iterator to the beginning of the known-categories
1015  /// list.
1016  known_categories_iterator known_categories_begin() const {
1017    return known_categories_iterator(getCategoryListRaw());
1018  }
1019
1020  /// \brief Retrieve an iterator to the end of the known-categories list.
1021  known_categories_iterator known_categories_end() const {
1022    return known_categories_iterator();
1023  }
1024
1025  /// \brief Determine whether the known-categories list is empty.
1026  bool known_categories_empty() const {
1027    return known_categories_begin() == known_categories_end();
1028  }
1029
1030private:
1031  /// \brief Test whether the given category is a visible extension.
1032  ///
1033  /// Used in the \c visible_extensions_iterator.
1034  static bool isVisibleExtension(ObjCCategoryDecl *Cat);
1035
1036public:
1037  /// \brief Iterator that walks over all of the visible extensions, skipping
1038  /// any that are known but hidden.
1039  typedef filtered_category_iterator<isVisibleExtension>
1040    visible_extensions_iterator;
1041
1042  /// \brief Retrieve an iterator to the beginning of the visible-extensions
1043  /// list.
1044  visible_extensions_iterator visible_extensions_begin() const {
1045    return visible_extensions_iterator(getCategoryListRaw());
1046  }
1047
1048  /// \brief Retrieve an iterator to the end of the visible-extensions list.
1049  visible_extensions_iterator visible_extensions_end() const {
1050    return visible_extensions_iterator();
1051  }
1052
1053  /// \brief Determine whether the visible-extensions list is empty.
1054  bool visible_extensions_empty() const {
1055    return visible_extensions_begin() == visible_extensions_end();
1056  }
1057
1058private:
1059  /// \brief Test whether the given category is an extension.
1060  ///
1061  /// Used in the \c known_extensions_iterator.
1062  static bool isKnownExtension(ObjCCategoryDecl *Cat);
1063
1064public:
1065  /// \brief Iterator that walks over all of the known extensions.
1066  typedef filtered_category_iterator<isKnownExtension>
1067    known_extensions_iterator;
1068
1069  /// \brief Retrieve an iterator to the beginning of the known-extensions
1070  /// list.
1071  known_extensions_iterator known_extensions_begin() const {
1072    return known_extensions_iterator(getCategoryListRaw());
1073  }
1074
1075  /// \brief Retrieve an iterator to the end of the known-extensions list.
1076  known_extensions_iterator known_extensions_end() const {
1077    return known_extensions_iterator();
1078  }
1079
1080  /// \brief Determine whether the known-extensions list is empty.
1081  bool known_extensions_empty() const {
1082    return known_extensions_begin() == known_extensions_end();
1083  }
1084
1085  /// \brief Retrieve the raw pointer to the start of the category/extension
1086  /// list.
1087  ObjCCategoryDecl* getCategoryListRaw() const {
1088    // FIXME: Should make sure no callers ever do this.
1089    if (!hasDefinition())
1090      return 0;
1091
1092    if (data().ExternallyCompleted)
1093      LoadExternalDefinition();
1094
1095    return data().CategoryList;
1096  }
1097
1098  /// \brief Set the raw pointer to the start of the category/extension
1099  /// list.
1100  void setCategoryListRaw(ObjCCategoryDecl *category) {
1101    data().CategoryList = category;
1102  }
1103
1104  ObjCPropertyDecl
1105    *FindPropertyVisibleInPrimaryClass(IdentifierInfo *PropertyId) const;
1106
1107  virtual void collectPropertiesToImplement(PropertyMap &PM,
1108                                            PropertyDeclOrder &PO) const;
1109
1110  /// isSuperClassOf - Return true if this class is the specified class or is a
1111  /// super class of the specified interface class.
1112  bool isSuperClassOf(const ObjCInterfaceDecl *I) const {
1113    // If RHS is derived from LHS it is OK; else it is not OK.
1114    while (I != NULL) {
1115      if (declaresSameEntity(this, I))
1116        return true;
1117
1118      I = I->getSuperClass();
1119    }
1120    return false;
1121  }
1122
1123  /// isArcWeakrefUnavailable - Checks for a class or one of its super classes
1124  /// to be incompatible with __weak references. Returns true if it is.
1125  bool isArcWeakrefUnavailable() const;
1126
1127  /// isObjCRequiresPropertyDefs - Checks that a class or one of its super
1128  /// classes must not be auto-synthesized. Returns class decl. if it must not
1129  /// be; 0, otherwise.
1130  const ObjCInterfaceDecl *isObjCRequiresPropertyDefs() const;
1131
1132  ObjCIvarDecl *lookupInstanceVariable(IdentifierInfo *IVarName,
1133                                       ObjCInterfaceDecl *&ClassDeclared);
1134  ObjCIvarDecl *lookupInstanceVariable(IdentifierInfo *IVarName) {
1135    ObjCInterfaceDecl *ClassDeclared;
1136    return lookupInstanceVariable(IVarName, ClassDeclared);
1137  }
1138
1139  ObjCProtocolDecl *lookupNestedProtocol(IdentifierInfo *Name);
1140
1141  // Lookup a method. First, we search locally. If a method isn't
1142  // found, we search referenced protocols and class categories.
1143  ObjCMethodDecl *lookupMethod(Selector Sel, bool isInstance,
1144                               bool shallowCategoryLookup= false,
1145                               const ObjCCategoryDecl *C= 0) const;
1146  ObjCMethodDecl *lookupInstanceMethod(Selector Sel,
1147                            bool shallowCategoryLookup = false) const {
1148    return lookupMethod(Sel, true/*isInstance*/, shallowCategoryLookup);
1149  }
1150  ObjCMethodDecl *lookupClassMethod(Selector Sel,
1151                     bool shallowCategoryLookup = false) const {
1152    return lookupMethod(Sel, false/*isInstance*/, shallowCategoryLookup);
1153  }
1154  ObjCInterfaceDecl *lookupInheritedClass(const IdentifierInfo *ICName);
1155
1156  /// \brief Lookup a method in the classes implementation hierarchy.
1157  ObjCMethodDecl *lookupPrivateMethod(const Selector &Sel,
1158                                      bool Instance=true) const;
1159
1160  ObjCMethodDecl *lookupPrivateClassMethod(const Selector &Sel) {
1161    return lookupPrivateMethod(Sel, false);
1162  }
1163
1164  /// \brief Lookup a setter or getter in the class hierarchy,
1165  /// including in all categories except for category passed
1166  /// as argument.
1167  ObjCMethodDecl *lookupPropertyAccessor(const Selector Sel,
1168                                         const ObjCCategoryDecl *Cat) const {
1169    return lookupMethod(Sel, true/*isInstance*/,
1170                        false/*shallowCategoryLookup*/, Cat);
1171  }
1172
1173  SourceLocation getEndOfDefinitionLoc() const {
1174    if (!hasDefinition())
1175      return getLocation();
1176
1177    return data().EndLoc;
1178  }
1179
1180  void setEndOfDefinitionLoc(SourceLocation LE) { data().EndLoc = LE; }
1181
1182  void setSuperClassLoc(SourceLocation Loc) { data().SuperClassLoc = Loc; }
1183  SourceLocation getSuperClassLoc() const { return data().SuperClassLoc; }
1184
1185  /// isImplicitInterfaceDecl - check that this is an implicitly declared
1186  /// ObjCInterfaceDecl node. This is for legacy objective-c \@implementation
1187  /// declaration without an \@interface declaration.
1188  bool isImplicitInterfaceDecl() const {
1189    return hasDefinition() ? data().Definition->isImplicit() : isImplicit();
1190  }
1191
1192  /// ClassImplementsProtocol - Checks that 'lProto' protocol
1193  /// has been implemented in IDecl class, its super class or categories (if
1194  /// lookupCategory is true).
1195  bool ClassImplementsProtocol(ObjCProtocolDecl *lProto,
1196                               bool lookupCategory,
1197                               bool RHSIsQualifiedID = false);
1198
1199  typedef redeclarable_base::redecl_iterator redecl_iterator;
1200  using redeclarable_base::redecls_begin;
1201  using redeclarable_base::redecls_end;
1202  using redeclarable_base::getPreviousDecl;
1203  using redeclarable_base::getMostRecentDecl;
1204  using redeclarable_base::isFirstDecl;
1205
1206  /// Retrieves the canonical declaration of this Objective-C class.
1207  ObjCInterfaceDecl *getCanonicalDecl() { return getFirstDecl(); }
1208  const ObjCInterfaceDecl *getCanonicalDecl() const { return getFirstDecl(); }
1209
1210  // Low-level accessor
1211  const Type *getTypeForDecl() const { return TypeForDecl; }
1212  void setTypeForDecl(const Type *TD) const { TypeForDecl = TD; }
1213
1214  static bool classof(const Decl *D) { return classofKind(D->getKind()); }
1215  static bool classofKind(Kind K) { return K == ObjCInterface; }
1216
1217  friend class ASTReader;
1218  friend class ASTDeclReader;
1219  friend class ASTDeclWriter;
1220};
1221
1222/// ObjCIvarDecl - Represents an ObjC instance variable. In general, ObjC
1223/// instance variables are identical to C. The only exception is Objective-C
1224/// supports C++ style access control. For example:
1225///
1226///   \@interface IvarExample : NSObject
1227///   {
1228///     id defaultToProtected;
1229///   \@public:
1230///     id canBePublic; // same as C++.
1231///   \@protected:
1232///     id canBeProtected; // same as C++.
1233///   \@package:
1234///     id canBePackage; // framework visibility (not available in C++).
1235///   }
1236///
1237class ObjCIvarDecl : public FieldDecl {
1238  virtual void anchor();
1239
1240public:
1241  enum AccessControl {
1242    None, Private, Protected, Public, Package
1243  };
1244
1245private:
1246  ObjCIvarDecl(ObjCContainerDecl *DC, SourceLocation StartLoc,
1247               SourceLocation IdLoc, IdentifierInfo *Id,
1248               QualType T, TypeSourceInfo *TInfo, AccessControl ac, Expr *BW,
1249               bool synthesized,
1250               bool backingIvarReferencedInAccessor)
1251    : FieldDecl(ObjCIvar, DC, StartLoc, IdLoc, Id, T, TInfo, BW,
1252                /*Mutable=*/false, /*HasInit=*/ICIS_NoInit),
1253      NextIvar(0), DeclAccess(ac), Synthesized(synthesized),
1254      BackingIvarReferencedInAccessor(backingIvarReferencedInAccessor) {}
1255
1256public:
1257  static ObjCIvarDecl *Create(ASTContext &C, ObjCContainerDecl *DC,
1258                              SourceLocation StartLoc, SourceLocation IdLoc,
1259                              IdentifierInfo *Id, QualType T,
1260                              TypeSourceInfo *TInfo,
1261                              AccessControl ac, Expr *BW = NULL,
1262                              bool synthesized=false,
1263                              bool backingIvarReferencedInAccessor=false);
1264
1265  static ObjCIvarDecl *CreateDeserialized(ASTContext &C, unsigned ID);
1266
1267  /// \brief Return the class interface that this ivar is logically contained
1268  /// in; this is either the interface where the ivar was declared, or the
1269  /// interface the ivar is conceptually a part of in the case of synthesized
1270  /// ivars.
1271  const ObjCInterfaceDecl *getContainingInterface() const;
1272
1273  ObjCIvarDecl *getNextIvar() { return NextIvar; }
1274  const ObjCIvarDecl *getNextIvar() const { return NextIvar; }
1275  void setNextIvar(ObjCIvarDecl *ivar) { NextIvar = ivar; }
1276
1277  void setAccessControl(AccessControl ac) { DeclAccess = ac; }
1278
1279  AccessControl getAccessControl() const { return AccessControl(DeclAccess); }
1280
1281  AccessControl getCanonicalAccessControl() const {
1282    return DeclAccess == None ? Protected : AccessControl(DeclAccess);
1283  }
1284
1285  void setBackingIvarReferencedInAccessor(bool val) {
1286    BackingIvarReferencedInAccessor = val;
1287  }
1288  bool getBackingIvarReferencedInAccessor() const {
1289    return BackingIvarReferencedInAccessor;
1290  }
1291
1292  void setSynthesize(bool synth) { Synthesized = synth; }
1293  bool getSynthesize() const { return Synthesized; }
1294
1295  // Implement isa/cast/dyncast/etc.
1296  static bool classof(const Decl *D) { return classofKind(D->getKind()); }
1297  static bool classofKind(Kind K) { return K == ObjCIvar; }
1298private:
1299  /// NextIvar - Next Ivar in the list of ivars declared in class; class's
1300  /// extensions and class's implementation
1301  ObjCIvarDecl *NextIvar;
1302
1303  // NOTE: VC++ treats enums as signed, avoid using the AccessControl enum
1304  unsigned DeclAccess : 3;
1305  unsigned Synthesized : 1;
1306  unsigned BackingIvarReferencedInAccessor : 1;
1307};
1308
1309
1310/// \brief Represents a field declaration created by an \@defs(...).
1311class ObjCAtDefsFieldDecl : public FieldDecl {
1312  virtual void anchor();
1313  ObjCAtDefsFieldDecl(DeclContext *DC, SourceLocation StartLoc,
1314                      SourceLocation IdLoc, IdentifierInfo *Id,
1315                      QualType T, Expr *BW)
1316    : FieldDecl(ObjCAtDefsField, DC, StartLoc, IdLoc, Id, T,
1317                /*TInfo=*/0, // FIXME: Do ObjCAtDefs have declarators ?
1318                BW, /*Mutable=*/false, /*HasInit=*/ICIS_NoInit) {}
1319
1320public:
1321  static ObjCAtDefsFieldDecl *Create(ASTContext &C, DeclContext *DC,
1322                                     SourceLocation StartLoc,
1323                                     SourceLocation IdLoc, IdentifierInfo *Id,
1324                                     QualType T, Expr *BW);
1325
1326  static ObjCAtDefsFieldDecl *CreateDeserialized(ASTContext &C, unsigned ID);
1327
1328  // Implement isa/cast/dyncast/etc.
1329  static bool classof(const Decl *D) { return classofKind(D->getKind()); }
1330  static bool classofKind(Kind K) { return K == ObjCAtDefsField; }
1331};
1332
1333/// \brief Represents an Objective-C protocol declaration.
1334///
1335/// Objective-C protocols declare a pure abstract type (i.e., no instance
1336/// variables are permitted).  Protocols originally drew inspiration from
1337/// C++ pure virtual functions (a C++ feature with nice semantics and lousy
1338/// syntax:-). Here is an example:
1339///
1340/// \code
1341/// \@protocol NSDraggingInfo <refproto1, refproto2>
1342/// - (NSWindow *)draggingDestinationWindow;
1343/// - (NSImage *)draggedImage;
1344/// \@end
1345/// \endcode
1346///
1347/// This says that NSDraggingInfo requires two methods and requires everything
1348/// that the two "referenced protocols" 'refproto1' and 'refproto2' require as
1349/// well.
1350///
1351/// \code
1352/// \@interface ImplementsNSDraggingInfo : NSObject \<NSDraggingInfo>
1353/// \@end
1354/// \endcode
1355///
1356/// ObjC protocols inspired Java interfaces. Unlike Java, ObjC classes and
1357/// protocols are in distinct namespaces. For example, Cocoa defines both
1358/// an NSObject protocol and class (which isn't allowed in Java). As a result,
1359/// protocols are referenced using angle brackets as follows:
1360///
1361/// id \<NSDraggingInfo> anyObjectThatImplementsNSDraggingInfo;
1362///
1363class ObjCProtocolDecl : public ObjCContainerDecl,
1364                         public Redeclarable<ObjCProtocolDecl> {
1365  virtual void anchor();
1366
1367  struct DefinitionData {
1368    // \brief The declaration that defines this protocol.
1369    ObjCProtocolDecl *Definition;
1370
1371    /// \brief Referenced protocols
1372    ObjCProtocolList ReferencedProtocols;
1373  };
1374
1375  /// \brief Contains a pointer to the data associated with this class,
1376  /// which will be NULL if this class has not yet been defined.
1377  ///
1378  /// The bit indicates when we don't need to check for out-of-date
1379  /// declarations. It will be set unless modules are enabled.
1380  llvm::PointerIntPair<DefinitionData *, 1, bool> Data;
1381
1382  DefinitionData &data() const {
1383    assert(Data.getPointer() && "Objective-C protocol has no definition!");
1384    return *Data.getPointer();
1385  }
1386
1387  ObjCProtocolDecl(DeclContext *DC, IdentifierInfo *Id,
1388                   SourceLocation nameLoc, SourceLocation atStartLoc,
1389                   ObjCProtocolDecl *PrevDecl);
1390
1391  void allocateDefinitionData();
1392
1393  typedef Redeclarable<ObjCProtocolDecl> redeclarable_base;
1394  virtual ObjCProtocolDecl *getNextRedeclaration() {
1395    return RedeclLink.getNext();
1396  }
1397  virtual ObjCProtocolDecl *getPreviousDeclImpl() {
1398    return getPreviousDecl();
1399  }
1400  virtual ObjCProtocolDecl *getMostRecentDeclImpl() {
1401    return getMostRecentDecl();
1402  }
1403
1404public:
1405  static ObjCProtocolDecl *Create(ASTContext &C, DeclContext *DC,
1406                                  IdentifierInfo *Id,
1407                                  SourceLocation nameLoc,
1408                                  SourceLocation atStartLoc,
1409                                  ObjCProtocolDecl *PrevDecl);
1410
1411  static ObjCProtocolDecl *CreateDeserialized(ASTContext &C, unsigned ID);
1412
1413  const ObjCProtocolList &getReferencedProtocols() const {
1414    assert(hasDefinition() && "No definition available!");
1415    return data().ReferencedProtocols;
1416  }
1417  typedef ObjCProtocolList::iterator protocol_iterator;
1418  protocol_iterator protocol_begin() const {
1419    if (!hasDefinition())
1420      return protocol_iterator();
1421
1422    return data().ReferencedProtocols.begin();
1423  }
1424  protocol_iterator protocol_end() const {
1425    if (!hasDefinition())
1426      return protocol_iterator();
1427
1428    return data().ReferencedProtocols.end();
1429  }
1430  typedef ObjCProtocolList::loc_iterator protocol_loc_iterator;
1431  protocol_loc_iterator protocol_loc_begin() const {
1432    if (!hasDefinition())
1433      return protocol_loc_iterator();
1434
1435    return data().ReferencedProtocols.loc_begin();
1436  }
1437  protocol_loc_iterator protocol_loc_end() const {
1438    if (!hasDefinition())
1439      return protocol_loc_iterator();
1440
1441    return data().ReferencedProtocols.loc_end();
1442  }
1443  unsigned protocol_size() const {
1444    if (!hasDefinition())
1445      return 0;
1446
1447    return data().ReferencedProtocols.size();
1448  }
1449
1450  /// setProtocolList - Set the list of protocols that this interface
1451  /// implements.
1452  void setProtocolList(ObjCProtocolDecl *const*List, unsigned Num,
1453                       const SourceLocation *Locs, ASTContext &C) {
1454    assert(hasDefinition() && "Protocol is not defined");
1455    data().ReferencedProtocols.set(List, Num, Locs, C);
1456  }
1457
1458  ObjCProtocolDecl *lookupProtocolNamed(IdentifierInfo *PName);
1459
1460  // Lookup a method. First, we search locally. If a method isn't
1461  // found, we search referenced protocols and class categories.
1462  ObjCMethodDecl *lookupMethod(Selector Sel, bool isInstance) const;
1463  ObjCMethodDecl *lookupInstanceMethod(Selector Sel) const {
1464    return lookupMethod(Sel, true/*isInstance*/);
1465  }
1466  ObjCMethodDecl *lookupClassMethod(Selector Sel) const {
1467    return lookupMethod(Sel, false/*isInstance*/);
1468  }
1469
1470  /// \brief Determine whether this protocol has a definition.
1471  bool hasDefinition() const {
1472    // If the name of this protocol is out-of-date, bring it up-to-date, which
1473    // might bring in a definition.
1474    // Note: a null value indicates that we don't have a definition and that
1475    // modules are enabled.
1476    if (!Data.getOpaqueValue()) {
1477      if (IdentifierInfo *II = getIdentifier()) {
1478        if (II->isOutOfDate()) {
1479          updateOutOfDate(*II);
1480        }
1481      }
1482    }
1483
1484    return Data.getPointer();
1485  }
1486
1487  /// \brief Retrieve the definition of this protocol, if any.
1488  ObjCProtocolDecl *getDefinition() {
1489    return hasDefinition()? Data.getPointer()->Definition : 0;
1490  }
1491
1492  /// \brief Retrieve the definition of this protocol, if any.
1493  const ObjCProtocolDecl *getDefinition() const {
1494    return hasDefinition()? Data.getPointer()->Definition : 0;
1495  }
1496
1497  /// \brief Determine whether this particular declaration is also the
1498  /// definition.
1499  bool isThisDeclarationADefinition() const {
1500    return getDefinition() == this;
1501  }
1502
1503  /// \brief Starts the definition of this Objective-C protocol.
1504  void startDefinition();
1505
1506  virtual SourceRange getSourceRange() const LLVM_READONLY {
1507    if (isThisDeclarationADefinition())
1508      return ObjCContainerDecl::getSourceRange();
1509
1510    return SourceRange(getAtStartLoc(), getLocation());
1511  }
1512
1513  typedef redeclarable_base::redecl_iterator redecl_iterator;
1514  using redeclarable_base::redecls_begin;
1515  using redeclarable_base::redecls_end;
1516  using redeclarable_base::getPreviousDecl;
1517  using redeclarable_base::getMostRecentDecl;
1518  using redeclarable_base::isFirstDecl;
1519
1520  /// Retrieves the canonical declaration of this Objective-C protocol.
1521  ObjCProtocolDecl *getCanonicalDecl() { return getFirstDecl(); }
1522  const ObjCProtocolDecl *getCanonicalDecl() const { return getFirstDecl(); }
1523
1524  virtual void collectPropertiesToImplement(PropertyMap &PM,
1525                                            PropertyDeclOrder &PO) const;
1526
1527void collectInheritedProtocolProperties(const ObjCPropertyDecl *Property,
1528                                        ProtocolPropertyMap &PM) const;
1529
1530  static bool classof(const Decl *D) { return classofKind(D->getKind()); }
1531  static bool classofKind(Kind K) { return K == ObjCProtocol; }
1532
1533  friend class ASTReader;
1534  friend class ASTDeclReader;
1535  friend class ASTDeclWriter;
1536};
1537
1538/// ObjCCategoryDecl - Represents a category declaration. A category allows
1539/// you to add methods to an existing class (without subclassing or modifying
1540/// the original class interface or implementation:-). Categories don't allow
1541/// you to add instance data. The following example adds "myMethod" to all
1542/// NSView's within a process:
1543///
1544/// \@interface NSView (MyViewMethods)
1545/// - myMethod;
1546/// \@end
1547///
1548/// Categories also allow you to split the implementation of a class across
1549/// several files (a feature more naturally supported in C++).
1550///
1551/// Categories were originally inspired by dynamic languages such as Common
1552/// Lisp and Smalltalk.  More traditional class-based languages (C++, Java)
1553/// don't support this level of dynamism, which is both powerful and dangerous.
1554///
1555class ObjCCategoryDecl : public ObjCContainerDecl {
1556  virtual void anchor();
1557
1558  /// Interface belonging to this category
1559  ObjCInterfaceDecl *ClassInterface;
1560
1561  /// referenced protocols in this category.
1562  ObjCProtocolList ReferencedProtocols;
1563
1564  /// Next category belonging to this class.
1565  /// FIXME: this should not be a singly-linked list.  Move storage elsewhere.
1566  ObjCCategoryDecl *NextClassCategory;
1567
1568  /// \brief The location of the category name in this declaration.
1569  SourceLocation CategoryNameLoc;
1570
1571  /// class extension may have private ivars.
1572  SourceLocation IvarLBraceLoc;
1573  SourceLocation IvarRBraceLoc;
1574
1575  ObjCCategoryDecl(DeclContext *DC, SourceLocation AtLoc,
1576                   SourceLocation ClassNameLoc, SourceLocation CategoryNameLoc,
1577                   IdentifierInfo *Id, ObjCInterfaceDecl *IDecl,
1578                   SourceLocation IvarLBraceLoc=SourceLocation(),
1579                   SourceLocation IvarRBraceLoc=SourceLocation())
1580    : ObjCContainerDecl(ObjCCategory, DC, Id, ClassNameLoc, AtLoc),
1581      ClassInterface(IDecl), NextClassCategory(0),
1582      CategoryNameLoc(CategoryNameLoc),
1583      IvarLBraceLoc(IvarLBraceLoc), IvarRBraceLoc(IvarRBraceLoc) {
1584  }
1585
1586public:
1587
1588  static ObjCCategoryDecl *Create(ASTContext &C, DeclContext *DC,
1589                                  SourceLocation AtLoc,
1590                                  SourceLocation ClassNameLoc,
1591                                  SourceLocation CategoryNameLoc,
1592                                  IdentifierInfo *Id,
1593                                  ObjCInterfaceDecl *IDecl,
1594                                  SourceLocation IvarLBraceLoc=SourceLocation(),
1595                                  SourceLocation IvarRBraceLoc=SourceLocation());
1596  static ObjCCategoryDecl *CreateDeserialized(ASTContext &C, unsigned ID);
1597
1598  ObjCInterfaceDecl *getClassInterface() { return ClassInterface; }
1599  const ObjCInterfaceDecl *getClassInterface() const { return ClassInterface; }
1600
1601  ObjCCategoryImplDecl *getImplementation() const;
1602  void setImplementation(ObjCCategoryImplDecl *ImplD);
1603
1604  /// setProtocolList - Set the list of protocols that this interface
1605  /// implements.
1606  void setProtocolList(ObjCProtocolDecl *const*List, unsigned Num,
1607                       const SourceLocation *Locs, ASTContext &C) {
1608    ReferencedProtocols.set(List, Num, Locs, C);
1609  }
1610
1611  const ObjCProtocolList &getReferencedProtocols() const {
1612    return ReferencedProtocols;
1613  }
1614
1615  typedef ObjCProtocolList::iterator protocol_iterator;
1616  protocol_iterator protocol_begin() const {return ReferencedProtocols.begin();}
1617  protocol_iterator protocol_end() const { return ReferencedProtocols.end(); }
1618  unsigned protocol_size() const { return ReferencedProtocols.size(); }
1619  typedef ObjCProtocolList::loc_iterator protocol_loc_iterator;
1620  protocol_loc_iterator protocol_loc_begin() const {
1621    return ReferencedProtocols.loc_begin();
1622  }
1623  protocol_loc_iterator protocol_loc_end() const {
1624    return ReferencedProtocols.loc_end();
1625  }
1626
1627  ObjCCategoryDecl *getNextClassCategory() const { return NextClassCategory; }
1628
1629  /// \brief Retrieve the pointer to the next stored category (or extension),
1630  /// which may be hidden.
1631  ObjCCategoryDecl *getNextClassCategoryRaw() const {
1632    return NextClassCategory;
1633  }
1634
1635  bool IsClassExtension() const { return getIdentifier() == 0; }
1636
1637  typedef specific_decl_iterator<ObjCIvarDecl> ivar_iterator;
1638  ivar_iterator ivar_begin() const {
1639    return ivar_iterator(decls_begin());
1640  }
1641  ivar_iterator ivar_end() const {
1642    return ivar_iterator(decls_end());
1643  }
1644  unsigned ivar_size() const {
1645    return std::distance(ivar_begin(), ivar_end());
1646  }
1647  bool ivar_empty() const {
1648    return ivar_begin() == ivar_end();
1649  }
1650
1651  SourceLocation getCategoryNameLoc() const { return CategoryNameLoc; }
1652  void setCategoryNameLoc(SourceLocation Loc) { CategoryNameLoc = Loc; }
1653
1654  void setIvarLBraceLoc(SourceLocation Loc) { IvarLBraceLoc = Loc; }
1655  SourceLocation getIvarLBraceLoc() const { return IvarLBraceLoc; }
1656  void setIvarRBraceLoc(SourceLocation Loc) { IvarRBraceLoc = Loc; }
1657  SourceLocation getIvarRBraceLoc() const { return IvarRBraceLoc; }
1658
1659  static bool classof(const Decl *D) { return classofKind(D->getKind()); }
1660  static bool classofKind(Kind K) { return K == ObjCCategory; }
1661
1662  friend class ASTDeclReader;
1663  friend class ASTDeclWriter;
1664};
1665
1666class ObjCImplDecl : public ObjCContainerDecl {
1667  virtual void anchor();
1668
1669  /// Class interface for this class/category implementation
1670  ObjCInterfaceDecl *ClassInterface;
1671
1672protected:
1673  ObjCImplDecl(Kind DK, DeclContext *DC,
1674               ObjCInterfaceDecl *classInterface,
1675               SourceLocation nameLoc, SourceLocation atStartLoc)
1676    : ObjCContainerDecl(DK, DC,
1677                        classInterface? classInterface->getIdentifier() : 0,
1678                        nameLoc, atStartLoc),
1679      ClassInterface(classInterface) {}
1680
1681public:
1682  const ObjCInterfaceDecl *getClassInterface() const { return ClassInterface; }
1683  ObjCInterfaceDecl *getClassInterface() { return ClassInterface; }
1684  void setClassInterface(ObjCInterfaceDecl *IFace);
1685
1686  void addInstanceMethod(ObjCMethodDecl *method) {
1687    // FIXME: Context should be set correctly before we get here.
1688    method->setLexicalDeclContext(this);
1689    addDecl(method);
1690  }
1691  void addClassMethod(ObjCMethodDecl *method) {
1692    // FIXME: Context should be set correctly before we get here.
1693    method->setLexicalDeclContext(this);
1694    addDecl(method);
1695  }
1696
1697  void addPropertyImplementation(ObjCPropertyImplDecl *property);
1698
1699  ObjCPropertyImplDecl *FindPropertyImplDecl(IdentifierInfo *propertyId) const;
1700  ObjCPropertyImplDecl *FindPropertyImplIvarDecl(IdentifierInfo *ivarId) const;
1701
1702  // Iterator access to properties.
1703  typedef specific_decl_iterator<ObjCPropertyImplDecl> propimpl_iterator;
1704  propimpl_iterator propimpl_begin() const {
1705    return propimpl_iterator(decls_begin());
1706  }
1707  propimpl_iterator propimpl_end() const {
1708    return propimpl_iterator(decls_end());
1709  }
1710
1711  static bool classof(const Decl *D) { return classofKind(D->getKind()); }
1712  static bool classofKind(Kind K) {
1713    return K >= firstObjCImpl && K <= lastObjCImpl;
1714  }
1715};
1716
1717/// ObjCCategoryImplDecl - An object of this class encapsulates a category
1718/// \@implementation declaration. If a category class has declaration of a
1719/// property, its implementation must be specified in the category's
1720/// \@implementation declaration. Example:
1721/// \@interface I \@end
1722/// \@interface I(CATEGORY)
1723///    \@property int p1, d1;
1724/// \@end
1725/// \@implementation I(CATEGORY)
1726///  \@dynamic p1,d1;
1727/// \@end
1728///
1729/// ObjCCategoryImplDecl
1730class ObjCCategoryImplDecl : public ObjCImplDecl {
1731  virtual void anchor();
1732
1733  // Category name
1734  IdentifierInfo *Id;
1735
1736  // Category name location
1737  SourceLocation CategoryNameLoc;
1738
1739  ObjCCategoryImplDecl(DeclContext *DC, IdentifierInfo *Id,
1740                       ObjCInterfaceDecl *classInterface,
1741                       SourceLocation nameLoc, SourceLocation atStartLoc,
1742                       SourceLocation CategoryNameLoc)
1743    : ObjCImplDecl(ObjCCategoryImpl, DC, classInterface, nameLoc, atStartLoc),
1744      Id(Id), CategoryNameLoc(CategoryNameLoc) {}
1745public:
1746  static ObjCCategoryImplDecl *Create(ASTContext &C, DeclContext *DC,
1747                                      IdentifierInfo *Id,
1748                                      ObjCInterfaceDecl *classInterface,
1749                                      SourceLocation nameLoc,
1750                                      SourceLocation atStartLoc,
1751                                      SourceLocation CategoryNameLoc);
1752  static ObjCCategoryImplDecl *CreateDeserialized(ASTContext &C, unsigned ID);
1753
1754  /// getIdentifier - Get the identifier that names the category
1755  /// interface associated with this implementation.
1756  /// FIXME: This is a bad API, we are overriding the NamedDecl::getIdentifier()
1757  /// to mean something different. For example:
1758  /// ((NamedDecl *)SomeCategoryImplDecl)->getIdentifier()
1759  /// returns the class interface name, whereas
1760  /// ((ObjCCategoryImplDecl *)SomeCategoryImplDecl)->getIdentifier()
1761  /// returns the category name.
1762  IdentifierInfo *getIdentifier() const {
1763    return Id;
1764  }
1765  void setIdentifier(IdentifierInfo *II) { Id = II; }
1766
1767  ObjCCategoryDecl *getCategoryDecl() const;
1768
1769  SourceLocation getCategoryNameLoc() const { return CategoryNameLoc; }
1770
1771  /// getName - Get the name of identifier for the class interface associated
1772  /// with this implementation as a StringRef.
1773  //
1774  // FIXME: This is a bad API, we are overriding the NamedDecl::getName, to mean
1775  // something different.
1776  StringRef getName() const {
1777    return Id ? Id->getNameStart() : "";
1778  }
1779
1780  /// @brief Get the name of the class associated with this interface.
1781  //
1782  // FIXME: Deprecated, move clients to getName().
1783  std::string getNameAsString() const {
1784    return getName();
1785  }
1786
1787  static bool classof(const Decl *D) { return classofKind(D->getKind()); }
1788  static bool classofKind(Kind K) { return K == ObjCCategoryImpl;}
1789
1790  friend class ASTDeclReader;
1791  friend class ASTDeclWriter;
1792};
1793
1794raw_ostream &operator<<(raw_ostream &OS, const ObjCCategoryImplDecl &CID);
1795
1796/// ObjCImplementationDecl - Represents a class definition - this is where
1797/// method definitions are specified. For example:
1798///
1799/// @code
1800/// \@implementation MyClass
1801/// - (void)myMethod { /* do something */ }
1802/// \@end
1803/// @endcode
1804///
1805/// Typically, instance variables are specified in the class interface,
1806/// *not* in the implementation. Nevertheless (for legacy reasons), we
1807/// allow instance variables to be specified in the implementation.  When
1808/// specified, they need to be *identical* to the interface.
1809///
1810class ObjCImplementationDecl : public ObjCImplDecl {
1811  virtual void anchor();
1812  /// Implementation Class's super class.
1813  ObjCInterfaceDecl *SuperClass;
1814  SourceLocation SuperLoc;
1815
1816  /// \@implementation may have private ivars.
1817  SourceLocation IvarLBraceLoc;
1818  SourceLocation IvarRBraceLoc;
1819
1820  /// Support for ivar initialization.
1821  /// IvarInitializers - The arguments used to initialize the ivars
1822  CXXCtorInitializer **IvarInitializers;
1823  unsigned NumIvarInitializers;
1824
1825  /// Do the ivars of this class require initialization other than
1826  /// zero-initialization?
1827  bool HasNonZeroConstructors : 1;
1828
1829  /// Do the ivars of this class require non-trivial destruction?
1830  bool HasDestructors : 1;
1831
1832  ObjCImplementationDecl(DeclContext *DC,
1833                         ObjCInterfaceDecl *classInterface,
1834                         ObjCInterfaceDecl *superDecl,
1835                         SourceLocation nameLoc, SourceLocation atStartLoc,
1836                         SourceLocation superLoc = SourceLocation(),
1837                         SourceLocation IvarLBraceLoc=SourceLocation(),
1838                         SourceLocation IvarRBraceLoc=SourceLocation())
1839    : ObjCImplDecl(ObjCImplementation, DC, classInterface, nameLoc, atStartLoc),
1840       SuperClass(superDecl), SuperLoc(superLoc), IvarLBraceLoc(IvarLBraceLoc),
1841       IvarRBraceLoc(IvarRBraceLoc),
1842       IvarInitializers(0), NumIvarInitializers(0),
1843       HasNonZeroConstructors(false), HasDestructors(false) {}
1844public:
1845  static ObjCImplementationDecl *Create(ASTContext &C, DeclContext *DC,
1846                                        ObjCInterfaceDecl *classInterface,
1847                                        ObjCInterfaceDecl *superDecl,
1848                                        SourceLocation nameLoc,
1849                                        SourceLocation atStartLoc,
1850                                     SourceLocation superLoc = SourceLocation(),
1851                                        SourceLocation IvarLBraceLoc=SourceLocation(),
1852                                        SourceLocation IvarRBraceLoc=SourceLocation());
1853
1854  static ObjCImplementationDecl *CreateDeserialized(ASTContext &C, unsigned ID);
1855
1856  /// init_iterator - Iterates through the ivar initializer list.
1857  typedef CXXCtorInitializer **init_iterator;
1858
1859  /// init_const_iterator - Iterates through the ivar initializer list.
1860  typedef CXXCtorInitializer * const * init_const_iterator;
1861
1862  /// init_begin() - Retrieve an iterator to the first initializer.
1863  init_iterator       init_begin()       { return IvarInitializers; }
1864  /// begin() - Retrieve an iterator to the first initializer.
1865  init_const_iterator init_begin() const { return IvarInitializers; }
1866
1867  /// init_end() - Retrieve an iterator past the last initializer.
1868  init_iterator       init_end()       {
1869    return IvarInitializers + NumIvarInitializers;
1870  }
1871  /// end() - Retrieve an iterator past the last initializer.
1872  init_const_iterator init_end() const {
1873    return IvarInitializers + NumIvarInitializers;
1874  }
1875  /// getNumArgs - Number of ivars which must be initialized.
1876  unsigned getNumIvarInitializers() const {
1877    return NumIvarInitializers;
1878  }
1879
1880  void setNumIvarInitializers(unsigned numNumIvarInitializers) {
1881    NumIvarInitializers = numNumIvarInitializers;
1882  }
1883
1884  void setIvarInitializers(ASTContext &C,
1885                           CXXCtorInitializer ** initializers,
1886                           unsigned numInitializers);
1887
1888  /// Do any of the ivars of this class (not counting its base classes)
1889  /// require construction other than zero-initialization?
1890  bool hasNonZeroConstructors() const { return HasNonZeroConstructors; }
1891  void setHasNonZeroConstructors(bool val) { HasNonZeroConstructors = val; }
1892
1893  /// Do any of the ivars of this class (not counting its base classes)
1894  /// require non-trivial destruction?
1895  bool hasDestructors() const { return HasDestructors; }
1896  void setHasDestructors(bool val) { HasDestructors = val; }
1897
1898  /// getIdentifier - Get the identifier that names the class
1899  /// interface associated with this implementation.
1900  IdentifierInfo *getIdentifier() const {
1901    return getClassInterface()->getIdentifier();
1902  }
1903
1904  /// getName - Get the name of identifier for the class interface associated
1905  /// with this implementation as a StringRef.
1906  //
1907  // FIXME: This is a bad API, we are overriding the NamedDecl::getName, to mean
1908  // something different.
1909  StringRef getName() const {
1910    assert(getIdentifier() && "Name is not a simple identifier");
1911    return getIdentifier()->getName();
1912  }
1913
1914  /// @brief Get the name of the class associated with this interface.
1915  //
1916  // FIXME: Move to StringRef API.
1917  std::string getNameAsString() const {
1918    return getName();
1919  }
1920
1921  const ObjCInterfaceDecl *getSuperClass() const { return SuperClass; }
1922  ObjCInterfaceDecl *getSuperClass() { return SuperClass; }
1923  SourceLocation getSuperClassLoc() const { return SuperLoc; }
1924
1925  void setSuperClass(ObjCInterfaceDecl * superCls) { SuperClass = superCls; }
1926
1927  void setIvarLBraceLoc(SourceLocation Loc) { IvarLBraceLoc = Loc; }
1928  SourceLocation getIvarLBraceLoc() const { return IvarLBraceLoc; }
1929  void setIvarRBraceLoc(SourceLocation Loc) { IvarRBraceLoc = Loc; }
1930  SourceLocation getIvarRBraceLoc() const { return IvarRBraceLoc; }
1931
1932  typedef specific_decl_iterator<ObjCIvarDecl> ivar_iterator;
1933  ivar_iterator ivar_begin() const {
1934    return ivar_iterator(decls_begin());
1935  }
1936  ivar_iterator ivar_end() const {
1937    return ivar_iterator(decls_end());
1938  }
1939  unsigned ivar_size() const {
1940    return std::distance(ivar_begin(), ivar_end());
1941  }
1942  bool ivar_empty() const {
1943    return ivar_begin() == ivar_end();
1944  }
1945
1946  static bool classof(const Decl *D) { return classofKind(D->getKind()); }
1947  static bool classofKind(Kind K) { return K == ObjCImplementation; }
1948
1949  friend class ASTDeclReader;
1950  friend class ASTDeclWriter;
1951};
1952
1953raw_ostream &operator<<(raw_ostream &OS, const ObjCImplementationDecl &ID);
1954
1955/// ObjCCompatibleAliasDecl - Represents alias of a class. This alias is
1956/// declared as \@compatibility_alias alias class.
1957class ObjCCompatibleAliasDecl : public NamedDecl {
1958  virtual void anchor();
1959  /// Class that this is an alias of.
1960  ObjCInterfaceDecl *AliasedClass;
1961
1962  ObjCCompatibleAliasDecl(DeclContext *DC, SourceLocation L, IdentifierInfo *Id,
1963                          ObjCInterfaceDecl* aliasedClass)
1964    : NamedDecl(ObjCCompatibleAlias, DC, L, Id), AliasedClass(aliasedClass) {}
1965public:
1966  static ObjCCompatibleAliasDecl *Create(ASTContext &C, DeclContext *DC,
1967                                         SourceLocation L, IdentifierInfo *Id,
1968                                         ObjCInterfaceDecl* aliasedClass);
1969
1970  static ObjCCompatibleAliasDecl *CreateDeserialized(ASTContext &C,
1971                                                     unsigned ID);
1972
1973  const ObjCInterfaceDecl *getClassInterface() const { return AliasedClass; }
1974  ObjCInterfaceDecl *getClassInterface() { return AliasedClass; }
1975  void setClassInterface(ObjCInterfaceDecl *D) { AliasedClass = D; }
1976
1977  static bool classof(const Decl *D) { return classofKind(D->getKind()); }
1978  static bool classofKind(Kind K) { return K == ObjCCompatibleAlias; }
1979
1980};
1981
1982/// \brief Represents one property declaration in an Objective-C interface.
1983///
1984/// For example:
1985/// \code{.mm}
1986/// \@property (assign, readwrite) int MyProperty;
1987/// \endcode
1988class ObjCPropertyDecl : public NamedDecl {
1989  virtual void anchor();
1990public:
1991  enum PropertyAttributeKind {
1992    OBJC_PR_noattr    = 0x00,
1993    OBJC_PR_readonly  = 0x01,
1994    OBJC_PR_getter    = 0x02,
1995    OBJC_PR_assign    = 0x04,
1996    OBJC_PR_readwrite = 0x08,
1997    OBJC_PR_retain    = 0x10,
1998    OBJC_PR_copy      = 0x20,
1999    OBJC_PR_nonatomic = 0x40,
2000    OBJC_PR_setter    = 0x80,
2001    OBJC_PR_atomic    = 0x100,
2002    OBJC_PR_weak      = 0x200,
2003    OBJC_PR_strong    = 0x400,
2004    OBJC_PR_unsafe_unretained = 0x800
2005    // Adding a property should change NumPropertyAttrsBits
2006  };
2007
2008  enum {
2009    /// \brief Number of bits fitting all the property attributes.
2010    NumPropertyAttrsBits = 12
2011  };
2012
2013  enum SetterKind { Assign, Retain, Copy, Weak };
2014  enum PropertyControl { None, Required, Optional };
2015private:
2016  SourceLocation AtLoc;   // location of \@property
2017  SourceLocation LParenLoc; // location of '(' starting attribute list or null.
2018  TypeSourceInfo *DeclType;
2019  unsigned PropertyAttributes : NumPropertyAttrsBits;
2020  unsigned PropertyAttributesAsWritten : NumPropertyAttrsBits;
2021  // \@required/\@optional
2022  unsigned PropertyImplementation : 2;
2023
2024  Selector GetterName;    // getter name of NULL if no getter
2025  Selector SetterName;    // setter name of NULL if no setter
2026
2027  ObjCMethodDecl *GetterMethodDecl; // Declaration of getter instance method
2028  ObjCMethodDecl *SetterMethodDecl; // Declaration of setter instance method
2029  ObjCIvarDecl *PropertyIvarDecl;   // Synthesize ivar for this property
2030
2031  ObjCPropertyDecl(DeclContext *DC, SourceLocation L, IdentifierInfo *Id,
2032                   SourceLocation AtLocation,  SourceLocation LParenLocation,
2033                   TypeSourceInfo *T)
2034    : NamedDecl(ObjCProperty, DC, L, Id), AtLoc(AtLocation),
2035      LParenLoc(LParenLocation), DeclType(T),
2036      PropertyAttributes(OBJC_PR_noattr),
2037      PropertyAttributesAsWritten(OBJC_PR_noattr),
2038      PropertyImplementation(None),
2039      GetterName(Selector()),
2040      SetterName(Selector()),
2041      GetterMethodDecl(0), SetterMethodDecl(0) , PropertyIvarDecl(0) {}
2042public:
2043  static ObjCPropertyDecl *Create(ASTContext &C, DeclContext *DC,
2044                                  SourceLocation L,
2045                                  IdentifierInfo *Id, SourceLocation AtLocation,
2046                                  SourceLocation LParenLocation,
2047                                  TypeSourceInfo *T,
2048                                  PropertyControl propControl = None);
2049
2050  static ObjCPropertyDecl *CreateDeserialized(ASTContext &C, unsigned ID);
2051
2052  SourceLocation getAtLoc() const { return AtLoc; }
2053  void setAtLoc(SourceLocation L) { AtLoc = L; }
2054
2055  SourceLocation getLParenLoc() const { return LParenLoc; }
2056  void setLParenLoc(SourceLocation L) { LParenLoc = L; }
2057
2058  TypeSourceInfo *getTypeSourceInfo() const { return DeclType; }
2059  QualType getType() const { return DeclType->getType(); }
2060  void setType(TypeSourceInfo *T) { DeclType = T; }
2061
2062  PropertyAttributeKind getPropertyAttributes() const {
2063    return PropertyAttributeKind(PropertyAttributes);
2064  }
2065  void setPropertyAttributes(PropertyAttributeKind PRVal) {
2066    PropertyAttributes |= PRVal;
2067  }
2068
2069  PropertyAttributeKind getPropertyAttributesAsWritten() const {
2070    return PropertyAttributeKind(PropertyAttributesAsWritten);
2071  }
2072
2073  bool hasWrittenStorageAttribute() const {
2074    return PropertyAttributesAsWritten & (OBJC_PR_assign | OBJC_PR_copy |
2075        OBJC_PR_unsafe_unretained | OBJC_PR_retain | OBJC_PR_strong |
2076        OBJC_PR_weak);
2077  }
2078
2079  void setPropertyAttributesAsWritten(PropertyAttributeKind PRVal) {
2080    PropertyAttributesAsWritten = PRVal;
2081  }
2082
2083 void makeitReadWriteAttribute() {
2084    PropertyAttributes &= ~OBJC_PR_readonly;
2085    PropertyAttributes |= OBJC_PR_readwrite;
2086 }
2087
2088  // Helper methods for accessing attributes.
2089
2090  /// isReadOnly - Return true iff the property has a setter.
2091  bool isReadOnly() const {
2092    return (PropertyAttributes & OBJC_PR_readonly);
2093  }
2094
2095  /// isAtomic - Return true if the property is atomic.
2096  bool isAtomic() const {
2097    return (PropertyAttributes & OBJC_PR_atomic);
2098  }
2099
2100  /// isRetaining - Return true if the property retains its value.
2101  bool isRetaining() const {
2102    return (PropertyAttributes &
2103            (OBJC_PR_retain | OBJC_PR_strong | OBJC_PR_copy));
2104  }
2105
2106  /// getSetterKind - Return the method used for doing assignment in
2107  /// the property setter. This is only valid if the property has been
2108  /// defined to have a setter.
2109  SetterKind getSetterKind() const {
2110    if (PropertyAttributes & OBJC_PR_strong)
2111      return getType()->isBlockPointerType() ? Copy : Retain;
2112    if (PropertyAttributes & OBJC_PR_retain)
2113      return Retain;
2114    if (PropertyAttributes & OBJC_PR_copy)
2115      return Copy;
2116    if (PropertyAttributes & OBJC_PR_weak)
2117      return Weak;
2118    return Assign;
2119  }
2120
2121  Selector getGetterName() const { return GetterName; }
2122  void setGetterName(Selector Sel) { GetterName = Sel; }
2123
2124  Selector getSetterName() const { return SetterName; }
2125  void setSetterName(Selector Sel) { SetterName = Sel; }
2126
2127  ObjCMethodDecl *getGetterMethodDecl() const { return GetterMethodDecl; }
2128  void setGetterMethodDecl(ObjCMethodDecl *gDecl) { GetterMethodDecl = gDecl; }
2129
2130  ObjCMethodDecl *getSetterMethodDecl() const { return SetterMethodDecl; }
2131  void setSetterMethodDecl(ObjCMethodDecl *gDecl) { SetterMethodDecl = gDecl; }
2132
2133  // Related to \@optional/\@required declared in \@protocol
2134  void setPropertyImplementation(PropertyControl pc) {
2135    PropertyImplementation = pc;
2136  }
2137  PropertyControl getPropertyImplementation() const {
2138    return PropertyControl(PropertyImplementation);
2139  }
2140
2141  void setPropertyIvarDecl(ObjCIvarDecl *Ivar) {
2142    PropertyIvarDecl = Ivar;
2143  }
2144  ObjCIvarDecl *getPropertyIvarDecl() const {
2145    return PropertyIvarDecl;
2146  }
2147
2148  virtual SourceRange getSourceRange() const LLVM_READONLY {
2149    return SourceRange(AtLoc, getLocation());
2150  }
2151
2152  /// Get the default name of the synthesized ivar.
2153  IdentifierInfo *getDefaultSynthIvarName(ASTContext &Ctx) const;
2154
2155  /// Lookup a property by name in the specified DeclContext.
2156  static ObjCPropertyDecl *findPropertyDecl(const DeclContext *DC,
2157                                            IdentifierInfo *propertyID);
2158
2159  static bool classof(const Decl *D) { return classofKind(D->getKind()); }
2160  static bool classofKind(Kind K) { return K == ObjCProperty; }
2161};
2162
2163/// ObjCPropertyImplDecl - Represents implementation declaration of a property
2164/// in a class or category implementation block. For example:
2165/// \@synthesize prop1 = ivar1;
2166///
2167class ObjCPropertyImplDecl : public Decl {
2168public:
2169  enum Kind {
2170    Synthesize,
2171    Dynamic
2172  };
2173private:
2174  SourceLocation AtLoc;   // location of \@synthesize or \@dynamic
2175
2176  /// \brief For \@synthesize, the location of the ivar, if it was written in
2177  /// the source code.
2178  ///
2179  /// \code
2180  /// \@synthesize int a = b
2181  /// \endcode
2182  SourceLocation IvarLoc;
2183
2184  /// Property declaration being implemented
2185  ObjCPropertyDecl *PropertyDecl;
2186
2187  /// Null for \@dynamic. Required for \@synthesize.
2188  ObjCIvarDecl *PropertyIvarDecl;
2189
2190  /// Null for \@dynamic. Non-null if property must be copy-constructed in
2191  /// getter.
2192  Expr *GetterCXXConstructor;
2193
2194  /// Null for \@dynamic. Non-null if property has assignment operator to call
2195  /// in Setter synthesis.
2196  Expr *SetterCXXAssignment;
2197
2198  ObjCPropertyImplDecl(DeclContext *DC, SourceLocation atLoc, SourceLocation L,
2199                       ObjCPropertyDecl *property,
2200                       Kind PK,
2201                       ObjCIvarDecl *ivarDecl,
2202                       SourceLocation ivarLoc)
2203    : Decl(ObjCPropertyImpl, DC, L), AtLoc(atLoc),
2204      IvarLoc(ivarLoc), PropertyDecl(property), PropertyIvarDecl(ivarDecl),
2205      GetterCXXConstructor(0), SetterCXXAssignment(0) {
2206    assert (PK == Dynamic || PropertyIvarDecl);
2207  }
2208
2209public:
2210  static ObjCPropertyImplDecl *Create(ASTContext &C, DeclContext *DC,
2211                                      SourceLocation atLoc, SourceLocation L,
2212                                      ObjCPropertyDecl *property,
2213                                      Kind PK,
2214                                      ObjCIvarDecl *ivarDecl,
2215                                      SourceLocation ivarLoc);
2216
2217  static ObjCPropertyImplDecl *CreateDeserialized(ASTContext &C, unsigned ID);
2218
2219  virtual SourceRange getSourceRange() const LLVM_READONLY;
2220
2221  SourceLocation getLocStart() const LLVM_READONLY { return AtLoc; }
2222  void setAtLoc(SourceLocation Loc) { AtLoc = Loc; }
2223
2224  ObjCPropertyDecl *getPropertyDecl() const {
2225    return PropertyDecl;
2226  }
2227  void setPropertyDecl(ObjCPropertyDecl *Prop) { PropertyDecl = Prop; }
2228
2229  Kind getPropertyImplementation() const {
2230    return PropertyIvarDecl ? Synthesize : Dynamic;
2231  }
2232
2233  ObjCIvarDecl *getPropertyIvarDecl() const {
2234    return PropertyIvarDecl;
2235  }
2236  SourceLocation getPropertyIvarDeclLoc() const { return IvarLoc; }
2237
2238  void setPropertyIvarDecl(ObjCIvarDecl *Ivar,
2239                           SourceLocation IvarLoc) {
2240    PropertyIvarDecl = Ivar;
2241    this->IvarLoc = IvarLoc;
2242  }
2243
2244  /// \brief For \@synthesize, returns true if an ivar name was explicitly
2245  /// specified.
2246  ///
2247  /// \code
2248  /// \@synthesize int a = b; // true
2249  /// \@synthesize int a; // false
2250  /// \endcode
2251  bool isIvarNameSpecified() const {
2252    return IvarLoc.isValid() && IvarLoc != getLocation();
2253  }
2254
2255  Expr *getGetterCXXConstructor() const {
2256    return GetterCXXConstructor;
2257  }
2258  void setGetterCXXConstructor(Expr *getterCXXConstructor) {
2259    GetterCXXConstructor = getterCXXConstructor;
2260  }
2261
2262  Expr *getSetterCXXAssignment() const {
2263    return SetterCXXAssignment;
2264  }
2265  void setSetterCXXAssignment(Expr *setterCXXAssignment) {
2266    SetterCXXAssignment = setterCXXAssignment;
2267  }
2268
2269  static bool classof(const Decl *D) { return classofKind(D->getKind()); }
2270  static bool classofKind(Decl::Kind K) { return K == ObjCPropertyImpl; }
2271
2272  friend class ASTDeclReader;
2273};
2274
2275template<bool (*Filter)(ObjCCategoryDecl *)>
2276void
2277ObjCInterfaceDecl::filtered_category_iterator<Filter>::
2278findAcceptableCategory() {
2279  while (Current && !Filter(Current))
2280    Current = Current->getNextClassCategoryRaw();
2281}
2282
2283template<bool (*Filter)(ObjCCategoryDecl *)>
2284inline ObjCInterfaceDecl::filtered_category_iterator<Filter> &
2285ObjCInterfaceDecl::filtered_category_iterator<Filter>::operator++() {
2286  Current = Current->getNextClassCategoryRaw();
2287  findAcceptableCategory();
2288  return *this;
2289}
2290
2291inline bool ObjCInterfaceDecl::isVisibleCategory(ObjCCategoryDecl *Cat) {
2292  return !Cat->isHidden();
2293}
2294
2295inline bool ObjCInterfaceDecl::isVisibleExtension(ObjCCategoryDecl *Cat) {
2296  return Cat->IsClassExtension() && !Cat->isHidden();
2297}
2298
2299inline bool ObjCInterfaceDecl::isKnownExtension(ObjCCategoryDecl *Cat) {
2300  return Cat->IsClassExtension();
2301}
2302
2303}  // end namespace clang
2304#endif
2305