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