d4837084a5f0ddf7276d22fe13f15ea0fa205a36
[blender.git] / source / blender / editors / transform / transform.c
1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  *
16  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
17  * All rights reserved.
18  */
19
20 /** \file
21  * \ingroup edtransform
22  */
23
24 #include <stdlib.h>
25 #include <stdio.h>
26 #include <string.h>
27 #include <math.h>
28 #include <float.h>
29
30 #include "MEM_guardedalloc.h"
31
32 #include "DNA_anim_types.h"
33 #include "DNA_armature_types.h"
34 #include "DNA_constraint_types.h"
35 #include "DNA_mask_types.h"
36 #include "DNA_mesh_types.h"
37 #include "DNA_movieclip_types.h"
38 #include "DNA_scene_types.h" /* PET modes */
39 #include "DNA_workspace_types.h"
40 #include "DNA_gpencil_types.h"
41
42 #include "BLI_alloca.h"
43 #include "BLI_utildefines.h"
44 #include "BLI_math.h"
45 #include "BLI_rect.h"
46 #include "BLI_listbase.h"
47 #include "BLI_string.h"
48 #include "BLI_ghash.h"
49 #include "BLI_utildefines_stack.h"
50 #include "BLI_memarena.h"
51
52 #include "BKE_nla.h"
53 #include "BKE_editmesh.h"
54 #include "BKE_editmesh_bvh.h"
55 #include "BKE_context.h"
56 #include "BKE_constraint.h"
57 #include "BKE_particle.h"
58 #include "BKE_unit.h"
59 #include "BKE_scene.h"
60 #include "BKE_mask.h"
61 #include "BKE_mesh.h"
62 #include "BKE_report.h"
63 #include "BKE_workspace.h"
64
65 #include "DEG_depsgraph.h"
66
67 #include "GPU_immediate.h"
68 #include "GPU_immediate_util.h"
69 #include "GPU_matrix.h"
70 #include "GPU_state.h"
71
72 #include "ED_image.h"
73 #include "ED_keyframing.h"
74 #include "ED_screen.h"
75 #include "ED_space_api.h"
76 #include "ED_markers.h"
77 #include "ED_view3d.h"
78 #include "ED_mesh.h"
79 #include "ED_clip.h"
80 #include "ED_node.h"
81 #include "ED_gpencil.h"
82
83 #include "WM_types.h"
84 #include "WM_api.h"
85
86 #include "UI_view2d.h"
87 #include "UI_interface.h"
88 #include "UI_interface_icons.h"
89 #include "UI_resources.h"
90
91 #include "RNA_access.h"
92 #include "RNA_define.h"
93
94 #include "BLF_api.h"
95 #include "BLT_translation.h"
96
97 #include "transform.h"
98
99 /* Disabling, since when you type you know what you are doing,
100  * and being able to set it to zero is handy. */
101 // #define USE_NUM_NO_ZERO
102
103 static void drawTransformApply(const struct bContext *C, ARegion *ar, void *arg);
104 static void doEdgeSlide(TransInfo *t, float perc);
105 static void doVertSlide(TransInfo *t, float perc);
106
107 static void drawEdgeSlide(TransInfo *t);
108 static void drawVertSlide(TransInfo *t);
109 static void postInputRotation(TransInfo *t, float values[3]);
110
111 static void ElementRotation(
112     TransInfo *t, TransDataContainer *tc, TransData *td, float mat[3][3], const short around);
113 static void initSnapSpatial(TransInfo *t, float r_snap[3]);
114
115 static void storeCustomLNorValue(TransDataContainer *t, BMesh *bm);
116
117 /* Transform Callbacks */
118 static void initBend(TransInfo *t);
119 static eRedrawFlag handleEventBend(TransInfo *t, const struct wmEvent *event);
120 static void Bend(TransInfo *t, const int mval[2]);
121
122 static void initShear(TransInfo *t);
123 static eRedrawFlag handleEventShear(TransInfo *t, const struct wmEvent *event);
124 static void applyShear(TransInfo *t, const int mval[2]);
125
126 static void initResize(TransInfo *t);
127 static void applyResize(TransInfo *t, const int mval[2]);
128
129 static void initSkinResize(TransInfo *t);
130 static void applySkinResize(TransInfo *t, const int mval[2]);
131
132 static void initTranslation(TransInfo *t);
133 static void applyTranslation(TransInfo *t, const int mval[2]);
134
135 static void initToSphere(TransInfo *t);
136 static void applyToSphere(TransInfo *t, const int mval[2]);
137
138 static void initRotation(TransInfo *t);
139 static void applyRotation(TransInfo *t, const int mval[2]);
140
141 static void initNormalRotation(TransInfo *t);
142 static void applyNormalRotation(TransInfo *t, const int mval[2]);
143
144 static void initShrinkFatten(TransInfo *t);
145 static void applyShrinkFatten(TransInfo *t, const int mval[2]);
146
147 static void initTilt(TransInfo *t);
148 static void applyTilt(TransInfo *t, const int mval[2]);
149
150 static void initCurveShrinkFatten(TransInfo *t);
151 static void applyCurveShrinkFatten(TransInfo *t, const int mval[2]);
152
153 static void initMaskShrinkFatten(TransInfo *t);
154 static void applyMaskShrinkFatten(TransInfo *t, const int mval[2]);
155
156 static void initGPShrinkFatten(TransInfo *t);
157 static void applyGPShrinkFatten(TransInfo *t, const int mval[2]);
158
159 static void initTrackball(TransInfo *t);
160 static void applyTrackball(TransInfo *t, const int mval[2]);
161
162 static void initPushPull(TransInfo *t);
163 static void applyPushPull(TransInfo *t, const int mval[2]);
164
165 static void initBevelWeight(TransInfo *t);
166 static void applyBevelWeight(TransInfo *t, const int mval[2]);
167
168 static void initCrease(TransInfo *t);
169 static void applyCrease(TransInfo *t, const int mval[2]);
170
171 static void initBoneSize(TransInfo *t);
172 static void applyBoneSize(TransInfo *t, const int mval[2]);
173
174 static void initBoneEnvelope(TransInfo *t);
175 static void applyBoneEnvelope(TransInfo *t, const int mval[2]);
176
177 static void initBoneRoll(TransInfo *t);
178 static void applyBoneRoll(TransInfo *t, const int mval[2]);
179
180 static void initEdgeSlide_ex(
181     TransInfo *t, bool use_double_side, bool use_even, bool flipped, bool use_clamp);
182 static void initEdgeSlide(TransInfo *t);
183 static eRedrawFlag handleEventEdgeSlide(TransInfo *t, const struct wmEvent *event);
184 static void applyEdgeSlide(TransInfo *t, const int mval[2]);
185
186 static void initVertSlide_ex(TransInfo *t, bool use_even, bool flipped, bool use_clamp);
187 static void initVertSlide(TransInfo *t);
188 static eRedrawFlag handleEventVertSlide(TransInfo *t, const struct wmEvent *event);
189 static void applyVertSlide(TransInfo *t, const int mval[2]);
190
191 static void initTimeTranslate(TransInfo *t);
192 static void applyTimeTranslate(TransInfo *t, const int mval[2]);
193
194 static void initTimeSlide(TransInfo *t);
195 static void applyTimeSlide(TransInfo *t, const int mval[2]);
196
197 static void initTimeScale(TransInfo *t);
198 static void applyTimeScale(TransInfo *t, const int mval[2]);
199
200 static void initBakeTime(TransInfo *t);
201 static void applyBakeTime(TransInfo *t, const int mval[2]);
202
203 static void initMirror(TransInfo *t);
204 static void applyMirror(TransInfo *t, const int mval[2]);
205
206 static void initAlign(TransInfo *t);
207 static void applyAlign(TransInfo *t, const int mval[2]);
208
209 static void initSeqSlide(TransInfo *t);
210 static void applySeqSlide(TransInfo *t, const int mval[2]);
211
212 static void initGPOpacity(TransInfo *t);
213 static void applyGPOpacity(TransInfo *t, const int mval[2]);
214 /* end transform callbacks */
215
216 static bool transdata_check_local_center(TransInfo *t, short around)
217 {
218   return ((around == V3D_AROUND_LOCAL_ORIGINS) &&
219           ((t->flag & (T_OBJECT | T_POSE)) ||
220            /* implicit: (t->flag & T_EDIT) */
221            (ELEM(t->obedit_type, OB_MESH, OB_CURVE, OB_MBALL, OB_ARMATURE, OB_GPENCIL)) ||
222            (t->spacetype == SPACE_GRAPH) ||
223            (t->options & (CTX_MOVIECLIP | CTX_MASK | CTX_PAINT_CURVE))));
224 }
225
226 bool transdata_check_local_islands(TransInfo *t, short around)
227 {
228   return ((around == V3D_AROUND_LOCAL_ORIGINS) && ((ELEM(t->obedit_type, OB_MESH))));
229 }
230
231 /* ************************** SPACE DEPENDENT CODE **************************** */
232
233 void setTransformViewMatrices(TransInfo *t)
234 {
235   if (t->spacetype == SPACE_VIEW3D && t->ar && t->ar->regiontype == RGN_TYPE_WINDOW) {
236     RegionView3D *rv3d = t->ar->regiondata;
237
238     copy_m4_m4(t->viewmat, rv3d->viewmat);
239     copy_m4_m4(t->viewinv, rv3d->viewinv);
240     copy_m4_m4(t->persmat, rv3d->persmat);
241     copy_m4_m4(t->persinv, rv3d->persinv);
242     t->persp = rv3d->persp;
243   }
244   else {
245     unit_m4(t->viewmat);
246     unit_m4(t->viewinv);
247     unit_m4(t->persmat);
248     unit_m4(t->persinv);
249     t->persp = RV3D_ORTHO;
250   }
251
252   calculateCenter2D(t);
253   calculateCenterLocal(t, t->center_global);
254 }
255
256 void setTransformViewAspect(TransInfo *t, float r_aspect[3])
257 {
258   copy_v3_fl(r_aspect, 1.0f);
259
260   if (t->spacetype == SPACE_IMAGE) {
261     SpaceImage *sima = t->sa->spacedata.first;
262
263     if (t->options & CTX_MASK) {
264       ED_space_image_get_aspect(sima, &r_aspect[0], &r_aspect[1]);
265     }
266     else if (t->options & CTX_PAINT_CURVE) {
267       /* pass */
268     }
269     else {
270       ED_space_image_get_uv_aspect(sima, &r_aspect[0], &r_aspect[1]);
271     }
272   }
273   else if (t->spacetype == SPACE_CLIP) {
274     SpaceClip *sclip = t->sa->spacedata.first;
275
276     if (t->options & CTX_MOVIECLIP) {
277       ED_space_clip_get_aspect_dimension_aware(sclip, &r_aspect[0], &r_aspect[1]);
278     }
279     else {
280       ED_space_clip_get_aspect(sclip, &r_aspect[0], &r_aspect[1]);
281     }
282   }
283   else if (t->spacetype == SPACE_GRAPH) {
284     /* depemds on context of usage */
285   }
286 }
287
288 static void convertViewVec2D(View2D *v2d, float r_vec[3], int dx, int dy)
289 {
290   float divx = BLI_rcti_size_x(&v2d->mask);
291   float divy = BLI_rcti_size_y(&v2d->mask);
292
293   r_vec[0] = BLI_rctf_size_x(&v2d->cur) * dx / divx;
294   r_vec[1] = BLI_rctf_size_y(&v2d->cur) * dy / divy;
295   r_vec[2] = 0.0f;
296 }
297
298 static void convertViewVec2D_mask(View2D *v2d, float r_vec[3], int dx, int dy)
299 {
300   float divx = BLI_rcti_size_x(&v2d->mask);
301   float divy = BLI_rcti_size_y(&v2d->mask);
302
303   float mulx = BLI_rctf_size_x(&v2d->cur);
304   float muly = BLI_rctf_size_y(&v2d->cur);
305
306   /* difference with convertViewVec2D */
307   /* clamp w/h, mask only */
308   if (mulx / divx < muly / divy) {
309     divy = divx;
310     muly = mulx;
311   }
312   else {
313     divx = divy;
314     mulx = muly;
315   }
316   /* end difference */
317
318   r_vec[0] = mulx * dx / divx;
319   r_vec[1] = muly * dy / divy;
320   r_vec[2] = 0.0f;
321 }
322
323 void convertViewVec(TransInfo *t, float r_vec[3], double dx, double dy)
324 {
325   if ((t->spacetype == SPACE_VIEW3D) && (t->ar->regiontype == RGN_TYPE_WINDOW)) {
326     if (t->options & CTX_PAINT_CURVE) {
327       r_vec[0] = dx;
328       r_vec[1] = dy;
329     }
330     else {
331       const float mval_f[2] = {(float)dx, (float)dy};
332       ED_view3d_win_to_delta(t->ar, mval_f, r_vec, t->zfac);
333     }
334   }
335   else if (t->spacetype == SPACE_IMAGE) {
336     if (t->options & CTX_MASK) {
337       convertViewVec2D_mask(t->view, r_vec, dx, dy);
338     }
339     else if (t->options & CTX_PAINT_CURVE) {
340       r_vec[0] = dx;
341       r_vec[1] = dy;
342     }
343     else {
344       convertViewVec2D(t->view, r_vec, dx, dy);
345     }
346
347     r_vec[0] *= t->aspect[0];
348     r_vec[1] *= t->aspect[1];
349   }
350   else if (ELEM(t->spacetype, SPACE_GRAPH, SPACE_NLA)) {
351     convertViewVec2D(t->view, r_vec, dx, dy);
352   }
353   else if (ELEM(t->spacetype, SPACE_NODE, SPACE_SEQ)) {
354     convertViewVec2D(&t->ar->v2d, r_vec, dx, dy);
355   }
356   else if (t->spacetype == SPACE_CLIP) {
357     if (t->options & CTX_MASK) {
358       convertViewVec2D_mask(t->view, r_vec, dx, dy);
359     }
360     else {
361       convertViewVec2D(t->view, r_vec, dx, dy);
362     }
363
364     r_vec[0] *= t->aspect[0];
365     r_vec[1] *= t->aspect[1];
366   }
367   else {
368     printf("%s: called in an invalid context\n", __func__);
369     zero_v3(r_vec);
370   }
371 }
372
373 void projectIntViewEx(TransInfo *t, const float vec[3], int adr[2], const eV3DProjTest flag)
374 {
375   if (t->spacetype == SPACE_VIEW3D) {
376     if (t->ar->regiontype == RGN_TYPE_WINDOW) {
377       if (ED_view3d_project_int_global(t->ar, vec, adr, flag) != V3D_PROJ_RET_OK) {
378         /* this is what was done in 2.64, perhaps we can be smarter? */
379         adr[0] = (int)2140000000.0f;
380         adr[1] = (int)2140000000.0f;
381       }
382     }
383   }
384   else if (t->spacetype == SPACE_IMAGE) {
385     SpaceImage *sima = t->sa->spacedata.first;
386
387     if (t->options & CTX_MASK) {
388       float v[2];
389
390       v[0] = vec[0] / t->aspect[0];
391       v[1] = vec[1] / t->aspect[1];
392
393       BKE_mask_coord_to_image(sima->image, &sima->iuser, v, v);
394
395       ED_image_point_pos__reverse(sima, t->ar, v, v);
396
397       adr[0] = v[0];
398       adr[1] = v[1];
399     }
400     else if (t->options & CTX_PAINT_CURVE) {
401       adr[0] = vec[0];
402       adr[1] = vec[1];
403     }
404     else {
405       float v[2];
406
407       v[0] = vec[0] / t->aspect[0];
408       v[1] = vec[1] / t->aspect[1];
409
410       UI_view2d_view_to_region(t->view, v[0], v[1], &adr[0], &adr[1]);
411     }
412   }
413   else if (t->spacetype == SPACE_ACTION) {
414     int out[2] = {0, 0};
415 #if 0
416     SpaceAction *sact = t->sa->spacedata.first;
417
418     if (sact->flag & SACTION_DRAWTIME) {
419       //vec[0] = vec[0]/((t->scene->r.frs_sec / t->scene->r.frs_sec_base));
420       /* same as below */
421       UI_view2d_view_to_region((View2D *)t->view, vec[0], vec[1], &out[0], &out[1]);
422     }
423     else
424 #endif
425     {
426       UI_view2d_view_to_region((View2D *)t->view, vec[0], vec[1], &out[0], &out[1]);
427     }
428
429     adr[0] = out[0];
430     adr[1] = out[1];
431   }
432   else if (ELEM(t->spacetype, SPACE_GRAPH, SPACE_NLA)) {
433     int out[2] = {0, 0};
434
435     UI_view2d_view_to_region((View2D *)t->view, vec[0], vec[1], &out[0], &out[1]);
436     adr[0] = out[0];
437     adr[1] = out[1];
438   }
439   else if (t->spacetype == SPACE_SEQ) { /* XXX not tested yet, but should work */
440     int out[2] = {0, 0};
441
442     UI_view2d_view_to_region((View2D *)t->view, vec[0], vec[1], &out[0], &out[1]);
443     adr[0] = out[0];
444     adr[1] = out[1];
445   }
446   else if (t->spacetype == SPACE_CLIP) {
447     SpaceClip *sc = t->sa->spacedata.first;
448
449     if (t->options & CTX_MASK) {
450       MovieClip *clip = ED_space_clip_get_clip(sc);
451
452       if (clip) {
453         float v[2];
454
455         v[0] = vec[0] / t->aspect[0];
456         v[1] = vec[1] / t->aspect[1];
457
458         BKE_mask_coord_to_movieclip(sc->clip, &sc->user, v, v);
459
460         ED_clip_point_stable_pos__reverse(sc, t->ar, v, v);
461
462         adr[0] = v[0];
463         adr[1] = v[1];
464       }
465       else {
466         adr[0] = 0;
467         adr[1] = 0;
468       }
469     }
470     else if (t->options & CTX_MOVIECLIP) {
471       float v[2];
472
473       v[0] = vec[0] / t->aspect[0];
474       v[1] = vec[1] / t->aspect[1];
475
476       UI_view2d_view_to_region(t->view, v[0], v[1], &adr[0], &adr[1]);
477     }
478     else {
479       BLI_assert(0);
480     }
481   }
482   else if (t->spacetype == SPACE_NODE) {
483     UI_view2d_view_to_region((View2D *)t->view, vec[0], vec[1], &adr[0], &adr[1]);
484   }
485 }
486 void projectIntView(TransInfo *t, const float vec[3], int adr[2])
487 {
488   projectIntViewEx(t, vec, adr, V3D_PROJ_TEST_NOP);
489 }
490
491 void projectFloatViewEx(TransInfo *t, const float vec[3], float adr[2], const eV3DProjTest flag)
492 {
493   switch (t->spacetype) {
494     case SPACE_VIEW3D: {
495       if (t->options & CTX_PAINT_CURVE) {
496         adr[0] = vec[0];
497         adr[1] = vec[1];
498       }
499       else if (t->ar->regiontype == RGN_TYPE_WINDOW) {
500         /* allow points behind the view [#33643] */
501         if (ED_view3d_project_float_global(t->ar, vec, adr, flag) != V3D_PROJ_RET_OK) {
502           /* XXX, 2.64 and prior did this, weak! */
503           adr[0] = t->ar->winx / 2.0f;
504           adr[1] = t->ar->winy / 2.0f;
505         }
506         return;
507       }
508       break;
509     }
510     default: {
511       int a[2] = {0, 0};
512       projectIntView(t, vec, a);
513       adr[0] = a[0];
514       adr[1] = a[1];
515       break;
516     }
517   }
518 }
519 void projectFloatView(TransInfo *t, const float vec[3], float adr[2])
520 {
521   projectFloatViewEx(t, vec, adr, V3D_PROJ_TEST_NOP);
522 }
523
524 void applyAspectRatio(TransInfo *t, float vec[2])
525 {
526   if ((t->spacetype == SPACE_IMAGE) && (t->mode == TFM_TRANSLATION) &&
527       !(t->options & CTX_PAINT_CURVE)) {
528     SpaceImage *sima = t->sa->spacedata.first;
529
530     if ((sima->flag & SI_COORDFLOATS) == 0) {
531       int width, height;
532       ED_space_image_get_size(sima, &width, &height);
533
534       vec[0] *= width;
535       vec[1] *= height;
536     }
537
538     vec[0] /= t->aspect[0];
539     vec[1] /= t->aspect[1];
540   }
541   else if ((t->spacetype == SPACE_CLIP) && (t->mode == TFM_TRANSLATION)) {
542     if (t->options & (CTX_MOVIECLIP | CTX_MASK)) {
543       vec[0] /= t->aspect[0];
544       vec[1] /= t->aspect[1];
545     }
546   }
547 }
548
549 void removeAspectRatio(TransInfo *t, float vec[2])
550 {
551   if ((t->spacetype == SPACE_IMAGE) && (t->mode == TFM_TRANSLATION)) {
552     SpaceImage *sima = t->sa->spacedata.first;
553
554     if ((sima->flag & SI_COORDFLOATS) == 0) {
555       int width, height;
556       ED_space_image_get_size(sima, &width, &height);
557
558       vec[0] /= width;
559       vec[1] /= height;
560     }
561
562     vec[0] *= t->aspect[0];
563     vec[1] *= t->aspect[1];
564   }
565   else if ((t->spacetype == SPACE_CLIP) && (t->mode == TFM_TRANSLATION)) {
566     if (t->options & (CTX_MOVIECLIP | CTX_MASK)) {
567       vec[0] *= t->aspect[0];
568       vec[1] *= t->aspect[1];
569     }
570   }
571 }
572
573 static void viewRedrawForce(const bContext *C, TransInfo *t)
574 {
575   if (t->options & CTX_GPENCIL_STROKES) {
576     bGPdata *gpd = ED_gpencil_data_get_active(C);
577     if (gpd) {
578       DEG_id_tag_update(&gpd->id, ID_RECALC_GEOMETRY);
579     }
580     WM_event_add_notifier(C, NC_GPENCIL | NA_EDITED, NULL);
581   }
582   else if (t->spacetype == SPACE_VIEW3D) {
583     if (t->options & CTX_PAINT_CURVE) {
584       wmWindow *window = CTX_wm_window(C);
585       WM_paint_cursor_tag_redraw(window, t->ar);
586     }
587     else {
588       /* Do we need more refined tags? */
589       if (t->flag & T_POSE)
590         WM_event_add_notifier(C, NC_OBJECT | ND_POSE, NULL);
591       else
592         WM_event_add_notifier(C, NC_OBJECT | ND_TRANSFORM, NULL);
593
594       /* For real-time animation record - send notifiers recognized by animation editors */
595       // XXX: is this notifier a lame duck?
596       if ((t->animtimer) && IS_AUTOKEY_ON(t->scene))
597         WM_event_add_notifier(C, NC_OBJECT | ND_KEYS, NULL);
598     }
599   }
600   else if (t->spacetype == SPACE_ACTION) {
601     //SpaceAction *saction = (SpaceAction *)t->sa->spacedata.first;
602     WM_event_add_notifier(C, NC_ANIMATION | ND_KEYFRAME | NA_EDITED, NULL);
603   }
604   else if (t->spacetype == SPACE_GRAPH) {
605     //SpaceGraph *sipo = (SpaceGraph *)t->sa->spacedata.first;
606     WM_event_add_notifier(C, NC_ANIMATION | ND_KEYFRAME | NA_EDITED, NULL);
607   }
608   else if (t->spacetype == SPACE_NLA) {
609     WM_event_add_notifier(C, NC_ANIMATION | ND_NLA | NA_EDITED, NULL);
610   }
611   else if (t->spacetype == SPACE_NODE) {
612     //ED_area_tag_redraw(t->sa);
613     WM_event_add_notifier(C, NC_SPACE | ND_SPACE_NODE_VIEW, NULL);
614   }
615   else if (t->spacetype == SPACE_SEQ) {
616     WM_event_add_notifier(C, NC_SCENE | ND_SEQUENCER, NULL);
617   }
618   else if (t->spacetype == SPACE_IMAGE) {
619     if (t->options & CTX_MASK) {
620       Mask *mask = CTX_data_edit_mask(C);
621
622       WM_event_add_notifier(C, NC_MASK | NA_EDITED, mask);
623     }
624     else if (t->options & CTX_PAINT_CURVE) {
625       wmWindow *window = CTX_wm_window(C);
626       WM_paint_cursor_tag_redraw(window, t->ar);
627     }
628     else if (t->flag & T_CURSOR) {
629       ED_area_tag_redraw(t->sa);
630     }
631     else {
632       // XXX how to deal with lock?
633       SpaceImage *sima = (SpaceImage *)t->sa->spacedata.first;
634       if (sima->lock) {
635         WM_event_add_notifier(C, NC_GEOM | ND_DATA, OBEDIT_FROM_VIEW_LAYER(t->view_layer)->data);
636       }
637       else {
638         ED_area_tag_redraw(t->sa);
639       }
640     }
641   }
642   else if (t->spacetype == SPACE_CLIP) {
643     SpaceClip *sc = (SpaceClip *)t->sa->spacedata.first;
644
645     if (ED_space_clip_check_show_trackedit(sc)) {
646       MovieClip *clip = ED_space_clip_get_clip(sc);
647
648       /* objects could be parented to tracking data, so send this for viewport refresh */
649       WM_event_add_notifier(C, NC_OBJECT | ND_TRANSFORM, NULL);
650
651       WM_event_add_notifier(C, NC_MOVIECLIP | NA_EDITED, clip);
652     }
653     else if (ED_space_clip_check_show_maskedit(sc)) {
654       Mask *mask = CTX_data_edit_mask(C);
655
656       WM_event_add_notifier(C, NC_MASK | NA_EDITED, mask);
657     }
658   }
659 }
660
661 static void viewRedrawPost(bContext *C, TransInfo *t)
662 {
663   ED_area_status_text(t->sa, NULL);
664
665   if (t->spacetype == SPACE_VIEW3D) {
666     /* if autokeying is enabled, send notifiers that keyframes were added */
667     if (IS_AUTOKEY_ON(t->scene))
668       WM_main_add_notifier(NC_ANIMATION | ND_KEYFRAME | NA_EDITED, NULL);
669
670     /* redraw UV editor */
671     if (ELEM(t->mode, TFM_VERT_SLIDE, TFM_EDGE_SLIDE) &&
672         (t->settings->uvcalc_flag & UVCALC_TRANSFORM_CORRECT)) {
673       WM_event_add_notifier(C, NC_GEOM | ND_DATA, NULL);
674     }
675
676     /* XXX temp, first hack to get auto-render in compositor work (ton) */
677     WM_event_add_notifier(C, NC_SCENE | ND_TRANSFORM_DONE, CTX_data_scene(C));
678   }
679
680 #if 0  // TRANSFORM_FIX_ME
681   if (t->spacetype == SPACE_VIEW3D) {
682     allqueue(REDRAWBUTSOBJECT, 0);
683     allqueue(REDRAWVIEW3D, 0);
684   }
685   else if (t->spacetype == SPACE_IMAGE) {
686     allqueue(REDRAWIMAGE, 0);
687     allqueue(REDRAWVIEW3D, 0);
688   }
689   else if (ELEM(t->spacetype, SPACE_ACTION, SPACE_NLA, SPACE_GRAPH)) {
690     allqueue(REDRAWVIEW3D, 0);
691     allqueue(REDRAWACTION, 0);
692     allqueue(REDRAWNLA, 0);
693     allqueue(REDRAWIPO, 0);
694     allqueue(REDRAWTIME, 0);
695     allqueue(REDRAWBUTSOBJECT, 0);
696   }
697
698   scrarea_queue_headredraw(curarea);
699 #endif
700 }
701
702 /* ************************** TRANSFORMATIONS **************************** */
703
704 static void view_editmove(unsigned short UNUSED(event))
705 {
706 #if 0  // TRANSFORM_FIX_ME
707   int refresh = 0;
708   /* Regular:   Zoom in */
709   /* Shift:     Scroll up */
710   /* Ctrl:      Scroll right */
711   /* Alt-Shift: Rotate up */
712   /* Alt-Ctrl:  Rotate right */
713
714   /* only work in 3D window for now
715    * In the end, will have to send to event to a 2D window handler instead
716    */
717   if (Trans.flag & T_2D_EDIT)
718     return;
719
720   switch (event) {
721     case WHEELUPMOUSE:
722       if (G.qual & LR_SHIFTKEY) {
723         if (G.qual & LR_ALTKEY) {
724           G.qual &= ~LR_SHIFTKEY;
725           persptoetsen(PAD2);
726           G.qual |= LR_SHIFTKEY;
727         }
728         else {
729           persptoetsen(PAD2);
730         }
731       }
732       else if (G.qual & LR_CTRLKEY) {
733         if (G.qual & LR_ALTKEY) {
734           G.qual &= ~LR_CTRLKEY;
735           persptoetsen(PAD4);
736           G.qual |= LR_CTRLKEY;
737         }
738         else {
739           persptoetsen(PAD4);
740         }
741       }
742       else if (U.uiflag & USER_WHEELZOOMDIR)
743         persptoetsen(PADMINUS);
744       else
745         persptoetsen(PADPLUSKEY);
746
747       refresh = 1;
748       break;
749     case WHEELDOWNMOUSE:
750       if (G.qual & LR_SHIFTKEY) {
751         if (G.qual & LR_ALTKEY) {
752           G.qual &= ~LR_SHIFTKEY;
753           persptoetsen(PAD8);
754           G.qual |= LR_SHIFTKEY;
755         }
756         else {
757           persptoetsen(PAD8);
758         }
759       }
760       else if (G.qual & LR_CTRLKEY) {
761         if (G.qual & LR_ALTKEY) {
762           G.qual &= ~LR_CTRLKEY;
763           persptoetsen(PAD6);
764           G.qual |= LR_CTRLKEY;
765         }
766         else {
767           persptoetsen(PAD6);
768         }
769       }
770       else if (U.uiflag & USER_WHEELZOOMDIR)
771         persptoetsen(PADPLUSKEY);
772       else
773         persptoetsen(PADMINUS);
774
775       refresh = 1;
776       break;
777   }
778
779   if (refresh)
780     setTransformViewMatrices(&Trans);
781 #endif
782 }
783
784 /* ************************************************* */
785
786 /* NOTE: these defines are saved in keymap files, do not change values but just add new ones */
787 enum {
788   TFM_MODAL_CANCEL = 1,
789   TFM_MODAL_CONFIRM = 2,
790   TFM_MODAL_TRANSLATE = 3,
791   TFM_MODAL_ROTATE = 4,
792   TFM_MODAL_RESIZE = 5,
793   TFM_MODAL_SNAP_INV_ON = 6,
794   TFM_MODAL_SNAP_INV_OFF = 7,
795   TFM_MODAL_SNAP_TOGGLE = 8,
796   TFM_MODAL_AXIS_X = 9,
797   TFM_MODAL_AXIS_Y = 10,
798   TFM_MODAL_AXIS_Z = 11,
799   TFM_MODAL_PLANE_X = 12,
800   TFM_MODAL_PLANE_Y = 13,
801   TFM_MODAL_PLANE_Z = 14,
802   TFM_MODAL_CONS_OFF = 15,
803   TFM_MODAL_ADD_SNAP = 16,
804   TFM_MODAL_REMOVE_SNAP = 17,
805
806   /* 18 and 19 used by numinput, defined in transform.h */
807
808   TFM_MODAL_PROPSIZE_UP = 20,
809   TFM_MODAL_PROPSIZE_DOWN = 21,
810   TFM_MODAL_AUTOIK_LEN_INC = 22,
811   TFM_MODAL_AUTOIK_LEN_DEC = 23,
812
813   TFM_MODAL_EDGESLIDE_UP = 24,
814   TFM_MODAL_EDGESLIDE_DOWN = 25,
815
816   /* for analog input, like trackpad */
817   TFM_MODAL_PROPSIZE = 26,
818   /* node editor insert offset (aka auto-offset) direction toggle */
819   TFM_MODAL_INSERTOFS_TOGGLE_DIR = 27,
820 };
821
822 static bool transform_modal_item_poll(const wmOperator *op, int value)
823 {
824   const TransInfo *t = op->customdata;
825   switch (value) {
826     case TFM_MODAL_CANCEL: {
827       if ((t->flag & T_RELEASE_CONFIRM) && ISMOUSE(t->launch_event)) {
828         return false;
829       }
830       break;
831     }
832     case TFM_MODAL_PROPSIZE:
833     case TFM_MODAL_PROPSIZE_UP:
834     case TFM_MODAL_PROPSIZE_DOWN: {
835       if ((t->flag & T_PROP_EDIT) == 0) {
836         return false;
837       }
838       break;
839     }
840     case TFM_MODAL_ADD_SNAP:
841     case TFM_MODAL_REMOVE_SNAP: {
842       if (t->spacetype != SPACE_VIEW3D) {
843         return false;
844       }
845       else if (t->tsnap.mode & (SCE_SNAP_MODE_INCREMENT | SCE_SNAP_MODE_GRID)) {
846         return false;
847       }
848       else if (!validSnap(t)) {
849         return false;
850       }
851       break;
852     }
853     case TFM_MODAL_AXIS_X:
854     case TFM_MODAL_AXIS_Y:
855     case TFM_MODAL_AXIS_Z:
856     case TFM_MODAL_PLANE_X:
857     case TFM_MODAL_PLANE_Y:
858     case TFM_MODAL_PLANE_Z: {
859       if (t->flag & T_NO_CONSTRAINT) {
860         return false;
861       }
862       if (!ELEM(value, TFM_MODAL_AXIS_X, TFM_MODAL_AXIS_Y)) {
863         if (t->flag & T_2D_EDIT) {
864           return false;
865         }
866       }
867       break;
868     }
869     case TFM_MODAL_CONS_OFF: {
870       if ((t->con.mode & CON_APPLY) == 0) {
871         return false;
872       }
873       break;
874     }
875     case TFM_MODAL_EDGESLIDE_UP:
876     case TFM_MODAL_EDGESLIDE_DOWN: {
877       if (t->mode != TFM_EDGE_SLIDE) {
878         return false;
879       }
880       break;
881     }
882     case TFM_MODAL_INSERTOFS_TOGGLE_DIR: {
883       if (t->spacetype != SPACE_NODE) {
884         return false;
885       }
886       break;
887     }
888     case TFM_MODAL_AUTOIK_LEN_INC:
889     case TFM_MODAL_AUTOIK_LEN_DEC: {
890       if ((t->flag & T_AUTOIK) == 0) {
891         return false;
892       }
893       break;
894     }
895   }
896   return true;
897 }
898
899 /* called in transform_ops.c, on each regeneration of keymaps */
900 wmKeyMap *transform_modal_keymap(wmKeyConfig *keyconf)
901 {
902   static const EnumPropertyItem modal_items[] = {
903       {TFM_MODAL_CONFIRM, "CONFIRM", 0, "Confirm", ""},
904       {TFM_MODAL_CANCEL, "CANCEL", 0, "Cancel", ""},
905       {TFM_MODAL_AXIS_X, "AXIS_X", 0, "X axis", ""},
906       {TFM_MODAL_AXIS_Y, "AXIS_Y", 0, "Y axis", ""},
907       {TFM_MODAL_AXIS_Z, "AXIS_Z", 0, "Z axis", ""},
908       {TFM_MODAL_PLANE_X, "PLANE_X", 0, "X plane", ""},
909       {TFM_MODAL_PLANE_Y, "PLANE_Y", 0, "Y plane", ""},
910       {TFM_MODAL_PLANE_Z, "PLANE_Z", 0, "Z plane", ""},
911       {TFM_MODAL_CONS_OFF, "CONS_OFF", 0, "Clear Constraints", ""},
912       {TFM_MODAL_SNAP_INV_ON, "SNAP_INV_ON", 0, "Snap Invert", ""},
913       {TFM_MODAL_SNAP_INV_OFF, "SNAP_INV_OFF", 0, "Snap Invert (Off)", ""},
914       {TFM_MODAL_SNAP_TOGGLE, "SNAP_TOGGLE", 0, "Snap Toggle", ""},
915       {TFM_MODAL_ADD_SNAP, "ADD_SNAP", 0, "Add Snap Point", ""},
916       {TFM_MODAL_REMOVE_SNAP, "REMOVE_SNAP", 0, "Remove Last Snap Point", ""},
917       {NUM_MODAL_INCREMENT_UP, "INCREMENT_UP", 0, "Numinput Increment Up", ""},
918       {NUM_MODAL_INCREMENT_DOWN, "INCREMENT_DOWN", 0, "Numinput Increment Down", ""},
919       {TFM_MODAL_PROPSIZE_UP, "PROPORTIONAL_SIZE_UP", 0, "Increase Proportional Influence", ""},
920       {TFM_MODAL_PROPSIZE_DOWN,
921        "PROPORTIONAL_SIZE_DOWN",
922        0,
923        "Decrease Proportional Influence",
924        ""},
925       {TFM_MODAL_AUTOIK_LEN_INC, "AUTOIK_CHAIN_LEN_UP", 0, "Increase Max AutoIK Chain Length", ""},
926       {TFM_MODAL_AUTOIK_LEN_DEC,
927        "AUTOIK_CHAIN_LEN_DOWN",
928        0,
929        "Decrease Max AutoIK Chain Length",
930        ""},
931       {TFM_MODAL_EDGESLIDE_UP, "EDGESLIDE_EDGE_NEXT", 0, "Select next Edge Slide Edge", ""},
932       {TFM_MODAL_EDGESLIDE_DOWN, "EDGESLIDE_PREV_NEXT", 0, "Select previous Edge Slide Edge", ""},
933       {TFM_MODAL_PROPSIZE, "PROPORTIONAL_SIZE", 0, "Adjust Proportional Influence", ""},
934       {TFM_MODAL_INSERTOFS_TOGGLE_DIR,
935        "INSERTOFS_TOGGLE_DIR",
936        0,
937        "Toggle Direction for Node Auto-offset",
938        ""},
939       {TFM_MODAL_TRANSLATE, "TRANSLATE", 0, "Move", ""},
940       {TFM_MODAL_ROTATE, "ROTATE", 0, "Rotate", ""},
941       {TFM_MODAL_RESIZE, "RESIZE", 0, "Resize", ""},
942       {0, NULL, 0, NULL, NULL},
943   };
944
945   wmKeyMap *keymap = WM_modalkeymap_get(keyconf, "Transform Modal Map");
946
947   keymap = WM_modalkeymap_add(keyconf, "Transform Modal Map", modal_items);
948   keymap->poll_modal_item = transform_modal_item_poll;
949
950   return keymap;
951 }
952
953 static void transform_event_xyz_constraint(TransInfo *t, short key_type, char cmode, bool is_plane)
954 {
955   if (!(t->flag & T_NO_CONSTRAINT)) {
956     int constraint_axis, constraint_plane;
957     const bool edit_2d = (t->flag & T_2D_EDIT) != 0;
958     const char *msg1 = "", *msg2 = "", *msg3 = "";
959     char axis;
960
961     /* Initialize */
962     switch (key_type) {
963       case XKEY:
964         msg1 = IFACE_("along X");
965         msg2 = IFACE_("along %s X");
966         msg3 = IFACE_("locking %s X");
967         axis = 'X';
968         constraint_axis = CON_AXIS0;
969         break;
970       case YKEY:
971         msg1 = IFACE_("along Y");
972         msg2 = IFACE_("along %s Y");
973         msg3 = IFACE_("locking %s Y");
974         axis = 'Y';
975         constraint_axis = CON_AXIS1;
976         break;
977       case ZKEY:
978         msg1 = IFACE_("along Z");
979         msg2 = IFACE_("along %s Z");
980         msg3 = IFACE_("locking %s Z");
981         axis = 'Z';
982         constraint_axis = CON_AXIS2;
983         break;
984       default:
985         /* Invalid key */
986         return;
987     }
988     constraint_plane = ((CON_AXIS0 | CON_AXIS1 | CON_AXIS2) & (~constraint_axis));
989
990     if (edit_2d && (key_type != ZKEY)) {
991       if (cmode == axis) {
992         stopConstraint(t);
993       }
994       else {
995         setUserConstraint(t, V3D_ORIENT_GLOBAL, constraint_axis, msg1);
996       }
997     }
998     else if (!edit_2d) {
999       if (cmode != axis) {
1000         /* First press, constraint to an axis. */
1001         t->orientation.index = 0;
1002         const short *orientation_ptr = t->orientation.types[t->orientation.index];
1003         const short orientation = orientation_ptr ? *orientation_ptr : V3D_ORIENT_GLOBAL;
1004         if (is_plane == false) {
1005           setUserConstraint(t, orientation, constraint_axis, msg2);
1006         }
1007         else {
1008           setUserConstraint(t, orientation, constraint_plane, msg3);
1009         }
1010       }
1011       else {
1012         /* Successive presses on existing axis, cycle orientation modes. */
1013         t->orientation.index = (t->orientation.index + 1) % ARRAY_SIZE(t->orientation.types);
1014
1015         if (t->orientation.index == 0) {
1016           stopConstraint(t);
1017         }
1018         else {
1019           const short *orientation_ptr = t->orientation.types[t->orientation.index];
1020           const short orientation = orientation_ptr ? *orientation_ptr : V3D_ORIENT_GLOBAL;
1021           if (is_plane == false) {
1022             setUserConstraint(t, orientation, constraint_axis, msg2);
1023           }
1024           else {
1025             setUserConstraint(t, orientation, constraint_plane, msg3);
1026           }
1027         }
1028       }
1029     }
1030     t->redraw |= TREDRAW_HARD;
1031   }
1032 }
1033
1034 int transformEvent(TransInfo *t, const wmEvent *event)
1035 {
1036   char cmode = constraintModeToChar(t);
1037   bool handled = false;
1038   const int modifiers_prev = t->modifiers;
1039
1040   t->redraw |= handleMouseInput(t, &t->mouse, event);
1041
1042   /* Handle modal numinput events first, if already activated. */
1043   if (((event->val == KM_PRESS) || (event->type == EVT_MODAL_MAP)) && hasNumInput(&t->num) &&
1044       handleNumInput(t->context, &(t->num), event)) {
1045     t->redraw |= TREDRAW_HARD;
1046     handled = true;
1047   }
1048   else if (event->type == MOUSEMOVE) {
1049     if (t->modifiers & MOD_CONSTRAINT_SELECT)
1050       t->con.mode |= CON_SELECT;
1051
1052     copy_v2_v2_int(t->mval, event->mval);
1053
1054     /* Use this for soft redraw. Might cause flicker in object mode */
1055     // t->redraw |= TREDRAW_SOFT;
1056     t->redraw |= TREDRAW_HARD;
1057
1058     if (t->state == TRANS_STARTING) {
1059       t->state = TRANS_RUNNING;
1060     }
1061
1062     applyMouseInput(t, &t->mouse, t->mval, t->values);
1063
1064     // Snapping mouse move events
1065     t->redraw |= handleSnapping(t, event);
1066     handled = true;
1067   }
1068   /* handle modal keymap first */
1069   else if (event->type == EVT_MODAL_MAP) {
1070     switch (event->val) {
1071       case TFM_MODAL_CANCEL:
1072         t->state = TRANS_CANCEL;
1073         handled = true;
1074         break;
1075       case TFM_MODAL_CONFIRM:
1076         t->state = TRANS_CONFIRM;
1077         handled = true;
1078         break;
1079       case TFM_MODAL_TRANSLATE:
1080         /* only switch when... */
1081         if (ELEM(t->mode,
1082                  TFM_ROTATION,
1083                  TFM_RESIZE,
1084                  TFM_TRACKBALL,
1085                  TFM_EDGE_SLIDE,
1086                  TFM_VERT_SLIDE)) {
1087           restoreTransObjects(t);
1088           resetTransModal(t);
1089           resetTransRestrictions(t);
1090           initTranslation(t);
1091           initSnapping(t, NULL);  // need to reinit after mode change
1092           t->redraw |= TREDRAW_HARD;
1093           WM_event_add_mousemove(t->context);
1094           handled = true;
1095         }
1096         else if (t->mode == TFM_SEQ_SLIDE) {
1097           t->flag ^= T_ALT_TRANSFORM;
1098           t->redraw |= TREDRAW_HARD;
1099           handled = true;
1100         }
1101         else {
1102           if (t->obedit_type == OB_MESH) {
1103             if ((t->mode == TFM_TRANSLATION) && (t->spacetype == SPACE_VIEW3D)) {
1104               restoreTransObjects(t);
1105               resetTransModal(t);
1106               resetTransRestrictions(t);
1107
1108               /* first try edge slide */
1109               initEdgeSlide(t);
1110               /* if that fails, do vertex slide */
1111               if (t->state == TRANS_CANCEL) {
1112                 resetTransModal(t);
1113                 t->state = TRANS_STARTING;
1114                 initVertSlide(t);
1115               }
1116               /* vert slide can fail on unconnected vertices (rare but possible) */
1117               if (t->state == TRANS_CANCEL) {
1118                 resetTransModal(t);
1119                 t->mode = TFM_TRANSLATION;
1120                 t->state = TRANS_STARTING;
1121                 restoreTransObjects(t);
1122                 resetTransRestrictions(t);
1123                 initTranslation(t);
1124               }
1125               initSnapping(t, NULL);  // need to reinit after mode change
1126               t->redraw |= TREDRAW_HARD;
1127               handled = true;
1128               WM_event_add_mousemove(t->context);
1129             }
1130           }
1131           else if (t->options & (CTX_MOVIECLIP | CTX_MASK)) {
1132             if (t->mode == TFM_TRANSLATION) {
1133               restoreTransObjects(t);
1134
1135               t->flag ^= T_ALT_TRANSFORM;
1136               t->redraw |= TREDRAW_HARD;
1137               handled = true;
1138             }
1139           }
1140         }
1141         break;
1142       case TFM_MODAL_ROTATE:
1143         /* only switch when... */
1144         if (!(t->options & CTX_TEXTURE) && !(t->options & (CTX_MOVIECLIP | CTX_MASK))) {
1145           if (ELEM(t->mode,
1146                    TFM_ROTATION,
1147                    TFM_RESIZE,
1148                    TFM_TRACKBALL,
1149                    TFM_TRANSLATION,
1150                    TFM_EDGE_SLIDE,
1151                    TFM_VERT_SLIDE)) {
1152             restoreTransObjects(t);
1153             resetTransModal(t);
1154             resetTransRestrictions(t);
1155
1156             if (t->mode == TFM_ROTATION) {
1157               initTrackball(t);
1158             }
1159             else {
1160               initRotation(t);
1161             }
1162             initSnapping(t, NULL);  // need to reinit after mode change
1163             t->redraw |= TREDRAW_HARD;
1164             handled = true;
1165           }
1166         }
1167         break;
1168       case TFM_MODAL_RESIZE:
1169         /* only switch when... */
1170         if (ELEM(t->mode,
1171                  TFM_ROTATION,
1172                  TFM_TRANSLATION,
1173                  TFM_TRACKBALL,
1174                  TFM_EDGE_SLIDE,
1175                  TFM_VERT_SLIDE)) {
1176
1177           /* Scale isn't normally very useful after extrude along normals, see T39756 */
1178           if ((t->con.mode & CON_APPLY) && (t->con.orientation == V3D_ORIENT_NORMAL)) {
1179             stopConstraint(t);
1180           }
1181
1182           restoreTransObjects(t);
1183           resetTransModal(t);
1184           resetTransRestrictions(t);
1185           initResize(t);
1186           initSnapping(t, NULL);  // need to reinit after mode change
1187           t->redraw |= TREDRAW_HARD;
1188           handled = true;
1189         }
1190         else if (t->mode == TFM_SHRINKFATTEN) {
1191           t->flag ^= T_ALT_TRANSFORM;
1192           t->redraw |= TREDRAW_HARD;
1193           handled = true;
1194         }
1195         else if (t->mode == TFM_RESIZE) {
1196           if (t->options & CTX_MOVIECLIP) {
1197             restoreTransObjects(t);
1198
1199             t->flag ^= T_ALT_TRANSFORM;
1200             t->redraw |= TREDRAW_HARD;
1201             handled = true;
1202           }
1203         }
1204         break;
1205
1206       case TFM_MODAL_SNAP_INV_ON:
1207         t->modifiers |= MOD_SNAP_INVERT;
1208         t->redraw |= TREDRAW_HARD;
1209         handled = true;
1210         break;
1211       case TFM_MODAL_SNAP_INV_OFF:
1212         t->modifiers &= ~MOD_SNAP_INVERT;
1213         t->redraw |= TREDRAW_HARD;
1214         handled = true;
1215         break;
1216       case TFM_MODAL_SNAP_TOGGLE:
1217         t->modifiers ^= MOD_SNAP;
1218         t->redraw |= TREDRAW_HARD;
1219         handled = true;
1220         break;
1221       case TFM_MODAL_AXIS_X:
1222         if (!(t->flag & T_NO_CONSTRAINT)) {
1223           transform_event_xyz_constraint(t, XKEY, cmode, false);
1224           t->redraw |= TREDRAW_HARD;
1225           handled = true;
1226         }
1227         break;
1228       case TFM_MODAL_AXIS_Y:
1229         if ((t->flag & T_NO_CONSTRAINT) == 0) {
1230           transform_event_xyz_constraint(t, YKEY, cmode, false);
1231           t->redraw |= TREDRAW_HARD;
1232           handled = true;
1233         }
1234         break;
1235       case TFM_MODAL_AXIS_Z:
1236         if ((t->flag & (T_NO_CONSTRAINT)) == 0) {
1237           transform_event_xyz_constraint(t, ZKEY, cmode, false);
1238           t->redraw |= TREDRAW_HARD;
1239           handled = true;
1240         }
1241         break;
1242       case TFM_MODAL_PLANE_X:
1243         if ((t->flag & (T_NO_CONSTRAINT | T_2D_EDIT)) == 0) {
1244           transform_event_xyz_constraint(t, XKEY, cmode, true);
1245           t->redraw |= TREDRAW_HARD;
1246           handled = true;
1247         }
1248         break;
1249       case TFM_MODAL_PLANE_Y:
1250         if ((t->flag & (T_NO_CONSTRAINT | T_2D_EDIT)) == 0) {
1251           transform_event_xyz_constraint(t, YKEY, cmode, true);
1252           t->redraw |= TREDRAW_HARD;
1253           handled = true;
1254         }
1255         break;
1256       case TFM_MODAL_PLANE_Z:
1257         if ((t->flag & (T_NO_CONSTRAINT | T_2D_EDIT)) == 0) {
1258           transform_event_xyz_constraint(t, ZKEY, cmode, true);
1259           t->redraw |= TREDRAW_HARD;
1260           handled = true;
1261         }
1262         break;
1263       case TFM_MODAL_CONS_OFF:
1264         if ((t->flag & T_NO_CONSTRAINT) == 0) {
1265           stopConstraint(t);
1266           t->redraw |= TREDRAW_HARD;
1267           handled = true;
1268         }
1269         break;
1270       case TFM_MODAL_ADD_SNAP:
1271         addSnapPoint(t);
1272         t->redraw |= TREDRAW_HARD;
1273         handled = true;
1274         break;
1275       case TFM_MODAL_REMOVE_SNAP:
1276         removeSnapPoint(t);
1277         t->redraw |= TREDRAW_HARD;
1278         handled = true;
1279         break;
1280       case TFM_MODAL_PROPSIZE:
1281         /* MOUSEPAN usage... */
1282         if (t->flag & T_PROP_EDIT) {
1283           float fac = 1.0f + 0.005f * (event->y - event->prevy);
1284           t->prop_size *= fac;
1285           if (t->spacetype == SPACE_VIEW3D && t->persp != RV3D_ORTHO) {
1286             t->prop_size = max_ff(min_ff(t->prop_size, ((View3D *)t->view)->clip_end),
1287                                   T_PROP_SIZE_MIN);
1288           }
1289           else {
1290             t->prop_size = max_ff(min_ff(t->prop_size, T_PROP_SIZE_MAX), T_PROP_SIZE_MIN);
1291           }
1292           calculatePropRatio(t);
1293           t->redraw |= TREDRAW_HARD;
1294           handled = true;
1295         }
1296         break;
1297       case TFM_MODAL_PROPSIZE_UP:
1298         if (t->flag & T_PROP_EDIT) {
1299           t->prop_size *= (t->modifiers & MOD_PRECISION) ? 1.01f : 1.1f;
1300           if (t->spacetype == SPACE_VIEW3D && t->persp != RV3D_ORTHO) {
1301             t->prop_size = min_ff(t->prop_size, ((View3D *)t->view)->clip_end);
1302           }
1303           else {
1304             t->prop_size = min_ff(t->prop_size, T_PROP_SIZE_MAX);
1305           }
1306           calculatePropRatio(t);
1307           t->redraw |= TREDRAW_HARD;
1308           handled = true;
1309         }
1310         break;
1311       case TFM_MODAL_PROPSIZE_DOWN:
1312         if (t->flag & T_PROP_EDIT) {
1313           t->prop_size /= (t->modifiers & MOD_PRECISION) ? 1.01f : 1.1f;
1314           t->prop_size = max_ff(t->prop_size, T_PROP_SIZE_MIN);
1315           calculatePropRatio(t);
1316           t->redraw |= TREDRAW_HARD;
1317           handled = true;
1318         }
1319         break;
1320       case TFM_MODAL_AUTOIK_LEN_INC:
1321         if (t->flag & T_AUTOIK) {
1322           transform_autoik_update(t, 1);
1323           t->redraw |= TREDRAW_HARD;
1324           handled = true;
1325         }
1326         break;
1327       case TFM_MODAL_AUTOIK_LEN_DEC:
1328         if (t->flag & T_AUTOIK) {
1329           transform_autoik_update(t, -1);
1330           t->redraw |= TREDRAW_HARD;
1331           handled = true;
1332         }
1333         break;
1334       case TFM_MODAL_INSERTOFS_TOGGLE_DIR:
1335         if (t->spacetype == SPACE_NODE) {
1336           SpaceNode *snode = (SpaceNode *)t->sa->spacedata.first;
1337
1338           BLI_assert(t->sa->spacetype == t->spacetype);
1339
1340           if (snode->insert_ofs_dir == SNODE_INSERTOFS_DIR_RIGHT) {
1341             snode->insert_ofs_dir = SNODE_INSERTOFS_DIR_LEFT;
1342           }
1343           else if (snode->insert_ofs_dir == SNODE_INSERTOFS_DIR_LEFT) {
1344             snode->insert_ofs_dir = SNODE_INSERTOFS_DIR_RIGHT;
1345           }
1346           else {
1347             BLI_assert(0);
1348           }
1349
1350           t->redraw |= TREDRAW_SOFT;
1351         }
1352         break;
1353       /* Those two are only handled in transform's own handler, see T44634! */
1354       case TFM_MODAL_EDGESLIDE_UP:
1355       case TFM_MODAL_EDGESLIDE_DOWN:
1356       default:
1357         break;
1358     }
1359   }
1360   /* else do non-mapped events */
1361   else if (event->val == KM_PRESS) {
1362     switch (event->type) {
1363       case RIGHTMOUSE:
1364         t->state = TRANS_CANCEL;
1365         handled = true;
1366         break;
1367       /* enforce redraw of transform when modifiers are used */
1368       case LEFTSHIFTKEY:
1369       case RIGHTSHIFTKEY:
1370         t->modifiers |= MOD_CONSTRAINT_PLANE;
1371         t->redraw |= TREDRAW_HARD;
1372         handled = true;
1373         break;
1374
1375       case SPACEKEY:
1376         t->state = TRANS_CONFIRM;
1377         handled = true;
1378         break;
1379
1380       case MIDDLEMOUSE:
1381         if ((t->flag & T_NO_CONSTRAINT) == 0) {
1382           /* exception for switching to dolly, or trackball, in camera view */
1383           if (t->flag & T_CAMERA) {
1384             if (t->mode == TFM_TRANSLATION)
1385               setLocalConstraint(t, (CON_AXIS2), IFACE_("along local Z"));
1386             else if (t->mode == TFM_ROTATION) {
1387               restoreTransObjects(t);
1388               initTrackball(t);
1389             }
1390           }
1391           else {
1392             t->modifiers |= MOD_CONSTRAINT_SELECT;
1393             if (t->con.mode & CON_APPLY) {
1394               stopConstraint(t);
1395             }
1396             else {
1397               if (event->shift) {
1398                 /* bit hackish... but it prevents mmb select to print the
1399                  * orientation from menu */
1400                 float mati[3][3];
1401                 strcpy(t->spacename, "global");
1402                 unit_m3(mati);
1403                 initSelectConstraint(t, mati);
1404               }
1405               else {
1406                 initSelectConstraint(t, t->spacemtx);
1407               }
1408               postSelectConstraint(t);
1409             }
1410           }
1411           t->redraw |= TREDRAW_HARD;
1412           handled = true;
1413         }
1414         break;
1415       case ESCKEY:
1416         t->state = TRANS_CANCEL;
1417         handled = true;
1418         break;
1419       case PADENTER:
1420       case RETKEY:
1421         t->state = TRANS_CONFIRM;
1422         handled = true;
1423         break;
1424       case GKEY:
1425         /* only switch when... */
1426         if (ELEM(t->mode, TFM_ROTATION, TFM_RESIZE, TFM_TRACKBALL)) {
1427           restoreTransObjects(t);
1428           resetTransModal(t);
1429           resetTransRestrictions(t);
1430           initTranslation(t);
1431           initSnapping(t, NULL);  // need to reinit after mode change
1432           t->redraw |= TREDRAW_HARD;
1433           handled = true;
1434         }
1435         break;
1436       case SKEY:
1437         /* only switch when... */
1438         if (ELEM(t->mode, TFM_ROTATION, TFM_TRANSLATION, TFM_TRACKBALL)) {
1439           restoreTransObjects(t);
1440           resetTransModal(t);
1441           resetTransRestrictions(t);
1442           initResize(t);
1443           initSnapping(t, NULL);  // need to reinit after mode change
1444           t->redraw |= TREDRAW_HARD;
1445           handled = true;
1446         }
1447         break;
1448       case RKEY:
1449         /* only switch when... */
1450         if (!(t->options & CTX_TEXTURE)) {
1451           if (ELEM(t->mode, TFM_ROTATION, TFM_RESIZE, TFM_TRACKBALL, TFM_TRANSLATION)) {
1452             restoreTransObjects(t);
1453             resetTransModal(t);
1454             resetTransRestrictions(t);
1455
1456             if (t->mode == TFM_ROTATION) {
1457               initTrackball(t);
1458             }
1459             else {
1460               initRotation(t);
1461             }
1462             initSnapping(t, NULL);  // need to reinit after mode change
1463             t->redraw |= TREDRAW_HARD;
1464             handled = true;
1465           }
1466         }
1467         break;
1468       case CKEY:
1469         if (event->alt) {
1470           if (!(t->options & CTX_NO_PET)) {
1471             t->flag ^= T_PROP_CONNECTED;
1472             sort_trans_data_dist(t);
1473             calculatePropRatio(t);
1474             t->redraw = TREDRAW_HARD;
1475             handled = true;
1476           }
1477         }
1478         break;
1479       case OKEY:
1480         if (t->flag & T_PROP_EDIT && event->shift) {
1481           t->prop_mode = (t->prop_mode + 1) % PROP_MODE_MAX;
1482           calculatePropRatio(t);
1483           t->redraw |= TREDRAW_HARD;
1484           handled = true;
1485         }
1486         break;
1487       case PADPLUSKEY:
1488         if (event->alt && t->flag & T_PROP_EDIT) {
1489           t->prop_size *= (t->modifiers & MOD_PRECISION) ? 1.01f : 1.1f;
1490           if (t->spacetype == SPACE_VIEW3D && t->persp != RV3D_ORTHO)
1491             t->prop_size = min_ff(t->prop_size, ((View3D *)t->view)->clip_end);
1492           calculatePropRatio(t);
1493           t->redraw = TREDRAW_HARD;
1494           handled = true;
1495         }
1496         break;
1497       case PAGEUPKEY:
1498       case WHEELDOWNMOUSE:
1499         if (t->flag & T_AUTOIK) {
1500           transform_autoik_update(t, 1);
1501         }
1502         else {
1503           view_editmove(event->type);
1504         }
1505         t->redraw = TREDRAW_HARD;
1506         handled = true;
1507         break;
1508       case PADMINUS:
1509         if (event->alt && t->flag & T_PROP_EDIT) {
1510           t->prop_size /= (t->modifiers & MOD_PRECISION) ? 1.01f : 1.1f;
1511           calculatePropRatio(t);
1512           t->redraw = TREDRAW_HARD;
1513           handled = true;
1514         }
1515         break;
1516       case PAGEDOWNKEY:
1517       case WHEELUPMOUSE:
1518         if (t->flag & T_AUTOIK) {
1519           transform_autoik_update(t, -1);
1520         }
1521         else {
1522           view_editmove(event->type);
1523         }
1524         t->redraw = TREDRAW_HARD;
1525         handled = true;
1526         break;
1527       case LEFTALTKEY:
1528       case RIGHTALTKEY:
1529         if (ELEM(t->spacetype, SPACE_SEQ, SPACE_VIEW3D)) {
1530           t->flag |= T_ALT_TRANSFORM;
1531           t->redraw |= TREDRAW_HARD;
1532           handled = true;
1533         }
1534         break;
1535       case NKEY:
1536         if (ELEM(t->mode, TFM_ROTATION)) {
1537           if ((t->flag & T_EDIT) && t->obedit_type == OB_MESH) {
1538             restoreTransObjects(t);
1539             resetTransModal(t);
1540             resetTransRestrictions(t);
1541             initNormalRotation(t);
1542             t->redraw = TREDRAW_HARD;
1543             handled = true;
1544           }
1545         }
1546         break;
1547       default:
1548         break;
1549     }
1550
1551     /* Snapping key events */
1552     t->redraw |= handleSnapping(t, event);
1553   }
1554   else if (event->val == KM_RELEASE) {
1555     switch (event->type) {
1556       case LEFTSHIFTKEY:
1557       case RIGHTSHIFTKEY:
1558         t->modifiers &= ~MOD_CONSTRAINT_PLANE;
1559         t->redraw |= TREDRAW_HARD;
1560         handled = true;
1561         break;
1562
1563       case MIDDLEMOUSE:
1564         if ((t->flag & T_NO_CONSTRAINT) == 0) {
1565           t->modifiers &= ~MOD_CONSTRAINT_SELECT;
1566           postSelectConstraint(t);
1567           t->redraw |= TREDRAW_HARD;
1568           handled = true;
1569         }
1570         break;
1571       case LEFTALTKEY:
1572       case RIGHTALTKEY:
1573         if (ELEM(t->spacetype, SPACE_SEQ, SPACE_VIEW3D)) {
1574           t->flag &= ~T_ALT_TRANSFORM;
1575           t->redraw |= TREDRAW_HARD;
1576           handled = true;
1577         }
1578         break;
1579       default:
1580         break;
1581     }
1582
1583     /* confirm transform if launch key is released after mouse move */
1584     if (t->flag & T_RELEASE_CONFIRM) {
1585       /* XXX Keyrepeat bug in Xorg messes this up, will test when fixed */
1586       if ((event->type == t->launch_event) && ISMOUSE(t->launch_event)) {
1587         t->state = TRANS_CONFIRM;
1588       }
1589     }
1590   }
1591
1592   /* if we change snap options, get the unsnapped values back */
1593   if ((t->modifiers & (MOD_SNAP | MOD_SNAP_INVERT)) !=
1594       (modifiers_prev & (MOD_SNAP | MOD_SNAP_INVERT))) {
1595     applyMouseInput(t, &t->mouse, t->mval, t->values);
1596   }
1597
1598   /* Per transform event, if present */
1599   if (t->handleEvent && (!handled ||
1600                          /* Needed for vertex slide, see [#38756] */
1601                          (event->type == MOUSEMOVE))) {
1602     t->redraw |= t->handleEvent(t, event);
1603   }
1604
1605   /* Try to init modal numinput now, if possible. */
1606   if (!(handled || t->redraw) && ((event->val == KM_PRESS) || (event->type == EVT_MODAL_MAP)) &&
1607       handleNumInput(t->context, &(t->num), event)) {
1608     t->redraw |= TREDRAW_HARD;
1609     handled = true;
1610   }
1611
1612   if (t->redraw && !ELEM(event->type, MOUSEMOVE, INBETWEEN_MOUSEMOVE)) {
1613     WM_window_status_area_tag_redraw(CTX_wm_window(t->context));
1614   }
1615
1616   if (handled || t->redraw) {
1617     return 0;
1618   }
1619   else {
1620     return OPERATOR_PASS_THROUGH;
1621   }
1622 }
1623
1624 bool calculateTransformCenter(bContext *C, int centerMode, float cent3d[3], float cent2d[2])
1625 {
1626   TransInfo *t = MEM_callocN(sizeof(TransInfo), "TransInfo data");
1627   bool success;
1628
1629   t->context = C;
1630
1631   t->state = TRANS_RUNNING;
1632
1633   /* avoid calculating PET */
1634   t->options = CTX_NO_PET;
1635
1636   t->mode = TFM_DUMMY;
1637
1638   initTransInfo(C, t, NULL, NULL);
1639
1640   /* avoid doing connectivity lookups (when V3D_AROUND_LOCAL_ORIGINS is set) */
1641   t->around = V3D_AROUND_CENTER_BOUNDS;
1642
1643   createTransData(C, t);  // make TransData structs from selection
1644
1645   t->around = centerMode;  // override userdefined mode
1646
1647   if (t->data_len_all == 0) {
1648     success = false;
1649   }
1650   else {
1651     success = true;
1652
1653     calculateCenter(t);
1654
1655     if (cent2d) {
1656       copy_v2_v2(cent2d, t->center2d);
1657     }
1658
1659     if (cent3d) {
1660       // Copy center from constraint center. Transform center can be local
1661       copy_v3_v3(cent3d, t->center_global);
1662     }
1663   }
1664
1665   /* aftertrans does insert keyframes, and clears base flags; doesn't read transdata */
1666   special_aftertrans_update(C, t);
1667
1668   postTrans(C, t);
1669
1670   MEM_freeN(t);
1671
1672   return success;
1673 }
1674
1675 typedef enum {
1676   UP,
1677   DOWN,
1678   LEFT,
1679   RIGHT,
1680 } ArrowDirection;
1681
1682 #define POS_INDEX 0
1683 /* NOTE: this --^ is a bit hackish, but simplifies GPUVertFormat usage among functions
1684  * private to this file  - merwin
1685  */
1686
1687 static void drawArrow(ArrowDirection d, short offset, short length, short size)
1688 {
1689   immBegin(GPU_PRIM_LINES, 6);
1690
1691   switch (d) {
1692     case LEFT:
1693       offset = -offset;
1694       length = -length;
1695       size = -size;
1696       ATTR_FALLTHROUGH;
1697     case RIGHT:
1698       immVertex2f(POS_INDEX, offset, 0);
1699       immVertex2f(POS_INDEX, offset + length, 0);
1700       immVertex2f(POS_INDEX, offset + length, 0);
1701       immVertex2f(POS_INDEX, offset + length - size, -size);
1702       immVertex2f(POS_INDEX, offset + length, 0);
1703       immVertex2f(POS_INDEX, offset + length - size, size);
1704       break;
1705
1706     case DOWN:
1707       offset = -offset;
1708       length = -length;
1709       size = -size;
1710       ATTR_FALLTHROUGH;
1711     case UP:
1712       immVertex2f(POS_INDEX, 0, offset);
1713       immVertex2f(POS_INDEX, 0, offset + length);
1714       immVertex2f(POS_INDEX, 0, offset + length);
1715       immVertex2f(POS_INDEX, -size, offset + length - size);
1716       immVertex2f(POS_INDEX, 0, offset + length);
1717       immVertex2f(POS_INDEX, size, offset + length - size);
1718       break;
1719   }
1720
1721   immEnd();
1722 }
1723
1724 static void drawArrowHead(ArrowDirection d, short size)
1725 {
1726   immBegin(GPU_PRIM_LINES, 4);
1727
1728   switch (d) {
1729     case LEFT:
1730       size = -size;
1731       ATTR_FALLTHROUGH;
1732     case RIGHT:
1733       immVertex2f(POS_INDEX, 0, 0);
1734       immVertex2f(POS_INDEX, -size, -size);
1735       immVertex2f(POS_INDEX, 0, 0);
1736       immVertex2f(POS_INDEX, -size, size);
1737       break;
1738
1739     case DOWN:
1740       size = -size;
1741       ATTR_FALLTHROUGH;
1742     case UP:
1743       immVertex2f(POS_INDEX, 0, 0);
1744       immVertex2f(POS_INDEX, -size, -size);
1745       immVertex2f(POS_INDEX, 0, 0);
1746       immVertex2f(POS_INDEX, size, -size);
1747       break;
1748   }
1749
1750   immEnd();
1751 }
1752
1753 static void drawArc(float size, float angle_start, float angle_end, int segments)
1754 {
1755   float delta = (angle_end - angle_start) / segments;
1756   float angle;
1757   int a;
1758
1759   immBegin(GPU_PRIM_LINE_STRIP, segments + 1);
1760
1761   for (angle = angle_start, a = 0; a < segments; angle += delta, a++) {
1762     immVertex2f(POS_INDEX, cosf(angle) * size, sinf(angle) * size);
1763   }
1764   immVertex2f(POS_INDEX, cosf(angle_end) * size, sinf(angle_end) * size);
1765
1766   immEnd();
1767 }
1768
1769 static bool helpline_poll(bContext *C)
1770 {
1771   ARegion *ar = CTX_wm_region(C);
1772
1773   if (ar && ar->regiontype == RGN_TYPE_WINDOW)
1774     return 1;
1775   return 0;
1776 }
1777
1778 static void drawHelpline(bContext *UNUSED(C), int x, int y, void *customdata)
1779 {
1780   TransInfo *t = (TransInfo *)customdata;
1781
1782   if (t->helpline != HLP_NONE) {
1783     float cent[2];
1784     float mval[3] = {
1785         x,
1786         y,
1787         0.0f,
1788     };
1789     float tmval[2] = {
1790         (float)t->mval[0],
1791         (float)t->mval[1],
1792     };
1793
1794     projectFloatViewEx(t, t->center_global, cent, V3D_PROJ_TEST_CLIP_ZERO);
1795     /* Offset the values for the area region. */
1796     const float offset[2] = {
1797         t->ar->winrct.xmin,
1798         t->ar->winrct.ymin,
1799     };
1800
1801     for (int i = 0; i < 2; i++) {
1802       cent[i] += offset[i];
1803       tmval[i] += offset[i];
1804     }
1805
1806     GPU_matrix_push();
1807
1808     /* Dashed lines first. */
1809     if (ELEM(t->helpline, HLP_SPRING, HLP_ANGLE)) {
1810       const uint shdr_pos = GPU_vertformat_attr_add(
1811           immVertexFormat(), "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
1812
1813       UNUSED_VARS_NDEBUG(shdr_pos); /* silence warning */
1814       BLI_assert(shdr_pos == POS_INDEX);
1815
1816       GPU_line_width(1.0f);
1817
1818       immBindBuiltinProgram(GPU_SHADER_2D_LINE_DASHED_UNIFORM_COLOR);
1819
1820       float viewport_size[4];
1821       GPU_viewport_size_get_f(viewport_size);
1822       immUniform2f("viewport_size", viewport_size[2], viewport_size[3]);
1823
1824       immUniform1i("colors_len", 0); /* "simple" mode */
1825       immUniformThemeColor(TH_VIEW_OVERLAY);
1826       immUniform1f("dash_width", 6.0f);
1827       immUniform1f("dash_factor", 0.5f);
1828
1829       immBegin(GPU_PRIM_LINES, 2);
1830       immVertex2fv(POS_INDEX, cent);
1831       immVertex2f(POS_INDEX, tmval[0], tmval[1]);
1832       immEnd();
1833
1834       immUnbindProgram();
1835     }
1836
1837     /* And now, solid lines. */
1838     uint pos = GPU_vertformat_attr_add(immVertexFormat(), "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
1839     UNUSED_VARS_NDEBUG(pos); /* silence warning */
1840     BLI_assert(pos == POS_INDEX);
1841     immBindBuiltinProgram(GPU_SHADER_2D_UNIFORM_COLOR);
1842
1843     switch (t->helpline) {
1844       case HLP_SPRING:
1845         immUniformThemeColor(TH_VIEW_OVERLAY);
1846
1847         GPU_matrix_translate_3fv(mval);
1848         GPU_matrix_rotate_axis(-RAD2DEGF(atan2f(cent[0] - tmval[0], cent[1] - tmval[1])), 'Z');
1849
1850         GPU_line_width(3.0f);
1851         drawArrow(UP, 5, 10, 5);
1852         drawArrow(DOWN, 5, 10, 5);
1853         break;
1854       case HLP_HARROW:
1855         immUniformThemeColor(TH_VIEW_OVERLAY);
1856         GPU_matrix_translate_3fv(mval);
1857
1858         GPU_line_width(3.0f);
1859         drawArrow(RIGHT, 5, 10, 5);
1860         drawArrow(LEFT, 5, 10, 5);
1861         break;
1862       case HLP_VARROW:
1863         immUniformThemeColor(TH_VIEW_OVERLAY);
1864
1865         GPU_matrix_translate_3fv(mval);
1866
1867         GPU_line_width(3.0f);
1868         drawArrow(UP, 5, 10, 5);
1869         drawArrow(DOWN, 5, 10, 5);
1870         break;
1871       case HLP_CARROW: {
1872         /* Draw arrow based on direction defined by custom-points. */
1873         immUniformThemeColor(TH_VIEW_OVERLAY);
1874
1875         GPU_matrix_translate_3fv(mval);
1876
1877         GPU_line_width(3.0f);
1878
1879         const int *data = t->mouse.data;
1880         const float dx = data[2] - data[0], dy = data[3] - data[1];
1881         const float angle = -atan2f(dx, dy);
1882
1883         GPU_matrix_push();
1884
1885         GPU_matrix_rotate_axis(RAD2DEGF(angle), 'Z');
1886
1887         drawArrow(UP, 5, 10, 5);
1888         drawArrow(DOWN, 5, 10, 5);
1889
1890         GPU_matrix_pop();
1891         break;
1892       }
1893       case HLP_ANGLE: {
1894         float dx = tmval[0] - cent[0], dy = tmval[1] - cent[1];
1895         float angle = atan2f(dy, dx);
1896         float dist = hypotf(dx, dy);
1897         float delta_angle = min_ff(15.0f / dist, (float)M_PI / 4.0f);
1898         float spacing_angle = min_ff(5.0f / dist, (float)M_PI / 12.0f);
1899
1900         immUniformThemeColor(TH_VIEW_OVERLAY);
1901
1902         GPU_matrix_translate_3f(cent[0] - tmval[0] + mval[0], cent[1] - tmval[1] + mval[1], 0);
1903
1904         GPU_line_width(3.0f);
1905         drawArc(dist, angle - delta_angle, angle - spacing_angle, 10);
1906         drawArc(dist, angle + spacing_angle, angle + delta_angle, 10);
1907
1908         GPU_matrix_push();
1909
1910         GPU_matrix_translate_3f(
1911             cosf(angle - delta_angle) * dist, sinf(angle - delta_angle) * dist, 0);
1912         GPU_matrix_rotate_axis(RAD2DEGF(angle - delta_angle), 'Z');
1913
1914         drawArrowHead(DOWN, 5);
1915
1916         GPU_matrix_pop();
1917
1918         GPU_matrix_translate_3f(
1919             cosf(angle + delta_angle) * dist, sinf(angle + delta_angle) * dist, 0);
1920         GPU_matrix_rotate_axis(RAD2DEGF(angle + delta_angle), 'Z');
1921
1922         drawArrowHead(UP, 5);
1923         break;
1924       }
1925       case HLP_TRACKBALL: {
1926         unsigned char col[3], col2[3];
1927         UI_GetThemeColor3ubv(TH_GRID, col);
1928
1929         GPU_matrix_translate_3fv(mval);
1930
1931         GPU_line_width(3.0f);
1932
1933         UI_make_axis_color(col, col2, 'X');
1934         immUniformColor3ubv(col2);
1935
1936         drawArrow(RIGHT, 5, 10, 5);
1937         drawArrow(LEFT, 5, 10, 5);
1938
1939         UI_make_axis_color(col, col2, 'Y');
1940         immUniformColor3ubv(col2);
1941
1942         drawArrow(UP, 5, 10, 5);
1943         drawArrow(DOWN, 5, 10, 5);
1944         break;
1945       }
1946     }
1947
1948     immUnbindProgram();
1949     GPU_matrix_pop();
1950   }
1951 }
1952
1953 static bool transinfo_show_overlay(const struct bContext *C, TransInfo *t, ARegion *ar)
1954 {
1955   /* Don't show overlays when not the active view and when overlay is disabled: T57139 */
1956   bool ok = false;
1957   if (ar == t->ar) {
1958     ok = true;
1959   }
1960   else {
1961     ScrArea *sa = CTX_wm_area(C);
1962     if (sa->spacetype == SPACE_VIEW3D) {
1963       View3D *v3d = sa->spacedata.first;
1964       if ((v3d->flag2 & V3D_HIDE_OVERLAYS) == 0) {
1965         ok = true;
1966       }
1967     }
1968   }
1969   return ok;
1970 }
1971
1972 static void drawTransformView(const struct bContext *C, ARegion *ar, void *arg)
1973 {
1974   TransInfo *t = arg;
1975
1976   if (!transinfo_show_overlay(C, t, ar)) {
1977     return;
1978   }
1979
1980   GPU_line_width(1.0f);
1981
1982   drawConstraint(t);
1983   drawPropCircle(C, t);
1984   drawSnapping(C, t);
1985
1986   if (ar == t->ar) {
1987     /* edge slide, vert slide */
1988     drawEdgeSlide(t);
1989     drawVertSlide(t);
1990
1991     /* Rotation */
1992     drawDial3d(t);
1993   }
1994 }
1995
1996 /* just draw a little warning message in the top-right corner of the viewport
1997  * to warn that autokeying is enabled */
1998 static void drawAutoKeyWarning(TransInfo *UNUSED(t), ARegion *ar)
1999 {
2000   rcti rect;
2001   const char *printable = IFACE_("Auto Keying On");
2002   float printable_size[2];
2003   int xco, yco;
2004
2005   ED_region_visible_rect(ar, &rect);
2006
2007   const int font_id = BLF_default();
2008   BLF_width_and_height(
2009       font_id, printable, BLF_DRAW_STR_DUMMY_MAX, &printable_size[0], &printable_size[1]);
2010
2011   xco = (rect.xmax - U.widget_unit) - (int)printable_size[0];
2012   yco = (rect.ymax - U.widget_unit);
2013
2014   /* warning text (to clarify meaning of overlays)
2015    * - original color was red to match the icon, but that clashes badly with a less nasty border
2016    */
2017   unsigned char color[3];
2018   UI_GetThemeColorShade3ubv(TH_TEXT_HI, -50, color);
2019   BLF_color3ubv(font_id, color);
2020 #ifdef WITH_INTERNATIONAL
2021   BLF_draw_default(xco, yco, 0.0f, printable, BLF_DRAW_STR_DUMMY_MAX);
2022 #else
2023   BLF_draw_default_ascii(xco, yco, 0.0f, printable, BLF_DRAW_STR_DUMMY_MAX);
2024 #endif
2025
2026   /* autokey recording icon... */
2027   GPU_blend_set_func_separate(
2028       GPU_SRC_ALPHA, GPU_ONE_MINUS_SRC_ALPHA, GPU_ONE, GPU_ONE_MINUS_SRC_ALPHA);
2029   GPU_blend(true);
2030
2031   xco -= U.widget_unit;
2032   yco -= (int)printable_size[1] / 2;
2033
2034   UI_icon_draw(xco, yco, ICON_REC);
2035
2036   GPU_blend(false);
2037 }
2038
2039 static void drawTransformPixel(const struct bContext *C, ARegion *ar, void *arg)
2040 {
2041   TransInfo *t = arg;
2042
2043   if (!transinfo_show_overlay(C, t, ar)) {
2044     return;
2045   }
2046
2047   if (ar == t->ar) {
2048     Scene *scene = t->scene;
2049     ViewLayer *view_layer = t->view_layer;
2050     Object *ob = OBACT(view_layer);
2051
2052     /* draw auto-key-framing hint in the corner
2053      * - only draw if enabled (advanced users may be distracted/annoyed),
2054      *   for objects that will be autokeyframed (no point otherwise),
2055      *   AND only for the active region (as showing all is too overwhelming)
2056      */
2057     if ((U.autokey_flag & AUTOKEY_FLAG_NOWARNING) == 0) {
2058       if (ar == t->ar) {
2059         if (t->flag & (T_OBJECT | T_POSE)) {
2060           if (ob && autokeyframe_cfra_can_key(scene, &ob->id)) {
2061             drawAutoKeyWarning(t, ar);
2062           }
2063         }
2064       }
2065     }
2066   }
2067 }
2068
2069 /**
2070  * \see #initTransform which reads values from the operator.
2071  */
2072 void saveTransform(bContext *C, TransInfo *t, wmOperator *op)
2073 {
2074   ToolSettings *ts = CTX_data_tool_settings(C);
2075   int proportional = 0;
2076   PropertyRNA *prop;
2077
2078   // Save back mode in case we're in the generic operator
2079   if ((prop = RNA_struct_find_property(op->ptr, "mode"))) {
2080     RNA_property_enum_set(op->ptr, prop, t->mode);
2081   }
2082
2083   if ((prop = RNA_struct_find_property(op->ptr, "value"))) {
2084     float values[4];
2085
2086     copy_v4_v4(values, (t->flag & T_AUTOVALUES) ? t->auto_values : t->values);
2087
2088     if (RNA_property_array_check(prop)) {
2089       RNA_property_float_set_array(op->ptr, prop, values);
2090     }
2091     else {
2092       RNA_property_float_set(op->ptr, prop, values[0]);
2093     }
2094   }
2095
2096   /* convert flag to enum */
2097   switch (t->flag & T_PROP_EDIT_ALL) {
2098     case T_PROP_EDIT:
2099       proportional = PROP_EDIT_ON;
2100       break;
2101     case (T_PROP_EDIT | T_PROP_CONNECTED):
2102       proportional = PROP_EDIT_CONNECTED;
2103       break;
2104     case (T_PROP_EDIT | T_PROP_PROJECTED):
2105       proportional = PROP_EDIT_PROJECTED;
2106       break;
2107     default:
2108       proportional = PROP_EDIT_OFF;
2109       break;
2110   }
2111
2112   // If modal, save settings back in scene if not set as operator argument
2113   if ((t->flag & T_MODAL) || (op->flag & OP_IS_REPEAT)) {
2114     /* save settings if not set in operator */
2115
2116     /* skip saving proportional edit if it was not actually used */
2117     if (!(t->options & CTX_NO_PET)) {
2118       if ((prop = RNA_struct_find_property(op->ptr, "proportional")) &&
2119           !RNA_property_is_set(op->ptr, prop)) {
2120         if (t->spacetype == SPACE_GRAPH)
2121           ts->proportional_fcurve = proportional;
2122         else if (t->spacetype == SPACE_ACTION)
2123           ts->proportional_action = proportional;
2124         else if (t->obedit_type != -1)
2125           ts->proportional = proportional;
2126         else if (t->options & CTX_MASK)
2127           ts->proportional_mask = (proportional != PROP_EDIT_OFF);
2128         else
2129           ts->proportional_objects = (proportional != PROP_EDIT_OFF);
2130       }
2131
2132       if ((prop = RNA_struct_find_property(op->ptr, "proportional_size"))) {
2133         ts->proportional_size = RNA_property_is_set(op->ptr, prop) ?
2134                                     RNA_property_float_get(op->ptr, prop) :
2135                                     t->prop_size;
2136       }
2137
2138       if ((prop = RNA_struct_find_property(op->ptr, "proportional_edit_falloff")) &&
2139           !RNA_property_is_set(op->ptr, prop)) {
2140         ts->prop_mode = t->prop_mode;
2141       }
2142     }
2143
2144     /* do we check for parameter? */
2145     if (transformModeUseSnap(t)) {
2146       if (t->modifiers & MOD_SNAP) {
2147         ts->snap_flag |= SCE_SNAP;
2148       }
2149       else {
2150         ts->snap_flag &= ~SCE_SNAP;
2151       }
2152     }
2153
2154     if (t->spacetype == SPACE_VIEW3D) {
2155       if ((prop = RNA_struct_find_property(op->ptr, "orient_type")) &&
2156           !RNA_property_is_set(op->ptr, prop) &&
2157           (t->orientation.user != V3D_ORIENT_CUSTOM_MATRIX)) {
2158         TransformOrientationSlot *orient_slot = &t->scene->orientation_slots[SCE_ORIENT_DEFAULT];
2159         orient_slot->type = t->orientation.user;
2160         BLI_assert(((orient_slot->index_custom == -1) && (t->orientation.custom == NULL)) ||
2161                    (BKE_scene_transform_orientation_get_index(t->scene, t->orientation.custom) ==
2162                     orient_slot->index_custom));
2163       }
2164     }
2165   }
2166
2167   if ((prop = RNA_struct_find_property(op->ptr, "proportional"))) {
2168     RNA_property_enum_set(op->ptr, prop, proportional);
2169     RNA_enum_set(op->ptr, "proportional_edit_falloff", t->prop_mode);
2170     RNA_float_set(op->ptr, "proportional_size", t->prop_size);
2171   }
2172
2173   if ((prop = RNA_struct_find_property(op->ptr, "mirror"))) {
2174     RNA_property_boolean_set(op->ptr, prop, (t->flag & T_NO_MIRROR) == 0);
2175   }
2176
2177   /* Orientation used for redo. */
2178   const bool use_orient_axis = (t->orient_matrix_is_set &&
2179                                 (RNA_struct_find_property(op->ptr, "orient_axis") != NULL));
2180   short orientation;
2181   if (t->con.mode & CON_APPLY) {
2182     orientation = t->con.orientation;
2183     if (orientation == V3D_ORIENT_CUSTOM) {
2184       const int orientation_index_custom = BKE_scene_transform_orientation_get_index(
2185           t->scene, t->orientation.custom);
2186       /* Maybe we need a t->con.custom_orientation?
2187        * Seems like it would always match t->orientation.custom. */
2188       orientation = V3D_ORIENT_CUSTOM + orientation_index_custom;
2189       BLI_assert(orientation >= V3D_ORIENT_CUSTOM);
2190     }
2191   }
2192   else if ((t->orientation.user == V3D_ORIENT_CUSTOM_MATRIX) &&
2193            (prop = RNA_struct_find_property(op->ptr, "orient_matrix_type"))) {
2194     orientation = RNA_property_enum_get(op->ptr, prop);
2195   }
2196   else if (use_orient_axis) {
2197     /* We're not using an orientation, use the fallback. */
2198     orientation = t->orientation.unset;
2199   }
2200   else {
2201     orientation = V3D_ORIENT_GLOBAL;
2202   }
2203
2204   if ((prop = RNA_struct_find_property(op->ptr, "orient_axis"))) {
2205     if (t->flag & T_MODAL) {
2206       if (t->con.mode & CON_APPLY) {
2207         int orient_axis = constraintModeToIndex(t);
2208         if (orient_axis != -1) {
2209           RNA_property_enum_set(op->ptr, prop, orient_axis);
2210         }
2211       }
2212       else {
2213         RNA_property_enum_set(op->ptr, prop, t->orient_axis);
2214       }
2215     }
2216   }
2217   if ((prop = RNA_struct_find_property(op->ptr, "orient_axis_ortho"))) {
2218     if (t->flag & T_MODAL) {
2219       RNA_property_enum_set(op->ptr, prop, t->orient_axis_ortho);
2220     }
2221   }
2222
2223   if ((prop = RNA_struct_find_property(op->ptr, "orient_matrix"))) {
2224     if (t->flag & T_MODAL) {
2225       if (orientation != V3D_ORIENT_CUSTOM_MATRIX) {
2226         if (t->flag & T_MODAL) {
2227           RNA_enum_set(op->ptr, "orient_matrix_type", orientation);
2228         }
2229       }
2230       if (t->con.mode & CON_APPLY) {
2231         RNA_float_set_array(op->ptr, "orient_matrix", &t->con.mtx[0][0]);
2232       }
2233       else if (use_orient_axis) {
2234         RNA_float_set_array(op->ptr, "orient_matrix", &t->orient_matrix[0][0]);
2235       }
2236       else {
2237         RNA_float_set_array(op->ptr, "orient_matrix", &t->spacemtx[0][0]);
2238       }
2239     }
2240   }
2241
2242   if ((prop = RNA_struct_find_property(op->ptr, "orient_type"))) {
2243     /* constraint orientation can be global, even if user selects something else
2244      * so use the orientation in the constraint if set */
2245
2246     /* Use 'orient_matrix' instead. */
2247     if (t->flag & T_MODAL) {
2248       if (orientation != V3D_ORIENT_CUSTOM_MATRIX) {
2249         RNA_property_enum_set(op->ptr, prop, orientation);
2250       }
2251     }
2252   }
2253
2254   if ((prop = RNA_struct_find_property(op->ptr, "constraint_axis"))) {
2255     bool constraint_axis[3] = {false, false, false};
2256     if (t->flag & T_MODAL) {
2257       /* Only set if needed, so we can hide in the UI when nothing is set.
2258        * See 'transform_poll_property'. */
2259       if (t->con.mode & CON_APPLY) {
2260         if (t->con.mode & CON_AXIS0) {
2261           constraint_axis[0] = true;
2262         }
2263         if (t->con.mode & CON_AXIS1) {
2264           constraint_axis[1] = true;
2265         }
2266         if (t->con.mode & CON_AXIS2) {
2267           constraint_axis[2] = true;
2268         }
2269       }
2270       if (ELEM(true, UNPACK3(constraint_axis))) {
2271         RNA_property_boolean_set_array(op->ptr, prop, constraint_axis);
2272       }
2273     }
2274   }
2275
2276   {
2277     const char *prop_id = NULL;
2278     bool prop_state = true;
2279     if (t->mode == TFM_SHRINKFATTEN) {
2280       prop_id = "use_even_offset";
2281       prop_state = false;
2282     }
2283
2284     if (prop_id && (prop = RNA_struct_find_property(op->ptr, prop_id))) {
2285       RNA_property_boolean_set(op->ptr, prop, ((t->flag & T_ALT_TRANSFORM) == 0) == prop_state);
2286     }
2287   }
2288
2289   if ((prop = RNA_struct_find_property(op->ptr, "correct_uv"))) {
2290     RNA_property_boolean_set(
2291         op->ptr, prop, (t->settings->uvcalc_flag & UVCALC_TRANSFORM_CORRECT) != 0);
2292   }
2293
2294   if (t->mode == TFM_SHEAR) {
2295     prop = RNA_struct_find_property(op->ptr, "shear_axis");
2296     t->custom.mode.data = POINTER_FROM_INT(RNA_property_enum_get(op->ptr, prop));
2297     RNA_property_enum_set(op->ptr, prop, POINTER_AS_INT(t->custom.mode.data));
2298   }
2299 }
2300
2301 /**
2302  * \note  caller needs to free 't' on a 0 return
2303  * \warning  \a event might be NULL (when tweaking from redo panel)
2304  * \see #saveTransform which writes these values back.
2305  */
2306 bool initTransform(bContext *C, TransInfo *t, wmOperator *op, const wmEvent *event, int mode)
2307 {
2308   int options = 0;
2309   PropertyRNA *prop;
2310
2311   t->context = C;
2312
2313   /* added initialize, for external calls to set stuff in TransInfo, like undo string */
2314
2315   t->state = TRANS_STARTING;
2316
2317   if ((prop = RNA_struct_find_property(op->ptr, "cursor_transform")) &&
2318       RNA_property_is_set(op->ptr, prop)) {
2319     if (RNA_property_boolean_get(op->ptr, prop)) {
2320       options |= CTX_CURSOR;
2321     }
2322   }
2323
2324   if ((prop = RNA_struct_find_property(op->ptr, "texture_space")) &&
2325       RNA_property_is_set(op->ptr, prop)) {
2326     if (RNA_property_boolean_get(op->ptr, prop)) {
2327       options |= CTX_TEXTURE;
2328     }
2329   }
2330
2331   if ((prop = RNA_struct_find_property(op->ptr, "gpencil_strokes")) &&
2332       RNA_property_is_set(op->ptr, prop)) {
2333     if (RNA_property_boolean_get(op->ptr, prop)) {
2334       options |= CTX_GPENCIL_STROKES;
2335     }
2336   }
2337
2338   t->options = options;
2339
2340   t->mode = mode;
2341
2342   /* Needed to translate tweak events to mouse buttons. */
2343   t->launch_event = event ? WM_userdef_event_type_from_keymap_type(event->type) : -1;
2344
2345   /* XXX Remove this when wm_operator_call_internal doesn't use window->eventstate
2346    * (which can have type = 0) */
2347   /* For gizmo only, so assume LEFTMOUSE. */
2348   if (t->launch_event == 0) {
2349     t->launch_event = LEFTMOUSE;
2350   }
2351
2352   unit_m3(t->spacemtx);
2353
2354   initTransInfo(C, t, op, event);
2355   initTransformOrientation(C, t);
2356
2357   if (t->spacetype == SPACE_VIEW3D) {
2358     t->draw_handle_apply = ED_region_draw_cb_activate(
2359         t->ar->type, drawTransformApply, t, REGION_DRAW_PRE_VIEW);
2360     t->draw_handle_view = ED_region_draw_cb_activate(
2361         t->ar->type, drawTransformView, t, REGION_DRAW_POST_VIEW);
2362     t->draw_handle_pixel = ED_region_draw_cb_activate(
2363         t->ar->type, drawTransformPixel, t, REGION_DRAW_POST_PIXEL);
2364     t->draw_handle_cursor = WM_paint_cursor_activate(
2365         CTX_wm_manager(C), SPACE_TYPE_ANY, RGN_TYPE_ANY, helpline_poll, drawHelpline, t);
2366   }
2367   else if (t->spacetype == SPACE_IMAGE) {
2368     t->draw_handle_view = ED_region_draw_cb_activate(
2369         t->ar->type, drawTransformView, t, REGION_DRAW_POST_VIEW);
2370     t->draw_handle_cursor = WM_paint_cursor_activate(
2371         CTX_wm_manager(C), SPACE_TYPE_ANY, RGN_TYPE_ANY, helpline_poll, drawHelpline, t);
2372   }
2373   else if (t->spacetype == SPACE_CLIP) {
2374     t->draw_handle_view = ED_region_draw_cb_activate(
2375         t->ar->type, drawTransformView, t, REGION_DRAW_POST_VIEW);
2376     t->draw_handle_cursor = WM_paint_cursor_activate(
2377         CTX_wm_manager(C), SPACE_TYPE_ANY, RGN_TYPE_ANY, helpline_poll, drawHelpline, t);
2378   }
2379   else if (t->spacetype == SPACE_NODE) {
2380     t->draw_handle_view = ED_region_draw_cb_activate(
2381         t->ar->type, drawTransformView, t, REGION_DRAW_POST_VIEW);
2382     t->draw_handle_cursor = WM_paint_cursor_activate(
2383         CTX_wm_manager(C), SPACE_TYPE_ANY, RGN_TYPE_ANY, helpline_poll, drawHelpline, t);
2384   }
2385   else if (t->spacetype == SPACE_GRAPH) {
2386     t->draw_handle_view = ED_region_draw_cb_activate(
2387         t->ar->type, drawTransformView, t, REGION_DRAW_POST_VIEW);
2388     t->draw_handle_cursor = WM_paint_cursor_activate(
2389         CTX_wm_manager(C), SPACE_TYPE_ANY, RGN_TYPE_ANY, helpline_poll, drawHelpline, t);
2390   }
2391   else if (t->spacetype == SPACE_ACTION) {
2392     t->draw_handle_view = ED_region_draw_cb_activate(
2393         t->ar->type, drawTransformView, t, REGION_DRAW_POST_VIEW);
2394     t->draw_handle_cursor = WM_paint_cursor_activate(
2395         CTX_wm_manager(C), SPACE_TYPE_ANY, RGN_TYPE_ANY, helpline_poll, drawHelpline, t);
2396   }
2397
2398   createTransData(C, t);  // make TransData structs from selection
2399
2400   if (t->data_len_all == 0) {
2401     postTrans(C, t);
2402     return 0;
2403   }
2404
2405   if (event) {
2406     /* keymap for shortcut header prints */
2407     t->keymap = WM_keymap_active(CTX_wm_manager(C), op->type->modalkeymap);
2408
2409     /* Stupid code to have Ctrl-Click on gizmo work ok.
2410      *
2411      * Do this only for translation/rotation/resize because only these
2412      * modes are available from gizmo and doing such check could
2413      * lead to keymap conflicts for other modes (see #31584)
2414      */
2415     if (ELEM(mode, TFM_TRANSLATION, TFM_ROTATION, TFM_RESIZE)) {
2416       wmKeyMapItem *kmi;
2417
2418       for (kmi = t->keymap->items.first; kmi; kmi = kmi->next) {
2419         if (kmi->flag & KMI_INACTIVE) {
2420           continue;
2421         }
2422
2423         if (kmi->propvalue == TFM_MODAL_SNAP_INV_ON && kmi->val == KM_PRESS) {
2424           if ((ELEM(kmi->type, LEFTCTRLKEY, RIGHTCTRLKEY) && event->ctrl) ||
2425               (ELEM(kmi->type, LEFTSHIFTKEY, RIGHTSHIFTKEY) && event->shift) ||
2426               (ELEM(kmi->type, LEFTALTKEY, RIGHTALTKEY) && event->alt) ||
2427               ((kmi->type == OSKEY) && event->oskey)) {
2428             t->modifiers |= MOD_SNAP_INVERT;
2429           }
2430           break;
2431         }
2432       }
2433     }
2434   }
2435
2436   initSnapping(t, op);  // Initialize snapping data AFTER mode flags
2437
2438   initSnapSpatial(t, t->snap_spatial);
2439
2440   /* EVIL! posemode code can switch translation to rotate when 1 bone is selected.
2441    * will be removed (ton) */
2442
2443   /* EVIL2: we gave as argument also texture space context bit... was cleared */
2444
2445   /* EVIL3: extend mode for animation editors also switches modes...
2446    * but is best way to avoid duplicate code */
2447   mode = t->mode;
2448
2449   calculatePropRatio(t);
2450   calculateCenter(t);
2451
2452   /* Overwrite initial values if operator supplied a non-null vector.
2453    *
2454    * Run before init functions so 'values_modal_offset' can be applied on mouse input.
2455    */
2456   BLI_assert(is_zero_v4(t->values_modal_offset));
2457   if ((prop = RNA_struct_find_property(op->ptr, "value")) && RNA_property_is_set(op->ptr, prop)) {
2458     float values[4] = {0}; /* in case value isn't length 4, avoid uninitialized memory  */
2459
2460     if (RNA_property_array_check(prop)) {
2461       RNA_float_get_array(op->ptr, "value", values);
2462     }
2463     else {
2464       values[0] = RNA_float_get(op->ptr, "value");
2465     }
2466
2467     copy_v4_v4(t->values, values);
2468
2469     if (t->flag & T_MODAL) {
2470       copy_v4_v4(t->values_modal_offset, values);
2471       t->redraw = TREDRAW_HARD;
2472     }
2473     else {
2474       copy_v4_v4(t->auto_values, values);
2475       t->flag |= T_AUTOVALUES;
2476     }
2477   }
2478
2479   if (event) {
2480     /* Initialize accurate transform to settings requested by keymap. */
2481     bool use_accurate = false;
2482     if ((prop = RNA_struct_find_property(op->ptr, "use_accurate")) &&
2483         RNA_property_is_set(op->ptr, prop)) {
2484       if (RNA_property_boolean_get(op->ptr, prop)) {
2485         use_accurate = true;
2486       }
2487     }
2488     initMouseInput(t, &t->mouse, t->center2d, event->mval, use_accurate);
2489   }
2490
2491   switch (mode) {
2492     case TFM_TRANSLATION:
2493       initTranslation(t);
2494       break;
2495     case TFM_ROTATION:
2496       initRotation(t);
2497       break;
2498     case TFM_RESIZE:
2499       initResize(t);
2500       break;
2501     case TFM_SKIN_RESIZE:
2502       initSkinResize(t);
2503       break;
2504     case TFM_TOSPHERE:
2505       initToSphere(t);
2506       break;
2507     case TFM_SHEAR:
2508       prop = RNA_struct_find_property(op->ptr, "shear_axis");
2509       t->custom.mode.data = POINTER_FROM_INT(RNA_property_enum_get(op->ptr, prop));
2510       initShear(t);
2511       break;
2512     case TFM_BEND:
2513       initBend(t);
2514       break;
2515     case TFM_SHRINKFATTEN:
2516       initShrinkFatten(t);
2517       break;
2518     case TFM_TILT:
2519       initTilt(t);
2520       break;
2521     case TFM_CURVE_SHRINKFATTEN:
2522       initCurveShrinkFatten(t);
2523       break;
2524     case TFM_MASK_SHRINKFATTEN:
2525       initMaskShrinkFatten(t);
2526       break;
2527     case TFM_GPENCIL_SHRINKFATTEN:
2528       initGPShrinkFatten(t);
2529       break;
2530     case TFM_TRACKBALL:
2531       initTrackball(t);
2532       break;
2533     case TFM_PUSHPULL:
2534       initPushPull(t);
2535       break;
2536     case TFM_CREASE:
2537       initCrease(t);
2538       break;
2539     case TFM_BONESIZE: { /* used for both B-Bone width (bonesize) as for deform-dist (envelope) */
2540       /* Note: we have to pick one, use the active object. */
2541       TransDataContainer *tc = TRANS_DATA_CONTAINER_FIRST_OK(t);
2542       bArmature *arm = tc->poseobj->data;
2543       if (arm->drawtype == ARM_ENVELOPE) {
2544         initBoneEnvelope(t);
2545         t->mode = TFM_BONE_ENVELOPE_DIST;
2546       }
2547       else {
2548         initBoneSize(t);
2549       }
2550       break;
2551     }
2552     case TFM_BONE_ENVELOPE:
2553       initBoneEnvelope(t);
2554       break;
2555     case TFM_BONE_ENVELOPE_DIST:
2556       initBoneEnvelope(t);
2557       t->mode = TFM_BONE_ENVELOPE_DIST;
2558       break;
2559     case TFM_EDGE_SLIDE:
2560     case TFM_VERT_SLIDE: {
2561       const bool use_even = (op ? RNA_boolean_get(op->ptr, "use_even") : false);
2562       const bool flipped = (op ? RNA_boolean_get(op->ptr, "flipped") : false);
2563       const bool use_clamp = (op ? RNA_boolean_get(op->ptr, "use_clamp") : true);
2564       if (mode == TFM_EDGE_SLIDE) {
2565         const bool use_double_side = (op ? !RNA_boolean_get(op->ptr, "single_side") : true);
2566         initEdgeSlide_ex(t, use_double_side, use_even, flipped, use_clamp);
2567       }
2568       else {
2569         initVertSlide_ex(t, use_even, flipped, use_clamp);
2570       }
2571       break;
2572     }
2573     case TFM_BONE_ROLL:
2574       initBoneRoll(t);
2575       break;
2576     case TFM_TIME_TRANSLATE:
2577       initTimeTranslate(t);
2578       break;
2579     case TFM_TIME_SLIDE:
2580       initTimeSlide(t);
2581       break;
2582     case TFM_TIME_SCALE:
2583       initTimeScale(t);
2584       break;
2585     case TFM_TIME_DUPLICATE:
2586       /* same as TFM_TIME_EXTEND, but we need the mode info for later
2587        * so that duplicate-culling will work properly
2588        */
2589       if (ELEM(t->spacetype, SPACE_GRAPH, SPACE_NLA))
2590         initTranslation(t);
2591       else
2592         initTimeTranslate(t);
2593       t->mode = mode;
2594       break;
2595     case TFM_TIME_EXTEND:
2596       /* now that transdata has been made, do like for TFM_TIME_TRANSLATE (for most Animation
2597        * Editors because they have only 1D transforms for time values) or TFM_TRANSLATION
2598        * (for Graph/NLA Editors only since they uses 'standard' transforms to get 2D movement)
2599        * depending on which editor this was called from
2600        */
2601       if (ELEM(t->spacetype, SPACE_GRAPH, SPACE_NLA))
2602         initTranslation(t);
2603       else
2604         initTimeTranslate(t);
2605       break;
2606     case TFM_BAKE_TIME:
2607       initBakeTime(t);
2608       break;
2609     case TFM_MIRROR:
2610       initMirror(t);
2611       break;
2612     case TFM_BWEIGHT:
2613       initBevelWeight(t);
2614       break;
2615     case TFM_ALIGN:
2616       initAlign(t);
2617       break;
2618     case TFM_SEQ_SLIDE:
2619       initSeqSlide(t);
2620       break;
2621     case TFM_NORMAL_ROTATION:
2622       initNormalRotation(t);
2623       break;
2624     case TFM_GPENCIL_OPACITY:
2625       initGPOpacity(t);
2626       break;
2627   }
2628
2629   if (t->state == TRANS_CANCEL) {
2630     postTrans(C, t);
2631     return 0;
2632   }
2633
2634   /* Transformation axis from operator */
2635   if ((prop = RNA_struct_find_property(op->ptr, "orient_axis")) &&
2636       RNA_property_is_set(op->ptr, prop)) {
2637     t->orient_axis = RNA_property_enum_get(op->ptr, prop);
2638   }
2639   if ((prop = RNA_struct_find_property(op->ptr, "orient_axis_ortho")) &&
2640       RNA_property_is_set(op->ptr, prop)) {
2641     t->orient_axis_ortho = RNA_property_enum_get(op->ptr, prop);
2642   }
2643
2644   /* Constraint init from operator */
2645   if ((t->flag & T_MODAL) ||
2646       /* For mirror operator the constraint axes are effectively the values. */
2647       (RNA_struct_find_property(op->ptr, "values") == NULL)) {
2648     if ((prop = RNA_struct_find_property(op->ptr, "constraint_axis")) &&
2649         RNA_property_is_set(op->ptr, prop)) {
2650       bool constraint_axis[3];
2651
2652       RNA_property_boolean_get_array(op->ptr, prop, constraint_axis);
2653
2654       if (constraint_axis[0] || constraint_axis[1] || constraint_axis[2]) {
2655         t->con.mode |= CON_APPLY;
2656
2657         if (constraint_axis[0]) {
2658           t->con.mode |= CON_AXIS0;
2659         }
2660         if (constraint_axis[1]) {
2661           t->con.mode |= CON_AXIS1;
2662         }
2663         if (constraint_axis[2]) {
2664           t->con.mode |= CON_AXIS2;
2665         }
2666
2667         setUserConstraint(t, t->orientation.user, t->con.mode, "%s");
2668       }
2669     }
2670   }
2671   else {
2672     /* So we can adjust in non global orientation. */
2673     if (t->orientation.user != V3D_ORIENT_GLOBAL) {
2674       t->con.mode |= CON_APPLY | CON_AXIS0 | CON_AXIS1 | CON_AXIS2;
2675       setUserConstraint(t, t->orientation.user, t->con.mode, "%s");
2676     }
2677   }
2678
2679   /* Don't write into the values when non-modal because they are already set from operator redo
2680    * values. */
2681   if (t->flag & T_MODAL) {
2682     /* Setup the mouse input with initial values. */
2683     applyMouseInput(t, &t->mouse, t->mouse.imval, t->values);
2684   }
2685
2686   if ((prop = RNA_struct_find_property(op->ptr, "preserve_clnor"))) {
2687     if ((t->flag & T_EDIT) && t->obedit_type == OB_MESH) {
2688
2689       FOREACH_TRANS_DATA_CONTAINER(t, tc)
2690       {
2691         if ((((Mesh *)(tc->obedit->data))->flag & ME_AUTOSMOOTH)) {
2692           BMEditMesh *em = NULL;  // BKE_editmesh_from_object(t->obedit);
2693           bool do_skip = false;
2694
2695           /* Currently only used for two of three most frequent transform ops,
2696            * can include more ops.
2697            * Note that scaling cannot be included here,
2698            * non-uniform scaling will affect normals. */
2699           if (ELEM(t->mode, TFM_TRANSLATION, TFM_ROTATION)) {
2700             if (em->bm->totvertsel == em->bm->totvert) {
2701               /* No need to invalidate if whole mesh is selected. */
2702               do_skip = true;
2703             }
2704           }
2705
2706           if (t->flag & T_MODAL) {
2707             RNA_property_boolean_set(op->ptr, prop, false);
2708           }
2709           else if (!do_skip) {
2710             const bool preserve_clnor = RNA_property_boolean_get(op->ptr, prop);
2711             if (preserve_clnor) {
2712               BKE_editmesh_lnorspace_update(em);
2713               t->flag |= T_CLNOR_REBUILD;
2714             }
2715             BM_lnorspace_invalidate(em->bm, true);
2716           }
2717         }
2718       }
2719     }
2720   }
2721
2722   t->context = NULL;
2723
2724   return 1;
2725 }
2726
2727 void transformApply(bContext *C, TransInfo *t)
2728 {
2729   t->context = C;
2730
2731   if ((t->redraw & TREDRAW_HARD) || (t->draw_handle_apply == NULL && (t->redraw & TREDRAW_SOFT))) {
2732     selectConstraint(t);
2733     if (t->transform) {
2734       t->transform(t, t->mval);  // calls recalcData()
2735       viewRedrawForce(C, t);
2736     }
2737     t->redraw = TREDRAW_NOTHING;
2738   }
2739   else if (t->redraw & TREDRAW_SOFT) {
2740     viewRedrawForce(C, t);
2741   }
2742
2743   /* If auto confirm is on, break after one pass */
2744   if (t->options & CTX_AUTOCONFIRM) {
2745     t->state = TRANS_CONFIRM;
2746   }
2747
2748   t->context = NULL;
2749 }
2750
2751 static void drawTransformApply(const bContext *C, ARegion *UNUSED(ar), void *arg)
2752 {
2753   TransInfo *t = arg;
2754
2755   if (t->redraw & TREDRAW_SOFT) {
2756     t->redraw |= TREDRAW_HARD;
2757     transformApply((bContext *)C, t);
2758   }
2759 }
2760
2761 int transformEnd(bContext *C, TransInfo *t)
2762 {
2763   int exit_code = OPERATOR_RUNNING_MODAL;
2764
2765   t->context = C;
2766
2767   if (t->state != TRANS_STARTING && t->state != TRANS_RUNNING) {
2768     /* handle restoring objects */
2769     if (t->state == TRANS_CANCEL) {
2770       /* exception, edge slide transformed UVs too */
2771       if (t->mode == TFM_EDGE_SLIDE) {
2772         doEdgeSlide(t, 0.0f);
2773       }
2774       else if (t->mode == TFM_VERT_SLIDE) {
2775         doVertSlide(t, 0.0f);
2776       }
2777
2778       exit_code = OPERATOR_CANCELLED;
2779       restoreTransObjects(t);  // calls recalcData()
2780     }
2781     else {
2782       if (t->flag & T_CLNOR_REBUILD) {
2783         FOREACH_TRANS_DATA_CONTAINER(t, tc)
2784         {
2785           BMEditMesh *em = BKE_editmesh_from_object(tc->obedit);
2786           BM_lnorspace_rebuild(em->bm, true);
2787         }
2788       }
2789       exit_code = OPERATOR_FINISHED;
2790     }
2791
2792     /* aftertrans does insert keyframes, and clears base flags; doesn't read transdata */
2793     special_aftertrans_update(C, t);
2794
2795     /* free data */
2796     postTrans(C, t);
2797
2798     /* send events out for redraws */
2799     viewRedrawPost(C, t);
2800
2801     viewRedrawForce(C, t);
2802   }
2803
2804   t->context = NULL;
2805
2806   return exit_code;
2807 }
2808
2809 /* ************************** TRANSFORM LOCKS **************************** */
2810
2811 static void protectedTransBits(short protectflag, float vec[3])
2812 {
2813   if (protectflag & OB_LOCK_LOCX)
2814     vec[0] = 0.0f;
2815   if (protectflag & OB_LOCK_LOCY)
2816     vec[1] = 0.0f;
2817   if (protectflag & OB_LOCK_LOCZ)
2818     vec[2] = 0.0f;
2819 }
2820
2821 static void protectedSizeBits(short protectflag, float size[3])
2822 {
2823   if (protectflag & OB_LOCK_SCALEX)
2824     size[0] = 1.0f;
2825   if (protectflag & OB_LOCK_SCALEY)
2826     size[1] = 1.0f;
2827   if (protectflag & OB_LOCK_SCALEZ)
2828     size[2] = 1.0f;
2829 }
2830
2831 static void protectedRotateBits(short protectflag, float eul[3], const float oldeul[3])
2832 {
2833   if (protectflag & OB_LOCK_ROTX)
2834     eul[0] = oldeul[0];
2835   if (protectflag & OB_LOCK_ROTY)
2836     eul[1] = oldeul[1];
2837   if (protectflag & OB_LOCK_ROTZ)
2838     eul[2] = oldeul[2];
2839 }
2840
2841 /* this function only does the delta rotation */
2842 /* axis-angle is usually internally stored as quats... */
2843 static void protectedAxisAngleBits(
2844     short protectflag, float axis[3], float *angle, float oldAxis[3], float oldAngle)
2845 {
2846   /* check that protection flags are set */
2847   if ((protectflag & (OB_LOCK_ROTX | OB_LOCK_ROTY | OB_LOCK_ROTZ | OB_LOCK_ROTW)) == 0)
2848     return;
2849
2850   if (protectflag & OB_LOCK_ROT4D) {
2851     /* axis-angle getting limited as 4D entities that they are... */
2852     if (protectflag & OB_LOCK_ROTW)
2853       *angle = oldAngle;
2854     if (protectflag & OB_LOCK_ROTX)
2855       axis[0] = oldAxis[0];
2856     if (protectflag & OB_LOCK_ROTY)
2857       axis[1] = oldAxis[1];
2858     if (protectflag & OB_LOCK_ROTZ)
2859       axis[2] = oldAxis[2];
2860   }
2861   else {
2862     /* axis-angle get limited with euler... */
2863     float eul[3], oldeul[3];
2864
2865     axis_angle_to_eulO(eul, EULER_ORDER_DEFAULT, axis, *angle);
2866     axis_angle_to_eulO(oldeul, EULER_ORDER_DEFAULT, oldAxis, oldAngle);
2867
2868     if (protectflag & OB_LOCK_ROTX)
2869       eul[0] = oldeul[0];
2870     if (protectflag & OB_LOCK_ROTY)
2871       eul[1] = oldeul[1];
2872     if (protectflag & OB_LOCK_ROTZ)
2873       eul[2] = oldeul[2];
2874
2875     eulO_to_axis_angle(axis, angle, eul, EULER_ORDER_DEFAULT);
2876
2877     /* When converting to axis-angle,
2878      * we need a special exception for the case when there is no axis. */
2879     if (IS_EQF(axis[0], axis[1]) && IS_EQF(axis[1], axis[2])) {
2880       /* for now, rotate around y-axis then (so that it simply becomes the roll) */
2881       axis[1] = 1.0f;
2882     }
2883   }
2884 }
2885
2886 /* this function only does the delta rotation */
2887 static void protectedQuaternionBits(short protectflag, float quat[4], const float oldquat[4])
2888 {
2889   /* check that protection flags are set */
2890   if ((protectflag & (OB_LOCK_ROTX | OB_LOCK_ROTY | OB_LOCK_ROTZ | OB_LOCK_ROTW)) == 0)
2891     return;
2892
2893   if (protectflag & OB_LOCK_ROT4D) {
2894     /* quaternions getting limited as 4D entities that they are... */
2895     if (protectflag & OB_LOCK_ROTW)
2896       quat[0] = oldquat[0];
2897     if (protectflag & OB_LOCK_ROTX)
2898       quat[1] = oldquat[1];
2899     if (protectflag & OB_LOCK_ROTY)
2900       quat[2] = oldquat[2];
2901     if (protectflag & OB_LOCK_ROTZ)
2902       quat[3] = oldquat[3];
2903   }
2904   else {
2905     /* quaternions get limited with euler... (compatibility mode) */
2906     float eul[3], oldeul[3], nquat[4], noldquat[4];
2907     float qlen;
2908
2909     qlen = normalize_qt_qt(nquat, quat);
2910     normalize_qt_qt(noldquat, oldquat);
2911
2912     quat_to_eul(eul, nquat);
2913     quat_to_eul(oldeul, noldquat);
2914
2915     if (protectflag & OB_LOCK_ROTX)
2916       eul[0] = oldeul[0];
2917     if (protectflag & OB_LOCK_ROTY)
2918       eul[1] = oldeul[1];
2919     if (protectflag & OB_LOCK_ROTZ)
2920       eul[2] = oldeul[2];
2921
2922     eul_to_quat(quat, eul);
2923
2924     /* restore original quat size */
2925     mul_qt_fl(quat, qlen);
2926
2927     /* quaternions flip w sign to accumulate rotations correctly */
2928     if ((nquat[0] < 0.0f && quat[0] > 0.0f) || (nquat[0] > 0.0f && quat[0] < 0.0f)) {
2929       mul_qt_fl(quat, -1.0f);
2930     }
2931   }
2932 }
2933
2934 /* ******************* TRANSFORM LIMITS ********************** */
2935
2936 static void constraintTransLim(TransInfo *t, TransData *td)
2937 {
2938   if (td->con) {
2939     const bConstraintTypeInfo *ctiLoc = BKE_constraint_typeinfo_from_type(
2940         CONSTRAINT_TYPE_LOCLIMIT);
2941     const bConstraintTypeInfo *ctiDist = BKE_constraint_typeinfo_from_type(
2942         CONSTRAINT_TYPE_DISTLIMIT);
2943
2944     bConstraintOb cob = {NULL};
2945     bConstraint *con;
2946     float ctime = (float)(t->scene->r.cfra);
2947
2948     /* Make a temporary bConstraintOb for using these limit constraints
2949      * - they only care that cob->matrix is correctly set ;-)
2950      * - current space should be local
2951      */
2952     unit_m4(cob.matrix);
2953     copy_v3_v3(cob.matrix[3], td->loc);
2954
2955     /* Evaluate valid constraints */
2956     for (con = td->con; con; con = con->next) {
2957       const bConstraintTypeInfo *cti = NULL;
2958       ListBase targets = {NULL, NULL};
2959
2960       /* only consider constraint if enabled */
2961       if (con->flag & (CONSTRAINT_DISABLE | CONSTRAINT_OFF))
2962         continue;
2963       if (con->enforce == 0.0f)
2964         continue;
2965
2966       /* only use it if it's tagged for this purpose (and the right type) */
2967       if (con->type == CONSTRAINT_TYPE_LOCLIMIT) {
2968         bLocLimitConstraint *data = con->data;
2969
2970         if ((data->flag2 & LIMIT_TRANSFORM) == 0)
2971           continue;
2972         cti = ctiLoc;
2973       }
2974       else if (con->type == CONSTRAINT_TYPE_DISTLIMIT) {
2975         bDistLimitConstraint *data = con->data;
2976
2977         if ((data->flag & LIMITDIST_TRANSFORM) == 0)
2978           continue;
2979         cti = ctiDist;
2980       }
2981
2982       if (cti) {
2983         /* do space conversions */
2984         if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
2985           /* just multiply by td->mtx (this should be ok) */
2986           mul_m4_m3m4(cob.matrix, td->mtx, cob.matrix);
2987         }
2988         else if (con->ownspace != CONSTRAINT_SPACE_LOCAL) {
2989           /* skip... incompatible spacetype */
2990           continue;
2991         }
2992
2993         /* get constraint targets if needed */
2994         BKE_constraint_targets_for_solving_get(t->depsgraph, con, &cob, &targets, ctime);
2995
2996         /* do constraint */
2997         cti->evaluate_constraint(con, &cob, &targets);
2998
2999         /* convert spaces again */
3000         if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
3001           /* just multiply by td->smtx (this should be ok) */
3002           mul_m4_m3m4(cob.matrix, td->smtx, cob.matrix);
3003         }
3004
3005         /* free targets list */
3006         BLI_freelistN(&targets);
3007       }
3008     }
3009
3010     /* copy results from cob->matrix */
3011     copy_v3_v3(td->loc, cob.matrix[3]);
3012   }
3013 }
3014
3015 static void constraintob_from_transdata(bConstraintOb *cob, TransData *td)
3016 {
3017   /* Make a temporary bConstraintOb for use by limit constraints
3018    * - they only care that cob->matrix is correctly set ;-)
3019    * - current space should be local
3020    */
3021   memset(cob, 0, sizeof(bConstraintOb));
3022   if (td->ext) {
3023     if (td->ext->rotOrder == ROT_MODE_QUAT) {
3024       /* quats */
3025       /* objects and bones do normalization first too, otherwise
3026        * we don't necessarily end up with a rotation matrix, and
3027        * then conversion back to quat gives a different result */
3028       float quat[4];
3029       normalize_qt_qt(quat, td->ext->quat);
3030       quat_to_mat4(cob->matrix, quat);
3031     }
3032     else if (td->ext->rotOrder == ROT_MODE_AXISANGLE) {
3033       /* axis angle */
3034       axis_angle_to_mat4(cob->matrix, td->ext->rotAxis, *td->ext->rotAngle);
3035     }
3036     else {
3037       /* eulers */
3038       eulO_to_mat4(cob->matrix, td->ext->rot, td->ext->rotOrder);
3039     }
3040   }
3041 }
3042
3043 static void constraintRotLim(TransInfo *UNUSED(t), TransData *td)
3044 {
3045   if (td->con) {
3046     const bConstraintTypeInfo *cti = BKE_constraint_typeinfo_from_type(CONSTRAINT_TYPE_ROTLIMIT);
3047     bConstraintOb cob;
3048     bConstraint *con;
3049     bool do_limit = false;
3050
3051     /* Evaluate valid constraints */
3052     for (con = td->con; con; con = con->next) {
3053       /* only consider constraint if enabled */
3054       if (con->flag & (CONSTRAINT_DISABLE | CONSTRAINT_OFF))
3055         continue;
3056       if (con->enforce == 0.0f)
3057         continue;
3058
3059       /* we're only interested in Limit-Rotation constraints */
3060       if (con->type == CONSTRAINT_TYPE_ROTLIMIT) {
3061         bRotLimitConstraint *data = con->data;
3062
3063         /* only use it if it's tagged for this purpose */
3064         if ((data->flag2 & LIMIT_TRANSFORM) == 0)
3065           continue;
3066
3067         /* skip incompatible spacetypes */
3068         if (!ELEM(con->ownspace, CONSTRAINT_SPACE_WORLD, CONSTRAINT_SPACE_LOCAL))
3069           continue;
3070
3071         /* only do conversion if necessary, to preserve quats and eulers */
3072         if (do_limit == false) {
3073           constraintob_from_transdata(&cob, td);
3074           do_limit = true;
3075         }
3076
3077         /* do space conversions */
3078         if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
3079           /* just multiply by td->mtx (this should be ok) */
3080           mul_m4_m3m4(cob.matrix, td->mtx, cob.matrix);
3081         }
3082
3083         /* do constraint */
3084         cti->evaluate_constraint(con, &cob, NULL);
3085
3086         /* convert spaces again */
3087         if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
3088           /* just multiply by td->smtx (this should be ok) */
3089           mul_m4_m3m4(cob.matrix, td->smtx, cob.matrix);
3090         }
3091       }
3092     }
3093
3094     if (do_limit) {
3095       /* copy results from cob->matrix */
3096       if (td->ext->rotOrder == ROT_MODE_QUAT) {
3097         /* quats */
3098         mat4_to_quat(td->ext->quat, cob.matrix);
3099       }
3100       else if (td->ext->rotOrder == ROT_MODE_AXISANGLE) {
3101         /* axis angle */
3102         mat4_to_axis_angle(td->ext->rotAxis, td->ext->rotAngle, cob.matrix);
3103       }
3104       else {
3105         /* eulers */
3106         mat4_to_eulO(td->ext->rot, td->ext->rotOrder, cob.matrix);
3107       }
3108     }
3109   }
3110 }
3111
3112 static void constraintSizeLim(TransInfo *t, TransData *td)
3113 {
3114   if (td->con && td->ext) {
3115     const bConstraintTypeInfo *cti = BKE_constraint_typeinfo_from_type(CONSTRAINT_TYPE_SIZELIMIT);
3116     bConstraintOb cob = {NULL};
3117     bConstraint *con;
3118     float size_sign[3], size_abs[3];
3119     int i;
3120
3121     /* Make a temporary bConstraintOb for using these limit constraints
3122      * - they only care that cob->matrix is correctly set ;-)
3123      * - current space should be local
3124      */
3125     if ((td->flag & TD_SINGLESIZE) && !(t->con.mode & CON_APPLY)) {
3126       /* scale val and reset size */
3127       return;  // TODO: fix this case
3128     }
3129     else {
3130       /* Reset val if SINGLESIZE but using a constraint */
3131       if (td->flag & TD_SINGLESIZE)
3132         return;
3133
3134       /* separate out sign to apply back later */
3135       for (i = 0; i < 3; i++) {
3136         size_sign[i] = signf(td->ext->size[i]);
3137         size_abs[i] = fabsf(td->ext->size[i]);
3138       }
3139
3140       size_to_mat4(cob.matrix, size_abs);
3141     }
3142
3143     /* Evaluate valid constraints */
3144     for (con = td->con; con; con = con->next) {
3145       /* only consider constraint if enabled */
3146       if (con->flag & (CONSTRAINT_DISABLE | CONSTRAINT_OFF))
3147         continue;
3148       if (con->enforce == 0.0f)
3149         continue;
3150
3151       /* we're only interested in Limit-Scale constraints */
3152       if (con->type == CONSTRAINT_TYPE_SIZELIMIT) {
3153         bSizeLimitConstraint *data = con->data;
3154
3155         /* only use it if it's tagged for this purpose */
3156         if ((data->flag2 & LIMIT_TRANSFORM) == 0)
3157           continue;
3158
3159         /* do space conversions */
3160         if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
3161           /* just multiply by td->mtx (this should be ok) */
3162           mul_m4_m3m4(cob.matrix, td->mtx, cob.matrix);
3163         }
3164         else if (con->ownspace != CONSTRAINT_SPACE_LOCAL) {
3165           /* skip... incompatible spacetype */
3166           continue;
3167         }
3168
3169         /* do constraint */
3170         cti->evaluate_constraint(con, &cob, NULL);
3171
3172         /* convert spaces again */
3173         if (con->ownspace == CONSTRAINT_SPACE_WORLD) {
3174           /* just multiply by td->smtx (this should be ok) */
3175           mul_m4_m3m4(cob.matrix, td->smtx, cob.matrix);
3176         }
3177       }
3178     }
3179
3180     /* copy results from cob->matrix */
3181     if ((td->flag & TD_SINGLESIZE) && !(t->con.mode & CON_APPLY)) {
3182       /* scale val and reset size */
3183       return;  // TODO: fix this case
3184     }
3185     else {
3186       /* Reset val if SINGLESIZE but using a constraint */
3187       if (td->flag & TD_SINGLESIZE)
3188         return;
3189
3190       /* extrace scale from matrix and apply back sign */
3191       mat4_to_size(td->ext->size, cob.matrix);
3192       mul_v3_v3(td->ext->size, size_sign);
3193     }
3194   }
3195 }
3196
3197 /* -------------------------------------------------------------------- */
3198 /* Transform (Bend) */
3199
3200 /** \name Transform Bend
3201  * \{ */
3202
3203 struct BendCustomData {
3204   /* All values are in global space. */
3205   float warp_sta[3];
3206   float warp_end[3];
3207
3208   float warp_nor[3];
3209   float warp_tan[3];
3210
3211   /* for applying the mouse distance */
3212   float warp_init_dist;
3213 };
3214
3215 static void initBend(TransInfo *t)
3216 {
3217   const float mval_fl[2] = {UNPACK2(t->mval)};
3218   const float *curs;
3219   float tvec[3];
3220   struct BendCustomData *data;
3221
3222   t->mode = TFM_BEND;
3223   t->transform = Bend;
3224   t->handleEvent = handleEventBend;
3225
3226   setInputPostFct(&t->mouse, postInputRotation);
3227   initMouseInputMode(t, &t->mouse, INPUT_ANGLE_SPRING);
3228
3229   t->idx_max = 1;
3230   t->num.idx_max = 1;
3231   t->snap[0] = 0.0f;
3232   t->snap[1] = SNAP_INCREMENTAL_ANGLE;
3233   t->snap[2] = t->snap[1] * 0.2;
3234
3235   copy_v3_fl(t->num.val_inc, t->snap[1]);
3236   t->num.unit_sys = t->scene->unit.system;
3237   t->num.unit_use_radians = (t->scene->unit.system_rotation == USER_UNIT_ROT_RADIANS);
3238   t->num.unit_type[0] = B_UNIT_ROTATION;
3239   t->num.unit_type[1] = B_UNIT_LENGTH;
3240
3241   t->flag |= T_NO_CONSTRAINT;
3242
3243   //copy_v3_v3(t->center, ED_view3d_cursor3d_get(t->scene, t->view));
3244   if ((t->flag & T_OVERRIDE_CENTER) == 0) {
3245     calculateCenterCursor(t, t->center_global);
3246   }
3247   calculateCenterLocal(t, t->center_global);
3248
3249   t->val = 0.0f;
3250
3251   data = MEM_callocN(sizeof(*data), __func__);
3252
3253   curs = t->scene->cursor.location;
3254   copy_v3_v3(data->warp_sta, curs);
3255   ED_view3d_win_to_3d(t->sa->spacedata.first, t->ar, curs, mval_fl, data->warp_end);
3256
3257   copy_v3_v3(data->warp_nor, t->viewinv[2]);
3258   normalize_v3(data->warp_nor);
3259
3260   /* tangent */
3261   sub_v3_v3v3(tvec, data->warp_end, data->warp_sta);
3262   cross_v3_v3v3(data->warp_tan, tvec, data->warp_nor);
3263   normalize_v3(data->warp_tan);
3264
3265   data->warp_init_dist = len_v3v3(data->warp_end, data->warp_sta);
3266
3267   t->custom.mode.data = data;
3268   t->custom.mode.use_free = true;
3269 }
3270
3271 static eRedrawFlag handleEventBend(TransInfo *UNUSED(t), const wmEvent *event)
3272 {
3273   eRedrawFlag status = TREDRAW_NOTHING;
3274
3275   if (event->type == MIDDLEMOUSE && event->val == KM_PRESS) {
3276     status = TREDRAW_HARD;
3277   }
3278
3279   return status;
3280 }
3281
3282 static void Bend(TransInfo *t, const int UNUSED(mval[2]))
3283 {
3284   float vec[3];
3285   float pivot_global[3];
3286   float warp_end_radius_global[3];
3287   int i;
3288   char str[UI_MAX_DRAW_STR];
3289   const struct BendCustomData *data = t->custom.mode.data;
3290   const bool is_clamp = (t->flag & T_ALT_TRANSFORM) == 0;
3291
3292   union {
3293     struct {
3294       float angle, scale;
3295     };
3296     float vector[2];
3297   } values;
3298
3299   /* amount of radians for bend */
3300   copy_v2_v2(values.vector, t->values);
3301
3302 #if 0
3303   snapGrid(t, angle_rad);
3304 #else
3305   /* hrmf, snapping radius is using 'angle' steps, need to convert to something else
3306    * this isnt essential but nicer to give reasonable snapping values for radius */
3307   if (t->tsnap.mode & SCE_SNAP_MODE_INCREMENT) {
3308     const float radius_snap = 0.1f;
3309     const float snap_hack = (t->snap[1] * data->warp_init_dist) / radius_snap;
3310     values.scale *= snap_hack;
3311     snapGridIncrement(t, values.vector);
3312     values.scale /= snap_hack;
3313   }
3314 #endif
3315
3316   if (applyNumInput(&t->num, values.vector)) {
3317     values.scale = values.scale / data->warp_init_dist;
3318   }
3319
3320   copy_v2_v2(t->values, values.vector);
3321
3322   /* header print for NumInput */
3323   if (hasNumInput(&t->num)) {
3324     char c[NUM_STR_REP_LEN * 2];
3325
3326     outputNumInput(&(t->num), c, &t->scene->unit);
3327
3328     BLI_snprintf(str,
3329                  sizeof(str),
3330                  IFACE_("Bend Angle: %s Radius: %s Alt, Clamp %s"),
3331                  &c[0],
3332                  &c[NUM_STR_REP_LEN],
3333                  WM_bool_as_string(is_clamp));
3334   }
3335   else {
3336     /* default header print */
3337     BLI_snprintf(str,
3338                  sizeof(str),
3339                  IFACE_("Bend Angle: %.3f Radius: %.4f, Alt, Clamp %s"),
3340                  RAD2DEGF(values.angle),
3341                  values.scale * data->warp_init_dist,
3342                  WM_bool_as_string(is_clamp));
3343   }
3344
3345   values.angle *= -1.0f;
3346   values.scale *= data->warp_init_dist;
3347
3348   /* calc 'data->warp_end' from 'data->warp_end_init' */
3349   copy_v3_v3(warp_end_radius_global, data->warp_end);
3350   dist_ensure_v3_v3fl(warp_end_radius_global, data->warp_sta, values.scale);
3351   /* done */
3352
3353   /* calculate pivot */
3354   copy_v3_v3(pivot_global, data->warp_sta);
3355   if (values.angle > 0.0f) {
3356     madd_v3_v3fl(pivot_global,
3357                  data->warp_tan,
3358                  -values.scale * shell_angle_to_dist((float)M_PI_2 - values.angle));
3359   }
3360   else {
3361     madd_v3_v3fl(pivot_global,
3362                  data->warp_tan,
3363                  +values.scale * shell_angle_to_dist((float)M_PI_2 + values.angle));
3364   }
3365
3366   /* TODO(campbell): xform, compensate object center. */
3367   FOREACH_TRANS_DATA_CONTAINER(t, tc)
3368   {
3369     TransData *td = tc->data;
3370
3371     float warp_sta_local[3];
3372     float warp_end_local[3];
3373     float warp_end_radius_local[3];
3374     float pivot_local[3];
3375
3376     if (tc->use_local_mat) {
3377       sub_v3_v3v3(warp_sta_local, data->warp_sta, tc->mat[3]);
3378       sub_v3_v3v3(warp_end_local, data->warp_end, tc->mat[3]);
3379       sub_v3_v3v3(warp_end_radius_local, warp_end_radius_global, tc->mat[3]);
3380       sub_v3_v3v3(pivot_local, pivot_global, tc->mat[3]);
3381     }
3382     else {
3383       copy_v3_v3(warp_sta_local, data->warp_sta);
3384       copy_v3_v3(warp_end_local, data->warp_end);
3385       copy_v3_v3(warp_end_radius_local, warp_end_radius_global);
3386       copy_v3_v3(pivot_local, pivot_global);
3387     }
3388
3389     for (i = 0; i < tc->data_len; i++, td++) {
3390       float mat[3][3];
3391       float delta[3];
3392       float fac, fac_scaled;
3393
3394       if (td->flag & TD_NOACTION)
3395         break;
3396
3397       if (td->flag & TD_SKIP)
3398         continue;
3399
3400       if (UNLIKELY(values.angle == 0.0f)) {
3401         copy_v3_v3(td->loc, td->iloc);
3402         continue;
3403       }
3404
3405       copy_v3_v3(vec, td->iloc);
3406       mul_m3_v3(td->mtx, vec);
3407
3408       fac = line_point_factor_v3(vec, warp_sta_local, warp_end_radius_local);
3409       if (is_clamp) {
3410         CLAMP(fac, 0.0f, 1.0f);
3411       }
3412
3413       if (t->options & CTX_GPENCIL_STROKES) {
3414         /* grease pencil multiframe falloff */
3415         bGPDstroke *gps = (bGPDstroke *)td->extra;
3416         if (gps != NULL) {
3417           fac_scaled = fac * td->factor * gps->runtime.multi_frame_falloff;
3418         }
3419         else {
3420           fac_scaled = fac * td->factor;
3421         }
3422       }
3423       else {
3424         fac_scaled = fac * td->factor;
3425       }
3426
3427       axis_angle_normalized_to_mat3(mat, data->warp_nor, values.angle * fac_scaled);
3428       interp_v3_v3v3(delta, warp_sta_local, warp_end_radius_local, fac_scaled);
3429       sub_v3_v3(delta, warp_sta_local);
3430
3431       /* delta is subtracted, rotation adds back this offset */
3432       sub_v3_v3(vec, delta);
3433
3434       sub_v3_v3(vec, pivot_local);
3435       mul_m3_v3(mat, vec);
3436       add_v3_v3(vec, pivot_local);
3437
3438       mul_m3_v3(td->smtx, vec);
3439
3440       /* rotation */
3441       if ((t->flag & T_POINTS) == 0) {
3442         ElementRotation(t, tc, td, mat, V3D_AROUND_LOCAL_ORIGINS);
3443       }
3444
3445       /* location */
3446       copy_v3_v3(td->loc, vec);
3447     }
3448   }
3449
3450   recalcData(t);
3451
3452   ED_area_status_text(t->sa, str);
3453 }
3454 /** \} */
3455
3456 /* -------------------------------------------------------------------- */
3457 /* Transform (Shear) */
3458
3459 /** \name Transform Shear
3460  * \{ */
3461
3462 static void initShear_mouseInputMode(TransInfo *t)
3463 {
3464   float dir[3];
3465
3466   if (t->custom.mode.data == NULL) {
3467     copy_v3_v3(dir, t->orient_matrix[t->orient_axis_ortho]);
3468   }
3469   else {
3470     cross_v3_v3v3(dir, t->orient_matrix[t->orient_axis_ortho], t->orient_matrix[t->orient_axis]);
3471   }
3472
3473   /* Without this, half the gizmo handles move in the opposite direction. */
3474   if ((t->orient_axis_ortho + 1) % 3 != t->orient_axis) {
3475     negate_v3(dir);
3476   }
3477
3478   mul_mat3_m4_v3(t->viewmat, dir);
3479   if (normalize_v2(dir) == 0.0f) {
3480     dir[0] = 1.0f;
3481   }
3482   setCustomPointsFromDirection(t, &t->mouse, dir);
3483
3484   initMouseInputMode(t, &t->mouse, INPUT_CUSTOM_RATIO);
3485 }
3486
3487 static void initShear(TransInfo *t)
3488 {
3489   t->mode = TFM_SHEAR;
3490   t->transform = applyShear;
3491   t->handleEvent = handleEventShear;
3492
3493   if (t->orient_axis == t->orient_axis_ortho) {
3494     t->orient_axis = 2;
3495     t->orient_axis_ortho = 1;
3496   }
3497
3498   initShear_mouseInputMode(t);
3499
3500   t->idx_max = 0;
3501   t->num.idx_max = 0;
3502   t->snap[0] = 0.0f;
3503   t->snap[1] = 0.1f;
3504   t->snap[2] = t->snap[1] * 0.1f;
3505
3506   copy_v3_fl(t->num.val_inc, t->snap[1]);
3507   t->num.unit_sys = t->scene->unit.system;
3508   t->num.unit_type[0] = B_UNIT_NONE; /* Don't think we have any unit here? */
3509
3510   t->flag |= T_NO_CONSTRAINT;
3511 }
3512
3513 static eRedrawFlag handleEventShear(TransInfo *t, const wmEvent *event)
3514 {
3515   eRedrawFlag status = TREDRAW_NOTHING;
3516
3517   if (event->type == MIDDLEMOUSE && event->val == KM_PRESS) {
3518     /* Use custom.mode.data pointer to signal Shear direction */
3519     if (t->custom.mode.data == NULL) {
3520       t->custom.mode.data = (void *)1;
3521     }
3522     else {
3523       t->custom.mode.data = NULL;
3524     }
3525     initShear_mouseInputMode(t);
3526
3527     status = TREDRAW_HARD;
3528   }
3529   else if (event->type == XKEY && event->val == KM_PRESS) {
3530     t->custom.mode.data = NULL;
3531     initShear_mouseInputMode(t);
3532
3533     status = TREDRAW_HARD;
3534   }
3535   else if (event->type == YKEY && event->val == KM_PRESS) {
3536     t->custom.mode.data = (void *)1;
3537     initShear_mouseInputMode(t);
3538
3539     status = TREDRAW_HARD;
3540   }
3541
3542   return status;
3543 }
3544
3545 static void applyShear(TransInfo *t, const int UNUSED(mval[2]))
3546 {
3547   float vec[3];
3548   float smat[3][3], tmat[3][3], totmat[3][3], axismat[3][3], axismat_inv[3][3];
3549   float value;
3550   int i;
3551   char str[UI_MAX_DRAW_STR];
3552   const bool is_local_center = transdata_check_local_center(t, t->around);
3553
3554   value = t->values[0];
3555
3556   snapGridIncrement(t, &value);
3557
3558   applyNumInput(&t->num, &value);
3559
3560   t->values[0] = value;
3561
3562   /* header print for NumInput */
3563   if (hasNumInput(&t->num)) {
3564     char c[NUM_STR_REP_LEN];
3565
3566     outputNumInput(&(t->num), c, &t->scene->unit);
3567
3568     BLI_snprintf(str, sizeof(str), IFACE_("Shear: %s %s"), c, t->proptext);
3569   }
3570   else {
3571     /* default header print */
3572     BLI_snprintf(str,
3573                  sizeof(str),
3574                  IFACE_("Shear: %.3f %s (Press X or Y to set shear axis)"),
3575                  value,
3576                  t->proptext);
3577   }
3578
3579   unit_m3(smat);
3580
3581   // Custom data signals shear direction
3582   if (t->custom.mode.data == NULL)
3583     smat[1][0] = value;
3584   else
3585     smat[0][1] = value;
3586
3587   copy_v3_v3(axismat_inv[0], t->orient_matrix[t->orient_axis_ortho]);
3588   copy_v3_v3(axismat_inv[2], t->orient_matrix[t->orient_axis]);
3589   cross_v3_v3v3(axismat_inv[1], axismat_inv[0], axismat_inv[2]);
3590   invert_m3_m3(axismat, axismat_inv);
3591
3592   mul_m3_series(totmat, axismat_inv, smat, axismat);
3593
3594   FOREACH_TRANS_DATA_CONTAINER(t, tc)
3595   {
3596     TransData *td = tc->data;
3597     for (i = 0; i < tc->data_len; i++, td++) {
3598       const float *center, *co;
3599
3600       if (td->flag & TD_NOACTION)
3601         break;
3602
3603       if (td->flag & TD_SKIP)
3604         continue;
3605
3606       if (t->flag & T_EDIT) {
3607         mul_m3_series(tmat, td->smtx, totmat, td->mtx);
3608       }
3609       else {
3610         copy_m3_m3(tmat, totmat);
3611       }
3612
3613       if (is_local_center) {
3614         center = td->center;
3615         co = td->loc;
3616       }
3617       else {
3618         center = tc->center_local;
3619         co = td->center;
3620       }
3621
3622       sub_v3_v3v3(vec, co, center);
3623
3624       mul_m3_v3(tmat, vec);
3625
3626       add_v3_v3(vec, center);
3627       sub_v3_v3(vec, co);
3628
3629       if (t->options & CTX_GPENCIL_STROKES) {
3630         /* grease pencil multiframe falloff */
3631         bGPDstroke *gps = (bGPDstroke *)td->extra;
3632         if (gps != NULL) {
3633           mul_v3_fl(vec, td->factor * gps->runtime.multi_frame_falloff);
3634         }
3635         else {
3636           mul_v3_fl(vec, td->factor);
3637         }
3638       }
3639       else {
3640         mul_v3_fl(vec, td->factor);
3641       }
3642
3643       add_v3_v3v3(td->loc, td->iloc, vec);
3644     }
3645   }
3646
3647   recalcData(t);
3648
3649   ED_area_status_text(t->sa, str);
3650 }
3651 /** \} */
3652
3653 /* -------------------------------------------------------------------- */
3654 /* Transform (Resize) */
3655
3656 /** \name Transform Resize
3657  * \{ */
3658
3659 static void initResize(TransInfo *t)
3660 {
3661   t->mode = TFM_RESIZE;
3662   t->transform = applyResize;
3663
3664   initMouseInputMode(t, &t->mouse, INPUT_SPRING_FLIP);
3665