GPUMaterial: Add a domain property.

This let us know efficiently if a material has a dedicated nodetree for each of it's output node input.

Only works for Eevee at this moment.

2 files changed, 32 insertions, 0 deletions

diff --git a/source/blender/gpu/GPU_material.h b/source/blender/gpu/GPU_material.h
index 7bc256a70fa..1508941f706 100644
--- a/source/blender/gpu/GPU_material.h
+++ b/source/blender/gpu/GPU_material.h
@@ -267,6 +267,8 @@ void GPU_material_vertex_attributes(GPUMaterial *material,
 bool GPU_material_do_color_management(GPUMaterial *mat);
 bool GPU_material_use_new_shading_nodes(GPUMaterial *mat);
 bool GPU_material_use_world_space_shading(GPUMaterial *mat);
+bool GPU_material_use_domain_surface(GPUMaterial *mat);
+bool GPU_material_use_domain_volume(GPUMaterial *mat);
 
 /* Exported shading */
 
diff --git a/source/blender/gpu/intern/gpu_material.c b/source/blender/gpu/intern/gpu_material.c
index 55157246c19..312cd7bc20a 100644
--- a/source/blender/gpu/intern/gpu_material.c
+++ b/source/blender/gpu/intern/gpu_material.c
@@ -135,9 +135,20 @@ struct GPUMaterial {
 	bool bound;
 
 	bool is_opensubdiv;
+
+	/* XXX: Should be in Material. But it depends on the output node
+	 * used and since the output selection is difference for GPUMaterial...
+	 */
+	int domain;
+
 	GPUUniformBuffer *ubo; /* UBOs for shader uniforms. */
 };
 
+enum {
+	GPU_DOMAIN_SURFACE    = (1 << 0),
+	GPU_DOMAIN_VOLUME     = (1 << 1)
+};
+
 /* Forward declaration so shade_light_textures() can use this, while still keeping the code somewhat organized */
 static void texture_rgb_blend(
         GPUMaterial *mat, GPUNodeLink *tex, GPUNodeLink *out, GPUNodeLink *fact, GPUNodeLink *facg,
@@ -509,6 +520,16 @@ bool GPU_material_use_world_space_shading(GPUMaterial *mat)
 	return BKE_scene_use_world_space_shading(mat->scene);
 }
 
+bool GPU_material_use_domain_surface(GPUMaterial *mat)
+{
+	return (mat->domain & GPU_DOMAIN_SURFACE);
+}
+
+bool GPU_material_use_domain_volume(GPUMaterial *mat)
+{
+	return (mat->domain & GPU_DOMAIN_VOLUME);
+}
+
 static GPUNodeLink *lamp_get_visibility(GPUMaterial *mat, GPULamp *lamp, GPUNodeLink **lv, GPUNodeLink **dist)
 {
 	GPUNodeLink *visifac;
@@ -2145,6 +2166,7 @@ GPUMaterial *GPU_material_from_nodetree(
 	GPUMaterial *mat;
 	GPUNodeLink *outlink;
 	LinkData *link;
+	bool has_volume_output, has_surface_output;
 
 	/* Caller must re-use materials. */
 	BLI_assert(GPU_material_from_nodetree_find(gpumaterials, engine_type, options) == NULL);
@@ -2156,6 +2178,14 @@ GPUMaterial *GPU_material_from_nodetree(
 	mat->options = options;
 
 	ntreeGPUMaterialNodes(ntree, mat, NODE_NEW_SHADING | NODE_NEWER_SHADING);
+	ntreeGPUMaterialDomain(ntree, &has_surface_output, &has_volume_output);
+
+	if (has_surface_output) {
+		mat->domain |= GPU_DOMAIN_SURFACE;
+	}
+	if (has_volume_output) {
+		mat->domain |= GPU_DOMAIN_VOLUME;
+	}
 
 	/* Let Draw manager finish the construction. */
 	if (mat->outlink) {