style cleanup: comments
[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 #ifndef _WIN32
36 #include <unistd.h>
37 #else
38 #include <io.h>
39 #endif
40
41
42 #include "BLI_blenlib.h"
43 #include "BLI_math.h"
44 #include "BLI_utildefines.h"
45
46 #include "DNA_anim_types.h"
47 #include "DNA_object_types.h"
48 #include "DNA_screen_types.h"
49 #include "DNA_space_types.h"
50 #include "DNA_windowmanager_types.h"
51 #include "DNA_userdef_types.h"
52
53 #include "BKE_context.h"
54 #include "BKE_curve.h"
55 #include "BKE_fcurve.h"
56
57
58 #include "BIF_gl.h"
59 #include "BIF_glutil.h"
60
61 #include "ED_anim_api.h"
62
63 #include "graph_intern.h"
64
65 #include "UI_interface.h"
66 #include "UI_resources.h"
67 #include "UI_view2d.h"
68
69 /* *************************** */
70 /* Utility Drawing Defines */
71
72 /* determine the alpha value that should be used when 
73  * drawing components for some F-Curve (fcu)
74  *      - selected F-Curves should be more visible than partially visible ones
75  */
76 static float fcurve_display_alpha(FCurve *fcu)
77 {
78         return (fcu->flag & FCURVE_SELECTED) ? 1.0f : U.fcu_inactive_alpha;
79 }
80
81 /* *************************** */
82 /* F-Curve Modifier Drawing */
83
84 /* Envelope -------------- */
85
86 /* TODO: draw a shaded poly showing the region of influence too!!! */
87 static void draw_fcurve_modifier_controls_envelope(FModifier *fcm, View2D *v2d)
88 {
89         FMod_Envelope *env = (FMod_Envelope *)fcm->data;
90         FCM_EnvelopeData *fed;
91         const float fac = 0.05f * BLI_rctf_size_x(&v2d->cur);
92         int i;
93         
94         /* draw two black lines showing the standard reference levels */
95         glColor3f(0.0f, 0.0f, 0.0f);
96         setlinestyle(5);
97         
98         glBegin(GL_LINES);
99         glVertex2f(v2d->cur.xmin, env->midval + env->min);
100         glVertex2f(v2d->cur.xmax, env->midval + env->min);
101                 
102         glVertex2f(v2d->cur.xmin, env->midval + env->max);
103         glVertex2f(v2d->cur.xmax, env->midval + env->max);
104         glEnd();  /* GL_LINES */
105         setlinestyle(0);
106         
107         /* set size of vertices (non-adjustable for now) */
108         glPointSize(2.0f);
109         
110         /* for now, point color is fixed, and is white */
111         glColor3f(1.0f, 1.0f, 1.0f);
112         
113         /* we use bgl points not standard gl points, to workaround vertex 
114          * drawing bugs that some drivers have (probably legacy ones only though)
115          */
116         bglBegin(GL_POINTS);
117         for (i = 0, fed = env->data; i < env->totvert; i++, fed++) {
118                 /* only draw if visible
119                  *      - min/max here are fixed, not relative
120                  */
121                 if (IN_RANGE(fed->time, (v2d->cur.xmin - fac), (v2d->cur.xmax + fac))) {
122                         glVertex2f(fed->time, fed->min);
123                         glVertex2f(fed->time, fed->max);
124                 }
125         }
126         bglEnd();  /* GL_POINTS */
127         
128         glPointSize(1.0f);
129 }
130
131 /* *************************** */
132 /* F-Curve Drawing */
133
134 /* Points ---------------- */
135
136 /* helper func - draw keyframe vertices only for an F-Curve */
137 static void draw_fcurve_vertices_keyframes(FCurve *fcu, SpaceIpo *UNUSED(sipo), View2D *v2d, short edit, short sel)
138 {
139         BezTriple *bezt = fcu->bezt;
140         const float fac = 0.05f * BLI_rctf_size_x(&v2d->cur);
141         int i;
142         
143         /* we use bgl points not standard gl points, to workaround vertex 
144          * drawing bugs that some drivers have (probably legacy ones only though)
145          */
146         bglBegin(GL_POINTS);
147         
148         for (i = 0; i < fcu->totvert; i++, bezt++) {
149                 /* as an optimization step, only draw those in view 
150                  *      - we apply a correction factor to ensure that points don't pop in/out due to slight twitches of view size
151                  */
152                 if (IN_RANGE(bezt->vec[1][0], (v2d->cur.xmin - fac), (v2d->cur.xmax + fac))) {
153                         if (edit) {
154                                 /* 'Keyframe' vertex only, as handle lines and handles have already been drawn
155                                  *      - only draw those with correct selection state for the current drawing color
156                                  *      - 
157                                  */
158                                 if ((bezt->f2 & SELECT) == sel)
159                                         bglVertex3fv(bezt->vec[1]);
160                         }
161                         else {
162                                 /* no check for selection here, as curve is not editable... */
163                                 /* XXX perhaps we don't want to even draw points?   maybe add an option for that later */
164                                 bglVertex3fv(bezt->vec[1]);
165                         }
166                 }
167         }
168         
169         bglEnd();  /* GL_POINTS */
170 }
171
172
173 /* helper func - draw handle vertex for an F-Curve as a round unfilled circle 
174  * NOTE: the caller MUST HAVE GL_LINE_SMOOTH & GL_BLEND ENABLED, otherwise, the controls don't 
175  * have a consistent appearance (due to off-pixel alignments)...
176  */
177 static void draw_fcurve_handle_control(float x, float y, float xscale, float yscale, float hsize)
178 {
179         static GLuint displist = 0;
180         
181         /* initialize round circle shape */
182         if (displist == 0) {
183                 GLUquadricObj *qobj;
184                 
185                 displist = glGenLists(1);
186                 glNewList(displist, GL_COMPILE);
187                 
188                 qobj    = gluNewQuadric();
189                 gluQuadricDrawStyle(qobj, GLU_SILHOUETTE); 
190                 gluDisk(qobj, 0,  0.7, 8, 1);
191                 gluDeleteQuadric(qobj);  
192                 
193                 glEndList();
194         }
195         
196         /* adjust view transform before starting */
197         glTranslatef(x, y, 0.0f);
198         glScalef(1.0f / xscale * hsize, 1.0f / yscale * hsize, 1.0f);
199         
200         /* draw! */
201         glCallList(displist);
202         
203         /* restore view transform */
204         glScalef(xscale / hsize, yscale / hsize, 1.0);
205         glTranslatef(-x, -y, 0.0f);
206 }
207
208 /* helper func - draw handle vertices only for an F-Curve (if it is not protected) */
209 static void draw_fcurve_vertices_handles(FCurve *fcu, SpaceIpo *sipo, View2D *v2d, short sel, short sel_handle_only)
210 {
211         BezTriple *bezt = fcu->bezt;
212         BezTriple *prevbezt = NULL;
213         float hsize, xscale, yscale;
214         int i;
215         
216         /* get view settings */
217         hsize = UI_GetThemeValuef(TH_HANDLE_VERTEX_SIZE);
218         UI_view2d_getscale(v2d, &xscale, &yscale);
219         
220         /* set handle color */
221         if (sel) UI_ThemeColor(TH_HANDLE_VERTEX_SELECT);
222         else UI_ThemeColor(TH_HANDLE_VERTEX);
223         
224         /* anti-aliased lines for more consistent appearance */
225         if ((sipo->flag & SIPO_BEAUTYDRAW_OFF) == 0) glEnable(GL_LINE_SMOOTH);
226         glEnable(GL_BLEND);
227         
228         for (i = 0; i < fcu->totvert; i++, prevbezt = bezt, bezt++) {
229                 /* Draw the editmode handles for a bezier curve (others don't have handles) 
230                  * if their selection status matches the selection status we're drawing for
231                  *      - first handle only if previous beztriple was bezier-mode
232                  *      - second handle only if current beztriple is bezier-mode
233                  *
234                  * Also, need to take into account whether the keyframe was selected
235                  * if a Graph Editor option to only show handles of selected keys is on.
236                  */
237                 if (!sel_handle_only || BEZSELECTED(bezt) ) {
238                         if ((!prevbezt && (bezt->ipo == BEZT_IPO_BEZ)) || (prevbezt && (prevbezt->ipo == BEZT_IPO_BEZ))) {
239                                 if ((bezt->f1 & SELECT) == sel) /* && v2d->cur.xmin < bezt->vec[0][0] < v2d->cur.xmax)*/
240                                         draw_fcurve_handle_control(bezt->vec[0][0], bezt->vec[0][1], xscale, yscale, hsize);
241                         }
242                         
243                         if (bezt->ipo == BEZT_IPO_BEZ) {
244                                 if ((bezt->f3 & SELECT) == sel) /* && v2d->cur.xmin < bezt->vec[2][0] < v2d->cur.xmax)*/
245                                         draw_fcurve_handle_control(bezt->vec[2][0], bezt->vec[2][1], xscale, yscale, hsize);
246                         }
247                 }
248         }
249         
250         if ((sipo->flag & SIPO_BEAUTYDRAW_OFF) == 0) glDisable(GL_LINE_SMOOTH);
251         glDisable(GL_BLEND);
252 }
253
254 /* helper func - set color to draw F-Curve data with */
255 static void set_fcurve_vertex_color(FCurve *fcu, short sel)
256 {
257         /* Fade the 'intensity' of the vertices based on the selection of the curves too */
258         int alphaOffset = (int)((fcurve_display_alpha(fcu) - 1.0f) * 255);
259         
260         /* Set color of curve vertex based on state of curve (i.e. 'Edit' Mode) */
261         if ((fcu->flag & FCURVE_PROTECTED) == 0) {
262                 /* Curve's points ARE BEING edited */
263                 if (sel) UI_ThemeColorShadeAlpha(TH_VERTEX_SELECT, 0, alphaOffset); 
264                 else UI_ThemeColorShadeAlpha(TH_VERTEX, 0, alphaOffset);
265         } 
266         else {
267                 /* Curve's points CANNOT BE edited */
268                 if (sel) UI_ThemeColorShadeAlpha(TH_TEXT_HI, 0, alphaOffset);
269                 else UI_ThemeColorShadeAlpha(TH_TEXT, 0, alphaOffset);
270         }
271 }
272
273
274 static void draw_fcurve_vertices(SpaceIpo *sipo, ARegion *ar, FCurve *fcu, short do_handles, short sel_handle_only)
275 {
276         View2D *v2d = &ar->v2d;
277         
278         /* only draw points if curve is visible 
279          *  - draw unselected points before selected points as separate passes to minimize color-changing overhead
280          *         (XXX dunno if this is faster than drawing all in one pass though) 
281          *     and also to make sure in the case of overlapping points that the selected is always visible
282          *      - draw handles before keyframes, so that keyframes will overlap handles (keyframes are more important for users)
283          */
284         
285         glPointSize(UI_GetThemeValuef(TH_VERTEX_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                 set_fcurve_vertex_color(fcu, 0);
290                 draw_fcurve_vertices_handles(fcu, sipo, v2d, 0, sel_handle_only);
291                 
292                 set_fcurve_vertex_color(fcu, 1);
293                 draw_fcurve_vertices_handles(fcu, sipo, v2d, 1, sel_handle_only);
294         }
295                 
296         /* draw keyframes over the handles */
297         set_fcurve_vertex_color(fcu, 0);
298         draw_fcurve_vertices_keyframes(fcu, sipo, v2d, !(fcu->flag & FCURVE_PROTECTED), 0);
299         
300         set_fcurve_vertex_color(fcu, 1);
301         draw_fcurve_vertices_keyframes(fcu, sipo, v2d, !(fcu->flag & FCURVE_PROTECTED), 1);
302         
303         glPointSize(1.0f);
304 }
305
306 /* Handles ---------------- */
307
308 static int 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 0;
321         } 
322         else {
323                 return 1;
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         /* a single call to GL_LINES here around these calls should be sufficient to still
334          * get separate line segments, but which aren't wrapped with GL_LINE_STRIP every time we
335          * want a single line
336          */
337         glBegin(GL_LINES);
338         
339         /* slightly hacky, but we want to draw unselected points before selected ones 
340          * so that selected points are clearly visible
341          */
342         for (sel = 0; sel < 2; sel++) {
343                 BezTriple *bezt = fcu->bezt, *prevbezt = NULL;
344                 int basecol = (sel) ? TH_HANDLE_SEL_FREE : TH_HANDLE_FREE;
345                 float *fp;
346                 unsigned char col[4];
347                 
348                 for (b = 0; b < fcu->totvert; b++, prevbezt = bezt, bezt++) {
349                         /* if only selected keyframes can get their handles shown, 
350                          * check that keyframe is selected
351                          */
352                         if (sipo->flag & SIPO_SELVHANDLESONLY) {
353                                 if (BEZSELECTED(bezt) == 0)
354                                         continue;
355                         }
356                         
357                         /* draw handle with appropriate set of colors if selection is ok */
358                         if ((bezt->f2 & SELECT) == sel) {
359                                 fp = bezt->vec[0];
360                                 
361                                 /* only draw first handle if previous segment had handles */
362                                 if ((!prevbezt && (bezt->ipo == BEZT_IPO_BEZ)) || (prevbezt && (prevbezt->ipo == BEZT_IPO_BEZ))) {
363                                         UI_GetThemeColor3ubv(basecol + bezt->h1, col);
364                                         col[3] = fcurve_display_alpha(fcu) * 255;
365                                         glColor4ubv((GLubyte *)col);
366                                         
367                                         glVertex2fv(fp); glVertex2fv(fp + 3);
368                                 }
369                                 
370                                 /* only draw second handle if this segment is bezier */
371                                 if (bezt->ipo == BEZT_IPO_BEZ) {
372                                         UI_GetThemeColor3ubv(basecol + bezt->h2, col);
373                                         col[3] = fcurve_display_alpha(fcu) * 255;
374                                         glColor4ubv((GLubyte *)col);
375                                         
376                                         glVertex2fv(fp + 3); glVertex2fv(fp + 6);
377                                 }
378                         }
379                         else {
380                                 /* only draw first handle if previous segment was had handles, and selection is ok */
381                                 if (((bezt->f1 & SELECT) == sel) &&
382                                     ((!prevbezt && (bezt->ipo == BEZT_IPO_BEZ)) || (prevbezt && (prevbezt->ipo == BEZT_IPO_BEZ))))
383                                 {
384                                         fp = bezt->vec[0];
385                                         UI_GetThemeColor3ubv(basecol + bezt->h1, col);
386                                         col[3] = fcurve_display_alpha(fcu) * 255;
387                                         glColor4ubv((GLubyte *)col);
388                                         
389                                         glVertex2fv(fp); glVertex2fv(fp + 3);
390                                 }
391                                 
392                                 /* only draw second handle if this segment is bezier, and selection is ok */
393                                 if (((bezt->f3 & SELECT) == sel) &&
394                                     (bezt->ipo == BEZT_IPO_BEZ))
395                                 {
396                                         fp = bezt->vec[1];
397                                         UI_GetThemeColor3ubv(basecol + bezt->h2, col);
398                                         col[3] = fcurve_display_alpha(fcu) * 255;
399                                         glColor4ubv((GLubyte *)col);
400                                         
401                                         glVertex2fv(fp); glVertex2fv(fp + 3);
402                                 }
403                         }
404                 }
405         }
406         
407         glEnd();  /* GL_LINES */
408 }
409
410 /* Samples ---------------- */
411
412 /* helper func - draw sample-range marker for an F-Curve as a cross 
413  * NOTE: the caller MUST HAVE GL_LINE_SMOOTH & GL_BLEND ENABLED, otherwise, the controls don't 
414  * have a consistent appearance (due to off-pixel alignments)...
415  */
416 static void draw_fcurve_sample_control(float x, float y, float xscale, float yscale, float hsize)
417 {
418         static GLuint displist = 0;
419         
420         /* initialize X shape */
421         if (displist == 0) {
422                 displist = glGenLists(1);
423                 glNewList(displist, GL_COMPILE);
424                 
425                 glBegin(GL_LINES);
426                 glVertex2f(-0.7f, -0.7f);
427                 glVertex2f(+0.7f, +0.7f);
428                         
429                 glVertex2f(-0.7f, +0.7f);
430                 glVertex2f(+0.7f, -0.7f);
431                 glEnd();  /* GL_LINES */
432                 
433                 glEndList();
434         }
435         
436         /* adjust view transform before starting */
437         glTranslatef(x, y, 0.0f);
438         glScalef(1.0f / xscale * hsize, 1.0f / yscale * hsize, 1.0f);
439         
440         /* draw! */
441         glCallList(displist);
442         
443         /* restore view transform */
444         glScalef(xscale / hsize, yscale / hsize, 1.0);
445         glTranslatef(-x, -y, 0.0f);
446 }
447
448 /* helper func - draw keyframe vertices only for an F-Curve */
449 static void draw_fcurve_samples(SpaceIpo *sipo, ARegion *ar, FCurve *fcu)
450 {
451         FPoint *first, *last;
452         float hsize, xscale, yscale;
453         
454         /* get view settings */
455         hsize = UI_GetThemeValuef(TH_VERTEX_SIZE);
456         UI_view2d_getscale(&ar->v2d, &xscale, &yscale);
457         
458         /* set vertex color */
459         if (fcu->flag & (FCURVE_ACTIVE | FCURVE_SELECTED)) UI_ThemeColor(TH_TEXT_HI);
460         else UI_ThemeColor(TH_TEXT);
461         
462         /* get verts */
463         first = fcu->fpt;
464         last = (first) ? (first + (fcu->totvert - 1)) : (NULL);
465         
466         /* draw */
467         if (first && last) {
468                 /* anti-aliased lines for more consistent appearance */
469                 if ((sipo->flag & SIPO_BEAUTYDRAW_OFF) == 0) glEnable(GL_LINE_SMOOTH);
470                 glEnable(GL_BLEND);
471                 
472                 draw_fcurve_sample_control(first->vec[0], first->vec[1], xscale, yscale, hsize);
473                 draw_fcurve_sample_control(last->vec[0], last->vec[1], xscale, yscale, hsize);
474                 
475                 glDisable(GL_BLEND);
476                 if ((sipo->flag & SIPO_BEAUTYDRAW_OFF) == 0) glDisable(GL_LINE_SMOOTH);
477         }
478 }
479
480 /* Curve ---------------- */
481
482 /* helper func - just draw the F-Curve by sampling the visible region (for drawing curves with modifiers) */
483 static void draw_fcurve_curve(bAnimContext *ac, ID *id, FCurve *fcu, View2D *v2d, View2DGrid *grid)
484 {
485         ChannelDriver *driver;
486         float samplefreq, ctime;
487         float stime, etime;
488         float unitFac;
489         float dx, dy;
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, 0);
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;
520         if (samplefreq < 0.00001f) samplefreq = 0.00001f;
521         
522         
523         /* the start/end times are simply the horizontal extents of the 'cur' rect */
524         stime = v2d->cur.xmin;
525         etime = v2d->cur.xmax + samplefreq; /* + samplefreq here so that last item gets included... */
526         
527         
528         /* at each sampling interval, add a new vertex 
529          *      - apply the unit correction factor to the calculated values so that 
530          *        the displayed values appear correctly in the viewport
531          */
532         glBegin(GL_LINE_STRIP);
533         
534         for (ctime = stime; ctime <= etime; ctime += samplefreq)
535                 glVertex2f(ctime, evaluate_fcurve(fcu, ctime) * unitFac);
536         
537         glEnd();
538         
539         /* restore driver */
540         fcu->driver = driver;
541 }
542
543 /* helper func - draw a samples-based F-Curve */
544 static void draw_fcurve_curve_samples(bAnimContext *ac, ID *id, FCurve *fcu, View2D *v2d)
545 {
546         FPoint *prevfpt = fcu->fpt;
547         FPoint *fpt = prevfpt + 1;
548         float fac, v[2];
549         int b = fcu->totvert - 1;
550         
551         glBegin(GL_LINE_STRIP);
552         
553         /* apply unit mapping */
554         ANIM_unit_mapping_apply_fcurve(ac->scene, id, fcu, 0);
555         
556         /* extrapolate to left? - left-side of view comes before first keyframe? */
557         if (prevfpt->vec[0] > v2d->cur.xmin) {
558                 v[0] = v2d->cur.xmin;
559                 
560                 /* y-value depends on the interpolation */
561                 if ((fcu->extend == FCURVE_EXTRAPOLATE_CONSTANT) || (fcu->flag & FCURVE_INT_VALUES) || (fcu->totvert == 1)) {
562                         /* just extend across the first keyframe's value */
563                         v[1] = prevfpt->vec[1];
564                 } 
565                 else {
566                         /* extrapolate linear dosnt use the handle, use the next points center instead */
567                         fac = (prevfpt->vec[0] - fpt->vec[0]) / (prevfpt->vec[0] - v[0]);
568                         if (fac) fac = 1.0f / fac;
569                         v[1] = prevfpt->vec[1] - fac * (prevfpt->vec[1] - fpt->vec[1]);
570                 }
571                 
572                 glVertex2fv(v);
573         }
574         
575         /* if only one sample, add it now */
576         if (fcu->totvert == 1)
577                 glVertex2fv(prevfpt->vec);
578         
579         /* loop over samples, drawing segments */
580         /* draw curve between first and last keyframe (if there are enough to do so) */
581         while (b--) {
582                 /* Linear interpolation: just add one point (which should add a new line segment) */
583                 glVertex2fv(prevfpt->vec);
584                 
585                 /* get next pointers */
586                 prevfpt = fpt;
587                 fpt++;
588                 
589                 /* last point? */
590                 if (b == 0)
591                         glVertex2fv(prevfpt->vec);
592         }
593         
594         /* extrapolate to right? (see code for left-extrapolation above too) */
595         if (prevfpt->vec[0] < v2d->cur.xmax) {
596                 v[0] = v2d->cur.xmax;
597                 
598                 /* y-value depends on the interpolation */
599                 if ((fcu->extend == FCURVE_EXTRAPOLATE_CONSTANT) || (fcu->flag & FCURVE_INT_VALUES) || (fcu->totvert == 1)) {
600                         /* based on last keyframe's value */
601                         v[1] = prevfpt->vec[1];
602                 } 
603                 else {
604                         /* extrapolate linear dosnt use the handle, use the previous points center instead */
605                         fpt = prevfpt - 1;
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                 glVertex2fv(v);
612         }
613         
614         /* unapply unit mapping */
615         ANIM_unit_mapping_apply_fcurve(ac->scene, id, fcu, ANIM_UNITCONV_RESTORE);
616         
617         glEnd();
618 }
619
620 /* helper func - draw one repeat of an F-Curve */
621 static void draw_fcurve_curve_bezts(bAnimContext *ac, ID *id, FCurve *fcu, View2D *v2d)
622 {
623         BezTriple *prevbezt = fcu->bezt;
624         BezTriple *bezt = prevbezt + 1;
625         float v1[2], v2[2], v3[2], v4[2];
626         float *fp, data[120];
627         float fac = 0.0f;
628         int b = fcu->totvert - 1;
629         int resol;
630         
631         glBegin(GL_LINE_STRIP);
632         
633         /* apply unit mapping */
634         ANIM_unit_mapping_apply_fcurve(ac->scene, id, fcu, 0);
635         
636         /* extrapolate to left? */
637         if (prevbezt->vec[1][0] > v2d->cur.xmin) {
638                 /* left-side of view comes before first keyframe, so need to extend as not cyclic */
639                 v1[0] = v2d->cur.xmin;
640                 
641                 /* y-value depends on the interpolation */
642                 if ((fcu->extend == FCURVE_EXTRAPOLATE_CONSTANT) || (prevbezt->ipo == BEZT_IPO_CONST) || (fcu->totvert == 1)) {
643                         /* just extend across the first keyframe's value */
644                         v1[1] = prevbezt->vec[1][1];
645                 } 
646                 else if (prevbezt->ipo == BEZT_IPO_LIN) {
647                         /* extrapolate linear dosnt use the handle, use the next points center instead */
648                         fac = (prevbezt->vec[1][0] - bezt->vec[1][0]) / (prevbezt->vec[1][0] - v1[0]);
649                         if (fac) fac = 1.0f / fac;
650                         v1[1] = prevbezt->vec[1][1] - fac * (prevbezt->vec[1][1] - bezt->vec[1][1]);
651                 } 
652                 else {
653                         /* based on angle of handle 1 (relative to keyframe) */
654                         fac = (prevbezt->vec[0][0] - prevbezt->vec[1][0]) / (prevbezt->vec[1][0] - v1[0]);
655                         if (fac) fac = 1.0f / fac;
656                         v1[1] = prevbezt->vec[1][1] - fac * (prevbezt->vec[0][1] - prevbezt->vec[1][1]);
657                 }
658                 
659                 glVertex2fv(v1);
660         }
661         
662         /* if only one keyframe, add it now */
663         if (fcu->totvert == 1) {
664                 v1[0] = prevbezt->vec[1][0];
665                 v1[1] = prevbezt->vec[1][1];
666                 glVertex2fv(v1);
667         }
668         
669         /* draw curve between first and last keyframe (if there are enough to do so) */
670         /* TODO: optimize this to not have to calc stuff out of view too? */
671         while (b--) {
672                 if (prevbezt->ipo == BEZT_IPO_CONST) {
673                         /* Constant-Interpolation: draw segment between previous keyframe and next, but holding same value */
674                         v1[0] = prevbezt->vec[1][0];
675                         v1[1] = prevbezt->vec[1][1];
676                         glVertex2fv(v1);
677                         
678                         v1[0] = bezt->vec[1][0];
679                         v1[1] = prevbezt->vec[1][1];
680                         glVertex2fv(v1);
681                 }
682                 else if (prevbezt->ipo == BEZT_IPO_LIN) {
683                         /* Linear interpolation: just add one point (which should add a new line segment) */
684                         v1[0] = prevbezt->vec[1][0];
685                         v1[1] = prevbezt->vec[1][1];
686                         glVertex2fv(v1);
687                 }
688                 else {
689                         /* Bezier-Interpolation: draw curve as series of segments between keyframes 
690                          *      - resol determines number of points to sample in between keyframes
691                          */
692                         
693                         /* resol depends on distance between points (not just horizontal) OR is a fixed high res */
694                         /* TODO: view scale should factor into this someday too... */
695                         if (fcu->driver) 
696                                 resol = 32;
697                         else 
698                                 resol = (int)(5.0f * len_v2v2(bezt->vec[1], prevbezt->vec[1]));
699                         
700                         if (resol < 2) {
701                                 /* only draw one */
702                                 v1[0] = prevbezt->vec[1][0];
703                                 v1[1] = prevbezt->vec[1][1];
704                                 glVertex2fv(v1);
705                         }
706                         else {
707                                 /* clamp resolution to max of 32 */
708                                 /* NOTE: higher values will crash */
709                                 if (resol > 32) resol = 32;
710                                 
711                                 v1[0] = prevbezt->vec[1][0];
712                                 v1[1] = prevbezt->vec[1][1];
713                                 v2[0] = prevbezt->vec[2][0];
714                                 v2[1] = prevbezt->vec[2][1];
715
716                                 v3[0] = bezt->vec[0][0];
717                                 v3[1] = bezt->vec[0][1];
718                                 v4[0] = bezt->vec[1][0];
719                                 v4[1] = bezt->vec[1][1];
720                                 
721                                 correct_bezpart(v1, v2, v3, v4);
722                                 
723                                 BKE_curve_forward_diff_bezier(v1[0], v2[0], v3[0], v4[0], data, resol, sizeof(float) * 3);
724                                 BKE_curve_forward_diff_bezier(v1[1], v2[1], v3[1], v4[1], data + 1, resol, sizeof(float) * 3);
725                                 
726                                 for (fp = data; resol; resol--, fp += 3)
727                                         glVertex2fv(fp);
728                         }
729                 }
730                 
731                 /* get next pointers */
732                 prevbezt = bezt;
733                 bezt++;
734                 
735                 /* last point? */
736                 if (b == 0) {
737                         v1[0] = prevbezt->vec[1][0];
738                         v1[1] = prevbezt->vec[1][1];
739                         glVertex2fv(v1);
740                 }
741         }
742         
743         /* extrapolate to right? (see code for left-extrapolation above too) */
744         if (prevbezt->vec[1][0] < v2d->cur.xmax) {
745                 v1[0] = v2d->cur.xmax;
746                 
747                 /* y-value depends on the interpolation */
748                 if ((fcu->extend == FCURVE_EXTRAPOLATE_CONSTANT) || (fcu->flag & FCURVE_INT_VALUES) || (prevbezt->ipo == BEZT_IPO_CONST) || (fcu->totvert == 1)) {
749                         /* based on last keyframe's value */
750                         v1[1] = prevbezt->vec[1][1];
751                 } 
752                 else if (prevbezt->ipo == BEZT_IPO_LIN) {
753                         /* extrapolate linear dosnt use the handle, use the previous points center instead */
754                         bezt = prevbezt - 1;
755                         fac = (prevbezt->vec[1][0] - bezt->vec[1][0]) / (prevbezt->vec[1][0] - v1[0]);
756                         if (fac) fac = 1.0f / fac;
757                         v1[1] = prevbezt->vec[1][1] - fac * (prevbezt->vec[1][1] - bezt->vec[1][1]);
758                 } 
759                 else {
760                         /* based on angle of handle 1 (relative to keyframe) */
761                         fac = (prevbezt->vec[2][0] - prevbezt->vec[1][0]) / (prevbezt->vec[1][0] - v1[0]);
762                         if (fac) fac = 1.0f / fac;
763                         v1[1] = prevbezt->vec[1][1] - fac * (prevbezt->vec[2][1] - prevbezt->vec[1][1]);
764                 }
765                 
766                 glVertex2fv(v1);
767         }
768         
769         /* unapply unit mapping */
770         ANIM_unit_mapping_apply_fcurve(ac->scene, id, fcu, ANIM_UNITCONV_RESTORE);
771         
772         glEnd();
773
774
775 /* Public Curve-Drawing API  ---------------- */
776
777 /* Draw the 'ghost' F-Curves (i.e. snapshots of the curve) 
778  * NOTE: unit mapping has already been applied to the values, so do not try and apply again
779  */
780 void graph_draw_ghost_curves(bAnimContext *ac, SpaceIpo *sipo, ARegion *ar)
781 {
782         FCurve *fcu;
783         
784         /* draw with thick dotted lines */
785         setlinestyle(10);
786         glLineWidth(3.0f);
787         
788         /* anti-aliased lines for less jagged appearance */
789         if ((sipo->flag & SIPO_BEAUTYDRAW_OFF) == 0) glEnable(GL_LINE_SMOOTH);
790         glEnable(GL_BLEND);
791         
792         /* the ghost curves are simply sampled F-Curves stored in sipo->ghostCurves */
793         for (fcu = sipo->ghostCurves.first; fcu; fcu = fcu->next) {
794                 /* set whatever color the curve has set 
795                  *  - this is set by the function which creates these
796                  *      - draw with a fixed opacity of 2
797                  */
798                 glColor4f(fcu->color[0], fcu->color[1], fcu->color[2], 0.5f);
799                 
800                 /* simply draw the stored samples */
801                 draw_fcurve_curve_samples(ac, NULL, fcu, &ar->v2d);
802         }
803         
804         /* restore settings */
805         setlinestyle(0);
806         glLineWidth(1.0f);
807         
808         if ((sipo->flag & SIPO_BEAUTYDRAW_OFF) == 0) glDisable(GL_LINE_SMOOTH);
809         glDisable(GL_BLEND);
810 }
811
812 /* This is called twice from space_graph.c -> graph_main_area_draw()
813  * Unselected then selected F-Curves are drawn so that they do not occlude each other.
814  */
815 void graph_draw_curves(bAnimContext *ac, SpaceIpo *sipo, ARegion *ar, View2DGrid *grid, short sel)
816 {
817         ListBase anim_data = {NULL, NULL};
818         bAnimListElem *ale;
819         int filter;
820         
821         /* build list of curves to draw */
822         filter = (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_CURVE_VISIBLE);
823         filter |= ((sel) ? (ANIMFILTER_SEL) : (ANIMFILTER_UNSEL));
824         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
825                 
826         /* for each curve:
827          *      draw curve, then handle-lines, and finally vertices in this order so that 
828          *  the data will be layered correctly
829          */
830         for (ale = anim_data.first; ale; ale = ale->next) {
831                 FCurve *fcu = (FCurve *)ale->key_data;
832                 FModifier *fcm = find_active_fmodifier(&fcu->modifiers);
833                 AnimData *adt = ANIM_nla_mapping_get(ac, ale);
834                 
835                 /* map keyframes for drawing if scaled F-Curve */
836                 if (adt)
837                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 0); 
838                 
839                 /* draw curve:
840                  *      - curve line may be result of one or more destructive modifiers or just the raw data,
841                  *        so we need to check which method should be used
842                  *      - controls from active modifier take precedence over keyframes
843                  *        (XXX! editing tools need to take this into account!)
844                  */
845                  
846                 /* 1) draw curve line */
847                 {
848                         /* set color/drawing style for curve itself */
849                         if (((fcu->grp) && (fcu->grp->flag & AGRP_PROTECTED)) || (fcu->flag & FCURVE_PROTECTED)) {
850                                 /* protected curves (non editable) are drawn with dotted lines */
851                                 setlinestyle(2);
852                         }
853                         if (((fcu->grp) && (fcu->grp->flag & AGRP_MUTED)) || (fcu->flag & FCURVE_MUTED)) {
854                                 /* muted curves are drawn in a grayish hue */
855                                 /* XXX should we have some variations? */
856                                 UI_ThemeColorShade(TH_HEADER, 50);
857                         }
858                         else {
859                                 /* set whatever color the curve has set 
860                                  *      - unselected curves draw less opaque to help distinguish the selected ones
861                                  */
862                                 glColor4f(fcu->color[0], fcu->color[1], fcu->color[2], fcurve_display_alpha(fcu));
863                         }
864                         
865                         /* draw active F-Curve thicker than the rest to make it stand out */
866                         if (fcu->flag & FCURVE_ACTIVE) {
867                                 glLineWidth(2.0);
868                         }
869                         
870                         /* anti-aliased lines for less jagged appearance */
871                         if ((sipo->flag & SIPO_BEAUTYDRAW_OFF) == 0) glEnable(GL_LINE_SMOOTH);
872                         glEnable(GL_BLEND);
873                         
874                         /* draw F-Curve */
875                         if ((fcu->modifiers.first) || (fcu->flag & FCURVE_INT_VALUES)) {
876                                 /* draw a curve affected by modifiers or only allowed to have integer values 
877                                  * by sampling it at various small-intervals over the visible region 
878                                  */
879                                 draw_fcurve_curve(ac, ale->id, fcu, &ar->v2d, grid);
880                         }
881                         else if (((fcu->bezt) || (fcu->fpt)) && (fcu->totvert)) {
882                                 /* just draw curve based on defined data (i.e. no modifiers) */
883                                 if (fcu->bezt)
884                                         draw_fcurve_curve_bezts(ac, ale->id, fcu, &ar->v2d);
885                                 else if (fcu->fpt)
886                                         draw_fcurve_curve_samples(ac, ale->id, fcu, &ar->v2d);
887                         }
888                         
889                         /* restore settings */
890                         setlinestyle(0);
891                         glLineWidth(1.0);
892                         
893                         if ((sipo->flag & SIPO_BEAUTYDRAW_OFF) == 0) glDisable(GL_LINE_SMOOTH);
894                         glDisable(GL_BLEND);
895                 }
896                 
897                 /* 2) draw handles and vertices as appropriate based on active 
898                  *      - if the option to only show controls if the F-Curve is selected is enabled, we must obey this
899                  */
900                 if (!(sipo->flag & SIPO_SELCUVERTSONLY) || (fcu->flag & FCURVE_SELECTED)) {
901                         if (fcurve_are_keyframes_usable(fcu) == 0) {
902                                 /* only draw controls if this is the active modifier */
903                                 if ((fcu->flag & FCURVE_ACTIVE) && (fcm)) {
904                                         switch (fcm->type) {
905                                                 case FMODIFIER_TYPE_ENVELOPE: /* envelope */
906                                                         draw_fcurve_modifier_controls_envelope(fcm, &ar->v2d);
907                                                         break;
908                                         }
909                                 }
910                         }
911                         else if (((fcu->bezt) || (fcu->fpt)) && (fcu->totvert)) {
912                                 /* apply unit mapping */
913                                 ANIM_unit_mapping_apply_fcurve(ac->scene, ale->id, fcu, 0);
914                                 
915                                 if (fcu->bezt) {
916                                         int do_handles = draw_fcurve_handles_check(sipo, fcu);
917                                         
918                                         if (do_handles) {
919                                                 /* only draw handles/vertices on keyframes */
920                                                 glEnable(GL_BLEND);
921                                                 draw_fcurve_handles(sipo, fcu);
922                                                 glDisable(GL_BLEND);
923                                         }
924                                         
925                                         draw_fcurve_vertices(sipo, ar, fcu, do_handles, (sipo->flag & SIPO_SELVHANDLESONLY));
926                                 }
927                                 else {
928                                         /* samples: only draw two indicators at either end as indicators */
929                                         draw_fcurve_samples(sipo, ar, fcu);
930                                 }
931                                 
932                                 /* unapply unit mapping */
933                                 ANIM_unit_mapping_apply_fcurve(ac->scene, ale->id, fcu, ANIM_UNITCONV_RESTORE);
934                         }
935                 }
936                 
937                 /* undo mapping of keyframes for drawing if scaled F-Curve */
938                 if (adt)
939                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 0); 
940         }
941         
942         /* free list of curves */
943         BLI_freelistN(&anim_data);
944 }
945
946 /* ************************************************************************* */
947 /* Channel List */
948
949 /* left hand part */
950 void graph_draw_channel_names(bContext *C, bAnimContext *ac, ARegion *ar) 
951 {
952         ListBase anim_data = {NULL, NULL};
953         bAnimListElem *ale;
954         int filter;
955         
956         View2D *v2d = &ar->v2d;
957         float y = 0.0f, height;
958         size_t items;
959         int i = 0;
960         
961         /* build list of channels to draw */
962         filter = (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE | ANIMFILTER_LIST_CHANNELS);
963         items = ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
964         
965         /* Update max-extent of channels here (taking into account scrollers):
966          *  - this is done to allow the channel list to be scrollable, but must be done here
967          *    to avoid regenerating the list again and/or also because channels list is drawn first
968          *      - offset of ACHANNEL_HEIGHT*2 is added to the height of the channels, as first is for 
969          *        start of list offset, and the second is as a correction for the scrollers.
970          */
971         height = (float)((items * ACHANNEL_STEP) + (ACHANNEL_HEIGHT * 2));
972         UI_view2d_totRect_set(v2d, ar->winx, height);
973         
974         /* loop through channels, and set up drawing depending on their type  */        
975         {   /* first pass: just the standard GL-drawing for backdrop + text */
976                 y = (float)ACHANNEL_FIRST;
977                 
978                 for (ale = anim_data.first, i = 0; ale; ale = ale->next, i++) {
979                         const float yminc = (float)(y - ACHANNEL_HEIGHT_HALF);
980                         const float ymaxc = (float)(y + ACHANNEL_HEIGHT_HALF);
981                         
982                         /* check if visible */
983                         if (IN_RANGE(yminc, v2d->cur.ymin, v2d->cur.ymax) ||
984                             IN_RANGE(ymaxc, v2d->cur.ymin, v2d->cur.ymax) )
985                         {
986                                 /* draw all channels using standard channel-drawing API */
987                                 ANIM_channel_draw(ac, ale, yminc, ymaxc);
988                         }
989                         
990                         /* adjust y-position for next one */
991                         y -= ACHANNEL_STEP;
992                 }
993         }
994         {   /* second pass: widgets */
995                 uiBlock *block = uiBeginBlock(C, ar, __func__, UI_EMBOSS);
996                 size_t channel_index = 0;
997                 
998                 y = (float)ACHANNEL_FIRST;
999                 
1000                 /* set blending again, as may not be set in previous step */
1001                 glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
1002                 glEnable(GL_BLEND);
1003                 
1004                 for (ale = anim_data.first, i = 0; ale; ale = ale->next, i++) {
1005                         const float yminc = (float)(y - ACHANNEL_HEIGHT_HALF);
1006                         const float ymaxc = (float)(y + ACHANNEL_HEIGHT_HALF);
1007                         
1008                         /* check if visible */
1009                         if (IN_RANGE(yminc, v2d->cur.ymin, v2d->cur.ymax) ||
1010                             IN_RANGE(ymaxc, v2d->cur.ymin, v2d->cur.ymax) )
1011                         {
1012                                 /* draw all channels using standard channel-drawing API */
1013                                 ANIM_channel_draw_widgets(C, ac, ale, block, yminc, ymaxc, channel_index);
1014                         }
1015                         
1016                         /* adjust y-position for next one */
1017                         y -= ACHANNEL_STEP;
1018                         channel_index++;
1019                 }
1020                 
1021                 uiEndBlock(C, block);
1022                 uiDrawBlock(C, block);
1023                 
1024                 glDisable(GL_BLEND);
1025         }
1026         
1027         /* free tempolary channels */
1028         BLI_freelistN(&anim_data);
1029 }