Tools__MethodRewriter.st
branchjv
changeset 12287 400a99059170
parent 12207 17656fc80e6c
child 12401 4714b9640528
--- a/Tools__MethodRewriter.st	Fri Sep 07 12:37:19 2012 +0100
+++ b/Tools__MethodRewriter.st	Fri Sep 07 18:28:09 2012 +0100
@@ -909,7 +909,7 @@
 
 canDropClasses: dropContext
 
-    ^dropContext dropObjects allSatisfy: [:obj|obj isClassObject]
+    ^dropContext dropObjects allSatisfy: [:obj|obj isClassObject or:[obj isMethodObject]]
 
     "Created: / 20-07-2007 / 16:58:13 / janfrog"
 !
@@ -918,8 +918,13 @@
 
     | newClasses |
     newClasses := dropContext dropObjects 
-                    select:[:obj|obj isClassObject]
-                    thenCollect:[:obj|obj theObject].
+                    select:[:obj|obj isClassObject or:[obj isMethodObject]].
+    newClasses := newClasses    
+                    collect:[:obj| |clsOrMethod| 
+                        clsOrMethod := obj theObject.
+                        clsOrMethod isMethod ifTrue:[clsOrMethod containingClass] ifFalse:[clsOrMethod]
+                    ] as:IdentitySet.
+
     newClasses := newClasses reject:[:cls|self classes includes:cls].
     self classes 
         addAll: newClasses;
@@ -1074,9 +1079,9 @@
 !MethodRewriter class methodsFor:'documentation'!
 
 version_CVS
-    ^ '§Header: /cvs/stx/stx/libtool/Tools__MethodRewriter.st,v 1.15 2011/08/04 19:21:34 cg Exp §'
+    ^ '§Header: /cvs/stx/stx/libtool/Tools__MethodRewriter.st,v 1.16 2012/08/03 19:21:17 stefan Exp §'
 !
 
 version_SVN
-    ^ '$Id: Tools__MethodRewriter.st 7954 2012-03-26 15:34:31Z vranyj1 $'
+    ^ '$Id: Tools__MethodRewriter.st 8048 2012-09-07 17:28:09Z vranyj1 $'
 ! !