216 lines
9.6 KiB
Python
216 lines
9.6 KiB
Python
##-*****************************************************************************
|
|
##
|
|
## Copyright (c) 2009-2011,
|
|
## Sony Pictures Imageworks, Inc. and
|
|
## Industrial Light & Magic, a division of Lucasfilm Entertainment Company Ltd.
|
|
##
|
|
## All rights reserved.
|
|
##
|
|
## Redistribution and use in source and binary forms, with or without
|
|
## modification, are permitted provided that the following conditions are
|
|
## met:
|
|
## * Redistributions of source code must retain the above copyright
|
|
## notice, this list of conditions and the following disclaimer.
|
|
## * Redistributions in binary form must reproduce the above
|
|
## copyright notice, this list of conditions and the following disclaimer
|
|
## in the documentation and/or other materials provided with the
|
|
## distribution.
|
|
## * Neither the name of Sony Pictures Imageworks, nor
|
|
## Industrial Light & Magic nor the names of their contributors may be used
|
|
## to endorse or promote products derived from this software without specific
|
|
## prior written permission.
|
|
##
|
|
## THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
## "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
## LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
## A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
## OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
## SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
## LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
## DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
## THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
## (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
## OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
##
|
|
##-*****************************************************************************
|
|
|
|
import maya.cmds as MayaCmds
|
|
import os
|
|
import subprocess
|
|
import unittest
|
|
import util
|
|
|
|
class animPropTest(unittest.TestCase):
|
|
|
|
#-------------------------------------------------------------------------
|
|
def setUp(self):
|
|
MayaCmds.file(new=True, force=True)
|
|
self.__abcStitcher = [os.environ['AbcStitcher']]
|
|
self.__files = [util.expandFileName('animProp.abc'),
|
|
util.expandFileName('animProp01_14.abc'),
|
|
util.expandFileName('animProp15_24.abc')]
|
|
|
|
#-------------------------------------------------------------------------
|
|
def tearDown(self):
|
|
for f in self.__files :
|
|
os.remove(f)
|
|
|
|
#-------------------------------------------------------------------------
|
|
def setAndKeyProps( self, nodeName ):
|
|
|
|
MayaCmds.select(nodeName)
|
|
MayaCmds.addAttr(longName='SPT_int8', defaultValue=0,
|
|
attributeType='byte', keyable=True)
|
|
MayaCmds.addAttr(longName='SPT_int16', defaultValue=0,
|
|
attributeType='short', keyable=True)
|
|
MayaCmds.addAttr(longName='SPT_int32', defaultValue=0,
|
|
attributeType='long', keyable=True)
|
|
MayaCmds.addAttr(longName='SPT_float', defaultValue=0,
|
|
attributeType='float', keyable=True)
|
|
MayaCmds.addAttr(longName='SPT_double', defaultValue=0,
|
|
attributeType='double', keyable=True)
|
|
MayaCmds.setKeyframe(nodeName, value=0, attribute='SPT_int8',
|
|
t=[1, 24])
|
|
MayaCmds.setKeyframe(nodeName, value=100, attribute='SPT_int16',
|
|
t=[1, 24])
|
|
MayaCmds.setKeyframe(nodeName, value=1000, attribute='SPT_int32',
|
|
t=[1, 24])
|
|
MayaCmds.setKeyframe(nodeName, value=0.57777777, attribute='SPT_float',
|
|
t=[1, 24])
|
|
MayaCmds.setKeyframe(nodeName, value=5.045643545457,
|
|
attribute='SPT_double', t=[1, 24])
|
|
MayaCmds.setKeyframe(nodeName, value=8, attribute='SPT_int8', t=12)
|
|
MayaCmds.setKeyframe(nodeName, value=16, attribute='SPT_int16', t=12)
|
|
MayaCmds.setKeyframe(nodeName, value=32, attribute='SPT_int32', t=12)
|
|
MayaCmds.setKeyframe(nodeName, value=3.141592654,
|
|
attribute='SPT_float', t=12)
|
|
MayaCmds.setKeyframe(nodeName, value=3.141592654,
|
|
attribute='SPT_double', t=12)
|
|
|
|
def verifyProps( self, root, nodeName, wfgAndMoblur ):
|
|
|
|
# write to files
|
|
if wfgAndMoblur:
|
|
# reset __files so we wont try to tear down files that dont exist
|
|
self.__files = [self.__files[0]]
|
|
MayaCmds.AbcExport(j='-atp SPT_ -fr 1 24 -wfg -frs -0.25 -frs 0.0 -frs 0.25 -root %s -file %s' % (root, self.__files[0]))
|
|
else:
|
|
MayaCmds.AbcExport(j='-atp SPT_ -fr 1 14 -root %s -file %s' % (root, self.__files[1]))
|
|
MayaCmds.AbcExport(j='-atp SPT_ -fr 15 24 -root %s -file %s' % (root, self.__files[2]))
|
|
|
|
subprocess.call(self.__abcStitcher + self.__files)
|
|
|
|
# read file and verify data
|
|
MayaCmds.AbcImport(self.__files[0], mode='open')
|
|
|
|
t = 1 # frame 1
|
|
MayaCmds.currentTime(t, update=True)
|
|
self.failUnlessEqual(0, MayaCmds.getAttr(nodeName+'.SPT_int8'),
|
|
'%s.SPT_int8 != 0 at frame %d' %( nodeName, t))
|
|
self.failUnlessEqual(100, MayaCmds.getAttr(nodeName+'.SPT_int16'),
|
|
'%s.SPT_int16 != 100 at frame %d' %( nodeName, t))
|
|
self.failUnlessEqual(1000, MayaCmds.getAttr(nodeName+'.SPT_int32'),
|
|
'%s.SPT_int32 != 1000 at frame %d' %( nodeName, t))
|
|
self.failUnlessAlmostEqual(0.57777777,
|
|
MayaCmds.getAttr(nodeName+'.SPT_float'), 4,
|
|
'%s.SPT_float != 0.57777777 at frame %d' %( nodeName, t))
|
|
self.failUnlessAlmostEqual(5.045643545457,
|
|
MayaCmds.getAttr(nodeName+'.SPT_double'), 7,
|
|
'%s.SPT_double != 5.045643545457 at frame %d' %( nodeName, t))
|
|
|
|
t = 12 # frame 12
|
|
MayaCmds.currentTime(t, update=True)
|
|
self.failUnlessEqual(8, MayaCmds.getAttr(nodeName+'.SPT_int8'),
|
|
'%s.SPT_int8 != 8 at frame %d' %( nodeName, t))
|
|
self.failUnlessEqual(16, MayaCmds.getAttr(nodeName+'.SPT_int16'),
|
|
'%s.SPT_int16 != 16 at frame %d' %( nodeName, t))
|
|
self.failUnlessEqual(32, MayaCmds.getAttr(nodeName+'.SPT_int32'),
|
|
'%s.SPT_int32 != 32 at frame %d' %( nodeName, t))
|
|
self.failUnlessAlmostEqual(3.141592654,
|
|
MayaCmds.getAttr(nodeName+'.SPT_float'), 4,
|
|
'%s.SPT_float != 3.141592654 at frame %d' %( nodeName, t))
|
|
self.failUnlessAlmostEqual(3.1415926547,
|
|
MayaCmds.getAttr(nodeName+'.SPT_double'), 7,
|
|
'%s.SPT_double != 3.141592654 at frame %d' %( nodeName, t))
|
|
|
|
t = 24 # frame 24
|
|
MayaCmds.currentTime(t, update=True)
|
|
self.failUnlessEqual(0,MayaCmds.getAttr(nodeName+'.SPT_int8'),
|
|
'%s.SPT_int8 != 0 at frame %d' % (nodeName, t))
|
|
self.failUnlessEqual(100, MayaCmds.getAttr(nodeName+'.SPT_int16'),
|
|
'%s.SPT_int16 != 100 at frame %d' % (nodeName, t))
|
|
self.failUnlessEqual(1000, MayaCmds.getAttr(nodeName+'.SPT_int32'),
|
|
'%s.SPT_int32 != 1000 at frame %d' % (nodeName, t))
|
|
self.failUnlessAlmostEqual(0.57777777,
|
|
MayaCmds.getAttr(nodeName+'.SPT_float'), 4,
|
|
'%s.SPT_float != 0.57777777 at frame %d' % (nodeName, t))
|
|
self.failUnlessAlmostEqual(5.045643545457,
|
|
MayaCmds.getAttr(nodeName+'.SPT_double'), 7,
|
|
'%s.SPT_double != 5.045643545457 at frame %d' % (nodeName, t))
|
|
|
|
def testAnimTransformProp(self):
|
|
nodeName = MayaCmds.createNode('transform')
|
|
self.setAndKeyProps(nodeName)
|
|
self.verifyProps(nodeName, nodeName, False)
|
|
|
|
def testAnimCameraProp(self):
|
|
root = MayaCmds.camera()
|
|
nodeName = root[0]
|
|
shapeName = root[1]
|
|
self.setAndKeyProps(shapeName)
|
|
self.verifyProps(nodeName, shapeName, False)
|
|
|
|
def testAnimMeshProp(self):
|
|
nodeName = 'polyCube'
|
|
shapeName = 'polyCubeShape'
|
|
MayaCmds.polyCube(name=nodeName, ch=False)
|
|
self.setAndKeyProps(shapeName)
|
|
self.verifyProps(nodeName, shapeName, False)
|
|
|
|
def testAnimNurbsCurvePropReadWrite(self):
|
|
nodeName = 'nCurve'
|
|
shapeName = 'curveShape1'
|
|
MayaCmds.curve(p=[(0, 0, 0), (3, 5, 6), (5, 6, 7), (9, 9, 9)], name=nodeName)
|
|
self.setAndKeyProps(shapeName)
|
|
self.verifyProps(nodeName, shapeName, False)
|
|
|
|
def testAnimNurbsSurfaceProp(self):
|
|
nodeName = MayaCmds.sphere(ch=False)[0]
|
|
nodeNameList = MayaCmds.pickWalk(d='down')
|
|
shapeName = nodeNameList[0]
|
|
self.setAndKeyProps(shapeName)
|
|
self.verifyProps(nodeName, shapeName, False)
|
|
|
|
def testWFGAnimTransformProp(self):
|
|
nodeName = MayaCmds.createNode('transform')
|
|
self.setAndKeyProps(nodeName)
|
|
self.verifyProps(nodeName, nodeName, True)
|
|
|
|
def testWFGAnimCameraProp(self):
|
|
root = MayaCmds.camera()
|
|
nodeName = root[0]
|
|
shapeName = root[1]
|
|
self.setAndKeyProps(shapeName)
|
|
self.verifyProps(nodeName, shapeName, True)
|
|
|
|
def testWFGAnimMeshProp(self):
|
|
nodeName = 'polyCube'
|
|
shapeName = 'polyCubeShape'
|
|
MayaCmds.polyCube(name=nodeName, ch=False)
|
|
self.setAndKeyProps(shapeName)
|
|
self.verifyProps(nodeName, shapeName, True)
|
|
|
|
def testWFGAnimNurbsCurvePropReadWrite(self):
|
|
nodeName = 'nCurve'
|
|
shapeName = 'curveShape1'
|
|
MayaCmds.curve(p=[(0, 0, 0), (3, 5, 6), (5, 6, 7), (9, 9, 9)], name=nodeName)
|
|
self.setAndKeyProps(shapeName)
|
|
self.verifyProps(nodeName, shapeName, True)
|
|
|
|
def testWFGAnimNurbsSurfaceProp(self):
|
|
nodeName = MayaCmds.sphere(ch=False)[0]
|
|
nodeNameList = MayaCmds.pickWalk(d='down')
|
|
shapeName = nodeNameList[0]
|
|
self.setAndKeyProps(shapeName)
|
|
self.verifyProps(nodeName, shapeName, True)
|