Move resolvePath in the new file fileUtils.py of meshroom core

Author: demoulinv

meshroom/core/fileUtils.py

@@ -0,0 +1,59 @@
+import os
+import re
+
+def getFileElements(inputFilePath: str):
+
+    filename = os.path.basename(inputFilePath)
+    pattern = r"(?P<FILESTEM>.*?)(?P<FRAME_ID>[-._]\d+)?(?P<EXTENSION>\.\w{3,4})"
+    match = re.search(pattern, filename)
+    frameId = match.group("FRAME_ID")
+    
+    fileElements = {}
+    if match:
+        fileElements = {
+            "<PATH>": inputFilePath,
+            "<FILENAME>": filename,
+            "<FILESTEM>": match.group("FILESTEM"),
+            "<FILESTEM_PREFIX>": match.group("FILESTEM"),
+            "<EXTENSION>": match.group("EXTENSION"),
+        }
+    if frameId is not None:
+        fileElements["<FRAMEID>"] = frameId
+        fileElements["<FILESTEM>"] += frameId
+
+    return fileElements
+
+
+def getViewElements(vp):
+
+    vpPath = vp.childAttribute("path").value
+
+    viewElements = getFileElements(vpPath)
+
+    viewElements["<VIEW_ID>"] = str(vp.childAttribute("viewId").value)
+    viewElements["<INTRINSIC_ID>"] = str(vp.childAttribute("intrinsicId").value)
+    viewElements["<POSE_ID>"] = str(vp.childAttribute("poseId").value)
+
+    return viewElements
+
+
+def replacePatterns(input, pattern, replacements):
+    # Use all substrings of "input" matching the regex "pattern" as a key to substitute themselves by their value in the dictionnary "replacements".
+    # If "replacements" does not contain the key, the key is removed from "input" to build the resolved string.
+    def replaceMatch(match):
+        key = match.group()
+        return replacements.get(key, "")
+    return re.sub(pattern, replaceMatch, input)
+
+
+def resolvePath(input, outputTemplate: str) -> str:
+
+    if isinstance(input, str):
+        replacements = getFileElements(input)
+    else:
+        replacements = getViewElements(input)
+
+    resolved = replacePatterns(outputTemplate, r"<\w*>", replacements)
+
+    return resolved
+

meshroom/ui/components/filepath.py

@@ -111,34 +111,17 @@ def fileSizeMB(self, path):
     @Slot(str, QObject, result=str)
     def resolve(self, path, vp):
         # Resolve dynamic path that depends on viewpoint
+        from meshroom.core import fileUtils
 
-        replacements = {}
         if vp == None:
             replacements = FilepathHelper.getFilenamesFromFolder(FilepathHelper, FilepathHelper.dirname(FilepathHelper, path), FilepathHelper.extension(FilepathHelper, path))
             resolved = [path for i in range(len(replacements))]
             for key in replacements:
                 for i in range(len(resolved)):
                     resolved[i] = resolved[i].replace("<FRAMEID>", replacements[i])
             return resolved
-        else:
 
-            vpPath = vp.childAttribute("path").value
-            filename = FilepathHelper.basename(FilepathHelper, vpPath)
-            replacements = {
-                "<VIEW_ID>": str(vp.childAttribute("viewId").value),
-                "<INTRINSIC_ID>": str(vp.childAttribute("intrinsicId").value),
-                "<POSE_ID>": str(vp.childAttribute("poseId").value),
-                "<PATH>": vpPath,
-                "<FILENAME>": filename,
-                "<FILESTEM>": FilepathHelper.removeExtension(FilepathHelper, filename),
-                "<EXTENSION>": FilepathHelper.extension(FilepathHelper, filename),
-            }
-
-        resolved = path
-        for key in replacements:
-            resolved = resolved.replace(key, replacements[key])
-
-        return resolved
+        return fileUtils.resolvePath(vp, path)
 
     @Slot(str, result="QVariantList")
     @Slot(str, str, result="QVariantList")