source/blender/editors/transform/transform.c

   1 /**
   2  * $Id$
   3  *
   4  * ***** BEGIN GPL LICENSE BLOCK *****
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License
   8  * as published by the Free Software Foundation; either version 2
   9  * of the License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  19  *
  20  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
  21  * All rights reserved.
  22  *
  23  * The Original Code is: all of this file.
  24  *
  25  * Contributor(s): none yet.
  26  *
  27  * ***** END GPL LICENSE BLOCK *****
  28  */
  29
  30 #include <stdlib.h>
  31 #include <stdio.h>
  32 #include <string.h>
  33 #include <math.h>
  34 #include <float.h>
  35
  36 #ifdef HAVE_CONFIG_H
  37 #include <config.h>
  38 #endif
  39
  40 #ifndef WIN32
  41 #include <unistd.h>
  42 #else
  43 #include <io.h>
  44 #endif
  45
  46 #include "MEM_guardedalloc.h"
  47
  48 #include "DNA_anim_types.h"
  49 #include "DNA_armature_types.h"
  50 #include "DNA_action_types.h"  /* for some special action-editor settings */
  51 #include "DNA_constraint_types.h"
  52 #include "DNA_ipo_types.h"              /* some silly ipo flag  */
  53 #include "DNA_listBase.h"
  54 #include "DNA_meshdata_types.h"
  55 #include "DNA_mesh_types.h"
  56 #include "DNA_object_types.h"
  57 #include "DNA_scene_types.h"            /* PET modes                    */
  58 #include "DNA_screen_types.h"   /* area dimensions              */
  59 #include "DNA_texture_types.h"
  60 #include "DNA_userdef_types.h"
  61 #include "DNA_view3d_types.h"
  62 #include "DNA_space_types.h"
  63 #include "DNA_windowmanager_types.h"
  64
  65 #include "RNA_access.h"
  66
  67 //#include "BIF_editview.h"             /* arrows_move_cursor   */
  68 #include "BIF_gl.h"
  69 #include "BIF_glutil.h"
  70 //#include "BIF_mywindow.h"
  71 //#include "BIF_resources.h"
  72 //#include "BIF_screen.h"
  73 //#include "BIF_space.h"                        /* undo                                 */
  74 //#include "BIF_toets.h"                        /* persptoetsen                 */
  75 //#include "BIF_mywindow.h"             /* warp_pointer                 */
  76 //#include "BIF_toolbox.h"                      /* notice                               */
  77 //#include "BIF_editmesh.h"
  78 //#include "BIF_editsima.h"
  79 //#include "BIF_editparticle.h"
  80
  81 #include "BKE_action.h"
  82 #include "BKE_nla.h"
  83 //#include "BKE_bad_level_calls.h"/* popmenu and error  */
  84 #include "BKE_bmesh.h"
  85 #include "BKE_context.h"
  86 #include "BKE_constraint.h"
  87 #include "BKE_global.h"
  88 #include "BKE_particle.h"
  89 #include "BKE_pointcache.h"
  90 #include "BKE_utildefines.h"
  91 #include "BKE_context.h"
  92 #include "BKE_unit.h"
  93
  94 //#include "BSE_view.h"
  95
  96 #include "ED_image.h"
  97 #include "ED_keyframing.h"
  98 #include "ED_screen.h"
  99 #include "ED_space_api.h"
 100 #include "ED_markers.h"
 101 #include "ED_util.h"
 102 #include "ED_view3d.h"
 103
 104 #include "UI_view2d.h"
 105 #include "WM_types.h"
 106 #include "WM_api.h"
 107
 108 #include "BLI_arithb.h"
 109 #include "BLI_blenlib.h"
 110 #include "BLI_editVert.h"
 111
 112 #include "PIL_time.h"                   /* sleep                                */
 113
 114 #include "UI_resources.h"
 115
 116 //#include "blendef.h"
 117 //
 118 //#include "mydevice.h"
 119
 120 #include "transform.h"
 121
 122 /* ************************** SPACE DEPENDANT CODE **************************** */
 123
 124 void setTransformViewMatrices(TransInfo *t)
 125 {
 126         if(t->spacetype==SPACE_VIEW3D && t->ar->regiontype == RGN_TYPE_WINDOW) {
 127                 RegionView3D *rv3d = t->ar->regiondata;
 128
 129                 Mat4CpyMat4(t->viewmat, rv3d->viewmat);
 130                 Mat4CpyMat4(t->viewinv, rv3d->viewinv);
 131                 Mat4CpyMat4(t->persmat, rv3d->persmat);
 132                 Mat4CpyMat4(t->persinv, rv3d->persinv);
 133                 t->persp = rv3d->persp;
 134         }
 135         else {
 136                 Mat4One(t->viewmat);
 137                 Mat4One(t->viewinv);
 138                 Mat4One(t->persmat);
 139                 Mat4One(t->persinv);
 140                 t->persp = V3D_ORTHO;
 141         }
 142
 143         calculateCenter2D(t);
 144 }
 145
 146 void convertViewVec(TransInfo *t, float *vec, short dx, short dy)
 147 {
 148         if (t->spacetype==SPACE_VIEW3D) {
 149                 if (t->ar->regiontype == RGN_TYPE_WINDOW)
 150                 {
 151                         window_to_3d_delta(t->ar, vec, dx, dy);
 152                 }
 153         }
 154         else if(t->spacetype==SPACE_IMAGE) {
 155                 View2D *v2d = t->view;
 156                 float divx, divy, aspx, aspy;
 157
 158                 ED_space_image_uv_aspect(t->sa->spacedata.first, &aspx, &aspy);
 159
 160                 divx= v2d->mask.xmax-v2d->mask.xmin;
 161                 divy= v2d->mask.ymax-v2d->mask.ymin;
 162
 163                 vec[0]= aspx*(v2d->cur.xmax-v2d->cur.xmin)*(dx)/divx;
 164                 vec[1]= aspy*(v2d->cur.ymax-v2d->cur.ymin)*(dy)/divy;
 165                 vec[2]= 0.0f;
 166         }
 167         else if(ELEM(t->spacetype, SPACE_IPO, SPACE_NLA)) {
 168                 View2D *v2d = t->view;
 169                 float divx, divy;
 170
 171                 divx= v2d->mask.xmax-v2d->mask.xmin;
 172                 divy= v2d->mask.ymax-v2d->mask.ymin;
 173
 174                 vec[0]= (v2d->cur.xmax-v2d->cur.xmin)*(dx) / (divx);
 175                 vec[1]= (v2d->cur.ymax-v2d->cur.ymin)*(dy) / (divy);
 176                 vec[2]= 0.0f;
 177         }
 178         else if(t->spacetype==SPACE_NODE) {
 179                 View2D *v2d = &t->ar->v2d;
 180                 float divx, divy;
 181
 182                 divx= v2d->mask.xmax-v2d->mask.xmin;
 183                 divy= v2d->mask.ymax-v2d->mask.ymin;
 184
 185                 vec[0]= (v2d->cur.xmax-v2d->cur.xmin)*(dx)/divx;
 186                 vec[1]= (v2d->cur.ymax-v2d->cur.ymin)*(dy)/divy;
 187                 vec[2]= 0.0f;
 188         }
 189         else if(t->spacetype==SPACE_SEQ) {
 190                 View2D *v2d = &t->ar->v2d;
 191                 float divx, divy;
 192
 193                 divx= v2d->mask.xmax-v2d->mask.xmin;
 194                 divy= v2d->mask.ymax-v2d->mask.ymin;
 195
 196                 vec[0]= (v2d->cur.xmax-v2d->cur.xmin)*(dx)/divx;
 197                 vec[1]= (v2d->cur.ymax-v2d->cur.ymin)*(dy)/divy;
 198                 vec[2]= 0.0f;
 199         }
 200 }
 201
 202 void projectIntView(TransInfo *t, float *vec, int *adr)
 203 {
 204         if (t->spacetype==SPACE_VIEW3D) {
 205                 if(t->ar->regiontype == RGN_TYPE_WINDOW)
 206                         project_int_noclip(t->ar, vec, adr);
 207         }
 208         else if(t->spacetype==SPACE_IMAGE) {
 209                 float aspx, aspy, v[2];
 210
 211                 ED_space_image_uv_aspect(t->sa->spacedata.first, &aspx, &aspy);
 212                 v[0]= vec[0]/aspx;
 213                 v[1]= vec[1]/aspy;
 214
 215                 UI_view2d_to_region_no_clip(t->view, v[0], v[1], adr, adr+1);
 216         }
 217         else if(ELEM(t->spacetype, SPACE_IPO, SPACE_NLA)) {
 218                 int out[2] = {0, 0};
 219
 220                 UI_view2d_view_to_region((View2D *)t->view, vec[0], vec[1], out, out+1);
 221                 adr[0]= out[0];
 222                 adr[1]= out[1];
 223         }
 224         else if(t->spacetype==SPACE_SEQ) { /* XXX not tested yet, but should work */
 225                 int out[2] = {0, 0};
 226
 227                 UI_view2d_view_to_region((View2D *)t->view, vec[0], vec[1], out, out+1);
 228                 adr[0]= out[0];
 229                 adr[1]= out[1];
 230         }
 231 }
 232
 233 void projectFloatView(TransInfo *t, float *vec, float *adr)
 234 {
 235         if (t->spacetype==SPACE_VIEW3D) {
 236                 if(t->ar->regiontype == RGN_TYPE_WINDOW)
 237                         project_float_noclip(t->ar, vec, adr);
 238         }
 239         else if(t->spacetype==SPACE_IMAGE) {
 240                 int a[2];
 241
 242                 projectIntView(t, vec, a);
 243                 adr[0]= a[0];
 244                 adr[1]= a[1];
 245         }
 246         else if(ELEM(t->spacetype, SPACE_IPO, SPACE_NLA)) {
 247                 int a[2];
 248
 249                 projectIntView(t, vec, a);
 250                 adr[0]= a[0];
 251                 adr[1]= a[1];
 252         }
 253 }
 254
 255 void applyAspectRatio(TransInfo *t, float *vec)
 256 {
 257         SpaceImage *sima= t->sa->spacedata.first;
 258
 259         if ((t->spacetype==SPACE_IMAGE) && (t->mode==TFM_TRANSLATION)) {
 260                 float aspx, aspy;
 261
 262                 if((sima->flag & SI_COORDFLOATS)==0) {
 263                         int width, height;
 264                         ED_space_image_size(sima, &width, &height);
 265
 266                         vec[0] *= width;
 267                         vec[1] *= height;
 268                 }
 269
 270                 ED_space_image_uv_aspect(sima, &aspx, &aspy);
 271                 vec[0] /= aspx;
 272                 vec[1] /= aspy;
 273         }
 274 }
 275
 276 void removeAspectRatio(TransInfo *t, float *vec)
 277 {
 278         SpaceImage *sima= t->sa->spacedata.first;
 279
 280         if ((t->spacetype==SPACE_IMAGE) && (t->mode==TFM_TRANSLATION)) {
 281                 float aspx, aspy;
 282
 283                 if((sima->flag & SI_COORDFLOATS)==0) {
 284                         int width, height;
 285                         ED_space_image_size(sima, &width, &height);
 286
 287                         vec[0] /= width;
 288                         vec[1] /= height;
 289                 }
 290
 291                 ED_space_image_uv_aspect(sima, &aspx, &aspy);
 292                 vec[0] *= aspx;
 293                 vec[1] *= aspy;
 294         }
 295 }
 296
 297 static void viewRedrawForce(bContext *C, TransInfo *t)
 298 {
 299         if (t->spacetype == SPACE_VIEW3D)
 300         {
 301                 /* Do we need more refined tags? */
 302                 WM_event_add_notifier(C, NC_OBJECT|ND_TRANSFORM, NULL);
 303
 304                 /* for realtime animation record - send notifiers recognised by animation editors */
 305                 if ((t->animtimer) && IS_AUTOKEY_ON(t->scene))
 306                         WM_event_add_notifier(C, NC_OBJECT|ND_KEYS, NULL);
 307         }
 308         else if (t->spacetype == SPACE_ACTION) {
 309                 //SpaceAction *saction= (SpaceAction *)t->sa->spacedata.first;
 310                 WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_EDIT, NULL);
 311         }
 312         else if (t->spacetype == SPACE_IPO) {
 313                 //SpaceIpo *sipo= (SpaceIpo *)t->sa->spacedata.first;
 314                 WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_EDIT, NULL);
 315         }
 316         else if (t->spacetype == SPACE_NLA) {
 317                 WM_event_add_notifier(C, NC_ANIMATION|ND_NLA_EDIT, NULL);
 318         }
 319         else if(t->spacetype == SPACE_NODE)
 320         {
 321                 //ED_area_tag_redraw(t->sa);
 322                 WM_event_add_notifier(C, NC_SCENE|ND_NODES, NULL);
 323         }
 324         else if(t->spacetype == SPACE_SEQ)
 325         {
 326                 WM_event_add_notifier(C, NC_SCENE|ND_SEQUENCER, NULL);
 327         }
 328         else if (t->spacetype==SPACE_IMAGE) {
 329                 // XXX how to deal with lock?
 330 #if 0
 331                 SpaceImage *sima= (SpaceImage*)t->sa->spacedata.first;
 332                 if(sima->lock) force_draw_plus(SPACE_VIEW3D, 0);
 333                 else force_draw(0);
 334 #endif
 335
 336                 WM_event_add_notifier(C, NC_OBJECT|ND_GEOM_DATA, t->obedit);
 337         }
 338 }
 339
 340 static void viewRedrawPost(TransInfo *t)
 341 {
 342         ED_area_headerprint(t->sa, NULL);
 343
 344 #if 0 // TRANSFORM_FIX_ME
 345         if(t->spacetype==SPACE_VIEW3D) {
 346                 allqueue(REDRAWBUTSOBJECT, 0);
 347                 allqueue(REDRAWVIEW3D, 0);
 348         }
 349         else if(t->spacetype==SPACE_IMAGE) {
 350                 allqueue(REDRAWIMAGE, 0);
 351                 allqueue(REDRAWVIEW3D, 0);
 352         }
 353         else if(ELEM3(t->spacetype, SPACE_ACTION, SPACE_NLA, SPACE_IPO)) {
 354                 allqueue(REDRAWVIEW3D, 0);
 355                 allqueue(REDRAWACTION, 0);
 356                 allqueue(REDRAWNLA, 0);
 357                 allqueue(REDRAWIPO, 0);
 358                 allqueue(REDRAWTIME, 0);
 359                 allqueue(REDRAWBUTSOBJECT, 0);
 360         }
 361
 362         scrarea_queue_headredraw(curarea);
 363 #endif
 364 }
 365
 366 /* ************************** TRANSFORMATIONS **************************** */
 367
 368 void BIF_selectOrientation() {
 369 #if 0 // TRANSFORM_FIX_ME
 370         short val;
 371         char *str_menu = BIF_menustringTransformOrientation("Orientation");
 372         val= pupmenu(str_menu);
 373         MEM_freeN(str_menu);
 374
 375         if(val >= 0) {
 376                 G.vd->twmode = val;
 377         }
 378 #endif
 379 }
 380
 381 static void view_editmove(unsigned short event)
 382 {
 383 #if 0 // TRANSFORM_FIX_ME
 384         int refresh = 0;
 385         /* Regular:   Zoom in */
 386         /* Shift:     Scroll up */
 387         /* Ctrl:      Scroll right */
 388         /* Alt-Shift: Rotate up */
 389         /* Alt-Ctrl:  Rotate right */
 390
 391         /* only work in 3D window for now
 392          * In the end, will have to send to event to a 2D window handler instead
 393          */
 394         if (Trans.flag & T_2D_EDIT)
 395                 return;
 396
 397         switch(event) {
 398                 case WHEELUPMOUSE:
 399
 400                         if( G.qual & LR_SHIFTKEY ) {
 401                                 if( G.qual & LR_ALTKEY ) {
 402                                         G.qual &= ~LR_SHIFTKEY;
 403                                         persptoetsen(PAD2);
 404                                         G.qual |= LR_SHIFTKEY;
 405                                 } else {
 406                                         persptoetsen(PAD2);
 407                                 }
 408                         } else if( G.qual & LR_CTRLKEY ) {
 409                                 if( G.qual & LR_ALTKEY ) {
 410                                         G.qual &= ~LR_CTRLKEY;
 411                                         persptoetsen(PAD4);
 412                                         G.qual |= LR_CTRLKEY;
 413                                 } else {
 414                                         persptoetsen(PAD4);
 415                                 }
 416                         } else if(U.uiflag & USER_WHEELZOOMDIR)
 417                                 persptoetsen(PADMINUS);
 418                         else
 419                                 persptoetsen(PADPLUSKEY);
 420
 421                         refresh = 1;
 422                         break;
 423                 case WHEELDOWNMOUSE:
 424                         if( G.qual & LR_SHIFTKEY ) {
 425                                 if( G.qual & LR_ALTKEY ) {
 426                                         G.qual &= ~LR_SHIFTKEY;
 427                                         persptoetsen(PAD8);
 428                                         G.qual |= LR_SHIFTKEY;
 429                                 } else {
 430                                         persptoetsen(PAD8);
 431                                 }
 432                         } else if( G.qual & LR_CTRLKEY ) {
 433                                 if( G.qual & LR_ALTKEY ) {
 434                                         G.qual &= ~LR_CTRLKEY;
 435                                         persptoetsen(PAD6);
 436                                         G.qual |= LR_CTRLKEY;
 437                                 } else {
 438                                         persptoetsen(PAD6);
 439                                 }
 440                         } else if(U.uiflag & USER_WHEELZOOMDIR)
 441                                 persptoetsen(PADPLUSKEY);
 442                         else
 443                                 persptoetsen(PADMINUS);
 444
 445                         refresh = 1;
 446                         break;
 447         }
 448
 449         if (refresh)
 450                 setTransformViewMatrices(&Trans);
 451 #endif
 452 }
 453
 454 #if 0
 455 static char *transform_to_undostr(TransInfo *t)
 456 {
 457         switch (t->mode) {
 458                 case TFM_TRANSLATION:
 459                         return "Translate";
 460                 case TFM_ROTATION:
 461                         return "Rotate";
 462                 case TFM_RESIZE:
 463                         return "Scale";
 464                 case TFM_TOSPHERE:
 465                         return "To Sphere";
 466                 case TFM_SHEAR:
 467                         return "Shear";
 468                 case TFM_WARP:
 469                         return "Warp";
 470                 case TFM_SHRINKFATTEN:
 471                         return "Shrink/Fatten";
 472                 case TFM_TILT:
 473                         return "Tilt";
 474                 case TFM_TRACKBALL:
 475                         return "Trackball";
 476                 case TFM_PUSHPULL:
 477                         return "Push/Pull";
 478                 case TFM_BEVEL:
 479                         return "Bevel";
 480                 case TFM_BWEIGHT:
 481                         return "Bevel Weight";
 482                 case TFM_CREASE:
 483                         return "Crease";
 484                 case TFM_BONESIZE:
 485                         return "Bone Width";
 486                 case TFM_BONE_ENVELOPE:
 487                         return "Bone Envelope";
 488                 case TFM_TIME_TRANSLATE:
 489                         return "Translate Anim. Data";
 490                 case TFM_TIME_SCALE:
 491                         return "Scale Anim. Data";
 492                 case TFM_TIME_SLIDE:
 493                         return "Time Slide";
 494                 case TFM_BAKE_TIME:
 495                         return "Key Time";
 496                 case TFM_MIRROR:
 497                         return "Mirror";
 498         }
 499         return "Transform";
 500 }
 501 #endif
 502
 503 /* ************************************************* */
 504
 505 /* NOTE: these defines are saved in keymap files, do not change values but just add new ones */
 506 #define TFM_MODAL_CANCEL                        1
 507 #define TFM_MODAL_CONFIRM                       2
 508 #define TFM_MODAL_TRANSLATE                     3
 509 #define TFM_MODAL_ROTATE                        4
 510 #define TFM_MODAL_RESIZE                        5
 511 #define TFM_MODAL_SNAP_GEARS            6
 512 #define TFM_MODAL_SNAP_GEARS_OFF        7
 513
 514 /* called in transform_ops.c, on each regeneration of keymaps */
 515 void transform_modal_keymap(wmWindowManager *wm)
 516 {
 517         static EnumPropertyItem modal_items[] = {
 518         {TFM_MODAL_CANCEL, "CANCEL", 0, "Cancel", ""},
 519         {TFM_MODAL_CONFIRM, "CONFIRM", 0, "Confirm", ""},
 520         {TFM_MODAL_TRANSLATE, "TRANSLATE", 0, "Translate", ""},
 521         {TFM_MODAL_ROTATE, "ROTATE", 0, "Rotate", ""},
 522         {TFM_MODAL_RESIZE, "RESIZE", 0, "Resize", ""},
 523         {TFM_MODAL_SNAP_GEARS, "SNAP_GEARS", 0, "Snap On", ""},
 524         {TFM_MODAL_SNAP_GEARS_OFF, "SNAP_GEARS_OFF", 0, "Snap Off", ""},
 525         {0, NULL, 0, NULL, NULL}};
 526
 527         wmKeyMap *keymap= WM_modalkeymap_get(wm, "Transform Modal Map");
 528
 529         /* this function is called for each spacetype, only needs to add map once */
 530         if(keymap) return;
 531
 532         keymap= WM_modalkeymap_add(wm, "Transform Modal Map", modal_items);
 533
 534         /* items for modal map */
 535         WM_modalkeymap_add_item(keymap, ESCKEY,    KM_PRESS, KM_ANY, 0, TFM_MODAL_CANCEL);
 536         WM_modalkeymap_add_item(keymap, LEFTMOUSE, KM_ANY, KM_ANY, 0, TFM_MODAL_CONFIRM);
 537         WM_modalkeymap_add_item(keymap, RETKEY, KM_PRESS, KM_ANY, 0, TFM_MODAL_CONFIRM);
 538         WM_modalkeymap_add_item(keymap, PADENTER, KM_PRESS, KM_ANY, 0, TFM_MODAL_CONFIRM);
 539
 540         WM_modalkeymap_add_item(keymap, GKEY, KM_PRESS, 0, 0, TFM_MODAL_TRANSLATE);
 541         WM_modalkeymap_add_item(keymap, RKEY, KM_PRESS, 0, 0, TFM_MODAL_ROTATE);
 542         WM_modalkeymap_add_item(keymap, SKEY, KM_PRESS, 0, 0, TFM_MODAL_RESIZE);
 543
 544         WM_modalkeymap_add_item(keymap, LEFTCTRLKEY, KM_PRESS, KM_ANY, 0, TFM_MODAL_SNAP_GEARS);
 545         WM_modalkeymap_add_item(keymap, LEFTCTRLKEY, KM_RELEASE, KM_ANY, 0, TFM_MODAL_SNAP_GEARS_OFF);
 546
 547         /* assign map to operators */
 548         WM_modalkeymap_assign(keymap, "TFM_OT_transform");
 549         WM_modalkeymap_assign(keymap, "TFM_OT_translate");
 550         WM_modalkeymap_assign(keymap, "TFM_OT_rotate");
 551         WM_modalkeymap_assign(keymap, "TFM_OT_tosphere");
 552         WM_modalkeymap_assign(keymap, "TFM_OT_resize");
 553         WM_modalkeymap_assign(keymap, "TFM_OT_shear");
 554         WM_modalkeymap_assign(keymap, "TFM_OT_warp");
 555         WM_modalkeymap_assign(keymap, "TFM_OT_shrink_fatten");
 556         WM_modalkeymap_assign(keymap, "TFM_OT_tilt");
 557         WM_modalkeymap_assign(keymap, "TFM_OT_trackball");
 558
 559 }
 560
 561
 562 void transformEvent(TransInfo *t, wmEvent *event)
 563 {
 564         float mati[3][3] = {{1.0f, 0.0f, 0.0f}, {0.0f, 1.0f, 0.0f}, {0.0f, 0.0f, 1.0f}};
 565         char cmode = constraintModeToChar(t);
 566
 567         t->redraw |= handleMouseInput(t, &t->mouse, event);
 568
 569         if (event->type == MOUSEMOVE)
 570         {
 571                 t->mval[0] = event->x - t->ar->winrct.xmin;
 572                 t->mval[1] = event->y - t->ar->winrct.ymin;
 573
 574                 t->redraw = 1;
 575
 576                 applyMouseInput(t, &t->mouse, t->mval, t->values);
 577         }
 578
 579         /* handle modal keymap first */
 580         if (event->type == EVT_MODAL_MAP) {
 581                 switch (event->val) {
 582                         case TFM_MODAL_CANCEL:
 583                                 t->state = TRANS_CANCEL;
 584                                 break;
 585                         case TFM_MODAL_CONFIRM:
 586                                 t->state = TRANS_CONFIRM;
 587                                 break;
 588
 589                         case TFM_MODAL_TRANSLATE:
 590                                 /* only switch when... */
 591                                 if( ELEM3(t->mode, TFM_ROTATION, TFM_RESIZE, TFM_TRACKBALL) ) {
 592                                         resetTransRestrictions(t);
 593                                         restoreTransObjects(t);
 594                                         initTranslation(t);
 595                                         initSnapping(t, NULL); // need to reinit after mode change
 596                                         t->redraw = 1;
 597                                 }
 598                                 break;
 599                         case TFM_MODAL_ROTATE:
 600                                 /* only switch when... */
 601                                 if( ELEM4(t->mode, TFM_ROTATION, TFM_RESIZE, TFM_TRACKBALL, TFM_TRANSLATION) ) {
 602
 603                                         resetTransRestrictions(t);
 604
 605                                         if (t->mode == TFM_ROTATION) {
 606                                                 restoreTransObjects(t);
 607                                                 initTrackball(t);
 608                                         }
 609                                         else {
 610                                                 restoreTransObjects(t);
 611                                                 initRotation(t);
 612                                         }
 613                                         initSnapping(t, NULL); // need to reinit after mode change
 614                                         t->redraw = 1;
 615                                 }
 616                                 break;
 617                         case TFM_MODAL_RESIZE:
 618                                 /* only switch when... */
 619                                 if( ELEM3(t->mode, TFM_ROTATION, TFM_TRANSLATION, TFM_TRACKBALL) ) {
 620                                         resetTransRestrictions(t);
 621                                         restoreTransObjects(t);
 622                                         initResize(t);
 623                                         initSnapping(t, NULL); // need to reinit after mode change
 624                                         t->redraw = 1;
 625                                 }
 626                                 break;
 627
 628                         case TFM_MODAL_SNAP_GEARS:
 629                                 t->modifiers |= MOD_SNAP_GEARS;
 630                                 t->redraw = 1;
 631                                 break;
 632                         case TFM_MODAL_SNAP_GEARS_OFF:
 633                                 t->modifiers &= ~MOD_SNAP_GEARS;
 634                                 t->redraw = 1;
 635                                 break;
 636                 }
 637         }
 638         /* else do non-mapped events */
 639         else if (event->val==KM_PRESS) {
 640                 switch (event->type){
 641                 case RIGHTMOUSE:
 642                         t->state = TRANS_CANCEL;
 643                         break;
 644                 /* enforce redraw of transform when modifiers are used */
 645                 case LEFTCTRLKEY:
 646                 case RIGHTCTRLKEY:
 647                         t->modifiers |= MOD_SNAP_GEARS;
 648                         t->redraw = 1;
 649                         break;
 650
 651                 case LEFTSHIFTKEY:
 652                 case RIGHTSHIFTKEY:
 653                         t->modifiers |= MOD_CONSTRAINT_PLANE;
 654                         t->redraw = 1;
 655                         break;
 656
 657                 case SPACEKEY:
 658                         if ((t->spacetype==SPACE_VIEW3D) && event->alt) {
 659 #if 0 // TRANSFORM_FIX_ME
 660                                 short mval[2];
 661
 662                                 getmouseco_sc(mval);
 663                                 BIF_selectOrientation();
 664                                 calc_manipulator_stats(curarea);
 665                                 Mat3CpyMat4(t->spacemtx, G.vd->twmat);
 666                                 warp_pointer(mval[0], mval[1]);
 667 #endif
 668                         }
 669                         else {
 670                                 t->state = TRANS_CONFIRM;
 671                         }
 672                         break;
 673
 674                 case MIDDLEMOUSE:
 675                         if ((t->flag & T_NO_CONSTRAINT)==0) {
 676                                 /* exception for switching to dolly, or trackball, in camera view */
 677                                 if (t->flag & T_CAMERA) {
 678                                         if (t->mode==TFM_TRANSLATION)
 679                                                 setLocalConstraint(t, (CON_AXIS2), "along local Z");
 680                                         else if (t->mode==TFM_ROTATION) {
 681                                                 restoreTransObjects(t);
 682                                                 initTrackball(t);
 683                                         }
 684                                 }
 685                                 else {
 686                                         t->modifiers |= MOD_CONSTRAINT_SELECT;
 687                                         if (t->con.mode & CON_APPLY) {
 688                                                 stopConstraint(t);
 689                                         }
 690                                         else {
 691                                                 if (event->shift) {
 692                                                         initSelectConstraint(t, t->spacemtx);
 693                                                 }
 694                                                 else {
 695                                                         /* bit hackish... but it prevents mmb select to print the orientation from menu */
 696                                                         strcpy(t->spacename, "global");
 697                                                         initSelectConstraint(t, mati);
 698                                                 }
 699                                                 postSelectConstraint(t);
 700                                         }
 701                                 }
 702                                 t->redraw = 1;
 703                         }
 704                         break;
 705                 case ESCKEY:
 706                         t->state = TRANS_CANCEL;
 707                         break;
 708                 case PADENTER:
 709                 case RETKEY:
 710                         t->state = TRANS_CONFIRM;
 711                         break;
 712                 case GKEY:
 713                         /* only switch when... */
 714                         if( ELEM3(t->mode, TFM_ROTATION, TFM_RESIZE, TFM_TRACKBALL) ) {
 715                                 resetTransRestrictions(t);
 716                                 restoreTransObjects(t);
 717                                 initTranslation(t);
 718                                 initSnapping(t, NULL); // need to reinit after mode change
 719                                 t->redraw = 1;
 720                         }
 721                         break;
 722                 case SKEY:
 723                         /* only switch when... */
 724                         if( ELEM3(t->mode, TFM_ROTATION, TFM_TRANSLATION, TFM_TRACKBALL) ) {
 725                                 resetTransRestrictions(t);
 726                                 restoreTransObjects(t);
 727                                 initResize(t);
 728                                 initSnapping(t, NULL); // need to reinit after mode change
 729                                 t->redraw = 1;
 730                         }
 731                         break;
 732                 case RKEY:
 733                         /* only switch when... */
 734                         if( ELEM4(t->mode, TFM_ROTATION, TFM_RESIZE, TFM_TRACKBALL, TFM_TRANSLATION) ) {
 735
 736                                 resetTransRestrictions(t);
 737
 738                                 if (t->mode == TFM_ROTATION) {
 739                                         restoreTransObjects(t);
 740                                         initTrackball(t);
 741                                 }
 742                                 else {
 743                                         restoreTransObjects(t);
 744                                         initRotation(t);
 745                                 }
 746                                 initSnapping(t, NULL); // need to reinit after mode change
 747                                 t->redraw = 1;
 748                         }
 749                         break;
 750                 case CKEY:
 751                         if (event->alt) {
 752                                 t->flag ^= T_PROP_CONNECTED;
 753                                 sort_trans_data_dist(t);
 754                                 calculatePropRatio(t);
 755                                 t->redraw= 1;
 756                         }
 757                         else {
 758                                 stopConstraint(t);
 759                                 t->redraw = 1;
 760                         }
 761                         break;
 762                 case XKEY:
 763                         if ((t->flag & T_NO_CONSTRAINT)==0) {
 764                                 if (cmode == 'X') {
 765                                         if (t->flag & T_2D_EDIT) {
 766                                                 stopConstraint(t);
 767                                         }
 768                                         else {
 769                                                 if (t->con.mode & CON_USER) {
 770                                                         stopConstraint(t);
 771                                                 }
 772                                                 else {
 773                                                         if ((t->modifiers & MOD_CONSTRAINT_PLANE) == 0)
 774                                                                 setUserConstraint(t, (CON_AXIS0), "along %s X");
 775                                                         else if (t->modifiers & MOD_CONSTRAINT_PLANE)
 776                                                                 setUserConstraint(t, (CON_AXIS1|CON_AXIS2), "locking %s X");
 777                                                 }
 778                                         }
 779                                 }
 780                                 else {
 781                                         if (t->flag & T_2D_EDIT) {
 782                                                 setConstraint(t, mati, (CON_AXIS0), "along X axis");
 783                                         }
 784                                         else {
 785                                                 if ((t->modifiers & MOD_CONSTRAINT_PLANE) == 0)
 786                                                         setConstraint(t, mati, (CON_AXIS0), "along global X");
 787                                                 else if (t->modifiers & MOD_CONSTRAINT_PLANE)
 788                                                         setConstraint(t, mati, (CON_AXIS1|CON_AXIS2), "locking global X");
 789                                         }
 790                                 }
 791                                 t->redraw = 1;
 792                         }
 793                         break;
 794                 case YKEY:
 795                         if ((t->flag & T_NO_CONSTRAINT)==0) {
 796                                 if (cmode == 'Y') {
 797                                         if (t->flag & T_2D_EDIT) {
 798                                                 stopConstraint(t);
 799                                         }
 800                                         else {
 801                                                 if (t->con.mode & CON_USER) {
 802                                                         stopConstraint(t);
 803                                                 }
 804                                                 else {
 805                                                         if ((t->modifiers & MOD_CONSTRAINT_PLANE) == 0)
 806                                                                 setUserConstraint(t, (CON_AXIS1), "along %s Y");
 807                                                         else if (t->modifiers & MOD_CONSTRAINT_PLANE)
 808                                                                 setUserConstraint(t, (CON_AXIS0|CON_AXIS2), "locking %s Y");
 809                                                 }
 810                                         }
 811                                 }
 812                                 else {
 813                                         if (t->flag & T_2D_EDIT) {
 814                                                 setConstraint(t, mati, (CON_AXIS1), "along Y axis");
 815                                         }
 816                                         else {
 817                                                 if ((t->modifiers & MOD_CONSTRAINT_PLANE) == 0)
 818                                                         setConstraint(t, mati, (CON_AXIS1), "along global Y");
 819                                                 else if (t->modifiers & MOD_CONSTRAINT_PLANE)
 820                                                         setConstraint(t, mati, (CON_AXIS0|CON_AXIS2), "locking global Y");
 821                                         }
 822                                 }
 823                                 t->redraw = 1;
 824                         }
 825                         break;
 826                 case ZKEY:
 827                         if ((t->flag & T_NO_CONSTRAINT)==0) {
 828                                 if (cmode == 'Z') {
 829                                         if (t->con.mode & CON_USER) {
 830                                                 stopConstraint(t);
 831                                         }
 832                                         else {
 833                                                 if ((t->modifiers & MOD_CONSTRAINT_PLANE) == 0)
 834                                                         setUserConstraint(t, (CON_AXIS2), "along %s Z");
 835                                                 else if ((t->modifiers & MOD_CONSTRAINT_PLANE) && ((t->flag & T_2D_EDIT)==0))
 836                                                         setUserConstraint(t, (CON_AXIS0|CON_AXIS1), "locking %s Z");
 837                                         }
 838                                 }
 839                                 else if ((t->flag & T_2D_EDIT)==0) {
 840                                         if ((t->modifiers & MOD_CONSTRAINT_PLANE) == 0)
 841                                                 setConstraint(t, mati, (CON_AXIS2), "along global Z");
 842                                         else if (t->modifiers & MOD_CONSTRAINT_PLANE)
 843                                                 setConstraint(t, mati, (CON_AXIS0|CON_AXIS1), "locking global Z");
 844                                 }
 845                                 t->redraw = 1;
 846                         }
 847                         break;
 848                 case OKEY:
 849                         if (t->flag & T_PROP_EDIT && event->shift) {
 850                                 t->prop_mode = (t->prop_mode + 1) % 6;
 851                                 calculatePropRatio(t);
 852                                 t->redraw = 1;
 853                         }
 854                         break;
 855                 case PADPLUSKEY:
 856                         if(event->alt && t->flag & T_PROP_EDIT) {
 857                                 t->prop_size *= 1.1f;
 858                                 calculatePropRatio(t);
 859                         }
 860                         t->redraw= 1;
 861                         break;
 862                 case PAGEUPKEY:
 863                 case WHEELDOWNMOUSE:
 864                         if (t->flag & T_AUTOIK) {
 865                                 transform_autoik_update(t, 1);
 866                         }
 867                         else if(t->flag & T_PROP_EDIT) {
 868                                 t->prop_size*= 1.1f;
 869                                 calculatePropRatio(t);
 870                         }
 871                         else view_editmove(event->type);
 872                         t->redraw= 1;
 873                         break;
 874                 case PADMINUS:
 875                         if(event->alt && t->flag & T_PROP_EDIT) {
 876                                 t->prop_size*= 0.90909090f;
 877                                 calculatePropRatio(t);
 878                         }
 879                         t->redraw= 1;
 880                         break;
 881                 case PAGEDOWNKEY:
 882                 case WHEELUPMOUSE:
 883                         if (t->flag & T_AUTOIK) {
 884                                 transform_autoik_update(t, -1);
 885                         }
 886                         else if (t->flag & T_PROP_EDIT) {
 887                                 t->prop_size*= 0.90909090f;
 888                                 calculatePropRatio(t);
 889                         }
 890                         else view_editmove(event->type);
 891                         t->redraw= 1;
 892                         break;
 893 //              case NDOFMOTION:
 894 //            viewmoveNDOF(1);
 895   //         break;
 896                 }
 897
 898                 // Numerical input events
 899                 t->redraw |= handleNumInput(&(t->num), event);
 900
 901                 // NDof input events
 902                 switch(handleNDofInput(&(t->ndof), event))
 903                 {
 904                         case NDOF_CONFIRM:
 905                                 if ((t->options & CTX_NDOF) == 0)
 906                                 {
 907                                         /* Confirm on normal transform only */
 908                                         t->state = TRANS_CONFIRM;
 909                                 }
 910                                 break;
 911                         case NDOF_CANCEL:
 912                                 if (t->options & CTX_NDOF)
 913                                 {
 914                                         /* Cancel on pure NDOF transform */
 915                                         t->state = TRANS_CANCEL;
 916                                 }
 917                                 else
 918                                 {
 919                                         /* Otherwise, just redraw, NDof input was cancelled */
 920                                         t->redraw = 1;
 921                                 }
 922                                 break;
 923                         case NDOF_NOMOVE:
 924                                 if (t->options & CTX_NDOF)
 925                                 {
 926                                         /* Confirm on pure NDOF transform */
 927                                         t->state = TRANS_CONFIRM;
 928                                 }
 929                                 break;
 930                         case NDOF_REFRESH:
 931                                 t->redraw = 1;
 932                                 break;
 933
 934                 }
 935
 936                 // Snapping events
 937                 t->redraw |= handleSnapping(t, event);
 938
 939                 //arrows_move_cursor(event->type);
 940         }
 941         else {
 942                 switch (event->type){
 943                 case LEFTMOUSE:
 944                         t->state = TRANS_CONFIRM;
 945                         break;
 946                 case LEFTSHIFTKEY:
 947                 case RIGHTSHIFTKEY:
 948                         t->modifiers &= ~MOD_CONSTRAINT_PLANE;
 949                         t->redraw = 1;
 950                         break;
 951
 952                 case LEFTCTRLKEY:
 953                 case RIGHTCTRLKEY:
 954                         t->modifiers &= ~MOD_SNAP_GEARS;
 955                         /* no redraw on release modifier keys! this makes sure you can assign the 'grid' still
 956                            after releasing modifer key */
 957                         //t->redraw = 1;
 958                         break;
 959                 case MIDDLEMOUSE:
 960                         if ((t->flag & T_NO_CONSTRAINT)==0) {
 961                                 t->modifiers &= ~MOD_CONSTRAINT_SELECT;
 962                                 postSelectConstraint(t);
 963                                 t->redraw = 1;
 964                         }
 965                         break;
 966 //              case LEFTMOUSE:
 967 //              case RIGHTMOUSE:
 968 //                      if(WM_modal_tweak_exit(event, t->event_type))
 969 ////                    if (t->options & CTX_TWEAK)
 970 //                              t->state = TRANS_CONFIRM;
 971 //                      break;
 972                 }
 973         }
 974
 975         // Per transform event, if present
 976         if (t->handleEvent)
 977                 t->redraw |= t->handleEvent(t, event);
 978 }
 979
 980 int calculateTransformCenter(bContext *C, wmEvent *event, int centerMode, float *vec)
 981 {
 982         TransInfo *t = MEM_callocN(sizeof(TransInfo), "TransInfo data");
 983         int success = 1;
 984
 985         t->state = TRANS_RUNNING;
 986
 987         t->options = CTX_NONE;
 988
 989         t->mode = TFM_DUMMY;
 990
 991         initTransInfo(C, t, NULL, event);                                       // internal data, mouse, vectors
 992
 993         createTransData(C, t);                  // make TransData structs from selection
 994
 995         t->around = centerMode;                         // override userdefined mode
 996
 997         if (t->total == 0) {
 998                 success = 0;
 999         }
1000         else {
1001                 success = 1;
1002
1003                 calculateCenter(t);
1004
1005                 // Copy center from constraint center. Transform center can be local
1006                 VECCOPY(vec, t->con.center);
1007         }
1008
1009         postTrans(t);
1010
1011         /* aftertrans does insert ipos and action channels, and clears base flags, doesnt read transdata */
1012         special_aftertrans_update(t);
1013
1014         MEM_freeN(t);
1015
1016         return success;
1017 }
1018
1019 typedef enum {
1020         UP,
1021         DOWN,
1022         LEFT,
1023         RIGHT
1024 } ArrowDirection;
1025 static void drawArrow(ArrowDirection d, short offset, short length, short size)
1026 {
1027         switch(d)
1028         {
1029                 case LEFT:
1030                         offset = -offset;
1031                         length = -length;
1032                         size = -size;
1033                 case RIGHT:
1034                         glBegin(GL_LINES);
1035                         glVertex2s( offset, 0);
1036                         glVertex2s( offset + length, 0);
1037                         glVertex2s( offset + length, 0);
1038                         glVertex2s( offset + length - size, -size);
1039                         glVertex2s( offset + length, 0);
1040                         glVertex2s( offset + length - size,  size);
1041                         glEnd();
1042                         break;
1043                 case DOWN:
1044                         offset = -offset;
1045                         length = -length;
1046                         size = -size;
1047                 case UP:
1048                         glBegin(GL_LINES);
1049                         glVertex2s( 0, offset);
1050                         glVertex2s( 0, offset + length);
1051                         glVertex2s( 0, offset + length);
1052                         glVertex2s(-size, offset + length - size);
1053                         glVertex2s( 0, offset + length);
1054                         glVertex2s( size, offset + length - size);
1055                         glEnd();
1056                         break;
1057         }
1058 }
1059
1060 static void drawArrowHead(ArrowDirection d, short size)
1061 {
1062         switch(d)
1063         {
1064                 case LEFT:
1065                         size = -size;
1066                 case RIGHT:
1067                         glBegin(GL_LINES);
1068                         glVertex2s( 0, 0);
1069                         glVertex2s( -size, -size);
1070                         glVertex2s( 0, 0);
1071                         glVertex2s( -size,  size);
1072                         glEnd();
1073                         break;
1074                 case DOWN:
1075                         size = -size;
1076                 case UP:
1077                         glBegin(GL_LINES);
1078                         glVertex2s( 0, 0);
1079                         glVertex2s(-size, -size);
1080                         glVertex2s( 0, 0);
1081                         glVertex2s( size, -size);
1082                         glEnd();
1083                         break;
1084         }
1085 }
1086
1087 static void drawArc(float size, float angle_start, float angle_end, int segments)
1088 {
1089         float delta = (angle_end - angle_start) / segments;
1090         float angle;
1091
1092         glBegin(GL_LINE_STRIP);
1093
1094         for( angle = angle_start; angle < angle_end; angle += delta)
1095         {
1096                 glVertex2f( cosf(angle) * size, sinf(angle) * size);
1097         }
1098         glVertex2f( cosf(angle_end) * size, sinf(angle_end) * size);
1099
1100         glEnd();
1101 }
1102
1103 void drawHelpline(const struct bContext *C, TransInfo *t)
1104 {
1105         if (t->helpline != HLP_NONE && !(t->flag & T_USES_MANIPULATOR))
1106         {
1107                 float vecrot[3], cent[2];
1108
1109                 VECCOPY(vecrot, t->center);
1110                 if(t->flag & T_EDIT) {
1111                         Object *ob= t->obedit;
1112                         if(ob) Mat4MulVecfl(ob->obmat, vecrot);
1113                 }
1114                 else if(t->flag & T_POSE) {
1115                         Object *ob=t->poseobj;
1116                         if(ob) Mat4MulVecfl(ob->obmat, vecrot);
1117                 }
1118
1119                 projectFloatView(t, vecrot, cent);      // no overflow in extreme cases
1120
1121                 glDisable(GL_DEPTH_TEST);
1122
1123                 glMatrixMode(GL_PROJECTION);
1124                 glPushMatrix();
1125                 glMatrixMode(GL_MODELVIEW);
1126                 glPushMatrix();
1127
1128                 ED_region_pixelspace(t->ar);
1129
1130                 switch(t->helpline)
1131                 {
1132                         case HLP_SPRING:
1133                                 UI_ThemeColor(TH_WIRE);
1134
1135                                 setlinestyle(3);
1136                                 glBegin(GL_LINE_STRIP);
1137                                 glVertex2sv(t->mval);
1138                                 glVertex2fv(cent);
1139                                 glEnd();
1140
1141                                 glTranslatef(t->mval[0], t->mval[1], 0);
1142                                 glRotatef(-180 / M_PI * atan2f(cent[0] - t->mval[0], cent[1] - t->mval[1]), 0, 0, 1);
1143
1144                                 setlinestyle(0);
1145                                 glLineWidth(3.0);
1146                                 drawArrow(UP, 5, 10, 5);
1147                                 drawArrow(DOWN, 5, 10, 5);
1148                                 glLineWidth(1.0);
1149                                 break;
1150                         case HLP_HARROW:
1151                                 UI_ThemeColor(TH_WIRE);
1152
1153                                 glTranslatef(t->mval[0], t->mval[1], 0);
1154
1155                                 glLineWidth(3.0);
1156                                 drawArrow(RIGHT, 5, 10, 5);
1157                                 drawArrow(LEFT, 5, 10, 5);
1158                                 glLineWidth(1.0);
1159                                 break;
1160                         case HLP_VARROW:
1161                                 UI_ThemeColor(TH_WIRE);
1162
1163                                 glTranslatef(t->mval[0], t->mval[1], 0);
1164
1165                                 glLineWidth(3.0);
1166                                 glBegin(GL_LINES);
1167                                 drawArrow(UP, 5, 10, 5);
1168                                 drawArrow(DOWN, 5, 10, 5);
1169                                 glLineWidth(1.0);
1170                                 break;
1171                         case HLP_ANGLE:
1172                                 {
1173                                         float dx = t->mval[0] - cent[0], dy = t->mval[1] - cent[1];
1174                                         float angle = atan2f(dy, dx);
1175                                         float dist = sqrtf(dx*dx + dy*dy);
1176                                         float delta_angle = MIN2(15 / dist, M_PI/4);
1177                                         float spacing_angle = MIN2(5 / dist, M_PI/12);
1178                                         UI_ThemeColor(TH_WIRE);
1179
1180                                         setlinestyle(3);
1181                                         glBegin(GL_LINE_STRIP);
1182                                         glVertex2sv(t->mval);
1183                                         glVertex2fv(cent);
1184                                         glEnd();
1185
1186                                         glTranslatef(cent[0], cent[1], 0);
1187
1188                                         setlinestyle(0);
1189                                         glLineWidth(3.0);
1190                                         drawArc(dist, angle - delta_angle, angle - spacing_angle, 10);
1191                                         drawArc(dist, angle + spacing_angle, angle + delta_angle, 10);
1192
1193                                         glPushMatrix();
1194
1195                                         glTranslatef(cosf(angle - delta_angle) * dist, sinf(angle - delta_angle) * dist, 0);
1196                                         glRotatef(180 / M_PI * (angle - delta_angle), 0, 0, 1);
1197
1198                                         drawArrowHead(DOWN, 5);
1199
1200                                         glPopMatrix();
1201
1202                                         glTranslatef(cosf(angle + delta_angle) * dist, sinf(angle + delta_angle) * dist, 0);
1203                                         glRotatef(180 / M_PI * (angle + delta_angle), 0, 0, 1);
1204
1205                                         drawArrowHead(UP, 5);
1206
1207                                         glLineWidth(1.0);
1208                                         break;
1209                                 }
1210                                 case HLP_TRACKBALL:
1211                                 {
1212                                         char col[3], col2[3];
1213                                         UI_GetThemeColor3ubv(TH_GRID, col);
1214
1215                                         glTranslatef(t->mval[0], t->mval[1], 0);
1216
1217                                         glLineWidth(3.0);
1218
1219                                         UI_make_axis_color(col, col2, 'x');
1220                                         glColor3ubv((GLubyte *)col2);
1221
1222                                         drawArrow(RIGHT, 5, 10, 5);
1223                                         drawArrow(LEFT, 5, 10, 5);
1224
1225                                         UI_make_axis_color(col, col2, 'y');
1226                                         glColor3ubv((GLubyte *)col2);
1227
1228                                         drawArrow(UP, 5, 10, 5);
1229                                         drawArrow(DOWN, 5, 10, 5);
1230                                         glLineWidth(1.0);
1231                                         break;
1232                                 }
1233                 }
1234
1235                 glMatrixMode(GL_PROJECTION);
1236                 glPopMatrix();
1237                 glMatrixMode(GL_MODELVIEW);
1238                 glPopMatrix();
1239
1240                 glEnable(GL_DEPTH_TEST);
1241         }
1242 }
1243
1244 void drawTransform(const struct bContext *C, struct ARegion *ar, void *arg)
1245 {
1246         TransInfo *t = arg;
1247
1248         drawConstraint(C, t);
1249         drawPropCircle(C, t);
1250         drawSnapping(C, t);
1251         drawHelpline(C, t);
1252 }
1253
1254 void saveTransform(bContext *C, TransInfo *t, wmOperator *op)
1255 {
1256         ToolSettings *ts = CTX_data_tool_settings(C);
1257         int constraint_axis[3] = {0, 0, 0};
1258         int proportional = 0;
1259
1260         if (t->flag & T_AUTOVALUES)
1261         {
1262                 RNA_float_set_array(op->ptr, "value", t->auto_values);
1263         }
1264         else
1265         {
1266                 RNA_float_set_array(op->ptr, "value", t->values);
1267         }
1268
1269         /* XXX convert stupid flag to enum */
1270         switch(t->flag & (T_PROP_EDIT|T_PROP_CONNECTED))
1271         {
1272         case (T_PROP_EDIT|T_PROP_CONNECTED):
1273                 proportional = 2;
1274                 break;
1275         case T_PROP_EDIT:
1276                 proportional = 1;
1277                 break;
1278         default:
1279                 proportional = 0;
1280         }
1281
1282         if (RNA_struct_find_property(op->ptr, "proportional"))
1283         {
1284                 RNA_enum_set(op->ptr, "proportional", proportional);
1285                 RNA_enum_set(op->ptr, "proportional_editing_falloff", t->prop_mode);
1286                 RNA_float_set(op->ptr, "proportional_size", t->prop_size);
1287         }
1288
1289         if (RNA_struct_find_property(op->ptr, "mirror"))
1290         {
1291                 RNA_boolean_set(op->ptr, "mirror", t->flag & T_MIRROR);
1292         }
1293
1294         if (RNA_struct_find_property(op->ptr, "constraint_axis"))
1295         {
1296                 RNA_enum_set(op->ptr, "constraint_orientation", t->current_orientation);
1297
1298                 if (t->con.mode & CON_APPLY)
1299                 {
1300                         if (t->con.mode & CON_AXIS0) {
1301                                 constraint_axis[0] = 1;
1302                         }
1303                         if (t->con.mode & CON_AXIS1) {
1304                                 constraint_axis[1] = 1;
1305                         }
1306                         if (t->con.mode & CON_AXIS2) {
1307                                 constraint_axis[2] = 1;
1308                         }
1309                 }
1310
1311                 RNA_boolean_set_array(op->ptr, "constraint_axis", constraint_axis);
1312         }
1313
1314         // XXX If modal, save settings back in scene
1315         if (t->flag & T_MODAL)
1316         {
1317                 ts->prop_mode = t->prop_mode;
1318                 ts->proportional = proportional;
1319
1320                 if(t->spacetype == SPACE_VIEW3D)
1321                 {
1322                         View3D *v3d = t->view;
1323
1324                         v3d->twmode = t->current_orientation;
1325                 }
1326         }
1327 }
1328
1329 int initTransform(bContext *C, TransInfo *t, wmOperator *op, wmEvent *event, int mode)
1330 {
1331         int options = 0;
1332
1333         /* added initialize, for external calls to set stuff in TransInfo, like undo string */
1334
1335         t->state = TRANS_RUNNING;
1336
1337         t->options = options;
1338
1339         t->mode = mode;
1340
1341         if (!initTransInfo(C, t, op, event))                                    // internal data, mouse, vectors
1342         {
1343                 return 0;
1344         }
1345
1346         if(t->spacetype == SPACE_VIEW3D)
1347         {
1348                 //calc_manipulator_stats(curarea);
1349                 initTransformOrientation(C, t);
1350
1351                 t->draw_handle = ED_region_draw_cb_activate(t->ar->type, drawTransform, t, REGION_DRAW_POST);
1352         }
1353         else if(t->spacetype == SPACE_IMAGE) {
1354                 Mat3One(t->spacemtx);
1355                 t->draw_handle = ED_region_draw_cb_activate(t->ar->type, drawTransform, t, REGION_DRAW_POST);
1356         }
1357         else
1358                 Mat3One(t->spacemtx);
1359
1360         createTransData(C, t);                  // make TransData structs from selection
1361
1362         if (t->total == 0) {
1363                 postTrans(t);
1364                 return 0;
1365         }
1366
1367         initSnapping(t, op); // Initialize snapping data AFTER mode flags
1368
1369         /* EVIL! posemode code can switch translation to rotate when 1 bone is selected. will be removed (ton) */
1370         /* EVIL2: we gave as argument also texture space context bit... was cleared */
1371         /* EVIL3: extend mode for animation editors also switches modes... but is best way to avoid duplicate code */
1372         mode = t->mode;
1373
1374         calculatePropRatio(t);
1375         calculateCenter(t);
1376
1377         initMouseInput(t, &t->mouse, t->center2d, t->imval);
1378
1379         switch (mode) {
1380         case TFM_TRANSLATION:
1381                 initTranslation(t);
1382                 break;
1383         case TFM_ROTATION:
1384                 initRotation(t);
1385                 break;
1386         case TFM_RESIZE:
1387                 initResize(t);
1388                 break;
1389         case TFM_TOSPHERE:
1390                 initToSphere(t);
1391                 break;
1392         case TFM_SHEAR:
1393                 initShear(t);
1394                 break;
1395         case TFM_WARP:
1396                 initWarp(t);
1397                 break;
1398         case TFM_SHRINKFATTEN:
1399                 initShrinkFatten(t);
1400                 break;
1401         case TFM_TILT:
1402                 initTilt(t);
1403                 break;
1404         case TFM_CURVE_SHRINKFATTEN:
1405                 initCurveShrinkFatten(t);
1406                 break;
1407         case TFM_TRACKBALL:
1408                 initTrackball(t);
1409                 break;
1410         case TFM_PUSHPULL:
1411                 initPushPull(t);
1412                 break;
1413         case TFM_CREASE:
1414                 initCrease(t);
1415                 break;
1416         case TFM_BONESIZE:
1417                 {       /* used for both B-Bone width (bonesize) as for deform-dist (envelope) */
1418                         bArmature *arm= t->poseobj->data;
1419                         if(arm->drawtype==ARM_ENVELOPE)
1420                                 initBoneEnvelope(t);
1421                         else
1422                                 initBoneSize(t);
1423                 }
1424                 break;
1425         case TFM_BONE_ENVELOPE:
1426                 initBoneEnvelope(t);
1427                 break;
1428         case TFM_BONE_ROLL:
1429                 initBoneRoll(t);
1430                 break;
1431         case TFM_TIME_TRANSLATE:
1432                 initTimeTranslate(t);
1433                 break;
1434         case TFM_TIME_SLIDE:
1435                 initTimeSlide(t);
1436                 break;
1437         case TFM_TIME_SCALE:
1438                 initTimeScale(t);
1439                 break;
1440         case TFM_TIME_EXTEND:
1441                 /* now that transdata has been made, do like for TFM_TIME_TRANSLATE (for most Animation
1442                  * Editors because they have only 1D transforms for time values) or TFM_TRANSLATION
1443                  * (for Graph/NLA Editors only since they uses 'standard' transforms to get 2D movement)
1444                  * depending on which editor this was called from
1445                  */
1446                 if ELEM(t->spacetype, SPACE_IPO, SPACE_NLA)
1447                         initTranslation(t);
1448                 else
1449                         initTimeTranslate(t);
1450                 break;
1451         case TFM_BAKE_TIME:
1452                 initBakeTime(t);
1453                 break;
1454         case TFM_MIRROR:
1455                 initMirror(t);
1456                 break;
1457         case TFM_BEVEL:
1458                 initBevel(t);
1459                 break;
1460         case TFM_BWEIGHT:
1461                 initBevelWeight(t);
1462                 break;
1463         case TFM_ALIGN:
1464                 initAlign(t);
1465                 break;
1466         }
1467
1468         /* overwrite initial values if operator supplied a non-null vector */
1469         if (RNA_property_is_set(op->ptr, "value"))
1470         {
1471                 float values[4];
1472                 RNA_float_get_array(op->ptr, "value", values);
1473                 QUATCOPY(t->values, values);
1474                 QUATCOPY(t->auto_values, values);
1475                 t->flag |= T_AUTOVALUES;
1476         }
1477
1478         /* Constraint init from operator */
1479         if (RNA_struct_find_property(op->ptr, "constraint_axis") && RNA_property_is_set(op->ptr, "constraint_axis"))
1480         {
1481                 int constraint_axis[3];
1482
1483                 RNA_boolean_get_array(op->ptr, "constraint_axis", constraint_axis);
1484
1485                 if (constraint_axis[0] || constraint_axis[1] || constraint_axis[2])
1486                 {
1487                         t->con.mode |= CON_APPLY;
1488
1489                         if (constraint_axis[0]) {
1490                                 t->con.mode |= CON_AXIS0;
1491                         }
1492                         if (constraint_axis[1]) {
1493                                 t->con.mode |= CON_AXIS1;
1494                         }
1495                         if (constraint_axis[2]) {
1496                                 t->con.mode |= CON_AXIS2;
1497                         }
1498
1499                         setUserConstraint(t, t->con.mode, "%s");
1500                 }
1501         }
1502
1503         return 1;
1504 }
1505
1506 void transformApply(bContext *C, TransInfo *t)
1507 {
1508         if (t->redraw)
1509         {
1510                 if (t->modifiers & MOD_CONSTRAINT_SELECT)
1511                         t->con.mode |= CON_SELECT;
1512
1513                 selectConstraint(t);
1514                 if (t->transform) {
1515                         t->transform(t, t->mval);  // calls recalcData()
1516                         viewRedrawForce(C, t);
1517                 }
1518                 t->redraw = 0;
1519         }
1520
1521         /* If auto confirm is on, break after one pass */
1522         if (t->options & CTX_AUTOCONFIRM)
1523         {
1524                 t->state = TRANS_CONFIRM;
1525         }
1526
1527         if (BKE_ptcache_get_continue_physics())
1528         {
1529                 // TRANSFORM_FIX_ME
1530                 //do_screenhandlers(G.curscreen);
1531                 t->redraw = 1;
1532         }
1533 }
1534
1535 int transformEnd(bContext *C, TransInfo *t)
1536 {
1537         int exit_code = OPERATOR_RUNNING_MODAL;
1538
1539         if (t->state != TRANS_RUNNING)
1540         {
1541                 /* handle restoring objects */
1542                 if(t->state == TRANS_CANCEL)
1543                 {
1544                         exit_code = OPERATOR_CANCELLED;
1545                         restoreTransObjects(t); // calls recalcData()
1546                 }
1547                 else
1548                 {
1549                         exit_code = OPERATOR_FINISHED;
1550                 }
1551
1552                 /* free data */
1553                 postTrans(t);
1554
1555                 /* aftertrans does insert keyframes, and clears base flags, doesnt read transdata */
1556                 special_aftertrans_update(t);
1557
1558                 /* send events out for redraws */
1559                 viewRedrawPost(t);
1560
1561                 /*  Undo as last, certainly after special_trans_update! */
1562
1563                 if(t->state == TRANS_CANCEL) {
1564 //                      if(t->undostr) ED_undo_push(C, t->undostr);
1565                 }
1566                 else {
1567 //                      if(t->undostr) ED_undo_push(C, t->undostr);
1568 //                      else ED_undo_push(C, transform_to_undostr(t));
1569                 }
1570                 t->undostr= NULL;
1571
1572                 viewRedrawForce(C, t);
1573         }
1574
1575         return exit_code;
1576 }
1577
1578 /* ************************** TRANSFORM LOCKS **************************** */
1579
1580 static void protectedTransBits(short protectflag, float *vec)
1581 {
1582         if(protectflag & OB_LOCK_LOCX)
1583                 vec[0]= 0.0f;
1584         if(protectflag & OB_LOCK_LOCY)
1585                 vec[1]= 0.0f;
1586         if(protectflag & OB_LOCK_LOCZ)
1587                 vec[2]= 0.0f;
1588 }
1589
1590 static void protectedSizeBits(short protectflag, float *size)
1591 {
1592         if(protectflag & OB_LOCK_SCALEX)
1593                 size[0]= 1.0f;
1594         if(protectflag & OB_LOCK_SCALEY)
1595                 size[1]= 1.0f;
1596         if(protectflag & OB_LOCK_SCALEZ)
1597                 size[2]= 1.0f;
1598 }
1599
1600 static void protectedRotateBits(short protectflag, float *eul, float *oldeul)
1601 {
1602         if(protectflag & OB_LOCK_ROTX)
1603                 eul[0]= oldeul[0];
1604         if(protectflag & OB_LOCK_ROTY)
1605                 eul[1]= oldeul[1];
1606         if(protectflag & OB_LOCK_ROTZ)
1607                 eul[2]= oldeul[2];
1608 }
1609
1610 static void protectedQuaternionBits(short protectflag, float *quat, float *oldquat)
1611 {
1612         /* quaternions get limited with euler... */
1613         /* this function only does the delta rotation */
1614
1615         if(protectflag) {
1616                 float eul[3], oldeul[3], quat1[4];
1617
1618                 QUATCOPY(quat1, quat);
1619                 QuatToEul(quat, eul);
1620                 QuatToEul(oldquat, oldeul);
1621
1622                 if(protectflag & OB_LOCK_ROTX)
1623                         eul[0]= oldeul[0];
1624                 if(protectflag & OB_LOCK_ROTY)
1625                         eul[1]= oldeul[1];
1626                 if(protectflag & OB_LOCK_ROTZ)
1627                         eul[2]= oldeul[2];
1628
1629                 EulToQuat(eul, quat);
1630                 /* quaternions flip w sign to accumulate rotations correctly */
1631                 if( (quat1[0]<0.0f && quat[0]>0.0f) || (quat1[0]>0.0f && quat[0]<0.0f) ) {
1632                         QuatMulf(quat, -1.0f);
1633                 }
1634         }
1635 }
1636
1637 /* ******************* TRANSFORM LIMITS ********************** */
1638
1639 static void constraintTransLim(TransInfo *t, TransData *td)
1640 {
1641         if (td->con) {
1642                 bConstraintTypeInfo *cti= get_constraint_typeinfo(CONSTRAINT_TYPE_LOCLIMIT);
1643                 bConstraintOb cob;
1644                 bConstraint *con;
1645
1646                 /* Make a temporary bConstraintOb for using these limit constraints
1647                  *      - they only care that cob->matrix is correctly set ;-)
1648                  *      - current space should be local
1649                  */
1650                 memset(&cob, 0, sizeof(bConstraintOb));
1651                 Mat4One(cob.matrix);
1652                 if (td->tdi) {
1653                         TransDataIpokey *tdi= td->tdi;
1654                         cob.matrix[3][0]= tdi->locx[0];
1655                         cob.matrix[3][1]= tdi->locy[0];
1656                         cob.matrix[3][2]= tdi->locz[0];
1657                 }
1658                 else {
1659                         VECCOPY(cob.matrix[3], td->loc);
1660                 }
1661
1662                 /* Evaluate valid constraints */
1663                 for (con= td->con; con; con= con->next) {
1664                         float tmat[4][4];
1665
1666                         /* only consider constraint if enabled */
1667                         if (con->flag & CONSTRAINT_DISABLE) continue;
1668                         if (con->enforce == 0.0f) continue;
1669
1670                         /* only use it if it's tagged for this purpose (and the right type) */
1671                         if (con->type == CONSTRAINT_TYPE_LOCLIMIT) {
1672                                 bLocLimitConstraint *data= con->data;
1673
1674                                 if ((data->flag2 & LIMIT_TRANSFORM)==0)
1675                                         continue;
1676
1677                                 /* do space conversions */
1678                                 if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
1679                                         /* just multiply by td->mtx (this should be ok) */
1680                                         Mat4CpyMat4(tmat, cob.matrix);
1681                                         Mat4MulMat34(cob.matrix, td->mtx, tmat);
1682                                 }
1683                                 else if (con->ownspace != CONSTRAINT_SPACE_LOCAL) {
1684                                         /* skip... incompatable spacetype */
1685                                         continue;
1686                                 }
1687
1688                                 /* do constraint */
1689                                 cti->evaluate_constraint(con, &cob, NULL);
1690
1691                                 /* convert spaces again */
1692                                 if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
1693                                         /* just multiply by td->mtx (this should be ok) */
1694                                         Mat4CpyMat4(tmat, cob.matrix);
1695                                         Mat4MulMat34(cob.matrix, td->smtx, tmat);
1696                                 }
1697                         }
1698                 }
1699
1700                 /* copy results from cob->matrix */
1701                 if (td->tdi) {
1702                         TransDataIpokey *tdi= td->tdi;
1703                         tdi->locx[0]= cob.matrix[3][0];
1704                         tdi->locy[0]= cob.matrix[3][1];
1705                         tdi->locz[0]= cob.matrix[3][2];
1706                 }
1707                 else {
1708                         VECCOPY(td->loc, cob.matrix[3]);
1709                 }
1710         }
1711 }
1712
1713 static void constraintRotLim(TransInfo *t, TransData *td)
1714 {
1715         if (td->con) {
1716                 bConstraintTypeInfo *cti= get_constraint_typeinfo(CONSTRAINT_TYPE_ROTLIMIT);
1717                 bConstraintOb cob;
1718                 bConstraint *con;
1719
1720                 /* Make a temporary bConstraintOb for using these limit constraints
1721                  *      - they only care that cob->matrix is correctly set ;-)
1722                  *      - current space should be local
1723                  */
1724                 memset(&cob, 0, sizeof(bConstraintOb));
1725                 if (td->flag & TD_USEQUAT) {
1726                         /* quats */
1727                         if (td->ext)
1728                                 QuatToMat4(td->ext->quat, cob.matrix);
1729                         else
1730                                 return;
1731                 }
1732                 else if (td->tdi) {
1733                         /* ipo-keys eulers */
1734                         TransDataIpokey *tdi= td->tdi;
1735                         float eul[3];
1736
1737                         eul[0]= tdi->rotx[0];
1738                         eul[1]= tdi->roty[0];
1739                         eul[2]= tdi->rotz[0];
1740
1741                         EulToMat4(eul, cob.matrix);
1742                 }
1743                 else {
1744                         /* eulers */
1745                         if (td->ext)
1746                                 EulToMat4(td->ext->rot, cob.matrix);
1747                         else
1748                                 return;
1749                 }
1750
1751                 /* Evaluate valid constraints */
1752                 for (con= td->con; con; con= con->next) {
1753                         /* only consider constraint if enabled */
1754                         if (con->flag & CONSTRAINT_DISABLE) continue;
1755                         if (con->enforce == 0.0f) continue;
1756
1757                         /* we're only interested in Limit-Rotation constraints */
1758                         if (con->type == CONSTRAINT_TYPE_ROTLIMIT) {
1759                                 bRotLimitConstraint *data= con->data;
1760                                 float tmat[4][4];
1761
1762                                 /* only use it if it's tagged for this purpose */
1763                                 if ((data->flag2 & LIMIT_TRANSFORM)==0)
1764                                         continue;
1765
1766                                 /* do space conversions */
1767                                 if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
1768                                         /* just multiply by td->mtx (this should be ok) */
1769                                         Mat4CpyMat4(tmat, cob.matrix);
1770                                         Mat4MulMat34(cob.matrix, td->mtx, tmat);
1771                                 }
1772                                 else if (con->ownspace != CONSTRAINT_SPACE_LOCAL) {
1773                                         /* skip... incompatable spacetype */
1774                                         continue;
1775                                 }
1776
1777                                 /* do constraint */
1778                                 cti->evaluate_constraint(con, &cob, NULL);
1779
1780                                 /* convert spaces again */
1781                                 if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
1782                                         /* just multiply by td->mtx (this should be ok) */
1783                                         Mat4CpyMat4(tmat, cob.matrix);
1784                                         Mat4MulMat34(cob.matrix, td->smtx, tmat);
1785                                 }
1786                         }
1787                 }
1788
1789                 /* copy results from cob->matrix */
1790                 if (td->flag & TD_USEQUAT) {
1791                         /* quats */
1792                         Mat4ToQuat(cob.matrix, td->ext->quat);
1793                 }
1794                 else if (td->tdi) {
1795                         /* ipo-keys eulers */
1796                         TransDataIpokey *tdi= td->tdi;
1797                         float eul[3];
1798
1799                         Mat4ToEul(cob.matrix, eul);
1800
1801                         tdi->rotx[0]= eul[0];
1802                         tdi->roty[0]= eul[1];
1803                         tdi->rotz[0]= eul[2];
1804                 }
1805                 else {
1806                         /* eulers */
1807                         Mat4ToEul(cob.matrix, td->ext->rot);
1808                 }
1809         }
1810 }
1811
1812 static void constraintSizeLim(TransInfo *t, TransData *td)
1813 {
1814         if (td->con && td->ext) {
1815                 bConstraintTypeInfo *cti= get_constraint_typeinfo(CONSTRAINT_TYPE_SIZELIMIT);
1816                 bConstraintOb cob;
1817                 bConstraint *con;
1818
1819                 /* Make a temporary bConstraintOb for using these limit constraints
1820                  *      - they only care that cob->matrix is correctly set ;-)
1821                  *      - current space should be local
1822                  */
1823                 memset(&cob, 0, sizeof(bConstraintOb));
1824                 if (td->tdi) {
1825                         TransDataIpokey *tdi= td->tdi;
1826                         float size[3];
1827
1828                         size[0]= tdi->sizex[0];
1829                         size[1]= tdi->sizey[0];
1830                         size[2]= tdi->sizez[0];
1831                         SizeToMat4(size, cob.matrix);
1832                 }
1833                 else if ((td->flag & TD_SINGLESIZE) && !(t->con.mode & CON_APPLY)) {
1834                         /* scale val and reset size */
1835                         return; // TODO: fix this case
1836                 }
1837                 else {
1838                         /* Reset val if SINGLESIZE but using a constraint */
1839                         if (td->flag & TD_SINGLESIZE)
1840                                 return;
1841
1842                         SizeToMat4(td->ext->size, cob.matrix);
1843                 }
1844
1845                 /* Evaluate valid constraints */
1846                 for (con= td->con; con; con= con->next) {
1847                         /* only consider constraint if enabled */
1848                         if (con->flag & CONSTRAINT_DISABLE) continue;
1849                         if (con->enforce == 0.0f) continue;
1850
1851                         /* we're only interested in Limit-Scale constraints */
1852                         if (con->type == CONSTRAINT_TYPE_SIZELIMIT) {
1853                                 bSizeLimitConstraint *data= con->data;
1854                                 float tmat[4][4];
1855
1856                                 /* only use it if it's tagged for this purpose */
1857                                 if ((data->flag2 & LIMIT_TRANSFORM)==0)
1858                                         continue;
1859
1860                                 /* do space conversions */
1861                                 if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
1862                                         /* just multiply by td->mtx (this should be ok) */
1863                                         Mat4CpyMat4(tmat, cob.matrix);
1864                                         Mat4MulMat34(cob.matrix, td->mtx, tmat);
1865                                 }
1866                                 else if (con->ownspace != CONSTRAINT_SPACE_LOCAL) {
1867                                         /* skip... incompatable spacetype */
1868                                         continue;
1869                                 }
1870
1871                                 /* do constraint */
1872                                 cti->evaluate_constraint(con, &cob, NULL);
1873
1874                                 /* convert spaces again */
1875                                 if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
1876                                         /* just multiply by td->mtx (this should be ok) */
1877                                         Mat4CpyMat4(tmat, cob.matrix);
1878                                         Mat4MulMat34(cob.matrix, td->smtx, tmat);
1879                                 }
1880                         }
1881                 }
1882
1883                 /* copy results from cob->matrix */
1884                 if (td->tdi) {
1885                         TransDataIpokey *tdi= td->tdi;
1886                         float size[3];
1887
1888                         Mat4ToSize(cob.matrix, size);
1889
1890                         tdi->sizex[0]= size[0];
1891                         tdi->sizey[0]= size[1];
1892                         tdi->sizez[0]= size[2];
1893                 }
1894                 else if ((td->flag & TD_SINGLESIZE) && !(t->con.mode & CON_APPLY)) {
1895                         /* scale val and reset size */
1896                         return; // TODO: fix this case
1897                 }
1898                 else {
1899                         /* Reset val if SINGLESIZE but using a constraint */
1900                         if (td->flag & TD_SINGLESIZE)
1901                                 return;
1902
1903                         Mat4ToSize(cob.matrix, td->ext->size);
1904                 }
1905         }
1906 }
1907
1908 /* ************************** WARP *************************** */
1909
1910 void initWarp(TransInfo *t)
1911 {
1912         float max[3], min[3];
1913         int i;
1914
1915         t->mode = TFM_WARP;
1916         t->transform = Warp;
1917         t->handleEvent = handleEventWarp;
1918
1919         initMouseInputMode(t, &t->mouse, INPUT_HORIZONTAL_RATIO);
1920
1921         t->idx_max = 0;
1922         t->num.idx_max = 0;
1923         t->snap[0] = 0.0f;
1924         t->snap[1] = 5.0f;
1925         t->snap[2] = 1.0f;
1926
1927         t->flag |= T_NO_CONSTRAINT;
1928
1929         /* we need min/max in view space */
1930         for(i = 0; i < t->total; i++) {
1931                 float center[3];
1932                 VECCOPY(center, t->data[i].center);
1933                 Mat3MulVecfl(t->data[i].mtx, center);
1934                 Mat4MulVecfl(t->viewmat, center);
1935                 VecSubf(center, center, t->viewmat[3]);
1936                 if (i)
1937                         MinMax3(min, max, center);
1938                 else {
1939                         VECCOPY(max, center);
1940                         VECCOPY(min, center);
1941                 }
1942         }
1943
1944         t->center[0]= (min[0]+max[0])/2.0f;
1945         t->center[1]= (min[1]+max[1])/2.0f;
1946         t->center[2]= (min[2]+max[2])/2.0f;
1947
1948         if (max[0] == min[0]) max[0] += 0.1; /* not optimal, but flipping is better than invalid garbage (i.e. division by zero!) */
1949         t->val= (max[0]-min[0])/2.0f; /* t->val is X dimension projected boundbox */
1950 }
1951
1952 int handleEventWarp(TransInfo *t, wmEvent *event)
1953 {
1954         int status = 0;
1955
1956         if (event->type == MIDDLEMOUSE && event->val==KM_PRESS)
1957         {
1958                 // Use customData pointer to signal warp direction
1959                 if      (t->customData == 0)
1960                         t->customData = (void*)1;
1961                 else
1962                         t->customData = 0;
1963
1964                 status = 1;
1965         }
1966
1967         return status;
1968 }
1969
1970 int Warp(TransInfo *t, short mval[2])
1971 {
1972         TransData *td = t->data;
1973         float vec[3], circumfac, dist, phi0, co, si, *curs, cursor[3], gcursor[3];
1974         int i;
1975         char str[50];
1976
1977         curs= give_cursor(t->scene, t->view);
1978         /*
1979          * gcursor is the one used for helpline.
1980          * It has to be in the same space as the drawing loop
1981          * (that means it needs to be in the object's space when in edit mode and
1982          *  in global space in object mode)
1983          *
1984          * cursor is used for calculations.
1985          * It needs to be in view space, but we need to take object's offset
1986          * into account if in Edit mode.
1987          */
1988         VECCOPY(cursor, curs);
1989         VECCOPY(gcursor, cursor);
1990         if (t->flag & T_EDIT) {
1991                 VecSubf(cursor, cursor, t->obedit->obmat[3]);
1992                 VecSubf(gcursor, gcursor, t->obedit->obmat[3]);
1993                 Mat3MulVecfl(t->data->smtx, gcursor);
1994         }
1995         Mat4MulVecfl(t->viewmat, cursor);
1996         VecSubf(cursor, cursor, t->viewmat[3]);
1997
1998         /* amount of degrees for warp */
1999         circumfac = 360.0f * t->values[0];
2000
2001         if (t->customData) /* non-null value indicates reversed input */
2002         {
2003                 circumfac *= -1;
2004         }
2005
2006         snapGrid(t, &circumfac);
2007         applyNumInput(&t->num, &circumfac);
2008
2009         /* header print for NumInput */
2010         if (hasNumInput(&t->num)) {
2011                 char c[20];
2012
2013                 outputNumInput(&(t->num), c);
2014
2015                 sprintf(str, "Warp: %s", c);
2016         }
2017         else {
2018                 /* default header print */
2019                 sprintf(str, "Warp: %.3f", circumfac);
2020         }
2021
2022         circumfac*= (float)(-M_PI/360.0);
2023
2024         for(i = 0; i < t->total; i++, td++) {
2025                 float loc[3];
2026                 if (td->flag & TD_NOACTION)
2027                         break;
2028
2029                 if (td->flag & TD_SKIP)
2030                         continue;
2031
2032                 /* translate point to center, rotate in such a way that outline==distance */
2033                 VECCOPY(vec, td->iloc);
2034                 Mat3MulVecfl(td->mtx, vec);
2035                 Mat4MulVecfl(t->viewmat, vec);
2036                 VecSubf(vec, vec, t->viewmat[3]);
2037
2038                 dist= vec[0]-cursor[0];
2039
2040                 /* t->val is X dimension projected boundbox */
2041                 phi0= (circumfac*dist/t->val);
2042
2043                 vec[1]= (vec[1]-cursor[1]);
2044
2045                 co= (float)cos(phi0);
2046                 si= (float)sin(phi0);
2047                 loc[0]= -si*vec[1]+cursor[0];
2048                 loc[1]= co*vec[1]+cursor[1];
2049                 loc[2]= vec[2];
2050
2051                 Mat4MulVecfl(t->viewinv, loc);
2052                 VecSubf(loc, loc, t->viewinv[3]);
2053                 Mat3MulVecfl(td->smtx, loc);
2054
2055                 VecSubf(loc, loc, td->iloc);
2056                 VecMulf(loc, td->factor);
2057                 VecAddf(td->loc, td->iloc, loc);
2058         }
2059
2060         recalcData(t);
2061
2062         ED_area_headerprint(t->sa, str);
2063
2064         return 1;
2065 }
2066
2067 /* ************************** SHEAR *************************** */
2068
2069 void initShear(TransInfo *t)
2070 {
2071         t->mode = TFM_SHEAR;
2072         t->transform = Shear;
2073         t->handleEvent = handleEventShear;
2074
2075         initMouseInputMode(t, &t->mouse, INPUT_HORIZONTAL_ABSOLUTE);
2076
2077         t->idx_max = 0;
2078         t->num.idx_max = 0;
2079         t->snap[0] = 0.0f;
2080         t->snap[1] = 0.1f;
2081         t->snap[2] = t->snap[1] * 0.1f;
2082
2083         t->flag |= T_NO_CONSTRAINT;
2084 }
2085
2086 int handleEventShear(TransInfo *t, wmEvent *event)
2087 {
2088         int status = 0;
2089
2090         if (event->type == MIDDLEMOUSE && event->val==KM_PRESS)
2091         {
2092                 // Use customData pointer to signal Shear direction
2093                 if      (t->customData == 0)
2094                 {
2095                         initMouseInputMode(t, &t->mouse, INPUT_VERTICAL_ABSOLUTE);
2096                         t->customData = (void*)1;
2097                 }
2098                 else
2099                 {
2100                         initMouseInputMode(t, &t->mouse, INPUT_HORIZONTAL_ABSOLUTE);
2101                         t->customData = 0;
2102                 }
2103
2104                 status = 1;
2105         }
2106
2107         return status;
2108 }
2109
2110
2111 int Shear(TransInfo *t, short mval[2])
2112 {
2113         TransData *td = t->data;
2114         float vec[3];
2115         float smat[3][3], tmat[3][3], totmat[3][3], persmat[3][3], persinv[3][3];
2116         float value;
2117         int i;
2118         char str[50];
2119
2120         Mat3CpyMat4(persmat, t->viewmat);
2121         Mat3Inv(persinv, persmat);
2122
2123         value = 0.05f * t->values[0];
2124
2125         snapGrid(t, &value);
2126
2127         applyNumInput(&t->num, &value);
2128
2129         /* header print for NumInput */
2130         if (hasNumInput(&t->num)) {
2131                 char c[20];
2132
2133                 outputNumInput(&(t->num), c);
2134
2135                 sprintf(str, "Shear: %s %s", c, t->proptext);
2136         }
2137         else {
2138                 /* default header print */
2139                 sprintf(str, "Shear: %.3f %s", value, t->proptext);
2140         }
2141
2142         Mat3One(smat);
2143
2144         // Custom data signals shear direction
2145         if (t->customData == 0)
2146                 smat[1][0] = value;
2147         else
2148                 smat[0][1] = value;
2149
2150         Mat3MulMat3(tmat, smat, persmat);
2151         Mat3MulMat3(totmat, persinv, tmat);
2152
2153         for(i = 0 ; i < t->total; i++, td++) {
2154                 if (td->flag & TD_NOACTION)
2155                         break;
2156
2157                 if (td->flag & TD_SKIP)
2158                         continue;
2159
2160                 if (t->obedit) {
2161                         float mat3[3][3];
2162                         Mat3MulMat3(mat3, totmat, td->mtx);
2163                         Mat3MulMat3(tmat, td->smtx, mat3);
2164                 }
2165                 else {
2166                         Mat3CpyMat3(tmat, totmat);
2167                 }
2168                 VecSubf(vec, td->center, t->center);
2169
2170                 Mat3MulVecfl(tmat, vec);
2171
2172                 VecAddf(vec, vec, t->center);
2173                 VecSubf(vec, vec, td->center);
2174
2175                 VecMulf(vec, td->factor);
2176
2177                 VecAddf(td->loc, td->iloc, vec);
2178         }
2179
2180         recalcData(t);
2181
2182         ED_area_headerprint(t->sa, str);
2183
2184         return 1;
2185 }
2186
2187 /* ************************** RESIZE *************************** */
2188
2189 void initResize(TransInfo *t)
2190 {
2191         t->mode = TFM_RESIZE;
2192         t->transform = Resize;
2193
2194         initMouseInputMode(t, &t->mouse, INPUT_SPRING_FLIP);
2195
2196         t->flag |= T_NULL_ONE;
2197         t->num.flag |= NUM_NULL_ONE;
2198         t->num.flag |= NUM_AFFECT_ALL;
2199         if (!t->obedit) {
2200                 t->flag |= T_NO_ZERO;
2201                 t->num.flag |= NUM_NO_ZERO;
2202         }
2203
2204         t->idx_max = 2;
2205         t->num.idx_max = 2;
2206         t->snap[0] = 0.0f;
2207         t->snap[1] = 0.1f;
2208         t->snap[2] = t->snap[1] * 0.1f;
2209 }
2210
2211 static void headerResize(TransInfo *t, float vec[3], char *str) {
2212         char tvec[60];
2213         if (hasNumInput(&t->num)) {
2214                 outputNumInput(&(t->num), tvec);
2215         }
2216         else {
2217                 sprintf(&tvec[0], "%.4f", vec[0]);
2218                 sprintf(&tvec[20], "%.4f", vec[1]);
2219                 sprintf(&tvec[40], "%.4f", vec[2]);
2220         }
2221
2222         if (t->con.mode & CON_APPLY) {
2223                 switch(t->num.idx_max) {
2224                 case 0:
2225                         sprintf(str, "Scale: %s%s %s", &tvec[0], t->con.text, t->proptext);
2226                         break;
2227                 case 1:
2228                         sprintf(str, "Scale: %s : %s%s %s", &tvec[0], &tvec[20], t->con.text, t->proptext);
2229                         break;
2230                 case 2:
2231                         sprintf(str, "Scale: %s : %s : %s%s %s", &tvec[0], &tvec[20], &tvec[40], t->con.text, t->proptext);
2232                 }
2233         }
2234         else {
2235                 if (t->flag & T_2D_EDIT)
2236                         sprintf(str, "Scale X: %s   Y: %s%s %s", &tvec[0], &tvec[20], t->con.text, t->proptext);
2237                 else
2238                         sprintf(str, "Scale X: %s   Y: %s  Z: %s%s %s", &tvec[0], &tvec[20], &tvec[40], t->con.text, t->proptext);
2239         }
2240 }
2241
2242 #define SIGN(a)         (a<-FLT_EPSILON?1:a>FLT_EPSILON?2:3)
2243 #define VECSIGNFLIP(a, b) ((SIGN(a[0]) & SIGN(b[0]))==0 || (SIGN(a[1]) & SIGN(b[1]))==0 || (SIGN(a[2]) & SIGN(b[2]))==0)
2244
2245 /* smat is reference matrix, only scaled */
2246 static void TransMat3ToSize( float mat[][3], float smat[][3], float *size)
2247 {
2248         float vec[3];
2249
2250         VecCopyf(vec, mat[0]);
2251         size[0]= Normalize(vec);
2252         VecCopyf(vec, mat[1]);
2253         size[1]= Normalize(vec);
2254         VecCopyf(vec, mat[2]);
2255         size[2]= Normalize(vec);
2256
2257         /* first tried with dotproduct... but the sign flip is crucial */
2258         if( VECSIGNFLIP(mat[0], smat[0]) ) size[0]= -size[0];
2259         if( VECSIGNFLIP(mat[1], smat[1]) ) size[1]= -size[1];
2260         if( VECSIGNFLIP(mat[2], smat[2]) ) size[2]= -size[2];
2261 }
2262
2263
2264 static void ElementResize(TransInfo *t, TransData *td, float mat[3][3]) {
2265         float tmat[3][3], smat[3][3], center[3];
2266         float vec[3];
2267
2268         if (t->flag & T_EDIT) {
2269                 Mat3MulMat3(smat, mat, td->mtx);
2270                 Mat3MulMat3(tmat, td->smtx, smat);
2271         }
2272         else {
2273                 Mat3CpyMat3(tmat, mat);
2274         }
2275
2276         if (t->con.applySize) {
2277                 t->con.applySize(t, td, tmat);
2278         }
2279
2280         /* local constraint shouldn't alter center */
2281         if (t->around == V3D_LOCAL) {
2282                 if (t->flag & T_OBJECT) {
2283                         VECCOPY(center, td->center);
2284                 }
2285                 else if (t->flag & T_EDIT) {
2286
2287                         if(t->around==V3D_LOCAL && (t->settings->selectmode & SCE_SELECT_FACE)) {
2288                                 VECCOPY(center, td->center);
2289                         }
2290                         else {
2291                                 VECCOPY(center, t->center);
2292                         }
2293                 }
2294                 else {
2295                         VECCOPY(center, t->center);
2296                 }
2297         }
2298         else {
2299                 VECCOPY(center, t->center);
2300         }
2301
2302         if (td->ext) {
2303                 float fsize[3];
2304
2305                 if (t->flag & (T_OBJECT|T_TEXTURE|T_POSE)) {
2306                         float obsizemat[3][3];
2307                         // Reorient the size mat to fit the oriented object.
2308                         Mat3MulMat3(obsizemat, tmat, td->axismtx);
2309                         //printmatrix3("obsizemat", obsizemat);
2310                         TransMat3ToSize(obsizemat, td->axismtx, fsize);
2311                         //printvecf("fsize", fsize);
2312                 }
2313                 else {
2314                         Mat3ToSize(tmat, fsize);
2315                 }
2316
2317                 protectedSizeBits(td->protectflag, fsize);
2318
2319                 if ((t->flag & T_V3D_ALIGN)==0) {       // align mode doesn't resize objects itself
2320                         /* handle ipokeys? */
2321                         if(td->tdi) {
2322                                 TransDataIpokey *tdi= td->tdi;
2323                                 /* calculate delta size (equal for size and dsize) */
2324
2325                                 vec[0]= (tdi->oldsize[0])*(fsize[0] -1.0f) * td->factor;
2326                                 vec[1]= (tdi->oldsize[1])*(fsize[1] -1.0f) * td->factor;
2327                                 vec[2]= (tdi->oldsize[2])*(fsize[2] -1.0f) * td->factor;
2328
2329                                 add_tdi_poin(tdi->sizex, tdi->oldsize,   vec[0]);
2330                                 add_tdi_poin(tdi->sizey, tdi->oldsize+1, vec[1]);
2331                                 add_tdi_poin(tdi->sizez, tdi->oldsize+2, vec[2]);
2332
2333                         }
2334                         else if((td->flag & TD_SINGLESIZE) && !(t->con.mode & CON_APPLY)){
2335                                 /* scale val and reset size */
2336                                 *td->val = td->ival * fsize[0] * td->factor;
2337
2338                                 td->ext->size[0] = td->ext->isize[0];
2339                                 td->ext->size[1] = td->ext->isize[1];
2340                                 td->ext->size[2] = td->ext->isize[2];
2341                         }
2342                         else {
2343                                 /* Reset val if SINGLESIZE but using a constraint */
2344                                 if (td->flag & TD_SINGLESIZE)
2345                                         *td->val = td->ival;
2346
2347                                 td->ext->size[0] = td->ext->isize[0] * (fsize[0]) * td->factor;
2348                                 td->ext->size[1] = td->ext->isize[1] * (fsize[1]) * td->factor;
2349                                 td->ext->size[2] = td->ext->isize[2] * (fsize[2]) * td->factor;
2350                         }
2351                 }
2352
2353                 constraintSizeLim(t, td);
2354         }
2355
2356         /* For individual element center, Editmode need to use iloc */
2357         if (t->flag & T_POINTS)
2358                 VecSubf(vec, td->iloc, center);
2359         else
2360                 VecSubf(vec, td->center, center);
2361
2362         Mat3MulVecfl(tmat, vec);
2363
2364         VecAddf(vec, vec, center);
2365         if (t->flag & T_POINTS)
2366                 VecSubf(vec, vec, td->iloc);
2367         else
2368                 VecSubf(vec, vec, td->center);
2369
2370         VecMulf(vec, td->factor);
2371
2372         if (t->flag & (T_OBJECT|T_POSE)) {
2373                 Mat3MulVecfl(td->smtx, vec);
2374         }
2375
2376         protectedTransBits(td->protectflag, vec);
2377
2378         if(td->tdi) {
2379                 TransDataIpokey *tdi= td->tdi;
2380                 add_tdi_poin(tdi->locx, tdi->oldloc, vec[0]);
2381                 add_tdi_poin(tdi->locy, tdi->oldloc+1, vec[1]);
2382                 add_tdi_poin(tdi->locz, tdi->oldloc+2, vec[2]);
2383         }
2384         else VecAddf(td->loc, td->iloc, vec);
2385
2386         constraintTransLim(t, td);
2387 }
2388
2389 int Resize(TransInfo *t, short mval[2])
2390 {
2391         TransData *td;
2392         float size[3], mat[3][3];
2393         float ratio;
2394         int i;
2395         char str[200];
2396
2397         /* for manipulator, center handle, the scaling can't be done relative to center */
2398         if( (t->flag & T_USES_MANIPULATOR) && t->con.mode==0)
2399         {
2400                 ratio = 1.0f - ((t->imval[0] - mval[0]) + (t->imval[1] - mval[1]))/100.0f;
2401         }
2402         else
2403         {
2404                 ratio = t->values[0];
2405         }
2406
2407         size[0] = size[1] = size[2] = ratio;
2408
2409         snapGrid(t, size);
2410
2411         if (hasNumInput(&t->num)) {
2412                 applyNumInput(&t->num, size);
2413                 constraintNumInput(t, size);
2414         }
2415
2416         applySnapping(t, size);
2417
2418         if (t->flag & T_AUTOVALUES)
2419         {
2420                 VECCOPY(size, t->auto_values);
2421         }
2422
2423         VECCOPY(t->values, size);
2424
2425         SizeToMat3(size, mat);
2426
2427         if (t->con.applySize) {
2428                 t->con.applySize(t, NULL, mat);
2429         }
2430
2431         Mat3CpyMat3(t->mat, mat);       // used in manipulator
2432
2433         headerResize(t, size, str);
2434
2435         for(i = 0, td=t->data; i < t->total; i++, td++) {
2436                 if (td->flag & TD_NOACTION)
2437                         break;
2438
2439                 if (td->flag & TD_SKIP)
2440                         continue;
2441
2442                 ElementResize(t, td, mat);
2443         }
2444
2445         /* evil hack - redo resize if cliping needed */
2446         if (t->flag & T_CLIP_UV && clipUVTransform(t, size, 1)) {
2447                 SizeToMat3(size, mat);
2448
2449                 if (t->con.applySize)
2450                         t->con.applySize(t, NULL, mat);
2451
2452                 for(i = 0, td=t->data; i < t->total; i++, td++)
2453                         ElementResize(t, td, mat);
2454         }
2455
2456         recalcData(t);
2457
2458         ED_area_headerprint(t->sa, str);
2459
2460         return 1;
2461 }
2462
2463 /* ************************** TOSPHERE *************************** */
2464
2465 void initToSphere(TransInfo *t)
2466 {
2467         TransData *td = t->data;
2468         int i;
2469
2470         t->mode = TFM_TOSPHERE;
2471         t->transform = ToSphere;
2472
2473         initMouseInputMode(t, &t->mouse, INPUT_HORIZONTAL_RATIO);
2474
2475         t->idx_max = 0;
2476         t->num.idx_max = 0;
2477         t->snap[0] = 0.0f;
2478         t->snap[1] = 0.1f;
2479         t->snap[2] = t->snap[1] * 0.1f;
2480
2481         t->num.flag |= NUM_NULL_ONE | NUM_NO_NEGATIVE;
2482         t->flag |= T_NO_CONSTRAINT;
2483
2484         // Calculate average radius
2485         for(i = 0 ; i < t->total; i++, td++) {
2486                 t->val += VecLenf(t->center, td->iloc);
2487         }
2488
2489         t->val /= (float)t->total;
2490 }
2491
2492 int ToSphere(TransInfo *t, short mval[2])
2493 {
2494         float vec[3];
2495         float ratio, radius;
2496         int i;
2497         char str[64];
2498         TransData *td = t->data;
2499
2500         ratio = t->values[0];
2501
2502         snapGrid(t, &ratio);
2503
2504         applyNumInput(&t->num, &ratio);
2505
2506         if (ratio < 0)
2507                 ratio = 0.0f;
2508         else if (ratio > 1)
2509                 ratio = 1.0f;
2510
2511         /* header print for NumInput */
2512         if (hasNumInput(&t->num)) {
2513                 char c[20];
2514
2515                 outputNumInput(&(t->num), c);
2516
2517                 sprintf(str, "To Sphere: %s %s", c, t->proptext);
2518         }
2519         else {
2520                 /* default header print */
2521                 sprintf(str, "To Sphere: %.4f %s", ratio, t->proptext);
2522         }
2523
2524
2525         for(i = 0 ; i < t->total; i++, td++) {
2526                 float tratio;
2527                 if (td->flag & TD_NOACTION)
2528                         break;
2529
2530                 if (td->flag & TD_SKIP)
2531                         continue;
2532
2533                 VecSubf(vec, td->iloc, t->center);
2534
2535                 radius = Normalize(vec);
2536
2537                 tratio = ratio * td->factor;
2538
2539                 VecMulf(vec, radius * (1.0f - tratio) + t->val * tratio);
2540
2541                 VecAddf(td->loc, t->center, vec);
2542         }
2543
2544
2545         recalcData(t);
2546
2547         ED_area_headerprint(t->sa, str);
2548
2549         return 1;
2550 }
2551
2552 /* ************************** ROTATION *************************** */
2553
2554
2555 void initRotation(TransInfo *t)
2556 {
2557         t->mode = TFM_ROTATION;
2558         t->transform = Rotation;
2559
2560         initMouseInputMode(t, &t->mouse, INPUT_ANGLE);
2561
2562         t->ndof.axis = 16;
2563         /* Scale down and flip input for rotation */
2564         t->ndof.factor[0] = -0.2f;
2565
2566         t->idx_max = 0;
2567         t->num.idx_max = 0;
2568         t->snap[0] = 0.0f;
2569         t->snap[1] = (float)((5.0/180)*M_PI);
2570         t->snap[2] = t->snap[1] * 0.2f;
2571
2572         if (t->flag & T_2D_EDIT)
2573                 t->flag |= T_NO_CONSTRAINT;
2574 }
2575
2576 static void ElementRotation(TransInfo *t, TransData *td, float mat[3][3], short around) {
2577         float vec[3], totmat[3][3], smat[3][3];
2578         float eul[3], fmat[3][3], quat[4];
2579         float *center = t->center;
2580
2581         /* local constraint shouldn't alter center */
2582         if (around == V3D_LOCAL) {
2583                 if (t->flag & (T_OBJECT|T_POSE)) {
2584                         center = td->center;
2585                 }
2586                 else {
2587                         /* !TODO! Make this if not rely on G */
2588                         if(around==V3D_LOCAL && (t->settings->selectmode & SCE_SELECT_FACE)) {
2589                                 center = td->center;
2590                         }
2591                 }
2592         }
2593
2594         if (t->flag & T_POINTS) {
2595                 Mat3MulMat3(totmat, mat, td->mtx);
2596                 Mat3MulMat3(smat, td->smtx, totmat);
2597
2598                 VecSubf(vec, td->iloc, center);
2599                 Mat3MulVecfl(smat, vec);
2600
2601                 VecAddf(td->loc, vec, center);
2602
2603                 VecSubf(vec,td->loc,td->iloc);
2604                 protectedTransBits(td->protectflag, vec);
2605                 VecAddf(td->loc, td->iloc, vec);
2606
2607                 if(td->flag & TD_USEQUAT) {
2608                         Mat3MulSerie(fmat, td->mtx, mat, td->smtx, 0, 0, 0, 0, 0);
2609                         Mat3ToQuat(fmat, quat); // Actual transform
2610
2611                         if(td->ext->quat){
2612                                 QuatMul(td->ext->quat, quat, td->ext->iquat);
2613
2614                                 /* is there a reason not to have this here? -jahka */
2615                                 protectedQuaternionBits(td->protectflag, td->ext->quat, td->ext->iquat);
2616                         }
2617                 }
2618         }
2619         /**
2620          * HACK WARNING
2621          *
2622          * This is some VERY ugly special case to deal with pose mode.
2623          *
2624          * The problem is that mtx and smtx include each bone orientation.
2625          *
2626          * That is needed to rotate each bone properly, HOWEVER, to calculate
2627          * the translation component, we only need the actual armature object's
2628          * matrix (and inverse). That is not all though. Once the proper translation
2629          * has been computed, it has to be converted back into the bone's space.
2630          */
2631         else if (t->flag & T_POSE) {
2632                 float pmtx[3][3], imtx[3][3];
2633
2634                 // Extract and invert armature object matrix
2635                 Mat3CpyMat4(pmtx, t->poseobj->obmat);
2636                 Mat3Inv(imtx, pmtx);
2637
2638                 if ((td->flag & TD_NO_LOC) == 0)
2639                 {
2640                         VecSubf(vec, td->center, center);
2641
2642                         Mat3MulVecfl(pmtx, vec);        // To Global space
2643                         Mat3MulVecfl(mat, vec);         // Applying rotation
2644                         Mat3MulVecfl(imtx, vec);        // To Local space
2645
2646                         VecAddf(vec, vec, center);
2647                         /* vec now is the location where the object has to be */
2648
2649                         VecSubf(vec, vec, td->center); // Translation needed from the initial location
2650
2651                         Mat3MulVecfl(pmtx, vec);        // To Global space
2652                         Mat3MulVecfl(td->smtx, vec);// To Pose space
2653
2654                         protectedTransBits(td->protectflag, vec);
2655
2656                         VecAddf(td->loc, td->iloc, vec);
2657
2658                         constraintTransLim(t, td);
2659                 }
2660
2661                 /* rotation */
2662                 if ((t->flag & T_V3D_ALIGN)==0) { // align mode doesn't rotate objects itself
2663                         /* euler or quaternion? */
2664                         if (td->flag & TD_USEQUAT) {
2665                                 Mat3MulSerie(fmat, td->mtx, mat, td->smtx, 0, 0, 0, 0, 0);
2666
2667                                 Mat3ToQuat(fmat, quat); // Actual transform
2668
2669                                 QuatMul(td->ext->quat, quat, td->ext->iquat);
2670                                 /* this function works on end result */
2671                                 protectedQuaternionBits(td->protectflag, td->ext->quat, td->ext->iquat);
2672                         }
2673                         else {
2674                                 float eulmat[3][3];
2675
2676                                 Mat3MulMat3(totmat, mat, td->mtx);
2677                                 Mat3MulMat3(smat, td->smtx, totmat);
2678
2679                                 /* calculate the total rotatation in eulers */
2680                                 VECCOPY(eul, td->ext->irot);
2681                                 EulOToMat3(eul, td->rotOrder, eulmat);
2682
2683                                 /* mat = transform, obmat = bone rotation */
2684                                 Mat3MulMat3(fmat, smat, eulmat);
2685
2686                                 Mat3ToCompatibleEulO(fmat, eul, td->ext->rot, td->rotOrder);
2687
2688                                 /* and apply (to end result only) */
2689                                 protectedRotateBits(td->protectflag, eul, td->ext->irot);
2690                                 VECCOPY(td->ext->rot, eul);
2691                         }
2692
2693                         constraintRotLim(t, td);
2694                 }
2695         }
2696         else {
2697                 if ((td->flag & TD_NO_LOC) == 0)
2698                 {
2699                         /* translation */
2700                         VecSubf(vec, td->center, center);
2701                         Mat3MulVecfl(mat, vec);
2702                         VecAddf(vec, vec, center);
2703                         /* vec now is the location where the object has to be */
2704                         VecSubf(vec, vec, td->center);
2705                         Mat3MulVecfl(td->smtx, vec);
2706
2707                         protectedTransBits(td->protectflag, vec);
2708
2709                         if(td->tdi) {
2710                                 TransDataIpokey *tdi= td->tdi;
2711                                 add_tdi_poin(tdi->locx, tdi->oldloc, vec[0]);
2712                                 add_tdi_poin(tdi->locy, tdi->oldloc+1, vec[1]);
2713                                 add_tdi_poin(tdi->locz, tdi->oldloc+2, vec[2]);
2714                         }
2715                         else VecAddf(td->loc, td->iloc, vec);
2716                 }
2717
2718
2719                 constraintTransLim(t, td);
2720
2721                 /* rotation */
2722                 if ((t->flag & T_V3D_ALIGN)==0) { // align mode doesn't rotate objects itself
2723                         /* euler or quaternion? */
2724                     if (td->flag & TD_USEQUAT) {
2725                                 Mat3MulSerie(fmat, td->mtx, mat, td->smtx, 0, 0, 0, 0, 0);
2726                                 Mat3ToQuat(fmat, quat); // Actual transform
2727
2728                                 QuatMul(td->ext->quat, quat, td->ext->iquat);
2729                                 /* this function works on end result */
2730                                 protectedQuaternionBits(td->protectflag, td->ext->quat, td->ext->iquat);
2731                         }
2732                         else {
2733                                 float obmat[3][3];
2734
2735                                 /* are there ipo keys? */
2736                                 if(td->tdi) {
2737                                         TransDataIpokey *tdi= td->tdi;
2738                                         float current_rot[3];
2739                                         float rot[3];
2740
2741                                         /* current IPO value for compatible euler */
2742                                         current_rot[0] = (tdi->rotx) ? tdi->rotx[0] : 0.0f;
2743                                         current_rot[1] = (tdi->roty) ? tdi->roty[0] : 0.0f;
2744                                         current_rot[2] = (tdi->rotz) ? tdi->rotz[0] : 0.0f;
2745                                         VecMulf(current_rot, (float)(M_PI_2 / 9.0));
2746
2747                                         /* calculate the total rotatation in eulers */
2748                                         VecAddf(eul, td->ext->irot, td->ext->drot);
2749                                         EulToMat3(eul, obmat);
2750                                         /* mat = transform, obmat = object rotation */
2751                                         Mat3MulMat3(fmat, mat, obmat);
2752
2753                                         Mat3ToCompatibleEul(fmat, eul, current_rot);
2754
2755                                         /* correct back for delta rot */
2756                                         if(tdi->flag & TOB_IPODROT) {
2757                                                 VecSubf(rot, eul, td->ext->irot);
2758                                         }
2759                                         else {
2760                                                 VecSubf(rot, eul, td->ext->drot);
2761                                         }
2762
2763                                         VecMulf(rot, (float)(9.0/M_PI_2));
2764                                         VecSubf(rot, rot, tdi->oldrot);
2765
2766                                         protectedRotateBits(td->protectflag, rot, tdi->oldrot);
2767
2768                                         add_tdi_poin(tdi->rotx, tdi->oldrot, rot[0]);
2769                                         add_tdi_poin(tdi->roty, tdi->oldrot+1, rot[1]);
2770                                         add_tdi_poin(tdi->rotz, tdi->oldrot+2, rot[2]);
2771                                 }
2772                                 else {
2773                                         Mat3MulMat3(totmat, mat, td->mtx);
2774                                         Mat3MulMat3(smat, td->smtx, totmat);
2775
2776                                         /* calculate the total rotatation in eulers */
2777                                         VecAddf(eul, td->ext->irot, td->ext->drot); /* we have to correct for delta rot */
2778                                         EulToMat3(eul, obmat);
2779                                         /* mat = transform, obmat = object rotation */
2780                                         Mat3MulMat3(fmat, smat, obmat);
2781
2782                                         Mat3ToCompatibleEul(fmat, eul, td->ext->rot);
2783
2784                                         /* correct back for delta rot */
2785                                         VecSubf(eul, eul, td->ext->drot);
2786
2787                                         /* and apply */
2788                                         protectedRotateBits(td->protectflag, eul, td->ext->irot);
2789                                         VECCOPY(td->ext->rot, eul);
2790                                 }
2791                         }
2792
2793                         constraintRotLim(t, td);
2794                 }
2795         }
2796 }
2797
2798 static void applyRotation(TransInfo *t, float angle, float axis[3])
2799 {
2800         TransData *td = t->data;
2801         float mat[3][3];
2802         int i;
2803
2804         VecRotToMat3(axis, angle, mat);
2805
2806         for(i = 0 ; i < t->total; i++, td++) {
2807
2808                 if (td->flag & TD_NOACTION)
2809                         break;
2810
2811                 if (td->flag & TD_SKIP)
2812                         continue;
2813
2814                 if (t->con.applyRot) {
2815                         t->con.applyRot(t, td, axis, NULL);
2816                         VecRotToMat3(axis, angle * td->factor, mat);
2817                 }
2818                 else if (t->flag & T_PROP_EDIT) {
2819                         VecRotToMat3(axis, angle * td->factor, mat);
2820                 }
2821
2822                 ElementRotation(t, td, mat, t->around);
2823         }
2824 }
2825
2826 int Rotation(TransInfo *t, short mval[2])
2827 {
2828         char str[64];
2829
2830         float final;
2831
2832         float axis[3];
2833         float mat[3][3];
2834
2835         VECCOPY(axis, t->viewinv[2]);
2836         VecMulf(axis, -1.0f);
2837         Normalize(axis);
2838
2839         final = t->values[0];
2840
2841         applyNDofInput(&t->ndof, &final);
2842
2843         snapGrid(t, &final);
2844
2845         if (t->con.applyRot) {
2846                 t->con.applyRot(t, NULL, axis, &final);
2847         }
2848
2849         applySnapping(t, &final);
2850
2851         if (hasNumInput(&t->num)) {
2852                 char c[20];
2853
2854                 applyNumInput(&t->num, &final);
2855
2856                 outputNumInput(&(t->num), c);
2857
2858                 sprintf(str, "Rot: %s %s %s", &c[0], t->con.text, t->proptext);
2859
2860                 /* Clamp between -180 and 180 */
2861                 while (final >= 180.0)
2862                         final -= 360.0;
2863
2864                 while (final <= -180.0)
2865                         final += 360.0;
2866
2867                 final *= (float)(M_PI / 180.0);
2868         }
2869         else {
2870                 sprintf(str, "Rot: %.2f%s %s", 180.0*final/M_PI, t->con.text, t->proptext);
2871         }
2872
2873         VecRotToMat3(axis, final, mat);
2874
2875         // TRANSFORM_FIX_ME
2876 //      t->values[0] = final;           // used in manipulator
2877 //      Mat3CpyMat3(t->mat, mat);       // used in manipulator
2878
2879         applyRotation(t, final, axis);
2880
2881         recalcData(t);
2882
2883         ED_area_headerprint(t->sa, str);
2884
2885         return 1;
2886 }
2887
2888
2889 /* ************************** TRACKBALL *************************** */
2890
2891 void initTrackball(TransInfo *t)
2892 {
2893         t->mode = TFM_TRACKBALL;
2894         t->transform = Trackball;
2895
2896         initMouseInputMode(t, &t->mouse, INPUT_TRACKBALL);
2897
2898         t->ndof.axis = 40;
2899         /* Scale down input for rotation */
2900         t->ndof.factor[0] = 0.2f;
2901         t->ndof.factor[1] = 0.2f;
2902
2903         t->idx_max = 1;
2904         t->num.idx_max = 1;
2905         t->snap[0] = 0.0f;
2906         t->snap[1] = (float)((5.0/180)*M_PI);
2907         t->snap[2] = t->snap[1] * 0.2f;
2908
2909         t->flag |= T_NO_CONSTRAINT;
2910 }
2911
2912 static void applyTrackball(TransInfo *t, float axis1[3], float axis2[3], float angles[2])
2913 {
2914         TransData *td = t->data;
2915         float mat[3][3], smat[3][3], totmat[3][3];
2916         int i;
2917
2918         VecRotToMat3(axis1, angles[0], smat);
2919         VecRotToMat3(axis2, angles[1], totmat);
2920
2921         Mat3MulMat3(mat, smat, totmat);
2922
2923         for(i = 0 ; i < t->total; i++, td++) {
2924                 if (td->flag & TD_NOACTION)
2925                         break;
2926
2927                 if (td->flag & TD_SKIP)
2928                         continue;
2929
2930                 if (t->flag & T_PROP_EDIT) {
2931                         VecRotToMat3(axis1, td->factor * angles[0], smat);
2932                         VecRotToMat3(axis2, td->factor * angles[1], totmat);
2933
2934                         Mat3MulMat3(mat, smat, totmat);
2935                 }
2936
2937                 ElementRotation(t, td, mat, t->around);
2938         }
2939 }
2940
2941 int Trackball(TransInfo *t, short mval[2])
2942 {
2943         char str[128];
2944         float axis1[3], axis2[3];
2945         float mat[3][3], totmat[3][3], smat[3][3];
2946         float phi[2];
2947
2948         VECCOPY(axis1, t->persinv[0]);
2949         VECCOPY(axis2, t->persinv[1]);
2950         Normalize(axis1);
2951         Normalize(axis2);
2952
2953         phi[0] = t->values[0];
2954         phi[1] = t->values[1];
2955
2956         applyNDofInput(&t->ndof, phi);
2957
2958         snapGrid(t, phi);
2959
2960         if (hasNumInput(&t->num)) {
2961                 char c[40];
2962
2963                 applyNumInput(&t->num, phi);
2964
2965                 outputNumInput(&(t->num), c);
2966
2967                 sprintf(str, "Trackball: %s %s %s", &c[0], &c[20], t->proptext);
2968
2969                 phi[0] *= (float)(M_PI / 180.0);
2970                 phi[1] *= (float)(M_PI / 180.0);
2971         }
2972         else {
2973                 sprintf(str, "Trackball: %.2f %.2f %s", 180.0*phi[0]/M_PI, 180.0*phi[1]/M_PI, t->proptext);
2974         }
2975
2976         VecRotToMat3(axis1, phi[0], smat);
2977         VecRotToMat3(axis2, phi[1], totmat);
2978
2979         Mat3MulMat3(mat, smat, totmat);