mirror of
https://github.com/3b1b/manim.git
synced 2025-07-27 12:03:03 +08:00
82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
import os
|
|
import numpy as np
|
|
|
|
|
|
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 os.path.abspath(path)
|
|
|
|
|
|
# def get_scene_output_directory(scene_class):
|
|
# return guarantee_existance(os.path.join(
|
|
# VIDEO_DIR,
|
|
# scene_class.__module__.replace(".", os.path.sep)
|
|
# ))
|
|
|
|
|
|
# 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_partial_movie_output_directory(scene, camera_config, frame_duration):
|
|
# directory = get_movie_output_directory(
|
|
# scene.__class__, camera_config, frame_duration
|
|
# )
|
|
# return guarantee_existance(
|
|
# os.path.join(
|
|
# directory,
|
|
# "partial_movie_files",
|
|
# scene.get_output_file_name(),
|
|
# )
|
|
# )
|
|
|
|
|
|
# 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))
|
|
|
|
|
|
def get_sorted_integer_files(directory,
|
|
min_index=0,
|
|
max_index=np.inf,
|
|
remove_non_integer_files=False,
|
|
remove_indices_greater_than=None,
|
|
extension=None,
|
|
):
|
|
indexed_files = []
|
|
for file in os.listdir(directory):
|
|
if '.' in file:
|
|
index_str = file[:file.index('.')]
|
|
else:
|
|
index_str = file
|
|
|
|
full_path = os.path.join(directory, file)
|
|
if index_str.isdigit():
|
|
index = int(index_str)
|
|
if remove_indices_greater_than is not None:
|
|
if index > remove_indices_greater_than:
|
|
os.remove(full_path)
|
|
continue
|
|
if extension is not None and not file.endswith(extension):
|
|
continue
|
|
if index >= min_index and index < max_index:
|
|
indexed_files.append((index, file))
|
|
elif remove_non_integer_files:
|
|
os.remove(full_path)
|
|
indexed_files.sort(key=lambda p: p[0])
|
|
return list(map(lambda p: os.path.join(directory, p[1]), indexed_files))
|