From 83e28f9efb6ca7fc7fb45f565841a867612a52ce Mon Sep 17 00:00:00 2001 From: Grant Sanderson Date: Wed, 17 Jan 2018 12:17:58 -0800 Subject: [PATCH] Improved gradient_hightlight of PointCloudMobject --- mobject/point_cloud_mobject.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/mobject/point_cloud_mobject.py b/mobject/point_cloud_mobject.py index 7ce367c5..08dfe7c8 100644 --- a/mobject/point_cloud_mobject.py +++ b/mobject/point_cloud_mobject.py @@ -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()