Workbench: Smoke: Add support for Color Mappping for smoke debugging

1 files changed, 51 insertions, 30 deletions

diff --git a/source/blender/draw/engines/workbench/workbench_volume.c b/source/blender/draw/engines/workbench/workbench_volume.c
index ba1d3d9ff5c..b0c08a2dc28 100644
--- a/source/blender/draw/engines/workbench/workbench_volume.c
+++ b/source/blender/draw/engines/workbench/workbench_volume.c
@@ -35,7 +35,9 @@
 
 static struct {
 	struct GPUShader *volume_sh;
+	struct GPUShader *volume_coba_sh;
 	struct GPUShader *volume_slice_sh;
+	struct GPUShader *volume_slice_coba_sh;
 	struct GPUTexture *dummy_tex;
 	struct GPUTexture *dummy_coba_tex;
 } e_data = {NULL};
@@ -49,9 +51,19 @@ void workbench_volume_engine_init(void)
 		e_data.volume_sh = DRW_shader_create(
 		        datatoc_workbench_volume_vert_glsl, NULL,
 		        datatoc_workbench_volume_frag_glsl, NULL);
+		e_data.volume_coba_sh = DRW_shader_create(
+		        datatoc_workbench_volume_vert_glsl, NULL,
+		        datatoc_workbench_volume_frag_glsl,
+		        "#define USE_COBA\n");
 		e_data.volume_slice_sh = DRW_shader_create(
 		        datatoc_workbench_volume_vert_glsl, NULL,
-		        datatoc_workbench_volume_frag_glsl, "#define VOLUME_SLICE");
+		        datatoc_workbench_volume_frag_glsl,
+		        "#define VOLUME_SLICE\n");
+		e_data.volume_slice_coba_sh = DRW_shader_create(
+		        datatoc_workbench_volume_vert_glsl, NULL,
+		        datatoc_workbench_volume_frag_glsl,
+		        "#define VOLUME_SLICE\n"
+		        "#define USE_COBA\n");
 
 		float pixel[4] = {0.0f, 0.0f, 0.0f, 0.0f};
 		e_data.dummy_tex = GPU_texture_create_3D(1, 1, 1, GPU_RGBA8, pixel, NULL);
@@ -62,7 +74,9 @@ void workbench_volume_engine_init(void)
 void workbench_volume_engine_free(void)
 {
 	DRW_SHADER_FREE_SAFE(e_data.volume_sh);
+	DRW_SHADER_FREE_SAFE(e_data.volume_coba_sh);
 	DRW_SHADER_FREE_SAFE(e_data.volume_slice_sh);
+	DRW_SHADER_FREE_SAFE(e_data.volume_slice_coba_sh);
 	DRW_TEXTURE_FREE_SAFE(e_data.dummy_tex);
 	DRW_TEXTURE_FREE_SAFE(e_data.dummy_coba_tex);
 }
@@ -78,6 +92,7 @@ void workbench_volume_cache_populate(WORKBENCH_Data *vedata, Scene *scene, Objec
 	SmokeDomainSettings *sds = smd->domain;
 	WORKBENCH_PrivateData *wpd = vedata->stl->g_data;
 	DefaultTextureList *dtxl = DRW_viewport_texture_list_get();
+	DRWShadingGroup *grp = NULL;
 
 	/* Don't show smoke before simulation starts, this could be made an option in the future. */
 	if (!sds->fluid || CFRA < sds->point_cache[0]->startframe) {
@@ -85,21 +100,26 @@ void workbench_volume_cache_populate(WORKBENCH_Data *vedata, Scene *scene, Objec
 	}
 
 	wpd->volumes_do = true;
-
-	if (!sds->wt || !(sds->viewsettings & MOD_SMOKE_VIEW_SHOWBIG)) {
+	if (sds->use_coba) {
+		GPU_create_smoke_coba_field(smd);
+	}
+	else if (!sds->wt || !(sds->viewsettings & MOD_SMOKE_VIEW_SHOWBIG)) {
 		GPU_create_smoke(smd, 0);
 	}
 	else if (sds->wt && (sds->viewsettings & MOD_SMOKE_VIEW_SHOWBIG)) {
 		GPU_create_smoke(smd, 1);
 	}
 
-	if (sds->tex == NULL) {
+	if ((!sds->use_coba && sds->tex == NULL) ||
+	    (sds->use_coba && sds->tex_field == NULL))
+	{
 		return;
 	}
 
-	if (sds->slice_method == MOD_SMOKE_SLICE_AXIS_ALIGNED &&
-	    sds->axis_slice_method == AXIS_SLICE_SINGLE)
-	{
+	const bool use_slice = (sds->slice_method == MOD_SMOKE_SLICE_AXIS_ALIGNED &&
+	                        sds->axis_slice_method == AXIS_SLICE_SINGLE);
+
+	if (use_slice) {
 		float invviewmat[4][4];
 		DRW_viewport_matrix_get(invviewmat, DRW_MAT_VIEWINV);
 
@@ -107,45 +127,46 @@ void workbench_volume_cache_populate(WORKBENCH_Data *vedata, Scene *scene, Objec
 		                  ? axis_dominant_v3_single(invviewmat[2])
 		                  : sds->slice_axis - 1;
 
-		DRWShadingGroup *grp = DRW_shgroup_create(e_data.volume_slice_sh, vedata->psl->volume_pass);
-		DRW_shgroup_uniform_texture(grp, "densityTexture", sds->tex);
-		DRW_shgroup_uniform_texture(grp, "shadowTexture", sds->tex_shadow);
-		DRW_shgroup_uniform_texture(grp, "flameTexture", (sds->tex_flame) ? sds->tex_flame : e_data.dummy_tex);
-		DRW_shgroup_uniform_texture(grp, "flameColorTexture", (sds->tex_flame) ? sds->tex_flame_coba : e_data.dummy_coba_tex);
-		DRW_shgroup_uniform_texture_ref(grp, "depthBuffer", &dtxl->depth);
-		DRW_shgroup_uniform_float_copy(grp, "densityScale", 10.0f * sds->display_thickness);
+		GPUShader *sh = (sds->use_coba) ? e_data.volume_slice_coba_sh : e_data.volume_slice_sh;
+		grp = DRW_shgroup_create(sh, vedata->psl->volume_pass);
 		DRW_shgroup_uniform_float_copy(grp, "slicePosition", sds->slice_depth);
 		DRW_shgroup_uniform_int_copy(grp, "sliceAxis", axis);
-		DRW_shgroup_state_disable(grp, DRW_STATE_CULL_FRONT);
-		BLI_addtail(&wpd->smoke_domains, BLI_genericNodeN(smd));
-
-		/* TODO COBA Rendering */
-
-		DRW_shgroup_call_object_add(grp, DRW_cache_quad_get(), ob);
 	}
 	else {
 		int max_slices = max_iii(sds->res[0], sds->res[1], sds->res[2]) * sds->slice_per_voxel;
 
-		DRWShadingGroup *grp = DRW_shgroup_create(e_data.volume_sh, vedata->psl->volume_pass);
+		GPUShader *sh = (sds->use_coba) ? e_data.volume_coba_sh : e_data.volume_sh;
+		grp = DRW_shgroup_create(sh, vedata->psl->volume_pass);
 		DRW_shgroup_uniform_vec4(grp, "viewvecs[0]", (float *)wpd->viewvecs, 3);
-		DRW_shgroup_uniform_texture_ref(grp, "depthBuffer", &dtxl->depth);
-		DRW_shgroup_uniform_texture(grp, "densityTexture", sds->tex);
-		DRW_shgroup_uniform_texture(grp, "shadowTexture", sds->tex_shadow);
-		DRW_shgroup_uniform_texture(grp, "flameTexture", (sds->tex_flame) ? sds->tex_flame : e_data.dummy_tex);
-		DRW_shgroup_uniform_texture(grp, "flameColorTexture", (sds->tex_flame) ? sds->tex_flame_coba : e_data.dummy_coba_tex);
-		DRW_shgroup_uniform_float_copy(grp, "densityScale", 10.0f * sds->display_thickness);
 		DRW_shgroup_uniform_int_copy(grp, "samplesLen", max_slices);
 		/* TODO FIXME : This step size is in object space but the ray itself
 		 * is NOT unit length in object space so the required number of subdivisions
 		 * is tricky to get. */
 		DRW_shgroup_uniform_float_copy(grp, "stepLength", 8.0f / max_slices);
-		DRW_shgroup_state_enable(grp, DRW_STATE_CULL_FRONT);
-		BLI_addtail(&wpd->smoke_domains, BLI_genericNodeN(smd));
+	}
 
-		/* TODO COBA Rendering */
+	if (sds->use_coba) {
+		DRW_shgroup_uniform_texture(grp, "densityTexture", sds->tex_field);
+		DRW_shgroup_uniform_texture(grp, "transferTexture", sds->tex_coba);
+	}
+	else {
+		DRW_shgroup_uniform_texture(grp, "densityTexture", sds->tex);
+		DRW_shgroup_uniform_texture(grp, "shadowTexture", sds->tex_shadow);
+		DRW_shgroup_uniform_texture(grp, "flameTexture", (sds->tex_flame) ? sds->tex_flame : e_data.dummy_tex);
+		DRW_shgroup_uniform_texture(grp, "flameColorTexture", (sds->tex_flame) ? sds->tex_flame_coba : e_data.dummy_coba_tex);
+	}
+	DRW_shgroup_uniform_texture_ref(grp, "depthBuffer", &dtxl->depth);
+	DRW_shgroup_uniform_float_copy(grp, "densityScale", 10.0f * sds->display_thickness);
+	DRW_shgroup_state_disable(grp, DRW_STATE_CULL_FRONT);
 
+	if (use_slice) {
+		DRW_shgroup_call_object_add(grp, DRW_cache_quad_get(), ob);
+	}
+	else {
 		DRW_shgroup_call_object_add(grp, DRW_cache_cube_get(), ob);
 	}
+
+	BLI_addtail(&wpd->smoke_domains, BLI_genericNodeN(smd));
 }
 
 void workbench_volume_smoke_textures_free(WORKBENCH_PrivateData *wpd)