tictacsync 0.1a14__py3-none-any.whl → 1.4.4b0__py3-none-any.whl

tictacsync/mamsync.py

@@ -0,0 +1,343 @@
+
+# I know, the following is ugly, but I need those try's to
+# run the command in my dev setting AND from
+# a deployment set-up... surely I'm setting
+# things wrong [TODO]: find why and clean up this mess
+
+try:
+    from . import yaltc
+    from . import device_scanner
+    from . import timeline
+    from . import multi2polywav
+    from . import mamconf
+    from . import entry
+except:
+    import yaltc
+    import device_scanner
+    import timeline
+    import multi2polywav
+    import mamconf
+    import entry
+
+import argparse, tempfile, configparser, re
+from loguru import logger
+from pathlib import Path
+# import os, sys
+import os, sys, sox, platformdirs, shutil, filecmp
+from rich.progress import track
+# from pprint import pprint
+from rich.console import Console
+# from rich.text import Text
+from rich.table import Table
+from rich import print
+from pprint import pprint, pformat
+import numpy as np
+
+DEL_TEMP = False
+# CONF_FILE = 'mamsync.cfg'
+# LOG_FILE = 'mamdone.txt'
+
+av_file_extensions = \
+"""MOV webm mkv flv flv vob ogv ogg drc gif gifv mng avi MTS M2TS TS mov qt
+wmv yuv rm rmvb viv asf amv mp4 m4p m4v mpg mp2 mpeg mpe mpv mpg mpeg m2v
+m4v svi 3gp 3g2 mxf roq nsv flv f4v f4p f4a f4b 3gp aa aac aax act aiff alac
+amr ape au awb dss dvf flac gsm iklax ivs m4a m4b m4p mmf mp3 mpc msv nmf
+ogg oga mogg opus ra rm raw rf64 sln tta voc vox wav wma wv webm 8svx cda""".split()
+
+logger.remove()
+logger.level("DEBUG", color="<yellow>")
+# logger.add(sys.stdout, level="DEBUG")
+# logger.add(sys.stdout, filter=lambda r: r["function"] == "__init__" and r["module"] == "yaltc")
+# logger.add(sys.stdout, filter=lambda r: r["function"] == "_fit_length")
+# logger.add(sys.stdout, filter=lambda r: r["function"] == "_write_ISOs")
+
+def copy_to_syncedroot(raw_root, synced_root):
+    # args are str
+    # copy dirs and non AV files
+    logger.debug(f'raw_root {raw_root}')
+    logger.debug(f'synced_root {synced_root}')
+    for raw_path in Path(raw_root).rglob('*'):
+        ext = raw_path.suffix[1:]
+        is_DS_Store = raw_path.name ==  '.DS_Store'# mac os
+        if ext not in av_file_extensions and not is_DS_Store:
+            logger.debug(f'raw_path: {raw_path}')
+            # dont copy WAVs either, they will be in ISOs 
+            rel = raw_path.relative_to(raw_root)
+            logger.debug(f'relative path {rel}')
+            synced_path = Path(synced_root)/Path(raw_root).name/rel
+            logger.debug(f'synced_path: {synced_path}')
+            if raw_path.is_dir():
+                    synced_path.mkdir(parents=True, exist_ok=True)
+                    continue
+            # if here, it's a file
+            if not synced_path.exists():
+                print(f'will mirror non AV file {synced_path}')
+                logger.debug(f'will mirror non AV file at {synced_path}')
+                shutil.copy2(raw_path, synced_path)
+                continue
+            # file exists, check if same
+            same = filecmp.cmp(raw_path, synced_path, shallow=False)
+            logger.debug(f'copy exists of:\n{raw_path}\n{synced_path}')
+            if not same:
+                print(f'file changed, copying again\n{raw_path}')
+                shutil.copy2(raw_path, synced_path)
+            else:
+                logger.debug('same content, next')
+                continue # next raw_path in loop
+
+# logger.add(sys.stdout, filter=lambda r: r["function"] == "clean_synced")
+# def clean_synced(raw_root, synced_root):
+#     # removes <>v<capitalLetter>.mov if any
+#     # returns a list of deleted pathlib.Path
+#     project = Path(raw_root).name
+#     logger.debug(f'project {project}')
+#     synced_proj = Path(synced_root)/project
+#     if not synced_proj.exists():
+#         synced_proj.mkdir()
+#     deleted = []
+#     for raw_path in Path(synced_proj).rglob('*'):
+#         m = re.match(r'.*(v[A-Z])\..+', raw_path.name)
+#         if m != None:
+#             logger.debug(f'cleaning {raw_path}')
+#             raw_path.unlink()
+#             deleted.append(raw_path)
+#     return deleted
+
+def copy_raw_root_tree_to_sndroot(raw_root, snd_root):
+    # args are str
+    # copy only tree structure, no files
+    for raw_path in Path(raw_root).rglob('*'):
+        synced_path = Path(snd_root)/str(raw_path)[1:] # cant join abs. paths
+        if raw_path.is_dir():
+            synced_path.mkdir(parents=True, exist_ok=True)
+
+def new_parser():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--resync',
+                    action='store_true',
+                    dest='resync',
+                    help='Resync previously done clips.')
+    parser.add_argument(
+            "sub_dir",
+            type=str,
+            nargs='?',
+            help="Sub directory to scan, should under RAWROOT."
+            )
+    parser.add_argument('--terse',
+                    action='store_true',
+                    dest='terse',
+                    help='Terse output')
+    # parser.add_argument('--isos', # default True in mamsync
+    #                 action='store_true',
+    #                 dest='write_ISOs',
+    #                 help='Cut ISO sound files')
+    parser.add_argument('-t','--timelineoffset',
+                    nargs=1,
+                    default=['00:00:00:00'],
+                    dest='timelineoffset',
+                    help='When processing multicam, where to place clips on NLE timeline (HH:MM:SS:FF)')
+    return parser
+
+def clear_log():
+    # clear the file logging clips already synced
+    data_dir = platformdirs.user_data_dir('mamsync', 'plutz', ensure_exists=True)
+    log_file = Path(data_dir)/mamconf.LOG_FILE
+    print('Clearing log file of synced clips: "%s"'%log_file)
+    with open(log_file, 'w') as fh:
+        fh.write('done:\n')
+
+def main():
+    parser = new_parser()
+    args = parser.parse_args()
+    logger.debug(f'arguments from argparse {args}')
+    roots_strings = mamconf.get_proj(False)
+    roots_pathlibPaths = [Path(s) for s in mamconf.get_proj(False)]
+    logger.debug(f'roots_strings from mamconf.get_proj {roots_strings}')
+    logger.debug(f'roots_pathlibPaths from mamconf.get_proj {roots_pathlibPaths}')
+    # check all have values, except for PROXIES, the last one
+    if any([r == '' for r in roots_strings][:-1]):
+        print("Can't sync if some folders are not set:")
+        mamconf.print_out_conf(*mamconf.get_proj())
+        print('Bye.')
+        sys.exit(0)
+    # because optional PROXIES folder '' yields a '.' path, exclude it
+    for r in [rp for rp in roots_pathlibPaths if rp != Path('.')]:
+        if not r.is_absolute():
+            print(f'\rError: folder {r} must be an absolute path. Bye')
+            sys.exit(0)
+        if not r.exists():
+            print(f'\rError: folder {r} does not exist. Bye')
+            sys.exit(0)
+        if not r.is_dir():
+            print(f'\rError: path {r} is not a folder. Bye')
+            sys.exit(0)
+    raw_root, synced_root, snd_root, _ = roots_pathlibPaths
+    if args.sub_dir != None:
+        top_dir = args.sub_dir
+        logger.debug(f'sub _dir: {args.sub_dir}')
+        if not Path(top_dir).exists():
+            print(f"\rError: folder {top_dir} doesn't exist, bye.")
+            sys.exit(0)
+    else:
+        top_dir = raw_root
+    if args.resync:
+        clear_log()
+        # deleted = clean_synced(raw_root, synced_root)
+        # logger.debug(f'deleted older clip versions: {deleted}')
+    # go, mamsync!
+    copy_to_syncedroot(raw_root, synced_root)
+    # copy_raw_root_tree_to_sndroot(raw_root, snd_root) # why?
+    multi2polywav.poly_all(top_dir)
+    scanner = device_scanner.Scanner(top_dir, stay_silent=args.terse)
+    scanner.scan_media_and_build_devices_UID(synced_root=synced_root)
+    for m in scanner.found_media_files:
+        if m.device.tracks:
+            if not all([lv == None for lv in m.device.tracks.lag_values]):
+                logger.debug('%s has lag_values %s'%(
+                        m.path, m.device.tracks.lag_values))
+                # any lag for a channel is specified by user in tracks.txt
+                entry.process_lag_adjustement(m)
+    audio_REC_only = all([m.device.dev_type == 'REC' for m
+                in scanner.found_media_files])
+    if not args.terse:
+        if scanner.input_structure == 'ordered':
+            print('\nDetected structured folders')
+            # if scanner.top_dir_has_multicam:
+            #     print(', multicam')
+            # else:
+            #     print()
+        else:
+            print('\nDetected loose structure')
+            if scanner.CAM_numbers() > 1:
+                print('\nNote: different CAMs are present, will sync audio for each of them but if you want to set their')
+                print('respective timecode for NLE timeline alignement you should regroup clips by CAM under their own DIR.')
+        print('\nFound [gold1]%i[/gold1] media files '%(
+            len(scanner.found_media_files)), end='')
+        print('from [gold1]%i[/gold1] devices:\n'%(
+            scanner.get_devices_number()))
+        all_devices = scanner.get_devices()
+        for dev in all_devices:
+            dt = 'Camera' if dev.dev_type == 'CAM' else 'Recorder'
+            print('%s [gold1]%s[/gold1] with files:'%(dt, dev.name), end = ' ')
+            medias = scanner.get_media_for_device(dev)
+            for m in medias[:-1]: # last printed out of loop
+                print('[gold1]%s[/gold1]'%m.path.name, end=', ')
+            print('[gold1]%s[/gold1]'%medias[-1].path.name)
+            a_media = medias[0]
+        # check if all audio recorders have same sampling freq
+        freqs = [dev.sampling_freq for dev in all_devices if dev.dev_type == 'REC']
+        same = np.isclose(np.std(freqs),0)
+        logger.debug('sampling freqs from audio recorders %s, same:%s'%(freqs, same))
+        if not same:
+            print('some audio recorders have different sampling frequencies:')
+            print(freqs)
+            print('resulting in undefined results: quitting...')
+            quit()
+        print()
+    recordings = [yaltc.Recording(m, do_plots=False) for m 
+                        in scanner.found_media_files]
+    recordings_with_time =  [
+        rec 
+        for rec in recordings
+        if rec.get_start_time()
+        ]
+    [r.load_track_info() for r in recordings_with_time if r.is_audio()]
+    if not args.terse:    
+        table = Table(title="tictacsync results")
+        table.add_column("Recording\n", justify="center", style='gold1')
+        table.add_column("TTC chan\n (1st=#0)", justify="center", style='gold1')
+        # table.add_column("Device\n", justify="center", style='gold1')
+        table.add_column("UTC times\nstart:end", justify="center", style='gold1')
+        table.add_column("Clock drift\n(ppm)", justify="right", style='gold1')
+        # table.add_column("SN ratio\n(dB)", justify="center", style='gold1')
+        table.add_column("Date\n", justify="center", style='gold1')
+        rec_WO_time = [
+            rec.AVpath.name
+            for rec in recordings
+            if rec not in recordings_with_time]
+        if rec_WO_time:
+            print('No time found for: ',end='')
+            [print(rec, end=' ') for rec in rec_WO_time]
+            print('\n')
+        for r in recordings_with_time:
+            date = r.get_start_time().strftime("%y-%m-%d")
+            start_HHMMSS = r.get_start_time().strftime("%Hh%Mm%Ss")
+            end_MMSS = r.get_end_time().strftime("%Mm%Ss")
+            times_range = start_HHMMSS + ':' + end_MMSS
+            table.add_row(
+                str(r.AVpath.name),
+                str(r.TicTacCode_channel),
+                # r.device,
+                times_range,
+                # '%.6f'%(r.true_samplerate/1e3),
+                '%2i'%(r.get_samplerate_drift()),
+                # '%.0f'%r.decoder.SN_ratio,
+                date
+                )
+        console = Console()
+        console.print(table)
+        print()
+    n_devices = scanner.get_devices_number()
+    if len(recordings_with_time) < 2:
+        if not args.terse:
+            print('\nNothing to sync, exiting.\n')
+        sys.exit(1)
+    matcher = timeline.Matcher(recordings_with_time)
+    matcher.scan_audio_for_each_videoclip()
+    if not matcher.mergers:
+        if not args.terse:
+            print('\nNothing to sync, bye.\n')
+        sys.exit(1)
+    if scanner.input_structure != 'ordered':
+        print('Warning, can\'t run mamsync without structured folders: [gold1]--isos[/gold1] option ignored.\n')
+    asked_ISOs = True # par defaut
+    dont_write_cam_folder = False # write them
+    for merger in track(matcher.mergers, description="Merging..."):
+        merger._build_audio_and_write_video(top_dir,
+        dont_write_cam_folder,
+        asked_ISOs,
+        synced_root = synced_root,
+        snd_root = snd_root,
+        raw_root = raw_root)
+    if not args.terse:
+        print("\n")
+    # find out where files were written
+    # a_merger = matcher.mergers[0]
+    # log file
+    p = Path(platformdirs.user_data_dir('mamsync', 'plutz'))/mamconf.LOG_FILE
+    log_filehandle = open(p, 'a')
+    for merger in matcher.mergers:
+        print('[gold1]%s[/gold1]'%merger.videoclip.AVpath.name, end='')
+        for audio in merger.get_matched_audio_recs():
+            print(' + [gold1]%s[/gold1]'%audio.AVpath.name, end='')
+        new_file = merger.videoclip.final_synced_file.parts
+        final_p = merger.videoclip.final_synced_file
+        nameAnd2Parents = Path('').joinpath(*final_p.parts[-2:])
+        print(' became [gold1]%s[/gold1]'%nameAnd2Parents)
+        # add full path to log file
+        log_filehandle.write(f'{merger.videoclip.AVpath}\n')
+        # matcher._build_otio_tracks_for_cam()
+    log_filehandle.close()
+    matcher.set_up_clusters() # multicam
+    matcher.shrink_gaps_between_takes(args.timelineoffset)
+    logger.debug('matcher.multicam_clips_clusters %s'%
+                        pformat(matcher.multicam_clips_clusters))
+    # clusters is list of {'end': t1, 'start': t2, 'vids': [r1,r3]}
+    # really_clusters is True if one of them has len() > 1
+    really_clusters = any([len(cl['vids']) > 1 for cl
+                        in matcher.multicam_clips_clusters])
+    if really_clusters:
+        if scanner.input_structure == 'loose':
+            print('\nThere are synced multicam clips but without structured folders')
+            print('they were not grouped together under the same folder.')
+        else:
+            matcher.move_multicam_to_dir(raw_root=raw_root, synced_root=synced_root)
+    else:
+        logger.debug('not really a multicam cluster, nothing to move')
+    sys.exit(0)
+    
+if __name__ == '__main__':
+    main()
+
+
+

tictacsync/multi2polywav.py

@@ -1,8 +1,9 @@
-import argparse, wave, subprocess
+import argparse, wave, subprocess, sys
 from loguru import logger
 from pathlib import Path
 from itertools import groupby
 import sox, tempfile, os, ffmpeg
+from rich import print
 
 
 """
@@ -30,11 +31,11 @@ def print_grby(grby):
 def wav_recursive_scan(top_directory):
     files_lower_case = Path(top_directory).rglob('*.wav') 
     files_upper_case = Path(top_directory).rglob('*.WAV')
-    files = list(files_lower_case) + list(files_upper_case)
+    files = set(list(files_lower_case) + list(files_upper_case))
     paths = [
         p
         for p in files
-        # if BACKUPFOLDER not in p.parts
+        if 'SyncedMedia' not in p.parts
     ]
     return paths
 
@@ -55,34 +56,35 @@ def nframes(path):
         return duration_ts
     return n_frames
 
-
 def build_poly_name(multifiles):
     """
     Returns string of polywav filename, constructed from similitudes between
     multifile names. Ex:
-    /full/path/4CH002I.wav and /full/path/4CH002M.wav returns 4CH002X.wav
+    4CH002I.wav and 4CH002M.wav returns 4CH002X.wav
     """
     s1 = str(multifiles[0].stem)
     s2 = str(multifiles[1].stem)
     if len(s1) != len(s2):
-        print('Can not build compound name with %s.wav and %s.wav'%(s1,s2))
+        print('\nCan not build compound name with %s.wav and %s.wav'%(s1,s2))
         print('names lengths differ.')
         print('In folder "%s", quitting.'%multifiles[0].parent)
-        quit()
+        sys.exit(1)
     pairs = list(zip(s1, s2))
     not_same = [a for a, b in pairs if a != b ]
     if len(not_same) > 2:
-        print('Can not build compound name with %s.wav and %s.wav'%(s1,s2))
+        print('\nCan not build compound name with %s.wav and %s.wav'%(s1,s2))
         print('names differ by more than two characters.')
         print('In folder "%s", quitting.'%multifiles[0].parent)
-        quit()
+        sys.exit(1)
     compound = [a if a == b else 'X' for a, b in pairs ]
     return ''.join(compound) + '.wav'
 
 def jump_metadata(from_file, to_file):
     tempfile_for_metadata = tempfile.NamedTemporaryFile(suffix='.wav', delete=True)
     tempfile_for_metadata = tempfile_for_metadata.name
-    process_list = ['ffmpeg', '-loglevel', 'quiet', '-nostats', '-hide_banner', '-i', from_file, '-i', to_file, '-map_metadata', '0', '-c', 'copy', tempfile_for_metadata]
+    # ffmpeg -i 32ch-44100-bwf.wav -i onechan.wav -map 1 -map_metadata 0 -c copy outmeta.wav
+    process_list = ['ffmpeg', '-loglevel', 'quiet', '-nostats', '-hide_banner', '-i', from_file, '-i', to_file, '-map', '1',
+                                     '-map_metadata', '0', '-c', 'copy', tempfile_for_metadata]
     # ss = shlex.split("ffmpeg  -i %s -i %s -map_metadata 0 -c copy %s"%(from_file, to_file, tempfile_for_metadata))
     # print(ss)
     # logger.debug('process %s'%process_list)
@@ -95,8 +97,9 @@ def build_poly(multifiles):
     # multifiles is list of Path
     # change extensions to mfw (multifile wav)
     dir_multi = multifiles[0].parent
-    poly_name = build_poly_name(multifiles) # base only
-    poly_name = str(dir_multi/Path(poly_name))
+    multifiles.reverse()
+    poly_name_b = build_poly_name(multifiles) # base only
+    poly_name = str(dir_multi/Path(poly_name_b))
     filenames = [str(p) for p in multifiles]
     logger.debug('sox.build args: %s %s'%(
         filenames,
@@ -110,10 +113,14 @@ def build_poly(multifiles):
     logger.debug('sox.build status: %s'%status)
     if status != True:
         print('sox did not merge files')
-        quit()
+        sys.exit(1)
     jump_metadata(filenames[0], poly_name)
+    print('\nMerging multifiles recordings:', end = ' ')
+    [print('[gold1]%s[/gold1]'%p.name, end=' + ') for p in multifiles[:-1]]
+    print('[gold1]%s[/gold1]'%multifiles[-1].name, end=' = ')
+    print('[gold1]%s[/gold1]'%poly_name_b, end=' ')
+    print('(backups will have .MFW as extension)')
     [p.rename(str(p.parent/p.stem) + '.mfw') for p in multifiles]
-    # [os.remove(f) for f in filenames] # not anymore, non destructive..
 
 def getch():
     """Read single character from standard input without echo."""