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