Differential D3101 Diff 10392 source/blender/editors/armature/pose_select.c

Changeset View

Standalone View

source/blender/editors/armature/pose_select.c

Show All 39 Lines
#include "BLI_blenlib.h"		#include "BLI_blenlib.h"

#include "BKE_action.h"		#include "BKE_action.h"
#include "BKE_armature.h"		#include "BKE_armature.h"
#include "BKE_constraint.h"		#include "BKE_constraint.h"
#include "BKE_context.h"		#include "BKE_context.h"
#include "BKE_object.h"		#include "BKE_object.h"
#include "BKE_report.h"		#include "BKE_report.h"
		#include "BKE_layer.h"

#include "DEG_depsgraph.h"		#include "DEG_depsgraph.h"

#include "RNA_access.h"		#include "RNA_access.h"
#include "RNA_define.h"		#include "RNA_define.h"

#include "WM_api.h"		#include "WM_api.h"
#include "WM_types.h"		#include "WM_types.h"
▲ Show 20 Lines • Show All 84 Lines • ▼ Show 20 Lines	bool ED_armature_pose_select_pick_with_buffer(
Object *ob = base->object;		Object *ob = base->object;
Bone *nearBone;		Bone *nearBone;

if (!ob \|\| !ob->pose) return 0;		if (!ob \|\| !ob->pose) return 0;

Object *ob_act = OBACT(view_layer);		Object *ob_act = OBACT(view_layer);
Object *obedit = OBEDIT_FROM_VIEW_LAYER(view_layer);		Object *obedit = OBEDIT_FROM_VIEW_LAYER(view_layer);

nearBone = get_bone_from_selectbuffer(base, obedit, buffer, hits, 1, do_nearest);		/* Callers happen to already get the active base */
		Base *base_dummy = NULL;
		nearBone = get_bone_from_selectbuffer(&base, 1, obedit != NULL, buffer, hits, 1, do_nearest, &base_dummy);

/* if the bone cannot be affected, don't do anything */		/* if the bone cannot be affected, don't do anything */
if ((nearBone) && !(nearBone->flag & BONE_UNSELECTABLE)) {		if ((nearBone) && !(nearBone->flag & BONE_UNSELECTABLE)) {
bArmature *arm = ob->data;		bArmature *arm = ob->data;

/* since we do unified select, we don't shift+select a bone if the		/* since we do unified select, we don't shift+select a bone if the
* armature object was not active yet.		* armature object was not active yet.
* note, special exception for armature mode so we can do multi-select		* note, special exception for armature mode so we can do multi-select
* we could check for multi-select explicitly but think its fine to		* we could check for multi-select explicitly but think its fine to
* always give predictable behavior in weight paint mode - campbell */		* always give predictable behavior in weight paint mode - campbell */
if ((ob_act == NULL) \|\| ((ob_act != ob) && (ob_act->mode & OB_MODE_WEIGHT_PAINT) == 0)) {		if ((ob_act == NULL) \|\| ((ob_act != ob) && (ob_act->mode & OB_MODE_WEIGHT_PAINT) == 0)) {
/* when we are entering into posemode via toggle-select,		/* when we are entering into posemode via toggle-select,
* from another active object - always select the bone. */		* from another active object - always select the bone. */
if (!extend && !deselect && toggle) {		if (!extend && !deselect && toggle) {
/* re-select below */		/* re-select below */
nearBone->flag &= ~BONE_SELECTED;		nearBone->flag &= ~BONE_SELECTED;
}		}
}		}

if (!extend && !deselect && !toggle) {		if (!extend && !deselect && !toggle) {
ED_pose_deselect_all(ob, SEL_DESELECT, true);		{
		uint objects_len = 0;
		Object **objects = BKE_view_layer_array_from_objects_in_mode(
		view_layer, &objects_len, {
		.object_mode = OB_MODE_POSE,
		.no_dup_data = true});
		ED_pose_deselect_all_multi(objects, objects_len, SEL_DESELECT, true);
		MEM_SAFE_FREE(objects);
		}
nearBone->flag \|= (BONE_SELECTED \| BONE_TIPSEL \| BONE_ROOTSEL);		nearBone->flag \|= (BONE_SELECTED \| BONE_TIPSEL \| BONE_ROOTSEL);
arm->act_bone = nearBone;		arm->act_bone = nearBone;
}		}
else {		else {
if (extend) {		if (extend) {
nearBone->flag \|= (BONE_SELECTED \| BONE_TIPSEL \| BONE_ROOTSEL);		nearBone->flag \|= (BONE_SELECTED \| BONE_TIPSEL \| BONE_ROOTSEL);
arm->act_bone = nearBone;		arm->act_bone = nearBone;
}		}
▲ Show 20 Lines • Show All 69 Lines • ▼ Show 20 Lines	void ED_pose_deselect_all(Object *ob, int select_mode, const bool ignore_visibility)
for (pchan = ob->pose->chanbase.first; pchan; pchan = pchan->next) {		for (pchan = ob->pose->chanbase.first; pchan; pchan = pchan->next) {
/* ignore the pchan if it isn't visible or if its selection cannot be changed */		/* ignore the pchan if it isn't visible or if its selection cannot be changed */
if (ignore_visibility \|\| PBONE_VISIBLE(arm, pchan->bone)) {		if (ignore_visibility \|\| PBONE_VISIBLE(arm, pchan->bone)) {
pose_do_bone_select(pchan, select_mode);		pose_do_bone_select(pchan, select_mode);
}		}
}		}
}		}

		static bool ed_pose_is_any_selected(Object *ob, bool ignore_visibility)
		{
		bArmature *arm = ob->data;
		for (bPoseChannel *pchan = ob->pose->chanbase.first; pchan; pchan = pchan->next) {
		if (ignore_visibility \|\| PBONE_VISIBLE(arm, pchan->bone)) {
		if (pchan->bone->flag & BONE_SELECTED) {
		return true;
		}
		}
		}
		return false;
		}

		static bool ed_pose_is_any_selected_multi(Object **objects, uint objects_len, bool ignore_visibility)
		{
		for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
		Object *ob_iter = objects[ob_index];
		if (ed_pose_is_any_selected(ob_iter, ignore_visibility)) {
		return true;
		}
		}
		return false;
		}

		void ED_pose_deselect_all_multi(Object **objects, uint objects_len, int select_mode, const bool ignore_visibility)
		{
		if (select_mode == SEL_TOGGLE) {
		select_mode = ed_pose_is_any_selected_multi(
		objects, objects_len, ignore_visibility) ? SEL_DESELECT : SEL_SELECT;
		}

		for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
		Object *ob_iter = objects[ob_index];
		ED_pose_deselect_all(ob_iter, select_mode, ignore_visibility);
		}
		}

/* *************** Selections ******************** */		/* *************** Selections ******************** */

static void selectconnected_posebonechildren(Object ob, Bone bone, int extend)		static void selectconnected_posebonechildren(Object ob, Bone bone, int extend)
{		{
Bone *curBone;		Bone *curBone;

/* stop when unconnected child is encountered, or when unselectable bone is encountered */		/* stop when unconnected child is encountered, or when unselectable bone is encountered */
if (!(bone->flag & BONE_CONNECTED) \|\| (bone->flag & BONE_UNSELECTABLE))		if (!(bone->flag & BONE_CONNECTED) \|\| (bone->flag & BONE_UNSELECTABLE))
Show All 10 Lines	static void selectconnected_posebonechildren(Object ob, Bone bone, int extend)
for (curBone = bone->childbase.first; curBone; curBone = curBone->next)		for (curBone = bone->childbase.first; curBone; curBone = curBone->next)
selectconnected_posebonechildren(ob, curBone, extend);		selectconnected_posebonechildren(ob, curBone, extend);
}		}

/* within active object context */		/* within active object context */
/* previously known as "selectconnected_posearmature" */		/* previously known as "selectconnected_posearmature" */
static int pose_select_connected_invoke(bContext C, wmOperator op, const wmEvent *event)		static int pose_select_connected_invoke(bContext C, wmOperator op, const wmEvent *event)
{		{
Object *ob = BKE_object_pose_armature_get(CTX_data_active_object(C));
bArmature arm = (bArmature )ob->data;
Bone bone, curBone, *next = NULL;		Bone bone, curBone, *next = NULL;
const bool extend = RNA_boolean_get(op->ptr, "extend");		const bool extend = RNA_boolean_get(op->ptr, "extend");

view3d_operator_needs_opengl(C);		view3d_operator_needs_opengl(C);

bone = get_nearest_bone(C, event->mval, !extend);		Base *base = NULL;
		bone = get_nearest_bone(C, event->mval, !extend, &base);

if (!bone)		if (!bone)
return OPERATOR_CANCELLED;		return OPERATOR_CANCELLED;

		bArmature *arm = base->object->data;

/* Select parents */		/* Select parents */
for (curBone = bone; curBone; curBone = next) {		for (curBone = bone; curBone; curBone = next) {
/* ignore bone if cannot be selected */		/* ignore bone if cannot be selected */
if ((curBone->flag & BONE_UNSELECTABLE) == 0) {		if ((curBone->flag & BONE_UNSELECTABLE) == 0) {
if (extend)		if (extend)
curBone->flag &= ~BONE_SELECTED;		curBone->flag &= ~BONE_SELECTED;
else		else
curBone->flag \|= BONE_SELECTED;		curBone->flag \|= BONE_SELECTED;

if (curBone->flag & BONE_CONNECTED)		if (curBone->flag & BONE_CONNECTED)
next = curBone->parent;		next = curBone->parent;
else		else
next = NULL;		next = NULL;
}		}
else		else
next = NULL;		next = NULL;
}		}

/* Select children */		/* Select children */
for (curBone = bone->childbase.first; curBone; curBone = next)		for (curBone = bone->childbase.first; curBone; curBone = next)
selectconnected_posebonechildren(ob, curBone, extend);		selectconnected_posebonechildren(base->object, curBone, extend);

/* updates */		/* updates */
WM_event_add_notifier(C, NC_OBJECT \| ND_BONE_SELECT, ob);		WM_event_add_notifier(C, NC_OBJECT \| ND_BONE_SELECT, base->object);

if (arm->flag & ARM_HAS_VIZ_DEPS) {		if (arm->flag & ARM_HAS_VIZ_DEPS) {
/* mask modifier ('armature' mode), etc. */		/* mask modifier ('armature' mode), etc. */
DEG_id_tag_update(&ob->id, OB_RECALC_DATA);		DEG_id_tag_update(&base->object->id, OB_RECALC_DATA);
}		}

return OPERATOR_FINISHED;		return OPERATOR_FINISHED;
}		}

static int pose_select_linked_poll(bContext *C)		static int pose_select_linked_poll(bContext *C)
{		{
return (ED_operator_view3d_active(C) && ED_operator_posemode(C));		return (ED_operator_view3d_active(C) && ED_operator_posemode(C));
Show All 20 Lines

/* -------------------------------------- */		/* -------------------------------------- */

static int pose_de_select_all_exec(bContext C, wmOperator op)		static int pose_de_select_all_exec(bContext C, wmOperator op)
{		{
int action = RNA_enum_get(op->ptr, "action");		int action = RNA_enum_get(op->ptr, "action");

Scene *scene = CTX_data_scene(C);		Scene *scene = CTX_data_scene(C);
Object *ob = ED_object_context(C);
bArmature *arm = ob->data;
int multipaint = scene->toolsettings->multipaint;		int multipaint = scene->toolsettings->multipaint;

if (action == SEL_TOGGLE) {		if (action == SEL_TOGGLE) {
action = CTX_DATA_COUNT(C, selected_pose_bones) ? SEL_DESELECT : SEL_SELECT;		action = CTX_DATA_COUNT(C, selected_pose_bones) ? SEL_DESELECT : SEL_SELECT;
}		}

		Object *ob_prev = NULL;

/* Set the flags */		/* Set the flags */
CTX_DATA_BEGIN(C, bPoseChannel *, pchan, visible_pose_bones)		CTX_DATA_BEGIN_WITH_ID(C, bPoseChannel , pchan, visible_pose_bones, Object , ob)
{		{
		bArmature *arm = ob->data;
pose_do_bone_select(pchan, action);		pose_do_bone_select(pchan, action);
}
CTX_DATA_END;

WM_event_add_notifier(C, NC_OBJECT \| ND_BONE_SELECT, NULL);

		if (ob_prev != ob) {
/* weightpaint or mask modifiers need depsgraph updates */		/* weightpaint or mask modifiers need depsgraph updates */
if (multipaint \|\| (arm->flag & ARM_HAS_VIZ_DEPS)) {		if (multipaint \|\| (arm->flag & ARM_HAS_VIZ_DEPS)) {
DEG_id_tag_update(&ob->id, OB_RECALC_DATA);		DEG_id_tag_update(&ob->id, OB_RECALC_DATA);
}		}
		ob_prev = ob;
		}
		}
		CTX_DATA_END;

		WM_event_add_notifier(C, NC_OBJECT \| ND_BONE_SELECT, NULL);

return OPERATOR_FINISHED;		return OPERATOR_FINISHED;
}		}

void POSE_OT_select_all(wmOperatorType *ot)		void POSE_OT_select_all(wmOperatorType *ot)
{		{
/* identifiers */		/* identifiers */
ot->name = "(De)select All";		ot->name = "(De)select All";
▲ Show 20 Lines • Show All 59 Lines • ▼ Show 20 Lines	void POSE_OT_select_parent(wmOperatorType *ot)
/* flags */		/* flags */
ot->flag = OPTYPE_REGISTER \| OPTYPE_UNDO;		ot->flag = OPTYPE_REGISTER \| OPTYPE_UNDO;
}		}

/* -------------------------------------- */		/* -------------------------------------- */

static int pose_select_constraint_target_exec(bContext C, wmOperator UNUSED(op))		static int pose_select_constraint_target_exec(bContext C, wmOperator UNUSED(op))
{		{
Object *ob = BKE_object_pose_armature_get(CTX_data_active_object(C));
bArmature arm = (bArmature )ob->data;
bConstraint *con;		bConstraint *con;
int found = 0;		int found = 0;
		Object *ob_prev = NULL;

CTX_DATA_BEGIN (C, bPoseChannel *, pchan, visible_pose_bones)		CTX_DATA_BEGIN_WITH_ID (C, bPoseChannel , pchan, visible_pose_bones, Object , ob)
{		{
		bArmature *arm = ob->data;
if (pchan->bone->flag & BONE_SELECTED) {		if (pchan->bone->flag & BONE_SELECTED) {
for (con = pchan->constraints.first; con; con = con->next) {		for (con = pchan->constraints.first; con; con = con->next) {
const bConstraintTypeInfo *cti = BKE_constraint_typeinfo_get(con);		const bConstraintTypeInfo *cti = BKE_constraint_typeinfo_get(con);
ListBase targets = {NULL, NULL};		ListBase targets = {NULL, NULL};
bConstraintTarget *ct;		bConstraintTarget *ct;

if (cti && cti->get_constraint_targets) {		if (cti && cti->get_constraint_targets) {
cti->get_constraint_targets(con, &targets);		cti->get_constraint_targets(con, &targets);

for (ct = targets.first; ct; ct = ct->next) {		for (ct = targets.first; ct; ct = ct->next) {
if ((ct->tar == ob) && (ct->subtarget[0])) {		if ((ct->tar == ob) && (ct->subtarget[0])) {
bPoseChannel *pchanc = BKE_pose_channel_find_name(ob->pose, ct->subtarget);		bPoseChannel *pchanc = BKE_pose_channel_find_name(ob->pose, ct->subtarget);
if ((pchanc) && !(pchanc->bone->flag & BONE_UNSELECTABLE)) {		if ((pchanc) && !(pchanc->bone->flag & BONE_UNSELECTABLE)) {
pchanc->bone->flag \|= BONE_SELECTED \| BONE_TIPSEL \| BONE_ROOTSEL;		pchanc->bone->flag \|= BONE_SELECTED \| BONE_TIPSEL \| BONE_ROOTSEL;
found = 1;		found = 1;

		if (ob != ob_prev) {
		/* updates */
		WM_event_add_notifier(C, NC_OBJECT \| ND_BONE_SELECT, ob);
		if (arm->flag & ARM_HAS_VIZ_DEPS) {
		/* mask modifier ('armature' mode), etc. */
		DEG_id_tag_update(&ob->id, OB_RECALC_DATA);
		}
		ob_prev = ob;
		}
}		}
}		}
}		}

if (cti->flush_constraint_targets)		if (cti->flush_constraint_targets)
cti->flush_constraint_targets(con, &targets, 1);		cti->flush_constraint_targets(con, &targets, 1);
}		}
}		}
}		}
}		}
CTX_DATA_END;		CTX_DATA_END;

if (!found)		if (!found)
return OPERATOR_CANCELLED;		return OPERATOR_CANCELLED;

/* updates */
WM_event_add_notifier(C, NC_OBJECT \| ND_BONE_SELECT, ob);

if (arm->flag & ARM_HAS_VIZ_DEPS) {
/* mask modifier ('armature' mode), etc. */
DEG_id_tag_update(&ob->id, OB_RECALC_DATA);
}

return OPERATOR_FINISHED;		return OPERATOR_FINISHED;
}		}

void POSE_OT_select_constraint_target(wmOperatorType *ot)		void POSE_OT_select_constraint_target(wmOperatorType *ot)
{		{
/* identifiers */		/* identifiers */
ot->name = "Select Constraint Target";		ot->name = "Select Constraint Target";
ot->idname = "POSE_OT_select_constraint_target";		ot->idname = "POSE_OT_select_constraint_target";
▲ Show 20 Lines • Show All 449 Lines • Show Last 20 Lines