opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-03 04:04:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Slightly faster sort_points method, and bug fix in fade

Browse Source
This commit is contained in:
Grant Sanderson
2015-12-21 12:44:54 -08:00
parent 7f45044baf
commit 7ae5a0eccb
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mobject/mobject.py
View File

@ -194,12 +194,10 @@ class Mobject(object):
function is any map from R^3 to R
"""
for mob in self.get_full_submobject_family():
indices = range(len(mob.points))
indices.sort(
lambda *pair : cmp(*map(function, mob.points[pair, :]))
indices = np.argsort(
np.apply_along_axis(function, 1, mob.points)
)
mob.points = mob.points[indices]
mob.rgbs = mob.rgbs[indices]
mob.apply_over_attr_arrays(lambda arr : arr[indices])
return self
def repeat(self, count):
@ -309,7 +307,7 @@ class Mobject(object):
return self
def fade_to(self, color, alpha):
self.rgbs = interpolate(self.rgbs, np.array(Color(color).rgb), alpha)
self.rgbs = interpolate(self.rgbs, np.array(Color(color).rgb), 1-alpha)
for mob in self.sub_mobjects:
mob.fade_to(color, alpha)
return self