def __init__( self, nodeName, parent=None ): super( componentWidget, self ).__init__( parent ) self.parent = parent def on_context_menu( point, inNodeName ): popMenu = QtGui.QMenu() deleteAction = QtGui.QAction( 'Delete Component', popMenu, triggered=lambda a=inNodeName:self.deleteComponentFromObject( a ) ) popMenu.addAction( deleteAction ) popMenu.exec_( self.componentLabel.mapToGlobal( point ) ) # Setup layout. verticalLayout = QtGui.QVBoxLayout() verticalLayout.setContentsMargins( 0,0,0,0 ) verticalLayout.setSpacing( 0 ) verticalLayout.setAlignment( QtCore.Qt.AlignTop ) # Label for component componentLabel = QTWidgets.basicLabel( nodeName, 'bold', 10, 'black', '6E9094', inIndent=20 ) componentLabel.setMinimumHeight( 18 ) componentLabel.setContextMenuPolicy( QtCore.Qt.CustomContextMenu ) componentLabel.customContextMenuRequested.connect( lambda point, nodeName=nodeName:on_context_menu( point, nodeName ) ) # Properties propertyStack = QtGui.QVBoxLayout() propertyFrame = QTWidgets.basicFrame() propertyFrame.setMinimumHeight( 40 ) propertyFrame.setMaximumHeight( 40 ) # Add string edit property modulePlug = NodeUtility.getPlug( nodeName, 'moduleName' ) moduleValue = NodeUtility.getPlugValue( modulePlug ) moduleTextLayout = QTWidgets.stringProperty( 'Module Name', moduleValue ) ''' ADD EDIT FIELDS FOR PRIORITY AND CHARACTER ROOT!!!!!!!!! ''' # Add everything to the vertical layout. propertyStack.addLayout( moduleTextLayout ) propertyFrame.setLayout( propertyStack ) verticalLayout.addWidget( componentLabel ) verticalLayout.addWidget( propertyFrame ) # Connections moduleTextBox = propertyFrame.findChild( QtGui.QLineEdit, 'Module Name' ) moduleTextBox.editingFinished.connect( lambda inPlugName='moduleName', inQTType='QLineEdit', inPlugValue=moduleTextBox, inNodeName=nodeName :ModuleRootComponent( inNodeName ).setComponentAttributeFromQT( inPlugName, inQTType, inPlugValue, inNodeName ) ) #return mainWidget self.setLayout( verticalLayout )
def __init__( self, nodeName, parent=None ): super( componentWidget, self ).__init__( parent ) self.parent = parent def on_context_menu( point, inNodeName ): popMenu = QtGui.QMenu() buildAction = QtGui.QAction( 'Build Joint', popMenu, triggered=lambda a=inNodeName:self.buildNode( a ) ) popMenu.addAction( buildAction ) deleteAction = QtGui.QAction( 'Delete Component', popMenu, triggered=lambda a=inNodeName:self.deleteComponentFromObject( a ) ) popMenu.addAction( deleteAction ) popMenu.exec_( componentLabel.mapToGlobal( point ) ) # Setup layout. verticalLayout = QtGui.QVBoxLayout() verticalLayout.setContentsMargins( 0,0,0,0 ) verticalLayout.setSpacing( 0 ) verticalLayout.setAlignment( QtCore.Qt.AlignTop ) # Label for component componentLabel = QTWidgets.basicLabel( nodeName, 'bold', 10, 'black', '6E9094', inIndent=20 ) componentLabel.setMinimumHeight( 18 ) componentLabel.setContextMenuPolicy( QtCore.Qt.CustomContextMenu ) componentLabel.customContextMenuRequested.connect( lambda point, node=nodeName:on_context_menu( point, node ) ) # Properties propertyStack = QtGui.QVBoxLayout() propertyFrame = QTWidgets.basicFrame() propertyFrame.setMinimumHeight( 40 ) propertyFrame.setMaximumHeight( 40 ) # Add string edit property propertyPlug = NodeUtility.getPlug( nodeName, 'jointName' ) propertyValue = NodeUtility.getPlugValue( propertyPlug ) jointTextLayout = QTWidgets.stringProperty( 'Joint Name', propertyValue ) propertyStack.addLayout( jointTextLayout ) propertyFrame.setLayout( propertyStack ) verticalLayout.addWidget( componentLabel ) verticalLayout.addWidget( propertyFrame ) # Connections textBox = propertyFrame.findChild( QtGui.QLineEdit ) textBox.editingFinished.connect( lambda inPlugName='jointName', inQTType='QLineEdit', inPlugValue=textBox, inNodeName=nodeName :BasicJointComponent( inNodeName ).setComponentAttributeFromQT( inPlugName, inQTType, inPlugValue, inNodeName ) ) self.setLayout( verticalLayout )
def componentGui( inNodeName ): def on_context_menu( point ): popMenu = QtGui.QMenu() popMenu.addAction( 'actionEdit' ) popMenu.addAction( 'actionDelete' ) popMenu.addSeparator() popMenu.addAction( 'actionAdd' ) popMenu.exec_( componentLabel.mapToGlobal( point ) ) mainWidget = QtGui.QWidget() verticalLayout = QtGui.QVBoxLayout( mainWidget ) verticalLayout.setContentsMargins( 0,0,0,0 ) verticalLayout.setSpacing( 0 ) verticalLayout.setAlignment( QtCore.Qt.AlignTop ) # Label for component componentLabel = QTWidgets.basicLabel( inNodeName, 'bold', 10, 'black', 448094 ) componentLabel.setContextMenuPolicy( QtCore.Qt.CustomContextMenu ) componentLabel.customContextMenuRequested.connect( on_context_menu ) propertyFrame = QTWidgets.basicFrame() propertyFrame.setMinimumHeight( 40 ) propertyFrame.setMaximumHeight( 40 ) # Add string edit property plug = NodeUtility.getPlug( inNodeName, 'jointName' ) plugValue = NodeUtility.getPlugValue( plug ) QTWidgets.stringProperty( propertyFrame, 'Joint Name', plugValue ) # Add everything to the vertical layout. verticalLayout.addWidget( componentLabel ) verticalLayout.addWidget( propertyFrame ) #print 'test2: {0}'.format( prop.takeAt(0).widget() ) #print propertyFrame.count() #print propertyFrame.findChild( QtGui.QLineEdit ) #for item in xrange( propertyFrame.count() ): # print propertyFrame.itemAt( item ).widget() return mainWidget
def __init__( self, nodeName, parent=None ): super( componentWidget, self ).__init__( parent ) self.parent = parent def on_context_menu( point, inNodeName ): popMenu = QtGui.QMenu() deleteAction = QtGui.QAction( 'Delete Component', popMenu, triggered=lambda a=inNodeName:self.deleteComponentFromObject( a ) ) popMenu.addAction( deleteAction ) popMenu.exec_( self.componentLabel.mapToGlobal( point ) ) # Colors. userColors = GeneralUtility.getUserDefinedColors( inType=1 ) self.colorList = [] for color in userColors: self.colorList.append( GeneralUtility.convertRGB( color, inType=1 ) ) # Setup layout. verticalLayout = QtGui.QVBoxLayout() verticalLayout.setContentsMargins( 0,0,0,0 ) verticalLayout.setSpacing( 0 ) verticalLayout.setAlignment( QtCore.Qt.AlignTop ) # Label for component #self.componentLabel = QTWidgets.basicLabel( nodeName, 'bold', 10, 'black', '6E9094', inIndent=20 ) self.componentLabel = QtGui.QLabel() self.componentLabel.setText( nodeName ) self.componentLabel.setIndent( 20 ) controlColorPlug = NodeUtility.getPlug( nodeName, 'controlColor' ) self.COLOR = NodeUtility.getPlugValue( controlColorPlug ) self.componentLabel.setStyleSheet( 'font:bold; font-size:10px; color:black; background-color:rgb({0},{1},{2})'.format(self.colorList[self.COLOR-1][0], self.colorList[self.COLOR-1][1], self.colorList[self.COLOR-1][2] ) ) self.componentLabel.setMinimumHeight( 18 ) self.componentLabel.setContextMenuPolicy( QtCore.Qt.CustomContextMenu ) self.componentLabel.customContextMenuRequested.connect( lambda point, nodeName=nodeName:on_context_menu( point, nodeName ) ) propertyFrame = QTWidgets.basicFrame() propertyFrame.setMinimumHeight( 100 ) propertyFrame.setMaximumHeight( 100 ) propertyStack = QtGui.QVBoxLayout() # Add string edit property propertyRow = QtGui.QHBoxLayout() propertyPlug = NodeUtility.getPlug( nodeName, 'controlName' ) propertyValue = NodeUtility.getPlugValue( propertyPlug ) self.textBox = QtGui.QLineEdit() self.textBox.setAlignment( QtCore.Qt.AlignLeft ) self.textBox.setMinimumHeight( 20 ) if propertyValue: self.textBox.setText( propertyValue ) textBoxLabel = QtGui.QLabel() textBoxLabel.setText( 'Control Name' ) textBoxLabel.setAlignment( QtCore.Qt.AlignCenter ) textBoxLabel.setMinimumHeight( 12 ) propertyRow.addWidget( self.textBox ) propertyRow.addWidget( textBoxLabel ) # Colors. colorRow = QtGui.QHBoxLayout() color1Btn = QtGui.QPushButton() color1Btn.setStyleSheet( 'background-color:rgb({0},{1},{2}); width:20'.format( self.colorList[0][0], self.colorList[0][1], self.colorList[0][2] ) ) color1Btn.clicked.connect( lambda a=nodeName, b=1:self.colorChange( a, b ) ) color2Btn = QtGui.QPushButton() color2Btn.setStyleSheet( 'background-color:rgb({0},{1},{2}); width:20'.format( self.colorList[1][0], self.colorList[1][1], self.colorList[1][2] ) ) color2Btn.clicked.connect( lambda a=nodeName, b=2:self.colorChange( a, b ) ) color3Btn = QtGui.QPushButton() color3Btn.setStyleSheet( 'background-color:rgb({0},{1},{2}); width:20'.format( self.colorList[2][0], self.colorList[2][1], self.colorList[2][2] ) ) color3Btn.clicked.connect( lambda a=nodeName, b=3:self.colorChange( a, b ) ) color4Btn = QtGui.QPushButton() color4Btn.setStyleSheet( 'background-color:rgb({0},{1},{2}); width:20'.format( self.colorList[3][0], self.colorList[3][1], self.colorList[3][2] ) ) color4Btn.clicked.connect( lambda a=nodeName, b=4:self.colorChange( a, b ) ) color5Btn = QtGui.QPushButton() color5Btn.setStyleSheet( 'background-color:rgb({0},{1},{2}); width:20'.format( self.colorList[4][0], self.colorList[4][1], self.colorList[4][2] ) ) color5Btn.clicked.connect( lambda a=nodeName, b=5:self.colorChange( a, b ) ) color6Btn = QtGui.QPushButton() color6Btn.setStyleSheet( 'background-color:rgb({0},{1},{2}); width:20'.format( self.colorList[5][0], self.colorList[5][1], self.colorList[5][2] ) ) color6Btn.clicked.connect( lambda a=nodeName, b=6:self.colorChange( a, b ) ) color7Btn = QtGui.QPushButton() color7Btn.setStyleSheet( 'background-color:rgb({0},{1},{2}); width:20'.format( self.colorList[6][0], self.colorList[6][1], self.colorList[6][2] ) ) color7Btn.clicked.connect( lambda a=nodeName, b=7:self.colorChange( a, b ) ) color8Btn = QtGui.QPushButton() color8Btn.setStyleSheet( 'background-color:rgb({0},{1},{2}); width:20'.format( self.colorList[7][0], self.colorList[7][1], self.colorList[7][2] ) ) color8Btn.clicked.connect( lambda a=nodeName, b=8:self.colorChange( a, b ) ) colorRow.addWidget( color1Btn ) colorRow.addWidget( color2Btn ) colorRow.addWidget( color3Btn ) colorRow.addWidget( color4Btn ) colorRow.addWidget( color5Btn ) colorRow.addWidget( color6Btn ) colorRow.addWidget( color7Btn ) colorRow.addWidget( color8Btn ) # Edit buttons. self.editButton = QtGui.QToolButton() self.editButton.setCheckable( True ) self.editButton.setText( 'Edit Control' ) self.editButton.toggled.connect( lambda:self.editCurveProperties() ) propertyStack.addLayout( propertyRow ) propertyStack.addLayout( colorRow ) propertyStack.addWidget( self.editButton ) propertyFrame.setLayout( propertyStack ) # Add everything to the vertical layout. verticalLayout.addWidget( self.componentLabel ) verticalLayout.addWidget( propertyFrame ) # Connections self.textBox.editingFinished.connect( lambda inPlugName='controlName', inQTType='QLineEdit', inPlugValue=self.textBox, inNodeName=nodeName :CurveControlComponent( inNodeName ).setComponentAttributeFromQT( inPlugName, inQTType, inPlugValue, inNodeName ) ) #return mainWidget self.setLayout( verticalLayout )
def __init__( self, nodeName, parent=None ): super( componentWidget, self ).__init__( parent ) self.parent = parent def on_context_menu( point, inNodeName ): popMenu = QtGui.QMenu() deleteAction = QtGui.QAction( 'Delete Component', popMenu, triggered=lambda a=inNodeName:self.deleteComponentFromObject( a ) ) popMenu.addAction( deleteAction ) popMenu.exec_( self.componentLabel.mapToGlobal( point ) ) # Setup layout. verticalLayout = QtGui.QVBoxLayout() verticalLayout.setContentsMargins( 0,0,0,0 ) verticalLayout.setSpacing( 0 ) verticalLayout.setAlignment( QtCore.Qt.AlignTop ) # Label for component componentLabel = QTWidgets.basicLabel( nodeName, 'bold', 10, 'black', '6E9094', inIndent=20 ) componentLabel.setMinimumHeight( 18 ) componentLabel.setContextMenuPolicy( QtCore.Qt.CustomContextMenu ) componentLabel.customContextMenuRequested.connect( lambda point, nodeName=nodeName:on_context_menu( point, nodeName ) ) # Properties propertyStack = QtGui.QVBoxLayout() propertyFrame = QTWidgets.basicFrame() propertyFrame.setMinimumHeight( 120 ) propertyFrame.setMaximumHeight( 120 ) # Add string edit property characterPlug = NodeUtility.getPlug( nodeName, 'characterName' ) characterValue = NodeUtility.getPlugValue( characterPlug ) characterTextLayout = QTWidgets.stringProperty( 'Character Name', characterValue ) skelGroupPlug = NodeUtility.getPlug( nodeName, 'skeletonGroupName' ) skelGroupValue = NodeUtility.getPlugValue( skelGroupPlug ) skelGroupTextLayout = QTWidgets.stringProperty( 'Skeleton Group Name', skelGroupValue ) rigGroupPlug = NodeUtility.getPlug( nodeName, 'rigGroupName' ) rigGroupValue = NodeUtility.getPlugValue( rigGroupPlug ) rigGroupTextLayout = QTWidgets.stringProperty( 'Rigging Group Name', rigGroupValue ) ''' ADD MODULES BUTTON. PROBABLY HAVE A POPUP LISTING UNCONNECTED MODULES IN THE SCENE?? ''' # Add everything to the vertical layout. propertyStack.addLayout( characterTextLayout ) propertyStack.addLayout( skelGroupTextLayout ) propertyStack.addLayout( rigGroupTextLayout ) propertyFrame.setLayout( propertyStack ) verticalLayout.addWidget( componentLabel ) verticalLayout.addWidget( propertyFrame ) # Connections charTextBox = propertyFrame.findChild( QtGui.QLineEdit, 'Character Name' ) charTextBox.editingFinished.connect( lambda inPlugName='characterName', inQTType='QLineEdit', inPlugValue=charTextBox, inNodeName=nodeName :CharacterRootComponent( inNodeName ).setComponentAttributeFromQT( inPlugName, inQTType, inPlugValue, inNodeName ) ) skelTextBox = propertyFrame.findChild( QtGui.QLineEdit, 'Skeleton Group Name' ) skelTextBox.editingFinished.connect( lambda inPlugName='skeletonGroupName', inQTType='QLineEdit', inPlugValue=skelTextBox, inNodeName=nodeName :CharacterRootComponent( inNodeName ).setComponentAttributeFromQT( inPlugName, inQTType, inPlugValue, inNodeName ) ) rigTextBox = propertyFrame.findChild( QtGui.QLineEdit, 'Rigging Group Name' ) rigTextBox.editingFinished.connect( lambda inPlugName='rigGroupName', inQTType='QLineEdit', inPlugValue=rigTextBox, inNodeName=nodeName :CharacterRootComponent( inNodeName ).setComponentAttributeFromQT( inPlugName, inQTType, inPlugValue, inNodeName ) ) #return mainWidget self.setLayout( verticalLayout )
def __init__( self ): super( UIBitsTools, self ).__init__() layout = QtGui.QVBoxLayout( self ) # Build bits controls # Presets moduleRootBtn = QTWidgets.imageButton( 'Create Module', ':/riggingUI/icons/preset_module.png', [32,32] ) moduleRootBtn.clicked.connect( lambda a=1:self.preset_buttons(a) ) characterRootBtn = QTWidgets.imageButton( 'Create Character', ':/riggingUI/icons/preset_character.png', [32,32] ) characterRootBtn.clicked.connect( lambda a=2:self.preset_buttons(a) ) # Primitive buttons. sphereBtn = QTWidgets.imageButton( 'Sphere', ':/riggingUI/icons/bit_sphere.png', [32,32] ) sphereBtn.clicked.connect( self.primitive_buttons ) boxBtn = QTWidgets.imageButton( 'Box', ':/riggingUI/icons/bit_box.png', [32,32] ) boxBtn.clicked.connect( self.primitive_buttons ) cylinderBtn = QTWidgets.imageButton( 'Cylinder', ':/riggingUI/icons/bit_cylinder.png', [32,32] ) cylinderBtn.clicked.connect( self.primitive_buttons ) coneBtn = QTWidgets.imageButton( 'Cone', ':/riggingUI/icons/bit_cone.png', [32,32] ) coneBtn.clicked.connect( self.primitive_buttons ) torusBtn = QTWidgets.imageButton( 'Torus', ':/riggingUI/icons/bit_torus.png', [32,32] ) torusBtn.clicked.connect( self.primitive_buttons ) # Bit tools. matchTranslationBtn = QTWidgets.imageTextButton( 'Match Translation', ':/riggingUI/icons/icon_match_translation.png', [16,16] ) matchTranslationBtn.clicked.connect( lambda a='tran':TransformUtility.matchTransforms( a ) ) matchRotationBtn = QTWidgets.imageTextButton( 'Match Rotation', ':/riggingUI/icons/icon_match_rotation.png', [16,16] ) matchRotationBtn.clicked.connect( lambda a='rot':TransformUtility.matchTransforms( a ) ) matchAllBtn = QTWidgets.imageTextButton( 'Match All', ':/riggingUI/icons/icon_match_all.png', [16,16] ) matchAllBtn.clicked.connect( lambda a='all':TransformUtility.matchTransforms( a ) ) copyBitSettingsBtn = QTWidgets.imageTextButton( 'Copy Bit Settings', ':/riggingUI/icons/bit_copy_settings.png', [16,16] ) copyBitSettingsBtn.clicked.connect( lambda:NodeUtility.copyBitSettings() ) addChildBtn = QTWidgets.imageTextButton( 'Add Child', ':/riggingUI/icons/bit_add_child.png', [16,16] ) addChildBtn.clicked.connect( lambda:NodeUtility.setBitChild() ) deleteChildBtn = QTWidgets.imageTextButton( 'Delete Child', ':/riggingUI/icons/bit_delete_child.png', [16,16] ) deleteChildBtn.clicked.connect( lambda:NodeUtility.deleteBitChild() ) # Build bits layout. # Presets. presetsHeader = QTWidgets.basicLabel( 'Bit Presets', 'bold', 14, 'white', '2B2B30' ) presetsHeader.setMinimumHeight( 30 ) presetsHeader.setAlignment( QtCore.Qt.AlignCenter ) presetsFrame = QTWidgets.basicFrame() presetsLayout = QtGui.QHBoxLayout() presetsLayout.addWidget( moduleRootBtn ) presetsLayout.addWidget( characterRootBtn ) presetsFrame.setLayout( presetsLayout ) # Primitive buttons bitPrimitivesHeader = QTWidgets.basicLabel( 'Bit Primitives', 'bold', 14, 'white', '2B2B30' ) bitPrimitivesHeader.setMinimumHeight( 30 ) bitPrimitivesHeader.setAlignment( QtCore.Qt.AlignCenter ) primitiveFrame = QTWidgets.basicFrame() primitiveLayout = QtGui.QHBoxLayout() primitiveLayout.addWidget( sphereBtn ) primitiveLayout.addWidget( boxBtn ) primitiveLayout.addWidget( cylinderBtn ) primitiveLayout.addWidget( coneBtn ) primitiveLayout.addWidget( torusBtn ) primitiveFrame.setLayout( primitiveLayout ) # Bit tools. bitToolsHeader = QTWidgets.basicLabel( 'Bit Tools', 'bold', 14, 'white', '2B2B30' ) bitToolsHeader.setMinimumHeight( 30 ) bitToolsHeader.setAlignment( QtCore.Qt.AlignCenter ) bitToolsGrid = QtGui.QGridLayout() bitToolsGrid.setColumnMinimumWidth( 0, 100 ) bitToolsGrid.setColumnMinimumWidth( 1, 100 ) bitToolsGrid.setColumnMinimumWidth( 2, 100 ) bitToolsGrid.setSpacing( 2 ) bitToolsGrid.setContentsMargins( 0,0,0,0 ) # widget, row, col bitToolsGrid.addWidget( matchTranslationBtn, 0, 0 ) bitToolsGrid.addWidget( matchRotationBtn, 0, 1 ) bitToolsGrid.addWidget( matchAllBtn, 0, 2 ) bitToolsGrid.addWidget( copyBitSettingsBtn, 1, 0 ) bitToolsGrid.addWidget( addChildBtn, 1, 1 ) bitToolsGrid.addWidget( deleteChildBtn, 1, 2 ) # Added the bits widgets to the sub-layout of the main window. layout.addWidget( presetsHeader ) layout.addWidget( presetsFrame ) layout.addWidget( bitPrimitivesHeader ) layout.addWidget( primitiveFrame ) layout.addWidget( bitToolsHeader ) layout.addLayout( bitToolsGrid )