def requiredAttributes( self, *args, **kwargs ):
'''
kwargs
curveType: Type of curve control to make. Square, triangle, arrow, plus, pyramid, circle, ringDirection.
'''
super( CurveControlComponent, self ).requiredAttributes()
NodeUtility.addPlug( self.newNode, 'controlName', 'dataType', 'string' )
NodeUtility.addPlug( self.newNode, 'curveType', 'dataType', 'string' )
self.setAttribute( 'curveType', kwargs['curveType'], self.newNode )
# Add attribute for control color.
NodeUtility.addPlug( self.newNode, 'controlColor', 'attributeType', 'byte' )
self.setAttribute( 'controlColor', 1, self.newNode )
# Control transform relative to the parent.
NodeUtility.addPlug( self.newNode, 'controlPosition', 'attributeType', 'float3' )
NodeUtility.addPlug( self.newNode, 'controlRotation', 'attributeType', 'float3' )
NodeUtility.addPlug( self.newNode, 'controlScale', 'attributeType', 'float3' )
# Add attributes for the curve CVs.
tempCurve = self.createCurveControl( '{0}TempCurve'.format( self.newNode ), kwargs['curveType'] )
tempCurveName = OpenMaya.MDagPath.getAPathTo( tempCurve ).fullPathName()
tempCurvePoints = NurbsCurveUtility.getCurveCvs( tempCurveName )
NurbsCurveUtility.addCurveValues( self.newNode, tempCurvePoints )
# Update curve transform values.
storeControlTransforms( tempCurveName, self.newNode )
# Clean up.
cmds.delete( tempCurveName )
def editCurveProperties( self ):
'''
Activates the control so the user can edit it's properties.
'''
if self.editButton.isChecked():
if self.textBox.text():
# Lock the components UI.
if self.parent.selectedLockActive is False:
self.parent.lockSelection()
# Create the curve.
curveType = CurveControlComponent( self.componentLabel.text() ).curveType
node = CurveControlComponent( self.componentLabel.text() ).createCurveControl( self.textBox.text(), curveType )
controlName = OpenMaya.MDagPath.getAPathTo( node ).fullPathName()
# Parent the control to the bit.
parentNode = CurveControlComponent( self.componentLabel.text() ).parentNode[0]
cmds.parent( controlName, parentNode )
self.CONTROL = OpenMaya.MDagPath.getAPathTo( node ).fullPathName()
# Set the control to the transform matrix.
applyStoredTransforms( self.componentLabel.text(), self.CONTROL )
# Get the saved properties and apply them to the curve.
cvList = NurbsCurveUtility.readCurveValues( self.componentLabel.text() )
cvPointArray = NurbsCurveUtility.buildCVPointArray( cvList )
NurbsCurveUtility.setCurveCvs( self.CONTROL, cvPointArray )
# Color.
GeneralUtility.setUserColor( self.CONTROL, userColor=self.COLOR )
else:
raise ValueError( '{0} does not have a Control Name set.'.format( self.componentLabel.text() ) )
else:
if self.CONTROL is not None:
# Read the control properties and save them to the component node.
cvList = NurbsCurveUtility.getCurveCvs( self.CONTROL )
NurbsCurveUtility.writeCurveValues( self.componentLabel.text(), cvList )
# Update the transform matrix.
storeControlTransforms( self.CONTROL, self.componentLabel.text() )
# Delete the control
cmds.delete( self.CONTROL )
self.CONTROL = None
# Unlock the UI.
if self.parent.selectedLockActive:
self.parent.lockSelection()
