mirror of
https://github.com/3b1b/manim.git
synced 2025-04-13 09:47:07 +00:00
96 lines
3.3 KiB
Python
96 lines
3.3 KiB
Python
#!/usr/bin/env python
|
|
import inspect
|
|
import os
|
|
import sys
|
|
import importlib
|
|
|
|
from manimlib.constants import PRODUCTION_QUALITY_CAMERA_CONFIG
|
|
from manimlib.constants import PRODUCTION_QUALITY_FRAME_DURATION
|
|
from manimlib.config import get_module
|
|
from manimlib.extract_scene import is_child_scene
|
|
from manimlib.utils.output_directory_getters import get_movie_output_directory
|
|
from manimlib.utils.output_directory_getters import get_partial_movie_output_directory
|
|
from manimlib.utils.output_directory_getters import get_sorted_integer_files
|
|
|
|
|
|
def get_sorted_scene_classes(module_name):
|
|
module = get_module(module_name)
|
|
importlib.import_module(module.__name__)
|
|
line_to_scene = {}
|
|
name_scene_list = inspect.getmembers(
|
|
module,
|
|
lambda obj: is_child_scene(obj, module)
|
|
)
|
|
for name, scene_class in name_scene_list:
|
|
if inspect.getmodule(scene_class).__name__ != module.__name__:
|
|
continue
|
|
lines, line_no = inspect.getsourcelines(scene_class)
|
|
line_to_scene[line_no] = scene_class
|
|
return [
|
|
line_to_scene[index]
|
|
for index in sorted(line_to_scene.keys())
|
|
]
|
|
|
|
|
|
def stage_animations(module_name):
|
|
scene_classes = get_sorted_scene_classes(module_name)
|
|
if len(scene_classes) == 0:
|
|
print("There are no rendered animations from this module")
|
|
return
|
|
output_directory_kwargs = {
|
|
"camera_config": PRODUCTION_QUALITY_CAMERA_CONFIG,
|
|
"frame_duration": PRODUCTION_QUALITY_FRAME_DURATION,
|
|
}
|
|
animation_dir = get_movie_output_directory(
|
|
scene_classes[0], **output_directory_kwargs
|
|
)
|
|
files = os.listdir(animation_dir)
|
|
sorted_files = []
|
|
for scene_class in scene_classes:
|
|
scene_name = scene_class.__name__
|
|
# Partial movie file directory
|
|
pmf_dir = get_partial_movie_output_directory(
|
|
scene_class, **output_directory_kwargs
|
|
)
|
|
if os.path.exists(pmf_dir):
|
|
for extension in [".mov", ".mp4"]:
|
|
int_files = get_sorted_integer_files(
|
|
pmf_dir, extension=extension
|
|
)
|
|
for file in int_files:
|
|
sorted_files.append(os.path.join(pmf_dir, file))
|
|
else:
|
|
for clip in [f for f in files if f.startswith(scene_name + ".")]:
|
|
sorted_files.append(os.path.join(animation_dir, clip))
|
|
|
|
animation_subdir = os.path.dirname(animation_dir)
|
|
count = 0
|
|
while True:
|
|
staged_scenes_dir = os.path.join(
|
|
animation_subdir, "staged_scenes_{}".format(count)
|
|
)
|
|
if not os.path.exists(staged_scenes_dir):
|
|
os.makedirs(staged_scenes_dir)
|
|
break
|
|
# Otherwise, keep trying new names until
|
|
# there is a free one
|
|
count += 1
|
|
for count, f in reversed(list(enumerate(sorted_files))):
|
|
# Going in reversed order means that when finder
|
|
# sorts by date modified, it shows up in the
|
|
# correct order
|
|
symlink_name = os.path.join(
|
|
staged_scenes_dir,
|
|
"Scene_{:03}_{}".format(
|
|
count,
|
|
"".join(f.split(os.sep)[-2:])
|
|
)
|
|
)
|
|
os.symlink(f, symlink_name)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) < 2:
|
|
raise Exception("No module given.")
|
|
module_name = sys.argv[1]
|
|
stage_animations(module_name)
|