Merge with -r 22620:23107.
[blender.git] / 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_GEOM|ND_DATA, t->obedit->data);
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 (RNA_struct_find_property(op->ptr, "value"))
1261         {
1262                 if (t->flag & T_AUTOVALUES)
1263                 {
1264                         RNA_float_set_array(op->ptr, "value", t->auto_values);
1265                 }
1266                 else
1267                 {
1268                         RNA_float_set_array(op->ptr, "value", t->values);
1269                 }
1270         }
1271
1272         /* XXX convert stupid flag to enum */
1273         switch(t->flag & (T_PROP_EDIT|T_PROP_CONNECTED))
1274         {
1275         case (T_PROP_EDIT|T_PROP_CONNECTED):
1276                 proportional = 2;
1277                 break;
1278         case T_PROP_EDIT:
1279                 proportional = 1;
1280                 break;
1281         default:
1282                 proportional = 0;
1283         }
1284
1285         if (RNA_struct_find_property(op->ptr, "proportional"))
1286         {
1287                 RNA_enum_set(op->ptr, "proportional", proportional);
1288                 RNA_enum_set(op->ptr, "proportional_editing_falloff", t->prop_mode);
1289                 RNA_float_set(op->ptr, "proportional_size", t->prop_size);
1290         }
1291
1292         if (RNA_struct_find_property(op->ptr, "mirror"))
1293         {
1294                 RNA_boolean_set(op->ptr, "mirror", t->flag & T_MIRROR);
1295         }
1296
1297         if (RNA_struct_find_property(op->ptr, "constraint_axis"))
1298         {
1299                 RNA_enum_set(op->ptr, "constraint_orientation", t->current_orientation);
1300
1301                 if (t->con.mode & CON_APPLY)
1302                 {
1303                         if (t->con.mode & CON_AXIS0) {
1304                                 constraint_axis[0] = 1;
1305                         }
1306                         if (t->con.mode & CON_AXIS1) {
1307                                 constraint_axis[1] = 1;
1308                         }
1309                         if (t->con.mode & CON_AXIS2) {
1310                                 constraint_axis[2] = 1;
1311                         }
1312                 }
1313
1314                 RNA_boolean_set_array(op->ptr, "constraint_axis", constraint_axis);
1315         }
1316
1317         // XXX If modal, save settings back in scene
1318         if (t->flag & T_MODAL)
1319         {
1320                 ts->prop_mode = t->prop_mode;
1321                 ts->proportional = proportional;
1322
1323                 if(t->spacetype == SPACE_VIEW3D)
1324                 {
1325                         View3D *v3d = t->view;
1326
1327                         v3d->twmode = t->current_orientation;
1328                 }
1329         }
1330 }
1331
1332 int initTransform(bContext *C, TransInfo *t, wmOperator *op, wmEvent *event, int mode)
1333 {
1334         int options = 0;
1335
1336         /* added initialize, for external calls to set stuff in TransInfo, like undo string */
1337
1338         t->state = TRANS_RUNNING;
1339
1340         t->options = options;
1341
1342         t->mode = mode;
1343
1344         if (!initTransInfo(C, t, op, event))                                    // internal data, mouse, vectors
1345         {
1346                 return 0;
1347         }
1348
1349         if(t->spacetype == SPACE_VIEW3D)
1350         {
1351                 //calc_manipulator_stats(curarea);
1352                 initTransformOrientation(C, t);
1353
1354                 t->draw_handle = ED_region_draw_cb_activate(t->ar->type, drawTransform, t, REGION_DRAW_POST);
1355         }
1356         else if(t->spacetype == SPACE_IMAGE) {
1357                 Mat3One(t->spacemtx);
1358                 t->draw_handle = ED_region_draw_cb_activate(t->ar->type, drawTransform, t, REGION_DRAW_POST);
1359         }
1360         else
1361                 Mat3One(t->spacemtx);
1362
1363         createTransData(C, t);                  // make TransData structs from selection
1364
1365         if (t->total == 0) {
1366                 postTrans(t);
1367                 return 0;
1368         }
1369
1370         initSnapping(t, op); // Initialize snapping data AFTER mode flags
1371
1372         /* EVIL! posemode code can switch translation to rotate when 1 bone is selected. will be removed (ton) */
1373         /* EVIL2: we gave as argument also texture space context bit... was cleared */
1374         /* EVIL3: extend mode for animation editors also switches modes... but is best way to avoid duplicate code */
1375         mode = t->mode;
1376
1377         calculatePropRatio(t);
1378         calculateCenter(t);
1379
1380         initMouseInput(t, &t->mouse, t->center2d, t->imval);
1381
1382         switch (mode) {
1383         case TFM_TRANSLATION:
1384                 initTranslation(t);
1385                 break;
1386         case TFM_ROTATION:
1387                 initRotation(t);
1388                 break;
1389         case TFM_RESIZE:
1390                 initResize(t);
1391                 break;
1392         case TFM_TOSPHERE:
1393                 initToSphere(t);
1394                 break;
1395         case TFM_SHEAR:
1396                 initShear(t);
1397                 break;
1398         case TFM_WARP:
1399                 initWarp(t);
1400                 break;
1401         case TFM_SHRINKFATTEN:
1402                 initShrinkFatten(t);
1403                 break;
1404         case TFM_TILT:
1405                 initTilt(t);
1406                 break;
1407         case TFM_CURVE_SHRINKFATTEN:
1408                 initCurveShrinkFatten(t);
1409                 break;
1410         case TFM_TRACKBALL:
1411                 initTrackball(t);
1412                 break;
1413         case TFM_PUSHPULL:
1414                 initPushPull(t);
1415                 break;
1416         case TFM_CREASE:
1417                 initCrease(t);
1418                 break;
1419         case TFM_BONESIZE:
1420                 {       /* used for both B-Bone width (bonesize) as for deform-dist (envelope) */
1421                         bArmature *arm= t->poseobj->data;
1422                         if(arm->drawtype==ARM_ENVELOPE)
1423                                 initBoneEnvelope(t);
1424                         else
1425                                 initBoneSize(t);
1426                 }
1427                 break;
1428         case TFM_BONE_ENVELOPE:
1429                 initBoneEnvelope(t);
1430                 break;
1431         case TFM_BONE_ROLL:
1432                 initBoneRoll(t);
1433                 break;
1434         case TFM_TIME_TRANSLATE:
1435                 initTimeTranslate(t);
1436                 break;
1437         case TFM_TIME_SLIDE:
1438                 initTimeSlide(t);
1439                 break;
1440         case TFM_TIME_SCALE:
1441                 initTimeScale(t);
1442                 break;
1443         case TFM_TIME_EXTEND:
1444                 /* now that transdata has been made, do like for TFM_TIME_TRANSLATE (for most Animation
1445                  * Editors because they have only 1D transforms for time values) or TFM_TRANSLATION
1446                  * (for Graph/NLA Editors only since they uses 'standard' transforms to get 2D movement)
1447                  * depending on which editor this was called from
1448                  */
1449                 if ELEM(t->spacetype, SPACE_IPO, SPACE_NLA)
1450                         initTranslation(t);
1451                 else
1452                         initTimeTranslate(t);
1453                 break;
1454         case TFM_BAKE_TIME:
1455                 initBakeTime(t);
1456                 break;
1457         case TFM_MIRROR:
1458                 initMirror(t);
1459                 break;
1460         case TFM_BEVEL:
1461                 initBevel(t);
1462                 break;
1463         case TFM_BWEIGHT:
1464                 initBevelWeight(t);
1465                 break;
1466         case TFM_ALIGN:
1467                 initAlign(t);
1468                 break;
1469         }
1470
1471         /* overwrite initial values if operator supplied a non-null vector */
1472         if (RNA_property_is_set(op->ptr, "value"))
1473         {
1474                 float values[4];
1475                 RNA_float_get_array(op->ptr, "value", values);
1476                 QUATCOPY(t->values, values);
1477                 QUATCOPY(t->auto_values, values);
1478                 t->flag |= T_AUTOVALUES;
1479         }
1480
1481         /* Constraint init from operator */
1482         if (RNA_struct_find_property(op->ptr, "constraint_axis") && RNA_property_is_set(op->ptr, "constraint_axis"))
1483         {
1484                 int constraint_axis[3];
1485
1486                 RNA_boolean_get_array(op->ptr, "constraint_axis", constraint_axis);
1487
1488                 if (constraint_axis[0] || constraint_axis[1] || constraint_axis[2])
1489                 {
1490                         t->con.mode |= CON_APPLY;
1491
1492                         if (constraint_axis[0]) {
1493                                 t->con.mode |= CON_AXIS0;
1494                         }
1495                         if (constraint_axis[1]) {
1496                                 t->con.mode |= CON_AXIS1;
1497                         }
1498                         if (constraint_axis[2]) {
1499                                 t->con.mode |= CON_AXIS2;
1500                         }
1501
1502                         setUserConstraint(t, t->con.mode, "%s");
1503                 }
1504         }
1505
1506         return 1;
1507 }
1508
1509 void transformApply(bContext *C, TransInfo *t)
1510 {
1511         if (t->redraw)
1512         {
1513                 if (t->modifiers & MOD_CONSTRAINT_SELECT)
1514                         t->con.mode |= CON_SELECT;
1515
1516                 selectConstraint(t);
1517                 if (t->transform) {
1518                         t->transform(t, t->mval);  // calls recalcData()
1519                         viewRedrawForce(C, t);
1520                 }
1521                 t->redraw = 0;
1522         }
1523
1524         /* If auto confirm is on, break after one pass */
1525         if (t->options & CTX_AUTOCONFIRM)
1526         {
1527                 t->state = TRANS_CONFIRM;
1528         }
1529
1530         if (BKE_ptcache_get_continue_physics())
1531         {
1532                 // TRANSFORM_FIX_ME
1533                 //do_screenhandlers(G.curscreen);
1534                 t->redraw = 1;
1535         }
1536 }
1537
1538 int transformEnd(bContext *C, TransInfo *t)
1539 {
1540         int exit_code = OPERATOR_RUNNING_MODAL;
1541
1542         if (t->state != TRANS_RUNNING)
1543         {
1544                 /* handle restoring objects */
1545                 if(t->state == TRANS_CANCEL)
1546                 {
1547                         exit_code = OPERATOR_CANCELLED;
1548                         restoreTransObjects(t); // calls recalcData()
1549                 }
1550                 else
1551                 {
1552                         exit_code = OPERATOR_FINISHED;
1553                 }
1554
1555                 /* free data */
1556                 postTrans(t);
1557
1558                 /* aftertrans does insert keyframes, and clears base flags, doesnt read transdata */
1559                 special_aftertrans_update(t);
1560
1561                 /* send events out for redraws */
1562                 viewRedrawPost(t);
1563
1564                 /*  Undo as last, certainly after special_trans_update! */
1565
1566                 if(t->state == TRANS_CANCEL) {
1567 //                      if(t->undostr) ED_undo_push(C, t->undostr);
1568                 }
1569                 else {
1570 //                      if(t->undostr) ED_undo_push(C, t->undostr);
1571 //                      else ED_undo_push(C, transform_to_undostr(t));
1572                 }
1573                 t->undostr= NULL;
1574
1575                 viewRedrawForce(C, t);
1576         }
1577
1578         return exit_code;
1579 }
1580
1581 /* ************************** TRANSFORM LOCKS **************************** */
1582
1583 static void protectedTransBits(short protectflag, float *vec)
1584 {
1585         if(protectflag & OB_LOCK_LOCX)
1586                 vec[0]= 0.0f;
1587         if(protectflag & OB_LOCK_LOCY)
1588                 vec[1]= 0.0f;
1589         if(protectflag & OB_LOCK_LOCZ)
1590                 vec[2]= 0.0f;
1591 }
1592
1593 static void protectedSizeBits(short protectflag, float *size)
1594 {
1595         if(protectflag & OB_LOCK_SCALEX)
1596                 size[0]= 1.0f;
1597         if(protectflag & OB_LOCK_SCALEY)
1598                 size[1]= 1.0f;
1599         if(protectflag & OB_LOCK_SCALEZ)
1600                 size[2]= 1.0f;
1601 }
1602
1603 static void protectedRotateBits(short protectflag, float *eul, float *oldeul)
1604 {
1605         if(protectflag & OB_LOCK_ROTX)
1606                 eul[0]= oldeul[0];
1607         if(protectflag & OB_LOCK_ROTY)
1608                 eul[1]= oldeul[1];
1609         if(protectflag & OB_LOCK_ROTZ)
1610                 eul[2]= oldeul[2];
1611 }
1612
1613 static void protectedQuaternionBits(short protectflag, float *quat, float *oldquat)
1614 {
1615         /* quaternions get limited with euler... */
1616         /* this function only does the delta rotation */
1617
1618         if(protectflag) {
1619                 float eul[3], oldeul[3], quat1[4];
1620
1621                 QUATCOPY(quat1, quat);
1622                 QuatToEul(quat, eul);
1623                 QuatToEul(oldquat, oldeul);
1624
1625                 if(protectflag & OB_LOCK_ROTX)
1626                         eul[0]= oldeul[0];
1627                 if(protectflag & OB_LOCK_ROTY)
1628                         eul[1]= oldeul[1];
1629                 if(protectflag & OB_LOCK_ROTZ)
1630                         eul[2]= oldeul[2];
1631
1632                 EulToQuat(eul, quat);
1633                 /* quaternions flip w sign to accumulate rotations correctly */
1634                 if( (quat1[0]<0.0f && quat[0]>0.0f) || (quat1[0]>0.0f && quat[0]<0.0f) ) {
1635                         QuatMulf(quat, -1.0f);
1636                 }
1637         }
1638 }
1639
1640 /* ******************* TRANSFORM LIMITS ********************** */
1641
1642 static void constraintTransLim(TransInfo *t, TransData *td)
1643 {
1644         if (td->con) {
1645                 bConstraintTypeInfo *cti= get_constraint_typeinfo(CONSTRAINT_TYPE_LOCLIMIT);
1646                 bConstraintOb cob;
1647                 bConstraint *con;
1648
1649                 /* Make a temporary bConstraintOb for using these limit constraints
1650                  *      - they only care that cob->matrix is correctly set ;-)
1651                  *      - current space should be local
1652                  */
1653                 memset(&cob, 0, sizeof(bConstraintOb));
1654                 Mat4One(cob.matrix);
1655                 if (td->tdi) {
1656                         TransDataIpokey *tdi= td->tdi;
1657                         cob.matrix[3][0]= tdi->locx[0];
1658                         cob.matrix[3][1]= tdi->locy[0];
1659                         cob.matrix[3][2]= tdi->locz[0];
1660                 }
1661                 else {
1662                         VECCOPY(cob.matrix[3], td->loc);
1663                 }
1664
1665                 /* Evaluate valid constraints */
1666                 for (con= td->con; con; con= con->next) {
1667                         float tmat[4][4];
1668
1669                         /* only consider constraint if enabled */
1670                         if (con->flag & CONSTRAINT_DISABLE) continue;
1671                         if (con->enforce == 0.0f) continue;
1672
1673                         /* only use it if it's tagged for this purpose (and the right type) */
1674                         if (con->type == CONSTRAINT_TYPE_LOCLIMIT) {
1675                                 bLocLimitConstraint *data= con->data;
1676
1677                                 if ((data->flag2 & LIMIT_TRANSFORM)==0)
1678                                         continue;
1679
1680                                 /* do space conversions */
1681                                 if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
1682                                         /* just multiply by td->mtx (this should be ok) */
1683                                         Mat4CpyMat4(tmat, cob.matrix);
1684                                         Mat4MulMat34(cob.matrix, td->mtx, tmat);
1685                                 }
1686                                 else if (con->ownspace != CONSTRAINT_SPACE_LOCAL) {
1687                                         /* skip... incompatable spacetype */
1688                                         continue;
1689                                 }
1690
1691                                 /* do constraint */
1692                                 cti->evaluate_constraint(con, &cob, NULL);
1693
1694                                 /* convert spaces again */
1695                                 if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
1696                                         /* just multiply by td->mtx (this should be ok) */
1697                                         Mat4CpyMat4(tmat, cob.matrix);
1698                                         Mat4MulMat34(cob.matrix, td->smtx, tmat);
1699                                 }
1700                         }
1701                 }
1702
1703                 /* copy results from cob->matrix */
1704                 if (td->tdi) {
1705                         TransDataIpokey *tdi= td->tdi;
1706                         tdi->locx[0]= cob.matrix[3][0];
1707                         tdi->locy[0]= cob.matrix[3][1];
1708                         tdi->locz[0]= cob.matrix[3][2];
1709                 }
1710                 else {
1711                         VECCOPY(td->loc, cob.matrix[3]);
1712                 }
1713         }
1714 }
1715
1716 static void constraintRotLim(TransInfo *t, TransData *td)
1717 {
1718         if (td->con) {
1719                 bConstraintTypeInfo *cti= get_constraint_typeinfo(CONSTRAINT_TYPE_ROTLIMIT);
1720                 bConstraintOb cob;
1721                 bConstraint *con;
1722
1723                 /* Make a temporary bConstraintOb for using these limit constraints
1724                  *      - they only care that cob->matrix is correctly set ;-)
1725                  *      - current space should be local
1726                  */
1727                 memset(&cob, 0, sizeof(bConstraintOb));
1728                 if (td->flag & TD_USEQUAT) {
1729                         /* quats */
1730                         if (td->ext)
1731                                 QuatToMat4(td->ext->quat, cob.matrix);
1732                         else
1733                                 return;
1734                 }
1735                 else if (td->tdi) {
1736                         /* ipo-keys eulers */
1737                         TransDataIpokey *tdi= td->tdi;
1738                         float eul[3];
1739
1740                         eul[0]= tdi->rotx[0];
1741                         eul[1]= tdi->roty[0];
1742                         eul[2]= tdi->rotz[0];
1743
1744                         EulOToMat4(eul, td->rotOrder, cob.matrix);
1745                 }
1746                 else {
1747                         /* eulers */
1748                         if (td->ext)
1749                                 EulOToMat4(td->ext->rot, td->rotOrder, cob.matrix);
1750                         else
1751                                 return;
1752                 }
1753
1754                 /* Evaluate valid constraints */
1755                 for (con= td->con; con; con= con->next) {
1756                         /* only consider constraint if enabled */
1757                         if (con->flag & CONSTRAINT_DISABLE) continue;
1758                         if (con->enforce == 0.0f) continue;
1759
1760                         /* we're only interested in Limit-Rotation constraints */
1761                         if (con->type == CONSTRAINT_TYPE_ROTLIMIT) {
1762                                 bRotLimitConstraint *data= con->data;
1763                                 float tmat[4][4];
1764
1765                                 /* only use it if it's tagged for this purpose */
1766                                 if ((data->flag2 & LIMIT_TRANSFORM)==0)
1767                                         continue;
1768
1769                                 /* do space conversions */
1770                                 if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
1771                                         /* just multiply by td->mtx (this should be ok) */
1772                                         Mat4CpyMat4(tmat, cob.matrix);
1773                                         Mat4MulMat34(cob.matrix, td->mtx, tmat);
1774                                 }
1775                                 else if (con->ownspace != CONSTRAINT_SPACE_LOCAL) {
1776                                         /* skip... incompatable spacetype */
1777                                         continue;
1778                                 }
1779
1780                                 /* do constraint */
1781                                 cti->evaluate_constraint(con, &cob, NULL);
1782
1783                                 /* convert spaces again */
1784                                 if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
1785                                         /* just multiply by td->mtx (this should be ok) */
1786                                         Mat4CpyMat4(tmat, cob.matrix);
1787                                         Mat4MulMat34(cob.matrix, td->smtx, tmat);
1788                                 }
1789                         }
1790                 }
1791
1792                 /* copy results from cob->matrix */
1793                 if (td->flag & TD_USEQUAT) {
1794                         /* quats */
1795                         Mat4ToQuat(cob.matrix, td->ext->quat);
1796                 }
1797                 else if (td->tdi) {
1798                         /* ipo-keys eulers */
1799                         TransDataIpokey *tdi= td->tdi;
1800                         float eul[3];
1801
1802                         Mat4ToEulO(cob.matrix, eul, td->rotOrder);
1803
1804                         tdi->rotx[0]= eul[0];
1805                         tdi->roty[0]= eul[1];
1806                         tdi->rotz[0]= eul[2];
1807                 }
1808                 else {
1809                         /* eulers */
1810                         Mat4ToEulO(cob.matrix, td->ext->rot, td->rotOrder);
1811                 }
1812         }
1813 }
1814
1815 static void constraintSizeLim(TransInfo *t, TransData *td)
1816 {
1817         if (td->con && td->ext) {
1818                 bConstraintTypeInfo *cti= get_constraint_typeinfo(CONSTRAINT_TYPE_SIZELIMIT);
1819                 bConstraintOb cob;
1820                 bConstraint *con;
1821
1822                 /* Make a temporary bConstraintOb for using these limit constraints
1823                  *      - they only care that cob->matrix is correctly set ;-)
1824                  *      - current space should be local
1825                  */
1826                 memset(&cob, 0, sizeof(bConstraintOb));
1827                 if (td->tdi) {
1828                         TransDataIpokey *tdi= td->tdi;
1829                         float size[3];
1830
1831                         size[0]= tdi->sizex[0];
1832                         size[1]= tdi->sizey[0];
1833                         size[2]= tdi->sizez[0];
1834                         SizeToMat4(size, cob.matrix);
1835                 }
1836                 else if ((td->flag & TD_SINGLESIZE) && !(t->con.mode & CON_APPLY)) {
1837                         /* scale val and reset size */
1838                         return; // TODO: fix this case
1839                 }
1840                 else {
1841                         /* Reset val if SINGLESIZE but using a constraint */
1842                         if (td->flag & TD_SINGLESIZE)
1843                                 return;
1844
1845                         SizeToMat4(td->ext->size, cob.matrix);
1846                 }
1847
1848                 /* Evaluate valid constraints */
1849                 for (con= td->con; con; con= con->next) {
1850                         /* only consider constraint if enabled */
1851                         if (con->flag & CONSTRAINT_DISABLE) continue;
1852                         if (con->enforce == 0.0f) continue;
1853
1854                         /* we're only interested in Limit-Scale constraints */
1855                         if (con->type == CONSTRAINT_TYPE_SIZELIMIT) {
1856                                 bSizeLimitConstraint *data= con->data;
1857                                 float tmat[4][4];
1858
1859                                 /* only use it if it's tagged for this purpose */
1860                                 if ((data->flag2 & LIMIT_TRANSFORM)==0)
1861                                         continue;
1862
1863                                 /* do space conversions */
1864                                 if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
1865                                         /* just multiply by td->mtx (this should be ok) */
1866                                         Mat4CpyMat4(tmat, cob.matrix);
1867                                         Mat4MulMat34(cob.matrix, td->mtx, tmat);
1868                                 }
1869                                 else if (con->ownspace != CONSTRAINT_SPACE_LOCAL) {
1870                                         /* skip... incompatable spacetype */
1871                                         continue;
1872                                 }
1873
1874                                 /* do constraint */
1875                                 cti->evaluate_constraint(con, &cob, NULL);
1876
1877                                 /* convert spaces again */
1878                                 if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
1879                                         /* just multiply by td->mtx (this should be ok) */
1880                                         Mat4CpyMat4(tmat, cob.matrix);
1881                                         Mat4MulMat34(cob.matrix, td->smtx, tmat);
1882                                 }
1883                         }
1884                 }
1885
1886                 /* copy results from cob->matrix */
1887                 if (td->tdi) {
1888                         TransDataIpokey *tdi= td->tdi;
1889                         float size[3];
1890
1891                         Mat4ToSize(cob.matrix, size);
1892
1893                         tdi->sizex[0]= size[0];
1894                         tdi->sizey[0]= size[1];
1895                         tdi->sizez[0]= size[2];
1896                 }
1897                 else if ((td->flag & TD_SINGLESIZE) && !(t->con.mode & CON_APPLY)) {
1898                         /* scale val and reset size */
1899                         return; // TODO: fix this case
1900                 }
1901                 else {
1902                         /* Reset val if SINGLESIZE but using a constraint */
1903                         if (td->flag & TD_SINGLESIZE)
1904                                 return;
1905
1906                         Mat4ToSize(cob.matrix, td->ext->size);
1907                 }
1908         }
1909 }
1910
1911 /* ************************** WARP *************************** */
1912
1913 void initWarp(TransInfo *t)
1914 {
1915         float max[3], min[3];
1916         int i;
1917
1918         t->mode = TFM_WARP;
1919         t->transform = Warp;
1920         t->handleEvent = handleEventWarp;
1921
1922         initMouseInputMode(t, &t->mouse, INPUT_HORIZONTAL_RATIO);
1923
1924         t->idx_max = 0;
1925         t->num.idx_max = 0;
1926         t->snap[0] = 0.0f;
1927         t->snap[1] = 5.0f;
1928         t->snap[2] = 1.0f;
1929
1930         t->flag |= T_NO_CONSTRAINT;
1931
1932         /* we need min/max in view space */
1933         for(i = 0; i < t->total; i++) {
1934                 float center[3];
1935                 VECCOPY(center, t->data[i].center);
1936                 Mat3MulVecfl(t->data[i].mtx, center);
1937                 Mat4MulVecfl(t->viewmat, center);
1938                 VecSubf(center, center, t->viewmat[3]);
1939                 if (i)
1940                         MinMax3(min, max, center);
1941                 else {
1942                         VECCOPY(max, center);
1943                         VECCOPY(min, center);
1944                 }
1945         }
1946
1947         t->center[0]= (min[0]+max[0])/2.0f;
1948         t->center[1]= (min[1]+max[1])/2.0f;
1949         t->center[2]= (min[2]+max[2])/2.0f;
1950
1951         if (max[0] == min[0]) max[0] += 0.1; /* not optimal, but flipping is better than invalid garbage (i.e. division by zero!) */
1952         t->val= (max[0]-min[0])/2.0f; /* t->val is X dimension projected boundbox */
1953 }
1954
1955 int handleEventWarp(TransInfo *t, wmEvent *event)
1956 {
1957         int status = 0;
1958
1959         if (event->type == MIDDLEMOUSE && event->val==KM_PRESS)
1960         {
1961                 // Use customData pointer to signal warp direction
1962                 if      (t->customData == 0)
1963                         t->customData = (void*)1;
1964                 else
1965                         t->customData = 0;
1966
1967                 status = 1;
1968         }
1969
1970         return status;
1971 }
1972
1973 int Warp(TransInfo *t, short mval[2])
1974 {
1975         TransData *td = t->data;
1976         float vec[3], circumfac, dist, phi0, co, si, *curs, cursor[3], gcursor[3];
1977         int i;
1978         char str[50];
1979
1980         curs= give_cursor(t->scene, t->view);
1981         /*
1982          * gcursor is the one used for helpline.
1983          * It has to be in the same space as the drawing loop
1984          * (that means it needs to be in the object's space when in edit mode and
1985          *  in global space in object mode)
1986          *
1987          * cursor is used for calculations.
1988          * It needs to be in view space, but we need to take object's offset
1989          * into account if in Edit mode.
1990          */
1991         VECCOPY(cursor, curs);
1992         VECCOPY(gcursor, cursor);
1993         if (t->flag & T_EDIT) {
1994                 VecSubf(cursor, cursor, t->obedit->obmat[3]);
1995                 VecSubf(gcursor, gcursor, t->obedit->obmat[3]);
1996                 Mat3MulVecfl(t->data->smtx, gcursor);
1997         }
1998         Mat4MulVecfl(t->viewmat, cursor);
1999         VecSubf(cursor, cursor, t->viewmat[3]);
2000
2001         /* amount of degrees for warp */
2002         circumfac = 360.0f * t->values[0];
2003
2004         if (t->customData) /* non-null value indicates reversed input */
2005         {
2006                 circumfac *= -1;
2007         }
2008
2009         snapGrid(t, &circumfac);
2010         applyNumInput(&t->num, &circumfac);
2011
2012         /* header print for NumInput */
2013         if (hasNumInput(&t->num)) {
2014                 char c[20];
2015
2016                 outputNumInput(&(t->num), c);
2017
2018                 sprintf(str, "Warp: %s", c);
2019         }
2020         else {
2021                 /* default header print */
2022                 sprintf(str, "Warp: %.3f", circumfac);
2023         }
2024
2025         circumfac*= (float)(-M_PI/360.0);
2026
2027         for(i = 0; i < t->total; i++, td++) {
2028                 float loc[3];
2029                 if (td->flag & TD_NOACTION)
2030                         break;
2031
2032                 if (td->flag & TD_SKIP)
2033                         continue;
2034
2035                 /* translate point to center, rotate in such a way that outline==distance */
2036                 VECCOPY(vec, td->iloc);
2037                 Mat3MulVecfl(td->mtx, vec);
2038                 Mat4MulVecfl(t->viewmat, vec);
2039                 VecSubf(vec, vec, t->viewmat[3]);
2040
2041                 dist= vec[0]-cursor[0];
2042
2043                 /* t->val is X dimension projected boundbox */
2044                 phi0= (circumfac*dist/t->val);
2045
2046                 vec[1]= (vec[1]-cursor[1]);
2047
2048                 co= (float)cos(phi0);
2049                 si= (float)sin(phi0);
2050                 loc[0]= -si*vec[1]+cursor[0];
2051                 loc[1]= co*vec[1]+cursor[1];
2052                 loc[2]= vec[2];
2053
2054                 Mat4MulVecfl(t->viewinv, loc);
2055                 VecSubf(loc, loc, t->viewinv[3]);
2056                 Mat3MulVecfl(td->smtx, loc);
2057
2058                 VecSubf(loc, loc, td->iloc);
2059                 VecMulf(loc, td->factor);
2060                 VecAddf(td->loc, td->iloc, loc);
2061         }
2062
2063         recalcData(t);
2064
2065         ED_area_headerprint(t->sa, str);
2066
2067         return 1;
2068 }
2069
2070 /* ************************** SHEAR *************************** */
2071
2072 void initShear(TransInfo *t)
2073 {
2074         t->mode = TFM_SHEAR;
2075         t->transform = Shear;
2076         t->handleEvent = handleEventShear;
2077
2078         initMouseInputMode(t, &t->mouse, INPUT_HORIZONTAL_ABSOLUTE);
2079
2080         t->idx_max = 0;
2081         t->num.idx_max = 0;
2082         t->snap[0] = 0.0f;
2083         t->snap[1] = 0.1f;
2084         t->snap[2] = t->snap[1] * 0.1f;
2085
2086         t->flag |= T_NO_CONSTRAINT;
2087 }
2088
2089 int handleEventShear(TransInfo *t, wmEvent *event)
2090 {
2091         int status = 0;
2092
2093         if (event->type == MIDDLEMOUSE && event->val==KM_PRESS)
2094         {
2095                 // Use customData pointer to signal Shear direction
2096                 if      (t->customData == 0)
2097                 {
2098                         initMouseInputMode(t, &t->mouse, INPUT_VERTICAL_ABSOLUTE);
2099                         t->customData = (void*)1;
2100                 }
2101                 else
2102                 {
2103                         initMouseInputMode(t, &t->mouse, INPUT_HORIZONTAL_ABSOLUTE);
2104                         t->customData = 0;
2105                 }
2106
2107                 status = 1;
2108         }
2109
2110         return status;
2111 }
2112
2113
2114 int Shear(TransInfo *t, short mval[2])
2115 {
2116         TransData *td = t->data;
2117         float vec[3];
2118         float smat[3][3], tmat[3][3], totmat[3][3], persmat[3][3], persinv[3][3];
2119         float value;
2120         int i;
2121         char str[50];
2122
2123         Mat3CpyMat4(persmat, t->viewmat);
2124         Mat3Inv(persinv, persmat);
2125
2126         value = 0.05f * t->values[0];
2127
2128         snapGrid(t, &value);
2129
2130         applyNumInput(&t->num, &value);
2131
2132         /* header print for NumInput */
2133         if (hasNumInput(&t->num)) {
2134                 char c[20];
2135
2136                 outputNumInput(&(t->num), c);
2137
2138                 sprintf(str, "Shear: %s %s", c, t->proptext);
2139         }
2140         else {
2141                 /* default header print */
2142                 sprintf(str, "Shear: %.3f %s", value, t->proptext);
2143         }
2144
2145         Mat3One(smat);
2146
2147         // Custom data signals shear direction
2148         if (t->customData == 0)
2149                 smat[1][0] = value;
2150         else
2151                 smat[0][1] = value;
2152
2153         Mat3MulMat3(tmat, smat, persmat);
2154         Mat3MulMat3(totmat, persinv, tmat);
2155
2156         for(i = 0 ; i < t->total; i++, td++) {
2157                 if (td->flag & TD_NOACTION)
2158                         break;
2159
2160                 if (td->flag & TD_SKIP)
2161                         continue;
2162
2163                 if (t->obedit) {
2164                         float mat3[3][3];
2165                         Mat3MulMat3(mat3, totmat, td->mtx);
2166                         Mat3MulMat3(tmat, td->smtx, mat3);
2167                 }
2168                 else {
2169                         Mat3CpyMat3(tmat, totmat);
2170                 }
2171                 VecSubf(vec, td->center, t->center);
2172
2173                 Mat3MulVecfl(tmat, vec);
2174
2175                 VecAddf(vec, vec, t->center);
2176                 VecSubf(vec, vec, td->center);
2177
2178                 VecMulf(vec, td->factor);
2179
2180                 VecAddf(td->loc, td->iloc, vec);
2181         }
2182
2183         recalcData(t);
2184
2185         ED_area_headerprint(t->sa, str);
2186
2187         return 1;
2188 }
2189
2190 /* ************************** RESIZE *************************** */
2191
2192 void initResize(TransInfo *t)
2193 {
2194         t->mode = TFM_RESIZE;
2195         t->transform = Resize;
2196
2197         initMouseInputMode(t, &t->mouse, INPUT_SPRING_FLIP);
2198
2199         t->flag |= T_NULL_ONE;
2200         t->num.flag |= NUM_NULL_ONE;
2201         t->num.flag |= NUM_AFFECT_ALL;
2202         if (!t->obedit) {
2203                 t->flag |= T_NO_ZERO;
2204                 t->num.flag |= NUM_NO_ZERO;
2205         }
2206
2207         t->idx_max = 2;
2208         t->num.idx_max = 2;
2209         t->snap[0] = 0.0f;
2210         t->snap[1] = 0.1f;
2211         t->snap[2] = t->snap[1] * 0.1f;
2212 }
2213
2214 static void headerResize(TransInfo *t, float vec[3], char *str) {
2215         char tvec[60];
2216         if (hasNumInput(&t->num)) {
2217                 outputNumInput(&(t->num), tvec);
2218         }
2219         else {
2220                 sprintf(&tvec[0], "%.4f", vec[0]);
2221                 sprintf(&tvec[20], "%.4f", vec[1]);
2222                 sprintf(&tvec[40], "%.4f", vec[2]);
2223         }
2224
2225         if (t->con.mode & CON_APPLY) {
2226                 switch(t->num.idx_max) {
2227                 case 0:
2228                         sprintf(str, "Scale: %s%s %s", &tvec[0], t->con.text, t->proptext);
2229                         break;
2230                 case 1:
2231                         sprintf(str, "Scale: %s : %s%s %s", &tvec[0], &tvec[20], t->con.text, t->proptext);
2232                         break;
2233                 case 2:
2234                         sprintf(str, "Scale: %s : %s : %s%s %s", &tvec[0], &tvec[20], &tvec[40], t->con.text, t->proptext);
2235                 }
2236         }
2237         else {
2238                 if (t->flag & T_2D_EDIT)
2239                         sprintf(str, "Scale X: %s   Y: %s%s %s", &tvec[0], &tvec[20], t->con.text, t->proptext);
2240                 else
2241                         sprintf(str, "Scale X: %s   Y: %s  Z: %s%s %s", &tvec[0], &tvec[20], &tvec[40], t->con.text, t->proptext);
2242         }
2243 }
2244
2245 #define SIGN(a)         (a<-FLT_EPSILON?1:a>FLT_EPSILON?2:3)
2246 #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)
2247
2248 /* smat is reference matrix, only scaled */
2249 static void TransMat3ToSize( float mat[][3], float smat[][3], float *size)
2250 {
2251         float vec[3];
2252
2253         VecCopyf(vec, mat[0]);
2254         size[0]= Normalize(vec);
2255         VecCopyf(vec, mat[1]);
2256         size[1]= Normalize(vec);
2257         VecCopyf(vec, mat[2]);
2258         size[2]= Normalize(vec);
2259
2260         /* first tried with dotproduct... but the sign flip is crucial */
2261         if( VECSIGNFLIP(mat[0], smat[0]) ) size[0]= -size[0];
2262         if( VECSIGNFLIP(mat[1], smat[1]) ) size[1]= -size[1];
2263         if( VECSIGNFLIP(mat[2], smat[2]) ) size[2]= -size[2];
2264 }
2265
2266
2267 static void ElementResize(TransInfo *t, TransData *td, float mat[3][3]) {
2268         float tmat[3][3], smat[3][3], center[3];
2269         float vec[3];
2270
2271         if (t->flag & T_EDIT) {
2272                 Mat3MulMat3(smat, mat, td->mtx);
2273                 Mat3MulMat3(tmat, td->smtx, smat);
2274         }
2275         else {
2276                 Mat3CpyMat3(tmat, mat);
2277         }
2278
2279         if (t->con.applySize) {
2280                 t->con.applySize(t, td, tmat);
2281         }
2282
2283         /* local constraint shouldn't alter center */
2284         if (t->around == V3D_LOCAL) {
2285                 if (t->flag & T_OBJECT) {
2286                         VECCOPY(center, td->center);
2287                 }
2288                 else if (t->flag & T_EDIT) {
2289
2290                         if(t->around==V3D_LOCAL && (t->settings->selectmode & SCE_SELECT_FACE)) {
2291                                 VECCOPY(center, td->center);
2292                         }
2293                         else {
2294                                 VECCOPY(center, t->center);
2295                         }
2296                 }
2297                 else {
2298                         VECCOPY(center, t->center);
2299                 }
2300         }
2301         else {
2302                 VECCOPY(center, t->center);
2303         }
2304
2305         if (td->ext) {
2306                 float fsize[3];
2307
2308                 if (t->flag & (T_OBJECT|T_TEXTURE|T_POSE)) {
2309                         float obsizemat[3][3];
2310                         // Reorient the size mat to fit the oriented object.
2311                         Mat3MulMat3(obsizemat, tmat, td->axismtx);
2312                         //printmatrix3("obsizemat", obsizemat);
2313                         TransMat3ToSize(obsizemat, td->axismtx, fsize);
2314                         //printvecf("fsize", fsize);
2315                 }
2316                 else {
2317                         Mat3ToSize(tmat, fsize);
2318                 }
2319
2320                 protectedSizeBits(td->protectflag, fsize);
2321
2322                 if ((t->flag & T_V3D_ALIGN)==0) {       // align mode doesn't resize objects itself
2323                         /* handle ipokeys? */
2324                         if(td->tdi) {
2325                                 TransDataIpokey *tdi= td->tdi;
2326                                 /* calculate delta size (equal for size and dsize) */
2327
2328                                 vec[0]= (tdi->oldsize[0])*(fsize[0] -1.0f) * td->factor;
2329                                 vec[1]= (tdi->oldsize[1])*(fsize[1] -1.0f) * td->factor;
2330                                 vec[2]= (tdi->oldsize[2])*(fsize[2] -1.0f) * td->factor;
2331
2332                                 add_tdi_poin(tdi->sizex, tdi->oldsize,   vec[0]);
2333                                 add_tdi_poin(tdi->sizey, tdi->oldsize+1, vec[1]);
2334                                 add_tdi_poin(tdi->sizez, tdi->oldsize+2, vec[2]);
2335
2336                         }
2337                         else if((td->flag & TD_SINGLESIZE) && !(t->con.mode & CON_APPLY)){
2338                                 /* scale val and reset size */
2339                                 *td->val = td->ival * fsize[0] * td->factor;
2340
2341                                 td->ext->size[0] = td->ext->isize[0];
2342                                 td->ext->size[1] = td->ext->isize[1];
2343                                 td->ext->size[2] = td->ext->isize[2];
2344                         }
2345                         else {
2346                                 /* Reset val if SINGLESIZE but using a constraint */
2347                                 if (td->flag & TD_SINGLESIZE)
2348                                         *td->val = td->ival;
2349
2350                                 td->ext->size[0] = td->ext->isize[0] * (fsize[0]) * td->factor;
2351                                 td->ext->size[1] = td->ext->isize[1] * (fsize[1]) * td->factor;
2352                                 td->ext->size[2] = td->ext->isize[2] * (fsize[2]) * td->factor;
2353                         }
2354                 }
2355
2356                 constraintSizeLim(t, td);
2357         }
2358
2359         /* For individual element center, Editmode need to use iloc */
2360         if (t->flag & T_POINTS)
2361                 VecSubf(vec, td->iloc, center);
2362         else
2363                 VecSubf(vec, td->center, center);
2364
2365         Mat3MulVecfl(tmat, vec);
2366
2367         VecAddf(vec, vec, center);
2368         if (t->flag & T_POINTS)
2369                 VecSubf(vec, vec, td->iloc);
2370         else
2371                 VecSubf(vec, vec, td->center);
2372
2373         VecMulf(vec, td->factor);
2374
2375         if (t->flag & (T_OBJECT|T_POSE)) {
2376                 Mat3MulVecfl(td->smtx, vec);
2377         }
2378
2379         protectedTransBits(td->protectflag, vec);
2380
2381         if(td->tdi) {
2382                 TransDataIpokey *tdi= td->tdi;
2383                 add_tdi_poin(tdi->locx, tdi->oldloc, vec[0]);
2384                 add_tdi_poin(tdi->locy, tdi->oldloc+1, vec[1]);
2385                 add_tdi_poin(tdi->locz, tdi->oldloc+2, vec[2]);
2386         }
2387         else VecAddf(td->loc, td->iloc, vec);
2388
2389         constraintTransLim(t, td);
2390 }
2391
2392 int Resize(TransInfo *t, short mval[2])
2393 {
2394         TransData *td;
2395         float size[3], mat[3][3];
2396         float ratio;
2397         int i;
2398         char str[200];
2399
2400         /* for manipulator, center handle, the scaling can't be done relative to center */
2401         if( (t->flag & T_USES_MANIPULATOR) && t->con.mode==0)
2402         {
2403                 ratio = 1.0f - ((t->imval[0] - mval[0]) + (t->imval[1] - mval[1]))/100.0f;
2404         }
2405         else
2406         {
2407                 ratio = t->values[0];
2408         }
2409
2410         size[0] = size[1] = size[2] = ratio;
2411
2412         snapGrid(t, size);
2413
2414         if (hasNumInput(&t->num)) {
2415                 applyNumInput(&t->num, size);
2416                 constraintNumInput(t, size);
2417         }
2418
2419         applySnapping(t, size);
2420
2421         if (t->flag & T_AUTOVALUES)
2422         {
2423                 VECCOPY(size, t->auto_values);
2424         }
2425
2426         VECCOPY(t->values, size);
2427
2428         SizeToMat3(size, mat);
2429
2430         if (t->con.applySize) {
2431                 t->con.applySize(t, NULL, mat);
2432         }
2433
2434         Mat3CpyMat3(t->mat, mat);       // used in manipulator
2435
2436         headerResize(t, size, str);
2437
2438         for(i = 0, td=t->data; i < t->total; i++, td++) {
2439                 if (td->flag & TD_NOACTION)
2440                         break;
2441
2442                 if (td->flag & TD_SKIP)
2443                         continue;
2444
2445                 ElementResize(t, td, mat);
2446         }
2447
2448         /* evil hack - redo resize if cliping needed */
2449         if (t->flag & T_CLIP_UV && clipUVTransform(t, size, 1)) {
2450                 SizeToMat3(size, mat);
2451
2452                 if (t->con.applySize)
2453                         t->con.applySize(t, NULL, mat);
2454
2455                 for(i = 0, td=t->data; i < t->total; i++, td++)
2456                         ElementResize(t, td, mat);
2457         }
2458
2459         recalcData(t);
2460
2461         ED_area_headerprint(t->sa, str);
2462
2463         return 1;
2464 }
2465
2466 /* ************************** TOSPHERE *************************** */
2467
2468 void initToSphere(TransInfo *t)
2469 {
2470         TransData *td = t->data;
2471         int i;
2472
2473         t->mode = TFM_TOSPHERE;
2474         t->transform = ToSphere;
2475
2476         initMouseInputMode(t, &t->mouse, INPUT_HORIZONTAL_RATIO);
2477
2478         t->idx_max = 0;
2479         t->num.idx_max = 0;
2480         t->snap[0] = 0.0f;
2481         t->snap[1] = 0.1f;
2482         t->snap[2] = t->snap[1] * 0.1f;
2483
2484         t->num.flag |= NUM_NULL_ONE | NUM_NO_NEGATIVE;
2485         t->flag |= T_NO_CONSTRAINT;
2486
2487         // Calculate average radius
2488         for(i = 0 ; i < t->total; i++, td++) {
2489                 t->val += VecLenf(t->center, td->iloc);
2490         }
2491
2492         t->val /= (float)t->total;
2493 }
2494
2495 int ToSphere(TransInfo *t, short mval[2])
2496 {
2497         float vec[3];
2498         float ratio, radius;
2499         int i;
2500         char str[64];
2501         TransData *td = t->data;
2502
2503         ratio = t->values[0];
2504
2505         snapGrid(t, &ratio);
2506
2507         applyNumInput(&t->num, &ratio);
2508
2509         if (ratio < 0)
2510                 ratio = 0.0f;
2511         else if (ratio > 1)
2512                 ratio = 1.0f;
2513
2514         /* header print for NumInput */
2515         if (hasNumInput(&t->num)) {
2516                 char c[20];
2517
2518                 outputNumInput(&(t->num), c);
2519
2520                 sprintf(str, "To Sphere: %s %s", c, t->proptext);
2521         }
2522         else {
2523                 /* default header print */
2524                 sprintf(str, "To Sphere: %.4f %s", ratio, t->proptext);
2525         }
2526
2527
2528         for(i = 0 ; i < t->total; i++, td++) {
2529                 float tratio;
2530                 if (td->flag & TD_NOACTION)
2531                         break;
2532
2533                 if (td->flag & TD_SKIP)
2534                         continue;
2535
2536                 VecSubf(vec, td->iloc, t->center);
2537
2538                 radius = Normalize(vec);
2539
2540                 tratio = ratio * td->factor;
2541
2542                 VecMulf(vec, radius * (1.0f - tratio) + t->val * tratio);
2543
2544                 VecAddf(td->loc, t->center, vec);
2545         }
2546
2547
2548         recalcData(t);
2549
2550         ED_area_headerprint(t->sa, str);
2551
2552         return 1;
2553 }
2554
2555 /* ************************** ROTATION *************************** */
2556
2557
2558 void initRotation(TransInfo *t)
2559 {
2560         t->mode = TFM_ROTATION;
2561         t->transform = Rotation;
2562
2563         initMouseInputMode(t, &t->mouse, INPUT_ANGLE);
2564
2565         t->ndof.axis = 16;
2566         /* Scale down and flip input for rotation */
2567         t->ndof.factor[0] = -0.2f;
2568
2569         t->idx_max = 0;
2570         t->num.idx_max = 0;
2571         t->snap[0] = 0.0f;
2572         t->snap[1] = (float)((5.0/180)*M_PI);
2573         t->snap[2] = t->snap[1] * 0.2f;
2574
2575         if (t->flag & T_2D_EDIT)
2576                 t->flag |= T_NO_CONSTRAINT;
2577 }
2578
2579 static void ElementRotation(TransInfo *t, TransData *td, float mat[3][3], short around) {
2580         float vec[3], totmat[3][3], smat[3][3];
2581         float eul[3], fmat[3][3], quat[4];
2582         float *center = t->center;
2583
2584         /* local constraint shouldn't alter center */
2585         if (around == V3D_LOCAL) {
2586                 if (t->flag & (T_OBJECT|T_POSE)) {
2587                         center = td->center;
2588                 }
2589                 else {
2590                         /* !TODO! Make this if not rely on G */
2591                         if(around==V3D_LOCAL && (t->settings->selectmode & SCE_SELECT_FACE)) {
2592                                 center = td->center;
2593                         }
2594                 }
2595         }
2596
2597         if (t->flag & T_POINTS) {
2598                 Mat3MulMat3(totmat, mat, td->mtx);
2599                 Mat3MulMat3(smat, td->smtx, totmat);
2600
2601                 VecSubf(vec, td->iloc, center);
2602                 Mat3MulVecfl(smat, vec);
2603
2604                 VecAddf(td->loc, vec, center);
2605
2606                 VecSubf(vec,td->loc,td->iloc);
2607                 protectedTransBits(td->protectflag, vec);
2608                 VecAddf(td->loc, td->iloc, vec);
2609
2610                 if(td->flag & TD_USEQUAT) {
2611                         Mat3MulSerie(fmat, td->mtx, mat, td->smtx, 0, 0, 0, 0, 0);
2612                         Mat3ToQuat(fmat, quat); // Actual transform
2613
2614                         if(td->ext->quat){
2615                                 QuatMul(td->ext->quat, quat, td->ext->iquat);
2616
2617                                 /* is there a reason not to have this here? -jahka */
2618                                 protectedQuaternionBits(td->protectflag, td->ext->quat, td->ext->iquat);
2619                         }
2620                 }
2621         }
2622         /**
2623          * HACK WARNING
2624          *
2625          * This is some VERY ugly special case to deal with pose mode.
2626          *
2627          * The problem is that mtx and smtx include each bone orientation.
2628          *
2629          * That is needed to rotate each bone properly, HOWEVER, to calculate
2630          * the translation component, we only need the actual armature object's
2631          * matrix (and inverse). That is not all though. Once the proper translation
2632          * has been computed, it has to be converted back into the bone's space.
2633          */
2634         else if (t->flag & T_POSE) {
2635                 float pmtx[3][3], imtx[3][3];
2636
2637                 // Extract and invert armature object matrix
2638                 Mat3CpyMat4(pmtx, t->poseobj->obmat);
2639                 Mat3Inv(imtx, pmtx);
2640
2641                 if ((td->flag & TD_NO_LOC) == 0)
2642                 {
2643                         VecSubf(vec, td->center, center);
2644
2645                         Mat3MulVecfl(pmtx, vec);        // To Global space
2646                         Mat3MulVecfl(mat, vec);         // Applying rotation
2647                         Mat3MulVecfl(imtx, vec);        // To Local space
2648
2649                         VecAddf(vec, vec, center);
2650                         /* vec now is the location where the object has to be */
2651
2652                         VecSubf(vec, vec, td->center); // Translation needed from the initial location
2653
2654                         Mat3MulVecfl(pmtx, vec);        // To Global space
2655                         Mat3MulVecfl(td->smtx, vec);// To Pose space
2656
2657                         protectedTransBits(td->protectflag, vec);
2658
2659                         VecAddf(td->loc, td->iloc, vec);
2660
2661                         constraintTransLim(t, td);
2662                 }
2663
2664                 /* rotation */
2665                 if ((t->flag & T_V3D_ALIGN)==0) { // align mode doesn't rotate objects itself
2666                         /* euler or quaternion? */
2667                         if (td->flag & TD_USEQUAT) {
2668                                 Mat3MulSerie(fmat, td->mtx, mat, td->smtx, 0, 0, 0, 0, 0);
2669
2670                                 Mat3ToQuat(fmat, quat); // Actual transform
2671
2672                                 QuatMul(td->ext->quat, quat, td->ext->iquat);
2673                                 /* this function works on end result */
2674                                 protectedQuaternionBits(td->protectflag, td->ext->quat, td->ext->iquat);
2675                         }
2676                         else { 
2677                                 float eulmat[3][3];
2678                                 
2679                                 Mat3MulMat3(totmat, mat, td->mtx);
2680                                 Mat3MulMat3(smat, td->smtx, totmat);
2681                                 
2682                                 /* calculate the total rotatation in eulers */
2683                                 VECCOPY(eul, td->ext->irot);
2684                                 EulOToMat3(eul, td->rotOrder, eulmat);
2685                                 
2686                                 /* mat = transform, obmat = bone rotation */
2687                                 Mat3MulMat3(fmat, smat, eulmat);
2688                                 
2689                                 Mat3ToCompatibleEulO(fmat, eul, td->ext->rot, td->rotOrder);
2690                                 
2691                                 /* and apply (to end result only) */
2692                                 protectedRotateBits(td->protectflag, eul, td->ext->irot);
2693                                 VECCOPY(td->ext->rot, eul);
2694                         }
2695
2696                         constraintRotLim(t, td);
2697                 }
2698         }
2699         else {
2700                 if ((td->flag & TD_NO_LOC) == 0)
2701                 {
2702                         /* translation */
2703                         VecSubf(vec, td->center, center);
2704                         Mat3MulVecfl(mat, vec);
2705                         VecAddf(vec, vec, center);
2706                         /* vec now is the location where the object has to be */
2707                         VecSubf(vec, vec, td->center);
2708                         Mat3MulVecfl(td->smtx, vec);
2709
2710                         protectedTransBits(td->protectflag, vec);
2711
2712                         if(td->tdi) {
2713                                 TransDataIpokey *tdi= td->tdi;
2714                                 add_tdi_poin(tdi->locx, tdi->oldloc, vec[0]);
2715                                 add_tdi_poin(tdi->locy, tdi->oldloc+1, vec[1]);
2716                                 add_tdi_poin(tdi->locz, tdi->oldloc+2, vec[2]);
2717                         }
2718                         else VecAddf(td->loc, td->iloc, vec);
2719                 }
2720
2721
2722                 constraintTransLim(t, td);
2723
2724                 /* rotation */
2725                 if ((t->flag & T_V3D_ALIGN)==0) { // align mode doesn't rotate objects itself
2726                         /* euler or quaternion? */
2727                     if (td->flag & TD_USEQUAT) {
2728                                 Mat3MulSerie(fmat, td->mtx, mat, td->smtx, 0, 0, 0, 0, 0);
2729                                 Mat3ToQuat(fmat, quat); // Actual transform
2730
2731                                 QuatMul(td->ext->quat, quat, td->ext->iquat);
2732                                 /* this function works on end result */
2733                                 protectedQuaternionBits(td->protectflag, td->ext->quat, td->ext->iquat);
2734                         }
2735                         else {
2736                                 float obmat[3][3];
2737
2738                                 /* are there ipo keys? */
2739                                 if(td->tdi) {
2740                                         TransDataIpokey *tdi= td->tdi;
2741                                         float current_rot[3];
2742                                         float rot[3];
2743
2744                                         /* current IPO value for compatible euler */
2745                                         current_rot[0] = (tdi->rotx) ? tdi->rotx[0] : 0.0f;
2746                                         current_rot[1] = (tdi->roty) ? tdi->roty[0] : 0.0f;
2747                                         current_rot[2] = (tdi->rotz) ? tdi->rotz[0] : 0.0f;
2748                                         VecMulf(current_rot, (float)(M_PI_2 / 9.0));
2749
2750                                         /* calculate the total rotatation in eulers */
2751                                         VecAddf(eul, td->ext->irot, td->ext->drot);
2752                                         EulToMat3(eul, obmat);
2753                                         /* mat = transform, obmat = object rotation */
2754                                         Mat3MulMat3(fmat, mat, obmat);
2755
2756                                         Mat3ToCompatibleEul(fmat, eul, current_rot);
2757
2758                                         /* correct back for delta rot */
2759                                         if(tdi->flag & TOB_IPODROT) {
2760                                                 VecSubf(rot, eul, td->ext->irot);
2761                                         }
2762                                         else {
2763                                                 VecSubf(rot, eul, td->ext->drot);
2764                                         }
2765
2766                                         VecMulf(rot, (float)(9.0/M_PI_2));
2767                                         VecSubf(rot, rot, tdi->oldrot);
2768
2769                                         protectedRotateBits(td->protectflag, rot, tdi->oldrot);
2770
2771                                         add_tdi_poin(tdi->rotx, tdi->oldrot, rot[0]);
2772                                         add_tdi_poin(tdi->roty, tdi->oldrot+1, rot[1]);
2773                                         add_tdi_poin(tdi->rotz, tdi->oldrot+2, rot[2]);
2774                                 }
2775                                 else {
2776                                         Mat3MulMat3(totmat, mat, td->mtx);
2777                                         Mat3MulMat3(smat, td->smtx, totmat);
2778
2779                                         /* calculate the total rotatation in eulers */
2780                                         VecAddf(eul, td->ext->irot, td->ext->drot); /* we have to correct for delta rot */
2781                                         EulToMat3(eul, obmat);
2782                                         /* mat = transform, obmat = object rotation */
2783                                         Mat3MulMat3(fmat, smat, obmat);
2784
2785                                         Mat3ToCompatibleEul(fmat, eul, td->ext->rot);
2786
2787                                         /* correct back for delta rot */
2788                                         VecSubf(eul, eul, td->ext->drot);
2789
2790                                         /* and apply */
2791                                         protectedRotateBits(td->protectflag, eul, td->ext->irot);
2792                                         VECCOPY(td->ext->rot, eul);
2793                                 }
2794                         }
2795
2796                         constraintRotLim(t, td);
2797                 }
2798         }
2799 }
2800
2801 static void applyRotation(TransInfo *t, float angle, float axis[3])
2802 {
2803         TransData *td = t->data;
2804         float mat[3][3];
2805         int i;
2806
2807         VecRotToMat3(axis, angle, mat);
2808
2809         for(i = 0 ; i < t->total; i++, td++) {
2810
2811                 if (td->flag & TD_NOACTION)
2812                         break;
2813
2814                 if (td->flag & TD_SKIP)
2815                         continue;
2816
2817                 if (t->con.applyRot) {
2818                         t->con.applyRot(t, td, axis, NULL);
2819                         VecRotToMat3(axis, angle * td->factor, mat);
2820                 }
2821                 else if (t->flag & T_PROP_EDIT) {
2822                         VecRotToMat3(axis, angle * td->factor, mat);
2823                 }
2824
2825                 ElementRotation(t, td, mat, t->around);
2826         }
2827 }
2828
2829 int Rotation(TransInfo *t, short mval[2])
2830 {
2831         char str[64];
2832
2833         float final;
2834
2835         float axis[3];
2836         float mat[3][3];
2837
2838         VECCOPY(axis, t->viewinv[2]);
2839         VecMulf(axis, -1.0f);
2840         Normalize(axis);
2841
2842         final = t->values[0];
2843
2844         applyNDofInput(&t->ndof, &final);
2845
2846         snapGrid(t, &final);
2847
2848         if (t->con.applyRot) {
2849                 t->con.applyRot(t, NULL, axis, &final);
2850         }
2851
2852         applySnapping(t, &final);
2853
2854         if (hasNumInput(&t->num)) {
2855                 char c[20];
2856
2857                 applyNumInput(&t->num, &final);
2858
2859                 outputNumInput(&(t->num), c);
2860
2861                 sprintf(str, "Rot: %s %s %s", &c[0], t->con.text, t->proptext);
2862
2863                 /* Clamp between -180 and 180 */
2864                 while (final >= 180.0)
2865                         final -= 360.0;
2866
2867                 while (final <= -180.0)
2868                         final += 360.0;
2869
2870                 final *= (float)(M_PI / 180.0);
2871         }
2872         else {
2873                 sprintf(str, "Rot: %.2f%s %s", 180.0*final/M_PI, t->con.text, t->proptext);
2874         }
2875
2876         VecRotToMat3(axis, final, mat);
2877
2878         // TRANSFORM_FIX_ME
2879 //      t->values[0] = final;           // used in manipulator
2880 //      Mat3CpyMat3(t->mat, mat);       // used in manipulator
2881
2882         applyRotation(t, final, axis);
2883
2884         recalcData(t);
2885
2886         ED_area_headerprint(t->sa, str);
2887
2888         return 1;
2889 }
2890
2891
2892 /* ************************** TRACKBALL *************************** */
2893
2894 void initTrackball(TransInfo *t)
2895 {
2896         t->mode = TFM_TRACKBALL;
2897         t->transform = Trackball;
2898
2899         initMouseInputMode(t, &t->mouse, INPUT_TRACKBALL);
2900
2901         t->ndof.axis = 40;
2902         /* Scale down input for rotation */
2903         t->ndof.factor[0] = 0.2f;
2904         t->ndof.factor[1] = 0.2f;
2905
2906         t->idx_max = 1;
2907         t->num.idx_max = 1;
2908         t->snap[0] = 0.0f;
2909         t->snap[1] = (float)((5.0/180)*M_PI);
2910         t->snap[2] = t->snap[1] * 0.2f;
2911
2912         t->flag |= T_NO_CONSTRAINT;
2913 }
2914
2915 static void applyTrackball(TransInfo *t, float axis1[3], float axis2[3], float angles[2])
2916 {
2917         TransData *td = t->data;
2918         float mat[3][3], smat[3][3], totmat[3][3];
2919         int i;
2920
2921         VecRotToMat3(axis1, angles[0], smat);
2922         VecRotToMat3(axis2, angles[1], totmat);
2923
2924         Mat3MulMat3(mat, smat, totmat);
2925
2926         for(i = 0 ; i < t->total; i++, td++) {
2927                 if (td->flag & TD_NOACTION)
2928                         break;
2929
2930                 if (td->flag & TD_SKIP)
2931                         continue;
2932
2933                 if (t->flag & T_PROP_EDIT) {
2934                         VecRotToMat3(axis1, td->factor * angles[0], smat);
2935                         VecRotToMat3(axis2, td->factor * angles[1], totmat);
2936
2937                         Mat3MulMat3(mat, smat, totmat);
2938                 }
2939
2940                 ElementRotation(t, td, mat, t->around);
2941         }
2942 }
2943
2944 int Trackball(TransInfo *t, short mval[2])
2945 {
2946         char str[128];
2947         float axis1[3], axis2[3];
2948         float mat[3][3], totmat[3][3], smat[3][3];
2949         float phi[2];
2950
2951         VECCOPY(axis1, t->persinv[0]);
2952         VECCOPY(axis2, t->persinv[1]);
2953         Normalize(axis1);
2954         Normalize(axis2);
2955
2956         phi[0] = t->values[0];
2957         phi[1] = t->values[1];
2958
2959         applyNDofInput(&t->ndof, phi);
2960
2961         snapGrid(t, phi);
2962
2963         if (hasNumInput(&t->num)) {
2964                 char c[40];
2965
2966                 applyNumInput(&t->num, phi);
2967
2968                 outputNumInput(&(t->num), c);
2969
2970                 sprintf(str, "Trackball: %s %s %s", &c[0], &c[20], t->proptext);
2971
2972                 phi[0] *= (float)(M_PI / 180.0);
2973                 phi[1] *= (float)(M_PI / 180.0);
2974         }
2975         else {
2976                 sprintf(str, "Trackball: %.2f %.2f %s", 180.0*phi[0]/M_PI, 180.0*phi[1]/M_PI, t->proptext);
2977         }
2978
2979         VecRotToMat3(axis1, phi[0], smat);
2980         VecRotToMat3(axis2, phi[1], totmat);
2981
2982         Mat3MulMat3(mat, smat, totmat);
2983
2984         // TRANSFORM_FIX_ME
2985         //Mat3CpyMat3(t->mat, mat);     // used in manipulator
2986
2987         applyTrackball(t, axis1, axis2, phi);
2988
2989         recalcData(t);
2990
2991         ED_area_headerprint(t->sa, str);
2992
2993         return 1;
2994 }
2995
2996 /* ************************** TRANSLATION *************************** */
2997
2998 void initTranslation(TransInfo *t)
2999 {
3000         t->mode = TFM_TRANSLATION;
3001         t->transform = Translation;
3002
3003         initMouseInputMode(t, &t->mouse, INPUT_VECTOR);
3004
3005         t->idx_max = (t->flag & T_2D_EDIT)? 1: 2;
3006         t->num.flag = 0;
3007         t->num.idx_max = t->idx_max;
3008
3009         t->ndof.axis = (t->flag & T_2D_EDIT)? 1|2: 1|2|4;
3010
3011         if(t->spacetype == SPACE_VIEW3D) {
3012                 View3D *v3d = t->view;
3013
3014                 t->snap[0] = 0.0f;
3015                 t->snap[1] = v3d->gridview * 1.0f;
3016                 t->snap[2] = t->snap[1] * 0.1f;
3017         }
3018         else if(t->spacetype == SPACE_IMAGE) {
3019                 t->snap[0] = 0.0f;
3020                 t->snap[1] = 0.125f;
3021                 t->snap[2] = 0.0625f;
3022         }
3023         else {
3024                 t->snap[0] = 0.0f;
3025                 t->snap[1] = t->snap[2] = 1.0f;
3026         }
3027 }
3028
3029 static void headerTranslation(TransInfo *t, float vec[3], char *str) {
3030         char tvec[60];
3031         char distvec[20];
3032         char autoik[20];
3033         float dist;
3034
3035         if (hasNumInput(&t->num)) {
3036                 outputNumInput(&(t->num), tvec);
3037                 dist = VecLength(t->num.val);
3038         }
3039         else {
3040                 float dvec[3];
3041
3042                 VECCOPY(dvec, vec);
3043                 applyAspectRatio(t, dvec);
3044
3045                 dist = VecLength(vec);
3046                 if(t->scene->unit.system) {
3047                         int i, do_split= t->scene->unit.flag & USER_UNIT_OPT_SPLIT ? 1:0;
3048
3049                         for(i=0; i<3; i++)
3050                                 bUnit_AsString(&tvec[i*20], 20, dvec[i]*t->scene->unit.scale_length, 4, t->scene->unit.system, B_UNIT_LENGTH, do_split, 1);
3051                 }
3052                 else {
3053                         sprintf(&tvec[0], "%.4f", dvec[0]);
3054                         sprintf(&tvec[20], "%.4f", dvec[1]);
3055                         sprintf(&tvec[40], "%.4f", dvec[2]);
3056                 }