GSOC 2013 paint
[blender-staging.git] / source / blender / editors / transform / transform_constraints.c
index 256ae90cbdc5336826208053158d575a2c514351..d8f17315c0103c7f6409435740b215bbb861972f 100644 (file)
 #include <string.h>
 #include <math.h>
 
-#ifndef WIN32
-#  include <unistd.h>
-#else
-#  include <io.h>
-#endif
-
 #include "DNA_object_types.h"
 #include "DNA_scene_types.h"
 #include "DNA_screen_types.h"
 #include "BIF_gl.h"
 #include "BIF_glutil.h"
 
-#include "BKE_context.h"
-
 #include "BLI_math.h"
 #include "BLI_utildefines.h"
 #include "BLI_string.h"
 
+#include "BKE_context.h"
+
 #include "ED_image.h"
 #include "ED_view3d.h"
 
@@ -135,9 +129,9 @@ static void postConstraintChecks(TransInfo *t, float vec[3], float pvec[3])
 
        mul_m3_v3(t->con.imtx, vec);
 
-       snapGrid(t, vec);
+       snapGridIncrement(t, vec);
 
-       if (t->num.flag & T_NULL_ONE) {
+       if (t->flag & T_NULL_ONE) {
                if (!(t->con.mode & CON_AXIS0))
                        vec[0] = 1.0f;
 
@@ -148,8 +142,7 @@ static void postConstraintChecks(TransInfo *t, float vec[3], float pvec[3])
                        vec[2] = 1.0f;
        }
 
-       if (hasNumInput(&t->num)) {
-               applyNumInput(&t->num, vec);
+       if (applyNumInput(&t->num, vec)) {
                constraintNumInput(t, vec);
                removeAspectRatio(t, vec);
        }
@@ -393,7 +386,11 @@ static void applyObjectConstraintVec(TransInfo *t, TransData *td, const float in
                        if (t->con.mode & CON_AXIS2) {
                                out[2] = in[i++];
                        }
+
                        mul_m3_v3(td->axismtx, out);
+                       if (t->flag & T_EDIT) {
+                               mul_m3_v3(t->obedit_mat, out);
+                       }
                }
        }
 }
@@ -445,6 +442,9 @@ static void applyObjectConstraintSize(TransInfo *t, TransData *td, float smat[3]
                }
 
                mul_m3_m3m3(tmat, smat, imat);
+               if (t->flag & T_EDIT) {
+                       mul_m3_m3m3(smat, t->obedit_mat, smat);
+               }
                mul_m3_m3m3(smat, td->axismtx, tmat);
        }
 }
@@ -509,24 +509,34 @@ static void applyObjectConstraintRot(TransInfo *t, TransData *td, float vec[3],
 {
        if (t->con.mode & CON_APPLY) {
                int mode = t->con.mode & (CON_AXIS0 | CON_AXIS1 | CON_AXIS2);
+               float tmp_axismtx[3][3];
+               float (*axismtx)[3];
 
                /* on setup call, use first object */
                if (td == NULL) {
                        td = t->data;
                }
 
+               if (t->flag & T_EDIT) {
+                       mul_m3_m3m3(tmp_axismtx, t->obedit_mat, td->axismtx);
+                       axismtx = tmp_axismtx;
+               }
+               else {
+                       axismtx = td->axismtx;
+               }
+
                switch (mode) {
                        case CON_AXIS0:
                        case (CON_AXIS1 | CON_AXIS2):
-                               copy_v3_v3(vec, td->axismtx[0]);
+                               copy_v3_v3(vec, axismtx[0]);
                                break;
                        case CON_AXIS1:
                        case (CON_AXIS0 | CON_AXIS2):
-                               copy_v3_v3(vec, td->axismtx[1]);
+                               copy_v3_v3(vec, axismtx[1]);
                                break;
                        case CON_AXIS2:
                        case (CON_AXIS0 | CON_AXIS1):
-                               copy_v3_v3(vec, td->axismtx[2]);
+                               copy_v3_v3(vec, axismtx[2]);
                                break;
                }
                if (angle && (mode & CON_NOFLIP) == 0 && hasNumInput(&t->num) == 0) {
@@ -552,35 +562,47 @@ void setConstraint(TransInfo *t, float space[3][3], int mode, const char text[])
        t->con.applyVec = applyAxisConstraintVec;
        t->con.applySize = applyAxisConstraintSize;
        t->con.applyRot = applyAxisConstraintRot;
-       t->redraw = 1;
+       t->redraw = TREDRAW_HARD;
+}
+
+/* applies individual td->axismtx constraints */
+void setAxisMatrixConstraint(TransInfo *t, int mode, const char text[])
+{
+       if (t->total == 1) {
+               float axismtx[3][3];
+               if (t->flag & T_EDIT) {
+                       mul_m3_m3m3(axismtx, t->obedit_mat, t->data->axismtx);
+               }
+               else {
+                       copy_m3_m3(axismtx, t->data->axismtx);
+               }
+
+               setConstraint(t, axismtx, mode, text);
+       }
+       else {
+               BLI_strncpy(t->con.text + 1, text, sizeof(t->con.text) - 1);
+               copy_m3_m3(t->con.mtx, t->data->axismtx);
+               t->con.mode = mode;
+               getConstraintMatrix(t);
+
+               startConstraint(t);
+
+               t->con.drawExtra = drawObjectConstraint;
+               t->con.applyVec = applyObjectConstraintVec;
+               t->con.applySize = applyObjectConstraintSize;
+               t->con.applyRot = applyObjectConstraintRot;
+               t->redraw = TREDRAW_HARD;
+       }
 }
 
 void setLocalConstraint(TransInfo *t, int mode, const char text[])
 {
+       /* edit-mode now allows local transforms too */
        if (t->flag & T_EDIT) {
-               float obmat[3][3];
-               copy_m3_m4(obmat, t->scene->obedit->obmat);
-               normalize_m3(obmat);
-               setConstraint(t, obmat, mode, text);
+               setConstraint(t, t->obedit_mat, mode, text);
        }
        else {
-               if (t->total == 1) {
-                       setConstraint(t, t->data->axismtx, mode, text);
-               }
-               else {
-                       BLI_strncpy(t->con.text + 1, text, sizeof(t->con.text) - 1);
-                       copy_m3_m3(t->con.mtx, t->data->axismtx);
-                       t->con.mode = mode;
-                       getConstraintMatrix(t);
-
-                       startConstraint(t);
-
-                       t->con.drawExtra = drawObjectConstraint;
-                       t->con.applyVec = applyObjectConstraintVec;
-                       t->con.applySize = applyObjectConstraintSize;
-                       t->con.applyRot = applyObjectConstraintRot;
-                       t->redraw = 1;
-               }
+               setAxisMatrixConstraint(t, mode, text);
        }
 }
 
@@ -597,18 +619,24 @@ void setUserConstraint(TransInfo *t, short orientation, int mode, const char fte
        switch (orientation) {
                case V3D_MANIP_GLOBAL:
                {
-                       float mtx[3][3] = MAT3_UNITY;
+                       float mtx[3][3];
                        BLI_snprintf(text, sizeof(text), ftext, IFACE_("global"));
+                       unit_m3(mtx);
                        setConstraint(t, mtx, mode, text);
+                       break;
                }
-               break;
                case V3D_MANIP_LOCAL:
                        BLI_snprintf(text, sizeof(text), ftext, IFACE_("local"));
                        setLocalConstraint(t, mode, text);
                        break;
                case V3D_MANIP_NORMAL:
                        BLI_snprintf(text, sizeof(text), ftext, IFACE_("normal"));
-                       setConstraint(t, t->spacemtx, mode, text);
+                       if (checkUseAxisMatrix(t)) {
+                               setAxisMatrixConstraint(t, mode, text);
+                       }
+                       else {
+                               setConstraint(t, t->spacemtx, mode, text);
+                       }
                        break;
                case V3D_MANIP_VIEW:
                        BLI_snprintf(text, sizeof(text), ftext, IFACE_("view"));
@@ -635,7 +663,7 @@ void drawConstraint(TransInfo *t)
 {
        TransCon *tc = &(t->con);
 
-       if (!ELEM3(t->spacetype, SPACE_VIEW3D, SPACE_IMAGE, SPACE_NODE))
+       if (!ELEM(t->spacetype, SPACE_VIEW3D, SPACE_IMAGE, SPACE_NODE))
                return;
        if (!(tc->mode & CON_APPLY))
                return;
@@ -701,6 +729,7 @@ void drawPropCircle(const struct bContext *C, TransInfo *t)
                RegionView3D *rv3d = CTX_wm_region_view3d(C);
                float tmat[4][4], imat[4][4];
                float center[3];
+               int depth_test_enabled;
 
                UI_ThemeColor(TH_GRID);
 
@@ -727,53 +756,83 @@ void drawPropCircle(const struct bContext *C, TransInfo *t)
                                /* untested - mask aspect is TODO */
                                ED_space_image_get_aspect(t->sa->spacedata.first, &aspx, &aspy);
                        }
+                       else if (t->options & CTX_PAINT_CURVE) {
+                               aspx = aspy = 1.0;
+                       }
                        else {
                                ED_space_image_get_uv_aspect(t->sa->spacedata.first, &aspx, &aspy);
                        }
                        glScalef(1.0f / aspx, 1.0f / aspy, 1.0);
                }
 
+               depth_test_enabled = glIsEnabled(GL_DEPTH_TEST);
+               if (depth_test_enabled)
+                       glDisable(GL_DEPTH_TEST);
+
                set_inverted_drawing(1);
                drawcircball(GL_LINE_LOOP, center, t->prop_size, imat);
                set_inverted_drawing(0);
 
+               if (depth_test_enabled)
+                       glEnable(GL_DEPTH_TEST);
+
                glPopMatrix();
        }
 }
 
 static void drawObjectConstraint(TransInfo *t)
 {
-       int i;
-       TransData *td = t->data;
-
        /* Draw the first one lighter because that's the one who controls the others.
         * Meaning the transformation is projected on that one and just copied on the others
         * constraint space.
         * In a nutshell, the object with light axis is controlled by the user and the others follow.
         * Without drawing the first light, users have little clue what they are doing.
         */
-       if (t->con.mode & CON_AXIS0) {
-               drawLine(t, td->ob->obmat[3], td->axismtx[0], 'X', DRAWLIGHT);
-       }
-       if (t->con.mode & CON_AXIS1) {
-               drawLine(t, td->ob->obmat[3], td->axismtx[1], 'Y', DRAWLIGHT);
-       }
-       if (t->con.mode & CON_AXIS2) {
-               drawLine(t, td->ob->obmat[3], td->axismtx[2], 'Z', DRAWLIGHT);
-       }
+       short options = DRAWLIGHT;
+       TransData *td = t->data;
+       int i;
+       float tmp_axismtx[3][3];
+
+       for (i = 0; i < t->total; i++, td++) {
+               float co[3];
+               float (*axismtx)[3];
+
+               if (t->flag & T_PROP_EDIT) {
+                       /* we're sorted, so skip the rest */
+                       if (td->factor == 0.0f) {
+                               break;
+                       }
+               }
 
-       td++;
+               if (t->flag & T_OBJECT) {
+                       copy_v3_v3(co, td->ob->obmat[3]);
+                       axismtx = td->axismtx;
+               }
+               else if (t->flag & T_EDIT) {
+                       mul_v3_m4v3(co, t->obedit->obmat, td->center);
+
+                       mul_m3_m3m3(tmp_axismtx, t->obedit_mat, td->axismtx);
+                       axismtx = tmp_axismtx;
+               }
+               else if (t->flag & T_POSE) {
+                       mul_v3_m4v3(co, t->poseobj->obmat, td->center);
+                       axismtx = td->axismtx;
+               }
+               else {
+                       copy_v3_v3(co, td->center);
+                       axismtx = td->axismtx;
+               }
 
-       for (i = 1; i < t->total; i++, td++) {
                if (t->con.mode & CON_AXIS0) {
-                       drawLine(t, td->ob->obmat[3], td->axismtx[0], 'X', 0);
+                       drawLine(t, co, axismtx[0], 'X', options);
                }
                if (t->con.mode & CON_AXIS1) {
-                       drawLine(t, td->ob->obmat[3], td->axismtx[1], 'Y', 0);
+                       drawLine(t, co, axismtx[1], 'Y', options);
                }
                if (t->con.mode & CON_AXIS2) {
-                       drawLine(t, td->ob->obmat[3], td->axismtx[2], 'Z', 0);
+                       drawLine(t, co, axismtx[2], 'Z', options);
                }
+               options &= ~DRAWLIGHT;
        }
 }
 
@@ -857,28 +916,28 @@ void postSelectConstraint(TransInfo *t)
        setNearestAxis(t);
 
        startConstraint(t);
-       t->redraw = 1;
+       t->redraw = TREDRAW_HARD;
 }
 
 static void setNearestAxis2d(TransInfo *t)
 {
        /* no correction needed... just use whichever one is lower */
-       if (abs(t->mval[0] - t->con.imval[0]) < abs(t->mval[1] - t->con.imval[1]) ) {
+       if (abs(t->mval[0] - t->con.imval[0]) < abs(t->mval[1] - t->con.imval[1])) {
                t->con.mode |= CON_AXIS1;
-               BLI_snprintf(t->con.text, sizeof(t->con.text), IFACE_(" along Y axis"));
+               BLI_strncpy(t->con.text, IFACE_(" along Y axis"), sizeof(t->con.text));
        }
        else {
                t->con.mode |= CON_AXIS0;
-               BLI_snprintf(t->con.text, sizeof(t->con.text), IFACE_(" along X axis"));
+               BLI_strncpy(t->con.text, IFACE_(" along X axis"), sizeof(t->con.text));
        }
 }
 
 static void setNearestAxis3d(TransInfo *t)
 {
        float zfac;
-       float mvec[3], axis[3], proj[3];
+       float mvec[3], proj[3];
        float len[3];
-       int i, icoord[2];
+       int i;
 
        /* calculate mouse movement */
        mvec[0] = (float)(t->mval[0] - t->con.imval[0]);
@@ -890,30 +949,31 @@ static void setNearestAxis3d(TransInfo *t)
         * and to overflow the short integers.
         * The formula used is a bit stupid, just a simplification of the subtraction
         * of two 2D points 30 pixels apart (that's the last factor in the formula) after
-        * projecting them with window_to_3d_delta and then get the length of that vector.
+        * projecting them with ED_view3d_win_to_delta and then get the length of that vector.
         */
-       zfac = t->persmat[0][3] * t->center[0] + t->persmat[1][3] * t->center[1] + t->persmat[2][3] * t->center[2] + t->persmat[3][3];
+       zfac = mul_project_m4_v3_zfac(t->persmat, t->center);
        zfac = len_v3(t->persinv[0]) * 2.0f / t->ar->winx * zfac * 30.0f;
 
        for (i = 0; i < 3; i++) {
+               float axis[3], axis_2d[2];
+
                copy_v3_v3(axis, t->con.mtx[i]);
 
                mul_v3_fl(axis, zfac);
                /* now we can project to get window coordinate */
                add_v3_v3(axis, t->con.center);
-               projectIntView(t, axis, icoord);
+               projectFloatView(t, axis, axis_2d);
 
-               axis[0] = (float)(icoord[0] - t->center2d[0]);
-               axis[1] = (float)(icoord[1] - t->center2d[1]);
+               sub_v2_v2v2(axis, axis_2d, t->center2d);
                axis[2] = 0.0f;
 
-               if (normalize_v3(axis) != 0.0f) {
+               if (normalize_v3(axis) > 1e-3f) {
                        project_v3_v3v3(proj, mvec, axis);
                        sub_v3_v3v3(axis, mvec, proj);
                        len[i] = normalize_v3(axis);
                }
                else {
-                       len[i] = 10000000000.0f;
+                       len[i] = 1e10f;
                }
        }
 
@@ -992,20 +1052,20 @@ char constraintModeToChar(TransInfo *t)
 }
 
 
-int isLockConstraint(TransInfo *t)
+bool isLockConstraint(TransInfo *t)
 {
        int mode = t->con.mode;
 
        if ((mode & (CON_AXIS0 | CON_AXIS1)) == (CON_AXIS0 | CON_AXIS1))
-               return 1;
+               return true;
 
        if ((mode & (CON_AXIS1 | CON_AXIS2)) == (CON_AXIS1 | CON_AXIS2))
-               return 1;
+               return true;
 
        if ((mode & (CON_AXIS0 | CON_AXIS2)) == (CON_AXIS0 | CON_AXIS2))
-               return 1;
+               return true;
 
-       return 0;
+       return false;
 }
 
 /*