YACWC

util.py

@@ -0,0 +1,112 @@
+
+import glob
+import json
+import textwrap
+import sys
+import os
+from collections import defaultdict
+def json_per_row_to_json(fpath = '/home/thebears/Videos/Winslow/saved/marks'):
+    fid = open(fpath,'r')
+    d = list()
+    for x in fid:
+        d.append(json.loads(x))
+        
+    json.dump(d,open(fpath+'.json','w'),indent=4)
+
+
+def is_ipython():
+    import __main__ as main_ff
+    if hasattr(main_ff,'__file__'):
+        return False
+    else:
+        return True
+    
+def exit_if_not_ipython():
+    import __main__ as main_ff
+    if hasattr(main_ff,'__file__'):
+        print('Running in non-interactive mode, exiting...')
+        sys.exit()
+    else:
+        print('Running in interactive mode, not exiting...')
+
+
+
+def append_json_no_file_read(obj_append, jpath = '/home/thebears/Videos/Winslow/saved/marks.json'):
+    fid = open(jpath,'r+')
+    fid.seek(0,2)
+    position = fid.tell() -2
+    fid.seek(position)
+    fid.write(",\n")
+    fid.write(textwrap.indent(json.dumps(obj_append, indent=4),'    '))
+    fid.write('\n]')
+    fid.close()
+
+
+
+def get_cset_match(input_in):
+    if not os.path.exists(input_in):
+        return {}
+            
+    if os.path.isdir(input_in):
+        return get_cset_match_dir(input_in)
+    else:
+        dir_list = os.path.dirname(input_in)
+        summ = get_cset_match_dir(dir_list)
+        name, ext = os.path.splitext(input_in)
+        tstamp = int(name.replace('_trimmed','').split('_')[-1])
+        return summ[tstamp]
+
+def get_cset_for_file_matching(file_in):
+    name, ext = os.path.splitext(file_in)
+    cset = dict()
+    files =  glob.glob(name+"*")
+    for f in files:
+        [f_name, f_ext ] =f.split(".",1)
+        cset['.'+f_ext] = f
+
+    return cset
+    
+    
+def get_cset_match_dir(rtdir):
+    summary = dict()
+
+    other_files = dict()
+
+    for f in os.walk(rtdir):
+        c_rt = f[0]
+        c_files = f[2]
+
+        for c_file in c_files:
+            name, ext = os.path.splitext(c_file)
+            try:
+                tstamp = int(name.replace('_trimmed','').split('_')[-1])
+                if ext in {'.mp4'}:
+                    summary[tstamp] = dict()
+                    summary[tstamp][ext] = os.path.join(c_rt, c_file)
+                else:
+                    if tstamp not in other_files:
+                        other_files[tstamp] = dict()
+                    other_files[tstamp][ext] = os.path.join(c_rt, c_file)
+            except:
+                pass
+
+    mp4_tstamps = sorted(summary)
+    other_tstamps = sorted(other_files)
+    idx_other = 0
+    
+    merge_map = defaultdict(list)
+    merged = 0
+    for idx_mp4, m_t in enumerate(mp4_tstamps):
+#        for idx_o_test in range(len(other_tstamps)))
+        for idx_o_test, o_t in enumerate(other_tstamps):
+            if abs(m_t - o_t) < 25:
+                merge_map[m_t].append(o_t)
+
+
+
+    for k, v in merge_map.items():
+        for v_join in v:
+            summary[k].update(other_files[v_join])
+
+
+    return summary