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