Merge branch 'master' into blender2.8
[blender.git] / source / blender / blenkernel / BKE_node.h
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2005 Blender Foundation.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): Bob Holcomb.
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 #ifndef __BKE_NODE_H__
29 #define __BKE_NODE_H__
30
31 /** \file BKE_node.h
32  *  \ingroup bke
33  */
34
35 #include "BLI_utildefines.h"
36 #include "BLI_ghash.h"
37
38 #include "DNA_listBase.h"
39
40 /* for FOREACH_NODETREE */
41 #include "DNA_lamp_types.h"
42 #include "DNA_material_types.h"
43 #include "DNA_node_types.h"
44 #include "DNA_scene_types.h"
45 #include "DNA_texture_types.h"
46 #include "DNA_world_types.h"
47 #include "DNA_linestyle_types.h"
48
49 #include "RNA_types.h"
50
51 /* not very important, but the stack solver likes to know a maximum */
52 #define MAX_SOCKET      512
53
54 struct bContext;
55 struct bNode;
56 struct bNodeLink;
57 struct bNodeSocket;
58 struct bNodeStack;
59 struct bNodeTree;
60 struct bNodeTreeType;
61 struct bNodeTreeExec;
62 struct bNodeExecContext;
63 struct bNodeExecData;
64 struct GPUMaterial;
65 struct GPUNodeStack;
66 struct ID;
67 struct ImBuf;
68 struct ImageFormatData;
69 struct ListBase;
70 struct Main;
71 struct uiLayout;
72 struct MTex;
73 struct PointerRNA;
74 struct RenderData;
75 struct Scene;
76 struct Tex;
77 struct ViewRender;
78 struct SpaceNode;
79 struct ARegion;
80 struct ColorManagedViewSettings;
81 struct ColorManagedDisplaySettings;
82 struct bNodeInstanceHash;
83
84 /* -------------------------------------------------------------------- */
85 /** \name Node Type Definitions
86  * \{ */
87
88 /** Compact definition of a node socket.
89  * Can be used to quickly define a list of static sockets for a node,
90  * which are added to each new node of that type.
91  *
92  * \deprecated This struct is used by C nodes to define templates as simple
93  * static struct lists. These are converted to the new template collections
94  * in RNA types automatically.
95  */
96 typedef struct bNodeSocketTemplate {
97         int type, limit;
98         char name[64];  /* MAX_NAME */
99         float val1, val2, val3, val4;   /* default alloc value for inputs */
100         float min, max;
101         int subtype;  /* would use PropertySubType but this is a bad level include to use RNA */
102         int flag;
103         
104         /* after this line is used internal only */
105         struct bNodeSocket *sock;               /* used to hold verified socket */
106         char identifier[64];                    /* generated from name */
107 } bNodeSocketTemplate;
108
109 /** Defines a socket type.
110  * Defines the appearance and behavior of a socket in the UI.
111  */
112 typedef struct bNodeSocketType {
113         char idname[64];                                /* identifier name */
114         
115         void (*draw)(struct bContext *C, struct uiLayout *layout, struct PointerRNA *ptr, struct PointerRNA *node_ptr, const char *text);
116         void (*draw_color)(struct bContext *C, struct PointerRNA *ptr, struct PointerRNA *node_ptr, float *r_color);
117         
118         void (*interface_draw)(struct bContext *C, struct uiLayout *layout, struct PointerRNA *ptr);
119         void (*interface_draw_color)(struct bContext *C, struct PointerRNA *ptr, float *r_color);
120         void (*interface_register_properties)(struct bNodeTree *ntree, struct bNodeSocket *stemp, struct StructRNA *data_srna);
121         void (*interface_init_socket)(struct bNodeTree *ntree, struct bNodeSocket *stemp, struct bNode *node, struct bNodeSocket *sock, const char *data_path);
122         void (*interface_verify_socket)(struct bNodeTree *ntree, struct bNodeSocket *stemp, struct bNode *node, struct bNodeSocket *sock, const char *data_path);
123         void (*interface_from_socket)(struct bNodeTree *ntree, struct bNodeSocket *stemp, struct bNode *node, struct bNodeSocket *sock);
124         
125         /* RNA integration */
126         ExtensionRNA ext_socket;
127         ExtensionRNA ext_interface;
128         
129         /* for standard socket types in C */
130         int type, subtype;
131 } bNodeSocketType;
132
133 typedef void *(*NodeInitExecFunction)(struct bNodeExecContext *context, struct bNode *node, bNodeInstanceKey key);
134 typedef void (*NodeFreeExecFunction)(void *nodedata);
135 typedef void (*NodeExecFunction)(void *data, int thread, struct bNode *, struct bNodeExecData *execdata, struct bNodeStack **in, struct bNodeStack **out);
136 typedef int (*NodeGPUExecFunction)(struct GPUMaterial *mat, struct bNode *node, struct bNodeExecData *execdata, struct GPUNodeStack *in, struct GPUNodeStack *out);
137
138 /** Defines a node type.
139  * Initial attributes and constants for a node as well as callback functions
140  * implementing the node behavior.
141  */
142 typedef struct bNodeType {
143         void *next, *prev;
144         short needs_free;               /* set for allocated types that need to be freed */
145         
146         char idname[64];                                /* identifier name */
147         int type;
148
149         char ui_name[64];       /* MAX_NAME */
150         char ui_description[256];
151         int ui_icon;
152         
153         float width, minwidth, maxwidth;
154         float height, minheight, maxheight;
155         short nclass, flag, compatibility;
156         
157         /* templates for static sockets */
158         bNodeSocketTemplate *inputs, *outputs;
159         
160         char storagename[64];                   /* struct name for DNA */
161         
162         /* Main draw function for the node */
163         void (*draw_nodetype)(const struct bContext *C, struct ARegion *ar, struct SpaceNode *snode,
164                          struct bNodeTree *ntree, struct bNode *node, bNodeInstanceKey key);
165         /* Updates the node geometry attributes according to internal state before actual drawing */
166         void (*draw_nodetype_prepare)(const struct bContext *C, struct bNodeTree *ntree, struct bNode *node);
167
168         /* Draw the option buttons on the node */
169         void (*draw_buttons)(struct uiLayout *, struct bContext *C, struct PointerRNA *ptr);
170         /* Additional parameters in the side panel */
171         void (*draw_buttons_ex)(struct uiLayout *, struct bContext *C, struct PointerRNA *ptr);
172
173         /* Additional drawing on backdrop */
174         void (*draw_backdrop)(struct SpaceNode *snode, struct ImBuf *backdrop, struct bNode *node, int x, int y);
175
176         /**
177          * Optional custom label function for the node header.
178          * \note Used as a fallback when #bNode.label isn't set.
179          */
180         void (*labelfunc)(struct bNodeTree *ntree, struct bNode *node, char *label, int maxlen);
181         /// Optional custom resize handle polling.
182         int (*resize_area_func)(struct bNode *node, int x, int y);
183         /// Optional selection area polling.
184         int (*select_area_func)(struct bNode *node, int x, int y);
185         /// Optional tweak area polling (for grabbing).
186         int (*tweak_area_func)(struct bNode *node, int x, int y);
187         
188         /// Called when the node is updated in the editor.
189         void (*updatefunc)(struct bNodeTree *ntree, struct bNode *node);
190         /// Check and update if internal ID data has changed.
191         void (*verifyfunc)(struct bNodeTree *ntree, struct bNode *node, struct ID *id);
192         
193         /// Initialize a new node instance of this type after creation.
194         void (*initfunc)(struct bNodeTree *ntree, struct bNode *node);
195         /// Free the node instance.
196         void (*freefunc)(struct bNode *node);
197         /// Make a copy of the node instance.
198         void (*copyfunc)(struct bNodeTree *dest_ntree, struct bNode *dest_node, struct bNode *src_node);
199         
200         /* Registerable API callback versions, called in addition to C callbacks */
201         void (*initfunc_api)(const struct bContext *C, struct PointerRNA *ptr);
202         void (*freefunc_api)(struct PointerRNA *ptr);
203         void (*copyfunc_api)(struct PointerRNA *ptr, struct bNode *src_node);
204         
205         /* can this node type be added to a node tree */
206         int (*poll)(struct bNodeType *ntype, struct bNodeTree *nodetree);
207         /* can this node be added to a node tree */
208         int (*poll_instance)(struct bNode *node, struct bNodeTree *nodetree);
209         
210         /* optional handling of link insertion */
211         void (*insert_link)(struct bNodeTree *ntree, struct bNode *node, struct bNodeLink *link);
212         /* Update the internal links list, for muting and disconnect operators. */
213         void (*update_internal_links)(struct bNodeTree *, struct bNode *node);
214         
215         /* **** execution callbacks **** */
216         NodeInitExecFunction initexecfunc;
217         NodeFreeExecFunction freeexecfunc;
218         NodeExecFunction execfunc;
219         /* gpu */
220         NodeGPUExecFunction gpufunc;
221         
222         /* RNA integration */
223         ExtensionRNA ext;
224 } bNodeType;
225
226 /* nodetype->nclass, for add-menu and themes */
227 #define NODE_CLASS_INPUT                        0
228 #define NODE_CLASS_OUTPUT                       1
229 #define NODE_CLASS_OP_COLOR                     3
230 #define NODE_CLASS_OP_VECTOR            4
231 #define NODE_CLASS_OP_FILTER            5
232 #define NODE_CLASS_GROUP                        6
233 // #define NODE_CLASS_FILE                              7
234 #define NODE_CLASS_CONVERTOR            8
235 #define NODE_CLASS_MATTE                        9
236 #define NODE_CLASS_DISTORT                      10
237 // #define NODE_CLASS_OP_DYNAMIC                11 /* deprecated */
238 #define NODE_CLASS_PATTERN                      12
239 #define NODE_CLASS_TEXTURE                      13
240 // #define NODE_CLASS_EXECUTION         14
241 // #define NODE_CLASS_GETDATA                   15
242 // #define NODE_CLASS_SETDATA                   16
243 // #define NODE_CLASS_MATH                              17
244 // #define NODE_CLASS_MATH_VECTOR               18
245 // #define NODE_CLASS_MATH_ROTATION     19
246 // #define NODE_CLASS_PARTICLES         25
247 // #define NODE_CLASS_TRANSFORM         30
248 // #define NODE_CLASS_COMBINE                   31
249 #define NODE_CLASS_SCRIPT                       32
250 #define NODE_CLASS_INTERFACE            33
251 #define NODE_CLASS_SHADER                       40
252 #define NODE_CLASS_LAYOUT                       100
253
254 /* nodetype->compatibility */
255 #define NODE_OLD_SHADING        (1 << 0)
256 #define NODE_NEW_SHADING        (1 << 1)
257 #define NODE_NEWER_SHADING      (1 << 2)
258
259 /* node resize directions */
260 #define NODE_RESIZE_TOP         1
261 #define NODE_RESIZE_BOTTOM      2
262 #define NODE_RESIZE_RIGHT       4
263 #define NODE_RESIZE_LEFT        8
264
265 typedef enum eNodeSizePreset {
266         NODE_SIZE_DEFAULT,
267         NODE_SIZE_SMALL,
268         NODE_SIZE_MIDDLE,
269         NODE_SIZE_LARGE
270 } eNodeSizePreset;
271
272 struct bNodeTreeExec;
273
274 typedef void (*bNodeClassCallback)(void *calldata, int nclass, const char *name);
275 typedef struct bNodeTreeType {
276         int type;                                               /* type identifier */
277         char idname[64];                                /* identifier name */
278
279         char ui_name[64];
280         char ui_description[256];
281         int ui_icon;
282         
283         /* callbacks */
284         void (*free_cache)(struct bNodeTree *ntree);
285         void (*free_node_cache)(struct bNodeTree *ntree, struct bNode *node);
286         void (*foreach_nodeclass)(struct Scene *scene, void *calldata, bNodeClassCallback func);        /* iteration over all node classes */
287         /* Check visibility in the node editor */
288         int (*poll)(const struct bContext *C, struct bNodeTreeType *ntreetype);
289         /* Select a node tree from the context */
290         void (*get_from_context)(const struct bContext *C, struct bNodeTreeType *ntreetype,
291                                  struct bNodeTree **r_ntree, struct ID **r_id, struct ID **r_from);
292
293         /* calls allowing threaded composite */
294         void (*localize)(struct bNodeTree *localtree, struct bNodeTree *ntree);
295         void (*local_sync)(struct bNodeTree *localtree, struct bNodeTree *ntree);
296         void (*local_merge)(struct Main *bmain, struct bNodeTree *localtree, struct bNodeTree *ntree);
297
298         /* Tree update. Overrides nodetype->updatetreefunc! */
299         void (*update)(struct bNodeTree *ntree);
300         
301         int (*validate_link)(struct bNodeTree *ntree, struct bNodeLink *link);
302
303         void (*node_add_init)(struct bNodeTree *ntree, struct bNode *bnode);
304         
305         /* RNA integration */
306         ExtensionRNA ext;
307 } bNodeTreeType;
308
309 /** \} */
310
311 /* -------------------------------------------------------------------- */
312 /** \name Generic API, Trees
313  * \{ */
314
315 struct bNodeTreeType *ntreeTypeFind(const char *idname);
316 void ntreeTypeAdd(struct bNodeTreeType *nt);
317 void ntreeTypeFreeLink(const struct bNodeTreeType *nt);
318 bool ntreeIsRegistered(struct bNodeTree *ntree);
319 struct GHashIterator *ntreeTypeGetIterator(void);
320
321 /* helper macros for iterating over tree types */
322 #define NODE_TREE_TYPES_BEGIN(ntype) \
323 { \
324         GHashIterator *__node_tree_type_iter__ = ntreeTypeGetIterator(); \
325         for (; !BLI_ghashIterator_done(__node_tree_type_iter__); BLI_ghashIterator_step(__node_tree_type_iter__)) { \
326                 bNodeTreeType *ntype = BLI_ghashIterator_getValue(__node_tree_type_iter__);
327
328 #define NODE_TREE_TYPES_END \
329         } \
330         BLI_ghashIterator_free(__node_tree_type_iter__); \
331 } (void)0
332
333 void ntreeSetTypes(const struct bContext *C, struct bNodeTree *ntree);
334
335 void ntreeInitDefault(struct bNodeTree *ntree);
336 struct bNodeTree *ntreeAddTree(struct Main *bmain, const char *name, const char *idname);
337
338 /* copy/free funcs, need to manage ID users */
339 void              ntreeFreeTree(struct bNodeTree *ntree);
340 void BKE_node_tree_copy_data(struct Main *bmain, struct bNodeTree *ntree_dst, const struct bNodeTree *ntree_src, const int flag);
341 struct bNodeTree *ntreeCopyTree_ex(const struct bNodeTree *ntree, struct Main *bmain, const bool do_id_user);
342 struct bNodeTree *ntreeCopyTree(struct Main *bmain, const struct bNodeTree *ntree);
343 /* node->id user count */
344 void              ntreeUserIncrefID(struct bNodeTree *ntree);
345 void              ntreeUserDecrefID(struct bNodeTree *ntree);
346
347
348 struct bNodeTree *ntreeFromID(const struct ID *id);
349
350 void              ntreeMakeLocal(struct Main *bmain, struct bNodeTree *ntree, bool id_in_mainlist, const bool lib_local);
351 struct bNode     *ntreeFindType(const struct bNodeTree *ntree, int type);
352 bool              ntreeHasType(const struct bNodeTree *ntree, int type);
353 bool              ntreeHasTree(const struct bNodeTree *ntree, const struct bNodeTree *lookup);
354 void              ntreeUpdateTree(struct Main *main, struct bNodeTree *ntree);
355 /* XXX Currently each tree update call does call to ntreeVerifyNodes too.
356  * Some day this should be replaced by a decent depsgraph automatism!
357  */
358 void              ntreeVerifyNodes(struct Main *main, struct ID *id);
359
360 void              ntreeGetDependencyList(struct bNodeTree *ntree, struct bNode ***deplist, int *totnodes);
361
362 /* XXX old trees handle output flags automatically based on special output node types and last active selection.
363  * new tree types have a per-output socket flag to indicate the final output to use explicitly.
364  */
365 void            ntreeSetOutput(struct bNodeTree *ntree);
366
367 void            ntreeFreeCache(struct bNodeTree *ntree);
368
369 int             ntreeNodeExists(struct bNodeTree *ntree, struct bNode *testnode);
370 int             ntreeOutputExists(struct bNode *node, struct bNodeSocket *testsock);
371 void            ntreeNodeFlagSet(const bNodeTree *ntree, const int flag, const bool enable);
372 struct bNodeTree *ntreeLocalize(struct bNodeTree *ntree);
373 void            ntreeLocalSync(struct bNodeTree *localtree, struct bNodeTree *ntree);
374 void            ntreeLocalMerge(struct Main *bmain, struct bNodeTree *localtree, struct bNodeTree *ntree);
375
376 /** \} */
377
378
379 /* -------------------------------------------------------------------- */
380 /** \name Node Tree Interface
381  * \{ */
382 struct bNodeSocket *ntreeFindSocketInterface(struct bNodeTree *ntree, int in_out, const char *identifier);
383 struct bNodeSocket *ntreeAddSocketInterface(struct bNodeTree *ntree, int in_out, const char *idname, const char *name);
384 struct bNodeSocket *ntreeInsertSocketInterface(struct bNodeTree *ntree, int in_out, const char *idname,
385                                                struct bNodeSocket *next_sock, const char *name);
386 struct bNodeSocket *ntreeAddSocketInterfaceFromSocket(struct bNodeTree *ntree, struct bNode *from_node, struct bNodeSocket *from_sock);
387 struct bNodeSocket *ntreeInsertSocketInterfaceFromSocket(struct bNodeTree *ntree, struct bNodeSocket *next_sock,
388                                                          struct bNode *from_node, struct bNodeSocket *from_sock);
389 void            ntreeRemoveSocketInterface(struct bNodeTree *ntree, struct bNodeSocket *sock);
390
391 struct StructRNA *ntreeInterfaceTypeGet(struct bNodeTree *ntree, int create);
392 void ntreeInterfaceTypeFree(struct bNodeTree *ntree);
393 void ntreeInterfaceTypeUpdate(struct bNodeTree *ntree);
394
395 /** \} */
396
397
398 /* -------------------------------------------------------------------- */
399 /** \name Generic API, Nodes
400  * \{ */
401
402 struct bNodeType *nodeTypeFind(const char *idname);
403 void                    nodeRegisterType(struct bNodeType *ntype);
404 void                    nodeUnregisterType(struct bNodeType *ntype);
405 bool                    nodeIsRegistered(struct bNode *node);
406 struct GHashIterator *nodeTypeGetIterator(void);
407
408 /* helper macros for iterating over node types */
409 #define NODE_TYPES_BEGIN(ntype) \
410 { \
411         GHashIterator *__node_type_iter__ = nodeTypeGetIterator(); \
412         for (; !BLI_ghashIterator_done(__node_type_iter__); BLI_ghashIterator_step(__node_type_iter__)) { \
413                 bNodeType *ntype = BLI_ghashIterator_getValue(__node_type_iter__);
414
415 #define NODE_TYPES_END \
416         } \
417         BLI_ghashIterator_free(__node_type_iter__); \
418 }
419
420 struct bNodeSocketType *nodeSocketTypeFind(const char *idname);
421 void                    nodeRegisterSocketType(struct bNodeSocketType *stype);
422 void                    nodeUnregisterSocketType(struct bNodeSocketType *stype);
423 bool                    nodeSocketIsRegistered(struct bNodeSocket *sock);
424 struct GHashIterator *nodeSocketTypeGetIterator(void);
425 const char *nodeStaticSocketType(int type, int subtype);
426 const char *nodeStaticSocketInterfaceType(int type, int subtype);
427
428 /* helper macros for iterating over node types */
429 #define NODE_SOCKET_TYPES_BEGIN(stype) \
430 { \
431         GHashIterator *__node_socket_type_iter__ = nodeSocketTypeGetIterator(); \
432         for (; !BLI_ghashIterator_done(__node_socket_type_iter__); BLI_ghashIterator_step(__node_socket_type_iter__)) { \
433                 bNodeSocketType *stype = BLI_ghashIterator_getValue(__node_socket_type_iter__);
434
435 #define NODE_SOCKET_TYPES_END \
436         } \
437         BLI_ghashIterator_free(__node_socket_type_iter__); \
438 }
439
440 struct bNodeSocket *nodeFindSocket(struct bNode *node, int in_out, const char *identifier);
441 struct bNodeSocket *nodeAddSocket(struct bNodeTree *ntree, struct bNode *node, int in_out, const char *idname,
442                                   const char *identifier, const char *name);
443 struct bNodeSocket *nodeInsertSocket(struct bNodeTree *ntree, struct bNode *node, int in_out, const char *idname,
444                                      struct bNodeSocket *next_sock, const char *identifier, const char *name);
445 struct bNodeSocket *nodeAddStaticSocket(struct bNodeTree *ntree, struct bNode *node, int in_out, int type, int subtype,
446                                         const char *identifier, const char *name);
447 struct bNodeSocket *nodeInsertStaticSocket(struct bNodeTree *ntree, struct bNode *node, int in_out, int type, int subtype,
448                                            struct bNodeSocket *next_sock, const char *identifier, const char *name);
449 void nodeRemoveSocket(struct bNodeTree *ntree, struct bNode *node, struct bNodeSocket *sock);
450 void nodeRemoveAllSockets(struct bNodeTree *ntree, struct bNode *node);
451 void nodeModifySocketType(struct bNodeTree *ntree, struct bNode *node, struct bNodeSocket *sock, int type, int subtype);
452
453 struct bNode    *nodeAddNode(const struct bContext *C, struct bNodeTree *ntree, const char *idname);
454 struct bNode    *nodeAddStaticNode(const struct bContext *C, struct bNodeTree *ntree, int type);
455 void            nodeUnlinkNode(struct bNodeTree *ntree, struct bNode *node);
456 void            nodeUniqueName(struct bNodeTree *ntree, struct bNode *node);
457
458 void            nodeFreeNode(struct bNodeTree *ntree, struct bNode *node);
459 struct bNode    *BKE_node_copy_ex(struct bNodeTree *ntree, struct bNode *node_src, const int flag);
460 struct bNode    *nodeCopyNode(struct bNodeTree *ntree, struct bNode *node);
461
462 struct bNodeLink *nodeAddLink(struct bNodeTree *ntree, struct bNode *fromnode, struct bNodeSocket *fromsock, struct bNode *tonode, struct bNodeSocket *tosock);
463 void            nodeRemLink(struct bNodeTree *ntree, struct bNodeLink *link);
464 void            nodeRemSocketLinks(struct bNodeTree *ntree, struct bNodeSocket *sock);
465 bool            nodeLinkIsHidden(struct bNodeLink *link);
466 void            nodeInternalRelink(struct bNodeTree *ntree, struct bNode *node);
467
468 void            nodeToView(struct bNode *node, float x, float y, float *rx, float *ry);
469 void            nodeFromView(struct bNode *node, float x, float y, float *rx, float *ry);
470 bool            nodeAttachNodeCheck(struct bNode *node, struct bNode *parent);
471 void            nodeAttachNode(struct bNode *node, struct bNode *parent);
472 void            nodeDetachNode(struct bNode *node);
473
474 struct bNode   *nodeFindNodebyName(struct bNodeTree *ntree, const char *name);
475 int             nodeFindNode(struct bNodeTree *ntree, struct bNodeSocket *sock, struct bNode **nodep, int *sockindex);
476 struct bNode   *nodeFindRootParent(bNode *node);
477
478 bool            nodeIsChildOf(const bNode *parent, const bNode *child);
479
480 void nodeChainIter(
481         const bNodeTree *ntree, const bNode *node_start,
482         bool (*callback)(bNode *, bNode *, void *, const bool), void *userdata,
483         const bool reversed);
484 void nodeParentsIter(bNode *node, bool (*callback)(bNode *, void *), void *userdata);
485
486 struct bNodeLink *nodeFindLink(struct bNodeTree *ntree, struct bNodeSocket *from, struct bNodeSocket *to);
487 int             nodeCountSocketLinks(struct bNodeTree *ntree, struct bNodeSocket *sock);
488
489 void            nodeSetSelected(struct bNode *node, bool select);
490 void            nodeSetActive(struct bNodeTree *ntree, struct bNode *node);
491 struct bNode   *nodeGetActive(struct bNodeTree *ntree);
492 struct bNode   *nodeGetActiveID(struct bNodeTree *ntree, short idtype);
493 bool            nodeSetActiveID(struct bNodeTree *ntree, short idtype, struct ID *id);
494 void            nodeClearActive(struct bNodeTree *ntree);
495 void            nodeClearActiveID(struct bNodeTree *ntree, short idtype);
496 struct bNode   *nodeGetActiveTexture(struct bNodeTree *ntree);
497
498 void            nodeUpdate(struct bNodeTree *ntree, struct bNode *node);
499 bool            nodeUpdateID(struct bNodeTree *ntree, struct ID *id);
500 void            nodeUpdateInternalLinks(struct bNodeTree *ntree, struct bNode *node);
501
502 int             nodeSocketIsHidden(struct bNodeSocket *sock);
503 void            ntreeTagUsedSockets(struct bNodeTree *ntree);
504
505 /* Node Clipboard */
506 void                   BKE_node_clipboard_init(struct bNodeTree *ntree);
507 void                   BKE_node_clipboard_clear(void);
508 bool                   BKE_node_clipboard_validate(void);
509 void                   BKE_node_clipboard_add_node(struct bNode *node);
510 void                   BKE_node_clipboard_add_link(struct bNodeLink *link);
511 const struct ListBase *BKE_node_clipboard_get_nodes(void);
512 const struct ListBase *BKE_node_clipboard_get_links(void);
513 int                    BKE_node_clipboard_get_type(void);
514
515 /* Node Instance Hash */
516 typedef struct bNodeInstanceHash {
517         GHash *ghash;   /* XXX should be made a direct member, GHash allocation needs to support it */
518 } bNodeInstanceHash;
519
520 typedef void (*bNodeInstanceValueFP)(void *value);
521
522 extern const bNodeInstanceKey NODE_INSTANCE_KEY_BASE;
523 extern const bNodeInstanceKey NODE_INSTANCE_KEY_NONE;
524
525 bNodeInstanceKey       BKE_node_instance_key(bNodeInstanceKey parent_key, struct bNodeTree *ntree, struct bNode *node);
526
527 bNodeInstanceHash     *BKE_node_instance_hash_new(const char *info);
528 void                   BKE_node_instance_hash_free(bNodeInstanceHash *hash, bNodeInstanceValueFP valfreefp);
529 void                   BKE_node_instance_hash_insert(bNodeInstanceHash *hash, bNodeInstanceKey key, void *value);
530 void                  *BKE_node_instance_hash_lookup(bNodeInstanceHash *hash, bNodeInstanceKey key);
531 int                    BKE_node_instance_hash_remove(bNodeInstanceHash *hash, bNodeInstanceKey key, bNodeInstanceValueFP valfreefp);
532 void                   BKE_node_instance_hash_clear(bNodeInstanceHash *hash, bNodeInstanceValueFP valfreefp);
533 void                  *BKE_node_instance_hash_pop(bNodeInstanceHash *hash, bNodeInstanceKey key);
534 int                    BKE_node_instance_hash_haskey(bNodeInstanceHash *hash, bNodeInstanceKey key);
535 int                    BKE_node_instance_hash_size(bNodeInstanceHash *hash);
536
537 void                   BKE_node_instance_hash_clear_tags(bNodeInstanceHash *hash);
538 void                   BKE_node_instance_hash_tag(bNodeInstanceHash *hash, void *value);
539 bool                   BKE_node_instance_hash_tag_key(bNodeInstanceHash *hash, bNodeInstanceKey key);
540 void                   BKE_node_instance_hash_remove_untagged(bNodeInstanceHash *hash, bNodeInstanceValueFP valfreefp);
541
542 typedef GHashIterator bNodeInstanceHashIterator;
543
544 BLI_INLINE bNodeInstanceHashIterator *BKE_node_instance_hash_iterator_new(bNodeInstanceHash *hash) { return BLI_ghashIterator_new(hash->ghash); }
545 BLI_INLINE void                       BKE_node_instance_hash_iterator_init(bNodeInstanceHashIterator *iter, bNodeInstanceHash *hash) { BLI_ghashIterator_init(iter, hash->ghash); }
546 BLI_INLINE void                       BKE_node_instance_hash_iterator_free(bNodeInstanceHashIterator *iter) { BLI_ghashIterator_free(iter); }
547 BLI_INLINE bNodeInstanceKey           BKE_node_instance_hash_iterator_get_key(bNodeInstanceHashIterator *iter) { return *(bNodeInstanceKey *)BLI_ghashIterator_getKey(iter); }
548 BLI_INLINE void                      *BKE_node_instance_hash_iterator_get_value(bNodeInstanceHashIterator *iter) { return BLI_ghashIterator_getValue(iter); }
549 BLI_INLINE void                       BKE_node_instance_hash_iterator_step(bNodeInstanceHashIterator *iter) { BLI_ghashIterator_step(iter); }
550 BLI_INLINE bool                       BKE_node_instance_hash_iterator_done(bNodeInstanceHashIterator *iter) { return BLI_ghashIterator_done(iter); }
551
552 #define NODE_INSTANCE_HASH_ITER(iter_, hash_) \
553         for (BKE_node_instance_hash_iterator_init(&iter_, hash_); \
554              BKE_node_instance_hash_iterator_done(&iter_) == false; \
555              BKE_node_instance_hash_iterator_step(&iter_))
556
557
558 /* Node Previews */
559
560 int             BKE_node_preview_used(struct bNode *node);
561 bNodePreview   *BKE_node_preview_verify(struct bNodeInstanceHash *previews, bNodeInstanceKey key, int xsize, int ysize, bool create);
562 bNodePreview   *BKE_node_preview_copy(struct bNodePreview *preview);
563 void            BKE_node_preview_free(struct bNodePreview *preview);
564 void            BKE_node_preview_init_tree(struct bNodeTree *ntree, int xsize, int ysize, int create_previews);
565 void            BKE_node_preview_free_tree(struct bNodeTree *ntree);
566 void            BKE_node_preview_remove_unused(struct bNodeTree *ntree);
567 void            BKE_node_preview_clear(struct bNodePreview *preview);
568 void            BKE_node_preview_clear_tree(struct bNodeTree *ntree);
569
570 void            BKE_node_preview_sync_tree(struct bNodeTree *to_ntree, struct bNodeTree *from_ntree);
571 void            BKE_node_preview_merge_tree(struct bNodeTree *to_ntree, struct bNodeTree *from_ntree, bool remove_old);
572
573 void            BKE_node_preview_set_pixel(struct bNodePreview *preview, const float col[4], int x, int y, bool do_manage);
574
575 /** \} */
576
577 /* -------------------------------------------------------------------- */
578 /** \name Node Type Access
579  * \{ */
580
581 void            nodeLabel(struct bNodeTree *ntree, struct bNode *node, char *label, int maxlen);
582
583 int                             nodeGroupPoll(struct bNodeTree *nodetree, struct bNodeTree *grouptree);
584
585 /* Init a new node type struct with default values and callbacks */
586 void            node_type_base(struct bNodeType *ntype, int type, const char *name, short nclass, short flag);
587 void            node_type_base_custom(struct bNodeType *ntype, const char *idname, const char *name, short nclass, short flag);
588 void            node_type_socket_templates(struct bNodeType *ntype, struct bNodeSocketTemplate *inputs, struct bNodeSocketTemplate *outputs);
589 void            node_type_size(struct bNodeType *ntype, int width, int minwidth, int maxwidth);
590 void            node_type_size_preset(struct bNodeType *ntype, eNodeSizePreset size);
591 void            node_type_init(struct bNodeType *ntype, void (*initfunc)(struct bNodeTree *ntree, struct bNode *node));
592 void            node_type_storage(struct bNodeType *ntype,
593                                   const char *storagename,
594                                   void (*freefunc)(struct bNode *node),
595                                   void (*copyfunc)(struct bNodeTree *dest_ntree, struct bNode *dest_node, struct bNode *src_node));
596 void            node_type_label(struct bNodeType *ntype, void (*labelfunc)(struct bNodeTree *ntree, struct bNode *, char *label, int maxlen));
597 void            node_type_update(struct bNodeType *ntype,
598                                  void (*updatefunc)(struct bNodeTree *ntree, struct bNode *node),
599                                  void (*verifyfunc)(struct bNodeTree *ntree, struct bNode *node, struct ID *id));
600
601 void            node_type_exec(struct bNodeType *ntype, NodeInitExecFunction initexecfunc, NodeFreeExecFunction freeexecfunc, NodeExecFunction execfunc);
602 void            node_type_gpu(struct bNodeType *ntype, NodeGPUExecFunction gpufunc);
603 void            node_type_internal_links(struct bNodeType *ntype, void (*update_internal_links)(struct bNodeTree *, struct bNode *));
604 void            node_type_compatibility(struct bNodeType *ntype, short compatibility);
605
606 /** \} */
607
608 /* -------------------------------------------------------------------- */
609 /** \name Node Generic Functions
610  * \{ */
611
612 bool BKE_node_is_connected_to_output(struct bNodeTree *ntree, struct bNode *node);
613
614 /* ************** COMMON NODES *************** */
615
616 #define NODE_UNDEFINED  -2              /* node type is not registered */
617 #define NODE_CUSTOM             -1              /* for dynamically registered custom types */
618 #define NODE_GROUP              2
619 // #define NODE_FORLOOP 3               /* deprecated */
620 // #define NODE_WHILELOOP       4       /* deprecated */
621 #define NODE_FRAME              5
622 #define NODE_REROUTE    6
623 #define NODE_GROUP_INPUT        7
624 #define NODE_GROUP_OUTPUT       8
625
626 void BKE_node_tree_unlink_id(ID *id, struct bNodeTree *ntree);
627
628 /** \} */
629
630 /* -------------------------------------------------------------------- */
631 /** \name Node Tree Iterator
632  *
633  * Utility macro for visiting every node tree in the library data, including local bNodeTree blocks in other IDs.
634  * This avoids the need for callback functions and allows executing code in a single inner code block.
635  *
636  * Variables:
637  *
638  *   nodetree:  The actual bNodeTree data block.
639  *              Check nodetree->idname or nodetree->typeinfo to use only specific types.
640  *
641  *   id:        The owner of the bNodeTree data block.
642  *              Same as nodetree if it's a linkable node tree from the library.
643  *
644  * Examples:
645  *
646  * \code{.c}
647  * FOREACH_NODETREE(bmain, nodetree, id) {
648  *     if (id == nodetree)
649  *         printf("This is a linkable node tree");
650  * } FOREACH_NODETREE_END
651  *
652  * FOREACH_NODETREE(bmain, nodetree, id) {
653  *     if (nodetree->idname == "ShaderNodeTree")
654  *         printf("This is a shader node tree);
655  *     if (GS(id) == ID_MA)
656  *         printf(" and it's owned by a material");
657  * } FOREACH_NODETREE_END
658  * \endcode
659  *
660  * \{
661  */
662
663 /* should be an opaque type, only for internal use by BKE_node_tree_iter_*** */
664 struct NodeTreeIterStore {
665         bNodeTree *ngroup;
666         Scene *scene;
667         Material *mat;
668         Tex *tex;
669         Lamp *lamp;
670         World *world;
671         FreestyleLineStyle *linestyle;
672 };
673
674 void BKE_node_tree_iter_init(struct NodeTreeIterStore *ntreeiter, struct Main *bmain);
675 bool BKE_node_tree_iter_step(struct NodeTreeIterStore *ntreeiter,
676                              struct bNodeTree **r_nodetree, struct ID **r_id);
677
678 #define FOREACH_NODETREE(bmain, _nodetree, _id) \
679 { \
680         struct NodeTreeIterStore _nstore; \
681         bNodeTree *_nodetree; \
682         ID *_id; \
683         /* avoid compiler warning about unused variables */ \
684         BKE_node_tree_iter_init(&_nstore, bmain); \
685         while (BKE_node_tree_iter_step(&_nstore, &_nodetree, &_id) == true) { \
686                 if (_nodetree) {
687
688 #define FOREACH_NODETREE_END \
689                 } \
690         } \
691 }
692 /** \} */
693
694
695 /* -------------------------------------------------------------------- */
696 /** \name Node Tree
697  */
698
699 void BKE_nodetree_remove_layer_n(struct bNodeTree *ntree, struct Scene *scene, const int layer_index);
700
701 /* -------------------------------------------------------------------- */
702 /** \name Shader Nodes
703  * \{ */
704
705 /* note: types are needed to restore callbacks, don't change values */
706 /* range 1 - 100 is reserved for common nodes */
707 /* using toolbox, we add node groups by assuming the values below don't exceed NODE_GROUP_MENU for now */
708
709 //#define SH_NODE_OUTPUT                1
710
711 //#define SH_NODE_MATERIAL      100
712 #define SH_NODE_RGB                     101
713 #define SH_NODE_VALUE           102
714 #define SH_NODE_MIX_RGB         103
715 #define SH_NODE_VALTORGB        104
716 #define SH_NODE_RGBTOBW         105
717 #define SH_NODE_SHADERTORGB     106
718 //#define SH_NODE_TEXTURE               106
719 #define SH_NODE_NORMAL          107
720 //#define SH_NODE_GEOMETRY      108
721 #define SH_NODE_MAPPING         109
722 #define SH_NODE_CURVE_VEC       110
723 #define SH_NODE_CURVE_RGB       111
724 #define SH_NODE_CAMERA          114
725 #define SH_NODE_MATH            115
726 #define SH_NODE_VECT_MATH       116
727 #define SH_NODE_SQUEEZE         117
728 //#define SH_NODE_MATERIAL_EXT  118
729 #define SH_NODE_INVERT          119
730 #define SH_NODE_SEPRGB          120
731 #define SH_NODE_COMBRGB         121
732 #define SH_NODE_HUE_SAT         122
733 #define NODE_DYNAMIC            123
734
735 #define SH_NODE_OUTPUT_MATERIAL                 124
736 #define SH_NODE_OUTPUT_WORLD                    125
737 #define SH_NODE_OUTPUT_LAMP                             126
738 #define SH_NODE_FRESNEL                                 127
739 #define SH_NODE_MIX_SHADER                              128
740 #define SH_NODE_ATTRIBUTE                               129
741 #define SH_NODE_BACKGROUND                              130
742 #define SH_NODE_BSDF_ANISOTROPIC                131
743 #define SH_NODE_BSDF_DIFFUSE                    132
744 #define SH_NODE_BSDF_GLOSSY                             133
745 #define SH_NODE_BSDF_GLASS                              134
746 #define SH_NODE_BSDF_TRANSLUCENT                137
747 #define SH_NODE_BSDF_TRANSPARENT                138
748 #define SH_NODE_BSDF_VELVET                             139
749 #define SH_NODE_EMISSION                                140
750 #define SH_NODE_NEW_GEOMETRY                    141
751 #define SH_NODE_LIGHT_PATH                              142
752 #define SH_NODE_TEX_IMAGE                               143
753 #define SH_NODE_TEX_SKY                                 145
754 #define SH_NODE_TEX_GRADIENT                    146
755 #define SH_NODE_TEX_VORONOI                             147
756 #define SH_NODE_TEX_MAGIC                               148
757 #define SH_NODE_TEX_WAVE                                149
758 #define SH_NODE_TEX_NOISE                               150
759 #define SH_NODE_TEX_MUSGRAVE                    152
760 #define SH_NODE_TEX_COORD                               155
761 #define SH_NODE_ADD_SHADER                              156
762 #define SH_NODE_TEX_ENVIRONMENT                 157
763 #define SH_NODE_OUTPUT_TEXTURE                  158
764 #define SH_NODE_HOLDOUT                                 159
765 #define SH_NODE_LAYER_WEIGHT                    160
766 #define SH_NODE_VOLUME_ABSORPTION               161
767 #define SH_NODE_VOLUME_SCATTER                  162
768 #define SH_NODE_GAMMA                                   163
769 #define SH_NODE_TEX_CHECKER                             164
770 #define SH_NODE_BRIGHTCONTRAST                  165
771 #define SH_NODE_LIGHT_FALLOFF                   166
772 #define SH_NODE_OBJECT_INFO                             167
773 #define SH_NODE_PARTICLE_INFO                   168
774 #define SH_NODE_TEX_BRICK                               169
775 #define SH_NODE_BUMP                                    170
776 #define SH_NODE_SCRIPT                                  171
777 #define SH_NODE_AMBIENT_OCCLUSION               172
778 #define SH_NODE_BSDF_REFRACTION                 173
779 #define SH_NODE_TANGENT                                 174
780 #define SH_NODE_NORMAL_MAP                              175
781 #define SH_NODE_HAIR_INFO                               176
782 #define SH_NODE_SUBSURFACE_SCATTERING   177
783 #define SH_NODE_WIREFRAME                               178
784 #define SH_NODE_BSDF_TOON                               179
785 #define SH_NODE_WAVELENGTH                              180
786 #define SH_NODE_BLACKBODY                               181
787 #define SH_NODE_VECT_TRANSFORM                  182
788 #define SH_NODE_SEPHSV                                  183
789 #define SH_NODE_COMBHSV                                 184
790 #define SH_NODE_BSDF_HAIR                               185
791 #define SH_NODE_LAMP                                    186
792 #define SH_NODE_UVMAP                                   187
793 #define SH_NODE_SEPXYZ                                  188
794 #define SH_NODE_COMBXYZ                                 189
795 #define SH_NODE_OUTPUT_LINESTYLE                190
796 #define SH_NODE_UVALONGSTROKE                   191
797 #define SH_NODE_TEX_POINTDENSITY                192
798 #define SH_NODE_BSDF_PRINCIPLED         193
799 #define SH_NODE_TEX_IES                 194
800 #define SH_NODE_EEVEE_SPECULAR                  195
801 #define SH_NODE_BEVEL                   197
802 #define SH_NODE_DISPLACEMENT            198
803 #define SH_NODE_VECTOR_DISPLACEMENT     199
804 #define SH_NODE_VOLUME_PRINCIPLED       200
805
806 /* custom defines options for Material node */
807 #define SH_NODE_MAT_DIFF   1
808 #define SH_NODE_MAT_SPEC   2
809 #define SH_NODE_MAT_NEG    4
810
811 /* API */
812
813 struct bNodeTreeExec *ntreeShaderBeginExecTree(struct bNodeTree *ntree);
814 void            ntreeShaderEndExecTree(struct bNodeTreeExec *exec);
815 bool            ntreeShaderExecTree(struct bNodeTree *ntree, int thread);
816
817 void            ntreeGPUMaterialNodes(struct bNodeTree *ntree, struct GPUMaterial *mat, short compatibility);
818 void            ntreeGPUMaterialDomain(struct bNodeTree *ntree, bool *has_surface_output, bool *has_volume_output);
819
820 /** \} */
821
822 /* -------------------------------------------------------------------- */
823 /** \name Composite Nodes
824  * \{ */
825
826 /* output socket defines */
827 #define RRES_OUT_IMAGE                                  0
828 #define RRES_OUT_ALPHA                                  1
829 #define RRES_OUT_Z                                              2
830 #define RRES_OUT_NORMAL                                 3
831 #define RRES_OUT_UV                                             4
832 #define RRES_OUT_VEC                                    5
833 #define RRES_OUT_RGBA                                   6
834 #define RRES_OUT_DIFF                                   7
835 #define RRES_OUT_SPEC                                   8
836 #define RRES_OUT_SHADOW                                 9
837 #define RRES_OUT_AO                                             10
838 #define RRES_OUT_REFLECT                                11
839 #define RRES_OUT_REFRACT                                12
840 #define RRES_OUT_INDIRECT                               13
841 #define RRES_OUT_INDEXOB                                14
842 #define RRES_OUT_INDEXMA                                15
843 #define RRES_OUT_MIST                                   16
844 #define RRES_OUT_EMIT                                   17
845 #define RRES_OUT_ENV                                    18
846 #define RRES_OUT_DIFF_DIRECT                    19
847 #define RRES_OUT_DIFF_INDIRECT                  20
848 #define RRES_OUT_DIFF_COLOR                             21
849 #define RRES_OUT_GLOSSY_DIRECT                  22
850 #define RRES_OUT_GLOSSY_INDIRECT                23
851 #define RRES_OUT_GLOSSY_COLOR                   24
852 #define RRES_OUT_TRANSM_DIRECT                  25
853 #define RRES_OUT_TRANSM_INDIRECT                26
854 #define RRES_OUT_TRANSM_COLOR                   27
855 #define RRES_OUT_SUBSURFACE_DIRECT              28
856 #define RRES_OUT_SUBSURFACE_INDIRECT    29
857 #define RRES_OUT_SUBSURFACE_COLOR               30
858 #define RRES_OUT_DEBUG                          31
859
860 /* note: types are needed to restore callbacks, don't change values */
861 #define CMP_NODE_VIEWER         201
862 #define CMP_NODE_RGB            202
863 #define CMP_NODE_VALUE          203
864 #define CMP_NODE_MIX_RGB        204
865 #define CMP_NODE_VALTORGB       205
866 #define CMP_NODE_RGBTOBW        206
867 #define CMP_NODE_NORMAL         207
868 #define CMP_NODE_CURVE_VEC      208
869 #define CMP_NODE_CURVE_RGB      209
870 #define CMP_NODE_ALPHAOVER      210
871 #define CMP_NODE_BLUR           211
872 #define CMP_NODE_FILTER         212
873 #define CMP_NODE_MAP_VALUE      213
874 #define CMP_NODE_TIME           214
875 #define CMP_NODE_VECBLUR        215
876 #define CMP_NODE_SEPRGBA        216
877 #define CMP_NODE_SEPHSVA        217
878 #define CMP_NODE_SETALPHA       218
879 #define CMP_NODE_HUE_SAT        219
880 #define CMP_NODE_IMAGE          220
881 #define CMP_NODE_R_LAYERS       221
882 #define CMP_NODE_COMPOSITE      222
883 #define CMP_NODE_OUTPUT_FILE    223
884 #define CMP_NODE_TEXTURE        224
885 #define CMP_NODE_TRANSLATE      225
886 #define CMP_NODE_ZCOMBINE       226
887 #define CMP_NODE_COMBRGBA       227
888 #define CMP_NODE_DILATEERODE    228
889 #define CMP_NODE_ROTATE         229
890 #define CMP_NODE_SCALE          230
891 #define CMP_NODE_SEPYCCA        231
892 #define CMP_NODE_COMBYCCA       232
893 #define CMP_NODE_SEPYUVA        233
894 #define CMP_NODE_COMBYUVA       234
895 #define CMP_NODE_DIFF_MATTE     235
896 #define CMP_NODE_COLOR_SPILL    236
897 #define CMP_NODE_CHROMA_MATTE   237
898 #define CMP_NODE_CHANNEL_MATTE  238
899 #define CMP_NODE_FLIP           239
900 #define CMP_NODE_SPLITVIEWER    240
901 // #define CMP_NODE_INDEX_MASK  241
902 #define CMP_NODE_MAP_UV         242
903 #define CMP_NODE_ID_MASK        243
904 #define CMP_NODE_DEFOCUS        244
905 #define CMP_NODE_DISPLACE       245
906 #define CMP_NODE_COMBHSVA       246
907 #define CMP_NODE_MATH           247
908 #define CMP_NODE_LUMA_MATTE     248
909 #define CMP_NODE_BRIGHTCONTRAST 249
910 #define CMP_NODE_GAMMA          250
911 #define CMP_NODE_INVERT         251
912 #define CMP_NODE_NORMALIZE      252
913 #define CMP_NODE_CROP           253
914 #define CMP_NODE_DBLUR          254
915 #define CMP_NODE_BILATERALBLUR  255
916 #define CMP_NODE_PREMULKEY  256
917 #define CMP_NODE_DIST_MATTE     257
918 #define CMP_NODE_VIEW_LEVELS    258
919 #define CMP_NODE_COLOR_MATTE 259
920 #define CMP_NODE_COLORBALANCE 260
921 #define CMP_NODE_HUECORRECT 261
922 #define CMP_NODE_MOVIECLIP      262
923 #define CMP_NODE_STABILIZE2D    263
924 #define CMP_NODE_TRANSFORM      264
925 #define CMP_NODE_MOVIEDISTORTION        265
926 #define CMP_NODE_DOUBLEEDGEMASK    266
927 #define CMP_NODE_OUTPUT_MULTI_FILE__DEPRECATED  267     /* DEPRECATED multi file node has been merged into regular CMP_NODE_OUTPUT_FILE */
928 #define CMP_NODE_MASK           268
929 #define CMP_NODE_KEYINGSCREEN           269
930 #define CMP_NODE_KEYING                 270
931 #define CMP_NODE_TRACKPOS               271
932 #define CMP_NODE_INPAINT    272
933 #define CMP_NODE_DESPECKLE  273
934
935 #define CMP_NODE_GLARE          301
936 #define CMP_NODE_TONEMAP        302
937 #define CMP_NODE_LENSDIST       303
938 #define CMP_NODE_SUNBEAMS       304
939
940 #define CMP_NODE_COLORCORRECTION 312
941 #define CMP_NODE_MASK_BOX       313
942 #define CMP_NODE_MASK_ELLIPSE   314
943 #define CMP_NODE_BOKEHIMAGE     315
944 #define CMP_NODE_BOKEHBLUR      316
945 #define CMP_NODE_SWITCH         317
946 #define CMP_NODE_PIXELATE       318
947
948 #define CMP_NODE_MAP_RANGE      319
949 #define CMP_NODE_PLANETRACKDEFORM       320
950 #define CMP_NODE_CORNERPIN          321
951 #define CMP_NODE_SWITCH_VIEW    322
952
953 /* channel toggles */
954 #define CMP_CHAN_RGB            1
955 #define CMP_CHAN_A                      2
956
957 /* filter types */
958 #define CMP_FILT_SOFT           0
959 #define CMP_FILT_SHARP          1
960 #define CMP_FILT_LAPLACE        2
961 #define CMP_FILT_SOBEL          3
962 #define CMP_FILT_PREWITT        4
963 #define CMP_FILT_KIRSCH         5
964 #define CMP_FILT_SHADOW         6
965
966 /* scale node type, in custom1 */
967 #define CMP_SCALE_RELATIVE              0
968 #define CMP_SCALE_ABSOLUTE              1
969 #define CMP_SCALE_SCENEPERCENT  2
970 #define CMP_SCALE_RENDERPERCENT 3
971 /* custom2 */
972 #define CMP_SCALE_RENDERSIZE_FRAME_ASPECT  (1 << 0)
973 #define CMP_SCALE_RENDERSIZE_FRAME_CROP    (1 << 1)
974
975 /* track position node, in custom1 */
976 #define CMP_TRACKPOS_ABSOLUTE                   0
977 #define CMP_TRACKPOS_RELATIVE_START     1
978 #define CMP_TRACKPOS_RELATIVE_FRAME     2
979 #define CMP_TRACKPOS_ABSOLUTE_FRAME     3
980
981 /* API */
982 void ntreeCompositExecTree(struct Scene *scene, struct bNodeTree *ntree, struct RenderData *rd, int rendering, int do_previews,
983                            const struct ColorManagedViewSettings *view_settings, const struct ColorManagedDisplaySettings *display_settings,
984                            const char *view_name);
985 void ntreeCompositTagRender(struct Scene *sce);
986 int ntreeCompositTagAnimated(struct bNodeTree *ntree);
987 void ntreeCompositTagGenerators(struct bNodeTree *ntree);
988 void ntreeCompositUpdateRLayers(struct bNodeTree *ntree);
989 void ntreeCompositRegisterPass(struct bNodeTree *ntree, struct Scene *scene, struct ViewLayer *view_layer, const char *name, int type);
990 void ntreeCompositClearTags(struct bNodeTree *ntree);
991
992 struct bNodeSocket *ntreeCompositOutputFileAddSocket(struct bNodeTree *ntree, struct bNode *node,
993                                                      const char *name, struct ImageFormatData *im_format);
994 int ntreeCompositOutputFileRemoveActiveSocket(struct bNodeTree *ntree, struct bNode *node);
995 void ntreeCompositOutputFileSetPath(struct bNode *node, struct bNodeSocket *sock, const char *name);
996 void ntreeCompositOutputFileSetLayer(struct bNode *node, struct bNodeSocket *sock, const char *name);
997 /* needed in do_versions */
998 void ntreeCompositOutputFileUniquePath(struct ListBase *list, struct bNodeSocket *sock, const char defname[], char delim);
999 void ntreeCompositOutputFileUniqueLayer(struct ListBase *list, struct bNodeSocket *sock, const char defname[], char delim);
1000
1001 void ntreeCompositColorBalanceSyncFromLGG(bNodeTree *ntree, bNode *node);
1002 void ntreeCompositColorBalanceSyncFromCDL(bNodeTree *ntree, bNode *node);
1003
1004 /** \} */
1005
1006 /* -------------------------------------------------------------------- */
1007 /** \name Texture Nodes
1008  * \{ */
1009
1010 struct TexResult;
1011
1012 #define TEX_NODE_OUTPUT     401
1013 #define TEX_NODE_CHECKER    402
1014 #define TEX_NODE_TEXTURE    403
1015 #define TEX_NODE_BRICKS     404
1016 #define TEX_NODE_MATH       405
1017 #define TEX_NODE_MIX_RGB    406
1018 #define TEX_NODE_RGBTOBW    407
1019 #define TEX_NODE_VALTORGB   408
1020 #define TEX_NODE_IMAGE      409
1021 #define TEX_NODE_CURVE_RGB  410
1022 #define TEX_NODE_INVERT     411
1023 #define TEX_NODE_HUE_SAT    412
1024 #define TEX_NODE_CURVE_TIME 413
1025 #define TEX_NODE_ROTATE     414
1026 #define TEX_NODE_VIEWER     415
1027 #define TEX_NODE_TRANSLATE  416
1028 #define TEX_NODE_COORD      417
1029 #define TEX_NODE_DISTANCE   418
1030 #define TEX_NODE_COMPOSE    419
1031 #define TEX_NODE_DECOMPOSE  420
1032 #define TEX_NODE_VALTONOR   421
1033 #define TEX_NODE_SCALE      422
1034 #define TEX_NODE_AT         423
1035
1036 /* 501-599 reserved. Use like this: TEX_NODE_PROC + TEX_CLOUDS, etc */
1037 #define TEX_NODE_PROC      500
1038 #define TEX_NODE_PROC_MAX  600
1039
1040 /* API */
1041 int  ntreeTexTagAnimated(struct bNodeTree *ntree);
1042 void ntreeTexCheckCyclics(struct bNodeTree *ntree);
1043
1044 struct bNodeTreeExec *ntreeTexBeginExecTree(struct bNodeTree *ntree);
1045 void ntreeTexEndExecTree(struct bNodeTreeExec *exec);
1046 int ntreeTexExecTree(struct bNodeTree *ntree, struct TexResult *target,
1047                      float coord[3], float dxt[3], float dyt[3], int osatex, const short thread,
1048                      struct Tex *tex, short which_output, int cfra, int preview, struct MTex *mtex);
1049 /** \} */
1050
1051 void init_nodesystem(void);
1052 void free_nodesystem(void);
1053
1054 /* -------------------------------------------------------------------- */
1055 /* evaluation support, */
1056
1057 struct Depsgraph;
1058
1059 void BKE_nodetree_shading_params_eval(struct Depsgraph *depsgraph,
1060                                       struct bNodeTree *ntree_dst,
1061                                       const struct bNodeTree *ntree_src);
1062
1063 #endif  /* __BKE_NODE_H__ */