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