Update

2023/scripts/rigging_tools/ngskintools2/ui/actions.py

@@ -0,0 +1,146 @@
+from ngSkinTools2 import signal
+from ngSkinTools2.api import PasteOperation
+from ngSkinTools2.api.pyside import QAction, QtGui, QtWidgets
+from ngSkinTools2.api.python_compatibility import Object
+from ngSkinTools2.api.session import Session
+from ngSkinTools2.operations import import_export_actions, import_v1_actions
+from ngSkinTools2.operations.layers import (
+    ToggleEnabledAction,
+    build_action_initialize_layers,
+)
+from ngSkinTools2.operations.paint import FloodAction, PaintAction
+from ngSkinTools2.operations.website_links import WebsiteLinksActions
+from ngSkinTools2.ui import action
+from ngSkinTools2.ui.updatewindow import build_action_check_for_updates
+
+
+def define_action(parent, label, callback=None, icon=None, shortcut=None, tooltip=None):
+    result = QAction(label, parent)
+    if icon is not None:
+        result.setIcon(QtGui.QIcon(icon))
+    if callback is not None:
+        result.triggered.connect(callback)
+    if shortcut is not None:
+        if not isinstance(shortcut, QtGui.QKeySequence):
+            shortcut = QtGui.QKeySequence(shortcut)
+        result.setShortcut(shortcut)
+    if tooltip is not None:
+        result.setToolTip(tooltip)
+        result.setStatusTip(tooltip)
+    return result
+
+
+def build_action_delete_custom_nodes_for_selection(parent, session):
+    from ngSkinTools2.operations import removeLayerData
+
+    result = define_action(
+        parent,
+        "Delete Custom Nodes For Selection",
+        callback=lambda: removeLayerData.remove_custom_nodes_from_selection(interactive=True, session=session),
+    )
+
+    @signal.on(session.events.nodeSelectionChanged)
+    def update():
+        result.setEnabled(bool(session.state.selection))
+
+    update()
+    return result
+
+
+class Actions(Object):
+    def separator(self, parent, label=""):
+        separator = QAction(parent)
+        separator.setText(label)
+        separator.setSeparator(True)
+        return separator
+
+    def __init__(self, parent, session):
+        """
+        :type session: Session
+        """
+        qt_action = lambda a: action.qt_action(a, session, parent)
+        from ngSkinTools2.operations import layers, removeLayerData, tools
+        from ngSkinTools2.ui.transferDialog import build_transfer_action
+
+        self.initialize = build_action_initialize_layers(session, parent)
+        self.exportFile = import_export_actions.buildAction_export(session, parent)
+        self.importFile = import_export_actions.buildAction_import(session, parent)
+        self.import_v1 = import_v1_actions.build_action_import_v1(session, parent)
+
+        self.addLayer = layers.buildAction_createLayer(session, parent)
+        self.deleteLayer = layers.buildAction_deleteLayer(session, parent)
+        self.toggle_layer_enabled = qt_action(ToggleEnabledAction)
+
+        # self.moveLayerUp = defineCallbackAction(u"Move Layer Up", None, icon=":/moveLayerUp.png")
+        # self.moveLayerDown = defineCallbackAction(u"Move Layer Down", None, icon=":/moveLayerDown.png")
+        self.paint = qt_action(PaintAction)
+        self.flood = qt_action(FloodAction)
+
+        self.toolsAssignFromClosestJoint, self.toolsAssignFromClosestJointOptions = tools.create_action__from_closest_joint(parent, session)
+        (
+            self.toolsAssignFromClosestJointSelectedInfluences,
+            self.toolsAssignFromClosestJointOptionsSelectedInfluences,
+        ) = tools.create_action__from_closest_joint(parent, session)
+        self.toolsAssignFromClosestJointOptionsSelectedInfluences.all_influences.set(False)
+        self.toolsAssignFromClosestJointOptionsSelectedInfluences.create_new_layer.set(False)
+        self.toolsUnifyWeights, self.toolsUnifyWeightsOptions = tools.create_action__unify_weights(parent, session)
+
+        self.toolsDeleteCustomNodes = define_action(
+            parent, "Delete All Custom Nodes", callback=lambda: removeLayerData.remove_custom_nodes(interactive=True, session=session)
+        )
+
+        self.toolsDeleteCustomNodesOnSelection = build_action_delete_custom_nodes_for_selection(parent, session)
+
+        self.transfer = build_transfer_action(session=session, parent=parent)
+
+        # self.setLayerMirrored = defineAction(u"Mirrored", icon=":/polyMirrorGeometry.png")
+        # self.setLayerMirrored.setCheckable(True)
+
+        self.documentation = WebsiteLinksActions(parent=parent)
+
+        self.check_for_updates = build_action_check_for_updates(parent=parent)
+
+        from ngSkinTools2.operations import copy_paste_actions
+
+        self.cut_influences = copy_paste_actions.action_copy_cut(session, parent, True)
+        self.copy_influences = copy_paste_actions.action_copy_cut(session, parent, False)
+        self.paste_weights = copy_paste_actions.action_paste(session, parent, PasteOperation.replace)
+        self.paste_weights_add = copy_paste_actions.action_paste(session, parent, PasteOperation.add)
+        self.paste_weights_sub = copy_paste_actions.action_paste(session, parent, PasteOperation.subtract)
+
+        self.copy_components = tools.create_action__copy_component_weights(parent=parent, session=session)
+        self.paste_component_average = tools.create_action__paste_average_component_weight(parent=parent, session=session)
+
+        self.merge_layer = tools.create_action__merge_layers(parent=parent, session=session)
+        self.duplicate_layer = tools.create_action__duplicate_layer(parent=parent, session=session)
+        self.fill_layer_transparency = tools.create_action__fill_transparency(parent=parent, session=session)
+
+        self.add_influences = tools.create_action__add_influences(parent=parent, session=session)
+        from ngSkinTools2.ui import influencesview
+
+        self.show_used_influences_only = influencesview.build_used_influences_action(parent)
+        self.set_influences_sorted = influencesview.build_set_influences_sorted_action(parent)
+        self.randomize_influences_colors = layers.build_action_randomize_influences_colors(parent=parent, session=session)
+
+        self.select_affected_vertices = tools.create_action__select_affected_vertices(parent=parent, session=session)
+
+    def addLayersActions(self, context):
+        context.addAction(self.addLayer)
+        context.addAction(self.deleteLayer)
+        context.addAction(self.separator(context))
+        context.addAction(self.merge_layer)
+        context.addAction(self.duplicate_layer)
+        context.addAction(self.fill_layer_transparency)
+        context.addAction(self.separator(context))
+        context.addAction(self.toggle_layer_enabled)
+
+    def addInfluencesActions(self, context):
+        context.addAction(self.separator(context, "Actions"))
+        context.addAction(self.toolsAssignFromClosestJointSelectedInfluences)
+        context.addAction(self.select_affected_vertices)
+        context.addAction(self.separator(context, "Clipboard"))
+        context.addAction(self.cut_influences)
+        context.addAction(self.copy_influences)
+        context.addAction(self.paste_weights)
+        context.addAction(self.paste_weights_add)
+        context.addAction(self.paste_weights_sub)