ArnoldShaderUI : use mechanism provided by ShaderUI for userDefault r…

…egistration
GafferHQ · Aug 14, 2019 · 780b4ca · 780b4ca
1 parent b239d1c
commit 780b4ca
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/python/GafferArnoldUI/ArnoldShaderUI.py b/python/GafferArnoldUI/ArnoldShaderUI.py
@@ -283,8 +283,9 @@ def __translateNodeMetadata( nodeEntry ) :
 			userDefault = __aiMetadataGetStr( nodeEntry, paramName, "gaffer.userDefault" )
 
 		if userDefault:
-			__metadata[paramPath]["userDefault"] = userDefault
-
+			nodeName, _, plugName = paramPath.split( "." )
+			Gaffer.Metadata.registerValue( "ai:surface:%s:%s" % ( nodeName, plugName ), "userDefault", userDefault )
+
 
 with IECoreArnold.UniverseBlock( writable = False ) :
 
@@ -350,7 +351,8 @@ def __plugMetadata( plug, name ) :
 	parameterPlugComponentKeys = set()
 
 	for name, metadata in __metadata.items() :
-		keys = ( nodeKeys, parametersPlugKeys, parameterPlugKeys, parameterPlugComponentKeys )[name.count( "." )]
+		level = name.count( "." )
+		keys = ( nodeKeys, parametersPlugKeys, parameterPlugKeys, parameterPlugComponentKeys )[level]
 		keys.update( metadata.keys() )
 
 	for key in nodeKeys :
@@ -360,6 +362,7 @@ def __plugMetadata( plug, name ) :
 		Gaffer.Metadata.registerValue( nodeType, "parameters", key, functools.partial( __plugMetadata, name = key ) )
 
 	for key in parameterPlugKeys :
+
 		Gaffer.Metadata.registerValue( nodeType, "parameters.*", key, functools.partial( __plugMetadata, name = key ) )
 
 	for key in parameterPlugComponentKeys :