opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-01 08:54:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

New MobjectFromImageArray class and MobjectFromRegion class

Browse Source
This commit is contained in:
Grant Sanderson
2015-12-20 16:44:22 -08:00
parent 83c8d8bbd4
commit c7389e008d
1 changed files with 23 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
mobject/image_mobject.py
View File

@ -6,6 +6,7 @@ from random import random
from helpers import *
from mobject import Mobject
import displayer as disp
class ImageMobject(Mobject):
"""
@ -22,7 +23,6 @@ class ImageMobject(Mobject):
def __init__(self, image_file, **kwargs):
digest_locals(self)
Mobject.__init__(self, **kwargs)
self.filter_rgb = 255 * np.array(Color(self.filter_color).get_rgb()).astype('uint8')
self.name = to_cammel_case(
os.path.split(image_file)[-1].split(".")[0]
)
@ -82,7 +82,9 @@ class ImageMobject(Mobject):
height, width = image_array.shape[:2]
#Flatten array, and find indices where rgb is not filter_rgb
array = image_array.reshape((height * width, 3))
bools = array == self.filter_rgb
filter_rgb = np.array(Color(self.filter_color).get_rgb())
filter_rgb = 255*filter_rgb.astype('uint8')
bools = array == filter_rgb
bools = bools[:,0]*bools[:,1]*bools[:,2]
indices = np.arange(height * width, dtype = 'int')[~bools]
rgbs = array[indices, :].astype('float') / 255.0
@ -98,3 +100,22 @@ class ImageMobject(Mobject):
points *= 2 * SPACE_WIDTH / width
self.add_points(points, rgbs = rgbs)
class MobjectFromPixelArray(ImageMobject):
def __init__(self, image_array, **kwargs):
Mobject.__init__(self, **kwargs)
self.generate_points_from_image_array(image_array)
class MobjectFromRegion(MobjectFromPixelArray):
def __init__(self, region, **kwargs):
MobjectFromPixelArray.__init__(
self,
disp.paint_region(region),
**kwargs
)