def testMotion( self ) : p1 = IECoreScene.PointsPrimitive( IECore.V3fVectorData( [ imath.V3f( 10 ) ] * 10 ) ) p1["width"] = IECoreScene.PrimitiveVariable( IECoreScene.PrimitiveVariable.Interpolation.Vertex, IECore.FloatVectorData( [ 1 ] * 10 ), ) p2 = IECoreScene.PointsPrimitive( IECore.V3fVectorData( [ imath.V3f( 20 ) ] * 10 ) ) p2["width"] = IECoreScene.PrimitiveVariable( IECoreScene.PrimitiveVariable.Interpolation.Vertex, IECore.FloatVectorData( [ 2 ] * 10 ), ) with IECoreArnold.UniverseBlock( writable = True ) : n = IECoreArnold.NodeAlgo.convert( [ p1, p2 ], -0.25, 0.25, "testPoints" ) a = arnold.AiNodeGetArray( n, "points" ) self.assertEqual( arnold.AiArrayGetNumElements( a.contents ), 10 ) self.assertEqual( arnold.AiArrayGetNumKeys( a.contents ), 2 ) r = arnold.AiNodeGetArray( n, "radius" ) self.assertEqual( arnold.AiArrayGetNumElements( r.contents ), 10 ) self.assertEqual( arnold.AiArrayGetNumKeys( r.contents ), 2 ) for i in range( 0, 10 ) : self.assertEqual( arnold.AiArrayGetVec( a, i ), arnold.AtVector( 10 ) ) self.assertEqual( arnold.AiArrayGetFlt( r, i ), 0.5 ) for i in range( 11, 20 ) : self.assertEqual( arnold.AiArrayGetVec( a, i ), arnold.AtVector( 20 ) ) self.assertEqual( arnold.AiArrayGetFlt( r, i ), 1 ) self.assertEqual( arnold.AiNodeGetFlt( n, "motion_start" ), -0.25 ) self.assertEqual( arnold.AiNodeGetFlt( n, "motion_end" ), 0.25 )
def testSampleDeduplication(self): camera = IECoreScene.Camera() camera.setProjection("perspective") with IECoreArnold.UniverseBlock(writable=True): animatedNode = IECoreArnold.NodeAlgo.convert([camera, camera], 1.0, 2.0, "samples") node = IECoreArnold.NodeAlgo.convert(camera, "sample") for parameter in [ "screen_window_min", "screen_window_max", "fov", "aperture_size", "focus_distance", ]: if parameter.startswith("screen_"): self.assertEqual( arnold.AiNodeGetVec2(animatedNode, parameter).x, arnold.AiNodeGetVec2(node, parameter).x) self.assertEqual( arnold.AiNodeGetVec2(animatedNode, parameter).y, arnold.AiNodeGetVec2(node, parameter).y) else: self.assertEqual( arnold.AiNodeGetFlt(animatedNode, parameter), arnold.AiNodeGetFlt(node, parameter)) array = arnold.AiNodeGetArray(animatedNode, parameter) self.assertEqual(arnold.AiArrayGetNumElements(array), 1) self.assertEqual(arnold.AiArrayGetNumKeys(array), 1)
def testMotion(self): c1 = IECoreScene.CurvesPrimitive(IECore.IntVectorData([4])) c2 = IECoreScene.CurvesPrimitive(IECore.IntVectorData([4])) c1["P"] = IECoreScene.PrimitiveVariable( IECoreScene.PrimitiveVariable.Interpolation.Vertex, IECore.V3fVectorData([imath.V3f(1)] * 4), ) c2["P"] = IECoreScene.PrimitiveVariable( IECoreScene.PrimitiveVariable.Interpolation.Vertex, IECore.V3fVectorData([imath.V3f(2)] * 4), ) with IECoreArnold.UniverseBlock(writable=True) as universe: n = IECoreArnold.NodeAlgo.convert([c1, c2], -0.25, 0.25, universe, "testCurve") a = arnold.AiNodeGetArray(n, "points") self.assertEqual(arnold.AiArrayGetNumElements(a.contents), 4) self.assertEqual(arnold.AiArrayGetNumKeys(a.contents), 2) for i in range(0, 4): self.assertEqual(arnold.AiArrayGetVec(a, i), arnold.AtVector(1)) for i in range(4, 8): self.assertEqual(arnold.AiArrayGetVec(a, i), arnold.AtVector(2)) self.assertEqual(arnold.AiNodeGetFlt(n, "motion_start"), -0.25) self.assertEqual(arnold.AiNodeGetFlt(n, "motion_end"), 0.25)
def testMotion(self): with IECoreArnold.UniverseBlock(writable=True): c = IECoreArnold.InstancingConverter() m1 = IECore.MeshPrimitive.createPlane( IECore.Box2f(IECore.V2f(-1), IECore.V2f(1))) m2 = IECore.MeshPrimitive.createPlane( IECore.Box2f(IECore.V2f(-2), IECore.V2f(2))) n1 = c.convert([m1, m2], -0.25, 0.25, "testMesh") self.assertEqual( arnold.AiNodeEntryGetName(arnold.AiNodeGetNodeEntry(n1)), "polymesh") self.assertEqual( arnold.AiArrayGetNumKeys( arnold.AiNodeGetArray(n1, "vlist").contents), 2) n2 = c.convert([m1, m2], -0.25, 0.25, "testInstance") self.assertEqual( arnold.AiNodeEntryGetName(arnold.AiNodeGetNodeEntry(n2)), "ginstance") self.assertEqual(arnold.AiNodeGetPtr(n2, "node"), ctypes.addressof(n1.contents)) n3 = c.convert([m2, m1], -0.25, 0.25, "testMesh") self.assertEqual( arnold.AiNodeEntryGetName(arnold.AiNodeGetNodeEntry(n1)), "polymesh") self.assertEqual( arnold.AiArrayGetNumKeys( arnold.AiNodeGetArray(n1, "vlist").contents), 2) n4 = c.convert([m1, m2], -0.5, 0.5, "testInstance") self.assertEqual( arnold.AiNodeEntryGetName(arnold.AiNodeGetNodeEntry(n1)), "polymesh") self.assertEqual( arnold.AiArrayGetNumKeys( arnold.AiNodeGetArray(n1, "vlist").contents), 2)
def testMotion(self): m1 = IECore.MeshPrimitive.createPlane( IECore.Box2f(IECore.V2f(-1), IECore.V2f(1))) IECore.MeshNormalsOp()(input=m1, copyInput=False) m2 = m1.copy() m2["P"].data[0] -= IECore.V3f(0, 0, 1) m2["P"].data[1] -= IECore.V3f(0, 0, 1) IECore.MeshNormalsOp()(input=m2, copyInput=False) with IECoreArnold.UniverseBlock(writable=True): node = IECoreArnold.NodeAlgo.convert([m1, m2], -0.25, 0.25, "testMesh") vList = arnold.AiNodeGetArray(node, "vlist") self.assertEqual(arnold.AiArrayGetNumElements(vList.contents), 4) self.assertEqual(arnold.AiArrayGetNumKeys(vList.contents), 2) nList = arnold.AiNodeGetArray(node, "nlist") self.assertEqual(arnold.AiArrayGetNumElements(nList.contents), 4) self.assertEqual(arnold.AiArrayGetNumKeys(nList.contents), 2) for i in range(0, 4): p = arnold.AiArrayGetVec(vList, i) self.assertEqual(IECore.V3f(p.x, p.y, p.z), m1["P"].data[i]) n = arnold.AiArrayGetVec(nList, i) self.assertEqual(IECore.V3f(n.x, n.y, n.z), m1["N"].data[i]) for i in range(4, 8): p = arnold.AiArrayGetVec(vList, i) self.assertEqual(IECore.V3f(p.x, p.y, p.z), m2["P"].data[i - 4]) n = arnold.AiArrayGetVec(nList, i) self.assertEqual(IECore.V3f(n.x, n.y, n.z), m2["N"].data[i - 4]) self.assertEqual(arnold.AiNodeGetFlt(node, "motion_start"), -0.25) self.assertEqual(arnold.AiNodeGetFlt(node, "motion_end"), 0.25)
def testNPrimitiveVariable( self ) : c = IECore.CurvesPrimitive( IECore.IntVectorData( [ 4 ] ), IECore.CubicBasisf.catmullRom() ) c["P"] = IECore.PrimitiveVariable( IECore.PrimitiveVariable.Interpolation.Vertex, IECore.V3fVectorData( [ IECore.V3f( x, 0, 0 ) for x in range( 0, 4 ) ] ) ) with IECoreArnold.UniverseBlock( writable = True ) : # No N - should be a ribbon n = IECoreArnold.NodeAlgo.convert( c, "testCurve" ) self.assertEqual( arnold.AiNodeGetStr( n, "mode" ), "ribbon" ) self.assertEqual( arnold.AiArrayGetNumElements( arnold.AiNodeGetArray( n, "orientations" ).contents ), 0 ) # N - should be oriented c["N"] = IECore.PrimitiveVariable( IECore.PrimitiveVariable.Interpolation.Vertex, IECore.V3fVectorData( [ IECore.V3f( 0, math.sin( x ), math.cos( x ) ) for x in range( 0, 4 ) ] ) ) n = IECoreArnold.NodeAlgo.convert( c, "testCurve" ) self.assertEqual( arnold.AiNodeGetStr( n, "mode" ), "oriented" ) orientations = arnold.AiNodeGetArray( n, "orientations" ) self.assertEqual( arnold.AiArrayGetNumElements( orientations.contents ), 4 ) for i in range( 0, 4 ) : self.assertEqual( arnold.AiArrayGetVec( orientations, i ), arnold.AtVector( 0, math.sin( i ), math.cos( i ) ) ) # Motion blurred N - should be oriented and deforming c2 = c.copy() c2["N"] = IECore.PrimitiveVariable( IECore.PrimitiveVariable.Interpolation.Vertex, IECore.V3fVectorData( [ IECore.V3f( 0, math.sin( x + 0.2 ), math.cos( x + 0.2 ) ) for x in range( 0, 4 ) ] ) ) n = IECoreArnold.NodeAlgo.convert( [ c, c2 ], 0.0, 1.0, "testCurve" ) self.assertEqual( arnold.AiNodeGetStr( n, "mode" ), "oriented" ) orientations = arnold.AiNodeGetArray( n, "orientations" ) self.assertEqual( arnold.AiArrayGetNumElements( orientations.contents ), 4 ) self.assertEqual( arnold.AiArrayGetNumKeys( orientations.contents ), 2 ) for i in range( 0, 4 ) : self.assertEqual( arnold.AiArrayGetVec( orientations, i ), arnold.AtVector( 0, math.sin( i ), math.cos( i ) ) ) self.assertEqual( arnold.AiArrayGetVec( orientations, i + 4 ), arnold.AtVector( 0, math.sin( i + 0.2 ), math.cos( i + 0.2 ) ) )
def testTransformMotion( self ) : s = Gaffer.ScriptNode() s["plane"] = GafferScene.Plane() s["sphere"] = GafferScene.Sphere() s["group"] = GafferScene.Group() s["group"]["in"][0].setInput( s["plane"]["out"] ) s["group"]["in"][1].setInput( s["sphere"]["out"] ) s["expression"] = Gaffer.Expression() s["expression"].setExpression( inspect.cleandoc( """ parent["plane"]["transform"]["translate"]["x"] = context.getFrame() parent["sphere"]["transform"]["translate"]["y"] = context.getFrame() * 2 parent["group"]["transform"]["translate"]["z"] = context.getFrame() - 1 """ ) ) s["planeFilter"] = GafferScene.PathFilter() s["planeFilter"]["paths"].setValue( IECore.StringVectorData( [ "/group/plane" ] ) ) s["attributes"] = GafferScene.StandardAttributes() s["attributes"]["in"].setInput( s["group"]["out"] ) s["attributes"]["filter"].setInput( s["planeFilter"]["out"] ) s["attributes"]["attributes"]["transformBlur"]["enabled"].setValue( True ) s["attributes"]["attributes"]["transformBlur"]["value"].setValue( False ) s["options"] = GafferScene.StandardOptions() s["options"]["in"].setInput( s["attributes"]["out"] ) s["options"]["options"]["shutter"]["enabled"].setValue( True ) s["options"]["options"]["transformBlur"]["enabled"].setValue( True ) s["render"] = GafferArnold.ArnoldRender() s["render"]["in"].setInput( s["options"]["out"] ) s["render"]["mode"].setValue( s["render"].Mode.SceneDescriptionMode ) s["render"]["fileName"].setValue( self.temporaryDirectory() + "/test.ass" ) # No motion blur s["options"]["options"]["transformBlur"]["value"].setValue( False ) s["render"]["task"].execute() with IECoreArnold.UniverseBlock( writable = True ) : arnold.AiASSLoad( self.temporaryDirectory() + "/test.ass" ) camera = arnold.AiNodeLookUpByName( "gaffer:defaultCamera" ) sphere = arnold.AiNodeLookUpByName( "/group/sphere" ) sphereMotionStart = arnold.AiNodeGetFlt( sphere, "motion_start" ) sphereMotionEnd = arnold.AiNodeGetFlt( sphere, "motion_end" ) sphereMatrix = arnold.AiNodeGetMatrix( sphere, "matrix" ) plane = arnold.AiNodeLookUpByName( "/group/plane" ) planeMotionStart = arnold.AiNodeGetFlt( plane, "motion_start" ) planeMotionEnd = arnold.AiNodeGetFlt( plane, "motion_end" ) planeMatrix = arnold.AiNodeGetMatrix( plane, "matrix" ) # Motion parameters should be left at default self.assertEqual( sphereMotionStart, 0 ) self.assertEqual( sphereMotionEnd, 1 ) self.assertEqual( planeMotionStart, 0 ) self.assertEqual( planeMotionEnd, 1 ) expectedSphereMatrix = arnold.AiM4Translation( arnold.AtVector( 0, 2, 0 ) ) expectedPlaneMatrix = arnold.AiM4Translation( arnold.AtVector( 1, 0, 0 ) ) self.assertEqual( self.__m44f( sphereMatrix ), self.__m44f( expectedSphereMatrix ) ) self.assertEqual( self.__m44f( planeMatrix ), self.__m44f( expectedPlaneMatrix ) ) self.assertEqual( arnold.AiNodeGetFlt( camera, "shutter_start" ), 1 ) self.assertEqual( arnold.AiNodeGetFlt( camera, "shutter_end" ), 1 ) # Motion blur s["options"]["options"]["transformBlur"]["value"].setValue( True ) s["render"]["task"].execute() with IECoreArnold.UniverseBlock( writable = True ) : arnold.AiASSLoad( self.temporaryDirectory() + "/test.ass" ) camera = arnold.AiNodeLookUpByName( "gaffer:defaultCamera" ) sphere = arnold.AiNodeLookUpByName( "/group/sphere" ) sphereMotionStart = arnold.AiNodeGetFlt( sphere, "motion_start" ) sphereMotionEnd = arnold.AiNodeGetFlt( sphere, "motion_end" ) sphereMatrices = arnold.AiNodeGetArray( sphere, "matrix" ) plane = arnold.AiNodeLookUpByName( "/group/plane" ) planeMotionStart = arnold.AiNodeGetFlt( plane, "motion_start" ) planeMotionEnd = arnold.AiNodeGetFlt( plane, "motion_end" ) planeMatrices = arnold.AiNodeGetArray( plane, "matrix" ) self.assertEqual( sphereMotionStart, 0.75 ) self.assertEqual( sphereMotionEnd, 1.25 ) self.assertEqual( arnold.AiArrayGetNumElements( sphereMatrices.contents ), 1 ) self.assertEqual( arnold.AiArrayGetNumKeys( sphereMatrices.contents ), 2 ) self.assertEqual( planeMotionStart, 0.75 ) self.assertEqual( planeMotionEnd, 1.25 ) self.assertEqual( arnold.AiArrayGetNumElements( planeMatrices.contents ), 1 ) self.assertEqual( arnold.AiArrayGetNumKeys( planeMatrices.contents ), 2 ) for i in range( 0, 2 ) : frame = 0.75 + 0.5 * i sphereMatrix = arnold.AiArrayGetMtx( sphereMatrices, i ) expectedSphereMatrix = arnold.AiM4Translation( arnold.AtVector( 0, frame * 2, frame - 1 ) ) planeMatrix = arnold.AiArrayGetMtx( planeMatrices, i ) expectedPlaneMatrix = arnold.AiM4Translation( arnold.AtVector( 1, 0, frame - 1 ) ) self.assertEqual( self.__m44f( sphereMatrix ), self.__m44f( expectedSphereMatrix ) ) self.assertEqual( self.__m44f( planeMatrix ), self.__m44f( expectedPlaneMatrix ) ) self.assertEqual( arnold.AiNodeGetFlt( camera, "shutter_start" ), 0.75 ) self.assertEqual( arnold.AiNodeGetFlt( camera, "shutter_end" ), 1.25 ) # Motion blur on, but sampleMotion off s["options"]["options"]["sampleMotion"]["enabled"].setValue( True ) s["options"]["options"]["sampleMotion"]["value"].setValue( False ) s["render"]["task"].execute() with IECoreArnold.UniverseBlock( writable = True ) : arnold.AiASSLoad( self.temporaryDirectory() + "/test.ass" ) camera = arnold.AiNodeLookUpByName( "gaffer:defaultCamera" ) sphere = arnold.AiNodeLookUpByName( "/group/sphere" ) sphereMotionStart = arnold.AiNodeGetFlt( sphere, "motion_start" ) sphereMotionEnd = arnold.AiNodeGetFlt( sphere, "motion_end" ) sphereMatrices = arnold.AiNodeGetArray( sphere, "matrix" ) plane = arnold.AiNodeLookUpByName( "/group/plane" ) planeMotionStart = arnold.AiNodeGetFlt( plane, "motion_start" ) planeMotionEnd = arnold.AiNodeGetFlt( plane, "motion_end" ) planeMatrices = arnold.AiNodeGetArray( plane, "matrix" ) self.assertEqual( sphereMotionStart, 0.75 ) self.assertEqual( sphereMotionEnd, 1.25 ) self.assertEqual( arnold.AiArrayGetNumElements( sphereMatrices.contents ), 1 ) self.assertEqual( arnold.AiArrayGetNumKeys( sphereMatrices.contents ), 2 ) self.assertEqual( planeMotionStart, 0.75 ) self.assertEqual( planeMotionEnd, 1.25 ) self.assertEqual( arnold.AiArrayGetNumElements( planeMatrices.contents ), 1 ) self.assertEqual( arnold.AiArrayGetNumKeys( planeMatrices.contents ), 2 ) for i in range( 0, 2 ) : frame = 0.75 + 0.5 * i sphereMatrix = arnold.AiArrayGetMtx( sphereMatrices, i ) expectedSphereMatrix = arnold.AiM4Translation( arnold.AtVector( 0, frame * 2, frame - 1 ) ) planeMatrix = arnold.AiArrayGetMtx( planeMatrices, i ) expectedPlaneMatrix = arnold.AiM4Translation( arnold.AtVector( 1, 0, frame - 1 ) ) self.assertEqual( self.__m44f( sphereMatrix ), self.__m44f( expectedSphereMatrix ) ) self.assertEqual( self.__m44f( planeMatrix ), self.__m44f( expectedPlaneMatrix ) ) self.assertEqual( arnold.AiNodeGetFlt( camera, "shutter_start" ), 0.75 ) self.assertEqual( arnold.AiNodeGetFlt( camera, "shutter_end" ), 0.75 )
def testConvertAnimatedParameters(self): with IECoreArnold.UniverseBlock(writable=True): samples = [] for i in range(0, 2): camera = IECoreScene.Camera() camera.setProjection("perspective") camera.setFocalLengthFromFieldOfView(45 * (i + 1)) camera.setAperture(imath.V2f(10, 10 + i)) camera.setFStop(i + 1) camera.setFocusDistance(i + 100) samples.append(camera) animatedNode = IECoreArnold.NodeAlgo.convert( samples, 1.0, 2.0, "samples") nodes = [ IECoreArnold.NodeAlgo.convert(samples[i], "sample{}".format(i)) for i, sample in enumerate(samples) ] self.assertEqual(arnold.AiNodeGetFlt(animatedNode, "motion_start"), 1.0) self.assertEqual(arnold.AiNodeGetFlt(animatedNode, "motion_start"), 1.0) for i, node in enumerate(nodes): animatedScreenWindowMin = arnold.AiArrayGetVec2( arnold.AiNodeGetArray(animatedNode, "screen_window_min"), i) animatedScreenWindowMax = arnold.AiArrayGetVec2( arnold.AiNodeGetArray(animatedNode, "screen_window_max"), i) self.assertEqual( animatedScreenWindowMin.x, arnold.AiNodeGetVec2(node, "screen_window_min").x) self.assertEqual( animatedScreenWindowMin.y, arnold.AiNodeGetVec2(node, "screen_window_min").y) self.assertEqual( animatedScreenWindowMax.x, arnold.AiNodeGetVec2(node, "screen_window_max").x) self.assertEqual( animatedScreenWindowMax.y, arnold.AiNodeGetVec2(node, "screen_window_max").y) self.assertEqual( arnold.AiArrayGetFlt( arnold.AiNodeGetArray(animatedNode, "fov"), i), arnold.AiNodeGetFlt(node, "fov")) self.assertEqual( arnold.AiArrayGetFlt( arnold.AiNodeGetArray(animatedNode, "aperture_size"), i), arnold.AiNodeGetFlt(node, "aperture_size")) self.assertEqual( arnold.AiArrayGetFlt( arnold.AiNodeGetArray(animatedNode, "focus_distance"), i), arnold.AiNodeGetFlt(node, "focus_distance")) for parameter in [ "screen_window_min", "screen_window_max", "fov", "aperture_size", "focus_distance", ]: array = arnold.AiNodeGetArray(animatedNode, "fov") self.assertEqual(arnold.AiArrayGetNumElements(array), 1) self.assertEqual(arnold.AiArrayGetNumKeys(array), 2)
def testVertexToVaryingConversion(self): c = IECoreScene.CurvesPrimitive(IECore.IntVectorData([2, 2]), IECore.CubicBasisf.linear()) c["P"] = IECoreScene.PrimitiveVariable( IECoreScene.PrimitiveVariable.Interpolation.Vertex, IECore.V3fVectorData([imath.V3f(x, 0, 0) for x in range(0, 4)])) c["width"] = IECoreScene.PrimitiveVariable( IECoreScene.PrimitiveVariable.Interpolation.Vertex, IECore.FloatVectorData([1] * 4), ) c["foo"] = IECoreScene.PrimitiveVariable( IECoreScene.PrimitiveVariable.Interpolation.Vertex, IECore.FloatVectorData([1.5] * 4)) self.assertTrue(c.arePrimitiveVariablesValid()) c2 = c.copy() c2["width"] = IECoreScene.PrimitiveVariable( IECoreScene.PrimitiveVariable.Interpolation.Vertex, IECore.FloatVectorData([2] * 4), ) self.assertTrue(c2.arePrimitiveVariablesValid()) c3 = IECoreScene.CurvesPrimitive(IECore.IntVectorData([8, 8]), IECore.CubicBasisf.bSpline()) c3["P"] = IECoreScene.PrimitiveVariable( IECoreScene.PrimitiveVariable.Interpolation.Vertex, IECore.V3fVectorData([ imath.V3f(0, 0, 0), imath.V3f(0, 0, 0), imath.V3f(0, 0, 0), imath.V3f(0, 0.75, 0), imath.V3f(0.25, 1, 0), imath.V3f(1, 1, 0), imath.V3f(1, 1, 0), imath.V3f(1, 1, 0), imath.V3f(0, 0, 1), imath.V3f(0, 0, 1), imath.V3f(0, 0, 1), imath.V3f(0, 0.75, 1), imath.V3f(0.25, 1, 1), imath.V3f(1, 1, 1), imath.V3f(1, 1, 1), imath.V3f(1, 1, 1) ]), ) c3["width"] = IECoreScene.PrimitiveVariable( IECoreScene.PrimitiveVariable.Interpolation.Vertex, IECore.FloatVectorData([1] * 16), ) c3["foo"] = IECoreScene.PrimitiveVariable( IECoreScene.PrimitiveVariable.Interpolation.Vertex, IECore.FloatVectorData([1.5] * 16)) self.assertTrue(c3.arePrimitiveVariablesValid()) c4 = c3.copy() c4["width"] = IECoreScene.PrimitiveVariable( IECoreScene.PrimitiveVariable.Interpolation.Vertex, IECore.FloatVectorData([2] * 16), ) self.assertTrue(c4.arePrimitiveVariablesValid()) with IECoreArnold.UniverseBlock(writable=True) as universe: n = IECoreArnold.NodeAlgo.convert(c, universe, "testLinearCurve") r = arnold.AiNodeGetArray(n, "radius") self.assertEqual(arnold.AiArrayGetNumElements(r.contents), 4) self.assertEqual(arnold.AiArrayGetNumKeys(r.contents), 1) foo = arnold.AiNodeGetArray(n, "foo") self.assertEqual(arnold.AiArrayGetNumElements(foo.contents), 4) self.assertEqual(arnold.AiArrayGetNumKeys(foo.contents), 1) for i in range(0, 4): self.assertEqual(arnold.AiArrayGetFlt(r, i), 0.5) self.assertEqual(arnold.AiArrayGetFlt(foo, i), 1.5) n2 = IECoreArnold.NodeAlgo.convert([c, c2], -0.25, 0.25, universe, "testLinearCurves") r2 = arnold.AiNodeGetArray(n2, "radius") self.assertEqual(arnold.AiArrayGetNumElements(r2.contents), 4) self.assertEqual(arnold.AiArrayGetNumKeys(r2.contents), 2) foo2 = arnold.AiNodeGetArray(n2, "foo") self.assertEqual(arnold.AiArrayGetNumElements(foo2.contents), 4) # arbitrary userdata is not sampled self.assertEqual(arnold.AiArrayGetNumKeys(foo2.contents), 1) for i in range(0, 4): self.assertEqual(arnold.AiArrayGetFlt(r2, i), 0.5) self.assertEqual(arnold.AiArrayGetFlt(foo2, i), 1.5) for i in range(4, 8): self.assertEqual(arnold.AiArrayGetFlt(r2, i), 1) # for cubic curves, radius will have been converted to Varying, so it will have fewer elements n3 = IECoreArnold.NodeAlgo.convert(c3, universe, "testBSplineCurve") r3 = arnold.AiNodeGetArray(n3, "radius") self.assertEqual(arnold.AiArrayGetNumElements(r3.contents), 12) self.assertEqual(arnold.AiArrayGetNumKeys(r3.contents), 1) foo3 = arnold.AiNodeGetArray(n3, "foo") self.assertEqual(arnold.AiArrayGetNumElements(foo3.contents), 12) self.assertEqual(arnold.AiArrayGetNumKeys(foo3.contents), 1) for i in range(0, 12): self.assertEqual(arnold.AiArrayGetFlt(r3, i), 0.5) self.assertEqual(arnold.AiArrayGetFlt(foo3, i), 1.5) n4 = IECoreArnold.NodeAlgo.convert([c3, c4], -0.25, 0.25, universe, "testBSplineCurves") r4 = arnold.AiNodeGetArray(n4, "radius") self.assertEqual(arnold.AiArrayGetNumElements(r4.contents), 12) self.assertEqual(arnold.AiArrayGetNumKeys(r4.contents), 2) foo4 = arnold.AiNodeGetArray(n4, "foo") self.assertEqual(arnold.AiArrayGetNumElements(foo4.contents), 12) # arbitrary userdata is not sampled self.assertEqual(arnold.AiArrayGetNumKeys(foo4.contents), 1) for i in range(0, 12): self.assertEqual(arnold.AiArrayGetFlt(r4, i), 0.5) self.assertEqual(arnold.AiArrayGetFlt(foo4, i), 1.5) for i in range(12, 24): self.assertEqual(arnold.AiArrayGetFlt(r4, i), 1)