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