Initial Submission

First submission of current state of ARTv2. Currently considered to be in Alpha. There are a couple of animation tools not implemented yet, and one module not implemented yet, as well as incomplete documentation.

1 files changed, 287 insertions, 0 deletions

diff --git a/Core/Scripts/Interfaces/ART_MatchOverRangeUI.py b/Core/Scripts/Interfaces/ART_MatchOverRangeUI.py
new file mode 100644
index 0000000..a713ece
--- /dev/null
+++ b/Core/Scripts/Interfaces/ART_MatchOverRangeUI.py
@@ -0,0 +1,287 @@
+
+import os
+from functools import partial
+
+import maya.cmds as cmds
+
+import System.interfaceUtils as interfaceUtils
+import System.utils as utils
+from ThirdParty.Qt import QtGui, QtCore, QtWidgets
+
+
+class ART_MatchOverRange(object):
+    def __init__(self, animPickerUI, parent=None):
+
+        super(ART_MatchOverRange, self).__init__()
+
+        # get the directory path of the tools
+        settings = QtCore.QSettings("Epic Games", "ARTv2")
+        self.toolsPath = settings.value("toolsPath")
+        self.iconsPath = settings.value("iconPath")
+        self.scriptPath = settings.value("scriptPath")
+        self.projectPath = settings.value("projectPath")
+
+        self.pickerUI = animPickerUI
+
+        # write out qss based on user settings
+        stylesheetDir = utils.returnNicePath(self.scriptPath, "Interfaces/StyleSheets/")
+        stylesheets = os.listdir(stylesheetDir)
+
+        for sheet in stylesheets:
+            interfaceUtils.writeQSS(os.path.join(stylesheetDir, sheet))
+
+        # build the UI
+        self.buildUI()
+
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def buildUI(self):
+
+        if cmds.window("pyART_MatchOverRangeWIN", exists=True):
+            cmds.deleteUI("pyART_MatchOverRangeWIN", wnd=True)
+
+        # create the main window
+        self.mainWin = QtWidgets.QMainWindow(self.pickerUI)
+
+        # create the main widget
+        self.mainWidget = QtWidgets.QFrame()
+        self.mainWidget.setObjectName("dark")
+        self.mainWin.setCentralWidget(self.mainWidget)
+
+        # load stylesheet
+        styleSheetFile = utils.returnNicePath(self.toolsPath, "Core/Scripts/Interfaces/StyleSheets/animPicker.qss")
+        f = open(styleSheetFile, "r")
+        self.style = f.read()
+        f.close()
+        self.mainWin.setStyleSheet(self.style)
+
+        # set window size
+        self.mainWin.setMinimumSize(QtCore.QSize(600, 350))
+        self.mainWin.setMaximumSize(QtCore.QSize(600, 350))
+        self.mainWin.resize(600, 350)
+
+        # set qt object name
+        self.mainWin.setObjectName("pyART_MatchOverRangeWIN")
+        self.mainWin.setWindowTitle("Match Over Frame Range")
+
+        # horizontal layout
+        self.mainLayout = QtWidgets.QHBoxLayout(self.mainWidget)
+
+        # LEFT SIDE
+        # module list widget
+        self.moduleList = QtWidgets.QListWidget()
+        self.mainLayout.addWidget(self.moduleList)
+        # self.moduleList.setMinimumSize(QtCore.QSize(300, 280))
+        # self.moduleList.setMaximumSize(QtCore.QSize(300, 280))
+        self.moduleList.setSelectionMode(QtWidgets.QAbstractItemView.NoSelection)
+        self.moduleList.setSpacing(20)
+
+        # RIGHT SIDE
+        self.rightLayout = QtWidgets.QVBoxLayout()
+        self.mainLayout.addLayout(self.rightLayout)
+
+        self.rightLayout.addSpacerItem(
+            QtWidgets.QSpacerItem(0, 25, QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed))
+        self.characterCombo = QtWidgets.QComboBox()
+        self.rightLayout.addWidget(self.characterCombo)
+        self.characterCombo.setMinimumHeight(50)
+        self.characterCombo.setMaximumHeight(50)
+        self.characterCombo.setIconSize(QtCore.QSize(50, 50))
+        self.characterCombo.currentIndexChanged.connect(partial(self.findCharacterModules))
+
+        # frame ranges
+
+        # SPACER!
+        self.rightLayout.addSpacerItem(
+            QtWidgets.QSpacerItem(0, 40, QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed))
+
+        self.rangeLayout = QtWidgets.QHBoxLayout()
+        self.rightLayout.addLayout(self.rangeLayout)
+
+        label1 = QtWidgets.QLabel("Start:")
+        label1.setStyleSheet("background: transparent;")
+        self.rangeLayout.addWidget(label1)
+
+        self.startFrame = QtWidgets.QSpinBox()
+        self.startFrame.setButtonSymbols(QtWidgets.QAbstractSpinBox.NoButtons)
+        self.rangeLayout.addWidget(self.startFrame)
+        self.startFrame.setRange(-10000, 10000)
+
+        label2 = QtWidgets.QLabel("End:")
+        label2.setStyleSheet("background: transparent;")
+        self.rangeLayout.addWidget(label2)
+
+        self.endFrame = QtWidgets.QSpinBox()
+        self.endFrame.setButtonSymbols(QtWidgets.QAbstractSpinBox.NoButtons)
+        self.rangeLayout.addWidget(self.endFrame)
+        self.endFrame.setRange(-10000, 10000)
+
+        # SPACER!
+        self.rightLayout.addSpacerItem(
+            QtWidgets.QSpacerItem(0, 0, QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Expanding))
+
+        # Button
+        self.matchButton = QtWidgets.QPushButton("Match")
+        self.rightLayout.addWidget(self.matchButton)
+        self.matchButton.setObjectName("blueButton")
+        self.matchButton.setMinimumHeight(50)
+        self.matchButton.clicked.connect(self.match)
+
+        # SPACER!
+        self.rightLayout.addSpacerItem(
+            QtWidgets.QSpacerItem(0, 25, QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed))
+
+        # show the window
+        self.mainWin.show()
+
+        # populate UI
+        self.findCharacters()
+        self.findCharacterModules()
+
+        startFrame = cmds.playbackOptions(q=True, min=True)
+        endFrame = cmds.playbackOptions(q=True, max=True)
+
+        self.startFrame.setValue(startFrame)
+        self.endFrame.setValue(endFrame)
+
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def findCharacters(self):
+
+        allNodes = cmds.ls(type="network")
+        characterNodes = []
+        for node in allNodes:
+            attrs = cmds.listAttr(node)
+            if "rigModules" in attrs:
+                characterNodes.append(node)
+
+        # go through each node, find the character name, the namespace on the node, and the picker attribute
+        for node in characterNodes:
+            try:
+                namespace = cmds.getAttr(node + ".namespace")
+            except:
+                namespace = cmds.getAttr(node + ".name")
+
+            # add the icon found on the node's icon path attribute to the tab
+            iconPath = cmds.getAttr(node + ".iconPath")
+            iconPath = utils.returnNicePath(self.projectPath, iconPath)
+            icon = QtGui.QIcon(iconPath)
+
+            self.characterCombo.addItem(icon, namespace)
+
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+
+    def findCharacterModules(self, *args):
+
+        self.moduleList.clear()
+
+        # current character
+        selectedChar = self.characterCombo.currentText()
+
+        # get rig modules
+        if cmds.objExists(selectedChar + ":" + "ART_RIG_ROOT"):
+            modules = cmds.listConnections(selectedChar + ":" + "ART_RIG_ROOT.rigModules")
+
+            for module in modules:
+                niceName = cmds.getAttr(module + ".moduleName")
+                moduleType = cmds.getAttr(module + ".moduleType")
+
+                # create widget
+                item = QtWidgets.QListWidgetItem()
+
+                widgetItem = QtWidgets.QGroupBox()
+                widgetItem.setMinimumHeight(50)
+                widgetItem.setProperty("module", module)
+                widgetItem.setObjectName("light")
+
+                layout = QtWidgets.QHBoxLayout(widgetItem)
+
+                checkBox = QtWidgets.QCheckBox(niceName)
+                checkBox.setChecked(False)
+                layout.addWidget(checkBox)
+
+                comboBox = QtWidgets.QComboBox()
+                layout.addWidget(comboBox)
+
+                # add items to combo box bases on module class var
+                mod = __import__("RigModules." + moduleType, {}, {}, [moduleType])
+                matchData = mod.matchData
+
+                if matchData[0] is True:
+                    for each in matchData[1]:
+                        comboBox.addItem(each)
+
+                    comboBox.setCurrentIndex(1)
+
+                    self.moduleList.addItem(item)
+                    self.moduleList.setItemWidget(item, widgetItem)
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+
+    def match(self):
+
+        # get the current character
+        character = self.characterCombo.currentText()
+
+        # go through each module in list, find import method, and setup constraints accordingly
+        moduleItems = []
+        for i in range(self.moduleList.count()):
+            item = self.moduleList.item(i)
+            itemWidget = self.moduleList.itemWidget(item)
+            itemModule = itemWidget.property("module")
+
+            children = itemWidget.children()
+            for child in children:
+
+                if type(child) == QtWidgets.QCheckBox:
+                    value = child.isChecked()
+                    if value == False:
+                        break
+
+                if type(child) == QtWidgets.QComboBox:
+                    matchMethod = child.currentText()
+                    moduleItems.append([itemModule, matchMethod])
+
+        # get frame range
+        start = self.startFrame.value()
+        end = self.endFrame.value()
+
+        # loop through frame range, calling each module's match function given the match method
+        if len(moduleItems) > 0:
+            for i in range(start, end + 1):
+                cmds.currentTime(i)
+                for each in moduleItems:
+                    # get inst
+                    modType = cmds.getAttr(each[0] + ".moduleType")
+                    modName = cmds.getAttr(each[0] + ".moduleName")
+                    mod = __import__("RigModules." + modType, {}, {}, [modType])
+                    reload(mod)
+
+                    # get the class name from that module file (returns Modules.ART_Root.ART_Root for example)
+                    moduleClass = getattr(mod, mod.className)
+
+                    # find the instance of that module
+                    moduleInst = moduleClass(self, modName)
+
+                    # set namespace for instance
+                    moduleInst.namespace = character + ":"
+
+                    # call on module's match function (method, checkbox, match over range)
+                    moduleInst.switchMode(each[1], None, True)