code cleanup: split scons includes onto multiple lines, reduce chance of include...
[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 "BLO_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                                                 mult_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                 MovieTrackingTrack *track;
663                 int framenr = ED_space_clip_get_clip_frame_number(sc);
664
665                 flushTransTracking(t);
666
667                 track = tracksbase->first;
668                 while (track) {
669                         if (TRACK_VIEW_SELECTED(sc, track) && (track->flag & TRACK_LOCKED) == 0) {
670                                 MovieTrackingMarker *marker = BKE_tracking_marker_get(track, framenr);
671
672                                 if (t->mode == TFM_TRANSLATION) {
673                                         if (TRACK_AREA_SELECTED(track, TRACK_AREA_PAT))
674                                                 BKE_tracking_marker_clamp(marker, CLAMP_PAT_POS);
675                                         if (TRACK_AREA_SELECTED(track, TRACK_AREA_SEARCH))
676                                                 BKE_tracking_marker_clamp(marker, CLAMP_SEARCH_POS);
677                                 }
678                                 else if (t->mode == TFM_RESIZE) {
679                                         if (TRACK_AREA_SELECTED(track, TRACK_AREA_PAT))
680                                                 BKE_tracking_marker_clamp(marker, CLAMP_PAT_DIM);
681                                         if (TRACK_AREA_SELECTED(track, TRACK_AREA_SEARCH))
682                                                 BKE_tracking_marker_clamp(marker, CLAMP_SEARCH_DIM);
683                                 }
684                                 else if (t->mode == TFM_ROTATION) {
685                                         if (TRACK_AREA_SELECTED(track, TRACK_AREA_PAT))
686                                                 BKE_tracking_marker_clamp(marker, CLAMP_PAT_POS);
687                                 }
688                         }
689
690                         track = track->next;
691                 }
692
693                 DAG_id_tag_update(&clip->id, 0);
694         }
695         else if (t->options & CTX_MASK) {
696                 recalcData_mask_common(t);
697         }
698 }
699
700 /* helper for recalcData() - for 3d-view transforms */
701 static void recalcData_view3d(TransInfo *t)
702 {
703         Base *base = t->scene->basact;
704         
705         if (t->obedit) {
706                 if (ELEM(t->obedit->type, OB_CURVE, OB_SURF)) {
707                         Curve *cu = t->obedit->data;
708                         ListBase *nurbs = BKE_curve_editNurbs_get(cu);
709                         Nurb *nu = nurbs->first;
710                         
711                         if (t->state != TRANS_CANCEL) {
712                                 clipMirrorModifier(t, t->obedit);
713                                 applyProject(t);
714                         }
715                         
716                         DAG_id_tag_update(t->obedit->data, 0);  /* sets recalc flags */
717                                 
718                         if (t->state == TRANS_CANCEL) {
719                                 while (nu) {
720                                         BKE_nurb_handles_calc(nu); /* Cant do testhandlesNurb here, it messes up the h1 and h2 flags */
721                                         nu = nu->next;
722                                 }
723                         }
724                         else {
725                                 /* Normal updating */
726                                 while (nu) {
727                                         BKE_nurb_test2D(nu);
728                                         BKE_nurb_handles_calc(nu);
729                                         nu = nu->next;
730                                 }
731                         }
732                 }
733                 else if (t->obedit->type == OB_LATTICE) {
734                         Lattice *la = t->obedit->data;
735                         
736                         if (t->state != TRANS_CANCEL) {
737                                 applyProject(t);
738                         }
739                         
740                         DAG_id_tag_update(t->obedit->data, 0);  /* sets recalc flags */
741                         
742                         if (la->editlatt->latt->flag & LT_OUTSIDE) outside_lattice(la->editlatt->latt);
743                 }
744                 else if (t->obedit->type == OB_MESH) {
745                         BMEditMesh *em = BKE_editmesh_from_object(t->obedit);
746                         /* mirror modifier clipping? */
747                         if (t->state != TRANS_CANCEL) {
748                                 /* apply clipping after so we never project past the clip plane [#25423] */
749                                 applyProject(t);
750                                 clipMirrorModifier(t, t->obedit);
751                         }
752                         if ((t->options & CTX_NO_MIRROR) == 0 && (t->flag & T_MIRROR))
753                                 editbmesh_apply_to_mirror(t);
754                                 
755                         DAG_id_tag_update(t->obedit->data, 0);  /* sets recalc flags */
756                         
757                         EDBM_mesh_normals_update(em);
758                         BKE_editmesh_tessface_calc(em);
759                 }
760                 else if (t->obedit->type == OB_ARMATURE) { /* no recalc flag, does pose */
761                         bArmature *arm = t->obedit->data;
762                         ListBase *edbo = arm->edbo;
763                         EditBone *ebo;
764                         TransData *td = t->data;
765                         int i;
766                         
767                         if (t->state != TRANS_CANCEL) {
768                                 applyProject(t);
769                         }
770                         
771                         /* Ensure all bones are correctly adjusted */
772                         for (ebo = edbo->first; ebo; ebo = ebo->next) {
773                                 
774                                 if ((ebo->flag & BONE_CONNECTED) && ebo->parent) {
775                                         /* If this bone has a parent tip that has been moved */
776                                         if (ebo->parent->flag & BONE_TIPSEL) {
777                                                 copy_v3_v3(ebo->head, ebo->parent->tail);
778                                                 if (t->mode == TFM_BONE_ENVELOPE) ebo->rad_head = ebo->parent->rad_tail;
779                                         }
780                                         /* If this bone has a parent tip that has NOT been moved */
781                                         else {
782                                                 copy_v3_v3(ebo->parent->tail, ebo->head);
783                                                 if (t->mode == TFM_BONE_ENVELOPE) ebo->parent->rad_tail = ebo->rad_head;
784                                         }
785                                 }
786                                 
787                                 /* on extrude bones, oldlength==0.0f, so we scale radius of points */
788                                 ebo->length = len_v3v3(ebo->head, ebo->tail);
789                                 if (ebo->oldlength == 0.0f) {
790                                         ebo->rad_head = 0.25f * ebo->length;
791                                         ebo->rad_tail = 0.10f * ebo->length;
792                                         ebo->dist = 0.25f * ebo->length;
793                                         if (ebo->parent) {
794                                                 if (ebo->rad_head > ebo->parent->rad_tail)
795                                                         ebo->rad_head = ebo->parent->rad_tail;
796                                         }
797                                 }
798                                 else if (t->mode != TFM_BONE_ENVELOPE) {
799                                         /* if bones change length, lets do that for the deform distance as well */
800                                         ebo->dist *= ebo->length / ebo->oldlength;
801                                         ebo->rad_head *= ebo->length / ebo->oldlength;
802                                         ebo->rad_tail *= ebo->length / ebo->oldlength;
803                                         ebo->oldlength = ebo->length;
804                                 }
805                         }
806                         
807                         
808                         if (t->mode != TFM_BONE_ROLL) {
809                                 /* fix roll */
810                                 for (i = 0; i < t->total; i++, td++) {
811                                         if (td->extra) {
812                                                 float vec[3], up_axis[3];
813                                                 float qrot[4];
814                                                 float roll;
815                                                 
816                                                 ebo = td->extra;
817                                                 copy_v3_v3(up_axis, td->axismtx[2]);
818                                                 
819                                                 if (t->mode != TFM_ROTATION) {
820                                                         sub_v3_v3v3(vec, ebo->tail, ebo->head);
821                                                         normalize_v3(vec);
822                                                         rotation_between_vecs_to_quat(qrot, td->axismtx[1], vec);
823                                                         mul_qt_v3(qrot, up_axis);
824                                                 }
825                                                 else {
826                                                         mul_m3_v3(t->mat, up_axis);
827                                                 }
828                                                 
829                                                 /* roll has a tendency to flip in certain orientations - [#34283], [#33974] */
830                                                 roll = ED_rollBoneToVector(ebo, up_axis, false);
831                                                 ebo->roll = angle_compat_rad(roll, ebo->roll);
832                                         }
833                                 }
834                         }
835                         
836                         if (arm->flag & ARM_MIRROR_EDIT)
837                                 transform_armature_mirror_update(t->obedit);
838                         
839                 }
840                 else {
841                         if (t->state != TRANS_CANCEL) {
842                                 applyProject(t);
843                         }
844                         DAG_id_tag_update(t->obedit->data, 0);  /* sets recalc flags */
845                 }
846         }
847         else if ((t->flag & T_POSE) && t->poseobj) {
848                 Object *ob = t->poseobj;
849                 bArmature *arm = ob->data;
850                 
851                 /* if animtimer is running, and the object already has animation data,
852                  * check if the auto-record feature means that we should record 'samples'
853                  * (i.e. uneditable animation values)
854                  *
855                  * context is needed for keying set poll() functions.
856                  */
857                 // TODO: autokeyframe calls need some setting to specify to add samples (FPoints) instead of keyframes?
858                 if ((t->animtimer) && (t->context) && IS_AUTOKEY_ON(t->scene)) {
859                         int targetless_ik = (t->flag & T_AUTOIK); // XXX this currently doesn't work, since flags aren't set yet!
860                         
861                         animrecord_check_state(t->scene, &ob->id, t->animtimer);
862                         autokeyframe_pose_cb_func(t->context, t->scene, (View3D *)t->view, ob, t->mode, targetless_ik);
863                 }
864                 
865                 /* old optimize trick... this enforces to bypass the depgraph */
866                 if (!(arm->flag & ARM_DELAYDEFORM)) {
867                         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);  /* sets recalc flags */
868                         /* transformation of pose may affect IK tree, make sure it is rebuilt */
869                         BIK_clear_data(ob->pose);
870                 }
871                 else
872                         BKE_pose_where_is(t->scene, ob);
873         }
874         else if (base && (base->object->mode & OB_MODE_PARTICLE_EDIT) && PE_get_current(t->scene, base->object)) {
875                 if (t->state != TRANS_CANCEL) {
876                         applyProject(t);
877                 }
878                 flushTransParticles(t);
879         }
880         else {
881                 int i;
882                 
883                 if (t->state != TRANS_CANCEL) {
884                         applyProject(t);
885                 }
886                 
887                 for (i = 0; i < t->total; i++) {
888                         TransData *td = t->data + i;
889                         Object *ob = td->ob;
890                         
891                         if (td->flag & TD_NOACTION)
892                                 break;
893                         
894                         if (td->flag & TD_SKIP)
895                                 continue;
896                         
897                         /* if animtimer is running, and the object already has animation data,
898                          * check if the auto-record feature means that we should record 'samples'
899                          * (i.e. uneditable animation values)
900                          */
901                         // TODO: autokeyframe calls need some setting to specify to add samples (FPoints) instead of keyframes?
902                         if ((t->animtimer) && IS_AUTOKEY_ON(t->scene)) {
903                                 animrecord_check_state(t->scene, &ob->id, t->animtimer);
904                                 autokeyframe_ob_cb_func(t->context, t->scene, (View3D *)t->view, ob, t->mode);
905                         }
906                         
907                         /* sets recalc flags fully, instead of flushing existing ones 
908                          * otherwise proxies don't function correctly
909                          */
910                         DAG_id_tag_update(&ob->id, OB_RECALC_OB);
911                 }
912         }
913 }
914
915 /* helper for recalcData() - for sequencer transforms */
916 static void recalcData_sequencer(TransInfo *t)
917 {
918         TransData *td;
919         int a;
920         Sequence *seq_prev = NULL;
921
922         for (a = 0, td = t->data; a < t->total; a++, td++) {
923                 TransDataSeq *tdsq = (TransDataSeq *) td->extra;
924                 Sequence *seq = tdsq->seq;
925
926                 if (seq != seq_prev) {
927                         BKE_sequence_invalidate_dependent(t->scene, seq);
928                 }
929
930                 seq_prev = seq;
931         }
932
933         BKE_sequencer_preprocessed_cache_cleanup();
934
935         flushTransSeq(t);
936 }
937
938 /* called for updating while transform acts, once per redraw */
939 void recalcData(TransInfo *t)
940 {
941         if (t->spacetype == SPACE_NODE) {
942                 flushTransNodes(t);
943         }
944         else if (t->spacetype == SPACE_SEQ) {
945                 recalcData_sequencer(t);
946         }
947         else if (t->spacetype == SPACE_ACTION) {
948                 recalcData_actedit(t);
949         }
950         else if (t->spacetype == SPACE_IPO) {
951                 recalcData_graphedit(t);
952         }
953         else if (t->spacetype == SPACE_NLA) {
954                 recalcData_nla(t);
955         }
956         else if (t->spacetype == SPACE_IMAGE) {
957                 recalcData_image(t);
958         }
959         else if (t->spacetype == SPACE_VIEW3D) {
960                 recalcData_view3d(t);
961         }
962         else if (t->spacetype == SPACE_CLIP) {
963                 recalcData_spaceclip(t);
964         }
965 }
966
967 void drawLine(TransInfo *t, const float center[3], const float dir[3], char axis, short options)
968 {
969         float v1[3], v2[3], v3[3];
970         unsigned char col[3], col2[3];
971
972         if (t->spacetype == SPACE_VIEW3D) {
973                 View3D *v3d = t->view;
974                 
975                 glPushMatrix();
976                 
977                 //if (t->obedit) glLoadMatrixf(t->obedit->obmat);       // sets opengl viewing
978                 
979                 
980                 copy_v3_v3(v3, dir);
981                 mul_v3_fl(v3, v3d->far);
982                 
983                 sub_v3_v3v3(v2, center, v3);
984                 add_v3_v3v3(v1, center, v3);
985                 
986                 if (options & DRAWLIGHT) {
987                         col[0] = col[1] = col[2] = 220;
988                 }
989                 else {
990                         UI_GetThemeColor3ubv(TH_GRID, col);
991                 }
992                 UI_make_axis_color(col, col2, axis);
993                 glColor3ubv(col2);
994                 
995                 setlinestyle(0);
996                 glBegin(GL_LINE_STRIP);
997                 glVertex3fv(v1);
998                 glVertex3fv(v2);
999                 glEnd();
1000                 
1001                 glPopMatrix();
1002         }
1003 }
1004
1005 /**
1006  * Free data before switching to another mode.
1007  */
1008 void resetTransModal(TransInfo *t)
1009 {
1010         if (t->mode == TFM_EDGE_SLIDE) {
1011                 freeEdgeSlideVerts(t);
1012         }
1013         else if (t->mode == TFM_VERT_SLIDE) {
1014                 freeVertSlideVerts(t);
1015         }
1016 }
1017
1018 void resetTransRestrictions(TransInfo *t)
1019 {
1020         t->flag &= ~T_ALL_RESTRICTIONS;
1021 }
1022
1023 /* the *op can be NULL */
1024 int initTransInfo(bContext *C, TransInfo *t, wmOperator *op, const wmEvent *event)
1025 {
1026         Scene *sce = CTX_data_scene(C);
1027         ToolSettings *ts = CTX_data_tool_settings(C);
1028         ARegion *ar = CTX_wm_region(C);
1029         ScrArea *sa = CTX_wm_area(C);
1030         Object *obedit = CTX_data_edit_object(C);
1031         
1032         /* moving: is shown in drawobject() (transform color) */
1033 //  TRANSFORM_FIX_ME
1034 //      if (obedit || (t->flag & T_POSE) ) G.moving = G_TRANSFORM_EDIT;
1035 //      else if (G.f & G_PARTICLEEDIT) G.moving = G_TRANSFORM_PARTICLE;
1036 //      else G.moving = G_TRANSFORM_OBJ;
1037         
1038         t->scene = sce;
1039         t->sa = sa;
1040         t->ar = ar;
1041         t->obedit = obedit;
1042         t->settings = ts;
1043
1044         if (obedit) {
1045                 copy_m3_m4(t->obedit_mat, obedit->obmat);
1046                 normalize_m3(t->obedit_mat);
1047         }
1048
1049         t->data = NULL;
1050         t->ext = NULL;
1051         
1052         t->helpline = HLP_NONE;
1053         
1054         t->flag = 0;
1055         
1056         t->redraw = 1; /* redraw first time */
1057         
1058         if (event) {
1059                 copy_v2_v2_int(t->imval, event->mval);
1060                 t->event_type = event->type;
1061         }
1062         else {
1063                 t->imval[0] = 0;
1064                 t->imval[1] = 0;
1065         }
1066         
1067         t->con.imval[0] = t->imval[0];
1068         t->con.imval[1] = t->imval[1];
1069         
1070         t->mval[0] = t->imval[0];
1071         t->mval[1] = t->imval[1];
1072         
1073         t->transform        = NULL;
1074         t->handleEvent      = NULL;
1075         
1076         t->total            = 0;
1077         
1078         t->val = 0.0f;
1079
1080         zero_v3(t->vec);
1081         zero_v3(t->center);
1082
1083         unit_m3(t->mat);
1084         
1085         /* if there's an event, we're modal */
1086         if (event) {
1087                 t->flag |= T_MODAL;
1088         }
1089
1090         /* Crease needs edge flag */
1091         if (ELEM(t->mode, TFM_CREASE, TFM_BWEIGHT)) {
1092                 t->options |= CTX_EDGE;
1093         }
1094
1095
1096         /* Assign the space type, some exceptions for running in different mode */
1097         if (sa == NULL) {
1098                 /* background mode */
1099                 t->spacetype = SPACE_EMPTY;
1100         }
1101         else if ((ar == NULL) && (sa->spacetype == SPACE_VIEW3D)) {
1102                 /* running in the text editor */
1103                 t->spacetype = SPACE_EMPTY;
1104         }
1105         else {
1106                 /* normal operation */
1107                 t->spacetype = sa->spacetype;
1108         }
1109
1110
1111         if (t->spacetype == SPACE_VIEW3D) {
1112                 View3D *v3d = sa->spacedata.first;
1113                 bScreen *animscreen = ED_screen_animation_playing(CTX_wm_manager(C));
1114                 
1115                 t->view = v3d;
1116                 t->animtimer = (animscreen) ? animscreen->animtimer : NULL;
1117                 
1118                 /* turn manipulator off during transform */
1119                 // FIXME: but don't do this when USING the manipulator...
1120                 if (t->flag & T_MODAL) {
1121                         t->twtype = v3d->twtype;
1122                         v3d->twtype = 0;
1123                 }
1124
1125                 if (v3d->flag & V3D_ALIGN) t->flag |= T_V3D_ALIGN;
1126                 t->around = v3d->around;
1127                 
1128                 if (op && (RNA_struct_find_property(op->ptr, "constraint_orientation") &&
1129                            RNA_struct_property_is_set(op->ptr, "constraint_orientation")))
1130                 {
1131                         t->current_orientation = RNA_enum_get(op->ptr, "constraint_orientation");
1132
1133                         if (t->current_orientation >= V3D_MANIP_CUSTOM + BIF_countTransformOrientation(C)) {
1134                                 t->current_orientation = V3D_MANIP_GLOBAL;
1135                         }
1136                 }
1137                 else {
1138                         t->current_orientation = v3d->twmode;
1139                 }
1140
1141                 /* exceptional case */
1142                 if (t->around == V3D_LOCAL && (t->settings->selectmode & SCE_SELECT_FACE)) {
1143                         if (ELEM3(t->mode, TFM_ROTATION, TFM_RESIZE, TFM_TRACKBALL)) {
1144                                 t->options |= CTX_NO_PET;
1145                         }
1146                 }
1147
1148                 /* initialize UV transform from */
1149                 if (op && RNA_struct_find_property(op->ptr, "correct_uv")) {
1150                         if (RNA_struct_property_is_set(op->ptr, "correct_uv")) {
1151                                 if (RNA_boolean_get(op->ptr, "correct_uv")) {
1152                                         t->settings->uvcalc_flag |= UVCALC_TRANSFORM_CORRECT;
1153                                 }
1154                                 else {
1155                                         t->settings->uvcalc_flag &= ~UVCALC_TRANSFORM_CORRECT;
1156                                 }
1157                         }
1158                         else {
1159                                 RNA_boolean_set(op->ptr, "correct_uv", t->settings->uvcalc_flag & UVCALC_TRANSFORM_CORRECT);
1160                         }
1161                 }
1162
1163         }
1164         else if (t->spacetype == SPACE_IMAGE) {
1165                 SpaceImage *sima = sa->spacedata.first;
1166                 // XXX for now, get View2D from the active region
1167                 t->view = &ar->v2d;
1168                 t->around = sima->around;
1169
1170                 if (ED_space_image_show_uvedit(sima, t->obedit)) {
1171                         /* UV transform */
1172                 }
1173                 else if (sima->mode == SI_MODE_MASK) {
1174                         t->options |= CTX_MASK;
1175                 }
1176                 else {
1177                         /* image not in uv edit, nor in mask mode, can happen for some tools */
1178                 }
1179         }
1180         else if (t->spacetype == SPACE_NODE) {
1181                 // XXX for now, get View2D from the active region
1182                 t->view = &ar->v2d;
1183                 t->around = V3D_CENTER;
1184         }
1185         else if (t->spacetype == SPACE_IPO) {
1186                 SpaceIpo *sipo = sa->spacedata.first;
1187                 t->view = &ar->v2d;
1188                 t->around = sipo->around;
1189         }
1190         else if (t->spacetype == SPACE_CLIP) {
1191                 SpaceClip *sclip = sa->spacedata.first;
1192                 t->view = &ar->v2d;
1193                 t->around = sclip->around;
1194
1195                 if (ED_space_clip_check_show_trackedit(sclip))
1196                         t->options |= CTX_MOVIECLIP;
1197                 else if (ED_space_clip_check_show_maskedit(sclip))
1198                         t->options |= CTX_MASK;
1199         }
1200         else {
1201                 if (ar) {
1202                         // XXX for now, get View2D  from the active region
1203                         t->view = &ar->v2d;
1204                         // XXX for now, the center point is the midpoint of the data
1205                 }
1206                 else {
1207                         t->view = NULL;
1208                 }
1209                 t->around = V3D_CENTER;
1210         }
1211         
1212         if (op && RNA_struct_property_is_set(op->ptr, "release_confirm")) {
1213                 if (RNA_boolean_get(op->ptr, "release_confirm")) {
1214                         t->flag |= T_RELEASE_CONFIRM;
1215                 }
1216         }
1217         else {
1218                 if (U.flag & USER_RELEASECONFIRM) {
1219                         t->flag |= T_RELEASE_CONFIRM;
1220                 }
1221         }
1222
1223         if (op && (RNA_struct_find_property(op->ptr, "mirror") &&
1224                    RNA_struct_property_is_set(op->ptr, "mirror")))
1225         {
1226                 if (RNA_boolean_get(op->ptr, "mirror")) {
1227                         t->flag |= T_MIRROR;
1228                         t->mirror = 1;
1229                 }
1230         }
1231         // Need stuff to take it from edit mesh or whatnot here
1232         else if (t->spacetype == SPACE_VIEW3D) {
1233                 if (t->obedit && t->obedit->type == OB_MESH && (((Mesh *)t->obedit->data)->editflag & ME_EDIT_MIRROR_X)) {
1234                         t->flag |= T_MIRROR;
1235                         t->mirror = 1;
1236                 }
1237         }
1238         
1239         /* setting PET flag only if property exist in operator. Otherwise, assume it's not supported */
1240         if (op && RNA_struct_find_property(op->ptr, "proportional")) {
1241                 if (RNA_struct_property_is_set(op->ptr, "proportional")) {
1242                         switch (RNA_enum_get(op->ptr, "proportional")) {
1243                                 case PROP_EDIT_CONNECTED:
1244                                         t->flag |= T_PROP_CONNECTED;
1245                                 case PROP_EDIT_ON:
1246                                         t->flag |= T_PROP_EDIT;
1247                                         break;
1248                         }
1249                 }
1250                 else {
1251                         /* use settings from scene only if modal */
1252                         if (t->flag & T_MODAL) {
1253                                 if ((t->options & CTX_NO_PET) == 0) {
1254                                         if (t->obedit && ts->proportional != PROP_EDIT_OFF) {
1255                                                 t->flag |= T_PROP_EDIT;
1256
1257                                                 if (ts->proportional == PROP_EDIT_CONNECTED) {
1258                                                         t->flag |= T_PROP_CONNECTED;
1259                                                 }
1260                                         }
1261                                         else if (t->options & CTX_MASK) {
1262                                                 if (ts->proportional_mask) {
1263                                                         t->flag |= T_PROP_EDIT;
1264
1265                                                         if (ts->proportional == PROP_EDIT_CONNECTED) {
1266                                                                 t->flag |= T_PROP_CONNECTED;
1267                                                         }
1268                                                 }
1269                                         }
1270                                         else if (t->obedit == NULL && ts->proportional_objects) {
1271                                                 t->flag |= T_PROP_EDIT;
1272                                         }
1273                                 }
1274                         }
1275                 }
1276                 
1277                 if (op && (RNA_struct_find_property(op->ptr, "proportional_size") &&
1278                            RNA_struct_property_is_set(op->ptr, "proportional_size")))
1279                 {
1280                         t->prop_size = RNA_float_get(op->ptr, "proportional_size");
1281                 }
1282                 else {
1283                         t->prop_size = ts->proportional_size;
1284                 }
1285                 
1286                 
1287                 /* TRANSFORM_FIX_ME rna restrictions */
1288                 if (t->prop_size <= 0.00001f) {
1289                         printf("Proportional size (%f) under 0.00001, resetting to 1!\n", t->prop_size);
1290                         t->prop_size = 1.0f;
1291                 }
1292                 
1293                 if (op && (RNA_struct_find_property(op->ptr, "proportional_edit_falloff") &&
1294                            RNA_struct_property_is_set(op->ptr, "proportional_edit_falloff")))
1295                 {
1296                         t->prop_mode = RNA_enum_get(op->ptr, "proportional_edit_falloff");
1297                 }
1298                 else {
1299                         t->prop_mode = ts->prop_mode;
1300                 }
1301         }
1302         else { /* add not pet option to context when not available */
1303                 t->options |= CTX_NO_PET;
1304         }
1305         
1306         // Mirror is not supported with PET, turn it off.
1307         if (t->flag & T_PROP_EDIT) {
1308                 t->flag &= ~T_MIRROR;
1309         }
1310
1311         setTransformViewMatrices(t);
1312         initNumInput(&t->num);
1313         
1314         return 1;
1315 }
1316
1317 /* Here I would suggest only TransInfo related issues, like free data & reset vars. Not redraws */
1318 void postTrans(bContext *C, TransInfo *t)
1319 {
1320         TransData *td;
1321         
1322         if (t->draw_handle_view)
1323                 ED_region_draw_cb_exit(t->ar->type, t->draw_handle_view);
1324         if (t->draw_handle_apply)
1325                 ED_region_draw_cb_exit(t->ar->type, t->draw_handle_apply);
1326         if (t->draw_handle_pixel)
1327                 ED_region_draw_cb_exit(t->ar->type, t->draw_handle_pixel);
1328         if (t->draw_handle_cursor)
1329                 WM_paint_cursor_end(CTX_wm_manager(C), t->draw_handle_cursor);
1330
1331         if (t->customFree) {
1332                 /* Can take over freeing t->data and data2d etc... */
1333                 t->customFree(t);
1334                 BLI_assert(t->customData == NULL);
1335         }
1336         else if ((t->customData != NULL) && (t->flag & T_FREE_CUSTOMDATA)) {
1337                 MEM_freeN(t->customData);
1338                 t->customData = NULL;
1339         }
1340
1341         /* postTrans can be called when nothing is selected, so data is NULL already */
1342         if (t->data) {
1343                 
1344                 /* free data malloced per trans-data */
1345                 if ((t->obedit && ELEM(t->obedit->type, OB_CURVE, OB_SURF)) ||
1346                     (t->spacetype == SPACE_IPO))
1347                 {
1348                         int a;
1349                         for (a = 0, td = t->data; a < t->total; a++, td++) {
1350                                 if (td->flag & TD_BEZTRIPLE) {
1351                                         MEM_freeN(td->hdata);
1352                                 }
1353                         }
1354                 }
1355                 MEM_freeN(t->data);
1356         }
1357         
1358         BLI_freelistN(&t->tsnap.points);
1359
1360         if (t->ext) MEM_freeN(t->ext);
1361         if (t->data2d) {
1362                 MEM_freeN(t->data2d);
1363                 t->data2d = NULL;
1364         }
1365         
1366         if (t->spacetype == SPACE_IMAGE) {
1367                 if (t->options & CTX_MASK) {
1368                         /* pass */
1369                 }
1370                 else {
1371                         SpaceImage *sima = t->sa->spacedata.first;
1372                         if (sima->flag & SI_LIVE_UNWRAP)
1373                                 ED_uvedit_live_unwrap_end(t->state == TRANS_CANCEL);
1374                 }
1375         }
1376         else if (t->spacetype == SPACE_VIEW3D) {
1377                 View3D *v3d = t->sa->spacedata.first;
1378                 /* restore manipulator */
1379                 if (t->flag & T_MODAL) {
1380                         v3d->twtype = t->twtype;
1381                 }
1382         }
1383         
1384         if (t->mouse.data) {
1385                 MEM_freeN(t->mouse.data);
1386         }
1387 }
1388
1389 void applyTransObjects(TransInfo *t)
1390 {
1391         TransData *td;
1392         
1393         for (td = t->data; td < t->data + t->total; td++) {
1394                 copy_v3_v3(td->iloc, td->loc);
1395                 if (td->ext->rot) {
1396                         copy_v3_v3(td->ext->irot, td->ext->rot);
1397                 }
1398                 if (td->ext->size) {
1399                         copy_v3_v3(td->ext->isize, td->ext->size);
1400                 }
1401         }
1402         recalcData(t);
1403 }
1404
1405 static void restoreElement(TransData *td)
1406 {
1407         /* TransData for crease has no loc */
1408         if (td->loc) {
1409                 copy_v3_v3(td->loc, td->iloc);
1410         }
1411         if (td->val) {
1412                 *td->val = td->ival;
1413         }
1414
1415         if (td->ext && (td->flag & TD_NO_EXT) == 0) {
1416                 if (td->ext->rot) {
1417                         copy_v3_v3(td->ext->rot, td->ext->irot);
1418                 }
1419                 if (td->ext->rotAngle) {
1420                         *td->ext->rotAngle = td->ext->irotAngle;
1421                 }
1422                 if (td->ext->rotAxis) {
1423                         copy_v3_v3(td->ext->rotAxis, td->ext->irotAxis);
1424                 }
1425                 /* XXX, drotAngle & drotAxis not used yet */
1426                 if (td->ext->size) {
1427                         copy_v3_v3(td->ext->size, td->ext->isize);
1428                 }
1429                 if (td->ext->quat) {
1430                         copy_qt_qt(td->ext->quat, td->ext->iquat);
1431                 }
1432         }
1433         
1434         if (td->flag & TD_BEZTRIPLE) {
1435                 *(td->hdata->h1) = td->hdata->ih1;
1436                 *(td->hdata->h2) = td->hdata->ih2;
1437         }
1438 }
1439
1440 void restoreTransObjects(TransInfo *t)
1441 {
1442         TransData *td;
1443         TransData2D *td2d;
1444
1445         for (td = t->data; td < t->data + t->total; td++) {
1446                 restoreElement(td);
1447         }
1448         
1449         for (td2d = t->data2d; t->data2d && td2d < t->data2d + t->total; td2d++) {
1450                 if (td2d->h1) {
1451                         td2d->h1[0] = td2d->ih1[0];
1452                         td2d->h1[1] = td2d->ih1[1];
1453                 }
1454                 if (td2d->h2) {
1455                         td2d->h2[0] = td2d->ih2[0];
1456                         td2d->h2[1] = td2d->ih2[1];
1457                 }
1458         }
1459
1460         unit_m3(t->mat);
1461         
1462         recalcData(t);
1463 }
1464
1465 void calculateCenter2D(TransInfo *t)
1466 {
1467         if (t->flag & (T_EDIT | T_POSE)) {
1468                 Object *ob = t->obedit ? t->obedit : t->poseobj;
1469                 float vec[3];
1470                 
1471                 copy_v3_v3(vec, t->center);
1472                 mul_m4_v3(ob->obmat, vec);
1473                 projectIntView(t, vec, t->center2d);
1474         }
1475         else {
1476                 projectIntView(t, t->center, t->center2d);
1477         }
1478 }
1479
1480 void calculateCenterCursor(TransInfo *t)
1481 {
1482         const float *cursor;
1483         
1484         cursor = give_cursor(t->scene, t->view);
1485         copy_v3_v3(t->center, cursor);
1486         
1487         /* If edit or pose mode, move cursor in local space */
1488         if (t->flag & (T_EDIT | T_POSE)) {
1489                 Object *ob = t->obedit ? t->obedit : t->poseobj;
1490                 float mat[3][3], imat[3][3];
1491                 
1492                 sub_v3_v3v3(t->center, t->center, ob->obmat[3]);
1493                 copy_m3_m4(mat, ob->obmat);
1494                 invert_m3_m3(imat, mat);
1495                 mul_m3_v3(imat, t->center);
1496         }
1497         
1498         calculateCenter2D(t);
1499 }
1500
1501 void calculateCenterCursor2D(TransInfo *t)
1502 {
1503         float aspx = 1.0, aspy = 1.0;
1504         float *cursor = NULL;
1505         
1506         if (t->spacetype == SPACE_IMAGE) {
1507                 SpaceImage *sima = (SpaceImage *)t->sa->spacedata.first;
1508                 /* only space supported right now but may change */
1509                 if (t->options & CTX_MASK) {
1510                         ED_space_image_get_aspect(sima, &aspx, &aspy);
1511                 }
1512                 else {
1513                         ED_space_image_get_uv_aspect(sima, &aspx, &aspy);
1514                 }
1515                 cursor = sima->cursor;
1516         }
1517         
1518         if (cursor) {
1519                 if (t->options & CTX_MASK) {
1520                         float co[2];
1521                         float frame_size[2];
1522                         SpaceImage *sima = (SpaceImage *)t->sa->spacedata.first;
1523                         ED_space_image_get_size_fl(sima, frame_size);
1524
1525                         BKE_mask_coord_from_frame(co, cursor, frame_size);
1526
1527                         ED_space_image_get_aspect(sima, &aspx, &aspy);
1528
1529                         t->center[0] = co[0] * aspx;
1530                         t->center[1] = co[1] * aspy;
1531                 }
1532                 else {
1533                         t->center[0] = cursor[0] * aspx;
1534                         t->center[1] = cursor[1] * aspy;
1535                 }
1536         }
1537         
1538         calculateCenter2D(t);
1539 }
1540
1541 static void calculateCenterCursorGraph2D(TransInfo *t)
1542 {
1543         SpaceIpo *sipo = (SpaceIpo *)t->sa->spacedata.first;
1544         Scene *scene = t->scene;
1545         
1546         /* cursor is combination of current frame, and graph-editor cursor value */
1547         t->center[0] = (float)(scene->r.cfra);
1548         t->center[1] = sipo->cursorVal;
1549         
1550         calculateCenter2D(t);
1551 }
1552
1553 void calculateCenterMedian(TransInfo *t)
1554 {
1555         float partial[3] = {0.0f, 0.0f, 0.0f};
1556         int total = 0;
1557         int i;
1558         
1559         for (i = 0; i < t->total; i++) {
1560                 if (t->data[i].flag & TD_SELECTED) {
1561                         if (!(t->data[i].flag & TD_NOCENTER)) {
1562                                 add_v3_v3(partial, t->data[i].center);
1563                                 total++;
1564                         }
1565                 }
1566         }
1567         if (i)
1568                 mul_v3_fl(partial, 1.0f / total);
1569         copy_v3_v3(t->center, partial);
1570         
1571         calculateCenter2D(t);
1572 }
1573
1574 void calculateCenterBound(TransInfo *t)
1575 {
1576         float max[3];
1577         float min[3];
1578         int i;
1579         for (i = 0; i < t->total; i++) {
1580                 if (i) {
1581                         if (t->data[i].flag & TD_SELECTED) {
1582                                 if (!(t->data[i].flag & TD_NOCENTER))
1583                                         minmax_v3v3_v3(min, max, t->data[i].center);
1584                         }
1585                 }
1586                 else {
1587                         copy_v3_v3(max, t->data[i].center);
1588                         copy_v3_v3(min, t->data[i].center);
1589                 }
1590         }
1591         mid_v3_v3v3(t->center, min, max);
1592
1593         calculateCenter2D(t);
1594 }
1595
1596 void calculateCenter(TransInfo *t)
1597 {
1598         switch (t->around) {
1599                 case V3D_CENTER:
1600                         calculateCenterBound(t);
1601                         break;
1602                 case V3D_CENTROID:
1603                         calculateCenterMedian(t);
1604                         break;
1605                 case V3D_CURSOR:
1606                         if (t->spacetype == SPACE_IMAGE)
1607                                 calculateCenterCursor2D(t);
1608                         else if (t->spacetype == SPACE_IPO)
1609                                 calculateCenterCursorGraph2D(t);
1610                         else
1611                                 calculateCenterCursor(t);
1612                         break;
1613                 case V3D_LOCAL:
1614                         /* Individual element center uses median center for helpline and such */
1615                         calculateCenterMedian(t);
1616                         break;
1617                 case V3D_ACTIVE:
1618                 {
1619                         /* set median, and if if if... do object center */
1620                 
1621                         /* EDIT MODE ACTIVE EDITMODE ELEMENT */
1622
1623                         if (t->obedit) {
1624                                 if (t->obedit && t->obedit->type == OB_MESH) {
1625                                         BMEditSelection ese;
1626                                         BMEditMesh *em = BKE_editmesh_from_object(t->obedit);
1627
1628                                         if (BM_select_history_active_get(em->bm, &ese)) {
1629                                                 BM_editselection_center(&ese, t->center);
1630                                                 calculateCenter2D(t);
1631                                                 break;
1632                                         }
1633                                 }
1634                                 else if (ELEM(t->obedit->type, OB_CURVE, OB_SURF)) {
1635                                         float center[3];
1636                                         Curve *cu = (Curve *)t->obedit->data;
1637
1638                                         if (ED_curve_actSelection(cu, center)) {
1639                                                 copy_v3_v3(t->center, center);
1640                                                 calculateCenter2D(t);
1641                                                 break;
1642                                         }
1643                                 }
1644                         } /* END EDIT MODE ACTIVE ELEMENT */
1645
1646                         calculateCenterMedian(t);
1647                         if ((t->flag & (T_EDIT | T_POSE)) == 0) {
1648                                 Scene *scene = t->scene;
1649                                 Object *ob = OBACT;
1650                                 if (ob) {
1651                                         copy_v3_v3(t->center, ob->obmat[3]);
1652                                         projectIntView(t, t->center, t->center2d);
1653                                 }
1654                         }
1655                 
1656                 }
1657         }
1658         
1659         /* setting constraint center */
1660         copy_v3_v3(t->con.center, t->center);
1661         if (t->flag & (T_EDIT | T_POSE)) {
1662                 Object *ob = t->obedit ? t->obedit : t->poseobj;
1663                 mul_m4_v3(ob->obmat, t->con.center);
1664         }
1665         
1666         /* for panning from cameraview */
1667         if (t->flag & T_OBJECT) {
1668                 if (t->spacetype == SPACE_VIEW3D && t->ar && t->ar->regiontype == RGN_TYPE_WINDOW) {
1669                         View3D *v3d = t->view;
1670                         Scene *scene = t->scene;
1671                         RegionView3D *rv3d = t->ar->regiondata;
1672                         
1673                         if (v3d->camera == OBACT && rv3d->persp == RV3D_CAMOB) {
1674                                 float axis[3];
1675                                 /* persinv is nasty, use viewinv instead, always right */
1676                                 copy_v3_v3(axis, t->viewinv[2]);
1677                                 normalize_v3(axis);
1678                                 
1679                                 /* 6.0 = 6 grid units */
1680                                 axis[0] = t->center[0] - 6.0f * axis[0];
1681                                 axis[1] = t->center[1] - 6.0f * axis[1];
1682                                 axis[2] = t->center[2] - 6.0f * axis[2];
1683                                 
1684                                 projectIntView(t, axis, t->center2d);
1685                                 
1686                                 /* rotate only needs correct 2d center, grab needs ED_view3d_calc_zfac() value */
1687                                 if (t->mode == TFM_TRANSLATION) {
1688                                         copy_v3_v3(t->center, axis);
1689                                         copy_v3_v3(t->con.center, t->center);
1690                                 }
1691                         }
1692                 }
1693         }
1694         
1695         if (t->spacetype == SPACE_VIEW3D) {
1696                 /* ED_view3d_calc_zfac() defines a factor for perspective depth correction, used in ED_view3d_win_to_delta() */
1697                 float vec[3];
1698                 if (t->flag & (T_EDIT | T_POSE)) {
1699                         Object *ob = t->obedit ? t->obedit : t->poseobj;
1700                         mul_v3_m4v3(vec, ob->obmat, t->center);
1701                 }
1702                 else {
1703                         copy_v3_v3(vec, t->center);
1704                 }
1705                 t->zfac = ED_view3d_calc_zfac(t->ar->regiondata, vec, NULL);
1706         }
1707 }
1708
1709 void calculatePropRatio(TransInfo *t)
1710 {
1711         TransData *td = t->data;
1712         int i;
1713         float dist;
1714         short connected = t->flag & T_PROP_CONNECTED;
1715
1716         if (t->flag & T_PROP_EDIT) {
1717                 for (i = 0; i < t->total; i++, td++) {
1718                         if (td->flag & TD_SELECTED) {
1719                                 td->factor = 1.0f;
1720                         }
1721                         else if (t->flag & T_MIRROR && td->loc[0] * t->mirror < -0.00001f) {
1722                                 td->flag |= TD_SKIP;
1723                                 td->factor = 0.0f;
1724                                 restoreElement(td);
1725                         }
1726                         else if ((connected && (td->flag & TD_NOTCONNECTED || td->dist > t->prop_size)) ||
1727                                  (connected == 0 && td->rdist > t->prop_size))
1728                         {
1729                                 /*
1730                                  * The elements are sorted according to their dist member in the array,
1731                                  * that means we can stop when it finds one element outside of the propsize.
1732                                  * do not set 'td->flag |= TD_NOACTION', the prop circle is being changed.
1733                                  */
1734                                 
1735                                 td->factor = 0.0f;
1736                                 restoreElement(td);
1737                         }
1738                         else {
1739                                 /* Use rdist for falloff calculations, it is the real distance */
1740                                 td->flag &= ~TD_NOACTION;
1741                                 
1742                                 if (connected)
1743                                         dist = (t->prop_size - td->dist) / t->prop_size;
1744                                 else
1745                                         dist = (t->prop_size - td->rdist) / t->prop_size;
1746
1747                                 /*
1748                                  * Clamp to positive numbers.
1749                                  * Certain corner cases with connectivity and individual centers
1750                                  * can give values of rdist larger than propsize.
1751                                  */
1752                                 if (dist < 0.0f)
1753                                         dist = 0.0f;
1754                                 
1755                                 switch (t->prop_mode) {
1756                                         case PROP_SHARP:
1757                                                 td->factor = dist * dist;
1758                                                 break;
1759                                         case PROP_SMOOTH:
1760                                                 td->factor = 3.0f * dist * dist - 2.0f * dist * dist * dist;
1761                                                 break;
1762                                         case PROP_ROOT:
1763                                                 td->factor = (float)sqrt(dist);
1764                                                 break;
1765                                         case PROP_LIN:
1766                                                 td->factor = dist;
1767                                                 break;
1768                                         case PROP_CONST:
1769                                                 td->factor = 1.0f;
1770                                                 break;
1771                                         case PROP_SPHERE:
1772                                                 td->factor = (float)sqrt(2 * dist - dist * dist);
1773                                                 break;
1774                                         case PROP_RANDOM:
1775                                                 td->factor = BLI_frand() * dist;
1776                                                 break;
1777                                         default:
1778                                                 td->factor = 1;
1779                                 }
1780                         }
1781                 }
1782                 switch (t->prop_mode) {
1783                         case PROP_SHARP:
1784                                 strcpy(t->proptext, IFACE_("(Sharp)"));
1785                                 break;
1786                         case PROP_SMOOTH:
1787                                 strcpy(t->proptext, IFACE_("(Smooth)"));
1788                                 break;
1789                         case PROP_ROOT:
1790                                 strcpy(t->proptext, IFACE_("(Root)"));
1791                                 break;
1792                         case PROP_LIN:
1793                                 strcpy(t->proptext, IFACE_("(Linear)"));
1794                                 break;
1795                         case PROP_CONST:
1796                                 strcpy(t->proptext, IFACE_("(Constant)"));
1797                                 break;
1798                         case PROP_SPHERE:
1799                                 strcpy(t->proptext, IFACE_("(Sphere)"));
1800                                 break;
1801                         case PROP_RANDOM:
1802                                 strcpy(t->proptext, IFACE_("(Random)"));
1803                                 break;
1804                         default:
1805                                 t->proptext[0] = '\0';
1806                 }
1807         }
1808         else {
1809                 for (i = 0; i < t->total; i++, td++) {
1810                         td->factor = 1.0;
1811                 }
1812                 t->proptext[0] = '\0';
1813         }
1814 }