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