RBF Manager not working correctly with new weightDriver kernel. Closes #514

Author: miquelcampos

release/scripts/mgear/rigbits/menu.py

@@ -29,8 +29,8 @@ def install():
         ("-----", None),
         ("Duplicate symmetrical", str_duplicateSym),
         ("-----", None),
-        ("RBF Manager2", str_rbf_manager2_ui),
-        ("SDK Manager (BETA)", str_SDK_manager_ui),
+        ("RBF Manager 2.1", str_rbf_manager2_ui),
+        ("SDK Manager", str_SDK_manager_ui),
         ("-----", None),
         ("Space Manager", str_space_manager),
         ("-----", None),

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

@@ -87,6 +87,17 @@
 from . import widget
 
 
+# Callback node snippet
+CALLBACK_NODE_CODE = r"""
+
+import mgear.pymaya as pm
+for nod in pm.ls(type="mGearWeightDriver"):
+    print( "RBF mgearWeightDriver node {} force evaluation".format(nod.name()))
+    nod.evaluate.set(True)
+
+"""
+
+
 # =============================================================================
 # general functions
 # =============================================================================
@@ -355,6 +366,45 @@ def update_rbf_solver_reference(self):
 
         return new_path
 
+    def add_RBF_evaluate_callback_node(self, rigTopNode=None):
+
+        if not rigTopNode:
+            sel = pm.selected()
+            if sel:
+                rigTopNode = sel[0]
+            else:
+                pm.displayWarning("Please select the rig top node")
+                return
+
+        if not pm.hasAttr(rigTopNode, "is_rig"):
+            pm.displayWarning("The selected object is not a rig top Node. It is missing the is_rig attribute")
+            return
+
+        def connections(rigTopNode):
+            i = 0
+            while True:
+                try:
+                    pm.connectAttr(
+                        pm.PyNode(cb_node).message,
+                        pm.PyNode(rigTopNode).attr(
+                            "rigScriptNodes[%s]" % str(i)))
+                    break
+                except:
+                    i += 1
+                    if i > 100:
+                        pm.displayWarning("next available reached limit 100")
+                        break
+
+        cb_node = pm.scriptNode(st=1,
+                                beforeScript=CALLBACK_NODE_CODE,
+                                n='rbf_force_eval_cb_node',
+                                stp='python')
+        pm.scriptNode(cb_node, executeBefore=True)
+
+        connections(rigTopNode)
+
+        return cb_node
+
 
 class RBFManagerUI(widget.RBFWidget):
     """A manager for creating, mirroring, importing/exporting poses created
@@ -534,6 +584,10 @@ def addRBFToSetup(self):
             drivenNode_name = drivenNode
 
         # Check if there is an existing rbf node attached
+        print(drivenNode_name)
+        if not drivenNode_name:
+            pm.displayWarning("Not Driven Channels Selected")
+            return
         if mc.objExists(drivenNode_name):
             if existing_rbf_setup(drivenNode_name):
                 msg = "Node is already driven by an RBF Setup."
@@ -868,7 +922,7 @@ def setNodeToField(self, lineEdit, multi=False):
             str: str set to the lineedit
         """
         selected = mc.ls(sl=True)
-        if not multi:
+        if not multi and selected:
             selected = [selected[0]]
         controlNameData = ", ".join(selected)
         lineEdit.setText(controlNameData)
@@ -1864,7 +1918,7 @@ def connectSignals(self):
             partial(self.setNodeToField, self.driverLineEdit)
         )
         self.setDrivenButton.clicked.connect(
-            partial(self.setNodeToField, self.drivenLineEdit, multi=True)
+            partial(self.setNodeToField, self.drivenLineEdit)
         )
         self.allButton.clicked.connect(self.setDriverControlLineEdit)
         self.addDrivenButton.clicked.connect(self.addNewDriven)
@@ -1968,6 +2022,10 @@ def createMenuBar(self, hideMenuBar=False):
         file.addAction(
             "Update file to mGear Weight Driver Variant",
             self.menuFunc.update_rbf_solver_reference)
+        file.addSeparator()
+        file.addAction(
+            "Add RBF Callback Node To Force Evaluation",
+            self.menuFunc.add_RBF_evaluate_callback_node)
         # mirror --------------------------------------------------------------
         mirrorMenu = mainMenuBar.addMenu("Mirror")
         mirrorMenu1 = mirrorMenu.addAction("Mirror Setup", self.mirrorSetup)

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

@@ -9,7 +9,7 @@
 from maya.app.general.mayaMixin import MayaQWidgetDockableMixin
 from mgear.rigbits.six import PY2
 
-__version__ = "2.0.1"
+__version__ = "2.1.0"
 TOOL_NAME = "RBF Manager"
 TOOL_TITLE = "{} v{} | mGear {}".format(
     TOOL_NAME, __version__, mgear.getVersion()