Animation Channels Drawing Tweak:
[blender.git] / source / blender / editors / include / ED_anim_api.h
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) 2008 Blender Foundation.
21  * All rights reserved.
22  *
23  * 
24  * Contributor(s): Joshua Leung
25  *
26  * ***** END GPL LICENSE BLOCK *****
27  */
28
29 #ifndef ED_ANIM_API_H
30 #define ED_ANIM_API_H
31
32 struct ID;
33 struct ListBase;
34 struct AnimData;
35
36 struct bContext;
37 struct wmKeyConfig;
38 struct ScrArea;
39 struct ARegion;
40 struct View2D;
41
42 struct Scene;
43 struct Object;
44
45 struct bDopeSheet;
46
47 struct bActionGroup;
48 struct FCurve;
49 struct FModifier;
50
51 struct uiBlock;
52 struct uiLayout;
53
54 /* ************************************************ */
55 /* ANIMATION CHANNEL FILTERING */
56 /* anim_filter.c */
57
58 /* --------------- Context --------------------- */
59
60 /* This struct defines a structure used for animation-specific 
61  * 'context' information
62  */
63 typedef struct bAnimContext {
64         void *data;                             /* data to be filtered for use in animation editor */
65         short datatype;                 /* type of data eAnimCont_Types */
66         
67         short mode;                             /* editor->mode */
68         short spacetype;                /* sa->spacetype */
69         short regiontype;               /* active region -> type (channels or main) */
70         struct ScrArea *sa;             /* editor */ 
71         struct ARegion *ar;             /* region within editor */
72         
73         struct Scene *scene;    /* active scene */
74         struct Object *obact;   /* active object */
75         ListBase *markers;              /* active set of markers */
76 } bAnimContext;
77
78 /* Main Data container types */
79 // XXX was ACTCONT_*
80 typedef enum eAnimCont_Types {
81         ANIMCONT_NONE = 0,              /* invalid or no data */
82         ANIMCONT_ACTION,                /* action (bAction) */
83         ANIMCONT_SHAPEKEY,              /* shapekey (Key) */
84         ANIMCONT_GPENCIL,               /* grease pencil (screen) */
85         ANIMCONT_DOPESHEET,             /* dopesheet (bDopesheet) */
86         ANIMCONT_FCURVES,               /* animation F-Curves (bDopesheet) */
87         ANIMCONT_DRIVERS,               /* drivers (bDopesheet) */
88         ANIMCONT_NLA,                   /* nla (bDopesheet) */
89 } eAnimCont_Types;
90
91 /* --------------- Channels -------------------- */
92
93 /* This struct defines a structure used for quick and uniform access for 
94  * channels of animation data
95  */
96 typedef struct bAnimListElem {
97         struct bAnimListElem *next, *prev;
98         
99         void    *data;                  /* source data this elem represents */
100         int     type;                   /* one of the ANIMTYPE_* values */
101         int             flag;                   /* copy of elem's flags for quick access */
102         int     index;                  /* for un-named data, the index of the data in it's collection */
103         
104         void    *key_data;              /* motion data - mostly F-Curves, but can be other types too */
105         short   datatype;               /* type of motion data to expect */
106         
107         struct ID *id;                  /* ID block that channel is attached to */
108         struct AnimData *adt;   /* source of the animation data attached to ID block (for convenience) */
109         
110         void    *owner;                 /* group or channel which acts as this channel's owner */
111         short   ownertype;              /* type of owner */
112 } bAnimListElem;
113
114
115 /* Some types for easier type-testing 
116  * NOTE: need to keep the order of these synchronised with the channels define code
117  *              which is used for drawing and handling channel lists for 
118  */
119 // XXX was ACTTYPE_*
120 typedef enum eAnim_ChannelType {
121         ANIMTYPE_NONE= 0,
122         ANIMTYPE_ANIMDATA,
123         ANIMTYPE_SPECIALDATA,
124         
125         ANIMTYPE_SUMMARY,
126         
127         ANIMTYPE_SCENE,
128         ANIMTYPE_OBJECT,
129         ANIMTYPE_GROUP,
130         ANIMTYPE_FCURVE,
131         
132         ANIMTYPE_FILLACTD,
133         ANIMTYPE_FILLDRIVERS,
134         ANIMTYPE_FILLMATD,
135         ANIMTYPE_FILLPARTD,
136         
137         ANIMTYPE_DSMAT,
138         ANIMTYPE_DSLAM,
139         ANIMTYPE_DSCAM,
140         ANIMTYPE_DSCUR,
141         ANIMTYPE_DSSKEY,
142         ANIMTYPE_DSWOR,
143         ANIMTYPE_DSNTREE,
144         ANIMTYPE_DSPART,
145         ANIMTYPE_DSMBALL,
146         ANIMTYPE_DSARM,
147         ANIMTYPE_DSMESH,
148         
149         ANIMTYPE_SHAPEKEY,
150         
151         ANIMTYPE_GPDATABLOCK,
152         ANIMTYPE_GPLAYER,
153         
154         ANIMTYPE_NLATRACK,
155         ANIMTYPE_NLAACTION,
156         
157                 /* always as last item, the total number of channel types... */
158         ANIMTYPE_NUM_TYPES,
159 } eAnim_ChannelType;
160
161 /* types of keyframe data in bAnimListElem */
162 typedef enum eAnim_KeyType {
163         ALE_NONE = 0,           /* no keyframe data */
164         ALE_FCURVE,                     /* F-Curve */
165         ALE_GPFRAME,            /* Grease Pencil Frames */
166         ALE_NLASTRIP,           /* NLA Strips */
167         
168         ALE_ALL,                        /* All channels summary */
169         ALE_SCE,                        /* Scene summary */
170         ALE_OB,                         /* Object summary */
171         ALE_ACT,                        /* Action summary */
172         ALE_GROUP,                      /* Action Group summary */
173 } eAnim_KeyType;
174
175 /* ----------------- Filtering -------------------- */
176
177 /* filtering flags  - under what circumstances should a channel be added */
178 // XXX was ACTFILTER_*
179 typedef enum eAnimFilter_Flags {
180         ANIMFILTER_VISIBLE              = (1<<0),       /* should channels be visible (in terms of hierarchy only) */
181         ANIMFILTER_SEL                  = (1<<1),       /* should channels be selected */
182         ANIMFILTER_UNSEL                = (1<<2),       /* should channels be NOT selected */
183         ANIMFILTER_FOREDIT              = (1<<3),       /* does editable status matter */
184         ANIMFILTER_CURVESONLY   = (1<<4),       /* don't include summary-channels, etc. */
185         ANIMFILTER_CHANNELS             = (1<<5),       /* make list for interface drawing */
186         ANIMFILTER_ACTGROUPED   = (1<<6),       /* belongs to the active actiongroup */
187         ANIMFILTER_CURVEVISIBLE = (1<<7),       /* F-Curve is visible for editing/viewing in Graph Editor */
188         ANIMFILTER_ACTIVE               = (1<<8),       /* channel should be 'active' */
189         ANIMFILTER_ANIMDATA             = (1<<9),       /* only return the underlying AnimData blocks (not the tracks, etc.) data comes from */
190         ANIMFILTER_NLATRACKS    = (1<<10),      /* only include NLA-tracks */
191         ANIMFILTER_SELEDIT              = (1<<11),      /* link editability with selected status */
192         
193         /* all filters - the power inside the bracket must be the last power for left-shifts + 1 */
194         ANIMFILTER_ALLFILTERS   = ((1<<12) - 1)
195 } eAnimFilter_Flags;
196
197
198 /* ---------- Flag Checking Macros ------------ */
199 // xxx check on all of these flags again...
200
201 /* Dopesheet only */
202         /* 'Scene' channels */
203 #define SEL_SCEC(sce) ((sce->flag & SCE_DS_SELECTED))
204 #define EXPANDED_SCEC(sce) ((sce->flag & SCE_DS_COLLAPSED)==0)
205         /* 'Sub-Scene' channels (flags stored in Data block) */
206 #define FILTER_WOR_SCED(wo) ((wo->flag & WO_DS_EXPAND))
207 #define FILTER_NTREE_SCED(ntree) ((ntree->flag & NTREE_DS_EXPAND))
208         /* 'Object' channels */
209 #define SEL_OBJC(base) ((base->flag & SELECT))
210 #define EXPANDED_OBJC(ob) ((ob->nlaflag & OB_ADS_COLLAPSED)==0)
211         /* 'Sub-object' channels (flags stored in Object block) */
212 #define FILTER_MAT_OBJC(ob) ((ob->nlaflag & OB_ADS_SHOWMATS))
213 #define FILTER_PART_OBJC(ob) ((ob->nlaflag & OB_ADS_SHOWPARTS))
214         /* 'Sub-object' channels (flags stored in Data block) */
215 #define FILTER_SKE_OBJD(key) ((key->flag & KEY_DS_EXPAND))
216 #define FILTER_MAT_OBJD(ma) ((ma->flag & MA_DS_EXPAND))
217 #define FILTER_LAM_OBJD(la) ((la->flag & LA_DS_EXPAND))
218 #define FILTER_CAM_OBJD(ca) ((ca->flag & CAM_DS_EXPAND))
219 #define FILTER_CUR_OBJD(cu) ((cu->flag & CU_DS_EXPAND))
220 #define FILTER_PART_OBJD(part) ((part->flag & PART_DS_EXPAND))
221 #define FILTER_MBALL_OBJD(mb) ((mb->flag2 & MB_DS_EXPAND))
222 #define FILTER_ARM_OBJD(arm) ((arm->flag & ARM_DS_EXPAND))
223 #define FILTER_MESH_OBJD(me) ((me->flag & ME_DS_EXPAND))
224         /* 'Sub-object/Action' channels (flags stored in Action) */
225 #define SEL_ACTC(actc) ((actc->flag & ACT_SELECTED))
226 #define EXPANDED_ACTC(actc) ((actc->flag & ACT_COLLAPSED)==0)
227         /* 'Sub-AnimData' chanenls */
228 #define EXPANDED_DRVD(adt) ((adt->flag & ADT_DRIVERS_COLLAPSED)==0)
229
230 /* Actions (also used for Dopesheet) */
231         /* Action Channel Group */
232 #define EDITABLE_AGRP(agrp) ((agrp->flag & AGRP_PROTECTED)==0)
233 #define EXPANDED_AGRP(agrp) (agrp->flag & AGRP_EXPANDED)
234 #define SEL_AGRP(agrp) ((agrp->flag & AGRP_SELECTED) || (agrp->flag & AGRP_ACTIVE))
235         /* F-Curve Channels */
236 #define EDITABLE_FCU(fcu) ((fcu->flag & FCURVE_PROTECTED)==0)
237 #define SEL_FCU(fcu) (fcu->flag & (FCURVE_ACTIVE|FCURVE_SELECTED))
238
239 /* ShapeKey mode only */
240 #define EDITABLE_SHAPEKEY(kb) ((kb->flag & KEYBLOCK_LOCKED)==0)
241 #define SEL_SHAPEKEY(kb) (kb->flag & KEYBLOCK_SEL)
242
243 /* Grease Pencil only */
244         /* Grease Pencil datablock settings */
245 #define EXPANDED_GPD(gpd) (gpd->flag & GP_DATA_EXPAND) 
246         /* Grease Pencil Layer settings */
247 #define EDITABLE_GPL(gpl) ((gpl->flag & GP_LAYER_LOCKED)==0)
248 #define SEL_GPL(gpl) ((gpl->flag & GP_LAYER_ACTIVE) || (gpl->flag & GP_LAYER_SELECT))
249
250 /* NLA only */
251 #define SEL_NLT(nlt) (nlt->flag & NLATRACK_SELECTED)
252 #define EDITABLE_NLT(nlt) ((nlt->flag & NLATRACK_PROTECTED)==0)
253
254 /* -------------- Channel Defines -------------- */
255
256 /* channel heights */
257 #define ACHANNEL_FIRST                  -16
258 #define ACHANNEL_HEIGHT                 16
259 #define ACHANNEL_HEIGHT_HALF    8
260 #define ACHANNEL_SKIP                   2
261 #define ACHANNEL_STEP                   (ACHANNEL_HEIGHT + ACHANNEL_SKIP)
262
263 /* channel widths */
264 #define ACHANNEL_NAMEWIDTH              200
265
266 /* channel toggle-buttons */
267 #define ACHANNEL_BUTTON_WIDTH   16
268
269
270 /* -------------- NLA Channel Defines -------------- */
271
272 /* NLA channel heights */
273 #define NLACHANNEL_FIRST                        -16
274 #define NLACHANNEL_HEIGHT                       24
275 #define NLACHANNEL_HEIGHT_HALF  12
276 #define NLACHANNEL_SKIP                 2
277 #define NLACHANNEL_STEP                 (NLACHANNEL_HEIGHT + NLACHANNEL_SKIP)
278
279 /* channel widths */
280 #define NLACHANNEL_NAMEWIDTH            200
281
282 /* channel toggle-buttons */
283 #define NLACHANNEL_BUTTON_WIDTH 16
284
285 /* ---------------- API  -------------------- */
286
287 /* Obtain list of filtered Animation channels to operate on.
288  * Returns the number of channels in the list
289  */
290 int ANIM_animdata_filter(bAnimContext *ac, ListBase *anim_data, int filter_mode, void *data, short datatype);
291
292 /* Obtain current anim-data context from Blender Context info.
293  * Returns whether the operation was successful. 
294  */
295 short ANIM_animdata_get_context(const struct bContext *C, bAnimContext *ac);
296
297 /* Obtain current anim-data context (from Animation Editor) given 
298  * that Blender Context info has already been set. 
299  * Returns whether the operation was successful.
300  */
301 short ANIM_animdata_context_getdata(bAnimContext *ac);
302
303 /* ************************************************ */
304 /* ANIMATION CHANNELS LIST */
305 /* anim_channels_*.c */
306
307 /* ------------------------ Drawing TypeInfo -------------------------- */
308
309 /* flag-setting behaviour */
310 typedef enum eAnimChannels_SetFlag {
311         ACHANNEL_SETFLAG_CLEAR = 0,
312         ACHANNEL_SETFLAG_ADD,
313         ACHANNEL_SETFLAG_TOGGLE
314 } eAnimChannels_SetFlag;
315
316 /* types of settings for AnimChannels */
317 typedef enum eAnimChannel_Settings {
318         ACHANNEL_SETTING_SELECT = 0,
319         ACHANNEL_SETTING_PROTECT,                       // warning: for drawing UI's, need to check if this is off (maybe inverse this later)
320         ACHANNEL_SETTING_MUTE,
321         ACHANNEL_SETTING_EXPAND,
322         ACHANNEL_SETTING_VISIBLE,                       /* only for Graph Editor */
323         ACHANNEL_SETTING_SOLO,                          /* only for NLA Tracks */
324 } eAnimChannel_Settings;
325
326
327 /* Drawing, mouse handling, and flag setting behaviour... */
328 typedef struct bAnimChannelType {
329         /* drawing */
330                 /* get RGB color that is used to draw the majority of the backdrop */
331         void (*get_backdrop_color)(bAnimContext *ac, bAnimListElem *ale, float *color);
332                 /* draw backdrop strip for channel */
333         void (*draw_backdrop)(bAnimContext *ac, bAnimListElem *ale, float yminc, float ymaxc);
334                 /* get depth of indention (relative to the depth channel is nested at) */
335         short (*get_indent_level)(bAnimContext *ac, bAnimListElem *ale);
336                 /* get offset in pixels for the start of the channel (in addition to the indent depth) */
337         short (*get_offset)(bAnimContext *ac, bAnimListElem *ale);
338         
339         /* get name (for channel lists) */
340         void (*name)(bAnimListElem *ale, char *name);
341         /* get icon (for channel lists) */
342         int (*icon)(bAnimListElem *ale);
343         
344         /* settings */
345                 /* check if the given setting is valid in the current context */
346         short (*has_setting)(bAnimContext *ac, bAnimListElem *ale, int setting);
347                 /* get the flag used for this setting */
348         int (*setting_flag)(int setting, short *neg);
349                 /* get the pointer to int/short where data is stored, 
350                  * with type being  sizeof(ptr_data) which should be fine for runtime use...
351                  *      - assume that setting has been checked to be valid for current context
352                  */
353         void *(*setting_ptr)(bAnimListElem *ale, int setting, short *type);
354 } bAnimChannelType;
355
356 /* ------------------------ Drawing API -------------------------- */
357
358 /* Get typeinfo for the given channel */
359 bAnimChannelType *ANIM_channel_get_typeinfo(bAnimListElem *ale);
360
361 /* Draw the given channel */
362 void ANIM_channel_draw(bAnimContext *ac, bAnimListElem *ale, float yminc, float ymaxc);
363 /* Draw the widgets for the given channel */
364 void ANIM_channel_draw_widgets(bAnimContext *ac, bAnimListElem *ale, struct uiBlock *block, float yminc, float ymaxc);
365
366
367 /* ------------------------ Editing API -------------------------- */
368
369 /* Check if some setting for a channel is enabled 
370  * Returns: 1 = On, 0 = Off, -1 = Invalid
371  *
372  *      - setting: eAnimChannel_Settings
373  */
374 short ANIM_channel_setting_get(bAnimContext *ac, bAnimListElem *ale, int setting);
375
376 /* Change value of some setting for a channel 
377  *      - setting: eAnimChannel_Settings
378  *      - mode: eAnimChannels_SetFlag
379  */
380 void ANIM_channel_setting_set(bAnimContext *ac, bAnimListElem *ale, int setting, short mode);
381
382
383 /* Flush visibility (for Graph Editor) changes up/down hierarchy for changes in the given setting 
384  *      - anim_data: list of the all the anim channels that can be chosen
385  *              -> filtered using ANIMFILTER_CHANNELS only, since if we took VISIBLE too,
386  *                then the channels under closed expanders get ignored...
387  *      - ale_setting: the anim channel (not in the anim_data list directly, though occuring there)
388  *              with the new state of the setting that we want flushed up/down the hierarchy 
389  *      - vizOn: whether the visibility setting has been enabled or disabled 
390  */
391 void ANIM_visibility_flush_anim_channels(bAnimContext *ac, ListBase *anim_data, bAnimListElem *ale_setting, short vizOn);
392
393
394 /* Deselect all animation channels */
395 void ANIM_deselect_anim_channels(void *data, short datatype, short test, short sel);
396
397 /* Set the 'active' channel of type channel_type, in the given action */
398 void ANIM_set_active_channel(bAnimContext *ac, void *data, short datatype, int filter, void *channel_data, short channel_type);
399
400
401 /* Delete the F-Curve from the given AnimData block (if possible), as appropriate according to animation context */
402 void ANIM_fcurve_delete_from_animdata(bAnimContext *ac, struct AnimData *adt, struct FCurve *fcu);
403
404 /* ************************************************ */
405 /* DRAWING API */
406 /* anim_draw.c */
407
408 /* ---------- Current Frame Drawing ---------------- */
409
410 /* flags for Current Frame Drawing */
411 enum {
412                 /* plain time indicator with no special indicators */
413         DRAWCFRA_PLAIN                  = 0,
414                 /* draw box indicating current frame number */
415         DRAWCFRA_SHOW_NUMBOX    = (1<<0),
416                 /* time indication in seconds or frames */
417         DRAWCFRA_UNIT_SECONDS   = (1<<1),
418                 /* show time-offset line */
419         DRAWCFRA_SHOW_TIMEOFS   = (1<<2),
420 } eAnimEditDraw_CurrentFrame; 
421
422 /* main call to draw current-frame indicator in an Animation Editor */
423 void ANIM_draw_cfra(const struct bContext *C, struct View2D *v2d, short flag);
424
425 /* ------------- Preview Range Drawing -------------- */
426
427 /* main call to draw preview range curtains */
428 void ANIM_draw_previewrange(const struct bContext *C, struct View2D *v2d);
429
430 /* ************************************************* */
431 /* F-MODIFIER TOOLS */
432
433 /* draw a given F-Modifier for some layout/UI-Block */
434 void ANIM_uiTemplate_fmodifier_draw(struct uiLayout *layout, struct ID *id, ListBase *modifiers, struct FModifier *fcm);
435
436 /* ************************************************* */
437 /* ASSORTED TOOLS */
438
439 /* ------------ Animation F-Curves <-> Icons/Names Mapping ------------ */
440 /* anim_ipo_utils.c */
441
442 /* Get icon + name for channel-list displays for F-Curve */
443 int getname_anim_fcurve(char *name, struct ID *id, struct FCurve *fcu);
444
445 /* Automatically determine a color for the nth F-Curve */
446 void getcolor_fcurve_rainbow(int cur, int tot, float *out);
447
448 /* ------------- NLA-Mapping ----------------------- */
449 /* anim_draw.c */
450
451 /* Obtain the AnimData block providing NLA-scaling for the given channel if applicable */
452 struct AnimData *ANIM_nla_mapping_get(bAnimContext *ac, bAnimListElem *ale);
453
454 /* Apply/Unapply NLA mapping to all keyframes in the nominated F-Curve */
455 void ANIM_nla_mapping_apply_fcurve(struct AnimData *adt, struct FCurve *fcu, short restore, short only_keys);
456
457 /* ..... */
458
459 /* Perform auto-blending/extend refreshes after some operations */
460 // NOTE: defined in space_nla/nla_edit.c, not in animation/
461 void ED_nla_postop_refresh(bAnimContext *ac);
462
463 /* ------------- Utility macros ----------------------- */
464
465 /* provide access to Keyframe Type info in BezTriple
466  * NOTE: this is so that we can change it from being stored in 'hide'
467  */
468 #define BEZKEYTYPE(bezt) ((bezt)->hide)
469
470 /* set/clear/toggle macro 
471  *      - channel - channel with a 'flag' member that we're setting
472  *      - smode - 0=clear, 1=set, 2=toggle
473  *      - sflag - bitflag to set
474  */
475 #define ACHANNEL_SET_FLAG(channel, smode, sflag) \
476         { \
477                 if (smode == ACHANNEL_SETFLAG_TOGGLE)   (channel)->flag ^= (sflag); \
478                 else if (smode == ACHANNEL_SETFLAG_ADD) (channel)->flag |= (sflag); \
479                 else                                                                    (channel)->flag &= ~(sflag); \
480         }
481         
482 /* set/clear/toggle macro, where the flag is negative 
483  *      - channel - channel with a 'flag' member that we're setting
484  *      - smode - 0=clear, 1=set, 2=toggle
485  *      - sflag - bitflag to set
486  */
487 #define ACHANNEL_SET_FLAG_NEG(channel, smode, sflag) \
488         { \
489                 if (smode == ACHANNEL_SETFLAG_TOGGLE)   (channel)->flag ^= (sflag); \
490                 else if (smode == ACHANNEL_SETFLAG_ADD) (channel)->flag &= ~(sflag); \
491                 else                                                                    (channel)->flag |= (sflag); \
492         }
493
494
495 /* --------- anim_deps.c, animation updates -------- */
496
497 void ANIM_id_update(struct Scene *scene, struct ID *id);
498 void ANIM_list_elem_update(struct Scene *scene, bAnimListElem *ale);
499
500 /* data -> channels syncing */
501 void ANIM_sync_animchannels_to_data(const struct bContext *C);
502
503 /* ************************************************* */
504 /* OPERATORS */
505         
506         /* generic animation channels */
507 void ED_operatortypes_animchannels(void);
508 void ED_keymap_animchannels(struct wmKeyConfig *keyconf);
509
510         /* generic time editing */
511 void ED_operatortypes_anim(void);
512 void ED_keymap_anim(struct wmKeyConfig *keyconf);
513
514 /* ************************************************ */
515
516 #endif /* ED_ANIM_API_H */
517