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