public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-09-01 00:48:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixes to Surface

Browse source
This commit is contained in:
Grant Sanderson 2023-06-10 09:22:15 -07:00
parent 63f6e9d84f
commit bae3b98c0b
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
manimlib/mobject/types/surface.py
View file

@ -130,11 +130,13 @@ class Surface(Mobject):
def get_unit_normals(self) -> Vect3Array: def get_unit_normals(self) -> Vect3Array:
nu, nv = self.resolution nu, nv = self.resolution
indices = np.arange(nu * nv) indices = np.arange(nu * nv)
if len(indices) == 0:
return np.zeros((3, 0))
left = indices - 1 left = indices - 1
right = indices + 1 right = indices + 1
up = indices - nv up = indices - nv
down = indices + nv down = indices + nv
left[0] = indices[0] left[0] = indices[0]
right[-1] = indices[-1] right[-1] = indices[-1]
@ -166,7 +168,7 @@ class Surface(Mobject):
nu, nv = smobject.resolution nu, nv = smobject.resolution
self.data['point'][:] = self.get_partial_points_array( self.data['point'][:] = self.get_partial_points_array(
self.data['point'], a, b, smobject.data['point'], a, b,
(nu, nv, 3), (nu, nv, 3),
axis=axis axis=axis
) )
@ -183,7 +185,7 @@ class Surface(Mobject):
if len(points) == 0: if len(points) == 0:
return points return points
nu, nv = resolution[:2] nu, nv = resolution[:2]
points = points.reshape(resolution) points = points.reshape(resolution).copy()
max_index = resolution[axis] - 1 max_index = resolution[axis] - 1
lower_index, lower_residue = integer_interpolate(0, max_index, a) lower_index, lower_residue = integer_interpolate(0, max_index, a)
upper_index, upper_residue = integer_interpolate(0, max_index, b) upper_index, upper_residue = integer_interpolate(0, max_index, b)