diff options
Diffstat (limited to 'source/blender/nodes/shader/node_shader_tree.c')
-rw-r--r-- | source/blender/nodes/shader/node_shader_tree.c | 1279 |
1 files changed, 0 insertions, 1279 deletions
diff --git a/source/blender/nodes/shader/node_shader_tree.c b/source/blender/nodes/shader/node_shader_tree.c deleted file mode 100644 index 46a2f7f1968..00000000000 --- a/source/blender/nodes/shader/node_shader_tree.c +++ /dev/null @@ -1,1279 +0,0 @@ -/* - * This program is free software; you can redistribute it and/or - * modify it under the terms of the GNU General Public License - * as published by the Free Software Foundation; either version 2 - * of the License, or (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program; if not, write to the Free Software Foundation, - * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. - * - * The Original Code is Copyright (C) 2007 Blender Foundation. - * All rights reserved. - */ - -/** \file - * \ingroup nodes - */ - -#include <string.h> - -#include "DNA_light_types.h" -#include "DNA_linestyle_types.h" -#include "DNA_material_types.h" -#include "DNA_node_types.h" -#include "DNA_scene_types.h" -#include "DNA_space_types.h" -#include "DNA_workspace_types.h" -#include "DNA_world_types.h" - -#include "BLI_alloca.h" -#include "BLI_linklist.h" -#include "BLI_listbase.h" -#include "BLI_threads.h" -#include "BLI_utildefines.h" - -#include "BLT_translation.h" - -#include "BKE_context.h" -#include "BKE_lib_id.h" -#include "BKE_linestyle.h" -#include "BKE_node.h" -#include "BKE_scene.h" - -#include "RNA_access.h" - -#include "GPU_material.h" - -#include "RE_texture.h" - -#include "NOD_common.h" - -#include "node_common.h" -#include "node_exec.h" -#include "node_shader_util.h" -#include "node_util.h" - -static bool shader_tree_poll(const bContext *C, bNodeTreeType *UNUSED(treetype)) -{ - Scene *scene = CTX_data_scene(C); - const char *engine_id = scene->r.engine; - - /* Allow empty engine string too, - * this is from older versions that didn't have registerable engines yet. */ - return (engine_id[0] == '\0' || STREQ(engine_id, RE_engine_id_CYCLES) || - !BKE_scene_use_shading_nodes_custom(scene)); -} - -static void shader_get_from_context(const bContext *C, - bNodeTreeType *UNUSED(treetype), - bNodeTree **r_ntree, - ID **r_id, - ID **r_from) -{ - SpaceNode *snode = CTX_wm_space_node(C); - Scene *scene = CTX_data_scene(C); - ViewLayer *view_layer = CTX_data_view_layer(C); - Object *ob = OBACT(view_layer); - - if (snode->shaderfrom == SNODE_SHADER_OBJECT) { - if (ob) { - *r_from = &ob->id; - if (ob->type == OB_LAMP) { - *r_id = ob->data; - *r_ntree = ((Light *)ob->data)->nodetree; - } - else { - Material *ma = BKE_object_material_get(ob, ob->actcol); - if (ma) { - *r_id = &ma->id; - *r_ntree = ma->nodetree; - } - } - } - } -#ifdef WITH_FREESTYLE - else if (snode->shaderfrom == SNODE_SHADER_LINESTYLE) { - FreestyleLineStyle *linestyle = BKE_linestyle_active_from_view_layer(view_layer); - if (linestyle) { - *r_from = NULL; - *r_id = &linestyle->id; - *r_ntree = linestyle->nodetree; - } - } -#endif - else { /* SNODE_SHADER_WORLD */ - if (scene->world) { - *r_from = NULL; - *r_id = &scene->world->id; - *r_ntree = scene->world->nodetree; - } - } -} - -static void foreach_nodeclass(Scene *UNUSED(scene), void *calldata, bNodeClassCallback func) -{ - func(calldata, NODE_CLASS_INPUT, N_("Input")); - func(calldata, NODE_CLASS_OUTPUT, N_("Output")); - func(calldata, NODE_CLASS_SHADER, N_("Shader")); - func(calldata, NODE_CLASS_TEXTURE, N_("Texture")); - func(calldata, NODE_CLASS_OP_COLOR, N_("Color")); - func(calldata, NODE_CLASS_OP_VECTOR, N_("Vector")); - func(calldata, NODE_CLASS_CONVERTER, N_("Converter")); - func(calldata, NODE_CLASS_SCRIPT, N_("Script")); - func(calldata, NODE_CLASS_GROUP, N_("Group")); - func(calldata, NODE_CLASS_INTERFACE, N_("Interface")); - func(calldata, NODE_CLASS_LAYOUT, N_("Layout")); -} - -static void localize(bNodeTree *localtree, bNodeTree *UNUSED(ntree)) -{ - bNode *node, *node_next; - - /* replace muted nodes and reroute nodes by internal links */ - for (node = localtree->nodes.first; node; node = node_next) { - node_next = node->next; - - if (node->flag & NODE_MUTED || node->type == NODE_REROUTE) { - nodeInternalRelink(localtree, node); - ntreeFreeLocalNode(localtree, node); - } - } -} - -static void local_sync(bNodeTree *localtree, bNodeTree *ntree) -{ - BKE_node_preview_sync_tree(ntree, localtree); -} - -static void local_merge(Main *UNUSED(bmain), bNodeTree *localtree, bNodeTree *ntree) -{ - BKE_node_preview_merge_tree(ntree, localtree, true); -} - -static void update(bNodeTree *ntree) -{ - ntreeSetOutput(ntree); - - ntree_update_reroute_nodes(ntree); - - if (ntree->update & NTREE_UPDATE_NODES) { - /* clean up preview cache, in case nodes have been removed */ - BKE_node_preview_remove_unused(ntree); - } -} - -static bool shader_validate_link(bNodeTree *UNUSED(ntree), bNodeLink *link) -{ - /* Can't connect shader into other socket types, other way around is fine - * since it will be interpreted as emission. */ - if (link->fromsock->type == SOCK_SHADER) { - return (link->tosock->type == SOCK_SHADER); - } - return true; -} - -static bool shader_node_tree_socket_type_valid(bNodeTreeType *UNUSED(ntreetype), - bNodeSocketType *socket_type) -{ - return nodeIsStaticSocketType(socket_type) && - ELEM(socket_type->type, SOCK_FLOAT, SOCK_VECTOR, SOCK_RGBA, SOCK_SHADER); -} - -bNodeTreeType *ntreeType_Shader; - -void register_node_tree_type_sh(void) -{ - bNodeTreeType *tt = ntreeType_Shader = MEM_callocN(sizeof(bNodeTreeType), - "shader node tree type"); - - tt->type = NTREE_SHADER; - strcpy(tt->idname, "ShaderNodeTree"); - strcpy(tt->ui_name, N_("Shader Editor")); - tt->ui_icon = 0; /* defined in drawnode.c */ - strcpy(tt->ui_description, N_("Shader nodes")); - - tt->foreach_nodeclass = foreach_nodeclass; - tt->localize = localize; - tt->local_sync = local_sync; - tt->local_merge = local_merge; - tt->update = update; - tt->poll = shader_tree_poll; - tt->get_from_context = shader_get_from_context; - tt->validate_link = shader_validate_link; - tt->valid_socket_type = shader_node_tree_socket_type_valid; - - tt->rna_ext.srna = &RNA_ShaderNodeTree; - - ntreeTypeAdd(tt); -} - -/* GPU material from shader nodes */ - -/* Find an output node of the shader tree. - * - * NOTE: it will only return output which is NOT in the group, which isn't how - * render engines works but it's how the GPU shader compilation works. This we - * can change in the future and make it a generic function, but for now it stays - * private here. - */ -bNode *ntreeShaderOutputNode(bNodeTree *ntree, int target) -{ - /* Make sure we only have single node tagged as output. */ - ntreeSetOutput(ntree); - - /* Find output node that matches type and target. If there are - * multiple, we prefer exact target match and active nodes. */ - bNode *output_node = NULL; - - LISTBASE_FOREACH (bNode *, node, &ntree->nodes) { - if (!ELEM(node->type, SH_NODE_OUTPUT_MATERIAL, SH_NODE_OUTPUT_WORLD, SH_NODE_OUTPUT_LIGHT)) { - continue; - } - - if (node->custom1 == SHD_OUTPUT_ALL) { - if (output_node == NULL) { - output_node = node; - } - else if (output_node->custom1 == SHD_OUTPUT_ALL) { - if ((node->flag & NODE_DO_OUTPUT) && !(output_node->flag & NODE_DO_OUTPUT)) { - output_node = node; - } - } - } - else if (node->custom1 == target) { - if (output_node == NULL) { - output_node = node; - } - else if (output_node->custom1 == SHD_OUTPUT_ALL) { - output_node = node; - } - else if ((node->flag & NODE_DO_OUTPUT) && !(output_node->flag & NODE_DO_OUTPUT)) { - output_node = node; - } - } - } - - return output_node; -} - -/* Find socket with a specified identifier. */ -static bNodeSocket *ntree_shader_node_find_socket(ListBase *sockets, const char *identifier) -{ - for (bNodeSocket *sock = sockets->first; sock != NULL; sock = sock->next) { - if (STREQ(sock->identifier, identifier)) { - return sock; - } - } - return NULL; -} - -/* Find input socket with a specified identifier. */ -static bNodeSocket *ntree_shader_node_find_input(bNode *node, const char *identifier) -{ - return ntree_shader_node_find_socket(&node->inputs, identifier); -} - -/* Find output socket with a specified identifier. */ -static bNodeSocket *ntree_shader_node_find_output(bNode *node, const char *identifier) -{ - return ntree_shader_node_find_socket(&node->outputs, identifier); -} - -/* Find input socket at a specific position. */ -static bNodeSocket *ntree_shader_node_input_get(bNode *node, int n) -{ - return BLI_findlink(&node->inputs, n); -} - -/* Find output socket at a specific position. */ -static bNodeSocket *ntree_shader_node_output_get(bNode *node, int n) -{ - return BLI_findlink(&node->outputs, n); -} - -/* Return true on success. */ -static bool ntree_shader_expand_socket_default(bNodeTree *localtree, - bNode *node, - bNodeSocket *socket) -{ - bNode *value_node; - bNodeSocket *value_socket; - bNodeSocketValueVector *src_vector; - bNodeSocketValueRGBA *src_rgba, *dst_rgba; - bNodeSocketValueFloat *src_float, *dst_float; - bNodeSocketValueInt *src_int; - - switch (socket->type) { - case SOCK_VECTOR: - value_node = nodeAddStaticNode(NULL, localtree, SH_NODE_RGB); - value_socket = ntree_shader_node_find_output(value_node, "Color"); - BLI_assert(value_socket != NULL); - src_vector = socket->default_value; - dst_rgba = value_socket->default_value; - copy_v3_v3(dst_rgba->value, src_vector->value); - dst_rgba->value[3] = 1.0f; /* should never be read */ - break; - case SOCK_RGBA: - value_node = nodeAddStaticNode(NULL, localtree, SH_NODE_RGB); - value_socket = ntree_shader_node_find_output(value_node, "Color"); - BLI_assert(value_socket != NULL); - src_rgba = socket->default_value; - dst_rgba = value_socket->default_value; - copy_v4_v4(dst_rgba->value, src_rgba->value); - break; - case SOCK_INT: - /* HACK: Support as float. */ - value_node = nodeAddStaticNode(NULL, localtree, SH_NODE_VALUE); - value_socket = ntree_shader_node_find_output(value_node, "Value"); - BLI_assert(value_socket != NULL); - src_int = socket->default_value; - dst_float = value_socket->default_value; - dst_float->value = (float)(src_int->value); - break; - case SOCK_FLOAT: - value_node = nodeAddStaticNode(NULL, localtree, SH_NODE_VALUE); - value_socket = ntree_shader_node_find_output(value_node, "Value"); - BLI_assert(value_socket != NULL); - src_float = socket->default_value; - dst_float = value_socket->default_value; - dst_float->value = src_float->value; - break; - default: - return false; - } - nodeAddLink(localtree, value_node, value_socket, node, socket); - return true; -} - -static void ntree_shader_unlink_hidden_value_sockets(bNode *group_node, bNodeSocket *isock) -{ - bNodeTree *group_ntree = (bNodeTree *)group_node->id; - bNode *node; - bool removed_link = false; - - for (node = group_ntree->nodes.first; node; node = node->next) { - const bool is_group = ELEM(node->type, NODE_GROUP, NODE_CUSTOM_GROUP) && (node->id != NULL); - - LISTBASE_FOREACH (bNodeSocket *, sock, &node->inputs) { - if (!is_group && (sock->flag & SOCK_HIDE_VALUE) == 0) { - continue; - } - /* If socket is linked to a group input node and sockets id match. */ - if (sock && sock->link && sock->link->fromnode->type == NODE_GROUP_INPUT) { - if (STREQ(isock->identifier, sock->link->fromsock->identifier)) { - if (is_group) { - /* Recursively unlink sockets within the nested group. */ - ntree_shader_unlink_hidden_value_sockets(node, sock); - } - else { - nodeRemLink(group_ntree, sock->link); - removed_link = true; - } - } - } - } - } - - if (removed_link) { - ntreeUpdateTree(G.main, group_ntree); - } -} - -/* Node groups once expanded looses their input sockets values. - * To fix this, link value/rgba nodes into the sockets and copy the group sockets values. */ -static void ntree_shader_groups_expand_inputs(bNodeTree *localtree) -{ - bool link_added = false; - - LISTBASE_FOREACH (bNode *, node, &localtree->nodes) { - const bool is_group = ELEM(node->type, NODE_GROUP, NODE_CUSTOM_GROUP) && (node->id != NULL); - const bool is_group_output = node->type == NODE_GROUP_OUTPUT && (node->flag & NODE_DO_OUTPUT); - - if (is_group) { - /* Do it recursively. */ - ntree_shader_groups_expand_inputs((bNodeTree *)node->id); - } - - if (is_group || is_group_output) { - LISTBASE_FOREACH (bNodeSocket *, socket, &node->inputs) { - if (socket->link != NULL && !(socket->link->flag & NODE_LINK_MUTED)) { - bNodeLink *link = socket->link; - /* Fix the case where the socket is actually converting the data. (see T71374) - * We only do the case of lossy conversion to float. */ - if ((socket->type == SOCK_FLOAT) && (link->fromsock->type != link->tosock->type)) { - if (link->fromsock->type == SOCK_RGBA) { - bNode *tmp = nodeAddStaticNode(NULL, localtree, SH_NODE_RGBTOBW); - nodeAddLink(localtree, link->fromnode, link->fromsock, tmp, tmp->inputs.first); - nodeAddLink(localtree, tmp, tmp->outputs.first, node, socket); - } - else if (link->fromsock->type == SOCK_VECTOR) { - bNode *tmp = nodeAddStaticNode(NULL, localtree, SH_NODE_VECTOR_MATH); - tmp->custom1 = NODE_VECTOR_MATH_DOT_PRODUCT; - bNodeSocket *dot_input1 = tmp->inputs.first; - bNodeSocket *dot_input2 = dot_input1->next; - bNodeSocketValueVector *input2_socket_value = dot_input2->default_value; - copy_v3_fl(input2_socket_value->value, 1.0f / 3.0f); - nodeAddLink(localtree, link->fromnode, link->fromsock, tmp, dot_input1); - nodeAddLink(localtree, tmp, tmp->outputs.last, node, socket); - } - } - continue; - } - - if (is_group) { - /* Detect the case where an input is plugged into a hidden value socket. - * In this case we should just remove the link to trigger the socket default override. */ - ntree_shader_unlink_hidden_value_sockets(node, socket); - } - - if (ntree_shader_expand_socket_default(localtree, node, socket)) { - link_added = true; - } - } - } - } - - if (link_added) { - ntreeUpdateTree(G.main, localtree); - } -} - -static void flatten_group_do(bNodeTree *ntree, bNode *gnode) -{ - bNodeLink *link, *linkn, *tlink; - bNode *node, *nextnode; - bNodeTree *ngroup; - LinkNode *group_interface_nodes = NULL; - - ngroup = (bNodeTree *)gnode->id; - - /* Add the nodes into the ntree */ - for (node = ngroup->nodes.first; node; node = nextnode) { - nextnode = node->next; - /* Remove interface nodes. - * This also removes remaining links to and from interface nodes. - * We must delay removal since sockets will reference this node. see: T52092 */ - if (ELEM(node->type, NODE_GROUP_INPUT, NODE_GROUP_OUTPUT)) { - BLI_linklist_prepend(&group_interface_nodes, node); - } - /* migrate node */ - BLI_remlink(&ngroup->nodes, node); - BLI_addtail(&ntree->nodes, node); - /* ensure unique node name in the node tree */ - /* This is very slow and it has no use for GPU nodetree. (see T70609) */ - // nodeUniqueName(ntree, node); - } - - /* Save first and last link to iterate over flattened group links. */ - bNodeLink *glinks_first = ntree->links.last; - - /* Add internal links to the ntree */ - for (link = ngroup->links.first; link; link = linkn) { - linkn = link->next; - BLI_remlink(&ngroup->links, link); - BLI_addtail(&ntree->links, link); - } - - bNodeLink *glinks_last = ntree->links.last; - - /* restore external links to and from the gnode */ - if (glinks_first != NULL) { - /* input links */ - for (link = glinks_first->next; link != glinks_last->next; link = link->next) { - if (link->fromnode->type == NODE_GROUP_INPUT) { - const char *identifier = link->fromsock->identifier; - /* find external links to this input */ - for (tlink = ntree->links.first; tlink != glinks_first->next; tlink = tlink->next) { - if (tlink->tonode == gnode && STREQ(tlink->tosock->identifier, identifier)) { - nodeAddLink(ntree, tlink->fromnode, tlink->fromsock, link->tonode, link->tosock); - } - } - } - } - /* Also iterate over the new links to cover passthrough links. */ - glinks_last = ntree->links.last; - /* output links */ - for (tlink = ntree->links.first; tlink != glinks_first->next; tlink = tlink->next) { - if (tlink->fromnode == gnode) { - const char *identifier = tlink->fromsock->identifier; - /* find internal links to this output */ - for (link = glinks_first->next; link != glinks_last->next; link = link->next) { - /* only use active output node */ - if (link->tonode->type == NODE_GROUP_OUTPUT && (link->tonode->flag & NODE_DO_OUTPUT)) { - if (STREQ(link->tosock->identifier, identifier)) { - nodeAddLink(ntree, link->fromnode, link->fromsock, tlink->tonode, tlink->tosock); - } - } - } - } - } - } - - while (group_interface_nodes) { - node = BLI_linklist_pop(&group_interface_nodes); - ntreeFreeLocalNode(ntree, node); - } - - ntree->update |= NTREE_UPDATE_NODES | NTREE_UPDATE_LINKS; -} - -/* Flatten group to only have a simple single tree */ -static void ntree_shader_groups_flatten(bNodeTree *localtree) -{ - /* This is effectively recursive as the flattened groups will add - * nodes at the end of the list, which will also get evaluated. */ - for (bNode *node = localtree->nodes.first, *node_next; node; node = node_next) { - if (ELEM(node->type, NODE_GROUP, NODE_CUSTOM_GROUP) && node->id != NULL) { - flatten_group_do(localtree, node); - /* Continue even on new flattened nodes. */ - node_next = node->next; - /* delete the group instance and its localtree. */ - bNodeTree *ngroup = (bNodeTree *)node->id; - ntreeFreeLocalNode(localtree, node); - ntreeFreeTree(ngroup); - BLI_assert(!ngroup->id.py_instance); /* Or call #BKE_libblock_free_data_py. */ - MEM_freeN(ngroup); - } - else { - node_next = node->next; - } - } - - ntreeUpdateTree(G.main, localtree); -} - -/* Check whether shader has a displacement. - * - * Will also return a node and its socket which is connected to a displacement - * output. Additionally, link which is attached to the displacement output is - * also returned. - */ -static bool ntree_shader_has_displacement(bNodeTree *ntree, - bNode *output_node, - bNode **r_node, - bNodeSocket **r_socket, - bNodeLink **r_link) -{ - if (output_node == NULL) { - /* We can't have displacement without output node, apparently. */ - return false; - } - /* Make sure sockets links pointers are correct. */ - ntreeUpdateTree(G.main, ntree); - bNodeSocket *displacement = ntree_shader_node_find_input(output_node, "Displacement"); - - if (displacement == NULL) { - /* Non-cycles node is used as an output. */ - return false; - } - - if ((displacement->link != NULL) && !(displacement->link->flag & NODE_LINK_MUTED)) { - *r_node = displacement->link->fromnode; - *r_socket = displacement->link->fromsock; - *r_link = displacement->link; - return true; - } - return false; -} - -static void ntree_shader_relink_node_normal(bNodeTree *ntree, - bNode *node, - bNode *node_from, - bNodeSocket *socket_from) -{ - /* TODO(sergey): Can we do something smarter here than just a name-based - * matching? - */ - LISTBASE_FOREACH (bNodeSocket *, sock, &node->inputs) { - if (STREQ(sock->identifier, "Normal") && sock->link == NULL) { - /* It's a normal input and nothing is connected to it. */ - nodeAddLink(ntree, node_from, socket_from, node, sock); - } - else if (sock->link) { - bNodeLink *link = sock->link; - if (ELEM(link->fromnode->type, SH_NODE_NEW_GEOMETRY, SH_NODE_TEX_COORD) && - STREQ(link->fromsock->identifier, "Normal")) { - /* Linked to a geometry node normal output. */ - nodeAddLink(ntree, node_from, socket_from, node, sock); - } - } - } -} - -/* Use specified node and socket as an input for unconnected normal sockets. */ -static void ntree_shader_link_builtin_normal(bNodeTree *ntree, - bNode *node_from, - bNodeSocket *socket_from) -{ - for (bNode *node = ntree->nodes.first; node != NULL; node = node->next) { - if (node == node_from) { - /* Don't connect node itself! */ - continue; - } - if (node->tmp_flag == -2) { - /* This node is used inside the displacement tree. Skip to avoid cycles. */ - continue; - } - ntree_shader_relink_node_normal(ntree, node, node_from, socket_from); - } -} - -static void ntree_shader_bypass_bump_link(bNodeTree *ntree, bNode *bump_node, bNodeLink *bump_link) -{ - /* Bypass bump nodes. This replicates cycles "implicit" behavior. */ - bNodeSocket *bump_normal_input = ntree_shader_node_find_input(bump_node, "Normal"); - bNode *fromnode; - bNodeSocket *fromsock; - /* Default to builtin normals if there is no link. */ - if (bump_normal_input->link) { - fromsock = bump_normal_input->link->fromsock; - fromnode = bump_normal_input->link->fromnode; - } - else { - fromnode = nodeAddStaticNode(NULL, ntree, SH_NODE_NEW_GEOMETRY); - fromsock = ntree_shader_node_find_output(fromnode, "Normal"); - } - /* Bypass the bump node by creating a link between the previous and next node. */ - nodeAddLink(ntree, fromnode, fromsock, bump_link->tonode, bump_link->tosock); - nodeRemLink(ntree, bump_link); -} - -static void ntree_shader_bypass_tagged_bump_nodes(bNodeTree *ntree) -{ - /* Bypass bump links inside copied nodes */ - LISTBASE_FOREACH_MUTABLE (bNodeLink *, link, &ntree->links) { - bNode *node = link->fromnode; - /* If node is a copy. */ - if (node->tmp_flag == -2 && node->type == SH_NODE_BUMP) { - ntree_shader_bypass_bump_link(ntree, node, link); - } - } - ntreeUpdateTree(G.main, ntree); -} - -static bool ntree_branch_count_and_tag_nodes(bNode *fromnode, bNode *tonode, void *userdata) -{ - int *node_count = (int *)userdata; - if (fromnode->tmp_flag == -1) { - fromnode->tmp_flag = *node_count; - (*node_count)++; - } - if (tonode->tmp_flag == -1) { - tonode->tmp_flag = *node_count; - (*node_count)++; - } - return true; -} - -/* Create a copy of a branch starting from a given node. - * callback is executed once for every copied node. - * Returns input node copy. */ -static bNode *ntree_shader_copy_branch(bNodeTree *ntree, - bNode *start_node, - void (*callback)(bNode *node, int user_data), - int user_data) -{ - /* Init tmp flag. */ - LISTBASE_FOREACH (bNode *, node, &ntree->nodes) { - node->tmp_flag = -1; - } - /* Count and tag all nodes inside the displacement branch of the tree. */ - start_node->tmp_flag = 0; - int node_count = 1; - nodeChainIterBackwards(ntree, start_node, ntree_branch_count_and_tag_nodes, &node_count, 1); - /* Make a full copy of the branch */ - bNode **nodes_copy = MEM_mallocN(sizeof(bNode *) * node_count, __func__); - LISTBASE_FOREACH (bNode *, node, &ntree->nodes) { - if (node->tmp_flag >= 0) { - int id = node->tmp_flag; - nodes_copy[id] = BKE_node_copy_ex( - ntree, node, LIB_ID_CREATE_NO_USER_REFCOUNT | LIB_ID_CREATE_NO_MAIN, false); - nodes_copy[id]->tmp_flag = -2; /* Copy */ - /* Make sure to clear all sockets links as they are invalid. */ - LISTBASE_FOREACH (bNodeSocket *, sock, &nodes_copy[id]->inputs) { - sock->link = NULL; - } - LISTBASE_FOREACH (bNodeSocket *, sock, &nodes_copy[id]->outputs) { - sock->link = NULL; - } - } - } - /* Recreate links between copied nodes. */ - LISTBASE_FOREACH (bNodeLink *, link, &ntree->links) { - if (link->fromnode->tmp_flag >= 0 && link->tonode->tmp_flag >= 0) { - bNode *fromnode = nodes_copy[link->fromnode->tmp_flag]; - bNode *tonode = nodes_copy[link->tonode->tmp_flag]; - bNodeSocket *fromsock = ntree_shader_node_find_output(fromnode, link->fromsock->identifier); - bNodeSocket *tosock = ntree_shader_node_find_input(tonode, link->tosock->identifier); - nodeAddLink(ntree, fromnode, fromsock, tonode, tosock); - } - } - /* Per node callback. */ - if (callback) { - for (int i = 0; i < node_count; i++) { - callback(nodes_copy[i], user_data); - } - } - bNode *start_node_copy = nodes_copy[start_node->tmp_flag]; - MEM_freeN(nodes_copy); - return start_node_copy; -} - -static void ntree_shader_copy_branch_displacement(bNodeTree *ntree, - bNode *displacement_node, - bNodeSocket *displacement_socket, - bNodeLink *displacement_link) -{ - /* Replace displacement socket/node/link. */ - bNode *tonode = displacement_link->tonode; - bNodeSocket *tosock = displacement_link->tosock; - displacement_node = ntree_shader_copy_branch(ntree, displacement_node, NULL, 0); - displacement_socket = ntree_shader_node_find_output(displacement_node, - displacement_socket->identifier); - nodeRemLink(ntree, displacement_link); - nodeAddLink(ntree, displacement_node, displacement_socket, tonode, tosock); - - ntreeUpdateTree(G.main, ntree); -} - -/* Re-link displacement output to unconnected normal sockets via bump node. - * This way material with have proper displacement in the viewport. - */ -static void ntree_shader_relink_displacement(bNodeTree *ntree, bNode *output_node) -{ - bNode *displacement_node; - bNodeSocket *displacement_socket; - bNodeLink *displacement_link; - if (!ntree_shader_has_displacement( - ntree, output_node, &displacement_node, &displacement_socket, &displacement_link)) { - /* There is no displacement output connected, nothing to re-link. */ - return; - } - - /* Copy the whole displacement branch to avoid cyclic dependency - * and issue when bypassing bump nodes. */ - ntree_shader_copy_branch_displacement( - ntree, displacement_node, displacement_socket, displacement_link); - /* Bypass bump nodes inside the copied branch to mimic cycles behavior. */ - ntree_shader_bypass_tagged_bump_nodes(ntree); - - /* Displacement Node may have changed because of branch copy and bump bypass. */ - ntree_shader_has_displacement( - ntree, output_node, &displacement_node, &displacement_socket, &displacement_link); - - /* We have to disconnect displacement output socket, otherwise we'll have - * cycles in the Cycles material :) - */ - nodeRemLink(ntree, displacement_link); - - /* Convert displacement vector to bump height. */ - bNode *dot_node = nodeAddStaticNode(NULL, ntree, SH_NODE_VECTOR_MATH); - bNode *geo_node = nodeAddStaticNode(NULL, ntree, SH_NODE_NEW_GEOMETRY); - bNodeSocket *normal_socket = ntree_shader_node_find_output(geo_node, "Normal"); - bNodeSocket *dot_input1 = dot_node->inputs.first; - bNodeSocket *dot_input2 = dot_input1->next; - dot_node->custom1 = NODE_VECTOR_MATH_DOT_PRODUCT; - - nodeAddLink(ntree, displacement_node, displacement_socket, dot_node, dot_input1); - nodeAddLink(ntree, geo_node, normal_socket, dot_node, dot_input2); - displacement_node = dot_node; - displacement_socket = ntree_shader_node_find_output(dot_node, "Value"); - - /* We can't connect displacement to normal directly, use bump node for that - * and hope that it gives good enough approximation. - */ - bNode *bump_node = nodeAddStaticNode(NULL, ntree, SH_NODE_BUMP); - bNodeSocket *bump_input_socket = ntree_shader_node_find_input(bump_node, "Height"); - bNodeSocket *bump_output_socket = ntree_shader_node_find_output(bump_node, "Normal"); - BLI_assert(bump_input_socket != NULL); - BLI_assert(bump_output_socket != NULL); - /* Connect bump node to where displacement output was originally - * connected to. - */ - nodeAddLink(ntree, displacement_node, displacement_socket, bump_node, bump_input_socket); - - /* Tag as part of the new displacement tree. */ - dot_node->tmp_flag = -2; - geo_node->tmp_flag = -2; - bump_node->tmp_flag = -2; - - ntreeUpdateTree(G.main, ntree); - - /* Connect all free-standing Normal inputs and relink geometry/coordinate nodes. */ - ntree_shader_link_builtin_normal(ntree, bump_node, bump_output_socket); - /* We modified the tree, it needs to be updated now. */ - ntreeUpdateTree(G.main, ntree); -} - -static void node_tag_branch_as_derivative(bNode *node, int dx) -{ - if (dx) { - node->branch_tag = 1; - } - else { - node->branch_tag = 2; - } -} - -static bool ntree_shader_bump_branches(bNode *fromnode, bNode *UNUSED(tonode), void *userdata) -{ - bNodeTree *ntree = (bNodeTree *)userdata; - - if (fromnode->type == SH_NODE_BUMP) { - bNodeSocket *height_dx_sock, *height_dy_sock, *bump_socket, *bump_dx_socket, *bump_dy_socket; - bNode *bump = fromnode; - bump_socket = ntree_shader_node_find_input(bump, "Height"); - bump_dx_socket = ntree_shader_node_find_input(bump, "Height_dx"); - bump_dy_socket = ntree_shader_node_find_input(bump, "Height_dy"); - if (bump_dx_socket->link) { - /* Avoid reconnecting the same bump twice. */ - } - else if (bump_socket && bump_socket->link) { - bNodeLink *link = bump_socket->link; - bNode *height = link->fromnode; - bNode *height_dx = ntree_shader_copy_branch(ntree, height, node_tag_branch_as_derivative, 1); - bNode *height_dy = ntree_shader_copy_branch(ntree, height, node_tag_branch_as_derivative, 0); - height_dx_sock = ntree_shader_node_find_output(height_dx, link->fromsock->identifier); - height_dy_sock = ntree_shader_node_find_output(height_dy, link->fromsock->identifier); - nodeAddLink(ntree, height_dx, height_dx_sock, bump, bump_dx_socket); - nodeAddLink(ntree, height_dy, height_dy_sock, bump, bump_dy_socket); - /* We could end iter here, but other bump node could be plugged into other input sockets. */ - } - } - return true; -} - -/* Generate emission node to convert regular data to closure sockets. - * Returns validity of the tree. - */ -static bool ntree_shader_implicit_closure_cast(bNodeTree *ntree) -{ - bool modified = false; - LISTBASE_FOREACH_MUTABLE (bNodeLink *, link, &ntree->links) { - if ((link->fromsock->type != SOCK_SHADER) && (link->tosock->type == SOCK_SHADER)) { - bNode *emission_node = nodeAddStaticNode(NULL, ntree, SH_NODE_EMISSION); - bNodeSocket *in_sock = ntree_shader_node_find_input(emission_node, "Color"); - bNodeSocket *out_sock = ntree_shader_node_find_output(emission_node, "Emission"); - nodeAddLink(ntree, link->fromnode, link->fromsock, emission_node, in_sock); - nodeAddLink(ntree, emission_node, out_sock, link->tonode, link->tosock); - nodeRemLink(ntree, link); - modified = true; - } - else if ((link->fromsock->type == SOCK_SHADER) && (link->tosock->type != SOCK_SHADER)) { - /* Meh. Not directly visible to the user. But better than nothing. */ - fprintf(stderr, "Shader Nodetree Error: Invalid implicit socket conversion\n"); - ntreeUpdateTree(G.main, ntree); - return false; - } - } - if (modified) { - ntreeUpdateTree(G.main, ntree); - } - return true; -} - -/* Socket already has a link to it. Add weights together. */ -static void ntree_weight_tree_merge_weight(bNodeTree *ntree, - bNode *UNUSED(fromnode), - bNodeSocket *fromsock, - bNode **tonode, - bNodeSocket **tosock) -{ - bNode *addnode = nodeAddStaticNode(NULL, ntree, SH_NODE_MATH); - addnode->custom1 = NODE_MATH_ADD; - addnode->tmp_flag = -2; /* Copy */ - bNodeSocket *addsock_out = ntree_shader_node_output_get(addnode, 0); - bNodeSocket *addsock_in0 = ntree_shader_node_input_get(addnode, 0); - bNodeSocket *addsock_in1 = ntree_shader_node_input_get(addnode, 1); - bNodeLink *oldlink = fromsock->link; - nodeAddLink(ntree, oldlink->fromnode, oldlink->fromsock, addnode, addsock_in0); - nodeAddLink(ntree, *tonode, *tosock, addnode, addsock_in1); - nodeRemLink(ntree, oldlink); - *tonode = addnode; - *tosock = addsock_out; -} - -static bool ntree_weight_tree_tag_nodes(bNode *fromnode, bNode *tonode, void *userdata) -{ - int *node_count = (int *)userdata; - bool to_node_from_weight_tree = ELEM(tonode->type, - SH_NODE_ADD_SHADER, - SH_NODE_MIX_SHADER, - SH_NODE_OUTPUT_WORLD, - SH_NODE_OUTPUT_MATERIAL); - if (tonode->tmp_flag == -1 && to_node_from_weight_tree) { - tonode->tmp_flag = *node_count; - *node_count += (tonode->type == SH_NODE_MIX_SHADER) ? 4 : 1; - } - if (fromnode->tmp_flag == -1 && ELEM(fromnode->type, SH_NODE_ADD_SHADER, SH_NODE_MIX_SHADER)) { - fromnode->tmp_flag = *node_count; - *node_count += (fromnode->type == SH_NODE_MIX_SHADER) ? 4 : 1; - } - return to_node_from_weight_tree; -} - -/* Invert evaluation order of the weight tree (add & mix closure nodes) to feed the closure nodes - * with their respective weights. */ -static void ntree_shader_weight_tree_invert(bNodeTree *ntree, bNode *output_node) -{ - bNodeLink *displace_link = NULL; - bNodeSocket *displace_output = ntree_shader_node_find_input(output_node, "Displacement"); - if (displace_output && displace_output->link) { - /* Remove any displacement link to avoid tagging it later on. */ - displace_link = displace_output->link; - displace_output->link = NULL; - } - bNodeLink *thickness_link = NULL; - bNodeSocket *thickness_output = ntree_shader_node_find_input(output_node, "Thickness"); - if (thickness_output && thickness_output->link) { - /* Remove any thickness link to avoid tagging it later on. */ - thickness_link = thickness_output->link; - thickness_output->link = NULL; - } - /* Init tmp flag. */ - LISTBASE_FOREACH (bNode *, node, &ntree->nodes) { - node->tmp_flag = -1; - } - /* Tag nodes from the weight tree. Only tag output node and mix/add shader nodes. */ - output_node->tmp_flag = 0; - int node_count = 1; - nodeChainIterBackwards(ntree, output_node, ntree_weight_tree_tag_nodes, &node_count, 0); - /* Make a mirror copy of the weight tree. */ - bNode **nodes_copy = MEM_mallocN(sizeof(bNode *) * node_count, __func__); - LISTBASE_FOREACH (bNode *, node, &ntree->nodes) { - if (node->tmp_flag >= 0) { - int id = node->tmp_flag; - - switch (node->type) { - case SH_NODE_OUTPUT_WORLD: - case SH_NODE_OUTPUT_MATERIAL: { - /* Start the tree with full weight. */ - nodes_copy[id] = nodeAddStaticNode(NULL, ntree, SH_NODE_VALUE); - nodes_copy[id]->tmp_flag = -2; /* Copy */ - ((bNodeSocketValueFloat *)ntree_shader_node_output_get(nodes_copy[id], 0)->default_value) - ->value = 1.0f; - break; - } - case SH_NODE_ADD_SHADER: { - /* Simple passthrough node. Each original inputs will get the same weight. */ - /* TODO(fclem) Better use some kind of reroute node? */ - nodes_copy[id] = nodeAddStaticNode(NULL, ntree, SH_NODE_MATH); - nodes_copy[id]->custom1 = NODE_MATH_ADD; - nodes_copy[id]->tmp_flag = -2; /* Copy */ - ((bNodeSocketValueFloat *)ntree_shader_node_input_get(nodes_copy[id], 0)->default_value) - ->value = 0.0f; - break; - } - case SH_NODE_MIX_SHADER: { - /* We need multiple nodes to emulate the mix node in reverse. */ - bNode *fromnode, *tonode; - bNodeSocket *fromsock, *tosock; - int id_start = id; - /* output = (factor * input_weight) */ - nodes_copy[id] = nodeAddStaticNode(NULL, ntree, SH_NODE_MATH); - nodes_copy[id]->custom1 = NODE_MATH_MULTIPLY; - nodes_copy[id]->tmp_flag = -2; /* Copy */ - id++; - /* output = ((1.0 - factor) * input_weight) <=> (input_weight - factor * input_weight) */ - nodes_copy[id] = nodeAddStaticNode(NULL, ntree, SH_NODE_MATH); - nodes_copy[id]->custom1 = NODE_MATH_SUBTRACT; - nodes_copy[id]->tmp_flag = -2; /* Copy */ - id++; - /* Node sanitizes the input mix factor by clamping it. */ - nodes_copy[id] = nodeAddStaticNode(NULL, ntree, SH_NODE_MATH); - nodes_copy[id]->custom1 = NODE_MATH_ADD; - nodes_copy[id]->custom2 = SHD_MATH_CLAMP; - nodes_copy[id]->tmp_flag = -2; /* Copy */ - ((bNodeSocketValueFloat *)ntree_shader_node_input_get(nodes_copy[id], 0)->default_value) - ->value = 0.0f; - /* Copy default value if no link present. */ - bNodeSocket *fac_sock = ntree_shader_node_find_input(node, "Fac"); - if (!fac_sock->link) { - float default_value = ((bNodeSocketValueFloat *)fac_sock->default_value)->value; - bNodeSocket *dst_sock = ntree_shader_node_input_get(nodes_copy[id], 1); - ((bNodeSocketValueFloat *)dst_sock->default_value)->value = default_value; - } - id++; - /* Reroute the weight input to the 3 processing nodes. Simplify linking later-on. */ - /* TODO(fclem) Better use some kind of reroute node? */ - nodes_copy[id] = nodeAddStaticNode(NULL, ntree, SH_NODE_MATH); - nodes_copy[id]->custom1 = NODE_MATH_ADD; - nodes_copy[id]->tmp_flag = -2; /* Copy */ - ((bNodeSocketValueFloat *)ntree_shader_node_input_get(nodes_copy[id], 0)->default_value) - ->value = 0.0f; - id++; - /* Link between nodes for the substraction. */ - fromnode = nodes_copy[id_start]; - tonode = nodes_copy[id_start + 1]; - fromsock = ntree_shader_node_output_get(fromnode, 0); - tosock = ntree_shader_node_input_get(tonode, 1); - nodeAddLink(ntree, fromnode, fromsock, tonode, tosock); - /* Link mix input to first node. */ - fromnode = nodes_copy[id_start + 2]; - tonode = nodes_copy[id_start]; - fromsock = ntree_shader_node_output_get(fromnode, 0); - tosock = ntree_shader_node_input_get(tonode, 1); - nodeAddLink(ntree, fromnode, fromsock, tonode, tosock); - /* Link weight input to both multiply nodes. */ - fromnode = nodes_copy[id_start + 3]; - fromsock = ntree_shader_node_output_get(fromnode, 0); - tonode = nodes_copy[id_start]; - tosock = ntree_shader_node_input_get(tonode, 0); - nodeAddLink(ntree, fromnode, fromsock, tonode, tosock); - tonode = nodes_copy[id_start + 1]; - tosock = ntree_shader_node_input_get(tonode, 0); - nodeAddLink(ntree, fromnode, fromsock, tonode, tosock); - break; - } - default: - BLI_assert(0); - break; - } - } - } - /* Recreate links between copied nodes. */ - LISTBASE_FOREACH (bNode *, node, &ntree->nodes) { - if (node->tmp_flag >= 0) { - /* Naming can be confusing here. We use original nodelink name for from/to prefix. - * The final link is in reversed order. */ - int socket_index; - LISTBASE_FOREACH_INDEX (bNodeSocket *, sock, &node->inputs, socket_index) { - bNodeSocket *tosock; - bNode *tonode; - - switch (node->type) { - case SH_NODE_OUTPUT_WORLD: - case SH_NODE_OUTPUT_MATERIAL: - case SH_NODE_ADD_SHADER: { - tonode = nodes_copy[node->tmp_flag]; - tosock = ntree_shader_node_output_get(tonode, 0); - break; - } - case SH_NODE_MIX_SHADER: { - if (socket_index == 0) { - /* Mix Factor. */ - tonode = nodes_copy[node->tmp_flag + 2]; - tosock = ntree_shader_node_input_get(tonode, 1); - } - else if (socket_index == 1) { - /* Shader 1. */ - tonode = nodes_copy[node->tmp_flag + 1]; - tosock = ntree_shader_node_output_get(tonode, 0); - } - else { - /* Shader 2. */ - tonode = nodes_copy[node->tmp_flag]; - tosock = ntree_shader_node_output_get(tonode, 0); - } - break; - } - default: - BLI_assert(0); - break; - } - - if (sock->link) { - bNodeSocket *fromsock; - bNode *fromnode = sock->link->fromnode; - - switch (fromnode->type) { - case SH_NODE_ADD_SHADER: { - fromnode = nodes_copy[fromnode->tmp_flag]; - fromsock = ntree_shader_node_input_get(fromnode, 1); - if (fromsock->link) { - ntree_weight_tree_merge_weight(ntree, fromnode, fromsock, &tonode, &tosock); - } - break; - } - case SH_NODE_MIX_SHADER: { - fromnode = nodes_copy[fromnode->tmp_flag + 3]; - fromsock = ntree_shader_node_input_get(fromnode, 1); - if (fromsock->link) { - ntree_weight_tree_merge_weight(ntree, fromnode, fromsock, &tonode, &tosock); - } - break; - } - case SH_NODE_BACKGROUND: - case SH_NODE_BSDF_ANISOTROPIC: - case SH_NODE_BSDF_DIFFUSE: - case SH_NODE_BSDF_GLASS: - case SH_NODE_BSDF_GLOSSY: - case SH_NODE_BSDF_HAIR_PRINCIPLED: - case SH_NODE_BSDF_HAIR: - case SH_NODE_BSDF_PRINCIPLED: - case SH_NODE_BSDF_REFRACTION: - case SH_NODE_BSDF_TOON: - case SH_NODE_BSDF_TRANSLUCENT: - case SH_NODE_BSDF_TRANSPARENT: - case SH_NODE_BSDF_VELVET: - case SH_NODE_EEVEE_SPECULAR: - case SH_NODE_EMISSION: - case SH_NODE_HOLDOUT: - case SH_NODE_SUBSURFACE_SCATTERING: - case SH_NODE_VOLUME_ABSORPTION: - case SH_NODE_VOLUME_PRINCIPLED: - case SH_NODE_VOLUME_SCATTER: - fromsock = ntree_shader_node_find_input(fromnode, "Weight"); - if (fromsock->link) { - ntree_weight_tree_merge_weight(ntree, fromnode, fromsock, &tonode, &tosock); - } - break; - default: - fromsock = sock->link->fromsock; - break; - } - - /* Manually add the link to the socket to avoid calling ntreeUpdateTree in the loop. */ - fromsock->link = nodeAddLink(ntree, fromnode, fromsock, tonode, tosock); - BLI_assert(fromsock->link); - } - } - } - } - /* Restore displacement & thickness link. */ - if (displace_link) { - nodeAddLink( - ntree, displace_link->fromnode, displace_link->fromsock, output_node, displace_output); - } - if (thickness_link) { - nodeAddLink( - ntree, thickness_link->fromnode, thickness_link->fromsock, output_node, thickness_output); - } - ntreeUpdateTree(G.main, ntree); - - MEM_freeN(nodes_copy); -} - -/* This one needs to work on a local tree. */ -void ntreeGPUMaterialNodes(bNodeTree *localtree, GPUMaterial *mat) -{ - bNodeTreeExec *exec; - - bNode *output = ntreeShaderOutputNode(localtree, SHD_OUTPUT_EEVEE); - - ntree_shader_groups_expand_inputs(localtree); - - ntree_shader_groups_flatten(localtree); - - if (output == NULL) { - /* Search again, now including flattened nodes. */ - output = ntreeShaderOutputNode(localtree, SHD_OUTPUT_EEVEE); - } - - /* Tree is valid if it contains no undefined implicit socket type cast. */ - bool valid_tree = ntree_shader_implicit_closure_cast(localtree); - - if (valid_tree && output != NULL) { - ntree_shader_weight_tree_invert(localtree, output); - } - - /* Perform all needed modifications on the tree in order to support - * displacement/bump mapping. - */ - ntree_shader_relink_displacement(localtree, output); - - /* Duplicate bump height branches for manual derivatives. - */ - nodeChainIterBackwards(localtree, output, ntree_shader_bump_branches, localtree, 0); - LISTBASE_FOREACH (bNode *, node, &localtree->nodes) { - if (node->type == SH_NODE_OUTPUT_AOV) { - nodeChainIterBackwards(localtree, node, ntree_shader_bump_branches, localtree, 0); - } - } - - exec = ntreeShaderBeginExecTree(localtree); - ntreeExecGPUNodes(exec, mat, output); - LISTBASE_FOREACH (bNode *, node, &localtree->nodes) { - if (node->type == SH_NODE_OUTPUT_AOV) { - ntreeExecGPUNodes(exec, mat, node); - } - } - ntreeShaderEndExecTree(exec); -} - -bNodeTreeExec *ntreeShaderBeginExecTree_internal(bNodeExecContext *context, - bNodeTree *ntree, - bNodeInstanceKey parent_key) -{ - bNodeTreeExec *exec; - bNode *node; - - /* ensures only a single output node is enabled */ - ntreeSetOutput(ntree); - - /* common base initialization */ - exec = ntree_exec_begin(context, ntree, parent_key); - - /* allocate the thread stack listbase array */ - exec->threadstack = MEM_callocN(BLENDER_MAX_THREADS * sizeof(ListBase), "thread stack array"); - - for (node = exec->nodetree->nodes.first; node; node = node->next) { - node->need_exec = 1; - } - - return exec; -} - -bNodeTreeExec *ntreeShaderBeginExecTree(bNodeTree *ntree) -{ - bNodeExecContext context; - bNodeTreeExec *exec; - - /* XXX hack: prevent exec data from being generated twice. - * this should be handled by the renderer! - */ - if (ntree->execdata) { - return ntree->execdata; - } - - context.previews = ntree->previews; - - exec = ntreeShaderBeginExecTree_internal(&context, ntree, NODE_INSTANCE_KEY_BASE); - - /* XXX this should not be necessary, but is still used for cmp/sha/tex nodes, - * which only store the ntree pointer. Should be fixed at some point! - */ - ntree->execdata = exec; - - return exec; -} - -void ntreeShaderEndExecTree_internal(bNodeTreeExec *exec) -{ - bNodeThreadStack *nts; - int a; - - if (exec->threadstack) { - for (a = 0; a < BLENDER_MAX_THREADS; a++) { - for (nts = exec->threadstack[a].first; nts; nts = nts->next) { - if (nts->stack) { - MEM_freeN(nts->stack); - } - } - BLI_freelistN(&exec->threadstack[a]); - } - - MEM_freeN(exec->threadstack); - exec->threadstack = NULL; - } - - ntree_exec_end(exec); -} - -void ntreeShaderEndExecTree(bNodeTreeExec *exec) -{ - if (exec) { - /* exec may get freed, so assign ntree */ - bNodeTree *ntree = exec->nodetree; - ntreeShaderEndExecTree_internal(exec); - - /* XXX clear nodetree backpointer to exec data, same problem as noted in ntreeBeginExecTree */ - ntree->execdata = NULL; - } -} |