diff options
author | Campbell Barton <ideasman42@gmail.com> | 2019-02-14 05:59:30 +0300 |
---|---|---|
committer | Campbell Barton <ideasman42@gmail.com> | 2019-02-14 07:22:58 +0300 |
commit | caa8e7ff2894ed22ea436bc61b06fb292d844108 (patch) | |
tree | e6f1bcd573118441d163144da3e9e74ec38e091e /source/blender/draw/modes/paint_vertex_mode.c | |
parent | 9478ac701be110e8d61085158bf92e356b001cd4 (diff) |
DRW: de-duplicate vertex & weight paint engines
These were almost exact duplicates, use one engine for drawing both
kinds of vertex color.
Diffstat (limited to 'source/blender/draw/modes/paint_vertex_mode.c')
-rw-r--r-- | source/blender/draw/modes/paint_vertex_mode.c | 152 |
1 files changed, 105 insertions, 47 deletions
diff --git a/source/blender/draw/modes/paint_vertex_mode.c b/source/blender/draw/modes/paint_vertex_mode.c index 51eeacd76c4..e58399c8cb9 100644 --- a/source/blender/draw/modes/paint_vertex_mode.c +++ b/source/blender/draw/modes/paint_vertex_mode.c @@ -26,17 +26,18 @@ #include "draw_common.h" - #include "DNA_mesh_types.h" #include "DNA_view3d_types.h" #include "DEG_depsgraph_query.h" +extern char datatoc_paint_face_vert_glsl[]; +extern char datatoc_paint_weight_vert_glsl[]; +extern char datatoc_paint_weight_frag_glsl[]; extern char datatoc_paint_vertex_vert_glsl[]; extern char datatoc_paint_vertex_frag_glsl[]; extern char datatoc_paint_wire_vert_glsl[]; extern char datatoc_paint_wire_frag_glsl[]; -extern char datatoc_paint_face_vert_glsl[]; extern char datatoc_paint_vert_frag_glsl[]; extern char datatoc_common_globals_lib_glsl[]; @@ -44,8 +45,16 @@ extern char datatoc_gpu_shader_uniform_color_frag_glsl[]; /* *********** LISTS *********** */ +enum { + VERTEX_MODE = 0, + WEIGHT_MODE = 1, +}; +#define MODE_LEN (WEIGHT_MODE + 1) + typedef struct PAINT_VERTEX_PassList { - struct DRWPass *vcolor_faces; + struct { + struct DRWPass *color_faces; + } by_mode[MODE_LEN]; struct DRWPass *wire_overlay; struct DRWPass *wire_select_overlay; struct DRWPass *face_select_overlay; @@ -65,9 +74,11 @@ typedef struct PAINT_VERTEX_Data { } PAINT_VERTEX_Data; typedef struct PAINT_VERTEX_Shaders { - struct GPUShader *vcolor_face; - struct GPUShader *wire_overlay; - struct GPUShader *wire_select_overlay; + struct { + struct GPUShader *color_face; + struct GPUShader *wire_overlay; + struct GPUShader *wire_select_overlay; + } by_mode[MODE_LEN]; struct GPUShader *face_select_overlay; struct GPUShader *vert_select_overlay; } PAINT_VERTEX_Shaders; @@ -76,12 +87,14 @@ typedef struct PAINT_VERTEX_Shaders { static struct { PAINT_VERTEX_Shaders sh_data[GPU_SHADER_CFG_LEN]; -} e_data = {{{NULL}}}; /* Engine data */ +} e_data = {{{{{NULL}}}}}; /* Engine data */ typedef struct PAINT_VERTEX_PrivateData { - DRWShadingGroup *fvcolor_shgrp; - DRWShadingGroup *lwire_shgrp; - DRWShadingGroup *lwire_select_shgrp; + struct { + DRWShadingGroup *color_shgrp; + DRWShadingGroup *lwire_shgrp; + DRWShadingGroup *lwire_select_shgrp; + } by_mode[MODE_LEN]; DRWShadingGroup *face_select_shgrp; DRWShadingGroup *vert_select_shgrp; } PAINT_VERTEX_PrivateData; /* Transient data */ @@ -98,23 +111,18 @@ static void PAINT_VERTEX_engine_init(void *UNUSED(vedata)) } const GPUShaderConfigData *sh_cfg_data = &GPU_shader_cfg_data[draw_ctx->sh_cfg]; - if (!sh_data->vcolor_face) { - - sh_data->vcolor_face = GPU_shader_create_from_arrays({ + if (!sh_data->face_select_overlay) { + sh_data->by_mode[VERTEX_MODE].color_face = GPU_shader_create_from_arrays({ .vert = (const char *[]){sh_cfg_data->lib, datatoc_paint_vertex_vert_glsl, NULL}, .frag = (const char *[]){datatoc_paint_vertex_frag_glsl, NULL}, .defs = (const char *[]){sh_cfg_data->def, NULL}, }); - sh_data->wire_overlay = GPU_shader_create_from_arrays({ - .vert = (const char *[]){sh_cfg_data->lib, datatoc_common_globals_lib_glsl, datatoc_paint_wire_vert_glsl, NULL}, - .frag = (const char *[]){datatoc_paint_wire_frag_glsl, NULL}, - .defs = (const char *[]){sh_cfg_data->def, "#define VERTEX_MODE\n", NULL}, - }); - sh_data->wire_select_overlay = GPU_shader_create_from_arrays({ - .vert = (const char *[]){sh_cfg_data->lib, datatoc_common_globals_lib_glsl, datatoc_paint_wire_vert_glsl, NULL}, - .frag = (const char *[]){datatoc_paint_wire_frag_glsl, NULL}, - .defs = (const char *[]){sh_cfg_data->def, "#define VERTEX_MODE\n" "#define USE_SELECT\n", NULL}, + sh_data->by_mode[WEIGHT_MODE].color_face = GPU_shader_create_from_arrays({ + .vert = (const char *[]){sh_cfg_data->lib, datatoc_common_globals_lib_glsl, datatoc_paint_weight_vert_glsl, NULL}, + .frag = (const char *[]){datatoc_common_globals_lib_glsl, datatoc_paint_weight_frag_glsl, NULL}, + .defs = (const char *[]){sh_cfg_data->def, NULL}, }); + sh_data->face_select_overlay = GPU_shader_create_from_arrays({ .vert = (const char *[]){sh_cfg_data->lib, datatoc_paint_face_vert_glsl, NULL}, .frag = (const char *[]){datatoc_gpu_shader_uniform_color_frag_glsl, NULL}, @@ -125,6 +133,23 @@ static void PAINT_VERTEX_engine_init(void *UNUSED(vedata)) .frag = (const char *[]){datatoc_paint_vert_frag_glsl, NULL}, .defs = (const char *[]){sh_cfg_data->def, "#define USE_SELECT\n", NULL}, }); + + const char *mode_defs[MODE_LEN] = { + "#define VERTEX_MODE\n", + "#define WEIGHT_MODE\n", + }; + for (int i = 0; i < MODE_LEN; i++) { + sh_data->by_mode[i].wire_overlay = GPU_shader_create_from_arrays({ + .vert = (const char *[]){sh_cfg_data->lib, datatoc_common_globals_lib_glsl, datatoc_paint_wire_vert_glsl, NULL}, + .frag = (const char *[]){datatoc_paint_wire_frag_glsl, NULL}, + .defs = (const char *[]){sh_cfg_data->def, mode_defs[i], NULL}, + }); + sh_data->by_mode[i].wire_select_overlay = GPU_shader_create_from_arrays({ + .vert = (const char *[]){sh_cfg_data->lib, datatoc_common_globals_lib_glsl, datatoc_paint_wire_vert_glsl, NULL}, + .frag = (const char *[]){datatoc_paint_wire_frag_glsl, NULL}, + .defs = (const char *[]){sh_cfg_data->def, mode_defs[i], "#define USE_SELECT\n", NULL}, + }); + } } } @@ -142,52 +167,74 @@ static void PAINT_VERTEX_cache_init(void *vedata) stl->g_data = MEM_mallocN(sizeof(*stl->g_data), __func__); } + /* Vertex color pass */ { - /* Create a pass */ DRWPass *pass = DRW_pass_create( "Vert Color Pass", DRW_STATE_WRITE_COLOR | DRW_STATE_DEPTH_EQUAL | DRW_STATE_MULTIPLY); - DRWShadingGroup *shgrp = DRW_shgroup_create(sh_data->vcolor_face, pass); + DRWShadingGroup *shgrp = DRW_shgroup_create(sh_data->by_mode[VERTEX_MODE].color_face, pass); DRW_shgroup_uniform_float_copy(shgrp, "white_factor", 1.0f - v3d->overlay.vertex_paint_mode_opacity); if (rv3d->rflag & RV3D_CLIPPING) { DRW_shgroup_world_clip_planes_from_rv3d(shgrp, rv3d); } - psl->vcolor_faces = pass; - stl->g_data->fvcolor_shgrp = shgrp; + psl->by_mode[VERTEX_MODE].color_faces = pass; + stl->g_data->by_mode[VERTEX_MODE].color_shgrp = shgrp; } + /* Weight color pass */ { DRWPass *pass = DRW_pass_create( - "Wire Pass", - DRW_STATE_WRITE_COLOR | DRW_STATE_WRITE_DEPTH | DRW_STATE_DEPTH_LESS_EQUAL | DRW_STATE_OFFSET_NEGATIVE); - DRWShadingGroup *shgrp = DRW_shgroup_create(sh_data->wire_overlay, pass); + "Weight Pass", + DRW_STATE_WRITE_COLOR | DRW_STATE_DEPTH_EQUAL | DRW_STATE_MULTIPLY); + DRWShadingGroup *shgrp = DRW_shgroup_create(sh_data->by_mode[WEIGHT_MODE].color_face, pass); + DRW_shgroup_uniform_bool_copy(shgrp, "drawContours", (v3d->overlay.wpaint_flag & V3D_OVERLAY_WPAINT_CONTOURS) != 0); + DRW_shgroup_uniform_float(shgrp, "opacity", &v3d->overlay.weight_paint_mode_opacity, 1); + DRW_shgroup_uniform_texture(shgrp, "colorramp", G_draw.weight_ramp); DRW_shgroup_uniform_block(shgrp, "globalsBlock", G_draw.block_ubo); if (rv3d->rflag & RV3D_CLIPPING) { DRW_shgroup_world_clip_planes_from_rv3d(shgrp, rv3d); } + psl->by_mode[WEIGHT_MODE].color_faces = pass; + stl->g_data->by_mode[WEIGHT_MODE].color_shgrp = shgrp; + } + + { + DRWPass *pass = DRW_pass_create( + "Wire Pass", + DRW_STATE_WRITE_COLOR | DRW_STATE_WRITE_DEPTH | DRW_STATE_DEPTH_LESS_EQUAL | DRW_STATE_OFFSET_NEGATIVE); + for (int i = 0; i < MODE_LEN; i++) { + DRWShadingGroup *shgrp = DRW_shgroup_create(sh_data->by_mode[i].wire_overlay, pass); + DRW_shgroup_uniform_block(shgrp, "globalsBlock", G_draw.block_ubo); + if (rv3d->rflag & RV3D_CLIPPING) { + DRW_shgroup_world_clip_planes_from_rv3d(shgrp, rv3d); + } + stl->g_data->by_mode[i].lwire_shgrp = shgrp; + } psl->wire_overlay = pass; - stl->g_data->lwire_shgrp = shgrp; } + { DRWPass *pass = DRW_pass_create( "Wire Mask Pass", DRW_STATE_WRITE_COLOR | DRW_STATE_WRITE_DEPTH | DRW_STATE_DEPTH_LESS_EQUAL | DRW_STATE_OFFSET_NEGATIVE); - DRWShadingGroup *shgrp = DRW_shgroup_create(sh_data->wire_select_overlay, pass); - DRW_shgroup_uniform_block(shgrp, "globalsBlock", G_draw.block_ubo); - if (rv3d->rflag & RV3D_CLIPPING) { - DRW_shgroup_world_clip_planes_from_rv3d(shgrp, rv3d); + for (int i = 0; i < MODE_LEN; i++) { + DRWShadingGroup *shgrp = DRW_shgroup_create(sh_data->by_mode[i].wire_select_overlay, pass); + DRW_shgroup_uniform_block(shgrp, "globalsBlock", G_draw.block_ubo); + if (rv3d->rflag & RV3D_CLIPPING) { + DRW_shgroup_world_clip_planes_from_rv3d(shgrp, rv3d); + } + stl->g_data->by_mode[i].lwire_select_shgrp = shgrp; } psl->wire_select_overlay = pass; - stl->g_data->lwire_select_shgrp = shgrp; } { + static float col[4] = {1.0f, 1.0f, 1.0f, 0.2f}; DRWPass *pass = DRW_pass_create( "Face Mask Pass", DRW_STATE_WRITE_COLOR | DRW_STATE_WRITE_DEPTH | DRW_STATE_DEPTH_LESS_EQUAL | DRW_STATE_BLEND); DRWShadingGroup *shgrp = DRW_shgroup_create(sh_data->face_select_overlay, pass); - static float col[4] = {1.0f, 1.0f, 1.0f, 0.2f}; DRW_shgroup_uniform_vec4(shgrp, "color", col, 1); if (rv3d->rflag & RV3D_CLIPPING) { DRW_shgroup_world_clip_planes_from_rv3d(shgrp, rv3d); @@ -217,25 +264,35 @@ static void PAINT_VERTEX_cache_populate(void *vedata, Object *ob) const View3D *v3d = draw_ctx->v3d; if ((ob->type == OB_MESH) && (ob == draw_ctx->obact)) { + const int draw_mode = (ob->mode == OB_MODE_VERTEX_PAINT) ? VERTEX_MODE : WEIGHT_MODE; const Mesh *me = ob->data; const Mesh *me_orig = DEG_get_original_object(ob)->data; const bool use_wire = (v3d->overlay.paint_flag & V3D_OVERLAY_PAINT_WIRE) != 0; - const bool use_surface = v3d->overlay.vertex_paint_mode_opacity != 0.0f; const bool use_face_sel = (me_orig->editflag & ME_EDIT_PAINT_FACE_SEL) != 0; const bool use_vert_sel = (me_orig->editflag & ME_EDIT_PAINT_VERT_SEL) != 0; - struct GPUBatch *geom; - if (me->mloopcol == NULL) { - return; + struct GPUBatch *geom = NULL; + if (draw_mode == VERTEX_MODE) { + if (me->mloopcol == NULL) { + return; + } + if (v3d->overlay.vertex_paint_mode_opacity != 0.0f) { + geom = DRW_cache_mesh_surface_vertpaint_get(ob); + } } - - if (use_surface) { - geom = DRW_cache_mesh_surface_vertpaint_get(ob); - DRW_shgroup_call_add(stl->g_data->fvcolor_shgrp, geom, ob->obmat); + else { + if (v3d->overlay.weight_paint_mode_opacity != 0.0f) { + geom = DRW_cache_mesh_surface_weights_get(ob); + } + } + if (geom != NULL) { + DRW_shgroup_call_add(stl->g_data->by_mode[draw_mode].color_shgrp, geom, ob->obmat); } if (use_face_sel || use_wire) { - DRWShadingGroup *shgrp = use_face_sel ? stl->g_data->lwire_select_shgrp : stl->g_data->lwire_shgrp; + DRWShadingGroup *shgrp = use_face_sel ? + stl->g_data->by_mode[draw_mode].lwire_select_shgrp : + stl->g_data->by_mode[draw_mode].lwire_shgrp; geom = DRW_cache_mesh_surface_edges_get(ob); DRW_shgroup_call_add(shgrp, geom, ob->obmat); } @@ -255,8 +312,9 @@ static void PAINT_VERTEX_cache_populate(void *vedata, Object *ob) static void PAINT_VERTEX_draw_scene(void *vedata) { PAINT_VERTEX_PassList *psl = ((PAINT_VERTEX_Data *)vedata)->psl; - - DRW_draw_pass(psl->vcolor_faces); + for (int i = 0; i < MODE_LEN; i++) { + DRW_draw_pass(psl->by_mode[i].color_faces); + } DRW_draw_pass(psl->wire_overlay); DRW_draw_pass(psl->wire_select_overlay); DRW_draw_pass(psl->vert_select_overlay); |