enhancement(vector_field): added fit_to_coordinate_system (#2728)

* enhancement(vector_field): added fit_to_coordinate_system

* tmp(vector_field) decision stage for implementation

* Update vector_field.py

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

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

* Fix OpenGL segfaulting when running self.play or self.wait (#2734)

* make play run only on main thread

* prefer dict.update

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Noam Zaks <[email protected]>
Co-authored-by: Benjamin Hackl <[email protected]>

Author: 4 people

manim/mobject/vector_field.py

@@ -19,6 +19,7 @@
 
 from manim.animation.updaters.update import UpdateFromAlphaFunc
 from manim.mobject.geometry.line import Vector
+from manim.mobject.graphing.coordinate_systems import CoordinateSystem
 from manim.mobject.opengl.opengl_vectorized_mobject import OpenGLVMobject
 
 from .. import config
@@ -168,6 +169,22 @@ def construct(self):
         """
         return lambda p: func(p * scalar)
 
+    def fit_to_coordinate_system(self, coordinate_system: CoordinateSystem):
+        """Scale the vector field to fit a coordinate system.
+
+        This method is useful when the vector field is defined in a coordinate system
+        different from the one used to display the vector field.
+
+        This method can only be used once because it transforms the origin of each vector.
+
+        Parameters
+        ----------
+        coordinate_system
+            The coordinate system to fit the vector field to.
+
+        """
+        self.apply_function(lambda pos: coordinate_system.coords_to_point(*pos))
+
     def nudge(
         self,
         mob: Mobject,