191 lines
8.9 KiB
Python
191 lines
8.9 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.
|
|
##
|
|
##-*****************************************************************************
|
|
|
|
from maya import cmds as MayaCmds
|
|
import os
|
|
import unittest
|
|
import util
|
|
|
|
class staticPropTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
MayaCmds.file(new=True, force=True)
|
|
self.__files = []
|
|
|
|
def tearDown(self):
|
|
for f in self.__files :
|
|
os.remove(f)
|
|
|
|
def setProps(self, nodeName):
|
|
MayaCmds.select(nodeName)
|
|
MayaCmds.addAttr(longName='SPT_int8', defaultValue=8,
|
|
attributeType='byte', keyable=True)
|
|
MayaCmds.addAttr(longName='SPT_int16', defaultValue=16,
|
|
attributeType='short', keyable=True)
|
|
MayaCmds.addAttr(longName='SPT_int32', defaultValue=32,
|
|
attributeType='long', keyable=True)
|
|
MayaCmds.addAttr(longName='SPT_float', defaultValue=3.2654,
|
|
attributeType='float', keyable=True)
|
|
|
|
MayaCmds.addAttr(longName='SPT_double', defaultValue=0.15724757,
|
|
attributeType='double', keyable=True)
|
|
MayaCmds.addAttr(longName='SPT_double_AbcGeomScope', dataType="string")
|
|
MayaCmds.setAttr(nodeName+'.SPT_double_AbcGeomScope', "vtx",
|
|
type="string")
|
|
|
|
MayaCmds.addAttr(longName='SPT_string', dataType="string")
|
|
MayaCmds.setAttr(nodeName+'.SPT_string', "empty", type="string")
|
|
MayaCmds.addAttr(longName='SPT_string_AbcGeomScope', dataType="string")
|
|
MayaCmds.setAttr(nodeName+'.SPT_string_AbcGeomScope', "potato",
|
|
type="string")
|
|
|
|
MayaCmds.addAttr(longName='SPT_int32_array', dataType='Int32Array')
|
|
MayaCmds.setAttr(nodeName+'.SPT_int32_array', [6, 7, 8, 9, 10],
|
|
type='Int32Array')
|
|
|
|
MayaCmds.addAttr(longName='SPT_vector_array', dataType='vectorArray')
|
|
MayaCmds.setAttr(nodeName+'.SPT_vector_array', 3,
|
|
(1,1,1), (2,2,2), (3,3,3), type='vectorArray')
|
|
MayaCmds.addAttr(longName='SPT_vector_array_AbcType', dataType="string")
|
|
MayaCmds.setAttr(nodeName+'.SPT_vector_array_AbcType', "normal2",
|
|
type="string")
|
|
|
|
MayaCmds.addAttr(longName='SPT_point_array', dataType='pointArray')
|
|
MayaCmds.setAttr(nodeName+'.SPT_point_array', 2,
|
|
(2,4,6,8), (20,40,60,80), type='pointArray')
|
|
|
|
MayaCmds.addAttr(longName='SPT_double_array', dataType='doubleArray')
|
|
MayaCmds.addAttr(longName='SPT_double_array_AbcGeomScope',
|
|
dataType="string")
|
|
MayaCmds.setAttr(nodeName+'.SPT_double_array',
|
|
[1.1, 2.2, 3.3, 4.4, 5.5], type='doubleArray')
|
|
MayaCmds.setAttr(nodeName+'.SPT_double_array_AbcGeomScope', "vtx",
|
|
type="string")
|
|
|
|
MayaCmds.addAttr(longName='SPT_float_array', dataType='floatArray')
|
|
MayaCmds.addAttr(longName='SPT_float_array_AbcGeomScope',
|
|
dataType="string")
|
|
MayaCmds.setAttr(nodeName+'.SPT_float_array',
|
|
[5.0, 4.75, 3.5, 2.25, 1.0], type='floatArray')
|
|
MayaCmds.setAttr(nodeName+'.SPT_float_array_AbcGeomScope', "vtx",
|
|
type="string")
|
|
|
|
MayaCmds.addAttr(longName='SPT_string_array', dataType='stringArray')
|
|
MayaCmds.setAttr(nodeName+'.SPT_string_array', 3, "string1", "string2", "string3",
|
|
type='stringArray')
|
|
|
|
def verifyProps(self, nodeName, fileName):
|
|
MayaCmds.AbcImport(fileName, mode='open')
|
|
self.failUnlessEqual(8, MayaCmds.getAttr(nodeName+'.SPT_int8'))
|
|
self.failUnlessEqual(16, MayaCmds.getAttr(nodeName+'.SPT_int16'))
|
|
self.failUnlessEqual(32, MayaCmds.getAttr(nodeName+'.SPT_int32'))
|
|
self.failUnlessAlmostEqual(3.2654,
|
|
MayaCmds.getAttr(nodeName+'.SPT_float'), 4)
|
|
self.failUnlessAlmostEqual(0.15724757,
|
|
MayaCmds.getAttr(nodeName+'.SPT_double'), 7)
|
|
self.failUnlessEqual('vtx',
|
|
MayaCmds.getAttr(nodeName+'.SPT_double_AbcGeomScope'))
|
|
self.failUnlessEqual('empty', MayaCmds.getAttr(nodeName+'.SPT_string'))
|
|
self.failUnlessEqual(0, MayaCmds.attributeQuery(
|
|
'SPT_string_AbcGeomScope', node=nodeName, exists=True))
|
|
self.failUnlessEqual([6, 7, 8, 9, 10],
|
|
MayaCmds.getAttr(nodeName+'.SPT_int32_array'))
|
|
self.failUnlessEqual(["string1", "string2", "string3"],
|
|
MayaCmds.getAttr(nodeName+'.SPT_string_array'))
|
|
self.failUnlessEqual([1.1, 2.2, 3.3, 4.4, 5.5],
|
|
MayaCmds.getAttr(nodeName+'.SPT_double_array'))
|
|
self.failUnlessEqual([5.0, 4.75, 3.5, 2.25, 1.0],
|
|
MayaCmds.getAttr(nodeName+'.SPT_float_array'))
|
|
self.failUnlessEqual([(1.0, 1.0, 0.0), (2.0, 2.0, 0.0), (3.0, 3.0, 0.0)],
|
|
MayaCmds.getAttr(nodeName+'.SPT_vector_array'))
|
|
self.failUnlessEqual('normal2',
|
|
MayaCmds.getAttr(nodeName+'.SPT_vector_array_AbcType'))
|
|
self.failUnlessEqual([(2.0, 4.0, 6.0, 1.0), (20.0, 40.0, 60.0, 1.0)],
|
|
MayaCmds.getAttr(nodeName+'.SPT_point_array'))
|
|
|
|
def testStaticTransformPropReadWrite(self):
|
|
nodeName = MayaCmds.createNode('transform')
|
|
self.setProps(nodeName)
|
|
self.__files.append(util.expandFileName('staticPropTransform.abc'))
|
|
MayaCmds.AbcExport(j='-atp SPT_ -root %s -file %s' % (nodeName, self.__files[-1]))
|
|
self.verifyProps(nodeName, self.__files[-1])
|
|
|
|
def testStaticCameraPropReadWrite(self):
|
|
root = MayaCmds.camera()
|
|
nodeName = root[0]
|
|
shapeName = root[1]
|
|
self.setProps(shapeName)
|
|
self.__files.append(util.expandFileName('staticPropCamera.abc'))
|
|
MayaCmds.AbcExport(j='-atp SPT_ -root %s -f %s' % (nodeName, self.__files[-1]))
|
|
self.verifyProps(shapeName, self.__files[-1])
|
|
|
|
def testStaticParticlePropReadWrite(self):
|
|
root = MayaCmds.particle(p=[(0, 0, 0), (1, 1, 1)])
|
|
nodeName = root[0]
|
|
shapeName = root[1]
|
|
self.setProps(shapeName)
|
|
self.__files.append(util.expandFileName('staticPropParticles.abc'))
|
|
MayaCmds.AbcExport(j='-atp SPT_ -root %s -f %s' % (nodeName, self.__files[-1]))
|
|
self.verifyProps(shapeName, self.__files[-1])
|
|
|
|
def testStaticMeshPropReadWrite(self):
|
|
nodeName = 'polyCube'
|
|
shapeName = 'polyCubeShape'
|
|
MayaCmds.polyCube(name=nodeName)
|
|
self.setProps(shapeName)
|
|
self.__files.append(util.expandFileName('staticPropMesh.abc'))
|
|
MayaCmds.AbcExport(j='-atp SPT_ -root %s -f %s' % (nodeName, self.__files[-1]))
|
|
self.verifyProps(shapeName, self.__files[-1])
|
|
|
|
def testStaticNurbsCurvePropReadWrite(self):
|
|
nodeName = 'nCurve'
|
|
shapeName = 'curveShape1'
|
|
MayaCmds.curve(p=[(0, 0, 0), (3, 5, 6), (5, 6, 7), (9, 9, 9)],
|
|
name=nodeName)
|
|
self.setProps(shapeName)
|
|
self.__files.append(util.expandFileName('staticPropCurve.abc'))
|
|
MayaCmds.AbcExport(j='-atp SPT_ -root %s -f %s' % (nodeName, self.__files[-1]))
|
|
self.verifyProps(shapeName, self.__files[-1])
|
|
|
|
def testStaticNurbsSurfacePropReadWrite(self):
|
|
nodeName = 'nSphere'
|
|
shapeName = 'nSphereShape'
|
|
MayaCmds.sphere(name=nodeName)
|
|
self.setProps(shapeName)
|
|
self.__files.append(util.expandFileName('staticPropNurbs.abc'))
|
|
MayaCmds.AbcExport(j='-atp SPT_ -root %s -file %s' % (nodeName, self.__files[-1]))
|
|
self.verifyProps(shapeName, self.__files[-1])
|