Fix #31288: Blender crash when select a material
[blender.git] / source / blender / nodes / texture / nodes / node_texture_common.c
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) 2006 Blender Foundation.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): Campbell Barton, Alfredo de Greef, David Millan Escriva,
24  * Juho Vepsäläinen
25  *
26  * ***** END GPL LICENSE BLOCK *****
27  */
28
29 /** \file blender/nodes/texture/nodes/node_texture_common.c
30  *  \ingroup texnodes
31  */
32
33
34 #include "DNA_node_types.h"
35
36 #include "BKE_node.h"
37
38 #include "node_texture_util.h"
39 #include "node_common.h"
40 #include "node_exec.h"
41
42 static void copy_stack(bNodeStack *to, bNodeStack *from)
43 {
44         if (to != from) {
45                 copy_v4_v4(to->vec, from->vec);
46                 to->data = from->data;
47                 to->datatype = from->datatype;
48                 
49                 /* tag as copy to prevent freeing */
50                 to->is_copy = 1;
51         }
52 }
53
54 /**** GROUP ****/
55
56 static void *group_initexec(bNode *node)
57 {
58         bNodeTree *ngroup= (bNodeTree*)node->id;
59         void *exec;
60         
61         if (!ngroup)
62                 return NULL;
63         
64         /* initialize the internal node tree execution */
65         exec = ntreeTexBeginExecTree(ngroup, 0);
66         
67         return exec;
68 }
69
70 static void group_freeexec(bNode *UNUSED(node), void *nodedata)
71 {
72         bNodeTreeExec*gexec= (bNodeTreeExec*)nodedata;
73         
74         ntreeTexEndExecTree(gexec, 0);
75 }
76
77 /* Copy inputs to the internal stack.
78  * This is a shallow copy, no buffers are duplicated here!
79  */
80 static void group_copy_inputs(bNode *node, bNodeStack **in, bNodeStack *gstack)
81 {
82         bNodeSocket *sock;
83         bNodeStack *ns;
84         int a;
85         for (sock=node->inputs.first, a=0; sock; sock=sock->next, ++a) {
86                 if (sock->groupsock) {
87                         ns = node_get_socket_stack(gstack, sock->groupsock);
88                         copy_stack(ns, in[a]);
89                 }
90         }
91 }
92
93 /* Copy internal results to the external outputs.
94  */
95 static void group_copy_outputs(bNode *node, bNodeStack **out, bNodeStack *gstack)
96 {
97         bNodeSocket *sock;
98         bNodeStack *ns;
99         int a;
100         for (sock=node->outputs.first, a=0; sock; sock=sock->next, ++a) {
101                 if (sock->groupsock) {
102                         ns = node_get_socket_stack(gstack, sock->groupsock);
103                         copy_stack(out[a], ns);
104                 }
105         }
106 }
107
108 static void group_execute(void *data, int thread, struct bNode *node, void *nodedata, struct bNodeStack **in, struct bNodeStack **out)
109 {
110         bNodeTreeExec *exec= (bNodeTreeExec*)nodedata;
111         bNodeThreadStack *nts;
112         
113         if (!exec)
114                 return;
115         
116         /* XXX same behavior as trunk: all nodes inside group are executed.
117          * it's stupid, but just makes it work. compo redesign will do this better.
118          */
119         {
120                 bNode *inode;
121                 for (inode=exec->nodetree->nodes.first; inode; inode=inode->next)
122                         inode->need_exec = 1;
123         }
124         
125         nts = ntreeGetThreadStack(exec, thread);
126         
127         group_copy_inputs(node, in, nts->stack);
128         ntreeExecThreadNodes(exec, nts, data, thread);
129         group_copy_outputs(node, out, nts->stack);
130         
131         ntreeReleaseThreadStack(nts);
132 }
133
134 void register_node_type_tex_group(bNodeTreeType *ttype)
135 {
136         static bNodeType ntype;
137
138         node_type_base(ttype, &ntype, NODE_GROUP, "Group", NODE_CLASS_GROUP, NODE_OPTIONS|NODE_CONST_OUTPUT);
139         node_type_socket_templates(&ntype, NULL, NULL);
140         node_type_size(&ntype, 120, 60, 200);
141         node_type_label(&ntype, node_group_label);
142         node_type_init(&ntype, node_group_init);
143         node_type_valid(&ntype, node_group_valid);
144         node_type_template(&ntype, node_group_template);
145         node_type_update(&ntype, NULL, node_group_verify);
146         node_type_group_edit(&ntype, node_group_edit_get, node_group_edit_set, node_group_edit_clear);
147         node_type_exec_new(&ntype, group_initexec, group_freeexec, group_execute);
148         
149         nodeRegisterType(ttype, &ntype);
150 }
151
152
153 /**** FOR LOOP ****/
154
155 #if 0 /* XXX loop nodes don't work nicely with current trees */
156 static void forloop_execute(void *data, int thread, struct bNode *node, void *nodedata, struct bNodeStack **in, struct bNodeStack **out)
157 {
158         bNodeTreeExec *exec= (bNodeTreeExec*)nodedata;
159         bNodeThreadStack *nts;
160         int iterations= (int)in[0]->vec[0];
161         bNodeSocket *sock;
162         bNodeStack *ns;
163         int iteration;
164         
165         /* XXX same behavior as trunk: all nodes inside group are executed.
166          * it's stupid, but just makes it work. compo redesign will do this better.
167          */
168         {
169                 bNode *inode;
170                 for (inode=exec->nodetree->nodes.first; inode; inode=inode->next)
171                         inode->need_exec = 1;
172         }
173         
174         nts = ntreeGetThreadStack(exec, thread);
175         
176         /* "Iteration" socket */
177         sock = exec->nodetree->inputs.first;
178         ns = node_get_socket_stack(nts->stack, sock);
179         
180 //      group_copy_inputs(node, in, nts->stack);
181         for (iteration=0; iteration < iterations; ++iteration) {
182                 /* first input contains current iteration counter */
183                 ns->vec[0] = (float)iteration;
184                 ns->vec[1]=ns->vec[2]=ns->vec[3] = 0.0f;
185                 
186 //              if (iteration > 0)
187 //                      loop_init_iteration(exec->nodetree, nts->stack);
188 //              ntreeExecThreadNodes(exec, nts, data, thread);
189         }
190 //      loop_copy_outputs(node, in, out, exec->stack);
191         
192         ntreeReleaseThreadStack(nts);
193 }
194
195 void register_node_type_tex_forloop(bNodeTreeType *ttype)
196 {
197         static bNodeType ntype;
198
199         node_type_base(ttype, &ntype, NODE_FORLOOP, "For", NODE_CLASS_GROUP, NODE_OPTIONS);
200         node_type_socket_templates(&ntype, NULL, NULL);
201         node_type_size(&ntype, 120, 60, 200);
202         node_type_label(&ntype, node_group_label);
203         node_type_init(&ntype, node_forloop_init);
204         node_type_valid(&ntype, node_group_valid);
205         node_type_template(&ntype, node_forloop_template);
206         node_type_update(&ntype, NULL, node_group_verify);
207         node_type_tree(&ntype, node_forloop_init_tree, node_loop_update_tree);
208         node_type_group_edit(&ntype, node_group_edit_get, node_group_edit_set, node_group_edit_clear);
209         node_type_exec_new(&ntype, group_initexec, group_freeexec, forloop_execute);
210         
211         nodeRegisterType(ttype, &ntype);
212 }
213 #endif
214
215 /**** WHILE LOOP ****/
216
217 #if 0 /* XXX loop nodes don't work nicely with current trees */
218 static void whileloop_execute(void *data, int thread, struct bNode *node, void *nodedata, struct bNodeStack **in, struct bNodeStack **out)
219 {
220         bNodeTreeExec *exec= (bNodeTreeExec*)nodedata;
221         bNodeThreadStack *nts;
222         int condition= (in[0]->vec[0] > 0.0f);
223         bNodeSocket *sock;
224         bNodeStack *ns;
225         int iteration;
226         
227         /* XXX same behavior as trunk: all nodes inside group are executed.
228          * it's stupid, but just makes it work. compo redesign will do this better.
229          */
230         {
231                 bNode *inode;
232                 for (inode=exec->nodetree->nodes.first; inode; inode=inode->next)
233                         inode->need_exec = 1;
234         }
235         
236         nts = ntreeGetThreadStack(exec, thread);
237         
238         /* "Condition" socket */
239         sock = exec->nodetree->outputs.first;
240         ns = node_get_socket_stack(nts->stack, sock);
241         
242         iteration = 0;
243 //      group_copy_inputs(node, in, nts->stack);
244         while (condition && iteration < node->custom1) {
245 //              if (iteration > 0)
246 //                      loop_init_iteration(exec->nodetree, nts->stack);
247 //              ntreeExecThreadNodes(exec, nts, data, thread);
248                 
249                 condition = (ns->vec[0] > 0.0f);
250                 ++iteration;
251         }
252 //      loop_copy_outputs(node, in, out, exec->stack);
253         
254         ntreeReleaseThreadStack(nts);
255 }
256
257 void register_node_type_tex_whileloop(bNodeTreeType *ttype)
258 {
259         static bNodeType ntype;
260
261         node_type_base(ttype, &ntype, NODE_WHILELOOP, "While", NODE_CLASS_GROUP, NODE_OPTIONS);
262         node_type_socket_templates(&ntype, NULL, NULL);
263         node_type_size(&ntype, 120, 60, 200);
264         node_type_label(&ntype, node_group_label);
265         node_type_init(&ntype, node_whileloop_init);
266         node_type_valid(&ntype, node_group_valid);
267         node_type_template(&ntype, node_whileloop_template);
268         node_type_update(&ntype, NULL, node_group_verify);
269         node_type_tree(&ntype, node_whileloop_init_tree, node_loop_update_tree);
270         node_type_group_edit(&ntype, node_group_edit_get, node_group_edit_set, node_group_edit_clear);
271         node_type_exec_new(&ntype, group_initexec, group_freeexec, whileloop_execute);
272         
273         nodeRegisterType(ttype, &ntype);
274 }
275 #endif