opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-07-31 14:03:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed bug in set_background_by_color_function

Browse Source
This commit is contained in:
Sridhar Ramesh
2018-01-30 18:06:05 -08:00
parent 634b28c485
commit c28b81a230
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
camera/camera.py
View File

@ -385,12 +385,12 @@ class Camera(object):
def get_points_of_all_pixels(self): def get_points_of_all_pixels(self):
""" """
Returns an array a such that a[i, j] gives the spatial Returns an array a such that a[i, j] gives the spatial
coordsinates associated with the pixel self.pixel_array[i, j] coordinates associated with the pixel self.pixel_array[i, j]
""" """
shape = self.pixel_array.shape shape = self.pixel_array.shape
indices = np.indices(shape[:2], dtype = 'float64') indices = np.indices(shape[:2], dtype = 'float64')
all_point_coords = np.zeros((shape[0], shape[1], 3)) all_point_coords = np.zeros((shape[0], shape[1], 3))
for i, space_dim in enumerate([SPACE_WIDTH, SPACE_HEIGHT]): for i, space_dim in enumerate([SPACE_HEIGHT, SPACE_WIDTH]):
all_point_coords[:,:,i] = \ all_point_coords[:,:,i] = \
indices[i,:,:]*2*space_dim/shape[i] - space_dim indices[i,:,:]*2*space_dim/shape[i] - space_dim
return all_point_coords return all_point_coords
@ -411,7 +411,7 @@ class Camera(object):
lambda p : float_rgba_to_int_rgba(point_to_rgba_func(p)), lambda p : float_rgba_to_int_rgba(point_to_rgba_func(p)),
2, points_of_all_pixels 2, points_of_all_pixels
)) ))
self.reset() self.reset() # Perhaps this really belongs in set_background?
class MovingCamera(Camera): class MovingCamera(Camera):