e229e678fe95c1cd2b112e5f6ca340d6c6115b9f
[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 ShadeInput_Type;
44
45 #define BPy_Node_Check(v) \
46     ((v)->ob_type == &Node_Type)
47
48 #define BPy_ShadeInput_Check(v) \
49     ((v)->ob_type == &ShadeInput_Type)
50
51 typedef struct BPy_ShadeInput {
52         PyObject_HEAD
53         ShadeInput *shi;
54 } BPy_ShadeInput;
55
56 typedef struct {
57         PyObject_VAR_HEAD
58         bNode* node;
59         bNodeStack **stack;
60 } BPy_SockMap;
61
62 typedef struct {
63         PyObject_HEAD
64         bNode *node;
65         PyObject *input;
66         PyObject *output;
67 } BPy_NodeSockets;
68
69 typedef struct BPy_Node {
70         PyObject_HEAD
71         bNode *node;
72         bNodeStack **in;
73         bNodeStack **out;
74         ShadeInput *shi;
75 } BPy_Node;
76
77 extern PyObject *Node_Init(void);
78 extern void InitNode(BPy_Node *self, bNode *node);
79 extern BPy_Node *Node_CreatePyObject(bNode *node);
80 extern BPy_NodeSockets *Node_CreateSockets(bNode *node);
81 extern void Node_SetStack(BPy_Node *self, bNodeStack **stack, int type);
82 extern void Node_SetShi(BPy_Node *self, ShadeInput *shi);
83 extern int pytype_is_pynode(PyObject *pyob);
84
85 #define NODE_INPUTSTACK         0
86 #define NODE_OUTPUTSTACK        1
87
88 #endif /* __NODE_H__*/
89