manual fix scenes

data/strange_stars/data.py

@@ -0,0 +1,3 @@
+FINAL_TITLE = "strange_stars"
+DE_VIDEO = "https://www.youtube.com/watch?v=-1FvAEaE0fc"
+EN_VIDEO = "https://www.youtube.com/watch?v=p_8yK2kmxoo"

main.py

@@ -6,14 +6,23 @@ from compare import linear_compare
 TIME_SEP = " --> "
 DECIMAL_SEP = '.'
 HMS_SEP = ':'
+SCENE_BREAK = '-------'
+EMPTY_IDX = '...'
+
 SYNC_STEP = 30
 
 FONT_SIZE = 18
 WORD_BREAK = 60
 
 MAX_ITER = 0
-SHOULD_REMOVE_TEMPS = False
-SHOULD_CUT_VIDEOS = True
+SHOULD_DOWNLOAD = False
+SHOULD_SYNC = False
+SHOULD_TIME = False
+SHOULD_FIX_CUTS = False
+SHOULD_CUT = True
+SHOULD_CONCAT = True
+
+SHOULD_REMOVE_TEMPS = True
 SHOULD_COPY_CODEC = False
 
 # FINAL_TITLE = "plastic"
@@ -62,6 +71,7 @@ def cut_videos(idx, en_start, en_end, de_start, de_end, en_txt, de_txt):
     print("cut_{:02d}: en: {:6.2f} -> {:6.2f}, de: {:6.2f} -> {:6.2f}".format(
         idx, en_start, en_end, de_start, de_end,
     ))
+    run_os_cmd("mkdir -p tmp")
     with open('tmp/concat_{}.txt'.format(idx), 'w+') as f:
         f.write('file de_{}.mp4\n'.format(idx))
         f.write('file en_{}.mp4\n'.format(idx))
@@ -117,6 +127,8 @@ def time_subs(syncs):
     en_subs = parse_subs('subs.en.vtt')
     latest_de_idx = 0
     iterable = en_subs[:MAX_ITER] if MAX_ITER else en_subs
+    with open('fix.txt', 'w+') as f:
+        f.write('')
     for i, en_sub in enumerate(iterable):
         if latest_de_idx >= len(de_subs):
             print("{:6.2f} {:6.2f} {:>6} {:>6} {:>90} {:>90}".format(
@@ -125,34 +137,72 @@ def time_subs(syncs):
             ))
             continue
         else:
-            print("{:6.2f} {:6.2f} {:6.2f} {:6.2f} {:>90} {:>90}".format(
+            print("{:6.2f} {:6.2f} {:6.2f} {:6.2f} {:>90} {:<90}".format(
                 en_sub[0], en_sub[1], de_subs[latest_de_idx][0], de_subs[latest_de_idx][1],
-                en_sub[2], de_subs[latest_de_idx][2],
+                en_sub[2],
+                de_subs[latest_de_idx][2],
             ))
-            de_first_start = de_subs[latest_de_idx][0]
+            with open('fix.txt', 'a+') as f:
+                f.write("{:03} ... {:>90} | {:<90}\n".format(i, en_sub[2], ''))
+            with open('fix.txt', 'a+') as f:
+                f.write("... {:03} {:>90} | {:<90}\n".format(latest_de_idx, '', de_subs[latest_de_idx][2]))
             de_txt = de_subs[latest_de_idx][2]
             latest_de_idx += 1
             en_sub_start, en_sub_end = en_sub[0], en_sub[1]
-            de_last_end = de_subs[latest_de_idx - 1][1]
             while latest_de_idx < len(de_subs):
                 de_next_sub_start, de_next_sub_end, txt = de_subs[latest_de_idx]
                 en_next_sub_start = iterable[i + 1][0]
                 sync_time = get_delta_at_time(en_sub_end, syncs)
                 if de_next_sub_end + sync_time < en_sub_end or de_next_sub_end + sync_time < en_next_sub_start:
+                    with open('fix.txt', 'a+') as f:
+                        f.write("... {:03} {:>90} | {:<90}\n".format(latest_de_idx, '', txt))
                     latest_de_idx += 1
                     de_txt += txt
-                    print("{:>6} {:>6} {:6.2f} {:6.2f} {:>90} {:>90}".format(
+                    print("{:>6} {:>6} {:6.2f} {:6.2f} {:>90} {:<90}".format(
                         '', '', de_next_sub_start, de_next_sub_end,
-                        '', txt,
+                        '',
+                        txt,
                     ))
                 else:
-                    de_last_end = de_subs[latest_de_idx - 1][1]
                     break
-        if SHOULD_CUT_VIDEOS:
-            cut_videos(i, en_sub[0], en_sub[1], de_first_start, de_last_end, en_sub[2], de_txt)
+        with open('fix.txt', 'a+') as f:
+            f.write('{}\n'.format(SCENE_BREAK))
     return len(iterable)
 
 
+def process_scene(scene_idx, scene):
+    if len(scene['en']) == 0 or len(scene['de']) == 0:
+        return
+    en_start, en_end = scene['en'][0][0], scene['en'][-1][1]
+    de_start, de_end = scene['de'][0][0], scene['de'][-1][1]
+    en_txt = ' '.join([entry[2] for entry in scene['en']])
+    de_txt = ' '.join([entry[2] for entry in scene['de']])
+    cut_videos(scene_idx, en_start, en_end, de_start, de_end, en_txt, de_txt)
+
+
+def do_cut_videos():
+    de_subs = parse_subs('subs.de.vtt')
+    en_subs = parse_subs('subs.en.vtt')
+    scene_idx, scene = 0, {'en': [], 'de': []}
+    with open('fix.txt') as f:
+        for line in f.readlines():
+            if len(line) < 7:
+                continue
+            line = line[:7]
+            if line == SCENE_BREAK:
+                process_scene(scene_idx, scene)
+                scene_idx, scene = scene_idx + 1, {'en': [], 'de': []}
+                continue
+            en_idx, de_idx = line.split()
+            if en_idx != EMPTY_IDX:
+                en_sub = en_subs[int(en_idx)]
+                scene['en'].append(en_sub)
+            if de_idx != EMPTY_IDX:
+                de_sub = de_subs[int(de_idx)]
+                scene['de'].append(de_sub)
+    return scene_idx
+
+
 def concat_all_videos(max_iters):
     with open('tmp/concat.txt', 'w+') as f:
         for idx in range(max_iters):
@@ -195,11 +245,20 @@ def sync():
 
 
 def main():
-    download_vids_and_subs()
-    syncs = sync()
-    syncs = [(0, 0)]
-    max_iters = time_subs(syncs)
-    concat_all_videos(max_iters)
+    if SHOULD_DOWNLOAD:
+        download_vids_and_subs()
+    if SHOULD_SYNC:
+        syncs = sync()
+    else:
+        syncs = [(0, 0)]
+    if SHOULD_TIME:
+        max_iters = time_subs(syncs)
+    if SHOULD_FIX_CUTS:
+        exit()
+    if SHOULD_CUT:
+        max_iters = do_cut_videos()
+    if SHOULD_CONCAT:
+        concat_all_videos(max_iters)
 
 
 if __name__ == "__main__":