opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-07-30 13:34:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Prepping for hilbert curve project

Browse Source
This commit is contained in:
Grant Sanderson
2015-12-13 15:41:45 -08:00
parent b2c5f71155
commit 399852a994
10 changed files with 57 additions and 381 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
extract_scene.py
View File

@ -45,7 +45,7 @@ def get_configuration(sys_argv):
print str(err)
sys.exit(2)
config = {
"module" : None,
"file" : None,
"scene_name" : "",
"args_extension" : "",
"display_config" : PRODUCTION_QUALITY_DISPLAY_CONFIG,
@ -83,7 +83,7 @@ def get_configuration(sys_argv):
if len(args) == 0:
print HELP_MESSAGE
sys.exit()
config["module"] = args[0].replace(".py", "")
config["file"] = args[0]
if len(args) > 1:
config["scene_name"] = args[1]
if len(args) > 2:
@ -180,17 +180,22 @@ def get_scene_classes(scene_names_to_classes, config):
return scene_names_to_classes.values()
return prompt_user_for_choice(scene_names_to_classes)
def get_module(file_name):
module_name = file_name.replace(".py", "")
last_module = imp.load_module(".", *imp.find_module("."))
for part in module_name.split(os.sep):
load_args = imp.find_module(part, last_module.__path__)
last_module = imp.load_module(part, *load_args)
return last_module
def main():
config = get_configuration(sys.argv)
module = imp.load_module(
config["module"],
*imp.find_module(config["module"])
)
module = get_module(config["file"])
scene_names_to_classes = dict(
inspect.getmembers(module, is_scene)
)
config["movie_prefix"] = config["module"]
config["movie_prefix"] = config["file"].replace(".py", "")
scene_kwargs = config["display_config"]
scene_kwargs["announce_construction"] = True
for SceneClass in get_scene_classes(scene_names_to_classes, config):