RBF Manager: Adding command to update files from weightDriver to mGearWeigthDriver #514

Author: miquelcampos

release/scripts/mgear/core/string.py

@@ -4,6 +4,7 @@
 # GLOBAL
 ##########################################################
 import re
+import os
 
 ##########################################################
 # FUNCTIONS
@@ -196,3 +197,36 @@ def convertRLName_old(name):
         name = re.sub(rePattern, rep, name)
 
     return name
+
+
+# String renamer in files
+def replace_string_in_file(search_string, replace_string, new_filename,
+                           source_path):
+    """Replace all occurrences of a string in an ASCII file and save a copy.
+
+    Args:
+        search_string (str): string to search in the file.
+        replace_string (str): string to replace the search_string with.
+        new_filename (str): Name for the modified file.
+        source_path (str): Path to the original ASCII file.
+
+    Returns:
+        tuple: (str) Path to the modified file,
+               (int) Number of replacements made.
+    """
+    if not os.path.isfile(source_path):
+        raise IOError("Source file does not exist: {}".format(source_path))
+
+    with open(source_path, 'r') as src_file:
+        content = src_file.read()
+
+    count = content.count(search_string)
+    content = content.replace(search_string, replace_string)
+
+    folder = os.path.dirname(source_path)
+    new_path = os.path.join(folder, new_filename)
+
+    with open(new_path, 'w') as new_file:
+        new_file.write(content)
+
+    return new_path, count

release/scripts/mgear/rigbits/rbf_manager2/rbf_manager_ui.py

@@ -316,6 +316,45 @@ def exportNodes(self, allSetups=True):
             return
         rbf_io.exportRBFs(nodesToExport, filePath)
 
+    def update_rbf_solver_reference(self):
+        """Open file dialog and update solver name in selected .rbf or .ma file.
+
+        Replaces all occurrences of 'weightDriver' with 'mGearWeightDriver' and
+        saves the modified file in the same folder with a '_RBF_solver_updated'
+        suffix. Displays how many replacements were made.
+
+        Returns:
+            str: Path to the updated file or None if cancelled.
+        """
+        file_path = mc.fileDialog2(
+            fileFilter="RBF/Maya ASCII (*.rbf *.ma)",
+            dialogStyle=2,
+            fileMode=1
+        )
+
+        if not file_path:
+            return None
+
+        source_path = file_path[0]
+        base, ext = os.path.splitext(source_path)
+        new_name = os.path.basename(base) + "_RBF_solver_updated" + ext
+
+        new_path, count = mString.replace_string_in_file(
+            "weightDriver",
+            "mGearWeightDriver",
+            new_name,
+            source_path
+        )
+
+        mc.confirmDialog(
+            title="RBF Solver Update",
+            message="Updated file saved:\n{}\n\nReplacements made: {}".format(
+                new_path, count),
+            button=["OK"]
+        )
+
+        return new_path
+
 
 class RBFManagerUI(widget.RBFWidget):
     """A manager for creating, mirroring, importing/exporting poses created
@@ -1925,6 +1964,10 @@ def createMenuBar(self, hideMenuBar=False):
             "Delete All Setup",
             partial(self.menuFunc.deleteSetup, allSetup=True),
         )
+        file.addSeparator()
+        file.addAction(
+            "Update file to mGear Weight Driver Variant",
+            self.menuFunc.update_rbf_solver_reference)
         # mirror --------------------------------------------------------------
         mirrorMenu = mainMenuBar.addMenu("Mirror")
         mirrorMenu1 = mirrorMenu.addAction("Mirror Setup", self.mirrorSetup)