359f6425126f7d9890ef8bcc32d40f052739ec13
[blender-staging.git] / source / blender / editors / transform / transform_generics.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): none yet.
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/editors/transform/transform_generics.c
29  *  \ingroup edtransform
30  */
31
32 #include <string.h>
33 #include <math.h>
34
35 #include "MEM_guardedalloc.h"
36
37 #include "BLI_sys_types.h" /* for intptr_t support */
38
39 #include "DNA_anim_types.h"
40 #include "DNA_armature_types.h"
41 #include "DNA_lattice_types.h"
42 #include "DNA_screen_types.h"
43 #include "DNA_sequence_types.h"
44 #include "DNA_space_types.h"
45 #include "DNA_scene_types.h"
46 #include "DNA_object_types.h"
47 #include "DNA_mesh_types.h"
48 #include "DNA_meshdata_types.h"
49 #include "DNA_view3d_types.h"
50 #include "DNA_modifier_types.h"
51 #include "DNA_movieclip_types.h"
52 #include "DNA_mask_types.h"
53
54 #include "BLI_math.h"
55 #include "BLI_blenlib.h"
56 #include "BLI_rand.h"
57 #include "BLI_utildefines.h"
58
59 #include "BLF_translation.h"
60
61 #include "RNA_access.h"
62
63 #include "BIF_gl.h"
64 #include "BIF_glutil.h"
65
66 #include "BIK_api.h"
67
68 #include "BKE_animsys.h"
69 #include "BKE_action.h"
70 #include "BKE_armature.h"
71 #include "BKE_curve.h"
72 #include "BKE_depsgraph.h"
73 #include "BKE_displist.h"
74 #include "BKE_fcurve.h"
75 #include "BKE_lattice.h"
76 #include "BKE_mesh.h"
77 #include "BKE_nla.h"
78 #include "BKE_context.h"
79 #include "BKE_sequencer.h"
80 #include "BKE_editmesh.h"
81 #include "BKE_tracking.h"
82 #include "BKE_mask.h"
83
84 #include "ED_anim_api.h"
85 #include "ED_armature.h"
86 #include "ED_image.h"
87 #include "ED_keyframing.h"
88 #include "ED_markers.h"
89 #include "ED_mesh.h"
90 #include "ED_particle.h"
91 #include "ED_screen_types.h"
92 #include "ED_space_api.h"
93 #include "ED_uvedit.h"
94 #include "ED_view3d.h"
95 #include "ED_curve.h" /* for curve_editnurbs */
96 #include "ED_clip.h"
97 #include "ED_screen.h"
98
99 #include "WM_types.h"
100 #include "WM_api.h"
101
102 #include "UI_resources.h"
103
104 #include "transform.h"
105
106 /* ************************** Functions *************************** */
107
108 void getViewVector(TransInfo *t, float coord[3], float vec[3])
109 {
110         if (t->persp != RV3D_ORTHO) {
111                 float p1[4], p2[4];
112                 
113                 copy_v3_v3(p1, coord);
114                 p1[3] = 1.0f;
115                 copy_v3_v3(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                 copy_v3_v3(vec, t->viewinv[2]);
129         }
130         normalize_v3(vec);
131 }
132
133 /* ************************** GENERICS **************************** */
134
135
136 static void clipMirrorModifier(TransInfo *t, Object *ob)
137 {
138         ModifierData *md = ob->modifiers.first;
139         float tolerance[3] = {0.0f, 0.0f, 0.0f};
140         int axis = 0;
141         
142         for (; md; md = md->next) {
143                 if ((md->type == eModifierType_Mirror) && (md->mode & eModifierMode_Realtime)) {
144                         MirrorModifierData *mmd = (MirrorModifierData *) md;
145                         
146                         if (mmd->flag & MOD_MIR_CLIPPING) {
147                                 axis = 0;
148                                 if (mmd->flag & MOD_MIR_AXIS_X) {
149                                         axis |= 1;
150                                         tolerance[0] = mmd->tolerance;
151                                 }
152                                 if (mmd->flag & MOD_MIR_AXIS_Y) {
153                                         axis |= 2;
154                                         tolerance[1] = mmd->tolerance;
155                                 }
156                                 if (mmd->flag & MOD_MIR_AXIS_Z) {
157                                         axis |= 4;
158                                         tolerance[2] = mmd->tolerance;
159                                 }
160                                 if (axis) {
161                                         float mtx[4][4], imtx[4][4];
162                                         int i;
163                                         TransData *td = t->data;
164                                         
165                                         if (mmd->mirror_ob) {
166                                                 float obinv[4][4];
167                                                 
168                                                 invert_m4_m4(obinv, mmd->mirror_ob->obmat);
169                                                 mul_m4_m4m4(mtx, obinv, ob->obmat);
170                                                 invert_m4_m4(imtx, mtx);
171                                         }
172                                         
173                                         for (i = 0; i < t->total; i++, td++) {
174                                                 int clip;
175                                                 float loc[3], iloc[3];
176                                                 
177                                                 if (td->flag & TD_NOACTION)
178                                                         break;
179                                                 if (td->loc == NULL)
180                                                         break;
181                                                 
182                                                 if (td->flag & TD_SKIP)
183                                                         continue;
184                                                 
185                                                 copy_v3_v3(loc,  td->loc);
186                                                 copy_v3_v3(iloc, td->iloc);
187                                                 
188                                                 if (mmd->mirror_ob) {
189                                                         mul_m4_v3(mtx, loc);
190                                                         mul_m4_v3(mtx, iloc);
191                                                 }
192                                                 
193                                                 clip = 0;
194                                                 if (axis & 1) {
195                                                         if (fabsf(iloc[0]) <= tolerance[0] ||
196                                                             loc[0] * iloc[0] < 0.0f)
197                                                         {
198                                                                 loc[0] = 0.0f;
199                                                                 clip = 1;
200                                                         }
201                                                 }
202                                                 
203                                                 if (axis & 2) {
204                                                         if (fabsf(iloc[1]) <= tolerance[1] ||
205                                                             loc[1] * iloc[1] < 0.0f)
206                                                         {
207                                                                 loc[1] = 0.0f;
208                                                                 clip = 1;
209                                                         }
210                                                 }
211                                                 if (axis & 4) {
212                                                         if (fabsf(iloc[2]) <= tolerance[2] ||
213                                                             loc[2] * iloc[2] < 0.0f)
214                                                         {
215                                                                 loc[2] = 0.0f;
216                                                                 clip = 1;
217                                                         }
218                                                 }
219                                                 if (clip) {
220                                                         if (mmd->mirror_ob) {
221                                                                 mul_m4_v3(imtx, loc);
222                                                         }
223                                                         copy_v3_v3(td->loc, loc);
224                                                 }
225                                         }
226                                 }
227                                 
228                         }
229                 }
230         }
231 }
232
233 /* assumes obedit set to mesh object */
234 static void editbmesh_apply_to_mirror(TransInfo *t)
235 {
236         TransData *td = t->data;
237         BMVert *eve;
238         int i;
239         
240         for (i = 0; i < t->total; i++, td++) {
241                 if (td->flag & TD_NOACTION)
242                         break;
243                 if (td->loc == NULL)
244                         break;
245                 if (td->flag & TD_SKIP)
246                         continue;
247                 
248                 eve = td->extra;
249                 if (eve) {
250                         eve->co[0] = -td->loc[0];
251                         eve->co[1] = td->loc[1];
252                         eve->co[2] = td->loc[2];
253                 }
254                 
255                 if (td->flag & TD_MIRROR_EDGE) {
256                         td->loc[0] = 0;
257                 }
258         }
259 }
260
261 /* for the realtime animation recording feature, handle overlapping data */
262 static void animrecord_check_state(Scene *scene, ID *id, wmTimer *animtimer)
263 {
264         ScreenAnimData *sad = (animtimer) ? animtimer->customdata : NULL;
265         
266         /* sanity checks */
267         if (ELEM3(NULL, scene, id, sad))
268                 return;
269         
270         /* check if we need a new strip if:
271          *  - if animtimer is running
272          *      - we're not only keying for available channels
273          *      - the option to add new actions for each round is not enabled
274          */
275         if (IS_AUTOKEY_FLAG(scene, INSERTAVAIL) == 0 && (scene->toolsettings->autokey_flag & ANIMRECORD_FLAG_WITHNLA)) {
276                 /* if playback has just looped around, we need to add a new NLA track+strip to allow a clean pass to occur */
277                 if ((sad) && (sad->flag & ANIMPLAY_FLAG_JUMPED)) {
278                         AnimData *adt = BKE_animdata_from_id(id);
279                         
280                         /* perform push-down manually with some differences 
281                          * NOTE: BKE_nla_action_pushdown() sync warning...
282                          */
283                         if ((adt->action) && !(adt->flag & ADT_NLA_EDIT_ON)) {
284                                 float astart, aend;
285                                 
286                                 /* only push down if action is more than 1-2 frames long */
287                                 calc_action_range(adt->action, &astart, &aend, 1);
288                                 if (aend > astart + 2.0f) {
289                                         NlaStrip *strip = add_nlastrip_to_stack(adt, adt->action);
290                                         
291                                         /* clear reference to action now that we've pushed it onto the stack */
292                                         adt->action->id.us--;
293                                         adt->action = NULL;
294                                         
295                                         /* adjust blending + extend so that they will behave correctly */
296                                         strip->extendmode = NLASTRIP_EXTEND_NOTHING;
297                                         strip->flag &= ~(NLASTRIP_FLAG_AUTO_BLENDS | NLASTRIP_FLAG_SELECT | NLASTRIP_FLAG_ACTIVE);
298                                         
299                                         /* also, adjust the AnimData's action extend mode to be on 
300                                          * 'nothing' so that previous result still play 
301                                          */
302                                         adt->act_extendmode = NLASTRIP_EXTEND_NOTHING;
303                                 }
304                         }
305                 }
306         }
307 }
308
309 static int fcu_test_selected(FCurve *fcu)
310 {
311         BezTriple *bezt = fcu->bezt;
312         unsigned int i;
313
314         if (bezt == NULL) /* ignore baked */
315                 return 0;
316
317         for (i = 0; i < fcu->totvert; i++, bezt++) {
318                 if (BEZSELECTED(bezt)) return 1;
319         }
320
321         return 0;
322 }
323
324 /* helper for recalcData() - for Action Editor transforms */
325 static void recalcData_actedit(TransInfo *t)
326 {
327         Scene *scene = t->scene;
328         SpaceAction *saction = (SpaceAction *)t->sa->spacedata.first;
329         
330         bAnimContext ac = {NULL};
331         ListBase anim_data = {NULL, NULL};
332         bAnimListElem *ale;
333         int filter;
334         
335         /* initialize relevant anim-context 'context' data from TransInfo data */
336         /* NOTE: sync this with the code in ANIM_animdata_get_context() */
337         ac.scene = t->scene;
338         ac.obact = OBACT;
339         ac.sa = t->sa;
340         ac.ar = t->ar;
341         ac.sl = (t->sa) ? t->sa->spacedata.first : NULL;
342         ac.spacetype = (t->sa) ? t->sa->spacetype : 0;
343         ac.regiontype = (t->ar) ? t->ar->regiontype : 0;
344         
345         ANIM_animdata_context_getdata(&ac);
346         
347         /* perform flush */
348         if (ELEM(ac.datatype, ANIMCONT_GPENCIL, ANIMCONT_MASK)) {
349                 /* flush transform values back to actual coordinates */
350                 flushTransIntFrameActionData(t);
351         }
352         else {
353                 /* get animdata blocks visible in editor, assuming that these will be the ones where things changed */
354                 filter = (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_ANIMDATA);
355                 ANIM_animdata_filter(&ac, &anim_data, filter, ac.data, ac.datatype);
356                 
357                 /* just tag these animdata-blocks to recalc, assuming that some data there changed 
358                  * BUT only do this if realtime updates are enabled
359                  */
360                 if ((saction->flag & SACTION_NOREALTIMEUPDATES) == 0) {
361                         for (ale = anim_data.first; ale; ale = ale->next) {
362                                 /* set refresh tags for objects using this animation */
363                                 ANIM_list_elem_update(t->scene, ale);
364                         }
365                 }
366                 
367                 /* now free temp channels */
368                 BLI_freelistN(&anim_data);
369         }
370 }
371 /* helper for recalcData() - for Graph Editor transforms */
372 static void recalcData_graphedit(TransInfo *t)
373 {
374         SpaceIpo *sipo = (SpaceIpo *)t->sa->spacedata.first;
375         Scene *scene;
376         
377         ListBase anim_data = {NULL, NULL};
378         bAnimContext ac = {NULL};
379         int filter;
380         
381         bAnimListElem *ale;
382         int dosort = 0;
383
384         const bool use_local_center = checkUseLocalCenter_GraphEdit(t);
385         
386         
387         /* initialize relevant anim-context 'context' data from TransInfo data */
388         /* NOTE: sync this with the code in ANIM_animdata_get_context() */
389         scene = ac.scene = t->scene;
390         ac.obact = OBACT;
391         ac.sa = t->sa;
392         ac.ar = t->ar;
393         ac.sl = (t->sa) ? t->sa->spacedata.first : NULL;
394         ac.spacetype = (t->sa) ? t->sa->spacetype : 0;
395         ac.regiontype = (t->ar) ? t->ar->regiontype : 0;
396         
397         ANIM_animdata_context_getdata(&ac);
398         
399         /* do the flush first */
400         flushTransGraphData(t);
401         
402         /* get curves to check if a re-sort is needed */
403         filter = (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVE_VISIBLE);
404         ANIM_animdata_filter(&ac, &anim_data, filter, ac.data, ac.datatype);
405         
406         /* now test if there is a need to re-sort */
407         for (ale = anim_data.first; ale; ale = ale->next) {
408                 FCurve *fcu = (FCurve *)ale->key_data;
409                 
410                 /* ignore unselected fcurves */
411                 if (!fcu_test_selected(fcu))
412                         continue;
413
414                 ANIM_unit_mapping_apply_fcurve(ac.scene, ale->id, ale->key_data,
415                                                ANIM_UNITCONV_ONLYSEL | ANIM_UNITCONV_SELVERTS | ANIM_UNITCONV_RESTORE |
416                                                (use_local_center ? ANIM_UNITCONV_SKIPKNOTS : 0));
417                 
418                 
419                 /* watch it: if the time is wrong: do not correct handles yet */
420                 if (test_time_fcurve(fcu))
421                         dosort++;
422                 else
423                         calchandles_fcurve(fcu);
424                 
425                 /* set refresh tags for objects using this animation,
426                  * BUT only if realtime updates are enabled  
427                  */
428                 if ((sipo->flag & SIPO_NOREALTIMEUPDATES) == 0)
429                         ANIM_list_elem_update(t->scene, ale);
430         }
431         
432         /* do resort and other updates? */
433         if (dosort) remake_graph_transdata(t, &anim_data);
434         
435         /* now free temp channels */
436         BLI_freelistN(&anim_data);
437 }
438
439 /* helper for recalcData() - for NLA Editor transforms */
440 static void recalcData_nla(TransInfo *t)
441 {
442         TransDataNla *tdn = (TransDataNla *)t->customData;
443         SpaceNla *snla = (SpaceNla *)t->sa->spacedata.first;
444         Scene *scene = t->scene;
445         double secf = FPS;
446         int i;
447         
448         /* for each strip we've got, perform some additional validation of the values that got set before
449          * using RNA to set the value (which does some special operations when setting these values to make
450          * sure that everything works ok)
451          */
452         for (i = 0; i < t->total; i++, tdn++) {
453                 NlaStrip *strip = tdn->strip;
454                 PointerRNA strip_ptr;
455                 short pExceeded, nExceeded, iter;
456                 int delta_y1, delta_y2;
457                 
458                 /* if this tdn has no handles, that means it is just a dummy that should be skipped */
459                 if (tdn->handle == 0)
460                         continue;
461                 
462                 /* set refresh tags for objects using this animation,
463                  * BUT only if realtime updates are enabled  
464                  */
465                 if ((snla->flag & SNLA_NOREALTIMEUPDATES) == 0)
466                         ANIM_id_update(t->scene, tdn->id);
467                 
468                 /* if canceling transform, just write the values without validating, then move on */
469                 if (t->state == TRANS_CANCEL) {
470                         /* clear the values by directly overwriting the originals, but also need to restore
471                          * endpoints of neighboring transition-strips
472                          */
473                         
474                         /* start */
475                         strip->start = tdn->h1[0];
476                         
477                         if ((strip->prev) && (strip->prev->type == NLASTRIP_TYPE_TRANSITION))
478                                 strip->prev->end = tdn->h1[0];
479                         
480                         /* end */
481                         strip->end = tdn->h2[0];
482                         
483                         if ((strip->next) && (strip->next->type == NLASTRIP_TYPE_TRANSITION))
484                                 strip->next->start = tdn->h2[0];
485                         
486                         /* flush transforms to child strips (since this should be a meta) */
487                         BKE_nlameta_flush_transforms(strip);
488                         
489                         /* restore to original track (if needed) */
490                         if (tdn->oldTrack != tdn->nlt) {
491                                 /* just append to end of list for now, since strips get sorted in special_aftertrans_update() */
492                                 BLI_remlink(&tdn->nlt->strips, strip);
493                                 BLI_addtail(&tdn->oldTrack->strips, strip);
494                         }
495                         
496                         continue;
497                 }
498                 
499                 /* firstly, check if the proposed transform locations would overlap with any neighboring strips
500                  * (barring transitions) which are absolute barriers since they are not being moved
501                  *
502                  * this is done as a iterative procedure (done 5 times max for now)
503                  */
504                 for (iter = 0; iter < 5; iter++) {
505                         pExceeded = ((strip->prev) && (strip->prev->type != NLASTRIP_TYPE_TRANSITION) && (tdn->h1[0] < strip->prev->end));
506                         nExceeded = ((strip->next) && (strip->next->type != NLASTRIP_TYPE_TRANSITION) && (tdn->h2[0] > strip->next->start));
507                         
508                         if ((pExceeded && nExceeded) || (iter == 4) ) {
509                                 /* both endpoints exceeded (or iteration ping-pong'd meaning that we need a compromise)
510                                  *      - simply crop strip to fit within the bounds of the strips bounding it
511                                  *      - if there were no neighbors, clear the transforms (make it default to the strip's current values)
512                                  */
513                                 if (strip->prev && strip->next) {
514                                         tdn->h1[0] = strip->prev->end;
515                                         tdn->h2[0] = strip->next->start;
516                                 }
517                                 else {
518                                         tdn->h1[0] = strip->start;
519                                         tdn->h2[0] = strip->end;
520                                 }
521                         }
522                         else if (nExceeded) {
523                                 /* move backwards */
524                                 float offset = tdn->h2[0] - strip->next->start;
525                                 
526                                 tdn->h1[0] -= offset;
527                                 tdn->h2[0] -= offset;
528                         }
529                         else if (pExceeded) {
530                                 /* more forwards */
531                                 float offset = strip->prev->end - tdn->h1[0];
532                                 
533                                 tdn->h1[0] += offset;
534                                 tdn->h2[0] += offset;
535                         }
536                         else /* all is fine and well */
537                                 break;
538                 }
539                 
540                 /* handle auto-snapping */
541                 switch (snla->autosnap) {
542                         case SACTSNAP_FRAME: /* snap to nearest frame/time  */
543                                 if (snla->flag & SNLA_DRAWTIME) {
544                                         tdn->h1[0] = (float)(floor(((double)tdn->h1[0] / secf) + 0.5) * secf);
545                                         tdn->h2[0] = (float)(floor(((double)tdn->h2[0] / secf) + 0.5) * secf);
546                                 }
547                                 else {
548                                         tdn->h1[0] = floorf(tdn->h1[0] + 0.5f);
549                                         tdn->h2[0] = floorf(tdn->h2[0] + 0.5f);
550                                 }
551                                 break;
552                         
553                         case SACTSNAP_MARKER: /* snap to nearest marker */
554                                 tdn->h1[0] = (float)ED_markers_find_nearest_marker_time(&t->scene->markers, tdn->h1[0]);
555                                 tdn->h2[0] = (float)ED_markers_find_nearest_marker_time(&t->scene->markers, tdn->h2[0]);
556                                 break;
557                 }
558                 
559                 /* Use RNA to write the values to ensure that constraints on these are obeyed
560                  * (e.g. for transition strips, the values are taken from the neighbours)
561                  * 
562                  * NOTE: we write these twice to avoid truncation errors which can arise when
563                  * moving the strips a large distance using numeric input [#33852] 
564                  */
565                 RNA_pointer_create(NULL, &RNA_NlaStrip, strip, &strip_ptr);
566                 
567                 RNA_float_set(&strip_ptr, "frame_start", tdn->h1[0]);
568                 RNA_float_set(&strip_ptr, "frame_end", tdn->h2[0]);
569                 
570                 RNA_float_set(&strip_ptr, "frame_start", tdn->h1[0]);
571                 RNA_float_set(&strip_ptr, "frame_end", tdn->h2[0]);
572                 
573                 /* flush transforms to child strips (since this should be a meta) */
574                 BKE_nlameta_flush_transforms(strip);
575                 
576                 
577                 /* now, check if we need to try and move track
578                  *      - we need to calculate both, as only one may have been altered by transform if only 1 handle moved
579                  */
580                 delta_y1 = ((int)tdn->h1[1] / NLACHANNEL_STEP(snla) - tdn->trackIndex);
581                 delta_y2 = ((int)tdn->h2[1] / NLACHANNEL_STEP(snla) - tdn->trackIndex);
582                 
583                 if (delta_y1 || delta_y2) {
584                         NlaTrack *track;
585                         int delta = (delta_y2) ? delta_y2 : delta_y1;
586                         int n;
587                         
588                         /* move in the requested direction, checking at each layer if there's space for strip to pass through,
589                          * stopping on the last track available or that we're able to fit in
590                          */
591                         if (delta > 0) {
592                                 for (track = tdn->nlt->next, n = 0; (track) && (n < delta); track = track->next, n++) {
593                                         /* check if space in this track for the strip */
594                                         if (BKE_nlatrack_has_space(track, strip->start, strip->end)) {
595                                                 /* move strip to this track */
596                                                 BLI_remlink(&tdn->nlt->strips, strip);
597                                                 BKE_nlatrack_add_strip(track, strip);
598                                                 
599                                                 tdn->nlt = track;
600                                                 tdn->trackIndex++;
601                                         }
602                                         else /* can't move any further */
603                                                 break;
604                                 }
605                         }
606                         else {
607                                 /* make delta 'positive' before using it, since we now know to go backwards */
608                                 delta = -delta;
609                                 
610                                 for (track = tdn->nlt->prev, n = 0; (track) && (n < delta); track = track->prev, n++) {
611                                         /* check if space in this track for the strip */
612                                         if (BKE_nlatrack_has_space(track, strip->start, strip->end)) {
613                                                 /* move strip to this track */
614                                                 BLI_remlink(&tdn->nlt->strips, strip);
615                                                 BKE_nlatrack_add_strip(track, strip);
616                                                 
617                                                 tdn->nlt = track;
618                                                 tdn->trackIndex--;
619                                         }
620                                         else /* can't move any further */
621                                                 break;
622                                 }
623                         }
624                 }
625         }
626 }
627
628 static void recalcData_mask_common(TransInfo *t)
629 {
630         Mask *mask = CTX_data_edit_mask(t->context);
631
632         flushTransMasking(t);
633
634         DAG_id_tag_update(&mask->id, 0);
635 }
636
637 /* helper for recalcData() - for Image Editor transforms */
638 static void recalcData_image(TransInfo *t)
639 {
640         if (t->options & CTX_MASK) {
641                 recalcData_mask_common(t);
642         }
643         else if (t->obedit && t->obedit->type == OB_MESH) {
644                 SpaceImage *sima = t->sa->spacedata.first;
645                 
646                 flushTransUVs(t);
647                 if (sima->flag & SI_LIVE_UNWRAP)
648                         ED_uvedit_live_unwrap_re_solve();
649                 
650                 DAG_id_tag_update(t->obedit->data, 0);
651         }
652 }
653
654 /* helper for recalcData() - for Movie Clip transforms */
655 static void recalcData_spaceclip(TransInfo *t)
656 {
657         SpaceClip *sc = t->sa->spacedata.first;
658
659         if (ED_space_clip_check_show_trackedit(sc)) {
660                 MovieClip *clip = ED_space_clip_get_clip(sc);
661                 ListBase *tracksbase = BKE_tracking_get_active_tracks(&clip->tracking);
662                 ListBase *plane_tracks_base = BKE_tracking_get_active_plane_tracks(&clip->tracking);
663                 MovieTrackingTrack *track;
664                 MovieTrackingPlaneTrack *plane_track;
665                 int framenr = ED_space_clip_get_clip_frame_number(sc);
666
667                 flushTransTracking(t);
668
669                 track = tracksbase->first;
670                 while (track) {
671                         if (TRACK_VIEW_SELECTED(sc, track) && (track->flag & TRACK_LOCKED) == 0) {
672                                 MovieTrackingMarker *marker = BKE_tracking_marker_get(track, framenr);
673
674                                 if (t->mode == TFM_TRANSLATION) {
675                                         if (TRACK_AREA_SELECTED(track, TRACK_AREA_PAT))
676                                                 BKE_tracking_marker_clamp(marker, CLAMP_PAT_POS);
677                                         if (TRACK_AREA_SELECTED(track, TRACK_AREA_SEARCH))
678                                                 BKE_tracking_marker_clamp(marker, CLAMP_SEARCH_POS);
679                                 }
680                                 else if (t->mode == TFM_RESIZE) {
681                                         if (TRACK_AREA_SELECTED(track, TRACK_AREA_PAT))
682                                                 BKE_tracking_marker_clamp(marker, CLAMP_PAT_DIM);
683                                         if (TRACK_AREA_SELECTED(track, TRACK_AREA_SEARCH))
684                                                 BKE_tracking_marker_clamp(marker, CLAMP_SEARCH_DIM);
685                                 }
686                                 else if (t->mode == TFM_ROTATION) {
687                                         if (TRACK_AREA_SELECTED(track, TRACK_AREA_PAT))
688                                                 BKE_tracking_marker_clamp(marker, CLAMP_PAT_POS);
689                                 }
690                         }
691
692                         track = track->next;
693                 }
694
695                 for (plane_track = plane_tracks_base->first;
696                      plane_track;
697                      plane_track = plane_track->next)
698                 {
699                         if (plane_track->flag & SELECT) {
700                                 BKE_tracking_track_plane_from_existing_motion(plane_track, framenr);
701                         }
702                 }
703
704                 DAG_id_tag_update(&clip->id, 0);
705         }
706         else if (t->options & CTX_MASK) {
707                 recalcData_mask_common(t);
708         }
709 }
710
711 /* helper for recalcData() - for 3d-view transforms */
712 static void recalcData_view3d(TransInfo *t)
713 {
714         Base *base = t->scene->basact;
715         
716         if (t->obedit) {
717                 if (ELEM(t->obedit->type, OB_CURVE, OB_SURF)) {
718                         Curve *cu = t->obedit->data;
719                         ListBase *nurbs = BKE_curve_editNurbs_get(cu);
720                         Nurb *nu = nurbs->first;
721                         
722                         if (t->state != TRANS_CANCEL) {
723                                 clipMirrorModifier(t, t->obedit);
724                                 applyProject(t);
725                         }
726                         
727                         DAG_id_tag_update(t->obedit->data, 0);  /* sets recalc flags */
728                                 
729                         if (t->state == TRANS_CANCEL) {
730                                 while (nu) {
731                                         BKE_nurb_handles_calc(nu); /* Cant do testhandlesNurb here, it messes up the h1 and h2 flags */
732                                         nu = nu->next;
733                                 }
734                         }
735                         else {
736                                 /* Normal updating */
737                                 while (nu) {
738                                         BKE_nurb_test2D(nu);
739                                         BKE_nurb_handles_calc(nu);
740                                         nu = nu->next;
741                                 }
742                         }
743                 }
744                 else if (t->obedit->type == OB_LATTICE) {
745                         Lattice *la = t->obedit->data;
746                         
747                         if (t->state != TRANS_CANCEL) {
748                                 applyProject(t);
749                         }
750                         
751                         DAG_id_tag_update(t->obedit->data, 0);  /* sets recalc flags */
752                         
753                         if (la->editlatt->latt->flag & LT_OUTSIDE) outside_lattice(la->editlatt->latt);
754                 }
755                 else if (t->obedit->type == OB_MESH) {
756                         BMEditMesh *em = BKE_editmesh_from_object(t->obedit);
757                         /* mirror modifier clipping? */
758                         if (t->state != TRANS_CANCEL) {
759                                 /* apply clipping after so we never project past the clip plane [#25423] */
760                                 applyProject(t);
761                                 clipMirrorModifier(t, t->obedit);
762                         }
763                         if ((t->options & CTX_NO_MIRROR) == 0 && (t->flag & T_MIRROR))
764                                 editbmesh_apply_to_mirror(t);
765                                 
766                         DAG_id_tag_update(t->obedit->data, 0);  /* sets recalc flags */
767                         
768                         EDBM_mesh_normals_update(em);
769                         BKE_editmesh_tessface_calc(em);
770                 }
771                 else if (t->obedit->type == OB_ARMATURE) { /* no recalc flag, does pose */
772                         bArmature *arm = t->obedit->data;
773                         ListBase *edbo = arm->edbo;
774                         EditBone *ebo;
775                         TransData *td = t->data;
776                         int i;
777                         
778                         if (t->state != TRANS_CANCEL) {
779                                 applyProject(t);
780                         }
781                         
782                         /* Ensure all bones are correctly adjusted */
783                         for (ebo = edbo->first; ebo; ebo = ebo->next) {
784                                 
785                                 if ((ebo->flag & BONE_CONNECTED) && ebo->parent) {
786                                         /* If this bone has a parent tip that has been moved */
787                                         if (ebo->parent->flag & BONE_TIPSEL) {
788                                                 copy_v3_v3(ebo->head, ebo->parent->tail);
789                                                 if (t->mode == TFM_BONE_ENVELOPE) ebo->rad_head = ebo->parent->rad_tail;
790                                         }
791                                         /* If this bone has a parent tip that has NOT been moved */
792                                         else {
793                                                 copy_v3_v3(ebo->parent->tail, ebo->head);
794                                                 if (t->mode == TFM_BONE_ENVELOPE) ebo->parent->rad_tail = ebo->rad_head;
795                                         }
796                                 }
797                                 
798                                 /* on extrude bones, oldlength==0.0f, so we scale radius of points */
799                                 ebo->length = len_v3v3(ebo->head, ebo->tail);
800                                 if (ebo->oldlength == 0.0f) {
801                                         ebo->rad_head = 0.25f * ebo->length;
802                                         ebo->rad_tail = 0.10f * ebo->length;
803                                         ebo->dist = 0.25f * ebo->length;
804                                         if (ebo->parent) {
805                                                 if (ebo->rad_head > ebo->parent->rad_tail)
806                                                         ebo->rad_head = ebo->parent->rad_tail;
807                                         }
808                                 }
809                                 else if (t->mode != TFM_BONE_ENVELOPE) {
810                                         /* if bones change length, lets do that for the deform distance as well */
811                                         ebo->dist *= ebo->length / ebo->oldlength;
812                                         ebo->rad_head *= ebo->length / ebo->oldlength;
813                                         ebo->rad_tail *= ebo->length / ebo->oldlength;
814                                         ebo->oldlength = ebo->length;
815                                 }
816                         }
817                         
818                         
819                         if (t->mode != TFM_BONE_ROLL) {
820                                 /* fix roll */
821                                 for (i = 0; i < t->total; i++, td++) {
822                                         if (td->extra) {
823                                                 float vec[3], up_axis[3];
824                                                 float qrot[4];
825                                                 float roll;
826                                                 
827                                                 ebo = td->extra;
828                                                 copy_v3_v3(up_axis, td->axismtx[2]);
829                                                 
830                                                 if (t->mode != TFM_ROTATION) {
831                                                         sub_v3_v3v3(vec, ebo->tail, ebo->head);
832                                                         normalize_v3(vec);
833                                                         rotation_between_vecs_to_quat(qrot, td->axismtx[1], vec);
834                                                         mul_qt_v3(qrot, up_axis);
835                                                 }
836                                                 else {
837                                                         mul_m3_v3(t->mat, up_axis);
838                                                 }
839                                                 
840                                                 /* roll has a tendency to flip in certain orientations - [#34283], [#33974] */
841                                                 roll = ED_rollBoneToVector(ebo, up_axis, false);
842                                                 ebo->roll = angle_compat_rad(roll, ebo->roll);
843                                         }
844                                 }
845                         }
846                         
847                         if (arm->flag & ARM_MIRROR_EDIT)
848                                 transform_armature_mirror_update(t->obedit);
849                         
850                 }
851                 else {
852                         if (t->state != TRANS_CANCEL) {
853                                 applyProject(t);
854                         }
855                         DAG_id_tag_update(t->obedit->data, 0);  /* sets recalc flags */
856                 }
857         }
858         else if ((t->flag & T_POSE) && t->poseobj) {
859                 Object *ob = t->poseobj;
860                 bArmature *arm = ob->data;
861                 
862                 /* if animtimer is running, and the object already has animation data,
863                  * check if the auto-record feature means that we should record 'samples'
864                  * (i.e. uneditable animation values)
865                  *
866                  * context is needed for keying set poll() functions.
867                  */
868                 // TODO: autokeyframe calls need some setting to specify to add samples (FPoints) instead of keyframes?
869                 if ((t->animtimer) && (t->context) && IS_AUTOKEY_ON(t->scene)) {
870                         int targetless_ik = (t->flag & T_AUTOIK); // XXX this currently doesn't work, since flags aren't set yet!
871                         
872                         animrecord_check_state(t->scene, &ob->id, t->animtimer);
873                         autokeyframe_pose_cb_func(t->context, t->scene, (View3D *)t->view, ob, t->mode, targetless_ik);
874                 }
875                 
876                 /* old optimize trick... this enforces to bypass the depgraph */
877                 if (!(arm->flag & ARM_DELAYDEFORM)) {
878                         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);  /* sets recalc flags */
879                         /* transformation of pose may affect IK tree, make sure it is rebuilt */
880                         BIK_clear_data(ob->pose);
881                 }
882                 else
883                         BKE_pose_where_is(t->scene, ob);
884         }
885         else if (base && (base->object->mode & OB_MODE_PARTICLE_EDIT) && PE_get_current(t->scene, base->object)) {
886                 if (t->state != TRANS_CANCEL) {
887                         applyProject(t);
888                 }
889                 flushTransParticles(t);
890         }
891         else {
892                 int i;
893                 
894                 if (t->state != TRANS_CANCEL) {
895                         applyProject(t);
896                 }
897                 
898                 for (i = 0; i < t->total; i++) {
899                         TransData *td = t->data + i;
900                         Object *ob = td->ob;
901                         
902                         if (td->flag & TD_NOACTION)
903                                 break;
904                         
905                         if (td->flag & TD_SKIP)
906                                 continue;
907                         
908                         /* if animtimer is running, and the object already has animation data,
909                          * check if the auto-record feature means that we should record 'samples'
910                          * (i.e. uneditable animation values)
911                          */
912                         // TODO: autokeyframe calls need some setting to specify to add samples (FPoints) instead of keyframes?
913                         if ((t->animtimer) && IS_AUTOKEY_ON(t->scene)) {
914                                 animrecord_check_state(t->scene, &ob->id, t->animtimer);
915                                 autokeyframe_ob_cb_func(t->context, t->scene, (View3D *)t->view, ob, t->mode);
916                         }
917                         
918                         /* sets recalc flags fully, instead of flushing existing ones 
919                          * otherwise proxies don't function correctly
920                          */
921                         DAG_id_tag_update(&ob->id, OB_RECALC_OB);
922
923                         if (t->flag & T_TEXTURE)
924                                 DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
925                 }
926         }
927 }
928
929 /* helper for recalcData() - for sequencer transforms */
930 static void recalcData_sequencer(TransInfo *t)
931 {
932         TransData *td;
933         int a;
934         Sequence *seq_prev = NULL;
935
936         for (a = 0, td = t->data; a < t->total; a++, td++) {
937                 TransDataSeq *tdsq = (TransDataSeq *) td->extra;
938                 Sequence *seq = tdsq->seq;
939
940                 if (seq != seq_prev) {
941                         BKE_sequence_invalidate_dependent(t->scene, seq);
942                 }
943
944                 seq_prev = seq;
945         }
946
947         BKE_sequencer_preprocessed_cache_cleanup();
948
949         flushTransSeq(t);
950 }
951
952 /* called for updating while transform acts, once per redraw */
953 void recalcData(TransInfo *t)
954 {
955         if (t->spacetype == SPACE_NODE) {
956                 flushTransNodes(t);
957         }
958         else if (t->spacetype == SPACE_SEQ) {
959                 recalcData_sequencer(t);
960         }
961         else if (t->spacetype == SPACE_ACTION) {
962                 recalcData_actedit(t);
963         }
964         else if (t->spacetype == SPACE_IPO) {
965                 recalcData_graphedit(t);
966         }
967         else if (t->spacetype == SPACE_NLA) {
968                 recalcData_nla(t);
969         }
970         else if (t->spacetype == SPACE_IMAGE) {
971                 recalcData_image(t);
972         }
973         else if (t->spacetype == SPACE_VIEW3D) {
974                 recalcData_view3d(t);
975         }
976         else if (t->spacetype == SPACE_CLIP) {
977                 recalcData_spaceclip(t);
978         }
979 }
980
981 void drawLine(TransInfo *t, const float center[3], const float dir[3], char axis, short options)
982 {
983         float v1[3], v2[3], v3[3];
984         unsigned char col[3], col2[3];
985
986         if (t->spacetype == SPACE_VIEW3D) {
987                 View3D *v3d = t->view;
988                 
989                 glPushMatrix();
990                 
991                 //if (t->obedit) glLoadMatrixf(t->obedit->obmat);       // sets opengl viewing
992                 
993                 
994                 copy_v3_v3(v3, dir);
995                 mul_v3_fl(v3, v3d->far);
996                 
997                 sub_v3_v3v3(v2, center, v3);
998                 add_v3_v3v3(v1, center, v3);
999                 
1000                 if (options & DRAWLIGHT) {
1001                         col[0] = col[1] = col[2] = 220;
1002                 }
1003                 else {
1004                         UI_GetThemeColor3ubv(TH_GRID, col);
1005                 }
1006                 UI_make_axis_color(col, col2, axis);
1007                 glColor3ubv(col2);
1008                 
1009                 setlinestyle(0);
1010                 glBegin(GL_LINE_STRIP);
1011                 glVertex3fv(v1);
1012                 glVertex3fv(v2);
1013                 glEnd();
1014                 
1015                 glPopMatrix();
1016         }
1017 }
1018
1019 /**
1020  * Free data before switching to another mode.
1021  */
1022 void resetTransModal(TransInfo *t)
1023 {
1024         if (t->mode == TFM_EDGE_SLIDE) {
1025                 freeEdgeSlideVerts(t);
1026         }
1027         else if (t->mode == TFM_VERT_SLIDE) {
1028                 freeVertSlideVerts(t);
1029         }
1030 }
1031
1032 void resetTransRestrictions(TransInfo *t)
1033 {
1034         t->flag &= ~T_ALL_RESTRICTIONS;
1035 }
1036
1037 static int initTransInfo_edit_pet_to_flag(const int proportional)
1038 {
1039         switch (proportional) {
1040                 case PROP_EDIT_ON:
1041                         return T_PROP_EDIT;
1042                 case PROP_EDIT_CONNECTED:
1043                         return T_PROP_EDIT | T_PROP_CONNECTED;
1044                 case PROP_EDIT_PROJECTED:
1045                         return T_PROP_EDIT | T_PROP_PROJECTED;
1046                 default:
1047                         return 0;
1048         }
1049 }
1050
1051 /* the *op can be NULL */
1052 int initTransInfo(bContext *C, TransInfo *t, wmOperator *op, const wmEvent *event)
1053 {
1054         Scene *sce = CTX_data_scene(C);
1055         ToolSettings *ts = CTX_data_tool_settings(C);
1056         ARegion *ar = CTX_wm_region(C);
1057         ScrArea *sa = CTX_wm_area(C);
1058         Object *obedit = CTX_data_edit_object(C);
1059         PropertyRNA *prop;
1060         
1061         t->scene = sce;
1062         t->sa = sa;
1063         t->ar = ar;
1064         t->obedit = obedit;
1065         t->settings = ts;
1066         t->reports = op ? op->reports : NULL;
1067
1068         if (obedit) {
1069                 copy_m3_m4(t->obedit_mat, obedit->obmat);
1070                 normalize_m3(t->obedit_mat);
1071         }
1072
1073         t->data = NULL;
1074         t->ext = NULL;
1075         
1076         t->helpline = HLP_NONE;
1077         
1078         t->flag = 0;
1079         
1080         t->redraw = 1; /* redraw first time */
1081         
1082         if (event) {
1083                 copy_v2_v2_int(t->imval, event->mval);
1084                 t->event_type = event->type;
1085         }
1086         else {
1087                 t->imval[0] = 0;
1088                 t->imval[1] = 0;
1089         }
1090         
1091         t->con.imval[0] = t->imval[0];
1092         t->con.imval[1] = t->imval[1];
1093         
1094         t->mval[0] = t->imval[0];
1095         t->mval[1] = t->imval[1];
1096         
1097         t->transform        = NULL;
1098         t->handleEvent      = NULL;
1099         
1100         t->total            = 0;
1101         
1102         t->val = 0.0f;
1103
1104         zero_v3(t->vec);
1105         zero_v3(t->center);
1106
1107         unit_m3(t->mat);
1108         
1109         /* if there's an event, we're modal */
1110         if (event) {
1111                 t->flag |= T_MODAL;
1112         }
1113
1114         /* Crease needs edge flag */
1115         if (ELEM(t->mode, TFM_CREASE, TFM_BWEIGHT)) {
1116                 t->options |= CTX_EDGE;
1117         }
1118
1119
1120         /* Assign the space type, some exceptions for running in different mode */
1121         if (sa == NULL) {
1122                 /* background mode */
1123                 t->spacetype = SPACE_EMPTY;
1124         }
1125         else if ((ar == NULL) && (sa->spacetype == SPACE_VIEW3D)) {
1126                 /* running in the text editor */
1127                 t->spacetype = SPACE_EMPTY;
1128         }
1129         else {
1130                 /* normal operation */
1131                 t->spacetype = sa->spacetype;
1132         }
1133
1134
1135         if (t->spacetype == SPACE_VIEW3D) {
1136                 View3D *v3d = sa->spacedata.first;
1137                 bScreen *animscreen = ED_screen_animation_playing(CTX_wm_manager(C));
1138                 
1139                 t->view = v3d;
1140                 t->animtimer = (animscreen) ? animscreen->animtimer : NULL;
1141                 
1142                 /* turn manipulator off during transform */
1143                 // FIXME: but don't do this when USING the manipulator...
1144                 if (t->flag & T_MODAL) {
1145                         t->twtype = v3d->twtype;
1146                         v3d->twtype = 0;
1147                 }
1148
1149                 if (v3d->flag & V3D_ALIGN) t->flag |= T_V3D_ALIGN;
1150                 t->around = v3d->around;
1151                 
1152                 if (op && ((prop = RNA_struct_find_property(op->ptr, "constraint_orientation")) &&
1153                            RNA_property_is_set(op->ptr, prop)))
1154                 {
1155                         t->current_orientation = RNA_property_enum_get(op->ptr, prop);
1156
1157                         if (t->current_orientation >= V3D_MANIP_CUSTOM + BIF_countTransformOrientation(C)) {
1158                                 t->current_orientation = V3D_MANIP_GLOBAL;
1159                         }
1160                 }
1161                 else {
1162                         t->current_orientation = v3d->twmode;
1163                 }
1164
1165                 /* exceptional case */
1166                 if (t->around == V3D_LOCAL && (t->settings->selectmode & SCE_SELECT_FACE)) {
1167                         if (ELEM3(t->mode, TFM_ROTATION, TFM_RESIZE, TFM_TRACKBALL)) {
1168                                 t->options |= CTX_NO_PET;
1169                         }
1170                 }
1171
1172                 /* initialize UV transform from */
1173                 if (op && ((prop = RNA_struct_find_property(op->ptr, "correct_uv")))) {
1174                         if (RNA_property_is_set(op->ptr, prop)) {
1175                                 if (RNA_property_boolean_get(op->ptr, prop)) {
1176                                         t->settings->uvcalc_flag |= UVCALC_TRANSFORM_CORRECT;
1177                                 }
1178                                 else {
1179                                         t->settings->uvcalc_flag &= ~UVCALC_TRANSFORM_CORRECT;
1180                                 }
1181                         }
1182                         else {
1183                                 RNA_property_boolean_set(op->ptr, prop, t->settings->uvcalc_flag & UVCALC_TRANSFORM_CORRECT);
1184                         }
1185                 }
1186
1187         }
1188         else if (t->spacetype == SPACE_IMAGE) {
1189                 SpaceImage *sima = sa->spacedata.first;
1190                 // XXX for now, get View2D from the active region
1191                 t->view = &ar->v2d;
1192                 t->around = sima->around;
1193
1194                 if (ED_space_image_show_uvedit(sima, t->obedit)) {
1195                         /* UV transform */
1196                 }
1197                 else if (sima->mode == SI_MODE_MASK) {
1198                         t->options |= CTX_MASK;
1199                 }
1200                 else {
1201                         /* image not in uv edit, nor in mask mode, can happen for some tools */
1202                 }
1203         }
1204         else if (t->spacetype == SPACE_NODE) {
1205                 // XXX for now, get View2D from the active region
1206                 t->view = &ar->v2d;
1207                 t->around = V3D_CENTER;
1208         }
1209         else if (t->spacetype == SPACE_IPO) {
1210                 SpaceIpo *sipo = sa->spacedata.first;
1211                 t->view = &ar->v2d;
1212                 t->around = sipo->around;
1213         }
1214         else if (t->spacetype == SPACE_CLIP) {
1215                 SpaceClip *sclip = sa->spacedata.first;
1216                 t->view = &ar->v2d;
1217                 t->around = sclip->around;
1218
1219                 if (ED_space_clip_check_show_trackedit(sclip))
1220                         t->options |= CTX_MOVIECLIP;
1221                 else if (ED_space_clip_check_show_maskedit(sclip))
1222                         t->options |= CTX_MASK;
1223         }
1224         else {
1225                 if (ar) {
1226                         // XXX for now, get View2D  from the active region
1227                         t->view = &ar->v2d;
1228                         // XXX for now, the center point is the midpoint of the data
1229                 }
1230                 else {
1231                         t->view = NULL;
1232                 }
1233                 t->around = V3D_CENTER;
1234         }
1235         
1236         if (op && ((prop = RNA_struct_find_property(op->ptr, "release_confirm")) &&
1237                    RNA_property_is_set(op->ptr, prop)))
1238         {
1239                 if (RNA_property_boolean_get(op->ptr, prop)) {
1240                         t->flag |= T_RELEASE_CONFIRM;
1241                 }
1242         }
1243         else {
1244                 if (U.flag & USER_RELEASECONFIRM) {
1245                         t->flag |= T_RELEASE_CONFIRM;
1246                 }
1247         }
1248
1249         if (op && ((prop = RNA_struct_find_property(op->ptr, "mirror")) &&
1250                    RNA_property_is_set(op->ptr, prop)))
1251         {
1252                 if (RNA_property_boolean_get(op->ptr, prop)) {
1253                         t->flag |= T_MIRROR;
1254                         t->mirror = 1;
1255                 }
1256         }
1257         // Need stuff to take it from edit mesh or whatnot here
1258         else if (t->spacetype == SPACE_VIEW3D) {
1259                 if (t->obedit && t->obedit->type == OB_MESH && (((Mesh *)t->obedit->data)->editflag & ME_EDIT_MIRROR_X)) {
1260                         t->flag |= T_MIRROR;
1261                         t->mirror = 1;
1262                 }
1263         }
1264         
1265         /* setting PET flag only if property exist in operator. Otherwise, assume it's not supported */
1266         if (op && (prop = RNA_struct_find_property(op->ptr, "proportional"))) {
1267                 if (RNA_property_is_set(op->ptr, prop)) {
1268                         t->flag |= initTransInfo_edit_pet_to_flag(RNA_property_enum_get(op->ptr, prop));
1269                 }
1270                 else {
1271                         /* use settings from scene only if modal */
1272                         if (t->flag & T_MODAL) {
1273                                 if ((t->options & CTX_NO_PET) == 0) {
1274                                         if (t->obedit) {
1275                                                 t->flag |= initTransInfo_edit_pet_to_flag(ts->proportional);
1276                                         }
1277                                         else if (t->options & CTX_MASK) {
1278                                                 if (ts->proportional_mask) {
1279                                                         t->flag |= T_PROP_EDIT;
1280
1281                                                         if (ts->proportional == PROP_EDIT_CONNECTED) {
1282                                                                 t->flag |= T_PROP_CONNECTED;
1283                                                         }
1284                                                 }
1285                                         }
1286                                         else if (t->obedit == NULL && ts->proportional_objects) {
1287                                                 t->flag |= T_PROP_EDIT;
1288                                         }
1289                                 }
1290                         }
1291                 }
1292                 
1293                 if (op && ((prop = RNA_struct_find_property(op->ptr, "proportional_size")) &&
1294                            RNA_property_is_set(op->ptr, prop)))
1295                 {
1296                         t->prop_size = RNA_property_float_get(op->ptr, prop);
1297                 }
1298                 else {
1299                         t->prop_size = ts->proportional_size;
1300                 }
1301                 
1302                 
1303                 /* TRANSFORM_FIX_ME rna restrictions */
1304                 if (t->prop_size <= 0.00001f) {
1305                         printf("Proportional size (%f) under 0.00001, resetting to 1!\n", t->prop_size);
1306                         t->prop_size = 1.0f;
1307                 }
1308                 
1309                 if (op && ((prop = RNA_struct_find_property(op->ptr, "proportional_edit_falloff")) &&
1310                            RNA_property_is_set(op->ptr, prop)))
1311                 {
1312                         t->prop_mode = RNA_property_enum_get(op->ptr, prop);
1313                 }
1314                 else {
1315                         t->prop_mode = ts->prop_mode;
1316                 }
1317         }
1318         else { /* add not pet option to context when not available */
1319                 t->options |= CTX_NO_PET;
1320         }
1321         
1322         // Mirror is not supported with PET, turn it off.
1323 #if 0
1324         if (t->flag & T_PROP_EDIT) {
1325                 t->flag &= ~T_MIRROR;
1326         }
1327 #endif
1328
1329         setTransformViewMatrices(t);
1330         initNumInput(&t->num);
1331         
1332         return 1;
1333 }
1334
1335 /* Here I would suggest only TransInfo related issues, like free data & reset vars. Not redraws */
1336 void postTrans(bContext *C, TransInfo *t)
1337 {
1338         TransData *td;
1339         
1340         if (t->draw_handle_view)
1341                 ED_region_draw_cb_exit(t->ar->type, t->draw_handle_view);
1342         if (t->draw_handle_apply)
1343                 ED_region_draw_cb_exit(t->ar->type, t->draw_handle_apply);
1344         if (t->draw_handle_pixel)
1345                 ED_region_draw_cb_exit(t->ar->type, t->draw_handle_pixel);
1346         if (t->draw_handle_cursor)
1347                 WM_paint_cursor_end(CTX_wm_manager(C), t->draw_handle_cursor);
1348
1349         if (t->customFree) {
1350                 /* Can take over freeing t->data and data2d etc... */
1351                 t->customFree(t);
1352                 BLI_assert(t->customData == NULL);
1353         }
1354         else if ((t->customData != NULL) && (t->flag & T_FREE_CUSTOMDATA)) {
1355                 MEM_freeN(t->customData);
1356                 t->customData = NULL;
1357         }
1358
1359         /* postTrans can be called when nothing is selected, so data is NULL already */
1360         if (t->data) {
1361                 
1362                 /* free data malloced per trans-data */
1363                 if ((t->obedit && ELEM(t->obedit->type, OB_CURVE, OB_SURF)) ||
1364                     (t->spacetype == SPACE_IPO))
1365                 {
1366                         int a;
1367                         for (a = 0, td = t->data; a < t->total; a++, td++) {
1368                                 if (td->flag & TD_BEZTRIPLE) {
1369                                         MEM_freeN(td->hdata);
1370                                 }
1371                         }
1372                 }
1373                 MEM_freeN(t->data);
1374         }
1375         
1376         BLI_freelistN(&t->tsnap.points);
1377
1378         if (t->ext) MEM_freeN(t->ext);
1379         if (t->data2d) {
1380                 MEM_freeN(t->data2d);
1381                 t->data2d = NULL;
1382         }
1383         
1384         if (t->spacetype == SPACE_IMAGE) {
1385                 if (t->options & CTX_MASK) {
1386                         /* pass */
1387                 }
1388                 else {
1389                         SpaceImage *sima = t->sa->spacedata.first;
1390                         if (sima->flag & SI_LIVE_UNWRAP)
1391                                 ED_uvedit_live_unwrap_end(t->state == TRANS_CANCEL);
1392                 }
1393         }
1394         else if (t->spacetype == SPACE_VIEW3D) {
1395                 View3D *v3d = t->sa->spacedata.first;
1396                 /* restore manipulator */
1397                 if (t->flag & T_MODAL) {
1398                         v3d->twtype = t->twtype;
1399                 }
1400         }
1401         
1402         if (t->mouse.data) {
1403                 MEM_freeN(t->mouse.data);
1404         }
1405 }
1406
1407 void applyTransObjects(TransInfo *t)
1408 {
1409         TransData *td;
1410         
1411         for (td = t->data; td < t->data + t->total; td++) {
1412                 copy_v3_v3(td->iloc, td->loc);
1413                 if (td->ext->rot) {
1414                         copy_v3_v3(td->ext->irot, td->ext->rot);
1415                 }
1416                 if (td->ext->size) {
1417                         copy_v3_v3(td->ext->isize, td->ext->size);
1418                 }
1419         }
1420         recalcData(t);
1421 }
1422
1423 static void restoreElement(TransData *td)
1424 {
1425         /* TransData for crease has no loc */
1426         if (td->loc) {
1427                 copy_v3_v3(td->loc, td->iloc);
1428         }
1429         if (td->val) {
1430                 *td->val = td->ival;
1431         }
1432
1433         if (td->ext && (td->flag & TD_NO_EXT) == 0) {
1434                 if (td->ext->rot) {
1435                         copy_v3_v3(td->ext->rot, td->ext->irot);
1436                 }
1437                 if (td->ext->rotAngle) {
1438                         *td->ext->rotAngle = td->ext->irotAngle;
1439                 }
1440                 if (td->ext->rotAxis) {
1441                         copy_v3_v3(td->ext->rotAxis, td->ext->irotAxis);
1442                 }
1443                 /* XXX, drotAngle & drotAxis not used yet */
1444                 if (td->ext->size) {
1445                         copy_v3_v3(td->ext->size, td->ext->isize);
1446                 }
1447                 if (td->ext->quat) {
1448                         copy_qt_qt(td->ext->quat, td->ext->iquat);
1449                 }
1450         }
1451         
1452         if (td->flag & TD_BEZTRIPLE) {
1453                 *(td->hdata->h1) = td->hdata->ih1;
1454                 *(td->hdata->h2) = td->hdata->ih2;
1455         }
1456 }
1457
1458 void restoreTransObjects(TransInfo *t)
1459 {
1460         TransData *td;
1461         TransData2D *td2d;
1462
1463         for (td = t->data; td < t->data + t->total; td++) {
1464                 restoreElement(td);
1465         }
1466         
1467         for (td2d = t->data2d; t->data2d && td2d < t->data2d + t->total; td2d++) {
1468                 if (td2d->h1) {
1469                         td2d->h1[0] = td2d->ih1[0];
1470                         td2d->h1[1] = td2d->ih1[1];
1471                 }
1472                 if (td2d->h2) {
1473                         td2d->h2[0] = td2d->ih2[0];
1474                         td2d->h2[1] = td2d->ih2[1];
1475                 }
1476         }
1477
1478         unit_m3(t->mat);
1479         
1480         recalcData(t);
1481 }
1482
1483 void calculateCenter2D(TransInfo *t)
1484 {
1485         if (t->flag & (T_EDIT | T_POSE)) {
1486                 Object *ob = t->obedit ? t->obedit : t->poseobj;
1487                 float vec[3];
1488                 
1489                 copy_v3_v3(vec, t->center);
1490                 mul_m4_v3(ob->obmat, vec);
1491                 projectIntView(t, vec, t->center2d);
1492         }
1493         else {
1494                 projectIntView(t, t->center, t->center2d);
1495         }
1496 }
1497
1498 void calculateCenterCursor(TransInfo *t)
1499 {
1500         const float *cursor;
1501         
1502         cursor = give_cursor(t->scene, t->view);
1503         copy_v3_v3(t->center, cursor);
1504         
1505         /* If edit or pose mode, move cursor in local space */
1506         if (t->flag & (T_EDIT | T_POSE)) {
1507                 Object *ob = t->obedit ? t->obedit : t->poseobj;
1508                 float mat[3][3], imat[3][3];
1509                 
1510                 sub_v3_v3v3(t->center, t->center, ob->obmat[3]);
1511                 copy_m3_m4(mat, ob->obmat);
1512                 invert_m3_m3(imat, mat);
1513                 mul_m3_v3(imat, t->center);
1514         }
1515         
1516         calculateCenter2D(t);
1517 }
1518
1519 void calculateCenterCursor2D(TransInfo *t)
1520 {
1521         float aspx = 1.0, aspy = 1.0;
1522         float *cursor = NULL;
1523         
1524         if (t->spacetype == SPACE_IMAGE) {
1525                 SpaceImage *sima = (SpaceImage *)t->sa->spacedata.first;
1526                 if (t->options & CTX_MASK) {
1527                         ED_space_image_get_aspect(sima, &aspx, &aspy);
1528                 }
1529                 else {
1530                         ED_space_image_get_uv_aspect(sima, &aspx, &aspy);
1531                 }
1532                 cursor = sima->cursor;
1533         }
1534         else if (t->spacetype == SPACE_CLIP) {
1535                 SpaceClip *space_clip = (SpaceClip *) t->sa->spacedata.first;
1536                 if (t->options & CTX_MOVIECLIP) {
1537                         ED_space_clip_get_aspect_dimension_aware(space_clip, &aspx, &aspy);
1538                 }
1539                 else {
1540                         ED_space_clip_get_aspect(space_clip, &aspx, &aspy);
1541                 }
1542                 cursor = space_clip->cursor;
1543         }
1544         
1545         if (cursor) {
1546                 if (t->options & CTX_MASK) {
1547                         float co[2];
1548                         float frame_size[2];
1549
1550                         if (t->spacetype == SPACE_IMAGE) {
1551                                 SpaceImage *sima = (SpaceImage *)t->sa->spacedata.first;
1552                                 ED_space_image_get_size_fl(sima, frame_size);
1553                                 BKE_mask_coord_from_frame(co, cursor, frame_size);
1554                                 ED_space_image_get_aspect(sima, &aspx, &aspy);
1555                         }
1556                         else if (t->spacetype == SPACE_CLIP) {
1557                                 SpaceClip *space_clip = (SpaceClip *) t->sa->spacedata.first;
1558                                 ED_space_clip_get_size_fl(space_clip, frame_size);
1559                                 BKE_mask_coord_from_frame(co, cursor, frame_size);
1560                                 ED_space_clip_get_aspect(space_clip, &aspx, &aspy);
1561                         }
1562                         else {
1563                                 BLI_assert(!"Shall not happen");
1564                         }
1565
1566                         t->center[0] = co[0] * aspx;
1567                         t->center[1] = co[1] * aspy;
1568                 }
1569                 else {
1570                         t->center[0] = cursor[0] * aspx;
1571                         t->center[1] = cursor[1] * aspy;
1572                 }
1573         }
1574         
1575         calculateCenter2D(t);
1576 }
1577
1578 static void calculateCenterCursorGraph2D(TransInfo *t)
1579 {
1580         SpaceIpo *sipo = (SpaceIpo *)t->sa->spacedata.first;
1581         Scene *scene = t->scene;
1582         
1583         /* cursor is combination of current frame, and graph-editor cursor value */
1584         t->center[0] = (float)(scene->r.cfra);
1585         t->center[1] = sipo->cursorVal;
1586         
1587         calculateCenter2D(t);
1588 }
1589
1590 void calculateCenterMedian(TransInfo *t)
1591 {
1592         float partial[3] = {0.0f, 0.0f, 0.0f};
1593         int total = 0;
1594         int i;
1595         
1596         for (i = 0; i < t->total; i++) {
1597                 if (t->data[i].flag & TD_SELECTED) {
1598                         if (!(t->data[i].flag & TD_NOCENTER)) {
1599                                 add_v3_v3(partial, t->data[i].center);
1600                                 total++;
1601                         }
1602                 }
1603         }
1604         if (i)
1605                 mul_v3_fl(partial, 1.0f / total);
1606         copy_v3_v3(t->center, partial);
1607         
1608         calculateCenter2D(t);
1609 }
1610
1611 void calculateCenterBound(TransInfo *t)
1612 {
1613         float max[3];
1614         float min[3];
1615         int i;
1616         for (i = 0; i < t->total; i++) {
1617                 if (i) {
1618                         if (t->data[i].flag & TD_SELECTED) {
1619                                 if (!(t->data[i].flag & TD_NOCENTER))
1620                                         minmax_v3v3_v3(min, max, t->data[i].center);
1621                         }
1622                 }
1623                 else {
1624                         copy_v3_v3(max, t->data[i].center);
1625                         copy_v3_v3(min, t->data[i].center);
1626                 }
1627         }
1628         mid_v3_v3v3(t->center, min, max);
1629
1630         calculateCenter2D(t);
1631 }
1632
1633 void calculateCenter(TransInfo *t)
1634 {
1635         switch (t->around) {
1636                 case V3D_CENTER:
1637                         calculateCenterBound(t);
1638                         break;
1639                 case V3D_CENTROID:
1640                         calculateCenterMedian(t);
1641                         break;
1642                 case V3D_CURSOR:
1643                         if (ELEM(t->spacetype, SPACE_IMAGE, SPACE_CLIP))
1644                                 calculateCenterCursor2D(t);
1645                         else if (t->spacetype == SPACE_IPO)
1646                                 calculateCenterCursorGraph2D(t);
1647                         else
1648                                 calculateCenterCursor(t);
1649                         break;
1650                 case V3D_LOCAL:
1651                         /* Individual element center uses median center for helpline and such */
1652                         calculateCenterMedian(t);
1653                         break;
1654                 case V3D_ACTIVE:
1655                 {
1656                         /* set median, and if if if... do object center */
1657                 
1658                         /* EDIT MODE ACTIVE EDITMODE ELEMENT */
1659
1660                         if (t->obedit) {
1661                                 if (t->obedit && t->obedit->type == OB_MESH) {
1662                                         BMEditSelection ese;
1663                                         BMEditMesh *em = BKE_editmesh_from_object(t->obedit);
1664
1665                                         if (BM_select_history_active_get(em->bm, &ese)) {
1666                                                 BM_editselection_center(&ese, t->center);
1667                                                 calculateCenter2D(t);
1668                                                 break;
1669                                         }
1670                                 }
1671                                 else if (ELEM(t->obedit->type, OB_CURVE, OB_SURF)) {
1672                                         float center[3];
1673                                         Curve *cu = (Curve *)t->obedit->data;
1674
1675                                         if (ED_curve_actSelection(cu, center)) {
1676                                                 copy_v3_v3(t->center, center);
1677                                                 calculateCenter2D(t);
1678                                                 break;
1679                                         }
1680                                 }
1681                                 else if (t->obedit && t->obedit->type == OB_LATTICE) {
1682                                         BPoint *actbp = BKE_lattice_active_point_get(t->obedit->data);
1683
1684                                         if (actbp) {
1685                                                 copy_v3_v3(t->center, actbp->vec);
1686                                                 calculateCenter2D(t);
1687                                                 break;
1688                                         }
1689                                 }
1690                         } /* END EDIT MODE ACTIVE ELEMENT */
1691
1692                         calculateCenterMedian(t);
1693                         if ((t->flag & (T_EDIT | T_POSE)) == 0) {
1694                                 Scene *scene = t->scene;
1695                                 Object *ob = OBACT;
1696                                 if (ob) {
1697                                         copy_v3_v3(t->center, ob->obmat[3]);
1698                                         projectIntView(t, t->center, t->center2d);
1699                                 }
1700                         }
1701                         break;
1702                 }
1703         }
1704         
1705         /* setting constraint center */
1706         copy_v3_v3(t->con.center, t->center);
1707         if (t->flag & (T_EDIT | T_POSE)) {
1708                 Object *ob = t->obedit ? t->obedit : t->poseobj;
1709                 mul_m4_v3(ob->obmat, t->con.center);
1710         }
1711         
1712         /* for panning from cameraview */
1713         if (t->flag & T_OBJECT) {
1714                 if (t->spacetype == SPACE_VIEW3D && t->ar && t->ar->regiontype == RGN_TYPE_WINDOW) {
1715                         View3D *v3d = t->view;
1716                         Scene *scene = t->scene;
1717                         RegionView3D *rv3d = t->ar->regiondata;
1718                         
1719                         if (v3d->camera == OBACT && rv3d->persp == RV3D_CAMOB) {
1720                                 float axis[3];
1721                                 /* persinv is nasty, use viewinv instead, always right */
1722                                 copy_v3_v3(axis, t->viewinv[2]);
1723                                 normalize_v3(axis);
1724                                 
1725                                 /* 6.0 = 6 grid units */
1726                                 axis[0] = t->center[0] - 6.0f * axis[0];
1727                                 axis[1] = t->center[1] - 6.0f * axis[1];
1728                                 axis[2] = t->center[2] - 6.0f * axis[2];
1729                                 
1730                                 projectIntView(t, axis, t->center2d);
1731                                 
1732                                 /* rotate only needs correct 2d center, grab needs ED_view3d_calc_zfac() value */
1733                                 if (t->mode == TFM_TRANSLATION) {
1734                                         copy_v3_v3(t->center, axis);
1735                                         copy_v3_v3(t->con.center, t->center);
1736                                 }
1737                         }
1738                 }
1739         }
1740         
1741         if (t->spacetype == SPACE_VIEW3D) {
1742                 /* ED_view3d_calc_zfac() defines a factor for perspective depth correction, used in ED_view3d_win_to_delta() */
1743                 float vec[3];
1744                 if (t->flag & (T_EDIT | T_POSE)) {
1745                         Object *ob = t->obedit ? t->obedit : t->poseobj;
1746                         mul_v3_m4v3(vec, ob->obmat, t->center);
1747                 }
1748                 else {
1749                         copy_v3_v3(vec, t->center);
1750                 }
1751
1752                 /* zfac is only used convertViewVec only in cases operator was invoked in RGN_TYPE_WINDOW
1753                  * and never used in other cases.
1754                  *
1755                  * We need special case here as well, since ED_view3d_calc_zfac will crahs when called
1756                  * for a region different from RGN_TYPE_WINDOW.
1757                  */
1758                 if (t->ar->regiontype == RGN_TYPE_WINDOW) {
1759                         t->zfac = ED_view3d_calc_zfac(t->ar->regiondata, vec, NULL);
1760                 }
1761                 else {
1762                         t->zfac = 0.0f;
1763                 }
1764         }
1765 }
1766
1767 void calculatePropRatio(TransInfo *t)
1768 {
1769         TransData *td = t->data;
1770         int i;
1771         float dist;
1772         short connected = t->flag & T_PROP_CONNECTED;
1773
1774         if (t->flag & T_PROP_EDIT) {
1775                 for (i = 0; i < t->total; i++, td++) {
1776                         if (td->flag & TD_SELECTED) {
1777                                 td->factor = 1.0f;
1778                         }
1779                         else if (t->flag & T_MIRROR && td->loc[0] * t->mirror < -0.00001f) {
1780                                 td->flag |= TD_SKIP;
1781                                 td->factor = 0.0f;
1782                                 restoreElement(td);
1783                         }
1784                         else if ((connected && (td->flag & TD_NOTCONNECTED || td->dist > t->prop_size)) ||
1785                                  (connected == 0 && td->rdist > t->prop_size))
1786                         {
1787                                 /*
1788                                  * The elements are sorted according to their dist member in the array,
1789                                  * that means we can stop when it finds one element outside of the propsize.
1790                                  * do not set 'td->flag |= TD_NOACTION', the prop circle is being changed.
1791                                  */
1792                                 
1793                                 td->factor = 0.0f;
1794                                 restoreElement(td);
1795                         }
1796                         else {
1797                                 /* Use rdist for falloff calculations, it is the real distance */
1798                                 td->flag &= ~TD_NOACTION;
1799                                 
1800                                 if (connected)
1801                                         dist = (t->prop_size - td->dist) / t->prop_size;
1802                                 else
1803                                         dist = (t->prop_size - td->rdist) / t->prop_size;
1804
1805                                 /*
1806                                  * Clamp to positive numbers.
1807                                  * Certain corner cases with connectivity and individual centers
1808                                  * can give values of rdist larger than propsize.
1809                                  */
1810                                 if (dist < 0.0f)
1811                                         dist = 0.0f;
1812                                 
1813                                 switch (t->prop_mode) {
1814                                         case PROP_SHARP:
1815                                                 td->factor = dist * dist;
1816                                                 break;
1817                                         case PROP_SMOOTH:
1818                                                 td->factor = 3.0f * dist * dist - 2.0f * dist * dist * dist;
1819                                                 break;
1820                                         case PROP_ROOT:
1821                                                 td->factor = (float)sqrt(dist);
1822                                                 break;
1823                                         case PROP_LIN:
1824                                                 td->factor = dist;
1825                                                 break;
1826                                         case PROP_CONST:
1827                                                 td->factor = 1.0f;
1828                                                 break;
1829                                         case PROP_SPHERE:
1830                                                 td->factor = (float)sqrt(2 * dist - dist * dist);
1831                                                 break;
1832                                         case PROP_RANDOM:
1833                                                 td->factor = BLI_frand() * dist;
1834                                                 break;
1835                                         default:
1836                                                 td->factor = 1;
1837                                                 break;
1838                                 }
1839                         }
1840                 }
1841                 switch (t->prop_mode) {
1842                         case PROP_SHARP:
1843                                 strcpy(t->proptext, IFACE_("(Sharp)"));
1844                                 break;
1845                         case PROP_SMOOTH:
1846                                 strcpy(t->proptext, IFACE_("(Smooth)"));
1847                                 break;
1848                         case PROP_ROOT:
1849                                 strcpy(t->proptext, IFACE_("(Root)"));
1850                                 break;
1851                         case PROP_LIN:
1852                                 strcpy(t->proptext, IFACE_("(Linear)"));
1853                                 break;
1854                         case PROP_CONST:
1855                                 strcpy(t->proptext, IFACE_("(Constant)"));
1856                                 break;
1857                         case PROP_SPHERE:
1858                                 strcpy(t->proptext, IFACE_("(Sphere)"));
1859                                 break;
1860                         case PROP_RANDOM:
1861                                 strcpy(t->proptext, IFACE_("(Random)"));
1862                                 break;
1863                         default:
1864                                 t->proptext[0] = '\0';
1865                                 break;
1866                 }
1867         }
1868         else {
1869                 for (i = 0; i < t->total; i++, td++) {
1870                         td->factor = 1.0;
1871                 }
1872                 t->proptext[0] = '\0';
1873         }
1874 }