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

Add point_to_rgb

Browse source
This commit is contained in:
Grant Sanderson 2020-08-12 13:23:21 -07:00
parent 150b03427d
commit ae590e7fde
1 changed files with 17 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
manimlib/mobject/types/image_mobject.py
View file

@ -5,6 +5,7 @@ from PIL import Image
from manimlib.constants import *
from manimlib.mobject.mobject import Mobject
from manimlib.utils.bezier import interpolate
from manimlib.utils.bezier import inverse_interpolate
from manimlib.utils.images import get_full_raster_image_path
from manimlib.utils.iterables import listify
@ -58,6 +59,22 @@ class ImageMobject(Mobject):
mobject1.opacity, mobject2.opacity, alpha
)
def point_to_rgb(self, point):
x0, y0 = self.get_corner(UL)[:2]
x1, y1 = self.get_corner(DR)[:2]
x_alpha = inverse_interpolate(x0, x1, point[0])
y_alpha = inverse_interpolate(y0, y1, point[1])
if not (0 <= x_alpha <= 1) and (0 <= y_alpha <= 1):
# TODO, raise smarter exception
raise Exception("Cannot sample color from outside an image")
pw, ph = self.image.size
rgb = self.image.getpixel((
int((pw - 1) * x_alpha),
int((ph - 1) * y_alpha),
))
return np.array(rgb) / 255
def get_shader_data(self):
data = self.get_blank_shader_data_array(len(self.points))
data["point"] = self.points