Merge branch 'master' into blender2.8
[blender.git] / source / blender / editors / space_graph / graph_draw.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  * The Original Code is Copyright (C) Blender Foundation
19  *
20  * Contributor(s): Joshua Leung (2009 Recode)
21  *
22  * ***** END GPL LICENSE BLOCK *****
23  */
24
25 /** \file blender/editors/space_graph/graph_draw.c
26  *  \ingroup spgraph
27  */
28
29
30 #include <stdio.h>
31 #include <math.h>
32 #include <string.h>
33 #include <float.h>
34
35 #include "BLI_blenlib.h"
36 #include "BLI_math.h"
37 #include "BLI_utildefines.h"
38
39 #include "DNA_anim_types.h"
40 #include "DNA_screen_types.h"
41 #include "DNA_space_types.h"
42 #include "DNA_windowmanager_types.h"
43 #include "DNA_userdef_types.h"
44
45 #include "BKE_context.h"
46 #include "BKE_curve.h"
47 #include "BKE_fcurve.h"
48
49 #include "BIF_glutil.h"
50
51 #include "GPU_draw.h"
52 #include "GPU_immediate.h"
53 #include "GPU_matrix.h"
54 #include "GPU_state.h"
55
56 #include "ED_anim_api.h"
57
58 #include "graph_intern.h"
59
60 #include "UI_interface.h"
61 #include "UI_resources.h"
62 #include "UI_view2d.h"
63
64 /* *************************** */
65 /* Utility Drawing Defines */
66
67 /* determine the alpha value that should be used when
68  * drawing components for some F-Curve (fcu)
69  * - selected F-Curves should be more visible than partially visible ones
70  */
71 static float fcurve_display_alpha(FCurve *fcu)
72 {
73         return (fcu->flag & FCURVE_SELECTED) ? 1.0f : U.fcu_inactive_alpha;
74 }
75
76 /* *************************** */
77 /* F-Curve Modifier Drawing */
78
79 /* Envelope -------------- */
80
81 /* TODO: draw a shaded poly showing the region of influence too!!! */
82 static void draw_fcurve_modifier_controls_envelope(FModifier *fcm, View2D *v2d)
83 {
84         FMod_Envelope *env = (FMod_Envelope *)fcm->data;
85         FCM_EnvelopeData *fed;
86         const float fac = 0.05f * BLI_rctf_size_x(&v2d->cur);
87         int i;
88
89         const uint shdr_pos = GPU_vertformat_attr_add(immVertexFormat(), "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
90
91         GPU_line_width(1.0f);
92
93         immBindBuiltinProgram(GPU_SHADER_2D_LINE_DASHED_UNIFORM_COLOR);
94
95         float viewport_size[4];
96         GPU_viewport_size_get_f(viewport_size);
97         immUniform2f("viewport_size", viewport_size[2] / UI_DPI_FAC, viewport_size[3] / UI_DPI_FAC);
98
99         immUniform1i("colors_len", 0);  /* Simple dashes. */
100         immUniformColor3f(0.0f, 0.0f, 0.0f);
101         immUniform1f("dash_width", 10.0f);
102         immUniform1f("dash_factor", 0.5f);
103
104         /* draw two black lines showing the standard reference levels */
105
106         immBegin(GPU_PRIM_LINES, 4);
107         immVertex2f(shdr_pos, v2d->cur.xmin, env->midval + env->min);
108         immVertex2f(shdr_pos, v2d->cur.xmax, env->midval + env->min);
109
110         immVertex2f(shdr_pos, v2d->cur.xmin, env->midval + env->max);
111         immVertex2f(shdr_pos, v2d->cur.xmax, env->midval + env->max);
112         immEnd();
113
114         immUnbindProgram();
115
116         if (env->totvert > 0) {
117                 /* set size of vertices (non-adjustable for now) */
118                 GPU_point_size(2.0f);
119
120                 immBindBuiltinProgram(GPU_SHADER_2D_UNIFORM_COLOR);
121
122                 /* for now, point color is fixed, and is white */
123                 immUniformColor3f(1.0f, 1.0f, 1.0f);
124
125                 immBeginAtMost(GPU_PRIM_POINTS, env->totvert * 2);
126
127                 for (i = 0, fed = env->data; i < env->totvert; i++, fed++) {
128                         /* only draw if visible
129                          * - min/max here are fixed, not relative
130                          */
131                         if (IN_RANGE(fed->time, (v2d->cur.xmin - fac), (v2d->cur.xmax + fac))) {
132                                 immVertex2f(shdr_pos, fed->time, fed->min);
133                                 immVertex2f(shdr_pos, fed->time, fed->max);
134                         }
135                 }
136
137                 immEnd();
138
139                 immUnbindProgram();
140         }
141 }
142
143 /* *************************** */
144 /* F-Curve Drawing */
145
146 /* Points ---------------- */
147
148 /* helper func - set color to draw F-Curve data with */
149 static void set_fcurve_vertex_color(FCurve *fcu, bool sel)
150 {
151         float color[4];
152         float diff;
153
154         /* Set color of curve vertex based on state of curve (i.e. 'Edit' Mode) */
155         if ((fcu->flag & FCURVE_PROTECTED) == 0) {
156                 /* Curve's points ARE BEING edited */
157                 UI_GetThemeColor3fv(sel ? TH_VERTEX_SELECT : TH_VERTEX, color);
158         }
159         else {
160                 /* Curve's points CANNOT BE edited */
161                 UI_GetThemeColor3fv(sel ? TH_TEXT_HI : TH_TEXT, color);
162         }
163
164         /* Fade the 'intensity' of the vertices based on the selection of the curves too
165          * - Only fade by 50% the amount the curves were faded by, so that the points
166          *   still stand out for easier selection
167          */
168         diff = 1.0f - fcurve_display_alpha(fcu);
169         color[3] = 1.0f - (diff * 0.5f);
170         CLAMP(color[3], 0.2f, 1.0f);
171
172         immUniformColor4fv(color);
173 }
174
175 static void draw_fcurve_selected_keyframe_vertices(FCurve *fcu, View2D *v2d, bool edit, bool sel, unsigned pos)
176 {
177         const float fac = 0.05f * BLI_rctf_size_x(&v2d->cur);
178
179         set_fcurve_vertex_color(fcu, sel);
180
181         immBeginAtMost(GPU_PRIM_POINTS, fcu->totvert);
182
183         BezTriple *bezt = fcu->bezt;
184         for (int i = 0; i < fcu->totvert; i++, bezt++) {
185                 /* as an optimization step, only draw those in view
186                  * - we apply a correction factor to ensure that points don't pop in/out due to slight twitches of view size
187                  */
188                 if (IN_RANGE(bezt->vec[1][0], (v2d->cur.xmin - fac), (v2d->cur.xmax + fac))) {
189                         if (edit) {
190                                 /* 'Keyframe' vertex only, as handle lines and handles have already been drawn
191                                  * - only draw those with correct selection state for the current drawing color
192                                  * -
193                                  */
194                                 if ((bezt->f2 & SELECT) == sel)
195                                         immVertex2fv(pos, bezt->vec[1]);
196                         }
197                         else {
198                                 /* no check for selection here, as curve is not editable... */
199                                 /* XXX perhaps we don't want to even draw points?   maybe add an option for that later */
200                                 immVertex2fv(pos, bezt->vec[1]);
201                         }
202                 }
203         }
204
205         immEnd();
206 }
207
208 /* helper func - draw keyframe vertices only for an F-Curve */
209 static void draw_fcurve_keyframe_vertices(FCurve *fcu, View2D *v2d, bool edit, unsigned pos)
210 {
211         immBindBuiltinProgram(GPU_SHADER_2D_POINT_UNIFORM_SIZE_UNIFORM_COLOR_AA);
212
213         immUniform1f("size", UI_GetThemeValuef(TH_VERTEX_SIZE) * U.pixelsize);
214
215         draw_fcurve_selected_keyframe_vertices(fcu, v2d, edit, false, pos);
216         draw_fcurve_selected_keyframe_vertices(fcu, v2d, edit, true, pos);
217
218         immUnbindProgram();
219 }
220
221
222 /* helper func - draw handle vertices only for an F-Curve (if it is not protected) */
223 static void draw_fcurve_selected_handle_vertices(FCurve *fcu, View2D *v2d, bool sel, bool sel_handle_only, unsigned pos)
224 {
225         (void) v2d; /* TODO: use this to draw only points in view */
226
227         /* set handle color */
228         float hcolor[3];
229         UI_GetThemeColor3fv(sel ? TH_HANDLE_VERTEX_SELECT : TH_HANDLE_VERTEX, hcolor);
230         immUniform4f("outlineColor", hcolor[0], hcolor[1], hcolor[2], 1.0f);
231         immUniformColor3fvAlpha(hcolor, 0.01f); /* almost invisible - only keep for smoothness */
232
233         immBeginAtMost(GPU_PRIM_POINTS, fcu->totvert * 2);
234
235         BezTriple *bezt = fcu->bezt;
236         BezTriple *prevbezt = NULL;
237         for (int i = 0; i < fcu->totvert; i++, prevbezt = bezt, bezt++) {
238                 /* Draw the editmode handles for a bezier curve (others don't have handles)
239                  * if their selection status matches the selection status we're drawing for
240                  * - first handle only if previous beztriple was bezier-mode
241                  * - second handle only if current beztriple is bezier-mode
242                  *
243                  * Also, need to take into account whether the keyframe was selected
244                  * if a Graph Editor option to only show handles of selected keys is on.
245                  */
246                 if (!sel_handle_only || BEZT_ISSEL_ANY(bezt)) {
247                         if ((!prevbezt && (bezt->ipo == BEZT_IPO_BEZ)) || (prevbezt && (prevbezt->ipo == BEZT_IPO_BEZ))) {
248                                 if ((bezt->f1 & SELECT) == sel) /* && v2d->cur.xmin < bezt->vec[0][0] < v2d->cur.xmax)*/
249                                         immVertex2fv(pos, bezt->vec[0]);
250                         }
251
252                         if (bezt->ipo == BEZT_IPO_BEZ) {
253                                 if ((bezt->f3 & SELECT) == sel) /* && v2d->cur.xmin < bezt->vec[2][0] < v2d->cur.xmax)*/
254                                         immVertex2fv(pos, bezt->vec[2]);
255                         }
256                 }
257         }
258
259         immEnd();
260 }
261
262 /* helper func - draw handle vertices only for an F-Curve (if it is not protected) */
263 static void draw_fcurve_handle_vertices(FCurve *fcu, View2D *v2d, bool sel_handle_only, unsigned pos)
264 {
265         /* smooth outlines for more consistent appearance */
266         immBindBuiltinProgram(GPU_SHADER_2D_POINT_UNIFORM_SIZE_UNIFORM_COLOR_OUTLINE_AA);
267
268         /* set handle size */
269         immUniform1f("size", (1.4f * UI_GetThemeValuef(TH_HANDLE_VERTEX_SIZE)) * U.pixelsize);
270         immUniform1f("outlineWidth", 1.5f * U.pixelsize);
271
272         draw_fcurve_selected_handle_vertices(fcu, v2d, false, sel_handle_only, pos);
273         draw_fcurve_selected_handle_vertices(fcu, v2d, true, sel_handle_only, pos);
274
275         immUnbindProgram();
276 }
277
278
279 static void draw_fcurve_vertices(ARegion *ar, FCurve *fcu, bool do_handles, bool sel_handle_only)
280 {
281         View2D *v2d = &ar->v2d;
282
283         /* only draw points if curve is visible
284          * - draw unselected points before selected points as separate passes
285          *    to make sure in the case of overlapping points that the selected is always visible
286          * - draw handles before keyframes, so that keyframes will overlap handles (keyframes are more important for users)
287          */
288
289         uint pos = GPU_vertformat_attr_add(immVertexFormat(), "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
290
291         GPU_blend(true);
292         GPU_enable_program_point_size();
293
294         /* draw the two handles first (if they're shown, the curve doesn't have just a single keyframe, and the curve is being edited) */
295         if (do_handles) {
296                 draw_fcurve_handle_vertices(fcu, v2d, sel_handle_only, pos);
297         }
298
299         /* draw keyframes over the handles */
300         draw_fcurve_keyframe_vertices(fcu, v2d, !(fcu->flag & FCURVE_PROTECTED), pos);
301
302         GPU_disable_program_point_size();
303         GPU_blend(false);
304 }
305
306 /* Handles ---------------- */
307
308 static bool draw_fcurve_handles_check(SpaceIpo *sipo, FCurve *fcu)
309 {
310         /* don't draw handle lines if handles are not to be shown */
311         if (    (sipo->flag & SIPO_NOHANDLES) || /* handles shouldn't be shown anywhere */
312                 (fcu->flag & FCURVE_PROTECTED) || /* keyframes aren't editable */
313 #if 0       /* handles can still be selected and handle types set, better draw - campbell */
314                 (fcu->flag & FCURVE_INT_VALUES) || /* editing the handles here will cause weird/incorrect interpolation issues */
315 #endif
316                 ((fcu->grp) && (fcu->grp->flag & AGRP_PROTECTED)) || /* group that curve belongs to is not editable */
317                 (fcu->totvert <= 1) /* do not show handles if there is only 1 keyframe, otherwise they all clump together in an ugly ball */
318                 )
319         {
320                 return false;
321         }
322         else {
323                 return true;
324         }
325 }
326
327 /* draw lines for F-Curve handles only (this is only done in EditMode)
328  * note: draw_fcurve_handles_check must be checked before running this. */
329 static void draw_fcurve_handles(SpaceIpo *sipo, FCurve *fcu)
330 {
331         int sel, b;
332
333         GPUVertFormat *format = immVertexFormat();
334         uint pos = GPU_vertformat_attr_add(format, "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
335         uint color = GPU_vertformat_attr_add(format, "color", GPU_COMP_U8, 4, GPU_FETCH_INT_TO_FLOAT_UNIT);
336         immBindBuiltinProgram(GPU_SHADER_2D_FLAT_COLOR);
337
338         immBeginAtMost(GPU_PRIM_LINES, 4 * 2 * fcu->totvert);
339
340         /* slightly hacky, but we want to draw unselected points before selected ones
341          * so that selected points are clearly visible
342          */
343         for (sel = 0; sel < 2; sel++) {
344                 BezTriple *bezt = fcu->bezt, *prevbezt = NULL;
345                 int basecol = (sel) ? TH_HANDLE_SEL_FREE : TH_HANDLE_FREE;
346                 const float *fp;
347                 unsigned char col[4];
348
349                 for (b = 0; b < fcu->totvert; b++, prevbezt = bezt, bezt++) {
350                         /* if only selected keyframes can get their handles shown,
351                          * check that keyframe is selected
352                          */
353                         if (sipo->flag & SIPO_SELVHANDLESONLY) {
354                                 if (BEZT_ISSEL_ANY(bezt) == 0)
355                                         continue;
356                         }
357
358                         /* draw handle with appropriate set of colors if selection is ok */
359                         if ((bezt->f2 & SELECT) == sel) {
360                                 fp = bezt->vec[0];
361
362                                 /* only draw first handle if previous segment had handles */
363                                 if ((!prevbezt && (bezt->ipo == BEZT_IPO_BEZ)) || (prevbezt && (prevbezt->ipo == BEZT_IPO_BEZ))) {
364                                         UI_GetThemeColor3ubv(basecol + bezt->h1, col);
365                                         col[3] = fcurve_display_alpha(fcu) * 255;
366                                         immAttr4ubv(color, col);
367                                         immVertex2fv(pos, fp);
368                                         immAttr4ubv(color, col);
369                                         immVertex2fv(pos, fp + 3);
370                                 }
371
372                                 /* only draw second handle if this segment is bezier */
373                                 if (bezt->ipo == BEZT_IPO_BEZ) {
374                                         UI_GetThemeColor3ubv(basecol + bezt->h2, col);
375                                         col[3] = fcurve_display_alpha(fcu) * 255;
376                                         immAttr4ubv(color, col);
377                                         immVertex2fv(pos, fp + 3);
378                                         immAttr4ubv(color, col);
379                                         immVertex2fv(pos, fp + 6);
380                                 }
381                         }
382                         else {
383                                 /* only draw first handle if previous segment was had handles, and selection is ok */
384                                 if (((bezt->f1 & SELECT) == sel) &&
385                                     ((!prevbezt && (bezt->ipo == BEZT_IPO_BEZ)) || (prevbezt && (prevbezt->ipo == BEZT_IPO_BEZ))))
386                                 {
387                                         fp = bezt->vec[0];
388                                         UI_GetThemeColor3ubv(basecol + bezt->h1, col);
389                                         col[3] = fcurve_display_alpha(fcu) * 255;
390                                         immAttr4ubv(color, col);
391                                         immVertex2fv(pos, fp);
392                                         immAttr4ubv(color, col);
393                                         immVertex2fv(pos, fp + 3);
394                                 }
395
396                                 /* only draw second handle if this segment is bezier, and selection is ok */
397                                 if (((bezt->f3 & SELECT) == sel) &&
398                                     (bezt->ipo == BEZT_IPO_BEZ))
399                                 {
400                                         fp = bezt->vec[1];
401                                         UI_GetThemeColor3ubv(basecol + bezt->h2, col);
402                                         col[3] = fcurve_display_alpha(fcu) * 255;
403                                         immAttr4ubv(color, col);
404                                         immVertex2fv(pos, fp);
405                                         immAttr4ubv(color, col);
406                                         immVertex2fv(pos, fp + 3);
407                                 }
408                         }
409                 }
410         }
411
412         immEnd();
413         immUnbindProgram();
414 }
415
416 /* Samples ---------------- */
417
418 /* helper func - draw sample-range marker for an F-Curve as a cross
419  * NOTE: the caller MUST HAVE GL_LINE_SMOOTH & GL_BLEND ENABLED, otherwise, the controls don't
420  * have a consistent appearance (due to off-pixel alignments)...
421  */
422 static void draw_fcurve_sample_control(float x, float y, float xscale, float yscale, float hsize, unsigned int pos)
423 {
424         /* adjust view transform before starting */
425         GPU_matrix_push();
426         GPU_matrix_translate_2f(x, y);
427         GPU_matrix_scale_2f(1.0f / xscale * hsize, 1.0f / yscale * hsize);
428
429         /* draw X shape */
430         immBegin(GPU_PRIM_LINES, 4);
431         immVertex2f(pos, -0.7f, -0.7f);
432         immVertex2f(pos, +0.7f, +0.7f);
433
434         immVertex2f(pos, -0.7f, +0.7f);
435         immVertex2f(pos, +0.7f, -0.7f);
436         immEnd();
437
438         /* restore view transform */
439         GPU_matrix_pop();
440 }
441
442 /* helper func - draw keyframe vertices only for an F-Curve */
443 static void draw_fcurve_samples(SpaceIpo *sipo, ARegion *ar, FCurve *fcu)
444 {
445         FPoint *first, *last;
446         float hsize, xscale, yscale;
447
448         /* get view settings */
449         hsize = UI_GetThemeValuef(TH_VERTEX_SIZE);
450         UI_view2d_scale_get(&ar->v2d, &xscale, &yscale);
451
452         /* get verts */
453         first = fcu->fpt;
454         last = (first) ? (first + (fcu->totvert - 1)) : (NULL);
455
456         /* draw */
457         if (first && last) {
458                 /* anti-aliased lines for more consistent appearance */
459                 if ((sipo->flag & SIPO_BEAUTYDRAW_OFF) == 0) GPU_line_smooth(true);
460                 GPU_blend(true);
461
462                 uint pos = GPU_vertformat_attr_add(immVertexFormat(), "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
463                 immBindBuiltinProgram(GPU_SHADER_2D_UNIFORM_COLOR);
464
465                 immUniformThemeColor((fcu->flag & FCURVE_SELECTED) ? TH_TEXT_HI : TH_TEXT);
466
467                 draw_fcurve_sample_control(first->vec[0], first->vec[1], xscale, yscale, hsize, pos);
468                 draw_fcurve_sample_control(last->vec[0], last->vec[1], xscale, yscale, hsize, pos);
469
470                 immUnbindProgram();
471
472                 GPU_blend(false);
473                 if ((sipo->flag & SIPO_BEAUTYDRAW_OFF) == 0) GPU_line_smooth(false);
474         }
475 }
476
477 /* Curve ---------------- */
478
479 /* helper func - just draw the F-Curve by sampling the visible region (for drawing curves with modifiers) */
480 static void draw_fcurve_curve(bAnimContext *ac, ID *id, FCurve *fcu, View2D *v2d, View2DGrid *grid, unsigned int pos)
481 {
482         SpaceIpo *sipo = (SpaceIpo *)ac->sl;
483         ChannelDriver *driver;
484         float samplefreq;
485         float stime, etime;
486         float unitFac, offset;
487         float dx, dy;
488         short mapping_flag = ANIM_get_normalization_flags(ac);
489         int i, n;
490
491         /* when opening a blend file on a different sized screen or while dragging the toolbar this can happen
492          * best just bail out in this case */
493         UI_view2d_grid_size(grid, &dx, &dy);
494         if (dx <= 0.0f)
495                 return;
496
497
498         /* disable any drivers temporarily */
499         driver = fcu->driver;
500         fcu->driver = NULL;
501
502         /* compute unit correction factor */
503         unitFac = ANIM_unit_mapping_get_factor(ac->scene, id, fcu, mapping_flag, &offset);
504
505         /* Note about sampling frequency:
506          * Ideally, this is chosen such that we have 1-2 pixels = 1 segment
507          * which means that our curves can be as smooth as possible. However,
508          * this does mean that curves may not be fully accurate (i.e. if they have
509          * sudden spikes which happen at the sampling point, we may have problems).
510          * Also, this may introduce lower performance on less densely detailed curves,
511          * though it is impossible to predict this from the modifiers!
512          *
513          * If the automatically determined sampling frequency is likely to cause an infinite
514          * loop (i.e. too close to 0), then clamp it to a determined "safe" value. The value
515          *  chosen here is just the coarsest value which still looks reasonable...
516          */
517         /* grid->dx represents the number of 'frames' between gridlines, but we divide by U.v2d_min_gridsize to get pixels-steps */
518         /* TODO: perhaps we should have 1.0 frames as upper limit so that curves don't get too distorted? */
519         samplefreq = dx / (U.v2d_min_gridsize * U.pixelsize);
520
521         if (sipo->flag & SIPO_BEAUTYDRAW_OFF) {
522                 /* Low Precision = coarse lower-bound clamping
523                  *
524                  * Although the "Beauty Draw" flag was originally for AA'd
525                  * line drawing, the sampling rate here has a much greater
526                  * impact on performance (e.g. for T40372)!
527                  *
528                  * This one still amounts to 10 sample-frames for each 1-frame interval
529                  * which should be quite a decent approximation in many situations.
530                  */
531                 if (samplefreq < 0.1f)
532                         samplefreq = 0.1f;
533         }
534         else {
535                 /* "Higher Precision" but slower - especially on larger windows (e.g. T40372) */
536                 if (samplefreq < 0.00001f)
537                         samplefreq = 0.00001f;
538         }
539
540
541         /* the start/end times are simply the horizontal extents of the 'cur' rect */
542         stime = v2d->cur.xmin;
543         etime = v2d->cur.xmax + samplefreq; /* + samplefreq here so that last item gets included... */
544
545
546         /* at each sampling interval, add a new vertex
547          * - apply the unit correction factor to the calculated values so that
548          *   the displayed values appear correctly in the viewport
549          */
550
551         n = (etime - stime) / samplefreq + 0.5f;
552
553         if (n > 0) {
554                 immBegin(GPU_PRIM_LINE_STRIP, (n + 1));
555
556                 for (i = 0; i <= n; i++) {
557                         float ctime = stime + i * samplefreq;
558                         immVertex2f(pos, ctime, (evaluate_fcurve(fcu, ctime) + offset) * unitFac);
559                 }
560
561                 immEnd();
562         }
563
564         /* restore driver */
565         fcu->driver = driver;
566 }
567
568 /* helper func - draw a samples-based F-Curve */
569 static void draw_fcurve_curve_samples(bAnimContext *ac, ID *id, FCurve *fcu, View2D *v2d, const uint shdr_pos)
570 {
571         FPoint *prevfpt = fcu->fpt;
572         FPoint *fpt = prevfpt + 1;
573         float fac, v[2];
574         int b = fcu->totvert;
575         float unit_scale, offset;
576         short mapping_flag = ANIM_get_normalization_flags(ac);
577         int count = fcu->totvert;
578
579         if (prevfpt->vec[0] > v2d->cur.xmin) {
580                 count++;
581         }
582
583         if ((prevfpt + b - 1)->vec[0] < v2d->cur.xmax) {
584                 count++;
585         }
586
587         /* apply unit mapping */
588         GPU_matrix_push();
589         unit_scale = ANIM_unit_mapping_get_factor(ac->scene, id, fcu, mapping_flag, &offset);
590         GPU_matrix_scale_2f(1.0f, unit_scale);
591         GPU_matrix_translate_2f(0.0f, offset);
592
593         immBegin(GPU_PRIM_LINE_STRIP, count);
594
595         /* extrapolate to left? - left-side of view comes before first keyframe? */
596         if (prevfpt->vec[0] > v2d->cur.xmin) {
597                 v[0] = v2d->cur.xmin;
598
599                 /* y-value depends on the interpolation */
600                 if ((fcu->extend == FCURVE_EXTRAPOLATE_CONSTANT) || (fcu->flag & FCURVE_INT_VALUES) || (fcu->totvert == 1)) {
601                         /* just extend across the first keyframe's value */
602                         v[1] = prevfpt->vec[1];
603                 }
604                 else {
605                         /* extrapolate linear doesn't use the handle, use the next points center instead */
606                         fac = (prevfpt->vec[0] - fpt->vec[0]) / (prevfpt->vec[0] - v[0]);
607                         if (fac) fac = 1.0f / fac;
608                         v[1] = prevfpt->vec[1] - fac * (prevfpt->vec[1] - fpt->vec[1]);
609                 }
610
611                 immVertex2fv(shdr_pos, v);
612         }
613
614         /* loop over samples, drawing segments */
615         /* draw curve between first and last keyframe (if there are enough to do so) */
616         while (b--) {
617                 /* Linear interpolation: just add one point (which should add a new line segment) */
618                 immVertex2fv(shdr_pos, prevfpt->vec);
619
620                 /* get next pointers */
621                 if (b > 0) {
622                         prevfpt++;
623                 }
624         }
625
626         /* extrapolate to right? (see code for left-extrapolation above too) */
627         if (prevfpt->vec[0] < v2d->cur.xmax) {
628                 v[0] = v2d->cur.xmax;
629
630                 /* y-value depends on the interpolation */
631                 if ((fcu->extend == FCURVE_EXTRAPOLATE_CONSTANT) || (fcu->flag & FCURVE_INT_VALUES) || (fcu->totvert == 1)) {
632                         /* based on last keyframe's value */
633                         v[1] = prevfpt->vec[1];
634                 }
635                 else {
636                         /* extrapolate linear doesn't use the handle, use the previous points center instead */
637                         fpt = prevfpt - 1;
638                         fac = (prevfpt->vec[0] - fpt->vec[0]) / (prevfpt->vec[0] - v[0]);
639                         if (fac) fac = 1.0f / fac;
640                         v[1] = prevfpt->vec[1] - fac * (prevfpt->vec[1] - fpt->vec[1]);
641                 }
642
643                 immVertex2fv(shdr_pos, v);
644         }
645
646         immEnd();
647
648         GPU_matrix_pop();
649 }
650
651 /* helper func - check if the F-Curve only contains easily drawable segments
652  * (i.e. no easing equation interpolations)
653  */
654 static bool fcurve_can_use_simple_bezt_drawing(FCurve *fcu)
655 {
656         BezTriple *bezt;
657         int i;
658
659         for (i = 0, bezt = fcu->bezt; i < fcu->totvert; i++, bezt++) {
660                 if (ELEM(bezt->ipo, BEZT_IPO_CONST, BEZT_IPO_LIN, BEZT_IPO_BEZ) == false) {
661                         return false;
662                 }
663         }
664
665         return true;
666 }
667
668 /* helper func - draw one repeat of an F-Curve (using Bezier curve approximations) */
669 static void draw_fcurve_curve_bezts(bAnimContext *ac, ID *id, FCurve *fcu, View2D *v2d, unsigned int pos)
670 {
671         BezTriple *prevbezt = fcu->bezt;
672         BezTriple *bezt = prevbezt + 1;
673         float v1[2], v2[2], v3[2], v4[2];
674         float *fp, data[120];
675         float fac = 0.0f;
676         int b = fcu->totvert - 1;
677         int resol;
678         float unit_scale, offset;
679         short mapping_flag = ANIM_get_normalization_flags(ac);
680
681         /* apply unit mapping */
682         GPU_matrix_push();
683         unit_scale = ANIM_unit_mapping_get_factor(ac->scene, id, fcu, mapping_flag, &offset);
684         GPU_matrix_scale_2f(1.0f, unit_scale);
685         GPU_matrix_translate_2f(0.0f, offset);
686
687         /* For now, this assumes the worst case scenario, where all the keyframes have
688          * bezier interpolation, and are drawn at full res.
689          * This is tricky to optimize, but maybe can be improved at some point... */
690         immBeginAtMost(GPU_PRIM_LINE_STRIP, (b * 32 + 3));
691
692         /* extrapolate to left? */
693         if (prevbezt->vec[1][0] > v2d->cur.xmin) {
694                 /* left-side of view comes before first keyframe, so need to extend as not cyclic */
695                 v1[0] = v2d->cur.xmin;
696
697                 /* y-value depends on the interpolation */
698                 if ((fcu->extend == FCURVE_EXTRAPOLATE_CONSTANT) || (prevbezt->ipo == BEZT_IPO_CONST) || (fcu->totvert == 1)) {
699                         /* just extend across the first keyframe's value */
700                         v1[1] = prevbezt->vec[1][1];
701                 }
702                 else if (prevbezt->ipo == BEZT_IPO_LIN) {
703                         /* extrapolate linear dosnt use the handle, use the next points center instead */
704                         fac = (prevbezt->vec[1][0] - bezt->vec[1][0]) / (prevbezt->vec[1][0] - v1[0]);
705                         if (fac) fac = 1.0f / fac;
706                         v1[1] = prevbezt->vec[1][1] - fac * (prevbezt->vec[1][1] - bezt->vec[1][1]);
707                 }
708                 else {
709                         /* based on angle of handle 1 (relative to keyframe) */
710                         fac = (prevbezt->vec[0][0] - prevbezt->vec[1][0]) / (prevbezt->vec[1][0] - v1[0]);
711                         if (fac) fac = 1.0f / fac;
712                         v1[1] = prevbezt->vec[1][1] - fac * (prevbezt->vec[0][1] - prevbezt->vec[1][1]);
713                 }
714
715                 immVertex2fv(pos, v1);
716         }
717
718         /* if only one keyframe, add it now */
719         if (fcu->totvert == 1) {
720                 v1[0] = prevbezt->vec[1][0];
721                 v1[1] = prevbezt->vec[1][1];
722                 immVertex2fv(pos, v1);
723         }
724
725         /* draw curve between first and last keyframe (if there are enough to do so) */
726         /* TODO: optimize this to not have to calc stuff out of view too? */
727         while (b--) {
728                 if (prevbezt->ipo == BEZT_IPO_CONST) {
729                         /* Constant-Interpolation: draw segment between previous keyframe and next, but holding same value */
730                         v1[0] = prevbezt->vec[1][0];
731                         v1[1] = prevbezt->vec[1][1];
732                         immVertex2fv(pos, v1);
733
734                         v1[0] = bezt->vec[1][0];
735                         v1[1] = prevbezt->vec[1][1];
736                         immVertex2fv(pos, v1);
737                 }
738                 else if (prevbezt->ipo == BEZT_IPO_LIN) {
739                         /* Linear interpolation: just add one point (which should add a new line segment) */
740                         v1[0] = prevbezt->vec[1][0];
741                         v1[1] = prevbezt->vec[1][1];
742                         immVertex2fv(pos, v1);
743                 }
744                 else if (prevbezt->ipo == BEZT_IPO_BEZ) {
745                         /* Bezier-Interpolation: draw curve as series of segments between keyframes
746                          * - resol determines number of points to sample in between keyframes
747                          */
748
749                         /* resol depends on distance between points (not just horizontal) OR is a fixed high res */
750                         /* TODO: view scale should factor into this someday too... */
751                         if (fcu->driver) {
752                                 resol = 32;
753                         }
754                         else {
755                                 resol = (int)(5.0f * len_v2v2(bezt->vec[1], prevbezt->vec[1]));
756                         }
757
758                         if (resol < 2) {
759                                 /* only draw one */
760                                 v1[0] = prevbezt->vec[1][0];
761                                 v1[1] = prevbezt->vec[1][1];
762                                 immVertex2fv(pos, v1);
763                         }
764                         else {
765                                 /* clamp resolution to max of 32 */
766                                 /* NOTE: higher values will crash */
767                                 if (resol > 32) resol = 32;
768
769                                 v1[0] = prevbezt->vec[1][0];
770                                 v1[1] = prevbezt->vec[1][1];
771                                 v2[0] = prevbezt->vec[2][0];
772                                 v2[1] = prevbezt->vec[2][1];
773
774                                 v3[0] = bezt->vec[0][0];
775                                 v3[1] = bezt->vec[0][1];
776                                 v4[0] = bezt->vec[1][0];
777                                 v4[1] = bezt->vec[1][1];
778
779                                 correct_bezpart(v1, v2, v3, v4);
780
781                                 BKE_curve_forward_diff_bezier(v1[0], v2[0], v3[0], v4[0], data, resol, sizeof(float) * 3);
782                                 BKE_curve_forward_diff_bezier(v1[1], v2[1], v3[1], v4[1], data + 1, resol, sizeof(float) * 3);
783
784                                 for (fp = data; resol; resol--, fp += 3) {
785                                         immVertex2fv(pos, fp);
786                                 }
787                         }
788                 }
789
790                 /* get next pointers */
791                 prevbezt = bezt;
792                 bezt++;
793
794                 /* last point? */
795                 if (b == 0) {
796                         v1[0] = prevbezt->vec[1][0];
797                         v1[1] = prevbezt->vec[1][1];
798                         immVertex2fv(pos, v1);
799                 }
800         }
801
802         /* extrapolate to right? (see code for left-extrapolation above too) */
803         if (prevbezt->vec[1][0] < v2d->cur.xmax) {
804                 v1[0] = v2d->cur.xmax;
805
806                 /* y-value depends on the interpolation */
807                 if ((fcu->extend == FCURVE_EXTRAPOLATE_CONSTANT) || (fcu->flag & FCURVE_INT_VALUES) || (prevbezt->ipo == BEZT_IPO_CONST) || (fcu->totvert == 1)) {
808                         /* based on last keyframe's value */
809                         v1[1] = prevbezt->vec[1][1];
810                 }
811                 else if (prevbezt->ipo == BEZT_IPO_LIN) {
812                         /* extrapolate linear dosnt use the handle, use the previous points center instead */
813                         bezt = prevbezt - 1;
814                         fac = (prevbezt->vec[1][0] - bezt->vec[1][0]) / (prevbezt->vec[1][0] - v1[0]);
815                         if (fac) fac = 1.0f / fac;
816                         v1[1] = prevbezt->vec[1][1] - fac * (prevbezt->vec[1][1] - bezt->vec[1][1]);
817                 }
818                 else {
819                         /* based on angle of handle 1 (relative to keyframe) */
820                         fac = (prevbezt->vec[2][0] - prevbezt->vec[1][0]) / (prevbezt->vec[1][0] - v1[0]);
821                         if (fac) fac = 1.0f / fac;
822                         v1[1] = prevbezt->vec[1][1] - fac * (prevbezt->vec[2][1] - prevbezt->vec[1][1]);
823                 }
824
825                 immVertex2fv(pos, v1);
826         }
827
828         immEnd();
829
830         GPU_matrix_pop();
831 }
832
833 /* Debugging -------------------------------- */
834
835 /* Draw indicators which show the value calculated from the driver,
836  * and how this is mapped to the value that comes out of it. This
837  * is handy for helping users better understand how to interpret
838  * the graphs, and also facilitates debugging.
839  */
840 static void graph_draw_driver_debug(bAnimContext *ac, ID *id, FCurve *fcu)
841 {
842         ChannelDriver *driver = fcu->driver;
843         View2D *v2d = &ac->ar->v2d;
844         short mapping_flag = ANIM_get_normalization_flags(ac);
845         float offset;
846         float unitfac = ANIM_unit_mapping_get_factor(ac->scene, id, fcu, mapping_flag, &offset);
847
848         /* for now, only show when debugging driver... */
849         //if ((driver->flag & DRIVER_FLAG_SHOWDEBUG) == 0)
850         //      return;
851
852         const uint shdr_pos = GPU_vertformat_attr_add(immVertexFormat(), "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
853         immBindBuiltinProgram(GPU_SHADER_2D_LINE_DASHED_UNIFORM_COLOR);
854
855         float viewport_size[4];
856         GPU_viewport_size_get_f(viewport_size);
857         immUniform2f("viewport_size", viewport_size[2] / UI_DPI_FAC, viewport_size[3] / UI_DPI_FAC);
858
859         immUniform1i("colors_len", 0);  /* Simple dashes. */
860
861         /* No curve to modify/visualize the result?
862          * => We still want to show the 1-1 default...
863          */
864         if ((fcu->totvert == 0) && BLI_listbase_is_empty(&fcu->modifiers)) {
865                 float t;
866
867                 /* draw with thin dotted lines in style of what curve would have been */
868                 immUniformColor3fv(fcu->color);
869
870                 immUniform1f("dash_width", 40.0f);
871                 immUniform1f("dash_factor", 0.5f);
872                 GPU_line_width(2.0f);
873
874                 /* draw 1-1 line, stretching just past the screen limits
875                  * NOTE: we need to scale the y-values to be valid for the units
876                  */
877                 immBegin(GPU_PRIM_LINES, 2);
878
879                 t = v2d->cur.xmin;
880                 immVertex2f(shdr_pos, t, (t + offset) * unitfac);
881
882                 t = v2d->cur.xmax;
883                 immVertex2f(shdr_pos, t, (t + offset) * unitfac);
884
885                 immEnd();
886         }
887
888         /* draw driver only if actually functional */
889         if ((driver->flag & DRIVER_FLAG_INVALID) == 0) {
890                 /* grab "coordinates" for driver outputs */
891                 float x = driver->curval;
892                 float y = fcu->curval * unitfac;
893
894                 /* only draw indicators if the point is in range*/
895                 if (x >= v2d->cur.xmin) {
896                         float co[2];
897
898                         /* draw dotted lines leading towards this point from both axes ....... */
899                         immUniformColor3f(0.9f, 0.9f, 0.9f);
900                         immUniform1f("dash_width", 10.0f);
901                         immUniform1f("dash_factor", 0.5f);
902
903                         immBegin(GPU_PRIM_LINES, (y >= v2d->cur.ymin) ? 4 : 2);
904
905                         /* x-axis lookup */
906                         co[0] = x;
907
908                         if (y >= v2d->cur.ymin) {
909                                 co[1] = v2d->cur.ymin - 1.0f;
910                                 immVertex2fv(shdr_pos, co);
911
912                                 co[1] = y;
913                                 immVertex2fv(shdr_pos, co);
914                         }
915
916                         /* y-axis lookup */
917                         co[1] = y;
918
919                         co[0] = v2d->cur.xmin - 1.0f;
920                         immVertex2fv(shdr_pos, co);
921
922                         co[0] = x;
923                         immVertex2fv(shdr_pos, co);
924
925                         immEnd();
926
927                         immUnbindProgram();
928
929                         /* GPU_PRIM_POINTS do not survive dashed line geometry shader... */
930                         immBindBuiltinProgram(GPU_SHADER_2D_UNIFORM_COLOR);
931
932                         /* x marks the spot .................................................... */
933                         /* -> outer frame */
934                         immUniformColor3f(0.9f, 0.9f, 0.9f);
935                         GPU_point_size(7.0);
936
937                         immBegin(GPU_PRIM_POINTS, 1);
938                         immVertex2f(shdr_pos, x, y);
939                         immEnd();
940
941                         /* inner frame */
942                         immUniformColor3f(0.9f, 0.0f, 0.0f);
943                         GPU_point_size(3.0);
944
945                         immBegin(GPU_PRIM_POINTS, 1);
946                         immVertex2f(shdr_pos, x, y);
947                         immEnd();
948                 }
949         }
950
951         immUnbindProgram();
952 }
953
954 /* Public Curve-Drawing API  ---------------- */
955
956 /* Draw the 'ghost' F-Curves (i.e. snapshots of the curve)
957  * NOTE: unit mapping has already been applied to the values, so do not try and apply again
958  */
959 void graph_draw_ghost_curves(bAnimContext *ac, SpaceIpo *sipo, ARegion *ar)
960 {
961         FCurve *fcu;
962
963         /* draw with thick dotted lines */
964         GPU_line_width(3.0f);
965
966         /* anti-aliased lines for less jagged appearance */
967         if ((sipo->flag & SIPO_BEAUTYDRAW_OFF) == 0) {
968                 GPU_line_smooth(true);
969         }
970         GPU_blend(true);
971
972         const uint shdr_pos = GPU_vertformat_attr_add(immVertexFormat(), "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
973
974         immBindBuiltinProgram(GPU_SHADER_2D_LINE_DASHED_UNIFORM_COLOR);
975
976         float viewport_size[4];
977         GPU_viewport_size_get_f(viewport_size);
978         immUniform2f("viewport_size", viewport_size[2] / UI_DPI_FAC, viewport_size[3] / UI_DPI_FAC);
979
980         immUniform1i("colors_len", 0);  /* Simple dashes. */
981         immUniform1f("dash_width", 20.0f);
982         immUniform1f("dash_factor", 0.5f);
983
984         /* the ghost curves are simply sampled F-Curves stored in sipo->runtime.ghost_curves */
985         for (fcu = sipo->runtime.ghost_curves.first; fcu; fcu = fcu->next) {
986                 /* set whatever color the curve has set
987                  * - this is set by the function which creates these
988                  * - draw with a fixed opacity of 2
989                  */
990                 immUniformColor3fvAlpha(fcu->color, 0.5f);
991
992                 /* simply draw the stored samples */
993                 draw_fcurve_curve_samples(ac, NULL, fcu, &ar->v2d, shdr_pos);
994         }
995
996         immUnbindProgram();
997
998         if ((sipo->flag & SIPO_BEAUTYDRAW_OFF) == 0) {
999                 GPU_line_smooth(false);
1000         }
1001         GPU_blend(false);
1002 }
1003
1004 /* This is called twice from space_graph.c -> graph_main_region_draw()
1005  * Unselected then selected F-Curves are drawn so that they do not occlude each other.
1006  */
1007 void graph_draw_curves(bAnimContext *ac, SpaceIpo *sipo, ARegion *ar, View2DGrid *grid, short sel)
1008 {
1009         ListBase anim_data = {NULL, NULL};
1010         bAnimListElem *ale;
1011         int filter;
1012
1013         /* build list of curves to draw */
1014         filter = (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_CURVE_VISIBLE);
1015         filter |= ((sel) ? (ANIMFILTER_SEL) : (ANIMFILTER_UNSEL));
1016         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
1017
1018         /* for each curve:
1019          * draw curve, then handle-lines, and finally vertices in this order so that
1020          * the data will be layered correctly
1021          */
1022         for (ale = anim_data.first; ale; ale = ale->next) {
1023                 FCurve *fcu = (FCurve *)ale->key_data;
1024                 FModifier *fcm = find_active_fmodifier(&fcu->modifiers);
1025                 AnimData *adt = ANIM_nla_mapping_get(ac, ale);
1026
1027                 /* map keyframes for drawing if scaled F-Curve */
1028                 if (adt) {
1029                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 0);
1030                 }
1031
1032                 /* draw curve:
1033                  * - curve line may be result of one or more destructive modifiers or just the raw data,
1034                  *   so we need to check which method should be used
1035                  * - controls from active modifier take precedence over keyframes
1036                  *   (XXX! editing tools need to take this into account!)
1037                  */
1038
1039                 /* 1) draw curve line */
1040                 if (((fcu->modifiers.first) || (fcu->flag & FCURVE_INT_VALUES)) ||
1041                     (((fcu->bezt) || (fcu->fpt)) && (fcu->totvert)))
1042                 {
1043                         /* set color/drawing style for curve itself */
1044                         /* draw active F-Curve thicker than the rest to make it stand out */
1045                         if (fcu->flag & FCURVE_ACTIVE) {
1046                                 GPU_line_width(2.5);
1047                         }
1048                         else {
1049                                 GPU_line_width(1.0);
1050                         }
1051
1052                         /* anti-aliased lines for less jagged appearance */
1053                         if ((sipo->flag & SIPO_BEAUTYDRAW_OFF) == 0) {
1054                                 GPU_line_smooth(true);
1055                         }
1056                         GPU_blend(true);
1057
1058                         const uint shdr_pos = GPU_vertformat_attr_add(immVertexFormat(), "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
1059
1060                         immBindBuiltinProgram(GPU_SHADER_2D_LINE_DASHED_UNIFORM_COLOR);
1061
1062                         float viewport_size[4];
1063                         GPU_viewport_size_get_f(viewport_size);
1064                         immUniform2f("viewport_size", viewport_size[2] / UI_DPI_FAC, viewport_size[3] / UI_DPI_FAC);
1065
1066                         immUniform1i("colors_len", 0);  /* Simple dashes. */
1067
1068                         if (BKE_fcurve_is_protected(fcu)) {
1069                                 /* protected curves (non editable) are drawn with dotted lines */
1070                                 immUniform1f("dash_width", 4.0f);
1071                                 immUniform1f("dash_factor", 0.5f);
1072                         }
1073                         else {
1074                                 immUniform1f("dash_factor", 2.0f);  /* solid line */
1075                         }
1076
1077                         if (((fcu->grp) && (fcu->grp->flag & AGRP_MUTED)) || (fcu->flag & FCURVE_MUTED)) {
1078                                 /* muted curves are drawn in a grayish hue */
1079                                 /* XXX should we have some variations? */
1080                                 immUniformThemeColorShade(TH_HEADER, 50);
1081                         }
1082                         else {
1083                                 /* set whatever color the curve has set
1084                                  * - unselected curves draw less opaque to help distinguish the selected ones
1085                                  */
1086                                 immUniformColor3fvAlpha(fcu->color, fcurve_display_alpha(fcu));
1087                         }
1088
1089                         /* draw F-Curve */
1090                         if ((fcu->modifiers.first) || (fcu->flag & FCURVE_INT_VALUES)) {
1091                                 /* draw a curve affected by modifiers or only allowed to have integer values
1092                                  * by sampling it at various small-intervals over the visible region
1093                                  */
1094                                 draw_fcurve_curve(ac, ale->id, fcu, &ar->v2d, grid, shdr_pos);
1095                         }
1096                         else if (((fcu->bezt) || (fcu->fpt)) && (fcu->totvert)) {
1097                                 /* just draw curve based on defined data (i.e. no modifiers) */
1098                                 if (fcu->bezt) {
1099                                         if (fcurve_can_use_simple_bezt_drawing(fcu)) {
1100                                                 draw_fcurve_curve_bezts(ac, ale->id, fcu, &ar->v2d, shdr_pos);
1101                                         }
1102                                         else {
1103                                                 draw_fcurve_curve(ac, ale->id, fcu, &ar->v2d, grid, shdr_pos);
1104                                         }
1105                                 }
1106                                 else if (fcu->fpt) {
1107                                         draw_fcurve_curve_samples(ac, ale->id, fcu, &ar->v2d, shdr_pos);
1108                                 }
1109                         }
1110
1111                         immUnbindProgram();
1112
1113                         if ((sipo->flag & SIPO_BEAUTYDRAW_OFF) == 0) {
1114                                 GPU_line_smooth(false);
1115                         }
1116                         GPU_blend(false);
1117                 }
1118
1119                 /* 2) draw handles and vertices as appropriate based on active
1120                  * - if the option to only show controls if the F-Curve is selected is enabled, we must obey this
1121                  */
1122                 if (!(sipo->flag & SIPO_SELCUVERTSONLY) || (fcu->flag & FCURVE_SELECTED)) {
1123                         if (!fcurve_are_keyframes_usable(fcu) && !(fcu->fpt && fcu->totvert)) {
1124                                 /* only draw controls if this is the active modifier */
1125                                 if ((fcu->flag & FCURVE_ACTIVE) && (fcm)) {
1126                                         switch (fcm->type) {
1127                                                 case FMODIFIER_TYPE_ENVELOPE: /* envelope */
1128                                                         draw_fcurve_modifier_controls_envelope(fcm, &ar->v2d);
1129                                                         break;
1130                                         }
1131                                 }
1132                         }
1133                         else if (((fcu->bezt) || (fcu->fpt)) && (fcu->totvert)) {
1134                                 short mapping_flag = ANIM_get_normalization_flags(ac);
1135                                 float offset;
1136                                 float unit_scale = ANIM_unit_mapping_get_factor(ac->scene, ale->id, fcu, mapping_flag, &offset);
1137
1138                                 /* apply unit-scaling to all values via OpenGL */
1139                                 GPU_matrix_push();
1140                                 GPU_matrix_scale_2f(1.0f, unit_scale);
1141                                 GPU_matrix_translate_2f(0.0f, offset);
1142
1143                                 /* set this once and for all - all handles and handle-verts should use the same thickness */
1144                                 GPU_line_width(1.0);
1145
1146                                 if (fcu->bezt) {
1147                                         bool do_handles = draw_fcurve_handles_check(sipo, fcu);
1148
1149                                         if (do_handles) {
1150                                                 /* only draw handles/vertices on keyframes */
1151                                                 GPU_blend(true);
1152                                                 draw_fcurve_handles(sipo, fcu);
1153                                                 GPU_blend(false);
1154                                         }
1155
1156                                         draw_fcurve_vertices(ar, fcu, do_handles, (sipo->flag & SIPO_SELVHANDLESONLY));
1157                                 }
1158                                 else {
1159                                         /* samples: only draw two indicators at either end as indicators */
1160                                         draw_fcurve_samples(sipo, ar, fcu);
1161                                 }
1162
1163                                 GPU_matrix_pop();
1164                         }
1165                 }
1166
1167                 /* 3) draw driver debugging stuff */
1168                 if ((ac->datatype == ANIMCONT_DRIVERS) && (fcu->flag & FCURVE_ACTIVE)) {
1169                         graph_draw_driver_debug(ac, ale->id, fcu);
1170                 }
1171
1172                 /* undo mapping of keyframes for drawing if scaled F-Curve */
1173                 if (adt)
1174                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 0);
1175         }
1176
1177         /* free list of curves */
1178         ANIM_animdata_freelist(&anim_data);
1179 }
1180
1181 /* ************************************************************************* */
1182 /* Channel List */
1183
1184 /* left hand part */
1185 void graph_draw_channel_names(bContext *C, bAnimContext *ac, ARegion *ar)
1186 {
1187         ListBase anim_data = {NULL, NULL};
1188         bAnimListElem *ale;
1189         int filter;
1190
1191         View2D *v2d = &ar->v2d;
1192         float y = 0.0f, height;
1193         size_t items;
1194         int i = 0;
1195
1196         /* build list of channels to draw */
1197         filter = (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE | ANIMFILTER_LIST_CHANNELS);
1198         items = ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
1199
1200         /* Update max-extent of channels here (taking into account scrollers):
1201          * - this is done to allow the channel list to be scrollable, but must be done here
1202          *   to avoid regenerating the list again and/or also because channels list is drawn first
1203          * - offset of ACHANNEL_HEIGHT*2 is added to the height of the channels, as first is for
1204          *   start of list offset, and the second is as a correction for the scrollers.
1205          */
1206         height = (float)((items * ACHANNEL_STEP(ac)) + (ACHANNEL_HEIGHT(ac) * 2));
1207         UI_view2d_totRect_set(v2d, BLI_rcti_size_x(&ar->v2d.mask), height);
1208
1209         /* loop through channels, and set up drawing depending on their type  */
1210         {   /* first pass: just the standard GL-drawing for backdrop + text */
1211                 size_t channel_index = 0;
1212
1213                 y = (float)ACHANNEL_FIRST(ac);
1214
1215                 for (ale = anim_data.first, i = 0; ale; ale = ale->next, i++) {
1216                         const float yminc = (float)(y - ACHANNEL_HEIGHT_HALF(ac));
1217                         const float ymaxc = (float)(y + ACHANNEL_HEIGHT_HALF(ac));
1218
1219                         /* check if visible */
1220                         if (IN_RANGE(yminc, v2d->cur.ymin, v2d->cur.ymax) ||
1221                             IN_RANGE(ymaxc, v2d->cur.ymin, v2d->cur.ymax) )
1222                         {
1223                                 /* draw all channels using standard channel-drawing API */
1224                                 ANIM_channel_draw(ac, ale, yminc, ymaxc, channel_index);
1225                         }
1226
1227                         /* adjust y-position for next one */
1228                         y -= ACHANNEL_STEP(ac);
1229                         channel_index++;
1230                 }
1231         }
1232         {   /* second pass: widgets */
1233                 uiBlock *block = UI_block_begin(C, ar, __func__, UI_EMBOSS);
1234                 size_t channel_index = 0;
1235
1236                 y = (float)ACHANNEL_FIRST(ac);
1237
1238                 /* set blending again, as may not be set in previous step */
1239                 GPU_blend_set_func_separate(GPU_SRC_ALPHA, GPU_ONE_MINUS_SRC_ALPHA, GPU_ONE, GPU_ONE_MINUS_SRC_ALPHA);
1240                 GPU_blend(true);
1241
1242                 for (ale = anim_data.first, i = 0; ale; ale = ale->next, i++) {
1243                         const float yminc = (float)(y - ACHANNEL_HEIGHT_HALF(ac));
1244                         const float ymaxc = (float)(y + ACHANNEL_HEIGHT_HALF(ac));
1245
1246                         /* check if visible */
1247                         if (IN_RANGE(yminc, v2d->cur.ymin, v2d->cur.ymax) ||
1248                             IN_RANGE(ymaxc, v2d->cur.ymin, v2d->cur.ymax) )
1249                         {
1250                                 /* draw all channels using standard channel-drawing API */
1251                                 ANIM_channel_draw_widgets(C, ac, ale, block, yminc, ymaxc, channel_index);
1252                         }
1253
1254                         /* adjust y-position for next one */
1255                         y -= ACHANNEL_STEP(ac);
1256                         channel_index++;
1257                 }
1258
1259                 UI_block_end(C, block);
1260                 UI_block_draw(C, block);
1261
1262                 GPU_blend(false);
1263         }
1264
1265         /* free tempolary channels */
1266         ANIM_animdata_freelist(&anim_data);
1267 }