Edit last commit
[blender.git] / source / blender / editors / object / object_constraint.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): Joshua Leung, Blender Foundation
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/editors/object/object_constraint.c
29  *  \ingroup edobj
30  */
31
32
33 #include <stdio.h>
34 #include <string.h>
35
36 #include "MEM_guardedalloc.h"
37
38 #include "BLI_blenlib.h"
39 #include "BLI_math.h"
40 #include "BLI_dynstr.h"
41 #include "BLI_utildefines.h"
42
43 #include "BLT_translation.h"
44
45 #include "DNA_anim_types.h"
46 #include "DNA_constraint_types.h"
47 #include "DNA_curve_types.h"
48 #include "DNA_scene_types.h"
49 #include "DNA_text_types.h"
50 #include "DNA_object_types.h"
51
52 #include "BKE_action.h"
53 #include "BKE_armature.h"
54 #include "BKE_constraint.h"
55 #include "BKE_context.h"
56 #include "BKE_fcurve.h"
57 #include "BKE_global.h"
58 #include "BKE_main.h"
59 #include "BKE_object.h"
60 #include "BKE_report.h"
61 #include "BKE_tracking.h"
62 #include "BIK_api.h"
63
64 #include "DEG_depsgraph.h"
65 #include "DEG_depsgraph_build.h"
66
67 #ifdef WITH_PYTHON
68 #include "BPY_extern.h"
69 #endif
70
71 #include "WM_api.h"
72 #include "WM_types.h"
73
74 #include "RNA_access.h"
75 #include "RNA_define.h"
76 #include "RNA_enum_types.h"
77
78 #include "ED_object.h"
79 #include "ED_keyframing.h"
80 #include "ED_screen.h"
81
82 #include "UI_interface.h"
83 #include "UI_resources.h"
84
85 #include "object_intern.h"
86
87 /* -------------- Get Active Constraint Data ---------------------- */
88
89 /* if object in posemode, active bone constraints, else object constraints */
90 ListBase *get_active_constraints(Object *ob)
91 {
92         if (ob == NULL)
93                 return NULL;
94         
95         if (ob->mode & OB_MODE_POSE) {
96                 bPoseChannel *pchan;
97                 
98                 pchan = BKE_pose_channel_active(ob);
99                 if (pchan)
100                         return &pchan->constraints;
101         }
102         else 
103                 return &ob->constraints;
104         
105         return NULL;
106 }
107
108 /* Find the list that a given constraint belongs to, and/or also get the posechannel this is from (if applicable) */
109 ListBase *get_constraint_lb(Object *ob, bConstraint *con, bPoseChannel **r_pchan)
110 {
111         if (r_pchan)
112                 *r_pchan = NULL;
113         
114         if (ELEM(NULL, ob, con))
115                 return NULL;
116         
117         /* try object constraints first */
118         if ((BLI_findindex(&ob->constraints, con) != -1)) {
119                 return &ob->constraints;
120         }
121         
122         /* if armature, try pose bones too */
123         if (ob->pose) {
124                 bPoseChannel *pchan;
125                 
126                 /* try each bone in order 
127                  * NOTE: it's not possible to directly look up the active bone yet, so this will have to do
128                  */
129                 for (pchan = ob->pose->chanbase.first; pchan; pchan = pchan->next) {
130                         if ((BLI_findindex(&pchan->constraints, con) != -1)) {
131                                 
132                                 if (r_pchan)
133                                         *r_pchan = pchan;
134                                 
135                                 return &pchan->constraints;
136                         }
137                 }
138         }
139         
140         /* done */
141         return NULL;
142 }
143
144 /* single constraint */
145 bConstraint *get_active_constraint(Object *ob)
146 {
147         return BKE_constraints_active_get(get_active_constraints(ob));
148 }
149
150 /* -------------- Constraint Management (Add New, Remove, Rename) -------------------- */
151 #ifdef WITH_PYTHON
152 /* ------------- PyConstraints ------------------ */
153
154 /* this callback sets the text-file to be used for selected menu item */
155 static void validate_pyconstraint_cb(void *arg1, void *arg2)
156 {
157         bPythonConstraint *data = arg1;
158         Text *text = NULL;
159         int index = *((int *)arg2);
160         int i;
161         
162         /* exception for no script */
163         if (index) {
164                 /* innovative use of a for...loop to search */
165                 for (text = G.main->text.first, i = 1; text && index != i; i++, text = text->id.next) ;
166         }
167         data->text = text;
168 }
169
170 /* this returns a string for the list of usable pyconstraint script names */
171 static char *buildmenu_pyconstraints(Text *con_text, int *pyconindex)
172 {
173         DynStr *pupds = BLI_dynstr_new();
174         Text *text;
175         char *str;
176         char buf[64];
177         int i;
178         
179         /* add title first */
180         sprintf(buf, "Scripts: %%t|[None]%%x0|");
181         BLI_dynstr_append(pupds, buf);
182         
183         /* init active-index first */
184         if (con_text == NULL)
185                 *pyconindex = 0;
186         
187         /* loop through markers, adding them */
188         for (text = G.main->text.first, i = 1; text; i++, text = text->id.next) {
189                 /* this is important to ensure that right script is shown as active */
190                 if (text == con_text) *pyconindex = i;
191                 
192                 /* only include valid pyconstraint scripts */
193                 if (BPY_is_pyconstraint(text)) {
194                         BLI_dynstr_append(pupds, text->id.name + 2);
195                         
196                         sprintf(buf, "%%x%d", i);
197                         BLI_dynstr_append(pupds, buf);
198                         
199                         if (text->id.next)
200                                 BLI_dynstr_append(pupds, "|");
201                 }
202         }
203         
204         /* convert to normal MEM_malloc'd string */
205         str = BLI_dynstr_get_cstring(pupds);
206         BLI_dynstr_free(pupds);
207         
208         return str;
209 }
210 #endif /* WITH_PYTHON */
211
212 #if 0 // UNUSED, until pyconstraints are added back.
213 /* this callback gets called when the 'refresh' button of a pyconstraint gets pressed */
214 static void update_pyconstraint_cb(void *arg1, void *arg2)
215 {
216 #ifndef WITH_PYTHON
217         (void)arg1; /* unused */
218         (void)arg2; /* unused */
219 #else
220         Object *owner = (Object *)arg1;
221         bConstraint *con = (bConstraint *)arg2;
222         if (owner && con)
223                 BPY_pyconstraint_update(owner, con);
224 #endif
225 }
226 #endif // UNUSED
227
228 /* helper function for add_constriant - sets the last target for the active constraint */
229 static void set_constraint_nth_target(bConstraint *con, Object *target, const char subtarget[], int index)
230 {
231         const bConstraintTypeInfo *cti = BKE_constraint_typeinfo_get(con);
232         ListBase targets = {NULL, NULL};
233         bConstraintTarget *ct;
234         int num_targets, i;
235         
236         if (cti && cti->get_constraint_targets) {
237                 cti->get_constraint_targets(con, &targets);
238                 num_targets = BLI_listbase_count(&targets);
239                 
240                 if (index < 0) {
241                         if (abs(index) < num_targets)
242                                 index = num_targets - abs(index);
243                         else
244                                 index = num_targets - 1;
245                 }
246                 else if (index >= num_targets) {
247                         index = num_targets - 1;
248                 }
249                 
250                 for (ct = targets.first, i = 0; ct; ct = ct->next, i++) {
251                         if (i == index) {
252                                 ct->tar = target;
253                                 BLI_strncpy(ct->subtarget, subtarget, sizeof(ct->subtarget));
254                                 break;
255                         }
256                 }
257                 
258                 if (cti->flush_constraint_targets)
259                         cti->flush_constraint_targets(con, &targets, 0);
260         }
261 }
262
263 /* ------------- Constraint Sanity Testing ------------------- */
264
265 static void test_constraint(Object *owner, bPoseChannel *pchan, bConstraint *con, int type)
266 {
267         const bConstraintTypeInfo *cti = BKE_constraint_typeinfo_get(con);
268         ListBase targets = {NULL, NULL};
269         bConstraintTarget *ct;
270         bool check_targets = true;
271
272         /* clear disabled-flag first */
273         con->flag &= ~CONSTRAINT_DISABLE;
274
275         if (con->type == CONSTRAINT_TYPE_KINEMATIC) {
276                 bKinematicConstraint *data = con->data;
277
278                 /* bad: we need a separate set of checks here as poletarget is
279                  *              optional... otherwise poletarget must exist too or else
280                  *              the constraint is deemed invalid
281                  */
282                 /* default IK check ... */
283                 if (BKE_object_exists_check(data->tar) == 0) {
284                         data->tar = NULL;
285                         con->flag |= CONSTRAINT_DISABLE;
286                 }
287                 else if (data->tar == owner) {
288                         if (!BKE_armature_find_bone_name(BKE_armature_from_object(owner), data->subtarget)) {
289                                 con->flag |= CONSTRAINT_DISABLE;
290                         }
291                 }
292
293                 if (data->poletar) {
294                         if (BKE_object_exists_check(data->poletar) == 0) {
295                                 data->poletar = NULL;
296                                 con->flag |= CONSTRAINT_DISABLE;
297                         }
298                         else if (data->poletar == owner) {
299                                 if (!BKE_armature_find_bone_name(BKE_armature_from_object(owner), data->polesubtarget)) {
300                                         con->flag |= CONSTRAINT_DISABLE;
301                                 }
302                         }
303                 }
304                 /* ... can be overwritten here */
305                 BIK_test_constraint(owner, con);
306                 /* targets have already been checked for this */
307                 check_targets = false;
308         }
309         else if (con->type == CONSTRAINT_TYPE_PIVOT) {
310                 bPivotConstraint *data = con->data;
311
312                 /* target doesn't have to exist, but if it is non-null, it must exist! */
313                 if (data->tar && BKE_object_exists_check(data->tar) == 0) {
314                         data->tar = NULL;
315                         con->flag |= CONSTRAINT_DISABLE;
316                 }
317                 else if (data->tar == owner) {
318                         if (!BKE_armature_find_bone_name(BKE_armature_from_object(owner), data->subtarget)) {
319                                 con->flag |= CONSTRAINT_DISABLE;
320                         }
321                 }
322
323                 /* targets have already been checked for this */
324                 check_targets = false;
325         }
326         else if (con->type == CONSTRAINT_TYPE_ACTION) {
327                 bActionConstraint *data = con->data;
328
329                 /* validate action */
330                 if (data->act == NULL) {
331                         /* must have action */
332                         con->flag |= CONSTRAINT_DISABLE;
333                 }
334                 else if (data->act->idroot != ID_OB) {
335                         /* only object-rooted actions can be used */
336                         data->act = NULL;
337                         con->flag |= CONSTRAINT_DISABLE;
338                 }
339         }
340         else if (con->type == CONSTRAINT_TYPE_FOLLOWPATH) {
341                 bFollowPathConstraint *data = con->data;
342
343                 /* don't allow track/up axes to be the same */
344                 if (data->upflag == data->trackflag)
345                         con->flag |= CONSTRAINT_DISABLE;
346                 if (data->upflag + 3 == data->trackflag)
347                         con->flag |= CONSTRAINT_DISABLE;
348         }
349         else if (con->type == CONSTRAINT_TYPE_TRACKTO) {
350                 bTrackToConstraint *data = con->data;
351
352                 /* don't allow track/up axes to be the same */
353                 if (data->reserved2 == data->reserved1)
354                         con->flag |= CONSTRAINT_DISABLE;
355                 if (data->reserved2 + 3 == data->reserved1)
356                         con->flag |= CONSTRAINT_DISABLE;
357         }
358         else if (con->type == CONSTRAINT_TYPE_LOCKTRACK) {
359                 bLockTrackConstraint *data = con->data;
360
361                 if (data->lockflag == data->trackflag)
362                         con->flag |= CONSTRAINT_DISABLE;
363                 if (data->lockflag + 3 == data->trackflag)
364                         con->flag |= CONSTRAINT_DISABLE;
365         }
366         else if (con->type == CONSTRAINT_TYPE_SPLINEIK) {
367                 bSplineIKConstraint *data = con->data;
368
369                 /* if the number of points does not match the amount required by the chain length,
370                  * free the points array and request a rebind...
371                  */
372                 if ((data->points == NULL) || (data->numpoints != data->chainlen + 1)) {
373                         /* free the points array */
374                         if (data->points) {
375                                 MEM_freeN(data->points);
376                                 data->points = NULL;
377                         }
378
379                         /* clear the bound flag, forcing a rebind next time this is evaluated */
380                         data->flag &= ~CONSTRAINT_SPLINEIK_BOUND;
381                 }
382         }
383         else if (con->type == CONSTRAINT_TYPE_FOLLOWTRACK) {
384                 bFollowTrackConstraint *data = con->data;
385
386                 if ((data->flag & CAMERASOLVER_ACTIVECLIP) == 0) {
387                         if (data->clip != NULL && data->track[0]) {
388                                 MovieTracking *tracking = &data->clip->tracking;
389                                 MovieTrackingObject *tracking_object;
390
391                                 if (data->object[0])
392                                         tracking_object = BKE_tracking_object_get_named(tracking, data->object);
393                                 else
394                                         tracking_object = BKE_tracking_object_get_camera(tracking);
395
396                                 if (!tracking_object) {
397                                         con->flag |= CONSTRAINT_DISABLE;
398                                 }
399                                 else {
400                                         if (!BKE_tracking_track_get_named(tracking, tracking_object, data->track))
401                                                 con->flag |= CONSTRAINT_DISABLE;
402                                 }
403                         }
404                         else {
405                                 con->flag |= CONSTRAINT_DISABLE;
406                         }
407                 }
408         }
409         else if (con->type == CONSTRAINT_TYPE_CAMERASOLVER) {
410                 bCameraSolverConstraint *data = con->data;
411
412                 if ((data->flag & CAMERASOLVER_ACTIVECLIP) == 0 && (data->clip == NULL))
413                         con->flag |= CONSTRAINT_DISABLE;
414         }
415         else if (con->type == CONSTRAINT_TYPE_OBJECTSOLVER) {
416                 bObjectSolverConstraint *data = con->data;
417
418                 if ((data->flag & CAMERASOLVER_ACTIVECLIP) == 0 && (data->clip == NULL))
419                         con->flag |= CONSTRAINT_DISABLE;
420         }
421         else if (con->type == CONSTRAINT_TYPE_TRANSFORM_CACHE) {
422                 bTransformCacheConstraint *data = con->data;
423
424                 if ((data->cache_file == NULL) || (data->object_path[0] == '\0')) {
425                         con->flag |= CONSTRAINT_DISABLE;
426                 }
427         }
428
429         /* Check targets for constraints */
430         if (check_targets && cti && cti->get_constraint_targets) {
431                 cti->get_constraint_targets(con, &targets);
432
433                 /* disable and clear constraints targets that are incorrect */
434                 for (ct = targets.first; ct; ct = ct->next) {
435                         /* general validity checks (for those constraints that need this) */
436                         if (BKE_object_exists_check(ct->tar) == 0) {
437                                 /* object doesn't exist, but constraint requires target */
438                                 ct->tar = NULL;
439                                 con->flag |= CONSTRAINT_DISABLE;
440                         }
441                         else if (ct->tar == owner) {
442                                 if (type == CONSTRAINT_OBTYPE_BONE) {
443                                         if (!BKE_armature_find_bone_name(BKE_armature_from_object(owner), ct->subtarget)) {
444                                                 /* bone must exist in armature... */
445                                                 /* TODO: clear subtarget? */
446                                                 con->flag |= CONSTRAINT_DISABLE;
447                                         }
448                                         else if (STREQ(pchan->name, ct->subtarget)) {
449                                                 /* cannot target self */
450                                                 ct->subtarget[0] = '\0';
451                                                 con->flag |= CONSTRAINT_DISABLE;
452                                         }
453                                 }
454                                 else {
455                                         /* cannot use self as target */
456                                         ct->tar = NULL;
457                                         con->flag |= CONSTRAINT_DISABLE;
458                                 }
459                         }
460
461                         /* target checks for specific constraints */
462                         if (ELEM(con->type, CONSTRAINT_TYPE_FOLLOWPATH, CONSTRAINT_TYPE_CLAMPTO, CONSTRAINT_TYPE_SPLINEIK)) {
463                                 if (ct->tar) {
464                                         if (ct->tar->type != OB_CURVE) {
465                                                 ct->tar = NULL;
466                                                 con->flag |= CONSTRAINT_DISABLE;
467                                         }
468                                         else {
469                                                 Curve *cu = ct->tar->data;
470
471                                                 /* auto-set 'Path' setting on curve so this works  */
472                                                 cu->flag |= CU_PATH;
473                                         }
474                                 }
475                         }
476                 }
477
478                 /* free any temporary targets */
479                 if (cti->flush_constraint_targets)
480                         cti->flush_constraint_targets(con, &targets, 0);
481         }
482 }
483
484 static int constraint_type_get(Object *owner, bPoseChannel *pchan)
485 {
486         int type;
487         /* Check parents */
488         if (pchan) {
489                 switch (owner->type) {
490                         case OB_ARMATURE:
491                                 type = CONSTRAINT_OBTYPE_BONE;
492                                 break;
493                         default:
494                                 type = CONSTRAINT_OBTYPE_OBJECT;
495                                 break;
496                 }
497         }
498         else
499                 type = CONSTRAINT_OBTYPE_OBJECT;
500         return type;
501 }
502
503 /* checks validity of object pointers, and NULLs,
504  * if Bone doesnt exist it sets the CONSTRAINT_DISABLE flag.
505  */
506 static void test_constraints(Object *owner, bPoseChannel *pchan)
507 {
508         bConstraint *curcon;
509         ListBase *conlist = NULL;
510         int type;
511         
512         if (owner == NULL) return;
513
514         type = constraint_type_get(owner, pchan);
515
516         /* Get the constraint list for this object */
517         switch (type) {
518                 case CONSTRAINT_OBTYPE_OBJECT:
519                         conlist = &owner->constraints;
520                         break;
521                 case CONSTRAINT_OBTYPE_BONE:
522                         conlist = &pchan->constraints;
523                         break;
524         }
525         
526         /* Check all constraints - is constraint valid? */
527         if (conlist) {
528                 for (curcon = conlist->first; curcon; curcon = curcon->next) {
529                         test_constraint(owner, pchan, curcon, type);
530                 }
531         }
532 }
533
534 void object_test_constraints(Object *owner)
535 {
536         if (owner->constraints.first)
537                 test_constraints(owner, NULL);
538         
539         if (owner->type == OB_ARMATURE && owner->pose) {
540                 bPoseChannel *pchan;
541                 
542                 for (pchan = owner->pose->chanbase.first; pchan; pchan = pchan->next) {
543                         if (pchan->constraints.first)
544                                 test_constraints(owner, pchan);
545                 }
546         }
547 }
548
549 static void object_test_constraint(Object *owner, bConstraint *con)
550 {
551         if (owner->type == OB_ARMATURE && owner->pose) {
552                 if (BLI_findindex(&owner->constraints, con) != -1) {
553                         test_constraint(owner, NULL, con, CONSTRAINT_OBTYPE_OBJECT);
554                 }
555                 else {
556                         bPoseChannel *pchan;
557                         for (pchan = owner->pose->chanbase.first; pchan; pchan = pchan->next) {
558                                 if (BLI_findindex(&pchan->constraints, con) != -1) {
559                                         test_constraint(owner, pchan, con, CONSTRAINT_OBTYPE_BONE);
560                                         break;
561                                 }
562                         }
563                 }
564         }
565         else {
566                 test_constraint(owner, NULL, con, CONSTRAINT_OBTYPE_OBJECT);
567         }
568 }
569
570 /************************ generic functions for operators using constraint names and data context *********************/
571
572 #define EDIT_CONSTRAINT_OWNER_OBJECT    0
573 #define EDIT_CONSTRAINT_OWNER_BONE      1
574
575 static const EnumPropertyItem constraint_owner_items[] = {
576         {EDIT_CONSTRAINT_OWNER_OBJECT, "OBJECT", 0, "Object", "Edit a constraint on the active object"},
577         {EDIT_CONSTRAINT_OWNER_BONE, "BONE", 0, "Bone", "Edit a constraint on the active bone"},
578         {0, NULL, 0, NULL, NULL}};
579
580
581 static int edit_constraint_poll_generic(bContext *C, StructRNA *rna_type)
582 {
583         PointerRNA ptr = CTX_data_pointer_get_type(C, "constraint", rna_type);
584         Object *ob = (ptr.id.data) ? ptr.id.data : ED_object_active_context(C);
585
586         if (!ptr.data) {
587                 CTX_wm_operator_poll_msg_set(C, "Context missing 'constraint'");
588                 return 0;
589         }
590
591         if (!ob) {
592                 CTX_wm_operator_poll_msg_set(C, "Context missing active object");
593                 return 0;
594         }
595
596         if (ID_IS_LINKED(ob) || (ptr.id.data && ID_IS_LINKED(ptr.id.data))) {
597                 CTX_wm_operator_poll_msg_set(C, "Cannot edit library data");
598                 return 0;
599         }
600
601         if (ID_IS_STATIC_OVERRIDE(ob)) {
602                 CTX_wm_operator_poll_msg_set(C, "Cannot edit constraints comming from static override");
603                 return (((bConstraint *)ptr.data)->flag & CONSTRAINT_STATICOVERRIDE_LOCAL) != 0;
604         }
605
606         return 1;
607 }
608
609 static int edit_constraint_poll(bContext *C)
610 {
611         return edit_constraint_poll_generic(C, &RNA_Constraint);
612 }
613
614 static void edit_constraint_properties(wmOperatorType *ot)
615 {
616         RNA_def_string(ot->srna, "constraint", NULL, MAX_NAME, "Constraint", "Name of the constraint to edit");
617         RNA_def_enum(ot->srna, "owner", constraint_owner_items, 0, "Owner", "The owner of this constraint");
618 }
619
620 static int edit_constraint_invoke_properties(bContext *C, wmOperator *op)
621 {
622         PointerRNA ptr = CTX_data_pointer_get_type(C, "constraint", &RNA_Constraint);
623         Object *ob = (ptr.id.data) ? ptr.id.data : ED_object_active_context(C);
624         bConstraint *con;
625         ListBase *list;
626         
627         if (RNA_struct_property_is_set(op->ptr, "constraint") && RNA_struct_property_is_set(op->ptr, "owner"))
628                 return 1;
629         
630         if (ptr.data) {
631                 con = ptr.data;
632                 RNA_string_set(op->ptr, "constraint", con->name);
633                 
634                 list = get_constraint_lb(ob, con, NULL);
635                 
636                 if (&ob->constraints == list)
637                         RNA_enum_set(op->ptr, "owner", EDIT_CONSTRAINT_OWNER_OBJECT);
638                 else
639                         RNA_enum_set(op->ptr, "owner", EDIT_CONSTRAINT_OWNER_BONE);
640                 
641                 return 1;
642         }
643         
644         return 0;
645 }
646
647 static bConstraint *edit_constraint_property_get(wmOperator *op, Object *ob, int type)
648 {
649         char constraint_name[MAX_NAME];
650         int owner = RNA_enum_get(op->ptr, "owner");
651         bConstraint *con;
652         ListBase *list = NULL;
653         
654         RNA_string_get(op->ptr, "constraint", constraint_name);
655         
656         if (owner == EDIT_CONSTRAINT_OWNER_OBJECT) {
657                 list = &ob->constraints;
658         }
659         else if (owner == EDIT_CONSTRAINT_OWNER_BONE) {
660                 bPoseChannel *pchan = BKE_pose_channel_active(ob);
661                 if (pchan)
662                         list = &pchan->constraints;
663                 else {
664                         //if (G.debug & G_DEBUG)
665                         //printf("edit_constraint_property_get: No active bone for object '%s'\n", (ob) ? ob->id.name + 2 : "<None>");
666                         return NULL;
667                 }
668         }
669         else {
670                 //if (G.debug & G_DEBUG)
671                 //printf("edit_constraint_property_get: defaulting to getting list in the standard way\n");
672                 list = get_active_constraints(ob);
673         }
674         
675         con = BKE_constraints_find_name(list, constraint_name);
676         //if (G.debug & G_DEBUG)
677         //printf("constraint found = %p, %s\n", (void *)con, (con) ? con->name : "<Not found>");
678
679         if (con && (type != 0) && (con->type != type))
680                 con = NULL;
681         
682         return con;
683 }
684
685 /* ********************** CONSTRAINT-SPECIFIC STUFF ********************* */
686
687 /* ---------- Distance-Dependent Constraints ---------- */
688 /* StretchTo, Limit Distance */
689
690 static int stretchto_reset_exec(bContext *C, wmOperator *op)
691 {
692         Object *ob = ED_object_active_context(C);
693         bConstraint *con = edit_constraint_property_get(op, ob, CONSTRAINT_TYPE_STRETCHTO);
694         bStretchToConstraint *data = (con) ? (bStretchToConstraint *)con->data : NULL;
695         
696         /* despite 3 layers of checks, we may still not be able to find a constraint */
697         if (data == NULL)
698                 return OPERATOR_CANCELLED;
699         
700         /* just set original length to 0.0, which will cause a reset on next recalc */
701         data->orglength = 0.0f;
702         ED_object_constraint_update(ob);
703         
704         WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT, NULL);
705         return OPERATOR_FINISHED;
706 }
707
708 static int stretchto_reset_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
709 {
710         if (edit_constraint_invoke_properties(C, op))
711                 return stretchto_reset_exec(C, op);
712         else
713                 return OPERATOR_CANCELLED;
714 }
715
716 void CONSTRAINT_OT_stretchto_reset(wmOperatorType *ot)
717 {
718         /* identifiers */
719         ot->name = "Reset Original Length";
720         ot->idname = "CONSTRAINT_OT_stretchto_reset";
721         ot->description = "Reset original length of bone for Stretch To Constraint";
722         
723         /* callbacks */
724         ot->invoke = stretchto_reset_invoke;
725         ot->exec = stretchto_reset_exec;
726         ot->poll = edit_constraint_poll;
727         
728         /* flags */
729         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
730         
731         /* properties */
732         edit_constraint_properties(ot);
733 }
734
735
736 static int limitdistance_reset_exec(bContext *C, wmOperator *op)
737 {
738         Object *ob = ED_object_active_context(C);
739         bConstraint *con = edit_constraint_property_get(op, ob, CONSTRAINT_TYPE_DISTLIMIT);
740         bDistLimitConstraint *data = (con) ? (bDistLimitConstraint *)con->data : NULL;
741         
742         /* despite 3 layers of checks, we may still not be able to find a constraint */
743         if (data == NULL)
744                 return OPERATOR_CANCELLED;
745         
746         /* just set original length to 0.0, which will cause a reset on next recalc */
747         data->dist = 0.0f;
748         ED_object_constraint_update(ob);
749         
750         WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT, NULL);
751         return OPERATOR_FINISHED;
752 }
753
754 static int limitdistance_reset_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
755 {
756         if (edit_constraint_invoke_properties(C, op))
757                 return limitdistance_reset_exec(C, op);
758         else
759                 return OPERATOR_CANCELLED;
760 }
761
762 void CONSTRAINT_OT_limitdistance_reset(wmOperatorType *ot)
763 {
764         /* identifiers */
765         ot->name = "Reset Distance";
766         ot->idname = "CONSTRAINT_OT_limitdistance_reset";
767         ot->description = "Reset limiting distance for Limit Distance Constraint";
768         
769         /* callbacks */
770         ot->invoke = limitdistance_reset_invoke;
771         ot->exec = limitdistance_reset_exec;
772         ot->poll = edit_constraint_poll;
773         
774         /* flags */
775         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
776         
777         /* properties */
778         edit_constraint_properties(ot);
779 }
780
781 /* ------------- Child-Of Constraint ------------------ */
782
783 static void child_get_inverse_matrix(const bContext *C, Scene *scene, Object *ob, bConstraint *con, float invmat[4][4], const int owner)
784 {
785         Depsgraph *depsgraph = CTX_data_depsgraph(C);
786
787         /* nullify inverse matrix first */
788         unit_m4(invmat);
789         
790         if (owner == EDIT_CONSTRAINT_OWNER_BONE) {
791                 bPoseChannel *pchan;
792                 /* try to find a pose channel - assume that this is the constraint owner */
793                 /* TODO: get from context instead? */
794                 if (ob && ob->pose && (pchan = BKE_pose_channel_active(ob))) {
795                         bConstraint *con_last;
796                         /* calculate/set inverse matrix:
797                          *  We just calculate all transform-stack eval up to but not including this constraint.
798                          *  This is because inverse should just inverse correct for just the constraint's influence
799                          *  when it gets applied; that is, at the time of application, we don't know anything about
800                          *  what follows.
801                          */
802                         float imat[4][4], tmat[4][4];
803                         float pmat[4][4];
804
805                         /* make sure we passed the correct constraint */
806                         BLI_assert(BLI_findindex(&pchan->constraints, con) != -1);
807
808                         /* 1. calculate posemat where inverse doesn't exist yet (inverse was cleared above),
809                          * to use as baseline ("pmat") to derive delta from. This extra calc saves users
810                          * from having pressing "Clear Inverse" first
811                          */
812                         BKE_pose_where_is(depsgraph, scene, ob);
813                         copy_m4_m4(pmat, pchan->pose_mat);
814
815                         /* 2. knock out constraints starting from this one */
816                         con_last = pchan->constraints.last;
817                         pchan->constraints.last = con->prev;
818
819                         if (con->prev) {
820                                 /* new end must not point to this one, else this chain cutting is useless */
821                                 con->prev->next = NULL;
822                         }
823                         else {
824                                 /* constraint was first */
825                                 pchan->constraints.first = NULL;
826                         }
827
828                         /* 3. solve pose without disabled constraints */
829                         BKE_pose_where_is(depsgraph, scene, ob);
830
831                         /* 4. determine effect of constraint by removing the newly calculated
832                          * pchan->pose_mat from the original pchan->pose_mat, thus determining
833                          * the effect of the constraint
834                          */
835                         invert_m4_m4(imat, pchan->pose_mat);
836                         mul_m4_m4m4(tmat, pmat, imat);
837                         invert_m4_m4(invmat, tmat);
838
839                         /* 5. restore constraints */
840                         pchan->constraints.last = con_last;
841
842                         if (con->prev) {
843                                 /* hook up prev to this one again */
844                                 con->prev->next = con;
845                         }
846                         else {
847                                 /* set as first again */
848                                 pchan->constraints.first = con;
849                         }
850
851                         /* 6. recalculate pose with new inv-mat applied */
852                         BKE_pose_where_is(depsgraph, scene, ob);
853                 }
854         }
855         if (owner == EDIT_CONSTRAINT_OWNER_OBJECT) {
856                 if (ob) {
857                         Object workob;
858
859                         /* make sure we passed the correct constraint */
860                         BLI_assert(BLI_findindex(&ob->constraints, con) != -1);
861
862                         /* use BKE_object_workob_calc_parent to find inverse - just like for normal parenting */
863                         BKE_object_workob_calc_parent(depsgraph, scene, ob, &workob);
864                         invert_m4_m4(invmat, workob.obmat);
865                 }
866         }
867 }
868
869 /* ChildOf Constraint - set inverse callback */
870 static int childof_set_inverse_exec(bContext *C, wmOperator *op)
871 {
872         Scene *scene = CTX_data_scene(C);
873         Object *ob = ED_object_active_context(C);
874         bConstraint *con = edit_constraint_property_get(op, ob, CONSTRAINT_TYPE_CHILDOF);
875         bChildOfConstraint *data = (con) ? (bChildOfConstraint *)con->data : NULL;
876         const int owner = RNA_enum_get(op->ptr, "owner");
877
878         /* despite 3 layers of checks, we may still not be able to find a constraint */
879         if (data == NULL) {
880                 printf("DEBUG: Child-Of Set Inverse - object = '%s'\n", (ob) ? ob->id.name + 2 : "<None>");
881                 BKE_report(op->reports, RPT_ERROR, "Could not find constraint data for Child-Of Set Inverse");
882                 return OPERATOR_CANCELLED;
883         }
884         
885         child_get_inverse_matrix(C, scene, ob, con, data->invmat, owner);
886
887         WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT, ob);
888         
889         return OPERATOR_FINISHED;
890 }
891
892 static int childof_set_inverse_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
893 {
894         if (edit_constraint_invoke_properties(C, op))
895                 return childof_set_inverse_exec(C, op);
896         else
897                 return OPERATOR_CANCELLED;
898 }
899
900 void CONSTRAINT_OT_childof_set_inverse(wmOperatorType *ot)
901 {
902         /* identifiers */
903         ot->name = "Set Inverse";
904         ot->idname = "CONSTRAINT_OT_childof_set_inverse";
905         ot->description = "Set inverse correction for ChildOf constraint";
906         
907         /* callbacks */
908         ot->invoke = childof_set_inverse_invoke;
909         ot->exec = childof_set_inverse_exec;
910         ot->poll = edit_constraint_poll;
911         
912         /* flags */
913         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
914         
915         /* properties */
916         edit_constraint_properties(ot);
917 }
918
919 /* ChildOf Constraint - clear inverse callback */
920 static int childof_clear_inverse_exec(bContext *C, wmOperator *op)
921 {
922         Object *ob = ED_object_active_context(C);
923         bConstraint *con = edit_constraint_property_get(op, ob, CONSTRAINT_TYPE_CHILDOF);
924         bChildOfConstraint *data = (con) ? (bChildOfConstraint *)con->data : NULL;
925         
926         if (data == NULL) {
927                 BKE_report(op->reports, RPT_ERROR, "Child Of constraint not found");
928                 return OPERATOR_CANCELLED;
929         }
930         
931         /* simply clear the matrix */
932         unit_m4(data->invmat);
933         
934         WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT, ob);
935         
936         return OPERATOR_FINISHED;
937 }
938
939 static int childof_clear_inverse_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
940 {
941         if (edit_constraint_invoke_properties(C, op))
942                 return childof_clear_inverse_exec(C, op);
943         else
944                 return OPERATOR_CANCELLED;
945 }
946
947 void CONSTRAINT_OT_childof_clear_inverse(wmOperatorType *ot)
948 {
949         /* identifiers */
950         ot->name = "Clear Inverse";
951         ot->idname = "CONSTRAINT_OT_childof_clear_inverse";
952         ot->description = "Clear inverse correction for ChildOf constraint";
953         
954         /* callbacks */
955         ot->invoke = childof_clear_inverse_invoke;
956         ot->exec = childof_clear_inverse_exec;
957         ot->poll = edit_constraint_poll;
958         
959         /* flags */
960         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
961         
962         /* properties */
963         edit_constraint_properties(ot);
964 }
965
966 /* --------------- Follow Path Constraint ------------------ */
967
968 static int followpath_path_animate_exec(bContext *C, wmOperator *op)
969 {
970         Object *ob = ED_object_active_context(C);
971         bConstraint *con = edit_constraint_property_get(op, ob, CONSTRAINT_TYPE_FOLLOWPATH);
972         bFollowPathConstraint *data = (con) ? (bFollowPathConstraint *)con->data : NULL;
973         
974         bAction *act = NULL;
975         FCurve *fcu = NULL;
976         int sfra = RNA_int_get(op->ptr, "frame_start");
977         int len  = RNA_int_get(op->ptr, "length");
978         float standardRange = 1.0;
979         
980         /* nearly impossible sanity check */
981         if (data == NULL) {
982                 BKE_report(op->reports, RPT_ERROR, "Follow Path constraint not found");
983                 return OPERATOR_CANCELLED;
984         }
985         
986         /* add F-Curve as appropriate */
987         if (data->tar) {
988                 Curve *cu = (Curve *)data->tar->data;
989                 
990                 if (ELEM(NULL, cu->adt, cu->adt->action) ||
991                     (list_find_fcurve(&cu->adt->action->curves, "eval_time", 0) == NULL))
992                 {
993                         /* create F-Curve for path animation */
994                         act = verify_adt_action(&cu->id, 1);
995                         fcu = verify_fcurve(act, NULL, NULL, "eval_time", 0, 1);
996                         
997                         /* standard vertical range - 1:1 = 100 frames */
998                         standardRange = 100.0f;
999                 }
1000                 else {
1001                         /* path anim exists already - abort for now as this may well be what was intended */
1002                         BKE_report(op->reports, RPT_WARNING, "Path is already animated");
1003                         return OPERATOR_CANCELLED;
1004                 }
1005         }
1006         else {
1007                 /* animate constraint's "fixed offset" */
1008                 PointerRNA ptr;
1009                 PropertyRNA *prop;
1010                 char *path;
1011                 
1012                 /* get RNA pointer to constraint's "offset_factor" property - to build RNA path */
1013                 RNA_pointer_create(&ob->id, &RNA_FollowPathConstraint, con, &ptr);
1014                 prop = RNA_struct_find_property(&ptr, "offset_factor");
1015                 
1016                 path = RNA_path_from_ID_to_property(&ptr, prop);
1017                 
1018                 /* create F-Curve for constraint */
1019                 act = verify_adt_action(&ob->id, 1);
1020                 fcu = verify_fcurve(act, NULL, NULL, path, 0, 1);
1021                 
1022                 /* standard vertical range - 0.0 to 1.0 */
1023                 standardRange = 1.0f;
1024                 
1025                 /* enable "Use Fixed Position" so that animating this has effect */
1026                 data->followflag |= FOLLOWPATH_STATIC;
1027                 
1028                 /* path needs to be freed */
1029                 if (path) 
1030                         MEM_freeN(path);
1031         }
1032         
1033         /* setup dummy 'generator' modifier here to get 1-1 correspondence still working
1034          * and define basic slope of this curve based on the properties
1035          */
1036         if (!fcu->bezt && !fcu->fpt && !fcu->modifiers.first) {
1037                 FModifier *fcm = add_fmodifier(&fcu->modifiers, FMODIFIER_TYPE_GENERATOR, fcu);
1038                 FMod_Generator *gen = fcm->data;
1039                 
1040                 /* Assume that we have the following equation:
1041                  *     y = Ax + B
1042                  *         1    0       <-- coefficients array indices
1043                  */
1044                 float A = standardRange / (float)(len);
1045                 float B = (float)(-sfra) * A;
1046                 
1047                 gen->coefficients[1] = A;
1048                 gen->coefficients[0] = B;
1049         }
1050         
1051         /* updates... */
1052         WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT, ob);
1053         return OPERATOR_FINISHED;
1054 }
1055
1056 static int followpath_path_animate_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1057 {
1058         /* hook up invoke properties for figuring out which constraint we're dealing with */
1059         if (edit_constraint_invoke_properties(C, op)) {
1060                 return followpath_path_animate_exec(C, op);
1061         }
1062         else {
1063                 return OPERATOR_CANCELLED;
1064         }
1065 }
1066
1067 void CONSTRAINT_OT_followpath_path_animate(wmOperatorType *ot)
1068 {
1069         /* identifiers */
1070         ot->name = "Auto Animate Path";
1071         ot->idname = "CONSTRAINT_OT_followpath_path_animate";
1072         ot->description = "Add default animation for path used by constraint if it isn't animated already";
1073         
1074         /* callbacks */
1075         ot->invoke = followpath_path_animate_invoke;
1076         ot->exec = followpath_path_animate_exec;
1077         ot->poll = edit_constraint_poll;
1078         
1079         /* flags */
1080         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1081         
1082         /* props */
1083         edit_constraint_properties(ot);
1084         RNA_def_int(ot->srna, "frame_start", 1, MINAFRAME, MAXFRAME, "Start Frame", 
1085                     "First frame of path animation", MINAFRAME, MAXFRAME);
1086         RNA_def_int(ot->srna, "length", 100, 0, MAXFRAME, "Length", 
1087                     "Number of frames that path animation should take", 0, MAXFRAME);
1088 }
1089
1090 /* ------------- Object Solver Constraint ------------------ */
1091
1092 static int objectsolver_set_inverse_exec(bContext *C, wmOperator *op)
1093 {
1094         Scene *scene = CTX_data_scene(C);
1095         Object *ob = ED_object_active_context(C);
1096         bConstraint *con = edit_constraint_property_get(op, ob, CONSTRAINT_TYPE_OBJECTSOLVER);
1097         bObjectSolverConstraint *data = (con) ? (bObjectSolverConstraint *)con->data : NULL;
1098         const int owner = RNA_enum_get(op->ptr, "owner");
1099
1100         /* despite 3 layers of checks, we may still not be able to find a constraint */
1101         if (data == NULL) {
1102                 printf("DEBUG: Child-Of Set Inverse - object = '%s'\n", (ob) ? ob->id.name + 2 : "<None>");
1103                 BKE_report(op->reports, RPT_ERROR, "Could not find constraint data for Child-Of Set Inverse");
1104                 return OPERATOR_CANCELLED;
1105         }
1106
1107         child_get_inverse_matrix(C, scene, ob, con, data->invmat, owner);
1108
1109         WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT, ob);
1110
1111         return OPERATOR_FINISHED;
1112 }
1113
1114 static int objectsolver_set_inverse_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1115 {
1116         if (edit_constraint_invoke_properties(C, op))
1117                 return objectsolver_set_inverse_exec(C, op);
1118         else
1119                 return OPERATOR_CANCELLED;
1120 }
1121
1122 void CONSTRAINT_OT_objectsolver_set_inverse(wmOperatorType *ot)
1123 {
1124         /* identifiers */
1125         ot->name = "Set Inverse";
1126         ot->idname = "CONSTRAINT_OT_objectsolver_set_inverse";
1127         ot->description = "Set inverse correction for ObjectSolver constraint";
1128         
1129         /* callbacks */
1130         ot->invoke = objectsolver_set_inverse_invoke;
1131         ot->exec = objectsolver_set_inverse_exec;
1132         ot->poll = edit_constraint_poll;
1133
1134         /* flags */
1135         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1136         
1137         /* properties */
1138         edit_constraint_properties(ot);
1139 }
1140
1141 static int objectsolver_clear_inverse_exec(bContext *C, wmOperator *op)
1142 {
1143         Object *ob = ED_object_active_context(C);
1144         bConstraint *con = edit_constraint_property_get(op, ob, CONSTRAINT_TYPE_OBJECTSOLVER);
1145         bObjectSolverConstraint *data = (con) ? (bObjectSolverConstraint *)con->data : NULL;
1146
1147         if (data == NULL) {
1148                 BKE_report(op->reports, RPT_ERROR, "Child Of constraint not found");
1149                 return OPERATOR_CANCELLED;
1150         }
1151
1152         /* simply clear the matrix */
1153         unit_m4(data->invmat);
1154
1155         WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT, ob);
1156
1157         return OPERATOR_FINISHED;
1158 }
1159
1160 static int objectsolver_clear_inverse_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1161 {
1162         if (edit_constraint_invoke_properties(C, op))
1163                 return objectsolver_clear_inverse_exec(C, op);
1164         else
1165                 return OPERATOR_CANCELLED;
1166 }
1167
1168 void CONSTRAINT_OT_objectsolver_clear_inverse(wmOperatorType *ot)
1169 {
1170         /* identifiers */
1171         ot->name = "Clear Inverse";
1172         ot->idname = "CONSTRAINT_OT_objectsolver_clear_inverse";
1173         ot->description = "Clear inverse correction for ObjectSolver constraint";
1174         
1175         /* callbacks */
1176         ot->invoke = objectsolver_clear_inverse_invoke;
1177         ot->exec = objectsolver_clear_inverse_exec;
1178         ot->poll = edit_constraint_poll;
1179
1180         /* flags */
1181         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1182         
1183         /* properties */
1184         edit_constraint_properties(ot);
1185 }
1186
1187 /***************************** BUTTONS ****************************/
1188
1189 void ED_object_constraint_set_active(Object *ob, bConstraint *con)
1190 {       
1191         ListBase *lb = get_constraint_lb(ob, con, NULL);
1192         
1193         /* lets be nice and escape if its active already */
1194         /* NOTE: this assumes that the stack doesn't have other active ones set... */
1195         if ((lb && con) && (con->flag & CONSTRAINT_ACTIVE))
1196                 return;
1197         
1198         BKE_constraints_active_set(lb, con);
1199 }
1200
1201 void ED_object_constraint_update(Object *ob)
1202 {
1203         if (ob->pose) BKE_pose_update_constraint_flags(ob->pose);
1204
1205         object_test_constraints(ob);
1206
1207         if (ob->type == OB_ARMATURE) 
1208                 DEG_id_tag_update(&ob->id, OB_RECALC_DATA | OB_RECALC_OB);
1209         else 
1210                 DEG_id_tag_update(&ob->id, OB_RECALC_OB);
1211 }
1212
1213 static void object_pose_tag_update(Main *bmain, Object *ob)
1214 {
1215         BKE_pose_tag_recalc(bmain, ob->pose);    /* Checks & sort pose channels. */
1216         if (ob->proxy && ob->adt) {
1217                 /* We need to make use of ugly POSE_ANIMATION_WORKAROUND here too, else anim data are not reloaded
1218                  * after calling `BKE_pose_rebuild()`, which causes T43872.
1219                  * Note that this is a bit wide here, since we cannot be sure whether there are some locked proxy bones
1220                  * or not...
1221                  * XXX Temp hack until new depsgraph hopefully solves this. */
1222                 ob->adt->recalc |= ADT_RECALC_ANIM;
1223         }
1224 }
1225
1226 void ED_object_constraint_dependency_update(Main *bmain, Object *ob)
1227 {
1228         ED_object_constraint_update(ob);
1229
1230         if (ob->pose) {
1231                 object_pose_tag_update(bmain, ob);
1232         }
1233         DEG_relations_tag_update(bmain);
1234 }
1235
1236 void ED_object_constraint_tag_update(Object *ob, bConstraint *con)
1237 {
1238         if (ob->pose) {
1239                 BKE_pose_tag_update_constraint_flags(ob->pose);
1240         }
1241
1242         object_test_constraint(ob, con);
1243
1244         if (ob->type == OB_ARMATURE)
1245                 DEG_id_tag_update(&ob->id, OB_RECALC_DATA | OB_RECALC_OB);
1246         else
1247                 DEG_id_tag_update(&ob->id, OB_RECALC_OB);
1248
1249         /* Do Copy-on-Write tag here too, otherwise constraint
1250          * influence/mute buttons in UI have no effect
1251          */
1252         DEG_id_tag_update(&ob->id, DEG_TAG_COPY_ON_WRITE);
1253 }
1254
1255 void ED_object_constraint_dependency_tag_update(Main *bmain, Object *ob, bConstraint *con)
1256 {
1257         ED_object_constraint_tag_update(ob, con);
1258
1259         if (ob->pose) {
1260                 object_pose_tag_update(bmain, ob);
1261         }
1262         DEG_relations_tag_update(bmain);
1263 }
1264
1265 static int constraint_poll(bContext *C)
1266 {
1267         PointerRNA ptr = CTX_data_pointer_get_type(C, "constraint", &RNA_Constraint);
1268         return (ptr.id.data && ptr.data);
1269 }
1270
1271
1272 static int constraint_delete_exec(bContext *C, wmOperator *UNUSED(op))
1273 {
1274         PointerRNA ptr = CTX_data_pointer_get_type(C, "constraint", &RNA_Constraint);
1275         Object *ob = ptr.id.data;
1276         bConstraint *con = ptr.data;
1277         ListBase *lb = get_constraint_lb(ob, con, NULL);
1278
1279         /* free the constraint */
1280         if (BKE_constraint_remove_ex(lb, ob, con, true)) {
1281                 /* there's no active constraint now, so make sure this is the case */
1282                 BKE_constraints_active_set(&ob->constraints, NULL);
1283                 ED_object_constraint_update(ob); /* needed to set the flags on posebones correctly */
1284
1285                 /* relatiols */
1286                 DEG_relations_tag_update(CTX_data_main(C));
1287
1288                 /* notifiers */
1289                 WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT | NA_REMOVED, ob);
1290                 
1291                 return OPERATOR_FINISHED;
1292         }
1293         else {
1294                 /* couldn't remove due to some invalid data */
1295                 return OPERATOR_CANCELLED;
1296         }
1297 }
1298
1299 void CONSTRAINT_OT_delete(wmOperatorType *ot)
1300 {
1301         /* identifiers */
1302         ot->name = "Delete Constraint";
1303         ot->idname = "CONSTRAINT_OT_delete";
1304         ot->description = "Remove constraint from constraint stack";
1305         
1306         /* callbacks */
1307         ot->exec = constraint_delete_exec;
1308         ot->poll = constraint_poll;
1309         
1310         /* flags */
1311         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1312 }
1313
1314
1315 static int constraint_move_down_exec(bContext *C, wmOperator *op)
1316 {
1317         Object *ob = ED_object_active_context(C);
1318         bConstraint *con = edit_constraint_property_get(op, ob, 0);
1319         
1320         if (con && con->next) {
1321                 ListBase *conlist = get_constraint_lb(ob, con, NULL);
1322                 bConstraint *nextCon = con->next;
1323                 
1324                 /* insert the nominated constraint after the one that used to be after it */
1325                 BLI_remlink(conlist, con);
1326                 BLI_insertlinkafter(conlist, nextCon, con);
1327                 
1328                 WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT, ob);
1329                 
1330                 return OPERATOR_FINISHED;
1331         }
1332         
1333         return OPERATOR_CANCELLED;
1334 }
1335
1336 static int constraint_move_down_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1337 {
1338         if (edit_constraint_invoke_properties(C, op))
1339                 return constraint_move_down_exec(C, op);
1340         else
1341                 return OPERATOR_CANCELLED;
1342 }
1343
1344 void CONSTRAINT_OT_move_down(wmOperatorType *ot)
1345 {
1346         /* identifiers */
1347         ot->name = "Move Constraint Down";
1348         ot->idname = "CONSTRAINT_OT_move_down";
1349         ot->description = "Move constraint down in constraint stack";
1350         
1351         /* callbacks */
1352         ot->invoke = constraint_move_down_invoke;
1353         ot->exec = constraint_move_down_exec;
1354         ot->poll = edit_constraint_poll;
1355         
1356         /* flags */
1357         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1358         
1359         /* properties */
1360         edit_constraint_properties(ot);
1361 }
1362
1363
1364 static int constraint_move_up_exec(bContext *C, wmOperator *op)
1365 {
1366         Object *ob = ED_object_active_context(C);
1367         bConstraint *con = edit_constraint_property_get(op, ob, 0);
1368         
1369         if (con && con->prev) {
1370                 ListBase *conlist = get_constraint_lb(ob, con, NULL);
1371                 bConstraint *prevCon = con->prev;
1372                 
1373                 /* insert the nominated constraint before the one that used to be before it */
1374                 BLI_remlink(conlist, con);
1375                 BLI_insertlinkbefore(conlist, prevCon, con);
1376                 
1377                 WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT, ob);
1378                 
1379                 return OPERATOR_FINISHED;
1380         }
1381         
1382         return OPERATOR_CANCELLED;
1383 }
1384
1385 static int constraint_move_up_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1386 {
1387         if (edit_constraint_invoke_properties(C, op))
1388                 return constraint_move_up_exec(C, op);
1389         else
1390                 return OPERATOR_CANCELLED;
1391 }
1392
1393 void CONSTRAINT_OT_move_up(wmOperatorType *ot)
1394 {
1395         /* identifiers */
1396         ot->name = "Move Constraint Up";
1397         ot->idname = "CONSTRAINT_OT_move_up";
1398         ot->description = "Move constraint up in constraint stack";
1399         
1400         /* callbacks */
1401         ot->exec = constraint_move_up_exec;
1402         ot->invoke = constraint_move_up_invoke;
1403         ot->poll = edit_constraint_poll;
1404         
1405         /* flags */
1406         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1407         edit_constraint_properties(ot);
1408 }
1409
1410 /***************************** OPERATORS ****************************/
1411
1412 /************************ remove constraint operators *********************/
1413
1414 static int pose_constraints_clear_exec(bContext *C, wmOperator *UNUSED(op))
1415 {
1416         Main *bmain = CTX_data_main(C);
1417         Object *ob = BKE_object_pose_armature_get(CTX_data_active_object(C));
1418         
1419         /* free constraints for all selected bones */
1420         CTX_DATA_BEGIN (C, bPoseChannel *, pchan, selected_pose_bones)
1421         {
1422                 BKE_constraints_free(&pchan->constraints);
1423                 pchan->constflag &= ~(PCHAN_HAS_IK | PCHAN_HAS_SPLINEIK | PCHAN_HAS_CONST);
1424         }
1425         CTX_DATA_END;
1426         
1427         /* force depsgraph to get recalculated since relationships removed */
1428         DEG_relations_tag_update(bmain);
1429         
1430         /* note, calling BIK_clear_data() isn't needed here */
1431
1432         /* do updates */
1433         DEG_id_tag_update(&ob->id, OB_RECALC_DATA);
1434         WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT | NA_REMOVED, ob);
1435         
1436         return OPERATOR_FINISHED;
1437 }
1438
1439 void POSE_OT_constraints_clear(wmOperatorType *ot)
1440 {
1441         /* identifiers */
1442         ot->name = "Clear Pose Constraints";
1443         ot->idname = "POSE_OT_constraints_clear";
1444         ot->description = "Clear all the constraints for the selected bones";
1445         
1446         /* callbacks */
1447         ot->exec = pose_constraints_clear_exec;
1448         ot->poll = ED_operator_posemode_exclusive; // XXX - do we want to ensure there are selected bones too?
1449 }
1450
1451
1452 static int object_constraints_clear_exec(bContext *C, wmOperator *UNUSED(op))
1453 {
1454         Main *bmain = CTX_data_main(C);
1455         
1456         /* do freeing */
1457         CTX_DATA_BEGIN (C, Object *, ob, selected_editable_objects)
1458         {
1459                 BKE_constraints_free(&ob->constraints);
1460                 DEG_id_tag_update(&ob->id, OB_RECALC_OB);
1461         }
1462         CTX_DATA_END;
1463         
1464         /* force depsgraph to get recalculated since relationships removed */
1465         DEG_relations_tag_update(bmain);
1466         
1467         /* do updates */
1468         WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT | NA_REMOVED, NULL);
1469         
1470         return OPERATOR_FINISHED;
1471 }
1472
1473 void OBJECT_OT_constraints_clear(wmOperatorType *ot)
1474 {
1475         /* identifiers */
1476         ot->name = "Clear Object Constraints";
1477         ot->idname = "OBJECT_OT_constraints_clear";
1478         ot->description = "Clear all the constraints for the active Object only";
1479         
1480         /* callbacks */
1481         ot->exec = object_constraints_clear_exec;
1482         ot->poll = ED_operator_object_active_editable;
1483 }
1484
1485 /************************ copy all constraints operators *********************/
1486
1487 static int pose_constraint_copy_exec(bContext *C, wmOperator *op)
1488 {
1489         Main *bmain = CTX_data_main(C);
1490         bPoseChannel *pchan = CTX_data_active_pose_bone(C);
1491         ListBase lb;
1492         CollectionPointerLink *link;
1493         
1494         /* don't do anything if bone doesn't exist or doesn't have any constraints */
1495         if (ELEM(NULL, pchan, pchan->constraints.first)) {
1496                 BKE_report(op->reports, RPT_ERROR, "No active bone with constraints for copying");
1497                 return OPERATOR_CANCELLED;
1498         }
1499         
1500         /* copy all constraints from active posebone to all selected posebones */
1501         CTX_data_selected_pose_bones(C, &lb);
1502         for (link = lb.first; link; link = link->next) {
1503                 Object *ob = link->ptr.id.data;
1504                 bPoseChannel *chan = link->ptr.data;
1505                 
1506                 /* if we're not handling the object we're copying from, copy all constraints over */
1507                 if (pchan != chan) {
1508                         BKE_constraints_copy(&chan->constraints, &pchan->constraints, true);
1509                         /* update flags (need to add here, not just copy) */
1510                         chan->constflag |= pchan->constflag;
1511
1512                         BKE_pose_tag_recalc(bmain, ob->pose);
1513                         DEG_id_tag_update((ID *)ob, OB_RECALC_DATA);
1514                 }
1515         }
1516         BLI_freelistN(&lb);
1517         
1518         /* force depsgraph to get recalculated since new relationships added */
1519         DEG_relations_tag_update(bmain);
1520
1521         WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT, NULL);
1522         
1523         return OPERATOR_FINISHED;
1524 }
1525
1526 void POSE_OT_constraints_copy(wmOperatorType *ot)
1527 {
1528         /* identifiers */
1529         ot->name = "Copy Constraints to Selected Bones";
1530         ot->idname = "POSE_OT_constraints_copy";
1531         ot->description = "Copy constraints to other selected bones";
1532         
1533         /* api callbacks */
1534         ot->exec = pose_constraint_copy_exec;
1535         ot->poll = ED_operator_posemode_exclusive;
1536
1537         /* flags */
1538         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1539 }
1540
1541 static int object_constraint_copy_exec(bContext *C, wmOperator *UNUSED(op))
1542 {
1543         Main *bmain = CTX_data_main(C);
1544         Object *obact = ED_object_active_context(C);
1545         
1546         /* copy all constraints from active object to all selected objects */
1547         CTX_DATA_BEGIN (C, Object *, ob, selected_editable_objects)
1548         {
1549                 /* if we're not handling the object we're copying from, copy all constraints over */
1550                 if (obact != ob) {
1551                         BKE_constraints_copy(&ob->constraints, &obact->constraints, true);
1552                         DEG_id_tag_update(&ob->id, OB_RECALC_DATA);
1553                 }
1554         }
1555         CTX_DATA_END;
1556         
1557         /* force depsgraph to get recalculated since new relationships added */
1558         DEG_relations_tag_update(bmain);
1559         
1560         /* notifiers for updates */
1561         WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT | NA_ADDED, NULL);
1562         
1563         return OPERATOR_FINISHED;
1564 }
1565
1566 void OBJECT_OT_constraints_copy(wmOperatorType *ot)
1567 {
1568         /* identifiers */
1569         ot->name = "Copy Constraints to Selected Objects";
1570         ot->idname = "OBJECT_OT_constraints_copy";
1571         ot->description = "Copy constraints to other selected objects";
1572         
1573         /* api callbacks */
1574         ot->exec = object_constraint_copy_exec;
1575         ot->poll = ED_operator_object_active_editable;
1576
1577         /* flags */
1578         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1579 }
1580
1581 /************************ add constraint operators *********************/
1582
1583 /* get the Object and/or PoseChannel to use as target */
1584 static bool get_new_constraint_target(bContext *C, int con_type, Object **tar_ob, bPoseChannel **tar_pchan, bool add)
1585 {
1586         Object *obact = ED_object_active_context(C);
1587         bPoseChannel *pchanact = BKE_pose_channel_active(obact);
1588         bool only_curve = false, only_mesh = false, only_ob = false;
1589         bool found = false;
1590         
1591         /* clear tar_ob and tar_pchan fields before use 
1592          *      - assume for now that both always exist...
1593          */
1594         *tar_ob = NULL;
1595         *tar_pchan = NULL;
1596         
1597         /* check if constraint type doesn't requires a target
1598          *      - if so, no need to get any targets 
1599          */
1600         switch (con_type) {
1601                 /* no-target constraints --------------------------- */
1602                 /* null constraint - shouldn't even be added! */
1603                 case CONSTRAINT_TYPE_NULL:
1604                 /* limit constraints - no targets needed */
1605                 case CONSTRAINT_TYPE_LOCLIMIT:
1606                 case CONSTRAINT_TYPE_ROTLIMIT:
1607                 case CONSTRAINT_TYPE_SIZELIMIT:
1608                 case CONSTRAINT_TYPE_SAMEVOL:
1609                         return false;
1610                         
1611                 /* restricted target-type constraints -------------- */
1612                 /* NOTE: for these, we cannot try to add a target object if no valid ones are found, since that doesn't work */
1613                 /* curve-based constraints - set the only_curve and only_ob flags */
1614                 case CONSTRAINT_TYPE_CLAMPTO:
1615                 case CONSTRAINT_TYPE_FOLLOWPATH:
1616                 case CONSTRAINT_TYPE_SPLINEIK:
1617                         only_curve = true;
1618                         only_ob = true;
1619                         add = false;
1620                         break;
1621                         
1622                 /* mesh only? */
1623                 case CONSTRAINT_TYPE_SHRINKWRAP:
1624                         only_mesh = true;
1625                         only_ob = true;
1626                         add = false;
1627                         break;
1628         }
1629         
1630         /* if the active Object is Armature, and we can search for bones, do so... */
1631         if ((obact->type == OB_ARMATURE) && (only_ob == false)) {
1632                 /* search in list of selected Pose-Channels for target */
1633                 CTX_DATA_BEGIN (C, bPoseChannel *, pchan, selected_pose_bones)
1634                 {
1635                         /* just use the first one that we encounter, as long as it is not the active one */
1636                         if (pchan != pchanact) {
1637                                 *tar_ob = obact;
1638                                 *tar_pchan = pchan;
1639                                 found = true;
1640                                 
1641                                 break;
1642                         }
1643                 }
1644                 CTX_DATA_END;
1645         }
1646         
1647         /* if not yet found, try selected Objects... */
1648         if (found == false) {
1649                 /* search in selected objects context */
1650                 CTX_DATA_BEGIN (C, Object *, ob, selected_objects)
1651                 {
1652                         /* just use the first object we encounter (that isn't the active object) 
1653                          * and which fulfills the criteria for the object-target that we've got 
1654                          */
1655                         if (ob != obact) {
1656                                 /* for armatures in pose mode, look inside the armature for the active bone
1657                                  * so that we set up cross-armature constraints with less effort
1658                                  */
1659                                 if ((ob->type == OB_ARMATURE) && (ob->mode & OB_MODE_POSE) &&
1660                                     (!only_curve && !only_mesh))
1661                                 {
1662                                         /* just use the active bone, and assume that it is visible + usable */
1663                                         *tar_ob = ob;
1664                                         *tar_pchan = BKE_pose_channel_active(ob);
1665                                         found = true;
1666                                         
1667                                         break;
1668                                 }
1669                                 else if (((!only_curve) || (ob->type == OB_CURVE)) &&
1670                                          ((!only_mesh) || (ob->type == OB_MESH)))
1671                                 {
1672                                         /* set target */
1673                                         *tar_ob = ob;
1674                                         found = true;
1675                                         
1676                                         /* perform some special operations on the target */
1677                                         if (only_curve) {
1678                                                 /* Curve-Path option must be enabled for follow-path constraints to be able to work */
1679                                                 Curve *cu = (Curve *)ob->data;
1680                                                 cu->flag |= CU_PATH;
1681                                         }
1682                                         
1683                                         break;
1684                                 }
1685                         }
1686                 }
1687                 CTX_DATA_END;
1688         }
1689         
1690         /* if still not found, add a new empty to act as a target (if allowed) */
1691         if ((found == false) && (add)) {
1692                 Main *bmain = CTX_data_main(C);
1693                 Scene *scene = CTX_data_scene(C);
1694                 ViewLayer *view_layer = CTX_data_view_layer(C);
1695                 Base *base = BASACT(view_layer), *newbase = NULL;
1696                 Object *obt;
1697                 
1698                 /* add new target object */
1699                 obt = BKE_object_add(bmain, scene, view_layer, OB_EMPTY, NULL);
1700                 
1701                 /* set layers OK */
1702                 newbase = BASACT(view_layer);
1703                 newbase->lay = base->lay;
1704                 obt->lay = newbase->lay;
1705                 
1706                 /* transform cent to global coords for loc */
1707                 if (pchanact) {
1708                         /* since by default, IK targets the tip of the last bone, use the tip of the active PoseChannel 
1709                          * if adding a target for an IK Constraint
1710                          */
1711                         if (con_type == CONSTRAINT_TYPE_KINEMATIC)
1712                                 mul_v3_m4v3(obt->loc, obact->obmat, pchanact->pose_tail);
1713                         else
1714                                 mul_v3_m4v3(obt->loc, obact->obmat, pchanact->pose_head);
1715                 }
1716                 else {
1717                         copy_v3_v3(obt->loc, obact->obmat[3]);
1718                 }
1719
1720                 /* restore, BKE_object_add sets active */
1721                 BASACT(view_layer) = base;
1722                 base->flag |= BASE_SELECTED;
1723                 
1724                 /* make our new target the new object */
1725                 *tar_ob = obt;
1726                 found = true;
1727         }
1728         
1729         /* return whether there's any target */
1730         return found;
1731 }
1732
1733 /* used by add constraint operators to add the constraint required */
1734 static int constraint_add_exec(bContext *C, wmOperator *op, Object *ob, ListBase *list, int type, const bool setTarget)
1735 {
1736         Main *bmain = CTX_data_main(C);
1737         bPoseChannel *pchan;
1738         bConstraint *con;
1739         
1740         if (list == &ob->constraints) {
1741                 pchan = NULL;
1742         }
1743         else {
1744                 pchan = BKE_pose_channel_active(ob);
1745                 
1746                 /* ensure not to confuse object/pose adding */
1747                 if (pchan == NULL) {
1748                         BKE_report(op->reports, RPT_ERROR, "No active pose bone to add a constraint to");
1749                         return OPERATOR_CANCELLED;
1750                 }
1751         }
1752         /* check if constraint to be added is valid for the given constraints stack */
1753         if (type == CONSTRAINT_TYPE_NULL) {
1754                 return OPERATOR_CANCELLED;
1755         }
1756         if ((type == CONSTRAINT_TYPE_KINEMATIC) && ((!pchan) || (list != &pchan->constraints))) {
1757                 BKE_report(op->reports, RPT_ERROR, "IK constraint can only be added to bones");
1758                 return OPERATOR_CANCELLED;
1759         }
1760         if ((type == CONSTRAINT_TYPE_SPLINEIK) && ((!pchan) || (list != &pchan->constraints))) {
1761                 BKE_report(op->reports, RPT_ERROR, "Spline IK constraint can only be added to bones");
1762                 return OPERATOR_CANCELLED;
1763         }
1764         
1765         /* create a new constraint of the type requried, and add it to the active/given constraints list */
1766         if (pchan)
1767                 con = BKE_constraint_add_for_pose(ob, pchan, NULL, type);
1768         else
1769                 con = BKE_constraint_add_for_object(ob, NULL, type);
1770         
1771         /* get the first selected object/bone, and make that the target
1772          *      - apart from the buttons-window add buttons, we shouldn't add in this way
1773          */
1774         if (setTarget) {
1775                 Object *tar_ob = NULL;
1776                 bPoseChannel *tar_pchan = NULL;
1777                 
1778                 /* get the target objects, adding them as need be */
1779                 if (get_new_constraint_target(C, type, &tar_ob, &tar_pchan, 1)) {
1780                         /* method of setting target depends on the type of target we've got 
1781                          *      - by default, just set the first target (distinction here is only for multiple-targeted constraints)
1782                          */
1783                         if (tar_pchan)
1784                                 set_constraint_nth_target(con, tar_ob, tar_pchan->name, 0);
1785                         else
1786                                 set_constraint_nth_target(con, tar_ob, "", 0);
1787                 }
1788         }
1789         
1790         /* do type-specific tweaking to the constraint settings  */
1791         switch (type) {
1792                 case CONSTRAINT_TYPE_PYTHON: /* FIXME: this code is not really valid anymore */
1793                 {
1794 #ifdef WITH_PYTHON
1795                         char *menustr;
1796                         int scriptint = 0;
1797                         /* popup a list of usable scripts */
1798                         menustr = buildmenu_pyconstraints(NULL, &scriptint);
1799                         /* XXX scriptint = pupmenu(menustr); */
1800                         MEM_freeN(menustr);
1801
1802                         /* only add constraint if a script was chosen */
1803                         if (scriptint) {
1804                                 /* add constraint */
1805                                 validate_pyconstraint_cb(con->data, &scriptint);
1806                                 
1807                                 /* make sure target allowance is set correctly */
1808                                 BPY_pyconstraint_update(ob, con);
1809                         }
1810 #endif
1811                         break;
1812                 }
1813
1814                 default:
1815                         break;
1816         }
1817         
1818         /* make sure all settings are valid - similar to above checks, but sometimes can be wrong */
1819         object_test_constraints(ob);
1820
1821         if (pchan)
1822                 BKE_pose_update_constraint_flags(ob->pose);
1823
1824
1825         /* force depsgraph to get recalculated since new relationships added */
1826         DEG_relations_tag_update(bmain);
1827         
1828         if ((ob->type == OB_ARMATURE) && (pchan)) {
1829                 BKE_pose_tag_recalc(bmain, ob->pose);  /* sort pose channels */
1830                 if (BKE_constraints_proxylocked_owner(ob, pchan) && ob->adt) {
1831                         /* We need to make use of ugly POSE_ANIMATION_WORKAROUND here too, else anim data are not reloaded
1832                          * after calling `BKE_pose_rebuild()`, which causes T43872.
1833                          * XXX Temp hack until new depsgraph hopefully solves this. */
1834                         ob->adt->recalc |= ADT_RECALC_ANIM;
1835                 }
1836                 DEG_id_tag_update(&ob->id, OB_RECALC_DATA | OB_RECALC_OB);
1837         }
1838         else
1839                 DEG_id_tag_update(&ob->id, OB_RECALC_OB);
1840         
1841         /* notifiers for updates */
1842         WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT | NA_ADDED, ob);
1843         
1844         return OPERATOR_FINISHED;
1845 }
1846
1847 /* ------------------ */
1848
1849 /* dummy operator callback */
1850 static int object_constraint_add_exec(bContext *C, wmOperator *op)
1851 {
1852         Object *ob = ED_object_active_context(C);
1853         int type = RNA_enum_get(op->ptr, "type");
1854         short with_targets = 0;
1855         
1856         if (!ob) {
1857                 BKE_report(op->reports, RPT_ERROR, "No active object to add constraint to");
1858                 return OPERATOR_CANCELLED;
1859         }
1860                 
1861         /* hack: set constraint targets from selected objects in context is allowed when
1862          *              operator name included 'with_targets', since the menu doesn't allow multiple properties
1863          */
1864         if (strstr(op->idname, "with_targets"))
1865                 with_targets = 1;
1866
1867         return constraint_add_exec(C, op, ob, &ob->constraints, type, with_targets);
1868 }
1869
1870 /* dummy operator callback */
1871 static int pose_constraint_add_exec(bContext *C, wmOperator *op)
1872 {
1873         Object *ob = BKE_object_pose_armature_get(ED_object_active_context(C));
1874         int type = RNA_enum_get(op->ptr, "type");
1875         short with_targets = 0;
1876         
1877         if (!ob) {
1878                 BKE_report(op->reports, RPT_ERROR, "No active object to add constraint to");
1879                 return OPERATOR_CANCELLED;
1880         }
1881                 
1882         /* hack: set constraint targets from selected objects in context is allowed when
1883          *              operator name included 'with_targets', since the menu doesn't allow multiple properties
1884          */
1885         if (strstr(op->idname, "with_targets"))
1886                 with_targets = 1;
1887         
1888         return constraint_add_exec(C, op, ob, get_active_constraints(ob), type, with_targets);
1889 }
1890
1891 /* ------------------ */
1892
1893 void OBJECT_OT_constraint_add(wmOperatorType *ot)
1894 {
1895         /* identifiers */
1896         ot->name = "Add Constraint";
1897         ot->description = "Add a constraint to the active object";
1898         ot->idname = "OBJECT_OT_constraint_add";
1899         
1900         /* api callbacks */
1901         ot->invoke = WM_menu_invoke;
1902         ot->exec = object_constraint_add_exec;
1903         ot->poll = ED_operator_object_active_editable;
1904         
1905         /* flags */
1906         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1907         
1908         /* properties */
1909         ot->prop = RNA_def_enum(ot->srna, "type", rna_enum_constraint_type_items, 0, "Type", "");
1910 }
1911
1912 void OBJECT_OT_constraint_add_with_targets(wmOperatorType *ot)
1913 {
1914         /* identifiers */
1915         ot->name = "Add Constraint (with Targets)";
1916         ot->description = "Add a constraint to the active object, with target (where applicable) set to the selected Objects/Bones";
1917         ot->idname = "OBJECT_OT_constraint_add_with_targets";
1918         
1919         /* api callbacks */
1920         ot->invoke = WM_menu_invoke;
1921         ot->exec = object_constraint_add_exec;
1922         ot->poll = ED_operator_object_active_editable;
1923         
1924         /* flags */
1925         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1926         
1927         /* properties */
1928         ot->prop = RNA_def_enum(ot->srna, "type", rna_enum_constraint_type_items, 0, "Type", "");
1929 }
1930
1931 void POSE_OT_constraint_add(wmOperatorType *ot)
1932 {
1933         /* identifiers */
1934         ot->name = "Add Constraint";
1935         ot->description = "Add a constraint to the active bone";
1936         ot->idname = "POSE_OT_constraint_add";
1937         
1938         /* api callbacks */
1939         ot->invoke = WM_menu_invoke;
1940         ot->exec = pose_constraint_add_exec;
1941         ot->poll = ED_operator_posemode_exclusive;
1942         
1943         /* flags */
1944         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1945         
1946         /* properties */
1947         ot->prop = RNA_def_enum(ot->srna, "type", rna_enum_constraint_type_items, 0, "Type", "");
1948 }
1949
1950 void POSE_OT_constraint_add_with_targets(wmOperatorType *ot)
1951 {
1952         /* identifiers */
1953         ot->name = "Add Constraint (with Targets)";
1954         ot->description = "Add a constraint to the active bone, with target (where applicable) set to the selected Objects/Bones";
1955         ot->idname = "POSE_OT_constraint_add_with_targets";
1956         
1957         /* api callbacks */
1958         ot->invoke = WM_menu_invoke;
1959         ot->exec = pose_constraint_add_exec;
1960         ot->poll = ED_operator_posemode_exclusive;
1961         
1962         /* flags */
1963         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
1964         
1965         /* properties */
1966         ot->prop = RNA_def_enum(ot->srna, "type", rna_enum_constraint_type_items, 0, "Type", "");
1967 }
1968
1969 /************************ IK Constraint operators *********************/
1970 /* NOTE: only for Pose-Channels */
1971 // TODO: should these be here, or back in editors/armature/poseobject.c again?
1972
1973 /* present menu with options + validation for targets to use */
1974 static int pose_ik_add_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
1975 {
1976         Object *ob = BKE_object_pose_armature_get(CTX_data_active_object(C));
1977         bPoseChannel *pchan = BKE_pose_channel_active(ob);
1978         bConstraint *con = NULL;
1979         
1980         uiPopupMenu *pup;
1981         uiLayout *layout;
1982         Object *tar_ob = NULL;
1983         bPoseChannel *tar_pchan = NULL;
1984         
1985         /* must have active bone */
1986         if (ELEM(NULL, ob, pchan)) {
1987                 BKE_report(op->reports, RPT_ERROR, "Must have an active bone to add IK constraint to");
1988                 return OPERATOR_CANCELLED;
1989         }
1990         
1991         /* bone must not have any constraints already */
1992         for (con = pchan->constraints.first; con; con = con->next) {
1993                 if (con->type == CONSTRAINT_TYPE_KINEMATIC) break;
1994         }
1995         if (con) {
1996                 BKE_report(op->reports, RPT_ERROR, "Bone already has an IK constraint");
1997                 return OPERATOR_CANCELLED;
1998         }
1999         
2000         /* prepare popup menu to choose targetting options */
2001         pup = UI_popup_menu_begin(C, IFACE_("Add IK"), ICON_NONE);
2002         layout = UI_popup_menu_layout(pup);
2003         
2004         /* the type of targets we'll set determines the menu entries to show... */
2005         if (get_new_constraint_target(C, CONSTRAINT_TYPE_KINEMATIC, &tar_ob, &tar_pchan, 0)) {
2006                 /* bone target, or object target? 
2007                  *      - the only thing that matters is that we want a target...
2008                  */
2009                 if (tar_pchan)
2010                         uiItemBooleanO(layout, IFACE_("To Active Bone"), ICON_NONE, "POSE_OT_ik_add", "with_targets", 1);
2011                 else
2012                         uiItemBooleanO(layout, IFACE_("To Active Object"), ICON_NONE, "POSE_OT_ik_add", "with_targets", 1);
2013         }
2014         else {
2015                 /* we have a choice of adding to a new empty, or not setting any target (targetless IK) */
2016                 uiItemBooleanO(layout, IFACE_("To New Empty Object"), ICON_NONE, "POSE_OT_ik_add", "with_targets", 1);
2017                 uiItemBooleanO(layout, IFACE_("Without Targets"), ICON_NONE, "POSE_OT_ik_add", "with_targets", 0);
2018         }
2019         
2020         /* finish building the menu, and process it (should result in calling self again) */
2021         UI_popup_menu_end(C, pup);
2022         
2023         return OPERATOR_INTERFACE;
2024 }
2025
2026 /* call constraint_add_exec() to add the IK constraint */
2027 static int pose_ik_add_exec(bContext *C, wmOperator *op)
2028 {
2029         Object *ob = CTX_data_active_object(C);
2030         const bool with_targets = RNA_boolean_get(op->ptr, "with_targets");
2031         
2032         /* add the constraint - all necessary checks should have been done by the invoke() callback already... */
2033         return constraint_add_exec(C, op, ob, get_active_constraints(ob), CONSTRAINT_TYPE_KINEMATIC, with_targets);
2034 }
2035
2036 void POSE_OT_ik_add(wmOperatorType *ot)
2037 {
2038         /* identifiers */
2039         ot->name = "Add IK to Bone";
2040         ot->description = "Add IK Constraint to the active Bone";
2041         ot->idname = "POSE_OT_ik_add";
2042         
2043         /* api callbacks */
2044         ot->invoke = pose_ik_add_invoke;
2045         ot->exec = pose_ik_add_exec;
2046         ot->poll = ED_operator_posemode_exclusive;
2047         
2048         /* flags */
2049         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
2050         
2051         /* properties */
2052         RNA_def_boolean(ot->srna, "with_targets", 1, "With Targets", "Assign IK Constraint with targets derived from the select bones/objects");
2053 }
2054
2055 /* ------------------ */
2056
2057 /* remove IK constraints from selected bones */
2058 static int pose_ik_clear_exec(bContext *C, wmOperator *UNUSED(op))
2059 {
2060         Object *ob = BKE_object_pose_armature_get(CTX_data_active_object(C));
2061         
2062         /* only remove IK Constraints */
2063         CTX_DATA_BEGIN (C, bPoseChannel *, pchan, selected_pose_bones)
2064         {
2065                 bConstraint *con, *next;
2066                 
2067                 /* TODO: should we be checking if these contraints were local before we try and remove them? */
2068                 for (con = pchan->constraints.first; con; con = next) {
2069                         next = con->next;
2070                         if (con->type == CONSTRAINT_TYPE_KINEMATIC) {
2071                                 BKE_constraint_remove(&pchan->constraints, con);
2072                         }
2073                 }
2074                 pchan->constflag &= ~(PCHAN_HAS_IK | PCHAN_HAS_TARGET);
2075         }
2076         CTX_DATA_END;
2077         
2078         /* refresh depsgraph */
2079         DEG_id_tag_update(&ob->id, OB_RECALC_DATA);
2080
2081         /* note, notifier might evolve */
2082         WM_event_add_notifier(C, NC_OBJECT | ND_CONSTRAINT | NA_REMOVED, ob);
2083         
2084         return OPERATOR_FINISHED;
2085 }
2086
2087 void POSE_OT_ik_clear(wmOperatorType *ot)
2088 {
2089         /* identifiers */
2090         ot->name = "Remove IK";
2091         ot->description = "Remove all IK Constraints from selected bones";
2092         ot->idname = "POSE_OT_ik_clear";
2093         
2094         /* api callbacks */
2095         ot->exec = pose_ik_clear_exec;
2096         ot->poll = ED_operator_posemode_exclusive;
2097         
2098         /* flags */
2099         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
2100 }
2101