public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-04-13 09:47:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update TimeVaryingVectorField to match new VectorField configuration

Browse source
This commit is contained in:
Grant Sanderson 2024-11-25 10:50:12 -07:00
parent 90ab2f64bb
commit c6b9826f84
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
manimlib/mobject/vector_field.py
View file

@ -318,13 +318,18 @@ class TimeVaryingVectorField(VectorField):
def __init__(
self,
# Takes in an array of points and a float for time
time_func,
time_func: Callable[[VectArray, float], VectArray],
coordinate_system: CoordinateSystem,
**kwargs
):
self.time = 0
super().__init__(func=lambda p: time_func(p, self.time), **kwargs)
def func(coords):
return time_func(coords, self.time)
super().__init__(func, coordinate_system, **kwargs)
self.add_updater(lambda m, dt: m.increment_time(dt))
always(self.update_vectors)
self.always.update_vectors()
def increment_time(self, dt):
self.time += dt