def getTrackManagerNode(cls, trackSetNode =None, createIfMissing =False):
""" Returns the name of the track manager nodeName for the current
Cadence scene.
trackSetNode: The track set nodeName on which to find the track
manager nodeName. If no nodeName is specified the method will look
it up internally.
createIfMissing: If true and no track manager nodeName is found one
will be created and connected to the track set nodeName, which will
also be created if one does not exist. """
if not trackSetNode:
trackSetNode = cls.getTrackSetNode(createIfMissing=createIfMissing)
connects = cmds.listConnections(
trackSetNode + '.usedBy',
source=True,
destination=False)
if connects:
for node in connects:
if cmds.nodeType(node) == 'trackManager':
return node
if createIfMissing:
node = cmds.createNode('trackManager')
cmds.connectAttr(
node + '.trackSet', trackSetNode + '.usedBy', force=True)
return node
return None
def _createMeshPointNode(self, shapeData):
self._removeMeshPointNode()
try:
node = cmds.createNode('closestPointOnMesh', skipSelect=True)
self._meshPointNode = node
except Exception as err:
print(Logger.createErrorMessage(u'ERROR: Unable to create mesh point node', err))
self._removeMeshPointNode()
return False
try:
cmds.connectAttr(shapeData['name'] + '.message', node + '.inMesh', force=True)
except Exception as err:
print(Logger.createErrorMessage(u'ERROR: Unable to connect mesh point node to shape', err))
self._removeMeshPointNode()
return False
return True
def _createMeshPointNode(self, shapeData):
self._removeMeshPointNode()
try:
node = cmds.createNode('closestPointOnMesh', skipSelect=True)
self._meshPointNode = node
except Exception as err:
print(Logger.createErrorMessage(u'ERROR: Unable to create mesh point node', err))
self._removeMeshPointNode()
return False
try:
cmds.connectAttr(shapeData['name'] + '.message', node + '.inMesh', force=True)
except Exception as err:
print(Logger.createErrorMessage(u'ERROR: Unable to connect mesh point node to shape', err))
self._removeMeshPointNode()
return False
return True
def createTrackNode(cls, uid, trackSetNode =None, props =None):
""" A track node consists of a triangular pointer (left = red, right =
green) which is selectable but only allows rotateY, translateX, and
translateZ. The node has a child, a transform called inverter, which
serves to counteract the scaling in x and z that is applied to the
triangular node. There are two orthogonal rulers (width and
length). Width and length uncertainty is represented by rectangular
bars at the ends of the rulers. In Maya one can directly adjust
track position (translateX and translateZ) and orientation
(rotationY); other attributes are adjusted only through the UI. """
if not trackSetNode:
trackSetNode = TrackSceneUtils.getTrackSetNode()
if not trackSetNode:
return None
node = cls.getTrackNode(uid, trackSetNode=trackSetNode)
if node:
return node
# Set up dimensional constants for the track node
nodeThickness = 1.0
thetaBreadth = 0.1
thetaThickness = 0.5
barBreadth = 2.0
barThickness = 0.5
rulerBreadth = 1.0
rulerThickness = 0.25
epsilon = 1.0
# Create an isoceles triangle pointer, with base aligned with X, and
# scaled by node.width. The midpoint of the base is centered on the
# 'track center' and the altitude extends from that center of the track
# 'anteriorly' to the perimeter of the track's profile (if present, else
# estimated). The node is scaled longitudinally (in z) based on the
# distance zN (the 'anterior' length of the track, in cm). The triangle
# is initially 1 cm on a side.
sideLength = 1.0
node = cmds.polyPrism(
length=nodeThickness,
sideLength=sideLength,
numberOfSides=3,
subdivisionsHeight=1,
subdivisionsCaps=0,
axis=(0, 1, 0),
createUVs=0,
constructionHistory=0,
name='Track0')[0]
# Point the triangle down the +Z axis
cmds.rotate(0.0, -90.0, 0.0)
# push it down below ground level so that the two rulers are just
# submerged, and scale the triangle in Z to match its width (1 cm) so it
# is ready to be scaled
cmds.move(0, -(nodeThickness/2.0 + rulerThickness), math.sqrt(3.0)/6.0)
# move the node's pivot to the 'base' of the triangle so it scales
# outward from that point
cmds.move(
0, 0, 0, node + ".scalePivot", node + ".rotatePivot", absolute=True)
cmds.scale(2.0/math.sqrt(3.0), 1.0, 100.0)
cmds.makeIdentity(
apply=True,
translate=True,
rotate=True,
scale=True,
normal=False)
# Set up the cadence attributes
cmds.addAttr(
longName='cadence_width',
shortName=TrackPropEnum.WIDTH.maya,
niceName='Width')
cmds.addAttr(
longName='cadence_widthUncertainty',
shortName=TrackPropEnum.WIDTH_UNCERTAINTY.maya,
niceName='Width Uncertainty')
cmds.addAttr(
longName='cadence_length',
shortName=TrackPropEnum.LENGTH.maya,
niceName='Length')
cmds.addAttr(
longName='cadence_lengthUncertainty',
shortName=TrackPropEnum.LENGTH_UNCERTAINTY.maya,
niceName='Length Uncertainty')
cmds.addAttr(
longName='cadence_lengthRatio',
shortName=TrackPropEnum.LENGTH_RATIO.maya,
niceName='Length Ratio')
cmds.addAttr(
longName='cadence_rotationUncertainty',
shortName=TrackPropEnum.ROTATION_UNCERTAINTY.maya,
niceName='Rotation Uncertainty')
cmds.addAttr(
longName='cadence_uniqueId',
shortName=TrackPropEnum.UID.maya,
dataType='string',
niceName='Unique ID')
# Construct a ruler representing track width, then push it down just
# below ground level, and ake it non-selectable. Drive its scale by the
# node's width attribute.
widthRuler = cmds.polyCube(
axis=(0, 1, 0),
width=100.0,
height=rulerThickness,
depth=rulerBreadth,
subdivisionsX=1,
subdivisionsY=1,
createUVs=0,
constructionHistory=0,
name='WidthRuler')[0]
# Push it down to just rest on the triangular node (which is already
# submerged by the thickness of the ruler and half the node thickness.
cmds.move(0.0, -rulerThickness/2.0, 0.0)
cmds.setAttr(widthRuler + '.overrideEnabled', 1)
cmds.setAttr(widthRuler + '.overrideDisplayType', 2)
# Construct a ruler representing track length and push it down the same
# as the width ruler, and make it non-selectable. Its length will be
# driven by the node's length attribute.
lengthRuler = cmds.polyCube(
axis=(0, 1, 0),
width=rulerBreadth,
height=rulerThickness,
depth=100.0,
subdivisionsX=1,
subdivisionsY=1,
createUVs=0,
constructionHistory=0,
name='LengthRuler')[0]
cmds.move(0.0, -rulerThickness/2.0, 0.0)
cmds.setAttr(lengthRuler + '.overrideEnabled', 1)
cmds.setAttr(lengthRuler + '.overrideDisplayType', 2)
# Now construct 'error bars' to the North, South, West, and East of the
# node, to visualize uncertainty in width (West and East bars) and
# length (North and South bars), and push them just below ground level,
# and make them non-selectable.
barN = cmds.polyCube(
axis=(0,1,0),
width=barBreadth,
height=barThickness,
depth=100.0,
subdivisionsX=1,
subdivisionsY=1,
createUVs=0,
constructionHistory=0,
name='BarN')[0]
cmds.move(0, -(barThickness/2 + rulerThickness), 0)
cmds.setAttr(barN + '.overrideEnabled', 1)
cmds.setAttr(barN + '.overrideDisplayType', 2)
barS = cmds.polyCube(
axis=(0, 1, 0),
width=barBreadth,
height=barThickness,
depth=100.0,
subdivisionsX=1,
subdivisionsY=1,
createUVs=0,
constructionHistory=0,
name='BarS')[0]
cmds.move(0, -(barThickness/2 + rulerThickness), 0)
cmds.setAttr(barS + '.overrideEnabled', 1)
cmds.setAttr(barS + '.overrideDisplayType', 2)
barW = cmds.polyCube(
axis=(0, 1, 0),
width=100.0,
height=barThickness,
depth=barBreadth,
subdivisionsX=1,
subdivisionsY=1,
createUVs=0,
constructionHistory=0,
name='BarW')[0]
cmds.move(0, -(barThickness/2 + rulerThickness), 0)
cmds.setAttr(barW + '.overrideEnabled', 1)
cmds.setAttr(barW + '.overrideDisplayType', 2)
barE = cmds.polyCube(
axis=(0, 1, 0),
width=100.0,
height=barThickness,
depth=barBreadth,
subdivisionsX=1,
subdivisionsY=1,
createUVs=0,
constructionHistory=0,
name='BarE')[0]
cmds.move(0, -(barThickness/2 + rulerThickness), 0)
cmds.setAttr(barE + '.overrideEnabled', 1)
cmds.setAttr(barE + '.overrideDisplayType', 2)
# Create two diverging lines that indicate rotation uncertainty (plus
# and minus), with their pivots placed so they extend from the node
# center, and each is made non-selectable. First make the indicator of
# maximum (counterclockwise) estimated track rotation
thetaPlus = cmds.polyCube(
axis=(0, 1, 0),
width=thetaBreadth,
height=thetaThickness,
depth=1.0,
subdivisionsX=1,
subdivisionsY=1,
createUVs=0,
constructionHistory=0,
name='ThetaPlus')[0]
cmds.setAttr(thetaPlus + '.overrideEnabled', 1)
cmds.setAttr(thetaPlus + '.overrideDisplayType', 2)
# Next, construct the indicator of the minimum (clockwise) estimate of
# track rotation
thetaMinus = cmds.polyCube(
axis=(0, 1, 0),
width=thetaBreadth,
height=thetaThickness,
depth=1.0,
subdivisionsX=1,
subdivisionsY=1,
createUVs=0,
constructionHistory=0,
name='ThetaMinus')[0]
cmds.setAttr(thetaMinus + '.overrideEnabled', 1)
cmds.setAttr(thetaMinus + '.overrideDisplayType', 2)
# The two width 'error bars' will be translated outward from the node
# center. First, the width attribute is converted from meters (as it
# comes from the database) to centimeters; the computation is available
# in the output of the node 'width'.
width = cmds.createNode('multiplyDivide', name='width')
cmds.setAttr(width + '.operation', 1)
cmds.setAttr(width + '.input1X', 100.0)
cmds.connectAttr(
node + '.' + TrackPropEnum.WIDTH.maya, width + '.input2X')
# Translate barW in x by width/2.0; output is in xW.outputX
xW = cmds.createNode('multiplyDivide', name = 'xW')
cmds.setAttr(xW + '.operation', 2)
cmds.connectAttr(width + '.outputX', xW + '.input1X')
cmds.setAttr(xW + '.input2X', 2.0)
cmds.connectAttr(xW + '.outputX', barW + '.translateX')
# Translate barE in x by -width/2.0; output is in xE.outputX
xE = cmds.createNode('multiplyDivide', name = 'xE')
cmds.setAttr(xE + '.operation', 2) # division operation
cmds.connectAttr(width + '.outputX', xE + '.input1X')
cmds.setAttr(xE + '.input2X', -2.0)
cmds.connectAttr(xE + '.outputX', barE + '.translateX')
# Now regarding length, first convert the node.length attribute from
# meters to centimeters. This computation is available in the output of
# the node 'length'
length = cmds.createNode('multiplyDivide', name='length')
cmds.setAttr(length + '.operation', 1)
cmds.setAttr(length + '.input1X', 100.0)
cmds.connectAttr(
node + '.' + TrackPropEnum.LENGTH.maya, length + '.input2X')
# scale thetaPlus and thetaMinus by length (since they are 1 cm,
# multiply by length in cm)
cmds.connectAttr(length + '.outputX', thetaPlus + '.scaleZ')
cmds.connectAttr(length + '.outputX', thetaMinus + '.scaleZ')
# Then barN is translated forward in z by zN = lengthRatio*length
# (centimeters)
zN = cmds.createNode('multiplyDivide', name='zN')
cmds.setAttr(zN + '.operation', 1)
cmds.connectAttr(
node + '.' + TrackPropEnum.LENGTH_RATIO.maya, zN + '.input1X')
cmds.connectAttr(length + '.outputX', zN + '.input2X')
cmds.connectAttr(zN + '.outputX', barN + '.translateZ')
# Next, translate barS backward in z by (zN - length); output is in
# zS.output1D
zS = cmds.createNode('plusMinusAverage', name='sZ')
cmds.setAttr(zS + '.operation', 2)
cmds.connectAttr(zN + '.outputX', zS + '.input1D[0]')
cmds.connectAttr(length + '.outputX', zS + '.input1D[1]')
cmds.connectAttr(zS + '.output1D', barS + '.translateZ')
# Next, compute the half length, hl = length/2.0 (centimeters)
hl = cmds.createNode('multiplyDivide', name='hl')
cmds.setAttr(hl + '.operation', 2)
cmds.connectAttr(length + '.outputX', hl + '.input1X')
cmds.setAttr(hl + '.input2X', 2.0)
# Translate lengthRuler along z by zL = (zN - hl) (centimeters)
zL = cmds.createNode('plusMinusAverage', name='zL')
cmds.setAttr(zL + '.operation', 2)
cmds.connectAttr(zN + '.outputX', zL + '.input1D[0]')
cmds.connectAttr(hl + '.outputX', zL + '.input1D[1]')
cmds.connectAttr(zL + '.output1D', lengthRuler + '.translateZ')
# Scale the four 'error bars' to represent the width and length
# uncertainties (centimeters)
cmds.connectAttr(
node + "." + TrackPropEnum.WIDTH_UNCERTAINTY.maya,
barW + '.scaleX')
cmds.connectAttr(
node + "." + TrackPropEnum.WIDTH_UNCERTAINTY.maya,
barE + '.scaleX')
cmds.connectAttr(
node + "." + TrackPropEnum.LENGTH_UNCERTAINTY.maya,
barN + '.scaleZ')
cmds.connectAttr(
node + "." + TrackPropEnum.LENGTH_UNCERTAINTY.maya,
barS + '.scaleZ')
# Create an 'inverter' transform under which all the other parts are
# hung as children, which counteracts scaling applied to its parent
# triangular node.
inverter = cmds.createNode('transform', name='inverter')
# drive the inverter's .scaleX and .scaleZ as the inverse of the parent
# node's scale values
sx = cmds.createNode('multiplyDivide', name='sx')
cmds.setAttr(sx + '.operation', 2)
cmds.setAttr(sx + '.input1X', 1.0)
cmds.connectAttr(node + '.scaleX', sx + '.input2X')
cmds.connectAttr(sx + '.outputX', inverter + '.scaleX')
sz = cmds.createNode('multiplyDivide', name='sz')
cmds.setAttr(sz + '.operation', 2)
cmds.setAttr(sz + '.input1X', 1.0)
cmds.connectAttr(node + '.scaleZ', sz + '.input2X')
cmds.connectAttr(sz + '.outputX', inverter + '.scaleZ')
# Assemble the parts as children under the scale inverter node
cmds.parent(lengthRuler, inverter)
cmds.parent(widthRuler, inverter)
cmds.parent(barN, inverter)
cmds.parent(barS, inverter)
cmds.parent(barW, inverter)
cmds.parent(barE, inverter)
cmds.parent(thetaPlus, inverter)
cmds.parent(thetaMinus, inverter)
cmds.parent(inverter, node)
# Rotate thetaPlus and thetaMinus about the Y axis to indicate
# rotational uncertainty
cmds.connectAttr(
node + '.' + TrackPropEnum.ROTATION_UNCERTAINTY.maya,
node + '|' + inverter + '|' + thetaPlus + '.rotateY')
neg = cmds.createNode('multiplyDivide', name='negative')
cmds.setAttr(neg + '.operation', 1)
cmds.setAttr(neg + '.input1X', -1.0)
cmds.connectAttr(
node + '.' + TrackPropEnum.ROTATION_UNCERTAINTY.maya,
neg + '.input2X')
cmds.connectAttr(
neg + '.outputX',
node + '|' + inverter + '|' + thetaMinus + '.rotateY')
# Disable some transforms of the node
cmds.setAttr(node + '.rotateX', lock=True)
cmds.setAttr(node + '.rotateZ', lock=True)
cmds.setAttr(node + '.scaleY', lock=True)
cmds.setAttr(node + '.translateY', lock=True)
# Now, the width of the triangle will be driven by its width attribute
# (driving .scaleX)
cmds.connectAttr(node + '.width', node + '.scaleX')
# The quantity zN is used to scale length of the triangle
cmds.connectAttr(zN + '.outputX', node + '.scaleZ')
# Scale the 'length' (in x) of the width ruler
cmds.connectAttr(
node + '.width', node + '|' + inverter + '|WidthRuler.scaleX')
# Scale the length of the length ruler
cmds.connectAttr(
node + '.length', node + '|' + inverter + '|LengthRuler.scaleZ')
# Translate the track node epsilon below ground level (to reveal the
# overlaid track siteMap)
cmds.move(0, -epsilon, 0, node)
# Initialize all the properties from the dictionary
if props:
cls.setTrackProps(node, props)
else:
print('in createTrackNode: properties not provided')
return node
# Add the new nodeName to the Cadence track scene set, color it, and
# we're done
cmds.sets(node, add=trackSetNode)
cls.colorTrackNode(node, props)
return node
