bf6dfe3938b7482ba78eb9ca9efedfdb90d84fd4
[blender.git] / source / blender / editors / animation / anim_draw.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) 2008 Blender Foundation.
17  * All rights reserved.
18  */
19
20 /** \file
21  * \ingroup edanimation
22  */
23
24 #include "BLI_sys_types.h"
25
26 #include "DNA_anim_types.h"
27 #include "DNA_scene_types.h"
28 #include "DNA_space_types.h"
29 #include "DNA_userdef_types.h"
30 #include "DNA_screen_types.h"
31 #include "DNA_object_types.h"
32 #include "DNA_gpencil_types.h"
33 #include "DNA_mask_types.h"
34
35 #include "BLI_math.h"
36 #include "BLI_timecode.h"
37 #include "BLI_utildefines.h"
38 #include "BLI_rect.h"
39 #include "BLI_dlrbTree.h"
40
41 #include "BKE_context.h"
42 #include "BKE_curve.h"
43 #include "BKE_fcurve.h"
44 #include "BKE_global.h"
45 #include "BKE_nla.h"
46 #include "BKE_mask.h"
47
48 #include "ED_anim_api.h"
49 #include "ED_keyframes_edit.h"
50 #include "ED_keyframes_draw.h"
51
52 #include "RNA_access.h"
53
54 #include "UI_interface.h"
55 #include "UI_resources.h"
56 #include "UI_view2d.h"
57
58 #include "GPU_immediate.h"
59 #include "GPU_matrix.h"
60 #include "GPU_state.h"
61
62 /* *************************************************** */
63 /* CURRENT FRAME DRAWING */
64
65 /* Draw current frame number in a little green box beside the current frame indicator */
66 void ANIM_draw_cfra_number(const bContext *C, View2D *v2d, short flag)
67 {
68   Scene *scene = CTX_data_scene(C);
69   const float time = scene->r.cfra + scene->r.subframe;
70   const float cfra = (float)(time * scene->r.framelen);
71   const bool show_time = (flag & DRAWCFRA_UNIT_SECONDS) != 0;
72
73   const uiFontStyle *fstyle = UI_FSTYLE_WIDGET;
74   unsigned char col[4];
75   float color[4];
76   float xscale, x, y;
77   char numstr[32] = "  t  "; /* t is the character to start replacing from */
78   float hlen;
79   int slen;
80
81   /* because the frame number text is subject to the same scaling as the contents of the view */
82   UI_view2d_scale_get(v2d, &xscale, NULL);
83   GPU_matrix_push();
84   GPU_matrix_scale_2f(1.0f / xscale, 1.0f);
85
86   /* get timecode string
87    * - padding on str-buf passed so that it doesn't sit on the frame indicator
88    */
89   if (show_time) {
90     BLI_timecode_string_from_time(
91         &numstr[2], sizeof(numstr) - 2, 0, FRA2TIME(cfra), FPS, U.timecode_style);
92   }
93   else {
94     BLI_timecode_string_from_time_seconds(&numstr[2], sizeof(numstr) - 2, 1, cfra);
95   }
96
97   slen = UI_fontstyle_string_width(fstyle, numstr) - 1;
98   hlen = slen * 0.5f;
99
100   /* get starting coordinates for drawing */
101   x = cfra * xscale;
102   y = -0.1f * U.widget_unit;
103
104   /* draw green box around/behind text */
105   UI_GetThemeColor4fv(TH_CFRAME, color);
106   color[3] = 3.0f;
107
108   UI_draw_roundbox_corner_set(UI_CNR_ALL);
109   UI_draw_roundbox_aa(true,
110                       x - hlen - 0.1f * U.widget_unit,
111                       y + 3.0f,
112                       x + hlen + 0.1f * U.widget_unit,
113                       y - 3.0f + U.widget_unit,
114                       0.1f * U.widget_unit,
115                       color);
116
117   /* draw current frame number */
118   UI_GetThemeColor4ubv(TH_TEXT_HI, col);
119   UI_fontstyle_draw_simple(
120       fstyle, x - hlen - 0.15f * U.widget_unit, y + 0.28f * U.widget_unit, numstr, col);
121
122   /* restore view transform */
123   GPU_matrix_pop();
124 }
125
126 /* General call for drawing current frame indicator in animation editor */
127 void ANIM_draw_cfra(const bContext *C, View2D *v2d, short flag)
128 {
129   Scene *scene = CTX_data_scene(C);
130
131   const float time = scene->r.cfra + scene->r.subframe;
132   const float x = (float)(time * scene->r.framelen);
133
134   GPU_line_width((flag & DRAWCFRA_WIDE) ? 3.0 : 2.0);
135
136   GPUVertFormat *format = immVertexFormat();
137   uint pos = GPU_vertformat_attr_add(format, "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
138
139   immBindBuiltinProgram(GPU_SHADER_2D_UNIFORM_COLOR);
140
141   /* Draw a light green line to indicate current frame */
142   immUniformThemeColor(TH_CFRAME);
143
144   immBegin(GPU_PRIM_LINES, 2);
145   immVertex2f(pos, x, v2d->cur.ymin - 500.0f); /* XXX arbitrary... want it go to bottom */
146   immVertex2f(pos, x, v2d->cur.ymax);
147   immEnd();
148   immUnbindProgram();
149 }
150
151 /* *************************************************** */
152 /* PREVIEW RANGE 'CURTAINS' */
153 /* Note: 'Preview Range' tools are defined in anim_ops.c */
154
155 /* Draw preview range 'curtains' for highlighting where the animation data is */
156 void ANIM_draw_previewrange(const bContext *C, View2D *v2d, int end_frame_width)
157 {
158   Scene *scene = CTX_data_scene(C);
159
160   /* only draw this if preview range is set */
161   if (PRVRANGEON) {
162     GPU_blend_set_func_separate(
163         GPU_SRC_ALPHA, GPU_ONE_MINUS_SRC_ALPHA, GPU_ONE, GPU_ONE_MINUS_SRC_ALPHA);
164     GPU_blend(true);
165
166     GPUVertFormat *format = immVertexFormat();
167     uint pos = GPU_vertformat_attr_add(format, "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
168
169     immBindBuiltinProgram(GPU_SHADER_2D_UNIFORM_COLOR);
170     immUniformThemeColorShadeAlpha(TH_ANIM_PREVIEW_RANGE, -25, -30);
171     /* XXX: Fix this hardcoded color (anim_active) */
172     //immUniformColor4f(0.8f, 0.44f, 0.1f, 0.2f);
173
174     /* only draw two separate 'curtains' if there's no overlap between them */
175     if (PSFRA < PEFRA + end_frame_width) {
176       immRectf(pos, v2d->cur.xmin, v2d->cur.ymin, (float)PSFRA, v2d->cur.ymax);
177       immRectf(pos, (float)(PEFRA + end_frame_width), v2d->cur.ymin, v2d->cur.xmax, v2d->cur.ymax);
178     }
179     else {
180       immRectf(pos, v2d->cur.xmin, v2d->cur.ymin, v2d->cur.xmax, v2d->cur.ymax);
181     }
182
183     immUnbindProgram();
184
185     GPU_blend(false);
186   }
187 }
188
189 /* *************************************************** */
190 /* SCENE FRAME RANGE */
191
192 /* Draw frame range guides (for scene frame range) in background */
193 // TODO: Should we still show these when preview range is enabled?
194 void ANIM_draw_framerange(Scene *scene, View2D *v2d)
195 {
196   /* draw darkened area outside of active timeline frame range */
197   GPU_blend_set_func_separate(
198       GPU_SRC_ALPHA, GPU_ONE_MINUS_SRC_ALPHA, GPU_ONE, GPU_ONE_MINUS_SRC_ALPHA);
199   GPU_blend(true);
200
201   GPUVertFormat *format = immVertexFormat();
202   uint pos = GPU_vertformat_attr_add(format, "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
203
204   immBindBuiltinProgram(GPU_SHADER_2D_UNIFORM_COLOR);
205   immUniformThemeColorShadeAlpha(TH_BACK, -25, -100);
206
207   if (SFRA < EFRA) {
208     immRectf(pos, v2d->cur.xmin, v2d->cur.ymin, (float)SFRA, v2d->cur.ymax);
209     immRectf(pos, (float)EFRA, v2d->cur.ymin, v2d->cur.xmax, v2d->cur.ymax);
210   }
211   else {
212     immRectf(pos, v2d->cur.xmin, v2d->cur.ymin, v2d->cur.xmax, v2d->cur.ymax);
213   }
214
215   GPU_blend(false);
216
217   /* thin lines where the actual frames are */
218   immUniformThemeColorShade(TH_BACK, -60);
219
220   immBegin(GPU_PRIM_LINES, 4);
221
222   immVertex2f(pos, (float)SFRA, v2d->cur.ymin);
223   immVertex2f(pos, (float)SFRA, v2d->cur.ymax);
224
225   immVertex2f(pos, (float)EFRA, v2d->cur.ymin);
226   immVertex2f(pos, (float)EFRA, v2d->cur.ymax);
227
228   immEnd();
229   immUnbindProgram();
230 }
231
232 /* *************************************************** */
233 /* NLA-MAPPING UTILITIES (required for drawing and also editing keyframes)  */
234
235 /* Obtain the AnimData block providing NLA-mapping for the given channel (if applicable) */
236 // TODO: do not supply return this if the animdata tells us that there is no mapping to perform
237 AnimData *ANIM_nla_mapping_get(bAnimContext *ac, bAnimListElem *ale)
238 {
239   /* sanity checks */
240   if (ac == NULL)
241     return NULL;
242
243   /* abort if rendering - we may get some race condition issues... */
244   if (G.is_rendering)
245     return NULL;
246
247   /* apart from strictly keyframe-related contexts, this shouldn't even happen */
248   // XXX: nla and channel here may not be necessary...
249   if (ELEM(ac->datatype,
250            ANIMCONT_ACTION,
251            ANIMCONT_SHAPEKEY,
252            ANIMCONT_DOPESHEET,
253            ANIMCONT_FCURVES,
254            ANIMCONT_NLA,
255            ANIMCONT_CHANNEL)) {
256     /* handling depends on the type of animation-context we've got */
257     if (ale) {
258       /* NLA Control Curves occur on NLA strips,
259        * and shouldn't be subjected to this kind of mapping. */
260       if (ale->type != ANIMTYPE_NLACURVE)
261         return ale->adt;
262     }
263   }
264
265   /* cannot handle... */
266   return NULL;
267 }
268
269 /* ------------------- */
270
271 /* Helper function for ANIM_nla_mapping_apply_fcurve() -> "restore",
272  * i.e. mapping points back to action-time. */
273 static short bezt_nlamapping_restore(KeyframeEditData *ked, BezTriple *bezt)
274 {
275   /* AnimData block providing scaling is stored in 'data', only_keys option is stored in i1 */
276   AnimData *adt = (AnimData *)ked->data;
277   short only_keys = (short)ked->i1;
278
279   /* adjust BezTriple handles only if allowed to */
280   if (only_keys == 0) {
281     bezt->vec[0][0] = BKE_nla_tweakedit_remap(adt, bezt->vec[0][0], NLATIME_CONVERT_UNMAP);
282     bezt->vec[2][0] = BKE_nla_tweakedit_remap(adt, bezt->vec[2][0], NLATIME_CONVERT_UNMAP);
283   }
284
285   bezt->vec[1][0] = BKE_nla_tweakedit_remap(adt, bezt->vec[1][0], NLATIME_CONVERT_UNMAP);
286
287   return 0;
288 }
289
290 /* helper function for ANIM_nla_mapping_apply_fcurve() -> "apply",
291  * i.e. mapping points to NLA-mapped global time */
292 static short bezt_nlamapping_apply(KeyframeEditData *ked, BezTriple *bezt)
293 {
294   /* AnimData block providing scaling is stored in 'data', only_keys option is stored in i1 */
295   AnimData *adt = (AnimData *)ked->data;
296   short only_keys = (short)ked->i1;
297
298   /* adjust BezTriple handles only if allowed to */
299   if (only_keys == 0) {
300     bezt->vec[0][0] = BKE_nla_tweakedit_remap(adt, bezt->vec[0][0], NLATIME_CONVERT_MAP);
301     bezt->vec[2][0] = BKE_nla_tweakedit_remap(adt, bezt->vec[2][0], NLATIME_CONVERT_MAP);
302   }
303
304   bezt->vec[1][0] = BKE_nla_tweakedit_remap(adt, bezt->vec[1][0], NLATIME_CONVERT_MAP);
305
306   return 0;
307 }
308
309 /* Apply/Unapply NLA mapping to all keyframes in the nominated F-Curve
310  * - restore = whether to map points back to non-mapped time
311  * - only_keys = whether to only adjust the location of the center point of beztriples
312  */
313 void ANIM_nla_mapping_apply_fcurve(AnimData *adt, FCurve *fcu, bool restore, bool only_keys)
314 {
315   KeyframeEditData ked = {{NULL}};
316   KeyframeEditFunc map_cb;
317
318   /* init edit data
319    * - AnimData is stored in 'data'
320    * - only_keys is stored in 'i1'
321    */
322   ked.data = (void *)adt;
323   ked.i1 = (int)only_keys;
324
325   /* get editing callback */
326   if (restore)
327     map_cb = bezt_nlamapping_restore;
328   else
329     map_cb = bezt_nlamapping_apply;
330
331   /* apply to F-Curve */
332   ANIM_fcurve_keyframes_loop(&ked, fcu, NULL, map_cb, NULL);
333 }
334
335 /* *************************************************** */
336 /* UNITS CONVERSION MAPPING (required for drawing and editing keyframes) */
337
338 /* Get flags used for normalization in ANIM_unit_mapping_get_factor. */
339 short ANIM_get_normalization_flags(bAnimContext *ac)
340 {
341   if (ac->sl->spacetype == SPACE_GRAPH) {
342     SpaceGraph *sipo = (SpaceGraph *)ac->sl;
343     bool use_normalization = (sipo->flag & SIPO_NORMALIZE) != 0;
344     bool freeze_normalization = (sipo->flag & SIPO_NORMALIZE_FREEZE) != 0;
345     return use_normalization ? (ANIM_UNITCONV_NORMALIZE |
346                                 (freeze_normalization ? ANIM_UNITCONV_NORMALIZE_FREEZE : 0)) :
347                                0;
348   }
349
350   return 0;
351 }
352
353 static float normalization_factor_get(Scene *scene, FCurve *fcu, short flag, float *r_offset)
354 {
355   float factor = 1.0f, offset = 0.0f;
356
357   if (flag & ANIM_UNITCONV_RESTORE) {
358     if (r_offset)
359       *r_offset = fcu->prev_offset;
360
361     return 1.0f / fcu->prev_norm_factor;
362   }
363
364   if (flag & ANIM_UNITCONV_NORMALIZE_FREEZE) {
365     if (r_offset)
366       *r_offset = fcu->prev_offset;
367     if (fcu->prev_norm_factor == 0.0f) {
368       /* Happens when Auto Normalize was disabled before
369        * any curves were displayed.
370        */
371       return 1.0f;
372     }
373     return fcu->prev_norm_factor;
374   }
375
376   if (G.moving & G_TRANSFORM_FCURVES) {
377     if (r_offset)
378       *r_offset = fcu->prev_offset;
379     if (fcu->prev_norm_factor == 0.0f) {
380       /* Same as above. */
381       return 1.0f;
382     }
383     return fcu->prev_norm_factor;
384   }
385
386   fcu->prev_norm_factor = 1.0f;
387   if (fcu->bezt) {
388     const bool use_preview_only = PRVRANGEON;
389     const BezTriple *bezt;
390     int i;
391     float max_coord = -FLT_MAX;
392     float min_coord = FLT_MAX;
393     float range;
394
395     if (fcu->totvert < 1) {
396       return 1.0f;
397     }
398
399     for (i = 0, bezt = fcu->bezt; i < fcu->totvert; i++, bezt++) {
400       if (use_preview_only && !IN_RANGE_INCL(bezt->vec[1][0], scene->r.psfra, scene->r.pefra)) {
401         continue;
402       }
403
404       if (i == 0) {
405         /* We ignore extrapolation flags and handle here, and use the
406          * control point position only. so we normalize "interesting"
407          * part of the curve.
408          *
409          * Here we handle left extrapolation.
410          */
411         max_coord = max_ff(max_coord, bezt->vec[1][1]);
412
413         min_coord = min_ff(min_coord, bezt->vec[1][1]);
414       }
415       else {
416         const BezTriple *prev_bezt = bezt - 1;
417         if (prev_bezt->ipo == BEZT_IPO_CONST) {
418           /* Constant interpolation: previous CV value is used up
419            * to the current keyframe.
420            */
421           max_coord = max_ff(max_coord, bezt->vec[1][1]);
422           min_coord = min_ff(min_coord, bezt->vec[1][1]);
423         }
424         else if (prev_bezt->ipo == BEZT_IPO_LIN) {
425           /* Linear interpolation: min/max using both previous and
426            * and current CV.
427            */
428           max_coord = max_ff(max_coord, bezt->vec[1][1]);
429           min_coord = min_ff(min_coord, bezt->vec[1][1]);
430           max_coord = max_ff(max_coord, prev_bezt->vec[1][1]);
431           min_coord = min_ff(min_coord, prev_bezt->vec[1][1]);
432         }
433         else if (prev_bezt->ipo == BEZT_IPO_BEZ) {
434           const int resol = fcu->driver ?
435                                 32 :
436                                 min_ii((int)(5.0f * len_v2v2(bezt->vec[1], prev_bezt->vec[1])),
437                                        32);
438           if (resol < 2) {
439             max_coord = max_ff(max_coord, prev_bezt->vec[1][1]);
440             min_coord = min_ff(min_coord, prev_bezt->vec[1][1]);
441           }
442           else {
443             float data[120];
444             float v1[2], v2[2], v3[2], v4[2];
445
446             v1[0] = prev_bezt->vec[1][0];
447             v1[1] = prev_bezt->vec[1][1];
448             v2[0] = prev_bezt->vec[2][0];
449             v2[1] = prev_bezt->vec[2][1];
450
451             v3[0] = bezt->vec[0][0];
452             v3[1] = bezt->vec[0][1];
453             v4[0] = bezt->vec[1][0];
454             v4[1] = bezt->vec[1][1];
455
456             correct_bezpart(v1, v2, v3, v4);
457
458             BKE_curve_forward_diff_bezier(
459                 v1[0], v2[0], v3[0], v4[0], data, resol, sizeof(float) * 3);
460             BKE_curve_forward_diff_bezier(
461                 v1[1], v2[1], v3[1], v4[1], data + 1, resol, sizeof(float) * 3);
462
463             for (int j = 0; j <= resol; ++j) {
464               const float *fp = &data[j * 3];
465               max_coord = max_ff(max_coord, fp[1]);
466               min_coord = min_ff(min_coord, fp[1]);
467             }
468           }
469         }
470       }
471     }
472
473     if (max_coord > min_coord) {
474       range = max_coord - min_coord;
475       if (range > FLT_EPSILON) {
476         factor = 2.0f / range;
477       }
478       offset = -min_coord - range / 2.0f;
479     }
480     else if (max_coord == min_coord) {
481       factor = 1.0f;
482       offset = -min_coord;
483     }
484   }
485   BLI_assert(factor != 0.0f);
486   if (r_offset) {
487     *r_offset = offset;
488   }
489
490   fcu->prev_norm_factor = factor;
491   fcu->prev_offset = offset;
492   return factor;
493 }
494
495 /* Get unit conversion factor for given ID + F-Curve */
496 float ANIM_unit_mapping_get_factor(Scene *scene, ID *id, FCurve *fcu, short flag, float *r_offset)
497 {
498   if (flag & ANIM_UNITCONV_NORMALIZE) {
499     return normalization_factor_get(scene, fcu, flag, r_offset);
500   }
501
502   if (r_offset)
503     *r_offset = 0.0f;
504
505   /* sanity checks */
506   if (id && fcu && fcu->rna_path) {
507     PointerRNA ptr, id_ptr;
508     PropertyRNA *prop;
509
510     /* get RNA property that F-Curve affects */
511     RNA_id_pointer_create(id, &id_ptr);
512     if (RNA_path_resolve_property(&id_ptr, fcu->rna_path, &ptr, &prop)) {
513       /* rotations: radians <-> degrees? */
514       if (RNA_SUBTYPE_UNIT(RNA_property_subtype(prop)) == PROP_UNIT_ROTATION) {
515         /* if the radians flag is not set, default to using degrees which need conversions */
516         if ((scene) && (scene->unit.system_rotation == USER_UNIT_ROT_RADIANS) == 0) {
517           if (flag & ANIM_UNITCONV_RESTORE)
518             return DEG2RADF(1.0f); /* degrees to radians */
519           else
520             return RAD2DEGF(1.0f); /* radians to degrees */
521         }
522       }
523
524       /* TODO: other rotation types here as necessary */
525     }
526   }
527
528   /* no mapping needs to occur... */
529   return 1.0f;
530 }
531
532 static bool find_prev_next_keyframes(struct bContext *C, int *nextfra, int *prevfra)
533 {
534   Scene *scene = CTX_data_scene(C);
535   Object *ob = CTX_data_active_object(C);
536   Mask *mask = CTX_data_edit_mask(C);
537   bDopeSheet ads = {NULL};
538   DLRBT_Tree keys;
539   ActKeyColumn *aknext, *akprev;
540   float cfranext, cfraprev;
541   bool donenext = false, doneprev = false;
542   int nextcount = 0, prevcount = 0;
543
544   cfranext = cfraprev = (float)(CFRA);
545
546   /* init binarytree-list for getting keyframes */
547   BLI_dlrbTree_init(&keys);
548
549   /* seed up dummy dopesheet context with flags to perform necessary filtering */
550   if ((scene->flag & SCE_KEYS_NO_SELONLY) == 0) {
551     /* only selected channels are included */
552     ads.filterflag |= ADS_FILTER_ONLYSEL;
553   }
554
555   /* populate tree with keyframe nodes */
556   scene_to_keylist(&ads, scene, &keys, 0);
557   gpencil_to_keylist(&ads, scene->gpd, &keys, false);
558
559   if (ob) {
560     ob_to_keylist(&ads, ob, &keys, 0);
561     gpencil_to_keylist(&ads, ob->data, &keys, false);
562   }
563
564   if (mask) {
565     MaskLayer *masklay = BKE_mask_layer_active(mask);
566     mask_to_keylist(&ads, masklay, &keys);
567   }
568
569   /* find matching keyframe in the right direction */
570   do {
571     aknext = (ActKeyColumn *)BLI_dlrbTree_search_next(&keys, compare_ak_cfraPtr, &cfranext);
572
573     if (aknext) {
574       if (CFRA == (int)aknext->cfra) {
575         /* make this the new starting point for the search and ignore */
576         cfranext = aknext->cfra;
577       }
578       else {
579         /* this changes the frame, so set the frame and we're done */
580         if (++nextcount == U.view_frame_keyframes)
581           donenext = true;
582       }
583       cfranext = aknext->cfra;
584     }
585   } while ((aknext != NULL) && (donenext == false));
586
587   do {
588     akprev = (ActKeyColumn *)BLI_dlrbTree_search_prev(&keys, compare_ak_cfraPtr, &cfraprev);
589
590     if (akprev) {
591       if (CFRA == (int)akprev->cfra) {
592         /* make this the new starting point for the search */
593       }
594       else {
595         /* this changes the frame, so set the frame and we're done */
596         if (++prevcount == U.view_frame_keyframes)
597           doneprev = true;
598       }
599       cfraprev = akprev->cfra;
600     }
601   } while ((akprev != NULL) && (doneprev == false));
602
603   /* free temp stuff */
604   BLI_dlrbTree_free(&keys);
605
606   /* any success? */
607   if (doneprev || donenext) {
608     if (doneprev)
609       *prevfra = cfraprev;
610     else
611       *prevfra = CFRA - (cfranext - CFRA);
612
613     if (donenext)
614       *nextfra = cfranext;
615     else
616       *nextfra = CFRA + (CFRA - cfraprev);
617
618     return true;
619   }
620
621   return false;
622 }
623
624 void ANIM_center_frame(struct bContext *C, int smooth_viewtx)
625 {
626   ARegion *ar = CTX_wm_region(C);
627   Scene *scene = CTX_data_scene(C);
628   float w = BLI_rctf_size_x(&ar->v2d.cur);
629   rctf newrct;
630   int nextfra, prevfra;
631
632   switch (U.view_frame_type) {
633     case ZOOM_FRAME_MODE_SECONDS: {
634       const float fps = FPS;
635       newrct.xmax = scene->r.cfra + U.view_frame_seconds * fps + 1;
636       newrct.xmin = scene->r.cfra - U.view_frame_seconds * fps - 1;
637       newrct.ymax = ar->v2d.cur.ymax;
638       newrct.ymin = ar->v2d.cur.ymin;
639       break;
640     }
641
642     /* hardest case of all, look for all keyframes around frame and display those */
643     case ZOOM_FRAME_MODE_KEYFRAMES:
644       if (find_prev_next_keyframes(C, &nextfra, &prevfra)) {
645         newrct.xmax = nextfra;
646         newrct.xmin = prevfra;
647         newrct.ymax = ar->v2d.cur.ymax;
648         newrct.ymin = ar->v2d.cur.ymin;
649         break;
650       }
651       /* else drop through, keep range instead */
652       ATTR_FALLTHROUGH;
653
654     case ZOOM_FRAME_MODE_KEEP_RANGE:
655     default:
656       newrct.xmax = scene->r.cfra + (w / 2);
657       newrct.xmin = scene->r.cfra - (w / 2);
658       newrct.ymax = ar->v2d.cur.ymax;
659       newrct.ymin = ar->v2d.cur.ymin;
660       break;
661   }
662
663   UI_view2d_smooth_view(C, ar, &newrct, smooth_viewtx);
664 }
665 /* *************************************************** */