Changeset View
Changeset View
Standalone View
Standalone View
source/blender/draw/modes/paint_vertex_mode.c
| Show All 40 Lines | |||||
| extern struct GPUUniformBuffer *globals_ubo; /* draw_common.c */ | extern struct GPUUniformBuffer *globals_ubo; /* draw_common.c */ | ||||
| extern struct GlobalsUboStorage ts; /* draw_common.c */ | extern struct GlobalsUboStorage ts; /* draw_common.c */ | ||||
| extern char datatoc_paint_vertex_vert_glsl[]; | extern char datatoc_paint_vertex_vert_glsl[]; | ||||
| extern char datatoc_paint_vertex_frag_glsl[]; | extern char datatoc_paint_vertex_frag_glsl[]; | ||||
| extern char datatoc_paint_wire_vert_glsl[]; | extern char datatoc_paint_wire_vert_glsl[]; | ||||
| extern char datatoc_paint_wire_frag_glsl[]; | extern char datatoc_paint_wire_frag_glsl[]; | ||||
| extern char datatoc_paint_face_vert_glsl[]; | extern char datatoc_paint_face_vert_glsl[]; | ||||
| extern char datatoc_paint_vert_frag_glsl[]; | |||||
| extern char datatoc_common_globals_lib_glsl[]; | extern char datatoc_common_globals_lib_glsl[]; | ||||
| extern char datatoc_gpu_shader_uniform_color_frag_glsl[]; | extern char datatoc_gpu_shader_uniform_color_frag_glsl[]; | ||||
| /* *********** LISTS *********** */ | /* *********** LISTS *********** */ | ||||
| typedef struct PAINT_VERTEX_PassList { | typedef struct PAINT_VERTEX_PassList { | ||||
| struct DRWPass *vcolor_faces; | struct DRWPass *vcolor_faces; | ||||
| struct DRWPass *wire_overlay; | struct DRWPass *wire_overlay; | ||||
| struct DRWPass *face_overlay; | struct DRWPass *face_overlay; | ||||
| struct DRWPass *vert_overlay; | |||||
| } PAINT_VERTEX_PassList; | } PAINT_VERTEX_PassList; | ||||
| typedef struct PAINT_VERTEX_StorageList { | typedef struct PAINT_VERTEX_StorageList { | ||||
| struct PAINT_VERTEX_PrivateData *g_data; | struct PAINT_VERTEX_PrivateData *g_data; | ||||
| } PAINT_VERTEX_StorageList; | } PAINT_VERTEX_StorageList; | ||||
| typedef struct PAINT_VERTEX_Data { | typedef struct PAINT_VERTEX_Data { | ||||
| void *engine_type; /* Required */ | void *engine_type; /* Required */ | ||||
| DRWViewportEmptyList *fbl; | DRWViewportEmptyList *fbl; | ||||
| DRWViewportEmptyList *txl; | DRWViewportEmptyList *txl; | ||||
| PAINT_VERTEX_PassList *psl; | PAINT_VERTEX_PassList *psl; | ||||
| PAINT_VERTEX_StorageList *stl; | PAINT_VERTEX_StorageList *stl; | ||||
| } PAINT_VERTEX_Data; | } PAINT_VERTEX_Data; | ||||
| /* *********** STATIC *********** */ | /* *********** STATIC *********** */ | ||||
| static struct { | static struct { | ||||
| struct GPUShader *vcolor_face_shader; | struct GPUShader *vcolor_face_shader; | ||||
| struct GPUShader *wire_overlay_shader; | struct GPUShader *wire_overlay_shader; | ||||
| struct GPUShader *face_overlay_shader; | struct GPUShader *face_overlay_shader; | ||||
| struct GPUShader *vert_overlay_shader; | |||||
| } e_data = {NULL}; /* Engine data */ | } e_data = {NULL}; /* Engine data */ | ||||
| typedef struct PAINT_VERTEX_PrivateData { | typedef struct PAINT_VERTEX_PrivateData { | ||||
| DRWShadingGroup *fvcolor_shgrp; | DRWShadingGroup *fvcolor_shgrp; | ||||
| DRWShadingGroup *lwire_shgrp; | DRWShadingGroup *lwire_shgrp; | ||||
| DRWShadingGroup *face_shgrp; | DRWShadingGroup *face_shgrp; | ||||
| DRWShadingGroup *vert_shgrp; | |||||
| } PAINT_VERTEX_PrivateData; /* Transient data */ | } PAINT_VERTEX_PrivateData; /* Transient data */ | ||||
| /* *********** FUNCTIONS *********** */ | /* *********** FUNCTIONS *********** */ | ||||
| static void PAINT_VERTEX_engine_init(void *UNUSED(vedata)) | static void PAINT_VERTEX_engine_init(void *UNUSED(vedata)) | ||||
| { | { | ||||
| if (!e_data.vcolor_face_shader) { | if (!e_data.vcolor_face_shader) { | ||||
| e_data.vcolor_face_shader = DRW_shader_create( | e_data.vcolor_face_shader = DRW_shader_create( | ||||
| datatoc_paint_vertex_vert_glsl, NULL, | datatoc_paint_vertex_vert_glsl, NULL, | ||||
| datatoc_paint_vertex_frag_glsl, NULL); | datatoc_paint_vertex_frag_glsl, NULL); | ||||
| e_data.wire_overlay_shader = DRW_shader_create_with_lib( | e_data.wire_overlay_shader = DRW_shader_create_with_lib( | ||||
| datatoc_paint_wire_vert_glsl, NULL, | datatoc_paint_wire_vert_glsl, NULL, | ||||
| datatoc_paint_wire_frag_glsl, | datatoc_paint_wire_frag_glsl, | ||||
| datatoc_common_globals_lib_glsl, "#define VERTEX_MODE\n"); | datatoc_common_globals_lib_glsl, "#define VERTEX_MODE\n"); | ||||
| e_data.face_overlay_shader = DRW_shader_create( | e_data.face_overlay_shader = DRW_shader_create( | ||||
| datatoc_paint_face_vert_glsl, NULL, | datatoc_paint_face_vert_glsl, NULL, | ||||
| datatoc_gpu_shader_uniform_color_frag_glsl, NULL); | datatoc_gpu_shader_uniform_color_frag_glsl, NULL); | ||||
| e_data.vert_overlay_shader = DRW_shader_create_with_lib( | |||||
| datatoc_paint_wire_vert_glsl, NULL, | |||||
| datatoc_paint_vert_frag_glsl, | |||||
| datatoc_common_globals_lib_glsl, NULL); | |||||
| } | } | ||||
| } | } | ||||
| static void PAINT_VERTEX_cache_init(void *vedata) | static void PAINT_VERTEX_cache_init(void *vedata) | ||||
| { | { | ||||
| PAINT_VERTEX_PassList *psl = ((PAINT_VERTEX_Data *)vedata)->psl; | PAINT_VERTEX_PassList *psl = ((PAINT_VERTEX_Data *)vedata)->psl; | ||||
| PAINT_VERTEX_StorageList *stl = ((PAINT_VERTEX_Data *)vedata)->stl; | PAINT_VERTEX_StorageList *stl = ((PAINT_VERTEX_Data *)vedata)->stl; | ||||
| const DRWContextState *draw_ctx = DRW_context_state_get(); | const DRWContextState *draw_ctx = DRW_context_state_get(); | ||||
| Show All 28 Lines | psl->face_overlay = DRW_pass_create( | ||||
| "Face Mask Pass", | "Face Mask Pass", | ||||
| DRW_STATE_WRITE_COLOR | DRW_STATE_WRITE_DEPTH | DRW_STATE_DEPTH_LESS_EQUAL | DRW_STATE_BLEND); | DRW_STATE_WRITE_COLOR | DRW_STATE_WRITE_DEPTH | DRW_STATE_DEPTH_LESS_EQUAL | DRW_STATE_BLEND); | ||||
| stl->g_data->face_shgrp = DRW_shgroup_create(e_data.face_overlay_shader, psl->face_overlay); | stl->g_data->face_shgrp = DRW_shgroup_create(e_data.face_overlay_shader, psl->face_overlay); | ||||
| static float col[4] = {1.0f, 1.0f, 1.0f, 0.2f}; | static float col[4] = {1.0f, 1.0f, 1.0f, 0.2f}; | ||||
| DRW_shgroup_uniform_vec4(stl->g_data->face_shgrp, "color", col, 1); | DRW_shgroup_uniform_vec4(stl->g_data->face_shgrp, "color", col, 1); | ||||
| } | } | ||||
| { | |||||
| psl->vert_overlay = DRW_pass_create( | |||||
| "Vert Mask Pass", | |||||
| DRW_STATE_WRITE_COLOR | DRW_STATE_WRITE_DEPTH | DRW_STATE_DEPTH_LESS_EQUAL | DRW_STATE_OFFSET_NEGATIVE); | |||||
| stl->g_data->vert_shgrp = DRW_shgroup_create(e_data.vert_overlay_shader, psl->vert_overlay); | |||||
| DRW_shgroup_uniform_block(stl->g_data->vert_shgrp, "globalsBlock", globals_ubo); | |||||
| } | |||||
| } | } | ||||
| static void PAINT_VERTEX_cache_populate(void *vedata, Object *ob) | static void PAINT_VERTEX_cache_populate(void *vedata, Object *ob) | ||||
| { | { | ||||
| PAINT_VERTEX_StorageList *stl = ((PAINT_VERTEX_Data *)vedata)->stl; | PAINT_VERTEX_StorageList *stl = ((PAINT_VERTEX_Data *)vedata)->stl; | ||||
| const DRWContextState *draw_ctx = DRW_context_state_get(); | const DRWContextState *draw_ctx = DRW_context_state_get(); | ||||
| const View3D *v3d = draw_ctx->v3d; | const View3D *v3d = draw_ctx->v3d; | ||||
| if ((ob->type == OB_MESH) && (ob == draw_ctx->obact)) { | if ((ob->type == OB_MESH) && (ob == draw_ctx->obact)) { | ||||
| const Mesh *me = ob->data; | 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_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_surface = v3d->overlay.vertex_paint_mode_opacity != 0.0f; | ||||
| const bool use_face_sel = (me->editflag & ME_EDIT_PAINT_FACE_SEL) != 0; | 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; | struct GPUBatch *geom; | ||||
| if (me->mloopcol == NULL) { | if (me->mloopcol == NULL) { | ||||
| return; | return; | ||||
| } | } | ||||
| if (use_surface) { | if (use_surface) { | ||||
| geom = DRW_cache_mesh_surface_vertpaint_get(ob); | geom = DRW_cache_mesh_surface_vertpaint_get(ob); | ||||
| DRW_shgroup_call_add(stl->g_data->fvcolor_shgrp, geom, ob->obmat); | DRW_shgroup_call_add(stl->g_data->fvcolor_shgrp, geom, ob->obmat); | ||||
| } | } | ||||
| if (use_face_sel || use_wire) { | if (use_face_sel || use_wire) { | ||||
| geom = DRW_cache_mesh_surface_edges_get(ob); | geom = DRW_cache_mesh_surface_edges_get(ob); | ||||
| DRW_shgroup_call_add(stl->g_data->lwire_shgrp, geom, ob->obmat); | DRW_shgroup_call_add(stl->g_data->lwire_shgrp, geom, ob->obmat); | ||||
| } | } | ||||
| if (use_face_sel) { | if (use_face_sel) { | ||||
| geom = DRW_cache_mesh_surface_get(ob); | geom = DRW_cache_mesh_surface_get(ob); | ||||
| DRW_shgroup_call_add(stl->g_data->face_shgrp, geom, ob->obmat); | DRW_shgroup_call_add(stl->g_data->face_shgrp, geom, ob->obmat); | ||||
| } | } | ||||
| if (use_vert_sel) { | |||||
| geom = DRW_cache_mesh_all_verts_get(ob); | |||||
| DRW_shgroup_call_add(stl->g_data->vert_shgrp, geom, ob->obmat); | |||||
| } | |||||
| } | } | ||||
| } | } | ||||
| static void PAINT_VERTEX_draw_scene(void *vedata) | static void PAINT_VERTEX_draw_scene(void *vedata) | ||||
| { | { | ||||
| PAINT_VERTEX_PassList *psl = ((PAINT_VERTEX_Data *)vedata)->psl; | PAINT_VERTEX_PassList *psl = ((PAINT_VERTEX_Data *)vedata)->psl; | ||||
| DRW_draw_pass(psl->vcolor_faces); | DRW_draw_pass(psl->vcolor_faces); | ||||
| DRW_draw_pass(psl->face_overlay); | DRW_draw_pass(psl->face_overlay); | ||||
| DRW_draw_pass(psl->wire_overlay); | DRW_draw_pass(psl->wire_overlay); | ||||
| DRW_draw_pass(psl->vert_overlay); | |||||
| } | } | ||||
| static void PAINT_VERTEX_engine_free(void) | static void PAINT_VERTEX_engine_free(void) | ||||
| { | { | ||||
| DRW_SHADER_FREE_SAFE(e_data.vcolor_face_shader); | DRW_SHADER_FREE_SAFE(e_data.vcolor_face_shader); | ||||
| DRW_SHADER_FREE_SAFE(e_data.wire_overlay_shader); | DRW_SHADER_FREE_SAFE(e_data.wire_overlay_shader); | ||||
| DRW_SHADER_FREE_SAFE(e_data.vert_overlay_shader); | |||||
| DRW_SHADER_FREE_SAFE(e_data.face_overlay_shader); | DRW_SHADER_FREE_SAFE(e_data.face_overlay_shader); | ||||
| } | } | ||||
| static const DrawEngineDataSize PAINT_VERTEX_data_size = DRW_VIEWPORT_DATA_SIZE(PAINT_VERTEX_Data); | static const DrawEngineDataSize PAINT_VERTEX_data_size = DRW_VIEWPORT_DATA_SIZE(PAINT_VERTEX_Data); | ||||
| DrawEngineType draw_engine_paint_vertex_type = { | DrawEngineType draw_engine_paint_vertex_type = { | ||||
| NULL, NULL, | NULL, NULL, | ||||
| N_("PaintVertexMode"), | N_("PaintVertexMode"), | ||||
| Show All 11 Lines | |||||