def testSamples( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/animatedCube.abc" ) self.assertEqual( a.numSamples(), 10 ) for i in range( 0, a.numSamples() ) : self.assertAlmostEqual( a.timeAtSample( i ), (i + 1) / 24.0 ) p = a.child( "persp" ) self.assertEqual( p.numSamples(), 1 ) self.assertEqual( p.timeAtSample( 0 ), 1 / 24.0 ) t = a.child( "pCube1" ) self.assertEqual( t.numSamples(), 10 ) for i in range( 0, t.numSamples() ) : self.assertAlmostEqual( t.timeAtSample( i ), (i + 1) / 24.0 ) m = t.child( "pCubeShape1" ) self.assertEqual( m.numSamples(), 10 ) for i in range( 0, m.numSamples() ) : self.assertAlmostEqual( m.timeAtSample( i ), (i + 1) / 24.0 ) a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/noTopLevelStoredBounds.abc" ) self.assertEqual( a.numSamples(), 0 ) # no time samples at the top level, so this should throw an exception: self.assertRaises( Exception, a.timeAtSample, 0 ) # should throw the RIGHT exceptions: try: a.timeAtSample(0) except Exception, e: self.assertEqual( str(e), "Invalid Argument : Sample index out of range" )
def testHasStoredBound( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/animatedCube.abc" ) self.assertEqual( a.hasStoredBound(), True ) self.assertEqual( a.child( "persp" ).hasStoredBound(), False ) self.assertEqual( a.child( "persp" ).child( "perspShape" ).hasStoredBound(), False ) self.assertEqual( a.child( "pCube1" ).hasStoredBound(), False ) self.assertEqual( a.child( "pCube1" ).child( "pCubeShape1" ).hasStoredBound(), True ) self.assertEqual( a.child( "front" ).hasStoredBound(), False ) self.assertEqual( a.child( "front" ).child( "frontShape" ).hasStoredBound(), False ) a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/noTopLevelStoredBounds.abc" ) self.assertEqual( a.hasStoredBound(), False )
def testHierarchy( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/cube.abc" ) self.assertEqual( a.numChildren(), 1 ) self.assertEqual( a.childNames(), IECore.StringVectorData( [ "group1" ] ) ) g1 = a.child( 0 ) g2 = a.child( "group1" ) self.assertEqual( g1.name(), g2.name() ) self.assertEqual( g1.name(), "group1" ) self.assertEqual( g1.fullName(), g2.fullName() ) self.assertEqual( g1.fullName(), "/group1" ) self.assertEqual( g1.numChildren(), 1 ) self.assertEqual( g1.childNames(), IECore.StringVectorData( [ "pCube1" ] ) ) c = g1.child( 0 ) self.assertEqual( c.name(), "pCube1" ) self.assertEqual( c.fullName(), "/group1/pCube1" ) self.assertEqual( c.numChildren(), 1 ) self.assertEqual( c.childNames(), IECore.StringVectorData( [ "pCubeShape1" ] ) ) cs = c.child( 0 ) self.assertEqual( cs.numChildren(), 0 ) self.assertEqual( cs.childNames(), IECore.StringVectorData() ) self.assertRaises( Exception, cs.child, 0 ) self.assertRaises( Exception, cs.child, "iDontExist" )
def testAnimation(self):
a = GafferScene.AlembicSource()
a["fileName"].setValue(
os.path.dirname(__file__) + "/alembicFiles/animatedCube.abc")
# We have to skip the test of built in sets, because our alembic file contains cameras
# and alembic doesn't provide a means of flagging them upfront.
self.assertSceneValid(a["out"], assertBuiltInSetsComplete=False)
b = IECoreAlembic.AlembicInput(
os.path.dirname(__file__) + "/alembicFiles/animatedCube.abc")
c = b.child("pCube1").child("pCubeShape1")
numSamples = b.numSamples()
startTime = b.timeAtSample(0)
endTime = b.timeAtSample(numSamples - 1)
for i in range(0, numSamples * 2):
time = startTime + (endTime -
startTime) * float(i) / (numSamples * 2 - 1)
c = Gaffer.Context()
c.setTime(time)
with c:
self.assertBoxesAlmostEqual(a["out"].bound("/"),
b.boundAtTime(time), 6)
self.assertBoxesAlmostEqual(
a["out"].bound("/pCube1/pCubeShape1"), b.boundAtTime(time),
6)
def testSampleInterval( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/animatedCube.abc" ) # persp has only one sample, so should always be reading from that regardless the time p = a.child( "persp" ) t = -1000 while t < 1000 : t += .01 self.assertEqual( p.sampleIntervalAtTime( t ), ( 0, 0, 0 ) ) # pCube1 has a sample per frame t = a.child( "pCube1" ) for i in range( 0, t.numSamples() ) : # reads on the frame should not need # interpolation. v = t.sampleIntervalAtTime( t.timeAtSample( i ) ) self.assertEqual( v[0], 0 ) self.assertEqual( v[1], i ) self.assertEqual( v[1], i ) # reads in between frames should need # interpolation if i < t.numSamples() -1 : v = t.sampleIntervalAtTime( t.timeAtSample( i ) + 1 / 48.0 ) self.assertAlmostEqual( v[0], 0.5 ) self.assertEqual( v[1], i ) self.assertEqual( v[2], i + 1 )
def testConvertArbGeomParams( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/coloredMesh.abc" ) m = a.child( "pPlane1" ).child( "pPlaneShape1" ).objectAtSample( 0, IECore.MeshPrimitive.staticTypeId() ) self.failUnless( m.arePrimitiveVariablesValid() ) self.failUnless( "colorSet1" in m ) self.assertEqual( m["colorSet1"].interpolation, IECore.PrimitiveVariable.Interpolation.FaceVarying ) self.failUnless( isinstance( m["colorSet1"].data, IECore.Color4fVectorData ) ) self.assertEqual( len( m["colorSet1"].data ), 4 ) self.assertEqual( m["colorSet1"].data, IECore.Color4fVectorData( [ IECore.Color4f( 0, 1, 0, 1 ), IECore.Color4f( 0, 0, 1, 1 ), IECore.Color4f( 0, 0, 0, 1 ), IECore.Color4f( 1, 0, 0, 1 ), ] ) ) self.failUnless( "ABC_int" in m ) self.assertEqual( m["ABC_int"].interpolation, IECore.PrimitiveVariable.Interpolation.Constant ) self.assertEqual( m["ABC_int"].data, IECore.IntVectorData( [ 10 ] ) )
def doOperation(self, args):
inFile = IECoreAlembic.AlembicInput(args["inputFile"].value)
outFile = IECore.ModelCache(args["outputFile"].value,
IECore.IndexedIO.OpenMode.Write)
time = inFile.timeAtSample(0)
def walk(alembicInput, modelCache):
o = alembicInput.objectAtTime(time,
IECore.Primitive.staticTypeId())
if o is not None:
modelCache.writeObject(o)
t = alembicInput.transformAtTime(time)
modelCache.writeTransform(t)
numChildren = alembicInput.numChildren()
for i in range(0, numChildren):
alembicChild = alembicInput.child(i)
modelCacheChild = modelCache.writableChild(alembicChild.name())
walk(alembicChild, modelCacheChild)
walk(inFile, outFile)
return args["outputFile"].value
def testAnimation(self):
a = GafferScene.AlembicSource()
a["fileName"].setValue(
os.path.dirname(__file__) + "/alembicFiles/animatedCube.abc")
self.assertSceneValid(a["out"])
b = IECoreAlembic.AlembicInput(
os.path.dirname(__file__) + "/alembicFiles/animatedCube.abc")
c = b.child("pCube1").child("pCubeShape1")
numSamples = b.numSamples()
startTime = b.timeAtSample(0)
endTime = b.timeAtSample(numSamples - 1)
for i in range(0, numSamples * 2):
time = startTime + (endTime -
startTime) * float(i) / (numSamples * 2 - 1)
c = Gaffer.Context()
c.setFrame(time * 24)
with c:
self.assertBoxesAlmostEqual(a["out"].bound("/"),
b.boundAtTime(time), 6)
self.assertBoxesAlmostEqual(
a["out"].bound("/pCube1/pCubeShape1"), b.boundAtTime(time),
6)
def testBound( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/cube.abc" ) self.assertEqual( a.boundAtSample(), IECore.Box3d( IECore.V3d( -2 ), IECore.V3d( 2 ) ) ) cs = a.child( "group1" ).child( "pCube1" ).child( "pCubeShape1" ) self.assertEqual( cs.boundAtSample(), IECore.Box3d( IECore.V3d( -1 ), IECore.V3d( 1 ) ) )
def testMetaData(self):
a = IECoreAlembic.AlembicInput(
os.path.dirname(__file__) + "/data/cube.abc")
m = a.metaData()
self.failUnless(isinstance(m, IECore.CompoundData))
def testPoints(self):
a = IECoreAlembic.AlembicInput(
os.path.dirname(__file__) + "/data/points.abc")
p = a.child("particle1").child("particleShape1")
points = p.objectAtSample(9)
self.assertTrue(isinstance(points, IECore.PointsPrimitive))
self.assertEqual(points.numPoints, 9)
self.assertTrue(points.arePrimitiveVariablesValid())
self.assertTrue(isinstance(points["P"].data, IECore.V3fVectorData))
self.assertEqual(points["P"].data.getInterpretation(),
IECore.GeometricData.Interpretation.Point)
self.assertEqual(points["velocity"].data.getInterpretation(),
IECore.GeometricData.Interpretation.Vector)
self.assertTrue(isinstance(points["id"].data, IECore.UInt64VectorData))
for i in range(0, 9):
self.assertTrue(
points["P"].data[i].normalized().equalWithAbsError(
points["velocity"].data[i].normalized(), 0.00001), )
self.assertEqual(points["id"].data[i], i)
def testConvertUVs(self):
a = IECoreAlembic.AlembicInput(
os.path.dirname(__file__) + "/data/coloredMesh.abc")
m = a.child("pPlane1").child("pPlaneShape1").objectAtSample(
0, IECore.MeshPrimitive.staticTypeId())
self.failUnless("uv" in m)
self.assertEqual(m["uv"].interpolation,
IECore.PrimitiveVariable.Interpolation.FaceVarying)
self.failUnless(isinstance(m["uv"].data, IECore.V2fVectorData))
self.assertEqual(m["uv"].data.getInterpretation(),
IECore.GeometricData.Interpretation.UV)
self.failUnless(isinstance(m["uv"].indices, IECore.IntVectorData))
self.assertEqual(
len(m["uv"].data),
m.variableSize(IECore.PrimitiveVariable.Interpolation.Vertex))
self.assertEqual(
len(m["uv"].indices),
m.variableSize(IECore.PrimitiveVariable.Interpolation.FaceVarying))
self.failUnless(m.isPrimitiveVariableValid(m["uv"]))
def testConverterAccess( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/animatedCube.abc" ) m = a.child( "pCube1" ).child( "pCubeShape1" ) c = m.converter() mesh = c.convert() self.failUnless( isinstance( mesh, IECore.MeshPrimitive ) )
def testRotatingTransformAtSample( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/rotatingCube.abc" ) t = a.child( "pCube1" ) for i in range( 0, 24 ) : ti = t.transformAtSample( i ) mi = IECore.M44d.createRotated( IECore.V3d( IECore.degreesToRadians( 90 * i ), 0, 0 ) ) self.failUnless( ti.equalWithAbsError( mi, 0.0000000000001 ) )
def testCamera( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/animatedCube.abc" ) c = a.child( "persp" ).child( "perspShape" ).objectAtSample( 0 ) self.failUnless( isinstance( c, IECore.Camera ) ) c = a.child( "persp" ).child( "perspShape" ).objectAtTime( 0 ) self.failUnless( isinstance( c, IECore.Camera ) )
def testConvertNormals( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/animatedCube.abc" ) m = a.child( "pCube1" ).child( "pCubeShape1" ) mesh = m.objectAtSample( 0 ) self.failUnless( "N" in mesh ) self.failUnless( isinstance( mesh["N"].data, IECore.V3fVectorData ) ) self.assertEqual( mesh["N"].interpolation, IECore.PrimitiveVariable.Interpolation.FaceVarying ) self.assertEqual( mesh["N"].data.getInterpretation(), IECore.GeometricData.Interpretation.Normal )
def testHierarchyIgnoresShadingGroups( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/sphereWithShadingGroups.abc" ) self.assertEqual( a.numChildren(), 1 ) self.assertEqual( a.childNames(), IECore.StringVectorData( [ "pSphere1" ] ) ) g = a.child( "pSphere1" ) m = g.child( "pSphereShape1" ) self.assertEqual( m.numChildren(), 0 ) self.assertEqual( m.childNames(), IECore.StringVectorData() )
def testNURBSCircle( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/curves.abc" ) c = a.child( "nurbsCircle" ).child( "nurbsCircleShape" ) curves = c.objectAtSample( 0 ) self.assertTrue( isinstance( curves, IECore.CurvesPrimitive ) ) self.assertEqual( curves.basis(), IECore.CubicBasisf.bSpline() ) self.assertEqual( curves.verticesPerCurve(), IECore.IntVectorData( [ 11 ] ) ) self.assertEqual( curves.periodic(), True ) self.assertTrue( curves.arePrimitiveVariablesValid() )
def testConvertMesh( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/cube.abc" ) c = a.child( "group1" ).child( "pCube1" ) self.assertEqual( c.objectAtSample( 0, IECore.MeshPrimitive.staticTypeId() ), None ) cs = c.child( "pCubeShape1" ) m = cs.objectAtSample( 0, IECore.MeshPrimitive.staticTypeId() ) self.failUnless( isinstance( m, IECore.MeshPrimitive ) )
def testInterpolatedRotate( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/rotatingCube.abc" ) t = a.child( "pCube1" ) for i in range( 0, t.numSamples() * 2 - 1 ) : frame = i / 2.0 + 1 time = frame / 24.0 matrix = t.transformAtTime( time ) expectedMatrix = IECore.M44d.createRotated( IECore.V3d( IECore.degreesToRadians( 90 * i * 0.5 ), 0, 0 ) ) self.failUnless( matrix.equalWithAbsError( expectedMatrix, 0.0000001 ) )
def testConvertSubD( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/subdPlane.abc" ) c = a.child( "pPlane1" ) self.assertEqual( c.objectAtSample( 0, IECore.MeshPrimitive.staticTypeId() ), None ) cs = c.child( "pPlaneShape1" ) m = cs.objectAtSample( 0, IECore.MeshPrimitive.staticTypeId() ) self.failUnless( isinstance( m, IECore.MeshPrimitive ) ) self.assertEqual( m.interpolation, "catmullClark" )
def testBoolGeomParam( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/bool.abc" ) p = a.child( "pPlane1" ).child( "pPlaneShape1" ) mesh = p.objectAtSample( 0 ) for n in ( "abc_testBoolTrue", "abc_testBoolFalse" ) : self.assertIn( n, mesh ) self.assertEqual( mesh[n].interpolation, IECore.PrimitiveVariable.Interpolation.Constant ) self.assertEqual( mesh["abc_testBoolTrue"].data, IECore.BoolVectorData( [ True ] ) ) self.assertEqual( mesh["abc_testBoolFalse"].data, IECore.BoolVectorData( [ False ] ) )
def testConvertInterpolated( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/animatedCube.abc" ) m = a.child( "pCube1" ).child( "pCubeShape1" ) mesh0 = m.objectAtSample( 0 ) mesh1 = m.objectAtSample( 1 ) mesh = m.objectAtTime( 1.5 / 24.0 ) self.failUnless( isinstance( mesh, IECore.MeshPrimitive ) ) self.assertEqual( mesh, IECore.linearObjectInterpolation( mesh0, mesh1, 0.5 ) )
def testMissingArchiveBounds( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/noTopLevelStoredBounds.abc" ) self.assertEqual( a.numSamples(), 0 ) # no time samples at the top level, so this should throw an exception self.assertRaisesRegexp( Exception, "Invalid Argument : Sample index out of range", a.timeAtSample, 0 ) # no stored bound at the top level self.assertFalse( a.hasStoredBound() ) self.assertRaises( Exception, a.boundAtSample, 0 )
def testBoundAtTime(self):
a = IECoreAlembic.AlembicInput(
os.path.dirname(__file__) + "/data/animatedCube.abc")
t = a.child("pCube1")
m = t.child("pCubeShape1")
startTime = a.timeAtSample(0)
endTime = a.timeAtSample(a.numSamples() - 1)
aStartBound = a.boundAtSample(0)
aEndBound = a.boundAtSample(a.numSamples() - 1)
mStartBound = m.boundAtSample(0)
mEndBound = m.boundAtSample(m.numSamples() - 1)
def lerp(a, b, x):
return a + (b - a) * x
def lerpBox(a, b, x):
r = a.__class__()
r.min = lerp(a.min, b.min, x)
r.max = lerp(a.max, b.max, x)
return r
numSteps = 100
for i in range(0, numSteps):
lerpFactor = (float(i) / (numSteps - 1))
time = lerp(startTime, endTime, lerpFactor)
aBound = a.boundAtTime(time)
expectedABound = lerpBox(aStartBound, aEndBound, lerpFactor)
self.failUnless(
aBound.min.equalWithAbsError(expectedABound.min, 0.000001))
self.failUnless(
aBound.max.equalWithAbsError(expectedABound.max, 0.000001))
mBound = m.boundAtTime(time)
expectedMBound = lerpBox(mStartBound, mEndBound, lerpFactor)
self.failUnless(
mBound.min.equalWithAbsError(expectedMBound.min, 0.000001))
self.failUnless(
mBound.max.equalWithAbsError(expectedMBound.max, 0.000001))
tBound = t.boundAtTime(time)
self.failUnless(
tBound.min.equalWithAbsError(expectedMBound.min, 0.000001))
self.failUnless(
tBound.max.equalWithAbsError(expectedMBound.max, 0.000001))
def testObjectAtSample( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/animatedCube.abc" ) m = a.child( "pCube1" ).child( "pCubeShape1" ) mesh = m.objectAtSample( 0 ) self.failUnless( isinstance( mesh, IECore.MeshPrimitive ) ) for i in range( 1, m.numSamples() ) : mesh2 = m.objectAtSample( i ) self.failUnless( isinstance( mesh2, IECore.MeshPrimitive ) ) self.assertEqual( mesh.verticesPerFace, mesh2.verticesPerFace ) self.assertNotEqual( mesh["P"], mesh2["P"] )
def testBoundAtSample( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/animatedCube.abc" ) self.assertEqual( a.boundAtSample( 0 ), IECore.Box3d( IECore.V3d( -0.5 ), IECore.V3d( 0.5 ) ) ) self.assertEqual( a.boundAtSample( a.numSamples()-1 ), IECore.Box3d( IECore.V3d( 0.5, -0.5, -0.5 ), IECore.V3d( 1.5, 2, 0.5 ) ) ) t = a.child( "pCube1" ) self.assertRaises( Exception, t.boundAtSample, 0 ) self.assertRaises( Exception, t.boundAtSample, t.numSamples() - 1 ) m = t.child( "pCubeShape1" ) self.assertEqual( m.boundAtSample( 0 ), IECore.Box3d( IECore.V3d( -0.5 ), IECore.V3d( 0.5 ) ) ) self.assertEqual( m.boundAtSample( m.numSamples()-1 ), IECore.Box3d( IECore.V3d( -0.5, -0.5, -0.5 ), IECore.V3d( 0.5, 2, 0.5 ) ) )
def testTransform( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/cube.abc" ) self.assertEqual( a.transformAtSample(), IECore.M44d() ) g = a.child( "group1" ) self.assertEqual( g.transformAtSample(), IECore.M44d.createScaled( IECore.V3d( 2 ) ) * IECore.M44d.createTranslated( IECore.V3d( 2, 0, 0 ) ) ) c = g.child( "pCube1" ) self.assertEqual( c.transformAtSample(), IECore.M44d.createTranslated( IECore.V3d( -1, 0, 0 ) ) ) cs = c.child( "pCubeShape1" ) self.assertEqual( cs.transformAtSample(), IECore.M44d() )
def testConvertUVs( self ) : a = IECoreAlembic.AlembicInput( os.path.dirname( __file__ ) + "/data/coloredMesh.abc" ) m = a.child( "pPlane1" ).child( "pPlaneShape1" ).objectAtSample( 0, IECore.MeshPrimitive.staticTypeId() ) self.failUnless( "s" in m ) self.failUnless( "t" in m ) self.assertEqual( m["s"].interpolation, IECore.PrimitiveVariable.Interpolation.FaceVarying ) self.assertEqual( m["t"].interpolation, IECore.PrimitiveVariable.Interpolation.FaceVarying ) self.failUnless( isinstance( m["s"].data, IECore.FloatVectorData ) ) self.failUnless( isinstance( m["t"].data, IECore.FloatVectorData ) )
def __alembicInput( self, args ) : if self.__input is not None and self.__inputFileName == args["fileName"].value : return self.__input self.__input = None self.__inputFileName = args["fileName"].value if args["fileName"].value : try : self.__input = IECoreAlembic.AlembicInput( args["fileName"].value ) except : IECore.msg( IECore.Msg.Level.Error, "AlembicProcedural", "Unable to open file \"%s\"" % args["fileName"].value ) return self.__input