opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-07-30 13:34:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

get_shader_data_array -> get_blank_shader_data_array

Browse Source
This commit is contained in:
Grant Sanderson
2020-02-13 15:51:16 -08:00
parent cfbcd9faa8
commit 0912043248
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
manimlib/mobject/mobject.py
View File

@ -1132,7 +1132,7 @@ class Mobject(Container):
def init_shader_data(self): def init_shader_data(self):
self.shader_data = np.zeros(len(self.points), dtype=self.shader_dtype) self.shader_data = np.zeros(len(self.points), dtype=self.shader_dtype)
def get_shader_data_array(self, size, name="shader_data"): def get_blank_shader_data_array(self, size, name="shader_data"):
# If possible, try to populate an existing array, rather # If possible, try to populate an existing array, rather
# than recreating it each frame # than recreating it each frame
arr = getattr(self, name) arr = getattr(self, name)

2
manimlib/mobject/types/image_mobject.py
View File

@ -42,7 +42,7 @@ class ImageMobject(Mobject):
self.set_opacity(self.opacity) self.set_opacity(self.opacity)
def get_shader_data(self): def get_shader_data(self):
data = self.get_shader_data_array(len(self.points)) data = self.get_blank_shader_data_array(len(self.points))
data["point"] = self.points data["point"] = self.points
data["im_coords"] = self.im_coords data["im_coords"] = self.im_coords
data["opacity"] = self.opacity data["opacity"] = self.opacity

4
manimlib/mobject/types/vectorized_mobject.py
View File

@ -849,7 +849,7 @@ class VMobject(Mobject):
if len(stroke_width) > 1: if len(stroke_width) > 1:
stroke_width = self.stretched_style_array_matching_points(stroke_width) stroke_width = self.stretched_style_array_matching_points(stroke_width)
data = self.get_shader_data_array(len(self.points), "stroke_data") data = self.get_blank_shader_data_array(len(self.points), "stroke_data")
data['point'] = self.points data['point'] = self.points
data['prev_point'][:3] = self.points[-3:] data['prev_point'][:3] = self.points[-3:]
data['prev_point'][3:] = self.points[:-3] data['prev_point'][3:] = self.points[:-3]
@ -945,7 +945,7 @@ class VMobject(Mobject):
# TODO, best way to enable multiple colors? # TODO, best way to enable multiple colors?
rgbas = self.get_fill_rgbas()[:1] rgbas = self.get_fill_rgbas()[:1]
data = self.get_shader_data_array(len(tri_indices), "fill_data") data = self.get_blank_shader_data_array(len(tri_indices), "fill_data")
data["point"] = points[tri_indices] data["point"] = points[tri_indices]
data["color"] = rgbas data["color"] = rgbas
# Assume the triangulation is such that the first n_points points # Assume the triangulation is such that the first n_points points