Simple animations for OpenGL geometry

Author: eulertour

manim/animation/animation.py

@@ -14,7 +14,9 @@
     from manim.scene.scene import Scene
 
 from .. import logger
-from ..mobject.mobject import Mobject, _AnimationBuilder
+from ..mobject import mobject
+from ..mobject import opengl_mobject
+from ..mobject.mobject import Mobject
 from ..utils.rate_functions import smooth
 from ..mobject.opengl_mobject import OpenGLMobject
 
@@ -198,7 +200,9 @@ def is_remover(self) -> bool:
         return self.remover
 
 
-def prepare_animation(anim: Union["Animation", "_AnimationBuilder"]) -> "Animation":
+def prepare_animation(
+    anim: Union["Animation", "mobject._AnimationBuilder"]
+) -> "Animation":
     r"""Returns either an unchanged animation, or the animation built
     from a passed animation factory.
 
@@ -225,7 +229,10 @@ def prepare_animation(anim: Union["Animation", "_AnimationBuilder"]) -> "Animati
         TypeError: Object 42 cannot be converted to an animation
 
     """
-    if isinstance(anim, _AnimationBuilder):
+    if isinstance(anim, mobject._AnimationBuilder):
+        return anim.build()
+
+    if isinstance(anim, opengl_mobject._AnimationBuilder):
         return anim.build()
 
     if isinstance(anim, Animation):

manim/animation/creation.py

@@ -82,6 +82,7 @@ def construct(self):
 from ..animation.composition import Succession
 from ..mobject.mobject import Group, Mobject
 from ..mobject.types.vectorized_mobject import VMobject
+from ..mobject.types.opengl_vectorized_mobject import OpenGLVMobject
 from ..utils.bezier import integer_interpolate
 from ..utils.rate_functions import double_smooth, linear, smooth
 
@@ -101,7 +102,9 @@ class ShowPartial(Animation):
     """
 
     def __init__(self, mobject: VMobject, **kwargs):
-        if not isinstance(mobject, VMobject):
+        if not isinstance(mobject, VMobject) and not isinstance(
+            mobject, OpenGLVMobject
+        ):
             raise TypeError("This Animation only works on vectorized mobjects")
         super().__init__(mobject, **kwargs)
 

manim/mobject/opengl_mobject.py

@@ -1617,7 +1617,7 @@ def update_target(*method_args, **method_kwargs):
         return update_target
 
     def build(self):
-        from manimlib.animation.transform import _MethodAnimation
+        from ..animation.transform import _MethodAnimation
 
         if self.overridden_animation:
             return self.overridden_animation

manim/renderer/opengl_renderer.py

@@ -54,14 +54,15 @@ def update_depth_test(self, context, shader_wrapper):
         else:
             self.context.disable(moderngl.DEPTH_TEST)
 
-    def render_mobject(self, mob):
+    def render_mobject(self, mobs):
         # shader_wrapper_list = self.get_shader_wrapper_list(mob)
-        shader_wrapper_list = mob.get_shader_wrapper_list()
-        render_group_list = map(
-            lambda x: self.get_render_group(self.context, x), shader_wrapper_list
-        )
-        for render_group in render_group_list:
-            self.render_render_group(render_group)
+        for mob in mobs:
+            shader_wrapper_list = mob.get_shader_wrapper_list()
+            render_group_list = map(
+                lambda x: self.get_render_group(self.context, x), shader_wrapper_list
+            )
+            for render_group in render_group_list:
+                self.render_render_group(render_group)
 
     def render_render_group(self, render_group):
         shader_wrapper = render_group["shader_wrapper"]
@@ -318,7 +319,7 @@ def play(self, scene, *args, **kwargs):
     def render(self, scene, frame_offset, moving_mobjects):
         def update_frame():
             self.frame_buffer_object.clear(*window_background_color)
-            self.render_mobject(scene.mobjects[0])
+            self.render_mobject(scene.mobjects)
             self.window.swap_buffers()
             self.animation_elapsed_time = time.time() - self.animation_start_time
 

manim/scene/scene.py

@@ -301,9 +301,15 @@ def get_mobject_family_members(self):
         list
             List of mobject family members.
         """
-        return extract_mobject_family_members(
-            self.mobjects, use_z_index=self.renderer.camera.use_z_index
-        )
+        if config["use_opengl_renderer"]:
+            family_members = []
+            for mob in self.mobjects:
+                family_members.extend(mob.get_family())
+            return family_members
+        else:
+            return extract_mobject_family_members(
+                self.mobjects, use_z_index=self.renderer.camera.use_z_index
+            )
 
     def add(self, *mobjects):
         """