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