def populateIcon(self, *args):
"""
Given the selected character, display the correct icon for that character in the UI.
"""
# default
self.characterIcon.setPixmap(self.defaultPixMap)
# 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()
fullPath = utils.returnNicePath(fullPath, selectedCharacter + ".png")
if os.path.exists(fullPath):
pixmap = QtGui.QPixmap(fullPath)
self.characterIcon.setPixmap(pixmap)
else:
self.characterIcon.setPixmap(self.defaultPixMap)
def populateTable(self):
pixmap = QtGui.QPixmap(20, 20)
pixmap.fill(QtGui.QColor(0, 255, 0))
iconOn = QtGui.QIcon(pixmap)
pixmapOff = QtGui.QPixmap(20, 20)
pixmapOff.fill(QtGui.QColor(255, 0, 0))
iconOff = QtGui.QIcon(pixmapOff)
modules = utils.returnRigModules()
self.modTable.setRowCount(100 + len(modules))
counter = 0
for module in modules:
aimState = False
pinState = False
#get module name
moduleName = cmds.getAttr(module + ".moduleName")
if cmds.objExists(module + ".aimMode"):
aimState = cmds.getAttr(module + ".aimMode")
if cmds.objExists(module + ".pinned"):
pinState = cmds.getAttr(module + ".pinned")
moduleItem = QtWidgets.QTableWidgetItem(moduleName)
self.modTable.setItem(counter, 0, moduleItem)
if aimState:
lockItem = QtWidgets.QTableWidgetItem(iconOn, "")
else:
lockItem = QtWidgets.QTableWidgetItem(iconOff, "")
self.modTable.setItem(counter, 2, lockItem)
if pinState:
pinItem = QtWidgets.QTableWidgetItem(iconOn, "")
else:
pinItem = QtWidgets.QTableWidgetItem(iconOff, "")
self.modTable.setItem(counter, 1, pinItem)
counter += 1
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 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 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()
