blender
/
blender
136
404
Fork 586
Code Issues 6.2k Pull Requests 753 Projects 19 Wiki Activity

Functions: add utility to find dependencies of input sockets

This commit is contained in:
Jacques Lucke 2020-07-10 14:22:35 +02:00
parent 60133ff98d
commit c806db6313
2 changed files with 35 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
source/blender/functions/FN_multi_function_network.hh
View File

@ -229,6 +229,10 @@ class MFNetwork : NonCopyable, NonMovable {
MFSocket *socket_or_null_by_id(uint id);
const MFSocket *socket_or_null_by_id(uint id) const;
void find_dependencies(Span<const MFInputSocket *> sockets,
VectorSet<const MFOutputSocket *> &r_dummy_sockets,
VectorSet<const MFInputSocket *> &r_unlinked_inputs) const;
std::string to_dot(Span<const MFNode *> marked_nodes = {}) const;
};

31
source/blender/functions/intern/multi_function_network.cc
View File

@ -15,6 +15,8 @@
*/
#include "BLI_dot_export.hh"
#include "BLI_stack.hh"
#include "FN_multi_function_network.hh"
namespace blender::fn {
@ -237,6 +239,35 @@ void MFNetwork::remove(Span<MFNode *> nodes)
}
}
void MFNetwork::find_dependencies(Span<const MFInputSocket *> sockets,
VectorSet<const MFOutputSocket *> &r_dummy_sockets,
VectorSet<const MFInputSocket *> &r_unlinked_inputs) const
{
Set<const MFNode *> visited_nodes;
Stack<const MFInputSocket *> sockets_to_check;
sockets_to_check.push_multiple(sockets);
while (!sockets_to_check.is_empty()) {
const MFInputSocket &socket = *sockets_to_check.pop();
const MFOutputSocket *origin_socket = socket.origin();
if (origin_socket == nullptr) {
r_unlinked_inputs.add(&socket);
continue;
}
const MFNode &origin_node = origin_socket->node();
if (origin_node.is_dummy()) {
r_dummy_sockets.add(origin_socket);
continue;
}
if (visited_nodes.add(&origin_node)) {
sockets_to_check.push_multiple(origin_node.inputs());
}
}
}
std::string MFNetwork::to_dot(Span<const MFNode *> marked_nodes) const
{
dot::DirectedGraph digraph;