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, 773 insertions, 0 deletions

diff --git a/Core/Scripts/Interfaces/ART_WeightWizard.py b/Core/Scripts/Interfaces/ART_WeightWizard.py
new file mode 100644
index 0000000..fd349e4
--- /dev/null
+++ b/Core/Scripts/Interfaces/ART_WeightWizard.py
@@ -0,0 +1,773 @@
+from ThirdParty.Qt import QtGui, QtCore, QtWidgets
+from functools import partial
+import maya.cmds as cmds
+import os
+import System.utils as utils
+import System.riggingUtils as riggingUtils
+
+#maya 2016< maya2017> compatability
+try:
+    import shiboken as shiboken
+except:
+    import shiboken2 as shiboken
+
+
+
+def getMainWindow():
+    import maya.OpenMayaUI as mui
+    pointer = mui.MQtUtil.mainWindow()
+    #pyside QMainWindow takes in a QWidget rather than QObject
+    return shiboken.wrapInstance(long(pointer), QtWidgets.QWidget)
+
+
+windowTitle = "Weight Wizard"
+windowObject = "pyArtWeightWizardWin"
+
+class WeightWizard(QtWidgets.QMainWindow):
+
+    def __init__(self, rigUiInst,  parent = None):
+        #call base class constructor
+        super(WeightWizard, self).__init__(parent)
+
+        #get the directory path of the tools
+        settings = QtCore.QSettings("Epic Games", "ARTv2")
+        self.toolsPath = settings.value("toolsPath")
+        self.iconsPath = settings.value("iconPath")
+
+        #store rigUiInst in class
+        self.rigUiInst = rigUiInst
+
+        #set size policy
+        mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed)
+
+        #create the main widget
+        self.mainWidget = QtWidgets.QWidget()
+        self.setCentralWidget(self.mainWidget)
+
+        #load toolbar stylesheet
+        styleSheetFile = utils.returnNicePath(self.toolsPath, "Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
+        f = open(styleSheetFile, "r")
+        self.style = f.read()
+        f.close()
+        self.setStyleSheet(self.style)
+
+        #set qt object name
+        self.setObjectName(windowObject)
+        self.setWindowTitle(windowTitle)
+
+        #create the mainLayout for the rig creator UI
+        self.layout = QtWidgets.QVBoxLayout(self.mainWidget)
+
+        self.resize(600, 400)
+        self.setSizePolicy(mainSizePolicy)
+        self.setMinimumSize(QtCore.QSize( 600, 400 ))
+        self.setMaximumSize(QtCore.QSize( 600, 400 ))
+
+        #Create a stackedWidget
+        self.stackWidget = QtWidgets.QStackedWidget()
+        self.layout.addWidget(self.stackWidget)
+
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        #PAGE ONE: SKIN PROXY GEO
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        self.page1 = QtWidgets.QFrame()
+        self.page1.setObjectName("dark")
+        self.stackWidget.addWidget(self.page1)
+        self.page1MainLayout = QtWidgets.QVBoxLayout(self.page1)
+
+
+        #label
+        pageOneLabel = QtWidgets.QLabel("Skin Weight Proxy Geometry?")
+        pageOneLabel.setStyleSheet("background: transparent;")
+        self.page1MainLayout.addWidget(pageOneLabel)
+        font = QtGui.QFont()
+        font.setPointSize(12)
+        font.setBold(True)
+        pageOneLabel.setFont(font)
+
+        #image
+        self.pageOneInfo = QtWidgets.QFrame()
+        self.pageOneInfo.setMinimumSize(QtCore.QSize( 560, 250 ))
+        self.pageOneInfo.setMaximumSize(QtCore.QSize( 560, 250 ))
+        self.page1MainLayout.addWidget(self.pageOneInfo)
+
+        image =  utils.returnNicePath(self.iconsPath, "System/weightProxy.png")
+        self.pageOneInfo.setStyleSheet("background-image: url(" + image + ");")
+
+        #buttons
+        self.pageOneButtonLayout = QtWidgets.QHBoxLayout()
+        self.page1MainLayout.addLayout(self.pageOneButtonLayout)
+        self.skinProxyFalseBtn = QtWidgets.QPushButton("No")
+        self.skinProxyFalseBtn.setMinimumHeight(50)
+        self.skinProxyFalseBtn.setFont(font)
+        self.skinProxyTrueBtn = QtWidgets.QPushButton("Yes")
+        self.skinProxyTrueBtn.setMinimumHeight(50)
+        self.skinProxyTrueBtn.setFont(font)
+        self.pageOneButtonLayout.addWidget(self.skinProxyFalseBtn)
+        self.pageOneButtonLayout.addWidget(self.skinProxyTrueBtn)
+
+        self.skinProxyFalseBtn.setObjectName("blueButton")
+        self.skinProxyTrueBtn.setObjectName("blueButton")
+
+
+        #button hookups
+        self.skinProxyFalseBtn.clicked.connect(self.checkForCustomMeshes)
+        self.skinProxyTrueBtn.clicked.connect(self.skinProxyGeo)
+
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        #PAGE TWO: NO CUSTOM GEO FOUND
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        self.page2 = QtWidgets.QFrame()
+        self.stackWidget.addWidget(self.page2)
+        self.page2MainLayout = QtWidgets.QVBoxLayout(self.page2)
+        self.page2.setObjectName("dark")
+
+        #label
+        pageTwoLabel = QtWidgets.QLabel("Skin Weight Custom Geometry?")
+        self.page2MainLayout.addWidget(pageTwoLabel)
+        font = QtGui.QFont()
+        font.setPointSize(12)
+        font.setBold(True)
+        pageTwoLabel.setFont(font)
+
+
+        #image
+        self.pageTwoInfo = QtWidgets.QFrame()
+        self.pageTwoInfo.setMinimumSize(QtCore.QSize( 560, 250 ))
+        self.pageTwoInfo.setMaximumSize(QtCore.QSize( 560, 250 ))
+        self.page2MainLayout.addWidget(self.pageTwoInfo)
+
+        image =  utils.returnNicePath(self.iconsPath, "System/geoNotFound.png")
+        self.pageTwoInfo.setStyleSheet("background-image: url(" + image + ");")
+
+        #buttons
+        self.pageTwoButtonLayout = QtWidgets.QHBoxLayout()
+        self.page2MainLayout.addLayout(self.pageTwoButtonLayout)
+        self.skipStepButton = QtWidgets.QPushButton("Skip This Step")
+        self.skipStepButton.setMinimumHeight(50)
+        self.skipStepButton.setFont(font)
+        self.addMeshesButton = QtWidgets.QPushButton("Add Meshes")
+        self.addMeshesButton.setMinimumHeight(50)
+        self.addMeshesButton.setFont(font)
+        self.pageTwoButtonLayout.addWidget(self.skipStepButton)
+        self.pageTwoButtonLayout.addWidget(self.addMeshesButton)
+        self.addMeshesButton.setObjectName("blueButton")
+        self.skipStepButton.setObjectName("blueButton")
+
+        #connect button
+        self.skipStepButton.clicked.connect(self.closeWizard)
+        self.addMeshesButton.clicked.connect(self.addCustomMeshes)
+
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        #PAGE THREE: SKIN CUSTOM GEO
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+
+        self.page3 = QtWidgets.QFrame()
+        self.stackWidget.addWidget(self.page3)
+        self.page3MainLayout = QtWidgets.QVBoxLayout(self.page3)
+        self.page3.setObjectName("dark")
+
+        #label
+        pageThreeLabel = QtWidgets.QLabel("Skin Weight Custom Geometry?")
+        self.page3MainLayout.addWidget(pageThreeLabel)
+        font = QtGui.QFont()
+        font.setPointSize(12)
+        font.setBold(True)
+        pageThreeLabel.setFont(font)
+
+
+        #image
+        self.pageThreeInfo = QtWidgets.QFrame()
+        self.pageThreeInfo.setMinimumSize(QtCore.QSize( 560, 250 ))
+        self.pageThreeInfo.setMaximumSize(QtCore.QSize( 560, 250 ))
+        self.page3MainLayout.addWidget(self.pageThreeInfo)
+
+        image =  utils.returnNicePath(self.iconsPath, "System/skinCustom.png")
+        self.pageThreeInfo.setStyleSheet("background-image: url(" + image + ");")
+
+        #buttons
+        self.pageThreeButtonLayout = QtWidgets.QHBoxLayout()
+        self.page3MainLayout.addLayout(self.pageThreeButtonLayout)
+        self.skinGeoFalseBtn = QtWidgets.QPushButton("No")
+        self.skinGeoFalseBtn.setMinimumHeight(50)
+        self.skinGeoFalseBtn.setFont(font)
+        self.skinGeoTrueBtn = QtWidgets.QPushButton("Yes")
+        self.skinGeoTrueBtn.setMinimumHeight(50)
+        self.skinGeoTrueBtn.setFont(font)
+        self.pageThreeButtonLayout.addWidget(self.skinGeoFalseBtn)
+        self.pageThreeButtonLayout.addWidget(self.skinGeoTrueBtn)
+        self.skinGeoTrueBtn.setObjectName("blueButton")
+        self.skinGeoFalseBtn.setObjectName("blueButton")
+
+
+        #connect buttons
+        self.skinGeoTrueBtn.clicked.connect(self.skinTools)
+        self.skinGeoFalseBtn.clicked.connect(self.closeWizard)
+
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        #PAGE FOUR: SKIN TOOLS
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+
+        self.page4 = QtWidgets.QFrame()
+        self.stackWidget.addWidget(self.page4)
+        self.page4TopLayout = QtWidgets.QVBoxLayout(self.page4)
+        self.page4.setObjectName("dark")
+        self.page4MainLayout = QtWidgets.QHBoxLayout()
+        self.page4TopLayout.addLayout(self.page4MainLayout)
+
+
+        #vertical layout for lists
+        self.page4VerticalLayout = QtWidgets.QVBoxLayout()
+        self.page4MainLayout.addLayout(self.page4VerticalLayout)
+
+        #list widgets for meshes/joints
+        self.splitter = QtWidgets.QSplitter()
+        self.page4VerticalLayout.addWidget(self.splitter)
+
+        #add geo list widget
+        self.geoLayoutWidget = QtWidgets.QWidget()
+        self.geoLayout = QtWidgets.QVBoxLayout(self.geoLayoutWidget)
+        label = QtWidgets.QLabel("Skinnable Geo:")
+        label.setFont(font)
+        self.geoLayout.addWidget(label)
+
+        self.geoList = QtWidgets.QListWidget()
+        self.geoLayout.addWidget(self.geoList)
+        self.splitter.addWidget(self.geoLayoutWidget)
+
+        #populate geo list
+        skinnableGeo = self.findCustomGeo()
+        for geo in skinnableGeo:
+            self.geoList.addItem(geo)
+
+        #geo list signals
+        self.geoList.itemDoubleClicked.connect(self.selectGeo)
+
+        #add joint list/layout
+        self.splitterWidget = QtWidgets.QWidget()
+        self.splitter.addWidget(self.splitterWidget)
+        self.splitterLayout = QtWidgets.QVBoxLayout(self.splitterWidget)
+
+        self.search = QtWidgets.QLineEdit()
+        self.search.setPlaceholderText("Search...")
+        self.splitterLayout.addWidget(self.search)
+        self.jointList = QtWidgets.QListWidget()
+        self.jointList.setSelectionMode(QtWidgets.QAbstractItemView.ExtendedSelection)
+        self.splitterLayout.addWidget(self.jointList)
+
+        #populate joint list
+        joints = self.findJoints()
+        for joint in joints:
+            self.jointList.addItem(joint)
+
+        #search bar
+        self.search.textChanged.connect(self.searchJoints)
+
+        #add bottom layout for options/filters
+        self.bottomLayout = QtWidgets.QHBoxLayout()
+        self.bottomLayoutFrame = QtWidgets.QFrame()
+        self.bottomLayoutFrame.setObjectName("dark")
+        self.bottomLayoutFrame.setMaximumHeight(70)
+        self.bottomLayout.addWidget(self.bottomLayoutFrame)
+        self.optionsLayout = QtWidgets.QHBoxLayout(self.bottomLayoutFrame)
+        self.page4VerticalLayout.addLayout(self.bottomLayout)
+
+        #add bottom options/smooth bind (max influences, maintain max, dropoff rate
+
+        #MAX INFLUENCES
+        self.maxInfluenceLayout = QtWidgets.QVBoxLayout()
+        self.optionsLayout.addLayout(self.maxInfluenceLayout)
+
+        label = QtWidgets.QLabel("Max Influences: ")
+        label.setMaximumHeight(20)
+        label.setStyleSheet("color: rgb(25,175,255); background:transparent;")
+        self.maxInfluenceLayout.addWidget(label)
+
+        self.maxInfOptionsLayout = QtWidgets.QHBoxLayout()
+        self.maxInfluenceLayout.addLayout(self.maxInfOptionsLayout)
+
+        self.maxInfluences = QtWidgets.QSlider()
+        self.maxInfluences.setRange(1, 8)
+        self.maxInfluences.setSingleStep(1)
+        self.maxInfluences.setPageStep(1)
+        self.maxInfluences.setOrientation(QtCore.Qt.Horizontal)
+        self.maxInfOptionsLayout.addWidget(self.maxInfluences)
+
+        self.maxInfluencesReadout = QtWidgets.QSpinBox()
+        self.maxInfOptionsLayout.addWidget(self.maxInfluencesReadout)
+        self.maxInfluencesReadout.setReadOnly(True)
+        self.maxInfluencesReadout.setButtonSymbols(QtWidgets.QAbstractSpinBox.NoButtons)
+        self.maxInfluences.valueChanged.connect(self.maxInfluencesReadout.setValue)
+
+
+        #DROPOFF RATE
+        self.dropoffRateLayout = QtWidgets.QVBoxLayout()
+        self.optionsLayout.addLayout(self.dropoffRateLayout)
+
+        label2 = QtWidgets.QLabel("Dropoff Rate:")
+        label2.setMaximumHeight(20)
+        label2.setStyleSheet("color: rgb(25,175,255); background:transparent;")
+        self.dropoffRateLayout.addWidget(label2)
+
+        self.dropRateOptionsLayout = QtWidgets.QHBoxLayout()
+        self.dropoffRateLayout.addLayout(self.dropRateOptionsLayout)
+
+        self.dropoffRate = QtWidgets.QSlider()
+        self.dropoffRate.setRange(.1, 100)
+        self.dropoffRate.setSingleStep(.1)
+        self.dropoffRate.setPageStep(1)
+        self.dropoffRate.setOrientation(QtCore.Qt.Horizontal)
+        self.dropRateOptionsLayout.addWidget(self.dropoffRate)
+
+        self.dropoffReadout = QtWidgets.QSpinBox()
+        self.dropRateOptionsLayout.addWidget(self.dropoffReadout)
+        self.dropoffReadout.setReadOnly(True)
+        self.dropoffReadout.setButtonSymbols(QtWidgets.QAbstractSpinBox.NoButtons)
+        self.dropoffRate.valueChanged.connect(self.dropoffReadout.setValue)
+
+        #MAINTAIN MAX
+        self.maintainMaxInf = QtWidgets.QPushButton("Maintain Max Influences")
+        self.optionsLayout.addWidget(self.maintainMaxInf)
+        self.maintainMaxInf.setMinimumWidth(160)
+        self.maintainMaxInf.setCheckable(True)
+        self.maintainMaxInf.setChecked(True)
+        self.maintainMaxInf.clicked.connect(partial(self.toggleButtonState, self.maintainMaxInf))
+        self.maintainMaxInf.setObjectName("blueButton")
+
+        #button layout
+        self.buttonLayout = QtWidgets.QVBoxLayout()
+        self.optionsLayout.addLayout(self.buttonLayout)
+
+        #SMOOTH BIND
+        self.smoothBind = QtWidgets.QPushButton("Smooth Bind")
+        self.smoothBind.setMinimumWidth(150)
+        self.smoothBind.setMinimumHeight(30)
+        self.buttonLayout.addWidget(self.smoothBind)
+        self.smoothBind.clicked.connect(self.smoothBindSelected)
+        self.smoothBind.setObjectName("blueButton")
+
+        #CLOSE
+        self.closeBtn = QtWidgets.QPushButton("Close")
+        self.closeBtn.setMinimumWidth(150)
+        self.closeBtn.setMinimumHeight(30)
+        self.buttonLayout.addWidget(self.closeBtn)
+        self.closeBtn.clicked.connect(self.closeWizard)
+        self.closeBtn.setObjectName("blueButton")
+
+
+        #set values
+        self.dropoffRate.setValue(4)
+        self.maxInfluences.setValue(4)
+
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        #PAGE  FIVE: SKIN WEIGHTS FOUND
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+        # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+
+        self.page5 = QtWidgets.QFrame()
+        self.page5.setObjectName("dark")
+        self.stackWidget.addWidget(self.page5)
+        self.page5MainLayout = QtWidgets.QVBoxLayout(self.page5)
+
+
+
+        #label
+        pageFiveLabel = QtWidgets.QLabel("Skin Weights Found For These Meshes:")
+        self.page5MainLayout.addWidget(pageFiveLabel)
+        font = QtGui.QFont()
+        font.setPointSize(12)
+        font.setBold(True)
+        pageFiveLabel.setFont(font)
+
+        #columnLayout
+        self.page5ColumnLayout = QtWidgets.QHBoxLayout()
+        self.page5MainLayout.addLayout(self.page5ColumnLayout)
+
+        #left column has list widget of meshes
+        self.page5MeshList = QtWidgets.QListWidget()
+        self.page5ColumnLayout.addWidget(self.page5MeshList)
+        self.page5MeshList.setSelectionMode(QtWidgets.QAbstractItemView.ExtendedSelection)
+        self.page5MeshList.setMinimumWidth(250)
+        self.page5MeshList.setMaximumWidth(250)
+
+
+        #right column has import method, and import/do not import buttons
+        self.page5VerticalLayout = QtWidgets.QVBoxLayout()
+        self.page5ColumnLayout.addLayout(self.page5VerticalLayout)
+
+        layout = QtWidgets.QHBoxLayout()
+        self.page5VerticalLayout.addLayout(layout)
+
+        label = QtWidgets.QLabel("Import Method: ")
+        label.setStyleSheet("background: transparent;")
+        layout.addWidget(label)
+        layout.setSpacing(10)
+        label.setMinimumWidth(130)
+        label.setMaximumWidth(130)
+
+        self.page5ImportOptions = QtWidgets.QComboBox()
+        self.page5ImportOptions.addItem("Vertex Order")
+        self.page5ImportOptions.addItem("World Position")
+        self.page5ImportOptions.addItem("Local Position")
+        self.page5ImportOptions.addItem("UV Position")
+        layout.addWidget(self.page5ImportOptions)
+        self.page5ImportOptions.setMinimumWidth(155)
+        self.page5ImportOptions.setMaximumWidth(155)
+
+        self.page5VerticalLayout.addSpacerItem(QtWidgets.QSpacerItem(0, 200, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Fixed))
+
+        self.page5ImportWeights = QtWidgets.QPushButton("Import Weights")
+        self.page5ImportWeights.setMinimumHeight(50)
+        self.page5ImportWeights.setFont(font)
+        self.page5VerticalLayout.addWidget(self.page5ImportWeights)
+        self.page5ImportWeights.clicked.connect(partial(self.importWeights))
+        self.page5ImportWeights.setObjectName("blueButton")
+
+        self.page5IgnoreWeights = QtWidgets.QPushButton("Skip")
+        self.page5IgnoreWeights.setMinimumHeight(50)
+        self.page5IgnoreWeights.setFont(font)
+        self.page5VerticalLayout.addWidget(self.page5IgnoreWeights)
+        self.page5IgnoreWeights.clicked.connect(partial(self.skipWeightImport))
+        self.page5IgnoreWeights.setObjectName("blueButton")
+
+
+
+
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def checkForCustomMeshes(self):
+
+        #check for custom geometry
+        skinnableGeo = self.findCustomGeo()
+        meshes = list(skinnableGeo)
+
+        if len(skinnableGeo) > 0:
+
+            matches = []
+
+            #check to see if weight files exist for this geo on disk
+            for i in range(len(meshes)):
+                filePath = utils.returnFriendlyPath(os.path.join(cmds.internalVar(utd = True), meshes[i] + ".WEIGHTS"))
+                if os.path.exists(filePath):
+                    matches.append(meshes[i])
+
+            if len(matches) > 0:
+                for match in matches:
+                    listWidgetItem = QtWidgets.QListWidgetItem(match)
+                    self.page5MeshList.addItem(listWidgetItem)
+                    listWidgetItem.setSelected(True)
+
+                #go to page 5
+                self.stackWidget.setCurrentIndex(4)
+
+            else:
+                #go to page 3
+                self.stackWidget.setCurrentIndex(2)
+
+        else:
+            #go to page 2
+            self.stackWidget.setCurrentIndex(1)
+
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def selectGeo(self):
+
+        selected = self.geoList.selectedItems()
+        selected = selected[0].text()
+
+        cmds.select(selected, r = True)
+
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def searchJoints(self):
+
+        #get current search terms
+        searchKey = self.search.text()
+
+        #get all items in the joint list
+        allItems = []
+        for i in range(self.jointList.count()):
+            item =  self.jointList.item(i)
+            itemName = item.text()
+            allItems.append([item, itemName])
+
+        #hide all items in list
+        for item in allItems:
+            item[0].setHidden(True)
+
+        #find items in list with search key and show item
+        if searchKey.find("*") == 0:
+            matchedItems = self.jointList.findItems(searchKey, QtCore.Qt.MatchFlag.MatchWildcard)
+            for item in matchedItems:
+                item.setHidden(False)
+
+        else:
+            matchedItems = self.jointList.findItems(searchKey, QtCore.Qt.MatchFlag.MatchContains)
+            for item in matchedItems:
+                item.setHidden(False)
+
+
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def skinProxyGeo(self):
+
+
+        #get modules in scene and call the skinProxyGeo function
+        for module in self.rigUiInst.moduleInstances:
+            module.skinProxyGeo()
+
+
+        self.checkForCustomMeshes()
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def addCustomMeshes(self):
+
+        try:
+            files = cmds.fileDialog2(fm = 4, okc = "Add Selected Meshes")
+
+            if len(files) > 0:
+                for f in files:
+                    try:
+                        cmds.file(f, i = True, dns = True)
+                    except:
+                        cmds.warning("Could not import file: " + str(f))
+
+                #add new meshes to list
+                skinnableGeo = self.findCustomGeo()
+                for geo in skinnableGeo:
+                    self.geoList.addItem(geo)
+
+                #next page
+                self.stackWidget.setCurrentIndex(2)
+
+        except:
+            pass
+
+
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def skinTools(self):
+        self.stackWidget.setCurrentIndex(3)
+
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def closeWizard(self):
+
+        mayaWindow = getMainWindow()
+        mayaWindow = mayaWindow.objectName()
+        if cmds.window(mayaWindow + "|pyArtWeightWizardWin", q = True, exists = True):
+            cmds.deleteUI(mayaWindow +"|pyArtWeightWizardWin")
+
+
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def findCustomGeo(self):
+        meshes = cmds.ls(type = "mesh")
+        skinnableGeo = []
+
+        for mesh in meshes:
+            parent = cmds.listRelatives(mesh, parent = True, type = "transform")[0]
+            if parent != None:
+                if parent.find("proxy_geo") == -1:
+                    if parent.find("lra") == -1:
+                        if parent.find("bone_geo") == -1:
+                            skinnableGeo.append(parent)
+        skinnableGeo = set(skinnableGeo)
+        return skinnableGeo
+
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def findJoints(self):
+        if cmds.objExists("root"):
+            cmds.select("root", hi = True)
+            selection = cmds.ls(sl = True)
+            joints = []
+
+            for each in selection:
+                if cmds.nodeType(each) == "joint":
+                    joints.append(each)
+
+            return joints
+
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def smoothBindSelected(self):
+
+        #get selected geo from list
+        selectedGeo = self.geoList.selectedItems()[0]
+        geoRow = self.geoList.row(selectedGeo)
+        geo = selectedGeo.text()
+
+        #selected joints
+        selectedJoints = self.jointList.selectedItems()
+        joints = []
+        for joint in selectedJoints:
+            joints.append(joint.text())
+
+        #get smooth skin settings
+        maxInf = self.maxInfluences.value()
+        dropoff = self.dropoffRate.value()
+        maintainMax = self.maintainMaxInf.isChecked()
+
+        #smooth bind
+        cmds.select(clear = True)
+        for joint in joints:
+            cmds.select(joint, add = True)
+        cmds.select(geo, add = True)
+
+        skinCluster = cmds.skinCluster(tsb = True, maximumInfluences = maxInf, obeyMaxInfluences = maintainMax, dropoffRate = dropoff, bindMethod = 0, skinMethod = 0, normalizeWeights = 1, name = geo + "_skinCluster")[0]
+
+        #clear selection and remove geo from list
+        cmds.select(clear = True)
+        self.geoList.takeItem(geoRow)
+
+
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def importWeights(self):
+
+        #get the selected items in the mesh list
+        selected = self.page5MeshList.selectedItems()
+
+        #get the import method
+        method = self.page5ImportOptions.currentText()
+
+        meshes = []
+        for each in selected:
+            meshes.append(each.text())
+
+        for mesh in meshes:
+            filePath = utils.returnFriendlyPath(os.path.join(cmds.internalVar(utd = True), mesh + ".WEIGHTS"))
+            if os.path.exists(filePath):
+                riggingUtils.import_skin_weights(filePath, mesh, True)
+
+        self.closeWizard()
+
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def skipWeightImport(self):
+        #go to page 3
+        self.stackWidget.setCurrentIndex(2)
+
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def toggleButtonState(self, button):
+
+        #button background image
+        buttonImageOn =  os.path.normcase(os.path.join(self.iconsPath, "System/blue_field_background.png"))
+        if buttonImageOn.partition("\\")[2] != "":
+            buttonImageOn = buttonImageOn.replace("\\", "/")
+
+        buttonImageOff =  os.path.normcase(os.path.join(self.iconsPath, "System/field_background.png"))
+        if buttonImageOff.partition("\\")[2] != "":
+            buttonImageOff = buttonImageOff.replace("\\", "/")
+
+        state = button.isChecked()
+        if state:
+            button.setStyleSheet("background-image: url(" + buttonImageOn + "); background-color: rgb(25,175,255)")
+
+        else:
+            button.setStyleSheet("background-image: url(" + buttonImageOff + "); background-color: rgb(0,0,0)")
+
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+    def closeWizard(self):
+        mayaWindow = getMainWindow()
+        mayaWindow = mayaWindow.objectName()
+        if cmds.window(mayaWindow + "|pyArtWeightWizardWin", q = True, exists = True):
+            cmds.deleteUI(mayaWindow +"|pyArtWeightWizardWin")
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+def run(rigUiInst):
+    mayaWindow = getMainWindow()
+    mayaWindow = mayaWindow.objectName()
+    if cmds.window(mayaWindow + "|pyArtWeightWizardWin", q = True, exists = True):
+        cmds.deleteUI(mayaWindow +"|pyArtWeightWizardWin")
+
+    gui = WeightWizard(rigUiInst, getMainWindow())
+    gui.show()
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
+def runOnlySkinTools(rigUiInst):
+    mayaWindow = getMainWindow()
+    mayaWindow = mayaWindow.objectName()
+    if cmds.window(mayaWindow + "|pyArtWeightWizardWin", q = True, exists = True):
+        cmds.deleteUI(mayaWindow +"|pyArtWeightWizardWin")
+
+    gui = WeightWizard(rigUiInst, getMainWindow())
+    gui.show()
+
+    gui.stackWidget.setCurrentIndex(3)