Multi-Objects: MBALL_OT_select_all

Based on D3893 by Habib Gahbiche.
This commit is contained in:
Dalai Felinto
2018-11-06 23:30:30 -02:00
parent f6bec570c5
commit 0a893c0901
3 changed files with 62 additions and 17 deletions

View File

@@ -49,6 +49,7 @@ struct MetaBall *BKE_mball_copy(struct Main *bmain, const struct MetaBall *mb);
void BKE_mball_make_local(struct Main *bmain, struct MetaBall *mb, const bool lib_local);
bool BKE_mball_is_any_selected(const struct MetaBall *mb);
bool BKE_mball_is_any_selected_multi(struct Object **objects, int objects_len);
bool BKE_mball_is_any_unselected(const struct MetaBall *mb);
bool BKE_mball_is_basis_for(struct Object *ob1, struct Object *ob2);
bool BKE_mball_is_basis(struct Object *ob);
@@ -71,8 +72,11 @@ void BKE_mball_translate(struct MetaBall *mb, const float offset[3]);
struct MetaElem *BKE_mball_element_add(struct MetaBall *mb, const int type);
void BKE_mball_select_all(struct MetaBall *mb);
void BKE_mball_select_all_multi(struct Object **objects, int objects_len);
void BKE_mball_deselect_all(struct MetaBall *mb);
void BKE_mball_deselect_all_multi(struct Object **objects, int objects_len);
void BKE_mball_select_swap(struct MetaBall *mb);
void BKE_mball_select_swap_multi(struct Object **objects, int objects_len);
/* **** Depsgraph evaluation **** */

View File

@@ -338,6 +338,19 @@ bool BKE_mball_is_any_selected(const MetaBall *mb)
return false;
}
bool BKE_mball_is_any_selected_multi(Object **objects, int objects_len)
{
for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
Object *obedit = objects[ob_index];
MetaBall *mb = (MetaBall *)obedit->data;
if (BKE_mball_is_any_selected(mb)) {
return true;
}
}
return false;
}
bool BKE_mball_is_any_unselected(const MetaBall *mb)
{
for (const MetaElem *ml = mb->editelems->first; ml != NULL; ml = ml->next) {
@@ -557,6 +570,15 @@ void BKE_mball_select_all(struct MetaBall *mb)
}
}
void BKE_mball_select_all_multi(Object **objects, int objects_len)
{
for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
Object *obedit = objects[ob_index];
MetaBall *mb = (MetaBall *)obedit->data;
BKE_mball_select_all(mb);
}
}
void BKE_mball_deselect_all(MetaBall *mb)
{
MetaElem *ml;
@@ -566,6 +588,16 @@ void BKE_mball_deselect_all(MetaBall *mb)
}
}
void BKE_mball_deselect_all_multi(Object **objects, int objects_len)
{
for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
Object *obedit = objects[ob_index];
MetaBall *mb = (MetaBall *)obedit->data;
BKE_mball_deselect_all(mb);
}
}
void BKE_mball_select_swap(struct MetaBall *mb)
{
MetaElem *ml;
@@ -575,6 +607,15 @@ void BKE_mball_select_swap(struct MetaBall *mb)
}
}
void BKE_mball_select_swap_multi(Object **objects, int objects_len)
{
for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
Object *obedit = objects[ob_index];
MetaBall *mb = (MetaBall *)obedit->data;
BKE_mball_select_swap(mb);
}
}
/* **** Depsgraph evaluation **** */
/* Draw Engine */

View File

@@ -125,38 +125,38 @@ MetaElem *ED_mball_add_primitive(bContext *UNUSED(C), Object *obedit, float mat[
/* Select or deselect all MetaElements */
static int mball_select_all_exec(bContext *C, wmOperator *op)
{
Object *obedit = CTX_data_edit_object(C);
MetaBall *mb = (MetaBall *)obedit->data;
MetaElem *ml;
int action = RNA_enum_get(op->ptr, "action");
if (BLI_listbase_is_empty(mb->editelems))
return OPERATOR_CANCELLED;
ViewLayer *view_layer = CTX_data_view_layer(C);
uint objects_len = 0;
Object **objects = BKE_view_layer_array_from_objects_in_edit_mode_unique_data(view_layer, &objects_len);
if (action == SEL_TOGGLE) {
action = SEL_SELECT;
for (ml = mb->editelems->first; ml; ml = ml->next) {
if (ml->flag & SELECT) {
action = SEL_DESELECT;
break;
}
}
action = BKE_mball_is_any_selected_multi(objects, objects_len) ?
SEL_DESELECT :
SEL_SELECT;
}
switch (action) {
case SEL_SELECT:
BKE_mball_select_all(mb);
BKE_mball_select_all_multi(objects, objects_len);
break;
case SEL_DESELECT:
BKE_mball_deselect_all(mb);
BKE_mball_deselect_all_multi(objects, objects_len);
break;
case SEL_INVERT:
BKE_mball_select_swap(mb);
BKE_mball_select_swap_multi(objects, objects_len);
break;
}
DEG_id_tag_update(&mb->id, DEG_TAG_SELECT_UPDATE);
WM_event_add_notifier(C, NC_GEOM | ND_SELECT, mb);
for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
Object *obedit = objects[ob_index];
MetaBall *mb = (MetaBall *)obedit->data;
DEG_id_tag_update(&mb->id, DEG_TAG_SELECT_UPDATE);
WM_event_add_notifier(C, NC_GEOM | ND_SELECT, mb);
}
MEM_freeN(objects);
return OPERATOR_FINISHED;
}