the quick function of setting and updating mgear component types

release/scripts/mgear/compatible/compatible_comp.py

@@ -0,0 +1,47 @@
+import importlib
+import mgear.pymaya as pm
+from mgear.vendor.Qt import QtCore, QtWidgets
+import mgear.compatible.compatible_comp_ui as rcUI
+
+importlib.reload(rcUI)
+
+
+class RelatedComponents(QtWidgets.QDialog, rcUI.Ui_Dialog):
+    def __init__(self, related_components, parent=None):
+        self.toolName = "RelatedComponents"
+        super(RelatedComponents, self).__init__(parent)
+        self.setupUi(self)
+        self.result_components = None
+        self.update_flag = False
+        self.components_comboBox.addItems(related_components)
+        self.create_connections()
+        self.setWindowTitle("Related Components")
+        self.setAttribute(QtCore.Qt.WA_DeleteOnClose, True)
+
+    def create_connections(self):
+        self.buttonBox.accepted.connect(self.ok)
+        self.update_checkBox.stateChanged.connect(self.on_update_changed)
+
+    def on_update_changed(self, state):
+        self.update_flag = state == QtCore.Qt.Checked
+
+    def ok(self):
+        self.result_components = str(self.components_comboBox.currentText())
+        self.update_flag = self.update_checkBox.isChecked()
+
+    def cancel(self):
+        pm.displayWarning("User cancels update.")
+
+
+def exec_window(related_components, *args):
+    windw = RelatedComponents(related_components)
+    if windw.exec_():
+        return windw
+
+
+if __name__ == "__main__":
+    sample_components = ["Arm", "Leg", "Spine"]
+    w = exec_window(sample_components)
+    if w:
+        print(f"Selected: {w.result_components}")
+        print(f"Update enabled: {w.update_flag}")

release/scripts/mgear/compatible/compatible_comp_dagmenu.py

@@ -0,0 +1,123 @@
+import re
+import importlib
+
+import maya.cmds as cmds
+import mgear.pymaya as pm
+from mgear.shifter import guide_template
+from mgear.compatible import compatible_comp as rc
+from mgear.compatible import (
+    guide_manager_compatible_comp as gmcr,
+)
+
+importlib.reload(rc)
+importlib.reload(gmcr)
+
+SPECIAL_TYPES = ["EPIC", "lite"]
+
+
+def update_component_type_and_update_guide_with_dagmenu(*args):
+    """
+    Updates the component type of selected guide components and optionally refreshes the guide.
+    Provides a user interface for selecting related component types and filtering methods.
+
+    Features:
+    - Validates selected components and their types
+    - Provides filtering options for related components
+    - Updates component types while maintaining guide structure
+    - Optionally refreshes the guide template after update
+
+    """
+    roots = gmcr.get_comp_root()
+
+    if not roots:
+        pm.displayWarning("No components selected.")
+        return
+
+    if not gmcr.are_comp_names_identical(roots):
+        pm.displayWarning("Selected components are not of the same comp type.")
+        return
+
+    root_comp_type = roots[0].getAttr("comp_type")
+    if not isinstance(root_comp_type, str):
+        cmds.error(
+            f"Expected string type for comp_type, but got {type(root_comp_type).__name__}"
+        )
+    elif not root_comp_type:
+        cmds.error("comp_type is empty or invalid")
+    if not re.fullmatch(r"^[a-zA-Z]+(?:_[a-zA-Z0-9_-]+)?$", root_comp_type):
+        cmds.error(
+            f"Invalid format: '{root_comp_type}'\n"
+            "Correct format should be: BASE_TYPE or BASE_TYPE_SUB_TYPE\n"
+            "Examples: EPIC_spine or arm_2jnt"
+        )
+
+    type_parts = root_comp_type.split("_")
+    base_type = type_parts[0]
+    sub_type = type_parts[1] if len(type_parts) > 1 else ""
+    all_components = gmcr.get_component_list()
+
+    buttons = []
+    if base_type in SPECIAL_TYPES:
+        buttons = [
+            f"Only match '{base_type}' type",
+            f"Match all containing '{sub_type}' type",
+            "Cancel",
+        ]
+    else:
+        buttons = [
+            f"Only match '{base_type}' type",
+            f"Match all containing '{base_type}' type",
+            "Cancel",
+        ]
+
+    choice = pm.confirmDialog(
+        title="Select Related Component Scope",
+        message="Please select the scope of related components to include",
+        button=buttons,
+        defaultButton=f"Match all containing '{base_type}' type",
+        cancelButton="Cancel",
+        dismissString="Cancel",
+    )
+    if choice == "Cancel":
+        return
+
+    related_components = []
+    for component in all_components:
+        comp_parts = component.split("_")
+        if choice == f"Only match '{base_type}' type":
+            if base_type in SPECIAL_TYPES:
+                if (
+                    len(comp_parts) > 1
+                    and comp_parts[0] == base_type
+                    and comp_parts[1] == sub_type
+                ):
+                    related_components.append(component)
+            else:
+                if comp_parts[0] == base_type:
+                    related_components.append(component)
+        else:
+            if base_type in SPECIAL_TYPES:
+                if len(comp_parts) > 1 and comp_parts[1] == sub_type:
+                    related_components.append(component)
+                elif sub_type and sub_type in component:
+                    related_components.append(component)
+            else:
+                if base_type in component:
+                    related_components.append(component)
+
+    if not related_components:
+        pm.displayWarning(f"No components related to '{root_comp_type}' found.")
+        return
+
+    custom_window = rc.exec_window(related_components)
+    if not custom_window or not custom_window.result_components:
+        return
+
+    gmcr.set_selected_component_type_is_manager_current_selected_Component(
+        roots, custom_window.result_components
+    )
+
+    if custom_window.update_flag:
+        pm.select(roots)
+        guide_template.updateGuide()
+        pm.select(clear=True)

release/scripts/mgear/compatible/compatible_comp_ui.py

@@ -0,0 +1,58 @@
+from mgear.vendor.Qt import QtCore, QtWidgets
+
+
+class Ui_Dialog(object):
+    def setupUi(self, Dialog):
+        Dialog.setObjectName("Dialog")
+        Dialog.resize(236, 113)
+        self.gridLayout = QtWidgets.QGridLayout(Dialog)
+        self.gridLayout.setObjectName("gridLayout")
+        self.buttonBox = QtWidgets.QDialogButtonBox(Dialog)
+        self.buttonBox.setOrientation(QtCore.Qt.Horizontal)
+        self.buttonBox.setStandardButtons(
+            QtWidgets.QDialogButtonBox.Cancel | QtWidgets.QDialogButtonBox.Ok
+        )
+        self.buttonBox.setObjectName("buttonBox")
+        self.gridLayout.addWidget(self.buttonBox, 3, 0, 1, 1)
+        self.horizontalLayout_2 = QtWidgets.QHBoxLayout()
+        self.horizontalLayout_2.setObjectName("horizontalLayout_2")
+        self.components_label = QtWidgets.QLabel(Dialog)
+        sizePolicy = QtWidgets.QSizePolicy(
+            QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Preferred
+        )
+        sizePolicy.setHorizontalStretch(0)
+        sizePolicy.setVerticalStretch(0)
+        sizePolicy.setHeightForWidth(
+            self.components_label.sizePolicy().hasHeightForWidth()
+        )
+        self.components_label.setSizePolicy(sizePolicy)
+        self.components_label.setObjectName("components_label")
+        self.horizontalLayout_2.addWidget(self.components_label)
+        self.components_comboBox = QtWidgets.QComboBox(Dialog)
+        self.components_comboBox.setObjectName("components_comboBox")
+        self.horizontalLayout_2.addWidget(self.components_comboBox)
+        self.gridLayout.addLayout(self.horizontalLayout_2, 0, 0, 1, 1)
+        spacerItem = QtWidgets.QSpacerItem(
+            20, 40, QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Expanding
+        )
+        self.gridLayout.addItem(spacerItem, 4, 0, 1, 1)
+        self.update_checkBox = QtWidgets.QCheckBox(Dialog)
+        self.update_checkBox.setLayoutDirection(QtCore.Qt.RightToLeft)
+        self.update_checkBox.setObjectName("update_checkBox")
+        self.update_checkBox.setChecked(True)
+        self.gridLayout.addWidget(self.update_checkBox, 1, 0, 1, 1)
+
+        self.retranslateUi(Dialog)
+        self.buttonBox.accepted.connect(Dialog.accept)
+        self.buttonBox.rejected.connect(Dialog.reject)
+
+    def retranslateUi(self, Dialog):
+        Dialog.setWindowTitle(
+            QtWidgets.QApplication.translate("Dialog", "Dialog", None, -1)
+        )
+        self.components_label.setText(
+            QtWidgets.QApplication.translate("Dialog", "Related components：", None, -1)
+        )
+        self.update_checkBox.setText(
+            QtWidgets.QApplication.translate("Dialog", "Update Guide", None, -1)
+        )

release/scripts/mgear/compatible/compatible_comp_ui.ui

@@ -0,0 +1,107 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<ui version="4.0">
+ <class>Dialog</class>
+ <widget class="QDialog" name="Dialog">
+  <property name="geometry">
+   <rect>
+    <x>0</x>
+    <y>0</y>
+    <width>236</width>
+    <height>113</height>
+   </rect>
+  </property>
+  <property name="windowTitle">
+   <string>Dialog</string>
+  </property>
+  <layout class="QGridLayout" name="gridLayout">
+   <item row="3" column="0">
+    <widget class="QDialogButtonBox" name="buttonBox">
+     <property name="orientation">
+      <enum>Qt::Horizontal</enum>
+     </property>
+     <property name="standardButtons">
+      <set>QDialogButtonBox::Cancel|QDialogButtonBox::Ok</set>
+     </property>
+    </widget>
+   </item>
+   <item row="0" column="0">
+    <layout class="QHBoxLayout" name="horizontalLayout_2">
+     <item>
+      <widget class="QLabel" name="components_label">
+       <property name="sizePolicy">
+        <sizepolicy hsizetype="Fixed" vsizetype="Preferred">
+         <horstretch>0</horstretch>
+         <verstretch>0</verstretch>
+        </sizepolicy>
+       </property>
+       <property name="text">
+        <string>Related components：</string>
+       </property>
+      </widget>
+     </item>
+     <item>
+      <widget class="QComboBox" name="components_comboBox"/>
+     </item>
+    </layout>
+   </item>
+   <item row="4" column="0">
+    <spacer name="verticalSpacer">
+     <property name="orientation">
+      <enum>Qt::Vertical</enum>
+     </property>
+     <property name="sizeHint" stdset="0">
+      <size>
+       <width>20</width>
+       <height>40</height>
+      </size>
+     </property>
+    </spacer>
+   </item>
+   <item row="1" column="0">
+    <widget class="QCheckBox" name="update_checkBox">
+     <property name="layoutDirection">
+      <enum>Qt::RightToLeft</enum>
+     </property>
+     <property name="text">
+      <string>Update Guide</string>
+     </property>
+    </widget>
+   </item>
+  </layout>
+ </widget>
+ <resources/>
+ <connections>
+  <connection>
+   <sender>buttonBox</sender>
+   <signal>accepted()</signal>
+   <receiver>Dialog</receiver>
+   <slot>accept()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>248</x>
+     <y>254</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>157</x>
+     <y>274</y>
+    </hint>
+   </hints>
+  </connection>
+  <connection>
+   <sender>buttonBox</sender>
+   <signal>rejected()</signal>
+   <receiver>Dialog</receiver>
+   <slot>reject()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>316</x>
+     <y>260</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>286</x>
+     <y>274</y>
+    </hint>
+   </hints>
+  </connection>
+ </connections>
+</ui>