opensource/manim-slides
1
0
Fork 0
mirror of https://github.com/jeertmans/manim-slides.git synced 2025-05-21 12:37:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(cli): reverse videos on the fly

Browse Source
This commit is contained in:
Jérome Eertmans
2022-09-07 23:24:54 +02:00
parent efbe488660 3520c42d7c
commit 9640605ae9
7 changed files with 94 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
manim_slides/slide.py
View File

@ -91,11 +91,12 @@ class Slide(Scene):
scene_name = type(self).__name__
scene_files_folder = os.path.join(files_folder, scene_name)
if os.path.exists(scene_files_folder):
shutil.rmtree(scene_files_folder)
old_animation_files = set()
if not os.path.exists(scene_files_folder):
os.mkdir(scene_files_folder)
else:
old_animation_files.update(os.listdir(scene_files_folder))
files = list()
for src_file in tqdm(
@ -105,10 +106,25 @@ class Slide(Scene):
ascii=True if platform.system() == "Windows" else None,
disable=config["progress_bar"] == "none",
):
dst_file = os.path.join(scene_files_folder, os.path.basename(src_file))
shutil.copyfile(src_file, dst_file)
rev_file = reverse_video_path(dst_file)
reverse_video_file(src_file, rev_file)
filename = os.path.basename(src_file)
_hash, ext = os.path.splitext(filename)
rev_filename = f"{_hash}_reversed{ext}"
dst_file = os.path.join(scene_files_folder, filename)
# We only copy animation if it was not present
if filename in old_animation_files:
old_animation_files.remove(filename)
else:
shutil.copyfile(src_file, dst_file)
# We only reverse video if it was not present
if rev_filename in old_animation_files:
old_animation_files.remove(rev_filename)
else:
rev_file = os.path.join(scene_files_folder, rev_filename)
reverse_video_file(src_file, rev_file)
files.append(dst_file)
logger.info(