DeclObjC.cpp revision 55d13b4d5530a14d5baa72adab32ae78ba256caf
11e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner//===--- DeclObjC.cpp - ObjC Declaration AST Node Implementation ----------===// 21e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// 31e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// The LLVM Compiler Infrastructure 41e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// 51e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// This file is distributed under the University of Illinois Open Source 61e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// License. See LICENSE.TXT for details. 71e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// 81e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner//===----------------------------------------------------------------------===// 91e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// 101e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// This file implements the Objective-C related Decl classes. 111e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// 121e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner//===----------------------------------------------------------------------===// 131e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 141e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner#include "clang/AST/DeclObjC.h" 151e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner#include "clang/AST/ASTContext.h" 161e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattnerusing namespace clang; 171e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 186c4ae5de0c356777446f823b573821fb95560d91Chris Lattner//===----------------------------------------------------------------------===// 196c4ae5de0c356777446f823b573821fb95560d91Chris Lattner// ObjC Decl Allocation/Deallocation Method Implementations 206c4ae5de0c356777446f823b573821fb95560d91Chris Lattner//===----------------------------------------------------------------------===// 216c4ae5de0c356777446f823b573821fb95560d91Chris Lattner 226c4ae5de0c356777446f823b573821fb95560d91Chris LattnerObjCMethodDecl *ObjCMethodDecl::Create(ASTContext &C, SourceLocation beginLoc, 236c4ae5de0c356777446f823b573821fb95560d91Chris Lattner SourceLocation endLoc, 246c4ae5de0c356777446f823b573821fb95560d91Chris Lattner Selector SelInfo, QualType T, 256c4ae5de0c356777446f823b573821fb95560d91Chris Lattner Decl *contextDecl, 266c4ae5de0c356777446f823b573821fb95560d91Chris Lattner AttributeList *M, bool isInstance, 276c4ae5de0c356777446f823b573821fb95560d91Chris Lattner bool isVariadic, 28b06fa3b86951b9f179c99c3768331536c32e902dChris Lattner ImplementationControl impControl) { 296c4ae5de0c356777446f823b573821fb95560d91Chris Lattner void *Mem = C.getAllocator().Allocate<ObjCMethodDecl>(); 306c4ae5de0c356777446f823b573821fb95560d91Chris Lattner return new (Mem) ObjCMethodDecl(beginLoc, endLoc, SelInfo, T, contextDecl, 31b06fa3b86951b9f179c99c3768331536c32e902dChris Lattner M, isInstance, 32b06fa3b86951b9f179c99c3768331536c32e902dChris Lattner isVariadic, impControl); 330e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner} 340e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner 350e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris LattnerObjCInterfaceDecl *ObjCInterfaceDecl::Create(ASTContext &C,SourceLocation atLoc, 360e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner unsigned numRefProtos, 370e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner IdentifierInfo *Id, 380e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner bool ForwardDecl, bool isInternal){ 390e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner void *Mem = C.getAllocator().Allocate<ObjCInterfaceDecl>(); 400e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner return new (Mem) ObjCInterfaceDecl(atLoc, numRefProtos, Id, ForwardDecl, 410e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner isInternal); 420e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner} 436c4ae5de0c356777446f823b573821fb95560d91Chris Lattner 440e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris LattnerObjCIvarDecl *ObjCIvarDecl::Create(ASTContext &C, SourceLocation L, 450e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner IdentifierInfo *Id, QualType T) { 460e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner void *Mem = C.getAllocator().Allocate<ObjCIvarDecl>(); 470e77ba0bf769e2e5a4a93c079f241b02aeb3ef93Chris Lattner return new (Mem) ObjCIvarDecl(L, Id, T); 486c4ae5de0c356777446f823b573821fb95560d91Chris Lattner} 496c4ae5de0c356777446f823b573821fb95560d91Chris Lattner 50cca59d77c4b84fd2da268018dbaf9431a621e75bChris LattnerObjCProtocolDecl *ObjCProtocolDecl::Create(ASTContext &C, SourceLocation L, 51cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner unsigned numRefProtos, 52c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner IdentifierInfo *Id) { 53cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner void *Mem = C.getAllocator().Allocate<ObjCProtocolDecl>(); 54c858105d41602a2dadb2efbc1af80a7b791ebac3Chris Lattner return new (Mem) ObjCProtocolDecl(L, numRefProtos, Id); 55cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner} 56cca59d77c4b84fd2da268018dbaf9431a621e75bChris Lattner 5761f9d41036e30ff80130f99b31c0626e3ef057ccChris LattnerObjCClassDecl *ObjCClassDecl::Create(ASTContext &C, SourceLocation L, 5861f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner ObjCInterfaceDecl **Elts, unsigned nElts) { 5961f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner void *Mem = C.getAllocator().Allocate<ObjCClassDecl>(); 6061f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner return new (Mem) ObjCClassDecl(L, Elts, nElts); 6161f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner} 6261f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner 6361f9d41036e30ff80130f99b31c0626e3ef057ccChris LattnerObjCForwardProtocolDecl * 6461f9d41036e30ff80130f99b31c0626e3ef057ccChris LattnerObjCForwardProtocolDecl::Create(ASTContext &C, SourceLocation L, 6561f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner ObjCProtocolDecl **Elts, unsigned NumElts) { 6661f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner void *Mem = C.getAllocator().Allocate<ObjCForwardProtocolDecl>(); 6761f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner return new (Mem) ObjCForwardProtocolDecl(L, Elts, NumElts); 6861f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner} 6961f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner 7061f9d41036e30ff80130f99b31c0626e3ef057ccChris LattnerObjCCategoryDecl *ObjCCategoryDecl::Create(ASTContext &C, SourceLocation L, 7161f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner IdentifierInfo *Id) { 7261f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner void *Mem = C.getAllocator().Allocate<ObjCCategoryDecl>(); 7368c82cf61228102aba1194efef222fa1478af2a8Chris Lattner return new (Mem) ObjCCategoryDecl(L, Id); 7461f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner} 7561f9d41036e30ff80130f99b31c0626e3ef057ccChris Lattner 7675c9cae5f85c72cbb1649e93849e16ede3f07522Chris LattnerObjCCategoryImplDecl * 7775c9cae5f85c72cbb1649e93849e16ede3f07522Chris LattnerObjCCategoryImplDecl::Create(ASTContext &C, SourceLocation L,IdentifierInfo *Id, 7875c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner ObjCInterfaceDecl *ClassInterface) { 7975c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner void *Mem = C.getAllocator().Allocate<ObjCCategoryImplDecl>(); 8075c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner return new (Mem) ObjCCategoryImplDecl(L, Id, ClassInterface); 8175c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner} 8275c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner 8375c9cae5f85c72cbb1649e93849e16ede3f07522Chris LattnerObjCImplementationDecl * 8475c9cae5f85c72cbb1649e93849e16ede3f07522Chris LattnerObjCImplementationDecl::Create(ASTContext &C, SourceLocation L, 8575c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner IdentifierInfo *Id, 8675c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner ObjCInterfaceDecl *ClassInterface, 8775c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner ObjCInterfaceDecl *SuperDecl) { 8875c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner void *Mem = C.getAllocator().Allocate<ObjCImplementationDecl>(); 8975c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner return new (Mem) ObjCImplementationDecl(L, Id, ClassInterface, SuperDecl); 9075c9cae5f85c72cbb1649e93849e16ede3f07522Chris Lattner} 911e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 92f8d17a59167d9c2026506ed8813ea434d93b662aChris LattnerObjCCompatibleAliasDecl * 93f8d17a59167d9c2026506ed8813ea434d93b662aChris LattnerObjCCompatibleAliasDecl::Create(ASTContext &C, SourceLocation L, 94f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner IdentifierInfo *Id, 95f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner ObjCInterfaceDecl* AliasedClass) { 96f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner void *Mem = C.getAllocator().Allocate<ObjCCompatibleAliasDecl>(); 97f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner return new (Mem) ObjCCompatibleAliasDecl(L, Id, AliasedClass); 98f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner} 99f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner 100f8d17a59167d9c2026506ed8813ea434d93b662aChris LattnerObjCPropertyDecl *ObjCPropertyDecl::Create(ASTContext &C, SourceLocation L) { 101f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner void *Mem = C.getAllocator().Allocate<ObjCPropertyDecl>(); 102f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner return new (Mem) ObjCPropertyDecl(L); 103f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner} 104f8d17a59167d9c2026506ed8813ea434d93b662aChris Lattner 1051e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner//===----------------------------------------------------------------------===// 1061e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// Objective-C Decl Implementation 1071e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner//===----------------------------------------------------------------------===// 1081e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 1091e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattnervoid ObjCMethodDecl::setMethodParams(ParmVarDecl **NewParamInfo, 1101e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner unsigned NumParams) { 1111e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner assert(ParamInfo == 0 && "Already has param info!"); 1121e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 1131e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner // Zero params -> null pointer. 1141e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (NumParams) { 1151e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ParamInfo = new ParmVarDecl*[NumParams]; 1161e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner memcpy(ParamInfo, NewParamInfo, sizeof(ParmVarDecl*)*NumParams); 1171e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner NumMethodParams = NumParams; 1181e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 1191e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 1201e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 1211e03a561f4bd96910cb31a8af53a6ad321a12b51Chris LattnerObjCMethodDecl::~ObjCMethodDecl() { 1221e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner delete[] ParamInfo; 1231e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 1241e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 1251e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// ObjCAddInstanceVariablesToClass - Inserts instance variables 1261e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// into ObjCInterfaceDecl's fields. 1271e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// 1281e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattnervoid ObjCInterfaceDecl::addInstanceVariablesToClass(ObjCIvarDecl **ivars, 1291e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner unsigned numIvars, 1301e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner SourceLocation RBrac) { 1311e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner NumIvars = numIvars; 1321e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (numIvars) { 1331e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner Ivars = new ObjCIvarDecl*[numIvars]; 1341e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner memcpy(Ivars, ivars, numIvars*sizeof(ObjCIvarDecl*)); 1351e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 1361e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner setLocEnd(RBrac); 1371e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 1381e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 1391e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// ObjCAddInstanceVariablesToClassImpl - Checks for correctness of Instance 1401e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// Variables (Ivars) relative to what declared in @implementation;s class. 1411e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// Ivars into ObjCImplementationDecl's fields. 1421e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// 1431e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattnervoid ObjCImplementationDecl::ObjCAddInstanceVariablesToClassImpl( 1441e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCIvarDecl **ivars, unsigned numIvars) { 1451e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner NumIvars = numIvars; 1461e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (numIvars) { 1471e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner Ivars = new ObjCIvarDecl*[numIvars]; 1481e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner memcpy(Ivars, ivars, numIvars*sizeof(ObjCIvarDecl*)); 1491e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 1501e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 1511e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 1521e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// addMethods - Insert instance and methods declarations into 1531e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// ObjCInterfaceDecl's InsMethods and ClsMethods fields. 1541e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// 1551e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattnervoid ObjCInterfaceDecl::addMethods(ObjCMethodDecl **insMethods, 1561e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner unsigned numInsMembers, 1571e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCMethodDecl **clsMethods, 1581e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner unsigned numClsMembers, 1591e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner SourceLocation endLoc) { 1601e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner NumInstanceMethods = numInsMembers; 1611e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (numInsMembers) { 1621e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner InstanceMethods = new ObjCMethodDecl*[numInsMembers]; 1631e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner memcpy(InstanceMethods, insMethods, numInsMembers*sizeof(ObjCMethodDecl*)); 1641e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 1651e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner NumClassMethods = numClsMembers; 1661e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (numClsMembers) { 1671e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ClassMethods = new ObjCMethodDecl*[numClsMembers]; 1681e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner memcpy(ClassMethods, clsMethods, numClsMembers*sizeof(ObjCMethodDecl*)); 1691e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 1701e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner AtEndLoc = endLoc; 1711e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 1721e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 1731e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// addMethods - Insert instance and methods declarations into 17455d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner/// ObjCInterfaceDecl's InsMethods and ClsMethods fields. 17555d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner/// 17655d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattnervoid ObjCInterfaceDecl::addProperties(ObjCPropertyDecl **Properties, 17755d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner unsigned NumProperties) { 17855d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner if (NumProperties == 0) return; 17955d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner 18055d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner NumPropertyDecl = NumProperties; 18155d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner PropertyDecl = new ObjCPropertyDecl*[NumProperties]; 18255d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner memcpy(PropertyDecl, Properties, NumProperties*sizeof(ObjCPropertyDecl*)); 18355d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner} 18455d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner 18555d13b4d5530a14d5baa72adab32ae78ba256cafChris Lattner/// addMethods - Insert instance and methods declarations into 1861e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// ObjCProtocolDecl's ProtoInsMethods and ProtoClsMethods fields. 1871e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// 1881e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattnervoid ObjCProtocolDecl::addMethods(ObjCMethodDecl **insMethods, 1891e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner unsigned numInsMembers, 1901e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCMethodDecl **clsMethods, 1911e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner unsigned numClsMembers, 1921e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner SourceLocation endLoc) { 1931e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner NumInstanceMethods = numInsMembers; 1941e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (numInsMembers) { 1951e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner InstanceMethods = new ObjCMethodDecl*[numInsMembers]; 1961e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner memcpy(InstanceMethods, insMethods, numInsMembers*sizeof(ObjCMethodDecl*)); 1971e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 1981e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner NumClassMethods = numClsMembers; 1991e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (numClsMembers) { 2001e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ClassMethods = new ObjCMethodDecl*[numClsMembers]; 2011e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner memcpy(ClassMethods, clsMethods, numClsMembers*sizeof(ObjCMethodDecl*)); 2021e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 2031e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner AtEndLoc = endLoc; 2041e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 2051e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 20668c82cf61228102aba1194efef222fa1478af2a8Chris Lattnervoid ObjCCategoryDecl::setReferencedProtocolList(ObjCProtocolDecl **List, 20768c82cf61228102aba1194efef222fa1478af2a8Chris Lattner unsigned NumRPs) { 20868c82cf61228102aba1194efef222fa1478af2a8Chris Lattner assert(NumReferencedProtocols == 0 && "Protocol list already set"); 20968c82cf61228102aba1194efef222fa1478af2a8Chris Lattner if (NumRPs == 0) return; 21068c82cf61228102aba1194efef222fa1478af2a8Chris Lattner 21168c82cf61228102aba1194efef222fa1478af2a8Chris Lattner ReferencedProtocols = new ObjCProtocolDecl*[NumRPs]; 21268c82cf61228102aba1194efef222fa1478af2a8Chris Lattner memcpy(ReferencedProtocols, List, NumRPs*sizeof(ObjCProtocolDecl*)); 21368c82cf61228102aba1194efef222fa1478af2a8Chris Lattner NumReferencedProtocols = NumRPs; 21468c82cf61228102aba1194efef222fa1478af2a8Chris Lattner} 21568c82cf61228102aba1194efef222fa1478af2a8Chris Lattner 21668c82cf61228102aba1194efef222fa1478af2a8Chris Lattner 2171e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// addMethods - Insert instance and methods declarations into 2181e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// ObjCCategoryDecl's CatInsMethods and CatClsMethods fields. 2191e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// 2201e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattnervoid ObjCCategoryDecl::addMethods(ObjCMethodDecl **insMethods, 2211e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner unsigned numInsMembers, 2221e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCMethodDecl **clsMethods, 2231e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner unsigned numClsMembers, 2241e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner SourceLocation endLoc) { 2251e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner NumInstanceMethods = numInsMembers; 2261e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (numInsMembers) { 2271e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner InstanceMethods = new ObjCMethodDecl*[numInsMembers]; 2281e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner memcpy(InstanceMethods, insMethods, numInsMembers*sizeof(ObjCMethodDecl*)); 2291e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 2301e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner NumClassMethods = numClsMembers; 2311e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (numClsMembers) { 2321e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ClassMethods = new ObjCMethodDecl*[numClsMembers]; 2331e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner memcpy(ClassMethods, clsMethods, numClsMembers*sizeof(ObjCMethodDecl*)); 2341e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 2351e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner AtEndLoc = endLoc; 2361e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 2371e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 2381e03a561f4bd96910cb31a8af53a6ad321a12b51Chris LattnerObjCIvarDecl *ObjCInterfaceDecl::lookupInstanceVariable( 2391e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner IdentifierInfo *ID, ObjCInterfaceDecl *&clsDeclared) { 2401e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCInterfaceDecl* ClassDecl = this; 2411e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner while (ClassDecl != NULL) { 2421e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner for (ivar_iterator I = ClassDecl->ivar_begin(), E = ClassDecl->ivar_end(); 2431e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner I != E; ++I) { 2441e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((*I)->getIdentifier() == ID) { 2451e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner clsDeclared = ClassDecl; 2461e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return *I; 2471e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 2481e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 2491e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ClassDecl = ClassDecl->getSuperClass(); 2501e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 2511e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return NULL; 2521e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 2531e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 2541e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// lookupInstanceMethod - This method returns an instance method by looking in 2551e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// the class, its categories, and its super classes (using a linear search). 2561e03a561f4bd96910cb31a8af53a6ad321a12b51Chris LattnerObjCMethodDecl *ObjCInterfaceDecl::lookupInstanceMethod(Selector Sel) { 2571e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCInterfaceDecl* ClassDecl = this; 2581e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCMethodDecl *MethodDecl = 0; 2591e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 2601e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner while (ClassDecl != NULL) { 2611e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((MethodDecl = ClassDecl->getInstanceMethod(Sel))) 2621e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return MethodDecl; 2631e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 2641e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner // Didn't find one yet - look through protocols. 2651e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCProtocolDecl **protocols = ClassDecl->getReferencedProtocols(); 2661e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner int numProtocols = ClassDecl->getNumIntfRefProtocols(); 2671e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner for (int pIdx = 0; pIdx < numProtocols; pIdx++) { 2681e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((MethodDecl = protocols[pIdx]->getInstanceMethod(Sel))) 2691e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return MethodDecl; 2701e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 2711e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner // Didn't find one yet - now look through categories. 2721e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCCategoryDecl *CatDecl = ClassDecl->getCategoryList(); 2731e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner while (CatDecl) { 2741e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((MethodDecl = CatDecl->getInstanceMethod(Sel))) 2751e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return MethodDecl; 2761e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner CatDecl = CatDecl->getNextClassCategory(); 2771e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 2781e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ClassDecl = ClassDecl->getSuperClass(); 2791e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 2801e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return NULL; 2811e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 2821e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 2831e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// lookupClassMethod - This method returns a class method by looking in the 2841e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// class, its categories, and its super classes (using a linear search). 2851e03a561f4bd96910cb31a8af53a6ad321a12b51Chris LattnerObjCMethodDecl *ObjCInterfaceDecl::lookupClassMethod(Selector Sel) { 2861e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCInterfaceDecl* ClassDecl = this; 2871e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCMethodDecl *MethodDecl = 0; 2881e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 2891e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner while (ClassDecl != NULL) { 2901e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((MethodDecl = ClassDecl->getClassMethod(Sel))) 2911e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return MethodDecl; 2921e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 2931e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner // Didn't find one yet - look through protocols. 2941e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCProtocolDecl **protocols = ClassDecl->getReferencedProtocols(); 2951e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner int numProtocols = ClassDecl->getNumIntfRefProtocols(); 2961e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner for (int pIdx = 0; pIdx < numProtocols; pIdx++) { 2971e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((MethodDecl = protocols[pIdx]->getClassMethod(Sel))) 2981e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return MethodDecl; 2991e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 3001e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner // Didn't find one yet - now look through categories. 3011e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCCategoryDecl *CatDecl = ClassDecl->getCategoryList(); 3021e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner while (CatDecl) { 3031e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((MethodDecl = CatDecl->getClassMethod(Sel))) 3041e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return MethodDecl; 3051e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner CatDecl = CatDecl->getNextClassCategory(); 3061e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 3071e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ClassDecl = ClassDecl->getSuperClass(); 3081e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 3091e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return NULL; 3101e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 3111e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 3121e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// lookupInstanceMethod - This method returns an instance method by looking in 3131e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// the class implementation. Unlike interfaces, we don't look outside the 3141e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// implementation. 3151e03a561f4bd96910cb31a8af53a6ad321a12b51Chris LattnerObjCMethodDecl *ObjCImplementationDecl::getInstanceMethod(Selector Sel) { 3161e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner for (instmeth_iterator I = instmeth_begin(), E = instmeth_end(); I != E; ++I) 3171e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((*I)->getSelector() == Sel) 3181e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return *I; 3191e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return NULL; 3201e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 3211e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 3221e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// lookupClassMethod - This method returns a class method by looking in 3231e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// the class implementation. Unlike interfaces, we don't look outside the 3241e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// implementation. 3251e03a561f4bd96910cb31a8af53a6ad321a12b51Chris LattnerObjCMethodDecl *ObjCImplementationDecl::getClassMethod(Selector Sel) { 3261e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner for (classmeth_iterator I = classmeth_begin(), E = classmeth_end(); 3271e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner I != E; ++I) 3281e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((*I)->getSelector() == Sel) 3291e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return *I; 3301e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return NULL; 3311e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 3321e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 3331e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// lookupInstanceMethod - This method returns an instance method by looking in 3341e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// the class implementation. Unlike interfaces, we don't look outside the 3351e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// implementation. 3361e03a561f4bd96910cb31a8af53a6ad321a12b51Chris LattnerObjCMethodDecl *ObjCCategoryImplDecl::getInstanceMethod(Selector Sel) { 3371e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner for (instmeth_iterator I = instmeth_begin(), E = instmeth_end(); I != E; ++I) 3381e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((*I)->getSelector() == Sel) 3391e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return *I; 3401e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return NULL; 3411e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 3421e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 3431e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// lookupClassMethod - This method returns an instance method by looking in 3441e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// the class implementation. Unlike interfaces, we don't look outside the 3451e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// implementation. 3461e03a561f4bd96910cb31a8af53a6ad321a12b51Chris LattnerObjCMethodDecl *ObjCCategoryImplDecl::getClassMethod(Selector Sel) { 3471e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner for (classmeth_iterator I = classmeth_begin(), E = classmeth_end(); 3481e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner I != E; ++I) 3491e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((*I)->getSelector() == Sel) 3501e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return *I; 3511e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return NULL; 3521e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 3531e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 3541e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// lookupInstanceMethod - Lookup a instance method in the protocol and protocols 3551e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// it inherited. 3561e03a561f4bd96910cb31a8af53a6ad321a12b51Chris LattnerObjCMethodDecl *ObjCProtocolDecl::lookupInstanceMethod(Selector Sel) { 3571e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCMethodDecl *MethodDecl = NULL; 3581e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 3591e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((MethodDecl = getInstanceMethod(Sel))) 3601e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return MethodDecl; 3611e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 3621e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (getNumReferencedProtocols() > 0) { 3631e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCProtocolDecl **RefPDecl = getReferencedProtocols(); 3641e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 3651e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner for (unsigned i = 0; i < getNumReferencedProtocols(); i++) { 3661e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((MethodDecl = RefPDecl[i]->getInstanceMethod(Sel))) 3671e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return MethodDecl; 3681e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 3691e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 3701e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return NULL; 3711e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 3721e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 3731e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// lookupInstanceMethod - Lookup a class method in the protocol and protocols 3741e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner// it inherited. 3751e03a561f4bd96910cb31a8af53a6ad321a12b51Chris LattnerObjCMethodDecl *ObjCProtocolDecl::lookupClassMethod(Selector Sel) { 3761e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCMethodDecl *MethodDecl = NULL; 3771e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 3781e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((MethodDecl = getClassMethod(Sel))) 3791e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return MethodDecl; 3801e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 3811e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (getNumReferencedProtocols() > 0) { 3821e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner ObjCProtocolDecl **RefPDecl = getReferencedProtocols(); 3831e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 3841e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner for(unsigned i = 0; i < getNumReferencedProtocols(); i++) { 3851e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if ((MethodDecl = RefPDecl[i]->getClassMethod(Sel))) 3861e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return MethodDecl; 3871e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 3881e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner } 3891e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return NULL; 3901e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 3911e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 3921e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// getSynthesizedMethodSize - Compute size of synthesized method name 3931e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// as done be the rewrite. 3941e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner/// 3951e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattnerunsigned ObjCMethodDecl::getSynthesizedMethodSize() const { 3961e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner // syntesized method name is a concatenation of -/+[class-name selector] 3971e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner // Get length of this name. 3981e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner unsigned length = 3; // _I_ or _C_ 3991e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner length += strlen(getClassInterface()->getName()) +1; // extra for _ 4001e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner NamedDecl *MethodContext = getMethodContext(); 4011e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (ObjCCategoryImplDecl *CID = 4021e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner dyn_cast<ObjCCategoryImplDecl>(MethodContext)) 4031e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner length += strlen(CID->getName()) +1; 4041e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner length += getSelector().getName().size(); // selector name 4051e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return length; 4061e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 4071e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner 4081e03a561f4bd96910cb31a8af53a6ad321a12b51Chris LattnerObjCInterfaceDecl *const ObjCMethodDecl::getClassInterface() const { 4091e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (ObjCInterfaceDecl *ID = dyn_cast<ObjCInterfaceDecl>(MethodContext)) 4101e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return ID; 4111e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (ObjCCategoryDecl *CD = dyn_cast<ObjCCategoryDecl>(MethodContext)) 4121e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return CD->getClassInterface(); 4131e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (ObjCImplementationDecl *IMD = 4141e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner dyn_cast<ObjCImplementationDecl>(MethodContext)) 4151e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return IMD->getClassInterface(); 4161e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner if (ObjCCategoryImplDecl *CID = 4171e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner dyn_cast<ObjCCategoryImplDecl>(MethodContext)) 4181e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return CID->getClassInterface(); 4191e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner assert(false && "unknown method context"); 4201e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner return 0; 4211e03a561f4bd96910cb31a8af53a6ad321a12b51Chris Lattner} 422