Two in one:
[blender.git] / source / blender / src / editconstraint.c
index 142587776c51a950adfd440095ae2bb415c583b8..bb921906d55bfe28541746a5ad9fb34eb5a34eb5 100644 (file)
 #include "MEM_guardedalloc.h"
 
 #include "BLI_blenlib.h"
+#include "BLI_arithb.h"
 
 #include "DNA_action_types.h"
 #include "DNA_armature_types.h"
+#include "DNA_constraint_types.h"
+#include "DNA_curve_types.h"
 #include "DNA_object_types.h"
 #include "DNA_scene_types.h"
 #include "DNA_screen_types.h"
-#include "DNA_constraint_types.h"
-#include "DNA_curve_types.h"
+#include "DNA_view3d_types.h"
 
-#include "BKE_utildefines.h"
 #include "BKE_action.h"
 #include "BKE_armature.h"
-#include "BKE_object.h"
-#include "BKE_global.h"
 #include "BKE_constraint.h"
+#include "BKE_depsgraph.h"
+#include "BKE_global.h"
 #include "BKE_ipo.h"
+#include "BKE_object.h"
+#include "BKE_utildefines.h"
 
 #include "BIF_editaction.h"
 #include "BIF_editarmature.h"
 #include "BIF_poseobject.h"
 #include "BIF_interface.h"
 #include "BIF_screen.h"
+#include "BIF_space.h"
 #include "BIF_toolbox.h"
 
 #include "blendef.h"
 #include "nla.h"
+#include "mydevice.h"
 
 
 ListBase *get_active_constraint_channels (Object *ob, int forcevalid)
@@ -85,11 +90,11 @@ ListBase *get_active_constraint_channels (Object *ob, int forcevalid)
                                if (!forcevalid)
                                        return NULL;
                                
-                               ob->action=add_empty_action();
+                               ob->action=add_empty_action(ID_PO);
                        }
                        
                        /* Make sure we have an actionchannel */
-                       achan = get_named_actionchannel(ob->action, pchan->name);
+                       achan = get_action_channel(ob->action, pchan->name);
                        if (!achan){
                                if (!forcevalid)
                                        return NULL;
@@ -109,29 +114,37 @@ ListBase *get_active_constraint_channels (Object *ob, int forcevalid)
                else return NULL;
        }
        /* else we return object constraints */
-       return &ob->constraintChannels;
+       else {
+               if(ob->ipoflag & OB_ACTION_OB) {
+                       bActionChannel *achan = get_action_channel(ob->action, "Object");
+                       if(achan)
+                               return &achan->constraintChannels;
+                       else 
+                               return NULL;
+               }
+               
+               return &ob->constraintChannels;
+       }
 }
 
 
 /* if object in posemode, active bone constraints, else object constraints */
 ListBase *get_active_constraints(Object *ob)
 {
-       ListBase *list;
-
        if (!ob)
                return NULL;
 
-       list = &ob->constraints;
-
        if (ob->flag & OB_POSEMODE) {
                bPoseChannel *pchan;
 
                pchan = get_active_posechannel(ob);
                if (pchan)
-                       list = &pchan->constraints;
+                       return &pchan->constraints;
        }
+       else 
+               return &ob->constraints;
 
-       return list;
+       return NULL;
 }
 
 /* single constraint */
@@ -164,7 +177,7 @@ bConstraintChannel *get_active_constraint_channel(Object *ob)
                                        if(con->flag & CONSTRAINT_ACTIVE)
                                                break;
                                if(con) {
-                                       bActionChannel *achan = get_named_actionchannel(ob->action, pchan->name);
+                                       bActionChannel *achan = get_action_channel(ob->action, pchan->name);
                                        if(achan) {
                                                for(chan= achan->constraintChannels.first; chan; chan= chan->next)
                                                        if(!strcmp(chan->name, con->name))
@@ -180,10 +193,14 @@ bConstraintChannel *get_active_constraint_channel(Object *ob)
                        if(con->flag & CONSTRAINT_ACTIVE)
                                break;
                if(con) {
-                       for(chan= ob->constraintChannels.first; chan; chan= chan->next)
-                               if(!strcmp(chan->name, con->name))
-                                       break;
-                       return chan;
+                       ListBase *lb= get_active_constraint_channels(ob, 0);
+
+                       if(lb) {
+                               for(chan= lb->first; chan; chan= chan->next)
+                                       if(!strcmp(chan->name, con->name))
+                                               break;
+                               return chan;
+                       }
                }
        }
        
@@ -191,7 +208,6 @@ bConstraintChannel *get_active_constraint_channel(Object *ob)
 }
 
 
-
 bConstraint *add_new_constraint(short type)
 {
        bConstraint *con;
@@ -250,6 +266,12 @@ char *get_con_subtarget_name(bConstraint *con, Object *target)
                        if (data->tar==target) return data->subtarget;
                }
                break;
+               case CONSTRAINT_TYPE_SIZELIKE:
+               {
+                       bSizeLikeConstraint *data = con->data;
+                       if (data->tar==target) return data->subtarget;
+               }
+               break;
                case CONSTRAINT_TYPE_KINEMATIC:
                {
                        bKinematicConstraint *data = con->data;
@@ -262,6 +284,12 @@ char *get_con_subtarget_name(bConstraint *con, Object *target)
                        if (data->tar==target) return data->subtarget;
                }
                break;
+               case CONSTRAINT_TYPE_MINMAX:
+               {
+                       bMinMaxConstraint *data = con->data;
+                       if (data->tar==target) return data->subtarget;
+               }
+               break;
                case CONSTRAINT_TYPE_LOCKTRACK:
                {
                        bLockTrackConstraint *data = con->data;
@@ -370,6 +398,24 @@ static void test_constraints (Object *owner, const char* substring)
                                                break;
                                        }
                                        
+                                       if ( (data->tar == owner) &&
+                                                (!get_named_bone(get_armature(owner), 
+                                                                                 data->subtarget))) {
+                                               curcon->flag |= CONSTRAINT_DISABLE;
+                                               break;
+                                       }
+                               }
+                                       break;
+                               case CONSTRAINT_TYPE_MINMAX:
+                               {
+                                       bMinMaxConstraint *data = curcon->data;
+                                       
+                                       if (!exist_object(data->tar)){
+                                               data->tar = NULL;
+                                               curcon->flag |= CONSTRAINT_DISABLE;
+                                               break;
+                                       }
+                                       
                                        if ( (data->tar == owner) &&
                                                 (!get_named_bone(get_armature(owner), 
                                                                                  data->subtarget))) {
@@ -388,6 +434,24 @@ static void test_constraints (Object *owner, const char* substring)
                                                break;
                                        }
                                        
+                                       if ( (data->tar == owner) &&
+                                                (!get_named_bone(get_armature(owner), 
+                                                                                 data->subtarget))) {
+                                               curcon->flag |= CONSTRAINT_DISABLE;
+                                               break;
+                                       }
+                               }
+                                       break;
+                               case CONSTRAINT_TYPE_SIZELIKE:
+                               {
+                                       bSizeLikeConstraint *data = curcon->data;
+                               
+                                       if (!exist_object(data->tar)){
+                                               data->tar = NULL;
+                                               curcon->flag |= CONSTRAINT_DISABLE;
+                                               break;
+                                       }
+                                       
                                        if ( (data->tar == owner) &&
                                                 (!get_named_bone(get_armature(owner), 
                                                                                  data->subtarget))) {
@@ -536,4 +600,290 @@ void object_test_constraints (Object *owner)
 
 }
 
+/* context: active object in posemode, active channel, optional selected channel */
+void add_constraint(int only_IK)
+{
+       Object *ob= OBACT, *obsel=NULL;
+       bPoseChannel *pchanact=NULL, *pchansel=NULL;
+       bConstraint *con=NULL;
+       Base *base;
+       short nr;
+       
+       /* paranoia checks */
+       if(ob==NULL || ob==G.obedit) return;
+
+       if(ob->pose && (ob->flag & OB_POSEMODE)) {
+               bArmature *arm= ob->data;
+               
+               /* find active channel */
+               pchanact= get_active_posechannel(ob);
+               if(pchanact==NULL) return;
+       
+               /* check protection */
+               if(ob->proxy && (pchanact->bone->layer & arm->layer_protected)) {
+                       error("Bone is Proxy protected");
+                       return;
+               }
+               
+               /* find selected bone */
+               for(pchansel= ob->pose->chanbase.first; pchansel; pchansel= pchansel->next) {
+                       if(pchansel!=pchanact)
+                               if(pchansel->bone->flag & BONE_SELECTED) 
+                                       if(pchansel->bone->layer & arm->layer)
+                                               break;
+               }
+       }
+       
+       /* find selected object */
+       for(base= FIRSTBASE; base; base= base->next)
+               if( TESTBASE(base) && base->object!=ob ) 
+                       obsel= base->object;
+       
+       /* the only_IK caller has checked for posemode! */
+       if(only_IK) {
+               for(con= pchanact->constraints.first; con; con= con->next) {
+                       if(con->type==CONSTRAINT_TYPE_KINEMATIC) break;
+               }
+               if(con) {
+                       error("Pose Channel already has IK");
+                       return;
+               }
+               
+               if(pchansel)
+                       nr= pupmenu("Add IK Constraint%t|To Active Bone%x10");
+               else if(obsel)
+                       nr= pupmenu("Add IK Constraint%t|To Active Object%x10");
+               else 
+                       nr= pupmenu("Add IK Constraint%t|To New Empty Object%x10|Without Target%x11");
+       }
+       else {
+               if(pchanact) {
+                       if(pchansel)
+                               nr= pupmenu("Add Constraint to Active Bone%t|Copy Location%x1|Copy Rotation%x2|Copy Scale%x8|Limit Location%x13|Limit Rotation%x14|Limit Scale%x15|Track To%x3|Floor%x4|Locked Track%x5|Stretch To%x7|Action%x16");
+                       else if(obsel && obsel->type==OB_CURVE)
+                               nr= pupmenu("Add Constraint to Active Object%t|Copy Location%x1|Copy Rotation%x2|Copy Scale%x8|Limit Location%x13|Limit Rotation%x14|Limit Scale%x15|Track To%x3|Floor%x4|Locked Track%x5|Follow Path%x6|Stretch To%x7|Action%x16");
+                       else if(obsel)
+                               nr= pupmenu("Add Constraint to Active Object%t|Copy Location%x1|Copy Rotation%x2|Copy Scale%x8|Limit Location%x13|Limit Rotation%x14|Limit Scale%x15|Track To%x3|Floor%x4|Locked Track%x5|Stretch To%x7|Action%x16");
+                       else
+                               nr= pupmenu("Add Constraint to New Empty Object%t|Copy Location%x1|Copy Rotation%x2|Copy Scale%x8|Limit Location%x13|Limit Rotation%x14|Limit Scale%x15|Track To%x3|Floor%x4|Locked Track%x5|Stretch To%x7");
+               }
+               else {
+                       if(obsel && obsel->type==OB_CURVE)
+                               nr= pupmenu("Add Constraint to Active Object%t|Copy Location%x1|Copy Rotation%x2|Copy Scale%x8|Limit Location%x13|Limit Rotation%x14|Limit Scale%x15|Track To%x3|Floor%x4|Locked Track%x5|Follow Path%x6");
+                       else if(obsel)
+                               nr= pupmenu("Add Constraint to Active Object%t|Copy Location%x1|Copy Rotation%x2|Copy Scale%x8|Limit Location%x13|Limit Rotation%x14|Limit Scale%x15|Track To%x3|Floor%x4|Locked Track%x5");
+                       else
+                               nr= pupmenu("Add Constraint to New Empty Object%t|Copy Location%x1|Copy Rotation%x2|Copy Scale%x8|Limit Location%x13|Limit Rotation%x14|Limit Scale%x15|Track To%x3|Floor%x4|Locked Track%x5");
+               }
+       }
+       
+       if(nr<1) return;
+       
+       /* handle IK separate */
+       if(nr==10 || nr==11) {
+               
+               /* prevent weird chains... */
+               if(pchansel) {
+                       bPoseChannel *pchan= pchanact;
+                       while(pchan) {
+                               if(pchan==pchansel) break;
+                               pchan= pchan->parent;
+                       }
+                       if(pchan) {
+                               error("IK root cannot be linked to IK tip");
+                               return;
+                       }
+                       pchan= pchansel;
+                       while(pchan) {
+                               if(pchan==pchanact) break;
+                               pchan= pchan->parent;
+                       }
+                       if(pchan) {
+                               error("IK tip cannot be linked to IK root");
+                               return;
+                       }               
+               }
+               
+               con = add_new_constraint(CONSTRAINT_TYPE_KINEMATIC);
+               BLI_addtail(&pchanact->constraints, con);
+               unique_constraint_name(con, &pchanact->constraints);
+               pchanact->constflag |= PCHAN_HAS_IK;    // for draw, but also for detecting while pose solving
+               if(nr==11) pchanact->constflag |= PCHAN_HAS_TARGET;
+       }
+       else {
+               
+               if(nr==1) con = add_new_constraint(CONSTRAINT_TYPE_LOCLIKE);
+               else if(nr==2) con = add_new_constraint(CONSTRAINT_TYPE_ROTLIKE);
+               else if(nr==3) con = add_new_constraint(CONSTRAINT_TYPE_TRACKTO);
+               else if(nr==4) con = add_new_constraint(CONSTRAINT_TYPE_MINMAX);
+               else if(nr==5) con = add_new_constraint(CONSTRAINT_TYPE_LOCKTRACK);
+               else if(nr==6) {
+                       Curve *cu= obsel->data;
+                       cu->flag |= CU_PATH;
+                       con = add_new_constraint(CONSTRAINT_TYPE_FOLLOWPATH);
+               }
+               else if(nr==7) con = add_new_constraint(CONSTRAINT_TYPE_STRETCHTO);
+               else if(nr==8) con = add_new_constraint(CONSTRAINT_TYPE_SIZELIKE);
+               else if(nr==13) con = add_new_constraint(CONSTRAINT_TYPE_LOCLIMIT);
+               else if(nr==14) con = add_new_constraint(CONSTRAINT_TYPE_ROTLIMIT);
+               else if(nr==15) con = add_new_constraint(CONSTRAINT_TYPE_SIZELIMIT);
+               else if(nr==16) con = add_new_constraint(CONSTRAINT_TYPE_ACTION);
+               
+               if(con==NULL) return;   /* paranoia */
+               
+               if(pchanact) {
+                       BLI_addtail(&pchanact->constraints, con);
+                       unique_constraint_name(con, &pchanact->constraints);
+                       pchanact->constflag |= PCHAN_HAS_CONST; /* for draw */
+               }
+               else {
+                       BLI_addtail(&ob->constraints, con);
+                       unique_constraint_name(con, &ob->constraints);
+               }
+       }
+       
+       /* set the target */
+       if(pchansel) {
+               set_constraint_target(con, ob, pchansel->name);
+       }
+       else if(obsel) {
+               set_constraint_target(con, obsel, NULL);
+       }
+       else if(ELEM4(nr, 11, 13, 14, 15)==0) { /* add new empty as target */
+               Base *base= BASACT, *newbase;
+               Object *obt;
+               
+               obt= add_object(OB_EMPTY);
+               /* set layers OK */
+               newbase= BASACT;
+               newbase->lay= base->lay;
+               obt->lay= newbase->lay;
+               
+               /* transform cent to global coords for loc */
+               if(pchanact) {
+                       if(only_IK)
+                               VecMat4MulVecfl(obt->loc, ob->obmat, pchanact->pose_tail);
+                       else
+                               VecMat4MulVecfl(obt->loc, ob->obmat, pchanact->pose_head);
+               }
+               else
+                       VECCOPY(obt->loc, ob->obmat[3]);
+               
+               set_constraint_target(con, obt, NULL);
+               
+               /* restore, add_object sets active */
+               BASACT= base;
+               base->flag |= SELECT;
+       }
+
+
+       /* active flag */
+       con->flag |= CONSTRAINT_ACTIVE;
+       for(con= con->prev; con; con= con->prev)
+               con->flag &= ~CONSTRAINT_ACTIVE;
+
+       DAG_scene_sort(G.scene);                // sort order of objects
+       
+       if(pchanact) {
+               ob->pose->flag |= POSE_RECALC;  // sort pose channels
+               DAG_object_flush_update(G.scene, ob, OB_RECALC_DATA);   // and all its relations
+       }
+       else
+               DAG_object_flush_update(G.scene, ob, OB_RECALC_OB);     // and all its relations
+
+       allqueue (REDRAWVIEW3D, 0);
+       allqueue (REDRAWBUTSOBJECT, 0);
+       allqueue (REDRAWOOPS, 0);
+       
+       if(only_IK)
+               BIF_undo_push("Add IK Constraint");
+       else
+               BIF_undo_push("Add Constraint");
+
+}
+
+void ob_clear_constraints(void)
+{
+       Object *ob= OBACT;
+       
+       /* paranoia checks */
+       if(!ob) return;
+       if(ob==G.obedit || (ob->flag & OB_POSEMODE)) return;
+       
+       if(okee("Clear Constraints")==0) return;
+       
+       free_constraints(&ob->constraints);
+       
+       DAG_object_flush_update(G.scene, ob, OB_RECALC_OB);
+       
+       allqueue (REDRAWVIEW3D, 0);
+       allqueue (REDRAWBUTSOBJECT, 0);
+       allqueue (REDRAWOOPS, 0);
+       
+       BIF_undo_push("Clear Constraint(s)");
+       
+}
+
+/* con already has the new name */
+void rename_constraint(Object *ob, bConstraint *con, char *oldname)
+{
+       bConstraint *tcon;
+       bConstraintChannel *conchan;
+       ListBase *conlist= NULL;
+       int from_object= 0;
+       char *channame="";
+       
+       /* get context by searching for con (primitive...) */
+       for(tcon= ob->constraints.first; tcon; tcon= tcon->next)
+               if(tcon==con)
+                       break;
+       
+       if(tcon) {
+               conlist= &ob->constraints;
+               channame= "Object";
+               from_object= 1;
+       }
+       else if(ob->pose) {
+               bPoseChannel *pchan;
+               
+               for(pchan= ob->pose->chanbase.first; pchan; pchan= pchan->next) {
+                       for(tcon= pchan->constraints.first; tcon; tcon= tcon->next) {
+                               if(tcon==con)
+                                       break;
+                       }
+                       if(tcon)
+                               break;
+               }
+               if(tcon) {
+                       conlist= &pchan->constraints;
+                       channame= pchan->name;
+               }
+       }
+       
+       if(conlist==NULL) {
+               printf("rename constraint failed\n");   /* should not happen in UI */
+               return;
+       }
+       
+       /* first make sure it's a unique name within context */
+       unique_constraint_name (con, conlist);
+
+       /* own channels */
+       if(from_object) {
+               for(conchan= ob->constraintChannels.first; conchan; conchan= conchan->next) {
+                       if( strcmp(oldname, conchan->name)==0 )
+                               BLI_strncpy(conchan->name, con->name, sizeof(conchan->name));
+               }
+       }
+       /* own action */
+       if(ob->action) {
+               bActionChannel *achan= get_action_channel(ob->action, channame);
+               if(achan) {
+                       conchan= get_constraint_channel(&achan->constraintChannels, oldname);
+                       if(conchan)
+                               BLI_strncpy(conchan->name, con->name, sizeof(conchan->name));
+               }
+       }
+       
+}