Maya插件教程 — 5.3 快速选择工具案例

这里介绍一个制作一个属于自己的快速选择工具。我们可以事先编辑快速选择的规则，然后下次相同规则的模型进来，能够快速批量选择需要的对象，比如控制器等。我们在为绑定角色制作动画的时候，一个人物模型存在大量的控制器。动画师有时候想一次性选中全部手指的控制器或者整个手臂的控制器。这个时候有个快速批量选择的工具，那动画师的工作效率就会提高很多。

这里是我这制作的界面，选择模式是一个下拉列表，不同的模式里面可以添加不同的选择组。因为可能不同的绑定规则，控制器命名方式是不一样的。所以我们添加一个模式的选择。

模式我们可以添加，每一个模式我们会把描述信息写入一个文件，这样就实现了我们的保存，下次打开的时候能够直接把自己添加的模式和选择组自动导入进来。同时我们也支持把现有的模式进行修改，然后把修改的内容会写进入配置文件中。这里的配置文件使用的是json,因为能Python和Json能够很方便的快速互转，这是我们的首选。

5.3.1 Json文件结构

{
  "CurrentMode": "Adv", 
  "ModeData": [
    {
      "list": [
        {
          "button": "\u5de6\u624b\u81c2", 
          "rule": [
            "FKShoulder_L", 
            "FKElbow_L", 
            "FKWrist_L"
          ]
        }, 
        {
          "button": "\u53f3\u624b\u63a7\u5236\u5668", 
          "rule": [
            "FKPinkyFinger1_R", 
          ]
        }
      ], 
      "name": "Adv"
    }
  ]
}

上面是我的存储文件结构，这个根据自己的需求和习惯自己设计即可。这里我来解释一下我为什么这样设计。

CurrentMode:这是当前选中的模式。因为我们下次打开此界面的时候，能够优先选中我们上次关闭时的选择。
ModeData:这是一个数组，每个数组对象是一个模式信息对象。
list：存放我们选择组的按钮信息，里面包括按钮的名字，按钮要选择的对象名字。
button:按钮的名字。
rule:需要选择的对象名字。
name:模式的名字。

5.3.2 解析配置文件

现在我们解析上面的配置文件，首先读取文件数据，把读取的字符串通过json库解析成Python对象。然后根据里面的信息初始化我们的模式Combox控件和按钮组。

    def initMode(self):
        self.ui.modeModifyBtn.setEnabled(False)
        curPath = self.GetCurrentDir()
        ruleFile = curPath + '/rule.json'
        if not os.path.exists(ruleFile):
            return
        with open(ruleFile,'r') as inFile:
            data = json.load(inFile)
            self.Data = data
        self.curtName = self.Data['CurrentMode']
        self.initModeComBox()
        self.refreshButtons()

    def initModeComBox(self):
        if self.Data == None:
            return
        self.ui.modeCombox.clear()
        for k in self.Data['ModeData']:
            name = k['name']
            if self.ui.modeCombox.findText(name) < 0:
                self.ui.modeCombox.addItem(name)
        self.ui.modeCombox.setCurrentText(self.curtName)
        if self.curtName == '':
            self.curtName = self.ui.modeCombox.currentText()
        if self.ui.modeCombox.currentText() != '':
            self.ui.modeModifyBtn.setEnabled(True)
        else:
            self.ui.modeModifyBtn.setEnabled(False)

def clearButtons(self):
        for k in self.buttons:
            self.ui.gridLayout.removeWidget(k)
            k.setParent(None)
        self.buttons = []
    
def refreshButtons(self):
        self.clearButtons()
        if self.Data == None:
            return
        item = self.findItem(self.Data,self.curtName)
        if not item:
            return
        row = 0
        col = 0
        mo = 2

        for n in item['list']:
            button = QuickSelectRuleButton.QuickSelectRuleButton()
            button.setRule(n['button'],n['rule'])
            self.ui.gridLayout.addWidget(button,row,col % mo)
            self.buttons.append(button)
            col += 1
            if col == mo:
                row += 1
                col = 0

这上面的代码都很简单，就是读取配置数据，然后动态创建对应的控件和还原控件中的数据。这里我们实现了一个自己的按钮QuickSelectRuleButton，这样我们方便响应自己的响应函数。

5.3.3 自定义控件

有时候为了方便，我们会实现一些自定义控件。比如我们这里想扩展一下我们的QPushButton，这样我们就能自定一些属性进去，然后方便我们操作。

# QuickSelectRuleButton.py
from PySide2 import QtWidgets, QtCore, QtGui
from PySide2.QtCore import Qt
import pymel.core as pm

class QuickSelectRuleButton(QtWidgets.QPushButton):
    def __init__(self):
        super(QuickSelectRuleButton, self).__init__()
        self.rule = []
        self.clicked.connect(self.onClick)

    def setRule(self,name,rule):
        self.rule = rule
        self.setText(name)

    def checkMatch(self,nodeName):
        for k in self.rule:
            if nodeName.endswith(k):
                return True
        return False

    def onClick(self):
        nodes = pm.ls()
        for k in nodes:
            if self.checkMatch(k.name()):
                pm.select(k.name(),add=1)

我们自定义的按钮，我们可以把选择规则设置进来，点击的响应就自动完成了，没必要我们使用QPushButton原生的按钮，每个按钮再单独设置点击响应。这里我们使用的是继承QPushButton来实现功能的扩展。

5.3.4 模式创建

上图是我们创建模式的一个窗口面板截图。我们有设置模板的名字，然后+号按钮是添加一个按钮，我们知道一个按钮代表一到多个的选择，所以我们需要创建一个空间，里面能设置按钮的名字和按钮选择的对象名字。

    def onAdd(self):
        itemWidget = QuickSelectRuleItem.QuickSelectRuleItem()
        itemWidget.setDelCallBack(self.delItemCallBack)
        item = QtWidgets.QListWidgetItem(self.ui.modeListWidget)
        item.setSizeHint(QtCore.QSize(0, 123));
        self.ui.modeListWidget.addItem(item)
        self.ui.modeListWidget.setItemWidget(item,itemWidget)
        self.itemList[itemWidget] = item

这里也很简单，我们需要知道怎么提取我们当前选中的对象的名字：

    def onSel(self):
        nodes = pm.selected()
        for n in nodes:
            name = n.name()
            ix = name.rfind(':')
            if ix > -1:
                name = name[ix+1:]
            self.ui.ruleNodesText.append(name)

我们使用rfind找到最后一个冒号，为了是把我们名字上的命名空间去掉。我们在选择的时候是不管命名空间的。

5.3.4 写配置文件

当我们点击确定按钮的时候，我们要把当前的内容保存出去，也就是把我们视图数据存储出去。上面已经介绍了我们配置文件的结构，我们构建出这样的对象即可。

def onOk(self):
        curPath = self.GetCurrentDir()
        ruleFile = curPath + '/rule.json'
        data = {}
        if os.path.exists(ruleFile):
            with open(ruleFile,'r') as inFile:
                data = json.load(inFile)
                inFile.close()
        else:
            data['CurrentMode'] = ''
            data['ModeData'] = []
        ruleName = self.ui.modeNameEdit.text()
        if ruleName == '':
            QtWidgets.QMessageBox.warning(self,'warning','需要设置名字',QtWidgets.QMessageBox.Ok)
            return
        if len(self.itemList) == 0:
            QtWidgets.QMessageBox.warning(self, 'warning', '需要至少选择一个对象', QtWidgets.QMessageBox.Ok)
            return
        if self.checkSame(data,ruleName):
            QtWidgets.QMessageBox.warning(self, 'warning', '名字重复，请重新设置', QtWidgets.QMessageBox.Ok)
            return
        item = {}
        item['name'] = ruleName
        item['list'] = []
        for t in self.itemList:
            ruleName = t.getRuleName()
            nodeNames = t.getNodeNames()
            if ruleName == '' or len(nodeNames) == '':
                continue
            item['list'].append({'button':ruleName,'rule':nodeNames})

        data['ModeData'].append(item)
        jsonStr = json.dumps(data, indent=2, sort_keys=True)
        with open(ruleFile,'w') as outFile:
            outFile.write(jsonStr)
            outFile.close()
        if self.okCallBack:
            self.okCallBack()
        self.close()

好了，这个里面的新知识点很少，都是前面介绍的综合使用。下面附上完整的代码：