[arcmt] In ARC default for properties is 'strong' so just remove a 'retain' if possible,

instead of changing it to 'strong'. rdar://9984862.

llvm-svn: 142304

clang/lib/ARCMigrate/TransProperties.cpp

@@ -132,7 +132,10 @@ private:
       return;
 
     if (propAttrs & ObjCPropertyDecl::OBJC_PR_retain) {
-      rewriteAttribute("retain", "strong", atLoc);
+      if (propAttrs & ObjCPropertyDecl::OBJC_PR_readonly)
+        rewriteAttribute("retain", "strong", atLoc);
+      else
+        removeAttribute("retain", atLoc); // strong is the default.
       return;
     }
 
@@ -215,6 +218,10 @@ private:
     }
   }
 
+  bool removeAttribute(StringRef fromAttr, SourceLocation atLoc) const {
+    return rewriteAttribute(fromAttr, StringRef(), atLoc);
+  }
+
   bool rewriteAttribute(StringRef fromAttr, StringRef toAttr,
                         SourceLocation atLoc) const {
     if (atLoc.isMacroID())
@@ -248,6 +255,11 @@ private:
     lexer.LexFromRawLexer(tok);
     if (tok.isNot(tok::l_paren)) return false;
     
+    Token BeforeTok = tok;
+    Token AfterTok;
+    AfterTok.startToken();
+    SourceLocation AttrLoc;
+    
     lexer.LexFromRawLexer(tok);
     if (tok.is(tok::r_paren))
       return false;
@@ -256,18 +268,40 @@ private:
       if (tok.isNot(tok::raw_identifier)) return false;
       StringRef ident(tok.getRawIdentifierData(), tok.getLength());
       if (ident == fromAttr) {
-        Pass.TA.replaceText(tok.getLocation(), fromAttr, toAttr);
-        return true;
+        if (!toAttr.empty()) {
+          Pass.TA.replaceText(tok.getLocation(), fromAttr, toAttr);
+          return true;
+        }
+        // We want to remove the attribute.
+        AttrLoc = tok.getLocation();
       }
 
       do {
         lexer.LexFromRawLexer(tok);
+        if (AttrLoc.isValid() && AfterTok.is(tok::unknown))
+          AfterTok = tok;
       } while (tok.isNot(tok::comma) && tok.isNot(tok::r_paren));
       if (tok.is(tok::r_paren))
         break;
+      if (AttrLoc.isInvalid())
+        BeforeTok = tok;
       lexer.LexFromRawLexer(tok);
     }
 
+    if (toAttr.empty() && AttrLoc.isValid() && AfterTok.isNot(tok::unknown)) {
+      // We want to remove the attribute.
+      if (BeforeTok.is(tok::l_paren) && AfterTok.is(tok::r_paren)) {
+        Pass.TA.remove(SourceRange(BeforeTok.getLocation(),
+                                   AfterTok.getLocation()));
+      } else if (BeforeTok.is(tok::l_paren) && AfterTok.is(tok::comma)) {
+        Pass.TA.remove(SourceRange(AttrLoc, AfterTok.getLocation()));
+      } else {
+        Pass.TA.remove(SourceRange(BeforeTok.getLocation(), AttrLoc));
+      }
+
+      return true;
+    }
+    
     return false;
   }
 

clang/test/ARCMT/assign-prop-with-arc-runtime.m

@@ -36,12 +36,18 @@ typedef _NSCachedAttributedString *BadClassForWeak;
 
 @property (assign) Foo *no_user_ivar1;
 @property (readonly) Foo *no_user_ivar2;
+
+@property (retain) id def1;
+@property (atomic,retain) id def2;
+@property (retain,atomic) id def3;
+
 @end
 
 @implementation Foo
 @synthesize x,w,q1,q2,oo,bcw,not_safe1,not_safe2,not_safe3;
 @synthesize no_user_ivar1, no_user_ivar2;
 @synthesize assign_plus1, assign_plus2, assign_plus3;
+@synthesize def1, def2, def3;
 
 -(void)test:(Foo *)parm {
   assign_plus1 = [[Foo alloc] init];

clang/test/ARCMT/assign-prop-with-arc-runtime.m.result

@@ -36,12 +36,18 @@ typedef _NSCachedAttributedString *BadClassForWeak;
 
 @property (weak) Foo *no_user_ivar1;
 @property (weak, readonly) Foo *no_user_ivar2;
+
+@property  id def1;
+@property (atomic) id def2;
+@property (atomic) id def3;
+
 @end
 
 @implementation Foo
 @synthesize x,w,q1,q2,oo,bcw,not_safe1,not_safe2,not_safe3;
 @synthesize no_user_ivar1, no_user_ivar2;
 @synthesize assign_plus1, assign_plus2, assign_plus3;
+@synthesize def1, def2, def3;
 
 -(void)test:(Foo *)parm {
   assign_plus1 = [[Foo alloc] init];

clang/test/ARCMT/releases-driver.m.result

@@ -20,7 +20,7 @@ id IhaveSideEffect();
 @interface Foo : NSObject {
   id bar;
 }
-@property (strong) id bar;
+@property  id bar;
 -(void)test:(id)obj;
 @end
 

clang/test/ARCMT/releases.m.result

@@ -20,7 +20,7 @@ id IhaveSideEffect();
 @interface Foo : NSObject {
   id bar;
 }
-@property (strong) id bar;
+@property  id bar;
 -(void)test:(id)obj;
 @end
 

clang/test/ARCMT/remove-dealloc-method.m.result

@@ -5,10 +5,10 @@
 #define nil ((void*) 0)
 
 @interface Foo 
-@property (strong) id x;
-@property (strong) id y;
-@property (strong) id w;
-@property (strong) id z;
+@property  id x;
+@property  id y;
+@property  id w;
+@property  id z;
 @end
 
 @implementation Foo 

clang/test/ARCMT/remove-dealloc-zerouts.m.result

@@ -3,10 +3,10 @@
 // RUN: diff %t %s.result
 
 @interface Foo 
-@property (strong) id x;
-@property (strong) id y;
-@property (strong) id w;
-@property (strong) id z;
+@property  id x;
+@property  id y;
+@property  id w;
+@property  id z;
 @property (strong) id q;
 @end
 
@@ -23,7 +23,7 @@
 @end
 
 @interface Bar
-@property (strong) Foo *a;
+@property  Foo *a;
 - (void) setA:(Foo*) val;
 - (id) a;
 @end

clang/test/ARCMT/retains.m.result

@@ -9,7 +9,7 @@ id IhaveSideEffect();
 @interface Foo : NSObject {
   id bar;
 }
-@property (strong) id bar;
+@property  id bar;
 -(id)test:(id)obj;
 -(id)something;
 @end