public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Tidy up ShaderWrapper.read_in

Browse source
This commit is contained in:
Grant Sanderson 2023-01-16 13:26:34 -08:00
parent 3f8c861973
commit f5480d02ff
1 changed files with 15 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
manimlib/shader_wrapper.py
View file

@ -155,21 +155,24 @@ class ShaderWrapper(object):
if total_len == 0:
return self
if indices_list is not None and self.vert_indices is not None:
# Stack the data
np.concatenate(data_list, out=self.vert_data)
if indices_list is None or self.vert_indices is None:
return self
total_verts = sum(len(vi) for vi in indices_list)
self.vert_indices = resize_array(self.vert_indices, total_verts)
# Stack vert_indices, but adding the appropriate offset
# alogn the way
n_points = 0
n_verts = 0
for k, data in enumerate(data_list):
new_n_points = n_points + len(data)
self.vert_data[n_points:new_n_points] = data
if self.vert_indices is not None and indices_list is not None:
vert_indices = indices_list[k]
new_n_verts = n_verts + len(vert_indices)
self.vert_indices[n_verts:new_n_verts] = vert_indices + n_points
for data, indices in zip(data_list, indices_list):
new_n_verts = n_verts + len(indices)
self.vert_indices[n_verts:new_n_verts] = indices + n_points
n_verts = new_n_verts
n_points = new_n_points
n_points += len(data)
return self