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