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

Speed up bind_graph_to_func

Browse source
This commit is contained in:
Grant Sanderson 2023-01-19 09:50:24 -08:00
parent 148898f983
commit e6abff4299
1 changed files with 9 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
manimlib/mobject/coordinate_systems.py
View file

@ -54,7 +54,7 @@ class CoordinateSystem(ABC):
self,
x_range: RangeSpecifier = DEFAULT_X_RANGE,
y_range: RangeSpecifier = DEFAULT_Y_RANGE,
num_sampled_graph_points_per_tick: int = 20,
num_sampled_graph_points_per_tick: int = 5,
):
self.x_range = x_range
self.y_range = y_range
@ -241,21 +241,20 @@ class CoordinateSystem(ABC):
Use for graphing functions which might change over time, or change with
conditions
"""
x_values = [self.x_axis.p2n(p) for p in graph.get_points()]
x_values = np.array([self.x_axis.p2n(p) for p in graph.get_points()])
def get_x_values():
def get_graph_points():
xs = x_values
if get_discontinuities:
ds = get_discontinuities()
ep = 1e-6
added_xs = it.chain(*((d - ep, d + ep) for d in ds))
return sorted([*x_values, *added_xs])[:len(x_values)]
else:
return x_values
xs[:] = sorted([*x_values, *added_xs])[:len(x_values)]
return self.c2p(xs, func(xs))
graph.add_updater(lambda g: g.set_points_as_corners([
self.c2p(x, func(x))
for x in get_x_values()
]))
graph.add_updater(lambda g: g.set_points_as_corners(
get_graph_points()
))
if not jagged:
graph.add_updater(lambda g: g.make_approximately_smooth())
return graph