blob: b056dbafad663c15f52f40fe82eb5fa2bd5a12a2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
#ifndef __FN_MULTI_FUNCTION_DEPENDENCIES_H__
#define __FN_MULTI_FUNCTION_DEPENDENCIES_H__
#include "BLI_set.h"
#include "DNA_image_types.h"
#include "DNA_object_types.h"
#include "FN_node_tree.h"
namespace FN {
using BLI::Set;
inline Set<Object *> get_objects_used_by_sockets(const FunctionTree &function_tree)
{
Set<Object *> objects;
for (const FSocket *fsocket : function_tree.all_sockets()) {
if (fsocket->idname() == "fn_ObjectSocket") {
Object *object = (Object *)RNA_pointer_get(fsocket->rna(), "value").data;
if (object != nullptr) {
objects.add(object);
}
}
}
for (const FGroupInput *group_input : function_tree.all_group_inputs()) {
if (group_input->vsocket().idname() == "fn_ObjectSocket") {
Object *object = (Object *)RNA_pointer_get(group_input->vsocket().rna(), "value").data;
if (object != nullptr) {
objects.add(object);
}
}
}
return objects;
}
inline Set<Image *> get_images_used_by_sockets(const FunctionTree &function_tree)
{
Set<Image *> images;
for (const FSocket *fsocket : function_tree.all_sockets()) {
if (fsocket->idname() == "fn_ImageSocket") {
Image *image = (Image *)RNA_pointer_get(fsocket->rna(), "value").data;
if (image != nullptr) {
images.add(image);
}
}
}
for (const FGroupInput *group_input : function_tree.all_group_inputs()) {
if (group_input->vsocket().idname() == "fn_ImageSocket") {
Image *image = (Image *)RNA_pointer_get(group_input->vsocket().rna(), "value").data;
if (image != nullptr) {
images.add(image);
}
}
}
return images;
}
inline void add_ids_used_by_nodes(IDHandleLookup &id_handle_lookup,
const FunctionTree &function_tree)
{
for (Object *object : get_objects_used_by_sockets(function_tree)) {
id_handle_lookup.add(object->id);
}
for (Image *image : get_images_used_by_sockets(function_tree)) {
id_handle_lookup.add(image->id);
}
}
} // namespace FN
#endif /* __FN_MULTI_FUNCTION_DEPENDENCIES_H__ */
|