mirror of
https://github.com/3b1b/manim.git
synced 2025-07-28 04:23:16 +08:00
31 lines
720 B
Python
31 lines
720 B
Python
import numpy as np
|
|
import os
|
|
|
|
from PIL import Image
|
|
|
|
from manimlib.constants import RASTER_IMAGE_DIR
|
|
from manimlib.utils.file_ops import seek_full_path_from_defaults
|
|
|
|
|
|
def get_full_raster_image_path(image_file_name):
|
|
return seek_full_path_from_defaults(
|
|
image_file_name,
|
|
default_dir=RASTER_IMAGE_DIR,
|
|
extensions=[".jpg", ".png", ".gif"]
|
|
)
|
|
|
|
|
|
def drag_pixels(frames):
|
|
curr = frames[0]
|
|
new_frames = []
|
|
for frame in frames:
|
|
curr += (curr == 0) * np.array(frame)
|
|
new_frames.append(np.array(curr))
|
|
return new_frames
|
|
|
|
|
|
def invert_image(image):
|
|
arr = np.array(image)
|
|
arr = (255 * np.ones(arr.shape)).astype(arr.dtype) - arr
|
|
return Image.fromarray(arr)
|