1 files changed, 47 insertions, 0 deletions
diff --git a/clang/lib/ARCMigrate/TransDeallocMethod.cpp b/clang/lib/ARCMigrate/TransDeallocMethod.cpp
new file mode 100644
index 00000000000..a7c3c1e9d64
--- /dev/null
+++ b/clang/lib/ARCMigrate/TransDeallocMethod.cpp
@@ -0,0 +1,47 @@
+//===--- TransDeallocMethod.cpp - Tranformations to ARC mode --------------===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+
+#include "Transforms.h"
+#include "Internals.h"
+
+using namespace clang;
+using namespace arcmt;
+using namespace trans;
+using llvm::StringRef;
+
+void trans::removeDeallocMethod(MigrationPass &pass) {
+  ASTContext &Ctx = pass.Ctx;
+  TransformActions &TA = pass.TA;
+  DeclContext *DC = Ctx.getTranslationUnitDecl();
+  ObjCMethodDecl *DeallocMethodDecl = 0;
+  IdentifierInfo *II = &Ctx.Idents.get("dealloc");
+
+  for (DeclContext::decl_iterator
+         I = DC->decls_begin(), E = DC->decls_end(); I != E; ++I) {
+    Decl *D = *I;
+    if (ObjCImplementationDecl *IMD = dyn_cast<ObjCImplementationDecl>(D)) {
+      DeallocMethodDecl = 0;
+      for (ObjCImplementationDecl::instmeth_iterator
+             I = IMD->instmeth_begin(), E = IMD->instmeth_end();
+          I != E; ++I) {
+        ObjCMethodDecl *OMD = *I;
+        if (OMD->isInstanceMethod() &&
+            OMD->getSelector() == Ctx.Selectors.getSelector(0, &II)) {
+          DeallocMethodDecl = OMD;
+          break;
+        }
+      }
+      if (DeallocMethodDecl && 
+          DeallocMethodDecl->getCompoundBody()->body_empty()) {
+        Transaction Trans(TA);
+        TA.remove(DeallocMethodDecl->getSourceRange());
+      }
+    }
+  }
+}