RNA:
[blender-staging.git] / source / blender / makesrna / intern / rna_nodetree.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  * Contributor(s): Blender Foundation (2008), Nathan Letwory
21  *
22  * ***** END GPL LICENSE BLOCK *****
23  */
24
25 #include <stdlib.h>
26
27 #include "RNA_define.h"
28 #include "RNA_types.h"
29
30 #include "rna_internal.h"
31
32 #include "DNA_node_types.h"
33 #include "BKE_node.h"
34
35 #ifdef RNA_RUNTIME
36
37 StructRNA *rna_Node_refine(struct PointerRNA *ptr)
38 {
39         bNode *node= (bNode*)ptr->data;
40
41         switch(node->type) {
42                 case SH_NODE_OUTPUT:
43                         return &RNA_ShaderNodeOutput;
44                 case SH_NODE_MATERIAL:
45                         return &RNA_ShaderNodeMaterial;
46                 /* XXX complete here */
47                 default:
48                         return &RNA_Node;
49         }
50 }
51
52 #else
53
54 static void rna_def_shader_node_output(BlenderRNA *brna)
55 {
56         StructRNA *srna;
57
58         srna= RNA_def_struct(brna, "ShaderNodeOutput", "ShaderNode");
59         RNA_def_struct_ui_text(srna, "Shader Node Output", "");
60         RNA_def_struct_sdna(srna, "bNode");
61 }
62
63 static void rna_def_shader_node_material(BlenderRNA *brna)
64 {
65         StructRNA *srna;
66         PropertyRNA *prop;
67
68         srna= RNA_def_struct(brna, "ShaderNodeMaterial", "ShaderNode");
69         RNA_def_struct_ui_text(srna, "Shader Node Material", "");
70         RNA_def_struct_sdna(srna, "bNode");
71
72         prop= RNA_def_property(srna, "material", PROP_POINTER, PROP_NONE);
73         RNA_def_property_pointer_sdna(prop, NULL, "id");
74         RNA_def_property_struct_type(prop, "Material");
75         RNA_def_property_ui_text(prop, "Material", "");
76
77         prop= RNA_def_property(srna, "diffuse", PROP_BOOLEAN, PROP_NONE);
78         RNA_def_property_boolean_sdna(prop, NULL, "custom1", SH_NODE_MAT_DIFF);
79         RNA_def_property_ui_text(prop, "Diffuse", "Material Node outputs Diffuse");
80
81         /* XXX add specular, negate normal */
82 }
83
84 static void rna_def_shader_node(BlenderRNA *brna)
85 {
86         StructRNA *srna;
87         PropertyRNA *prop;
88         static EnumPropertyItem node_type_items[] ={
89                 {SH_NODE_OUTPUT, "OUTPUT", "Output", ""},
90                 {SH_NODE_MATERIAL, "MATERIAL", "Material", ""},
91                 {SH_NODE_RGB, "RGB", "RGB", ""},
92                 {SH_NODE_VALUE, "VALUE", "Value", ""},
93                 {SH_NODE_MIX_RGB, "MIX_RGB", "Mix RGB", ""},
94                 {SH_NODE_VALTORGB, "VALUE_TO_RGB", "Value to RGB", ""},
95                 {SH_NODE_RGBTOBW, "RGB_TO_BW", "RGB to BW", ""},
96                 {SH_NODE_TEXTURE, "TEXTURE", "Texture", ""},
97                 {SH_NODE_NORMAL, "NORMAL", "Normal", ""},
98                 {SH_NODE_GEOMETRY, "GEOMETRY", "Geometry", ""},
99                 {SH_NODE_MAPPING, "MAPPING", "Mapping", ""},
100                 {SH_NODE_CURVE_VEC, "VECTOR_CURVES", "Vector Curves", ""},
101                 {SH_NODE_CURVE_RGB, "RGB_CURVES", "RGB Curves", ""},
102                 {SH_NODE_CAMERA, "CAMERA", "Camera", ""},
103                 {SH_NODE_MATH, "MATH", "Math", ""},
104                 {SH_NODE_VECT_MATH, "VECTOR_MATH", "Vector Math", ""},
105                 {SH_NODE_SQUEEZE, "SQUEEZE", "Squeeze", ""},
106                 {SH_NODE_MATERIAL_EXT, "MATERIAL_EXTENDED", "Material Extended", ""},
107                 {SH_NODE_INVERT, "INVERT", "Invert", ""},
108                 {SH_NODE_SEPRGB, "SEPARATE_RGB", "Seperate RGB", ""},
109                 {SH_NODE_COMBRGB, "COMBINE_RGB", "Combine RGB", ""},
110                 {SH_NODE_HUE_SAT, "HUE_SATURATION", "Hue/Saturation", ""},
111                 {NODE_DYNAMIC, "SCRIPT", "Script", ""},
112                 {0, NULL, NULL, NULL}};
113
114         srna= RNA_def_struct(brna, "ShaderNode", "Node");
115         RNA_def_struct_ui_text(srna, "Shader Node", "Material shader node.");
116         RNA_def_struct_sdna(srna, "bNode");
117
118         prop= RNA_def_property(srna, "type", PROP_ENUM, PROP_NONE);
119         RNA_def_property_flag(prop, PROP_NOT_EDITABLE);
120         RNA_def_property_enum_items(prop, node_type_items);
121         RNA_def_property_ui_text(prop, "Type", "");
122
123         /* specific types */
124         rna_def_shader_node_output(brna);
125         rna_def_shader_node_material(brna);
126 }
127
128 static void rna_def_node(BlenderRNA *brna)
129 {
130         StructRNA *srna;
131         PropertyRNA *prop;
132         
133         srna= RNA_def_struct(brna, "Node", NULL);
134         RNA_def_struct_ui_text(srna, "Node", "Node in a node tree.");
135         RNA_def_struct_sdna(srna, "bNode");
136         RNA_def_struct_refine_func(srna, "rna_Node_refine");
137         
138         prop= RNA_def_property(srna, "location", PROP_FLOAT, PROP_VECTOR);
139         RNA_def_property_float_sdna(prop, NULL, "locx");
140         RNA_def_property_array(prop, 2);
141         RNA_def_property_range(prop, -10000.0f, 10000.0f);
142         RNA_def_property_ui_text(prop, "Location", "");
143         
144         prop= RNA_def_property(srna, "name", PROP_STRING, PROP_NONE);
145         RNA_def_property_ui_text(prop, "Name", "Node name.");
146         RNA_def_struct_name_property(srna, prop);
147 }
148
149 static void rna_def_nodetree(BlenderRNA *brna)
150 {
151         StructRNA *srna;
152         PropertyRNA *prop;
153         
154         srna= RNA_def_struct(brna, "NodeTree", "ID");
155         RNA_def_struct_ui_text(srna, "Node Tree", "Node tree consisting of linked nodes used for materials, textures and compositing.");
156         RNA_def_struct_sdna(srna, "bNodeTree");
157
158         prop= RNA_def_property(srna, "nodes", PROP_COLLECTION, PROP_NONE);
159         RNA_def_property_collection_sdna(prop, NULL, "nodes", NULL);
160         RNA_def_property_struct_type(prop, "Node");
161         RNA_def_property_ui_text(prop, "Nodes", "");
162 }
163
164 void RNA_def_nodetree(BlenderRNA *brna)
165 {
166         rna_def_nodetree(brna);
167         rna_def_node(brna);
168         rna_def_shader_node(brna);
169 }
170
171 #endif
172