== PyNodes ==
[blender.git] / source / blender / python / api2_2x / Node.h
1 /* 
2  * $Id$
3  *
4  * ***** BEGIN GPL/BL DUAL 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. The Blender
10  * Foundation also sells licenses for use in proprietary software under
11  * the Blender License.  See http://www.blender.org/BL/ for information
12  * about this.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software Foundation,
21  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
22  *
23  * The Original Code is Copyright (C) 2006, Blender Foundation
24  * All rights reserved.
25  *
26  * Original code is this file
27  *
28  * Contributor(s): Nathan Letwory
29  *
30  * ***** END GPL/BL DUAL LICENSE BLOCK *****
31 */
32
33 #ifndef __NODE_H__
34 #define __NODE_H__
35
36 #include <Python.h>
37 #include "DNA_node_types.h"
38 #include "BKE_node.h"
39
40 #include "RE_shader_ext.h"              /* <- ShadeInput Shaderesult TexResult */
41
42 extern PyTypeObject Node_Type;
43 extern PyTypeObject NodeSocket_Type;
44 extern PyTypeObject NodeSocketLists_Type;
45 extern PyTypeObject ShadeInput_Type;
46
47 #define BPy_Node_Check(v) \
48     ((v)->ob_type == &Node_Type)
49
50 #define BPy_NodeSocket_Check(v) \
51     ((v)->ob_type == &NodeSocket_Type)
52
53 #define BPy_NodeSocketLists_Check(v) \
54     ((v)->ob_type == &NodeSocketLists_Type)
55
56 #define BPy_ShadeInput_Check(v) \
57     ((v)->ob_type == &ShadeInput_Type)
58
59 typedef struct BPy_ShadeInput {
60         PyObject_HEAD
61         ShadeInput *shi;
62 } BPy_ShadeInput;
63
64 typedef struct {
65         PyObject_VAR_HEAD
66         bNode* node;
67         bNodeStack **stack;
68 } BPy_SockMap;
69
70 typedef struct {
71         PyObject_HEAD
72         bNode *node;
73         PyObject *input;
74         PyObject *output;
75 } BPy_NodeSocketLists;
76
77 typedef struct BPy_Node {
78         PyObject_HEAD
79         bNode *node;
80         bNodeStack **in;
81         bNodeStack **out;
82         ShadeInput *shi;
83 } BPy_Node;
84
85 typedef struct BPy_NodeSocket {
86         PyObject_HEAD
87         char name[NODE_MAXSTR];
88         float min;
89         float max;
90         float val[4];
91         short type; /* VALUE, VECTOR or RGBA */
92 } BPy_NodeSocket;
93
94 extern PyObject *Node_Init(void);
95 extern void InitNode(BPy_Node *self, bNode *node);
96 extern BPy_Node *Node_CreatePyObject(bNode *node);
97 extern BPy_NodeSocketLists *Node_CreateSocketLists(bNode *node);
98 extern void Node_SetStack(BPy_Node *self, bNodeStack **stack, int type);
99 extern void Node_SetShi(BPy_Node *self, ShadeInput *shi);
100 extern int pytype_is_pynode(PyObject *pyob);
101
102 #define NODE_INPUTSTACK         0
103 #define NODE_OUTPUTSTACK        1
104
105 #endif /* __NODE_H__*/
106