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

Add option to insert embed line from the command line (mildly hacky)

Browse source
This commit is contained in:
Grant Sanderson 2021-11-14 12:31:56 -08:00
parent 7070777408
commit d065e1973d
1 changed files with 48 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

60
manimlib/config.py
View file

@ -5,6 +5,7 @@ import importlib
import os import os
import sys import sys
import yaml import yaml
from contextlib import contextmanager
from screeninfo import get_monitors from screeninfo import get_monitors
from manimlib.utils.config_ops import merge_dicts_recursively from manimlib.utils.config_ops import merge_dicts_recursively
@ -12,6 +13,9 @@ from manimlib.utils.init_config import init_customization
from manimlib.logger import log from manimlib.logger import log
__config_file__ = "custom_config.yml"
def parse_cli(): def parse_cli():
try: try:
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
@ -112,6 +116,12 @@ def parse_cli():
"in two comma separated values, e.g. \"3,6\", it will end" "in two comma separated values, e.g. \"3,6\", it will end"
"the rendering at the second value", "the rendering at the second value",
) )
parser.add_argument(
"-e", "--embed",
help="Takes a line number as an argument, and results"
"in the scene being called as if the line `self.embed()`"
"was inserted into the scene code at that line number."
)
parser.add_argument( parser.add_argument(
"-r", "--resolution", "-r", "--resolution",
help="Resolution, passed as \"WxH\", e.g. \"1920x1080\"", help="Resolution, passed as \"WxH\", e.g. \"1920x1080\"",
@ -162,14 +172,31 @@ def get_manim_dir():
def get_module(file_name): def get_module(file_name):
if file_name is None: if file_name is None:
return None return None
else: module_name = file_name.replace(os.sep, ".").replace(".py", "")
module_name = file_name.replace(os.sep, ".").replace(".py", "") spec = importlib.util.spec_from_file_location(module_name, file_name)
spec = importlib.util.spec_from_file_location(module_name, file_name) module = importlib.util.module_from_spec(spec)
module = importlib.util.module_from_spec(spec) spec.loader.exec_module(module)
spec.loader.exec_module(module) return module
return module
@contextmanager
def insert_embed_line(file_name, lineno):
with open(file_name, 'r') as fp:
lines = fp.readlines()
line = lines[lineno - 1]
n_spaces = len(line) - len(line.lstrip())
lines.insert(lineno - 1, " " * n_spaces + "self.embed()\n")
# Where should this really live?
alt_file = file_name.replace(".py", "_alt.py")
with open(alt_file, 'w') as fp:
fp.writelines(lines)
try:
yield alt_file
finally:
os.remove(alt_file)
__config_file__ = "custom_config.yml"
def get_custom_config(): def get_custom_config():
global __config_file__ global __config_file__
@ -197,9 +224,11 @@ def get_custom_config():
def check_temporary_storage(config): def check_temporary_storage(config):
if config["directories"]["temporary_storage"] == "" and sys.platform == "win32": if config["directories"]["temporary_storage"] == "" and sys.platform == "win32":
log.warning("You may be using Windows platform and have not specified the path of" log.warning(
"You may be using Windows platform and have not specified the path of"
" `temporary_storage`, which may cause OSError. So it is recommended" " `temporary_storage`, which may cause OSError. So it is recommended"
" to specify the `temporary_storage` in the config file (.yml)") " to specify the `temporary_storage` in the config file (.yml)"
)
def get_configuration(args): def get_configuration(args):
@ -229,8 +258,10 @@ def get_configuration(args):
elif not os.path.exists(__config_file__): elif not os.path.exists(__config_file__):
log.info(f"Using the default configuration file, which you can modify in `{global_defaults_file}`") log.info(f"Using the default configuration file, which you can modify in `{global_defaults_file}`")
log.info("If you want to create a local configuration file, you can create a file named" log.info(
f" `{__config_file__}`, or run `manimgl --config`") "If you want to create a local configuration file, you can create a file named"
f" `{__config_file__}`, or run `manimgl --config`"
)
custom_config = get_custom_config() custom_config = get_custom_config()
check_temporary_storage(custom_config) check_temporary_storage(custom_config)
@ -260,7 +291,12 @@ def get_configuration(args):
"quiet": args.quiet, "quiet": args.quiet,
} }
module = get_module(args.file) if args.embed is None:
module = get_module(args.file)
else:
with insert_embed_line(args.file, int(args.embed)) as alt_file:
module = get_module(alt_file)
config = { config = {
"module": module, "module": module,
"scene_names": args.scene_names, "scene_names": args.scene_names,