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