Merge pull request #4016 from Autodesk/gamaj/MAYA-136105/optimize_materialx_via_lobe_pruning

Lobe pruning shader optimization

Author: seando-adsk

lib/mayaUsd/render/MaterialXGenOgsXml/CMakeLists.txt

@@ -8,8 +8,10 @@ target_sources(${PROJECT_NAME}
         OgsFragment.cpp
         OgsXmlGenerator.cpp
         ShaderGenUtil.cpp
+        LobePruner.cpp
         Nodes/SurfaceNodeMaya.cpp
         Nodes/TexcoordNodeMaya.cpp
+        Nodes/MayaDarkClosureNode.cpp
         PugiXML/pugixml.cpp
 )
 
@@ -55,6 +57,7 @@ set(HEADERS
     OgsFragment.h
     OgsXmlGenerator.h
     ShaderGenUtil.h
+    LobePruner.h
 )
 
 # -----------------------------------------------------------------------------

lib/mayaUsd/render/MaterialXGenOgsXml/GlslFragmentGenerator.cpp

@@ -5,6 +5,7 @@
 
 #include "GlslFragmentGenerator.h"
 
+#include "Nodes/MayaDarkClosureNode.h"
 #include "Nodes/MayaShaderGraph.h"
 #include "Nodes/SurfaceNodeMaya.h"
 #include "Nodes/TexcoordNodeMaya.h"
@@ -24,6 +25,7 @@
 
 #include <mayaUsd/render/MaterialXGenOgsXml/CombinedMaterialXVersion.h>
 #include <mayaUsd/render/MaterialXGenOgsXml/GlslOcioNodeImpl.h>
+#include <mayaUsd/render/MaterialXGenOgsXml/LobePruner.h>
 #include <mayaUsd/render/MaterialXGenOgsXml/OgsXmlGenerator.h>
 
 #include <MaterialXGenGlsl/GlslShaderGenerator.h>
@@ -208,6 +210,13 @@ GlslFragmentGenerator::GlslFragmentGenerator()
             "IM_texcoord_vector3_" + GlslShaderGenerator::TARGET, TexcoordNodeGlslMaya::create);
     }
 
+    registerImplementation(
+        MaterialXMaya::ShaderGenUtil::LobePruner::getDarkBaseImplementationName(),
+        MayaDarkClosureNode::create);
+    registerImplementation(
+        MaterialXMaya::ShaderGenUtil::LobePruner::getDarkLayerImplementationName(),
+        MayaDarkClosureNode::create);
+
     // The MaterialX transform node will crash if one of the "space" inputs is empty. This will be
     // fixed in 1.38.9. In the meantime we use patched nodes to replace those previously added in
     // the base class.