Add parameter about_point for rotate (#1238)

* about_point parameter for rotate

* Update manim/mobject/mobject.py

* Update manim/mobject/mobject.py

Co-authored-by: Abhijith Muthyala <[email protected]>

* added about_point to opengl_mobject.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: Abhijith Muthyala <[email protected]>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: 3 people

manim/mobject/mobject.py

@@ -1055,18 +1055,24 @@ def rotate_about_origin(self, angle, axis=OUT, axes=[]):
         """Rotates the :class:`~.Mobject` about the ORIGIN, which is at [0,0,0]."""
         return self.rotate(angle, axis, about_point=ORIGIN)
 
-    def rotate(self, angle, axis=OUT, **kwargs):
+    def rotate(
+        self,
+        angle,
+        axis=OUT,
+        about_point: Union[np.ndarray, List, None] = None,
+        **kwargs,
+    ):
         """Rotates the :class:`~.Mobject` about a certain point."""
         if config.renderer == "opengl":
             rot_matrix_T = rotation_matrix_transpose(angle, axis)
             self.apply_points_function(
-                lambda points: np.dot(points, rot_matrix_T), **kwargs
+                lambda points: np.dot(points, rot_matrix_T), about_point, **kwargs
             )
             return self
         else:
             rot_matrix = rotation_matrix(angle, axis)
             self.apply_points_function_about_point(
-                lambda points: np.dot(points, rot_matrix.T), **kwargs
+                lambda points: np.dot(points, rot_matrix.T), about_point, **kwargs
             )
             return self
 

manim/mobject/opengl_mobject.py

@@ -3,6 +3,7 @@
 import random
 import sys
 from functools import wraps
+from typing import List, Union
 
 import moderngl
 import numpy as np
@@ -56,7 +57,7 @@ def __init__(
         # Must match in attributes of vert shader
         # Event listener
         listen_to_events=False,
-        **kwargs
+        **kwargs,
     ):
 
         self.color = color
@@ -614,7 +615,13 @@ def func(points):
     def rotate_about_origin(self, angle, axis=OUT):
         return self.rotate(angle, axis, about_point=ORIGIN)
 
-    def rotate(self, angle, axis=OUT, **kwargs):
+    def rotate(
+        self,
+        angle,
+        axis=OUT,
+        about_point: Union[np.ndarray, List, None] = None,
+        **kwargs,
+    ):
         rot_matrix_T = rotation_matrix_transpose(angle, axis)
         self.apply_points_function(
             lambda points: np.dot(points, rot_matrix_T), **kwargs
@@ -1404,7 +1411,7 @@ def get_shader_wrapper(self):
     def get_shader_wrapper_list(self):
         shader_wrappers = it.chain(
             [self.get_shader_wrapper()],
-            *[sm.get_shader_wrapper_list() for sm in self.submobjects]
+            *[sm.get_shader_wrapper_list() for sm in self.submobjects],
         )
         batches = batch_by_property(shader_wrappers, lambda sw: sw.get_id())