source/blender/editors/transform/transform_input.c

   1 /*
   2  * ***** BEGIN GPL LICENSE BLOCK *****
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License
   6  * as published by the Free Software Foundation; either version 2
   7  * of the License, or (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software Foundation,
  16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  17  *
  18  * Contributor(s): none yet.
  19  *
  20  * ***** END GPL LICENSE BLOCK *****
  21  */
  22
  23 /** \file blender/editors/transform/transform_input.c
  24  *  \ingroup edtransform
  25  */
  26
  27
  28 #include <stdlib.h>
  29 #include <math.h>
  30
  31 #include "DNA_screen_types.h"
  32
  33 #include "BLI_math.h"
  34 #include "BLI_utildefines.h"
  35
  36 #include "WM_types.h"
  37
  38 #include "transform.h"
  39
  40 #include "MEM_guardedalloc.h"
  41
  42 /* ************************** INPUT FROM MOUSE *************************** */
  43
  44 static void InputVector(TransInfo *t, MouseInput *mi, const int mval[2], float output[3])
  45 {
  46         float vec[3], dvec[3];
  47         if (mi->precision) {
  48                 /* calculate the main translation and the precise one separate */
  49                 convertViewVec(t, dvec, (mval[0] - mi->precision_mval[0]), (mval[1] - mi->precision_mval[1]));
  50                 mul_v3_fl(dvec, 0.1f);
  51                 convertViewVec(t, vec, (mi->precision_mval[0] - t->imval[0]), (mi->precision_mval[1] - t->imval[1]));
  52                 add_v3_v3v3(output, vec, dvec);
  53         }
  54         else {
  55                 convertViewVec(t, output, (mval[0] - t->imval[0]), (mval[1] - t->imval[1]));
  56         }
  57
  58 }
  59
  60 static void InputSpring(TransInfo *UNUSED(t), MouseInput *mi, const int mval[2], float output[3])
  61 {
  62         float ratio, precise_ratio, dx, dy;
  63         if (mi->precision) {
  64                 /* calculate ratio for shiftkey pos, and for total, and blend these for precision */
  65                 dx = (float)(mi->center[0] - mi->precision_mval[0]);
  66                 dy = (float)(mi->center[1] - mi->precision_mval[1]);
  67                 ratio = (float)sqrt( dx*dx + dy*dy);
  68
  69                 dx= (float)(mi->center[0] - mval[0]);
  70                 dy= (float)(mi->center[1] - mval[1]);
  71                 precise_ratio = (float)sqrt( dx*dx + dy*dy);
  72
  73                 ratio = (ratio + (precise_ratio - ratio) / 10.0f) / mi->factor;
  74         }
  75         else {
  76                 dx = (float)(mi->center[0] - mval[0]);
  77                 dy = (float)(mi->center[1] - mval[1]);
  78                 ratio = (float)sqrt( dx*dx + dy*dy) / mi->factor;
  79         }
  80
  81         output[0] = ratio;
  82 }
  83
  84 static void InputSpringFlip(TransInfo *t, MouseInput *mi, const int mval[2], float output[3])
  85 {
  86         InputSpring(t, mi, mval, output);
  87
  88         /* flip scale */
  89         /* values can become really big when zoomed in so use longs [#26598] */
  90         if      ((long long int)(mi->center[0] - mval[0]) * (long long int)(mi->center[0] - mi->imval[0]) +
  91                  (long long int)(mi->center[1] - mval[1]) * (long long int)(mi->center[1] - mi->imval[1]) < 0)
  92          {
  93                 output[0] *= -1.0f;
  94          }
  95 }
  96
  97 static void InputTrackBall(TransInfo *UNUSED(t), MouseInput *mi, const int mval[2], float output[3])
  98 {
  99
 100         if (mi->precision) {
 101                 output[0] = ( mi->imval[1] - mi->precision_mval[1] ) + ( mi->precision_mval[1] - mval[1] ) * 0.1f;
 102                 output[1] = ( mi->precision_mval[0] - mi->imval[0] ) + ( mval[0] - mi->precision_mval[0] ) * 0.1f;
 103         }
 104         else {
 105                 output[0] = (float)( mi->imval[1] - mval[1] );
 106                 output[1] = (float)( mval[0] - mi->imval[0] );
 107         }
 108
 109         output[0] *= mi->factor;
 110         output[1] *= mi->factor;
 111 }
 112
 113 static void InputHorizontalRatio(TransInfo *t, MouseInput *mi, const int mval[2], float output[3])
 114 {
 115         float x, pad;
 116
 117         pad = t->ar->winx / 10;
 118
 119         if (mi->precision) {
 120                 /* deal with Shift key by adding motion / 10 to motion before shift press */
 121                 x = mi->precision_mval[0] + (float)(mval[0] - mi->precision_mval[0]) / 10.0f;
 122         }
 123         else {
 124                 x = mval[0];
 125         }
 126
 127         output[0] = (x - pad) / (t->ar->winx - 2 * pad);
 128 }
 129
 130 static void InputHorizontalAbsolute(TransInfo *t, MouseInput *mi, const int mval[2], float output[3])
 131 {
 132         float vec[3];
 133
 134         InputVector(t, mi, mval, vec);
 135         project_v3_v3v3(vec, vec, t->viewinv[0]);
 136
 137         output[0] = dot_v3v3(t->viewinv[0], vec) * 2.0f;
 138 }
 139
 140 static void InputVerticalRatio(TransInfo *t, MouseInput *mi, const int mval[2], float output[3])
 141 {
 142         float y, pad;
 143
 144         pad = t->ar->winy / 10;
 145
 146         if (mi->precision) {
 147                 /* deal with Shift key by adding motion / 10 to motion before shift press */
 148                 y = mi->precision_mval[1] + (float)(mval[1] - mi->precision_mval[1]) / 10.0f;
 149         }
 150         else {
 151                 y = mval[0];
 152         }
 153
 154         output[0] = (y - pad) / (t->ar->winy - 2 * pad);
 155 }
 156
 157 static void InputVerticalAbsolute(TransInfo *t, MouseInput *mi, const int mval[2], float output[3])
 158 {
 159         float vec[3];
 160
 161         InputVector(t, mi, mval, vec);
 162         project_v3_v3v3(vec, vec, t->viewinv[1]);
 163
 164         output[0] = dot_v3v3(t->viewinv[1], vec) * 2.0f;
 165 }
 166
 167 void setCustomPoints(TransInfo *UNUSED(t), MouseInput *mi, int start[2], int end[2])
 168 {
 169         int *data;
 170
 171         if (mi->data == NULL) {
 172                 mi->data = MEM_callocN(sizeof(int) * 4, "custom points");
 173         }
 174
 175         data = mi->data;
 176
 177         data[0] = start[0];
 178         data[1] = start[1];
 179         data[2] = end[0];
 180         data[3] = end[1];
 181 }
 182
 183 static void InputCustomRatio(TransInfo *UNUSED(t), MouseInput *mi, const int mval[2], float output[3])
 184 {
 185         double length;
 186         double distance;
 187         double dx, dy;
 188         int *data = mi->data;
 189
 190         if (data) {
 191                 dx = data[2] - data[0];
 192                 dy = data[3] - data[1];
 193
 194                 length = sqrt(dx*dx + dy*dy);
 195
 196                 if (mi->precision) {
 197                         /* deal with Shift key by adding motion / 10 to motion before shift press */
 198                         int mdx, mdy;
 199                         mdx = (mi->precision_mval[0] + (float)(mval[0] - mi->precision_mval[0]) / 10.0f) - data[2];
 200                         mdy = (mi->precision_mval[1] + (float)(mval[1] - mi->precision_mval[1]) / 10.0f) - data[3];
 201
 202                         distance = (length != 0.0f)? (mdx*dx + mdy*dy) / length: 0.0f;
 203                 }
 204                 else {
 205                         int mdx, mdy;
 206                         mdx = mval[0] - data[2];
 207                         mdy = mval[1] - data[3];
 208
 209                         distance = (length != 0.0f)? (mdx*dx + mdy*dy) / length: 0.0f;
 210                 }
 211
 212                 output[0] = (float)((length != 0.0f)? distance / length: 0.0f);
 213         }
 214 }
 215
 216 static void InputAngle(TransInfo *UNUSED(t), MouseInput *mi, const int mval[2], float output[3])
 217 {
 218         double dx2 = mval[0] - mi->center[0];
 219         double dy2 = mval[1] - mi->center[1];
 220         double B = sqrt(dx2*dx2+dy2*dy2);
 221
 222         double dx1 = mi->imval[0] - mi->center[0];
 223         double dy1 = mi->imval[1] - mi->center[1];
 224         double A = sqrt(dx1*dx1+dy1*dy1);
 225
 226         double dx3 = mval[0] - mi->imval[0];
 227         double dy3 = mval[1] - mi->imval[1];
 228
 229         double *angle = mi->data;
 230
 231         /* use doubles here, to make sure a "1.0" (no rotation) doesnt become 9.999999e-01, which gives 0.02 for acos */
 232         double deler = ((dx1*dx1+dy1*dy1)+(dx2*dx2+dy2*dy2)-(dx3*dx3+dy3*dy3))
 233                 / (2.0 * ((A*B)?(A*B):1.0));
 234         /* ((A*B)?(A*B):1.0) this takes care of potential divide by zero errors */
 235
 236         float dphi;
 237
 238         dphi = saacos((float)deler);
 239         if ( (dx1*dy2-dx2*dy1)>0.0 ) dphi= -dphi;
 240
 241         /* If the angle is zero, because of lack of precision close to the 1.0 value in acos
 242          * approximate the angle with the opposite side of the normalized triangle
 243          * This is a good approximation here since the smallest acos value seems to be around
 244          * 0.02 degree and lower values don't even have a 0.01% error compared to the approximation
 245          */
 246         if (dphi == 0) {
 247                 double dx, dy;
 248
 249                 dx2 /= A;
 250                 dy2 /= A;
 251
 252                 dx1 /= B;
 253                 dy1 /= B;
 254
 255                 dx = dx1 - dx2;
 256                 dy = dy1 - dy2;
 257
 258                 dphi = sqrt(dx*dx + dy*dy);
 259                 if ( (dx1*dy2-dx2*dy1)>0.0 ) dphi= -dphi;
 260         }
 261
 262         if (mi->precision) {
 263                 dphi = dphi/30.0f;
 264         }
 265
 266         /* if no delta angle, don't update initial position */
 267         if (dphi != 0) {
 268                 mi->imval[0] = mval[0];
 269                 mi->imval[1] = mval[1];
 270         }
 271
 272         *angle += (double)dphi;
 273
 274         output[0] = *angle;
 275 }
 276
 277 void initMouseInput(TransInfo *UNUSED(t), MouseInput *mi, int center[2], int mval[2])
 278 {
 279         mi->factor = 0;
 280         mi->precision = 0;
 281
 282         mi->center[0] = center[0];
 283         mi->center[1] = center[1];
 284
 285         mi->imval[0] = mval[0];
 286         mi->imval[1] = mval[1];
 287
 288         mi->post = NULL;
 289 }
 290
 291 static void calcSpringFactor(MouseInput *mi)
 292 {
 293         mi->factor = (float)sqrt(
 294                 (
 295                         ((float)(mi->center[1] - mi->imval[1]))*((float)(mi->center[1] - mi->imval[1]))
 296                 +
 297                         ((float)(mi->center[0] - mi->imval[0]))*((float)(mi->center[0] - mi->imval[0]))
 298                 ) );
 299
 300         if (mi->factor==0.0f)
 301                 mi->factor= 1.0f; /* prevent Inf */
 302 }
 303
 304 void initMouseInputMode(TransInfo *t, MouseInput *mi, MouseInputMode mode)
 305 {
 306         /* may have been allocated previously */
 307         /* TODO, holding R-key can cause mem leak, but this causes [#28903]
 308          * disable for now. */
 309 #if 0
 310         if (mi->data) {
 311                 MEM_freeN(mi->data);
 312                 mi->data= NULL;
 313         }
 314 #endif
 315
 316         switch (mode) {
 317         case INPUT_VECTOR:
 318                 mi->apply = InputVector;
 319                 t->helpline = HLP_NONE;
 320                 break;
 321         case INPUT_SPRING:
 322                 calcSpringFactor(mi);
 323                 mi->apply = InputSpring;
 324                 t->helpline = HLP_SPRING;
 325                 break;
 326         case INPUT_SPRING_FLIP:
 327                 calcSpringFactor(mi);
 328                 mi->apply = InputSpringFlip;
 329                 t->helpline = HLP_SPRING;
 330                 break;
 331         case INPUT_ANGLE:
 332                 mi->data = MEM_callocN(sizeof(double), "angle accumulator");
 333                 mi->apply = InputAngle;
 334                 t->helpline = HLP_ANGLE;
 335                 break;
 336         case INPUT_TRACKBALL:
 337                 /* factor has to become setting or so */
 338                 mi->factor = 0.01f;
 339                 mi->apply = InputTrackBall;
 340                 t->helpline = HLP_TRACKBALL;
 341                 break;
 342         case INPUT_HORIZONTAL_RATIO:
 343                 mi->factor = (float)(mi->center[0] - mi->imval[0]);
 344                 mi->apply = InputHorizontalRatio;
 345                 t->helpline = HLP_HARROW;
 346                 break;
 347         case INPUT_HORIZONTAL_ABSOLUTE:
 348                 mi->apply = InputHorizontalAbsolute;
 349                 t->helpline = HLP_HARROW;
 350                 break;
 351         case INPUT_VERTICAL_RATIO:
 352                 mi->apply = InputVerticalRatio;
 353                 t->helpline = HLP_VARROW;
 354                 break;
 355         case INPUT_VERTICAL_ABSOLUTE:
 356                 mi->apply = InputVerticalAbsolute;
 357                 t->helpline = HLP_VARROW;
 358                 break;
 359         case INPUT_CUSTOM_RATIO:
 360                 mi->apply = InputCustomRatio;
 361                 t->helpline = HLP_NONE;
 362                 break;
 363         case INPUT_NONE:
 364         default:
 365                 mi->apply = NULL;
 366                 break;
 367         }
 368
 369         /* bootstrap mouse input with initial values */
 370         applyMouseInput(t, mi, mi->imval, t->values);
 371 }
 372
 373 void setInputPostFct(MouseInput *mi, void       (*post)(struct TransInfo *, float [3]))
 374 {
 375         mi->post = post;
 376 }
 377
 378 void applyMouseInput(TransInfo *t, MouseInput *mi, const int mval[2], float output[3])
 379 {
 380         if (mi->apply != NULL) {
 381                 mi->apply(t, mi, mval, output);
 382         }
 383
 384         if (mi->post) {
 385                 mi->post(t, output);
 386         }
 387 }
 388
 389 int handleMouseInput(TransInfo *t, MouseInput *mi, wmEvent *event)
 390 {
 391         int redraw = TREDRAW_NOTHING;
 392
 393         switch (event->type) {
 394         case LEFTSHIFTKEY:
 395         case RIGHTSHIFTKEY:
 396                 if (event->val == KM_PRESS) {
 397                         t->modifiers |= MOD_PRECISION;
 398                         /* shift is modifier for higher precision transform
 399                          * store the mouse position where the normal movement ended */
 400                         copy_v2_v2_int(mi->precision_mval, event->mval);
 401                         mi->precision = 1;
 402                 }
 403                 else {
 404                         t->modifiers &= ~MOD_PRECISION;
 405                         mi->precision = 0;
 406                 }
 407                 redraw = TREDRAW_HARD;
 408                 break;
 409         }
 410
 411         return redraw;
 412 }