From 240742a61324a99bce92d0d656c8ed3bc4344b87 Mon Sep 17 00:00:00 2001
From: Grant Sanderson <grantsanderson7@gmail.com>
Date: Wed, 23 Dec 2015 12:31:22 -0800
Subject: [PATCH] Added script to stage animations in nicely sorted order

---
 constants.py        |  4 +++-
 stage_animations.py | 45 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 48 insertions(+), 1 deletion(-)
 create mode 100644 stage_animations.py

diff --git a/constants.py b/constants.py
index cfdb0609..ef15bc13 100644
--- a/constants.py
+++ b/constants.py
@@ -63,13 +63,15 @@ FILE_DIR          = os.path.join(THIS_DIR, os.pardir, "animation_files")
 IMAGE_DIR         = os.path.join(FILE_DIR, "images")
 GIF_DIR           = os.path.join(FILE_DIR, "gifs")
 MOVIE_DIR         = os.path.join(FILE_DIR, "movies")
+STAGED_SCENES_DIR = os.path.join(FILE_DIR, "staged_scenes")
 TEX_DIR           = os.path.join(FILE_DIR, "Tex")
 TEX_IMAGE_DIR     = os.path.join(IMAGE_DIR, "Tex")
 MOBJECT_DIR       = os.path.join(FILE_DIR, "mobjects")
 IMAGE_MOBJECT_DIR = os.path.join(MOBJECT_DIR, "image")
 
 for folder in [IMAGE_DIR, GIF_DIR, MOVIE_DIR, TEX_DIR, 
-               TEX_IMAGE_DIR, MOBJECT_DIR, IMAGE_MOBJECT_DIR]:
+               TEX_IMAGE_DIR, MOBJECT_DIR, IMAGE_MOBJECT_DIR,
+               STAGED_SCENES_DIR]:
     if not os.path.exists(folder):
         os.mkdir(folder)
 
diff --git a/stage_animations.py b/stage_animations.py
new file mode 100644
index 00000000..ed4fbaeb
--- /dev/null
+++ b/stage_animations.py
@@ -0,0 +1,45 @@
+import sys
+import inspect
+import os
+import shutil
+from extract_scene import is_scene, get_module
+from constants import MOVIE_DIR, STAGED_SCENES_DIR
+
+
+def get_sorted_scene_names(module_name):
+    module = get_module(module_name)
+    line_to_scene = {}
+    for name, scene_class in inspect.getmembers(module, is_scene):
+        lines, line_no = inspect.getsourcelines(scene_class)
+        line_to_scene[line_no] = name
+    return [
+        line_to_scene[line_no]
+        for line_no in sorted(line_to_scene.keys())
+    ]
+
+
+
+def stage_animaions(module_name):
+    scene_names = get_sorted_scene_names(module_name)
+    movie_dir = os.path.join(
+        MOVIE_DIR, module_name.replace(".py", "")
+    )
+    files = os.listdir(movie_dir)
+    sorted_files = []
+    for scene in scene_names:
+        for clip in filter(lambda f : f.startswith(scene), files):
+            sorted_files.append(
+                os.path.join(movie_dir, clip)
+            )
+    for f in os.listdir(STAGED_SCENES_DIR):
+        os.remove(os.path.join(STAGED_SCENES_DIR, f))
+    for f in sorted_files:
+        shutil.copy(f, STAGED_SCENES_DIR)
+
+
+
+if __name__ == "__main__":
+    if len(sys.argv) < 2:
+        raise Exception("No module give.")
+    module_name = sys.argv[1]
+    stage_animaions(module_name)
\ No newline at end of file