def testMode(self):
with IECoreArnold.UniverseBlock(writable=True):
p = IECore.PointsPrimitive(
IECore.V3fVectorData([IECore.V3f(i) for i in range(0, 10)]))
n = IECoreArnold.NodeAlgo.convert(p)
self.assertEqual(arnold.AiNodeGetStr(n, "mode"), "disk")
p["type"] = IECore.PrimitiveVariable(
IECore.PrimitiveVariable.Interpolation.Constant, "particle")
n = IECoreArnold.NodeAlgo.convert(p)
self.assertEqual(arnold.AiNodeGetStr(n, "mode"), "disk")
p["type"] = IECore.PrimitiveVariable(
IECore.PrimitiveVariable.Interpolation.Constant, "disk")
n = IECoreArnold.NodeAlgo.convert(p)
self.assertEqual(arnold.AiNodeGetStr(n, "mode"), "disk")
p["type"] = IECore.PrimitiveVariable(
IECore.PrimitiveVariable.Interpolation.Constant, "sphere")
n = IECoreArnold.NodeAlgo.convert(p)
self.assertEqual(arnold.AiNodeGetStr(n, "mode"), "sphere")
p["type"] = IECore.PrimitiveVariable(
IECore.PrimitiveVariable.Interpolation.Constant, "patch")
n = IECoreArnold.NodeAlgo.convert(p)
self.assertEqual(arnold.AiNodeGetStr(n, "mode"), "quad")
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)
self.assertEqual(arnold.AiNodeGetStr(n, "mode"), "ribbon")
self.assertEqual(
arnold.AiNodeGetArray(n, "orientations").contents.nelements, 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)
self.assertEqual(arnold.AiNodeGetStr(n, "mode"), "oriented")
orientations = arnold.AiNodeGetArray(n, "orientations")
self.assertEqual(orientations.contents.nelements, 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])
self.assertEqual(arnold.AiNodeGetStr(n, "mode"), "oriented")
orientations = arnold.AiNodeGetArray(n, "orientations")
self.assertEqual(orientations.contents.nelements, 4)
self.assertEqual(orientations.contents.nkeys, 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 testProcedural(self):
r = IECoreArnold.Renderer("/tmp/test.ass")
with IECore.WorldBlock(r):
r.procedural(
r.ExternalProcedural(
"someVolumeThing.so",
IECore.Box3f(IECore.V3f(-1, -2, -3), IECore.V3f(4, 5, 6)),
{
"ai:nodeType": "volume",
"testFloat": 0.5
}))
volume = self.__allNodes(type=arnold.AI_NODE_SHAPE)[-1]
self.assertEqual(
arnold.AiNodeEntryGetName(arnold.AiNodeGetNodeEntry(volume)),
"volume")
self.assertEqual(arnold.AiNodeGetPnt(volume, "min"),
arnold.AtPoint(-1, -2, -3))
self.assertEqual(arnold.AiNodeGetPnt(volume, "max"),
arnold.AtPoint(4, 5, 6))
self.assertEqual(arnold.AiNodeGetStr(volume, "dso"),
"someVolumeThing.so")
self.assertEqual(arnold.AiNodeGetFlt(volume, "testFloat"), 0.5)
def testPrimitiveVariables( self ) : s = IECore.SpherePrimitive() s["v"] = IECore.PrimitiveVariable( IECore.PrimitiveVariable.Interpolation.Constant, IECore.V3f( 1, 2, 3 ) ) s["c"] = IECore.PrimitiveVariable( IECore.PrimitiveVariable.Interpolation.Constant, IECore.Color3f( 1, 2, 3 ) ) s["s"] = IECore.PrimitiveVariable( IECore.PrimitiveVariable.Interpolation.Constant, "test" ) s["i"] = IECore.PrimitiveVariable( IECore.PrimitiveVariable.Interpolation.Constant, 11 ) s["b"] = IECore.PrimitiveVariable( IECore.PrimitiveVariable.Interpolation.Constant, True ) s["f"] = IECore.PrimitiveVariable( IECore.PrimitiveVariable.Interpolation.Constant, 2.5 ) s["m"] = IECore.PrimitiveVariable( IECore.PrimitiveVariable.Interpolation.Constant, IECore.M44f( 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16) ) with IECoreArnold.UniverseBlock( writable = True ) : n = IECoreArnold.NodeAlgo.convert( s, "testSphere" ) self.assertEqual( arnold.AiNodeGetVec( n, "v" ), arnold.AtVector( 1, 2, 3 ) ) self.assertEqual( arnold.AiNodeGetRGB( n, "c" ), arnold.AtRGB( 1, 2, 3 ) ) self.assertEqual( arnold.AiNodeGetStr( n, "s" ), "test" ) self.assertEqual( arnold.AiNodeGetInt( n, "i" ), 11 ) self.assertEqual( arnold.AiNodeGetBool( n, "b" ), True ) self.assertEqual( arnold.AiNodeGetFlt( n, "f" ), 2.5 ) m = arnold.AiNodeGetMatrix( n, "m" ) self.assertEqual( [ list( i ) for i in m.data ], [ [1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16] ], )
def testSetParameter( self ) : with IECoreArnold.UniverseBlock( writable = True ) : n = arnold.AiNode( "standard_surface" ) IECoreArnold.ParameterAlgo.setParameter( n, "base", IECore.FloatData( 0.25 ) ) IECoreArnold.ParameterAlgo.setParameter( n, "customString", IECore.StringData( "test" ) ) self.assertEqual( arnold.AiNodeGetFlt( n, "base" ), 0.25 ) self.assertEqual( arnold.AiNodeGetStr( n, "customString" ), "test" )
def testBlindData(self):
flat = IECoreScene.Shader("flat")
flat.blindData().update({
"user:testInt":
IECore.IntData(1),
"user:testFloat":
IECore.FloatData(2.5),
"user:testV3f":
IECore.V3fData(imath.V3f(1, 2, 3)),
"user:testColor3f":
IECore.Color3fData(imath.Color3f(4, 5, 6)),
"user:testString":
IECore.StringData("we're all doomed"),
})
network = IECoreScene.ShaderNetwork(shaders={
"noiseHandle":
IECoreScene.Shader("noise"),
"flatHandle":
flat,
},
connections=[
(("noiseHandle", ""),
("flatHandle", "color")),
],
output="flatHandle")
with IECoreArnold.UniverseBlock(writable=True) as universe:
nodes = IECoreArnoldPreview.ShaderNetworkAlgo.convert(
network, universe, "test")
self.assertEqual(len(nodes), 2)
self.assertEqual(
arnold.AiNodeEntryGetName(arnold.AiNodeGetNodeEntry(nodes[0])),
"noise")
self.assertEqual(
arnold.AiNodeEntryGetName(arnold.AiNodeGetNodeEntry(nodes[1])),
"flat")
self.assertEqual(arnold.AiNodeGetName(nodes[0]),
"test:noiseHandle")
self.assertEqual(arnold.AiNodeGetName(nodes[1]), "test")
self.assertEqual(arnold.AiNodeGetInt(nodes[1], "user:testInt"), 1)
self.assertEqual(arnold.AiNodeGetFlt(nodes[1], "user:testFloat"),
2.5)
self.assertEqual(arnold.AiNodeGetVec(nodes[1], "user:testV3f"),
arnold.AtVector(1, 2, 3))
self.assertEqual(arnold.AiNodeGetRGB(nodes[1], "user:testColor3f"),
arnold.AtRGB(4, 5, 6))
self.assertEqual(arnold.AiNodeGetStr(nodes[1], "user:testString"),
"we're all doomed")
def testSetParameter(self):
with IECoreArnold.UniverseBlock(writable=True):
n = arnold.AiNode("standard")
IECoreArnold.ParameterAlgo.setParameter(n, "Kd",
IECore.FloatData(0.25))
IECoreArnold.ParameterAlgo.setParameter(n, "aov_emission",
IECore.StringData("test"))
self.assertEqual(arnold.AiNodeGetFlt(n, "Kd"), 0.25)
self.assertEqual(arnold.AiNodeGetStr(n, "aov_emission"), "test")
def Init(procName):
proc = arnold.AiNodeLookUpByName(procName)
if not proc:
print("No such procedural: %s" % procName)
return (0, None)
attrs = {}
it = arnold.AiNodeGetUserParamIterator(proc)
while not arnold.AiUserParamIteratorFinished(it):
param = arnold.AiUserParamIteratorGetNext(it)
pname = arnold.AiUserParamGetName(param)
pcat = arnold.AiUserParamGetCategory(param)
if pcat == arnold.AI_USERDEF_CONSTANT:
ptype = arnold.AiUserParamGetType(param)
pval = None
if ptype == arnold.AI_TYPE_BOOLEAN:
pval = arnold.AiNodeGetBool(proc, pname)
elif ptype == arnold.AI_TYPE_INT:
pval = arnold.AiNodeGetInt(proc, pname)
elif ptype == arnold.AI_TYPE_UINT:
pval = arnold.AiNodeGetUInt(proc, pname)
elif ptype == arnold.AI_TYPE_FLOAT:
pval = arnold.AiNodeGetFlt(proc, pname)
elif ptype == arnold.AI_TYPE_POINT:
pval = arnold.AiNodeGetPnt(proc, pname)
elif ptype == arnold.AI_TYPE_POINT2:
pval = arnold.AiNodeGetPnt2(proc, pname)
elif ptype == arnold.AI_TYPE_VECTOR:
pval = arnold.AiNodeGetVec(proc, pname)
elif ptype == arnold.AI_TYPE_RGB:
pval = arnold.AiNodeGetRGB(proc, pname)
elif ptype == arnold.AI_TYPE_RGBA:
pval = arnold.AiNodeGetRGBA(proc, pname)
elif ptype == arnold.AI_TYPE_STRING:
pval = arnold.AiNodeGetStr(proc, pname)
if pval != None:
attrs[pname] = (ptype, pval)
else:
print("Unsupported type (%d) for parameter \"%s\"" %
(ptype, pname))
else:
print("Ignore non constant parameter \"%s\"" % pname)
arnold.AiUserParamIteratorFinished(it)
return (1, attrs)
def render_ass(ass, remove_ass=False):
imagefilename = None
ai.AiBegin()
ai.AiLoadPlugins(os.environ['ARNOLD_PLUGIN_PATH'])
ai.AiASSLoad (ass, ai.AI_NODE_ALL)
ai.AiRender()
ai.AiEnd()
# read out file
ai.AiBegin()
ai.AiMsgSetConsoleFlags(ai.AI_LOG_ALL)
ai.AiASSLoad(ass, ai.AI_NODE_ALL)
iter = ai.AiUniverseGetNodeIterator(ai.AI_NODE_ALL)
while not ai.AiNodeIteratorFinished(iter):
node = ai.AiNodeIteratorGetNext(iter)
if ai.AiNodeIs( node, "driver_jpeg" ):
imagefilename = ai.AiNodeGetStr( node, "filename" )
ai.AiNodeIteratorDestroy(iter)
ai.AiEnd()
if remove_ass:
os.remove(ass)
return imagefilename
def testShaderSubstitutions( self ) : s = Gaffer.ScriptNode() s["plane"] = GafferScene.Plane() s["planeAttrs"] = GafferScene.CustomAttributes() s["planeAttrs"]["in"].setInput( s["plane"]["out"] ) s["planeAttrs"]["attributes"].addChild( Gaffer.NameValuePlug( "A", Gaffer.StringPlug( "value", defaultValue = 'bar' ) ) ) s["planeAttrs"]["attributes"].addChild( Gaffer.NameValuePlug( "B", Gaffer.StringPlug( "value", defaultValue = 'foo' ) ) ) s["cube"] = GafferScene.Cube() s["cubeAttrs"] = GafferScene.CustomAttributes() s["cubeAttrs"]["in"].setInput( s["cube"]["out"] ) s["cubeAttrs"]["attributes"].addChild( Gaffer.NameValuePlug( "B", Gaffer.StringPlug( "value", defaultValue = 'override' ) ) ) s["parent"] = GafferScene.Parent() s["parent"]["in"].setInput( s["planeAttrs"]["out"] ) s["parent"]["children"][0].setInput( s["cubeAttrs"]["out"] ) s["parent"]["parent"].setValue( "/plane" ) s["shader"] = GafferArnold.ArnoldShader() s["shader"].loadShader( "image" ) s["shader"]["parameters"]["filename"].setValue( "<attr:A>/path/<attr:B>.tx" ) s["filter"] = GafferScene.PathFilter() s["filter"]["paths"].setValue( IECore.StringVectorData( [ "/plane" ] ) ) s["shaderAssignment"] = GafferScene.ShaderAssignment() s["shaderAssignment"]["in"].setInput( s["parent"]["out"] ) s["shaderAssignment"]["filter"].setInput( s["filter"]["out"] ) s["shaderAssignment"]["shader"].setInput( s["shader"]["out"] ) s["light"] = GafferArnold.ArnoldLight() s["light"].loadShader( "photometric_light" ) s["light"]["parameters"]["filename"].setValue( "/path/<attr:A>.ies" ) s["goboTexture"] = GafferArnold.ArnoldShader() s["goboTexture"].loadShader( "image" ) s["goboTexture"]["parameters"]["filename"].setValue( "<attr:B>/gobo.tx" ) s["gobo"] = GafferArnold.ArnoldShader() s["gobo"].loadShader( "gobo" ) s["gobo"]["parameters"]["slidemap"].setInput( s["goboTexture"]["out"] ) s["goboAssign"] = GafferScene.ShaderAssignment() s["goboAssign"]["in"].setInput( s["light"]["out"] ) s["goboAssign"]["shader"].setInput( s["gobo"]["out"] ) s["lightBlocker"] = GafferArnold.ArnoldLightFilter() s["lightBlocker"].loadShader( "light_blocker" ) s["lightBlocker"]["parameters"]["geometry_type"].setValue( "<attr:geometryType>" ) s["lightGroup"] = GafferScene.Group() s["lightGroup"]["name"].setValue( "lightGroup" ) s["lightGroup"]["in"][0].setInput( s["goboAssign"]["out"] ) s["lightGroup"]["in"][1].setInput( s["lightBlocker"]["out"] ) s["parent2"] = GafferScene.Parent() s["parent2"]["in"].setInput( s["shaderAssignment"]["out"] ) s["parent2"]["children"][0].setInput( s["lightGroup"]["out"] ) s["parent2"]["parent"].setValue( "/" ) s["globalAttrs"] = GafferScene.CustomAttributes() s["globalAttrs"]["in"].setInput( s["parent2"]["out"] ) s["globalAttrs"]["global"].setValue( True ) s["globalAttrs"]["attributes"].addChild( Gaffer.NameValuePlug( "A", Gaffer.StringPlug( "value", defaultValue = 'default1' ) ) ) s["globalAttrs"]["attributes"].addChild( Gaffer.NameValuePlug( "B", Gaffer.StringPlug( "value", defaultValue = 'default2' ) ) ) s["globalAttrs"]["attributes"].addChild( Gaffer.NameValuePlug( "geometryType", Gaffer.StringPlug( "value", defaultValue = 'cylinder' ) ) ) s["render"] = GafferArnold.ArnoldRender() s["render"]["in"].setInput( s["globalAttrs"]["out"] ) s["render"]["mode"].setValue( s["render"].Mode.SceneDescriptionMode ) s["render"]["fileName"].setValue( self.temporaryDirectory() + "/test.ass" ) s["render"]["task"].execute() with IECoreArnold.UniverseBlock( writable = True ) : arnold.AiASSLoad( self.temporaryDirectory() + "/test.ass" ) plane = arnold.AiNodeLookUpByName( "/plane" ) shader = arnold.AiNodeGetPtr( plane, "shader" ) self.assertEqual( arnold.AiNodeGetStr( shader, "filename" ), "bar/path/foo.tx" ) cube = arnold.AiNodeLookUpByName( "/plane/cube" ) shader2 = arnold.AiNodeGetPtr( cube, "shader" ) self.assertEqual( arnold.AiNodeGetStr( shader2, "filename" ), "bar/path/override.tx" ) light = arnold.AiNodeLookUpByName( "light:/lightGroup/light" ) self.assertEqual( arnold.AiNodeGetStr( light, "filename" ), "/path/default1.ies" ) gobo = arnold.AiNodeGetPtr( light, "filters" ) goboTex = arnold.AiNodeGetLink( gobo, "slidemap" ) self.assertEqual( arnold.AiNodeGetStr( goboTex, "filename" ), "default2/gobo.tx" ) lightFilter = arnold.AiNodeLookUpByName( "lightFilter:/lightGroup/lightFilter" ) self.assertEqual( arnold.AiNodeGetStr( lightFilter, "geometry_type" ), "cylinder" )
