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 buildSymmetryModeUI(self, mainUI):
if cmds.window("ART_SymmetryModeWin", exists=True):
cmds.deleteUI("ART_SymmetryModeWin", wnd=True)
#launch a UI to get the name information
self.symmetryModeWin = QtWidgets.QMainWindow(mainUI)
#load stylesheet
styleSheetFile = utils.returnNicePath(
self.toolsPath,
"Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
style = f.read()
f.close()
self.symmetryModeWin.setStyleSheet(style)
#size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Fixed)
#create the main widget
self.symModeWin_mainWidget = QtWidgets.QWidget()
self.symmetryModeWin.setCentralWidget(self.symModeWin_mainWidget)
#set qt object name
self.symmetryModeWin.setObjectName("ART_SymmetryModeWin")
self.symmetryModeWin.setWindowTitle("Mass Mirror Mode")
#create the mainLayout for the rig creator UI
self.symModeWin_mainLayout = QtWidgets.QVBoxLayout(
self.symModeWin_mainWidget)
self.symModeWin_mainLayout.setContentsMargins(0, 0, 0, 0)
self.symmetryModeWin.setSizePolicy(mainSizePolicy)
self.symmetryModeWin.setMinimumSize(QtCore.QSize(600, 250))
self.symmetryModeWin.setMaximumSize(QtCore.QSize(600, 250))
#create the background
self.symModeWin_frame = QtWidgets.QFrame()
self.symModeWin_frame.setObjectName("mid")
self.symModeWin_mainLayout.addWidget(self.symModeWin_frame)
#create the layout for the widgets
self.symModeWin_widgetLayout = QtWidgets.QHBoxLayout(
self.symModeWin_frame)
self.symModeWin_widgetLayout.setContentsMargins(5, 5, 5, 5)
#add the QListWidget Frame
self.symModeWin_moduleListFrame = QtWidgets.QFrame()
self.symModeWin_moduleListFrame.setObjectName("mid")
self.symModeWin_moduleListFrame.setMinimumSize(QtCore.QSize(450, 200))
self.symModeWin_moduleListFrame.setMaximumSize(QtCore.QSize(450, 200))
self.symModeWin_moduleListFrame.setContentsMargins(20, 0, 20, 0)
#create the list widget
self.symModeWin_moduleList = QtWidgets.QListWidget(
self.symModeWin_moduleListFrame)
self.symModeWin_widgetLayout.addWidget(self.symModeWin_moduleListFrame)
self.symModeWin_moduleList.setMinimumSize(QtCore.QSize(450, 200))
self.symModeWin_moduleList.setMaximumSize(QtCore.QSize(450, 200))
self.symModeWin_moduleList.setSelectionMode(
QtWidgets.QAbstractItemView.NoSelection)
self.symModeWin_moduleList.setSpacing(3)
#add the layout for the buttons
self.symModeWin_buttonLayoutAll = QtWidgets.QVBoxLayout()
self.symModeWin_widgetLayout.addLayout(self.symModeWin_buttonLayoutAll)
self.symModeWin_buttonLayoutAll.setContentsMargins(5, 20, 5, 20)
#add the selection buttons
self.symModeWin_selectionButtonLayout = QtWidgets.QVBoxLayout()
self.symModeWin_buttonLayoutAll.addLayout(
self.symModeWin_selectionButtonLayout)
self.symModeWin_selectAllButton = QtWidgets.QPushButton("Select All")
self.symModeWin_selectAllButton.setMinimumSize(QtCore.QSize(115, 25))
self.symModeWin_selectAllButton.setMaximumSize(QtCore.QSize(115, 25))
self.symModeWin_selectionButtonLayout.addWidget(
self.symModeWin_selectAllButton)
self.symModeWin_selectAllButton.clicked.connect(
partial(self.symmetryMode_selectDeselect, True))
self.symModeWin_selectAllButton.setObjectName("blueButton")
self.symModeWin_selectNoneButton = QtWidgets.QPushButton(
"Clear Selection")
self.symModeWin_selectNoneButton.setMinimumSize(QtCore.QSize(115, 25))
self.symModeWin_selectNoneButton.setMaximumSize(QtCore.QSize(115, 25))
self.symModeWin_selectionButtonLayout.addWidget(
self.symModeWin_selectNoneButton)
self.symModeWin_selectNoneButton.clicked.connect(
partial(self.symmetryMode_selectDeselect, False))
self.symModeWin_selectNoneButton.setObjectName("blueButton")
#spacer
spacerItem = QtWidgets.QSpacerItem(20, 40,
QtWidgets.QSizePolicy.Minimum,
QtWidgets.QSizePolicy.Expanding)
self.symModeWin_buttonLayoutAll.addItem(spacerItem)
#add the mirror buttons
self.symModeWin_mirrorButtonLayout = QtWidgets.QVBoxLayout()
self.symModeWin_buttonLayoutAll.addLayout(
self.symModeWin_mirrorButtonLayout)
self.symModeWin_mirrorL2RButton = QtWidgets.QPushButton(
"Mirror Checked")
self.symModeWin_mirrorL2RButton.setToolTip(
"Mirror selected modules to unselected modules")
self.symModeWin_mirrorL2RButton.setMinimumSize(QtCore.QSize(115, 25))
self.symModeWin_mirrorL2RButton.setMaximumSize(QtCore.QSize(115, 25))
self.symModeWin_mirrorButtonLayout.addWidget(
self.symModeWin_mirrorL2RButton)
self.symModeWin_mirrorL2RButton.setObjectName("blueButton")
self.symModeWin_mirrorL2RButton.clicked.connect(
partial(self.symmetryMode_mirror))
#populate the list widget
modules = utils.returnRigModules()
entries = []
listMods = []
for mod in modules:
modName = cmds.getAttr(mod + ".moduleName")
mirrorModule = cmds.getAttr(mod + ".mirrorModule")
invalidTypes = [None, "None"]
if mirrorModule not in invalidTypes:
if modName not in listMods:
entries.append([modName, mirrorModule])
listMods.append(modName)
listMods.append(mirrorModule)
self.symModeWinModues = {}
if len(entries) == 0:
item = QtWidgets.QListWidgetItem(self.symModeWin_moduleList)
label = QtWidgets.QLabel("No modules with mirroring setup")
item.setSizeHint(label.sizeHint())
self.symModeWin_moduleList.addItem(item)
self.symModeWin_moduleList.setItemWidget(item, label)
for each in entries:
#create a custom widget to add to each entry in the listWidget
mainWidget = QtWidgets.QWidget()
buttonLayout = QtWidgets.QHBoxLayout(mainWidget)
#create the checkbox
checkbox = QtWidgets.QCheckBox()
checkbox.setMinimumSize(QtCore.QSize(12, 12))
checkbox.setMaximumSize(QtCore.QSize(12, 12))
checkbox.setChecked(True)
buttonLayout.addWidget(checkbox)
label = QtWidgets.QLabel("Mirror ")
buttonLayout.addWidget(label)
mirrorFrom = QtWidgets.QComboBox()
mirrorFrom.addItem(each[0])
mirrorFrom.addItem(each[1])
buttonLayout.addWidget(mirrorFrom)
mirrorFrom.setMinimumWidth(150)
label = QtWidgets.QLabel(" To ")
buttonLayout.addWidget(label)
label.setAlignment(QtCore.Qt.AlignCenter)
mirrorTo = QtWidgets.QComboBox()
mirrorTo.addItem(each[1])
mirrorTo.addItem(each[0])
buttonLayout.addWidget(mirrorTo)
mirrorTo.setMinimumWidth(150)
#signal/slots
mirrorFrom.currentIndexChanged.connect(
partial(self.toggleComboBoxFrom, mirrorFrom, mirrorTo))
mirrorTo.currentIndexChanged.connect(
partial(self.toggleComboBoxTo, mirrorFrom, mirrorTo))
#add this item widget to the list
item = QtWidgets.QListWidgetItem(self.symModeWin_moduleList)
index = entries.index(each)
if (index % 2) == 0:
item.setBackground(QtGui.QColor(106, 106, 108))
else:
item.setBackground(QtGui.QColor(46, 46, 48))
item.setSizeHint(mainWidget.sizeHint())
self.symModeWin_moduleList.addItem(item)
self.symModeWin_moduleList.setItemWidget(item, mainWidget)
#show the window
self.symmetryModeWin.show()
def buildImportWeightsUI(self):
"""
Build the interface for importing skin weights. The interface will create an entry for every piece of selected
geometry.
The interface will look like this:
.. image:: /images/importWeights.png
"""
if cmds.window("ART_importSkinWeightsUI", exists=True):
cmds.deleteUI("ART_importSkinWeightsUI", wnd=True)
# launch a UI to get the name information
self.importSkinWeights_Win = QtWidgets.QMainWindow(self.mainUI)
# load stylesheet
styleSheetFile = utils.returnNicePath(self.toolsPath, "Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
self.style = f.read()
f.close()
self.importSkinWeights_Win.setStyleSheet(self.style)
# create the main widget
self.importSkinWeights_mainWidget = QtWidgets.QWidget()
self.importSkinWeights_Win.setCentralWidget(self.importSkinWeights_mainWidget)
# set qt object name
self.importSkinWeights_Win.setObjectName("ART_importSkinWeightsUI")
self.importSkinWeights_Win.setWindowTitle("Import Skin Weights")
# create the mainLayout for the ui
self.importSkinWeights_mainLayout = QtWidgets.QVBoxLayout(self.importSkinWeights_mainWidget)
self.importSkinWeights_mainLayout.setContentsMargins(5, 5, 5, 5)
self.importSkinWeights_Win.resize(450, 400)
# self.importSkinWeights_Win.setSizePolicy(mainSizePolicy)
self.importSkinWeights_Win.setMinimumSize(QtCore.QSize(450, 400))
self.importSkinWeights_Win.setMaximumSize(QtCore.QSize(450, 400))
# create the background image
self.importSkinWeights_frame = QtWidgets.QFrame()
self.importSkinWeights_mainLayout.addWidget(self.importSkinWeights_frame)
self.importSkinWeights_frame.setObjectName("dark")
# create widgetLayout
self.importSkinWeights_widgetLayout = QtWidgets.QVBoxLayout(self.importSkinWeights_frame)
# import skin weights method
# self.importSkinWeights_methodForm = QtWidgets.QHBoxLayout()
# self.importSkinWeights_widgetLayout.addLayout(self.importSkinWeights_methodForm)
#
# label = QtWidgets.QLabel("Import Method: ")
# label.setStyleSheet("background: transparent;")
# self.importSkinWeights_methodForm.addWidget(label)
# self.importSkinWeights_importMethod = QtWidgets.QComboBox()
# self.importSkinWeights_methodForm.addWidget(self.importSkinWeights_importMethod)
# self.importSkinWeights_importMethod.addItem("Vertex Order")
# self.importSkinWeights_importMethod.addItem("World Position")
# self.importSkinWeights_importMethod.addItem("Local Position")
# self.importSkinWeights_importMethod.addItem("UV Position")
# scroll area contents
self.importSkinWeights_scrollContents = QtWidgets.QFrame()
self.importSkinWeights_scrollContents.setObjectName("light")
# Layout of Container Widget
self.importSkinWeights_VLayout = QtWidgets.QVBoxLayout()
self.importSkinWeights_VLayout.setSpacing(5)
# find selected geometry and populate scroll area
self.importSkinWeights_populate()
# add scrollArea for selected geo, skinFileName, and checkbox for importing
self.importSkinWeights_scrollLayout = QtWidgets.QScrollArea()
self.importSkinWeights_widgetLayout.addWidget(self.importSkinWeights_scrollLayout)
self.importSkinWeights_scrollLayout.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
self.importSkinWeights_scrollLayout.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOn)
self.importSkinWeights_scrollLayout.setWidgetResizable(False)
self.importSkinWeights_scrollLayout.setWidget(self.importSkinWeights_scrollContents)
# refresh and import button
font = QtGui.QFont()
font.setPointSize(8)
font.setBold(True)
self.importSkinWeights_BtnLayout = QtWidgets.QHBoxLayout()
self.importSkinWeights_widgetLayout.addLayout(self.importSkinWeights_BtnLayout)
self.importSkinWeights_RefreshBtn = QtWidgets.QPushButton("Refresh")
self.importSkinWeights_BtnLayout.addWidget(self.importSkinWeights_RefreshBtn)
self.importSkinWeights_RefreshBtn.setMinimumSize(QtCore.QSize(70, 50))
self.importSkinWeights_RefreshBtn.setMaximumSize(QtCore.QSize(70, 50))
self.importSkinWeights_RefreshBtn.setFont(font)
self.importSkinWeights_RefreshBtn.clicked.connect(partial(self.buildImportWeightsUI))
self.importSkinWeights_RefreshBtn.setObjectName("blueButton")
self.importSkinWeights_ImportBtn = QtWidgets.QPushButton("IMPORT WEIGHTS")
self.importSkinWeights_BtnLayout.addWidget(self.importSkinWeights_ImportBtn)
self.importSkinWeights_ImportBtn.setMinimumSize(QtCore.QSize(350, 50))
self.importSkinWeights_ImportBtn.setMaximumSize(QtCore.QSize(350, 50))
self.importSkinWeights_ImportBtn.setFont(font)
self.importSkinWeights_ImportBtn.clicked.connect(partial(self.importSkinWeights_DoImport))
self.importSkinWeights_ImportBtn.setObjectName("blueButton")
# lastly, progress bar
self.importSkinWeights_progBarTotal = QtWidgets.QProgressBar()
self.importSkinWeights_widgetLayout.addWidget(self.importSkinWeights_progBarTotal)
self.importSkinWeights_progBarTotal.setRange(0, self.importSkinWeights_VLayout.count() - 1)
self.importSkinWeights_progBarTotal.setValue(0)
self.importSkinWeights_progBarCurrent = QtWidgets.QProgressBar()
self.importSkinWeights_widgetLayout.addWidget(self.importSkinWeights_progBarCurrent)
self.importSkinWeights_progBarCurrent.setRange(0, 100)
self.importSkinWeights_progBarCurrent.setValue(0)
# show window
self.importSkinWeights_Win.show()
def createUI(self):
"""
Builds the UI, listing options for choosing a project and showing all assets belonging to that project for
edit or add.
"""
# fonts
font = QtGui.QFont()
font.setPointSize(12)
font.setBold(True)
fontSmall = QtGui.QFont()
fontSmall.setPointSize(9)
fontSmall.setBold(True)
# load 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)
# size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed)
# =======================================================================
# #create the main widget
# =======================================================================
self.mainWidget = QtWidgets.QWidget()
self.setCentralWidget(self.mainWidget)
# set qt object name
self.setObjectName(windowObject)
if self.edit:
self.setWindowTitle(windowTitle)
if self.add:
self.setWindowTitle("Add Rig For Animation")
self.setAttribute(QtCore.Qt.WA_DeleteOnClose)
# create the mainLayout
self.layout = QtWidgets.QVBoxLayout(self.mainWidget)
self.resize(400, 400)
self.setSizePolicy(mainSizePolicy)
self.setMinimumSize(QtCore.QSize(400, 400))
self.setMaximumSize(QtCore.QSize(400, 400))
# create the QFrame
self.frame = QtWidgets.QFrame()
self.layout.addWidget(self.frame)
self.widgetLayout = QtWidgets.QHBoxLayout(self.frame)
self.frame.setObjectName("mid")
# =======================================================================
# #create two VBox Layouts to create 2 columns
# =======================================================================
self.leftColumn = QtWidgets.QVBoxLayout()
self.widgetLayout.addLayout(self.leftColumn)
self.rightColumn = QtWidgets.QVBoxLayout()
self.widgetLayout.addLayout(self.rightColumn)
# =======================================================================
# #left column : project comboBox, group comboBox, listWidget of characters
# =======================================================================
self.projectMenu = QtWidgets.QComboBox()
self.leftColumn.addWidget(self.projectMenu)
self.projectMenu.setMinimumSize(150, 30)
self.projectMenu.setMaximumSize(150, 30)
self.projectMenu.currentIndexChanged.connect(self.populateGroups)
self.groupMenu = QtWidgets.QComboBox()
self.leftColumn.addWidget(self.groupMenu)
self.groupMenu.setMinimumSize(150, 30)
self.groupMenu.setMaximumSize(150, 30)
self.groupMenu.currentIndexChanged.connect(self.populateCharacters)
self.characterList = QtWidgets.QListWidget()
self.leftColumn.addWidget(self.characterList)
self.characterList.setMinimumSize(150, 200)
self.characterList.setMaximumSize(150, 200)
self.characterList.itemClicked.connect(partial(self.populateIcon))
self.populateProjects()
# =======================================================================
# #right column: icon frame, edit button/add button, close button
# =======================================================================
self.characterIcon = QtWidgets.QLabel()
self.characterIcon.setMinimumSize(200, 200)
self.characterIcon.setMaximumSize(200, 200)
self.rightColumn.addWidget(self.characterIcon)
# default image
self.defaultPixMap = QtGui.QPixmap(utils.returnNicePath(self.iconsPath, "System/noCharacter.png"))
self.characterIcon.setPixmap(self.defaultPixMap)
# if edit:
if self.edit:
self.editButton = QtWidgets.QPushButton("Edit Selected")
self.editButton.setFont(font)
self.rightColumn.addWidget(self.editButton)
self.editButton.setMinimumSize(200, 40)
self.editButton.setMaximumSize(200, 40)
self.editButton.clicked.connect(partial(self.editSelected))
self.editButton.setObjectName("blueButton")
# if add:
if self.add:
self.addButton = QtWidgets.QPushButton("Add Selected")
self.addButton.setFont(font)
self.rightColumn.addWidget(self.addButton)
self.addButton.setMinimumSize(200, 40)
self.addButton.setMaximumSize(200, 40)
self.addButton.clicked.connect(partial(self.addSelected))
self.addButton.setObjectName("blueButton")
self.closeButton = QtWidgets.QPushButton("Close")
self.closeButton.setFont(font)
self.rightColumn.addWidget(self.closeButton)
self.closeButton.setMinimumSize(200, 40)
self.closeButton.setMaximumSize(200, 40)
self.closeButton.clicked.connect(partial(self.closeUI))
self.closeButton.setObjectName("blueButton")
def exportSkinWeights_doExport(self):
"""
Gather the information from the interface and export the skin weights with that information.
For each piece of geometry, built the path of the .weight file by joining the output path and the .weight
file name, then call on export_skin_weights from riggingUtils, passing in the .weight file path and the mesh.
"""
# get the export path
exportPath = self.exportSkinWeights_lineEdit.text()
value = False
# find each lineEdit in the scrollArea and get the entered text
for i in range(self.exportSkinWeights_VLayout.count()):
hboxLayout = self.exportSkinWeights_VLayout.itemAt(i)
for x in range(hboxLayout.count()):
widget = hboxLayout.itemAt(x)
# go through each widget in the hboxLayout, and get values
if type(widget.widget()) == QtWidgets.QLabel:
geoName = widget.widget().text().partition(" :")[0]
dagname = widget.widget().property("dag")
# see if the user wants to export the weights for this entry
if type(widget.widget()) == QtWidgets.QCheckBox:
value = widget.widget().isChecked()
# get the fileName
if type(widget.widget()) == QtWidgets.QLineEdit:
fileName = widget.widget().text()
if fileName.find(":") is not -1:
# name contains invalid characters for file name.
cmds.warning(
"file name contains invalid characters: ':'")
return
# create the full path
fullPath = utils.returnNicePath(exportPath,
fileName + ".weights")
if value:
# if the checkbox is checked, export skin weights
# export the skin data
riggingUtils.export_skin_weights(fullPath, dagname)
# add exportPath attribute to geometry if it doesn't exist
if not cmds.objExists(dagname + ".weightFile"):
cmds.addAttr(dagname,
longName="weightFile",
dt="string",
keyable=True)
cmds.setAttr(dagname + ".weightFile",
fullPath,
type="string")
# close the UI
self.exportSkinWeights_Win.close()
# notify user
try:
cmds.inViewMessage(amg='<hl>All Weights Have Been Exported</hl>.',
pos='midCenter',
fade=True)
except:
print("All Weights Have Been Exported.")
def buildMoveInfsUI(self):
if cmds.window("ART_MoveInfluencesWin", exists = True):
cmds.deleteUI("ART_MoveInfluencesWin", wnd = True)
#launch a UI to get the name information
self.moveInfsWin = QtWidgets.QMainWindow(self.mainUI)
#load stylesheet
styleSheetFile = utils.returnNicePath(self.toolsPath, "Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
self.style = f.read()
f.close()
self.moveInfsWin.setStyleSheet(self.style)
#size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed)
#create the main widget
self.moveInfsWin_mainWidget = QtWidgets.QWidget()
self.moveInfsWin.setCentralWidget(self.moveInfsWin_mainWidget)
#set qt object name
self.moveInfsWin.setObjectName("ART_MoveInfluencesWin")
self.moveInfsWin.setWindowTitle("Move Influences")
#create the mainLayout for the rig creator UI
self.moveInfsWin_mainLayout = QtWidgets.QVBoxLayout(self.moveInfsWin_mainWidget)
self.moveInfsWin_mainLayout.setContentsMargins(0, 0, 0, 0)
self.moveInfsWin.resize(300, 100)
self.moveInfsWin.setSizePolicy(mainSizePolicy)
self.moveInfsWin.setMinimumSize(QtCore.QSize( 300, 100 ))
self.moveInfsWin.setMaximumSize(QtCore.QSize( 300, 100 ))
#create the background image
self.moveInfsWin_frame = QtWidgets.QFrame()
self.moveInfsWin_mainLayout.addWidget(self.moveInfsWin_frame)
#layout for the widgets
self.moveInfsWin_widgetLayout = QtWidgets.QVBoxLayout(self.moveInfsWin_frame)
#layout for the combo boxes
self.moveInfsWin_comboBoxLayout = QtWidgets.QHBoxLayout()
self.moveInfsWin_widgetLayout.addLayout(self.moveInfsWin_comboBoxLayout)
#combo boxes
self.moveFromComboBox = QtWidgets.QComboBox()
self.moveInfsWin_comboBoxLayout.addWidget(self.moveFromComboBox)
label = QtWidgets.QLabel(" --------> ")
label.setMaximumWidth(40)
label.setAlignment(QtCore.Qt.AlignCenter)
self.moveInfsWin_comboBoxLayout.addWidget(label)
self.moveToComboBox = QtWidgets.QComboBox()
self.moveInfsWin_comboBoxLayout.addWidget(self.moveToComboBox)
#process button
self.moveInfsGoBtn = QtWidgets.QPushButton("Move Influences")
self.moveInfsGoBtn.setObjectName("blueButton")
self.moveInfsWin_widgetLayout.addWidget(self.moveInfsGoBtn)
self.moveInfsGoBtn.setMinimumHeight(40)
self.moveInfsGoBtn.clicked.connect(partial(self.moveInfluences))
#populate the lists
status = self.findInfluencesOnSelection()
if not status:
return
#show the window
self.moveInfsWin.show()
def fbxImport(self):
try:
#Maya 2015 has one click dependency on FBX. super annoying
cmds.loadPlugin("OneClick.mll")
except:
pass
#get the file path from the UI
filePath = self.fbxFilePath.text()
if not os.path.exists(filePath):
cmds.warning("No such file exists")
return
#stripping namespace
if self.stripNamespace.isChecked():
#open maya standalone
mayaPath = None
for path in sys.path:
if path.find("bin") != -1:
if path.find("bin" + os.sep) == -1:
mayaPath = utils.returnFriendlyPath(
os.path.join(path, "mayapy.exe"))
#error checking
if mayaPath == None:
try:
msg = interfaceUtils.DialogMessage(
"Error", "Unable to locate mayapy.exe", [], 0)
msg.show()
except:
cmds.warning("Unable to locate mayapy.exe.")
return
scriptPath = utils.returnNicePath(
self.scriptPath, "System/ART_StripFbxNamespace.py")
#run a subprocess, opening mayapy/mayastandlone, running our stripNameSpace script
maya = subprocess.Popen(mayaPath + ' ' + scriptPath + ' ' +
filePath,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
out = maya.stdout.read()
err = maya.stderr.read()
print out
#get the current character
character = self.fbxCharacterCombo.currentText()
#duplicate the character's root
if cmds.objExists("root"):
cmds.warning(
"There is already a skeleton in the scene with the name \"root\". Aborting"
)
return
newSkeleton = cmds.duplicate(character + ":root")
cmds.select(newSkeleton)
cmds.delete(constraints=True)
#go through each module in list, find import method, and setup constraints accordingly
moduleItems = []
for i in range(self.fbxModuleList.count()):
item = self.fbxModuleList.item(i)
itemWidget = self.fbxModuleList.itemWidget(item)
itemModule = itemWidget.property("module")
children = itemWidget.children()
for child in children:
if type(child) == QtWidgets.QComboBox:
importMethod = child.currentText()
moduleItems.append([itemModule, importMethod])
controls = []
postModules = []
#setup the constraints
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)
#list of modules that have post bake operations needed
specialModules = ["ART_Leg_Standard"]
#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 + ":"
#run the module's pre import function
moduleInst.importFBX_pre(each[1], character)
if modType in specialModules:
postModules.append([each[1], character, moduleInst])
returnControls = moduleInst.importFBX(each[1], character)
if returnControls != None:
controls.extend(returnControls)
#ensure that the scene is in 30fps
cmds.currentUnit(time='ntsc')
cmds.playbackOptions(min=0,
max=100,
animationStartTime=0,
animationEndTime=100)
cmds.currentTime(0)
#import the FBX file
string = "FBXImportMode -v \"exmerge\";"
string += "FBXImport -file \"" + filePath + "\""
string += "FBXImportFillTimeline -v true"
mel.eval(string)
#ensure we're on the base layer
animLayers = cmds.ls(type="animLayer")
if animLayers != []:
for layer in animLayers:
cmds.animLayer(layer, edit=True, selected=False)
cmds.animLayer("BaseAnimation",
edit=True,
selected=True,
preferred=True)
#snap timeline to length of imported animation
cmds.select("root", hi=True)
firstFrame = cmds.findKeyframe(cmds.ls(sl=True), which='first')
lastFrame = cmds.findKeyframe(cmds.ls(sl=True), which='last')
if lastFrame == firstFrame:
lastFrame = lastFrame + 1
cmds.playbackOptions(min=firstFrame,
max=lastFrame,
animationStartTime=firstFrame,
animationEndTime=lastFrame)
#BAKE!
cmds.select(controls)
cmds.bakeResults(simulation=True, t=(firstFrame, lastFrame))
#Post Modules: Modules that have post-bake operations needing to be done
for each in postModules:
method = each[0]
character = each[1]
inst = each[2]
inst.importFBX_post(method, character)
#Clean up (delete duplicate skeleton)
cmds.delete("root")
#Look at frame offset, and offset animation based on that
frameOffset = self.frameOffsetField.value()
cmds.select(controls)
cmds.keyframe(timeChange=frameOffset, r=True)
firstFrame = cmds.findKeyframe(which='first')
lastFrame = cmds.findKeyframe(which='last')
cmds.playbackOptions(min=firstFrame,
max=lastFrame,
animationStartTime=firstFrame,
animationEndTime=lastFrame)
def buildSettingsUi(self):
#fonts
font = QtGui.QFont()
font.setPointSize(10)
font.setBold(True)
fontSmall = QtGui.QFont()
fontSmall.setPointSize(9)
fontSmall.setBold(True)
#images
frameBackground = os.path.normcase(
os.path.join(self.iconsPath, "System/field_background.png"))
if frameBackground.partition("\\")[2] != "":
frameBackground = frameBackground.replace("\\", "/")
imageBkgrd = os.path.normcase(
os.path.join(self.iconsPath, "System/toolbar_background.png"))
if imageBkgrd.partition("\\")[2] != "":
imageBkgrd = imageBkgrd.replace("\\", "/")
imageBtnBkrd = os.path.normcase(
os.path.join(self.iconsPath, "System/blue_field_background.png"))
if imageBtnBkrd.partition("\\")[2] != "":
imageBtnBkrd = imageBtnBkrd.replace("\\", "/")
#size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Fixed)
#create the main widget
self.mainWidget = QtWidgets.QWidget()
self.mainWidget.setStyleSheet(
"background-color: rgb(0, 0, 0);, color: rgb(0,0,0);")
self.setCentralWidget(self.mainWidget)
#set qt object name
self.setObjectName(windowObject)
self.setWindowTitle(windowTitle)
self.setAttribute(QtCore.Qt.WA_DeleteOnClose)
#create the mainLayout for the rig creator UI
self.layout = QtWidgets.QVBoxLayout(self.mainWidget)
self.resize(600, 260)
self.setSizePolicy(mainSizePolicy)
self.setMinimumSize(QtCore.QSize(600, 260))
self.setMaximumSize(QtCore.QSize(600, 260))
#create the QFrame
self.frame = QtWidgets.QFrame()
self.layout.addWidget(self.frame)
self.widgetLayout = QtWidgets.QVBoxLayout(self.frame)
#info page styling
self.frame.setStyleSheet("background-image: url(" + imageBkgrd + ");")
#MayaTools/Core : Sccipts, icons, jointmover, etc
#MayaTools/Projects: actual project files (animation rigs, thumbnails, poses, etc)
#location
self.locationLayout = QtWidgets.QHBoxLayout()
self.widgetLayout.addLayout(self.locationLayout)
#location -> label
label = QtWidgets.QLabel("Tools Location: ")
self.locationLayout.addWidget(label)
label.setFont(font)
label.setMinimumWidth(150)
#location -> line edit
path = utils.returnFriendlyPath(self.toolsPath)
self.locationPath = QtWidgets.QLineEdit(path)
self.locationLayout.addWidget(self.locationPath)
self.locationPath.setStyleSheet(
"background-image: url(" + frameBackground +
"); background-color: rgb(25,175,255);")
self.locationPath.setMinimumHeight(35)
#location -> browse button
self.locationBrowse = QtWidgets.QPushButton()
self.locationLayout.addWidget(self.locationBrowse)
self.locationBrowse.setMinimumSize(35, 35)
self.locationBrowse.setMaximumSize(35, 35)
btnBackground = utils.returnNicePath(self.iconsPath,
"System/fileBrowse.png")
self.locationBrowse.setStyleSheet("background-image: url(" +
btnBackground + ");")
self.locationBrowse.clicked.connect(
partial(self.browse, self.locationPath))
#scripts folder
self.scriptsLayout = QtWidgets.QHBoxLayout()
self.widgetLayout.addLayout(self.scriptsLayout)
#scripts -> label
label = QtWidgets.QLabel("Scripts: ")
self.scriptsLayout.addWidget(label)
label.setFont(fontSmall)
label.setMinimumWidth(150)
#scripts -> line edit
path = utils.returnFriendlyPath(self.scriptPath)
self.scriptsPath = QtWidgets.QLineEdit(path)
self.scriptsLayout.addWidget(self.scriptsPath)
self.scriptsPath.setStyleSheet("background-image: url(" +
frameBackground +
"); background-color: rgb(25,175,255);")
self.scriptsPath.setMinimumHeight(35)
#scripts -> browse button
self.scriptsBrowse = QtWidgets.QPushButton()
self.scriptsLayout.addWidget(self.scriptsBrowse)
self.scriptsBrowse.setMinimumSize(35, 35)
self.scriptsBrowse.setMaximumSize(35, 35)
self.scriptsBrowse.setStyleSheet("background-image: url(" +
btnBackground + ");")
self.scriptsBrowse.clicked.connect(
partial(self.browse, self.scriptsPath))
#icons folder
self.iconsLayout = QtWidgets.QHBoxLayout()
self.widgetLayout.addLayout(self.iconsLayout)
#icons -> label
label = QtWidgets.QLabel("Icons: ")
self.iconsLayout.addWidget(label)
label.setFont(fontSmall)
label.setMinimumWidth(150)
#icons -> line edit
path = utils.returnFriendlyPath(self.iconsPath)
self.iconPath = QtWidgets.QLineEdit(path)
self.iconsLayout.addWidget(self.iconPath)
self.iconPath.setStyleSheet("background-image: url(" +
frameBackground +
"); background-color: rgb(25,175,255);")
self.iconPath.setMinimumHeight(35)
#icons -> browse button
self.iconsBrowse = QtWidgets.QPushButton()
self.iconsLayout.addWidget(self.iconsBrowse)
self.iconsBrowse.setMinimumSize(35, 35)
self.iconsBrowse.setMaximumSize(35, 35)
self.iconsBrowse.setStyleSheet("background-image: url(" +
btnBackground + ");")
self.iconsBrowse.clicked.connect(partial(self.browse, self.iconsPath))
#projects folder
self.projectsLayout = QtWidgets.QHBoxLayout()
self.widgetLayout.addLayout(self.projectsLayout)
#projects -> label
label = QtWidgets.QLabel("Projects: ")
self.projectsLayout.addWidget(label)
label.setFont(fontSmall)
label.setMinimumWidth(150)
#projects -> line edit
path = utils.returnFriendlyPath(self.projPath)
self.projectsPath = QtWidgets.QLineEdit(path)
self.projectsLayout.addWidget(self.projectsPath)
self.projectsPath.setStyleSheet(
"background-image: url(" + frameBackground +
"); background-color: rgb(25,175,255);")
self.projectsPath.setMinimumHeight(35)
#projects -> browse button
self.projectsBrowse = QtWidgets.QPushButton()
self.projectsLayout.addWidget(self.projectsBrowse)
self.projectsBrowse.setMinimumSize(35, 35)
self.projectsBrowse.setMaximumSize(35, 35)
self.projectsBrowse.setStyleSheet("background-image: url(" +
btnBackground + ");")
self.projectsBrowse.clicked.connect(
partial(self.browse, self.projectsPath))
#Save button
self.saveChangesBtn = QtWidgets.QPushButton("Save Changes")
self.widgetLayout.addWidget(self.saveChangesBtn)
self.saveChangesBtn.setFont(font)
self.saveChangesBtn.setMinimumHeight(35)
self.saveChangesBtn.setStyleSheet(
"background-image: url(" + imageBtnBkrd +
");background-color: rgb(25, 175, 255);")
self.saveChangesBtn.clicked.connect(partial(self.saveSettings))
def buildUI(self):
if cmds.window("pyART_ResetModulesWIN", exists=True):
cmds.deleteUI("pyART_ResetModulesWIN", 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)
#create the mainLayout
self.mainLayout = QtWidgets.QVBoxLayout(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)
self.mainWin.setMinimumSize(QtCore.QSize(400, 250))
self.mainWin.setMaximumSize(QtCore.QSize(400, 250))
self.mainWin.resize(400, 250)
#set qt object name
self.mainWin.setObjectName("pyART_ResetModulesWIN")
self.mainWin.setWindowTitle("Reset Rig Controls")
self.layout = QtWidgets.QHBoxLayout()
self.mainLayout.addLayout(self.layout)
#LEFT SIDE
#list of modules
self.moduleList = QtWidgets.QListWidget()
self.moduleList.setMinimumSize(QtCore.QSize(180, 230))
self.moduleList.setMaximumSize(QtCore.QSize(180, 230))
self.layout.addWidget(self.moduleList)
self.moduleList.setSelectionMode(
QtWidgets.QAbstractItemView.MultiSelection)
#RIGHT SIDE
self.rightLayout = QtWidgets.QVBoxLayout()
self.layout.addLayout(self.rightLayout)
#character combo
self.characterCombo = QtWidgets.QComboBox()
self.rightLayout.addWidget(self.characterCombo)
self.characterCombo.setMinimumSize(QtCore.QSize(180, 60))
self.characterCombo.setMaximumSize(QtCore.QSize(180, 60))
self.characterCombo.setIconSize(QtCore.QSize(50, 50))
self.characterCombo.currentIndexChanged.connect(
partial(self.findCharacterModules))
#buttons
self.selAllBtn = QtWidgets.QPushButton("Select All")
self.selAllBtn.setMinimumWidth(180)
self.selAllBtn.setMaximumWidth(180)
self.rightLayout.addWidget(self.selAllBtn)
self.selAllBtn.setObjectName("blueButton")
self.selAllBtn.clicked.connect(partial(self.selectAllModules))
self.clearSelBtn = QtWidgets.QPushButton("Clear Selection")
self.clearSelBtn.setMinimumWidth(180)
self.clearSelBtn.setMaximumWidth(180)
self.rightLayout.addWidget(self.clearSelBtn)
self.clearSelBtn.setObjectName("blueButton")
self.clearSelBtn.clicked.connect(partial(self.clearModuleSelection))
self.rightLayout.addSpacerItem(
QtWidgets.QSpacerItem(0, 0, QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Expanding))
#reset transforms button
self.resetXformsBtn = QtWidgets.QPushButton("Reset Transformations")
self.resetXformsBtn.setMinimumWidth(180)
self.resetXformsBtn.setMaximumWidth(180)
self.rightLayout.addWidget(self.resetXformsBtn)
self.resetXformsBtn.setObjectName("blueButton")
self.resetXformsBtn.clicked.connect(partial(self.reset))
#show window
self.mainWin.show()
#populate UI
self.findCharacters()
self.findCharacterModules()
def buildUI(self):
if cmds.window("ART_PinModulesWin", exists = True):
cmds.deleteUI("ART_PinModulesWin", wnd = True)
#launch a UI to get the name information
self.window = QtWidgets.QMainWindow(self.mainUI)
#load stylesheet
styleSheetFile = utils.returnNicePath(self.toolsPath, "Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
self.style = f.read()
f.close()
self.window.setStyleSheet(self.style)
#size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed)
#create the main widget
self.mainWidget = QtWidgets.QWidget()
self.window.setCentralWidget(self.mainWidget)
#set qt object name
self.window.setObjectName("ART_PinModulesWin")
self.window.setWindowTitle("Pin Modules")
#create the mainLayout for the rig creator UI
self.mainLayout = QtWidgets.QVBoxLayout(self.mainWidget)
self.mainLayout.setContentsMargins(0, 0, 0, 0)
self.window.resize(400, 250)
self.window.setSizePolicy(mainSizePolicy)
self.window.setMinimumSize(QtCore.QSize( 400, 250 ))
self.window.setMaximumSize(QtCore.QSize( 400, 250 ))
#create the background image
self.frame = QtWidgets.QFrame()
self.mainLayout.addWidget(self.frame)
#create the layout for the widgets
self.widgetLayout = QtWidgets.QHBoxLayout(self.frame)
self.widgetLayout.setContentsMargins(5, 5, 5, 5)
#left side == list of modules in scene. for each item in list, will do something similar to aim mode, where we will toggle an icon for pin state
self.moduleList = QtWidgets.QListWidget()
self.widgetLayout.addWidget(self.moduleList)
self.moduleList.setMinimumSize(QtCore.QSize( 265, 200 ))
self.moduleList.setMaximumSize(QtCore.QSize( 265, 200 ))
self.moduleList.setSelectionMode(QtWidgets.QAbstractItemView.MultiSelection)
self.moduleList.setSpacing(3)
#right side layout == select all, clear selection, Pin Selected buttons
self.buttonLayout = QtWidgets.QVBoxLayout()
self.widgetLayout.addLayout(self.buttonLayout)
self.buttonLayout.setContentsMargins(5, 20, 5, 20)
#add the selection buttons
self.selectAllButton = QtWidgets.QPushButton("Select All")
self.selectAllButton.setMinimumSize(QtCore.QSize( 115, 25 ))
self.selectAllButton.setMaximumSize(QtCore.QSize( 115, 25 ))
self.buttonLayout.addWidget(self.selectAllButton)
self.selectAllButton.clicked.connect(self.moduleList.selectAll)
self.selectAllButton.setObjectName("blueButton")
self.selectNoneButton = QtWidgets.QPushButton("Clear Selection")
self.selectNoneButton.setMinimumSize(QtCore.QSize( 115, 25 ))
self.selectNoneButton.setMaximumSize(QtCore.QSize( 115, 25 ))
self.buttonLayout.addWidget(self.selectNoneButton)
self.selectNoneButton.clicked.connect(self.moduleList.clearSelection)
self.selectNoneButton.setObjectName("blueButton")
#spacer
spacerItem = QtWidgets.QSpacerItem(20, 80, QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Expanding)
self.buttonLayout.addItem(spacerItem)
#add the buttons for reset settings and reset transforms
self.pinBtn = QtWidgets.QPushButton("Pin Selected")
self.pinBtn.setMinimumSize(QtCore.QSize( 115, 25 ))
self.pinBtn.setMaximumSize(QtCore.QSize( 115, 25 ))
self.buttonLayout.addWidget(self.pinBtn)
self.pinBtn.setToolTip("Pin the selected modules so that parent module movements do not effect the pinned module")
self.pinBtn.clicked.connect(partial(self.toggleLock, True))
self.pinBtn.setObjectName("blueButton")
self.unpinBtn = QtWidgets.QPushButton("Unpin Selected")
self.unpinBtn.setMinimumSize(QtCore.QSize( 115, 25 ))
self.unpinBtn.setMaximumSize(QtCore.QSize( 115, 25 ))
self.buttonLayout.addWidget(self.unpinBtn)
self.unpinBtn.setToolTip("Unpin modules to resume normal module behavior")
self.unpinBtn.clicked.connect(partial(self.toggleLock, False))
self.unpinBtn.setObjectName("blueButton")
#populate the list widget
modules = utils.returnRigModules()
for module in modules:
#get module name
moduleName = cmds.getAttr(module + ".moduleName")
#font
headerFont = QtGui.QFont()
headerFont.setPointSize(10)
headerFont.setBold(True)
#create the listWidgetItem
icon = QtGui.QIcon(os.path.join(self.iconsPath, "System/locked.png"))
iconOff = QtGui.QIcon(os.path.join(self.iconsPath, "System/unlocked.png"))
item = QtWidgets.QListWidgetItem(iconOff, " " + moduleName)
item.setFont(headerFont)
item.setData(QtCore.Qt.UserRole, [icon, iconOff])
pinState = cmds.getAttr(module + ".pinned")
if pinState:
item.setIcon(icon)
self.moduleList.addItem(item)
#show the window
self.window.show()
def buildUI(self):
if cmds.window("pyART_SelectControlsWIN", exists=True):
cmds.deleteUI("pyART_SelectControlsWIN", 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)
# create the mainLayout
self.mainLayout = QtWidgets.QVBoxLayout(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)
self.mainWin.setMinimumSize(QtCore.QSize(400, 250))
self.mainWin.setMaximumSize(QtCore.QSize(400, 250))
self.mainWin.resize(400, 250)
# set qt object name
self.mainWin.setObjectName("pyART_SelectControlsWIN")
self.mainWin.setWindowTitle("Select Rig Controls")
self.layout = QtWidgets.QHBoxLayout()
self.mainLayout.addLayout(self.layout)
# LEFT SIDE
# list of modules
self.moduleList = QtWidgets.QListWidget()
self.moduleList.setMinimumSize(QtCore.QSize(180, 230))
self.moduleList.setMaximumSize(QtCore.QSize(180, 230))
self.layout.addWidget(self.moduleList)
self.moduleList.setSelectionMode(
QtWidgets.QAbstractItemView.MultiSelection)
self.moduleList.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
self.moduleList.customContextMenuRequested.connect(
self.createContextMenu)
# RIGHT SIDE
self.rightLayout = QtWidgets.QVBoxLayout()
self.layout.addLayout(self.rightLayout)
# character combo
self.characterCombo = QtWidgets.QComboBox()
self.rightLayout.addWidget(self.characterCombo)
self.characterCombo.setMinimumSize(QtCore.QSize(180, 60))
self.characterCombo.setMaximumSize(QtCore.QSize(180, 60))
self.characterCombo.setIconSize(QtCore.QSize(50, 50))
self.characterCombo.currentIndexChanged.connect(
partial(self.findCharacterModules))
# select options
self.fkControlsCB = QtWidgets.QCheckBox("FK Controls")
self.fkControlsCB.setChecked(True)
self.fkControlsCB.setToolTip(
"If this is checked, for the selected modules,\nall FK controls will be selected or added to the selection."
)
self.rightLayout.addWidget(self.fkControlsCB)
self.ikControlsCB = QtWidgets.QCheckBox("IK Controls")
self.ikControlsCB.setChecked(True)
self.ikControlsCB.setToolTip(
"If this is checked, for the selected modules,\nall IK controls will be selected or added to the selection."
)
self.rightLayout.addWidget(self.ikControlsCB)
self.specialControlsCB = QtWidgets.QCheckBox("Special Controls")
self.specialControlsCB.setChecked(True)
self.specialControlsCB.setToolTip(
"If this is checked, for the selected modules,\nany control that is not FK or IK (like dynamics for example),\nwill be selected or added to the selection."
)
self.rightLayout.addWidget(self.specialControlsCB)
self.selectSettingsCB = QtWidgets.QCheckBox("Include Settings")
self.selectSettingsCB.setChecked(True)
self.selectSettingsCB.setToolTip(
"If this is checked, for the selected modules,\nall settings nodes will be included in the selection."
)
self.rightLayout.addWidget(self.selectSettingsCB)
self.selectSpacesCB = QtWidgets.QCheckBox("Include Spaces")
self.selectSpacesCB.setChecked(True)
self.selectSpacesCB.setToolTip(
"If this is checked, for the selected modules,\nall space switching nodes will be included in the selection."
)
self.rightLayout.addWidget(self.selectSpacesCB)
self.rightLayout.addSpacerItem(
QtWidgets.QSpacerItem(0, 0, QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Expanding))
self.selectCtrlsBtn = QtWidgets.QPushButton("Select Controls")
self.selectCtrlsBtn.setMinimumWidth(180)
self.selectCtrlsBtn.setMaximumWidth(180)
self.rightLayout.addWidget(self.selectCtrlsBtn)
self.selectCtrlsBtn.setObjectName("blueButton")
self.selectCtrlsBtn.clicked.connect(self.selectControls)
# show window
if self.showUI:
self.mainWin.show()
# populate UI
self.findCharacters()
self.findCharacterModules()
def buildBakeOffsetsUI(self):
"""
Builds the interface, finding all modules and listing them for selection.
"""
if cmds.window("ART_BakeOffsetsWin", exists=True):
cmds.deleteUI("ART_BakeOffsetsWin", wnd=True)
# launch a UI to get the name information
self.bakeOffsetsWin = QtWidgets.QMainWindow(self.mainUI)
# load stylesheet
styleSheetFile = utils.returnNicePath(
self.toolsPath,
"Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
self.style = f.read()
f.close()
self.bakeOffsetsWin.setStyleSheet(self.style)
# size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Fixed)
# create the main widget
self.bakeOffsetsWin_mainWidget = QtWidgets.QWidget()
self.bakeOffsetsWin.setCentralWidget(self.bakeOffsetsWin_mainWidget)
# set qt object name
self.bakeOffsetsWin.setObjectName("ART_BakeOffsetsWin")
self.bakeOffsetsWin.setWindowTitle("Bake Offsets")
# create the mainLayout for the rig creator UI
self.bakeOffsetsWin_mainLayout = QtWidgets.QVBoxLayout(
self.bakeOffsetsWin_mainWidget)
self.bakeOffsetsWin_mainLayout.setContentsMargins(0, 0, 0, 0)
self.bakeOffsetsWin.resize(400, 250)
self.bakeOffsetsWin.setSizePolicy(mainSizePolicy)
self.bakeOffsetsWin.setMinimumSize(QtCore.QSize(400, 250))
self.bakeOffsetsWin.setMaximumSize(QtCore.QSize(400, 250))
# create the background
self.bakeOffsetsWin_frame = QtWidgets.QFrame()
self.bakeOffsetsWin_mainLayout.addWidget(self.bakeOffsetsWin_frame)
# create the layout for the widgets
self.bakeOffsetsWin_widgetLayout = QtWidgets.QHBoxLayout(
self.bakeOffsetsWin_frame)
self.bakeOffsetsWin_widgetLayout.setContentsMargins(5, 5, 5, 5)
# add the QListWidget Frame
self.bakeOffsetsWin_moduleListFrame = QtWidgets.QFrame()
self.bakeOffsetsWin_moduleListFrame.setMinimumSize(
QtCore.QSize(265, 200))
self.bakeOffsetsWin_moduleListFrame.setMaximumSize(
QtCore.QSize(265, 200))
self.bakeOffsetsWin_moduleListFrame.setContentsMargins(20, 0, 20, 0)
# create the list widget
self.bakeOffsetsWin_moduleList = QtWidgets.QListWidget(
self.bakeOffsetsWin_moduleListFrame)
self.bakeOffsetsWin_widgetLayout.addWidget(
self.bakeOffsetsWin_moduleListFrame)
self.bakeOffsetsWin_moduleList.setMinimumSize(QtCore.QSize(265, 200))
self.bakeOffsetsWin_moduleList.setMaximumSize(QtCore.QSize(265, 200))
self.bakeOffsetsWin_moduleList.setSelectionMode(
QtWidgets.QAbstractItemView.MultiSelection)
self.bakeOffsetsWin_moduleList.setSpacing(3)
# add the layout for the buttons
self.bakeOffsetsWin_buttonLayoutAll = QtWidgets.QVBoxLayout()
self.bakeOffsetsWin_widgetLayout.addLayout(
self.bakeOffsetsWin_buttonLayoutAll)
self.bakeOffsetsWin_buttonLayoutAll.setContentsMargins(5, 20, 5, 20)
# add the selection buttons
self.bakeOffsetsWin_selectionButtonLayout = QtWidgets.QVBoxLayout()
self.bakeOffsetsWin_buttonLayoutAll.addLayout(
self.bakeOffsetsWin_selectionButtonLayout)
self.bakeOffsetsWin_selectAllButton = QtWidgets.QPushButton(
"Select All")
self.bakeOffsetsWin_selectAllButton.setMinimumSize(
QtCore.QSize(115, 25))
self.bakeOffsetsWin_selectAllButton.setMaximumSize(
QtCore.QSize(115, 25))
self.bakeOffsetsWin_selectionButtonLayout.addWidget(
self.bakeOffsetsWin_selectAllButton)
self.bakeOffsetsWin_selectAllButton.clicked.connect(
self.bakeOffsetsWin_moduleList.selectAll)
self.bakeOffsetsWin_selectAllButton.setObjectName("blueButton")
self.bakeOffsetsWin_selectNoneButton = QtWidgets.QPushButton(
"Clear Selection")
self.bakeOffsetsWin_selectNoneButton.setMinimumSize(
QtCore.QSize(115, 25))
self.bakeOffsetsWin_selectNoneButton.setMaximumSize(
QtCore.QSize(115, 25))
self.bakeOffsetsWin_selectionButtonLayout.addWidget(
self.bakeOffsetsWin_selectNoneButton)
self.bakeOffsetsWin_selectNoneButton.clicked.connect(
self.bakeOffsetsWin_moduleList.clearSelection)
self.bakeOffsetsWin_selectNoneButton.setObjectName("blueButton")
# spacer
spacerItem = QtWidgets.QSpacerItem(20, 80,
QtWidgets.QSizePolicy.Minimum,
QtWidgets.QSizePolicy.Expanding)
self.bakeOffsetsWin_selectionButtonLayout.addItem(spacerItem)
# add the buttons for reset settings and reset transforms
self.bakeOffsetsWin_bakeOFfsetsBtn = QtWidgets.QPushButton(
"Bake Offsets")
self.bakeOffsetsWin_bakeOFfsetsBtn.setMinimumSize(QtCore.QSize(
115, 25))
self.bakeOffsetsWin_bakeOFfsetsBtn.setMaximumSize(QtCore.QSize(
115, 25))
self.bakeOffsetsWin_selectionButtonLayout.addWidget(
self.bakeOffsetsWin_bakeOFfsetsBtn)
self.bakeOffsetsWin_bakeOFfsetsBtn.setToolTip(
"Turn on Aim Mode for selected modules.")
self.bakeOffsetsWin_bakeOFfsetsBtn.clicked.connect(
partial(self.bakeOffsets))
self.bakeOffsetsWin_bakeOFfsetsBtn.setObjectName("blueButton")
# populate the list widget
modules = utils.returnRigModules()
for module in modules:
# get module name
moduleName = cmds.getAttr(module + ".moduleName")
if moduleName != "root":
self.bakeOffsetsWin_moduleList.addItem(moduleName)
# show the window
self.bakeOffsetsWin.show()
def buildUI(self):
if cmds.window("pyART_movePickerToTabWIN", exists=True):
cmds.deleteUI("pyART_movePickerToTabWIN", wnd=True)
#create the main window
self.mainWin = QtWidgets.QMainWindow(self.pickerUI)
#create the main widget
self.mainWidget = QtWidgets.QWidget()
self.mainWin.setCentralWidget(self.mainWidget)
#create the mainLayout
self.mainLayout = QtWidgets.QVBoxLayout(self.mainWidget)
self.layout = QtWidgets.QHBoxLayout()
self.mainLayout.addLayout(self.layout)
#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)
self.mainWin.setMinimumSize(QtCore.QSize(400, 400))
self.mainWin.setMaximumSize(QtCore.QSize(400, 400))
self.mainWin.resize(400, 400)
#set qt object name
self.mainWin.setObjectName("pyART_movePickerToTabWIN")
self.mainWin.setWindowTitle("Move Picker")
#create 2 columns
self.column1 = QtWidgets.QVBoxLayout()
self.layout.addLayout(self.column1)
self.column2 = QtWidgets.QVBoxLayout()
self.layout.addLayout(self.column2)
#create left side list widget, which will house the picker items
self.pickerItemsList = QtWidgets.QListWidget()
self.column1.addWidget(self.pickerItemsList)
self.pickerItemsList.setMinimumSize(180, 300)
self.pickerItemsList.setMaximumSize(180, 300)
#get the current tab index and the widget
index = self.pickerUI.characterTabs.currentIndex()
widget = self.pickerUI.characterTabs.widget(index)
#get the tab text
character = self.pickerUI.characterTabs.tabToolTip(index)
#find character nodes in the scene, and compare namespace to selected tab
characterMods = utils.returnCharacterModules()
nodeNamespace = ""
for each in characterMods:
if cmds.objExists(each + ".namespace"):
namespace = cmds.getAttr(each + ".namespace")
if namespace == character:
nodeNamespace = namespace + ":"
for module in self.modulesToAdd:
if module[2] == None:
modName = cmds.getAttr(nodeNamespace + module[0] +
".moduleName")
else:
modName = module[2]
qlistItem = QtWidgets.QListWidgetItem(modName)
qlistItem.setData(QtCore.Qt.UserRole, module[1])
self.pickerItemsList.addItem(qlistItem)
#create right side list widget, which will house the available tabs
self.tabList = QtWidgets.QListWidget()
self.column2.addWidget(self.tabList)
self.tabList.setMinimumSize(180, 300)
self.tabList.setMaximumSize(180, 300)
tabs = self.findTabs(False)
for tab in tabs:
qlistItem = QtWidgets.QListWidgetItem(tab[0])
qlistItem.setData(QtCore.Qt.UserRole, tab[1])
self.tabList.addItem(qlistItem)
#create button for move selected picker to selected tab
self.movePickerBtn = QtWidgets.QPushButton(
"Move Selected Picker To Selected Tab")
self.mainLayout.addWidget(self.movePickerBtn)
self.movePickerBtn.setObjectName("blueButton")
self.movePickerBtn.setMinimumHeight(50)
self.movePickerBtn.setMaximumHeight(50)
self.movePickerBtn.clicked.connect(self.moveToTab)
#show ui
self.mainWin.show()
def buildResetModeUI(self, mainUI):
if cmds.window("ART_ResetModeWin", exists=True):
cmds.deleteUI("ART_ResetModeWin", wnd=True)
#launch a UI to get the name information
self.resetModeWin = QtWidgets.QMainWindow(mainUI)
#load stylesheet
styleSheetFile = utils.returnNicePath(
self.toolsPath,
"Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
self.style = f.read()
f.close()
self.resetModeWin.setStyleSheet(self.style)
#size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Fixed)
#create the main widget
self.resetModeWin_mainWidget = QtWidgets.QWidget()
self.resetModeWin.setCentralWidget(self.resetModeWin_mainWidget)
#set qt object name
self.resetModeWin.setObjectName("ART_ResetModeWin")
self.resetModeWin.setWindowTitle("Reset Modules")
#create the mainLayout for the rig creator UI
self.resetModeWin_mainLayout = QtWidgets.QVBoxLayout(
self.resetModeWin_mainWidget)
self.resetModeWin_mainLayout.setContentsMargins(0, 0, 0, 0)
self.resetModeWin.resize(400, 250)
self.resetModeWin.setSizePolicy(mainSizePolicy)
self.resetModeWin.setMinimumSize(QtCore.QSize(400, 250))
self.resetModeWin.setMaximumSize(QtCore.QSize(400, 250))
#create the background image
self.resetModeWin_frame = QtWidgets.QFrame()
self.resetModeWin_mainLayout.addWidget(self.resetModeWin_frame)
#create the layout for the widgets
self.resetModeWin_widgetLayout = QtWidgets.QHBoxLayout(
self.resetModeWin_frame)
self.resetModeWin_widgetLayout.setContentsMargins(5, 5, 5, 5)
#add the QListWidget Frame
self.resetModeWin_moduleListFrame = QtWidgets.QFrame()
self.resetModeWin_moduleListFrame.setMinimumSize(QtCore.QSize(
275, 200))
self.resetModeWin_moduleListFrame.setMaximumSize(QtCore.QSize(
275, 200))
self.resetModeWin_moduleListFrame.setContentsMargins(20, 0, 20, 0)
#create the list widget
self.resetModeWin_moduleList = QtWidgets.QListWidget(
self.resetModeWin_moduleListFrame)
self.resetModeWin_widgetLayout.addWidget(
self.resetModeWin_moduleListFrame)
self.resetModeWin_moduleList.setMinimumSize(QtCore.QSize(265, 200))
self.resetModeWin_moduleList.setMaximumSize(QtCore.QSize(265, 200))
self.resetModeWin_moduleList.setSelectionMode(
QtWidgets.QAbstractItemView.MultiSelection)
self.resetModeWin_moduleList.setSpacing(3)
#add the layout for the buttons
self.resetModeWin_buttonLayoutAll = QtWidgets.QVBoxLayout()
self.resetModeWin_widgetLayout.addLayout(
self.resetModeWin_buttonLayoutAll)
self.resetModeWin_buttonLayoutAll.setContentsMargins(5, 20, 5, 20)
#button background image
image = utils.returnNicePath(self.iconsPath,
"System/blue_field_background.png")
#add the selection buttons
self.resetModeWin_selectionButtonLayout = QtWidgets.QVBoxLayout()
self.resetModeWin_buttonLayoutAll.addLayout(
self.resetModeWin_selectionButtonLayout)
self.resetModeWin_selectAllButton = QtWidgets.QPushButton("Select All")
self.resetModeWin_selectAllButton.setMinimumSize(QtCore.QSize(115, 25))
self.resetModeWin_selectAllButton.setMaximumSize(QtCore.QSize(115, 25))
self.resetModeWin_selectionButtonLayout.addWidget(
self.resetModeWin_selectAllButton)
self.resetModeWin_selectAllButton.clicked.connect(
self.resetModeWin_moduleList.selectAll)
self.resetModeWin_selectAllButton.setObjectName("blueButton")
self.resetModeWin_selectNoneButton = QtWidgets.QPushButton(
"Clear Selection")
self.resetModeWin_selectNoneButton.setMinimumSize(QtCore.QSize(
115, 25))
self.resetModeWin_selectNoneButton.setMaximumSize(QtCore.QSize(
115, 25))
self.resetModeWin_selectionButtonLayout.addWidget(
self.resetModeWin_selectNoneButton)
self.resetModeWin_selectNoneButton.clicked.connect(
self.resetModeWin_moduleList.clearSelection)
self.resetModeWin_selectNoneButton.setObjectName("blueButton")
#spacer
spacerItem = QtWidgets.QSpacerItem(20, 80,
QtWidgets.QSizePolicy.Minimum,
QtWidgets.QSizePolicy.Expanding)
self.resetModeWin_selectionButtonLayout.addItem(spacerItem)
#add the buttons for reset settings and reset transforms
self.resetModeWin_resetSettings = QtWidgets.QPushButton(
"Reset Settings")
self.resetModeWin_resetSettings.setMinimumSize(QtCore.QSize(115, 25))
self.resetModeWin_resetSettings.setMaximumSize(QtCore.QSize(115, 25))
self.resetModeWin_selectionButtonLayout.addWidget(
self.resetModeWin_resetSettings)
self.resetModeWin_resetSettings.clicked.connect(
partial(self.resetMode_resetSettings))
self.resetModeWin_resetSettings.setObjectName("blueButton")
self.resetModeWin_resetXforms = QtWidgets.QPushButton("Reset Xforms")
self.resetModeWin_resetXforms.setMinimumSize(QtCore.QSize(115, 25))
self.resetModeWin_resetXforms.setMaximumSize(QtCore.QSize(115, 25))
self.resetModeWin_selectionButtonLayout.addWidget(
self.resetModeWin_resetXforms)
self.resetModeWin_resetXforms.clicked.connect(
partial(self.resetMode_resetXformsUI))
self.resetModeWin_resetXforms.setObjectName("blueButton")
#populate the list widget
modules = utils.returnRigModules()
for module in modules:
moduleName = cmds.getAttr(module + ".moduleName")
self.resetModeWin_moduleList.addItem(moduleName)
#show the window
self.resetModeWin.show()
def buildUI(self):
if cmds.window("pyART_ImportMotionWIN", exists=True):
cmds.deleteUI("pyART_ImportMotionWIN", wnd=True)
#create the main window
self.mainWin = QtWidgets.QMainWindow(self.pickerUI)
#create the main widget
self.mainWidget = QtWidgets.QWidget()
self.mainWin.setCentralWidget(self.mainWidget)
#create the mainLayout
self.layout = QtWidgets.QVBoxLayout(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)
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_ImportMotionWIN")
self.mainWin.setWindowTitle("Import Motion")
#tabs
self.importTabs = QtWidgets.QTabWidget()
self.layout.addWidget(self.importTabs)
#style sheet
stylesheet = """
QTabBar::tab
{
background-color: qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 rgb(19,132,183), stop:1 rgb(30,30,30));
width: 100px;
padding-left: -10px;
}
QTabBar::tab:selected
{
background-color: rgb(14,100,143);
border: 2px solid black;
}
QTabBar::tab:hover
{
background: rgb(19,132,183);
}
QTabBar::tab:!selected
{
margin-top: 5px;
}
QTabWidget::pane
{
border-top: 2px solid rgb(19,132,183);
border-left: 2px solid rgb(19,132,183);
border-right: 2px solid rgb(19,132,183);
border-bottom: 2px solid rgb(19,132,183);
}
"""
self.importTabs.setStyleSheet(stylesheet)
#FBX Tab
self.fbxImportTab = QtWidgets.QWidget()
self.importTabs.addTab(self.fbxImportTab, "FBX")
#Anim Curve Tab
self.animImportTab = QtWidgets.QWidget()
self.importTabs.addTab(self.animImportTab, "Animation")
#=======================================================================
#=======================================================================
#=======================================================================
#=======================================================================
# #FBX TAB
#=======================================================================
#=======================================================================
#=======================================================================
#=======================================================================
#horizontal layout
self.fbxMainLayout = QtWidgets.QHBoxLayout(self.fbxImportTab)
#LEFT SIDE
#module list widget
self.fbxModuleList = QtWidgets.QListWidget()
self.fbxMainLayout.addWidget(self.fbxModuleList)
self.fbxModuleList.setMinimumSize(QtCore.QSize(300, 280))
self.fbxModuleList.setMaximumSize(QtCore.QSize(300, 280))
self.fbxModuleList.setSpacing(15)
self.fbxModuleList.setSelectionMode(
QtWidgets.QAbstractItemView.NoSelection)
#RIGHT SIDE
self.fbxRightLayout = QtWidgets.QVBoxLayout()
self.fbxMainLayout.addLayout(self.fbxRightLayout)
self.fbxCharacterCombo = QtWidgets.QComboBox()
self.fbxRightLayout.addWidget(self.fbxCharacterCombo)
self.fbxCharacterCombo.setMinimumSize(QtCore.QSize(250, 50))
self.fbxCharacterCombo.setMaximumSize(QtCore.QSize(250, 50))
self.fbxCharacterCombo.setIconSize(QtCore.QSize(45, 45))
self.fbxCharacterCombo.currentIndexChanged.connect(
partial(self.findCharacterModules))
self.fbxPathLayout = QtWidgets.QHBoxLayout()
self.fbxRightLayout.addLayout(self.fbxPathLayout)
self.fbxFilePath = QtWidgets.QLineEdit()
self.fbxFilePath.setMinimumWidth(210)
self.fbxFilePath.setMaximumWidth(210)
self.fbxPathLayout.addWidget(self.fbxFilePath)
self.fbxFilePath.setPlaceholderText("fbx file..")
browseBtn = QtWidgets.QPushButton()
browseBtn.setMinimumSize(25, 25)
browseBtn.setMaximumSize(25, 25)
self.fbxPathLayout.addWidget(browseBtn)
icon = QtGui.QIcon(
utils.returnNicePath(self.iconsPath, "System/fileBrowse.png"))
browseBtn.setIconSize(QtCore.QSize(25, 25))
browseBtn.setIcon(icon)
browseBtn.clicked.connect(self.fbxFileBrowse)
self.frameOffsetLayout = QtWidgets.QHBoxLayout()
self.fbxRightLayout.addLayout(self.frameOffsetLayout)
frameOffset = QtWidgets.QLabel("Frame Offset:")
frameOffset.setStyleSheet("background: transparent; font: bold;")
self.frameOffsetLayout.addWidget(frameOffset)
self.frameOffsetField = QtWidgets.QSpinBox()
self.frameOffsetField.setButtonSymbols(
QtWidgets.QAbstractSpinBox.NoButtons)
self.frameOffsetField.setRange(-1000, 10000)
self.frameOffsetLayout.addWidget(self.frameOffsetField)
#option to strip namespace
self.stripNamespace = QtWidgets.QCheckBox("Strip Incoming Namespace")
self.stripNamespace.setToolTip(
"If the incoming FBX has a namespace, checking this\noption will strip that namespace upon import"
)
self.stripNamespace.setChecked(True)
self.fbxRightLayout.addWidget(self.stripNamespace)
#Save/Load Settings
saveLoadLayout = QtWidgets.QHBoxLayout()
self.fbxRightLayout.addLayout(saveLoadLayout)
saveSettingsBtn = QtWidgets.QPushButton("Save Settings")
saveSettingsBtn.setMinimumSize(120, 30)
saveSettingsBtn.setMaximumSize(120, 30)
icon = QtGui.QIcon(
utils.returnNicePath(self.iconsPath, "System/save.png"))
saveSettingsBtn.setIconSize(QtCore.QSize(25, 25))
saveSettingsBtn.setIcon(icon)
saveLoadLayout.addWidget(saveSettingsBtn)
saveSettingsBtn.setObjectName("blueButton")
saveSettingsBtn.setToolTip("Save out module import settings")
saveSettingsBtn.clicked.connect(self.saveSettings)
loadSettingsBtn = QtWidgets.QPushButton("Load Settings")
loadSettingsBtn.setMinimumSize(120, 30)
loadSettingsBtn.setMaximumSize(120, 30)
icon = QtGui.QIcon(
utils.returnNicePath(self.iconsPath, "System/load.png"))
loadSettingsBtn.setIconSize(QtCore.QSize(25, 25))
loadSettingsBtn.setIcon(icon)
saveLoadLayout.addWidget(loadSettingsBtn)
loadSettingsBtn.setObjectName("blueButton")
loadSettingsBtn.setToolTip("Load and set module import settings")
loadSettingsBtn.clicked.connect(self.loadSettings)
#SPACER!
self.fbxRightLayout.addSpacerItem(
QtWidgets.QSpacerItem(0, 0, QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Expanding))
#Button
self.importFBXbutton = QtWidgets.QPushButton("Import")
self.fbxRightLayout.addWidget(self.importFBXbutton)
self.importFBXbutton.setObjectName("blueButton")
self.importFBXbutton.setMinimumHeight(50)
self.importFBXbutton.clicked.connect(self.fbxImport)
#show window
self.mainWin.show()
#populate UI
self.findCharacters()
self.findCharacterModules()
def buildUI(self):
"""
Build the UI, listing all modules in the scene that make up the asset for the user to select and build rigs
for the selected.
"""
# create the main window
self.mainWin = QtWidgets.QMainWindow(self.mainUI)
# load stylesheet
styleSheetFile = utils.returnNicePath(
self.toolsPath,
"Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
self.style = f.read()
f.close()
self.mainWin.setStyleSheet(self.style)
# create the main widget
self.mainWidget = QtWidgets.QWidget()
self.mainWin.setCentralWidget(self.mainWidget)
# set qt object name
self.mainWin.setObjectName("ART_DebugRigsWin")
self.mainWin.setWindowTitle("Build Rigs")
# font
headerFont = QtGui.QFont()
headerFont.setPointSize(8)
headerFont.setBold(True)
# set size policy
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Fixed)
# create the mainLayout for the rig creator UI
self.layout = QtWidgets.QVBoxLayout(self.mainWidget)
self.mainWin.resize(400, 300)
self.mainWin.setSizePolicy(mainSizePolicy)
self.mainWin.setMinimumSize(QtCore.QSize(400, 300))
self.mainWin.setMaximumSize(QtCore.QSize(400, 300))
# create the QFrame for this page
self.background = QtWidgets.QFrame()
self.background.setObjectName("mid")
self.layout.addWidget(self.background)
self.mainLayout = QtWidgets.QHBoxLayout(self.background)
# create the list on the left and add the modules to the list
self.moduleList = QtWidgets.QListWidget()
self.mainLayout.addWidget(self.moduleList)
for mod in self.mainUI.moduleInstances:
item = QtWidgets.QListWidgetItem(mod.name)
item.setData(QtCore.Qt.UserRole, mod)
self.moduleList.addItem(item)
# create our buttons on the right
self.rightLayout = QtWidgets.QVBoxLayout()
self.mainLayout.addLayout(self.rightLayout)
infoText = "This tool is only for testing rigs in development. "
infoText += "It will leave behind nodes in your scene that you do NOT want to publish with. "
infoText += "When using this tool, it is advised to open a clean scene to publish your final asset."
self.info = QtWidgets.QLabel()
self.rightLayout.addWidget(self.info)
self.info.setWordWrap(True)
self.info.setMinimumSize(150, 125)
self.info.setMaximumSize(150, 125)
self.info.setText(infoText)
self.rightLayout.addSpacerItem(
QtWidgets.QSpacerItem(0, 200, QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Expanding))
self.buildButton = QtWidgets.QPushButton("Build Rigs For Selected")
self.buildButton.setObjectName("blueButton")
self.rightLayout.addWidget(self.buildButton)
self.buildButton.setMinimumSize(150, 40)
self.buildButton.setMaximumSize(150, 40)
self.buildButton.clicked.connect(partial(self.buildRigs))
self.deleteButton = QtWidgets.QPushButton("Remove Selected Rigs")
self.deleteButton.setObjectName("blueButton")
self.rightLayout.addWidget(self.deleteButton)
self.deleteButton.setMinimumSize(150, 40)
self.deleteButton.setMaximumSize(150, 40)
self.deleteButton.clicked.connect(partial(self.deleteRig))
self.closeButton = QtWidgets.QPushButton("Close")
self.closeButton.setObjectName("blueButton")
self.rightLayout.addWidget(self.closeButton)
self.closeButton.setMinimumSize(150, 40)
self.closeButton.setMaximumSize(150, 40)
self.closeButton.clicked.connect(partial(self.close))
self.mainWin.show()
def buildUI(self):
"""
Builds the interface, which doesn't really have any user-interaction, but is there to display information
about the progress of the rig build. There are two QProgressBars that show current module build progress and
total build progress, then a QTextEdit which outputs information about what the build process is currently
working on.
After the interface is built, it sets the rig pose on the joints of each module.
"""
# create the main window
self.mainWin = QtWidgets.QMainWindow(interfaceUtils.getMainWindow())
self.mainWin.setStyleSheet("background-color: rgb(0, 0, 0);, color: rgb(0,0,0);")
styleSheetFile = utils.returnNicePath(self.toolsPath, "Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
self.style = f.read()
f.close()
self.mainWin.setStyleSheet(self.style)
# create the main widget
self.mainWidget = QtWidgets.QWidget()
self.mainWin.setCentralWidget(self.mainWidget)
# set qt object name
self.mainWin.setObjectName("ART_BuildProgressWin")
self.mainWin.setWindowTitle("Build Progress")
# font
headerFont = QtGui.QFont()
headerFont.setPointSize(8)
headerFont.setBold(True)
# set size policy
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed)
# create the mainLayout for the rig creator UI
self.layout = QtWidgets.QVBoxLayout(self.mainWidget)
self.mainWin.resize(500, 300)
self.mainWin.setSizePolicy(mainSizePolicy)
self.mainWin.setMinimumSize(QtCore.QSize(500, 300))
self.mainWin.setMaximumSize(QtCore.QSize(500, 300))
# create the QFrame for this page
self.background = QtWidgets.QFrame()
self.layout.addWidget(self.background)
self.mainLayout = QtWidgets.QVBoxLayout(self.background)
self.background.setObjectName("epic")
# build the progress bars:
self.currentTask = QtWidgets.QProgressBar()
self.mainLayout.addWidget(self.currentTask)
self.currentTask.setRange(0, 100)
self.currentTask.setTextVisible(True)
self.currentTask.setValue(0)
self.totalProgress = QtWidgets.QProgressBar()
self.mainLayout.addWidget(self.totalProgress)
self.totalProgress.setFormat("Total Progress..")
self.totalProgress.setRange(0, 12)
self.totalProgress.setTextVisible(True)
self.totalProgress.setValue(0)
# detailed information
self.infoText = QtWidgets.QTextEdit()
self.mainLayout.addWidget(self.infoText)
self.infoText.setMinimumHeight(200)
self.infoText.setMaximumHeight(200)
self.infoText.setText("Starting Build Process..")
self.infoText.setReadOnly(True)
# show the window
self.mainWin.show()
# start build
self.setRigPose()
def buildInterface(self):
"""
Builds the interface for the tool, finding all joints that compose the asset, comparing them to joints in the
skinCluster, then separating the initial list into joints in the cluster, and joints not in the cluster.
"""
if cmds.window("ART_addRemoveInfsWin", exists=True):
cmds.deleteUI("ART_addRemoveInfsWin", wnd=True)
# launch a UI to get the name information
self.addRemoveInfsWin = QtWidgets.QMainWindow(self.mainUI)
# size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Fixed)
# create the main widget
self.addRemoveInfsWin_mainWidget = QtWidgets.QWidget()
self.addRemoveInfsWin.setCentralWidget(
self.addRemoveInfsWin_mainWidget)
# load stylesheet
styleSheetFile = utils.returnNicePath(
self.toolsPath,
"Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
self.style = f.read()
f.close()
# set qt object name
self.addRemoveInfsWin.setObjectName("ART_addRemoveInfsWin")
self.addRemoveInfsWin.setWindowTitle("Add/Remove Influences")
# create the mainLayout for the ui
self.addRemoveInfsWin_mainLayout = QtWidgets.QVBoxLayout(
self.addRemoveInfsWin_mainWidget)
self.addRemoveInfsWin_mainLayout.setContentsMargins(5, 5, 5, 5)
self.addRemoveInfsWin.resize(300, 450)
self.addRemoveInfsWin.setSizePolicy(mainSizePolicy)
self.addRemoveInfsWin.setMinimumSize(QtCore.QSize(300, 450))
self.addRemoveInfsWin.setMaximumSize(QtCore.QSize(300, 450))
# create the background image
self.addRemoveInfsWin_frame = QtWidgets.QFrame()
self.addRemoveInfsWin_mainLayout.addWidget(self.addRemoveInfsWin_frame)
self.addRemoveInfsWin_frame.setObjectName("dark")
# create the main layout for the widgets
self.addRemoveInfsWin_widgetLayout = QtWidgets.QHBoxLayout(
self.addRemoveInfsWin_frame)
# two layouts needed for the widget layout. left side = vertical layout for filters,
# search, and list. right layout = vertical layout for buttons
self.addRemoveInfsWin_leftSideLayout = QtWidgets.QVBoxLayout()
self.addRemoveInfsWin_widgetLayout.addLayout(
self.addRemoveInfsWin_leftSideLayout)
self.addRemoveInfsWin_rightSideLayout = QtWidgets.QVBoxLayout()
self.addRemoveInfsWin_widgetLayout.addLayout(
self.addRemoveInfsWin_rightSideLayout)
# left side: filters, search, list
self.addRemoveInfsWin_filters = QtWidgets.QComboBox()
self.addRemoveInfsWin_leftSideLayout.addWidget(
self.addRemoveInfsWin_filters)
self.addRemoveInfsWin_filters.addItem("Show Influences In Skin")
self.addRemoveInfsWin_filters.addItem("Show Influences Not In Skin")
self.addRemoveInfsWin_filters.currentIndexChanged.connect(
partial(self.addOrRemoveInfs_ShowInfsFilter))
self.addRemoveInfsWin_search = QtWidgets.QLineEdit()
self.addRemoveInfsWin_leftSideLayout.addWidget(
self.addRemoveInfsWin_search)
self.addRemoveInfsWin_search.setPlaceholderText("Search...")
self.addRemoveInfsWin_search.textChanged.connect(
partial(self.addOrRemoveInfs_Search))
self.addRemoveInfsWin_infList = QtWidgets.QListWidget()
self.addRemoveInfsWin_leftSideLayout.addWidget(
self.addRemoveInfsWin_infList)
self.addRemoveInfsWin_infList.setSelectionMode(
QtWidgets.QAbstractItemView.ExtendedSelection)
# right side: add button, remove button, prune weights, remove unused button
font = QtGui.QFont()
font.setPointSize(8)
font.setBold(True)
self.addRemoveInfsWin_refreshSelBtn = QtWidgets.QPushButton("Refresh")
self.addRemoveInfsWin_rightSideLayout.addWidget(
self.addRemoveInfsWin_refreshSelBtn)
self.addRemoveInfsWin_refreshSelBtn.setMinimumSize(110, 35)
self.addRemoveInfsWin_refreshSelBtn.setMaximumSize(110, 35)
self.addRemoveInfsWin_refreshSelBtn.setFont(font)
self.addRemoveInfsWin_refreshSelBtn.clicked.connect(
partial(self.addOrRemoveInfs_RefreshSelection))
self.addRemoveInfsWin_refreshSelBtn.setObjectName("blueButton")
self.addRemoveInfsWin_rightSideLayout.addSpacerItem(
QtWidgets.QSpacerItem(100, 300, QtWidgets.QSizePolicy.Expanding,
QtWidgets.QSizePolicy.Fixed))
self.addRemoveInfsWin_addInfBtn = QtWidgets.QPushButton("Add")
self.addRemoveInfsWin_rightSideLayout.addWidget(
self.addRemoveInfsWin_addInfBtn)
self.addRemoveInfsWin_addInfBtn.setMinimumSize(110, 35)
self.addRemoveInfsWin_addInfBtn.setMaximumSize(110, 35)
self.addRemoveInfsWin_addInfBtn.setFont(font)
self.addRemoveInfsWin_addInfBtn.clicked.connect(
partial(self.addOrRemoveInfs_addInf, True, False))
self.addRemoveInfsWin_addInfBtn.setObjectName("blueButton")
self.addRemoveInfsWin_removeInfBtn = QtWidgets.QPushButton("Remove")
self.addRemoveInfsWin_rightSideLayout.addWidget(
self.addRemoveInfsWin_removeInfBtn)
self.addRemoveInfsWin_removeInfBtn.setMinimumSize(110, 35)
self.addRemoveInfsWin_removeInfBtn.setMaximumSize(110, 35)
self.addRemoveInfsWin_removeInfBtn.setFont(font)
self.addRemoveInfsWin_removeInfBtn.clicked.connect(
partial(self.addOrRemoveInfs_addInf, False, False))
self.addRemoveInfsWin_removeInfBtn.setObjectName("blueButton")
self.addRemoveInfsWin_removeUnusedInfBtn = QtWidgets.QPushButton(
"Remove Unused")
self.addRemoveInfsWin_rightSideLayout.addWidget(
self.addRemoveInfsWin_removeUnusedInfBtn)
self.addRemoveInfsWin_removeUnusedInfBtn.setMinimumSize(110, 35)
self.addRemoveInfsWin_removeUnusedInfBtn.setMaximumSize(110, 35)
self.addRemoveInfsWin_removeUnusedInfBtn.setFont(font)
self.addRemoveInfsWin_removeUnusedInfBtn.clicked.connect(
partial(self.addOrRemoveInfs_addInf, False, True))
self.addRemoveInfsWin_removeUnusedInfBtn.setObjectName("blueButton")
self.addRemoveInfsWin_pruneBtn = QtWidgets.QPushButton("Prune Weights")
self.addRemoveInfsWin_rightSideLayout.addWidget(
self.addRemoveInfsWin_pruneBtn)
self.addRemoveInfsWin_pruneBtn.setMinimumSize(110, 35)
self.addRemoveInfsWin_pruneBtn.setMaximumSize(110, 35)
self.addRemoveInfsWin_pruneBtn.setFont(font)
self.addRemoveInfsWin_pruneBtn.clicked.connect(
partial(self.addOrRemoveInfs_prune))
self.addRemoveInfsWin_pruneBtn.setObjectName("blueButton")
# populate infList
self.addOrRemoveInfs_RefreshSelection()
# show window
self.addRemoveInfsWin.show()
def __init__(self, moduleInst, rigUiInst, parent = None):
super(ART_SetMirrorModule_UI, 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")
#create class variables
self.modInst = moduleInst
self.rigUiInst = rigUiInst
#load stylesheet
styleSheetFile = utils.returnNicePath(self.toolsPath, "Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
style = f.read()
f.close()
self.setStyleSheet(style)
#size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed)
#create the main widget
self.mainWidget = QtWidgets.QWidget()
self.setCentralWidget(self.mainWidget)
#set qt object name
self.setObjectName(windowObject)
self.setWindowTitle(windowTitle)
#create the mainLayout for the rig creator UI
self.mainLayout = QtWidgets.QVBoxLayout(self.mainWidget)
self.mainLayout.setContentsMargins(0, 0, 0, 0)
self.setSizePolicy(mainSizePolicy)
self.setMinimumSize(QtCore.QSize( 250, 200 ))
self.setMaximumSize(QtCore.QSize( 250, 200 ))
#create the background image
self.frame = QtWidgets.QFrame()
self.mainLayout.addWidget(self.frame)
#create the layout for the widgets
self.widgetLayout = QtWidgets.QVBoxLayout(self.frame)
label = QtWidgets.QLabel("Choose Mirror Module:")
self.widgetLayout.addWidget(label)
font = QtGui.QFont()
font.setBold(True)
self.moduleList = QtWidgets.QListWidget()
self.moduleList.addItem("None")
self.widgetLayout.addWidget(self.moduleList)
#add items to comboBox
networkNode = self.modInst.returnNetworkNode
modules = utils.returnRigModules()
for mod in modules:
modName = cmds.getAttr(mod + ".moduleName")
modType = cmds.getAttr(mod + ".moduleType")
if modType == cmds.getAttr(networkNode + ".moduleType"):
if mod != networkNode:
self.moduleList.addItem(modName)
self.moduleList.setCurrentRow(0)
spacerItem1 = QtWidgets.QSpacerItem(20, 40, QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Expanding)
self.widgetLayout.addItem(spacerItem1)
#update button
self.updateBtn = QtWidgets.QPushButton("UPDATE")
self.widgetLayout.addWidget(self.updateBtn)
self.updateBtn.setMinimumSize(QtCore.QSize(230, 40))
self.updateBtn.setMaximumSize(QtCore.QSize(230, 40))
self.updateBtn.setSizePolicy(mainSizePolicy)
font = QtGui.QFont()
font.setPointSize(12)
self.updateBtn.setFont(font)
self.updateBtn.setObjectName("blueButton")
#hookup signal/slot on create button
self.updateBtn.clicked.connect(self.setMirrorModule)
def buildExportWeightsUI(self):
"""
Build the interface for exporting the skin weights. An entry is added for each piece of selected geometry.
The user then has the ability to specify a .weight file name for the associated geometry.
The user also specifies where they would like the weight files saved to.
.. image:: /images/exportWeights.png
"""
if cmds.window("ART_exportSkinWeightsUI", exists=True):
cmds.deleteUI("ART_exportSkinWeightsUI", wnd=True)
# launch a UI to get the name information
self.exportSkinWeights_Win = QtWidgets.QMainWindow(self.mainUI)
# size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Fixed)
# load stylesheet
styleSheetFile = utils.returnNicePath(
self.toolsPath,
"Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
self.style = f.read()
f.close()
# create the main widget
self.exportSkinWeights_mainWidget = QtWidgets.QWidget()
self.exportSkinWeights_Win.setCentralWidget(
self.exportSkinWeights_mainWidget)
# set qt object name
self.exportSkinWeights_Win.setObjectName("ART_exportSkinWeightsUI")
self.exportSkinWeights_Win.setWindowTitle("Export Skin Weights")
# create the mainLayout for the ui
self.exportSkinWeights_mainLayout = QtWidgets.QVBoxLayout(
self.exportSkinWeights_mainWidget)
self.exportSkinWeights_mainLayout.setContentsMargins(5, 5, 5, 5)
self.exportSkinWeights_Win.resize(450, 600)
self.exportSkinWeights_Win.setSizePolicy(mainSizePolicy)
self.exportSkinWeights_Win.setMinimumSize(QtCore.QSize(450, 600))
self.exportSkinWeights_Win.setMaximumSize(QtCore.QSize(450, 600))
# create the background image
self.exportSkinWeights_frame = QtWidgets.QFrame()
self.exportSkinWeights_mainLayout.addWidget(
self.exportSkinWeights_frame)
self.exportSkinWeights_frame.setObjectName("dark")
# create widgetLayout
self.exportSkinWeights_widgetLayout = QtWidgets.QVBoxLayout(
self.exportSkinWeights_frame)
# create the hboxLayout for lineEdit and browser button
self.exportSkinWeights_browseLayout = QtWidgets.QHBoxLayout()
self.exportSkinWeights_widgetLayout.addLayout(
self.exportSkinWeights_browseLayout)
# create the line edit for the export path
self.exportSkinWeights_lineEdit = QtWidgets.QLineEdit(
utils.returnFriendlyPath(self.toolsPath))
self.exportSkinWeights_browseLayout.addWidget(
self.exportSkinWeights_lineEdit)
self.exportSkinWeights_browseBtn = QtWidgets.QPushButton()
self.exportSkinWeights_browseLayout.addWidget(
self.exportSkinWeights_browseBtn)
self.exportSkinWeights_browseBtn.setMinimumSize(35, 35)
self.exportSkinWeights_browseBtn.setMaximumSize(35, 35)
icon = QtGui.QIcon(
os.path.join(self.iconsPath, "System/fileBrowse.png"))
self.exportSkinWeights_browseBtn.setIconSize(QtCore.QSize(30, 30))
self.exportSkinWeights_browseBtn.setIcon(icon)
self.exportSkinWeights_browseBtn.clicked.connect(
partial(self.exportSkinWeights_fileBrowse))
# scroll area contents
self.exportSkinWeights_scrollContents = QtWidgets.QFrame()
self.exportSkinWeights_scrollContents.setObjectName("light")
# Layout of Container Widget
self.exportSkinWeights_VLayout = QtWidgets.QVBoxLayout()
# find selected geometry and populate scroll area
self.exportSkinWeights_populate()
# add scrollArea for selected geo, skinFileName, and checkbox for exporting
self.exportSkinWeights_scrollLayout = QtWidgets.QScrollArea()
self.exportSkinWeights_widgetLayout.addWidget(
self.exportSkinWeights_scrollLayout)
self.exportSkinWeights_scrollLayout.setHorizontalScrollBarPolicy(
QtCore.Qt.ScrollBarAlwaysOff)
self.exportSkinWeights_scrollLayout.setVerticalScrollBarPolicy(
QtCore.Qt.ScrollBarAlwaysOn)
self.exportSkinWeights_scrollLayout.setWidgetResizable(False)
self.exportSkinWeights_scrollLayout.setWidget(
self.exportSkinWeights_scrollContents)
# lastly, export button
font = QtGui.QFont()
font.setPointSize(8)
font.setBold(True)
self.exportSkinWeights_exportBtnLayout = QtWidgets.QHBoxLayout()
self.exportSkinWeights_widgetLayout.addLayout(
self.exportSkinWeights_exportBtnLayout)
self.exportSkinWeights_RefreshBtn = QtWidgets.QPushButton("Refresh")
self.exportSkinWeights_exportBtnLayout.addWidget(
self.exportSkinWeights_RefreshBtn)
self.exportSkinWeights_RefreshBtn.setMinimumSize(QtCore.QSize(70, 50))
self.exportSkinWeights_RefreshBtn.setMaximumSize(QtCore.QSize(70, 50))
self.exportSkinWeights_RefreshBtn.setFont(font)
self.exportSkinWeights_RefreshBtn.clicked.connect(
partial(self.buildExportWeightsUI))
self.exportSkinWeights_RefreshBtn.setObjectName("blueButton")
self.exportSkinWeights_ExportBtn = QtWidgets.QPushButton(
"EXPORT WEIGHTS")
self.exportSkinWeights_exportBtnLayout.addWidget(
self.exportSkinWeights_ExportBtn)
self.exportSkinWeights_ExportBtn.setMinimumSize(QtCore.QSize(350, 50))
self.exportSkinWeights_ExportBtn.setMaximumSize(QtCore.QSize(350, 50))
self.exportSkinWeights_ExportBtn.setFont(font)
self.exportSkinWeights_ExportBtn.clicked.connect(
partial(self.exportSkinWeights_doExport))
self.exportSkinWeights_ExportBtn.setObjectName("blueButton")
# show window
self.exportSkinWeights_Win.show()
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 __init__(self,
baseName,
moduleInst,
rigUiInst,
prefix,
suffix,
parent=None):
"""
Instantiates the class, taking in current module information, and builds the interface.
:param baseName: The base name of the module, found on the network node attribute.
:param moduleInst: The instance in memory of the module whose name is to change.
:param rigUiInst: The instance in memory of the Rig Creator UI from which this class was called.
:param prefix: The existing prefix of the module name.
:param suffix: The existing suffix of the module name.
"""
super(ART_ChangeModuleName_UI, 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")
# create class variables
self.baseName = baseName
self.modInst = moduleInst
self.rigUiInst = rigUiInst
self.prefixInc = prefix
self.suffixInc = suffix
# load stylesheet
styleSheetFile = utils.returnNicePath(
self.toolsPath,
"Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
style = f.read()
f.close()
self.setStyleSheet(style)
# size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Fixed)
# create the main widget
self.mainWidget = QtWidgets.QWidget()
self.setCentralWidget(self.mainWidget)
# set qt object name
self.setObjectName(windowObject)
self.setWindowTitle(windowTitle)
# create the mainLayout for the rig creator UI
self.mainLayout = QtWidgets.QVBoxLayout(self.mainWidget)
self.mainLayout.setContentsMargins(0, 0, 0, 0)
self.resize(300, 150)
self.setSizePolicy(mainSizePolicy)
self.setMinimumSize(QtCore.QSize(300, 150))
self.setMaximumSize(QtCore.QSize(300, 150))
# create the background image
self.frame = QtWidgets.QFrame()
self.mainLayout.addWidget(self.frame)
# create the layout for the widgets
self.widgetLayout = QtWidgets.QVBoxLayout(self.frame)
# create the prefix pair of fields
self.prefixForm = QtWidgets.QFormLayout()
self.widgetLayout.addLayout(self.prefixForm)
self.prefixLabel = QtWidgets.QLabel("Prefix: ")
self.prefixForm.setWidget(0, QtWidgets.QFormLayout.LabelRole,
self.prefixLabel)
self.prefix = QtWidgets.QLineEdit(self.prefixInc)
self.prefixForm.setWidget(0, QtWidgets.QFormLayout.FieldRole,
self.prefix)
# hookup signal/slot connection
self.prefix.textChanged.connect(self.updatePreview)
# create the suffix pair of fields
self.suffixForm = QtWidgets.QFormLayout()
self.widgetLayout.addLayout(self.suffixForm)
self.suffixLabel = QtWidgets.QLabel("Suffix: ")
self.suffixForm.setWidget(0, QtWidgets.QFormLayout.LabelRole,
self.suffixLabel)
self.suffix = QtWidgets.QLineEdit(self.suffixInc)
self.suffixForm.setWidget(0, QtWidgets.QFormLayout.FieldRole,
self.suffix)
# hookup signal/slot connection
self.suffix.textChanged.connect(self.updatePreview)
# spacer
spacerItem = QtWidgets.QSpacerItem(20, 40,
QtWidgets.QSizePolicy.Minimum,
QtWidgets.QSizePolicy.Expanding)
self.widgetLayout.addItem(spacerItem)
# realtime preview of final module name
self.previewForm = QtWidgets.QFormLayout()
self.widgetLayout.addLayout(self.previewForm)
self.previewLabel = QtWidgets.QLabel("Preview: ")
self.previewName = QtWidgets.QLabel(self.prefixInc + self.baseName +
self.suffixInc)
self.previewName.setMinimumSize(QtCore.QSize(200, 20))
self.previewName.setMaximumSize(QtCore.QSize(200, 20))
self.previewName.setAlignment(QtCore.Qt.AlignHCenter)
self.previewForm.setWidget(0, QtWidgets.QFormLayout.LabelRole,
self.previewLabel)
self.previewForm.setWidget(0, QtWidgets.QFormLayout.FieldRole,
self.previewName)
# set preview font
font = QtGui.QFont()
font.setPointSize(12)
self.previewName.setFont(font)
spacerItem1 = QtWidgets.QSpacerItem(20, 40,
QtWidgets.QSizePolicy.Minimum,
QtWidgets.QSizePolicy.Expanding)
self.widgetLayout.addItem(spacerItem1)
# update button
self.updateBtn = QtWidgets.QPushButton("UPDATE")
self.widgetLayout.addWidget(self.updateBtn)
self.updateBtn.setMinimumSize(QtCore.QSize(285, 40))
self.updateBtn.setMaximumSize(QtCore.QSize(285, 40))
self.updateBtn.setSizePolicy(mainSizePolicy)
font = QtGui.QFont()
font.setPointSize(12)
self.updateBtn.setFont(font)
self.updateBtn.setObjectName("blueButton")
# hookup signal/slot on create button
self.updateBtn.clicked.connect(self.applyModuleNameChange)
def buildBoneCounterUI(self):
"""
Builds the interface for the bone counter tool, which is comprised off a QLineEdit that shows the current
bone count, a QPushButton to launch another simple UI that allows the user to set a bone count target,
and QProgressBar that shows a percentage to target bone count.
.. seealso:: ART_BoneCounter.setBoneCountTarget
"""
if cmds.window("ART_BoneCounterWin", exists=True):
cmds.deleteUI("ART_BoneCounterWin", wnd=True)
# launch a UI to get the name information
self.boneCounterWin = QtWidgets.QMainWindow(self.mainUI)
# size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed)
# create the main widget
self.boneCounterWin_mainWidget = QtWidgets.QWidget()
self.boneCounterWin.setCentralWidget(self.boneCounterWin_mainWidget)
# set qt object name
self.boneCounterWin.setObjectName("ART_BoneCounterWin")
self.boneCounterWin.setWindowTitle("Bone Counter")
# create the mainLayout for the rig creator UI
self.boneCounterWin_mainLayout = QtWidgets.QVBoxLayout(self.boneCounterWin_mainWidget)
self.boneCounterWin_mainLayout.setContentsMargins(0, 0, 0, 0)
self.boneCounterWin.resize(300, 100)
self.boneCounterWin.setSizePolicy(mainSizePolicy)
self.boneCounterWin.setMinimumSize(QtCore.QSize(300, 100))
self.boneCounterWin.setMaximumSize(QtCore.QSize(300, 100))
headerFont = QtGui.QFont()
headerFont.setPointSize(8)
headerFont.setBold(True)
# load toolbar stylesheet
styleSheetFile = utils.returnNicePath(self.toolsPath, "Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
self.style = f.read()
f.close()
# create the bone counter stylesheet
self.progBarStyle = """
QProgressBar{
border: 2px solid black;
border-radius: 5px;
text-align: center;
font: 87 10pt "Arial";
color: rgb(255,255,255);
}
QProgressBar::chunk {
background-color: qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 rgb(25,175,255), stop:1 rgb(9,62,98));
width: 10px;
margin: 0.5px;
}
"""
self.progBarStyleMax = """
QProgressBar{
border: 2px solid black;
border-radius: 5px;
text-align: center;
font: 87 10pt "Arial Black";
color: rgb(255,255,255);
}
QProgressBar::chunk {
background-color: qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 rgb(255,174,0), stop:1 rgb(30,30,30));
width: 10px;
margin: 0.5px;
}
"""
# create the background image
self.boneCounterWin_frame = QtWidgets.QFrame()
self.boneCounterWin_mainLayout.addWidget(self.boneCounterWin_frame)
self.boneCounterWin_frame.setObjectName("mid")
self.boneCounterWin.setStyleSheet(self.style)
# create the layout for the widgets
self.boneCounterWin_widgetLayoutMain = QtWidgets.QVBoxLayout(self.boneCounterWin_frame)
self.boneCounterWin_widgetLayoutMain.setContentsMargins(5, 5, 5, 5)
self.boneCounterWin_widgetLayout = QtWidgets.QHBoxLayout()
self.boneCounterWin_widgetLayoutMain.addLayout(self.boneCounterWin_widgetLayout)
# label creation
self.boneCount = QtWidgets.QLabel("Bone Count: ")
self.boneCount.setFont(headerFont)
self.boneCounterWin_widgetLayout.addWidget(self.boneCount)
self.boneCounter = QtWidgets.QSpinBox()
self.boneCounter.setReadOnly(True)
self.boneCounter.setButtonSymbols(QtWidgets.QAbstractSpinBox.NoButtons)
self.boneCounter.setRange(1, 9999)
self.boneCounter.setMinimumWidth(60)
self.boneCounter.setMaximumWidth(60)
self.boneCounterWin_widgetLayout.addWidget(self.boneCounter)
# create the button
self.boneMaxButton = QtWidgets.QPushButton("Set Target")
self.boneCounterWin_widgetLayout.addWidget(self.boneMaxButton)
self.boneMaxButton.clicked.connect(partial(self.setBoneCountTarget))
self.boneMaxButton.setMinimumHeight(30)
self.boneMaxButton.setStyleSheet(self.style)
self.boneMaxButton.setObjectName("blueButton")
# add the progress bar
self.boneCountBar = QtWidgets.QProgressBar()
self.boneCounterWin_widgetLayoutMain.addWidget(self.boneCountBar)
self.boneCountBar.setValue(1)
self.boneCountBar.setStyleSheet(self.progBarStyle)
self.boneCountBar.setRange(0, 100)
self.boneCountBar.setFormat("target = %m")
# add the max range of the progress bar to the main network node
if cmds.objExists("ART_Root_Module.target"):
value = cmds.getAttr("ART_Root_Module.target")
currentValue = self.boneCountBar.value()
self.boneCountBar.setMaximum(value)
if value < currentValue:
self.boneCountBar.setValue(value)
if currentValue <= value:
self.boneCountBar.setStyleSheet(self.progBarStyle)
if currentValue > value:
self.boneCountBar.setStyleSheet(self.progBarStyleMax)
else:
cmds.addAttr("ART_Root_Module", sn="target", keyable=False)
cmds.setAttr("ART_Root_Module.target", 100, lock=True)
# get the current bone count
modules = utils.returnRigModules()
allBones = []
for module in modules:
joints = cmds.getAttr(module + ".Created_Bones")
splitJoints = joints.split("::")
for bone in splitJoints:
if bone != "":
allBones.append(bone)
# update the spinBox and progress bar
self.boneCounter.setValue(len(allBones))
max = self.boneCountBar.maximum()
if len(allBones) <= max:
self.boneCountBar.setValue(len(allBones))
self.boneCountBar.setStyleSheet(self.progBarStyle)
if len(allBones) > max:
self.boneCountBar.setValue(max)
self.boneCountBar.setStyleSheet(self.progBarStyleMax)
# show window
self.boneCounterWin.show()
def addSelected(self, *args):
"""
Finds the selected character, and references that file into the currently opened scene. The path is
constructed using the QSettings info, with any group as a subfolder, and lastly the selected asset/character
as the last part of the path.
Also handles namespaces, adding the new namespace to the network node of the added asset, and launches the
animationUI.
"""
# get a list of the existing folders in projects
selectedProject = self.projectMenu.currentText()
fullPath = utils.returnNicePath(self.projectPath, selectedProject)
selectedGroup = self.groupMenu.currentText()
if len(selectedGroup) > 1:
fullPath = utils.returnNicePath(fullPath, selectedGroup)
selectedCharacter = self.characterList.currentItem().text()
mayaFile = utils.returnNicePath(fullPath, selectedCharacter + ".ma")
iconFile = utils.returnNicePath(fullPath, selectedCharacter + ".png")
# reference in the selected character if the file exists on drive
if os.path.exists(mayaFile):
if os.path.exists(iconFile):
# find existing namespaces in scene
namespaces = cmds.namespaceInfo(listOnlyNamespaces=True)
# reference the rig file
cmds.file(mayaFile, r=True, type="mayaAscii", loadReferenceDepth="all", namespace=selectedCharacter,
options="v=0")
# clear selection and fit view
cmds.select(clear=True)
cmds.viewFit()
panels = cmds.getPanel(type='modelPanel')
# turn on smooth shading
for panel in panels:
editor = cmds.modelPanel(panel, q=True, modelEditor=True)
cmds.modelEditor(editor, edit=True, displayAppearance="smoothShaded", displayTextures=True,
textures=True)
# find new namespaces in scene to figure out the namespace that was created upon referencing the
# character
newCharacterName = selectedCharacter
newNamespaces = cmds.namespaceInfo(listOnlyNamespaces=True)
for name in newNamespaces:
if name not in namespaces:
newCharacterName = name
# add an attribute to the rig root (if needed) and set the namespace attr to the newCharacterName
if cmds.objExists(newCharacterName + ":ART_RIG_ROOT"):
if not cmds.objExists(newCharacterName + ":ART_RIG_ROOT.namespace"):
cmds.addAttr(newCharacterName + ":ART_RIG_ROOT", ln="namespace", dt="string", keyable=False)
cmds.setAttr(newCharacterName + ":ART_RIG_ROOT.namespace", newCharacterName, type="string")
# delete any interfaces that may be up
self.closeUI()
if cmds.dockControl("pyArtRigCreatorDock", q=True, exists=True):
if cmds.window("pyArtRigCreatorUi", exists=True):
cmds.deleteUI("pyArtRigCreatorUi", wnd=True)
cmds.deleteUI("pyArtRigCreatorDock", control=True)
# launch anim UI
cmds.refresh(force=True)
stylesheetDir = utils.returnNicePath(self.scriptPath, "Interfaces/StyleSheets/")
stylesheets = os.listdir(stylesheetDir)
for sheet in stylesheets:
interfaceUtils.writeQSS(os.path.join(stylesheetDir, sheet))
import Interfaces.ART_AnimationUI as ART_AnimationUI
ART_AnimationUI.run()
def buildAimModeUI(self):
"""
Builds the Aim Mode interface, finding all modules that have the ability to aim, and listing those modules
as well as their current aim status.
"""
if cmds.window("ART_AimModeWin", exists=True):
cmds.deleteUI("ART_AimModeWin", wnd=True)
# launch a UI to get the name information
self.aimModeWin = QtWidgets.QMainWindow(self.mainUI)
# load stylesheet
styleSheetFile = utils.returnNicePath(
self.toolsPath,
"Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
self.style = f.read()
f.close()
self.aimModeWin.setStyleSheet(self.style)
# size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Fixed)
# create the main widget
self.aimModeWin_mainWidget = QtWidgets.QWidget()
self.aimModeWin.setCentralWidget(self.aimModeWin_mainWidget)
# set qt object name
self.aimModeWin.setObjectName("ART_AimModeWin")
self.aimModeWin.setWindowTitle("Aim Mode")
# create the mainLayout for the rig creator UI
self.aimModeWin_mainLayout = QtWidgets.QVBoxLayout(
self.aimModeWin_mainWidget)
self.aimModeWin_mainLayout.setContentsMargins(0, 0, 0, 0)
self.aimModeWin.resize(400, 250)
self.aimModeWin.setSizePolicy(mainSizePolicy)
self.aimModeWin.setMinimumSize(QtCore.QSize(400, 250))
self.aimModeWin.setMaximumSize(QtCore.QSize(400, 250))
# create the background image
self.aimModeWin_frame = QtWidgets.QFrame()
self.aimModeWin_mainLayout.addWidget(self.aimModeWin_frame)
# create the layout for the widgets
self.aimModeWin_widgetLayout = QtWidgets.QHBoxLayout(
self.aimModeWin_frame)
self.aimModeWin_widgetLayout.setContentsMargins(5, 5, 5, 5)
# add the QListWidget Frame
self.aimModeWin_moduleListFrame = QtWidgets.QFrame()
self.aimModeWin_moduleListFrame.setMinimumSize(QtCore.QSize(275, 200))
self.aimModeWin_moduleListFrame.setMaximumSize(QtCore.QSize(275, 200))
self.aimModeWin_moduleListFrame.setContentsMargins(20, 0, 20, 0)
# create the list widget
self.aimModeWin_moduleList = QtWidgets.QListWidget(
self.aimModeWin_moduleListFrame)
self.aimModeWin_widgetLayout.addWidget(self.aimModeWin_moduleListFrame)
self.aimModeWin_moduleList.setMinimumSize(QtCore.QSize(265, 200))
self.aimModeWin_moduleList.setMaximumSize(QtCore.QSize(265, 200))
self.aimModeWin_moduleList.setSelectionMode(
QtWidgets.QAbstractItemView.MultiSelection)
self.aimModeWin_moduleList.setSpacing(3)
# add the layout for the buttons
self.aimModeWin_buttonLayoutAll = QtWidgets.QVBoxLayout()
self.aimModeWin_widgetLayout.addLayout(self.aimModeWin_buttonLayoutAll)
self.aimModeWin_buttonLayoutAll.setContentsMargins(5, 20, 5, 20)
# add the selection buttons
self.aimModeWin_selectionButtonLayout = QtWidgets.QVBoxLayout()
self.aimModeWin_buttonLayoutAll.addLayout(
self.aimModeWin_selectionButtonLayout)
self.aimModeWin_selectAllButton = QtWidgets.QPushButton("Select All")
self.aimModeWin_selectAllButton.setMinimumSize(QtCore.QSize(115, 25))
self.aimModeWin_selectAllButton.setMaximumSize(QtCore.QSize(115, 25))
self.aimModeWin_selectionButtonLayout.addWidget(
self.aimModeWin_selectAllButton)
self.aimModeWin_selectAllButton.clicked.connect(
self.aimModeWin_moduleList.selectAll)
self.aimModeWin_selectAllButton.setObjectName("blueButton")
self.aimModeWin_selectNoneButton = QtWidgets.QPushButton(
"Clear Selection")
self.aimModeWin_selectNoneButton.setMinimumSize(QtCore.QSize(115, 25))
self.aimModeWin_selectNoneButton.setMaximumSize(QtCore.QSize(115, 25))
self.aimModeWin_selectionButtonLayout.addWidget(
self.aimModeWin_selectNoneButton)
self.aimModeWin_selectNoneButton.clicked.connect(
self.aimModeWin_moduleList.clearSelection)
self.aimModeWin_selectNoneButton.setObjectName("blueButton")
# spacer
spacerItem = QtWidgets.QSpacerItem(20, 80,
QtWidgets.QSizePolicy.Minimum,
QtWidgets.QSizePolicy.Expanding)
self.aimModeWin_selectionButtonLayout.addItem(spacerItem)
# add the buttons for reset settings and reset transforms
self.aimModeWin_turnOnAim = QtWidgets.QPushButton("On")
self.aimModeWin_turnOnAim.setMinimumSize(QtCore.QSize(115, 25))
self.aimModeWin_turnOnAim.setMaximumSize(QtCore.QSize(115, 25))
self.aimModeWin_selectionButtonLayout.addWidget(
self.aimModeWin_turnOnAim)
self.aimModeWin_turnOnAim.setToolTip(
"Turn on Aim Mode for selected modules.")
self.aimModeWin_turnOnAim.clicked.connect(
partial(self.aimModeUI_Toggle, True))
self.aimModeWin_turnOnAim.setObjectName("blueButton")
self.aimModeWin_turnOffAim = QtWidgets.QPushButton("Off")
self.aimModeWin_turnOffAim.setMinimumSize(QtCore.QSize(115, 25))
self.aimModeWin_turnOffAim.setMaximumSize(QtCore.QSize(115, 25))
self.aimModeWin_selectionButtonLayout.addWidget(
self.aimModeWin_turnOffAim)
self.aimModeWin_turnOffAim.setToolTip(
"Turn off Aim Mode for selected modules.")
self.aimModeWin_turnOffAim.clicked.connect(
partial(self.aimModeUI_Toggle, False))
self.aimModeWin_turnOffAim.setObjectName("blueButton")
# populate the list widget
modules = utils.returnRigModules()
for module in modules:
# get module name
moduleName = cmds.getAttr(module + ".moduleName")
# figure out if the module supports aimMode
canAim = False
if cmds.objExists(module + ".canAim"):
canAim = cmds.getAttr(module + ".canAim")
# see if it is currently in aimMode
aimMode = cmds.getAttr(module + ".aimMode")
# if it does, add it to the listwidget
if canAim:
# font
headerFont = QtGui.QFont()
headerFont.setPointSize(10)
headerFont.setBold(True)
# create the listWidgetItem
pixmap = QtGui.QPixmap(10, 10)
pixmap.fill(QtGui.QColor(0, 255, 0))
icon = QtGui.QIcon(pixmap)
pixmapOff = QtGui.QPixmap(10, 10)
pixmapOff.fill(QtGui.QColor(255, 0, 0))
iconOff = QtGui.QIcon(pixmapOff)
item = QtWidgets.QListWidgetItem(iconOff, moduleName)
item.setFont(headerFont)
item.setTextAlignment(QtCore.Qt.AlignCenter)
item.setData(QtCore.Qt.UserRole, [icon, iconOff])
if aimMode:
item.setIcon(icon)
self.aimModeWin_moduleList.addItem(item)
# show the window
self.aimModeWin.show()
def finalizeSetup_UI(self):
if cmds.window("ART_finalizeSetupWin", exists=True):
cmds.deleteUI("ART_finalizeSetupWin", wnd=True)
#launch a UI to get the name information
self.finalizeSetupWin = QtWidgets.QMainWindow(self.mainUI)
#size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Fixed)
#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.finalizeSetupWin.setStyleSheet(self.style)
#create the main widget
self.finalizeSetupWin_mainWidget = QtWidgets.QWidget()
self.finalizeSetupWin.setCentralWidget(
self.finalizeSetupWin_mainWidget)
#set qt object name
self.finalizeSetupWin.setObjectName("ART_finalizeSetupWin")
self.finalizeSetupWin.setWindowTitle("Finalize Setup")
#create the mainLayout for the rig creator UI
self.finalizeSetupWin_mainLayout = QtWidgets.QVBoxLayout(
self.finalizeSetupWin_mainWidget)
self.finalizeSetupWin_mainLayout.setContentsMargins(0, 0, 0, 0)
self.finalizeSetupWin.resize(450, 300)
self.finalizeSetupWin.setSizePolicy(mainSizePolicy)
self.finalizeSetupWin.setMinimumSize(QtCore.QSize(450, 300))
self.finalizeSetupWin.setMaximumSize(QtCore.QSize(450, 300))
#create the background image
self.finalizeSetupWin_frame = QtWidgets.QFrame()
self.finalizeSetupWin_mainLayout.addWidget(self.finalizeSetupWin_frame)
#create the main vertical layout inside the frame
self.finalizeSetupWin_mainVLayout = QtWidgets.QVBoxLayout(
self.finalizeSetupWin_frame)
# # # # TEXT EDIT # # # #
self.finalizeSetupWin_Text = QtWidgets.QTextEdit()
self.finalizeSetupWin_Text.setMinimumSize(QtCore.QSize(440, 230))
self.finalizeSetupWin_Text.setMaximumSize(QtCore.QSize(440, 230))
self.finalizeSetupWin_mainVLayout.addWidget(self.finalizeSetupWin_Text)
self.finalizeSetupWin_Text.setReadOnly(True)
self.finalizeSetupWin_Text.setAcceptRichText(True)
#text
text = "Finalizing the setup will create the skeleton that will be used for skin binding."
cursor = self.finalizeSetupWin_Text.textCursor()
cursor.insertText(text)
text = "\nIt is recommended that offsets are baked before continuing. "
self.finalizeSetupWin_Text.setTextColor(QtGui.QColor(236, 217, 0))
self.finalizeSetupWin_Text.setFontPointSize(10)
self.finalizeSetupWin_Text.append(text)
#image
image2 = utils.returnNicePath(self.iconsPath, "System/bakeOffsets.png")
icon = QtGui.QPixmap(image2)
image = icon.toImage()
cursor.insertImage(image)
text = "\n(You will still be able to edit your setup by coming back to this step using the 'Edit Setup' button seen in the deformation tools interface):\n\n"
self.finalizeSetupWin_Text.setTextColor(QtGui.QColor(255, 255, 255))
self.finalizeSetupWin_Text.setFontPointSize(8)
self.finalizeSetupWin_Text.append(text)
#image
image2 = utils.returnNicePath(self.iconsPath,
"System/finalizeSetup.png")
icon = QtGui.QPixmap(image2)
image = icon.toImage()
cursor.insertImage(image)
self.finalizeSetupWin_Text.setTextCursor(cursor)
end = "<br>"
fragment = QtGui.QTextDocumentFragment.fromHtml(end)
cursor.insertFragment(fragment)
# # # # END TEXT EDIT # # # #
# # # # BUTTON LAYOUT # # # #
self.finalizeSetupWin_buttonLayout = QtWidgets.QHBoxLayout()
self.finalizeSetupWin_mainVLayout.addLayout(
self.finalizeSetupWin_buttonLayout)
self.finalizeSetupWin_ContinueBtn = QtWidgets.QPushButton("Continue")
self.finalizeSetupWin_CancelBtn = QtWidgets.QPushButton("Cancel")
self.finalizeSetupWin_HelpBtn = QtWidgets.QPushButton("?")
self.finalizeSetupWin_HelpBtn.setMinimumSize(QtCore.QSize(25, 25))
self.finalizeSetupWin_HelpBtn.setMaximumSize(QtCore.QSize(25, 25))
self.finalizeSetupWin_buttonLayout.addWidget(
self.finalizeSetupWin_ContinueBtn)
self.finalizeSetupWin_buttonLayout.addWidget(
self.finalizeSetupWin_CancelBtn)
self.finalizeSetupWin_buttonLayout.addWidget(
self.finalizeSetupWin_HelpBtn)
self.finalizeSetupWin_ContinueBtn.clicked.connect(
partial(self.finalizeSetup_Continue))
self.finalizeSetupWin_CancelBtn.clicked.connect(
partial(self.finalizeSetup_Cancel))
self.finalizeSetupWin_HelpBtn.clicked.connect(
partial(self.finalizeSetup_Help))
self.finalizeSetupWin_ContinueBtn.setObjectName("blueButton")
self.finalizeSetupWin_CancelBtn.setObjectName("blueButton")
self.finalizeSetupWin_HelpBtn.setObjectName("blueButton")
# # # # END BUTTON LAYOUT # # # #
#show window
self.finalizeSetupWin_Text.moveCursor(QtGui.QTextCursor.Start)
self.finalizeSetupWin.show()
def __init__(self, baseName, className, rigUiInst, parent=None):
"""
Initialize the class, taking in the base name of the module to be added, the name of the class of the module
to be added, and the instance of the rig creator UI. Then build the interface for the tool.
:param baseName: The base name of the module to be added, defined in the module class file at the top.
:param className: The class name of the module to be added, so we can then initialize that module.
:param rigUiInst: The instance of the rig creator UI, from which this function was called.
"""
super(ART_AddModule_UI, 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")
# create class variables
self.baseName = baseName
self.className = className
self.rigUiInst = rigUiInst
# load stylesheet
styleSheetFile = utils.returnNicePath(
self.toolsPath,
"Core/Scripts/Interfaces/StyleSheets/mainScheme.qss")
f = open(styleSheetFile, "r")
style = f.read()
f.close()
self.setStyleSheet(style)
# size policies
mainSizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Fixed)
# create the main widget
self.mainWidget = QtWidgets.QWidget()
self.setCentralWidget(self.mainWidget)
# set qt object name
self.setObjectName(windowObject)
self.setWindowTitle(windowTitle)
# create the mainLayout for the rig creator UI
self.mainLayout = QtWidgets.QHBoxLayout(self.mainWidget)
self.mainLayout.setContentsMargins(0, 0, 0, 0)
self.setSizePolicy(mainSizePolicy)
self.setMinimumSize(QtCore.QSize(500, 220))
self.setMaximumSize(QtCore.QSize(500, 520))
# create the background
self.frame = QtWidgets.QFrame()
self.frame.setObjectName("mid")
self.mainLayout.addWidget(self.frame)
# create the layout for the widgets
self.column1layout = QtWidgets.QVBoxLayout(self.frame)
self.mainLayout.addLayout(self.column1layout)
self.column2Layout = QtWidgets.QVBoxLayout()
self.mainLayout.addLayout(self.column2Layout)
font = QtGui.QFont()
font.setBold(True)
label = QtWidgets.QLabel("Choose Parent Bone")
label.setFont(font)
label.setAlignment(QtCore.Qt.AlignCenter)
self.column2Layout.addWidget(label)
self.boneSearch = QtWidgets.QLineEdit()
self.column2Layout.addWidget(self.boneSearch)
self.boneSearch.setPlaceholderText("Search...")
self.boneSearch.textChanged.connect(self.searchList)
self.hierarchyTree = QtWidgets.QListWidget()
self.column2Layout.addWidget(self.hierarchyTree)
# add items to listWidget
parents = utils.getViableParents()
for bone in parents:
self.hierarchyTree.addItem(bone)
if bone == "root":
index = parents.index(bone)
self.hierarchyTree.setCurrentRow(index)
# create the prefix pair of fields
self.prefixForm = QtWidgets.QFormLayout()
self.column1layout.addLayout(self.prefixForm)
self.prefixLabel = QtWidgets.QLabel("Prefix: ")
self.prefixForm.setWidget(0, QtWidgets.QFormLayout.LabelRole,
self.prefixLabel)
self.prefix = QtWidgets.QLineEdit()
self.prefixForm.setWidget(0, QtWidgets.QFormLayout.FieldRole,
self.prefix)
# hookup signal/slot connection
self.prefix.textChanged.connect(self.updatePreview)
# create the suffix pair of fields
self.suffixForm = QtWidgets.QFormLayout()
self.column1layout.addLayout(self.suffixForm)
self.suffixLabel = QtWidgets.QLabel("Suffix: ")
self.suffixForm.setWidget(0, QtWidgets.QFormLayout.LabelRole,
self.suffixLabel)
self.suffix = QtWidgets.QLineEdit()
self.suffixForm.setWidget(0, QtWidgets.QFormLayout.FieldRole,
self.suffix)
# hookup signal/slot connection
self.suffix.textChanged.connect(self.updatePreview)
self.previewLabel = QtWidgets.QLabel("Module Name: ")
self.column1layout.addWidget(self.previewLabel)
self.previewName = QtWidgets.QLabel(self.baseName)
self.previewName.setMinimumSize(QtCore.QSize(255, 25))
self.previewName.setMaximumSize(QtCore.QSize(255, 25))
self.previewName.setAlignment(QtCore.Qt.AlignHCenter)
self.column1layout.addWidget(self.previewName)
# set preview font
font = QtGui.QFont()
font.setPointSize(12)
self.previewName.setFont(font)
spacerItem1 = QtWidgets.QSpacerItem(20, 40,
QtWidgets.QSizePolicy.Minimum,
QtWidgets.QSizePolicy.Expanding)
self.column1layout.addItem(spacerItem1)
# special cases (arms and legs)
specialCaseModules = ["ART_Leg_Standard", "ART_Arm_Standard"]
if className in specialCaseModules:
# spacer
groupBox = QtWidgets.QGroupBox("")
self.column1layout.addWidget(groupBox)
layout = QtWidgets.QVBoxLayout(groupBox)
self.radioButtonLayout = QtWidgets.QHBoxLayout()
layout.addLayout(self.radioButtonLayout)
self.rightRadioBtn = QtWidgets.QRadioButton("Right Side")
self.leftRadioBtn = QtWidgets.QRadioButton("Left Side")
self.radioButtonLayout.addWidget(self.rightRadioBtn)
self.radioButtonLayout.addWidget(self.leftRadioBtn)
self.leftRadioBtn.setChecked(True)
# spacer
spacerItem2 = QtWidgets.QSpacerItem(20, 80,
QtWidgets.QSizePolicy.Minimum,
QtWidgets.QSizePolicy.Expanding)
self.column1layout.addItem(spacerItem2)
# create button
self.createButton = QtWidgets.QPushButton("CREATE")
self.column1layout.addWidget(self.createButton)
self.createButton.setMinimumSize(QtCore.QSize(255, 40))
self.createButton.setMaximumSize(QtCore.QSize(255, 40))
self.createButton.setSizePolicy(mainSizePolicy)
self.createButton.setObjectName("blueButton")
font = QtGui.QFont()
font.setPointSize(12)
self.createButton.setFont(font)
# hookup signal/slot on create button
self.createButton.clicked.connect(self.createModule)
self.hierarchyTree.setFocus()
