Author: fjahanian
Date: Thu Aug 1 17:29:32 2013
New Revision: 187626
URL: http://llvm.org/viewvc/llvm-project?rev=187626&view=rev
Log:
ObjectiveC migrator. Migrate to instancetype return type
for mehods with certain prefix selector matching their class names'
suffix.
Added:
cfe/trunk/test/ARCMT/objcmt-instancetype-2.m.result
Modified:
cfe/trunk/include/clang/Basic/IdentifierTable.h
cfe/trunk/lib/ARCMigrate/ObjCMT.cpp
cfe/trunk/lib/Basic/IdentifierTable.cpp
cfe/trunk/test/ARCMT/objcmt-instancetype.m.result
Modified: cfe/trunk/include/clang/Basic/IdentifierTable.h
URL:
http://llvm.org/viewvc/llvm-project/cfe/trunk/include/clang/Basic/IdentifierTable.h?rev=187626&r1=187625&r2=187626&view=diff
==============================================================================
--- cfe/trunk/include/clang/Basic/IdentifierTable.h (original)
+++ cfe/trunk/include/clang/Basic/IdentifierTable.h Thu Aug 1 17:29:32 2013
@@ -586,10 +586,7 @@ enum ObjCInstanceTypeFamily {
OIT_None,
OIT_Array,
OIT_Dictionary,
- OIT_MemManage,
- OIT_NSString,
- OIT_NSSet,
- OIT_NSURL
+ OIT_MemManage
};
/// \brief Smart pointer class that efficiently represents Objective-C method
Modified: cfe/trunk/lib/ARCMigrate/ObjCMT.cpp
URL:
http://llvm.org/viewvc/llvm-project/cfe/trunk/lib/ARCMigrate/ObjCMT.cpp?rev=187626&r1=187625&r2=187626&view=diff
==============================================================================
--- cfe/trunk/lib/ARCMigrate/ObjCMT.cpp (original)
+++ cfe/trunk/lib/ARCMigrate/ObjCMT.cpp Thu Aug 1 17:29:32 2013
@@ -41,6 +41,8 @@ class ObjCMigrateASTConsumer : public AS
void migrateInstanceType(ASTContext &Ctx, ObjCContainerDecl *CDecl);
void migrateMethodInstanceType(ASTContext &Ctx, ObjCContainerDecl *CDecl,
ObjCMethodDecl *OM);
+ void migrateFactoryMethod(ASTContext &Ctx, ObjCContainerDecl *CDecl,
+ ObjCMethodDecl *OM);
public:
std::string MigrateDir;
@@ -549,13 +551,34 @@ void ObjCMigrateASTConsumer::migrateNSEn
Editor->commit(commit);
}
+static void ReplaceWithInstancetype(const ObjCMigrateASTConsumer &ASTC,
+ ObjCMethodDecl *OM) {
+ SourceRange R;
+ std::string ClassString;
+ if (TypeSourceInfo *TSInfo = OM->getResultTypeSourceInfo()) {
+ TypeLoc TL = TSInfo->getTypeLoc();
+ R = SourceRange(TL.getBeginLoc(), TL.getEndLoc());
+ ClassString = "instancetype";
+ }
+ else {
+ R = SourceRange(OM->getLocStart(), OM->getLocStart());
+ ClassString = OM->isInstanceMethod() ? '-' : '+';
+ ClassString += " (instancetype)";
+ }
+ edit::Commit commit(*ASTC.Editor);
+ commit.replace(R, ClassString);
+ ASTC.Editor->commit(commit);
+}
+
void ObjCMigrateASTConsumer::migrateMethodInstanceType(ASTContext &Ctx,
ObjCContainerDecl
*CDecl,
ObjCMethodDecl *OM) {
ObjCInstanceTypeFamily OIT_Family =
Selector::getInstTypeMethodFamily(OM->getSelector());
- if (OIT_Family == OIT_None)
+ if (OIT_Family == OIT_None) {
+ migrateFactoryMethod(Ctx, CDecl, OM);
return;
+ }
std::string ClassName;
switch (OIT_Family) {
case OIT_Array:
@@ -581,24 +604,11 @@ void ObjCMigrateASTConsumer::migrateMeth
IDecl = ImpDecl->getClassInterface();
}
if (!IDecl ||
- !IDecl->lookupInheritedClass(&Ctx.Idents.get(ClassName)))
+ !IDecl->lookupInheritedClass(&Ctx.Idents.get(ClassName))) {
+ migrateFactoryMethod(Ctx, CDecl, OM);
return;
-
- SourceRange R;
- std::string ClassString;
- if (TypeSourceInfo *TSInfo = OM->getResultTypeSourceInfo()) {
- TypeLoc TL = TSInfo->getTypeLoc();
- R = SourceRange(TL.getBeginLoc(), TL.getEndLoc());
- ClassString = "instancetype";
- }
- else {
- R = SourceRange(OM->getLocStart(), OM->getLocStart());
- ClassString = OM->isInstanceMethod() ? '-' : '+';
- ClassString += " (instancetype)";
}
- edit::Commit commit(*Editor);
- commit.replace(R, ClassString);
- Editor->commit(commit);
+ ReplaceWithInstancetype(*this, OM);
}
void ObjCMigrateASTConsumer::migrateInstanceType(ASTContext &Ctx,
@@ -612,6 +622,42 @@ void ObjCMigrateASTConsumer::migrateInst
}
}
+void ObjCMigrateASTConsumer::migrateFactoryMethod(ASTContext &Ctx,
+ ObjCContainerDecl *CDecl,
+ ObjCMethodDecl *OM) {
+ if (OM->isInstanceMethod() || !OM->getResultType()->isObjCIdType())
+ return;
+
+ // Candidate factory methods are + (id) NaMeXXX : ... which belong to a class
+ // NSYYYNamE with matching names be at least 3 characters long.
+ ObjCInterfaceDecl *IDecl = dyn_cast<ObjCInterfaceDecl>(CDecl);
+ if (!IDecl) {
+ if (ObjCCategoryDecl *CatDecl = dyn_cast<ObjCCategoryDecl>(CDecl))
+ IDecl = CatDecl->getClassInterface();
+ else if (ObjCImplDecl *ImpDecl = dyn_cast<ObjCImplDecl>(CDecl))
+ IDecl = ImpDecl->getClassInterface();
+ }
+ if (!IDecl)
+ return;
+
+ StringRef ClassName = IDecl->getName();
+ if (!ClassName.startswith("NS"))
+ return;
+
+ ClassName = ClassName.lower();
+ IdentifierInfo *MethodIdName = OM->getSelector().getIdentifierInfoForSlot(0);
+ StringRef MethodName = MethodIdName->getName();
+ StringRef MethodNamePrefix = MethodName.substr(0, 3).lower();
+ size_t Ix = ClassName.rfind(MethodNamePrefix);
+ if (Ix == StringRef::npos)
+ return;
+ StringRef ClassNamePostfix = ClassName.substr(Ix);
+ MethodName = MethodName.lower();
+ if (!MethodName.startswith(ClassNamePostfix))
+ return;
+ ReplaceWithInstancetype(*this, OM);
+}
+
namespace {
class RewritesReceiver : public edit::EditsReceiver {
Modified: cfe/trunk/lib/Basic/IdentifierTable.cpp
URL:
http://llvm.org/viewvc/llvm-project/cfe/trunk/lib/Basic/IdentifierTable.cpp?rev=187626&r1=187625&r2=187626&view=diff
==============================================================================
--- cfe/trunk/lib/Basic/IdentifierTable.cpp (original)
+++ cfe/trunk/lib/Basic/IdentifierTable.cpp Thu Aug 1 17:29:32 2013
@@ -474,14 +474,6 @@ ObjCInstanceTypeFamily Selector::getInst
case 'r':
if (startsWithWord(name, "retain")) return OIT_MemManage;
break;
- case 's':
- if (startsWithWord(name, "string")) return OIT_NSString;
- else
- if (startsWithWord(name, "set")) return OIT_NSSet;
- break;
- case 'U':
- if (startsWithWord(name, "URL")) return OIT_NSURL;
- break;
default:
break;
}
Added: cfe/trunk/test/ARCMT/objcmt-instancetype-2.m.result
URL:
http://llvm.org/viewvc/llvm-project/cfe/trunk/test/ARCMT/objcmt-instancetype-2.m.result?rev=187626&view=auto
==============================================================================
--- cfe/trunk/test/ARCMT/objcmt-instancetype-2.m.result (added)
+++ cfe/trunk/test/ARCMT/objcmt-instancetype-2.m.result Thu Aug 1 17:29:32 2013
@@ -0,0 +1,76 @@
+// RUN: rm -rf %t
+// RUN: %clang_cc1 -objcmt-migrate-property -mt-migrate-directory %t %s -x
objective-c -fobjc-runtime-has-weak -fobjc-arc
-fobjc-default-synthesize-properties -triple x86_64-apple-darwin11
+// RUN: c-arcmt-test -mt-migrate-directory %t | arcmt-test
-verify-transformed-files %s.result
+// RUN: %clang_cc1 -triple x86_64-apple-darwin11 -fsyntax-only -x objective-c
-fobjc-runtime-has-weak -fobjc-arc -fobjc-default-synthesize-properties
%s.result
+
+typedef unsigned int NSUInteger;
+typedef int NSInteger;
+typedef char BOOL;
+@class NSData, NSError, NSProtocolChecker, NSObject;
+@class NSPortNameServer, NSTimeZone;
+
+@interface NSMutableString
+@end
+
+@interface NSString @end
+
+@class NSString, NSURL;
+@interface NSString (NSStringDeprecated)
++ (instancetype)stringWithContentsOfFile:(NSString *)path
__attribute__((availability(macosx,introduced=10.0 ,deprecated=10.4,message=""
)));
++ (instancetype)stringWithContentsOfURL:(NSURL *)url
__attribute__((availability(macosx,introduced=10.0 ,deprecated=10.4,message=""
)));
++ (instancetype)stringWithCString:(const char *)bytes
length:(NSUInteger)length __attribute__((availability(macosx,introduced=10.0
,deprecated=10.4,message="" )));
++ (instancetype)stringWithCString:(const char *)bytes
__attribute__((availability(macosx,introduced=10.0 ,deprecated=10.4,message=""
)));
+@end
+
+
+typedef enum NSURLBookmarkResolutionOptions {
+ Bookmark
+} NSURLBookmarkResolutionOptions;
+
+@interface NSURL
++ (instancetype)URLWithString:(NSString *)URLString;
++ (instancetype)URLWithString:(NSString *)URLString relativeToURL:(NSURL
*)baseURL;
++ (instancetype)URLByResolvingBookmarkData:(NSData *)bookmarkData
options:(NSURLBookmarkResolutionOptions)options relativeToURL:(NSURL
*)relativeURL bookmarkDataIsStale:(BOOL *)isStale error:(NSError **)error
__attribute__((availability(macosx,introduced=10.6)));
+@end
+
+@class NSDictionary;
+@interface NSError
++ (instancetype)errorWithDomain:(NSString *)domain code:(NSInteger)code
userInfo:(NSDictionary *)dict;
+@end
+
+
+@interface NSMutableString (NSMutableStringExtensionMethods)
++ (instancetype)stringWithCapacity:(NSUInteger)capacity;
+@end
+
+@interface NSMutableData
++ (instancetype)dataWithCapacity:(NSUInteger)aNumItems;
++ (instancetype)dataWithLength:(NSUInteger)length;
+@end
+
+@interface NSMutableDictionary @end
+
+@interface NSMutableDictionary (NSSharedKeySetDictionary)
++ (instancetype )dictionaryWithSharedKeySet:(id)keyset
__attribute__((availability(macosx,introduced=10.8)));
+@end
+
+@interface NSProtocolChecker
++ (instancetype)protocolCheckerWithTarget:(NSObject *)anObject
protocol:(Protocol *)aProtocol;
+@end
+
+@interface NSConnection
++ (instancetype)connectionWithRegisteredName:(NSString *)name host:(NSString
*)hostName;
++ (instancetype)connectionWithRegisteredName:(NSString *)name host:(NSString
*)hostName usingNameServer:(NSPortNameServer *)server;
+@end
+
+@interface NSDate
++ (instancetype)dateWithString:(NSString *)aString
__attribute__((availability(macosx,introduced=10.4)));
+@end
+
+@interface NSCalendarDate : NSDate
++ (instancetype)calendarDate
__attribute__((availability(macosx,introduced=10.4)));
++ (instancetype)dateWithString:(NSString *)description
calendarFormat:(NSString *)format locale:(id)locale
__attribute__((availability(macosx,introduced=10.4)));
++ (instancetype)dateWithString:(NSString *)description
calendarFormat:(NSString *)format
__attribute__((availability(macosx,introduced=10.4)));
++ (instancetype)dateWithYear:(NSInteger)year month:(NSUInteger)month
day:(NSUInteger)day hour:(NSUInteger)hour minute:(NSUInteger)minute
second:(NSUInteger)second timeZone:(NSTimeZone *)aTimeZone
__attribute__((availability(macosx,introduced=10.4)));
+@end
+
Modified: cfe/trunk/test/ARCMT/objcmt-instancetype.m.result
URL:
http://llvm.org/viewvc/llvm-project/cfe/trunk/test/ARCMT/objcmt-instancetype.m.result?rev=187626&r1=187625&r2=187626&view=diff
==============================================================================
--- cfe/trunk/test/ARCMT/objcmt-instancetype.m.result (original)
+++ cfe/trunk/test/ARCMT/objcmt-instancetype.m.result Thu Aug 1 17:29:32 2013
@@ -11,7 +11,7 @@ typedef signed char BOOL;
@end
@interface NSString : NSObject
-+ (id)stringWithString:(NSString *)string;
++ (instancetype)stringWithString:(NSString *)string;
- (instancetype)initWithString:(NSString *)aString;
@end
_______________________________________________
cfe-commits mailing list
cfe-commits@cs.uiuc.edu
http://lists.cs.uiuc.edu/mailman/listinfo/cfe-commits
