mirror of
https://github.com/3b1b/manim.git
synced 2025-07-27 20:12:24 +08:00
af65c9d5d4
tl;dr: this is a significant performance improvement for many scenes. 1.7x - 2.6x improvement in animation it/s. This is a small change to some of the hotest paths in rendering objects. The biggest win comes from not using np.allclose() to check if two points are close enough. In general, NumPy is awesome at operating on large arrays, but overkill for very tiny questions like this. Created a small function to determine if two points are close using the same algorithm, and limited it to 2D points since that's all we need in set_cairo_context_path(). A couple of other minor tweaks to reduce or eliminate other uses of NumPy in this path. In general, it is better to avoid wrapping lists in np.array when a real NumPy array isn't actually needed. Added a new file for performance test scenes, with a single scene from the end of a video I've been working on. Data: MacBook Pro (16-inch, 2019) macOS Catalina 10.15.4 2.4 GHz 8-Core Intel Core i9 64 GB 2667 MHz DDR4 Python 3.7.3 (default, Mar 6 2020, 22:34:30) Profiler: yappi under Pycharm. Using the scene Perf1 from the included perf_scenes.py, averaged over 5 runs and rendered with: manim.py perf_scenes.py Perf1 -pl --leave_progress_bars Before: Animation 0: FadeInTextMobject, etc.: 8.93it/s Animation 1: ShowCreationParametricFunction, etc.: 84.66it/s Profiler shows 48.5% of the run spent under Camera.set_cairo_context_path() After Animation 0: FadeInTextMobject, etc.: 23.45it/s -- 2.63x improvement Animation 1: ShowCreationParametricFunction, etc.: 149.62it/s -- 1.77x improvement Profiler shows 19.9% of the run spent under Camera.set_cairo_context_path() Less improvement with production-quality renders, and percent improvement varies with scene of course. This appears to be a good win for every scene I'm working on though. I hope it will be for others, too. NB: there are more perf improvements to be had, of course, but this is the best one I currently have.
89 lines
2.6 KiB
Python
89 lines
2.6 KiB
Python
from manimlib.imports import *
|
|
|
|
"""
|
|
A set of scenes to be used for performance testing of Manim.
|
|
"""
|
|
|
|
|
|
class Perf1(GraphScene):
|
|
"""
|
|
A simple scene of two animations from the end of a video on recursion.
|
|
|
|
- Uses a graph in 1/4 of the scene.
|
|
- First fades in multiple lines of text and equations, and the graph axes.
|
|
- Next animates creation of two graphs and the creation of their text
|
|
labels.
|
|
"""
|
|
CONFIG = {
|
|
"x_axis_label":
|
|
"$n$",
|
|
"y_axis_label":
|
|
"$time$",
|
|
"x_axis_width":
|
|
FRAME_HEIGHT,
|
|
"y_axis_height":
|
|
FRAME_HEIGHT / 2,
|
|
"y_max":
|
|
50,
|
|
"y_min":
|
|
0,
|
|
"x_max":
|
|
100,
|
|
"x_min":
|
|
0,
|
|
"x_labeled_nums": [50, 100],
|
|
"y_labeled_nums":
|
|
range(0, 51, 10),
|
|
"y_tick_frequency":
|
|
10,
|
|
"x_tick_frequency":
|
|
10,
|
|
"axes_color":
|
|
BLUE,
|
|
"graph_origin":
|
|
np.array(
|
|
(-FRAME_X_RADIUS + LARGE_BUFF, -FRAME_Y_RADIUS + LARGE_BUFF, 0))
|
|
}
|
|
|
|
def construct(self):
|
|
t1 = TextMobject(
|
|
"Dividing a problem in half over and over means\\\\"
|
|
"the work done is proportional to $\\log_2{n}$").to_edge(UP)
|
|
|
|
t2 = TextMobject(
|
|
'\\textit{This is one of our\\\\favorite things to do in CS!}')
|
|
t2.to_edge(RIGHT)
|
|
|
|
t3 = TextMobject(
|
|
'The new \\texttt{power(x,n)} is \\underline{much}\\\\better than the old!'
|
|
)
|
|
t3.scale(0.8)
|
|
p1f = TexMobject('x^n=x \\times x^{n-1}').set_color(ORANGE)
|
|
t4 = TextMobject('\\textit{vs.}').scale(0.8)
|
|
p2f = TexMobject(
|
|
'x^n=x^{\\frac{n}{2}} \\times x^{\\frac{n}{2}}').set_color(GREEN)
|
|
p1v2g = VGroup(t3, p1f, t4, p2f).arrange(DOWN).center().to_edge(RIGHT)
|
|
|
|
self.setup_axes()
|
|
o_n = self.get_graph(lambda x: x, color=ORANGE, x_min=1, x_max=50)
|
|
o_log2n = self.get_graph(lambda x: math.log2(x),
|
|
color=GREEN,
|
|
x_min=2,
|
|
x_max=90)
|
|
onl = TexMobject('O(n)')
|
|
olog2nl = TexMobject('O(\\log_2{n})')
|
|
onl.next_to(o_n.get_point_from_function(0.6), UL)
|
|
olog2nl.next_to(o_log2n.get_point_from_function(0.8), UP)
|
|
self.play(
|
|
FadeIn(t1),
|
|
FadeIn(self.axes),
|
|
# FadeInFromDown(t2),
|
|
FadeIn(p1v2g),
|
|
)
|
|
self.play(ShowCreation(o_n),
|
|
ShowCreation(o_log2n),
|
|
ShowCreation(onl),
|
|
ShowCreation(olog2nl),
|
|
run_time=3)
|
|
self.wait(duration=5)
|