public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Simplify get_output_directory

Browse source
This commit is contained in:
Grant Sanderson 2024-12-12 17:38:15 -06:00
parent 7840cd6996
commit 5bf6fde16b
1 changed files with 10 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
manimlib/config.py
View file

@ -312,7 +312,7 @@ def update_scene_config(config: Dict, args: Namespace):
def update_run_config(config: Dict, args: Namespace): def update_run_config(config: Dict, args: Namespace):
config.run = dict( config.run = Dict(
file_name=args.file, file_name=args.file,
embed_line=(int(args.embed) if args.embed is not None else None), embed_line=(int(args.embed) if args.embed is not None else None),
is_reload=False, is_reload=False,
@ -380,17 +380,15 @@ def get_animations_numbers(args: Namespace) -> tuple[int | None, int | None]:
return int(stan), None return int(stan), None
def get_output_directory(args: Namespace, config: dict) -> str: def get_output_directory(args: Namespace, config: Dict) -> str:
dir_config = config["directories"] dir_config = config.directories
output_directory = args.video_dir or dir_config["output"] out_dir = args.video_dir or dir_config.output
if dir_config["mirror_module_path"] and args.file: if dir_config.mirror_module_path and args.file:
to_cut = dir_config["removed_mirror_prefix"] file_path = Path(args.file).absolute()
ext = os.path.abspath(args.file) rel_path = file_path.relative_to(dir_config.removed_mirror_prefix)
ext = ext.replace(to_cut, "").replace(".py", "") rel_path = Path(str(rel_path).lstrip("_"))
if ext.startswith("_"): out_dir = Path(out_dir, rel_path).with_suffix("")
ext = ext[1:] return out_dir
output_directory = os.path.join(output_directory, ext)
return output_directory
# Create global configuration # Create global configuration