mirror of
https://github.com/jeertmans/manim-slides.git
synced 2025-05-22 04:56:24 +08:00
67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
import os
|
|
import sys
|
|
from contextlib import contextmanager
|
|
from importlib.util import find_spec
|
|
|
|
|
|
@contextmanager
|
|
def suppress_stdout():
|
|
with open(os.devnull, "w") as devnull:
|
|
old_stdout = sys.stdout
|
|
sys.stdout = devnull
|
|
try:
|
|
yield
|
|
finally:
|
|
sys.stdout = old_stdout
|
|
|
|
|
|
MANIM_PACKAGE_NAME = "manim"
|
|
MANIM_AVAILABLE = find_spec(MANIM_PACKAGE_NAME) is not None
|
|
MANIM_IMPORTED = MANIM_PACKAGE_NAME in sys.modules
|
|
|
|
MANIMGL_PACKAGE_NAME = "manimlib"
|
|
MANIMGL_AVAILABLE = find_spec(MANIMGL_PACKAGE_NAME) is not None
|
|
MANIMGL_IMPORTED = MANIMGL_PACKAGE_NAME in sys.modules
|
|
|
|
if MANIM_IMPORTED and MANIMGL_IMPORTED:
|
|
from manim import logger
|
|
|
|
logger.warn(
|
|
"Both manim and manimgl are installed, therefore `manim-slide` needs to need which one to use. Please only import one of the two modules so that `manim-slide` knows which one to use. Here, manim is used by default"
|
|
)
|
|
MANIM = True
|
|
MANIMGL = False
|
|
elif MANIM_IMPORTED:
|
|
MANIM = True
|
|
MANIMGL = False
|
|
elif MANIMGL_IMPORTED:
|
|
MANIM = False
|
|
MANIMGL = True
|
|
elif MANIM_AVAILABLE:
|
|
MANIM = True
|
|
MANIMGL = False
|
|
elif MANIMGL_AVAILABLE:
|
|
MANIM = False
|
|
MANIMGL = True
|
|
else:
|
|
raise ModuleNotFoundError(
|
|
"Either manim (community) or manimgl (3b1b) package must be installed"
|
|
)
|
|
|
|
|
|
FFMPEG_BIN = None
|
|
|
|
if MANIMGL:
|
|
from manimlib import Scene, ThreeDScene, config
|
|
from manimlib.constants import FFMPEG_BIN
|
|
from manimlib.logger import log as logger
|
|
|
|
else:
|
|
with suppress_stdout(): # Avoids printing "Manim Community v..."
|
|
from manim import Scene, ThreeDScene, config, logger
|
|
|
|
try: # For manim<v0.16.0.post0
|
|
from manim.constants import FFMPEG_BIN as FFMPEG_BIN
|
|
except ImportError:
|
|
FFMPEG_BIN = config.ffmpeg_executable
|