mirror of
https://github.com/3b1b/manim.git
synced 2025-04-13 09:47:07 +00:00
44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
import numpy as np
|
|
from PIL import Image
|
|
|
|
from manimlib.utils.directories import get_raster_image_dir
|
|
from manimlib.utils.directories import get_vector_image_dir
|
|
from manimlib.utils.file_ops import find_file
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from typing import Iterable
|
|
|
|
|
|
def get_full_raster_image_path(image_file_name: str) -> str:
|
|
return find_file(
|
|
image_file_name,
|
|
directories=[get_raster_image_dir()],
|
|
extensions=[".jpg", ".jpeg", ".png", ".gif", ""]
|
|
)
|
|
|
|
|
|
def get_full_vector_image_path(image_file_name: str) -> str:
|
|
return find_file(
|
|
image_file_name,
|
|
directories=[get_vector_image_dir()],
|
|
extensions=[".svg", ".xdv", ""],
|
|
)
|
|
|
|
|
|
def drag_pixels(frames: Iterable) -> list:
|
|
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: Iterable) -> Image:
|
|
arr = np.array(image)
|
|
arr = (255 * np.ones(arr.shape)).astype(arr.dtype) - arr
|
|
return Image.fromarray(arr)
|