diff options
author | Campbell Barton <ideasman42@gmail.com> | 2018-09-09 09:11:02 +0300 |
---|---|---|
committer | Campbell Barton <ideasman42@gmail.com> | 2018-09-10 07:35:04 +0300 |
commit | 20634fd433b07bbd90358a625792292e9581a0f6 (patch) | |
tree | b6be45ddf341ce7d2dc8c85a67480e6815d4a6e6 /source/blender/editors/mesh/editmesh_preselect_elem.c | |
parent | 2b5d4d426a9b8483539d14b6e9ab90da262c2ecb (diff) |
Tool System: use preselect highlight w/ poly-build
- Poly build now uses a new gizmo for pre-selection
which has the same behavior as loop-cut.
This replaces hack where mouse-move set the active element
which was no longer working properly because of missing
depsgraph updates.
- Multi-object support for poly-build.
- Support for deformed cage.
- Fix error where changing active object wasn't properly
refreshing the preselect gizmo (for loopcut too).
Currently holding Alt to select non-boundary element's isn't working.
Diffstat (limited to 'source/blender/editors/mesh/editmesh_preselect_elem.c')
-rw-r--r-- | source/blender/editors/mesh/editmesh_preselect_elem.c | 218 |
1 files changed, 218 insertions, 0 deletions
diff --git a/source/blender/editors/mesh/editmesh_preselect_elem.c b/source/blender/editors/mesh/editmesh_preselect_elem.c new file mode 100644 index 00000000000..e0b06019de1 --- /dev/null +++ b/source/blender/editors/mesh/editmesh_preselect_elem.c @@ -0,0 +1,218 @@ +/* + * ***** BEGIN GPL LICENSE BLOCK ***** + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software Foundation, + * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + * + * ***** END GPL LICENSE BLOCK ***** + */ + +/** \file blender/editors/mesh/editmesh_preselect_elem.c + * \ingroup edmesh + */ + +#include "MEM_guardedalloc.h" + +#include "BLI_stack.h" +#include "BLI_math.h" + +#include "BKE_context.h" +#include "BKE_editmesh.h" + +#include "GPU_immediate.h" +#include "GPU_matrix.h" +#include "GPU_state.h" + +#include "DNA_object_types.h" + +#include "ED_mesh.h" +#include "ED_view3d.h" + +/* -------------------------------------------------------------------- */ +/** \name Mesh Element Pre-Select + * Public API: + * + * #EDBM_preselect_elem_create + * #EDBM_preselect_elem_destroy + * #EDBM_preselect_elem_clear + * #EDBM_preselect_elem_draw + * #EDBM_preselect_elem_update_from_single + * + * \{ */ + +static void vcos_get(BMVert *v, float r_co[3], const float (*coords)[3]) +{ + if (coords) { + copy_v3_v3(r_co, coords[BM_elem_index_get(v)]); + } + else { + copy_v3_v3(r_co, v->co); + } +} + +static void vcos_get_pair(BMVert *v[2], float r_cos[2][3], const float (*coords)[3]) +{ + if (coords) { + for (int j = 0; j < 2; j++) { + copy_v3_v3(r_cos[j], coords[BM_elem_index_get(v[j])]); + } + } + else { + for (int j = 0; j < 2; j++) { + copy_v3_v3(r_cos[j], v[j]->co); + } + } +} + +struct EditMesh_PreSelElem { + float (*edges)[2][3]; + int edges_len; + + float (*verts)[3]; + int verts_len; +}; + +struct EditMesh_PreSelElem *EDBM_preselect_elem_create(void) +{ + struct EditMesh_PreSelElem *psel = MEM_callocN(sizeof(*psel), __func__); + return psel; +} + +void EDBM_preselect_elem_destroy( + struct EditMesh_PreSelElem *psel) +{ + EDBM_preselect_elem_clear(psel); + MEM_freeN(psel); +} + +void EDBM_preselect_elem_clear( + struct EditMesh_PreSelElem *psel) +{ + MEM_SAFE_FREE(psel->edges); + psel->edges_len = 0; + + MEM_SAFE_FREE(psel->verts); + psel->verts_len = 0; +} + +void EDBM_preselect_elem_draw( + struct EditMesh_PreSelElem *psel, const float matrix[4][4]) +{ + if ((psel->edges_len == 0) && (psel->verts_len == 0)) { + return; + } + + GPU_depth_test(false); + + GPU_matrix_push(); + GPU_matrix_mul(matrix); + + uint pos = GPU_vertformat_attr_add(immVertexFormat(), "pos", GPU_COMP_F32, 3, GPU_FETCH_FLOAT); + + immBindBuiltinProgram(GPU_SHADER_3D_UNIFORM_COLOR); + immUniformColor3ub(255, 0, 255); + + if (psel->edges_len > 0) { + immBegin(GPU_PRIM_LINES, psel->edges_len * 2); + + for (int i = 0; i < psel->edges_len; i++) { + immVertex3fv(pos, psel->edges[i][0]); + immVertex3fv(pos, psel->edges[i][1]); + } + + immEnd(); + } + + if (psel->verts_len > 0) { + GPU_point_size(3.0f); + + immBegin(GPU_PRIM_POINTS, psel->verts_len); + + for (int i = 0; i < psel->verts_len; i++) { + immVertex3fv(pos, psel->verts[i]); + } + + immEnd(); + } + + immUnbindProgram(); + + GPU_matrix_pop(); + + /* Reset default */ + GPU_depth_test(true); +} + +static void view3d_preselect_mesh_elem_update_from_vert( + struct EditMesh_PreSelElem *psel, + BMesh *UNUSED(bm), BMVert *eve, const float (*coords)[3]) +{ + float (*verts)[3] = MEM_mallocN(sizeof(*psel->verts), __func__); + vcos_get(eve, verts[0], coords); + psel->verts = verts; + psel->verts_len = 1; +} + +static void view3d_preselect_mesh_elem_update_from_edge( + struct EditMesh_PreSelElem *psel, + BMesh *UNUSED(bm), BMEdge *eed, const float (*coords)[3]) +{ + float (*edges)[2][3] = MEM_mallocN(sizeof(*psel->edges), __func__); + vcos_get_pair(&eed->v1, edges[0], coords); + psel->edges = edges; + psel->edges_len = 1; +} + +static void view3d_preselect_mesh_elem_update_from_face( + struct EditMesh_PreSelElem *psel, + BMesh *UNUSED(bm), BMFace *efa, const float (*coords)[3]) +{ + float (*edges)[2][3] = MEM_mallocN(sizeof(*psel->edges) * efa->len, __func__); + BMLoop *l_iter, *l_first; + l_iter = l_first = BM_FACE_FIRST_LOOP(efa); + int i = 0; + do { + vcos_get_pair(&l_iter->e->v1, edges[i++], coords); + } while ((l_iter = l_iter->next) != l_first); + psel->edges = edges; + psel->edges_len = efa->len; +} + +void EDBM_preselect_elem_update_from_single( + struct EditMesh_PreSelElem *psel, + BMesh *bm, BMElem *ele, + const float (*coords)[3]) +{ + EDBM_preselect_elem_clear(psel); + + if (coords) { + BM_mesh_elem_index_ensure(bm, BM_VERT); + } + + switch (ele->head.htype) { + case BM_VERT: + view3d_preselect_mesh_elem_update_from_vert(psel, bm, (BMVert *)ele, coords); + break; + case BM_EDGE: + view3d_preselect_mesh_elem_update_from_edge(psel, bm, (BMEdge *)ele, coords); + break; + case BM_FACE: + view3d_preselect_mesh_elem_update_from_face(psel, bm, (BMFace *)ele, coords); + break; + default: + BLI_assert(0); + } +} + +/** \} */ |