Moved Auto-Keying "Insert for KeyingSet only" option from UserPrefs to
[blender.git] / source / blender / editors / transform / transform_generics.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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, 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 <string.h>
31 #include <math.h>
32
33 #include "MEM_guardedalloc.h"
34
35 #include "BLO_sys_types.h" // for intptr_t support
36
37 #include "DNA_anim_types.h"
38 #include "DNA_armature_types.h"
39 #include "DNA_lattice_types.h"
40 #include "DNA_screen_types.h"
41 #include "DNA_space_types.h"
42 #include "DNA_scene_types.h"
43 #include "DNA_object_types.h"
44 #include "DNA_meshdata_types.h"
45 #include "DNA_view3d_types.h"
46 #include "DNA_modifier_types.h"
47
48 #include "RNA_access.h"
49
50 //#include "BIF_screen.h"
51 //#include "BIF_mywindow.h"
52 #include "BIF_gl.h"
53 #include "BIF_glutil.h"
54 //#include "BIF_editmesh.h"
55 //#include "BIF_editsima.h"
56 //#include "BIF_editparticle.h"
57 //#include "BIF_meshtools.h"
58
59 #include "BKE_animsys.h"
60 #include "BKE_action.h"
61 #include "BKE_armature.h"
62 #include "BKE_curve.h"
63 #include "BKE_depsgraph.h"
64 #include "BKE_displist.h"
65 #include "BKE_fcurve.h"
66 #include "BKE_lattice.h"
67 #include "BKE_mesh.h"
68 #include "BKE_nla.h"
69 #include "BKE_context.h"
70
71 #include "ED_anim_api.h"
72 #include "ED_armature.h"
73 #include "ED_image.h"
74 #include "ED_keyframing.h"
75 #include "ED_markers.h"
76 #include "ED_mesh.h"
77 #include "ED_particle.h"
78 #include "ED_screen_types.h"
79 #include "ED_space_api.h"
80 #include "ED_uvedit.h"
81 #include "ED_view3d.h"
82 #include "ED_curve.h" /* for ED_curve_editnurbs */
83
84 //#include "BDR_unwrapper.h"
85
86 #include "BLI_math.h"
87 #include "BLI_blenlib.h"
88 #include "BLI_editVert.h"
89 #include "BLI_rand.h"
90 #include "BLI_utildefines.h"
91
92 #include "WM_types.h"
93 #include "WM_api.h"
94
95 #include "UI_resources.h"
96
97 //#include "blendef.h"
98 //
99 //#include "mydevice.h"
100
101 #include "transform.h"
102
103 extern ListBase editelems;
104
105 /* ************************** Functions *************************** */
106
107 void getViewVector(TransInfo *t, float coord[3], float vec[3])
108 {
109         if (t->persp != RV3D_ORTHO)
110         {
111                 float p1[4], p2[4];
112                 
113                 VECCOPY(p1, coord);
114                 p1[3] = 1.0f;
115                 VECCOPY(p2, p1);
116                 p2[3] = 1.0f;
117                 mul_m4_v4(t->viewmat, p2);
118                 
119                 p2[0] = 2.0f * p2[0];
120                 p2[1] = 2.0f * p2[1];
121                 p2[2] = 2.0f * p2[2];
122                 
123                 mul_m4_v4(t->viewinv, p2);
124                 
125                 sub_v3_v3v3(vec, p1, p2);
126         }
127         else {
128                 VECCOPY(vec, t->viewinv[2]);
129         }
130         normalize_v3(vec);
131 }
132
133 /* ************************** GENERICS **************************** */
134
135 static void clipMirrorModifier(TransInfo *t, Object *ob)
136 {
137         ModifierData *md= ob->modifiers.first;
138         float tolerance[3] = {0.0f, 0.0f, 0.0f};
139         int axis = 0;
140         
141         for (; md; md=md->next) {
142                 if (md->type==eModifierType_Mirror) {
143                         MirrorModifierData *mmd = (MirrorModifierData*) md;
144                         
145                         if(mmd->flag & MOD_MIR_CLIPPING) {
146                                 axis = 0;
147                                 if(mmd->flag & MOD_MIR_AXIS_X) {
148                                         axis |= 1;
149                                         tolerance[0] = mmd->tolerance;
150                                 }
151                                 if(mmd->flag & MOD_MIR_AXIS_Y) {
152                                         axis |= 2;
153                                         tolerance[1] = mmd->tolerance;
154                                 }
155                                 if(mmd->flag & MOD_MIR_AXIS_Z) {
156                                         axis |= 4;
157                                         tolerance[2] = mmd->tolerance;
158                                 }
159                                 if (axis) {
160                                         float mtx[4][4], imtx[4][4];
161                                         int i;
162                                         TransData *td = t->data;
163                                         
164                                         if (mmd->mirror_ob) {
165                                                 float obinv[4][4];
166                                                 
167                                                 invert_m4_m4(obinv, mmd->mirror_ob->obmat);
168                                                 mul_m4_m4m4(mtx, ob->obmat, obinv);
169                                                 invert_m4_m4(imtx, mtx);
170                                         }
171                                         
172                                         for(i = 0 ; i < t->total; i++, td++) {
173                                                 int clip;
174                                                 float loc[3], iloc[3];
175                                                 
176                                                 if (td->flag & TD_NOACTION)
177                                                         break;
178                                                 if (td->loc==NULL)
179                                                         break;
180                                                 
181                                                 if (td->flag & TD_SKIP)
182                                                         continue;
183                                                 
184                                                 copy_v3_v3(loc,  td->loc);
185                                                 copy_v3_v3(iloc, td->iloc);
186                                                 
187                                                 if (mmd->mirror_ob) {
188                                                         mul_m4_v3(mtx, loc);
189                                                         mul_m4_v3(mtx, iloc);
190                                                 }
191                                                 
192                                                 clip = 0;
193                                                 if(axis & 1) {
194                                                         if(fabs(iloc[0])<=tolerance[0] ||
195                                                            loc[0]*iloc[0]<0.0f) {
196                                                                 loc[0]= 0.0f;
197                                                                 clip = 1;
198                                                         }
199                                                 }
200                                                 
201                                                 if(axis & 2) {
202                                                         if(fabs(iloc[1])<=tolerance[1] ||
203                                                            loc[1]*iloc[1]<0.0f) {
204                                                                 loc[1]= 0.0f;
205                                                                 clip = 1;
206                                                         }
207                                                 }
208                                                 if(axis & 4) {
209                                                         if(fabs(iloc[2])<=tolerance[2] ||
210                                                            loc[2]*iloc[2]<0.0f) {
211                                                                 loc[2]= 0.0f;
212                                                                 clip = 1;
213                                                         }
214                                                 }
215                                                 if (clip) {
216                                                         if (mmd->mirror_ob) {
217                                                                 mul_m4_v3(imtx, loc);
218                                                         }
219                                                         copy_v3_v3(td->loc, loc);
220                                                 }
221                                         }
222                                 }
223                                 
224                         }
225                 }
226         }
227 }
228
229 /* assumes obedit set to mesh object */
230 static void editmesh_apply_to_mirror(TransInfo *t)
231 {
232         TransData *td = t->data;
233         EditVert *eve;
234         int i;
235         
236         for(i = 0 ; i < t->total; i++, td++) {
237                 if (td->flag & TD_NOACTION)
238                         break;
239                 if (td->loc==NULL)
240                         break;
241                 if (td->flag & TD_SKIP)
242                         continue;
243                 
244                 eve = td->extra;
245                 if (eve) {
246                         eve->co[0]= -td->loc[0];
247                         eve->co[1]= td->loc[1];
248                         eve->co[2]= td->loc[2];
249                 }
250                 
251                 if (td->flag & TD_MIRROR_EDGE)
252                 {
253                         td->loc[0] = 0;
254                 }
255         }
256 }
257
258 /* for the realtime animation recording feature, handle overlapping data */
259 static void animrecord_check_state (Scene *scene, ID *id, wmTimer *animtimer)
260 {
261         ScreenAnimData *sad= (animtimer) ? animtimer->customdata : NULL;
262         
263         /* sanity checks */
264         if ELEM3(NULL, scene, id, sad)
265                 return;
266         
267         /* check if we need a new strip if:
268          *      - if animtimer is running 
269          *      - we're not only keying for available channels
270          *      - the option to add new actions for each round is not enabled
271          */
272         if (IS_AUTOKEY_FLAG(scene, INSERTAVAIL)==0 && (scene->toolsettings->autokey_flag & ANIMRECORD_FLAG_WITHNLA)) {
273                 /* if playback has just looped around, we need to add a new NLA track+strip to allow a clean pass to occur */
274                 if ((sad) && (sad->flag & ANIMPLAY_FLAG_JUMPED)) {
275                         AnimData *adt= BKE_animdata_from_id(id);
276                         
277                         /* perform push-down manually with some differences 
278                          * NOTE: BKE_nla_action_pushdown() sync warning...
279                          */
280                         if ((adt->action) && !(adt->flag & ADT_NLA_EDIT_ON)) {
281                                 float astart, aend;
282                                 
283                                 /* only push down if action is more than 1-2 frames long */
284                                 calc_action_range(adt->action, &astart, &aend, 1);
285                                 if (aend > astart+2.0f) {
286                                         NlaStrip *strip= add_nlastrip_to_stack(adt, adt->action);
287                                         
288                                         /* clear reference to action now that we've pushed it onto the stack */
289                                         adt->action->id.us--;
290                                         adt->action= NULL;
291                                         
292                                         /* adjust blending + extend so that they will behave correctly */
293                                         strip->extendmode= NLASTRIP_EXTEND_NOTHING;
294                                         strip->flag &= ~(NLASTRIP_FLAG_AUTO_BLENDS|NLASTRIP_FLAG_SELECT|NLASTRIP_FLAG_ACTIVE);
295                                         
296                                         /* also, adjust the AnimData's action extend mode to be on 
297                                          * 'nothing' so that previous result still play 
298                                          */
299                                         adt->act_extendmode= NLASTRIP_EXTEND_NOTHING;
300                                 }
301                         }
302                 }
303         }
304 }
305
306 static int fcu_test_selected(FCurve *fcu)
307 {
308         BezTriple *bezt= fcu->bezt;
309         unsigned int i;
310
311         if (bezt==NULL) /* ignore baked */
312                 return 0;
313
314         for (i=0; i < fcu->totvert; i++, bezt++) {
315                 if (BEZSELECTED(bezt)) return 1;
316         }
317
318         return 0;
319 }
320
321 /* called for updating while transform acts, once per redraw */
322 void recalcData(TransInfo *t)
323 {
324         Base *base = t->scene->basact;
325
326         if (t->spacetype==SPACE_NODE) {
327                 flushTransNodes(t);
328         }
329         else if (t->spacetype==SPACE_SEQ) {
330                 flushTransSeq(t);
331         }
332         else if (t->spacetype == SPACE_ACTION) {
333                 Scene *scene= t->scene;
334                 SpaceAction *saction= (SpaceAction *)t->sa->spacedata.first;
335                 
336                 bAnimContext ac= {0};
337                 ListBase anim_data = {NULL, NULL};
338                 bAnimListElem *ale;
339                 int filter;
340                 
341                 /* initialise relevant anim-context 'context' data from TransInfo data */
342                         /* NOTE: sync this with the code in ANIM_animdata_get_context() */
343                 ac.scene= t->scene;
344                 ac.obact= OBACT;
345                 ac.sa= t->sa;
346                 ac.ar= t->ar;
347                 ac.spacetype= (t->sa)? t->sa->spacetype : 0;
348                 ac.regiontype= (t->ar)? t->ar->regiontype : 0;
349                 
350                 ANIM_animdata_context_getdata(&ac);
351                 
352                 /* perform flush */
353                 if (ac.datatype == ANIMCONT_GPENCIL) {
354                         /* flush transform values back to actual coordinates */
355                         flushTransGPactionData(t);
356                 }
357                 else {
358                         /* get animdata blocks visible in editor, assuming that these will be the ones where things changed */
359                         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_ANIMDATA);
360                         ANIM_animdata_filter(&ac, &anim_data, filter, ac.data, ac.datatype);
361                         
362                         /* just tag these animdata-blocks to recalc, assuming that some data there changed 
363                          * BUT only do this if realtime updates are enabled
364                          */
365                         if ((saction->flag & SACTION_NOREALTIMEUPDATES) == 0) {
366                                 for (ale= anim_data.first; ale; ale= ale->next) {
367                                         /* set refresh tags for objects using this animation */
368                                         ANIM_list_elem_update(t->scene, ale);
369                                 }
370                         }
371                         
372                         /* now free temp channels */
373                         BLI_freelistN(&anim_data);
374                 }
375         }
376         else if (t->spacetype == SPACE_IPO) {
377                 Scene *scene;
378                 SpaceIpo *sipo= (SpaceIpo *)t->sa->spacedata.first;
379                 
380                 ListBase anim_data = {NULL, NULL};
381                 bAnimContext ac;
382                 int filter;
383                 
384                 bAnimListElem *ale;
385                 int dosort = 0;
386                 
387                 
388                 /* initialise relevant anim-context 'context' data from TransInfo data */
389                         /* NOTE: sync this with the code in ANIM_animdata_get_context() */
390                 memset(&ac, 0, sizeof(bAnimContext));
391                 
392                 scene= ac.scene= t->scene;
393                 ac.obact= OBACT;
394                 ac.sa= t->sa;
395                 ac.ar= t->ar;
396                 ac.spacetype= (t->sa)? t->sa->spacetype : 0;
397                 ac.regiontype= (t->ar)? t->ar->regiontype : 0;
398                 
399                 ANIM_animdata_context_getdata(&ac);
400                 
401                 /* do the flush first */
402                 flushTransGraphData(t);
403                 
404                 /* get curves to check if a re-sort is needed */
405                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_CURVEVISIBLE);
406                 ANIM_animdata_filter(&ac, &anim_data, filter, ac.data, ac.datatype);
407                 
408                 /* now test if there is a need to re-sort */
409                 for (ale= anim_data.first; ale; ale= ale->next) {
410                         FCurve *fcu= (FCurve *)ale->key_data;
411                         
412                         /* ignore unselected fcurves */
413                         if (!fcu_test_selected(fcu))
414                                 continue;
415                         
416                         // fixme: only do this for selected verts...
417                         ANIM_unit_mapping_apply_fcurve(ac.scene, ale->id, ale->key_data, ANIM_UNITCONV_ONLYSEL|ANIM_UNITCONV_SELVERTS|ANIM_UNITCONV_RESTORE);
418                         
419                         
420                         /* watch it: if the time is wrong: do not correct handles yet */
421                         if (test_time_fcurve(fcu))
422                                 dosort++;
423                         else
424                                 calchandles_fcurve(fcu);
425                         
426                         /* set refresh tags for objects using this animation,
427                          * BUT only if realtime updates are enabled  
428                          */
429                         if ((sipo->flag & SIPO_NOREALTIMEUPDATES) == 0)
430                                 ANIM_list_elem_update(t->scene, ale);
431                 }
432                 
433                 /* do resort and other updates? */
434                 if (dosort) remake_graph_transdata(t, &anim_data);
435                 
436                 /* now free temp channels */
437                 BLI_freelistN(&anim_data);
438         }
439         else if (t->spacetype == SPACE_NLA) {
440                 TransDataNla *tdn= (TransDataNla *)t->customData;
441                 SpaceNla *snla= (SpaceNla *)t->sa->spacedata.first;
442                 Scene *scene= t->scene;
443                 double secf= FPS;
444                 int i;
445                 
446                 /* for each strip we've got, perform some additional validation of the values that got set before
447                  * using RNA to set the value (which does some special operations when setting these values to make
448                  * sure that everything works ok)
449                  */
450                 for (i = 0; i < t->total; i++, tdn++) {
451                         NlaStrip *strip= tdn->strip;
452                         PointerRNA strip_ptr;
453                         short pExceeded, nExceeded, iter;
454                         int delta_y1, delta_y2;
455                         
456                         /* if this tdn has no handles, that means it is just a dummy that should be skipped */
457                         if (tdn->handle == 0)
458                                 continue;
459                         
460                         /* set refresh tags for objects using this animation,
461                          * BUT only if realtime updates are enabled  
462                          */
463                         if ((snla->flag & SNLA_NOREALTIMEUPDATES) == 0)
464                                 ANIM_id_update(t->scene, tdn->id);
465                         
466                         /* if cancelling transform, just write the values without validating, then move on */
467                         if (t->state == TRANS_CANCEL) {
468                                 /* clear the values by directly overwriting the originals, but also need to restore
469                                  * endpoints of neighboring transition-strips
470                                  */
471                                 
472                                 /* start */
473                                 strip->start= tdn->h1[0];
474                                 
475                                 if ((strip->prev) && (strip->prev->type == NLASTRIP_TYPE_TRANSITION))
476                                         strip->prev->end= tdn->h1[0];
477                                 
478                                 /* end */
479                                 strip->end= tdn->h2[0];
480                                 
481                                 if ((strip->next) && (strip->next->type == NLASTRIP_TYPE_TRANSITION))
482                                         strip->next->start= tdn->h2[0];
483                                 
484                                 /* flush transforms to child strips (since this should be a meta) */
485                                 BKE_nlameta_flush_transforms(strip);
486                                 
487                                 /* restore to original track (if needed) */
488                                 if (tdn->oldTrack != tdn->nlt) {
489                                         /* just append to end of list for now, since strips get sorted in special_aftertrans_update() */
490                                         BLI_remlink(&tdn->nlt->strips, strip);
491                                         BLI_addtail(&tdn->oldTrack->strips, strip);
492                                 }
493                                 
494                                 continue;
495                         }
496                         
497                         /* firstly, check if the proposed transform locations would overlap with any neighbouring strips
498                          * (barring transitions) which are absolute barriers since they are not being moved
499                          *
500                          * this is done as a iterative procedure (done 5 times max for now)
501                          */
502                         for (iter=0; iter < 5; iter++) {
503                                 pExceeded= ((strip->prev) && (strip->prev->type != NLASTRIP_TYPE_TRANSITION) && (tdn->h1[0] < strip->prev->end));
504                                 nExceeded= ((strip->next) && (strip->next->type != NLASTRIP_TYPE_TRANSITION) && (tdn->h2[0] > strip->next->start));
505                                 
506                                 if ((pExceeded && nExceeded) || (iter == 4) ) {
507                                         /* both endpoints exceeded (or iteration ping-pong'd meaning that we need a compromise)
508                                          *      - simply crop strip to fit within the bounds of the strips bounding it
509                                          *      - if there were no neighbours, clear the transforms (make it default to the strip's current values)
510                                          */
511                                         if (strip->prev && strip->next) {
512                                                 tdn->h1[0]= strip->prev->end;
513                                                 tdn->h2[0]= strip->next->start;
514                                         }
515                                         else {
516                                                 tdn->h1[0]= strip->start;
517                                                 tdn->h2[0]= strip->end;
518                                         }
519                                 }
520                                 else if (nExceeded) {
521                                         /* move backwards */
522                                         float offset= tdn->h2[0] - strip->next->start;
523                                         
524                                         tdn->h1[0] -= offset;
525                                         tdn->h2[0] -= offset;
526                                 }
527                                 else if (pExceeded) {
528                                         /* more forwards */
529                                         float offset= strip->prev->end - tdn->h1[0];
530                                         
531                                         tdn->h1[0] += offset;
532                                         tdn->h2[0] += offset;
533                                 }
534                                 else /* all is fine and well */
535                                         break;
536                         }
537                         
538                         /* handle auto-snapping */
539                         switch (snla->autosnap) {
540                                 case SACTSNAP_FRAME: /* snap to nearest frame/time  */
541                                         if (snla->flag & SNLA_DRAWTIME) {
542                                                 tdn->h1[0]= (float)( floor((tdn->h1[0]/secf) + 0.5f) * secf );
543                                                 tdn->h2[0]= (float)( floor((tdn->h2[0]/secf) + 0.5f) * secf );
544                                         }
545                                         else {
546                                                 tdn->h1[0]= (float)( floor(tdn->h1[0]+0.5f) );
547                                                 tdn->h2[0]= (float)( floor(tdn->h2[0]+0.5f) );
548                                         }
549                                         break;
550                                 
551                                 case SACTSNAP_MARKER: /* snap to nearest marker */
552                                         tdn->h1[0]= (float)ED_markers_find_nearest_marker_time(&t->scene->markers, tdn->h1[0]);
553                                         tdn->h2[0]= (float)ED_markers_find_nearest_marker_time(&t->scene->markers, tdn->h2[0]);
554                                         break;
555                         }
556                         
557                         /* use RNA to write the values... */
558                         // TODO: do we need to write in 2 passes to make sure that no truncation goes on?
559                         RNA_pointer_create(NULL, &RNA_NlaStrip, strip, &strip_ptr);
560                         
561                         RNA_float_set(&strip_ptr, "frame_start", tdn->h1[0]);
562                         RNA_float_set(&strip_ptr, "frame_end", tdn->h2[0]);
563                         
564                         /* flush transforms to child strips (since this should be a meta) */
565                         BKE_nlameta_flush_transforms(strip);
566                         
567                         
568                         /* now, check if we need to try and move track
569                          *      - we need to calculate both, as only one may have been altered by transform if only 1 handle moved
570                          */
571                         delta_y1= ((int)tdn->h1[1] / NLACHANNEL_STEP - tdn->trackIndex);
572                         delta_y2= ((int)tdn->h2[1] / NLACHANNEL_STEP - tdn->trackIndex);
573                         
574                         if (delta_y1 || delta_y2) {
575                                 NlaTrack *track;
576                                 int delta = (delta_y2) ? delta_y2 : delta_y1;
577                                 int n;
578                                 
579                                 /* move in the requested direction, checking at each layer if there's space for strip to pass through,
580                                  * stopping on the last track available or that we're able to fit in
581                                  */
582                                 if (delta > 0) {
583                                         for (track=tdn->nlt->next, n=0; (track) && (n < delta); track=track->next, n++) {
584                                                 /* check if space in this track for the strip */
585                                                 if (BKE_nlatrack_has_space(track, strip->start, strip->end)) {
586                                                         /* move strip to this track */
587                                                         BLI_remlink(&tdn->nlt->strips, strip);
588                                                         BKE_nlatrack_add_strip(track, strip);
589                                                         
590                                                         tdn->nlt= track;
591                                                         tdn->trackIndex++;
592                                                 }
593                                                 else /* can't move any further */
594                                                         break;
595                                         }
596                                 }
597                                 else {
598                                         /* make delta 'positive' before using it, since we now know to go backwards */
599                                         delta= -delta;
600                                         
601                                         for (track=tdn->nlt->prev, n=0; (track) && (n < delta); track=track->prev, n++) {
602                                                 /* check if space in this track for the strip */
603                                                 if (BKE_nlatrack_has_space(track, strip->start, strip->end)) {
604                                                         /* move strip to this track */
605                                                         BLI_remlink(&tdn->nlt->strips, strip);
606                                                         BKE_nlatrack_add_strip(track, strip);
607                                                         
608                                                         tdn->nlt= track;
609                                                         tdn->trackIndex--;
610                                                 }
611                                                 else /* can't move any further */
612                                                         break;
613                                         }
614                                 }
615                         }
616                 }
617         }
618         else if (t->spacetype == SPACE_IMAGE) {
619                 if (t->obedit && t->obedit->type == OB_MESH) {
620                         SpaceImage *sima= t->sa->spacedata.first;
621                         
622                         flushTransUVs(t);
623                         if(sima->flag & SI_LIVE_UNWRAP)
624                                 ED_uvedit_live_unwrap_re_solve();
625                         
626                         DAG_id_tag_update(t->obedit->data, 0);
627                 }
628         }
629         else if (t->spacetype == SPACE_VIEW3D) {
630                 
631                 if (t->obedit) {
632                         if ELEM(t->obedit->type, OB_CURVE, OB_SURF) {
633                                 Curve *cu= t->obedit->data;
634                                 ListBase *nurbs= ED_curve_editnurbs(cu);
635                                 Nurb *nu= nurbs->first;
636
637                                 if(t->state != TRANS_CANCEL) {
638                                         clipMirrorModifier(t, t->obedit);
639                                         applyProject(t);
640                                 }
641
642                                 DAG_id_tag_update(t->obedit->data, 0);  /* sets recalc flags */
643
644                                 if (t->state == TRANS_CANCEL) {
645                                         while(nu) {
646                                                 calchandlesNurb(nu); /* Cant do testhandlesNurb here, it messes up the h1 and h2 flags */
647                                                 nu= nu->next;
648                                         }
649                                 } else {
650                                         /* Normal updating */
651                                         while(nu) {
652                                                 test2DNurb(nu);
653                                                 calchandlesNurb(nu);
654                                                 nu= nu->next;
655                                         }
656                                 }
657                         }
658                         else if(t->obedit->type==OB_LATTICE) {
659                                 Lattice *la= t->obedit->data;
660
661                                 if(t->state != TRANS_CANCEL) {
662                                         applyProject(t);
663                                 }
664
665                                 DAG_id_tag_update(t->obedit->data, 0);  /* sets recalc flags */
666         
667                                 if(la->editlatt->latt->flag & LT_OUTSIDE) outside_lattice(la->editlatt->latt);
668                         }
669                         else if (t->obedit->type == OB_MESH) {
670                                 EditMesh *em = ((Mesh*)t->obedit->data)->edit_mesh;
671                                 /* mirror modifier clipping? */
672                                 if(t->state != TRANS_CANCEL) {
673                                         clipMirrorModifier(t, t->obedit);
674                                         applyProject(t);
675                                 }
676                                 if((t->options & CTX_NO_MIRROR) == 0 && (t->flag & T_MIRROR))
677                                         editmesh_apply_to_mirror(t);
678                                         
679                                 DAG_id_tag_update(t->obedit->data, 0);  /* sets recalc flags */
680                                 
681                                 recalc_editnormals(em);
682                         }
683                         else if(t->obedit->type==OB_ARMATURE) { /* no recalc flag, does pose */
684                                 bArmature *arm= t->obedit->data;
685                                 ListBase *edbo = arm->edbo;
686                                 EditBone *ebo;
687                                 TransData *td = t->data;
688                                 int i;
689                                 
690                                 if(t->state != TRANS_CANCEL) {
691                                         applyProject(t);
692                                 }
693
694                                 /* Ensure all bones are correctly adjusted */
695                                 for (ebo = edbo->first; ebo; ebo = ebo->next){
696                                         
697                                         if ((ebo->flag & BONE_CONNECTED) && ebo->parent){
698                                                 /* If this bone has a parent tip that has been moved */
699                                                 if (ebo->parent->flag & BONE_TIPSEL){
700                                                         VECCOPY (ebo->head, ebo->parent->tail);
701                                                         if(t->mode==TFM_BONE_ENVELOPE) ebo->rad_head= ebo->parent->rad_tail;
702                                                 }
703                                                 /* If this bone has a parent tip that has NOT been moved */
704                                                 else{
705                                                         VECCOPY (ebo->parent->tail, ebo->head);
706                                                         if(t->mode==TFM_BONE_ENVELOPE) ebo->parent->rad_tail= ebo->rad_head;
707                                                 }
708                                         }
709                                         
710                                         /* on extrude bones, oldlength==0.0f, so we scale radius of points */
711                                         ebo->length= len_v3v3(ebo->head, ebo->tail);
712                                         if(ebo->oldlength==0.0f) {
713                                                 ebo->rad_head= 0.25f*ebo->length;
714                                                 ebo->rad_tail= 0.10f*ebo->length;
715                                                 ebo->dist= 0.25f*ebo->length;
716                                                 if(ebo->parent) {
717                                                         if(ebo->rad_head > ebo->parent->rad_tail)
718                                                                 ebo->rad_head= ebo->parent->rad_tail;
719                                                 }
720                                         }
721                                         else if(t->mode!=TFM_BONE_ENVELOPE) {
722                                                 /* if bones change length, lets do that for the deform distance as well */
723                                                 ebo->dist*= ebo->length/ebo->oldlength;
724                                                 ebo->rad_head*= ebo->length/ebo->oldlength;
725                                                 ebo->rad_tail*= ebo->length/ebo->oldlength;
726                                                 ebo->oldlength= ebo->length;
727                                         }
728                                 }
729                                 
730                                 
731                                 if (t->mode != TFM_BONE_ROLL)
732                                 {
733                                         /* fix roll */
734                                         for(i = 0; i < t->total; i++, td++)
735                                         {
736                                                 if (td->extra)
737                                                 {
738                                                         float vec[3], up_axis[3];
739                                                         float qrot[4];
740                                                         
741                                                         ebo = td->extra;
742                                                         VECCOPY(up_axis, td->axismtx[2]);
743                                                         
744                                                         if (t->mode != TFM_ROTATION)
745                                                         {
746                                                                 sub_v3_v3v3(vec, ebo->tail, ebo->head);
747                                                                 normalize_v3(vec);
748                                                                 rotation_between_vecs_to_quat(qrot, td->axismtx[1], vec);
749                                                                 mul_qt_v3(qrot, up_axis);
750                                                         }
751                                                         else
752                                                         {
753                                                                 mul_m3_v3(t->mat, up_axis);
754                                                         }
755                                                         
756                                                         ebo->roll = ED_rollBoneToVector(ebo, up_axis, FALSE);
757                                                 }
758                                         }
759                                 }
760                                 
761                                 if(arm->flag & ARM_MIRROR_EDIT)
762                                         transform_armature_mirror_update(t->obedit);
763                                 
764                         }
765                         else
766                         {
767                                 if(t->state != TRANS_CANCEL) {
768                                         applyProject(t);
769                                 }
770                                 DAG_id_tag_update(t->obedit->data, 0);  /* sets recalc flags */
771                         }
772                 }
773                 else if( (t->flag & T_POSE) && t->poseobj) {
774                         Object *ob= t->poseobj;
775                         bArmature *arm= ob->data;
776                         
777                         /* if animtimer is running, and the object already has animation data,
778                          * check if the auto-record feature means that we should record 'samples'
779                          * (i.e. uneditable animation values)
780                          *
781                          * context is needed for keying set poll() functions.
782                          */
783                         // TODO: autokeyframe calls need some setting to specify to add samples (FPoints) instead of keyframes?
784                         if ((t->animtimer) && (t->context) && IS_AUTOKEY_ON(t->scene)) {
785                                 int targetless_ik= (t->flag & T_AUTOIK); // XXX this currently doesn't work, since flags aren't set yet!
786                                 
787                                 animrecord_check_state(t->scene, &ob->id, t->animtimer);
788                                 autokeyframe_pose_cb_func(t->context, t->scene, (View3D *)t->view, ob, t->mode, targetless_ik);
789                         }
790                         
791                         /* old optimize trick... this enforces to bypass the depgraph */
792                         if (!(arm->flag & ARM_DELAYDEFORM)) {
793                                 DAG_id_tag_update(&ob->id, OB_RECALC_DATA);  /* sets recalc flags */
794                         }
795                         else
796                                 where_is_pose(t->scene, ob);
797                 }
798                 else if(base && (base->object->mode & OB_MODE_PARTICLE_EDIT) && PE_get_current(t->scene, base->object)) {
799                         if(t->state != TRANS_CANCEL) {
800                                 applyProject(t);
801                         }
802                         flushTransParticles(t);
803                 }
804                 else {
805                         int i;
806                         
807                         if(t->state != TRANS_CANCEL) {
808                                 applyProject(t);
809                         }
810
811                         for (i = 0; i < t->total; i++) {
812                                 TransData *td = t->data + i;
813                                 Object *ob = td->ob;
814                                 
815                                 if (td->flag & TD_NOACTION)
816                                         break;
817                                 
818                                 if (td->flag & TD_SKIP)
819                                         continue;
820                                 
821                                 /* if animtimer is running, and the object already has animation data,
822                                  * check if the auto-record feature means that we should record 'samples'
823                                  * (i.e. uneditable animation values)
824                                  */
825                                 // TODO: autokeyframe calls need some setting to specify to add samples (FPoints) instead of keyframes?
826                                 if ((t->animtimer) && IS_AUTOKEY_ON(t->scene)) {
827                                         animrecord_check_state(t->scene, &ob->id, t->animtimer);
828                                         autokeyframe_ob_cb_func(t->context, t->scene, (View3D *)t->view, ob, t->mode);
829                                 }
830                                 
831                                 /* sets recalc flags fully, instead of flushing existing ones 
832                                  * otherwise proxies don't function correctly
833                                  */
834                                 DAG_id_tag_update(&ob->id, OB_RECALC_OB);
835                         }
836                 }
837                 
838                 if(((View3D*)t->view)->drawtype == OB_SHADED)
839                         reshadeall_displist(t->scene);
840         }
841 }
842
843 void drawLine(TransInfo *t, float *center, float *dir, char axis, short options)
844 {
845         float v1[3], v2[3], v3[3];
846         unsigned char col[3], col2[3];
847
848         if (t->spacetype == SPACE_VIEW3D)
849         {
850                 View3D *v3d = t->view;
851                 
852                 glPushMatrix();
853                 
854                 //if(t->obedit) glLoadMatrixf(t->obedit->obmat);        // sets opengl viewing
855                 
856                 
857                 copy_v3_v3(v3, dir);
858                 mul_v3_fl(v3, v3d->far);
859                 
860                 sub_v3_v3v3(v2, center, v3);
861                 add_v3_v3v3(v1, center, v3);
862                 
863                 if (options & DRAWLIGHT) {
864                         col[0] = col[1] = col[2] = 220;
865                 }
866                 else {
867                         UI_GetThemeColor3ubv(TH_GRID, col);
868                 }
869                 UI_make_axis_color(col, col2, axis);
870                 glColor3ubv(col2);
871                 
872                 setlinestyle(0);
873                 glBegin(GL_LINE_STRIP);
874                         glVertex3fv(v1);
875                         glVertex3fv(v2);
876                 glEnd();
877                 
878                 glPopMatrix();
879         }
880 }
881
882 void resetTransRestrictions(TransInfo *t)
883 {
884         t->flag &= ~T_ALL_RESTRICTIONS;
885 }
886
887 int initTransInfo (bContext *C, TransInfo *t, wmOperator *op, wmEvent *event)
888 {
889         Scene *sce = CTX_data_scene(C);
890         ToolSettings *ts = CTX_data_tool_settings(C);
891         ARegion *ar = CTX_wm_region(C);
892         ScrArea *sa = CTX_wm_area(C);
893         Object *obedit = CTX_data_edit_object(C);
894         
895         /* moving: is shown in drawobject() (transform color) */
896 //  TRANSFORM_FIX_ME
897 //      if(obedit || (t->flag & T_POSE) ) G.moving= G_TRANSFORM_EDIT;
898 //      else if(G.f & G_PARTICLEEDIT) G.moving= G_TRANSFORM_PARTICLE;
899 //      else G.moving= G_TRANSFORM_OBJ;
900         
901         t->scene = sce;
902         t->sa = sa;
903         t->ar = ar;
904         t->obedit = obedit;
905         t->settings = ts;
906         
907         t->data = NULL;
908         t->ext = NULL;
909         
910         t->helpline = HLP_NONE;
911         
912         t->flag = 0;
913         
914         t->redraw = 1; /* redraw first time */
915         
916         if (event)
917         {
918                 t->imval[0] = event->x - t->ar->winrct.xmin;
919                 t->imval[1] = event->y - t->ar->winrct.ymin;
920                 
921                 t->event_type = event->type;
922         }
923         else
924         {
925                 t->imval[0] = 0;
926                 t->imval[1] = 0;
927         }
928         
929         t->con.imval[0] = t->imval[0];
930         t->con.imval[1] = t->imval[1];
931         
932         t->mval[0] = t->imval[0];
933         t->mval[1] = t->imval[1];
934         
935         t->transform            = NULL;
936         t->handleEvent          = NULL;
937         
938         t->total                        = 0;
939         
940         t->val = 0.0f;
941         
942         t->vec[0]                       =
943                 t->vec[1]               =
944                 t->vec[2]               = 0.0f;
945         
946         t->center[0]            =
947                 t->center[1]    =
948                 t->center[2]    = 0.0f;
949         
950         unit_m3(t->mat);
951         
952         /* if there's an event, we're modal */
953         if (event) {
954                 t->flag |= T_MODAL;
955         }
956
957         /* Crease needs edge flag */
958         if (t->mode == TFM_CREASE) {
959                 t->options |= CTX_EDGE;
960         }
961
962         t->spacetype = sa->spacetype;
963         if(t->spacetype == SPACE_VIEW3D)
964         {
965                 View3D *v3d = sa->spacedata.first;
966                 
967                 t->view = v3d;
968                 t->animtimer= CTX_wm_screen(C)->animtimer;
969                 
970                 /* turn manipulator off during transform */
971                 // FIXME: but don't do this when USING the manipulator...
972                 if (t->flag & T_MODAL) {
973                         t->twtype = v3d->twtype;
974                         v3d->twtype = 0;
975                 }
976
977                 if(v3d->flag & V3D_ALIGN) t->flag |= T_V3D_ALIGN;
978                 t->around = v3d->around;
979                 
980                 if (op && RNA_struct_find_property(op->ptr, "constraint_orientation") && RNA_property_is_set(op->ptr, "constraint_orientation"))
981                 {
982                         t->current_orientation = RNA_enum_get(op->ptr, "constraint_orientation");
983                         
984                         if (t->current_orientation >= V3D_MANIP_CUSTOM + BIF_countTransformOrientation(C))
985                         {
986                                 t->current_orientation = V3D_MANIP_GLOBAL;
987                         }
988                 }
989                 else
990                 {
991                         t->current_orientation = v3d->twmode;
992                 }
993
994                 /* exceptional case */
995                 if(t->around==V3D_LOCAL && (t->settings->selectmode & SCE_SELECT_FACE)) {
996                         if(ELEM3(t->mode, TFM_ROTATION, TFM_RESIZE, TFM_TRACKBALL)) {
997                                 t->options |= CTX_NO_PET;
998                         }
999                 }
1000         }
1001         else if(t->spacetype==SPACE_IMAGE || t->spacetype==SPACE_NODE)
1002         {
1003                 SpaceImage *sima = sa->spacedata.first;
1004                 // XXX for now, get View2D  from the active region
1005                 t->view = &ar->v2d;
1006                 t->around = sima->around;
1007         }
1008         else if(t->spacetype==SPACE_IPO) 
1009         {
1010                 SpaceIpo *sipo= sa->spacedata.first;
1011                 t->view = &ar->v2d;
1012                 t->around = sipo->around;
1013         }
1014         else
1015         {
1016                 // XXX for now, get View2D  from the active region
1017                 t->view = &ar->v2d;
1018                 // XXX for now, the center point is the midpoint of the data
1019                 t->around = V3D_CENTER;
1020         }
1021         
1022         if (op && RNA_property_is_set(op->ptr, "release_confirm"))
1023         {
1024                 if (RNA_boolean_get(op->ptr, "release_confirm"))
1025                 {
1026                         t->flag |= T_RELEASE_CONFIRM;
1027                 }
1028         }
1029         else
1030         {
1031                 if (U.flag & USER_RELEASECONFIRM)
1032                 {
1033                         t->flag |= T_RELEASE_CONFIRM;
1034                 }
1035         }
1036
1037         if (op && RNA_struct_find_property(op->ptr, "mirror") && RNA_property_is_set(op->ptr, "mirror"))
1038         {
1039                 if (RNA_boolean_get(op->ptr, "mirror"))
1040                 {
1041                         t->flag |= T_MIRROR;
1042                         t->mirror = 1;
1043                 }
1044         }
1045         // Need stuff to take it from edit mesh or whatnot here
1046         else
1047         {
1048                 if (t->obedit && t->obedit->type == OB_MESH && (((Mesh *)t->obedit->data)->editflag & ME_EDIT_MIRROR_X))
1049                 {
1050                         t->flag |= T_MIRROR;
1051                         t->mirror = 1;
1052                 }
1053         }
1054         
1055         /* setting PET flag only if property exist in operator. Otherwise, assume it's not supported */
1056         if (op && RNA_struct_find_property(op->ptr, "proportional"))
1057         {
1058                 if (RNA_property_is_set(op->ptr, "proportional"))
1059                 {
1060                         switch(RNA_enum_get(op->ptr, "proportional"))
1061                         {
1062                         case PROP_EDIT_CONNECTED:
1063                                 t->flag |= T_PROP_CONNECTED;
1064                         case PROP_EDIT_ON:
1065                                 t->flag |= T_PROP_EDIT;
1066                                 break;
1067                         }
1068                 }
1069                 else
1070                 {
1071                         /* use settings from scene only if modal */
1072                         if (t->flag & T_MODAL)
1073                         {
1074                                 if ((t->options & CTX_NO_PET) == 0)
1075                                 {
1076                                         if (t->obedit && ts->proportional != PROP_EDIT_OFF)
1077                                         {
1078                                                 t->flag |= T_PROP_EDIT;
1079
1080                                                 if(ts->proportional == PROP_EDIT_CONNECTED)
1081                                                         t->flag |= T_PROP_CONNECTED;
1082                                         }
1083                                         else if (t->obedit == NULL && ts->proportional_objects)
1084                                         {
1085                                                 t->flag |= T_PROP_EDIT;
1086                                         }
1087                                 }
1088                         }
1089                 }
1090                 
1091                 if (op && RNA_struct_find_property(op->ptr, "proportional_size") && RNA_property_is_set(op->ptr, "proportional_size"))
1092                 {
1093                         t->prop_size = RNA_float_get(op->ptr, "proportional_size");
1094                 }
1095                 else
1096                 {
1097                         t->prop_size = ts->proportional_size;
1098                 }
1099                 
1100                 
1101                 /* TRANSFORM_FIX_ME rna restrictions */
1102                 if (t->prop_size <= 0.00001f)
1103                 {
1104                         printf("Proportional size (%f) under 0.00001, reseting to 1!\n", t->prop_size);
1105                         t->prop_size = 1.0f;
1106                 }
1107                 
1108                 if (op && RNA_struct_find_property(op->ptr, "proportional_edit_falloff") && RNA_property_is_set(op->ptr, "proportional_edit_falloff"))
1109                 {
1110                         t->prop_mode = RNA_enum_get(op->ptr, "proportional_edit_falloff");
1111                 }
1112                 else
1113                 {
1114                         t->prop_mode = ts->prop_mode;
1115                 }
1116         }
1117         else /* add not pet option to context when not available */
1118         {
1119                 t->options |= CTX_NO_PET;
1120         }
1121         
1122         setTransformViewMatrices(t);
1123         initNumInput(&t->num);
1124         initNDofInput(&t->ndof);
1125         
1126         return 1;
1127 }
1128
1129 /* Here I would suggest only TransInfo related issues, like free data & reset vars. Not redraws */
1130 void postTrans (bContext *C, TransInfo *t)
1131 {
1132         TransData *td;
1133         
1134         if (t->draw_handle_view)
1135                 ED_region_draw_cb_exit(t->ar->type, t->draw_handle_view);
1136         if (t->draw_handle_apply)
1137                 ED_region_draw_cb_exit(t->ar->type, t->draw_handle_apply);
1138         if (t->draw_handle_pixel)
1139                 ED_region_draw_cb_exit(t->ar->type, t->draw_handle_pixel);
1140         if (t->draw_handle_cursor)
1141                 WM_paint_cursor_end(CTX_wm_manager(C), t->draw_handle_cursor);
1142
1143         if (t->customFree) {
1144                 /* Can take over freeing t->data and data2d etc... */
1145                 t->customFree(t);
1146         }
1147         else if (t->customData) {
1148                 MEM_freeN(t->customData);
1149         }
1150
1151         /* postTrans can be called when nothing is selected, so data is NULL already */
1152         if (t->data) {
1153                 int a;
1154                 
1155                 /* free data malloced per trans-data */
1156                 for(a=0, td= t->data; a<t->total; a++, td++) {
1157                         if (td->flag & TD_BEZTRIPLE) 
1158                                 MEM_freeN(td->hdata);
1159                 }
1160                 MEM_freeN(t->data);
1161         }
1162         
1163         BLI_freelistN(&t->tsnap.points);
1164
1165         if (t->ext) MEM_freeN(t->ext);
1166         if (t->data2d) {
1167                 MEM_freeN(t->data2d);
1168                 t->data2d= NULL;
1169         }
1170         
1171         if(t->spacetype==SPACE_IMAGE) {
1172                 SpaceImage *sima= t->sa->spacedata.first;
1173                 if(sima->flag & SI_LIVE_UNWRAP)
1174                         ED_uvedit_live_unwrap_end(t->state == TRANS_CANCEL);
1175         }
1176         else if(t->spacetype==SPACE_VIEW3D) {
1177                 View3D *v3d = t->sa->spacedata.first;
1178                 /* restore manipulator */
1179                 if (t->flag & T_MODAL) {
1180                         v3d->twtype = t->twtype;
1181                 }
1182         }
1183         
1184         if (t->mouse.data)
1185         {
1186                 MEM_freeN(t->mouse.data);
1187         }
1188 }
1189
1190 void applyTransObjects(TransInfo *t)
1191 {
1192         TransData *td;
1193         
1194         for (td = t->data; td < t->data + t->total; td++) {
1195                 VECCOPY(td->iloc, td->loc);
1196                 if (td->ext->rot) {
1197                         VECCOPY(td->ext->irot, td->ext->rot);
1198                 }
1199                 if (td->ext->size) {
1200                         VECCOPY(td->ext->isize, td->ext->size);
1201                 }
1202         }
1203         recalcData(t);
1204 }
1205
1206 static void restoreElement(TransData *td) {
1207         /* TransData for crease has no loc */
1208         if (td->loc) {
1209                 VECCOPY(td->loc, td->iloc);
1210         }
1211         if (td->val) {
1212                 *td->val = td->ival;
1213         }
1214
1215         if (td->ext && (td->flag&TD_NO_EXT)==0) {
1216                 if (td->ext->rot) {
1217                         VECCOPY(td->ext->rot, td->ext->irot);
1218                 }
1219                 if(td->ext->rotAngle) {
1220                         *td->ext->rotAngle= td->ext->irotAngle;
1221                 }
1222                 if(td->ext->rotAxis) {
1223                         VECCOPY(td->ext->rotAxis, td->ext->irotAxis);
1224                 }
1225                 /* XXX, drotAngle & drotAxis not used yet */
1226                 if (td->ext->size) {
1227                         VECCOPY(td->ext->size, td->ext->isize);
1228                 }
1229                 if (td->ext->quat) {
1230                         QUATCOPY(td->ext->quat, td->ext->iquat);
1231                 }
1232         }
1233         
1234         if (td->flag & TD_BEZTRIPLE) {
1235                 *(td->hdata->h1) = td->hdata->ih1;
1236                 *(td->hdata->h2) = td->hdata->ih2;
1237         }
1238 }
1239
1240 void restoreTransObjects(TransInfo *t)
1241 {
1242         TransData *td;
1243         TransData2D *td2d;
1244
1245         for (td = t->data; td < t->data + t->total; td++) {
1246                 restoreElement(td);
1247         }
1248         
1249         for (td2d=t->data2d; t->data2d && td2d < t->data2d + t->total; td2d++) {
1250                 if (td2d->h1) {
1251                         td2d->h1[0] = td2d->ih1[0];
1252                         td2d->h1[1] = td2d->ih1[1];
1253                 }
1254                 if (td2d->h2) {
1255                         td2d->h2[0] = td2d->ih2[0];
1256                         td2d->h2[1] = td2d->ih2[1];
1257                 }
1258         }
1259
1260         unit_m3(t->mat);
1261         
1262         recalcData(t);
1263 }
1264
1265 void calculateCenter2D(TransInfo *t)
1266 {
1267         if (t->flag & (T_EDIT|T_POSE)) {
1268                 Object *ob= t->obedit?t->obedit:t->poseobj;
1269                 float vec[3];
1270                 
1271                 VECCOPY(vec, t->center);
1272                 mul_m4_v3(ob->obmat, vec);
1273                 projectIntView(t, vec, t->center2d);
1274         }
1275         else {
1276                 projectIntView(t, t->center, t->center2d);
1277         }
1278 }
1279
1280 void calculateCenterCursor(TransInfo *t)
1281 {
1282         float *cursor;
1283         
1284         cursor = give_cursor(t->scene, t->view);
1285         VECCOPY(t->center, cursor);
1286         
1287         /* If edit or pose mode, move cursor in local space */
1288         if (t->flag & (T_EDIT|T_POSE)) {
1289                 Object *ob = t->obedit?t->obedit:t->poseobj;
1290                 float mat[3][3], imat[3][3];
1291                 
1292                 sub_v3_v3v3(t->center, t->center, ob->obmat[3]);
1293                 copy_m3_m4(mat, ob->obmat);
1294                 invert_m3_m3(imat, mat);
1295                 mul_m3_v3(imat, t->center);
1296         }
1297         
1298         calculateCenter2D(t);
1299 }
1300
1301 void calculateCenterCursor2D(TransInfo *t)
1302 {
1303         float aspx=1.0, aspy=1.0;
1304         float *cursor= NULL;
1305         
1306         if(t->spacetype==SPACE_IMAGE) {
1307                 SpaceImage *sima= (SpaceImage *)t->sa->spacedata.first;
1308                 /* only space supported right now but may change */
1309                 ED_space_image_uv_aspect(sima, &aspx, &aspy);
1310                 cursor = sima->cursor;
1311         }
1312         
1313         if (cursor) {
1314                 t->center[0] = cursor[0] * aspx;
1315                 t->center[1] = cursor[1] * aspy;
1316         }
1317         
1318         calculateCenter2D(t);
1319 }
1320
1321 void calculateCenterCursorGraph2D(TransInfo *t)
1322 {
1323         SpaceIpo *sipo= (SpaceIpo *)t->sa->spacedata.first;
1324         Scene *scene= t->scene;
1325         
1326         /* cursor is combination of current frame, and graph-editor cursor value */
1327         t->center[0]= (float)(scene->r.cfra);
1328         t->center[1]= sipo->cursorVal;
1329         
1330         calculateCenter2D(t);
1331 }
1332
1333 void calculateCenterMedian(TransInfo *t)
1334 {
1335         float partial[3] = {0.0f, 0.0f, 0.0f};
1336         int total = 0;
1337         int i;
1338         
1339         for(i = 0; i < t->total; i++) {
1340                 if (t->data[i].flag & TD_SELECTED) {
1341                         if (!(t->data[i].flag & TD_NOCENTER))
1342                         {
1343                                 add_v3_v3(partial, t->data[i].center);
1344                                 total++;
1345                         }
1346                 }
1347                 else {
1348                         /*
1349                            All the selected elements are at the head of the array
1350                            which means we can stop when it finds unselected data
1351                         */
1352                         break;
1353                 }
1354         }
1355         if(i)
1356                 mul_v3_fl(partial, 1.0f / total);
1357         VECCOPY(t->center, partial);
1358         
1359         calculateCenter2D(t);
1360 }
1361
1362 void calculateCenterBound(TransInfo *t)
1363 {
1364         float max[3];
1365         float min[3];
1366         int i;
1367         for(i = 0; i < t->total; i++) {
1368                 if (i) {
1369                         if (t->data[i].flag & TD_SELECTED) {
1370                                 if (!(t->data[i].flag & TD_NOCENTER))
1371                                         minmax_v3v3_v3(min, max, t->data[i].center);
1372                         }
1373                         else {
1374                                 /*
1375                                    All the selected elements are at the head of the array
1376                                    which means we can stop when it finds unselected data
1377                                 */
1378                                 break;
1379                         }
1380                 }
1381                 else {
1382                         VECCOPY(max, t->data[i].center);
1383                         VECCOPY(min, t->data[i].center);
1384                 }
1385         }
1386         add_v3_v3v3(t->center, min, max);
1387         mul_v3_fl(t->center, 0.5);
1388         
1389         calculateCenter2D(t);
1390 }
1391
1392 void calculateCenter(TransInfo *t)
1393 {
1394         switch(t->around) {
1395         case V3D_CENTER:
1396                 calculateCenterBound(t);
1397                 break;
1398         case V3D_CENTROID:
1399                 calculateCenterMedian(t);
1400                 break;
1401         case V3D_CURSOR:
1402                 if(t->spacetype==SPACE_IMAGE)
1403                         calculateCenterCursor2D(t);
1404                 else if(t->spacetype==SPACE_IPO)
1405                         calculateCenterCursorGraph2D(t);
1406                 else
1407                         calculateCenterCursor(t);
1408                 break;
1409         case V3D_LOCAL:
1410                 /* Individual element center uses median center for helpline and such */
1411                 calculateCenterMedian(t);
1412                 break;
1413         case V3D_ACTIVE:
1414                 {
1415                 /* set median, and if if if... do object center */
1416                 
1417                 /* EDIT MODE ACTIVE EDITMODE ELEMENT */
1418
1419                 if (t->obedit && t->obedit->type == OB_MESH) {
1420                         EditSelection ese;
1421                         EditMesh *em = BKE_mesh_get_editmesh(t->obedit->data);
1422                         
1423                         if (EM_get_actSelection(em, &ese)) {
1424                                 EM_editselection_center(t->center, &ese);
1425                                 calculateCenter2D(t);
1426                                 break;
1427                         }
1428                 } /* END EDIT MODE ACTIVE ELEMENT */
1429                 
1430                 calculateCenterMedian(t);
1431                 if((t->flag & (T_EDIT|T_POSE))==0)
1432                 {
1433                         Scene *scene = t->scene;
1434                         Object *ob= OBACT;
1435                         if(ob)
1436                         {
1437                                 VECCOPY(t->center, ob->obmat[3]);
1438                                 projectIntView(t, t->center, t->center2d);
1439                         }
1440                 }
1441                 
1442                 }
1443         }
1444         
1445         /* setting constraint center */
1446         VECCOPY(t->con.center, t->center);
1447         if(t->flag & (T_EDIT|T_POSE))
1448         {
1449                 Object *ob= t->obedit?t->obedit:t->poseobj;
1450                 mul_m4_v3(ob->obmat, t->con.center);
1451         }
1452         
1453         /* for panning from cameraview */
1454         if(t->flag & T_OBJECT)
1455         {
1456                 if(t->spacetype==SPACE_VIEW3D && t->ar && t->ar->regiontype == RGN_TYPE_WINDOW)
1457                 {
1458                         View3D *v3d = t->view;
1459                         Scene *scene = t->scene;
1460                         RegionView3D *rv3d = t->ar->regiondata;
1461                         
1462                         if(v3d->camera == OBACT && rv3d->persp==RV3D_CAMOB)
1463                         {
1464                                 float axis[3];
1465                                 /* persinv is nasty, use viewinv instead, always right */
1466                                 VECCOPY(axis, t->viewinv[2]);
1467                                 normalize_v3(axis);
1468                                 
1469                                 /* 6.0 = 6 grid units */
1470                                 axis[0]= t->center[0]- 6.0f*axis[0];
1471                                 axis[1]= t->center[1]- 6.0f*axis[1];
1472                                 axis[2]= t->center[2]- 6.0f*axis[2];
1473                                 
1474                                 projectIntView(t, axis, t->center2d);
1475                                 
1476                                 /* rotate only needs correct 2d center, grab needs initgrabz() value */
1477                                 if(t->mode==TFM_TRANSLATION)
1478                                 {
1479                                         VECCOPY(t->center, axis);
1480                                         VECCOPY(t->con.center, t->center);
1481                                 }
1482                         }
1483                 }
1484         }
1485         
1486         if(t->spacetype==SPACE_VIEW3D)
1487         {
1488                 /* initgrabz() defines a factor for perspective depth correction, used in window_to_3d_delta() */
1489                 if(t->flag & (T_EDIT|T_POSE)) {
1490                         Object *ob= t->obedit?t->obedit:t->poseobj;
1491                         float vec[3];
1492                         
1493                         VECCOPY(vec, t->center);
1494                         mul_m4_v3(ob->obmat, vec);
1495                         initgrabz(t->ar->regiondata, vec[0], vec[1], vec[2]);
1496                 }
1497                 else {
1498                         initgrabz(t->ar->regiondata, t->center[0], t->center[1], t->center[2]);
1499                 }
1500         }
1501 }
1502
1503 void calculatePropRatio(TransInfo *t)
1504 {
1505         TransData *td = t->data;
1506         int i;
1507         float dist;
1508         short connected = t->flag & T_PROP_CONNECTED;
1509
1510         if (t->flag & T_PROP_EDIT) {
1511                 for(i = 0 ; i < t->total; i++, td++) {
1512                         if (td->flag & TD_SELECTED) {
1513                                 td->factor = 1.0f;
1514                         }
1515                         else if (t->flag & T_MIRROR && td->loc[0] * t->mirror < -0.00001f)
1516                         {
1517                                 td->flag |= TD_SKIP;
1518                                 td->factor = 0.0f;
1519                                 restoreElement(td);
1520                         }
1521                         else if ((connected &&
1522                                                 (td->flag & TD_NOTCONNECTED || td->dist > t->prop_size))
1523                                 ||
1524                                         (connected == 0 &&
1525                                                 td->rdist > t->prop_size)) {
1526                                 /*
1527                                    The elements are sorted according to their dist member in the array,
1528                                    that means we can stop when it finds one element outside of the propsize.
1529                                 */
1530                                 td->flag |= TD_NOACTION;
1531                                 td->factor = 0.0f;
1532                                 restoreElement(td);
1533                         }
1534                         else {
1535                                 /* Use rdist for falloff calculations, it is the real distance */
1536                                 td->flag &= ~TD_NOACTION;
1537                                 
1538                                 if (connected)
1539                                         dist= (t->prop_size-td->dist)/t->prop_size;
1540                                 else
1541                                         dist= (t->prop_size-td->rdist)/t->prop_size;
1542
1543                                 /*
1544                                  * Clamp to positive numbers.
1545                                  * Certain corner cases with connectivity and individual centers
1546                                  * can give values of rdist larger than propsize.
1547                                  */
1548                                 if (dist < 0.0f)
1549                                         dist = 0.0f;
1550                                 
1551                                 switch(t->prop_mode) {
1552                                 case PROP_SHARP:
1553                                         td->factor= dist*dist;
1554                                         break;
1555                                 case PROP_SMOOTH:
1556                                         td->factor= 3.0f*dist*dist - 2.0f*dist*dist*dist;
1557                                         break;
1558                                 case PROP_ROOT:
1559                                         td->factor = (float)sqrt(dist);
1560                                         break;
1561                                 case PROP_LIN:
1562                                         td->factor = dist;
1563                                         break;
1564                                 case PROP_CONST:
1565                                         td->factor = 1.0f;
1566                                         break;
1567                                 case PROP_SPHERE:
1568                                         td->factor = (float)sqrt(2*dist - dist * dist);
1569                                         break;
1570                                 case PROP_RANDOM:
1571                                         BLI_srand( BLI_rand() ); /* random seed */
1572                                         td->factor = BLI_frand()*dist;
1573                                         break;
1574                                 default:
1575                                         td->factor = 1;
1576                                 }
1577                         }
1578                 }
1579                 switch(t->prop_mode) {
1580                 case PROP_SHARP:
1581                         strcpy(t->proptext, "(Sharp)");
1582                         break;
1583                 case PROP_SMOOTH:
1584                         strcpy(t->proptext, "(Smooth)");
1585                         break;
1586                 case PROP_ROOT:
1587                         strcpy(t->proptext, "(Root)");
1588                         break;
1589                 case PROP_LIN:
1590                         strcpy(t->proptext, "(Linear)");
1591                         break;
1592                 case PROP_CONST:
1593                         strcpy(t->proptext, "(Constant)");
1594                         break;
1595                 case PROP_SPHERE:
1596                         strcpy(t->proptext, "(Sphere)");
1597                         break;
1598                 case PROP_RANDOM:
1599                         strcpy(t->proptext, "(Random)");
1600                         break;
1601                 default:
1602                         strcpy(t->proptext, "");
1603                 }
1604         }
1605         else {
1606                 for(i = 0 ; i < t->total; i++, td++) {
1607                         td->factor = 1.0;
1608                 }
1609                 strcpy(t->proptext, "");
1610         }
1611 }