Overlays: Support for wireframes and edit mode overlay on xray objects

1 files changed, 52 insertions, 3 deletions

diff --git a/source/blender/draw/modes/edit_mesh_mode.c b/source/blender/draw/modes/edit_mesh_mode.c
index 67a424b8081..98fd1b249be 100644
--- a/source/blender/draw/modes/edit_mesh_mode.c
+++ b/source/blender/draw/modes/edit_mesh_mode.c
@@ -53,7 +53,7 @@ extern char datatoc_edit_mesh_overlay_geom_edge_glsl[];
 extern char datatoc_edit_mesh_overlay_loosevert_vert_glsl[];
 extern char datatoc_edit_mesh_overlay_facedot_frag_glsl[];
 extern char datatoc_edit_mesh_overlay_facedot_vert_glsl[];
-extern char datatoc_edit_mesh_overlay_mix_vert_glsl[];
+extern char datatoc_edit_mesh_overlay_ghost_clear_vert_glsl[];
 extern char datatoc_edit_mesh_overlay_mix_frag_glsl[];
 extern char datatoc_edit_mesh_overlay_facefill_vert_glsl[];
 extern char datatoc_edit_mesh_overlay_facefill_frag_glsl[];
@@ -67,6 +67,7 @@ extern char datatoc_gpu_shader_uniform_color_frag_glsl[];
 typedef struct EDIT_MESH_PassList {
 	struct DRWPass *vcolor_faces;
 	struct DRWPass *depth_hidden_wire;
+	struct DRWPass *ghost_clear_depth;
 	struct DRWPass *edit_face_overlay;
 	struct DRWPass *edit_face_occluded;
 	struct DRWPass *mix_occlude;
@@ -76,6 +77,7 @@ typedef struct EDIT_MESH_PassList {
 
 typedef struct EDIT_MESH_FramebufferList {
 	struct GPUFrameBuffer *occlude_wire_fb;
+	struct GPUFrameBuffer *ghost_wire_fb;
 } EDIT_MESH_FramebufferList;
 
 typedef struct EDIT_MESH_StorageList {
@@ -109,6 +111,7 @@ static struct {
 	GPUShader *normals_loop_sh;
 	GPUShader *normals_sh;
 	GPUShader *depth_sh;
+	GPUShader *ghost_clear_depth_sh;
 	/* temp buffer texture */
 	struct GPUTexture *occlude_wire_depth_tx;
 	struct GPUTexture *occlude_wire_color_tx;
@@ -134,6 +137,8 @@ typedef struct EDIT_MESH_PrivateData {
 	DRWShadingGroup *facedot_occluded_shgrp;
 	DRWShadingGroup *facefill_occluded_shgrp;
 
+	int ghost_ob;
+	int edit_ob;
 	bool do_zbufclip;
 } EDIT_MESH_PrivateData; /* Transient data */
 
@@ -300,6 +305,10 @@ static void EDIT_MESH_engine_init(void *vedata)
 	if (!e_data.depth_sh) {
 		e_data.depth_sh = DRW_shader_create_3D_depth_only();
 	}
+	if (!e_data.ghost_clear_depth_sh) {
+		e_data.ghost_clear_depth_sh = DRW_shader_create(datatoc_edit_mesh_overlay_ghost_clear_vert_glsl,
+		                                                NULL, NULL, NULL);
+	}
 }
 
 static DRWPass *edit_mesh_create_overlay_pass(
@@ -362,7 +371,8 @@ static void EDIT_MESH_cache_init(void *vedata)
 		/* Alloc transient pointers */
 		stl->g_data = MEM_mallocN(sizeof(*stl->g_data), __func__);
 	}
-
+	stl->g_data->ghost_ob = 0;
+	stl->g_data->edit_ob = 0;
 
 	const bool xray_enabled = ((draw_ctx->v3d->shading.flag & V3D_SHADING_XRAY) != 0) &&
 	                           (draw_ctx->v3d->shading.type < OB_MATERIAL);
@@ -392,6 +402,17 @@ static void EDIT_MESH_cache_init(void *vedata)
 	}
 
 	{
+		/* Depth clearing for ghosting. */
+		psl->ghost_clear_depth = DRW_pass_create(
+		        "Ghost Depth Clear",
+		        DRW_STATE_WRITE_DEPTH | DRW_STATE_DEPTH_ALWAYS | DRW_STATE_STENCIL_NEQUAL);
+
+		DRWShadingGroup *shgrp = DRW_shgroup_create(e_data.ghost_clear_depth_sh, psl->ghost_clear_depth);
+		DRW_shgroup_stencil_mask(shgrp, 0x00);
+		DRW_shgroup_call_add(shgrp, DRW_cache_fullscreen_quad_get(), NULL);
+	}
+
+	{
 		/* Normals */
 		psl->normals = DRW_pass_create(
 		        "Edit Mesh Normals Pass",
@@ -412,7 +433,7 @@ static void EDIT_MESH_cache_init(void *vedata)
 
 	if (!stl->g_data->do_zbufclip) {
 		psl->edit_face_overlay = edit_mesh_create_overlay_pass(
-		        &face_mod, DRW_STATE_DEPTH_LESS_EQUAL | DRW_STATE_WRITE_DEPTH | DRW_STATE_BLEND,
+		        &face_mod, DRW_STATE_DEPTH_LESS_EQUAL | DRW_STATE_BLEND,
 		        &stl->g_data->face_overlay_shgrp, &stl->g_data->ledges_overlay_shgrp,
 		        &stl->g_data->lverts_overlay_shgrp, &stl->g_data->facedot_overlay_shgrp);
 	}
@@ -538,6 +559,9 @@ static void EDIT_MESH_cache_populate(void *vedata, Object *ob)
 				        stl->g_data->lverts_overlay_shgrp, NULL, NULL);
 			}
 
+			stl->g_data->ghost_ob += (ob->dtx & OB_DRAWXRAY) ? 1 : 0;
+			stl->g_data->edit_ob += 1;
+
 			/* 3D text overlay */
 			if (me->drawflag & (ME_DRAWEXTRA_EDGELEN |
 			                    ME_DRAWEXTRA_FACEAREA |
@@ -560,6 +584,7 @@ static void EDIT_MESH_draw_scene(void *vedata)
 	EDIT_MESH_StorageList *stl = ((EDIT_MESH_Data *)vedata)->stl;
 	EDIT_MESH_FramebufferList *fbl = ((EDIT_MESH_Data *)vedata)->fbl;
 	DefaultFramebufferList *dfbl = DRW_viewport_framebuffer_list_get();
+	DefaultTextureList *dtxl = DRW_viewport_texture_list_get();
 
 	DRW_draw_pass(psl->vcolor_faces);
 
@@ -582,6 +607,30 @@ static void EDIT_MESH_draw_scene(void *vedata)
 	}
 	else {
 		DRW_draw_pass(psl->normals);
+
+		const DRWContextState *draw_ctx = DRW_context_state_get();
+		View3D *v3d = draw_ctx->v3d;
+
+		if (v3d->shading.type == OB_SOLID && (v3d->shading.flag & V3D_SHADING_XRAY) == 0) {
+			if (stl->g_data->ghost_ob == 1 && stl->g_data->edit_ob == 1) {
+				/* In the case of single ghost object edit (common case for retopology):
+				 * we duplicate the depht+stencil buffer and clear all depth to 1.0f where
+				 * the stencil buffer is no 0x00. */
+				const float *viewport_size = DRW_viewport_size_get();
+				const int size[2] = {(int)viewport_size[0], (int)viewport_size[1]};
+				struct GPUTexture *ghost_depth_tx = DRW_texture_pool_query_2D(size[0], size[1], GPU_DEPTH24_STENCIL8, &draw_engine_edit_mesh_type);
+				GPU_framebuffer_ensure_config(&fbl->ghost_wire_fb, {
+					GPU_ATTACHMENT_TEXTURE(ghost_depth_tx),
+					GPU_ATTACHMENT_TEXTURE(dtxl->color),
+				});
+
+				GPU_framebuffer_blit(dfbl->depth_only_fb, 0, fbl->ghost_wire_fb, 0, GPU_DEPTH_BIT | GPU_STENCIL_BIT);
+				GPU_framebuffer_bind(fbl->ghost_wire_fb);
+
+				DRW_draw_pass(psl->ghost_clear_depth);
+			}
+		}
+
 		DRW_draw_pass(psl->edit_face_overlay);
 	}
 }