tictacsync 1.0.2a0__py3-none-any.whl → 1.2.0b0__py3-none-any.whl

tictacsync/newmix.py

@@ -1,19 +1,99 @@
-import os, itertools, argparse, ffmpeg, tempfile
+import os, itertools, argparse, ffmpeg, tempfile, platformdirs
 from pathlib import Path
 from loguru import logger
-import shutil, sys, re, sox
+import shutil, sys, re, sox, configparser
 from pprint import pformat
 from rich import print
 
+try:
+    from . import mamconf
+except:
+    import mamconf
+
+
+LUA = True
 OUT_DIR_DEFAULT = 'SyncedMedia'
 MCCDIR = 'SyncedMulticamClips'
 SEC_DELAY_CHANGED_SND = 10 #sec, SND_DIR changed if diff time is bigger
 DEL_TEMP = True
+CONF_FILE = 'mamsync.cfg'
 
 logger.level("DEBUG", color="<yellow>")
 logger.remove()
-# logger.add(sys.stdout, filter=lambda r: r["function"] == "_change_audio4video")
-# logger.add(sys.stdout, filter=lambda r: r["function"] == "find_SND_vids_pairs_in_dir")
+# logger.add(sys.stdout, filter=lambda r: r["function"] == "main")
+# logger.add(sys.stdout, filter=lambda r: r["function"] == "find_SND_vids_pairs_in_udir")
+
+DAVINCI_RESOLVE_SCRIPT_LOCATION = '/Library/Application Support/Blackmagic Design/DaVinci Resolve/Fusion/Scripts/Utility/'
+DAVINCI_RESOLVE_SCRIPT_TEMPLATE = \
+"""def ClipWithFilePath(file_path):
+    media_pool = app.GetResolve().GetProjectManager().GetCurrentProject().GetMediaPool()
+    queue = [media_pool.GetRootFolder()]
+    while len(queue) > 0:
+        current = queue.pop(0)
+        queue += current.GetSubFolderList()
+        got_it = [cl for cl in current.GetClipList()
+            if cl.GetClipProperty('File Path') == file_path]
+        # print(got_it)
+        if got_it != []:
+            # print('got it')
+            return got_it[0]
+
+def findAndReplace(old_file, new_file):
+    clip_with_old_file = ClipWithFilePath(old_file)
+    if clip_with_old_file == None:
+        print(f'did not find clip with path {old_file}')
+        sys.exit(0)
+    else:
+        clip_with_old_file.ReplaceClip(new_file)
+    cfp, cn = [clip_with_old_file.GetClipProperty(p) for p in ['File Path', 'Clip Name']]
+    if cfp == new_file:
+        print(f'Loaded {cn} with a new sound track')
+    else:
+        print(f'findAndReplace {old_file} -> {new_file} failed')
+
+"""
+# python is deprecated... Lua is the way to go
+
+DAVINCI_RESOLVE_SCRIPT_TEMPLATE_LUA = """local function ClipWithPartialPath(partial_path)
+    local media_pool = app:GetResolve():GetProjectManager():GetCurrentProject():GetMediaPool()
+    local queue = {media_pool:GetRootFolder()}
+    while #queue > 0 do
+        local current = table.remove(queue, 1)
+        local subfolders = current:GetSubFolderList()
+        for _, folder in ipairs(subfolders) do
+            table.insert(queue, folder)
+        end
+        local got_it = {}
+        local clips = current:GetClipList()
+        for _, cl in ipairs(clips) do
+            if string.find(cl:GetClipProperty('File Path'), partial_path) then
+                table.insert(got_it, cl)
+            end
+        end
+        if #got_it > 0 then
+            return got_it[1]
+        end
+    end
+end
+
+local function findAndReplace(old_file, new_file)
+    local clip_with_old_file = ClipWithPartialPath(old_file)
+    if clip_with_old_file == nil then
+        print('did not find clip with path ' .. old_file)
+    else
+        clip_with_old_file:ReplaceClip(new_file)
+        local cfp = clip_with_old_file:GetClipProperty('File Path')
+        local cn = clip_with_old_file:GetClipProperty('Clip Name')
+        if cfp == new_file then
+            print('Loaded ' .. cn .. ' with a new sound track')
+        else
+            print('findAndReplace ' .. old_file .. ' -> ' .. new_file .. ' failed')
+        end
+    end
+end
+
+local changes = {
+"""
 
 video_extensions = \
 """webm mkv flv flv vob ogv  ogg drc gif gifv mng avi mov 
@@ -42,11 +122,12 @@ def is_synced_video(f):
     # logger.debug('ok_ext: %s ok_folders: %s'%(ok_ext, ok_folders))
     return ok_ext and ok_folders
 
-def find_SND_vids_pairs_in_dir(top):
-    # look for matching video name and SND dir name
+def find_SND_vids_pairs_in_udir(top):
+    # look for matching video name and SND in unique directory
+    # in alongside mode (vs MAM mode)
     # eg: IMG04.mp4 and IMG04_SND
     # maybe IMG04v2.mp4 if audio changed before (than it will be IMG04v3.mp4)
-    # returns list of matches
+    # returns list of (SND, vid) matches
     # recursively search from 'top' argument
     vids = []
     SNDs = []
@@ -80,59 +161,64 @@ def find_SND_vids_pairs_in_dir(top):
     logger.debug('matches: %s'%pformat(matches))
     return matches
 
+def find_SND_vids_pairs_in_dual_dir(synced_root, snd_root):
+    # look for matching video name and SND in unique directory
+    # in alongside mode (vs MAM mode)
+    # eg: IMG04.mp4 and directory IMG04_SND
+    # maybe IMG04v2.mp4 if audio changed before (than it will be IMG04v3.mp4)
+    # returns list of (SND, vid) matches
+    # recursively search from 'top' argument
+    vids = []
+    SNDs = []
+    print(f'Will look for new mix in {snd_root} compared to vids in {synced_root}')
+    for (root,dirs,files) in os.walk(synced_root):
+        for f in files:
+            pf = Path(root)/f
+            if pf.suffix[1:].lower() in video_extensions:
+                if not pf.is_symlink():
+                    vids.append(pf)
+    for (root,dirs,files) in os.walk(snd_root):
+        for d in dirs:
+            if d[-4:] == '_SND':
+                SNDs.append(Path(root)/d)
+    logger.debug('vids %s SNDs %s'%(pformat(vids), pformat(SNDs)))
+    matches = []
+    def _names_match(vidname, SND_name):
+        # vidname is a str and has no extension
+        # vidname could have vNN suffix as in DSC_8064v31 so matches DSC_8064
+        if vidname == SND_name: # no suffix presents
+            return True
+        m = re.match(SND_name + r'v(\d+)', vidname)
+        if m != None:
+            logger.debug('its a natch and N= %s'%m.groups()[0])
+        return m != None
+    for pair in list(itertools.product(SNDs, vids)):
+        # print(pair)
+        SND, vid = pair # Paths
+        vidname, ext = vid.name.split('.') # string
+        if _names_match(vidname, SND.name[:-4]):
+            logger.debug('SND %s matches video %s'%(
+                Path('').joinpath(*SND.parts[-2:]),
+                Path('').joinpath(*vid.parts[-3:])))
+            matches.append(pair) # list of Paths
+    logger.debug('matches: %s'%pformat(matches))
+    return matches
+
 def parse_and_check_arguments():
     # parses directories from command arguments
     # check for consistencies and warn user and exits,
-    # if returns, gives:
-    #  proxies_dir, originals_dir, audio_dir, both_audio_vid, scan_only
+    # returns parser.parse_args()
     parser = argparse.ArgumentParser()
-    # parser.add_argument('-v',
-    #                 nargs=1,
-    #                 dest='video_dirs',
-    #                 help='Where proxy clips and/or originals are stored')
-    # parser.add_argument('-a',
-    #                 nargs=1,
-    #                 dest='audio_dir',
-    #                 help='Contains newly changed mix files')
-    parser.add_argument('-b',
+    parser.add_argument('-u',
                     nargs=1,
-                    dest='both_audio_vid',
-                    help='Directory scanned for both audio and video, when tictacsync was used in "alongside mode"')
+                    dest='unique_directory',
+                    help='Directory scanned for both audio and video, when tictacsync is used rather than mamsync.')
     parser.add_argument('--dry',
                     action='store_true',
                     dest='scan_only',
                     help="Just display changed audio, don't merge")
     args = parser.parse_args()
     logger.debug('args %s'%args)
-    # ok cases:
-    # -p -o -a + no -b
-    # -o -a + no -b
-    # args_set = [args.originals_dir != None,
-    #                 args.audio_dir != None,
-    #                 args.both_audio_vid != None,
-    #                 ]
-    # p, o, a, b = args_set
-    # check that argument -b (both_audio_vid) is used alone
-    # if b and any([o, a, p]):
-    #     print("\nDon't specify other argument than -b if both audio and video searched in the same directory.\n") 
-    #     parser.print_help(sys.stderr)
-    #     sys.exit(0)
-    # check that if proxies (-p) are specified, orginals too (-o)
-    # if p and not o:
-    #     print("\nIf proxies directory is specified, so should originals directory.\n") 
-    #     parser.print_help(sys.stderr)
-    #     sys.exit(0)
-    # check that -o and -a are used together
-    # if not b and not (o and a):
-    #     print("\nAt least originals and audio directories must be given (-o and -a) when audio and video are in different dir.\n") 
-    #     parser.print_help(sys.stderr)
-    #     sys.exit(0)
-    # # work in progress (aug 2025), so limit to -b:
-    # if not b :
-    #     print("\nFor now, only -b argument is supported (a directory scanned for both audio and video) .\n") 
-    #     parser.print_help(sys.stderr)
-    #     sys.exit(0)
-    # list of singletons, so flatten. Keep None and False as is
     if args.scan_only:
         print('Sorry, --dry option not implemented yet, bye.')
         sys.exit(0)
@@ -147,7 +233,7 @@ def get_recent_mix(SND_dir, vid):
     logger.debug(f'wav_files {wav_files} in {SND_dir}')
     def is_mix(p):
         re_result = re.match(r'mix([lrLR])*', p.name)
-        logger.debug(f'for {p.name} re_result {re_result}')
+        logger.debug(f'for {p.name} re_result looking for mix {re_result}')
         return re_result is not None
     mix_files = [p for p in wav_files if is_mix(p)]
     if len(mix_files) == 0:
@@ -211,6 +297,9 @@ def _change_audio4video(audio_path: Path, video: Path):
     Replace audio in video (argument)  by the audio contained in
     audio_path (argument) returns nothing
     If name has version number, bump it (DSC_8064v13.MOV -> DSC_8064v14.MOV)
+    Returns string tuple of old and new video files
+    old file in incomplete e.g.: ../DSC_8064  without version and ext
+    new file is complete
 
     """
     vidname, video_ext = video.name.split('.')
@@ -223,14 +312,21 @@ def _change_audio4video(audio_path: Path, video: Path):
         # no suffix, add one
         out_path = video.parent / f'{vidname}v2.{video_ext}'
         out_n = _pathname(out_path)
+        # for Resolve search
+        partial_filename = str(video.parent / vidname)
     else:
         base, number_str = m.groups()
         logger.debug(f'base {base}, number_str {number_str}')
         up_tick =  1 + int(number_str)
         out_path = video.parent / f'{base}v{up_tick}.{video_ext}'
         out_n = _pathname(out_path)
+        # for Resolve search
+        partial_filename = str(video.parent / base)
     print(f'Video [bold]{video}[/bold] \nhas new sound and is now [bold]{out_path}[/bold]')
-    video.unlink()
+    logger.debug(f'partial filname {partial_filename}')
+    old_file = str(video)
+    new_file = str(out_path)
+    # video.unlink() # remove old one
     # building args for debug purpose only:
     ffmpeg_args = (
         ffmpeg
@@ -264,6 +360,7 @@ def _change_audio4video(audio_path: Path, video: Path):
         print(e)
         print(e.stderr.decode('UTF-8'))
         sys.exit(1)
+    return partial_filename, new_file
 
 def _sox_combine(paths) -> Path:
     """
@@ -299,15 +396,49 @@ def _sox_combine(paths) -> Path:
         (merged_duration, nchan))
     return out_file_handle
 
+def generate_script(new_mixes):
+    if LUA:
+        script = DAVINCI_RESOLVE_SCRIPT_TEMPLATE_LUA
+        postlude = ''
+        for a,b in new_mixes:
+            postlude += '{"%s","%s"},\n'%(a,b)
+            # postlude += f"('{a}','{b}'),\n"
+        postlude += '}\n\nfor _, pair in ipairs(changes) do\n'
+        postlude += '    findAndReplace(pair[1], pair[2])\n'
+        postlude += '    os.remove(pair[1])\n'
+        postlude += 'end\n'
+        # postlude += 'os.remove(pair[1])\n'
+    else: # python
+        script = DAVINCI_RESOLVE_SCRIPT_TEMPLATE
+        postlude = '\nchanges = [\n'
+        for a,b in new_mixes:
+            postlude += '("%s","%s"),\n'%(a,b)
+        postlude += ']\n'
+        postlude += '[findAndReplace(a,b) for a, b in changes]\n\n'
+    # foo = '[findAndReplace(a,b) for a, b in changes\n'
+    # print('foo',foo)
+    # print(postlude + foo)
+    return script + postlude
 
 def main():
-    # proxies_dir, originals_dir, audio_dir, both_audio_vid, scan_only = \
+    # proxies_dir, originals_dir, audio_dir, unique_directory, scan_only = \
     #     parse_and_check_arguments()
     args = parse_and_check_arguments()
-    matching_pairs = find_SND_vids_pairs_in_dir(args.both_audio_vid[0])
+    if args.unique_directory != None:
+        # alongside
+        matching_pairs = find_SND_vids_pairs_in_udir(args.unique_directory[0])
+    else:
+        # MAM mode
+        raw_root, synced_root, snd_root, proxies = mamconf.get_proj(False)
+        if raw_root == None:
+            print('Error: without -u option, mamsync should be configured before using newmix')
+            sys.exit(0)
+        matching_pairs = find_SND_vids_pairs_in_dual_dir(synced_root, snd_root)
+        logger.debug(f'matching_pairs {pformat(matching_pairs)}')
+    changes = []
     for SND_dir, vid in matching_pairs:
         new_mix_files = get_recent_mix(SND_dir, vid)
-        # logger.debug('new_mix_files: %s'%str(new_mix_files))
+        logger.debug('new_mix_files: %s'%str(new_mix_files))
         if new_mix_files != ():
             logger.debug(f'new mixes {new_mix_files} in {SND_dir} for {vid.name}')
             if len(new_mix_files) == 2:
@@ -315,9 +446,38 @@ def main():
                 logger.debug('stereo_wav: %s'%new_audio_wav)
             else: # len == 1, mono wav file
                 new_audio_wav = new_mix_files[0]
-            _change_audio4video(new_audio_wav, vid)
-            # print('\nVideo %s has new audio'%vid)
+            old_file, new_file = _change_audio4video(new_audio_wav, vid)
+            changes.append((old_file, new_file))
+    if len(changes) == 0:
+        print('No new mix.')
+        sys.exit(0)
+    logger.debug(f'changes {pformat(changes)}')
+    script = generate_script(changes)
+    fist_vid, _ = changes[0]
+    fist_name = Path(fist_vid).name
+    title = f'New sound for {fist_name}'
+    new_sound_clips = [p for p in Path(DAVINCI_RESOLVE_SCRIPT_LOCATION).iterdir() if 'New sound for' in str(p)]
+    if len(new_sound_clips) != 0:
+        print('There is already some "New sound for" Resolve scripts')
+        for p in new_sound_clips:
+            print(p)
+        while ( res:=input("Is it OK to delete them? (you could also do it manually). Enter y/n: ").lower() ) not in {"y", "n"}: pass
+        if res == 'y':
+             [p.unlink() for p in new_sound_clips]
+    # title = str(fist_vid)
+    if len(changes) > 1:
+        title += ' and others'
+    if LUA:        
+        script_path = Path(DAVINCI_RESOLVE_SCRIPT_LOCATION)/f'{title}.lua'
+        script += f'os.remove("{script_path}")\n'
+    else: 
+        # python version is deprecated...
+        script_path = Path(DAVINCI_RESOLVE_SCRIPT_LOCATION)/f'{title}.py'
+    with open(script_path, 'w') as fh:
+        fh.write(script)
+    print(f'Wrote script {script_path}')
+    # print(script)
 
 
 if __name__ == '__main__':
-    main()
+    main()