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