Bug fixed, code restructed

Former-commit-id: b6060a23f96851752feb0ea174ff87ed297b35c6

evaluation/alignment.py

@@ -3,87 +3,76 @@ import numpy as np
 sys.path.append('../src')
 from srt_util.srt import SrtScript
 
-
-def procedure(anchor,subsec,S_arr,subidx):
-
-    temp = subsec[subidx - 1]
-    print('------------------------------')    
-    print(anchor)
-    print(temp)
-
+def procedure(anchor, subsec, S_arr, subidx):
     cache_idx = 0
-    while subidx != cache_idx:  # loop until alignment stablized
-        cache_idx = subidx  # reinitialize cache
-        # Inside interval
-        if subidx >= len(subsec): continue
+    while subidx != cache_idx:
+        cache_idx = subidx
+        if subidx >= len(subsec):
+            break
         sub = subsec[subidx]
-        if (anchor.end < sub.start): continue
-        if (anchor.start < sub.start) & (sub.end < anchor.end):
-            S_arr[len(S_arr) - 1] += sub.source_text
+        if anchor.end < sub.start:
+            continue
+        if (anchor.start <= sub.start) and (sub.end <= anchor.end) or anchor.end - sub.start > sub.end - anchor.start:
+            S_arr[-1] += sub.source_text
             subidx += 1
-        elif anchor.end - sub.start > sub.end - anchor.start:
-            S_arr[len(S_arr) - 1] += sub.source_text
-            subidx += 1
-     
-    
-    print(sub)
-    print(S_arr[len(S_arr) - 1])
-    print('------------------------------') 
-    
-    subidx -= 1  # reset subidx to last segment   
-
+    return subidx - 1
 
-def alignment(pred_path,gt_path,threshold = 0.3):
+def alignment(pred_path, gt_path):
     pred = SrtScript.parse_from_srt_file(pred_path).segments
     gt = SrtScript.parse_from_srt_file(gt_path).segments
-    pred_arr = []
-    gt_arr = []
-    duration = 0
-    #count = 0
-    #for ps,gs in zip(pred,gt):
-    #    duration += ps.end + gs.end - ps.start - gs.start
-    #    count += len(ps.source_text) + len(gs.source_text)
-    #density = count / duration  #word density
-    idx_p, idx_t = -1, -1
+    pred_arr, gt_arr = [], []
+    idx_p, idx_t = 0, 0
+
     while idx_p < len(pred) or idx_t < len(gt):
-        idx_p += 1
-        idx_t += 1
-        try:
-            ps = pred[idx_p]
-            gs = gt[idx_t]
-        except IndexError:
-            if idx_t >= len(gt):
-                pred_arr.append(ps.source_text)
-                continue 
-            if idx_p >= len(pred):
-                gs = gt[idx_t]
-                gt_arr.append(gs.source_text)
-                continue
-        #print('init' + str(idx_t) + str(idx_p))
-        #duration
-        ps_dur = ps.end - ps.start
-        gs_dur = gs.end - gs.start
-        #forward/backward
-        if ps_dur <= gs_dur:
+        ps = pred[idx_p] if idx_p < len(pred) else None
+        gs = gt[idx_t] if idx_t < len(gt) else None
+
+        if not ps:
             gt_arr.append(gs.source_text)
-            if gs.end < ps.start:
-                idx_p -= 1  # reset idx if no match
-                continue
+            pred_arr.append('')
+            idx_t += 1
+            continue
+
+        if not gs:
             pred_arr.append(ps.source_text)
+            gt_arr.append('')
             idx_p += 1
-            procedure(gs,pred,pred_arr,idx_p)
-        else:
-            pred_arr.append(ps.source_text)
+            continue
+
+        ps_dur = ps.end - ps.start
+        gs_dur = gs.end - gs.start
+
+        if ps_dur <= gs_dur:
             if ps.end < gs.start:
-                idx_t -= 1  # reset idx if no match
-                continue
-            gt_arr.append(gs.source_text)
-            idx_t += 1
-            procedure(ps,gt,gt_arr,idx_t)
-    #print(pred_arr)
-    #print(gt_arr)
+                pred_arr.append(ps.source_text)
+                gt_arr.append('')
+                idx_t -= 1
+            else:
+                gt_arr.append(gs.source_text)
+                if gs.end >= ps.start:
+                    pred_arr.append(ps.source_text)
+                    idx_p = procedure(gs, pred, pred_arr, idx_p + 1)
+                else:
+                    pred_arr.append('')
+                    idx_p -= 1
+        else:
+            if gs.end < ps.start:
+                gt_arr.append(gs.source_text)
+                pred_arr.append('')
+                idx_p -= 1
+            else:
+                pred_arr.append(ps.source_text)
+                if ps.end >= gs.start:
+                    gt_arr.append(gs.source_text)
+                    idx_t = procedure(ps, gt, gt_arr, idx_t + 1)
+                else:
+                    gt_arr.append('')
+                    idx_t -= 1
 
-    return zip(pred_arr,gt_arr)
+        idx_p += 1
+        idx_t += 1
+    #print(gt_arr)
+    return zip(pred_arr, gt_arr)
 
-        
-alignment('../results/OVB/OVB_en.srt','../results/OVM/OVM_en.srt')
+#  Test Case
+#alignment('../results/...PATH1.../FILE.srt', '../results/PATH2/FILE.srt')