def loadFromRefLocatorShape(self, refLocatorShape):
self.refFilename = cleanupPath(m.getAttr('{}.{}'.format(refLocatorShape, ATTR_REF_FILENAME)))
self.idString = self.generateIdString(self.refFilename)
self.refShortName = self.generateShortName(self.refFilename)
self.refLocator = TransformHandle(shape=refLocatorShape)
self.setAnnotation(self.refShortName)
self.refNode = self.idString + REF_NODE_SUFFIX
self.instanceSource = '|{}|{}'.format(INSTANCES_SOURCE_GROUP, self.idString)
self.importSource = '|{}|{}'.format(IMPORT_SOURCE_GROUP, self.idString)
self.active = self.getActiveStateFromMaya()
def doReplacement(replaceDB):
notExistingSources = set()
createdRefs = []
for tr, path in replaceDB.items():
if path in notExistingSources:
continue
if not os.path.exists(os.path.expandvars(path)):
notExistingSources.add(path)
log.logAppend('Path does not exists: {}. Replacement skipped.'.format(path))
continue
refHandle = RefHandle()
refHandle.createNew(path)
createdRefs.append(refHandle)
worldRP = m.xform(tr, q=True, rotatePivot=True, worldSpace=True)
m.xform(refHandle.refLocator.transform, translation=worldRP, absolute=True, worldSpace=True)
transformParent = getParent(tr)
if transformParent:
newRefLocTransform = getLongName(m.parent(refHandle.refLocator.transform, transformParent)[0])
refHandle.setRefLocator(TransformHandle(transform=newRefLocTransform))
rotation = m.xform(tr, q=True, rotation=True, objectSpace=True)
scale = m.xform(tr, q=True, scale=True, objectSpace=True, relative=True)
shear = m.xform(tr, q=True, shear=True, objectSpace=True, relative=True)
m.xform(refHandle.refLocator.transform, rotation=rotation, scale=scale, shear=shear)
if m.objExists(tr):
m.delete(tr)
return createdRefs
def getAnnotationTransformHandle(self):
if self.isValid():
annotationShapes = self.refLocator.getChildren(allDescendants=True, typ='annotationShape')
if len(annotationShapes) == 1:
return TransformHandle(shape=annotationShapes[0])
else:
return None
def createAnnotation(self):
if not self.isValid():
return
annotationShapes = self.refLocator.getChildren(allDescendants=True, typ='annotationShape')
for s in annotationShapes:
m.delete(getParent(s))
annotationSh = m.annotate(self.refLocator.transform, p=(0, -0.5, 0))
annotationTr = getParent(annotationSh)
annotationTr = m.parent(annotationTr, self.refLocator.transform, relative=True)[0]
lockTransformations(annotationTr)
self.annotation = TransformHandle(transform=getLongName(annotationTr))
m.setAttr(self.annotation.shape + '.displayArrow', False)
m.setAttr(self.annotation.transform + '.overrideEnabled', True)
m.setAttr(self.annotation.transform + '.overrideDisplayType', 2)
class RefHandle(object):
def __init__(self):
self.refFilename = None
self.idString = None
self.refShortName = None
self.refLocator = None
self.annotation = None
self.refNode = None
self.instanceSource = None
self.importSource = None
self.active = False
def __str__(self):
return 'refFilename={}, idString={}, refShortName={}, refLocator=[{}], annotation=[{}],' \
'refNode={}, instanceSource={}, importSource={}, active={}' \
.format(
self.refFilename,
self.idString,
self.refShortName,
self.refLocator,
self.annotation,
self.refNode,
self.instanceSource,
self.importSource,
self.active,
)
def loadFromRefLocatorShape(self, refLocatorShape):
self.refFilename = cleanupPath(m.getAttr('{}.{}'.format(refLocatorShape, ATTR_REF_FILENAME)))
self.idString = self.generateIdString(self.refFilename)
self.refShortName = self.generateShortName(self.refFilename)
self.refLocator = TransformHandle(shape=refLocatorShape)
self.setAnnotation(self.refShortName)
self.refNode = self.idString + REF_NODE_SUFFIX
self.instanceSource = '|{}|{}'.format(INSTANCES_SOURCE_GROUP, self.idString)
self.importSource = '|{}|{}'.format(IMPORT_SOURCE_GROUP, self.idString)
self.active = self.getActiveStateFromMaya()
def createNew(self, refFilename):
refLocatorTr = m.spaceLocator(p=(0, 0, 0))[0]
refLocatorTr = getLongName(m.rename(refLocatorTr, self.generateShortName(refFilename) + REF_LOCATOR_SUFFIX))
refLocatorSh = getShape(refLocatorTr)
m.addAttr(
refLocatorSh,
at='message',
shortName=ATTR_REF_NODE_MESSAGE_NAMES[0],
longName=ATTR_REF_NODE_MESSAGE_NAMES[1],
niceName=ATTR_REF_NODE_MESSAGE_NAMES[2]
)
m.addAttr(
refLocatorSh,
dt='string',
shortName=ATTR_REF_FILENAME_NAMES[0],
longName=ATTR_REF_FILENAME_NAMES[1],
niceName=ATTR_REF_FILENAME_NAMES[2]
)
m.setAttr('{}.{}'.format(refLocatorSh, ATTR_REF_FILENAME), refFilename, typ='string')
self.loadFromRefLocatorShape(refLocatorSh)
# noinspection PyMethodMayBeStatic
def generateIdString(self, refFilename):
s = os.path.splitext(refFilename)[0]
if isPathRelative(refFilename):
s = REF_ROOT_VAR_NAME + s[len(REF_ROOT_VAR_NAME_P):]
return re.sub('[^0-9a-zA-Z_]+', '__', s).lower()
# noinspection PyMethodMayBeStatic
def generateShortName(self, longFilename):
return os.path.splitext(os.path.basename(longFilename))[0]
def getActiveStateFromMaya(self):
if not m.objExists(self.refNode):
return False
return m.isConnected(self.refNode + '.message', self.refLocator.shape + '.' + ATTR_REF_NODE_MESSAGE)
def getAnnotationTransformHandle(self):
if self.isValid():
annotationShapes = self.refLocator.getChildren(allDescendants=True, typ='annotationShape')
if len(annotationShapes) == 1:
return TransformHandle(shape=annotationShapes[0])
else:
return None
def isValid(self):
return (self.refLocator is not None) and (self.refLocator.exists())
def setAnnotation(self, text):
if (self.annotation is None) or (not self.annotation.exists()):
self.annotation = self.getAnnotationTransformHandle()
if self.annotation is None:
self.createAnnotation()
m.setAttr(self.annotation.shape + '.text', text, typ='string')
def createAnnotation(self):
if not self.isValid():
return
annotationShapes = self.refLocator.getChildren(allDescendants=True, typ='annotationShape')
for s in annotationShapes:
m.delete(getParent(s))
annotationSh = m.annotate(self.refLocator.transform, p=(0, -0.5, 0))
annotationTr = getParent(annotationSh)
annotationTr = m.parent(annotationTr, self.refLocator.transform, relative=True)[0]
lockTransformations(annotationTr)
self.annotation = TransformHandle(transform=getLongName(annotationTr))
m.setAttr(self.annotation.shape + '.displayArrow', False)
m.setAttr(self.annotation.transform + '.overrideEnabled', True)
m.setAttr(self.annotation.transform + '.overrideDisplayType', 2)
def activate(self):
if self.active:
return
if not self.refExists():
m.warning('{}: {}: Reference does not exists. Activation skipped.'.format(self.refLocator.shape, self.refFilename))
return
if not m.objExists(self.instanceSource):
self.createRefSource()
m.instance(
self.instanceSource,
name=REF_INST_NAME
)
inst = '|{}|{}'.format(INSTANCES_SOURCE_GROUP, REF_INST_NAME)
m.setAttr(inst + '.overrideEnabled', True)
m.setAttr(inst + '.overrideDisplayType', 2)
lockTransformations(inst, visibility=True)
parentAPI(inst, self.refLocator.transform, absolute=False)
m.connectAttr(self.refNode + '.message', self.refLocator.shape + '.refNodeMessage', force=True)
self.active = True
def createRefSource(self):
if m.objExists(self.refNode):
m.file(
referenceNode=self.refNode,
removeReference=True,
force=True
)
fileType = 'mayaAscii' if self.refFilename.endswith('.ma') else 'mayaBinary'
m.file(
self.refFilename,
reference=True,
typ=fileType,
referenceNode=self.refNode,
groupReference=True,
groupName=self.idString,
mergeNamespacesOnClash=False,
namespace=self.refShortName,
options='v=0;',
)
createInvisibleGroup(INSTANCES_SOURCE_GROUP)
m.parent('|' + self.idString, '|' + INSTANCES_SOURCE_GROUP)
def importRef(self):
if not self.refExists():
m.warning('{}: {}: Reference does not exists. Import skipped.'.format(self.refLocator.shape, self.refFilename))
return
if not m.objExists(self.importSource):
self.createRefImportSource()
importedRefGroup = '{}_{}'.format(REF_IMPORTED_GROUP, self.refShortName)
obj = m.duplicate(
self.importSource,
)
m.rename(obj[0], importedRefGroup)
impGroup = '|{}|{}'.format(IMPORT_SOURCE_GROUP, importedRefGroup)
m.addAttr(
impGroup,
dt='string',
shortName=ATTR_REF_SOURCE_PATH_NAMES[0],
longName=ATTR_REF_SOURCE_PATH_NAMES[1],
niceName=ATTR_REF_SOURCE_PATH_NAMES[2]
)
m.setAttr('{}.{}'.format(impGroup, ATTR_REF_SOURCE_PATH), self.refFilename, typ='string')
impGroup = parentAPI(impGroup, self.refLocator.transform, absolute=False)
refLocatorParents = self.refLocator.getParents()
refLocatorParent = refLocatorParents[0] if refLocatorParents else None
parentAPI(impGroup, refLocatorParent)
m.delete(self.refLocator.transform)
# m.parent(inst, self.refLocator.transform, relative=True)
def createRefImportSource(self):
fileType = 'mayaAscii' if self.refFilename.endswith('.ma') else 'mayaBinary'
m.file(
self.refFilename,
i=True,
typ=fileType,
groupReference=True,
groupName=self.idString,
mergeNamespacesOnClash=False,
namespace=self.refShortName,
options='v=0;',
)
createInvisibleGroup(IMPORT_SOURCE_GROUP)
m.parent('|' + self.idString, '|' + IMPORT_SOURCE_GROUP)
def deactivate(self):
refGeom = self.refLocator.getChildren()
for rg in refGeom:
if stripNamespaces(rg.split('|')[-1]).startswith(REF_INST_NAME):
m.delete(rg)
if self.getActiveStateFromMaya():
m.disconnectAttr(self.refNode + '.message', self.refLocator.shape + '.refNodeMessage')
self.active = False
def setRefFilename(self, refFilename):
m.setAttr('{}.{}'.format(self.refLocator.shape, ATTR_REF_FILENAME), refFilename, typ='string')
self.loadFromRefLocatorShape(self.refLocator.shape)
def getRefFilename(self):
return cleanupPath(self.refFilename)
def setRefLocator(self, transformHandle):
self.refLocator = transformHandle
def refExists(self):
return os.path.exists(os.path.expandvars(self.refFilename))
