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

Tiny rename in camera

Browse source
This commit is contained in:
Grant Sanderson 2018-01-17 15:13:30 -08:00
parent 96b48388dd
commit 1b67abd8a9
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
camera.py
View file

@ -193,12 +193,12 @@ class Camera(object):
pixel_coords = self.thickened_coordinates( pixel_coords = self.thickened_coordinates(
pixel_coords, thickness pixel_coords, thickness
) )
rgb_len = self.pixel_array.shape[2] rgba_len = self.pixel_array.shape[2]
rgbas = (255*rgbas).astype('uint8') rgbas = (255*rgbas).astype('uint8')
target_len = len(pixel_coords) target_len = len(pixel_coords)
factor = target_len/len(rgbas) factor = target_len/len(rgbas)
rgbas = np.array([rgbas]*factor).reshape((target_len, rgb_len)) rgbas = np.array([rgbas]*factor).reshape((target_len, rgba_len))
on_screen_indices = self.on_screen_pixels(pixel_coords) on_screen_indices = self.on_screen_pixels(pixel_coords)
pixel_coords = pixel_coords[on_screen_indices] pixel_coords = pixel_coords[on_screen_indices]
@ -211,9 +211,9 @@ class Camera(object):
indices = np.dot(pixel_coords, flattener)[:,0] indices = np.dot(pixel_coords, flattener)[:,0]
indices = indices.astype('int') indices = indices.astype('int')
new_pa = self.pixel_array.reshape((ph*pw, rgb_len)) new_pa = self.pixel_array.reshape((ph*pw, rgba_len))
new_pa[indices] = rgbas new_pa[indices] = rgbas
self.pixel_array = new_pa.reshape((ph, pw, rgb_len)) self.pixel_array = new_pa.reshape((ph, pw, rgba_len))
def display_image_mobject(self, image_mobject): def display_image_mobject(self, image_mobject):
corner_coords = self.points_to_pixel_coords(image_mobject.points) corner_coords = self.points_to_pixel_coords(image_mobject.points)