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