102 lines
3.2 KiB
Python
102 lines
3.2 KiB
Python
import itertools
|
||
|
||
from maya import cmds
|
||
|
||
from ngSkinTools2.api import PaintTool, target_info
|
||
from ngSkinTools2.api.session import Session
|
||
from ngSkinTools2.decorators import undoable
|
||
|
||
|
||
def as_list(arg):
|
||
return [] if arg is None else arg
|
||
|
||
|
||
customNodeTypes = ['ngst2MeshDisplay', 'ngst2SkinLayerData']
|
||
|
||
|
||
def list_custom_nodes():
|
||
"""
|
||
list all custom nodes in the scene
|
||
"""
|
||
|
||
result = []
|
||
for nodeType in customNodeTypes:
|
||
result.extend(as_list(cmds.ls(type=nodeType)))
|
||
return result
|
||
|
||
|
||
def list_custom_nodes_for_mesh(mesh=None):
|
||
"""
|
||
list custom nodes only related to provided mesh. None means current selection
|
||
"""
|
||
|
||
skin_cluster = target_info.get_related_skin_cluster(mesh)
|
||
if skin_cluster is None:
|
||
return []
|
||
|
||
# delete any ngSkinTools deformers from the history, and find upstream stuff from given skinCluster.
|
||
hist = as_list(cmds.listHistory(skin_cluster, future=True, levels=1))
|
||
return [i for i in hist if cmds.nodeType(i) in customNodeTypes]
|
||
|
||
|
||
def list_custom_nodes_for_meshes(meshes):
|
||
return list(itertools.chain.from_iterable([list_custom_nodes_for_mesh(i) for i in meshes]))
|
||
|
||
|
||
message_scene_noCustomNodes = '场景中不包含任何自定义的ngSkinTools节点。'
|
||
message_selection_noCustomNodes = '选择不包含任何自定义ngskinTools节点。'
|
||
message_scene_warning = (
|
||
'此命令删除所有自定义ngSkinTools节点。蒙皮权重“”将被保留，但所有图层数据都将丢失。您想继续吗？'
|
||
)
|
||
message_selection_warning = (
|
||
'此命令删除要选择的自定义ngSkinTools节点。皮肤重量'
|
||
'将被保留，但所有图层数据都将丢失。您想继续吗？'
|
||
)
|
||
|
||
|
||
@undoable
|
||
def remove_custom_nodes(interactive=False, session=None, meshes=None):
|
||
"""
|
||
Removes custom ngSkinTools2 nodes from the scene or selection.
|
||
|
||
:type meshes: list[str]
|
||
:param meshes: list of node names; if empty, operation will be scene-wide.
|
||
:type session: Session
|
||
:param session: optional; if specified, will fire events for current session about changed status of selected mesh
|
||
:type interactive: bool
|
||
:param interactive: if True, user warnings will be emited
|
||
"""
|
||
from ngSkinTools2.ui import dialogs
|
||
|
||
if meshes is None:
|
||
meshes = []
|
||
|
||
is_selection_mode = len(meshes) > 0
|
||
|
||
custom_nodes = list_custom_nodes() if not is_selection_mode else list_custom_nodes_for_meshes(meshes)
|
||
|
||
if not custom_nodes:
|
||
if interactive:
|
||
dialogs.info(message_selection_noCustomNodes if is_selection_mode else message_scene_noCustomNodes)
|
||
return
|
||
|
||
delete_confirmed = True
|
||
if interactive:
|
||
delete_confirmed = dialogs.yesNo(message_selection_warning if is_selection_mode else message_scene_warning)
|
||
|
||
if delete_confirmed:
|
||
cmds.delete(custom_nodes)
|
||
|
||
if PaintTool.is_painting():
|
||
# make sure that painting is canceled to restore mesh display etc
|
||
cmds.setToolTo("移除") # Move
|
||
|
||
if session is not None:
|
||
session.events.targetChanged.emitIfChanged()
|
||
|
||
|
||
@undoable
|
||
def remove_custom_nodes_from_selection(interactive=False, session=None):
|
||
selection = cmds.ls(sl=True)
|
||
remove_custom_nodes(interactive=interactive, session=session, meshes=as_list(selection))
|