mirror of
https://github.com/3b1b/manim.git
synced 2025-07-27 12:03:03 +08:00
70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
|
|
|
|
import inspect
|
|
import os
|
|
import sys
|
|
|
|
from constants import PRODUCTION_QUALITY_CAMERA_CONFIG
|
|
from constants import PRODUCTION_QUALITY_FRAME_DURATION
|
|
from extract_scene import get_module
|
|
from extract_scene import is_scene
|
|
from utils.output_directory_getters import get_movie_output_directory
|
|
|
|
|
|
def get_sorted_scene_classes(module_name):
|
|
module = get_module(module_name)
|
|
line_to_scene = {}
|
|
for name, scene_class in inspect.getmembers(module, is_scene):
|
|
if inspect.getmodule(scene_class) != module:
|
|
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
|
|
animation_dir = get_movie_output_directory(
|
|
scene_classes[0],
|
|
PRODUCTION_QUALITY_CAMERA_CONFIG,
|
|
PRODUCTION_QUALITY_FRAME_DURATION,
|
|
)
|
|
files = os.listdir(animation_dir)
|
|
sorted_files = []
|
|
for scene_class in scene_classes:
|
|
scene_name = scene_class.__name__
|
|
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_%d" % 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 enumerate(sorted_files):
|
|
symlink_name = os.path.join(
|
|
staged_scenes_dir,
|
|
"Scene_%03d" % count + f.split(os.sep)[-1]
|
|
)
|
|
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)
|