mirror of
https://github.com/3b1b/manim.git
synced 2025-07-28 20:43:56 +08:00
13d7228918
This reverts commit 17a1ea6db500eddfdec75cc727b70737d892e961, reversing changes made to c8c6e6d9ba77d42c8d50d8187ca2b0427da0079c.
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
import inspect
|
|
import os
|
|
|
|
from constants import ANIMATIONS_DIR
|
|
|
|
|
|
def add_extension_if_not_present(file_name, extension):
|
|
# This could conceivably be smarter about handling existing differing extensions
|
|
if(file_name[-len(extension):] != extension):
|
|
return file_name + extension
|
|
else:
|
|
return file_name
|
|
|
|
|
|
def guarantee_existance(path):
|
|
if not os.path.exists(path):
|
|
os.makedirs(path)
|
|
return path
|
|
|
|
|
|
def get_scene_output_directory(scene_class):
|
|
file_path = os.path.abspath(inspect.getfile(scene_class))
|
|
|
|
# TODO, is there a better way to do this?
|
|
parts = file_path.split(os.path.sep)
|
|
if "manim" in parts:
|
|
sub_parts = parts[parts.index("manim") + 1:]
|
|
file_path = os.path.join(*sub_parts)
|
|
file_path = file_path.replace(".pyc", "")
|
|
file_path = file_path.replace(".py", "")
|
|
return guarantee_existance(os.path.join(ANIMATIONS_DIR, file_path))
|
|
|
|
|
|
def get_movie_output_directory(scene_class, camera_config, frame_duration):
|
|
directory = get_scene_output_directory(scene_class)
|
|
sub_dir = "%dp%d" % (
|
|
camera_config["pixel_height"],
|
|
int(1.0 / frame_duration)
|
|
)
|
|
return guarantee_existance(os.path.join(directory, sub_dir))
|
|
|
|
|
|
def get_image_output_directory(scene_class, sub_dir="images"):
|
|
directory = get_scene_output_directory(scene_class)
|
|
return guarantee_existance(os.path.join(directory, sub_dir))
|