github/manim
1
1
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2026-04-26 03:00:23 -04:00
Code Issues Packages Projects Releases Wiki Activity

Improved gradient_hightlight of PointCloudMobject

Browse Source
This commit is contained in:
Grant Sanderson
2018-01-17 12:17:58 -08:00
parent 0bf1a9e303
commit 83e28f9efb
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
mobject/point_cloud_mobject.py
View File

@@ -37,7 +37,14 @@ class PMobject(Mobject):
mob.rgbas[:,:] = rgba
return self
def gradient_highlight(self, start_color, end_color):
# def gradient_highlight(self, start_color, end_color):
def gradient_highlight(self, *colors):
self.rgbas = np.array(map(
color_to_rgba,
color_gradient(colors, len(self.points))
))
return self
start_rgba, end_rgba = map(color_to_rgba, [start_color, end_color])
for mob in self.family_members_with_points():
num_points = mob.get_num_points()