Improve performance for :meth:.Mobject.become

manim/mobject/mobject.py

@@ -2471,10 +2471,11 @@ def construct(self):
         return self.shuffle(*args, **kwargs)
 
     # Alignment
-    def align_data(self, mobject: "Mobject"):
+    def align_data(self, mobject: "Mobject", skip_point_alignment: bool = False):
         self.null_point_align(mobject)
         self.align_submobjects(mobject)
-        self.align_points(mobject)
+        if not skip_point_alignment:
+            self.align_points(mobject)
         # Recurse
         for m1, m2 in zip(self.submobjects, mobject.submobjects):
             m1.align_data(m2)
@@ -2644,7 +2645,7 @@ def construct(self):
         if match_center:
             mobject.move_to(self.get_center())
 
-        self.align_data(mobject)
+        self.align_data(mobject, skip_point_alignment=True)
         for sm1, sm2 in zip(self.get_family(), mobject.get_family()):
             sm1.points = np.array(sm2.points)
             sm1.interpolate_color(sm1, sm2, 1)
@@ -2667,7 +2668,6 @@ def construct(self):
                     self.play(circ.animate.match_points(square))
                     self.wait(0.5)
         """
-        self.align_data(mobject)
         for sm1, sm2 in zip(self.get_family(), mobject.get_family()):
             sm1.points = np.array(sm2.points)
         return self

tests/test_vectorized_mobject.py

@@ -3,7 +3,7 @@
 import numpy as np
 import pytest
 
-from manim import Circle, Line, Mobject, Square, VDict, VGroup, VMobject
+from manim import Circle, Line, Mobject, RegularPolygon, Square, VDict, VGroup, VMobject
 from manim.constants import PI
 
 
@@ -265,3 +265,27 @@ def draw_circle(m: VMobject, n_points, x=0, y=0, r=1):
 
     # The number of points should be similar to the size of a and b
     assert len(o.points) <= (20 + 15 + 15) * 4
+
+
+def test_vmobject_same_points_become():
+    a = Square()
+    b = Circle()
+    a.become(b)
+    assert np.array_equal(a.points, b.points)
+    assert len(a.submobjects) == len(b.submobjects)
+
+
+def test_vmobject_same_num_submobjects_become():
+    a = Square()
+    b = RegularPolygon(n=6)
+    a.become(b)
+    assert np.array_equal(a.points, b.points)
+    assert len(a.submobjects) == len(b.submobjects)
+
+
+def test_vmobject_different_num_points_and_submobjects_become():
+    a = Square()
+    b = VGroup(Circle(), Square())
+    a.become(b)
+    assert np.array_equal(a.points, b.points)
+    assert len(a.submobjects) == len(b.submobjects)