Add utility function to copy default values from one tree to another
authorSergey Sharybin <sergey.vfx@gmail.com>
Tue, 29 Aug 2017 10:20:55 +0000 (12:20 +0200)
committerSergey Sharybin <sergey.vfx@gmail.com>
Tue, 29 Aug 2017 10:55:03 +0000 (12:55 +0200)
The trees should match in nodes layout completely.

source/blender/blenkernel/BKE_node.h
source/blender/blenkernel/intern/node.c
source/blender/nodes/NOD_socket.h
source/blender/nodes/intern/node_socket.c

index 5e50d37e1451fd7d151798332c9b610fe6bf85aa..5bc47f62b6401d03ddac6b30d3668f988de1e547 100644 (file)
@@ -1050,4 +1050,11 @@ int ntreeTexExecTree(struct bNodeTree *ntree, struct TexResult *target,
 void init_nodesystem(void);
 void free_nodesystem(void);
 
+/* -------------------------------------------------------------------- */
+/* evaluation support, */
+
+void BKE_nodetree_copy_default_values(struct bNodeTree *ntree_dst,
+                                      const struct bNodeTree *ntree_src);
+
+
 #endif  /* __BKE_NODE_H__ */
index f0b5fc9c610edde5c74dbf9304dde649000da80a..1c24d72759eb79e72e6290c2d606e9676c08e38d 100644 (file)
@@ -3804,3 +3804,33 @@ void BKE_nodetree_remove_layer_n(bNodeTree *ntree, Scene *scene, const int layer
                }
        }
 }
+
+static void node_copy_default_values_list(ListBase *sockets_dst,
+                                          const ListBase *sockets_src)
+{
+       bNodeSocket *sock_dst = sockets_dst->first;
+       const bNodeSocket *sock_src = sockets_src->first;
+       while (sock_dst != NULL) {
+               node_socket_copy_default_value(sock_dst, sock_src);
+               sock_dst = sock_dst->next;
+               sock_src = sock_src->next;
+       }
+}
+
+static void node_copy_default_values(bNode *node_dst, const bNode *node_src)
+{
+       node_copy_default_values_list(&node_dst->inputs, &node_src->inputs);
+       node_copy_default_values_list(&node_dst->outputs, &node_src->outputs);
+}
+
+void BKE_nodetree_copy_default_values(bNodeTree *ntree_dst,
+                                      const bNodeTree *ntree_src)
+{
+       bNode *node_dst = ntree_dst->nodes.first;
+       const bNode *node_src = ntree_src->nodes.first;
+       while (node_dst != NULL) {
+               node_copy_default_values(node_dst, node_src);
+               node_dst = node_dst->next;
+               node_src = node_src->next;
+       }
+}
index 5c8875593e7d22b4d867a128e56e81cd0b886b7c..e311adc45eb5bb06163ae9e1fa27f33e87edc9b3 100644 (file)
@@ -49,7 +49,7 @@ struct bNodeSocket *node_add_socket_from_template(struct bNodeTree *ntree, struc
 void node_verify_socket_templates(struct bNodeTree *ntree, struct bNode *node);
 
 void node_socket_init_default_value(struct bNodeSocket *sock);
-void node_socket_copy_default_value(struct bNodeSocket *to, struct bNodeSocket *from);
+void node_socket_copy_default_value(struct bNodeSocket *to, const struct bNodeSocket *from);
 void register_standard_node_socket_types(void);
 
 #endif
index 01c642b063de223e7a6cd882bb02480f48a4b2f3..ddd25b237a0915fff7a8d637466c57ffb12e6791 100644 (file)
@@ -267,7 +267,7 @@ void node_socket_init_default_value(bNodeSocket *sock)
        }
 }
 
-void node_socket_copy_default_value(bNodeSocket *to, bNodeSocket *from)
+void node_socket_copy_default_value(bNodeSocket *to, const bNodeSocket *from)
 {
        /* sanity check */
        if (to->type != from->type)