From db054b447de3636dc4e0f973e16ed22d39e85ce3 Mon Sep 17 00:00:00 2001 From: Hans Goudey Date: Wed, 17 Aug 2022 09:54:01 -0400 Subject: Cleanup: Move view3d_select.c to C++ Almost entirely adding casting and standard clang tidy changes. Also switch to `blender::Vector` instead of the macro-based `BLI_array`. --- source/blender/editors/space_view3d/CMakeLists.txt | 2 +- .../blender/editors/space_view3d/view3d_intern.h | 10 +- .../blender/editors/space_view3d/view3d_select.c | 4775 -------------------- .../blender/editors/space_view3d/view3d_select.cc | 4771 +++++++++++++++++++ 4 files changed, 4781 insertions(+), 4777 deletions(-) delete mode 100644 source/blender/editors/space_view3d/view3d_select.c create mode 100644 source/blender/editors/space_view3d/view3d_select.cc (limited to 'source/blender/editors/space_view3d') diff --git a/source/blender/editors/space_view3d/CMakeLists.txt b/source/blender/editors/space_view3d/CMakeLists.txt index 2b10e5cd17a..100266f4433 100644 --- a/source/blender/editors/space_view3d/CMakeLists.txt +++ b/source/blender/editors/space_view3d/CMakeLists.txt @@ -60,7 +60,7 @@ set(SRC view3d_ops.c view3d_placement.c view3d_project.c - view3d_select.c + view3d_select.cc view3d_snap.c view3d_utils.c view3d_view.c diff --git a/source/blender/editors/space_view3d/view3d_intern.h b/source/blender/editors/space_view3d/view3d_intern.h index 53fc450107a..4c9e2595023 100644 --- a/source/blender/editors/space_view3d/view3d_intern.h +++ b/source/blender/editors/space_view3d/view3d_intern.h @@ -9,6 +9,10 @@ #include "ED_view3d.h" +#ifdef __cplusplus +extern "C" { +#endif + /* internal exports only */ struct ARegion; @@ -83,7 +87,7 @@ void view3d_depths_rect_create(struct ARegion *region, struct rcti *rect, struct */ float view3d_depth_near(struct ViewDepths *d); -/* view3d_select.c */ +/* view3d_select.cc */ void VIEW3D_OT_select(struct wmOperatorType *ot); void VIEW3D_OT_select_circle(struct wmOperatorType *ot); @@ -241,3 +245,7 @@ void VIEW3D_GGT_placement(struct wmGizmoGroupType *gzgt); extern uchar view3d_camera_border_hack_col[3]; extern bool view3d_camera_border_hack_test; #endif + +#ifdef __cplusplus +} +#endif diff --git a/source/blender/editors/space_view3d/view3d_select.c b/source/blender/editors/space_view3d/view3d_select.c deleted file mode 100644 index 763848574ed..00000000000 --- a/source/blender/editors/space_view3d/view3d_select.c +++ /dev/null @@ -1,4775 +0,0 @@ -/* SPDX-License-Identifier: GPL-2.0-or-later - * Copyright 2008 Blender Foundation. All rights reserved. */ - -/** \file - * \ingroup spview3d - */ - -#include -#include -#include -#include - -#include "DNA_action_types.h" -#include "DNA_armature_types.h" -#include "DNA_curve_types.h" -#include "DNA_gpencil_types.h" -#include "DNA_mesh_types.h" -#include "DNA_meshdata_types.h" -#include "DNA_meta_types.h" -#include "DNA_object_types.h" -#include "DNA_scene_types.h" -#include "DNA_tracking_types.h" - -#include "MEM_guardedalloc.h" - -#include "BLI_array.h" -#include "BLI_bitmap.h" -#include "BLI_lasso_2d.h" -#include "BLI_linklist.h" -#include "BLI_listbase.h" -#include "BLI_math.h" -#include "BLI_rect.h" -#include "BLI_string.h" -#include "BLI_utildefines.h" - -#ifdef __BIG_ENDIAN__ -# include "BLI_endian_switch.h" -#endif - -/* vertex box select */ -#include "BKE_global.h" -#include "BKE_main.h" -#include "IMB_imbuf.h" -#include "IMB_imbuf_types.h" - -#include "BKE_action.h" -#include "BKE_armature.h" -#include "BKE_context.h" -#include "BKE_curve.h" -#include "BKE_editmesh.h" -#include "BKE_layer.h" -#include "BKE_mball.h" -#include "BKE_mesh.h" -#include "BKE_object.h" -#include "BKE_paint.h" -#include "BKE_scene.h" -#include "BKE_tracking.h" -#include "BKE_workspace.h" - -#include "WM_api.h" -#include "WM_toolsystem.h" -#include "WM_types.h" - -#include "RNA_access.h" -#include "RNA_define.h" -#include "RNA_enum_types.h" - -#include "ED_armature.h" -#include "ED_curve.h" -#include "ED_gpencil.h" -#include "ED_lattice.h" -#include "ED_mball.h" -#include "ED_mesh.h" -#include "ED_object.h" -#include "ED_outliner.h" -#include "ED_particle.h" -#include "ED_screen.h" -#include "ED_sculpt.h" -#include "ED_select_utils.h" - -#include "UI_interface.h" -#include "UI_resources.h" - -#include "GPU_matrix.h" -#include "GPU_select.h" - -#include "DEG_depsgraph.h" -#include "DEG_depsgraph_query.h" - -#include "DRW_engine.h" -#include "DRW_select_buffer.h" - -#include "view3d_intern.h" /* own include */ - -// #include "PIL_time_utildefines.h" - -/* -------------------------------------------------------------------- */ -/** \name Public Utilities - * \{ */ - -float ED_view3d_select_dist_px(void) -{ - return 75.0f * U.pixelsize; -} - -void ED_view3d_viewcontext_init(bContext *C, ViewContext *vc, Depsgraph *depsgraph) -{ - /* TODO: should return whether there is valid context to continue. */ - - memset(vc, 0, sizeof(ViewContext)); - vc->C = C; - vc->region = CTX_wm_region(C); - vc->bmain = CTX_data_main(C); - vc->depsgraph = depsgraph; - vc->scene = CTX_data_scene(C); - vc->view_layer = CTX_data_view_layer(C); - vc->v3d = CTX_wm_view3d(C); - vc->win = CTX_wm_window(C); - vc->rv3d = CTX_wm_region_view3d(C); - vc->obact = CTX_data_active_object(C); - vc->obedit = CTX_data_edit_object(C); -} - -void ED_view3d_viewcontext_init_object(ViewContext *vc, Object *obact) -{ - vc->obact = obact; - /* See public doc-string for rationale on checking the existing values first. */ - if (vc->obedit) { - BLI_assert(BKE_object_is_in_editmode(obact)); - vc->obedit = obact; - if (vc->em) { - vc->em = BKE_editmesh_from_object(vc->obedit); - } - } -} - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Internal Object Utilities - * \{ */ - -static bool object_deselect_all_visible(ViewLayer *view_layer, View3D *v3d) -{ - bool changed = false; - LISTBASE_FOREACH (Base *, base, &view_layer->object_bases) { - if (base->flag & BASE_SELECTED) { - if (BASE_SELECTABLE(v3d, base)) { - ED_object_base_select(base, BA_DESELECT); - changed = true; - } - } - } - return changed; -} - -/* deselect all except b */ -static bool object_deselect_all_except(ViewLayer *view_layer, Base *b) -{ - bool changed = false; - LISTBASE_FOREACH (Base *, base, &view_layer->object_bases) { - if (base->flag & BASE_SELECTED) { - if (b != base) { - ED_object_base_select(base, BA_DESELECT); - changed = true; - } - } - } - return changed; -} - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Internal Edit-Mesh Select Buffer Wrapper - * - * Avoid duplicate code when using edit-mode selection, - * actual logic is handled outside of this function. - * - * \note Currently this #EDBMSelectID_Context which is mesh specific - * however the logic could also be used for non-meshes too. - * - * \{ */ - -struct EditSelectBuf_Cache { - BLI_bitmap *select_bitmap; -}; - -static void editselect_buf_cache_init(ViewContext *vc, short select_mode) -{ - if (vc->obedit) { - uint bases_len = 0; - Base **bases = BKE_view_layer_array_from_bases_in_edit_mode( - vc->view_layer, vc->v3d, &bases_len); - - DRW_select_buffer_context_create(bases, bases_len, select_mode); - MEM_freeN(bases); - } - else { - /* Use for paint modes, currently only a single object at a time. */ - if (vc->obact) { - Base *base = BKE_view_layer_base_find(vc->view_layer, vc->obact); - DRW_select_buffer_context_create(&base, 1, select_mode); - } - } -} - -static void editselect_buf_cache_free(struct EditSelectBuf_Cache *esel) -{ - MEM_SAFE_FREE(esel->select_bitmap); -} - -static void editselect_buf_cache_free_voidp(void *esel_voidp) -{ - editselect_buf_cache_free(esel_voidp); - MEM_freeN(esel_voidp); -} - -static void editselect_buf_cache_init_with_generic_userdata(wmGenericUserData *wm_userdata, - ViewContext *vc, - short select_mode) -{ - struct EditSelectBuf_Cache *esel = MEM_callocN(sizeof(*esel), __func__); - wm_userdata->data = esel; - wm_userdata->free_fn = editselect_buf_cache_free_voidp; - wm_userdata->use_free = true; - editselect_buf_cache_init(vc, select_mode); -} - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Internal Edit-Mesh Utilities - * \{ */ - -static bool edbm_backbuf_check_and_select_verts(struct EditSelectBuf_Cache *esel, - Depsgraph *depsgraph, - Object *ob, - BMEditMesh *em, - const eSelectOp sel_op) -{ - BMVert *eve; - BMIter iter; - bool changed = false; - - const BLI_bitmap *select_bitmap = esel->select_bitmap; - uint index = DRW_select_buffer_context_offset_for_object_elem(depsgraph, ob, SCE_SELECT_VERTEX); - if (index == 0) { - return false; - } - - index -= 1; - BM_ITER_MESH (eve, &iter, em->bm, BM_VERTS_OF_MESH) { - if (!BM_elem_flag_test(eve, BM_ELEM_HIDDEN)) { - const bool is_select = BM_elem_flag_test(eve, BM_ELEM_SELECT); - const bool is_inside = BLI_BITMAP_TEST_BOOL(select_bitmap, index); - const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); - if (sel_op_result != -1) { - BM_vert_select_set(em->bm, eve, sel_op_result); - changed = true; - } - } - index++; - } - return changed; -} - -static bool edbm_backbuf_check_and_select_edges(struct EditSelectBuf_Cache *esel, - Depsgraph *depsgraph, - Object *ob, - BMEditMesh *em, - const eSelectOp sel_op) -{ - BMEdge *eed; - BMIter iter; - bool changed = false; - - const BLI_bitmap *select_bitmap = esel->select_bitmap; - uint index = DRW_select_buffer_context_offset_for_object_elem(depsgraph, ob, SCE_SELECT_EDGE); - if (index == 0) { - return false; - } - - index -= 1; - BM_ITER_MESH (eed, &iter, em->bm, BM_EDGES_OF_MESH) { - if (!BM_elem_flag_test(eed, BM_ELEM_HIDDEN)) { - const bool is_select = BM_elem_flag_test(eed, BM_ELEM_SELECT); - const bool is_inside = BLI_BITMAP_TEST_BOOL(select_bitmap, index); - const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); - if (sel_op_result != -1) { - BM_edge_select_set(em->bm, eed, sel_op_result); - changed = true; - } - } - index++; - } - return changed; -} - -static bool edbm_backbuf_check_and_select_faces(struct EditSelectBuf_Cache *esel, - Depsgraph *depsgraph, - Object *ob, - BMEditMesh *em, - const eSelectOp sel_op) -{ - BMFace *efa; - BMIter iter; - bool changed = false; - - const BLI_bitmap *select_bitmap = esel->select_bitmap; - uint index = DRW_select_buffer_context_offset_for_object_elem(depsgraph, ob, SCE_SELECT_FACE); - if (index == 0) { - return false; - } - - index -= 1; - BM_ITER_MESH (efa, &iter, em->bm, BM_FACES_OF_MESH) { - if (!BM_elem_flag_test(efa, BM_ELEM_HIDDEN)) { - const bool is_select = BM_elem_flag_test(efa, BM_ELEM_SELECT); - const bool is_inside = BLI_BITMAP_TEST_BOOL(select_bitmap, index); - const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); - if (sel_op_result != -1) { - BM_face_select_set(em->bm, efa, sel_op_result); - changed = true; - } - } - index++; - } - return changed; -} - -/* object mode, edbm_ prefix is confusing here, rename? */ -static bool edbm_backbuf_check_and_select_verts_obmode(Mesh *me, - struct EditSelectBuf_Cache *esel, - const eSelectOp sel_op) -{ - MVert *mv = me->mvert; - bool changed = false; - - const BLI_bitmap *select_bitmap = esel->select_bitmap; - - if (mv) { - const bool *hide_vert = (const bool *)CustomData_get_layer_named( - &me->vdata, CD_PROP_BOOL, ".hide_vert"); - - for (int index = 0; index < me->totvert; index++, mv++) { - if (!(hide_vert && hide_vert[index])) { - const bool is_select = mv->flag & SELECT; - const bool is_inside = BLI_BITMAP_TEST_BOOL(select_bitmap, index); - const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(mv->flag, sel_op_result, SELECT); - changed = true; - } - } - } - } - return changed; -} - -/* object mode, edbm_ prefix is confusing here, rename? */ -static bool edbm_backbuf_check_and_select_faces_obmode(Mesh *me, - struct EditSelectBuf_Cache *esel, - const eSelectOp sel_op) -{ - MPoly *mpoly = me->mpoly; - bool changed = false; - - const BLI_bitmap *select_bitmap = esel->select_bitmap; - - if (mpoly) { - const bool *hide_poly = (const bool *)CustomData_get_layer_named( - &me->pdata, CD_PROP_BOOL, ".hide_poly"); - - for (int index = 0; index < me->totpoly; index++, mpoly++) { - if (!(hide_poly && hide_poly[index])) { - const bool is_select = mpoly->flag & ME_FACE_SEL; - const bool is_inside = BLI_BITMAP_TEST_BOOL(select_bitmap, index); - const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(mpoly->flag, sel_op_result, ME_FACE_SEL); - changed = true; - } - } - } - } - return changed; -} - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Lasso Select - * \{ */ - -typedef struct LassoSelectUserData { - ViewContext *vc; - const rcti *rect; - const rctf *rect_fl; - rctf _rect_fl; - const int (*mcoords)[2]; - int mcoords_len; - eSelectOp sel_op; - eBezTriple_Flag select_flag; - - /* runtime */ - int pass; - bool is_done; - bool is_changed; -} LassoSelectUserData; - -static void view3d_userdata_lassoselect_init(LassoSelectUserData *r_data, - ViewContext *vc, - const rcti *rect, - const int (*mcoords)[2], - const int mcoords_len, - const eSelectOp sel_op) -{ - r_data->vc = vc; - - r_data->rect = rect; - r_data->rect_fl = &r_data->_rect_fl; - BLI_rctf_rcti_copy(&r_data->_rect_fl, rect); - - r_data->mcoords = mcoords; - r_data->mcoords_len = mcoords_len; - r_data->sel_op = sel_op; - /* SELECT by default, but can be changed if needed (only few cases use and respect this). */ - r_data->select_flag = SELECT; - - /* runtime */ - r_data->pass = 0; - r_data->is_done = false; - r_data->is_changed = false; -} - -static bool view3d_selectable_data(bContext *C) -{ - Object *ob = CTX_data_active_object(C); - - if (!ED_operator_region_view3d_active(C)) { - return 0; - } - - if (ob) { - if (ob->mode & OB_MODE_EDIT) { - if (ob->type == OB_FONT) { - return 0; - } - } - else { - if ((ob->mode & (OB_MODE_VERTEX_PAINT | OB_MODE_WEIGHT_PAINT | OB_MODE_TEXTURE_PAINT)) && - !BKE_paint_select_elem_test(ob)) { - return 0; - } - } - } - - return 1; -} - -/* helper also for box_select */ -static bool edge_fully_inside_rect(const rctf *rect, const float v1[2], const float v2[2]) -{ - return BLI_rctf_isect_pt_v(rect, v1) && BLI_rctf_isect_pt_v(rect, v2); -} - -static bool edge_inside_rect(const rctf *rect, const float v1[2], const float v2[2]) -{ - int d1, d2, d3, d4; - - /* check points in rect */ - if (edge_fully_inside_rect(rect, v1, v2)) { - return 1; - } - - /* check points completely out rect */ - if (v1[0] < rect->xmin && v2[0] < rect->xmin) { - return 0; - } - if (v1[0] > rect->xmax && v2[0] > rect->xmax) { - return 0; - } - if (v1[1] < rect->ymin && v2[1] < rect->ymin) { - return 0; - } - if (v1[1] > rect->ymax && v2[1] > rect->ymax) { - return 0; - } - - /* simple check lines intersecting. */ - d1 = (v1[1] - v2[1]) * (v1[0] - rect->xmin) + (v2[0] - v1[0]) * (v1[1] - rect->ymin); - d2 = (v1[1] - v2[1]) * (v1[0] - rect->xmin) + (v2[0] - v1[0]) * (v1[1] - rect->ymax); - d3 = (v1[1] - v2[1]) * (v1[0] - rect->xmax) + (v2[0] - v1[0]) * (v1[1] - rect->ymax); - d4 = (v1[1] - v2[1]) * (v1[0] - rect->xmax) + (v2[0] - v1[0]) * (v1[1] - rect->ymin); - - if (d1 < 0 && d2 < 0 && d3 < 0 && d4 < 0) { - return 0; - } - if (d1 > 0 && d2 > 0 && d3 > 0 && d4 > 0) { - return 0; - } - - return 1; -} - -static void do_lasso_select_pose__do_tag(void *userData, - struct bPoseChannel *pchan, - const float screen_co_a[2], - const float screen_co_b[2]) -{ - LassoSelectUserData *data = userData; - const bArmature *arm = data->vc->obact->data; - if (!PBONE_SELECTABLE(arm, pchan->bone)) { - return; - } - - if (BLI_rctf_isect_segment(data->rect_fl, screen_co_a, screen_co_b) && - BLI_lasso_is_edge_inside( - data->mcoords, data->mcoords_len, UNPACK2(screen_co_a), UNPACK2(screen_co_b), INT_MAX)) { - pchan->bone->flag |= BONE_DONE; - data->is_changed = true; - } -} -static void do_lasso_tag_pose(ViewContext *vc, - Object *ob, - const int mcoords[][2], - const int mcoords_len) -{ - ViewContext vc_tmp; - LassoSelectUserData data; - rcti rect; - - if ((ob->type != OB_ARMATURE) || (ob->pose == NULL)) { - return; - } - - vc_tmp = *vc; - vc_tmp.obact = ob; - - BLI_lasso_boundbox(&rect, mcoords, mcoords_len); - - view3d_userdata_lassoselect_init(&data, vc, &rect, mcoords, mcoords_len, 0); - - ED_view3d_init_mats_rv3d(vc_tmp.obact, vc->rv3d); - - /* Treat bones as clipped segments (no joints). */ - pose_foreachScreenBone(&vc_tmp, - do_lasso_select_pose__do_tag, - &data, - V3D_PROJ_TEST_CLIP_DEFAULT | V3D_PROJ_TEST_CLIP_CONTENT_DEFAULT); -} - -static bool do_lasso_select_objects(ViewContext *vc, - const int mcoords[][2], - const int mcoords_len, - const eSelectOp sel_op) -{ - View3D *v3d = vc->v3d; - Base *base; - - bool changed = false; - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - changed |= object_deselect_all_visible(vc->view_layer, vc->v3d); - } - - for (base = vc->view_layer->object_bases.first; base; base = base->next) { - if (BASE_SELECTABLE(v3d, base)) { /* Use this to avoid unnecessary lasso look-ups. */ - const bool is_select = base->flag & BASE_SELECTED; - const bool is_inside = ((ED_view3d_project_base(vc->region, base) == V3D_PROJ_RET_OK) && - BLI_lasso_is_point_inside( - mcoords, mcoords_len, base->sx, base->sy, IS_CLIPPED)); - const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); - if (sel_op_result != -1) { - ED_object_base_select(base, sel_op_result ? BA_SELECT : BA_DESELECT); - changed = true; - } - } - } - - if (changed) { - DEG_id_tag_update(&vc->scene->id, ID_RECALC_SELECT); - WM_main_add_notifier(NC_SCENE | ND_OB_SELECT, vc->scene); - } - return changed; -} - -/** - * Use for lasso & box select. - */ -static Base **do_pose_tag_select_op_prepare(ViewContext *vc, uint *r_bases_len) -{ - Base **bases = NULL; - BLI_array_declare(bases); - FOREACH_BASE_IN_MODE_BEGIN (vc->view_layer, vc->v3d, OB_ARMATURE, OB_MODE_POSE, base_iter) { - Object *ob_iter = base_iter->object; - bArmature *arm = ob_iter->data; - LISTBASE_FOREACH (bPoseChannel *, pchan, &ob_iter->pose->chanbase) { - Bone *bone = pchan->bone; - bone->flag &= ~BONE_DONE; - } - arm->id.tag |= LIB_TAG_DOIT; - ob_iter->id.tag &= ~LIB_TAG_DOIT; - BLI_array_append(bases, base_iter); - } - FOREACH_BASE_IN_MODE_END; - *r_bases_len = BLI_array_len(bases); - return bases; -} - -static bool do_pose_tag_select_op_exec(Base **bases, const uint bases_len, const eSelectOp sel_op) -{ - bool changed_multi = false; - - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - for (int i = 0; i < bases_len; i++) { - Base *base_iter = bases[i]; - Object *ob_iter = base_iter->object; - if (ED_pose_deselect_all(ob_iter, SEL_DESELECT, false)) { - ED_pose_bone_select_tag_update(ob_iter); - changed_multi = true; - } - } - } - - for (int i = 0; i < bases_len; i++) { - Base *base_iter = bases[i]; - Object *ob_iter = base_iter->object; - bArmature *arm = ob_iter->data; - - /* Don't handle twice. */ - if (arm->id.tag & LIB_TAG_DOIT) { - arm->id.tag &= ~LIB_TAG_DOIT; - } - else { - continue; - } - - bool changed = true; - LISTBASE_FOREACH (bPoseChannel *, pchan, &ob_iter->pose->chanbase) { - Bone *bone = pchan->bone; - if ((bone->flag & BONE_UNSELECTABLE) == 0) { - const bool is_select = bone->flag & BONE_SELECTED; - const bool is_inside = bone->flag & BONE_DONE; - const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(bone->flag, sel_op_result, BONE_SELECTED); - if (sel_op_result == 0) { - if (arm->act_bone == bone) { - arm->act_bone = NULL; - } - } - changed = true; - } - } - } - if (changed) { - ED_pose_bone_select_tag_update(ob_iter); - changed_multi = true; - } - } - return changed_multi; -} - -static bool do_lasso_select_pose(ViewContext *vc, - const int mcoords[][2], - const int mcoords_len, - const eSelectOp sel_op) -{ - uint bases_len; - Base **bases = do_pose_tag_select_op_prepare(vc, &bases_len); - - for (int i = 0; i < bases_len; i++) { - Base *base_iter = bases[i]; - Object *ob_iter = base_iter->object; - do_lasso_tag_pose(vc, ob_iter, mcoords, mcoords_len); - } - - const bool changed_multi = do_pose_tag_select_op_exec(bases, bases_len, sel_op); - if (changed_multi) { - DEG_id_tag_update(&vc->scene->id, ID_RECALC_SELECT); - WM_main_add_notifier(NC_SCENE | ND_OB_SELECT, vc->scene); - } - - MEM_freeN(bases); - return changed_multi; -} - -static void do_lasso_select_mesh__doSelectVert(void *userData, - BMVert *eve, - const float screen_co[2], - int UNUSED(index)) -{ - LassoSelectUserData *data = userData; - const bool is_select = BM_elem_flag_test(eve, BM_ELEM_SELECT); - const bool is_inside = - (BLI_rctf_isect_pt_v(data->rect_fl, screen_co) && - BLI_lasso_is_point_inside( - data->mcoords, data->mcoords_len, screen_co[0], screen_co[1], IS_CLIPPED)); - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - BM_vert_select_set(data->vc->em->bm, eve, sel_op_result); - data->is_changed = true; - } -} -struct LassoSelectUserData_ForMeshEdge { - LassoSelectUserData *data; - struct EditSelectBuf_Cache *esel; - uint backbuf_offset; -}; -static void do_lasso_select_mesh__doSelectEdge_pass0(void *user_data, - BMEdge *eed, - const float screen_co_a[2], - const float screen_co_b[2], - int index) -{ - struct LassoSelectUserData_ForMeshEdge *data_for_edge = user_data; - LassoSelectUserData *data = data_for_edge->data; - bool is_visible = true; - if (data_for_edge->backbuf_offset) { - uint bitmap_inedx = data_for_edge->backbuf_offset + index - 1; - is_visible = BLI_BITMAP_TEST_BOOL(data_for_edge->esel->select_bitmap, bitmap_inedx); - } - - const bool is_select = BM_elem_flag_test(eed, BM_ELEM_SELECT); - const bool is_inside = - (is_visible && edge_fully_inside_rect(data->rect_fl, screen_co_a, screen_co_b) && - BLI_lasso_is_point_inside( - data->mcoords, data->mcoords_len, UNPACK2(screen_co_a), IS_CLIPPED) && - BLI_lasso_is_point_inside( - data->mcoords, data->mcoords_len, UNPACK2(screen_co_b), IS_CLIPPED)); - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - BM_edge_select_set(data->vc->em->bm, eed, sel_op_result); - data->is_done = true; - data->is_changed = true; - } -} -static void do_lasso_select_mesh__doSelectEdge_pass1(void *user_data, - BMEdge *eed, - const float screen_co_a[2], - const float screen_co_b[2], - int index) -{ - struct LassoSelectUserData_ForMeshEdge *data_for_edge = user_data; - LassoSelectUserData *data = data_for_edge->data; - bool is_visible = true; - if (data_for_edge->backbuf_offset) { - uint bitmap_inedx = data_for_edge->backbuf_offset + index - 1; - is_visible = BLI_BITMAP_TEST_BOOL(data_for_edge->esel->select_bitmap, bitmap_inedx); - } - - const bool is_select = BM_elem_flag_test(eed, BM_ELEM_SELECT); - const bool is_inside = (is_visible && BLI_lasso_is_edge_inside(data->mcoords, - data->mcoords_len, - UNPACK2(screen_co_a), - UNPACK2(screen_co_b), - IS_CLIPPED)); - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - BM_edge_select_set(data->vc->em->bm, eed, sel_op_result); - data->is_changed = true; - } -} - -static void do_lasso_select_mesh__doSelectFace(void *userData, - BMFace *efa, - const float screen_co[2], - int UNUSED(index)) -{ - LassoSelectUserData *data = userData; - const bool is_select = BM_elem_flag_test(efa, BM_ELEM_SELECT); - const bool is_inside = - (BLI_rctf_isect_pt_v(data->rect_fl, screen_co) && - BLI_lasso_is_point_inside( - data->mcoords, data->mcoords_len, screen_co[0], screen_co[1], IS_CLIPPED)); - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - BM_face_select_set(data->vc->em->bm, efa, sel_op_result); - data->is_changed = true; - } -} - -static bool do_lasso_select_mesh(ViewContext *vc, - wmGenericUserData *wm_userdata, - const int mcoords[][2], - const int mcoords_len, - const eSelectOp sel_op) -{ - LassoSelectUserData data; - ToolSettings *ts = vc->scene->toolsettings; - rcti rect; - - /* set editmesh */ - vc->em = BKE_editmesh_from_object(vc->obedit); - - BLI_lasso_boundbox(&rect, mcoords, mcoords_len); - - view3d_userdata_lassoselect_init(&data, vc, &rect, mcoords, mcoords_len, sel_op); - - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - if (vc->em->bm->totvertsel) { - EDBM_flag_disable_all(vc->em, BM_ELEM_SELECT); - data.is_changed = true; - } - } - - /* for non zbuf projections, don't change the GL state */ - ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); - - GPU_matrix_set(vc->rv3d->viewmat); - - const bool use_zbuf = !XRAY_FLAG_ENABLED(vc->v3d); - - struct EditSelectBuf_Cache *esel = wm_userdata->data; - if (use_zbuf) { - if (wm_userdata->data == NULL) { - editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, ts->selectmode); - esel = wm_userdata->data; - esel->select_bitmap = DRW_select_buffer_bitmap_from_poly( - vc->depsgraph, vc->region, vc->v3d, mcoords, mcoords_len, &rect, NULL); - } - } - - if (ts->selectmode & SCE_SELECT_VERTEX) { - if (use_zbuf) { - data.is_changed |= edbm_backbuf_check_and_select_verts( - esel, vc->depsgraph, vc->obedit, vc->em, sel_op); - } - else { - mesh_foreachScreenVert( - vc, do_lasso_select_mesh__doSelectVert, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - } - } - if (ts->selectmode & SCE_SELECT_EDGE) { - /* Does both use_zbuf and non-use_zbuf versions (need screen cos for both) */ - struct LassoSelectUserData_ForMeshEdge data_for_edge = { - .data = &data, - .esel = use_zbuf ? esel : NULL, - .backbuf_offset = use_zbuf ? DRW_select_buffer_context_offset_for_object_elem( - vc->depsgraph, vc->obedit, SCE_SELECT_EDGE) : - 0, - }; - - const eV3DProjTest clip_flag = V3D_PROJ_TEST_CLIP_NEAR | - (use_zbuf ? 0 : V3D_PROJ_TEST_CLIP_BB); - /* Fully inside. */ - mesh_foreachScreenEdge_clip_bb_segment( - vc, do_lasso_select_mesh__doSelectEdge_pass0, &data_for_edge, clip_flag); - if (data.is_done == false) { - /* Fall back to partially inside. - * Clip content to account for edges partially behind the view. */ - mesh_foreachScreenEdge_clip_bb_segment(vc, - do_lasso_select_mesh__doSelectEdge_pass1, - &data_for_edge, - clip_flag | V3D_PROJ_TEST_CLIP_CONTENT_DEFAULT); - } - } - - if (ts->selectmode & SCE_SELECT_FACE) { - if (use_zbuf) { - data.is_changed |= edbm_backbuf_check_and_select_faces( - esel, vc->depsgraph, vc->obedit, vc->em, sel_op); - } - else { - mesh_foreachScreenFace( - vc, do_lasso_select_mesh__doSelectFace, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - } - } - - if (data.is_changed) { - EDBM_selectmode_flush(vc->em); - } - return data.is_changed; -} - -static void do_lasso_select_curve__doSelect(void *userData, - Nurb *UNUSED(nu), - BPoint *bp, - BezTriple *bezt, - int beztindex, - bool handles_visible, - const float screen_co[2]) -{ - LassoSelectUserData *data = userData; - - const bool is_inside = BLI_lasso_is_point_inside( - data->mcoords, data->mcoords_len, screen_co[0], screen_co[1], IS_CLIPPED); - if (bp) { - const bool is_select = bp->f1 & SELECT; - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(bp->f1, sel_op_result, data->select_flag); - data->is_changed = true; - } - } - else { - if (!handles_visible) { - /* can only be (beztindex == 1) here since handles are hidden */ - const bool is_select = bezt->f2 & SELECT; - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(bezt->f2, sel_op_result, data->select_flag); - } - bezt->f1 = bezt->f3 = bezt->f2; - data->is_changed = true; - } - else { - uint8_t *flag_p = (&bezt->f1) + beztindex; - const bool is_select = *flag_p & SELECT; - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(*flag_p, sel_op_result, data->select_flag); - data->is_changed = true; - } - } - } -} - -static bool do_lasso_select_curve(ViewContext *vc, - const int mcoords[][2], - const int mcoords_len, - const eSelectOp sel_op) -{ - const bool deselect_all = (sel_op == SEL_OP_SET); - LassoSelectUserData data; - rcti rect; - - BLI_lasso_boundbox(&rect, mcoords, mcoords_len); - - view3d_userdata_lassoselect_init(&data, vc, &rect, mcoords, mcoords_len, sel_op); - - Curve *curve = (Curve *)vc->obedit->data; - ListBase *nurbs = BKE_curve_editNurbs_get(curve); - - /* For deselect all, items to be selected are tagged with temp flag. Clear that first. */ - if (deselect_all) { - BKE_nurbList_flag_set(nurbs, BEZT_FLAG_TEMP_TAG, false); - data.select_flag = BEZT_FLAG_TEMP_TAG; - } - - ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); /* for foreach's screen/vert projection */ - nurbs_foreachScreenVert(vc, do_lasso_select_curve__doSelect, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - - /* Deselect items that were not added to selection (indicated by temp flag). */ - if (deselect_all) { - data.is_changed |= BKE_nurbList_flag_set_from_flag(nurbs, BEZT_FLAG_TEMP_TAG, SELECT); - } - - if (data.is_changed) { - BKE_curve_nurb_vert_active_validate(vc->obedit->data); - } - return data.is_changed; -} - -static void do_lasso_select_lattice__doSelect(void *userData, BPoint *bp, const float screen_co[2]) -{ - LassoSelectUserData *data = userData; - const bool is_select = bp->f1 & SELECT; - const bool is_inside = - (BLI_rctf_isect_pt_v(data->rect_fl, screen_co) && - BLI_lasso_is_point_inside( - data->mcoords, data->mcoords_len, screen_co[0], screen_co[1], IS_CLIPPED)); - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(bp->f1, sel_op_result, SELECT); - data->is_changed = true; - } -} -static bool do_lasso_select_lattice(ViewContext *vc, - const int mcoords[][2], - const int mcoords_len, - const eSelectOp sel_op) -{ - LassoSelectUserData data; - rcti rect; - - BLI_lasso_boundbox(&rect, mcoords, mcoords_len); - - view3d_userdata_lassoselect_init(&data, vc, &rect, mcoords, mcoords_len, sel_op); - - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - data.is_changed |= ED_lattice_flags_set(vc->obedit, 0); - } - - ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); /* for foreach's screen/vert projection */ - lattice_foreachScreenVert( - vc, do_lasso_select_lattice__doSelect, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - return data.is_changed; -} - -static void do_lasso_select_armature__doSelectBone(void *userData, - EditBone *ebone, - const float screen_co_a[2], - const float screen_co_b[2]) -{ - LassoSelectUserData *data = userData; - const bArmature *arm = data->vc->obedit->data; - if (!EBONE_VISIBLE(arm, ebone)) { - return; - } - - int is_ignore_flag = 0; - int is_inside_flag = 0; - - if (screen_co_a[0] != IS_CLIPPED) { - if (BLI_rcti_isect_pt(data->rect, UNPACK2(screen_co_a)) && - BLI_lasso_is_point_inside( - data->mcoords, data->mcoords_len, UNPACK2(screen_co_a), INT_MAX)) { - is_inside_flag |= BONESEL_ROOT; - } - } - else { - is_ignore_flag |= BONESEL_ROOT; - } - - if (screen_co_b[0] != IS_CLIPPED) { - if (BLI_rcti_isect_pt(data->rect, UNPACK2(screen_co_b)) && - BLI_lasso_is_point_inside( - data->mcoords, data->mcoords_len, UNPACK2(screen_co_b), INT_MAX)) { - is_inside_flag |= BONESEL_TIP; - } - } - else { - is_ignore_flag |= BONESEL_TIP; - } - - if (is_ignore_flag == 0) { - if (is_inside_flag == (BONE_ROOTSEL | BONE_TIPSEL) || - BLI_lasso_is_edge_inside(data->mcoords, - data->mcoords_len, - UNPACK2(screen_co_a), - UNPACK2(screen_co_b), - INT_MAX)) { - is_inside_flag |= BONESEL_BONE; - } - } - - ebone->temp.i = is_inside_flag | (is_ignore_flag >> 16); -} -static void do_lasso_select_armature__doSelectBone_clip_content(void *userData, - EditBone *ebone, - const float screen_co_a[2], - const float screen_co_b[2]) -{ - LassoSelectUserData *data = userData; - bArmature *arm = data->vc->obedit->data; - if (!EBONE_VISIBLE(arm, ebone)) { - return; - } - - const int is_ignore_flag = ebone->temp.i << 16; - int is_inside_flag = ebone->temp.i & ~0xFFFF; - - /* - When #BONESEL_BONE is set, there is nothing to do. - * - When #BONE_ROOTSEL or #BONE_TIPSEL have been set - they take priority over bone selection. - */ - if (is_inside_flag & (BONESEL_BONE | BONE_ROOTSEL | BONE_TIPSEL)) { - return; - } - - if (BLI_lasso_is_edge_inside( - data->mcoords, data->mcoords_len, UNPACK2(screen_co_a), UNPACK2(screen_co_b), INT_MAX)) { - is_inside_flag |= BONESEL_BONE; - } - - ebone->temp.i = is_inside_flag | (is_ignore_flag >> 16); -} - -static bool do_lasso_select_armature(ViewContext *vc, - const int mcoords[][2], - const int mcoords_len, - const eSelectOp sel_op) -{ - LassoSelectUserData data; - rcti rect; - - BLI_lasso_boundbox(&rect, mcoords, mcoords_len); - - view3d_userdata_lassoselect_init(&data, vc, &rect, mcoords, mcoords_len, sel_op); - - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - data.is_changed |= ED_armature_edit_deselect_all_visible(vc->obedit); - } - - bArmature *arm = vc->obedit->data; - - ED_armature_ebone_listbase_temp_clear(arm->edbo); - - ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); - - /* Operate on fully visible (non-clipped) points. */ - armature_foreachScreenBone( - vc, do_lasso_select_armature__doSelectBone, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - - /* Operate on bones as segments clipped to the viewport bounds - * (needed to handle bones with both points outside the view). - * A separate pass is needed since clipped coordinates can't be used for selecting joints. */ - armature_foreachScreenBone(vc, - do_lasso_select_armature__doSelectBone_clip_content, - &data, - V3D_PROJ_TEST_CLIP_DEFAULT | V3D_PROJ_TEST_CLIP_CONTENT_DEFAULT); - - data.is_changed |= ED_armature_edit_select_op_from_tagged(vc->obedit->data, sel_op); - - if (data.is_changed) { - WM_main_add_notifier(NC_OBJECT | ND_BONE_SELECT, vc->obedit); - } - return data.is_changed; -} - -static void do_lasso_select_mball__doSelectElem(void *userData, - struct MetaElem *ml, - const float screen_co[2]) -{ - LassoSelectUserData *data = userData; - const bool is_select = ml->flag & SELECT; - const bool is_inside = - (BLI_rctf_isect_pt_v(data->rect_fl, screen_co) && - BLI_lasso_is_point_inside( - data->mcoords, data->mcoords_len, screen_co[0], screen_co[1], INT_MAX)); - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(ml->flag, sel_op_result, SELECT); - data->is_changed = true; - } -} -static bool do_lasso_select_meta(ViewContext *vc, - const int mcoords[][2], - const int mcoords_len, - const eSelectOp sel_op) -{ - LassoSelectUserData data; - rcti rect; - - MetaBall *mb = (MetaBall *)vc->obedit->data; - - BLI_lasso_boundbox(&rect, mcoords, mcoords_len); - - view3d_userdata_lassoselect_init(&data, vc, &rect, mcoords, mcoords_len, sel_op); - - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - data.is_changed |= BKE_mball_deselect_all(mb); - } - - ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); - - mball_foreachScreenElem( - vc, do_lasso_select_mball__doSelectElem, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - - return data.is_changed; -} - -static void do_lasso_select_meshobject__doSelectVert(void *userData, - MVert *mv, - const float screen_co[2], - int UNUSED(index)) -{ - LassoSelectUserData *data = userData; - const bool is_select = mv->flag & SELECT; - const bool is_inside = - (BLI_rctf_isect_pt_v(data->rect_fl, screen_co) && - BLI_lasso_is_point_inside( - data->mcoords, data->mcoords_len, screen_co[0], screen_co[1], IS_CLIPPED)); - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(mv->flag, sel_op_result, SELECT); - data->is_changed = true; - } -} -static bool do_lasso_select_paintvert(ViewContext *vc, - wmGenericUserData *wm_userdata, - const int mcoords[][2], - const int mcoords_len, - const eSelectOp sel_op) -{ - const bool use_zbuf = !XRAY_ENABLED(vc->v3d); - Object *ob = vc->obact; - Mesh *me = ob->data; - rcti rect; - - if (me == NULL || me->totvert == 0) { - return false; - } - - bool changed = false; - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - /* flush selection at the end */ - changed |= paintvert_deselect_all_visible(ob, SEL_DESELECT, false); - } - - BLI_lasso_boundbox(&rect, mcoords, mcoords_len); - - struct EditSelectBuf_Cache *esel = wm_userdata->data; - if (use_zbuf) { - if (wm_userdata->data == NULL) { - editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, SCE_SELECT_VERTEX); - esel = wm_userdata->data; - esel->select_bitmap = DRW_select_buffer_bitmap_from_poly( - vc->depsgraph, vc->region, vc->v3d, mcoords, mcoords_len, &rect, NULL); - } - } - - if (use_zbuf) { - if (esel->select_bitmap != NULL) { - changed |= edbm_backbuf_check_and_select_verts_obmode(me, esel, sel_op); - } - } - else { - LassoSelectUserData data; - - view3d_userdata_lassoselect_init(&data, vc, &rect, mcoords, mcoords_len, sel_op); - - ED_view3d_init_mats_rv3d(vc->obact, vc->rv3d); - - meshobject_foreachScreenVert( - vc, do_lasso_select_meshobject__doSelectVert, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - - changed |= data.is_changed; - } - - if (changed) { - if (SEL_OP_CAN_DESELECT(sel_op)) { - BKE_mesh_mselect_validate(me); - } - paintvert_flush_flags(ob); - paintvert_tag_select_update(vc->C, ob); - } - - return changed; -} -static bool do_lasso_select_paintface(ViewContext *vc, - wmGenericUserData *wm_userdata, - const int mcoords[][2], - const int mcoords_len, - const eSelectOp sel_op) -{ - Object *ob = vc->obact; - Mesh *me = ob->data; - rcti rect; - - if (me == NULL || me->totpoly == 0) { - return false; - } - - bool changed = false; - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - /* flush selection at the end */ - changed |= paintface_deselect_all_visible(vc->C, ob, SEL_DESELECT, false); - } - - BLI_lasso_boundbox(&rect, mcoords, mcoords_len); - - struct EditSelectBuf_Cache *esel = wm_userdata->data; - if (esel == NULL) { - editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, SCE_SELECT_FACE); - esel = wm_userdata->data; - esel->select_bitmap = DRW_select_buffer_bitmap_from_poly( - vc->depsgraph, vc->region, vc->v3d, mcoords, mcoords_len, &rect, NULL); - } - - if (esel->select_bitmap) { - changed |= edbm_backbuf_check_and_select_faces_obmode(me, esel, sel_op); - } - - if (changed) { - paintface_flush_flags(vc->C, ob, true, false); - } - return changed; -} - -static bool view3d_lasso_select(bContext *C, - ViewContext *vc, - const int mcoords[][2], - const int mcoords_len, - const eSelectOp sel_op) -{ - Object *ob = CTX_data_active_object(C); - bool changed_multi = false; - - wmGenericUserData wm_userdata_buf = {0}; - wmGenericUserData *wm_userdata = &wm_userdata_buf; - - if (vc->obedit == NULL) { /* Object Mode */ - if (BKE_paint_select_face_test(ob)) { - changed_multi |= do_lasso_select_paintface(vc, wm_userdata, mcoords, mcoords_len, sel_op); - } - else if (BKE_paint_select_vert_test(ob)) { - changed_multi |= do_lasso_select_paintvert(vc, wm_userdata, mcoords, mcoords_len, sel_op); - } - else if (ob && - (ob->mode & (OB_MODE_VERTEX_PAINT | OB_MODE_WEIGHT_PAINT | OB_MODE_TEXTURE_PAINT))) { - /* pass */ - } - else if (ob && (ob->mode & OB_MODE_PARTICLE_EDIT)) { - changed_multi |= PE_lasso_select(C, mcoords, mcoords_len, sel_op); - } - else if (ob && (ob->mode & OB_MODE_POSE)) { - changed_multi |= do_lasso_select_pose(vc, mcoords, mcoords_len, sel_op); - if (changed_multi) { - ED_outliner_select_sync_from_pose_bone_tag(C); - } - } - else { - changed_multi |= do_lasso_select_objects(vc, mcoords, mcoords_len, sel_op); - if (changed_multi) { - ED_outliner_select_sync_from_object_tag(C); - } - } - } - else { /* Edit Mode */ - FOREACH_OBJECT_IN_MODE_BEGIN (vc->view_layer, vc->v3d, ob->type, ob->mode, ob_iter) { - ED_view3d_viewcontext_init_object(vc, ob_iter); - bool changed = false; - - switch (vc->obedit->type) { - case OB_MESH: - changed = do_lasso_select_mesh(vc, wm_userdata, mcoords, mcoords_len, sel_op); - break; - case OB_CURVES_LEGACY: - case OB_SURF: - changed = do_lasso_select_curve(vc, mcoords, mcoords_len, sel_op); - break; - case OB_LATTICE: - changed = do_lasso_select_lattice(vc, mcoords, mcoords_len, sel_op); - break; - case OB_ARMATURE: - changed = do_lasso_select_armature(vc, mcoords, mcoords_len, sel_op); - if (changed) { - ED_outliner_select_sync_from_edit_bone_tag(C); - } - break; - case OB_MBALL: - changed = do_lasso_select_meta(vc, mcoords, mcoords_len, sel_op); - break; - default: - BLI_assert_msg(0, "lasso select on incorrect object type"); - break; - } - - if (changed) { - DEG_id_tag_update(vc->obedit->data, ID_RECALC_SELECT); - WM_event_add_notifier(C, NC_GEOM | ND_SELECT, vc->obedit->data); - changed_multi = true; - } - } - FOREACH_OBJECT_IN_MODE_END; - } - - WM_generic_user_data_free(wm_userdata); - - return changed_multi; -} - -/* lasso operator gives properties, but since old code works - * with short array we convert */ -static int view3d_lasso_select_exec(bContext *C, wmOperator *op) -{ - ViewContext vc; - int mcoords_len; - const int(*mcoords)[2] = WM_gesture_lasso_path_to_array(C, op, &mcoords_len); - - if (mcoords) { - Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C); - view3d_operator_needs_opengl(C); - BKE_object_update_select_id(CTX_data_main(C)); - - /* setup view context for argument to callbacks */ - ED_view3d_viewcontext_init(C, &vc, depsgraph); - - eSelectOp sel_op = RNA_enum_get(op->ptr, "mode"); - bool changed_multi = view3d_lasso_select(C, &vc, mcoords, mcoords_len, sel_op); - - MEM_freeN((void *)mcoords); - - if (changed_multi) { - return OPERATOR_FINISHED; - } - return OPERATOR_CANCELLED; - } - return OPERATOR_PASS_THROUGH; -} - -void VIEW3D_OT_select_lasso(wmOperatorType *ot) -{ - ot->name = "Lasso Select"; - ot->description = "Select items using lasso selection"; - ot->idname = "VIEW3D_OT_select_lasso"; - - ot->invoke = WM_gesture_lasso_invoke; - ot->modal = WM_gesture_lasso_modal; - ot->exec = view3d_lasso_select_exec; - ot->poll = view3d_selectable_data; - ot->cancel = WM_gesture_lasso_cancel; - - /* flags */ - ot->flag = OPTYPE_UNDO | OPTYPE_DEPENDS_ON_CURSOR; - - /* properties */ - WM_operator_properties_gesture_lasso(ot); - WM_operator_properties_select_operation(ot); -} - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Cursor Picking - * \{ */ - -/* The max number of menu items in an object select menu */ -typedef struct SelMenuItemF { - char idname[MAX_ID_NAME - 2]; - int icon; - Base *base_ptr; - void *item_ptr; -} SelMenuItemF; - -#define SEL_MENU_SIZE 22 -static SelMenuItemF object_mouse_select_menu_data[SEL_MENU_SIZE]; - -/* special (crappy) operator only for menu select */ -static const EnumPropertyItem *object_select_menu_enum_itemf(bContext *C, - PointerRNA *UNUSED(ptr), - PropertyRNA *UNUSED(prop), - bool *r_free) -{ - EnumPropertyItem *item = NULL, item_tmp = {0}; - int totitem = 0; - int i = 0; - - /* Don't need context but avoid API doc-generation using this. */ - if (C == NULL || object_mouse_select_menu_data[i].idname[0] == '\0') { - return DummyRNA_NULL_items; - } - - for (; i < SEL_MENU_SIZE && object_mouse_select_menu_data[i].idname[0] != '\0'; i++) { - item_tmp.name = object_mouse_select_menu_data[i].idname; - item_tmp.identifier = object_mouse_select_menu_data[i].idname; - item_tmp.value = i; - item_tmp.icon = object_mouse_select_menu_data[i].icon; - RNA_enum_item_add(&item, &totitem, &item_tmp); - } - - RNA_enum_item_end(&item, &totitem); - *r_free = true; - - return item; -} - -static int object_select_menu_exec(bContext *C, wmOperator *op) -{ - const int name_index = RNA_enum_get(op->ptr, "name"); - const bool extend = RNA_boolean_get(op->ptr, "extend"); - const bool deselect = RNA_boolean_get(op->ptr, "deselect"); - const bool toggle = RNA_boolean_get(op->ptr, "toggle"); - bool changed = false; - const char *name = object_mouse_select_menu_data[name_index].idname; - - View3D *v3d = CTX_wm_view3d(C); - ViewLayer *view_layer = CTX_data_view_layer(C); - const Base *oldbasact = BASACT(view_layer); - - Base *basact = NULL; - CTX_DATA_BEGIN (C, Base *, base, selectable_bases) { - /* This is a bit dodgy, there should only be ONE object with this name, - * but library objects can mess this up. */ - if (STREQ(name, base->object->id.name + 2)) { - basact = base; - break; - } - } - CTX_DATA_END; - - if (basact == NULL) { - return OPERATOR_CANCELLED; - } - UNUSED_VARS_NDEBUG(v3d); - BLI_assert(BASE_SELECTABLE(v3d, basact)); - - if (extend) { - ED_object_base_select(basact, BA_SELECT); - changed = true; - } - else if (deselect) { - ED_object_base_select(basact, BA_DESELECT); - changed = true; - } - else if (toggle) { - if (basact->flag & BASE_SELECTED) { - if (basact == oldbasact) { - ED_object_base_select(basact, BA_DESELECT); - changed = true; - } - } - else { - ED_object_base_select(basact, BA_SELECT); - changed = true; - } - } - else { - object_deselect_all_except(view_layer, basact); - ED_object_base_select(basact, BA_SELECT); - changed = true; - } - - if ((oldbasact != basact)) { - ED_object_base_activate(C, basact); - } - - /* weak but ensures we activate menu again before using the enum */ - memset(object_mouse_select_menu_data, 0, sizeof(object_mouse_select_menu_data)); - - /* undo? */ - if (changed) { - Scene *scene = CTX_data_scene(C); - DEG_id_tag_update(&scene->id, ID_RECALC_SELECT); - WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, scene); - - ED_outliner_select_sync_from_object_tag(C); - - return OPERATOR_FINISHED; - } - return OPERATOR_CANCELLED; -} - -void VIEW3D_OT_select_menu(wmOperatorType *ot) -{ - PropertyRNA *prop; - - /* identifiers */ - ot->name = "Select Menu"; - ot->description = "Menu object selection"; - ot->idname = "VIEW3D_OT_select_menu"; - - /* api callbacks */ - ot->invoke = WM_menu_invoke; - ot->exec = object_select_menu_exec; - - /* flags */ - ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO; - - /* #Object.id.name to select (dynamic enum). */ - prop = RNA_def_enum(ot->srna, "name", DummyRNA_NULL_items, 0, "Object Name", ""); - RNA_def_enum_funcs(prop, object_select_menu_enum_itemf); - RNA_def_property_flag(prop, PROP_HIDDEN | PROP_ENUM_NO_TRANSLATE); - ot->prop = prop; - - prop = RNA_def_boolean(ot->srna, "extend", 0, "Extend", ""); - RNA_def_property_flag(prop, PROP_SKIP_SAVE); - prop = RNA_def_boolean(ot->srna, "deselect", 0, "Deselect", ""); - RNA_def_property_flag(prop, PROP_SKIP_SAVE); - prop = RNA_def_boolean(ot->srna, "toggle", 0, "Toggle", ""); - RNA_def_property_flag(prop, PROP_SKIP_SAVE); -} - -/** - * \return True when a menu was activated. - */ -static bool object_mouse_select_menu(bContext *C, - ViewContext *vc, - const GPUSelectResult *buffer, - const int hits, - const int mval[2], - const struct SelectPick_Params *params, - Base **r_basact) -{ - int base_count = 0; - bool ok; - LinkNodePair linklist = {NULL, NULL}; - - /* handle base->object->select_id */ - CTX_DATA_BEGIN (C, Base *, base, selectable_bases) { - ok = false; - - /* two selection methods, the CTRL select uses max dist of 15 */ - if (buffer) { - for (int a = 0; a < hits; a++) { - /* index was converted */ - if (base->object->runtime.select_id == (buffer[a].id & ~0xFFFF0000)) { - ok = true; - break; - } - } - } - else { - const int dist = 15 * U.pixelsize; - if (ED_view3d_project_base(vc->region, base) == V3D_PROJ_RET_OK) { - const int delta_px[2] = {base->sx - mval[0], base->sy - mval[1]}; - if (len_manhattan_v2_int(delta_px) < dist) { - ok = true; - } - } - } - - if (ok) { - base_count++; - BLI_linklist_append(&linklist, base); - - if (base_count == SEL_MENU_SIZE) { - break; - } - } - } - CTX_DATA_END; - - *r_basact = NULL; - - if (base_count == 0) { - return false; - } - if (base_count == 1) { - Base *base = (Base *)linklist.list->link; - BLI_linklist_free(linklist.list, NULL); - *r_basact = base; - return false; - } - - /* UI, full in static array values that we later use in an enum function */ - LinkNode *node; - int i; - - memset(object_mouse_select_menu_data, 0, sizeof(object_mouse_select_menu_data)); - - for (node = linklist.list, i = 0; node; node = node->next, i++) { - Base *base = node->link; - Object *ob = base->object; - const char *name = ob->id.name + 2; - - BLI_strncpy(object_mouse_select_menu_data[i].idname, name, MAX_ID_NAME - 2); - object_mouse_select_menu_data[i].icon = UI_icon_from_id(&ob->id); - } - - wmOperatorType *ot = WM_operatortype_find("VIEW3D_OT_select_menu", false); - PointerRNA ptr; - - WM_operator_properties_create_ptr(&ptr, ot); - RNA_boolean_set(&ptr, "extend", params->sel_op == SEL_OP_ADD); - RNA_boolean_set(&ptr, "deselect", params->sel_op == SEL_OP_SUB); - RNA_boolean_set(&ptr, "toggle", params->sel_op == SEL_OP_XOR); - WM_operator_name_call_ptr(C, ot, WM_OP_INVOKE_DEFAULT, &ptr, NULL); - WM_operator_properties_free(&ptr); - - BLI_linklist_free(linklist.list, NULL); - return true; -} - -static int bone_select_menu_exec(bContext *C, wmOperator *op) -{ - const int name_index = RNA_enum_get(op->ptr, "name"); - - const struct SelectPick_Params params = { - .sel_op = ED_select_op_from_operator(op->ptr), - }; - - View3D *v3d = CTX_wm_view3d(C); - ViewLayer *view_layer = CTX_data_view_layer(C); - const Base *oldbasact = BASACT(view_layer); - - Base *basact = object_mouse_select_menu_data[name_index].base_ptr; - - if (basact == NULL) { - return OPERATOR_CANCELLED; - } - - BLI_assert(BASE_SELECTABLE(v3d, basact)); - - if (basact->object->mode & OB_MODE_EDIT) { - EditBone *ebone = (EditBone *)object_mouse_select_menu_data[name_index].item_ptr; - ED_armature_edit_select_pick_bone(C, basact, ebone, BONE_SELECTED, ¶ms); - } - else { - bPoseChannel *pchan = (bPoseChannel *)object_mouse_select_menu_data[name_index].item_ptr; - ED_armature_pose_select_pick_bone(view_layer, v3d, basact->object, pchan->bone, ¶ms); - } - - /* Weak but ensures we activate the menu again before using the enum. */ - memset(object_mouse_select_menu_data, 0, sizeof(object_mouse_select_menu_data)); - - /* We make the armature selected: - * Not-selected active object in pose-mode won't work well for tools. */ - ED_object_base_select(basact, BA_SELECT); - - WM_event_add_notifier(C, NC_OBJECT | ND_BONE_SELECT, basact->object); - WM_event_add_notifier(C, NC_OBJECT | ND_BONE_ACTIVE, basact->object); - - /* In weight-paint, we use selected bone to select vertex-group, - * so don't switch to new active object. */ - if (oldbasact) { - if (basact->object->mode & OB_MODE_EDIT) { - /* Pass. */ - } - else if (oldbasact->object->mode & OB_MODE_ALL_WEIGHT_PAINT) { - /* Prevent activating. - * Selection causes this to be considered the 'active' pose in weight-paint mode. - * Eventually this limitation may be removed. - * For now, de-select all other pose objects deforming this mesh. */ - ED_armature_pose_select_in_wpaint_mode(view_layer, basact); - } - else { - if (oldbasact != basact) { - ED_object_base_activate(C, basact); - } - } - } - - /* Undo? */ - Scene *scene = CTX_data_scene(C); - DEG_id_tag_update(&scene->id, ID_RECALC_SELECT); - DEG_id_tag_update(&scene->id, ID_RECALC_BASE_FLAGS); - WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, scene); - - ED_outliner_select_sync_from_object_tag(C); - - return OPERATOR_FINISHED; -} - -void VIEW3D_OT_bone_select_menu(wmOperatorType *ot) -{ - PropertyRNA *prop; - - /* identifiers */ - ot->name = "Select Menu"; - ot->description = "Menu bone selection"; - ot->idname = "VIEW3D_OT_bone_select_menu"; - - /* api callbacks */ - ot->invoke = WM_menu_invoke; - ot->exec = bone_select_menu_exec; - - /* flags */ - ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO; - - /* #Object.id.name to select (dynamic enum). */ - prop = RNA_def_enum(ot->srna, "name", DummyRNA_NULL_items, 0, "Bone Name", ""); - RNA_def_enum_funcs(prop, object_select_menu_enum_itemf); - RNA_def_property_flag(prop, PROP_HIDDEN | PROP_ENUM_NO_TRANSLATE); - ot->prop = prop; - - prop = RNA_def_boolean(ot->srna, "extend", 0, "Extend", ""); - RNA_def_property_flag(prop, PROP_SKIP_SAVE); - prop = RNA_def_boolean(ot->srna, "deselect", 0, "Deselect", ""); - RNA_def_property_flag(prop, PROP_SKIP_SAVE); - prop = RNA_def_boolean(ot->srna, "toggle", 0, "Toggle", ""); - RNA_def_property_flag(prop, PROP_SKIP_SAVE); -} - -/** - * \return True when a menu was activated. - */ -static bool bone_mouse_select_menu(bContext *C, - const GPUSelectResult *buffer, - const int hits, - const bool is_editmode, - const struct SelectPick_Params *params) -{ - BLI_assert(buffer); - - int bone_count = 0; - LinkNodePair base_list = {NULL, NULL}; - LinkNodePair bone_list = {NULL, NULL}; - GSet *added_bones = BLI_gset_ptr_new("Bone mouse select menu"); - - /* Select logic taken from ed_armature_pick_bone_from_selectbuffer_impl in armature_select.c */ - for (int a = 0; a < hits; a++) { - void *bone_ptr = NULL; - Base *bone_base = NULL; - uint hitresult = buffer[a].id; - - if (!(hitresult & BONESEL_ANY)) { - /* To avoid including objects in selection. */ - continue; - } - - hitresult &= ~BONESEL_ANY; - const uint hit_object = hitresult & 0xFFFF; - - /* Find the hit bone base (armature object). */ - CTX_DATA_BEGIN (C, Base *, base, selectable_bases) { - if (base->object->runtime.select_id == hit_object) { - bone_base = base; - break; - } - } - CTX_DATA_END; - - if (!bone_base) { - continue; - } - - /* Determine what the current bone is */ - if (is_editmode) { - EditBone *ebone; - const uint hit_bone = (hitresult & ~BONESEL_ANY) >> 16; - bArmature *arm = bone_base->object->data; - ebone = BLI_findlink(arm->edbo, hit_bone); - if (ebone && !(ebone->flag & BONE_UNSELECTABLE)) { - bone_ptr = ebone; - } - } - else { - bPoseChannel *pchan; - const uint hit_bone = (hitresult & ~BONESEL_ANY) >> 16; - pchan = BLI_findlink(&bone_base->object->pose->chanbase, hit_bone); - if (pchan && !(pchan->bone->flag & BONE_UNSELECTABLE)) { - bone_ptr = pchan; - } - } - - if (!bone_ptr) { - continue; - } - /* We can hit a bone multiple times, so make sure we are not adding an already included bone - * to the list. */ - const bool is_duplicate_bone = BLI_gset_haskey(added_bones, bone_ptr); - - if (!is_duplicate_bone) { - bone_count++; - BLI_linklist_append(&base_list, bone_base); - BLI_linklist_append(&bone_list, bone_ptr); - BLI_gset_insert(added_bones, bone_ptr); - - if (bone_count == SEL_MENU_SIZE) { - break; - } - } - } - - BLI_gset_free(added_bones, NULL); - - if (bone_count == 0) { - return false; - } - if (bone_count == 1) { - BLI_linklist_free(base_list.list, NULL); - BLI_linklist_free(bone_list.list, NULL); - return false; - } - - /* UI, full in static array values that we later use in an enum function */ - LinkNode *bone_node, *base_node; - int i; - - memset(object_mouse_select_menu_data, 0, sizeof(object_mouse_select_menu_data)); - - for (base_node = base_list.list, bone_node = bone_list.list, i = 0; bone_node; - base_node = base_node->next, bone_node = bone_node->next, i++) { - char *name; - - object_mouse_select_menu_data[i].base_ptr = base_node->link; - - if (is_editmode) { - EditBone *ebone = bone_node->link; - object_mouse_select_menu_data[i].item_ptr = ebone; - name = ebone->name; - } - else { - bPoseChannel *pchan = bone_node->link; - object_mouse_select_menu_data[i].item_ptr = pchan; - name = pchan->name; - } - - BLI_strncpy(object_mouse_select_menu_data[i].idname, name, MAX_ID_NAME - 2); - object_mouse_select_menu_data[i].icon = ICON_BONE_DATA; - } - - wmOperatorType *ot = WM_operatortype_find("VIEW3D_OT_bone_select_menu", false); - PointerRNA ptr; - - WM_operator_properties_create_ptr(&ptr, ot); - RNA_boolean_set(&ptr, "extend", params->sel_op == SEL_OP_ADD); - RNA_boolean_set(&ptr, "deselect", params->sel_op == SEL_OP_SUB); - RNA_boolean_set(&ptr, "toggle", params->sel_op == SEL_OP_XOR); - WM_operator_name_call_ptr(C, ot, WM_OP_INVOKE_DEFAULT, &ptr, NULL); - WM_operator_properties_free(&ptr); - - BLI_linklist_free(base_list.list, NULL); - BLI_linklist_free(bone_list.list, NULL); - return true; -} - -static bool selectbuffer_has_bones(const GPUSelectResult *buffer, const uint hits) -{ - for (uint i = 0; i < hits; i++) { - if (buffer[i].id & 0xFFFF0000) { - return true; - } - } - return false; -} - -/* utility function for mixed_bones_object_selectbuffer */ -static int selectbuffer_ret_hits_15(GPUSelectResult *UNUSED(buffer), const int hits15) -{ - return hits15; -} - -static int selectbuffer_ret_hits_9(GPUSelectResult *buffer, const int hits15, const int hits9) -{ - const int ofs = hits15; - memcpy(buffer, buffer + ofs, hits9 * sizeof(GPUSelectResult)); - return hits9; -} - -static int selectbuffer_ret_hits_5(GPUSelectResult *buffer, - const int hits15, - const int hits9, - const int hits5) -{ - const int ofs = hits15 + hits9; - memcpy(buffer, buffer + ofs, hits5 * sizeof(GPUSelectResult)); - return hits5; -} - -/** - * Populate a select buffer with objects and bones, if there are any. - * Checks three selection levels and compare. - * - * \param do_nearest_xray_if_supported: When set, read in hits that don't stop - * at the nearest surface. The hits must still be ordered by depth. - * Needed so we can step to the next, non-active object when it's already selected, see: T76445. - */ -static int mixed_bones_object_selectbuffer(ViewContext *vc, - GPUSelectResult *buffer, - const int buffer_len, - const int mval[2], - eV3DSelectObjectFilter select_filter, - bool do_nearest, - bool do_nearest_xray_if_supported, - const bool do_material_slot_selection) -{ - rcti rect; - int hits15, hits9 = 0, hits5 = 0; - bool has_bones15 = false, has_bones9 = false, has_bones5 = false; - - int select_mode = (do_nearest ? VIEW3D_SELECT_PICK_NEAREST : VIEW3D_SELECT_PICK_ALL); - int hits = 0; - - if (do_nearest_xray_if_supported) { - if ((U.gpu_flag & USER_GPU_FLAG_NO_DEPT_PICK) == 0) { - select_mode = VIEW3D_SELECT_PICK_ALL; - } - } - - /* we _must_ end cache before return, use 'goto finally' */ - view3d_opengl_select_cache_begin(); - - BLI_rcti_init_pt_radius(&rect, mval, 14); - hits15 = view3d_opengl_select_ex( - vc, buffer, buffer_len, &rect, select_mode, select_filter, do_material_slot_selection); - if (hits15 == 1) { - hits = selectbuffer_ret_hits_15(buffer, hits15); - goto finally; - } - else if (hits15 > 0) { - int ofs; - has_bones15 = selectbuffer_has_bones(buffer, hits15); - - ofs = hits15; - BLI_rcti_init_pt_radius(&rect, mval, 9); - hits9 = view3d_opengl_select( - vc, buffer + ofs, buffer_len - ofs, &rect, select_mode, select_filter); - if (hits9 == 1) { - hits = selectbuffer_ret_hits_9(buffer, hits15, hits9); - goto finally; - } - else if (hits9 > 0) { - has_bones9 = selectbuffer_has_bones(buffer + ofs, hits9); - - ofs += hits9; - BLI_rcti_init_pt_radius(&rect, mval, 5); - hits5 = view3d_opengl_select( - vc, buffer + ofs, buffer_len - ofs, &rect, select_mode, select_filter); - if (hits5 == 1) { - hits = selectbuffer_ret_hits_5(buffer, hits15, hits9, hits5); - goto finally; - } - else if (hits5 > 0) { - has_bones5 = selectbuffer_has_bones(buffer + ofs, hits5); - } - } - - if (has_bones5) { - hits = selectbuffer_ret_hits_5(buffer, hits15, hits9, hits5); - goto finally; - } - else if (has_bones9) { - hits = selectbuffer_ret_hits_9(buffer, hits15, hits9); - goto finally; - } - else if (has_bones15) { - hits = selectbuffer_ret_hits_15(buffer, hits15); - goto finally; - } - - if (hits5 > 0) { - hits = selectbuffer_ret_hits_5(buffer, hits15, hits9, hits5); - goto finally; - } - else if (hits9 > 0) { - hits = selectbuffer_ret_hits_9(buffer, hits15, hits9); - goto finally; - } - else { - hits = selectbuffer_ret_hits_15(buffer, hits15); - goto finally; - } - } - -finally: - view3d_opengl_select_cache_end(); - return hits; -} - -static int mixed_bones_object_selectbuffer_extended(ViewContext *vc, - GPUSelectResult *buffer, - const int buffer_len, - const int mval[2], - eV3DSelectObjectFilter select_filter, - bool use_cycle, - bool enumerate, - bool *r_do_nearest) -{ - bool do_nearest = false; - View3D *v3d = vc->v3d; - - /* define if we use solid nearest select or not */ - if (use_cycle) { - /* Update the coordinates (even if the return value isn't used). */ - const bool has_motion = WM_cursor_test_motion_and_update(mval); - if (!XRAY_ACTIVE(v3d)) { - do_nearest = has_motion; - } - } - else { - if (!XRAY_ACTIVE(v3d)) { - do_nearest = true; - } - } - - if (r_do_nearest) { - *r_do_nearest = do_nearest; - } - - do_nearest = do_nearest && !enumerate; - - int hits = mixed_bones_object_selectbuffer( - vc, buffer, buffer_len, mval, select_filter, do_nearest, true, false); - - return hits; -} - -/** - * Compare result of 'GPU_select': 'GPUSelectResult', - * Needed for stable sorting, so cycling through all items near the cursor behaves predictably. - */ -static int gpu_select_buffer_depth_id_cmp(const void *sel_a_p, const void *sel_b_p) -{ - GPUSelectResult *a = (GPUSelectResult *)sel_a_p; - GPUSelectResult *b = (GPUSelectResult *)sel_b_p; - - if (a->depth < b->depth) { - return -1; - } - if (a->depth > b->depth) { - return 1; - } - - /* Depths match, sort by id. */ - uint sel_a = a->id; - uint sel_b = b->id; - -#ifdef __BIG_ENDIAN__ - BLI_endian_switch_uint32(&sel_a); - BLI_endian_switch_uint32(&sel_b); -#endif - - if (sel_a < sel_b) { - return -1; - } - if (sel_a > sel_b) { - return 1; - } - return 0; -} - -/** - * \param has_bones: When true, skip non-bone hits, also allow bases to be used - * that are visible but not select-able, - * since you may be in pose mode with an un-selectable object. - * - * \return the active base or NULL. - */ -static Base *mouse_select_eval_buffer(ViewContext *vc, - const GPUSelectResult *buffer, - int hits, - bool do_nearest, - bool has_bones, - bool do_bones_get_priotity, - int *r_select_id_subelem) -{ - ViewLayer *view_layer = vc->view_layer; - View3D *v3d = vc->v3d; - int a; - - bool found = false; - int select_id = 0; - int select_id_subelem = 0; - - if (do_nearest) { - uint min = 0xFFFFFFFF; - int hit_index = -1; - - if (has_bones && do_bones_get_priotity) { - /* we skip non-bone hits */ - for (a = 0; a < hits; a++) { - if (min > buffer[a].depth && (buffer[a].id & 0xFFFF0000)) { - min = buffer[a].depth; - hit_index = a; - } - } - } - else { - - for (a = 0; a < hits; a++) { - /* Any object. */ - if (min > buffer[a].depth) { - min = buffer[a].depth; - hit_index = a; - } - } - } - - if (hit_index != -1) { - select_id = buffer[hit_index].id & 0xFFFF; - select_id_subelem = (buffer[hit_index].id & 0xFFFF0000) >> 16; - found = true; - /* No need to set `min` to `buffer[hit_index].depth`, it's not used from now on. */ - } - } - else { - - { - GPUSelectResult *buffer_sorted = MEM_mallocN(sizeof(*buffer_sorted) * hits, __func__); - memcpy(buffer_sorted, buffer, sizeof(*buffer_sorted) * hits); - /* Remove non-bone objects. */ - if (has_bones && do_bones_get_priotity) { - /* Loop backwards to reduce re-ordering. */ - for (a = hits - 1; a >= 0; a--) { - if ((buffer_sorted[a].id & 0xFFFF0000) == 0) { - buffer_sorted[a] = buffer_sorted[--hits]; - } - } - } - qsort(buffer_sorted, hits, sizeof(GPUSelectResult), gpu_select_buffer_depth_id_cmp); - buffer = buffer_sorted; - } - - int hit_index = -1; - - /* It's possible there are no hits (all objects contained bones). */ - if (hits > 0) { - /* Only exclude active object when it is selected. */ - if (BASACT(view_layer) && (BASACT(view_layer)->flag & BASE_SELECTED)) { - const int select_id_active = BASACT(view_layer)->object->runtime.select_id; - for (int i_next = 0, i_prev = hits - 1; i_next < hits; i_prev = i_next++) { - if ((select_id_active == (buffer[i_prev].id & 0xFFFF)) && - (select_id_active != (buffer[i_next].id & 0xFFFF))) { - hit_index = i_next; - break; - } - } - } - - /* When the active object is unselected or not in `buffer`, use the nearest. */ - if (hit_index == -1) { - /* Just pick the nearest. */ - hit_index = 0; - } - } - - if (hit_index != -1) { - select_id = buffer[hit_index].id & 0xFFFF; - select_id_subelem = (buffer[hit_index].id & 0xFFFF0000) >> 16; - found = true; - } - MEM_freeN((void *)buffer); - } - - Base *basact = NULL; - if (found) { - for (Base *base = FIRSTBASE(view_layer); base; base = base->next) { - if (has_bones ? BASE_VISIBLE(v3d, base) : BASE_SELECTABLE(v3d, base)) { - if (base->object->runtime.select_id == select_id) { - basact = base; - break; - } - } - } - - if (basact && r_select_id_subelem) { - *r_select_id_subelem = select_id_subelem; - } - } - - return basact; -} - -static Base *mouse_select_object_center(ViewContext *vc, Base *startbase, const int mval[2]) -{ - ARegion *region = vc->region; - ViewLayer *view_layer = vc->view_layer; - View3D *v3d = vc->v3d; - - Base *oldbasact = BASACT(view_layer); - - const float mval_fl[2] = {(float)mval[0], (float)mval[1]}; - float dist = ED_view3d_select_dist_px() * 1.3333f; - Base *basact = NULL; - - /* Put the active object at a disadvantage to cycle through other objects. */ - const float penalty_dist = 10.0f * UI_DPI_FAC; - Base *base = startbase; - while (base) { - if (BASE_SELECTABLE(v3d, base)) { - float screen_co[2]; - if (ED_view3d_project_float_global( - region, base->object->obmat[3], screen_co, V3D_PROJ_TEST_CLIP_DEFAULT) == - V3D_PROJ_RET_OK) { - float dist_test = len_manhattan_v2v2(mval_fl, screen_co); - if (base == oldbasact) { - dist_test += penalty_dist; - } - if (dist_test < dist) { - dist = dist_test; - basact = base; - } - } - } - base = base->next; - - if (base == NULL) { - base = FIRSTBASE(view_layer); - } - if (base == startbase) { - break; - } - } - return basact; -} - -static Base *ed_view3d_give_base_under_cursor_ex(bContext *C, - const int mval[2], - int *r_material_slot) -{ - Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C); - ViewContext vc; - Base *basact = NULL; - GPUSelectResult buffer[MAXPICKELEMS]; - - /* setup view context for argument to callbacks */ - view3d_operator_needs_opengl(C); - BKE_object_update_select_id(CTX_data_main(C)); - - ED_view3d_viewcontext_init(C, &vc, depsgraph); - - const bool do_nearest = !XRAY_ACTIVE(vc.v3d); - const bool do_material_slot_selection = r_material_slot != NULL; - const int hits = mixed_bones_object_selectbuffer(&vc, - buffer, - ARRAY_SIZE(buffer), - mval, - VIEW3D_SELECT_FILTER_NOP, - do_nearest, - false, - do_material_slot_selection); - - if (hits > 0) { - const bool has_bones = (r_material_slot == NULL) && selectbuffer_has_bones(buffer, hits); - basact = mouse_select_eval_buffer( - &vc, buffer, hits, do_nearest, has_bones, true, r_material_slot); - } - - return basact; -} - -Base *ED_view3d_give_base_under_cursor(bContext *C, const int mval[2]) -{ - return ed_view3d_give_base_under_cursor_ex(C, mval, NULL); -} - -Object *ED_view3d_give_object_under_cursor(bContext *C, const int mval[2]) -{ - Base *base = ED_view3d_give_base_under_cursor(C, mval); - if (base) { - return base->object; - } - return NULL; -} - -struct Object *ED_view3d_give_material_slot_under_cursor(struct bContext *C, - const int mval[2], - int *r_material_slot) -{ - Base *base = ed_view3d_give_base_under_cursor_ex(C, mval, r_material_slot); - if (base) { - return base->object; - } - return NULL; -} - -bool ED_view3d_is_object_under_cursor(bContext *C, const int mval[2]) -{ - return ED_view3d_give_object_under_cursor(C, mval) != NULL; -} - -static void deselect_all_tracks(MovieTracking *tracking) -{ - MovieTrackingObject *object; - - object = tracking->objects.first; - while (object) { - ListBase *tracksbase = BKE_tracking_object_get_tracks(tracking, object); - MovieTrackingTrack *track = tracksbase->first; - - while (track) { - BKE_tracking_track_deselect(track, TRACK_AREA_ALL); - - track = track->next; - } - - object = object->next; - } -} - -static bool ed_object_select_pick_camera_track(bContext *C, - Scene *scene, - Base *basact, - MovieClip *clip, - const struct GPUSelectResult *buffer, - const short hits, - const struct SelectPick_Params *params) -{ - bool changed = false; - bool found = false; - - MovieTracking *tracking = &clip->tracking; - ListBase *tracksbase = NULL; - MovieTrackingTrack *track = NULL; - - for (int i = 0; i < hits; i++) { - const int hitresult = buffer[i].id; - - /* If there's bundles in buffer select bundles first, - * so non-camera elements should be ignored in buffer. */ - if (basact->object->runtime.select_id != (hitresult & 0xFFFF)) { - continue; - } - /* Index of bundle is 1<<16-based. if there's no "bone" index - * in height word, this buffer value belongs to camera. not to bundle. */ - if ((hitresult & 0xFFFF0000) == 0) { - continue; - } - - track = BKE_tracking_track_get_indexed(&clip->tracking, hitresult >> 16, &tracksbase); - found = true; - break; - } - - /* Note `params->deselect_all` is ignored for tracks as in this case - * all objects will be de-selected (not tracks). */ - if (params->sel_op == SEL_OP_SET) { - if ((found && params->select_passthrough) && TRACK_SELECTED(track)) { - found = false; - } - else if (found /* `|| params->deselect_all` */) { - /* Deselect everything. */ - deselect_all_tracks(tracking); - changed = true; - } - } - - if (found) { - switch (params->sel_op) { - case SEL_OP_ADD: { - BKE_tracking_track_select(tracksbase, track, TRACK_AREA_ALL, true); - break; - } - case SEL_OP_SUB: { - BKE_tracking_track_deselect(track, TRACK_AREA_ALL); - break; - } - case SEL_OP_XOR: { - if (TRACK_SELECTED(track)) { - BKE_tracking_track_deselect(track, TRACK_AREA_ALL); - } - else { - BKE_tracking_track_select(tracksbase, track, TRACK_AREA_ALL, true); - } - break; - } - case SEL_OP_SET: { - BKE_tracking_track_select(tracksbase, track, TRACK_AREA_ALL, false); - break; - } - case SEL_OP_AND: { - BLI_assert_unreachable(); /* Doesn't make sense for picking. */ - break; - } - } - - DEG_id_tag_update(&scene->id, ID_RECALC_SELECT); - DEG_id_tag_update(&clip->id, ID_RECALC_SELECT); - WM_event_add_notifier(C, NC_MOVIECLIP | ND_SELECT, track); - WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, scene); - - changed = true; - } - - return changed || found; -} - -/** - * Cursor selection picking for object & pose-mode. - * - * \param mval: Region relative cursor coordinates. - * \param params: Selection parameters. - * \param center: Select by the cursors on-screen distances to the center/origin - * instead of the geometry any other contents of the item being selected. - * This could be used to select by bones by their origin too, currently it's only used for objects. - * \param enumerate: Show a menu for objects at the cursor location. - * Otherwise fall-through to non-menu selection. - * \param object_only: Only select objects (not bones / track markers). - */ -static bool ed_object_select_pick(bContext *C, - const int mval[2], - const struct SelectPick_Params *params, - const bool center, - const bool enumerate, - const bool object_only) -{ - Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C); - ViewContext vc; - /* Setup view context for argument to callbacks. */ - ED_view3d_viewcontext_init(C, &vc, depsgraph); - - Scene *scene = vc.scene; - View3D *v3d = vc.v3d; - - /* Menu activation may find a base to make active (if it only finds a single item to select). */ - Base *basact_override = NULL; - - const bool is_obedit = (vc.obedit != NULL); - if (object_only) { - /* Signal for #view3d_opengl_select to skip edit-mode objects. */ - vc.obedit = NULL; - } - - /* Set for GPU depth buffer picking, leave NULL when selecting by center. */ - struct { - GPUSelectResult buffer[MAXPICKELEMS]; - int hits; - bool do_nearest; - bool has_bones; - } *gpu = NULL; - - /* First handle menu selection, early exit if a menu opens - * since this takes ownership of the selection action. - * - * Even when there is no menu `basact_override` may be set to avoid having to re-find - * the item under the cursor. */ - - if (center == false) { - gpu = MEM_mallocN(sizeof(*gpu), __func__); - gpu->do_nearest = false; - gpu->has_bones = false; - - /* If objects have pose-mode set, the bones are in the same selection buffer. */ - const eV3DSelectObjectFilter select_filter = ((object_only == false) ? - ED_view3d_select_filter_from_mode(scene, - vc.obact) : - VIEW3D_SELECT_FILTER_NOP); - gpu->hits = mixed_bones_object_selectbuffer_extended(&vc, - gpu->buffer, - ARRAY_SIZE(gpu->buffer), - mval, - select_filter, - true, - enumerate, - &gpu->do_nearest); - gpu->has_bones = (object_only && gpu->hits > 0) ? - false : - selectbuffer_has_bones(gpu->buffer, gpu->hits); - } - - /* First handle menu selection, early exit when a menu was opened. - * Otherwise fall through to regular selection. */ - if (enumerate) { - bool has_menu = false; - if (center) { - if (object_mouse_select_menu(C, &vc, NULL, 0, mval, params, &basact_override)) { - has_menu = true; - } - } - else { - if (gpu->hits != 0) { - if (gpu->has_bones && bone_mouse_select_menu(C, gpu->buffer, gpu->hits, false, params)) { - has_menu = true; - } - else if (object_mouse_select_menu( - C, &vc, gpu->buffer, gpu->hits, mval, params, &basact_override)) { - has_menu = true; - } - } - } - - /* Let the menu handle any further actions. */ - if (has_menu) { - if (gpu != NULL) { - MEM_freeN(gpu); - } - return false; - } - } - - /* No menu, continue with selection. */ - - ViewLayer *view_layer = vc.view_layer; - /* Don't set when the context has no active object (hidden), see: T60807. */ - const Base *oldbasact = vc.obact ? BASACT(view_layer) : NULL; - /* Always start list from `basact` when cycling the selection. */ - Base *startbase = (oldbasact && oldbasact->next) ? oldbasact->next : FIRSTBASE(view_layer); - - /* The next object's base to make active. */ - Base *basact = NULL; - const eObjectMode object_mode = oldbasact ? oldbasact->object->mode : OB_MODE_OBJECT; - - /* When enabled, don't attempt any further selection. */ - bool handled = false; - - /* Split `changed` into data-types so their associated updates can be properly performed. - * This is also needed as multiple changes may happen at once. - * Selecting a pose-bone or track can also select the object for e.g. */ - bool changed_object = false; - bool changed_pose = false; - bool changed_track = false; - - /* Handle setting the new base active (even when `handled == true`). */ - bool use_activate_selected_base = false; - - if (center) { - if (basact_override) { - basact = basact_override; - } - else { - basact = mouse_select_object_center(&vc, startbase, mval); - } - } - else { - if (basact_override) { - basact = basact_override; - } - else { - /* Regarding bone priority. - * - * - When in pose-bone, it's useful that any selection containing a bone - * gets priority over other geometry (background scenery for example). - * - * - When in object-mode, don't prioritize bones as it would cause - * pose-objects behind other objects to get priority - * (mainly noticeable when #SCE_OBJECT_MODE_LOCK is disabled). - * - * This way prioritizing based on pose-mode has a bias to stay in pose-mode - * without having to enforce this through locking the object mode. */ - bool do_bones_get_priotity = (object_mode & OB_MODE_POSE) != 0; - - basact = (gpu->hits > 0) ? mouse_select_eval_buffer(&vc, - gpu->buffer, - gpu->hits, - gpu->do_nearest, - gpu->has_bones, - do_bones_get_priotity, - NULL) : - NULL; - } - - /* Select pose-bones or camera-tracks. */ - if (((gpu->hits > 0) && gpu->has_bones) || - /* Special case, even when there are no hits, pose logic may de-select all bones. */ - ((gpu->hits == 0) && (object_mode & OB_MODE_POSE))) { - - if (basact && (gpu->has_bones && (basact->object->type == OB_CAMERA))) { - MovieClip *clip = BKE_object_movieclip_get(scene, basact->object, false); - if (clip != NULL) { - if (ed_object_select_pick_camera_track( - C, scene, basact, clip, gpu->buffer, gpu->hits, params)) { - ED_object_base_select(basact, BA_SELECT); - /* Don't set `handled` here as the object activation may be necessary. */ - changed_object = true; - - changed_track = true; - } - else { - /* Fallback to regular object selection if no new bundles were selected, - * allows to select object parented to reconstruction object. */ - basact = mouse_select_eval_buffer( - &vc, gpu->buffer, gpu->hits, gpu->do_nearest, false, false, NULL); - } - } - } - else if (ED_armature_pose_select_pick_with_buffer(view_layer, - v3d, - basact ? basact : (Base *)oldbasact, - gpu->buffer, - gpu->hits, - params, - gpu->do_nearest)) { - - changed_pose = true; - - /* When there is no `baseact` this will have operated on `oldbasact`, - * allowing #SelectPick_Params.deselect_all work in pose-mode. - * In this case no object operations are needed. */ - if (basact != NULL) { - /* By convention the armature-object is selected when in pose-mode. - * While leaving it unselected will work, leaving pose-mode would leave the object - * active + unselected which isn't ideal when performing other actions on the object. */ - ED_object_base_select(basact, BA_SELECT); - changed_object = true; - - WM_event_add_notifier(C, NC_OBJECT | ND_BONE_SELECT, basact->object); - WM_event_add_notifier(C, NC_OBJECT | ND_BONE_ACTIVE, basact->object); - - /* In weight-paint, we use selected bone to select vertex-group. - * In this case the active object mustn't change as it would leave weight-paint mode. */ - if (oldbasact) { - if (oldbasact->object->mode & OB_MODE_ALL_WEIGHT_PAINT) { - /* Prevent activating. - * Selection causes this to be considered the 'active' pose in weight-paint mode. - * Eventually this limitation may be removed. - * For now, de-select all other pose objects deforming this mesh. */ - ED_armature_pose_select_in_wpaint_mode(view_layer, basact); - - handled = true; - } - else if ((object_mode & OB_MODE_POSE) && (basact->object->mode & OB_MODE_POSE)) { - /* Within pose-mode, keep the current selection when switching pose bones, - * this is noticeable when in pose mode with multiple objects at once. - * Where selecting the bone of a different object would de-select this one. - * After that, exiting pose-mode would only have the active armature selected. - * This matches multi-object edit-mode behavior. */ - handled = true; - - if (oldbasact != basact) { - use_activate_selected_base = true; - } - } - else { - /* Don't set `handled` here as the object selection may be necessary - * when starting out in object-mode and moving into pose-mode, - * when moving from pose to object-mode using object selection also makes sense. */ - } - } - } - } - /* Prevent bone/track selecting to pass on to object selecting. */ - if (basact == oldbasact) { - handled = true; - } - } - } - - if (handled == false) { - if (scene->toolsettings->object_flag & SCE_OBJECT_MODE_LOCK) { - /* No special logic in edit-mode. */ - if (is_obedit == false) { - if (basact && !BKE_object_is_mode_compat(basact->object, object_mode)) { - if (object_mode == OB_MODE_OBJECT) { - struct Main *bmain = vc.bmain; - ED_object_mode_generic_exit(bmain, vc.depsgraph, scene, basact->object); - } - if (!BKE_object_is_mode_compat(basact->object, object_mode)) { - basact = NULL; - } - } - - /* Disallow switching modes, - * special exception for edit-mode - vertex-parent operator. */ - if (basact && oldbasact) { - if ((oldbasact->object->mode != basact->object->mode) && - (oldbasact->object->mode & basact->object->mode) == 0) { - basact = NULL; - } - } - } - } - } - - /* Ensure code above doesn't change the active base. This code is already fairly involved, - * it's best if changing the active object is localized to a single place. */ - BLI_assert(oldbasact == (vc.obact ? BASACT(view_layer) : NULL)); - - bool found = (basact != NULL); - if ((handled == false) && (vc.obedit == NULL)) { - /* Object-mode (pose mode will have been handled already). */ - if (params->sel_op == SEL_OP_SET) { - if ((found && params->select_passthrough) && (basact->flag & BASE_SELECTED)) { - found = false; - } - else if (found || params->deselect_all) { - /* Deselect everything. */ - /* `basact` may be NULL. */ - if (object_deselect_all_except(view_layer, basact)) { - changed_object = true; - } - } - } - } - - if ((handled == false) && found) { - - if (vc.obedit) { - /* Only do the select (use for setting vertex parents & hooks). - * In edit-mode do not activate. */ - object_deselect_all_except(view_layer, basact); - ED_object_base_select(basact, BA_SELECT); - - changed_object = true; - } - /* Also prevent making it active on mouse selection. */ - else if (BASE_SELECTABLE(v3d, basact)) { - use_activate_selected_base |= (oldbasact != basact) && (is_obedit == false); - - switch (params->sel_op) { - case SEL_OP_ADD: { - ED_object_base_select(basact, BA_SELECT); - break; - } - case SEL_OP_SUB: { - ED_object_base_select(basact, BA_DESELECT); - break; - } - case SEL_OP_XOR: { - if (basact->flag & BASE_SELECTED) { - /* Keep selected if the base is to be activated. */ - if (use_activate_selected_base == false) { - ED_object_base_select(basact, BA_DESELECT); - } - } - else { - ED_object_base_select(basact, BA_SELECT); - } - break; - } - case SEL_OP_SET: { - object_deselect_all_except(view_layer, basact); - ED_object_base_select(basact, BA_SELECT); - break; - } - case SEL_OP_AND: { - BLI_assert_unreachable(); /* Doesn't make sense for picking. */ - break; - } - } - - changed_object = true; - } - } - - /* Perform the activation even when 'handled', since this is used to ensure - * the object from the pose-bone selected is also activated. */ - if (use_activate_selected_base && (basact != NULL)) { - changed_object = true; - ED_object_base_activate(C, basact); /* adds notifier */ - if ((scene->toolsettings->object_flag & SCE_OBJECT_MODE_LOCK) == 0) { - WM_toolsystem_update_from_context_view3d(C); - } - } - - if (changed_object) { - DEG_id_tag_update(&scene->id, ID_RECALC_SELECT); - WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, scene); - - ED_outliner_select_sync_from_object_tag(C); - } - - if (changed_pose) { - ED_outliner_select_sync_from_pose_bone_tag(C); - } - - if (gpu != NULL) { - MEM_freeN(gpu); - } - - return (changed_object || changed_pose || changed_track); -} - -/** - * Mouse selection in weight paint. - * Called via generic mouse select operator. - * - * \return True when pick finds an element or the selection changed. - */ -static bool ed_wpaint_vertex_select_pick(bContext *C, - const int mval[2], - const struct SelectPick_Params *params, - Object *obact) -{ - View3D *v3d = CTX_wm_view3d(C); - const bool use_zbuf = !XRAY_ENABLED(v3d); - - Mesh *me = obact->data; /* already checked for NULL */ - uint index = 0; - MVert *mv; - bool changed = false; - - bool found = ED_mesh_pick_vert(C, obact, mval, ED_MESH_PICK_DEFAULT_VERT_DIST, use_zbuf, &index); - - if (params->sel_op == SEL_OP_SET) { - if ((found && params->select_passthrough) && (me->mvert[index].flag & SELECT)) { - found = false; - } - else if (found || params->deselect_all) { - /* Deselect everything. */ - changed |= paintface_deselect_all_visible(C, obact, SEL_DESELECT, false); - } - } - - if (found) { - mv = &me->mvert[index]; - switch (params->sel_op) { - case SEL_OP_ADD: { - mv->flag |= SELECT; - break; - } - case SEL_OP_SUB: { - mv->flag &= ~SELECT; - break; - } - case SEL_OP_XOR: { - mv->flag ^= SELECT; - break; - } - case SEL_OP_SET: { - paintvert_deselect_all_visible(obact, SEL_DESELECT, false); - mv->flag |= SELECT; - break; - } - case SEL_OP_AND: { - BLI_assert_unreachable(); /* Doesn't make sense for picking. */ - break; - } - } - - /* update mselect */ - if (mv->flag & SELECT) { - BKE_mesh_mselect_active_set(me, index, ME_VSEL); - } - else { - BKE_mesh_mselect_validate(me); - } - - paintvert_flush_flags(obact); - - changed = true; - } - - if (changed) { - paintvert_tag_select_update(C, obact); - } - - return changed || found; -} - -static int view3d_select_exec(bContext *C, wmOperator *op) -{ - Scene *scene = CTX_data_scene(C); - Object *obedit = CTX_data_edit_object(C); - Object *obact = CTX_data_active_object(C); - - struct SelectPick_Params params = {0}; - ED_select_pick_params_from_operator(op->ptr, ¶ms); - - const bool vert_without_handles = RNA_boolean_get(op->ptr, "vert_without_handles"); - bool center = RNA_boolean_get(op->ptr, "center"); - bool enumerate = RNA_boolean_get(op->ptr, "enumerate"); - /* Only force object select for edit-mode to support vertex parenting, - * or paint-select to allow pose bone select with vert/face select. */ - bool object_only = (RNA_boolean_get(op->ptr, "object") && - (obedit || BKE_paint_select_elem_test(obact) || - /* so its possible to select bones in weight-paint mode (LMB select) */ - (obact && (obact->mode & OB_MODE_ALL_WEIGHT_PAINT) && - BKE_object_pose_armature_get(obact)))); - - /* This could be called "changed_or_found" since this is true when there is an element - * under the cursor to select, even if it happens that the selection & active state doesn't - * actually change. This is important so undo pushes are predictable. */ - bool changed = false; - int mval[2]; - - if (object_only) { - obedit = NULL; - obact = NULL; - - /* ack, this is incorrect but to do this correctly we would need an - * alternative edit-mode/object-mode keymap, this copies the functionality - * from 2.4x where Ctrl+Select in edit-mode does object select only. */ - center = false; - } - - if (obedit && enumerate) { - /* Enumerate makes no sense in edit-mode unless also explicitly picking objects or bones. - * Pass the event through so the event may be handled by loop-select for e.g. see: T100204. */ - if (obedit->type != OB_ARMATURE) { - return OPERATOR_PASS_THROUGH | OPERATOR_CANCELLED; - } - } - - RNA_int_get_array(op->ptr, "location", mval); - - view3d_operator_needs_opengl(C); - BKE_object_update_select_id(CTX_data_main(C)); - - if (obedit && object_only == false) { - if (obedit->type == OB_MESH) { - changed = EDBM_select_pick(C, mval, ¶ms); - } - else if (obedit->type == OB_ARMATURE) { - if (enumerate) { - Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C); - ViewContext vc; - ED_view3d_viewcontext_init(C, &vc, depsgraph); - - GPUSelectResult buffer[MAXPICKELEMS]; - const int hits = mixed_bones_object_selectbuffer( - &vc, buffer, ARRAY_SIZE(buffer), mval, VIEW3D_SELECT_FILTER_NOP, false, true, false); - changed = bone_mouse_select_menu(C, buffer, hits, true, ¶ms); - } - if (!changed) { - changed = ED_armature_edit_select_pick(C, mval, ¶ms); - } - } - else if (obedit->type == OB_LATTICE) { - changed = ED_lattice_select_pick(C, mval, ¶ms); - } - else if (ELEM(obedit->type, OB_CURVES_LEGACY, OB_SURF)) { - changed = ED_curve_editnurb_select_pick( - C, mval, ED_view3d_select_dist_px(), vert_without_handles, ¶ms); - } - else if (obedit->type == OB_MBALL) { - changed = ED_mball_select_pick(C, mval, ¶ms); - } - else if (obedit->type == OB_FONT) { - changed = ED_curve_editfont_select_pick(C, mval, ¶ms); - } - } - else if (obact && obact->mode & OB_MODE_PARTICLE_EDIT) { - changed = PE_mouse_particles(C, mval, ¶ms); - } - else if (obact && BKE_paint_select_face_test(obact)) { - changed = paintface_mouse_select(C, mval, ¶ms, obact); - } - else if (BKE_paint_select_vert_test(obact)) { - changed = ed_wpaint_vertex_select_pick(C, mval, ¶ms, obact); - } - else { - changed = ed_object_select_pick(C, mval, ¶ms, center, enumerate, object_only); - } - - /* Pass-through flag may be cleared, see #WM_operator_flag_only_pass_through_on_press. */ - - /* Pass-through allows tweaks - * FINISHED to signal one operator worked */ - if (changed) { - WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, scene); - return OPERATOR_PASS_THROUGH | OPERATOR_FINISHED; - } - /* Nothing selected, just passthrough. */ - return OPERATOR_PASS_THROUGH | OPERATOR_CANCELLED; -} - -static int view3d_select_invoke(bContext *C, wmOperator *op, const wmEvent *event) -{ - RNA_int_set_array(op->ptr, "location", event->mval); - - const int retval = view3d_select_exec(C, op); - - return WM_operator_flag_only_pass_through_on_press(retval, event); -} - -void VIEW3D_OT_select(wmOperatorType *ot) -{ - PropertyRNA *prop; - - /* identifiers */ - ot->name = "Select"; - ot->description = "Select and activate item(s)"; - ot->idname = "VIEW3D_OT_select"; - - /* api callbacks */ - ot->invoke = view3d_select_invoke; - ot->exec = view3d_select_exec; - ot->poll = ED_operator_view3d_active; - ot->get_name = ED_select_pick_get_name; - - /* flags */ - ot->flag = OPTYPE_UNDO; - - /* properties */ - WM_operator_properties_mouse_select(ot); - - prop = RNA_def_boolean( - ot->srna, - "center", - 0, - "Center", - "Use the object center when selecting, in edit mode used to extend object selection"); - RNA_def_property_flag(prop, PROP_SKIP_SAVE); - prop = RNA_def_boolean( - ot->srna, "enumerate", 0, "Enumerate", "List objects under the mouse (object mode only)"); - RNA_def_property_flag(prop, PROP_SKIP_SAVE); - prop = RNA_def_boolean(ot->srna, "object", 0, "Object", "Use object selection (edit mode only)"); - RNA_def_property_flag(prop, PROP_SKIP_SAVE); - - /* Needed for select-through to usefully drag handles, see: T98254. - * NOTE: this option may be removed and become default behavior, see design task: T98552. */ - prop = RNA_def_boolean(ot->srna, - "vert_without_handles", - 0, - "Control Point Without Handles", - "Only select the curve control point, not it's handles"); - RNA_def_property_flag(prop, PROP_SKIP_SAVE); - - prop = RNA_def_int_vector(ot->srna, - "location", - 2, - NULL, - INT_MIN, - INT_MAX, - "Location", - "Mouse location", - INT_MIN, - INT_MAX); - RNA_def_property_flag(prop, PROP_HIDDEN); -} - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Box Select - * \{ */ - -typedef struct BoxSelectUserData { - ViewContext *vc; - const rcti *rect; - const rctf *rect_fl; - rctf _rect_fl; - eSelectOp sel_op; - eBezTriple_Flag select_flag; - - /* runtime */ - bool is_done; - bool is_changed; -} BoxSelectUserData; - -static void view3d_userdata_boxselect_init(BoxSelectUserData *r_data, - ViewContext *vc, - const rcti *rect, - const eSelectOp sel_op) -{ - r_data->vc = vc; - - r_data->rect = rect; - r_data->rect_fl = &r_data->_rect_fl; - BLI_rctf_rcti_copy(&r_data->_rect_fl, rect); - - r_data->sel_op = sel_op; - /* SELECT by default, but can be changed if needed (only few cases use and respect this). */ - r_data->select_flag = SELECT; - - /* runtime */ - r_data->is_done = false; - r_data->is_changed = false; -} - -bool edge_inside_circle(const float cent[2], - float radius, - const float screen_co_a[2], - const float screen_co_b[2]) -{ - const float radius_squared = radius * radius; - return (dist_squared_to_line_segment_v2(cent, screen_co_a, screen_co_b) < radius_squared); -} - -static void do_paintvert_box_select__doSelectVert(void *userData, - MVert *mv, - const float screen_co[2], - int UNUSED(index)) -{ - BoxSelectUserData *data = userData; - const bool is_select = mv->flag & SELECT; - const bool is_inside = BLI_rctf_isect_pt_v(data->rect_fl, screen_co); - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(mv->flag, sel_op_result, SELECT); - data->is_changed = true; - } -} -static bool do_paintvert_box_select(ViewContext *vc, - wmGenericUserData *wm_userdata, - const rcti *rect, - const eSelectOp sel_op) -{ - const bool use_zbuf = !XRAY_ENABLED(vc->v3d); - - Mesh *me; - - me = vc->obact->data; - if ((me == NULL) || (me->totvert == 0)) { - return OPERATOR_CANCELLED; - } - - bool changed = false; - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - changed |= paintvert_deselect_all_visible(vc->obact, SEL_DESELECT, false); - } - - if (BLI_rcti_is_empty(rect)) { - /* pass */ - } - else if (use_zbuf) { - struct EditSelectBuf_Cache *esel = wm_userdata->data; - if (wm_userdata->data == NULL) { - editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, SCE_SELECT_VERTEX); - esel = wm_userdata->data; - esel->select_bitmap = DRW_select_buffer_bitmap_from_rect( - vc->depsgraph, vc->region, vc->v3d, rect, NULL); - } - if (esel->select_bitmap != NULL) { - changed |= edbm_backbuf_check_and_select_verts_obmode(me, esel, sel_op); - } - } - else { - BoxSelectUserData data; - - view3d_userdata_boxselect_init(&data, vc, rect, sel_op); - - ED_view3d_init_mats_rv3d(vc->obact, vc->rv3d); - - meshobject_foreachScreenVert( - vc, do_paintvert_box_select__doSelectVert, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - changed |= data.is_changed; - } - - if (changed) { - if (SEL_OP_CAN_DESELECT(sel_op)) { - BKE_mesh_mselect_validate(me); - } - paintvert_flush_flags(vc->obact); - paintvert_tag_select_update(vc->C, vc->obact); - } - return changed; -} - -static bool do_paintface_box_select(ViewContext *vc, - wmGenericUserData *wm_userdata, - const rcti *rect, - int sel_op) -{ - Object *ob = vc->obact; - Mesh *me; - - me = BKE_mesh_from_object(ob); - if ((me == NULL) || (me->totpoly == 0)) { - return false; - } - - bool changed = false; - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - changed |= paintface_deselect_all_visible(vc->C, vc->obact, SEL_DESELECT, false); - } - - if (BLI_rcti_is_empty(rect)) { - /* pass */ - } - else { - struct EditSelectBuf_Cache *esel = wm_userdata->data; - if (wm_userdata->data == NULL) { - editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, SCE_SELECT_FACE); - esel = wm_userdata->data; - esel->select_bitmap = DRW_select_buffer_bitmap_from_rect( - vc->depsgraph, vc->region, vc->v3d, rect, NULL); - } - if (esel->select_bitmap != NULL) { - changed |= edbm_backbuf_check_and_select_faces_obmode(me, esel, sel_op); - } - } - - if (changed) { - paintface_flush_flags(vc->C, vc->obact, true, false); - } - return changed; -} - -static void do_nurbs_box_select__doSelect(void *userData, - Nurb *UNUSED(nu), - BPoint *bp, - BezTriple *bezt, - int beztindex, - bool handles_visible, - const float screen_co[2]) -{ - BoxSelectUserData *data = userData; - - const bool is_inside = BLI_rctf_isect_pt_v(data->rect_fl, screen_co); - if (bp) { - const bool is_select = bp->f1 & SELECT; - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(bp->f1, sel_op_result, data->select_flag); - data->is_changed = true; - } - } - else { - if (!handles_visible) { - /* can only be (beztindex == 1) here since handles are hidden */ - const bool is_select = bezt->f2 & SELECT; - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(bezt->f2, sel_op_result, data->select_flag); - data->is_changed = true; - } - bezt->f1 = bezt->f3 = bezt->f2; - } - else { - uint8_t *flag_p = (&bezt->f1) + beztindex; - const bool is_select = *flag_p & SELECT; - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(*flag_p, sel_op_result, data->select_flag); - data->is_changed = true; - } - } - } -} -static bool do_nurbs_box_select(ViewContext *vc, rcti *rect, const eSelectOp sel_op) -{ - const bool deselect_all = (sel_op == SEL_OP_SET); - BoxSelectUserData data; - - view3d_userdata_boxselect_init(&data, vc, rect, sel_op); - - Curve *curve = (Curve *)vc->obedit->data; - ListBase *nurbs = BKE_curve_editNurbs_get(curve); - - /* For deselect all, items to be selected are tagged with temp flag. Clear that first. */ - if (deselect_all) { - BKE_nurbList_flag_set(nurbs, BEZT_FLAG_TEMP_TAG, false); - data.select_flag = BEZT_FLAG_TEMP_TAG; - } - - ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); /* for foreach's screen/vert projection */ - nurbs_foreachScreenVert(vc, do_nurbs_box_select__doSelect, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - - /* Deselect items that were not added to selection (indicated by temp flag). */ - if (deselect_all) { - data.is_changed |= BKE_nurbList_flag_set_from_flag(nurbs, BEZT_FLAG_TEMP_TAG, SELECT); - } - - BKE_curve_nurb_vert_active_validate(vc->obedit->data); - - return data.is_changed; -} - -static void do_lattice_box_select__doSelect(void *userData, BPoint *bp, const float screen_co[2]) -{ - BoxSelectUserData *data = userData; - const bool is_select = bp->f1 & SELECT; - const bool is_inside = BLI_rctf_isect_pt_v(data->rect_fl, screen_co); - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(bp->f1, sel_op_result, SELECT); - data->is_changed = true; - } -} -static bool do_lattice_box_select(ViewContext *vc, rcti *rect, const eSelectOp sel_op) -{ - BoxSelectUserData data; - - view3d_userdata_boxselect_init(&data, vc, rect, sel_op); - - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - data.is_changed |= ED_lattice_flags_set(vc->obedit, 0); - } - - ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); /* for foreach's screen/vert projection */ - lattice_foreachScreenVert( - vc, do_lattice_box_select__doSelect, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - - return data.is_changed; -} - -static void do_mesh_box_select__doSelectVert(void *userData, - BMVert *eve, - const float screen_co[2], - int UNUSED(index)) -{ - BoxSelectUserData *data = userData; - const bool is_select = BM_elem_flag_test(eve, BM_ELEM_SELECT); - const bool is_inside = BLI_rctf_isect_pt_v(data->rect_fl, screen_co); - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - BM_vert_select_set(data->vc->em->bm, eve, sel_op_result); - data->is_changed = true; - } -} -struct BoxSelectUserData_ForMeshEdge { - BoxSelectUserData *data; - struct EditSelectBuf_Cache *esel; - uint backbuf_offset; -}; -/** - * Pass 0 operates on edges when fully inside. - */ -static void do_mesh_box_select__doSelectEdge_pass0( - void *userData, BMEdge *eed, const float screen_co_a[2], const float screen_co_b[2], int index) -{ - struct BoxSelectUserData_ForMeshEdge *data_for_edge = userData; - BoxSelectUserData *data = data_for_edge->data; - bool is_visible = true; - if (data_for_edge->backbuf_offset) { - uint bitmap_inedx = data_for_edge->backbuf_offset + index - 1; - is_visible = BLI_BITMAP_TEST_BOOL(data_for_edge->esel->select_bitmap, bitmap_inedx); - } - - const bool is_select = BM_elem_flag_test(eed, BM_ELEM_SELECT); - const bool is_inside = (is_visible && - edge_fully_inside_rect(data->rect_fl, screen_co_a, screen_co_b)); - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - BM_edge_select_set(data->vc->em->bm, eed, sel_op_result); - data->is_done = true; - data->is_changed = true; - } -} -/** - * Pass 1 operates on edges when partially inside. - */ -static void do_mesh_box_select__doSelectEdge_pass1( - void *userData, BMEdge *eed, const float screen_co_a[2], const float screen_co_b[2], int index) -{ - struct BoxSelectUserData_ForMeshEdge *data_for_edge = userData; - BoxSelectUserData *data = data_for_edge->data; - bool is_visible = true; - if (data_for_edge->backbuf_offset) { - uint bitmap_inedx = data_for_edge->backbuf_offset + index - 1; - is_visible = BLI_BITMAP_TEST_BOOL(data_for_edge->esel->select_bitmap, bitmap_inedx); - } - - const bool is_select = BM_elem_flag_test(eed, BM_ELEM_SELECT); - const bool is_inside = (is_visible && edge_inside_rect(data->rect_fl, screen_co_a, screen_co_b)); - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - BM_edge_select_set(data->vc->em->bm, eed, sel_op_result); - data->is_changed = true; - } -} -static void do_mesh_box_select__doSelectFace(void *userData, - BMFace *efa, - const float screen_co[2], - int UNUSED(index)) -{ - BoxSelectUserData *data = userData; - const bool is_select = BM_elem_flag_test(efa, BM_ELEM_SELECT); - const bool is_inside = BLI_rctf_isect_pt_v(data->rect_fl, screen_co); - const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); - if (sel_op_result != -1) { - BM_face_select_set(data->vc->em->bm, efa, sel_op_result); - data->is_changed = true; - } -} -static bool do_mesh_box_select(ViewContext *vc, - wmGenericUserData *wm_userdata, - const rcti *rect, - const eSelectOp sel_op) -{ - BoxSelectUserData data; - ToolSettings *ts = vc->scene->toolsettings; - - view3d_userdata_boxselect_init(&data, vc, rect, sel_op); - - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - if (vc->em->bm->totvertsel) { - EDBM_flag_disable_all(vc->em, BM_ELEM_SELECT); - data.is_changed = true; - } - } - - /* for non zbuf projections, don't change the GL state */ - ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); - - GPU_matrix_set(vc->rv3d->viewmat); - - const bool use_zbuf = !XRAY_FLAG_ENABLED(vc->v3d); - - struct EditSelectBuf_Cache *esel = wm_userdata->data; - if (use_zbuf) { - if (wm_userdata->data == NULL) { - editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, ts->selectmode); - esel = wm_userdata->data; - esel->select_bitmap = DRW_select_buffer_bitmap_from_rect( - vc->depsgraph, vc->region, vc->v3d, rect, NULL); - } - } - - if (ts->selectmode & SCE_SELECT_VERTEX) { - if (use_zbuf) { - data.is_changed |= edbm_backbuf_check_and_select_verts( - esel, vc->depsgraph, vc->obedit, vc->em, sel_op); - } - else { - mesh_foreachScreenVert( - vc, do_mesh_box_select__doSelectVert, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - } - } - if (ts->selectmode & SCE_SELECT_EDGE) { - /* Does both use_zbuf and non-use_zbuf versions (need screen cos for both) */ - struct BoxSelectUserData_ForMeshEdge cb_data = { - .data = &data, - .esel = use_zbuf ? esel : NULL, - .backbuf_offset = use_zbuf ? DRW_select_buffer_context_offset_for_object_elem( - vc->depsgraph, vc->obedit, SCE_SELECT_EDGE) : - 0, - }; - - const eV3DProjTest clip_flag = V3D_PROJ_TEST_CLIP_NEAR | - (use_zbuf ? 0 : V3D_PROJ_TEST_CLIP_BB); - /* Fully inside. */ - mesh_foreachScreenEdge_clip_bb_segment( - vc, do_mesh_box_select__doSelectEdge_pass0, &cb_data, clip_flag); - if (data.is_done == false) { - /* Fall back to partially inside. - * Clip content to account for edges partially behind the view. */ - mesh_foreachScreenEdge_clip_bb_segment(vc, - do_mesh_box_select__doSelectEdge_pass1, - &cb_data, - clip_flag | V3D_PROJ_TEST_CLIP_CONTENT_DEFAULT); - } - } - - if (ts->selectmode & SCE_SELECT_FACE) { - if (use_zbuf) { - data.is_changed |= edbm_backbuf_check_and_select_faces( - esel, vc->depsgraph, vc->obedit, vc->em, sel_op); - } - else { - mesh_foreachScreenFace( - vc, do_mesh_box_select__doSelectFace, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - } - } - - if (data.is_changed) { - EDBM_selectmode_flush(vc->em); - } - return data.is_changed; -} - -static bool do_meta_box_select(ViewContext *vc, const rcti *rect, const eSelectOp sel_op) -{ - Object *ob = vc->obedit; - MetaBall *mb = (MetaBall *)ob->data; - MetaElem *ml; - int a; - bool changed = false; - - GPUSelectResult buffer[MAXPICKELEMS]; - int hits; - - hits = view3d_opengl_select( - vc, buffer, MAXPICKELEMS, rect, VIEW3D_SELECT_ALL, VIEW3D_SELECT_FILTER_NOP); - - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - changed |= BKE_mball_deselect_all(mb); - } - - int metaelem_id = 0; - for (ml = mb->editelems->first; ml; ml = ml->next, metaelem_id += 0x10000) { - bool is_inside_radius = false; - bool is_inside_stiff = false; - - for (a = 0; a < hits; a++) { - const int hitresult = buffer[a].id; - - if (hitresult == -1) { - continue; - } - - const uint hit_object = hitresult & 0xFFFF; - if (vc->obedit->runtime.select_id != hit_object) { - continue; - } - - if (metaelem_id != (hitresult & 0xFFFF0000 & ~MBALLSEL_ANY)) { - continue; - } - - if (hitresult & MBALLSEL_RADIUS) { - is_inside_radius = true; - break; - } - - if (hitresult & MBALLSEL_STIFF) { - is_inside_stiff = true; - break; - } - } - const int flag_prev = ml->flag; - if (is_inside_radius) { - ml->flag |= MB_SCALE_RAD; - } - if (is_inside_stiff) { - ml->flag &= ~MB_SCALE_RAD; - } - - const bool is_select = (ml->flag & SELECT); - const bool is_inside = is_inside_radius || is_inside_stiff; - - const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); - if (sel_op_result != -1) { - SET_FLAG_FROM_TEST(ml->flag, sel_op_result, SELECT); - } - changed |= (flag_prev != ml->flag); - } - - return changed; -} - -static bool do_armature_box_select(ViewContext *vc, const rcti *rect, const eSelectOp sel_op) -{ - bool changed = false; - int a; - - GPUSelectResult buffer[MAXPICKELEMS]; - int hits; - - hits = view3d_opengl_select( - vc, buffer, MAXPICKELEMS, rect, VIEW3D_SELECT_ALL, VIEW3D_SELECT_FILTER_NOP); - - uint bases_len = 0; - Base **bases = BKE_view_layer_array_from_bases_in_edit_mode_unique_data( - vc->view_layer, vc->v3d, &bases_len); - - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - changed |= ED_armature_edit_deselect_all_visible_multi_ex(bases, bases_len); - } - - for (uint base_index = 0; base_index < bases_len; base_index++) { - Object *obedit = bases[base_index]->object; - obedit->id.tag &= ~LIB_TAG_DOIT; - - bArmature *arm = obedit->data; - ED_armature_ebone_listbase_temp_clear(arm->edbo); - } - - /* first we only check points inside the border */ - for (a = 0; a < hits; a++) { - const int select_id = buffer[a].id; - if (select_id != -1) { - if ((select_id & 0xFFFF0000) == 0) { - continue; - } - - EditBone *ebone; - Base *base_edit = ED_armature_base_and_ebone_from_select_buffer( - bases, bases_len, select_id, &ebone); - ebone->temp.i |= select_id & BONESEL_ANY; - base_edit->object->id.tag |= LIB_TAG_DOIT; - } - } - - for (uint base_index = 0; base_index < bases_len; base_index++) { - Object *obedit = bases[base_index]->object; - if (obedit->id.tag & LIB_TAG_DOIT) { - obedit->id.tag &= ~LIB_TAG_DOIT; - changed |= ED_armature_edit_select_op_from_tagged(obedit->data, sel_op); - } - } - - MEM_freeN(bases); - - return changed; -} - -/** - * Compare result of 'GPU_select': 'GPUSelectResult', - * needed for when we need to align with object draw-order. - */ -static int opengl_bone_select_buffer_cmp(const void *sel_a_p, const void *sel_b_p) -{ - uint sel_a = ((GPUSelectResult *)sel_a_p)->id; - uint sel_b = ((GPUSelectResult *)sel_b_p)->id; - -#ifdef __BIG_ENDIAN__ - BLI_endian_switch_uint32(&sel_a); - BLI_endian_switch_uint32(&sel_b); -#endif - - if (sel_a < sel_b) { - return -1; - } - if (sel_a > sel_b) { - return 1; - } - return 0; -} - -static bool do_object_box_select(bContext *C, ViewContext *vc, rcti *rect, const eSelectOp sel_op) -{ - View3D *v3d = vc->v3d; - int totobj = MAXPICKELEMS; /* XXX solve later */ - - /* Selection buffer has bones potentially too, so we add #MAXPICKELEMS. */ - GPUSelectResult *buffer = MEM_mallocN((totobj + MAXPICKELEMS) * sizeof(GPUSelectResult), - "selection buffer"); - const eV3DSelectObjectFilter select_filter = ED_view3d_select_filter_from_mode(vc->scene, - vc->obact); - const int hits = view3d_opengl_select( - vc, buffer, (totobj + MAXPICKELEMS), rect, VIEW3D_SELECT_ALL, select_filter); - - LISTBASE_FOREACH (Base *, base, &vc->view_layer->object_bases) { - base->object->id.tag &= ~LIB_TAG_DOIT; - } - - Base **bases = NULL; - BLI_array_declare(bases); - - bool changed = false; - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - changed |= object_deselect_all_visible(vc->view_layer, vc->v3d); - } - - if ((hits == -1) && !SEL_OP_USE_OUTSIDE(sel_op)) { - goto finally; - } - - LISTBASE_FOREACH (Base *, base, &vc->view_layer->object_bases) { - if (BASE_SELECTABLE(v3d, base)) { - if ((base->object->runtime.select_id & 0x0000FFFF) != 0) { - BLI_array_append(bases, base); - } - } - } - - /* The draw order doesn't always match the order we populate the engine, see: T51695. */ - qsort(buffer, hits, sizeof(GPUSelectResult), opengl_bone_select_buffer_cmp); - - for (const GPUSelectResult *buf_iter = buffer, *buf_end = buf_iter + hits; buf_iter < buf_end; - buf_iter++) { - bPoseChannel *pchan_dummy; - Base *base = ED_armature_base_and_pchan_from_select_buffer( - bases, BLI_array_len(bases), buf_iter->id, &pchan_dummy); - if (base != NULL) { - base->object->id.tag |= LIB_TAG_DOIT; - } - } - - for (Base *base = vc->view_layer->object_bases.first; base && hits; base = base->next) { - if (BASE_SELECTABLE(v3d, base)) { - const bool is_select = base->flag & BASE_SELECTED; - const bool is_inside = base->object->id.tag & LIB_TAG_DOIT; - const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); - if (sel_op_result != -1) { - ED_object_base_select(base, sel_op_result ? BA_SELECT : BA_DESELECT); - changed = true; - } - } - } - -finally: - if (bases != NULL) { - MEM_freeN(bases); - } - - MEM_freeN(buffer); - - if (changed) { - DEG_id_tag_update(&vc->scene->id, ID_RECALC_SELECT); - WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, vc->scene); - } - return changed; -} - -static bool do_pose_box_select(bContext *C, ViewContext *vc, rcti *rect, const eSelectOp sel_op) -{ - uint bases_len; - Base **bases = do_pose_tag_select_op_prepare(vc, &bases_len); - - int totobj = MAXPICKELEMS; /* XXX solve later */ - - /* Selection buffer has bones potentially too, so add #MAXPICKELEMS. */ - GPUSelectResult *buffer = MEM_mallocN((totobj + MAXPICKELEMS) * sizeof(GPUSelectResult), - "selection buffer"); - const eV3DSelectObjectFilter select_filter = ED_view3d_select_filter_from_mode(vc->scene, - vc->obact); - const int hits = view3d_opengl_select( - vc, buffer, (totobj + MAXPICKELEMS), rect, VIEW3D_SELECT_ALL, select_filter); - /* - * LOGIC NOTES (theeth): - * The buffer and ListBase have the same relative order, which makes the selection - * very simple. Loop through both data sets at the same time, if the color - * is the same as the object, we have a hit and can move to the next color - * and object pair, if not, just move to the next object, - * keeping the same color until we have a hit. - */ - - if (hits > 0) { - /* no need to loop if there's no hit */ - - /* The draw order doesn't always match the order we populate the engine, see: T51695. */ - qsort(buffer, hits, sizeof(GPUSelectResult), opengl_bone_select_buffer_cmp); - - for (const GPUSelectResult *buf_iter = buffer, *buf_end = buf_iter + hits; buf_iter < buf_end; - buf_iter++) { - Bone *bone; - Base *base = ED_armature_base_and_bone_from_select_buffer( - bases, bases_len, buf_iter->id, &bone); - - if (base == NULL) { - continue; - } - - /* Loop over contiguous bone hits for 'base'. */ - for (; buf_iter != buf_end; buf_iter++) { - /* should never fail */ - if (bone != NULL) { - base->object->id.tag |= LIB_TAG_DOIT; - bone->flag |= BONE_DONE; - } - - /* Select the next bone if we're not switching bases. */ - if (buf_iter + 1 != buf_end) { - const GPUSelectResult *col_next = buf_iter + 1; - if ((base->object->runtime.select_id & 0x0000FFFF) != (col_next->id & 0x0000FFFF)) { - break; - } - if (base->object->pose != NULL) { - const uint hit_bone = (col_next->id & ~BONESEL_ANY) >> 16; - bPoseChannel *pchan = BLI_findlink(&base->object->pose->chanbase, hit_bone); - bone = pchan ? pchan->bone : NULL; - } - else { - bone = NULL; - } - } - } - } - } - - const bool changed_multi = do_pose_tag_select_op_exec(bases, bases_len, sel_op); - if (changed_multi) { - DEG_id_tag_update(&vc->scene->id, ID_RECALC_SELECT); - WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, vc->scene); - } - - if (bases != NULL) { - MEM_freeN(bases); - } - MEM_freeN(buffer); - - return changed_multi; -} - -static int view3d_box_select_exec(bContext *C, wmOperator *op) -{ - Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C); - ViewContext vc; - rcti rect; - bool changed_multi = false; - - wmGenericUserData wm_userdata_buf = {0}; - wmGenericUserData *wm_userdata = &wm_userdata_buf; - - view3d_operator_needs_opengl(C); - BKE_object_update_select_id(CTX_data_main(C)); - - /* setup view context for argument to callbacks */ - ED_view3d_viewcontext_init(C, &vc, depsgraph); - - eSelectOp sel_op = RNA_enum_get(op->ptr, "mode"); - WM_operator_properties_border_to_rcti(op, &rect); - - if (vc.obedit) { - FOREACH_OBJECT_IN_MODE_BEGIN ( - vc.view_layer, vc.v3d, vc.obedit->type, vc.obedit->mode, ob_iter) { - ED_view3d_viewcontext_init_object(&vc, ob_iter); - bool changed = false; - - switch (vc.obedit->type) { - case OB_MESH: - vc.em = BKE_editmesh_from_object(vc.obedit); - changed = do_mesh_box_select(&vc, wm_userdata, &rect, sel_op); - if (changed) { - DEG_id_tag_update(vc.obedit->data, ID_RECALC_SELECT); - WM_event_add_notifier(C, NC_GEOM | ND_SELECT, vc.obedit->data); - } - break; - case OB_CURVES_LEGACY: - case OB_SURF: - changed = do_nurbs_box_select(&vc, &rect, sel_op); - if (changed) { - DEG_id_tag_update(vc.obedit->data, ID_RECALC_SELECT); - WM_event_add_notifier(C, NC_GEOM | ND_SELECT, vc.obedit->data); - } - break; - case OB_MBALL: - changed = do_meta_box_select(&vc, &rect, sel_op); - if (changed) { - DEG_id_tag_update(vc.obedit->data, ID_RECALC_SELECT); - WM_event_add_notifier(C, NC_GEOM | ND_SELECT, vc.obedit->data); - } - break; - case OB_ARMATURE: - changed = do_armature_box_select(&vc, &rect, sel_op); - if (changed) { - DEG_id_tag_update(&vc.obedit->id, ID_RECALC_SELECT); - WM_event_add_notifier(C, NC_OBJECT | ND_BONE_SELECT, vc.obedit); - ED_outliner_select_sync_from_edit_bone_tag(C); - } - break; - case OB_LATTICE: - changed = do_lattice_box_select(&vc, &rect, sel_op); - if (changed) { - DEG_id_tag_update(vc.obedit->data, ID_RECALC_SELECT); - WM_event_add_notifier(C, NC_GEOM | ND_SELECT, vc.obedit->data); - } - break; - default: - BLI_assert_msg(0, "box select on incorrect object type"); - break; - } - changed_multi |= changed; - } - FOREACH_OBJECT_IN_MODE_END; - } - else { /* No edit-mode, unified for bones and objects. */ - if (vc.obact && BKE_paint_select_face_test(vc.obact)) { - changed_multi = do_paintface_box_select(&vc, wm_userdata, &rect, sel_op); - } - else if (vc.obact && BKE_paint_select_vert_test(vc.obact)) { - changed_multi = do_paintvert_box_select(&vc, wm_userdata, &rect, sel_op); - } - else if (vc.obact && vc.obact->mode & OB_MODE_PARTICLE_EDIT) { - changed_multi = PE_box_select(C, &rect, sel_op); - } - else if (vc.obact && vc.obact->mode & OB_MODE_POSE) { - changed_multi = do_pose_box_select(C, &vc, &rect, sel_op); - if (changed_multi) { - ED_outliner_select_sync_from_pose_bone_tag(C); - } - } - else { /* object mode with none active */ - changed_multi = do_object_box_select(C, &vc, &rect, sel_op); - if (changed_multi) { - ED_outliner_select_sync_from_object_tag(C); - } - } - } - - WM_generic_user_data_free(wm_userdata); - - if (changed_multi) { - return OPERATOR_FINISHED; - } - return OPERATOR_CANCELLED; -} - -void VIEW3D_OT_select_box(wmOperatorType *ot) -{ - /* identifiers */ - ot->name = "Box Select"; - ot->description = "Select items using box selection"; - ot->idname = "VIEW3D_OT_select_box"; - - /* api callbacks */ - ot->invoke = WM_gesture_box_invoke; - ot->exec = view3d_box_select_exec; - ot->modal = WM_gesture_box_modal; - ot->poll = view3d_selectable_data; - ot->cancel = WM_gesture_box_cancel; - - /* flags */ - ot->flag = OPTYPE_UNDO; - - /* rna */ - WM_operator_properties_gesture_box(ot); - WM_operator_properties_select_operation(ot); -} - -/** \} */ - -/* -------------------------------------------------------------------- */ -/** \name Circle Select - * \{ */ - -typedef struct CircleSelectUserData { - ViewContext *vc; - bool select; - int mval[2]; - float mval_fl[2]; - float radius; - float radius_squared; - eBezTriple_Flag select_flag; - - /* runtime */ - bool is_changed; -} CircleSelectUserData; - -static void view3d_userdata_circleselect_init(CircleSelectUserData *r_data, - ViewContext *vc, - const bool select, - const int mval[2], - const float rad) -{ - r_data->vc = vc; - r_data->select = select; - copy_v2_v2_int(r_data->mval, mval); - r_data->mval_fl[0] = mval[0]; - r_data->mval_fl[1] = mval[1]; - - r_data->radius = rad; - r_data->radius_squared = rad * rad; - - /* SELECT by default, but can be changed if needed (only few cases use and respect this). */ - r_data->select_flag = SELECT; - - /* runtime */ - r_data->is_changed = false; -} - -static void mesh_circle_doSelectVert(void *userData, - BMVert *eve, - const float screen_co[2], - int UNUSED(index)) -{ - CircleSelectUserData *data = userData; - - if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { - BM_vert_select_set(data->vc->em->bm, eve, data->select); - data->is_changed = true; - } -} -static void mesh_circle_doSelectEdge(void *userData, - BMEdge *eed, - const float screen_co_a[2], - const float screen_co_b[2], - int UNUSED(index)) -{ - CircleSelectUserData *data = userData; - - if (edge_inside_circle(data->mval_fl, data->radius, screen_co_a, screen_co_b)) { - BM_edge_select_set(data->vc->em->bm, eed, data->select); - data->is_changed = true; - } -} -static void mesh_circle_doSelectFace(void *userData, - BMFace *efa, - const float screen_co[2], - int UNUSED(index)) -{ - CircleSelectUserData *data = userData; - - if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { - BM_face_select_set(data->vc->em->bm, efa, data->select); - data->is_changed = true; - } -} - -static bool mesh_circle_select(ViewContext *vc, - wmGenericUserData *wm_userdata, - eSelectOp sel_op, - const int mval[2], - float rad) -{ - ToolSettings *ts = vc->scene->toolsettings; - CircleSelectUserData data; - vc->em = BKE_editmesh_from_object(vc->obedit); - - bool changed = false; - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - if (vc->em->bm->totvertsel) { - EDBM_flag_disable_all(vc->em, BM_ELEM_SELECT); - vc->em->bm->totvertsel = 0; - vc->em->bm->totedgesel = 0; - vc->em->bm->totfacesel = 0; - changed = true; - } - } - const bool select = (sel_op != SEL_OP_SUB); - - ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); /* for foreach's screen/vert projection */ - - view3d_userdata_circleselect_init(&data, vc, select, mval, rad); - - const bool use_zbuf = !XRAY_FLAG_ENABLED(vc->v3d); - - if (use_zbuf) { - if (wm_userdata->data == NULL) { - editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, ts->selectmode); - } - } - struct EditSelectBuf_Cache *esel = wm_userdata->data; - - if (use_zbuf) { - if (esel->select_bitmap == NULL) { - esel->select_bitmap = DRW_select_buffer_bitmap_from_circle( - vc->depsgraph, vc->region, vc->v3d, mval, (int)(rad + 1.0f), NULL); - } - } - - if (ts->selectmode & SCE_SELECT_VERTEX) { - if (use_zbuf) { - if (esel->select_bitmap != NULL) { - changed |= edbm_backbuf_check_and_select_verts( - esel, vc->depsgraph, vc->obedit, vc->em, select ? SEL_OP_ADD : SEL_OP_SUB); - } - } - else { - mesh_foreachScreenVert(vc, mesh_circle_doSelectVert, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - } - } - - if (ts->selectmode & SCE_SELECT_EDGE) { - if (use_zbuf) { - if (esel->select_bitmap != NULL) { - changed |= edbm_backbuf_check_and_select_edges( - esel, vc->depsgraph, vc->obedit, vc->em, select ? SEL_OP_ADD : SEL_OP_SUB); - } - } - else { - mesh_foreachScreenEdge_clip_bb_segment( - vc, - mesh_circle_doSelectEdge, - &data, - (V3D_PROJ_TEST_CLIP_NEAR | V3D_PROJ_TEST_CLIP_BB | V3D_PROJ_TEST_CLIP_CONTENT_DEFAULT)); - } - } - - if (ts->selectmode & SCE_SELECT_FACE) { - if (use_zbuf) { - if (esel->select_bitmap != NULL) { - changed |= edbm_backbuf_check_and_select_faces( - esel, vc->depsgraph, vc->obedit, vc->em, select ? SEL_OP_ADD : SEL_OP_SUB); - } - } - else { - mesh_foreachScreenFace(vc, mesh_circle_doSelectFace, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - } - } - - changed |= data.is_changed; - - if (changed) { - BM_mesh_select_mode_flush_ex( - vc->em->bm, vc->em->selectmode, BM_SELECT_LEN_FLUSH_RECALC_NOTHING); - } - return changed; -} - -static bool paint_facesel_circle_select(ViewContext *vc, - wmGenericUserData *wm_userdata, - const eSelectOp sel_op, - const int mval[2], - float rad) -{ - BLI_assert(ELEM(sel_op, SEL_OP_SET, SEL_OP_ADD, SEL_OP_SUB)); - Object *ob = vc->obact; - Mesh *me = ob->data; - - bool changed = false; - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - /* flush selection at the end */ - changed |= paintface_deselect_all_visible(vc->C, ob, SEL_DESELECT, false); - } - - if (wm_userdata->data == NULL) { - editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, SCE_SELECT_FACE); - } - - { - struct EditSelectBuf_Cache *esel = wm_userdata->data; - esel->select_bitmap = DRW_select_buffer_bitmap_from_circle( - vc->depsgraph, vc->region, vc->v3d, mval, (int)(rad + 1.0f), NULL); - if (esel->select_bitmap != NULL) { - changed |= edbm_backbuf_check_and_select_faces_obmode(me, esel, sel_op); - MEM_freeN(esel->select_bitmap); - esel->select_bitmap = NULL; - } - } - - if (changed) { - paintface_flush_flags(vc->C, ob, true, false); - } - return changed; -} - -static void paint_vertsel_circle_select_doSelectVert(void *userData, - MVert *mv, - const float screen_co[2], - int UNUSED(index)) -{ - CircleSelectUserData *data = userData; - - if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { - SET_FLAG_FROM_TEST(mv->flag, data->select, SELECT); - data->is_changed = true; - } -} -static bool paint_vertsel_circle_select(ViewContext *vc, - wmGenericUserData *wm_userdata, - const eSelectOp sel_op, - const int mval[2], - float rad) -{ - BLI_assert(ELEM(sel_op, SEL_OP_SET, SEL_OP_ADD, SEL_OP_SUB)); - const bool use_zbuf = !XRAY_ENABLED(vc->v3d); - Object *ob = vc->obact; - Mesh *me = ob->data; - /* CircleSelectUserData data = {NULL}; */ /* UNUSED */ - - bool changed = false; - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - /* Flush selection at the end. */ - changed |= paintvert_deselect_all_visible(ob, SEL_DESELECT, false); - } - - const bool select = (sel_op != SEL_OP_SUB); - - if (use_zbuf) { - if (wm_userdata->data == NULL) { - editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, SCE_SELECT_VERTEX); - } - } - - if (use_zbuf) { - struct EditSelectBuf_Cache *esel = wm_userdata->data; - esel->select_bitmap = DRW_select_buffer_bitmap_from_circle( - vc->depsgraph, vc->region, vc->v3d, mval, (int)(rad + 1.0f), NULL); - if (esel->select_bitmap != NULL) { - changed |= edbm_backbuf_check_and_select_verts_obmode(me, esel, sel_op); - MEM_freeN(esel->select_bitmap); - esel->select_bitmap = NULL; - } - } - else { - CircleSelectUserData data; - - ED_view3d_init_mats_rv3d(vc->obact, vc->rv3d); /* for foreach's screen/vert projection */ - - view3d_userdata_circleselect_init(&data, vc, select, mval, rad); - meshobject_foreachScreenVert( - vc, paint_vertsel_circle_select_doSelectVert, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - changed |= data.is_changed; - } - - if (changed) { - if (sel_op == SEL_OP_SUB) { - BKE_mesh_mselect_validate(me); - } - paintvert_flush_flags(ob); - paintvert_tag_select_update(vc->C, ob); - } - return changed; -} - -static void nurbscurve_circle_doSelect(void *userData, - Nurb *UNUSED(nu), - BPoint *bp, - BezTriple *bezt, - int beztindex, - bool UNUSED(handles_visible), - const float screen_co[2]) -{ - CircleSelectUserData *data = userData; - - if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { - if (bp) { - SET_FLAG_FROM_TEST(bp->f1, data->select, data->select_flag); - } - else { - if (beztindex == 0) { - SET_FLAG_FROM_TEST(bezt->f1, data->select, data->select_flag); - } - else if (beztindex == 1) { - SET_FLAG_FROM_TEST(bezt->f2, data->select, data->select_flag); - } - else { - SET_FLAG_FROM_TEST(bezt->f3, data->select, data->select_flag); - } - } - data->is_changed = true; - } -} -static bool nurbscurve_circle_select(ViewContext *vc, - const eSelectOp sel_op, - const int mval[2], - float rad) -{ - const bool select = (sel_op != SEL_OP_SUB); - const bool deselect_all = (sel_op == SEL_OP_SET); - CircleSelectUserData data; - - view3d_userdata_circleselect_init(&data, vc, select, mval, rad); - - Curve *curve = (Curve *)vc->obedit->data; - ListBase *nurbs = BKE_curve_editNurbs_get(curve); - - /* For deselect all, items to be selected are tagged with temp flag. Clear that first. */ - if (deselect_all) { - BKE_nurbList_flag_set(nurbs, BEZT_FLAG_TEMP_TAG, false); - data.select_flag = BEZT_FLAG_TEMP_TAG; - } - - ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); /* for foreach's screen/vert projection */ - nurbs_foreachScreenVert(vc, nurbscurve_circle_doSelect, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - - /* Deselect items that were not added to selection (indicated by temp flag). */ - if (deselect_all) { - data.is_changed |= BKE_nurbList_flag_set_from_flag(nurbs, BEZT_FLAG_TEMP_TAG, SELECT); - } - - BKE_curve_nurb_vert_active_validate(vc->obedit->data); - - return data.is_changed; -} - -static void latticecurve_circle_doSelect(void *userData, BPoint *bp, const float screen_co[2]) -{ - CircleSelectUserData *data = userData; - - if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { - bp->f1 = data->select ? (bp->f1 | SELECT) : (bp->f1 & ~SELECT); - data->is_changed = true; - } -} -static bool lattice_circle_select(ViewContext *vc, - const eSelectOp sel_op, - const int mval[2], - float rad) -{ - CircleSelectUserData data; - const bool select = (sel_op != SEL_OP_SUB); - - view3d_userdata_circleselect_init(&data, vc, select, mval, rad); - - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - data.is_changed |= ED_lattice_flags_set(vc->obedit, 0); - } - ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); /* for foreach's screen/vert projection */ - - lattice_foreachScreenVert(vc, latticecurve_circle_doSelect, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - - return data.is_changed; -} - -/** - * \note logic is shared with the edit-bone case, see #armature_circle_doSelectJoint. - */ -static bool pchan_circle_doSelectJoint(void *userData, - bPoseChannel *pchan, - const float screen_co[2]) -{ - CircleSelectUserData *data = userData; - - if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { - if (data->select) { - pchan->bone->flag |= BONE_SELECTED; - } - else { - pchan->bone->flag &= ~BONE_SELECTED; - } - return 1; - } - return 0; -} -static void do_circle_select_pose__doSelectBone(void *userData, - struct bPoseChannel *pchan, - const float screen_co_a[2], - const float screen_co_b[2]) -{ - CircleSelectUserData *data = userData; - bArmature *arm = data->vc->obact->data; - if (!PBONE_SELECTABLE(arm, pchan->bone)) { - return; - } - - bool is_point_done = false; - int points_proj_tot = 0; - - /* project head location to screenspace */ - if (screen_co_a[0] != IS_CLIPPED) { - points_proj_tot++; - if (pchan_circle_doSelectJoint(data, pchan, screen_co_a)) { - is_point_done = true; - } - } - - /* project tail location to screenspace */ - if (screen_co_b[0] != IS_CLIPPED) { - points_proj_tot++; - if (pchan_circle_doSelectJoint(data, pchan, screen_co_b)) { - is_point_done = true; - } - } - - /* check if the head and/or tail is in the circle - * - the call to check also does the selection already - */ - - /* only if the endpoints didn't get selected, deal with the middle of the bone too - * It works nicer to only do this if the head or tail are not in the circle, - * otherwise there is no way to circle select joints alone */ - if ((is_point_done == false) && (points_proj_tot == 2) && - edge_inside_circle(data->mval_fl, data->radius, screen_co_a, screen_co_b)) { - if (data->select) { - pchan->bone->flag |= BONE_SELECTED; - } - else { - pchan->bone->flag &= ~BONE_SELECTED; - } - data->is_changed = true; - } - - data->is_changed |= is_point_done; -} -static bool pose_circle_select(ViewContext *vc, - const eSelectOp sel_op, - const int mval[2], - float rad) -{ - BLI_assert(ELEM(sel_op, SEL_OP_SET, SEL_OP_ADD, SEL_OP_SUB)); - CircleSelectUserData data; - const bool select = (sel_op != SEL_OP_SUB); - - view3d_userdata_circleselect_init(&data, vc, select, mval, rad); - - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - data.is_changed |= ED_pose_deselect_all(vc->obact, SEL_DESELECT, false); - } - - ED_view3d_init_mats_rv3d(vc->obact, vc->rv3d); /* for foreach's screen/vert projection */ - - /* Treat bones as clipped segments (no joints). */ - pose_foreachScreenBone(vc, - do_circle_select_pose__doSelectBone, - &data, - V3D_PROJ_TEST_CLIP_DEFAULT | V3D_PROJ_TEST_CLIP_CONTENT_DEFAULT); - - if (data.is_changed) { - ED_pose_bone_select_tag_update(vc->obact); - } - return data.is_changed; -} - -/** - * \note logic is shared with the pose-bone case, see #pchan_circle_doSelectJoint. - */ -static bool armature_circle_doSelectJoint(void *userData, - EditBone *ebone, - const float screen_co[2], - bool head) -{ - CircleSelectUserData *data = userData; - - if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { - if (head) { - if (data->select) { - ebone->flag |= BONE_ROOTSEL; - } - else { - ebone->flag &= ~BONE_ROOTSEL; - } - } - else { - if (data->select) { - ebone->flag |= BONE_TIPSEL; - } - else { - ebone->flag &= ~BONE_TIPSEL; - } - } - return 1; - } - return 0; -} -static void do_circle_select_armature__doSelectBone(void *userData, - struct EditBone *ebone, - const float screen_co_a[2], - const float screen_co_b[2]) -{ - CircleSelectUserData *data = userData; - const bArmature *arm = data->vc->obedit->data; - if (!(data->select ? EBONE_SELECTABLE(arm, ebone) : EBONE_VISIBLE(arm, ebone))) { - return; - } - - /* When true, ignore in the next pass. */ - ebone->temp.i = false; - - bool is_point_done = false; - bool is_edge_done = false; - int points_proj_tot = 0; - - /* project head location to screenspace */ - if (screen_co_a[0] != IS_CLIPPED) { - points_proj_tot++; - if (armature_circle_doSelectJoint(data, ebone, screen_co_a, true)) { - is_point_done = true; - } - } - - /* project tail location to screenspace */ - if (screen_co_b[0] != IS_CLIPPED) { - points_proj_tot++; - if (armature_circle_doSelectJoint(data, ebone, screen_co_b, false)) { - is_point_done = true; - } - } - - /* check if the head and/or tail is in the circle - * - the call to check also does the selection already - */ - - /* only if the endpoints didn't get selected, deal with the middle of the bone too - * It works nicer to only do this if the head or tail are not in the circle, - * otherwise there is no way to circle select joints alone */ - if ((is_point_done == false) && (points_proj_tot == 2) && - edge_inside_circle(data->mval_fl, data->radius, screen_co_a, screen_co_b)) { - SET_FLAG_FROM_TEST(ebone->flag, data->select, BONE_SELECTED | BONE_TIPSEL | BONE_ROOTSEL); - is_edge_done = true; - data->is_changed = true; - } - - if (is_point_done || is_edge_done) { - ebone->temp.i = true; - } - - data->is_changed |= is_point_done; -} -static void do_circle_select_armature__doSelectBone_clip_content(void *userData, - struct EditBone *ebone, - const float screen_co_a[2], - const float screen_co_b[2]) -{ - CircleSelectUserData *data = userData; - bArmature *arm = data->vc->obedit->data; - - if (!(data->select ? EBONE_SELECTABLE(arm, ebone) : EBONE_VISIBLE(arm, ebone))) { - return; - } - - /* Set in the first pass, needed so circle select prioritizes joints. */ - if (ebone->temp.i == true) { - return; - } - - if (edge_inside_circle(data->mval_fl, data->radius, screen_co_a, screen_co_b)) { - SET_FLAG_FROM_TEST(ebone->flag, data->select, BONE_SELECTED | BONE_TIPSEL | BONE_ROOTSEL); - data->is_changed = true; - } -} -static bool armature_circle_select(ViewContext *vc, - const eSelectOp sel_op, - const int mval[2], - float rad) -{ - CircleSelectUserData data; - bArmature *arm = vc->obedit->data; - - const bool select = (sel_op != SEL_OP_SUB); - - view3d_userdata_circleselect_init(&data, vc, select, mval, rad); - - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - data.is_changed |= ED_armature_edit_deselect_all_visible(vc->obedit); - } - - ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); - - /* Operate on fully visible (non-clipped) points. */ - armature_foreachScreenBone( - vc, do_circle_select_armature__doSelectBone, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - - /* Operate on bones as segments clipped to the viewport bounds - * (needed to handle bones with both points outside the view). - * A separate pass is needed since clipped coordinates can't be used for selecting joints. */ - armature_foreachScreenBone(vc, - do_circle_select_armature__doSelectBone_clip_content, - &data, - V3D_PROJ_TEST_CLIP_DEFAULT | V3D_PROJ_TEST_CLIP_CONTENT_DEFAULT); - - if (data.is_changed) { - ED_armature_edit_sync_selection(arm->edbo); - ED_armature_edit_validate_active(arm); - WM_main_add_notifier(NC_OBJECT | ND_BONE_SELECT, vc->obedit); - } - return data.is_changed; -} - -static void do_circle_select_mball__doSelectElem(void *userData, - struct MetaElem *ml, - const float screen_co[2]) -{ - CircleSelectUserData *data = userData; - - if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { - if (data->select) { - ml->flag |= SELECT; - } - else { - ml->flag &= ~SELECT; - } - data->is_changed = true; - } -} -static bool mball_circle_select(ViewContext *vc, - const eSelectOp sel_op, - const int mval[2], - float rad) -{ - CircleSelectUserData data; - - const bool select = (sel_op != SEL_OP_SUB); - - view3d_userdata_circleselect_init(&data, vc, select, mval, rad); - - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - data.is_changed |= BKE_mball_deselect_all(vc->obedit->data); - } - - ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); - - mball_foreachScreenElem( - vc, do_circle_select_mball__doSelectElem, &data, V3D_PROJ_TEST_CLIP_DEFAULT); - return data.is_changed; -} - -/** - * Callbacks for circle selection in Editmode - */ -static bool obedit_circle_select(bContext *C, - ViewContext *vc, - wmGenericUserData *wm_userdata, - const eSelectOp sel_op, - const int mval[2], - float rad) -{ - bool changed = false; - BLI_assert(ELEM(sel_op, SEL_OP_SET, SEL_OP_ADD, SEL_OP_SUB)); - switch (vc->obedit->type) { - case OB_MESH: - changed = mesh_circle_select(vc, wm_userdata, sel_op, mval, rad); - break; - case OB_CURVES_LEGACY: - case OB_SURF: - changed = nurbscurve_circle_select(vc, sel_op, mval, rad); - break; - case OB_LATTICE: - changed = lattice_circle_select(vc, sel_op, mval, rad); - break; - case OB_ARMATURE: - changed = armature_circle_select(vc, sel_op, mval, rad); - if (changed) { - ED_outliner_select_sync_from_edit_bone_tag(C); - } - break; - case OB_MBALL: - changed = mball_circle_select(vc, sel_op, mval, rad); - break; - default: - BLI_assert(0); - break; - } - - if (changed) { - DEG_id_tag_update(vc->obact->data, ID_RECALC_SELECT); - WM_main_add_notifier(NC_GEOM | ND_SELECT, vc->obact->data); - } - return changed; -} - -static bool object_circle_select(ViewContext *vc, - const eSelectOp sel_op, - const int mval[2], - float rad) -{ - BLI_assert(ELEM(sel_op, SEL_OP_SET, SEL_OP_ADD, SEL_OP_SUB)); - ViewLayer *view_layer = vc->view_layer; - View3D *v3d = vc->v3d; - - const float radius_squared = rad * rad; - const float mval_fl[2] = {mval[0], mval[1]}; - - bool changed = false; - if (SEL_OP_USE_PRE_DESELECT(sel_op)) { - changed |= object_deselect_all_visible(vc->view_layer, vc->v3d); - } - const bool select = (sel_op != SEL_OP_SUB); - const int select_flag = select ? BASE_SELECTED : 0; - - Base *base; - for (base = FIRSTBASE(view_layer); base; base = base->next) { - if (BASE_SELECTABLE(v3d, base) && ((base->flag & BASE_SELECTED) != select_flag)) { - float screen_co[2]; - if (ED_view3d_project_float_global( - vc->region, base->object->obmat[3], screen_co, V3D_PROJ_TEST_CLIP_DEFAULT) == - V3D_PROJ_RET_OK) { - if (len_squared_v2v2(mval_fl, screen_co) <= radius_squared) { - ED_object_base_select(base, select ? BA_SELECT : BA_DESELECT); - changed = true; - } - } - } - } - - return changed; -} - -/* not a real operator, only for circle test */ -static void view3d_circle_select_recalc(void *user_data) -{ - bContext *C = user_data; - Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C); - ViewContext vc; - ED_view3d_viewcontext_init(C, &vc, depsgraph); - em_setup_viewcontext(C, &vc); - - if (vc.obedit) { - switch (vc.obedit->type) { - case OB_MESH: { - FOREACH_OBJECT_IN_MODE_BEGIN ( - vc.view_layer, vc.v3d, vc.obact->type, vc.obact->mode, ob_iter) { - ED_view3d_viewcontext_init_object(&vc, ob_iter); - BM_mesh_select_mode_flush_ex( - vc.em->bm, vc.em->selectmode, BM_SELECT_LEN_FLUSH_RECALC_ALL); - } - FOREACH_OBJECT_IN_MODE_END; - break; - } - - default: - break; - } - } -} - -static int view3d_circle_select_modal(bContext *C, wmOperator *op, const wmEvent *event) -{ - int result = WM_gesture_circle_modal(C, op, event); - if (result & OPERATOR_FINISHED) { - view3d_circle_select_recalc(C); - } - return result; -} - -static void view3d_circle_select_cancel(bContext *C, wmOperator *op) -{ - WM_gesture_circle_cancel(C, op); - view3d_circle_select_recalc(C); -} - -static int view3d_circle_select_exec(bContext *C, wmOperator *op) -{ - Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C); - ViewContext vc; - const int radius = RNA_int_get(op->ptr, "radius"); - const int mval[2] = {RNA_int_get(op->ptr, "x"), RNA_int_get(op->ptr, "y")}; - - /* Allow each selection type to allocate their own data that's used between executions. */ - wmGesture *gesture = op->customdata; /* NULL when non-modal. */ - wmGenericUserData wm_userdata_buf = {0}; - wmGenericUserData *wm_userdata = gesture ? &gesture->user_data : &wm_userdata_buf; - - const eSelectOp sel_op = ED_select_op_modal(RNA_enum_get(op->ptr, "mode"), - WM_gesture_is_modal_first(gesture)); - - ED_view3d_viewcontext_init(C, &vc, depsgraph); - - Object *obact = vc.obact; - Object *obedit = vc.obedit; - - if (obedit || BKE_paint_select_elem_test(obact) || (obact && (obact->mode & OB_MODE_POSE))) { - view3d_operator_needs_opengl(C); - if (obedit == NULL) { - BKE_object_update_select_id(CTX_data_main(C)); - } - - FOREACH_OBJECT_IN_MODE_BEGIN (vc.view_layer, vc.v3d, obact->type, obact->mode, ob_iter) { - ED_view3d_viewcontext_init_object(&vc, ob_iter); - - obact = vc.obact; - obedit = vc.obedit; - - if (obedit) { - obedit_circle_select(C, &vc, wm_userdata, sel_op, mval, (float)radius); - } - else if (BKE_paint_select_face_test(obact)) { - paint_facesel_circle_select(&vc, wm_userdata, sel_op, mval, (float)radius); - } - else if (BKE_paint_select_vert_test(obact)) { - paint_vertsel_circle_select(&vc, wm_userdata, sel_op, mval, (float)radius); - } - else if (obact->mode & OB_MODE_POSE) { - pose_circle_select(&vc, sel_op, mval, (float)radius); - ED_outliner_select_sync_from_pose_bone_tag(C); - } - else { - BLI_assert(0); - } - } - FOREACH_OBJECT_IN_MODE_END; - } - else if (obact && (obact->mode & OB_MODE_PARTICLE_EDIT)) { - if (PE_circle_select(C, wm_userdata, sel_op, mval, (float)radius)) { - return OPERATOR_FINISHED; - } - return OPERATOR_CANCELLED; - } - else if (obact && obact->mode & OB_MODE_SCULPT) { - return OPERATOR_CANCELLED; - } - else { - if (object_circle_select(&vc, sel_op, mval, (float)radius)) { - DEG_id_tag_update(&vc.scene->id, ID_RECALC_SELECT); - WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, vc.scene); - - ED_outliner_select_sync_from_object_tag(C); - } - } - - /* Otherwise this is freed by the gesture. */ - if (wm_userdata == &wm_userdata_buf) { - WM_generic_user_data_free(wm_userdata); - } - else { - struct EditSelectBuf_Cache *esel = wm_userdata->data; - if (esel && esel->select_bitmap) { - MEM_freeN(esel->select_bitmap); - esel->select_bitmap = NULL; - } - } - - return OPERATOR_FINISHED; -} - -void VIEW3D_OT_select_circle(wmOperatorType *ot) -{ - ot->name = "Circle Select"; - ot->description = "Select items using circle selection"; - ot->idname = "VIEW3D_OT_select_circle"; - - ot->invoke = WM_gesture_circle_invoke; - ot->modal = view3d_circle_select_modal; - ot->exec = view3d_circle_select_exec; - ot->poll = view3d_selectable_data; - ot->cancel = view3d_circle_select_cancel; - ot->get_name = ED_select_circle_get_name; - - /* flags */ - ot->flag = OPTYPE_UNDO; - - /* properties */ - WM_operator_properties_gesture_circle(ot); - WM_operator_properties_select_operation_simple(ot); -} - -/** \} */ diff --git a/source/blender/editors/space_view3d/view3d_select.cc b/source/blender/editors/space_view3d/view3d_select.cc new file mode 100644 index 00000000000..085320a0e72 --- /dev/null +++ b/source/blender/editors/space_view3d/view3d_select.cc @@ -0,0 +1,4771 @@ +/* SPDX-License-Identifier: GPL-2.0-or-later + * Copyright 2008 Blender Foundation. All rights reserved. */ + +/** \file + * \ingroup spview3d + */ + +#include +#include +#include +#include + +#include "DNA_action_types.h" +#include "DNA_armature_types.h" +#include "DNA_curve_types.h" +#include "DNA_gpencil_types.h" +#include "DNA_mesh_types.h" +#include "DNA_meshdata_types.h" +#include "DNA_meta_types.h" +#include "DNA_object_types.h" +#include "DNA_scene_types.h" +#include "DNA_tracking_types.h" + +#include "MEM_guardedalloc.h" + +#include "BLI_bitmap.h" +#include "BLI_lasso_2d.h" +#include "BLI_linklist.h" +#include "BLI_listbase.h" +#include "BLI_math.h" +#include "BLI_rect.h" +#include "BLI_string.h" +#include "BLI_utildefines.h" +#include "BLI_vector.hh" + +#ifdef __BIG_ENDIAN__ +# include "BLI_endian_switch.h" +#endif + +/* vertex box select */ +#include "BKE_global.h" +#include "BKE_main.h" +#include "IMB_imbuf.h" +#include "IMB_imbuf_types.h" + +#include "BKE_action.h" +#include "BKE_armature.h" +#include "BKE_context.h" +#include "BKE_curve.h" +#include "BKE_editmesh.h" +#include "BKE_layer.h" +#include "BKE_mball.h" +#include "BKE_mesh.h" +#include "BKE_object.h" +#include "BKE_paint.h" +#include "BKE_scene.h" +#include "BKE_tracking.h" +#include "BKE_workspace.h" + +#include "WM_api.h" +#include "WM_toolsystem.h" +#include "WM_types.h" + +#include "RNA_access.h" +#include "RNA_define.h" +#include "RNA_enum_types.h" + +#include "ED_armature.h" +#include "ED_curve.h" +#include "ED_gpencil.h" +#include "ED_lattice.h" +#include "ED_mball.h" +#include "ED_mesh.h" +#include "ED_object.h" +#include "ED_outliner.h" +#include "ED_particle.h" +#include "ED_screen.h" +#include "ED_sculpt.h" +#include "ED_select_utils.h" + +#include "UI_interface.h" +#include "UI_resources.h" + +#include "GPU_matrix.h" +#include "GPU_select.h" + +#include "DEG_depsgraph.h" +#include "DEG_depsgraph_query.h" + +#include "DRW_engine.h" +#include "DRW_select_buffer.h" + +#include "view3d_intern.h" /* own include */ + +// #include "PIL_time_utildefines.h" + +/* -------------------------------------------------------------------- */ +/** \name Public Utilities + * \{ */ + +float ED_view3d_select_dist_px(void) +{ + return 75.0f * U.pixelsize; +} + +void ED_view3d_viewcontext_init(bContext *C, ViewContext *vc, Depsgraph *depsgraph) +{ + /* TODO: should return whether there is valid context to continue. */ + + memset(vc, 0, sizeof(ViewContext)); + vc->C = C; + vc->region = CTX_wm_region(C); + vc->bmain = CTX_data_main(C); + vc->depsgraph = depsgraph; + vc->scene = CTX_data_scene(C); + vc->view_layer = CTX_data_view_layer(C); + vc->v3d = CTX_wm_view3d(C); + vc->win = CTX_wm_window(C); + vc->rv3d = CTX_wm_region_view3d(C); + vc->obact = CTX_data_active_object(C); + vc->obedit = CTX_data_edit_object(C); +} + +void ED_view3d_viewcontext_init_object(ViewContext *vc, Object *obact) +{ + vc->obact = obact; + /* See public doc-string for rationale on checking the existing values first. */ + if (vc->obedit) { + BLI_assert(BKE_object_is_in_editmode(obact)); + vc->obedit = obact; + if (vc->em) { + vc->em = BKE_editmesh_from_object(vc->obedit); + } + } +} + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Internal Object Utilities + * \{ */ + +static bool object_deselect_all_visible(ViewLayer *view_layer, View3D *v3d) +{ + bool changed = false; + LISTBASE_FOREACH (Base *, base, &view_layer->object_bases) { + if (base->flag & BASE_SELECTED) { + if (BASE_SELECTABLE(v3d, base)) { + ED_object_base_select(base, BA_DESELECT); + changed = true; + } + } + } + return changed; +} + +/* deselect all except b */ +static bool object_deselect_all_except(ViewLayer *view_layer, Base *b) +{ + bool changed = false; + LISTBASE_FOREACH (Base *, base, &view_layer->object_bases) { + if (base->flag & BASE_SELECTED) { + if (b != base) { + ED_object_base_select(base, BA_DESELECT); + changed = true; + } + } + } + return changed; +} + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Internal Edit-Mesh Select Buffer Wrapper + * + * Avoid duplicate code when using edit-mode selection, + * actual logic is handled outside of this function. + * + * \note Currently this #EDBMSelectID_Context which is mesh specific + * however the logic could also be used for non-meshes too. + * + * \{ */ + +struct EditSelectBuf_Cache { + BLI_bitmap *select_bitmap; +}; + +static void editselect_buf_cache_init(ViewContext *vc, short select_mode) +{ + if (vc->obedit) { + uint bases_len = 0; + Base **bases = BKE_view_layer_array_from_bases_in_edit_mode( + vc->view_layer, vc->v3d, &bases_len); + + DRW_select_buffer_context_create(bases, bases_len, select_mode); + MEM_freeN(bases); + } + else { + /* Use for paint modes, currently only a single object at a time. */ + if (vc->obact) { + Base *base = BKE_view_layer_base_find(vc->view_layer, vc->obact); + DRW_select_buffer_context_create(&base, 1, select_mode); + } + } +} + +static void editselect_buf_cache_free(EditSelectBuf_Cache *esel) +{ + MEM_SAFE_FREE(esel->select_bitmap); +} + +static void editselect_buf_cache_free_voidp(void *esel_voidp) +{ + editselect_buf_cache_free(static_cast(esel_voidp)); + MEM_freeN(esel_voidp); +} + +static void editselect_buf_cache_init_with_generic_userdata(wmGenericUserData *wm_userdata, + ViewContext *vc, + short select_mode) +{ + EditSelectBuf_Cache *esel = MEM_cnew(__func__); + wm_userdata->data = esel; + wm_userdata->free_fn = editselect_buf_cache_free_voidp; + wm_userdata->use_free = true; + editselect_buf_cache_init(vc, select_mode); +} + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Internal Edit-Mesh Utilities + * \{ */ + +static bool edbm_backbuf_check_and_select_verts(EditSelectBuf_Cache *esel, + Depsgraph *depsgraph, + Object *ob, + BMEditMesh *em, + const eSelectOp sel_op) +{ + BMVert *eve; + BMIter iter; + bool changed = false; + + const BLI_bitmap *select_bitmap = esel->select_bitmap; + uint index = DRW_select_buffer_context_offset_for_object_elem(depsgraph, ob, SCE_SELECT_VERTEX); + if (index == 0) { + return false; + } + + index -= 1; + BM_ITER_MESH (eve, &iter, em->bm, BM_VERTS_OF_MESH) { + if (!BM_elem_flag_test(eve, BM_ELEM_HIDDEN)) { + const bool is_select = BM_elem_flag_test(eve, BM_ELEM_SELECT); + const bool is_inside = BLI_BITMAP_TEST_BOOL(select_bitmap, index); + const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); + if (sel_op_result != -1) { + BM_vert_select_set(em->bm, eve, sel_op_result); + changed = true; + } + } + index++; + } + return changed; +} + +static bool edbm_backbuf_check_and_select_edges(EditSelectBuf_Cache *esel, + Depsgraph *depsgraph, + Object *ob, + BMEditMesh *em, + const eSelectOp sel_op) +{ + BMEdge *eed; + BMIter iter; + bool changed = false; + + const BLI_bitmap *select_bitmap = esel->select_bitmap; + uint index = DRW_select_buffer_context_offset_for_object_elem(depsgraph, ob, SCE_SELECT_EDGE); + if (index == 0) { + return false; + } + + index -= 1; + BM_ITER_MESH (eed, &iter, em->bm, BM_EDGES_OF_MESH) { + if (!BM_elem_flag_test(eed, BM_ELEM_HIDDEN)) { + const bool is_select = BM_elem_flag_test(eed, BM_ELEM_SELECT); + const bool is_inside = BLI_BITMAP_TEST_BOOL(select_bitmap, index); + const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); + if (sel_op_result != -1) { + BM_edge_select_set(em->bm, eed, sel_op_result); + changed = true; + } + } + index++; + } + return changed; +} + +static bool edbm_backbuf_check_and_select_faces(EditSelectBuf_Cache *esel, + Depsgraph *depsgraph, + Object *ob, + BMEditMesh *em, + const eSelectOp sel_op) +{ + BMFace *efa; + BMIter iter; + bool changed = false; + + const BLI_bitmap *select_bitmap = esel->select_bitmap; + uint index = DRW_select_buffer_context_offset_for_object_elem(depsgraph, ob, SCE_SELECT_FACE); + if (index == 0) { + return false; + } + + index -= 1; + BM_ITER_MESH (efa, &iter, em->bm, BM_FACES_OF_MESH) { + if (!BM_elem_flag_test(efa, BM_ELEM_HIDDEN)) { + const bool is_select = BM_elem_flag_test(efa, BM_ELEM_SELECT); + const bool is_inside = BLI_BITMAP_TEST_BOOL(select_bitmap, index); + const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); + if (sel_op_result != -1) { + BM_face_select_set(em->bm, efa, sel_op_result); + changed = true; + } + } + index++; + } + return changed; +} + +/* object mode, edbm_ prefix is confusing here, rename? */ +static bool edbm_backbuf_check_and_select_verts_obmode(Mesh *me, + EditSelectBuf_Cache *esel, + const eSelectOp sel_op) +{ + MVert *mv = me->mvert; + bool changed = false; + + const BLI_bitmap *select_bitmap = esel->select_bitmap; + + if (mv) { + const bool *hide_vert = (const bool *)CustomData_get_layer_named( + &me->vdata, CD_PROP_BOOL, ".hide_vert"); + + for (int index = 0; index < me->totvert; index++, mv++) { + if (!(hide_vert && hide_vert[index])) { + const bool is_select = mv->flag & SELECT; + const bool is_inside = BLI_BITMAP_TEST_BOOL(select_bitmap, index); + const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(mv->flag, sel_op_result, SELECT); + changed = true; + } + } + } + } + return changed; +} + +/* object mode, edbm_ prefix is confusing here, rename? */ +static bool edbm_backbuf_check_and_select_faces_obmode(Mesh *me, + EditSelectBuf_Cache *esel, + const eSelectOp sel_op) +{ + MPoly *mpoly = me->mpoly; + bool changed = false; + + const BLI_bitmap *select_bitmap = esel->select_bitmap; + + if (mpoly) { + const bool *hide_poly = (const bool *)CustomData_get_layer_named( + &me->pdata, CD_PROP_BOOL, ".hide_poly"); + + for (int index = 0; index < me->totpoly; index++, mpoly++) { + if (!(hide_poly && hide_poly[index])) { + const bool is_select = mpoly->flag & ME_FACE_SEL; + const bool is_inside = BLI_BITMAP_TEST_BOOL(select_bitmap, index); + const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(mpoly->flag, sel_op_result, ME_FACE_SEL); + changed = true; + } + } + } + } + return changed; +} + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Lasso Select + * \{ */ + +struct LassoSelectUserData { + ViewContext *vc; + const rcti *rect; + const rctf *rect_fl; + rctf _rect_fl; + const int (*mcoords)[2]; + int mcoords_len; + eSelectOp sel_op; + eBezTriple_Flag select_flag; + + /* runtime */ + int pass; + bool is_done; + bool is_changed; +}; + +static void view3d_userdata_lassoselect_init(LassoSelectUserData *r_data, + ViewContext *vc, + const rcti *rect, + const int (*mcoords)[2], + const int mcoords_len, + const eSelectOp sel_op) +{ + r_data->vc = vc; + + r_data->rect = rect; + r_data->rect_fl = &r_data->_rect_fl; + BLI_rctf_rcti_copy(&r_data->_rect_fl, rect); + + r_data->mcoords = mcoords; + r_data->mcoords_len = mcoords_len; + r_data->sel_op = sel_op; + /* SELECT by default, but can be changed if needed (only few cases use and respect this). */ + r_data->select_flag = (eBezTriple_Flag)SELECT; + + /* runtime */ + r_data->pass = 0; + r_data->is_done = false; + r_data->is_changed = false; +} + +static bool view3d_selectable_data(bContext *C) +{ + Object *ob = CTX_data_active_object(C); + + if (!ED_operator_region_view3d_active(C)) { + return 0; + } + + if (ob) { + if (ob->mode & OB_MODE_EDIT) { + if (ob->type == OB_FONT) { + return 0; + } + } + else { + if ((ob->mode & (OB_MODE_VERTEX_PAINT | OB_MODE_WEIGHT_PAINT | OB_MODE_TEXTURE_PAINT)) && + !BKE_paint_select_elem_test(ob)) { + return 0; + } + } + } + + return 1; +} + +/* helper also for box_select */ +static bool edge_fully_inside_rect(const rctf *rect, const float v1[2], const float v2[2]) +{ + return BLI_rctf_isect_pt_v(rect, v1) && BLI_rctf_isect_pt_v(rect, v2); +} + +static bool edge_inside_rect(const rctf *rect, const float v1[2], const float v2[2]) +{ + int d1, d2, d3, d4; + + /* check points in rect */ + if (edge_fully_inside_rect(rect, v1, v2)) { + return 1; + } + + /* check points completely out rect */ + if (v1[0] < rect->xmin && v2[0] < rect->xmin) { + return 0; + } + if (v1[0] > rect->xmax && v2[0] > rect->xmax) { + return 0; + } + if (v1[1] < rect->ymin && v2[1] < rect->ymin) { + return 0; + } + if (v1[1] > rect->ymax && v2[1] > rect->ymax) { + return 0; + } + + /* simple check lines intersecting. */ + d1 = (v1[1] - v2[1]) * (v1[0] - rect->xmin) + (v2[0] - v1[0]) * (v1[1] - rect->ymin); + d2 = (v1[1] - v2[1]) * (v1[0] - rect->xmin) + (v2[0] - v1[0]) * (v1[1] - rect->ymax); + d3 = (v1[1] - v2[1]) * (v1[0] - rect->xmax) + (v2[0] - v1[0]) * (v1[1] - rect->ymax); + d4 = (v1[1] - v2[1]) * (v1[0] - rect->xmax) + (v2[0] - v1[0]) * (v1[1] - rect->ymin); + + if (d1 < 0 && d2 < 0 && d3 < 0 && d4 < 0) { + return 0; + } + if (d1 > 0 && d2 > 0 && d3 > 0 && d4 > 0) { + return 0; + } + + return 1; +} + +static void do_lasso_select_pose__do_tag(void *userData, + bPoseChannel *pchan, + const float screen_co_a[2], + const float screen_co_b[2]) +{ + LassoSelectUserData *data = static_cast(userData); + const bArmature *arm = static_cast(data->vc->obact->data); + if (!PBONE_SELECTABLE(arm, pchan->bone)) { + return; + } + + if (BLI_rctf_isect_segment(data->rect_fl, screen_co_a, screen_co_b) && + BLI_lasso_is_edge_inside( + data->mcoords, data->mcoords_len, UNPACK2(screen_co_a), UNPACK2(screen_co_b), INT_MAX)) { + pchan->bone->flag |= BONE_DONE; + data->is_changed = true; + } +} +static void do_lasso_tag_pose(ViewContext *vc, + Object *ob, + const int mcoords[][2], + const int mcoords_len) +{ + ViewContext vc_tmp; + LassoSelectUserData data; + rcti rect; + + if ((ob->type != OB_ARMATURE) || (ob->pose == nullptr)) { + return; + } + + vc_tmp = *vc; + vc_tmp.obact = ob; + + BLI_lasso_boundbox(&rect, mcoords, mcoords_len); + + view3d_userdata_lassoselect_init( + &data, vc, &rect, mcoords, mcoords_len, static_cast(0)); + + ED_view3d_init_mats_rv3d(vc_tmp.obact, vc->rv3d); + + /* Treat bones as clipped segments (no joints). */ + pose_foreachScreenBone(&vc_tmp, + do_lasso_select_pose__do_tag, + &data, + V3D_PROJ_TEST_CLIP_DEFAULT | V3D_PROJ_TEST_CLIP_CONTENT_DEFAULT); +} + +static bool do_lasso_select_objects(ViewContext *vc, + const int mcoords[][2], + const int mcoords_len, + const eSelectOp sel_op) +{ + View3D *v3d = vc->v3d; + Base *base; + + bool changed = false; + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + changed |= object_deselect_all_visible(vc->view_layer, vc->v3d); + } + + for (base = static_cast(vc->view_layer->object_bases.first); base; base = base->next) { + if (BASE_SELECTABLE(v3d, base)) { /* Use this to avoid unnecessary lasso look-ups. */ + const bool is_select = base->flag & BASE_SELECTED; + const bool is_inside = ((ED_view3d_project_base(vc->region, base) == V3D_PROJ_RET_OK) && + BLI_lasso_is_point_inside( + mcoords, mcoords_len, base->sx, base->sy, IS_CLIPPED)); + const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); + if (sel_op_result != -1) { + ED_object_base_select(base, sel_op_result ? BA_SELECT : BA_DESELECT); + changed = true; + } + } + } + + if (changed) { + DEG_id_tag_update(&vc->scene->id, ID_RECALC_SELECT); + WM_main_add_notifier(NC_SCENE | ND_OB_SELECT, vc->scene); + } + return changed; +} + +/** + * Use for lasso & box select. + */ +static blender::Vector do_pose_tag_select_op_prepare(ViewContext *vc) +{ + blender::Vector bases; + + FOREACH_BASE_IN_MODE_BEGIN (vc->view_layer, vc->v3d, OB_ARMATURE, OB_MODE_POSE, base_iter) { + Object *ob_iter = base_iter->object; + bArmature *arm = static_cast(ob_iter->data); + LISTBASE_FOREACH (bPoseChannel *, pchan, &ob_iter->pose->chanbase) { + Bone *bone = pchan->bone; + bone->flag &= ~BONE_DONE; + } + arm->id.tag |= LIB_TAG_DOIT; + ob_iter->id.tag &= ~LIB_TAG_DOIT; + bases.append(base_iter); + } + FOREACH_BASE_IN_MODE_END; + return bases; +} + +static bool do_pose_tag_select_op_exec(blender::MutableSpan bases, const eSelectOp sel_op) +{ + bool changed_multi = false; + + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + for (const int i : bases.index_range()) { + Base *base_iter = bases[i]; + Object *ob_iter = base_iter->object; + if (ED_pose_deselect_all(ob_iter, SEL_DESELECT, false)) { + ED_pose_bone_select_tag_update(ob_iter); + changed_multi = true; + } + } + } + + for (const int i : bases.index_range()) { + Base *base_iter = bases[i]; + Object *ob_iter = base_iter->object; + bArmature *arm = static_cast(ob_iter->data); + + /* Don't handle twice. */ + if (arm->id.tag & LIB_TAG_DOIT) { + arm->id.tag &= ~LIB_TAG_DOIT; + } + else { + continue; + } + + bool changed = true; + LISTBASE_FOREACH (bPoseChannel *, pchan, &ob_iter->pose->chanbase) { + Bone *bone = pchan->bone; + if ((bone->flag & BONE_UNSELECTABLE) == 0) { + const bool is_select = bone->flag & BONE_SELECTED; + const bool is_inside = bone->flag & BONE_DONE; + const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(bone->flag, sel_op_result, BONE_SELECTED); + if (sel_op_result == 0) { + if (arm->act_bone == bone) { + arm->act_bone = nullptr; + } + } + changed = true; + } + } + } + if (changed) { + ED_pose_bone_select_tag_update(ob_iter); + changed_multi = true; + } + } + return changed_multi; +} + +static bool do_lasso_select_pose(ViewContext *vc, + const int mcoords[][2], + const int mcoords_len, + const eSelectOp sel_op) +{ + blender::Vector bases = do_pose_tag_select_op_prepare(vc); + + for (const int i : bases.index_range()) { + Base *base_iter = bases[i]; + Object *ob_iter = base_iter->object; + do_lasso_tag_pose(vc, ob_iter, mcoords, mcoords_len); + } + + const bool changed_multi = do_pose_tag_select_op_exec(bases, sel_op); + if (changed_multi) { + DEG_id_tag_update(&vc->scene->id, ID_RECALC_SELECT); + WM_main_add_notifier(NC_SCENE | ND_OB_SELECT, vc->scene); + } + + return changed_multi; +} + +static void do_lasso_select_mesh__doSelectVert(void *userData, + BMVert *eve, + const float screen_co[2], + int UNUSED(index)) +{ + LassoSelectUserData *data = static_cast(userData); + const bool is_select = BM_elem_flag_test(eve, BM_ELEM_SELECT); + const bool is_inside = + (BLI_rctf_isect_pt_v(data->rect_fl, screen_co) && + BLI_lasso_is_point_inside( + data->mcoords, data->mcoords_len, screen_co[0], screen_co[1], IS_CLIPPED)); + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + BM_vert_select_set(data->vc->em->bm, eve, sel_op_result); + data->is_changed = true; + } +} +struct LassoSelectUserData_ForMeshEdge { + LassoSelectUserData *data; + EditSelectBuf_Cache *esel; + uint backbuf_offset; +}; +static void do_lasso_select_mesh__doSelectEdge_pass0(void *user_data, + BMEdge *eed, + const float screen_co_a[2], + const float screen_co_b[2], + int index) +{ + LassoSelectUserData_ForMeshEdge *data_for_edge = static_cast( + user_data); + LassoSelectUserData *data = data_for_edge->data; + bool is_visible = true; + if (data_for_edge->backbuf_offset) { + uint bitmap_inedx = data_for_edge->backbuf_offset + index - 1; + is_visible = BLI_BITMAP_TEST_BOOL(data_for_edge->esel->select_bitmap, bitmap_inedx); + } + + const bool is_select = BM_elem_flag_test(eed, BM_ELEM_SELECT); + const bool is_inside = + (is_visible && edge_fully_inside_rect(data->rect_fl, screen_co_a, screen_co_b) && + BLI_lasso_is_point_inside( + data->mcoords, data->mcoords_len, UNPACK2(screen_co_a), IS_CLIPPED) && + BLI_lasso_is_point_inside( + data->mcoords, data->mcoords_len, UNPACK2(screen_co_b), IS_CLIPPED)); + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + BM_edge_select_set(data->vc->em->bm, eed, sel_op_result); + data->is_done = true; + data->is_changed = true; + } +} +static void do_lasso_select_mesh__doSelectEdge_pass1(void *user_data, + BMEdge *eed, + const float screen_co_a[2], + const float screen_co_b[2], + int index) +{ + LassoSelectUserData_ForMeshEdge *data_for_edge = static_cast( + user_data); + LassoSelectUserData *data = data_for_edge->data; + bool is_visible = true; + if (data_for_edge->backbuf_offset) { + uint bitmap_inedx = data_for_edge->backbuf_offset + index - 1; + is_visible = BLI_BITMAP_TEST_BOOL(data_for_edge->esel->select_bitmap, bitmap_inedx); + } + + const bool is_select = BM_elem_flag_test(eed, BM_ELEM_SELECT); + const bool is_inside = (is_visible && BLI_lasso_is_edge_inside(data->mcoords, + data->mcoords_len, + UNPACK2(screen_co_a), + UNPACK2(screen_co_b), + IS_CLIPPED)); + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + BM_edge_select_set(data->vc->em->bm, eed, sel_op_result); + data->is_changed = true; + } +} + +static void do_lasso_select_mesh__doSelectFace(void *userData, + BMFace *efa, + const float screen_co[2], + int UNUSED(index)) +{ + LassoSelectUserData *data = static_cast(userData); + const bool is_select = BM_elem_flag_test(efa, BM_ELEM_SELECT); + const bool is_inside = + (BLI_rctf_isect_pt_v(data->rect_fl, screen_co) && + BLI_lasso_is_point_inside( + data->mcoords, data->mcoords_len, screen_co[0], screen_co[1], IS_CLIPPED)); + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + BM_face_select_set(data->vc->em->bm, efa, sel_op_result); + data->is_changed = true; + } +} + +static bool do_lasso_select_mesh(ViewContext *vc, + wmGenericUserData *wm_userdata, + const int mcoords[][2], + const int mcoords_len, + const eSelectOp sel_op) +{ + LassoSelectUserData data; + ToolSettings *ts = vc->scene->toolsettings; + rcti rect; + + /* set editmesh */ + vc->em = BKE_editmesh_from_object(vc->obedit); + + BLI_lasso_boundbox(&rect, mcoords, mcoords_len); + + view3d_userdata_lassoselect_init(&data, vc, &rect, mcoords, mcoords_len, sel_op); + + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + if (vc->em->bm->totvertsel) { + EDBM_flag_disable_all(vc->em, BM_ELEM_SELECT); + data.is_changed = true; + } + } + + /* for non zbuf projections, don't change the GL state */ + ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); + + GPU_matrix_set(vc->rv3d->viewmat); + + const bool use_zbuf = !XRAY_FLAG_ENABLED(vc->v3d); + + EditSelectBuf_Cache *esel = static_cast(wm_userdata->data); + if (use_zbuf) { + if (wm_userdata->data == nullptr) { + editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, ts->selectmode); + esel = static_cast(wm_userdata->data); + esel->select_bitmap = DRW_select_buffer_bitmap_from_poly( + vc->depsgraph, vc->region, vc->v3d, mcoords, mcoords_len, &rect, nullptr); + } + } + + if (ts->selectmode & SCE_SELECT_VERTEX) { + if (use_zbuf) { + data.is_changed |= edbm_backbuf_check_and_select_verts( + esel, vc->depsgraph, vc->obedit, vc->em, sel_op); + } + else { + mesh_foreachScreenVert( + vc, do_lasso_select_mesh__doSelectVert, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + } + } + if (ts->selectmode & SCE_SELECT_EDGE) { + /* Does both use_zbuf and non-use_zbuf versions (need screen cos for both) */ + LassoSelectUserData_ForMeshEdge data_for_edge{}; + data_for_edge.data = &data; + data_for_edge.esel = use_zbuf ? esel : nullptr; + data_for_edge.backbuf_offset = use_zbuf ? DRW_select_buffer_context_offset_for_object_elem( + vc->depsgraph, vc->obedit, SCE_SELECT_EDGE) : + 0; + + const eV3DProjTest clip_flag = V3D_PROJ_TEST_CLIP_NEAR | + (use_zbuf ? (eV3DProjTest)0 : V3D_PROJ_TEST_CLIP_BB); + /* Fully inside. */ + mesh_foreachScreenEdge_clip_bb_segment( + vc, do_lasso_select_mesh__doSelectEdge_pass0, &data_for_edge, clip_flag); + if (data.is_done == false) { + /* Fall back to partially inside. + * Clip content to account for edges partially behind the view. */ + mesh_foreachScreenEdge_clip_bb_segment(vc, + do_lasso_select_mesh__doSelectEdge_pass1, + &data_for_edge, + clip_flag | V3D_PROJ_TEST_CLIP_CONTENT_DEFAULT); + } + } + + if (ts->selectmode & SCE_SELECT_FACE) { + if (use_zbuf) { + data.is_changed |= edbm_backbuf_check_and_select_faces( + esel, vc->depsgraph, vc->obedit, vc->em, sel_op); + } + else { + mesh_foreachScreenFace( + vc, do_lasso_select_mesh__doSelectFace, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + } + } + + if (data.is_changed) { + EDBM_selectmode_flush(vc->em); + } + return data.is_changed; +} + +static void do_lasso_select_curve__doSelect(void *userData, + Nurb *UNUSED(nu), + BPoint *bp, + BezTriple *bezt, + int beztindex, + bool handles_visible, + const float screen_co[2]) +{ + LassoSelectUserData *data = static_cast(userData); + + const bool is_inside = BLI_lasso_is_point_inside( + data->mcoords, data->mcoords_len, screen_co[0], screen_co[1], IS_CLIPPED); + if (bp) { + const bool is_select = bp->f1 & SELECT; + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(bp->f1, sel_op_result, data->select_flag); + data->is_changed = true; + } + } + else { + if (!handles_visible) { + /* can only be (beztindex == 1) here since handles are hidden */ + const bool is_select = bezt->f2 & SELECT; + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(bezt->f2, sel_op_result, data->select_flag); + } + bezt->f1 = bezt->f3 = bezt->f2; + data->is_changed = true; + } + else { + uint8_t *flag_p = (&bezt->f1) + beztindex; + const bool is_select = *flag_p & SELECT; + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(*flag_p, sel_op_result, data->select_flag); + data->is_changed = true; + } + } + } +} + +static bool do_lasso_select_curve(ViewContext *vc, + const int mcoords[][2], + const int mcoords_len, + const eSelectOp sel_op) +{ + const bool deselect_all = (sel_op == SEL_OP_SET); + LassoSelectUserData data; + rcti rect; + + BLI_lasso_boundbox(&rect, mcoords, mcoords_len); + + view3d_userdata_lassoselect_init(&data, vc, &rect, mcoords, mcoords_len, sel_op); + + Curve *curve = (Curve *)vc->obedit->data; + ListBase *nurbs = BKE_curve_editNurbs_get(curve); + + /* For deselect all, items to be selected are tagged with temp flag. Clear that first. */ + if (deselect_all) { + BKE_nurbList_flag_set(nurbs, BEZT_FLAG_TEMP_TAG, false); + data.select_flag = BEZT_FLAG_TEMP_TAG; + } + + ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); /* for foreach's screen/vert projection */ + nurbs_foreachScreenVert(vc, do_lasso_select_curve__doSelect, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + + /* Deselect items that were not added to selection (indicated by temp flag). */ + if (deselect_all) { + data.is_changed |= BKE_nurbList_flag_set_from_flag(nurbs, BEZT_FLAG_TEMP_TAG, SELECT); + } + + if (data.is_changed) { + BKE_curve_nurb_vert_active_validate(static_cast(vc->obedit->data)); + } + return data.is_changed; +} + +static void do_lasso_select_lattice__doSelect(void *userData, BPoint *bp, const float screen_co[2]) +{ + LassoSelectUserData *data = static_cast(userData); + const bool is_select = bp->f1 & SELECT; + const bool is_inside = + (BLI_rctf_isect_pt_v(data->rect_fl, screen_co) && + BLI_lasso_is_point_inside( + data->mcoords, data->mcoords_len, screen_co[0], screen_co[1], IS_CLIPPED)); + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(bp->f1, sel_op_result, SELECT); + data->is_changed = true; + } +} +static bool do_lasso_select_lattice(ViewContext *vc, + const int mcoords[][2], + const int mcoords_len, + const eSelectOp sel_op) +{ + LassoSelectUserData data; + rcti rect; + + BLI_lasso_boundbox(&rect, mcoords, mcoords_len); + + view3d_userdata_lassoselect_init(&data, vc, &rect, mcoords, mcoords_len, sel_op); + + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + data.is_changed |= ED_lattice_flags_set(vc->obedit, 0); + } + + ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); /* for foreach's screen/vert projection */ + lattice_foreachScreenVert( + vc, do_lasso_select_lattice__doSelect, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + return data.is_changed; +} + +static void do_lasso_select_armature__doSelectBone(void *userData, + EditBone *ebone, + const float screen_co_a[2], + const float screen_co_b[2]) +{ + LassoSelectUserData *data = static_cast(userData); + const bArmature *arm = static_cast(data->vc->obedit->data); + if (!EBONE_VISIBLE(arm, ebone)) { + return; + } + + int is_ignore_flag = 0; + int is_inside_flag = 0; + + if (screen_co_a[0] != IS_CLIPPED) { + if (BLI_rcti_isect_pt(data->rect, UNPACK2(screen_co_a)) && + BLI_lasso_is_point_inside( + data->mcoords, data->mcoords_len, UNPACK2(screen_co_a), INT_MAX)) { + is_inside_flag |= BONESEL_ROOT; + } + } + else { + is_ignore_flag |= BONESEL_ROOT; + } + + if (screen_co_b[0] != IS_CLIPPED) { + if (BLI_rcti_isect_pt(data->rect, UNPACK2(screen_co_b)) && + BLI_lasso_is_point_inside( + data->mcoords, data->mcoords_len, UNPACK2(screen_co_b), INT_MAX)) { + is_inside_flag |= BONESEL_TIP; + } + } + else { + is_ignore_flag |= BONESEL_TIP; + } + + if (is_ignore_flag == 0) { + if (is_inside_flag == (BONE_ROOTSEL | BONE_TIPSEL) || + BLI_lasso_is_edge_inside(data->mcoords, + data->mcoords_len, + UNPACK2(screen_co_a), + UNPACK2(screen_co_b), + INT_MAX)) { + is_inside_flag |= BONESEL_BONE; + } + } + + ebone->temp.i = is_inside_flag | (is_ignore_flag >> 16); +} +static void do_lasso_select_armature__doSelectBone_clip_content(void *userData, + EditBone *ebone, + const float screen_co_a[2], + const float screen_co_b[2]) +{ + LassoSelectUserData *data = static_cast(userData); + bArmature *arm = static_cast(data->vc->obedit->data); + if (!EBONE_VISIBLE(arm, ebone)) { + return; + } + + const int is_ignore_flag = ebone->temp.i << 16; + int is_inside_flag = ebone->temp.i & ~0xFFFF; + + /* - When #BONESEL_BONE is set, there is nothing to do. + * - When #BONE_ROOTSEL or #BONE_TIPSEL have been set - they take priority over bone selection. + */ + if (is_inside_flag & (BONESEL_BONE | BONE_ROOTSEL | BONE_TIPSEL)) { + return; + } + + if (BLI_lasso_is_edge_inside( + data->mcoords, data->mcoords_len, UNPACK2(screen_co_a), UNPACK2(screen_co_b), INT_MAX)) { + is_inside_flag |= BONESEL_BONE; + } + + ebone->temp.i = is_inside_flag | (is_ignore_flag >> 16); +} + +static bool do_lasso_select_armature(ViewContext *vc, + const int mcoords[][2], + const int mcoords_len, + const eSelectOp sel_op) +{ + LassoSelectUserData data; + rcti rect; + + BLI_lasso_boundbox(&rect, mcoords, mcoords_len); + + view3d_userdata_lassoselect_init(&data, vc, &rect, mcoords, mcoords_len, sel_op); + + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + data.is_changed |= ED_armature_edit_deselect_all_visible(vc->obedit); + } + + bArmature *arm = static_cast(vc->obedit->data); + + ED_armature_ebone_listbase_temp_clear(arm->edbo); + + ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); + + /* Operate on fully visible (non-clipped) points. */ + armature_foreachScreenBone( + vc, do_lasso_select_armature__doSelectBone, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + + /* Operate on bones as segments clipped to the viewport bounds + * (needed to handle bones with both points outside the view). + * A separate pass is needed since clipped coordinates can't be used for selecting joints. */ + armature_foreachScreenBone(vc, + do_lasso_select_armature__doSelectBone_clip_content, + &data, + V3D_PROJ_TEST_CLIP_DEFAULT | V3D_PROJ_TEST_CLIP_CONTENT_DEFAULT); + + data.is_changed |= ED_armature_edit_select_op_from_tagged(arm, sel_op); + + if (data.is_changed) { + WM_main_add_notifier(NC_OBJECT | ND_BONE_SELECT, vc->obedit); + } + return data.is_changed; +} + +static void do_lasso_select_mball__doSelectElem(void *userData, + MetaElem *ml, + const float screen_co[2]) +{ + LassoSelectUserData *data = static_cast(userData); + const bool is_select = ml->flag & SELECT; + const bool is_inside = + (BLI_rctf_isect_pt_v(data->rect_fl, screen_co) && + BLI_lasso_is_point_inside( + data->mcoords, data->mcoords_len, screen_co[0], screen_co[1], INT_MAX)); + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(ml->flag, sel_op_result, SELECT); + data->is_changed = true; + } +} +static bool do_lasso_select_meta(ViewContext *vc, + const int mcoords[][2], + const int mcoords_len, + const eSelectOp sel_op) +{ + LassoSelectUserData data; + rcti rect; + + MetaBall *mb = (MetaBall *)vc->obedit->data; + + BLI_lasso_boundbox(&rect, mcoords, mcoords_len); + + view3d_userdata_lassoselect_init(&data, vc, &rect, mcoords, mcoords_len, sel_op); + + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + data.is_changed |= BKE_mball_deselect_all(mb); + } + + ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); + + mball_foreachScreenElem( + vc, do_lasso_select_mball__doSelectElem, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + + return data.is_changed; +} + +static void do_lasso_select_meshobject__doSelectVert(void *userData, + MVert *mv, + const float screen_co[2], + int UNUSED(index)) +{ + LassoSelectUserData *data = static_cast(userData); + const bool is_select = mv->flag & SELECT; + const bool is_inside = + (BLI_rctf_isect_pt_v(data->rect_fl, screen_co) && + BLI_lasso_is_point_inside( + data->mcoords, data->mcoords_len, screen_co[0], screen_co[1], IS_CLIPPED)); + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(mv->flag, sel_op_result, SELECT); + data->is_changed = true; + } +} +static bool do_lasso_select_paintvert(ViewContext *vc, + wmGenericUserData *wm_userdata, + const int mcoords[][2], + const int mcoords_len, + const eSelectOp sel_op) +{ + const bool use_zbuf = !XRAY_ENABLED(vc->v3d); + Object *ob = vc->obact; + Mesh *me = static_cast(ob->data); + rcti rect; + + if (me == nullptr || me->totvert == 0) { + return false; + } + + bool changed = false; + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + /* flush selection at the end */ + changed |= paintvert_deselect_all_visible(ob, SEL_DESELECT, false); + } + + BLI_lasso_boundbox(&rect, mcoords, mcoords_len); + + EditSelectBuf_Cache *esel = static_cast(wm_userdata->data); + if (use_zbuf) { + if (wm_userdata->data == nullptr) { + editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, SCE_SELECT_VERTEX); + esel = static_cast(wm_userdata->data); + esel->select_bitmap = DRW_select_buffer_bitmap_from_poly( + vc->depsgraph, vc->region, vc->v3d, mcoords, mcoords_len, &rect, nullptr); + } + } + + if (use_zbuf) { + if (esel->select_bitmap != nullptr) { + changed |= edbm_backbuf_check_and_select_verts_obmode(me, esel, sel_op); + } + } + else { + LassoSelectUserData data; + + view3d_userdata_lassoselect_init(&data, vc, &rect, mcoords, mcoords_len, sel_op); + + ED_view3d_init_mats_rv3d(vc->obact, vc->rv3d); + + meshobject_foreachScreenVert( + vc, do_lasso_select_meshobject__doSelectVert, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + + changed |= data.is_changed; + } + + if (changed) { + if (SEL_OP_CAN_DESELECT(sel_op)) { + BKE_mesh_mselect_validate(me); + } + paintvert_flush_flags(ob); + paintvert_tag_select_update(vc->C, ob); + } + + return changed; +} +static bool do_lasso_select_paintface(ViewContext *vc, + wmGenericUserData *wm_userdata, + const int mcoords[][2], + const int mcoords_len, + const eSelectOp sel_op) +{ + Object *ob = vc->obact; + Mesh *me = static_cast(ob->data); + rcti rect; + + if (me == nullptr || me->totpoly == 0) { + return false; + } + + bool changed = false; + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + /* flush selection at the end */ + changed |= paintface_deselect_all_visible(vc->C, ob, SEL_DESELECT, false); + } + + BLI_lasso_boundbox(&rect, mcoords, mcoords_len); + + EditSelectBuf_Cache *esel = static_cast(wm_userdata->data); + if (esel == nullptr) { + editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, SCE_SELECT_FACE); + esel = static_cast(wm_userdata->data); + esel->select_bitmap = DRW_select_buffer_bitmap_from_poly( + vc->depsgraph, vc->region, vc->v3d, mcoords, mcoords_len, &rect, nullptr); + } + + if (esel->select_bitmap) { + changed |= edbm_backbuf_check_and_select_faces_obmode(me, esel, sel_op); + } + + if (changed) { + paintface_flush_flags(vc->C, ob, true, false); + } + return changed; +} + +static bool view3d_lasso_select(bContext *C, + ViewContext *vc, + const int mcoords[][2], + const int mcoords_len, + const eSelectOp sel_op) +{ + Object *ob = CTX_data_active_object(C); + bool changed_multi = false; + + wmGenericUserData wm_userdata_buf = {0}; + wmGenericUserData *wm_userdata = &wm_userdata_buf; + + if (vc->obedit == nullptr) { /* Object Mode */ + if (BKE_paint_select_face_test(ob)) { + changed_multi |= do_lasso_select_paintface(vc, wm_userdata, mcoords, mcoords_len, sel_op); + } + else if (BKE_paint_select_vert_test(ob)) { + changed_multi |= do_lasso_select_paintvert(vc, wm_userdata, mcoords, mcoords_len, sel_op); + } + else if (ob && + (ob->mode & (OB_MODE_VERTEX_PAINT | OB_MODE_WEIGHT_PAINT | OB_MODE_TEXTURE_PAINT))) { + /* pass */ + } + else if (ob && (ob->mode & OB_MODE_PARTICLE_EDIT)) { + changed_multi |= PE_lasso_select(C, mcoords, mcoords_len, sel_op) != OPERATOR_CANCELLED; + } + else if (ob && (ob->mode & OB_MODE_POSE)) { + changed_multi |= do_lasso_select_pose(vc, mcoords, mcoords_len, sel_op); + if (changed_multi) { + ED_outliner_select_sync_from_pose_bone_tag(C); + } + } + else { + changed_multi |= do_lasso_select_objects(vc, mcoords, mcoords_len, sel_op); + if (changed_multi) { + ED_outliner_select_sync_from_object_tag(C); + } + } + } + else { /* Edit Mode */ + FOREACH_OBJECT_IN_MODE_BEGIN (vc->view_layer, vc->v3d, ob->type, ob->mode, ob_iter) { + ED_view3d_viewcontext_init_object(vc, ob_iter); + bool changed = false; + + switch (vc->obedit->type) { + case OB_MESH: + changed = do_lasso_select_mesh(vc, wm_userdata, mcoords, mcoords_len, sel_op); + break; + case OB_CURVES_LEGACY: + case OB_SURF: + changed = do_lasso_select_curve(vc, mcoords, mcoords_len, sel_op); + break; + case OB_LATTICE: + changed = do_lasso_select_lattice(vc, mcoords, mcoords_len, sel_op); + break; + case OB_ARMATURE: + changed = do_lasso_select_armature(vc, mcoords, mcoords_len, sel_op); + if (changed) { + ED_outliner_select_sync_from_edit_bone_tag(C); + } + break; + case OB_MBALL: + changed = do_lasso_select_meta(vc, mcoords, mcoords_len, sel_op); + break; + default: + BLI_assert_msg(0, "lasso select on incorrect object type"); + break; + } + + if (changed) { + DEG_id_tag_update(static_cast(vc->obedit->data), ID_RECALC_SELECT); + WM_event_add_notifier(C, NC_GEOM | ND_SELECT, vc->obedit->data); + changed_multi = true; + } + } + FOREACH_OBJECT_IN_MODE_END; + } + + WM_generic_user_data_free(wm_userdata); + + return changed_multi; +} + +/* lasso operator gives properties, but since old code works + * with short array we convert */ +static int view3d_lasso_select_exec(bContext *C, wmOperator *op) +{ + ViewContext vc; + int mcoords_len; + const int(*mcoords)[2] = WM_gesture_lasso_path_to_array(C, op, &mcoords_len); + + if (mcoords) { + Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C); + view3d_operator_needs_opengl(C); + BKE_object_update_select_id(CTX_data_main(C)); + + /* setup view context for argument to callbacks */ + ED_view3d_viewcontext_init(C, &vc, depsgraph); + + eSelectOp sel_op = static_cast(RNA_enum_get(op->ptr, "mode")); + bool changed_multi = view3d_lasso_select(C, &vc, mcoords, mcoords_len, sel_op); + + MEM_freeN((void *)mcoords); + + if (changed_multi) { + return OPERATOR_FINISHED; + } + return OPERATOR_CANCELLED; + } + return OPERATOR_PASS_THROUGH; +} + +void VIEW3D_OT_select_lasso(wmOperatorType *ot) +{ + ot->name = "Lasso Select"; + ot->description = "Select items using lasso selection"; + ot->idname = "VIEW3D_OT_select_lasso"; + + ot->invoke = WM_gesture_lasso_invoke; + ot->modal = WM_gesture_lasso_modal; + ot->exec = view3d_lasso_select_exec; + ot->poll = view3d_selectable_data; + ot->cancel = WM_gesture_lasso_cancel; + + /* flags */ + ot->flag = OPTYPE_UNDO | OPTYPE_DEPENDS_ON_CURSOR; + + /* properties */ + WM_operator_properties_gesture_lasso(ot); + WM_operator_properties_select_operation(ot); +} + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Cursor Picking + * \{ */ + +/* The max number of menu items in an object select menu */ +struct SelMenuItemF { + char idname[MAX_ID_NAME - 2]; + int icon; + Base *base_ptr; + void *item_ptr; +}; + +#define SEL_MENU_SIZE 22 +static SelMenuItemF object_mouse_select_menu_data[SEL_MENU_SIZE]; + +/* special (crappy) operator only for menu select */ +static const EnumPropertyItem *object_select_menu_enum_itemf(bContext *C, + PointerRNA *UNUSED(ptr), + PropertyRNA *UNUSED(prop), + bool *r_free) +{ + EnumPropertyItem *item = nullptr, item_tmp = {0}; + int totitem = 0; + int i = 0; + + /* Don't need context but avoid API doc-generation using this. */ + if (C == nullptr || object_mouse_select_menu_data[i].idname[0] == '\0') { + return DummyRNA_NULL_items; + } + + for (; i < SEL_MENU_SIZE && object_mouse_select_menu_data[i].idname[0] != '\0'; i++) { + item_tmp.name = object_mouse_select_menu_data[i].idname; + item_tmp.identifier = object_mouse_select_menu_data[i].idname; + item_tmp.value = i; + item_tmp.icon = object_mouse_select_menu_data[i].icon; + RNA_enum_item_add(&item, &totitem, &item_tmp); + } + + RNA_enum_item_end(&item, &totitem); + *r_free = true; + + return item; +} + +static int object_select_menu_exec(bContext *C, wmOperator *op) +{ + const int name_index = RNA_enum_get(op->ptr, "name"); + const bool extend = RNA_boolean_get(op->ptr, "extend"); + const bool deselect = RNA_boolean_get(op->ptr, "deselect"); + const bool toggle = RNA_boolean_get(op->ptr, "toggle"); + bool changed = false; + const char *name = object_mouse_select_menu_data[name_index].idname; + + View3D *v3d = CTX_wm_view3d(C); + ViewLayer *view_layer = CTX_data_view_layer(C); + const Base *oldbasact = BASACT(view_layer); + + Base *basact = nullptr; + CTX_DATA_BEGIN (C, Base *, base, selectable_bases) { + /* This is a bit dodgy, there should only be ONE object with this name, + * but library objects can mess this up. */ + if (STREQ(name, base->object->id.name + 2)) { + basact = base; + break; + } + } + CTX_DATA_END; + + if (basact == nullptr) { + return OPERATOR_CANCELLED; + } + UNUSED_VARS_NDEBUG(v3d); + BLI_assert(BASE_SELECTABLE(v3d, basact)); + + if (extend) { + ED_object_base_select(basact, BA_SELECT); + changed = true; + } + else if (deselect) { + ED_object_base_select(basact, BA_DESELECT); + changed = true; + } + else if (toggle) { + if (basact->flag & BASE_SELECTED) { + if (basact == oldbasact) { + ED_object_base_select(basact, BA_DESELECT); + changed = true; + } + } + else { + ED_object_base_select(basact, BA_SELECT); + changed = true; + } + } + else { + object_deselect_all_except(view_layer, basact); + ED_object_base_select(basact, BA_SELECT); + changed = true; + } + + if ((oldbasact != basact)) { + ED_object_base_activate(C, basact); + } + + /* weak but ensures we activate menu again before using the enum */ + memset(object_mouse_select_menu_data, 0, sizeof(object_mouse_select_menu_data)); + + /* undo? */ + if (changed) { + Scene *scene = CTX_data_scene(C); + DEG_id_tag_update(&scene->id, ID_RECALC_SELECT); + WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, scene); + + ED_outliner_select_sync_from_object_tag(C); + + return OPERATOR_FINISHED; + } + return OPERATOR_CANCELLED; +} + +void VIEW3D_OT_select_menu(wmOperatorType *ot) +{ + PropertyRNA *prop; + + /* identifiers */ + ot->name = "Select Menu"; + ot->description = "Menu object selection"; + ot->idname = "VIEW3D_OT_select_menu"; + + /* api callbacks */ + ot->invoke = WM_menu_invoke; + ot->exec = object_select_menu_exec; + + /* flags */ + ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO; + + /* #Object.id.name to select (dynamic enum). */ + prop = RNA_def_enum(ot->srna, "name", DummyRNA_NULL_items, 0, "Object Name", ""); + RNA_def_enum_funcs(prop, object_select_menu_enum_itemf); + RNA_def_property_flag(prop, (PropertyFlag)(PROP_HIDDEN | PROP_ENUM_NO_TRANSLATE)); + ot->prop = prop; + + prop = RNA_def_boolean(ot->srna, "extend", 0, "Extend", ""); + RNA_def_property_flag(prop, PROP_SKIP_SAVE); + prop = RNA_def_boolean(ot->srna, "deselect", 0, "Deselect", ""); + RNA_def_property_flag(prop, PROP_SKIP_SAVE); + prop = RNA_def_boolean(ot->srna, "toggle", 0, "Toggle", ""); + RNA_def_property_flag(prop, PROP_SKIP_SAVE); +} + +/** + * \return True when a menu was activated. + */ +static bool object_mouse_select_menu(bContext *C, + ViewContext *vc, + const GPUSelectResult *buffer, + const int hits, + const int mval[2], + const SelectPick_Params *params, + Base **r_basact) +{ + int base_count = 0; + bool ok; + LinkNodePair linklist = {nullptr, nullptr}; + + /* handle base->object->select_id */ + CTX_DATA_BEGIN (C, Base *, base, selectable_bases) { + ok = false; + + /* two selection methods, the CTRL select uses max dist of 15 */ + if (buffer) { + for (int a = 0; a < hits; a++) { + /* index was converted */ + if (base->object->runtime.select_id == (buffer[a].id & ~0xFFFF0000)) { + ok = true; + break; + } + } + } + else { + const int dist = 15 * U.pixelsize; + if (ED_view3d_project_base(vc->region, base) == V3D_PROJ_RET_OK) { + const int delta_px[2] = {base->sx - mval[0], base->sy - mval[1]}; + if (len_manhattan_v2_int(delta_px) < dist) { + ok = true; + } + } + } + + if (ok) { + base_count++; + BLI_linklist_append(&linklist, base); + + if (base_count == SEL_MENU_SIZE) { + break; + } + } + } + CTX_DATA_END; + + *r_basact = nullptr; + + if (base_count == 0) { + return false; + } + if (base_count == 1) { + Base *base = (Base *)linklist.list->link; + BLI_linklist_free(linklist.list, nullptr); + *r_basact = base; + return false; + } + + /* UI, full in static array values that we later use in an enum function */ + LinkNode *node; + int i; + + memset(object_mouse_select_menu_data, 0, sizeof(object_mouse_select_menu_data)); + + for (node = linklist.list, i = 0; node; node = node->next, i++) { + Base *base = static_cast(node->link); + Object *ob = base->object; + const char *name = ob->id.name + 2; + + BLI_strncpy(object_mouse_select_menu_data[i].idname, name, MAX_ID_NAME - 2); + object_mouse_select_menu_data[i].icon = UI_icon_from_id(&ob->id); + } + + wmOperatorType *ot = WM_operatortype_find("VIEW3D_OT_select_menu", false); + PointerRNA ptr; + + WM_operator_properties_create_ptr(&ptr, ot); + RNA_boolean_set(&ptr, "extend", params->sel_op == SEL_OP_ADD); + RNA_boolean_set(&ptr, "deselect", params->sel_op == SEL_OP_SUB); + RNA_boolean_set(&ptr, "toggle", params->sel_op == SEL_OP_XOR); + WM_operator_name_call_ptr(C, ot, WM_OP_INVOKE_DEFAULT, &ptr, nullptr); + WM_operator_properties_free(&ptr); + + BLI_linklist_free(linklist.list, nullptr); + return true; +} + +static int bone_select_menu_exec(bContext *C, wmOperator *op) +{ + const int name_index = RNA_enum_get(op->ptr, "name"); + + SelectPick_Params params{}; + params.sel_op = ED_select_op_from_operator(op->ptr); + + View3D *v3d = CTX_wm_view3d(C); + ViewLayer *view_layer = CTX_data_view_layer(C); + const Base *oldbasact = BASACT(view_layer); + + Base *basact = object_mouse_select_menu_data[name_index].base_ptr; + + if (basact == nullptr) { + return OPERATOR_CANCELLED; + } + + BLI_assert(BASE_SELECTABLE(v3d, basact)); + + if (basact->object->mode & OB_MODE_EDIT) { + EditBone *ebone = (EditBone *)object_mouse_select_menu_data[name_index].item_ptr; + ED_armature_edit_select_pick_bone(C, basact, ebone, BONE_SELECTED, ¶ms); + } + else { + bPoseChannel *pchan = (bPoseChannel *)object_mouse_select_menu_data[name_index].item_ptr; + ED_armature_pose_select_pick_bone(view_layer, v3d, basact->object, pchan->bone, ¶ms); + } + + /* Weak but ensures we activate the menu again before using the enum. */ + memset(object_mouse_select_menu_data, 0, sizeof(object_mouse_select_menu_data)); + + /* We make the armature selected: + * Not-selected active object in pose-mode won't work well for tools. */ + ED_object_base_select(basact, BA_SELECT); + + WM_event_add_notifier(C, NC_OBJECT | ND_BONE_SELECT, basact->object); + WM_event_add_notifier(C, NC_OBJECT | ND_BONE_ACTIVE, basact->object); + + /* In weight-paint, we use selected bone to select vertex-group, + * so don't switch to new active object. */ + if (oldbasact) { + if (basact->object->mode & OB_MODE_EDIT) { + /* Pass. */ + } + else if (oldbasact->object->mode & OB_MODE_ALL_WEIGHT_PAINT) { + /* Prevent activating. + * Selection causes this to be considered the 'active' pose in weight-paint mode. + * Eventually this limitation may be removed. + * For now, de-select all other pose objects deforming this mesh. */ + ED_armature_pose_select_in_wpaint_mode(view_layer, basact); + } + else { + if (oldbasact != basact) { + ED_object_base_activate(C, basact); + } + } + } + + /* Undo? */ + Scene *scene = CTX_data_scene(C); + DEG_id_tag_update(&scene->id, ID_RECALC_SELECT); + DEG_id_tag_update(&scene->id, ID_RECALC_BASE_FLAGS); + WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, scene); + + ED_outliner_select_sync_from_object_tag(C); + + return OPERATOR_FINISHED; +} + +void VIEW3D_OT_bone_select_menu(wmOperatorType *ot) +{ + PropertyRNA *prop; + + /* identifiers */ + ot->name = "Select Menu"; + ot->description = "Menu bone selection"; + ot->idname = "VIEW3D_OT_bone_select_menu"; + + /* api callbacks */ + ot->invoke = WM_menu_invoke; + ot->exec = bone_select_menu_exec; + + /* flags */ + ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO; + + /* #Object.id.name to select (dynamic enum). */ + prop = RNA_def_enum(ot->srna, "name", DummyRNA_NULL_items, 0, "Bone Name", ""); + RNA_def_enum_funcs(prop, object_select_menu_enum_itemf); + RNA_def_property_flag(prop, (PropertyFlag)(PROP_HIDDEN | PROP_ENUM_NO_TRANSLATE)); + ot->prop = prop; + + prop = RNA_def_boolean(ot->srna, "extend", 0, "Extend", ""); + RNA_def_property_flag(prop, PROP_SKIP_SAVE); + prop = RNA_def_boolean(ot->srna, "deselect", 0, "Deselect", ""); + RNA_def_property_flag(prop, PROP_SKIP_SAVE); + prop = RNA_def_boolean(ot->srna, "toggle", 0, "Toggle", ""); + RNA_def_property_flag(prop, PROP_SKIP_SAVE); +} + +/** + * \return True when a menu was activated. + */ +static bool bone_mouse_select_menu(bContext *C, + const GPUSelectResult *buffer, + const int hits, + const bool is_editmode, + const SelectPick_Params *params) +{ + BLI_assert(buffer); + + int bone_count = 0; + LinkNodePair base_list = {nullptr, nullptr}; + LinkNodePair bone_list = {nullptr, nullptr}; + GSet *added_bones = BLI_gset_ptr_new("Bone mouse select menu"); + + /* Select logic taken from ed_armature_pick_bone_from_selectbuffer_impl in armature_select.c */ + for (int a = 0; a < hits; a++) { + void *bone_ptr = nullptr; + Base *bone_base = nullptr; + uint hitresult = buffer[a].id; + + if (!(hitresult & BONESEL_ANY)) { + /* To avoid including objects in selection. */ + continue; + } + + hitresult &= ~BONESEL_ANY; + const uint hit_object = hitresult & 0xFFFF; + + /* Find the hit bone base (armature object). */ + CTX_DATA_BEGIN (C, Base *, base, selectable_bases) { + if (base->object->runtime.select_id == hit_object) { + bone_base = base; + break; + } + } + CTX_DATA_END; + + if (!bone_base) { + continue; + } + + /* Determine what the current bone is */ + if (is_editmode) { + EditBone *ebone; + const uint hit_bone = (hitresult & ~BONESEL_ANY) >> 16; + bArmature *arm = static_cast(bone_base->object->data); + ebone = static_cast(BLI_findlink(arm->edbo, hit_bone)); + if (ebone && !(ebone->flag & BONE_UNSELECTABLE)) { + bone_ptr = ebone; + } + } + else { + bPoseChannel *pchan; + const uint hit_bone = (hitresult & ~BONESEL_ANY) >> 16; + pchan = static_cast( + BLI_findlink(&bone_base->object->pose->chanbase, hit_bone)); + if (pchan && !(pchan->bone->flag & BONE_UNSELECTABLE)) { + bone_ptr = pchan; + } + } + + if (!bone_ptr) { + continue; + } + /* We can hit a bone multiple times, so make sure we are not adding an already included bone + * to the list. */ + const bool is_duplicate_bone = BLI_gset_haskey(added_bones, bone_ptr); + + if (!is_duplicate_bone) { + bone_count++; + BLI_linklist_append(&base_list, bone_base); + BLI_linklist_append(&bone_list, bone_ptr); + BLI_gset_insert(added_bones, bone_ptr); + + if (bone_count == SEL_MENU_SIZE) { + break; + } + } + } + + BLI_gset_free(added_bones, nullptr); + + if (bone_count == 0) { + return false; + } + if (bone_count == 1) { + BLI_linklist_free(base_list.list, nullptr); + BLI_linklist_free(bone_list.list, nullptr); + return false; + } + + /* UI, full in static array values that we later use in an enum function */ + LinkNode *bone_node, *base_node; + int i; + + memset(object_mouse_select_menu_data, 0, sizeof(object_mouse_select_menu_data)); + + for (base_node = base_list.list, bone_node = bone_list.list, i = 0; bone_node; + base_node = base_node->next, bone_node = bone_node->next, i++) { + char *name; + + object_mouse_select_menu_data[i].base_ptr = static_cast(base_node->link); + + if (is_editmode) { + EditBone *ebone = static_cast(bone_node->link); + object_mouse_select_menu_data[i].item_ptr = ebone; + name = ebone->name; + } + else { + bPoseChannel *pchan = static_cast(bone_node->link); + object_mouse_select_menu_data[i].item_ptr = pchan; + name = pchan->name; + } + + BLI_strncpy(object_mouse_select_menu_data[i].idname, name, MAX_ID_NAME - 2); + object_mouse_select_menu_data[i].icon = ICON_BONE_DATA; + } + + wmOperatorType *ot = WM_operatortype_find("VIEW3D_OT_bone_select_menu", false); + PointerRNA ptr; + + WM_operator_properties_create_ptr(&ptr, ot); + RNA_boolean_set(&ptr, "extend", params->sel_op == SEL_OP_ADD); + RNA_boolean_set(&ptr, "deselect", params->sel_op == SEL_OP_SUB); + RNA_boolean_set(&ptr, "toggle", params->sel_op == SEL_OP_XOR); + WM_operator_name_call_ptr(C, ot, WM_OP_INVOKE_DEFAULT, &ptr, nullptr); + WM_operator_properties_free(&ptr); + + BLI_linklist_free(base_list.list, nullptr); + BLI_linklist_free(bone_list.list, nullptr); + return true; +} + +static bool selectbuffer_has_bones(const GPUSelectResult *buffer, const uint hits) +{ + for (uint i = 0; i < hits; i++) { + if (buffer[i].id & 0xFFFF0000) { + return true; + } + } + return false; +} + +/* utility function for mixed_bones_object_selectbuffer */ +static int selectbuffer_ret_hits_15(GPUSelectResult *UNUSED(buffer), const int hits15) +{ + return hits15; +} + +static int selectbuffer_ret_hits_9(GPUSelectResult *buffer, const int hits15, const int hits9) +{ + const int ofs = hits15; + memcpy(buffer, buffer + ofs, hits9 * sizeof(GPUSelectResult)); + return hits9; +} + +static int selectbuffer_ret_hits_5(GPUSelectResult *buffer, + const int hits15, + const int hits9, + const int hits5) +{ + const int ofs = hits15 + hits9; + memcpy(buffer, buffer + ofs, hits5 * sizeof(GPUSelectResult)); + return hits5; +} + +/** + * Populate a select buffer with objects and bones, if there are any. + * Checks three selection levels and compare. + * + * \param do_nearest_xray_if_supported: When set, read in hits that don't stop + * at the nearest surface. The hits must still be ordered by depth. + * Needed so we can step to the next, non-active object when it's already selected, see: T76445. + */ +static int mixed_bones_object_selectbuffer(ViewContext *vc, + GPUSelectResult *buffer, + const int buffer_len, + const int mval[2], + eV3DSelectObjectFilter select_filter, + bool do_nearest, + bool do_nearest_xray_if_supported, + const bool do_material_slot_selection) +{ + rcti rect; + int hits15, hits9 = 0, hits5 = 0; + bool has_bones15 = false, has_bones9 = false, has_bones5 = false; + + eV3DSelectMode select_mode = (do_nearest ? VIEW3D_SELECT_PICK_NEAREST : VIEW3D_SELECT_PICK_ALL); + int hits = 0; + + if (do_nearest_xray_if_supported) { + if ((U.gpu_flag & USER_GPU_FLAG_NO_DEPT_PICK) == 0) { + select_mode = VIEW3D_SELECT_PICK_ALL; + } + } + + /* we _must_ end cache before return, use 'goto finally' */ + view3d_opengl_select_cache_begin(); + + BLI_rcti_init_pt_radius(&rect, mval, 14); + hits15 = view3d_opengl_select_ex( + vc, buffer, buffer_len, &rect, select_mode, select_filter, do_material_slot_selection); + if (hits15 == 1) { + hits = selectbuffer_ret_hits_15(buffer, hits15); + goto finally; + } + else if (hits15 > 0) { + int ofs; + has_bones15 = selectbuffer_has_bones(buffer, hits15); + + ofs = hits15; + BLI_rcti_init_pt_radius(&rect, mval, 9); + hits9 = view3d_opengl_select( + vc, buffer + ofs, buffer_len - ofs, &rect, select_mode, select_filter); + if (hits9 == 1) { + hits = selectbuffer_ret_hits_9(buffer, hits15, hits9); + goto finally; + } + else if (hits9 > 0) { + has_bones9 = selectbuffer_has_bones(buffer + ofs, hits9); + + ofs += hits9; + BLI_rcti_init_pt_radius(&rect, mval, 5); + hits5 = view3d_opengl_select( + vc, buffer + ofs, buffer_len - ofs, &rect, select_mode, select_filter); + if (hits5 == 1) { + hits = selectbuffer_ret_hits_5(buffer, hits15, hits9, hits5); + goto finally; + } + else if (hits5 > 0) { + has_bones5 = selectbuffer_has_bones(buffer + ofs, hits5); + } + } + + if (has_bones5) { + hits = selectbuffer_ret_hits_5(buffer, hits15, hits9, hits5); + goto finally; + } + else if (has_bones9) { + hits = selectbuffer_ret_hits_9(buffer, hits15, hits9); + goto finally; + } + else if (has_bones15) { + hits = selectbuffer_ret_hits_15(buffer, hits15); + goto finally; + } + + if (hits5 > 0) { + hits = selectbuffer_ret_hits_5(buffer, hits15, hits9, hits5); + goto finally; + } + else if (hits9 > 0) { + hits = selectbuffer_ret_hits_9(buffer, hits15, hits9); + goto finally; + } + else { + hits = selectbuffer_ret_hits_15(buffer, hits15); + goto finally; + } + } + +finally: + view3d_opengl_select_cache_end(); + return hits; +} + +static int mixed_bones_object_selectbuffer_extended(ViewContext *vc, + GPUSelectResult *buffer, + const int buffer_len, + const int mval[2], + eV3DSelectObjectFilter select_filter, + bool use_cycle, + bool enumerate, + bool *r_do_nearest) +{ + bool do_nearest = false; + View3D *v3d = vc->v3d; + + /* define if we use solid nearest select or not */ + if (use_cycle) { + /* Update the coordinates (even if the return value isn't used). */ + const bool has_motion = WM_cursor_test_motion_and_update(mval); + if (!XRAY_ACTIVE(v3d)) { + do_nearest = has_motion; + } + } + else { + if (!XRAY_ACTIVE(v3d)) { + do_nearest = true; + } + } + + if (r_do_nearest) { + *r_do_nearest = do_nearest; + } + + do_nearest = do_nearest && !enumerate; + + int hits = mixed_bones_object_selectbuffer( + vc, buffer, buffer_len, mval, select_filter, do_nearest, true, false); + + return hits; +} + +/** + * Compare result of 'GPU_select': 'GPUSelectResult', + * Needed for stable sorting, so cycling through all items near the cursor behaves predictably. + */ +static int gpu_select_buffer_depth_id_cmp(const void *sel_a_p, const void *sel_b_p) +{ + GPUSelectResult *a = (GPUSelectResult *)sel_a_p; + GPUSelectResult *b = (GPUSelectResult *)sel_b_p; + + if (a->depth < b->depth) { + return -1; + } + if (a->depth > b->depth) { + return 1; + } + + /* Depths match, sort by id. */ + uint sel_a = a->id; + uint sel_b = b->id; + +#ifdef __BIG_ENDIAN__ + BLI_endian_switch_uint32(&sel_a); + BLI_endian_switch_uint32(&sel_b); +#endif + + if (sel_a < sel_b) { + return -1; + } + if (sel_a > sel_b) { + return 1; + } + return 0; +} + +/** + * \param has_bones: When true, skip non-bone hits, also allow bases to be used + * that are visible but not select-able, + * since you may be in pose mode with an un-selectable object. + * + * \return the active base or nullptr. + */ +static Base *mouse_select_eval_buffer(ViewContext *vc, + const GPUSelectResult *buffer, + int hits, + bool do_nearest, + bool has_bones, + bool do_bones_get_priotity, + int *r_select_id_subelem) +{ + ViewLayer *view_layer = vc->view_layer; + View3D *v3d = vc->v3d; + int a; + + bool found = false; + int select_id = 0; + int select_id_subelem = 0; + + if (do_nearest) { + uint min = 0xFFFFFFFF; + int hit_index = -1; + + if (has_bones && do_bones_get_priotity) { + /* we skip non-bone hits */ + for (a = 0; a < hits; a++) { + if (min > buffer[a].depth && (buffer[a].id & 0xFFFF0000)) { + min = buffer[a].depth; + hit_index = a; + } + } + } + else { + + for (a = 0; a < hits; a++) { + /* Any object. */ + if (min > buffer[a].depth) { + min = buffer[a].depth; + hit_index = a; + } + } + } + + if (hit_index != -1) { + select_id = buffer[hit_index].id & 0xFFFF; + select_id_subelem = (buffer[hit_index].id & 0xFFFF0000) >> 16; + found = true; + /* No need to set `min` to `buffer[hit_index].depth`, it's not used from now on. */ + } + } + else { + + { + GPUSelectResult *buffer_sorted = static_cast( + MEM_mallocN(sizeof(*buffer_sorted) * hits, __func__)); + memcpy(buffer_sorted, buffer, sizeof(*buffer_sorted) * hits); + /* Remove non-bone objects. */ + if (has_bones && do_bones_get_priotity) { + /* Loop backwards to reduce re-ordering. */ + for (a = hits - 1; a >= 0; a--) { + if ((buffer_sorted[a].id & 0xFFFF0000) == 0) { + buffer_sorted[a] = buffer_sorted[--hits]; + } + } + } + qsort(buffer_sorted, hits, sizeof(GPUSelectResult), gpu_select_buffer_depth_id_cmp); + buffer = buffer_sorted; + } + + int hit_index = -1; + + /* It's possible there are no hits (all objects contained bones). */ + if (hits > 0) { + /* Only exclude active object when it is selected. */ + if (BASACT(view_layer) && (BASACT(view_layer)->flag & BASE_SELECTED)) { + const int select_id_active = BASACT(view_layer)->object->runtime.select_id; + for (int i_next = 0, i_prev = hits - 1; i_next < hits; i_prev = i_next++) { + if ((select_id_active == (buffer[i_prev].id & 0xFFFF)) && + (select_id_active != (buffer[i_next].id & 0xFFFF))) { + hit_index = i_next; + break; + } + } + } + + /* When the active object is unselected or not in `buffer`, use the nearest. */ + if (hit_index == -1) { + /* Just pick the nearest. */ + hit_index = 0; + } + } + + if (hit_index != -1) { + select_id = buffer[hit_index].id & 0xFFFF; + select_id_subelem = (buffer[hit_index].id & 0xFFFF0000) >> 16; + found = true; + } + MEM_freeN((void *)buffer); + } + + Base *basact = nullptr; + if (found) { + for (Base *base = FIRSTBASE(view_layer); base; base = base->next) { + if (has_bones ? BASE_VISIBLE(v3d, base) : BASE_SELECTABLE(v3d, base)) { + if (base->object->runtime.select_id == select_id) { + basact = base; + break; + } + } + } + + if (basact && r_select_id_subelem) { + *r_select_id_subelem = select_id_subelem; + } + } + + return basact; +} + +static Base *mouse_select_object_center(ViewContext *vc, Base *startbase, const int mval[2]) +{ + ARegion *region = vc->region; + ViewLayer *view_layer = vc->view_layer; + View3D *v3d = vc->v3d; + + Base *oldbasact = BASACT(view_layer); + + const float mval_fl[2] = {(float)mval[0], (float)mval[1]}; + float dist = ED_view3d_select_dist_px() * 1.3333f; + Base *basact = nullptr; + + /* Put the active object at a disadvantage to cycle through other objects. */ + const float penalty_dist = 10.0f * UI_DPI_FAC; + Base *base = startbase; + while (base) { + if (BASE_SELECTABLE(v3d, base)) { + float screen_co[2]; + if (ED_view3d_project_float_global( + region, base->object->obmat[3], screen_co, V3D_PROJ_TEST_CLIP_DEFAULT) == + V3D_PROJ_RET_OK) { + float dist_test = len_manhattan_v2v2(mval_fl, screen_co); + if (base == oldbasact) { + dist_test += penalty_dist; + } + if (dist_test < dist) { + dist = dist_test; + basact = base; + } + } + } + base = base->next; + + if (base == nullptr) { + base = FIRSTBASE(view_layer); + } + if (base == startbase) { + break; + } + } + return basact; +} + +static Base *ed_view3d_give_base_under_cursor_ex(bContext *C, + const int mval[2], + int *r_material_slot) +{ + Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C); + ViewContext vc; + Base *basact = nullptr; + GPUSelectResult buffer[MAXPICKELEMS]; + + /* setup view context for argument to callbacks */ + view3d_operator_needs_opengl(C); + BKE_object_update_select_id(CTX_data_main(C)); + + ED_view3d_viewcontext_init(C, &vc, depsgraph); + + const bool do_nearest = !XRAY_ACTIVE(vc.v3d); + const bool do_material_slot_selection = r_material_slot != nullptr; + const int hits = mixed_bones_object_selectbuffer(&vc, + buffer, + ARRAY_SIZE(buffer), + mval, + VIEW3D_SELECT_FILTER_NOP, + do_nearest, + false, + do_material_slot_selection); + + if (hits > 0) { + const bool has_bones = (r_material_slot == nullptr) && selectbuffer_has_bones(buffer, hits); + basact = mouse_select_eval_buffer( + &vc, buffer, hits, do_nearest, has_bones, true, r_material_slot); + } + + return basact; +} + +Base *ED_view3d_give_base_under_cursor(bContext *C, const int mval[2]) +{ + return ed_view3d_give_base_under_cursor_ex(C, mval, nullptr); +} + +Object *ED_view3d_give_object_under_cursor(bContext *C, const int mval[2]) +{ + Base *base = ED_view3d_give_base_under_cursor(C, mval); + if (base) { + return base->object; + } + return nullptr; +} + +Object *ED_view3d_give_material_slot_under_cursor(bContext *C, + const int mval[2], + int *r_material_slot) +{ + Base *base = ed_view3d_give_base_under_cursor_ex(C, mval, r_material_slot); + if (base) { + return base->object; + } + return nullptr; +} + +bool ED_view3d_is_object_under_cursor(bContext *C, const int mval[2]) +{ + return ED_view3d_give_object_under_cursor(C, mval) != nullptr; +} + +static void deselect_all_tracks(MovieTracking *tracking) +{ + MovieTrackingObject *object; + + object = static_cast(tracking->objects.first); + while (object) { + ListBase *tracksbase = BKE_tracking_object_get_tracks(tracking, object); + MovieTrackingTrack *track = static_cast(tracksbase->first); + + while (track) { + BKE_tracking_track_deselect(track, TRACK_AREA_ALL); + + track = track->next; + } + + object = object->next; + } +} + +static bool ed_object_select_pick_camera_track(bContext *C, + Scene *scene, + Base *basact, + MovieClip *clip, + const GPUSelectResult *buffer, + const short hits, + const SelectPick_Params *params) +{ + bool changed = false; + bool found = false; + + MovieTracking *tracking = &clip->tracking; + ListBase *tracksbase = nullptr; + MovieTrackingTrack *track = nullptr; + + for (int i = 0; i < hits; i++) { + const int hitresult = buffer[i].id; + + /* If there's bundles in buffer select bundles first, + * so non-camera elements should be ignored in buffer. */ + if (basact->object->runtime.select_id != (hitresult & 0xFFFF)) { + continue; + } + /* Index of bundle is 1<<16-based. if there's no "bone" index + * in height word, this buffer value belongs to camera. not to bundle. */ + if ((hitresult & 0xFFFF0000) == 0) { + continue; + } + + track = BKE_tracking_track_get_indexed(&clip->tracking, hitresult >> 16, &tracksbase); + found = true; + break; + } + + /* Note `params->deselect_all` is ignored for tracks as in this case + * all objects will be de-selected (not tracks). */ + if (params->sel_op == SEL_OP_SET) { + if ((found && params->select_passthrough) && TRACK_SELECTED(track)) { + found = false; + } + else if (found /* `|| params->deselect_all` */) { + /* Deselect everything. */ + deselect_all_tracks(tracking); + changed = true; + } + } + + if (found) { + switch (params->sel_op) { + case SEL_OP_ADD: { + BKE_tracking_track_select(tracksbase, track, TRACK_AREA_ALL, true); + break; + } + case SEL_OP_SUB: { + BKE_tracking_track_deselect(track, TRACK_AREA_ALL); + break; + } + case SEL_OP_XOR: { + if (TRACK_SELECTED(track)) { + BKE_tracking_track_deselect(track, TRACK_AREA_ALL); + } + else { + BKE_tracking_track_select(tracksbase, track, TRACK_AREA_ALL, true); + } + break; + } + case SEL_OP_SET: { + BKE_tracking_track_select(tracksbase, track, TRACK_AREA_ALL, false); + break; + } + case SEL_OP_AND: { + BLI_assert_unreachable(); /* Doesn't make sense for picking. */ + break; + } + } + + DEG_id_tag_update(&scene->id, ID_RECALC_SELECT); + DEG_id_tag_update(&clip->id, ID_RECALC_SELECT); + WM_event_add_notifier(C, NC_MOVIECLIP | ND_SELECT, track); + WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, scene); + + changed = true; + } + + return changed || found; +} + +/** + * Cursor selection picking for object & pose-mode. + * + * \param mval: Region relative cursor coordinates. + * \param params: Selection parameters. + * \param center: Select by the cursors on-screen distances to the center/origin + * instead of the geometry any other contents of the item being selected. + * This could be used to select by bones by their origin too, currently it's only used for objects. + * \param enumerate: Show a menu for objects at the cursor location. + * Otherwise fall-through to non-menu selection. + * \param object_only: Only select objects (not bones / track markers). + */ +static bool ed_object_select_pick(bContext *C, + const int mval[2], + const SelectPick_Params *params, + const bool center, + const bool enumerate, + const bool object_only) +{ + Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C); + ViewContext vc; + /* Setup view context for argument to callbacks. */ + ED_view3d_viewcontext_init(C, &vc, depsgraph); + + Scene *scene = vc.scene; + View3D *v3d = vc.v3d; + + /* Menu activation may find a base to make active (if it only finds a single item to select). */ + Base *basact_override = nullptr; + + const bool is_obedit = (vc.obedit != nullptr); + if (object_only) { + /* Signal for #view3d_opengl_select to skip edit-mode objects. */ + vc.obedit = nullptr; + } + + /* Set for GPU depth buffer picking, leave null when selecting by center. */ + struct GPUData { + GPUSelectResult buffer[MAXPICKELEMS]; + int hits; + bool do_nearest; + bool has_bones; + } *gpu = nullptr; + + /* First handle menu selection, early exit if a menu opens + * since this takes ownership of the selection action. + * + * Even when there is no menu `basact_override` may be set to avoid having to re-find + * the item under the cursor. */ + + if (center == false) { + gpu = MEM_new(__func__); + gpu->do_nearest = false; + gpu->has_bones = false; + + /* If objects have pose-mode set, the bones are in the same selection buffer. */ + const eV3DSelectObjectFilter select_filter = ((object_only == false) ? + ED_view3d_select_filter_from_mode(scene, + vc.obact) : + VIEW3D_SELECT_FILTER_NOP); + gpu->hits = mixed_bones_object_selectbuffer_extended(&vc, + gpu->buffer, + ARRAY_SIZE(gpu->buffer), + mval, + select_filter, + true, + enumerate, + &gpu->do_nearest); + gpu->has_bones = (object_only && gpu->hits > 0) ? + false : + selectbuffer_has_bones(gpu->buffer, gpu->hits); + } + + /* First handle menu selection, early exit when a menu was opened. + * Otherwise fall through to regular selection. */ + if (enumerate) { + bool has_menu = false; + if (center) { + if (object_mouse_select_menu(C, &vc, nullptr, 0, mval, params, &basact_override)) { + has_menu = true; + } + } + else { + if (gpu->hits != 0) { + if (gpu->has_bones && bone_mouse_select_menu(C, gpu->buffer, gpu->hits, false, params)) { + has_menu = true; + } + else if (object_mouse_select_menu( + C, &vc, gpu->buffer, gpu->hits, mval, params, &basact_override)) { + has_menu = true; + } + } + } + + /* Let the menu handle any further actions. */ + if (has_menu) { + if (gpu != nullptr) { + MEM_freeN(gpu); + } + return false; + } + } + + /* No menu, continue with selection. */ + + ViewLayer *view_layer = vc.view_layer; + /* Don't set when the context has no active object (hidden), see: T60807. */ + const Base *oldbasact = vc.obact ? BASACT(view_layer) : nullptr; + /* Always start list from `basact` when cycling the selection. */ + Base *startbase = (oldbasact && oldbasact->next) ? oldbasact->next : FIRSTBASE(view_layer); + + /* The next object's base to make active. */ + Base *basact = nullptr; + const eObjectMode object_mode = oldbasact ? static_cast(oldbasact->object->mode) : + OB_MODE_OBJECT; + + /* When enabled, don't attempt any further selection. */ + bool handled = false; + + /* Split `changed` into data-types so their associated updates can be properly performed. + * This is also needed as multiple changes may happen at once. + * Selecting a pose-bone or track can also select the object for e.g. */ + bool changed_object = false; + bool changed_pose = false; + bool changed_track = false; + + /* Handle setting the new base active (even when `handled == true`). */ + bool use_activate_selected_base = false; + + if (center) { + if (basact_override) { + basact = basact_override; + } + else { + basact = mouse_select_object_center(&vc, startbase, mval); + } + } + else { + if (basact_override) { + basact = basact_override; + } + else { + /* Regarding bone priority. + * + * - When in pose-bone, it's useful that any selection containing a bone + * gets priority over other geometry (background scenery for example). + * + * - When in object-mode, don't prioritize bones as it would cause + * pose-objects behind other objects to get priority + * (mainly noticeable when #SCE_OBJECT_MODE_LOCK is disabled). + * + * This way prioritizing based on pose-mode has a bias to stay in pose-mode + * without having to enforce this through locking the object mode. */ + bool do_bones_get_priotity = (object_mode & OB_MODE_POSE) != 0; + + basact = (gpu->hits > 0) ? mouse_select_eval_buffer(&vc, + gpu->buffer, + gpu->hits, + gpu->do_nearest, + gpu->has_bones, + do_bones_get_priotity, + nullptr) : + nullptr; + } + + /* Select pose-bones or camera-tracks. */ + if (((gpu->hits > 0) && gpu->has_bones) || + /* Special case, even when there are no hits, pose logic may de-select all bones. */ + ((gpu->hits == 0) && (object_mode & OB_MODE_POSE))) { + + if (basact && (gpu->has_bones && (basact->object->type == OB_CAMERA))) { + MovieClip *clip = BKE_object_movieclip_get(scene, basact->object, false); + if (clip != nullptr) { + if (ed_object_select_pick_camera_track( + C, scene, basact, clip, gpu->buffer, gpu->hits, params)) { + ED_object_base_select(basact, BA_SELECT); + /* Don't set `handled` here as the object activation may be necessary. */ + changed_object = true; + + changed_track = true; + } + else { + /* Fallback to regular object selection if no new bundles were selected, + * allows to select object parented to reconstruction object. */ + basact = mouse_select_eval_buffer( + &vc, gpu->buffer, gpu->hits, gpu->do_nearest, false, false, nullptr); + } + } + } + else if (ED_armature_pose_select_pick_with_buffer(view_layer, + v3d, + basact ? basact : (Base *)oldbasact, + gpu->buffer, + gpu->hits, + params, + gpu->do_nearest)) { + + changed_pose = true; + + /* When there is no `baseact` this will have operated on `oldbasact`, + * allowing #SelectPick_Params.deselect_all work in pose-mode. + * In this case no object operations are needed. */ + if (basact != nullptr) { + /* By convention the armature-object is selected when in pose-mode. + * While leaving it unselected will work, leaving pose-mode would leave the object + * active + unselected which isn't ideal when performing other actions on the object. */ + ED_object_base_select(basact, BA_SELECT); + changed_object = true; + + WM_event_add_notifier(C, NC_OBJECT | ND_BONE_SELECT, basact->object); + WM_event_add_notifier(C, NC_OBJECT | ND_BONE_ACTIVE, basact->object); + + /* In weight-paint, we use selected bone to select vertex-group. + * In this case the active object mustn't change as it would leave weight-paint mode. */ + if (oldbasact) { + if (oldbasact->object->mode & OB_MODE_ALL_WEIGHT_PAINT) { + /* Prevent activating. + * Selection causes this to be considered the 'active' pose in weight-paint mode. + * Eventually this limitation may be removed. + * For now, de-select all other pose objects deforming this mesh. */ + ED_armature_pose_select_in_wpaint_mode(view_layer, basact); + + handled = true; + } + else if ((object_mode & OB_MODE_POSE) && (basact->object->mode & OB_MODE_POSE)) { + /* Within pose-mode, keep the current selection when switching pose bones, + * this is noticeable when in pose mode with multiple objects at once. + * Where selecting the bone of a different object would de-select this one. + * After that, exiting pose-mode would only have the active armature selected. + * This matches multi-object edit-mode behavior. */ + handled = true; + + if (oldbasact != basact) { + use_activate_selected_base = true; + } + } + else { + /* Don't set `handled` here as the object selection may be necessary + * when starting out in object-mode and moving into pose-mode, + * when moving from pose to object-mode using object selection also makes sense. */ + } + } + } + } + /* Prevent bone/track selecting to pass on to object selecting. */ + if (basact == oldbasact) { + handled = true; + } + } + } + + if (handled == false) { + if (scene->toolsettings->object_flag & SCE_OBJECT_MODE_LOCK) { + /* No special logic in edit-mode. */ + if (is_obedit == false) { + if (basact && !BKE_object_is_mode_compat(basact->object, object_mode)) { + if (object_mode == OB_MODE_OBJECT) { + Main *bmain = vc.bmain; + ED_object_mode_generic_exit(bmain, vc.depsgraph, scene, basact->object); + } + if (!BKE_object_is_mode_compat(basact->object, object_mode)) { + basact = nullptr; + } + } + + /* Disallow switching modes, + * special exception for edit-mode - vertex-parent operator. */ + if (basact && oldbasact) { + if ((oldbasact->object->mode != basact->object->mode) && + (oldbasact->object->mode & basact->object->mode) == 0) { + basact = nullptr; + } + } + } + } + } + + /* Ensure code above doesn't change the active base. This code is already fairly involved, + * it's best if changing the active object is localized to a single place. */ + BLI_assert(oldbasact == (vc.obact ? BASACT(view_layer) : nullptr)); + + bool found = (basact != nullptr); + if ((handled == false) && (vc.obedit == nullptr)) { + /* Object-mode (pose mode will have been handled already). */ + if (params->sel_op == SEL_OP_SET) { + if ((found && params->select_passthrough) && (basact->flag & BASE_SELECTED)) { + found = false; + } + else if (found || params->deselect_all) { + /* Deselect everything. */ + /* `basact` may be nullptr. */ + if (object_deselect_all_except(view_layer, basact)) { + changed_object = true; + } + } + } + } + + if ((handled == false) && found) { + + if (vc.obedit) { + /* Only do the select (use for setting vertex parents & hooks). + * In edit-mode do not activate. */ + object_deselect_all_except(view_layer, basact); + ED_object_base_select(basact, BA_SELECT); + + changed_object = true; + } + /* Also prevent making it active on mouse selection. */ + else if (BASE_SELECTABLE(v3d, basact)) { + use_activate_selected_base |= (oldbasact != basact) && (is_obedit == false); + + switch (params->sel_op) { + case SEL_OP_ADD: { + ED_object_base_select(basact, BA_SELECT); + break; + } + case SEL_OP_SUB: { + ED_object_base_select(basact, BA_DESELECT); + break; + } + case SEL_OP_XOR: { + if (basact->flag & BASE_SELECTED) { + /* Keep selected if the base is to be activated. */ + if (use_activate_selected_base == false) { + ED_object_base_select(basact, BA_DESELECT); + } + } + else { + ED_object_base_select(basact, BA_SELECT); + } + break; + } + case SEL_OP_SET: { + object_deselect_all_except(view_layer, basact); + ED_object_base_select(basact, BA_SELECT); + break; + } + case SEL_OP_AND: { + BLI_assert_unreachable(); /* Doesn't make sense for picking. */ + break; + } + } + + changed_object = true; + } + } + + /* Perform the activation even when 'handled', since this is used to ensure + * the object from the pose-bone selected is also activated. */ + if (use_activate_selected_base && (basact != nullptr)) { + changed_object = true; + ED_object_base_activate(C, basact); /* adds notifier */ + if ((scene->toolsettings->object_flag & SCE_OBJECT_MODE_LOCK) == 0) { + WM_toolsystem_update_from_context_view3d(C); + } + } + + if (changed_object) { + DEG_id_tag_update(&scene->id, ID_RECALC_SELECT); + WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, scene); + + ED_outliner_select_sync_from_object_tag(C); + } + + if (changed_pose) { + ED_outliner_select_sync_from_pose_bone_tag(C); + } + + if (gpu != nullptr) { + MEM_freeN(gpu); + } + + return (changed_object || changed_pose || changed_track); +} + +/** + * Mouse selection in weight paint. + * Called via generic mouse select operator. + * + * \return True when pick finds an element or the selection changed. + */ +static bool ed_wpaint_vertex_select_pick(bContext *C, + const int mval[2], + const SelectPick_Params *params, + Object *obact) +{ + View3D *v3d = CTX_wm_view3d(C); + const bool use_zbuf = !XRAY_ENABLED(v3d); + + Mesh *me = static_cast(obact->data); /* already checked for nullptr */ + uint index = 0; + MVert *mv; + bool changed = false; + + bool found = ED_mesh_pick_vert(C, obact, mval, ED_MESH_PICK_DEFAULT_VERT_DIST, use_zbuf, &index); + + if (params->sel_op == SEL_OP_SET) { + if ((found && params->select_passthrough) && (me->mvert[index].flag & SELECT)) { + found = false; + } + else if (found || params->deselect_all) { + /* Deselect everything. */ + changed |= paintface_deselect_all_visible(C, obact, SEL_DESELECT, false); + } + } + + if (found) { + mv = &me->mvert[index]; + switch (params->sel_op) { + case SEL_OP_ADD: { + mv->flag |= SELECT; + break; + } + case SEL_OP_SUB: { + mv->flag &= ~SELECT; + break; + } + case SEL_OP_XOR: { + mv->flag ^= SELECT; + break; + } + case SEL_OP_SET: { + paintvert_deselect_all_visible(obact, SEL_DESELECT, false); + mv->flag |= SELECT; + break; + } + case SEL_OP_AND: { + BLI_assert_unreachable(); /* Doesn't make sense for picking. */ + break; + } + } + + /* update mselect */ + if (mv->flag & SELECT) { + BKE_mesh_mselect_active_set(me, index, ME_VSEL); + } + else { + BKE_mesh_mselect_validate(me); + } + + paintvert_flush_flags(obact); + + changed = true; + } + + if (changed) { + paintvert_tag_select_update(C, obact); + } + + return changed || found; +} + +static int view3d_select_exec(bContext *C, wmOperator *op) +{ + Scene *scene = CTX_data_scene(C); + Object *obedit = CTX_data_edit_object(C); + Object *obact = CTX_data_active_object(C); + + SelectPick_Params params{}; + ED_select_pick_params_from_operator(op->ptr, ¶ms); + + const bool vert_without_handles = RNA_boolean_get(op->ptr, "vert_without_handles"); + bool center = RNA_boolean_get(op->ptr, "center"); + bool enumerate = RNA_boolean_get(op->ptr, "enumerate"); + /* Only force object select for edit-mode to support vertex parenting, + * or paint-select to allow pose bone select with vert/face select. */ + bool object_only = (RNA_boolean_get(op->ptr, "object") && + (obedit || BKE_paint_select_elem_test(obact) || + /* so its possible to select bones in weight-paint mode (LMB select) */ + (obact && (obact->mode & OB_MODE_ALL_WEIGHT_PAINT) && + BKE_object_pose_armature_get(obact)))); + + /* This could be called "changed_or_found" since this is true when there is an element + * under the cursor to select, even if it happens that the selection & active state doesn't + * actually change. This is important so undo pushes are predictable. */ + bool changed = false; + int mval[2]; + + if (object_only) { + obedit = nullptr; + obact = nullptr; + + /* ack, this is incorrect but to do this correctly we would need an + * alternative edit-mode/object-mode keymap, this copies the functionality + * from 2.4x where Ctrl+Select in edit-mode does object select only. */ + center = false; + } + + if (obedit && enumerate) { + /* Enumerate makes no sense in edit-mode unless also explicitly picking objects or bones. + * Pass the event through so the event may be handled by loop-select for e.g. see: T100204. */ + if (obedit->type != OB_ARMATURE) { + return OPERATOR_PASS_THROUGH | OPERATOR_CANCELLED; + } + } + + RNA_int_get_array(op->ptr, "location", mval); + + view3d_operator_needs_opengl(C); + BKE_object_update_select_id(CTX_data_main(C)); + + if (obedit && object_only == false) { + if (obedit->type == OB_MESH) { + changed = EDBM_select_pick(C, mval, ¶ms); + } + else if (obedit->type == OB_ARMATURE) { + if (enumerate) { + Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C); + ViewContext vc; + ED_view3d_viewcontext_init(C, &vc, depsgraph); + + GPUSelectResult buffer[MAXPICKELEMS]; + const int hits = mixed_bones_object_selectbuffer( + &vc, buffer, ARRAY_SIZE(buffer), mval, VIEW3D_SELECT_FILTER_NOP, false, true, false); + changed = bone_mouse_select_menu(C, buffer, hits, true, ¶ms); + } + if (!changed) { + changed = ED_armature_edit_select_pick(C, mval, ¶ms); + } + } + else if (obedit->type == OB_LATTICE) { + changed = ED_lattice_select_pick(C, mval, ¶ms); + } + else if (ELEM(obedit->type, OB_CURVES_LEGACY, OB_SURF)) { + changed = ED_curve_editnurb_select_pick( + C, mval, ED_view3d_select_dist_px(), vert_without_handles, ¶ms); + } + else if (obedit->type == OB_MBALL) { + changed = ED_mball_select_pick(C, mval, ¶ms); + } + else if (obedit->type == OB_FONT) { + changed = ED_curve_editfont_select_pick(C, mval, ¶ms); + } + } + else if (obact && obact->mode & OB_MODE_PARTICLE_EDIT) { + changed = PE_mouse_particles(C, mval, ¶ms); + } + else if (obact && BKE_paint_select_face_test(obact)) { + changed = paintface_mouse_select(C, mval, ¶ms, obact); + } + else if (BKE_paint_select_vert_test(obact)) { + changed = ed_wpaint_vertex_select_pick(C, mval, ¶ms, obact); + } + else { + changed = ed_object_select_pick(C, mval, ¶ms, center, enumerate, object_only); + } + + /* Pass-through flag may be cleared, see #WM_operator_flag_only_pass_through_on_press. */ + + /* Pass-through allows tweaks + * FINISHED to signal one operator worked */ + if (changed) { + WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, scene); + return OPERATOR_PASS_THROUGH | OPERATOR_FINISHED; + } + /* Nothing selected, just passthrough. */ + return OPERATOR_PASS_THROUGH | OPERATOR_CANCELLED; +} + +static int view3d_select_invoke(bContext *C, wmOperator *op, const wmEvent *event) +{ + RNA_int_set_array(op->ptr, "location", event->mval); + + const int retval = view3d_select_exec(C, op); + + return WM_operator_flag_only_pass_through_on_press(retval, event); +} + +void VIEW3D_OT_select(wmOperatorType *ot) +{ + PropertyRNA *prop; + + /* identifiers */ + ot->name = "Select"; + ot->description = "Select and activate item(s)"; + ot->idname = "VIEW3D_OT_select"; + + /* api callbacks */ + ot->invoke = view3d_select_invoke; + ot->exec = view3d_select_exec; + ot->poll = ED_operator_view3d_active; + ot->get_name = ED_select_pick_get_name; + + /* flags */ + ot->flag = OPTYPE_UNDO; + + /* properties */ + WM_operator_properties_mouse_select(ot); + + prop = RNA_def_boolean( + ot->srna, + "center", + 0, + "Center", + "Use the object center when selecting, in edit mode used to extend object selection"); + RNA_def_property_flag(prop, PROP_SKIP_SAVE); + prop = RNA_def_boolean( + ot->srna, "enumerate", 0, "Enumerate", "List objects under the mouse (object mode only)"); + RNA_def_property_flag(prop, PROP_SKIP_SAVE); + prop = RNA_def_boolean(ot->srna, "object", 0, "Object", "Use object selection (edit mode only)"); + RNA_def_property_flag(prop, PROP_SKIP_SAVE); + + /* Needed for select-through to usefully drag handles, see: T98254. + * NOTE: this option may be removed and become default behavior, see design task: T98552. */ + prop = RNA_def_boolean(ot->srna, + "vert_without_handles", + 0, + "Control Point Without Handles", + "Only select the curve control point, not it's handles"); + RNA_def_property_flag(prop, PROP_SKIP_SAVE); + + prop = RNA_def_int_vector(ot->srna, + "location", + 2, + nullptr, + INT_MIN, + INT_MAX, + "Location", + "Mouse location", + INT_MIN, + INT_MAX); + RNA_def_property_flag(prop, PROP_HIDDEN); +} + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Box Select + * \{ */ + +struct BoxSelectUserData { + ViewContext *vc; + const rcti *rect; + const rctf *rect_fl; + rctf _rect_fl; + eSelectOp sel_op; + eBezTriple_Flag select_flag; + + /* runtime */ + bool is_done; + bool is_changed; +}; + +static void view3d_userdata_boxselect_init(BoxSelectUserData *r_data, + ViewContext *vc, + const rcti *rect, + const eSelectOp sel_op) +{ + r_data->vc = vc; + + r_data->rect = rect; + r_data->rect_fl = &r_data->_rect_fl; + BLI_rctf_rcti_copy(&r_data->_rect_fl, rect); + + r_data->sel_op = sel_op; + /* SELECT by default, but can be changed if needed (only few cases use and respect this). */ + r_data->select_flag = (eBezTriple_Flag)SELECT; + + /* runtime */ + r_data->is_done = false; + r_data->is_changed = false; +} + +bool edge_inside_circle(const float cent[2], + float radius, + const float screen_co_a[2], + const float screen_co_b[2]) +{ + const float radius_squared = radius * radius; + return (dist_squared_to_line_segment_v2(cent, screen_co_a, screen_co_b) < radius_squared); +} + +static void do_paintvert_box_select__doSelectVert(void *userData, + MVert *mv, + const float screen_co[2], + int UNUSED(index)) +{ + BoxSelectUserData *data = static_cast(userData); + const bool is_select = mv->flag & SELECT; + const bool is_inside = BLI_rctf_isect_pt_v(data->rect_fl, screen_co); + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(mv->flag, sel_op_result, SELECT); + data->is_changed = true; + } +} +static bool do_paintvert_box_select(ViewContext *vc, + wmGenericUserData *wm_userdata, + const rcti *rect, + const eSelectOp sel_op) +{ + const bool use_zbuf = !XRAY_ENABLED(vc->v3d); + + Mesh *me = static_cast(vc->obact->data); + if ((me == nullptr) || (me->totvert == 0)) { + return false; + } + + bool changed = false; + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + changed |= paintvert_deselect_all_visible(vc->obact, SEL_DESELECT, false); + } + + if (BLI_rcti_is_empty(rect)) { + /* pass */ + } + else if (use_zbuf) { + EditSelectBuf_Cache *esel = static_cast(wm_userdata->data); + if (wm_userdata->data == nullptr) { + editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, SCE_SELECT_VERTEX); + esel = static_cast(wm_userdata->data); + esel->select_bitmap = DRW_select_buffer_bitmap_from_rect( + vc->depsgraph, vc->region, vc->v3d, rect, nullptr); + } + if (esel->select_bitmap != nullptr) { + changed |= edbm_backbuf_check_and_select_verts_obmode(me, esel, sel_op); + } + } + else { + BoxSelectUserData data; + + view3d_userdata_boxselect_init(&data, vc, rect, sel_op); + + ED_view3d_init_mats_rv3d(vc->obact, vc->rv3d); + + meshobject_foreachScreenVert( + vc, do_paintvert_box_select__doSelectVert, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + changed |= data.is_changed; + } + + if (changed) { + if (SEL_OP_CAN_DESELECT(sel_op)) { + BKE_mesh_mselect_validate(me); + } + paintvert_flush_flags(vc->obact); + paintvert_tag_select_update(vc->C, vc->obact); + } + return changed; +} + +static bool do_paintface_box_select(ViewContext *vc, + wmGenericUserData *wm_userdata, + const rcti *rect, + eSelectOp sel_op) +{ + Object *ob = vc->obact; + Mesh *me; + + me = BKE_mesh_from_object(ob); + if ((me == nullptr) || (me->totpoly == 0)) { + return false; + } + + bool changed = false; + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + changed |= paintface_deselect_all_visible(vc->C, vc->obact, SEL_DESELECT, false); + } + + if (BLI_rcti_is_empty(rect)) { + /* pass */ + } + else { + EditSelectBuf_Cache *esel = static_cast(wm_userdata->data); + if (wm_userdata->data == nullptr) { + editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, SCE_SELECT_FACE); + esel = static_cast(wm_userdata->data); + esel->select_bitmap = DRW_select_buffer_bitmap_from_rect( + vc->depsgraph, vc->region, vc->v3d, rect, nullptr); + } + if (esel->select_bitmap != nullptr) { + changed |= edbm_backbuf_check_and_select_faces_obmode(me, esel, sel_op); + } + } + + if (changed) { + paintface_flush_flags(vc->C, vc->obact, true, false); + } + return changed; +} + +static void do_nurbs_box_select__doSelect(void *userData, + Nurb *UNUSED(nu), + BPoint *bp, + BezTriple *bezt, + int beztindex, + bool handles_visible, + const float screen_co[2]) +{ + BoxSelectUserData *data = static_cast(userData); + + const bool is_inside = BLI_rctf_isect_pt_v(data->rect_fl, screen_co); + if (bp) { + const bool is_select = bp->f1 & SELECT; + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(bp->f1, sel_op_result, data->select_flag); + data->is_changed = true; + } + } + else { + if (!handles_visible) { + /* can only be (beztindex == 1) here since handles are hidden */ + const bool is_select = bezt->f2 & SELECT; + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(bezt->f2, sel_op_result, data->select_flag); + data->is_changed = true; + } + bezt->f1 = bezt->f3 = bezt->f2; + } + else { + uint8_t *flag_p = (&bezt->f1) + beztindex; + const bool is_select = *flag_p & SELECT; + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(*flag_p, sel_op_result, data->select_flag); + data->is_changed = true; + } + } + } +} +static bool do_nurbs_box_select(ViewContext *vc, rcti *rect, const eSelectOp sel_op) +{ + const bool deselect_all = (sel_op == SEL_OP_SET); + BoxSelectUserData data; + + view3d_userdata_boxselect_init(&data, vc, rect, sel_op); + + Curve *curve = (Curve *)vc->obedit->data; + ListBase *nurbs = BKE_curve_editNurbs_get(curve); + + /* For deselect all, items to be selected are tagged with temp flag. Clear that first. */ + if (deselect_all) { + BKE_nurbList_flag_set(nurbs, BEZT_FLAG_TEMP_TAG, false); + data.select_flag = BEZT_FLAG_TEMP_TAG; + } + + ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); /* for foreach's screen/vert projection */ + nurbs_foreachScreenVert(vc, do_nurbs_box_select__doSelect, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + + /* Deselect items that were not added to selection (indicated by temp flag). */ + if (deselect_all) { + data.is_changed |= BKE_nurbList_flag_set_from_flag(nurbs, BEZT_FLAG_TEMP_TAG, SELECT); + } + + BKE_curve_nurb_vert_active_validate(curve); + + return data.is_changed; +} + +static void do_lattice_box_select__doSelect(void *userData, BPoint *bp, const float screen_co[2]) +{ + BoxSelectUserData *data = static_cast(userData); + const bool is_select = bp->f1 & SELECT; + const bool is_inside = BLI_rctf_isect_pt_v(data->rect_fl, screen_co); + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(bp->f1, sel_op_result, SELECT); + data->is_changed = true; + } +} +static bool do_lattice_box_select(ViewContext *vc, rcti *rect, const eSelectOp sel_op) +{ + BoxSelectUserData data; + + view3d_userdata_boxselect_init(&data, vc, rect, sel_op); + + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + data.is_changed |= ED_lattice_flags_set(vc->obedit, 0); + } + + ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); /* for foreach's screen/vert projection */ + lattice_foreachScreenVert( + vc, do_lattice_box_select__doSelect, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + + return data.is_changed; +} + +static void do_mesh_box_select__doSelectVert(void *userData, + BMVert *eve, + const float screen_co[2], + int UNUSED(index)) +{ + BoxSelectUserData *data = static_cast(userData); + const bool is_select = BM_elem_flag_test(eve, BM_ELEM_SELECT); + const bool is_inside = BLI_rctf_isect_pt_v(data->rect_fl, screen_co); + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + BM_vert_select_set(data->vc->em->bm, eve, sel_op_result); + data->is_changed = true; + } +} +struct BoxSelectUserData_ForMeshEdge { + BoxSelectUserData *data; + EditSelectBuf_Cache *esel; + uint backbuf_offset; +}; +/** + * Pass 0 operates on edges when fully inside. + */ +static void do_mesh_box_select__doSelectEdge_pass0( + void *userData, BMEdge *eed, const float screen_co_a[2], const float screen_co_b[2], int index) +{ + BoxSelectUserData_ForMeshEdge *data_for_edge = static_cast( + userData); + BoxSelectUserData *data = data_for_edge->data; + bool is_visible = true; + if (data_for_edge->backbuf_offset) { + uint bitmap_inedx = data_for_edge->backbuf_offset + index - 1; + is_visible = BLI_BITMAP_TEST_BOOL(data_for_edge->esel->select_bitmap, bitmap_inedx); + } + + const bool is_select = BM_elem_flag_test(eed, BM_ELEM_SELECT); + const bool is_inside = (is_visible && + edge_fully_inside_rect(data->rect_fl, screen_co_a, screen_co_b)); + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + BM_edge_select_set(data->vc->em->bm, eed, sel_op_result); + data->is_done = true; + data->is_changed = true; + } +} +/** + * Pass 1 operates on edges when partially inside. + */ +static void do_mesh_box_select__doSelectEdge_pass1( + void *userData, BMEdge *eed, const float screen_co_a[2], const float screen_co_b[2], int index) +{ + BoxSelectUserData_ForMeshEdge *data_for_edge = static_cast( + userData); + BoxSelectUserData *data = data_for_edge->data; + bool is_visible = true; + if (data_for_edge->backbuf_offset) { + uint bitmap_inedx = data_for_edge->backbuf_offset + index - 1; + is_visible = BLI_BITMAP_TEST_BOOL(data_for_edge->esel->select_bitmap, bitmap_inedx); + } + + const bool is_select = BM_elem_flag_test(eed, BM_ELEM_SELECT); + const bool is_inside = (is_visible && edge_inside_rect(data->rect_fl, screen_co_a, screen_co_b)); + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + BM_edge_select_set(data->vc->em->bm, eed, sel_op_result); + data->is_changed = true; + } +} +static void do_mesh_box_select__doSelectFace(void *userData, + BMFace *efa, + const float screen_co[2], + int UNUSED(index)) +{ + BoxSelectUserData *data = static_cast(userData); + const bool is_select = BM_elem_flag_test(efa, BM_ELEM_SELECT); + const bool is_inside = BLI_rctf_isect_pt_v(data->rect_fl, screen_co); + const int sel_op_result = ED_select_op_action_deselected(data->sel_op, is_select, is_inside); + if (sel_op_result != -1) { + BM_face_select_set(data->vc->em->bm, efa, sel_op_result); + data->is_changed = true; + } +} +static bool do_mesh_box_select(ViewContext *vc, + wmGenericUserData *wm_userdata, + const rcti *rect, + const eSelectOp sel_op) +{ + BoxSelectUserData data; + ToolSettings *ts = vc->scene->toolsettings; + + view3d_userdata_boxselect_init(&data, vc, rect, sel_op); + + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + if (vc->em->bm->totvertsel) { + EDBM_flag_disable_all(vc->em, BM_ELEM_SELECT); + data.is_changed = true; + } + } + + /* for non zbuf projections, don't change the GL state */ + ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); + + GPU_matrix_set(vc->rv3d->viewmat); + + const bool use_zbuf = !XRAY_FLAG_ENABLED(vc->v3d); + + EditSelectBuf_Cache *esel = static_cast(wm_userdata->data); + if (use_zbuf) { + if (wm_userdata->data == nullptr) { + editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, ts->selectmode); + esel = static_cast(wm_userdata->data); + esel->select_bitmap = DRW_select_buffer_bitmap_from_rect( + vc->depsgraph, vc->region, vc->v3d, rect, nullptr); + } + } + + if (ts->selectmode & SCE_SELECT_VERTEX) { + if (use_zbuf) { + data.is_changed |= edbm_backbuf_check_and_select_verts( + esel, vc->depsgraph, vc->obedit, vc->em, sel_op); + } + else { + mesh_foreachScreenVert( + vc, do_mesh_box_select__doSelectVert, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + } + } + if (ts->selectmode & SCE_SELECT_EDGE) { + /* Does both use_zbuf and non-use_zbuf versions (need screen cos for both) */ + struct BoxSelectUserData_ForMeshEdge cb_data {}; + cb_data.data = &data; + cb_data.esel = use_zbuf ? esel : nullptr; + cb_data.backbuf_offset = use_zbuf ? DRW_select_buffer_context_offset_for_object_elem( + vc->depsgraph, vc->obedit, SCE_SELECT_EDGE) : + 0; + + const eV3DProjTest clip_flag = V3D_PROJ_TEST_CLIP_NEAR | + (use_zbuf ? (eV3DProjTest)0 : V3D_PROJ_TEST_CLIP_BB); + /* Fully inside. */ + mesh_foreachScreenEdge_clip_bb_segment( + vc, do_mesh_box_select__doSelectEdge_pass0, &cb_data, clip_flag); + if (data.is_done == false) { + /* Fall back to partially inside. + * Clip content to account for edges partially behind the view. */ + mesh_foreachScreenEdge_clip_bb_segment(vc, + do_mesh_box_select__doSelectEdge_pass1, + &cb_data, + clip_flag | V3D_PROJ_TEST_CLIP_CONTENT_DEFAULT); + } + } + + if (ts->selectmode & SCE_SELECT_FACE) { + if (use_zbuf) { + data.is_changed |= edbm_backbuf_check_and_select_faces( + esel, vc->depsgraph, vc->obedit, vc->em, sel_op); + } + else { + mesh_foreachScreenFace( + vc, do_mesh_box_select__doSelectFace, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + } + } + + if (data.is_changed) { + EDBM_selectmode_flush(vc->em); + } + return data.is_changed; +} + +static bool do_meta_box_select(ViewContext *vc, const rcti *rect, const eSelectOp sel_op) +{ + Object *ob = vc->obedit; + MetaBall *mb = (MetaBall *)ob->data; + MetaElem *ml; + int a; + bool changed = false; + + GPUSelectResult buffer[MAXPICKELEMS]; + int hits; + + hits = view3d_opengl_select( + vc, buffer, MAXPICKELEMS, rect, VIEW3D_SELECT_ALL, VIEW3D_SELECT_FILTER_NOP); + + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + changed |= BKE_mball_deselect_all(mb); + } + + int metaelem_id = 0; + for (ml = static_cast(mb->editelems->first); ml; + ml = ml->next, metaelem_id += 0x10000) { + bool is_inside_radius = false; + bool is_inside_stiff = false; + + for (a = 0; a < hits; a++) { + const int hitresult = buffer[a].id; + + if (hitresult == -1) { + continue; + } + + const uint hit_object = hitresult & 0xFFFF; + if (vc->obedit->runtime.select_id != hit_object) { + continue; + } + + if (metaelem_id != (hitresult & 0xFFFF0000 & ~MBALLSEL_ANY)) { + continue; + } + + if (hitresult & MBALLSEL_RADIUS) { + is_inside_radius = true; + break; + } + + if (hitresult & MBALLSEL_STIFF) { + is_inside_stiff = true; + break; + } + } + const int flag_prev = ml->flag; + if (is_inside_radius) { + ml->flag |= MB_SCALE_RAD; + } + if (is_inside_stiff) { + ml->flag &= ~MB_SCALE_RAD; + } + + const bool is_select = (ml->flag & SELECT); + const bool is_inside = is_inside_radius || is_inside_stiff; + + const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); + if (sel_op_result != -1) { + SET_FLAG_FROM_TEST(ml->flag, sel_op_result, SELECT); + } + changed |= (flag_prev != ml->flag); + } + + return changed; +} + +static bool do_armature_box_select(ViewContext *vc, const rcti *rect, const eSelectOp sel_op) +{ + bool changed = false; + int a; + + GPUSelectResult buffer[MAXPICKELEMS]; + int hits; + + hits = view3d_opengl_select( + vc, buffer, MAXPICKELEMS, rect, VIEW3D_SELECT_ALL, VIEW3D_SELECT_FILTER_NOP); + + uint bases_len = 0; + Base **bases = BKE_view_layer_array_from_bases_in_edit_mode_unique_data( + vc->view_layer, vc->v3d, &bases_len); + + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + changed |= ED_armature_edit_deselect_all_visible_multi_ex(bases, bases_len); + } + + for (uint base_index = 0; base_index < bases_len; base_index++) { + Object *obedit = bases[base_index]->object; + obedit->id.tag &= ~LIB_TAG_DOIT; + + bArmature *arm = static_cast(obedit->data); + ED_armature_ebone_listbase_temp_clear(arm->edbo); + } + + /* first we only check points inside the border */ + for (a = 0; a < hits; a++) { + const int select_id = buffer[a].id; + if (select_id != -1) { + if ((select_id & 0xFFFF0000) == 0) { + continue; + } + + EditBone *ebone; + Base *base_edit = ED_armature_base_and_ebone_from_select_buffer( + bases, bases_len, select_id, &ebone); + ebone->temp.i |= select_id & BONESEL_ANY; + base_edit->object->id.tag |= LIB_TAG_DOIT; + } + } + + for (uint base_index = 0; base_index < bases_len; base_index++) { + Object *obedit = bases[base_index]->object; + if (obedit->id.tag & LIB_TAG_DOIT) { + obedit->id.tag &= ~LIB_TAG_DOIT; + changed |= ED_armature_edit_select_op_from_tagged(static_cast(obedit->data), + sel_op); + } + } + + MEM_freeN(bases); + + return changed; +} + +/** + * Compare result of 'GPU_select': 'GPUSelectResult', + * needed for when we need to align with object draw-order. + */ +static int opengl_bone_select_buffer_cmp(const void *sel_a_p, const void *sel_b_p) +{ + uint sel_a = ((GPUSelectResult *)sel_a_p)->id; + uint sel_b = ((GPUSelectResult *)sel_b_p)->id; + +#ifdef __BIG_ENDIAN__ + BLI_endian_switch_uint32(&sel_a); + BLI_endian_switch_uint32(&sel_b); +#endif + + if (sel_a < sel_b) { + return -1; + } + if (sel_a > sel_b) { + return 1; + } + return 0; +} + +static bool do_object_box_select(bContext *C, ViewContext *vc, rcti *rect, const eSelectOp sel_op) +{ + View3D *v3d = vc->v3d; + int totobj = MAXPICKELEMS; /* XXX solve later */ + + /* Selection buffer has bones potentially too, so we add #MAXPICKELEMS. */ + GPUSelectResult *buffer = static_cast( + MEM_mallocN((totobj + MAXPICKELEMS) * sizeof(GPUSelectResult), __func__)); + const eV3DSelectObjectFilter select_filter = ED_view3d_select_filter_from_mode(vc->scene, + vc->obact); + const int hits = view3d_opengl_select( + vc, buffer, (totobj + MAXPICKELEMS), rect, VIEW3D_SELECT_ALL, select_filter); + + LISTBASE_FOREACH (Base *, base, &vc->view_layer->object_bases) { + base->object->id.tag &= ~LIB_TAG_DOIT; + } + + blender::Vector bases; + + bool changed = false; + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + changed |= object_deselect_all_visible(vc->view_layer, vc->v3d); + } + + if ((hits == -1) && !SEL_OP_USE_OUTSIDE(sel_op)) { + goto finally; + } + + LISTBASE_FOREACH (Base *, base, &vc->view_layer->object_bases) { + if (BASE_SELECTABLE(v3d, base)) { + if ((base->object->runtime.select_id & 0x0000FFFF) != 0) { + bases.append(base); + } + } + } + + /* The draw order doesn't always match the order we populate the engine, see: T51695. */ + qsort(buffer, hits, sizeof(GPUSelectResult), opengl_bone_select_buffer_cmp); + + for (const GPUSelectResult *buf_iter = buffer, *buf_end = buf_iter + hits; buf_iter < buf_end; + buf_iter++) { + bPoseChannel *pchan_dummy; + Base *base = ED_armature_base_and_pchan_from_select_buffer( + bases.data(), bases.size(), buf_iter->id, &pchan_dummy); + if (base != nullptr) { + base->object->id.tag |= LIB_TAG_DOIT; + } + } + + for (Base *base = static_cast(vc->view_layer->object_bases.first); base && hits; + base = base->next) { + if (BASE_SELECTABLE(v3d, base)) { + const bool is_select = base->flag & BASE_SELECTED; + const bool is_inside = base->object->id.tag & LIB_TAG_DOIT; + const int sel_op_result = ED_select_op_action_deselected(sel_op, is_select, is_inside); + if (sel_op_result != -1) { + ED_object_base_select(base, sel_op_result ? BA_SELECT : BA_DESELECT); + changed = true; + } + } + } + +finally: + + MEM_freeN(buffer); + + if (changed) { + DEG_id_tag_update(&vc->scene->id, ID_RECALC_SELECT); + WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, vc->scene); + } + return changed; +} + +static bool do_pose_box_select(bContext *C, ViewContext *vc, rcti *rect, const eSelectOp sel_op) +{ + blender::Vector bases = do_pose_tag_select_op_prepare(vc); + + int totobj = MAXPICKELEMS; /* XXX solve later */ + + /* Selection buffer has bones potentially too, so add #MAXPICKELEMS. */ + GPUSelectResult *buffer = static_cast( + MEM_mallocN((totobj + MAXPICKELEMS) * sizeof(GPUSelectResult), __func__)); + const eV3DSelectObjectFilter select_filter = ED_view3d_select_filter_from_mode(vc->scene, + vc->obact); + const int hits = view3d_opengl_select( + vc, buffer, (totobj + MAXPICKELEMS), rect, VIEW3D_SELECT_ALL, select_filter); + /* + * LOGIC NOTES (theeth): + * The buffer and ListBase have the same relative order, which makes the selection + * very simple. Loop through both data sets at the same time, if the color + * is the same as the object, we have a hit and can move to the next color + * and object pair, if not, just move to the next object, + * keeping the same color until we have a hit. + */ + + if (hits > 0) { + /* no need to loop if there's no hit */ + + /* The draw order doesn't always match the order we populate the engine, see: T51695. */ + qsort(buffer, hits, sizeof(GPUSelectResult), opengl_bone_select_buffer_cmp); + + for (const GPUSelectResult *buf_iter = buffer, *buf_end = buf_iter + hits; buf_iter < buf_end; + buf_iter++) { + Bone *bone; + Base *base = ED_armature_base_and_bone_from_select_buffer( + bases.data(), bases.size(), buf_iter->id, &bone); + + if (base == nullptr) { + continue; + } + + /* Loop over contiguous bone hits for 'base'. */ + for (; buf_iter != buf_end; buf_iter++) { + /* should never fail */ + if (bone != nullptr) { + base->object->id.tag |= LIB_TAG_DOIT; + bone->flag |= BONE_DONE; + } + + /* Select the next bone if we're not switching bases. */ + if (buf_iter + 1 != buf_end) { + const GPUSelectResult *col_next = buf_iter + 1; + if ((base->object->runtime.select_id & 0x0000FFFF) != (col_next->id & 0x0000FFFF)) { + break; + } + if (base->object->pose != nullptr) { + const uint hit_bone = (col_next->id & ~BONESEL_ANY) >> 16; + bPoseChannel *pchan = static_cast( + BLI_findlink(&base->object->pose->chanbase, hit_bone)); + bone = pchan ? pchan->bone : nullptr; + } + else { + bone = nullptr; + } + } + } + } + } + + const bool changed_multi = do_pose_tag_select_op_exec(bases, sel_op); + if (changed_multi) { + DEG_id_tag_update(&vc->scene->id, ID_RECALC_SELECT); + WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, vc->scene); + } + + MEM_freeN(buffer); + + return changed_multi; +} + +static int view3d_box_select_exec(bContext *C, wmOperator *op) +{ + Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C); + ViewContext vc; + rcti rect; + bool changed_multi = false; + + wmGenericUserData wm_userdata_buf = {0}; + wmGenericUserData *wm_userdata = &wm_userdata_buf; + + view3d_operator_needs_opengl(C); + BKE_object_update_select_id(CTX_data_main(C)); + + /* setup view context for argument to callbacks */ + ED_view3d_viewcontext_init(C, &vc, depsgraph); + + eSelectOp sel_op = static_cast(RNA_enum_get(op->ptr, "mode")); + WM_operator_properties_border_to_rcti(op, &rect); + + if (vc.obedit) { + FOREACH_OBJECT_IN_MODE_BEGIN ( + vc.view_layer, vc.v3d, vc.obedit->type, vc.obedit->mode, ob_iter) { + ED_view3d_viewcontext_init_object(&vc, ob_iter); + bool changed = false; + + switch (vc.obedit->type) { + case OB_MESH: + vc.em = BKE_editmesh_from_object(vc.obedit); + changed = do_mesh_box_select(&vc, wm_userdata, &rect, sel_op); + if (changed) { + DEG_id_tag_update(static_cast(vc.obedit->data), ID_RECALC_SELECT); + WM_event_add_notifier(C, NC_GEOM | ND_SELECT, vc.obedit->data); + } + break; + case OB_CURVES_LEGACY: + case OB_SURF: + changed = do_nurbs_box_select(&vc, &rect, sel_op); + if (changed) { + DEG_id_tag_update(static_cast(vc.obedit->data), ID_RECALC_SELECT); + WM_event_add_notifier(C, NC_GEOM | ND_SELECT, vc.obedit->data); + } + break; + case OB_MBALL: + changed = do_meta_box_select(&vc, &rect, sel_op); + if (changed) { + DEG_id_tag_update(static_cast(vc.obedit->data), ID_RECALC_SELECT); + WM_event_add_notifier(C, NC_GEOM | ND_SELECT, vc.obedit->data); + } + break; + case OB_ARMATURE: + changed = do_armature_box_select(&vc, &rect, sel_op); + if (changed) { + DEG_id_tag_update(&vc.obedit->id, ID_RECALC_SELECT); + WM_event_add_notifier(C, NC_OBJECT | ND_BONE_SELECT, vc.obedit); + ED_outliner_select_sync_from_edit_bone_tag(C); + } + break; + case OB_LATTICE: + changed = do_lattice_box_select(&vc, &rect, sel_op); + if (changed) { + DEG_id_tag_update(static_cast(vc.obedit->data), ID_RECALC_SELECT); + WM_event_add_notifier(C, NC_GEOM | ND_SELECT, vc.obedit->data); + } + break; + default: + BLI_assert_msg(0, "box select on incorrect object type"); + break; + } + changed_multi |= changed; + } + FOREACH_OBJECT_IN_MODE_END; + } + else { /* No edit-mode, unified for bones and objects. */ + if (vc.obact && BKE_paint_select_face_test(vc.obact)) { + changed_multi = do_paintface_box_select(&vc, wm_userdata, &rect, sel_op); + } + else if (vc.obact && BKE_paint_select_vert_test(vc.obact)) { + changed_multi = do_paintvert_box_select(&vc, wm_userdata, &rect, sel_op); + } + else if (vc.obact && vc.obact->mode & OB_MODE_PARTICLE_EDIT) { + changed_multi = PE_box_select(C, &rect, sel_op); + } + else if (vc.obact && vc.obact->mode & OB_MODE_POSE) { + changed_multi = do_pose_box_select(C, &vc, &rect, sel_op); + if (changed_multi) { + ED_outliner_select_sync_from_pose_bone_tag(C); + } + } + else { /* object mode with none active */ + changed_multi = do_object_box_select(C, &vc, &rect, sel_op); + if (changed_multi) { + ED_outliner_select_sync_from_object_tag(C); + } + } + } + + WM_generic_user_data_free(wm_userdata); + + if (changed_multi) { + return OPERATOR_FINISHED; + } + return OPERATOR_CANCELLED; +} + +void VIEW3D_OT_select_box(wmOperatorType *ot) +{ + /* identifiers */ + ot->name = "Box Select"; + ot->description = "Select items using box selection"; + ot->idname = "VIEW3D_OT_select_box"; + + /* api callbacks */ + ot->invoke = WM_gesture_box_invoke; + ot->exec = view3d_box_select_exec; + ot->modal = WM_gesture_box_modal; + ot->poll = view3d_selectable_data; + ot->cancel = WM_gesture_box_cancel; + + /* flags */ + ot->flag = OPTYPE_UNDO; + + /* rna */ + WM_operator_properties_gesture_box(ot); + WM_operator_properties_select_operation(ot); +} + +/** \} */ + +/* -------------------------------------------------------------------- */ +/** \name Circle Select + * \{ */ + +struct CircleSelectUserData { + ViewContext *vc; + bool select; + int mval[2]; + float mval_fl[2]; + float radius; + float radius_squared; + eBezTriple_Flag select_flag; + + /* runtime */ + bool is_changed; +}; + +static void view3d_userdata_circleselect_init(CircleSelectUserData *r_data, + ViewContext *vc, + const bool select, + const int mval[2], + const float rad) +{ + r_data->vc = vc; + r_data->select = select; + copy_v2_v2_int(r_data->mval, mval); + r_data->mval_fl[0] = mval[0]; + r_data->mval_fl[1] = mval[1]; + + r_data->radius = rad; + r_data->radius_squared = rad * rad; + + /* SELECT by default, but can be changed if needed (only few cases use and respect this). */ + r_data->select_flag = (eBezTriple_Flag)SELECT; + + /* runtime */ + r_data->is_changed = false; +} + +static void mesh_circle_doSelectVert(void *userData, + BMVert *eve, + const float screen_co[2], + int UNUSED(index)) +{ + CircleSelectUserData *data = static_cast(userData); + + if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { + BM_vert_select_set(data->vc->em->bm, eve, data->select); + data->is_changed = true; + } +} +static void mesh_circle_doSelectEdge(void *userData, + BMEdge *eed, + const float screen_co_a[2], + const float screen_co_b[2], + int UNUSED(index)) +{ + CircleSelectUserData *data = static_cast(userData); + + if (edge_inside_circle(data->mval_fl, data->radius, screen_co_a, screen_co_b)) { + BM_edge_select_set(data->vc->em->bm, eed, data->select); + data->is_changed = true; + } +} +static void mesh_circle_doSelectFace(void *userData, + BMFace *efa, + const float screen_co[2], + int UNUSED(index)) +{ + CircleSelectUserData *data = static_cast(userData); + + if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { + BM_face_select_set(data->vc->em->bm, efa, data->select); + data->is_changed = true; + } +} + +static bool mesh_circle_select(ViewContext *vc, + wmGenericUserData *wm_userdata, + eSelectOp sel_op, + const int mval[2], + float rad) +{ + ToolSettings *ts = vc->scene->toolsettings; + CircleSelectUserData data; + vc->em = BKE_editmesh_from_object(vc->obedit); + + bool changed = false; + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + if (vc->em->bm->totvertsel) { + EDBM_flag_disable_all(vc->em, BM_ELEM_SELECT); + vc->em->bm->totvertsel = 0; + vc->em->bm->totedgesel = 0; + vc->em->bm->totfacesel = 0; + changed = true; + } + } + const bool select = (sel_op != SEL_OP_SUB); + + ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); /* for foreach's screen/vert projection */ + + view3d_userdata_circleselect_init(&data, vc, select, mval, rad); + + const bool use_zbuf = !XRAY_FLAG_ENABLED(vc->v3d); + + if (use_zbuf) { + if (wm_userdata->data == nullptr) { + editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, ts->selectmode); + } + } + EditSelectBuf_Cache *esel = static_cast(wm_userdata->data); + + if (use_zbuf) { + if (esel->select_bitmap == nullptr) { + esel->select_bitmap = DRW_select_buffer_bitmap_from_circle( + vc->depsgraph, vc->region, vc->v3d, mval, (int)(rad + 1.0f), nullptr); + } + } + + if (ts->selectmode & SCE_SELECT_VERTEX) { + if (use_zbuf) { + if (esel->select_bitmap != nullptr) { + changed |= edbm_backbuf_check_and_select_verts( + esel, vc->depsgraph, vc->obedit, vc->em, select ? SEL_OP_ADD : SEL_OP_SUB); + } + } + else { + mesh_foreachScreenVert(vc, mesh_circle_doSelectVert, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + } + } + + if (ts->selectmode & SCE_SELECT_EDGE) { + if (use_zbuf) { + if (esel->select_bitmap != nullptr) { + changed |= edbm_backbuf_check_and_select_edges( + esel, vc->depsgraph, vc->obedit, vc->em, select ? SEL_OP_ADD : SEL_OP_SUB); + } + } + else { + mesh_foreachScreenEdge_clip_bb_segment( + vc, + mesh_circle_doSelectEdge, + &data, + (V3D_PROJ_TEST_CLIP_NEAR | V3D_PROJ_TEST_CLIP_BB | V3D_PROJ_TEST_CLIP_CONTENT_DEFAULT)); + } + } + + if (ts->selectmode & SCE_SELECT_FACE) { + if (use_zbuf) { + if (esel->select_bitmap != nullptr) { + changed |= edbm_backbuf_check_and_select_faces( + esel, vc->depsgraph, vc->obedit, vc->em, select ? SEL_OP_ADD : SEL_OP_SUB); + } + } + else { + mesh_foreachScreenFace(vc, mesh_circle_doSelectFace, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + } + } + + changed |= data.is_changed; + + if (changed) { + BM_mesh_select_mode_flush_ex( + vc->em->bm, vc->em->selectmode, BM_SELECT_LEN_FLUSH_RECALC_NOTHING); + } + return changed; +} + +static bool paint_facesel_circle_select(ViewContext *vc, + wmGenericUserData *wm_userdata, + const eSelectOp sel_op, + const int mval[2], + float rad) +{ + BLI_assert(ELEM(sel_op, SEL_OP_SET, SEL_OP_ADD, SEL_OP_SUB)); + Object *ob = vc->obact; + Mesh *me = static_cast(ob->data); + + bool changed = false; + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + /* flush selection at the end */ + changed |= paintface_deselect_all_visible(vc->C, ob, SEL_DESELECT, false); + } + + if (wm_userdata->data == nullptr) { + editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, SCE_SELECT_FACE); + } + + { + EditSelectBuf_Cache *esel = static_cast(wm_userdata->data); + esel->select_bitmap = DRW_select_buffer_bitmap_from_circle( + vc->depsgraph, vc->region, vc->v3d, mval, (int)(rad + 1.0f), nullptr); + if (esel->select_bitmap != nullptr) { + changed |= edbm_backbuf_check_and_select_faces_obmode(me, esel, sel_op); + MEM_freeN(esel->select_bitmap); + esel->select_bitmap = nullptr; + } + } + + if (changed) { + paintface_flush_flags(vc->C, ob, true, false); + } + return changed; +} + +static void paint_vertsel_circle_select_doSelectVert(void *userData, + MVert *mv, + const float screen_co[2], + int UNUSED(index)) +{ + CircleSelectUserData *data = static_cast(userData); + + if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { + SET_FLAG_FROM_TEST(mv->flag, data->select, SELECT); + data->is_changed = true; + } +} +static bool paint_vertsel_circle_select(ViewContext *vc, + wmGenericUserData *wm_userdata, + const eSelectOp sel_op, + const int mval[2], + float rad) +{ + BLI_assert(ELEM(sel_op, SEL_OP_SET, SEL_OP_ADD, SEL_OP_SUB)); + const bool use_zbuf = !XRAY_ENABLED(vc->v3d); + Object *ob = vc->obact; + Mesh *me = static_cast(ob->data); + /* CircleSelectUserData data = {nullptr}; */ /* UNUSED */ + + bool changed = false; + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + /* Flush selection at the end. */ + changed |= paintvert_deselect_all_visible(ob, SEL_DESELECT, false); + } + + const bool select = (sel_op != SEL_OP_SUB); + + if (use_zbuf) { + if (wm_userdata->data == nullptr) { + editselect_buf_cache_init_with_generic_userdata(wm_userdata, vc, SCE_SELECT_VERTEX); + } + } + + if (use_zbuf) { + EditSelectBuf_Cache *esel = static_cast(wm_userdata->data); + esel->select_bitmap = DRW_select_buffer_bitmap_from_circle( + vc->depsgraph, vc->region, vc->v3d, mval, (int)(rad + 1.0f), nullptr); + if (esel->select_bitmap != nullptr) { + changed |= edbm_backbuf_check_and_select_verts_obmode(me, esel, sel_op); + MEM_freeN(esel->select_bitmap); + esel->select_bitmap = nullptr; + } + } + else { + CircleSelectUserData data; + + ED_view3d_init_mats_rv3d(vc->obact, vc->rv3d); /* for foreach's screen/vert projection */ + + view3d_userdata_circleselect_init(&data, vc, select, mval, rad); + meshobject_foreachScreenVert( + vc, paint_vertsel_circle_select_doSelectVert, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + changed |= data.is_changed; + } + + if (changed) { + if (sel_op == SEL_OP_SUB) { + BKE_mesh_mselect_validate(me); + } + paintvert_flush_flags(ob); + paintvert_tag_select_update(vc->C, ob); + } + return changed; +} + +static void nurbscurve_circle_doSelect(void *userData, + Nurb *UNUSED(nu), + BPoint *bp, + BezTriple *bezt, + int beztindex, + bool UNUSED(handles_visible), + const float screen_co[2]) +{ + CircleSelectUserData *data = static_cast(userData); + + if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { + if (bp) { + SET_FLAG_FROM_TEST(bp->f1, data->select, data->select_flag); + } + else { + if (beztindex == 0) { + SET_FLAG_FROM_TEST(bezt->f1, data->select, data->select_flag); + } + else if (beztindex == 1) { + SET_FLAG_FROM_TEST(bezt->f2, data->select, data->select_flag); + } + else { + SET_FLAG_FROM_TEST(bezt->f3, data->select, data->select_flag); + } + } + data->is_changed = true; + } +} +static bool nurbscurve_circle_select(ViewContext *vc, + const eSelectOp sel_op, + const int mval[2], + float rad) +{ + const bool select = (sel_op != SEL_OP_SUB); + const bool deselect_all = (sel_op == SEL_OP_SET); + CircleSelectUserData data; + + view3d_userdata_circleselect_init(&data, vc, select, mval, rad); + + Curve *curve = (Curve *)vc->obedit->data; + ListBase *nurbs = BKE_curve_editNurbs_get(curve); + + /* For deselect all, items to be selected are tagged with temp flag. Clear that first. */ + if (deselect_all) { + BKE_nurbList_flag_set(nurbs, BEZT_FLAG_TEMP_TAG, false); + data.select_flag = BEZT_FLAG_TEMP_TAG; + } + + ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); /* for foreach's screen/vert projection */ + nurbs_foreachScreenVert(vc, nurbscurve_circle_doSelect, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + + /* Deselect items that were not added to selection (indicated by temp flag). */ + if (deselect_all) { + data.is_changed |= BKE_nurbList_flag_set_from_flag(nurbs, BEZT_FLAG_TEMP_TAG, SELECT); + } + + BKE_curve_nurb_vert_active_validate(static_cast(vc->obedit->data)); + + return data.is_changed; +} + +static void latticecurve_circle_doSelect(void *userData, BPoint *bp, const float screen_co[2]) +{ + CircleSelectUserData *data = static_cast(userData); + + if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { + bp->f1 = data->select ? (bp->f1 | SELECT) : (bp->f1 & ~SELECT); + data->is_changed = true; + } +} +static bool lattice_circle_select(ViewContext *vc, + const eSelectOp sel_op, + const int mval[2], + float rad) +{ + CircleSelectUserData data; + const bool select = (sel_op != SEL_OP_SUB); + + view3d_userdata_circleselect_init(&data, vc, select, mval, rad); + + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + data.is_changed |= ED_lattice_flags_set(vc->obedit, 0); + } + ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); /* for foreach's screen/vert projection */ + + lattice_foreachScreenVert(vc, latticecurve_circle_doSelect, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + + return data.is_changed; +} + +/** + * \note logic is shared with the edit-bone case, see #armature_circle_doSelectJoint. + */ +static bool pchan_circle_doSelectJoint(void *userData, + bPoseChannel *pchan, + const float screen_co[2]) +{ + CircleSelectUserData *data = static_cast(userData); + + if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { + if (data->select) { + pchan->bone->flag |= BONE_SELECTED; + } + else { + pchan->bone->flag &= ~BONE_SELECTED; + } + return 1; + } + return 0; +} +static void do_circle_select_pose__doSelectBone(void *userData, + bPoseChannel *pchan, + const float screen_co_a[2], + const float screen_co_b[2]) +{ + CircleSelectUserData *data = static_cast(userData); + bArmature *arm = static_cast(data->vc->obact->data); + if (!PBONE_SELECTABLE(arm, pchan->bone)) { + return; + } + + bool is_point_done = false; + int points_proj_tot = 0; + + /* project head location to screenspace */ + if (screen_co_a[0] != IS_CLIPPED) { + points_proj_tot++; + if (pchan_circle_doSelectJoint(data, pchan, screen_co_a)) { + is_point_done = true; + } + } + + /* project tail location to screenspace */ + if (screen_co_b[0] != IS_CLIPPED) { + points_proj_tot++; + if (pchan_circle_doSelectJoint(data, pchan, screen_co_b)) { + is_point_done = true; + } + } + + /* check if the head and/or tail is in the circle + * - the call to check also does the selection already + */ + + /* only if the endpoints didn't get selected, deal with the middle of the bone too + * It works nicer to only do this if the head or tail are not in the circle, + * otherwise there is no way to circle select joints alone */ + if ((is_point_done == false) && (points_proj_tot == 2) && + edge_inside_circle(data->mval_fl, data->radius, screen_co_a, screen_co_b)) { + if (data->select) { + pchan->bone->flag |= BONE_SELECTED; + } + else { + pchan->bone->flag &= ~BONE_SELECTED; + } + data->is_changed = true; + } + + data->is_changed |= is_point_done; +} +static bool pose_circle_select(ViewContext *vc, + const eSelectOp sel_op, + const int mval[2], + float rad) +{ + BLI_assert(ELEM(sel_op, SEL_OP_SET, SEL_OP_ADD, SEL_OP_SUB)); + CircleSelectUserData data; + const bool select = (sel_op != SEL_OP_SUB); + + view3d_userdata_circleselect_init(&data, vc, select, mval, rad); + + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + data.is_changed |= ED_pose_deselect_all(vc->obact, SEL_DESELECT, false); + } + + ED_view3d_init_mats_rv3d(vc->obact, vc->rv3d); /* for foreach's screen/vert projection */ + + /* Treat bones as clipped segments (no joints). */ + pose_foreachScreenBone(vc, + do_circle_select_pose__doSelectBone, + &data, + V3D_PROJ_TEST_CLIP_DEFAULT | V3D_PROJ_TEST_CLIP_CONTENT_DEFAULT); + + if (data.is_changed) { + ED_pose_bone_select_tag_update(vc->obact); + } + return data.is_changed; +} + +/** + * \note logic is shared with the pose-bone case, see #pchan_circle_doSelectJoint. + */ +static bool armature_circle_doSelectJoint(void *userData, + EditBone *ebone, + const float screen_co[2], + bool head) +{ + CircleSelectUserData *data = static_cast(userData); + + if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { + if (head) { + if (data->select) { + ebone->flag |= BONE_ROOTSEL; + } + else { + ebone->flag &= ~BONE_ROOTSEL; + } + } + else { + if (data->select) { + ebone->flag |= BONE_TIPSEL; + } + else { + ebone->flag &= ~BONE_TIPSEL; + } + } + return 1; + } + return 0; +} +static void do_circle_select_armature__doSelectBone(void *userData, + EditBone *ebone, + const float screen_co_a[2], + const float screen_co_b[2]) +{ + CircleSelectUserData *data = static_cast(userData); + const bArmature *arm = static_cast(data->vc->obedit->data); + if (!(data->select ? EBONE_SELECTABLE(arm, ebone) : EBONE_VISIBLE(arm, ebone))) { + return; + } + + /* When true, ignore in the next pass. */ + ebone->temp.i = false; + + bool is_point_done = false; + bool is_edge_done = false; + int points_proj_tot = 0; + + /* project head location to screenspace */ + if (screen_co_a[0] != IS_CLIPPED) { + points_proj_tot++; + if (armature_circle_doSelectJoint(data, ebone, screen_co_a, true)) { + is_point_done = true; + } + } + + /* project tail location to screenspace */ + if (screen_co_b[0] != IS_CLIPPED) { + points_proj_tot++; + if (armature_circle_doSelectJoint(data, ebone, screen_co_b, false)) { + is_point_done = true; + } + } + + /* check if the head and/or tail is in the circle + * - the call to check also does the selection already + */ + + /* only if the endpoints didn't get selected, deal with the middle of the bone too + * It works nicer to only do this if the head or tail are not in the circle, + * otherwise there is no way to circle select joints alone */ + if ((is_point_done == false) && (points_proj_tot == 2) && + edge_inside_circle(data->mval_fl, data->radius, screen_co_a, screen_co_b)) { + SET_FLAG_FROM_TEST(ebone->flag, data->select, BONE_SELECTED | BONE_TIPSEL | BONE_ROOTSEL); + is_edge_done = true; + data->is_changed = true; + } + + if (is_point_done || is_edge_done) { + ebone->temp.i = true; + } + + data->is_changed |= is_point_done; +} +static void do_circle_select_armature__doSelectBone_clip_content(void *userData, + EditBone *ebone, + const float screen_co_a[2], + const float screen_co_b[2]) +{ + CircleSelectUserData *data = static_cast(userData); + bArmature *arm = static_cast(data->vc->obedit->data); + + if (!(data->select ? EBONE_SELECTABLE(arm, ebone) : EBONE_VISIBLE(arm, ebone))) { + return; + } + + /* Set in the first pass, needed so circle select prioritizes joints. */ + if (ebone->temp.i != 0) { + return; + } + + if (edge_inside_circle(data->mval_fl, data->radius, screen_co_a, screen_co_b)) { + SET_FLAG_FROM_TEST(ebone->flag, data->select, BONE_SELECTED | BONE_TIPSEL | BONE_ROOTSEL); + data->is_changed = true; + } +} +static bool armature_circle_select(ViewContext *vc, + const eSelectOp sel_op, + const int mval[2], + float rad) +{ + CircleSelectUserData data; + bArmature *arm = static_cast(vc->obedit->data); + + const bool select = (sel_op != SEL_OP_SUB); + + view3d_userdata_circleselect_init(&data, vc, select, mval, rad); + + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + data.is_changed |= ED_armature_edit_deselect_all_visible(vc->obedit); + } + + ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); + + /* Operate on fully visible (non-clipped) points. */ + armature_foreachScreenBone( + vc, do_circle_select_armature__doSelectBone, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + + /* Operate on bones as segments clipped to the viewport bounds + * (needed to handle bones with both points outside the view). + * A separate pass is needed since clipped coordinates can't be used for selecting joints. */ + armature_foreachScreenBone(vc, + do_circle_select_armature__doSelectBone_clip_content, + &data, + V3D_PROJ_TEST_CLIP_DEFAULT | V3D_PROJ_TEST_CLIP_CONTENT_DEFAULT); + + if (data.is_changed) { + ED_armature_edit_sync_selection(arm->edbo); + ED_armature_edit_validate_active(arm); + WM_main_add_notifier(NC_OBJECT | ND_BONE_SELECT, vc->obedit); + } + return data.is_changed; +} + +static void do_circle_select_mball__doSelectElem(void *userData, + MetaElem *ml, + const float screen_co[2]) +{ + CircleSelectUserData *data = static_cast(userData); + + if (len_squared_v2v2(data->mval_fl, screen_co) <= data->radius_squared) { + if (data->select) { + ml->flag |= SELECT; + } + else { + ml->flag &= ~SELECT; + } + data->is_changed = true; + } +} +static bool mball_circle_select(ViewContext *vc, + const eSelectOp sel_op, + const int mval[2], + float rad) +{ + CircleSelectUserData data; + + const bool select = (sel_op != SEL_OP_SUB); + + view3d_userdata_circleselect_init(&data, vc, select, mval, rad); + + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + data.is_changed |= BKE_mball_deselect_all(static_cast(vc->obedit->data)); + } + + ED_view3d_init_mats_rv3d(vc->obedit, vc->rv3d); + + mball_foreachScreenElem( + vc, do_circle_select_mball__doSelectElem, &data, V3D_PROJ_TEST_CLIP_DEFAULT); + return data.is_changed; +} + +/** + * Callbacks for circle selection in Editmode + */ +static bool obedit_circle_select(bContext *C, + ViewContext *vc, + wmGenericUserData *wm_userdata, + const eSelectOp sel_op, + const int mval[2], + float rad) +{ + bool changed = false; + BLI_assert(ELEM(sel_op, SEL_OP_SET, SEL_OP_ADD, SEL_OP_SUB)); + switch (vc->obedit->type) { + case OB_MESH: + changed = mesh_circle_select(vc, wm_userdata, sel_op, mval, rad); + break; + case OB_CURVES_LEGACY: + case OB_SURF: + changed = nurbscurve_circle_select(vc, sel_op, mval, rad); + break; + case OB_LATTICE: + changed = lattice_circle_select(vc, sel_op, mval, rad); + break; + case OB_ARMATURE: + changed = armature_circle_select(vc, sel_op, mval, rad); + if (changed) { + ED_outliner_select_sync_from_edit_bone_tag(C); + } + break; + case OB_MBALL: + changed = mball_circle_select(vc, sel_op, mval, rad); + break; + default: + BLI_assert(0); + break; + } + + if (changed) { + DEG_id_tag_update(static_cast(vc->obact->data), ID_RECALC_SELECT); + WM_main_add_notifier(NC_GEOM | ND_SELECT, vc->obact->data); + } + return changed; +} + +static bool object_circle_select(ViewContext *vc, + const eSelectOp sel_op, + const int mval[2], + float rad) +{ + BLI_assert(ELEM(sel_op, SEL_OP_SET, SEL_OP_ADD, SEL_OP_SUB)); + ViewLayer *view_layer = vc->view_layer; + View3D *v3d = vc->v3d; + + const float radius_squared = rad * rad; + const float mval_fl[2] = {static_cast(mval[0]), static_cast(mval[1])}; + + bool changed = false; + if (SEL_OP_USE_PRE_DESELECT(sel_op)) { + changed |= object_deselect_all_visible(vc->view_layer, vc->v3d); + } + const bool select = (sel_op != SEL_OP_SUB); + const int select_flag = select ? BASE_SELECTED : 0; + + Base *base; + for (base = FIRSTBASE(view_layer); base; base = base->next) { + if (BASE_SELECTABLE(v3d, base) && ((base->flag & BASE_SELECTED) != select_flag)) { + float screen_co[2]; + if (ED_view3d_project_float_global( + vc->region, base->object->obmat[3], screen_co, V3D_PROJ_TEST_CLIP_DEFAULT) == + V3D_PROJ_RET_OK) { + if (len_squared_v2v2(mval_fl, screen_co) <= radius_squared) { + ED_object_base_select(base, select ? BA_SELECT : BA_DESELECT); + changed = true; + } + } + } + } + + return changed; +} + +/* not a real operator, only for circle test */ +static void view3d_circle_select_recalc(void *user_data) +{ + bContext *C = static_cast(user_data); + Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C); + ViewContext vc; + ED_view3d_viewcontext_init(C, &vc, depsgraph); + em_setup_viewcontext(C, &vc); + + if (vc.obedit) { + switch (vc.obedit->type) { + case OB_MESH: { + FOREACH_OBJECT_IN_MODE_BEGIN ( + vc.view_layer, vc.v3d, vc.obact->type, vc.obact->mode, ob_iter) { + ED_view3d_viewcontext_init_object(&vc, ob_iter); + BM_mesh_select_mode_flush_ex( + vc.em->bm, vc.em->selectmode, BM_SELECT_LEN_FLUSH_RECALC_ALL); + } + FOREACH_OBJECT_IN_MODE_END; + break; + } + + default: + break; + } + } +} + +static int view3d_circle_select_modal(bContext *C, wmOperator *op, const wmEvent *event) +{ + int result = WM_gesture_circle_modal(C, op, event); + if (result & OPERATOR_FINISHED) { + view3d_circle_select_recalc(C); + } + return result; +} + +static void view3d_circle_select_cancel(bContext *C, wmOperator *op) +{ + WM_gesture_circle_cancel(C, op); + view3d_circle_select_recalc(C); +} + +static int view3d_circle_select_exec(bContext *C, wmOperator *op) +{ + Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C); + ViewContext vc; + const int radius = RNA_int_get(op->ptr, "radius"); + const int mval[2] = {RNA_int_get(op->ptr, "x"), RNA_int_get(op->ptr, "y")}; + + /* Allow each selection type to allocate their own data that's used between executions. */ + wmGesture *gesture = static_cast(op->customdata); /* nullptr when non-modal. */ + wmGenericUserData wm_userdata_buf = {0}; + wmGenericUserData *wm_userdata = gesture ? &gesture->user_data : &wm_userdata_buf; + + const eSelectOp sel_op = ED_select_op_modal( + static_cast(RNA_enum_get(op->ptr, "mode")), WM_gesture_is_modal_first(gesture)); + + ED_view3d_viewcontext_init(C, &vc, depsgraph); + + Object *obact = vc.obact; + Object *obedit = vc.obedit; + + if (obedit || BKE_paint_select_elem_test(obact) || (obact && (obact->mode & OB_MODE_POSE))) { + view3d_operator_needs_opengl(C); + if (obedit == nullptr) { + BKE_object_update_select_id(CTX_data_main(C)); + } + + FOREACH_OBJECT_IN_MODE_BEGIN (vc.view_layer, vc.v3d, obact->type, obact->mode, ob_iter) { + ED_view3d_viewcontext_init_object(&vc, ob_iter); + + obact = vc.obact; + obedit = vc.obedit; + + if (obedit) { + obedit_circle_select(C, &vc, wm_userdata, sel_op, mval, (float)radius); + } + else if (BKE_paint_select_face_test(obact)) { + paint_facesel_circle_select(&vc, wm_userdata, sel_op, mval, (float)radius); + } + else if (BKE_paint_select_vert_test(obact)) { + paint_vertsel_circle_select(&vc, wm_userdata, sel_op, mval, (float)radius); + } + else if (obact->mode & OB_MODE_POSE) { + pose_circle_select(&vc, sel_op, mval, (float)radius); + ED_outliner_select_sync_from_pose_bone_tag(C); + } + else { + BLI_assert(0); + } + } + FOREACH_OBJECT_IN_MODE_END; + } + else if (obact && (obact->mode & OB_MODE_PARTICLE_EDIT)) { + if (PE_circle_select(C, wm_userdata, sel_op, mval, (float)radius)) { + return OPERATOR_FINISHED; + } + return OPERATOR_CANCELLED; + } + else if (obact && obact->mode & OB_MODE_SCULPT) { + return OPERATOR_CANCELLED; + } + else { + if (object_circle_select(&vc, sel_op, mval, (float)radius)) { + DEG_id_tag_update(&vc.scene->id, ID_RECALC_SELECT); + WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, vc.scene); + + ED_outliner_select_sync_from_object_tag(C); + } + } + + /* Otherwise this is freed by the gesture. */ + if (wm_userdata == &wm_userdata_buf) { + WM_generic_user_data_free(wm_userdata); + } + else { + EditSelectBuf_Cache *esel = static_cast(wm_userdata->data); + if (esel && esel->select_bitmap) { + MEM_freeN(esel->select_bitmap); + esel->select_bitmap = nullptr; + } + } + + return OPERATOR_FINISHED; +} + +void VIEW3D_OT_select_circle(wmOperatorType *ot) +{ + ot->name = "Circle Select"; + ot->description = "Select items using circle selection"; + ot->idname = "VIEW3D_OT_select_circle"; + + ot->invoke = WM_gesture_circle_invoke; + ot->modal = view3d_circle_select_modal; + ot->exec = view3d_circle_select_exec; + ot->poll = view3d_selectable_data; + ot->cancel = view3d_circle_select_cancel; + ot->get_name = ED_select_circle_get_name; + + /* flags */ + ot->flag = OPTYPE_UNDO; + + /* properties */ + WM_operator_properties_gesture_circle(ot); + WM_operator_properties_select_operation_simple(ot); +} + +/** \} */ -- cgit v1.2.3