Object Mode: Use same empty (arrow) drawing as the bone axes display

4 files changed, 21 insertions, 88 deletions

diff --git a/source/blender/draw/intern/draw_cache.c b/source/blender/draw/intern/draw_cache.c
index c3fa9f5c1aa..d5d5972f153 100644
--- a/source/blender/draw/intern/draw_cache.c
+++ b/source/blender/draw/intern/draw_cache.c
@@ -985,77 +985,6 @@ GPUBatch *DRW_cache_empty_capsule_cap_get(void)
 #undef NSEGMENTS
 }
 
-GPUBatch *DRW_cache_arrows_get(void)
-{
-	if (!SHC.drw_arrows) {
-		GPUVertBuf *vbo = fill_arrows_vbo(1.0f);
-
-		SHC.drw_arrows = GPU_batch_create_ex(GPU_PRIM_LINES, vbo, NULL, GPU_BATCH_OWNS_VBO);
-	}
-	return SHC.drw_arrows;
-}
-
-GPUBatch *DRW_cache_axis_names_get(void)
-{
-	if (!SHC.drw_axis_names) {
-		const float size = 0.1f;
-		float v1[3], v2[3];
-
-		/* Position Only 3D format */
-		static GPUVertFormat format = { 0 };
-		static struct { uint pos; } attr_id;
-		if (format.attr_len == 0) {
-			/* Using 3rd component as axis indicator */
-			attr_id.pos = GPU_vertformat_attr_add(&format, "pos", GPU_COMP_F32, 3, GPU_FETCH_FLOAT);
-		}
-
-		/* Line */
-		GPUVertBuf *vbo = GPU_vertbuf_create_with_format(&format);
-		GPU_vertbuf_data_alloc(vbo, 14);
-
-		/* X */
-		copy_v3_fl3(v1, -size,  size, 0.0f);
-		copy_v3_fl3(v2,  size, -size, 0.0f);
-		GPU_vertbuf_attr_set(vbo, attr_id.pos, 0, v1);
-		GPU_vertbuf_attr_set(vbo, attr_id.pos, 1, v2);
-
-		copy_v3_fl3(v1,  size,  size, 0.0f);
-		copy_v3_fl3(v2, -size, -size, 0.0f);
-		GPU_vertbuf_attr_set(vbo, attr_id.pos, 2, v1);
-		GPU_vertbuf_attr_set(vbo, attr_id.pos, 3, v2);
-
-		/* Y */
-		copy_v3_fl3(v1, -size + 0.25f * size,  size, 1.0f);
-		copy_v3_fl3(v2,  0.0f,  0.0f, 1.0f);
-		GPU_vertbuf_attr_set(vbo, attr_id.pos, 4, v1);
-		GPU_vertbuf_attr_set(vbo, attr_id.pos, 5, v2);
-
-		copy_v3_fl3(v1,  size - 0.25f * size,  size, 1.0f);
-		copy_v3_fl3(v2, -size + 0.25f * size, -size, 1.0f);
-		GPU_vertbuf_attr_set(vbo, attr_id.pos, 6, v1);
-		GPU_vertbuf_attr_set(vbo, attr_id.pos, 7, v2);
-
-		/* Z */
-		copy_v3_fl3(v1, -size,  size, 2.0f);
-		copy_v3_fl3(v2,  size,  size, 2.0f);
-		GPU_vertbuf_attr_set(vbo, attr_id.pos, 8, v1);
-		GPU_vertbuf_attr_set(vbo, attr_id.pos, 9, v2);
-
-		copy_v3_fl3(v1,  size,  size, 2.0f);
-		copy_v3_fl3(v2, -size, -size, 2.0f);
-		GPU_vertbuf_attr_set(vbo, attr_id.pos, 10, v1);
-		GPU_vertbuf_attr_set(vbo, attr_id.pos, 11, v2);
-
-		copy_v3_fl3(v1, -size, -size, 2.0f);
-		copy_v3_fl3(v2,  size, -size, 2.0f);
-		GPU_vertbuf_attr_set(vbo, attr_id.pos, 12, v1);
-		GPU_vertbuf_attr_set(vbo, attr_id.pos, 13, v2);
-
-		SHC.drw_axis_names = GPU_batch_create_ex(GPU_PRIM_LINES, vbo, NULL, GPU_BATCH_OWNS_VBO);
-	}
-	return SHC.drw_axis_names;
-}
-
 GPUBatch *DRW_cache_image_plane_get(void)
 {
 	if (!SHC.drw_image_plane) {
diff --git a/source/blender/draw/intern/draw_cache.h b/source/blender/draw/intern/draw_cache.h
index 7d0996b3059..f5f4527ca13 100644
--- a/source/blender/draw/intern/draw_cache.h
+++ b/source/blender/draw/intern/draw_cache.h
@@ -70,8 +70,6 @@ struct GPUBatch *DRW_cache_empty_cylinder_get(void);
 struct GPUBatch *DRW_cache_empty_cone_get(void);
 struct GPUBatch *DRW_cache_empty_capsule_cap_get(void);
 struct GPUBatch *DRW_cache_empty_capsule_body_get(void);
-struct GPUBatch *DRW_cache_arrows_get(void);
-struct GPUBatch *DRW_cache_axis_names_get(void);
 struct GPUBatch *DRW_cache_image_plane_get(void);
 struct GPUBatch *DRW_cache_image_plane_wire_get(void);
 
diff --git a/source/blender/draw/intern/draw_common.c b/source/blender/draw/intern/draw_common.c
index ca4668af74f..217ddd3154f 100644
--- a/source/blender/draw/intern/draw_common.c
+++ b/source/blender/draw/intern/draw_common.c
@@ -186,6 +186,7 @@ extern char datatoc_gpu_shader_3D_smooth_color_frag_glsl[];
 extern char datatoc_gpu_shader_point_varying_color_frag_glsl[];
 
 extern char datatoc_object_mball_handles_vert_glsl[];
+extern char datatoc_object_empty_axes_vert_glsl[];
 
 static struct {
 	struct GPUShader *shape_outline;
@@ -203,6 +204,7 @@ static struct {
 
 	struct GPUShader *volume_velocity_needle_sh;
 	struct GPUShader *volume_velocity_sh;
+	struct GPUShader *empty_axes_sh;
 
 	struct GPUShader *mball_handles;
 } g_shaders = {NULL};
@@ -368,40 +370,43 @@ DRWShadingGroup *shgroup_instance_screen_aligned(DRWPass *pass, struct GPUBatch
 	return grp;
 }
 
-DRWShadingGroup *shgroup_instance_axis_names(DRWPass *pass, struct GPUBatch *geom)
+DRWShadingGroup *shgroup_instance_scaled(DRWPass *pass, struct GPUBatch *geom)
 {
-	GPUShader *sh = GPU_shader_get_builtin_shader(GPU_SHADER_3D_INSTANCE_SCREEN_ALIGNED_AXIS);
+	GPUShader *sh_inst = GPU_shader_get_builtin_shader(GPU_SHADER_INSTANCE_VARIYING_COLOR_VARIYING_SCALE);
 
-	DRW_shgroup_instance_format(g_formats.instance_screen_aligned, {
+	DRW_shgroup_instance_format(g_formats.instance_scaled, {
 		{"color",               DRW_ATTRIB_FLOAT, 3},
-		{"size",                DRW_ATTRIB_FLOAT, 1},
+		{"size",                DRW_ATTRIB_FLOAT, 3},
 		{"InstanceModelMatrix", DRW_ATTRIB_FLOAT, 16}
 	});
 
-	DRWShadingGroup *grp = DRW_shgroup_instance_create(sh, pass, geom, g_formats.instance_screen_aligned);
-	DRW_shgroup_uniform_vec3(grp, "screen_vecs[0]", DRW_viewport_screenvecs_get(), 2);
+	DRWShadingGroup *grp = DRW_shgroup_instance_create(sh_inst, pass, geom, g_formats.instance_scaled);
 
 	return grp;
 }
 
-DRWShadingGroup *shgroup_instance_scaled(DRWPass *pass, struct GPUBatch *geom)
+DRWShadingGroup *shgroup_instance(DRWPass *pass, struct GPUBatch *geom)
 {
-	GPUShader *sh_inst = GPU_shader_get_builtin_shader(GPU_SHADER_INSTANCE_VARIYING_COLOR_VARIYING_SCALE);
+	GPUShader *sh_inst = GPU_shader_get_builtin_shader(GPU_SHADER_INSTANCE_VARIYING_COLOR_VARIYING_SIZE);
 
-	DRW_shgroup_instance_format(g_formats.instance_scaled, {
+	DRW_shgroup_instance_format(g_formats.instance_sized, {
 		{"color",               DRW_ATTRIB_FLOAT, 3},
-		{"size",                DRW_ATTRIB_FLOAT, 3},
+		{"size",                DRW_ATTRIB_FLOAT, 1},
 		{"InstanceModelMatrix", DRW_ATTRIB_FLOAT, 16}
 	});
 
-	DRWShadingGroup *grp = DRW_shgroup_instance_create(sh_inst, pass, geom, g_formats.instance_scaled);
+	DRWShadingGroup *grp = DRW_shgroup_instance_create(sh_inst, pass, geom, g_formats.instance_sized);
 
 	return grp;
 }
 
-DRWShadingGroup *shgroup_instance(DRWPass *pass, struct GPUBatch *geom)
+DRWShadingGroup *shgroup_instance_empty_axes(DRWPass *pass, struct GPUBatch *geom)
 {
-	GPUShader *sh_inst = GPU_shader_get_builtin_shader(GPU_SHADER_INSTANCE_VARIYING_COLOR_VARIYING_SIZE);
+	if (g_shaders.empty_axes_sh == NULL) {
+		g_shaders.empty_axes_sh = DRW_shader_create(
+		        datatoc_object_empty_axes_vert_glsl, NULL,
+	            datatoc_gpu_shader_flat_color_frag_glsl, NULL);
+	}
 
 	DRW_shgroup_instance_format(g_formats.instance_sized, {
 		{"color",               DRW_ATTRIB_FLOAT, 3},
@@ -409,7 +414,8 @@ DRWShadingGroup *shgroup_instance(DRWPass *pass, struct GPUBatch *geom)
 		{"InstanceModelMatrix", DRW_ATTRIB_FLOAT, 16}
 	});
 
-	DRWShadingGroup *grp = DRW_shgroup_instance_create(sh_inst, pass, geom, g_formats.instance_sized);
+	DRWShadingGroup *grp = DRW_shgroup_instance_create(g_shaders.empty_axes_sh, pass, geom, g_formats.instance_sized);
+	DRW_shgroup_uniform_vec3(grp, "screenVecs[0]", DRW_viewport_screenvecs_get(), 2);
 
 	return grp;
 }
diff --git a/source/blender/draw/intern/draw_common.h b/source/blender/draw/intern/draw_common.h
index 0ad1402f29e..05c3da9f96c 100644
--- a/source/blender/draw/intern/draw_common.h
+++ b/source/blender/draw/intern/draw_common.h
@@ -116,7 +116,7 @@ struct DRWShadingGroup *shgroup_instance_screenspace(struct DRWPass *pass, struc
 struct DRWShadingGroup *shgroup_instance_solid(struct DRWPass *pass, struct GPUBatch *geom);
 struct DRWShadingGroup *shgroup_instance_wire(struct DRWPass *pass, struct GPUBatch *geom);
 struct DRWShadingGroup *shgroup_instance_screen_aligned(struct DRWPass *pass, struct GPUBatch *geom);
-struct DRWShadingGroup *shgroup_instance_axis_names(struct DRWPass *pass, struct GPUBatch *geom);
+struct DRWShadingGroup *shgroup_instance_empty_axes(struct DRWPass *pass, struct GPUBatch *geom);
 struct DRWShadingGroup *shgroup_instance_image_plane(struct DRWPass *pass, struct GPUBatch *geom);
 struct DRWShadingGroup *shgroup_instance_scaled(struct DRWPass *pass, struct GPUBatch *geom);
 struct DRWShadingGroup *shgroup_instance(struct DRWPass *pass, struct GPUBatch *geom);