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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
#ifndef __FN_VTREE_MULTI_FUNCTION_NETWORK_H__
#define __FN_VTREE_MULTI_FUNCTION_NETWORK_H__
#include "FN_node_tree.h"
#include "BLI_index_to_ref_map.h"
#include "BLI_multi_map.h"
#include "FN_multi_function_network.h"
namespace FN {
using BLI::IndexToRefMap;
using BLI::MultiMap;
class DummySocketMap {
private:
const FunctionTree *m_function_tree;
const MFNetwork *m_network;
IndexToRefMap<const MFSocket> m_dummy_socket_by_fsocket_id;
IndexToRefMap<const FSocket> m_fsocket_by_dummy_socket_id;
public:
DummySocketMap(const FunctionTree &function_tree,
const MFNetwork &network,
IndexToRefMap<const MFSocket> dummy_socket_by_fsocket_id,
IndexToRefMap<const FSocket> fsocket_by_dummy_socket_id)
: m_function_tree(&function_tree),
m_network(&network),
m_dummy_socket_by_fsocket_id(std::move(dummy_socket_by_fsocket_id)),
m_fsocket_by_dummy_socket_id(std::move(fsocket_by_dummy_socket_id))
{
}
bool is_mapped(const FSocket &fsocket) const
{
return m_dummy_socket_by_fsocket_id.contains(fsocket.id());
}
bool is_mapped(const MFSocket &socket) const
{
return m_fsocket_by_dummy_socket_id.contains(socket.id());
}
const MFInputSocket &lookup_singly_mapped_input_socket(const FInputSocket &fsocket) const
{
return m_dummy_socket_by_fsocket_id.lookup(fsocket.id()).as_input();
}
const MFOutputSocket &lookup_socket(const FOutputSocket &fsocket) const
{
return m_dummy_socket_by_fsocket_id.lookup(fsocket.id()).as_output();
}
const FInputSocket &lookup_fsocket(const MFInputSocket &socket) const
{
BLI_assert(socket.node().is_dummy());
return m_fsocket_by_dummy_socket_id.lookup(socket.id()).as_input();
}
const FOutputSocket &lookup_fsocket(const MFOutputSocket &socket) const
{
BLI_assert(socket.node().is_dummy());
return m_fsocket_by_dummy_socket_id.lookup(socket.id()).as_output();
}
};
class FunctionTreeMFNetwork {
private:
const FunctionTree &m_function_tree;
std::unique_ptr<MFNetwork> m_network;
DummySocketMap m_socket_map;
public:
FunctionTreeMFNetwork(const FunctionTree &function_tree,
std::unique_ptr<MFNetwork> network,
DummySocketMap socket_map)
: m_function_tree(function_tree),
m_network(std::move(network)),
m_socket_map(std::move(socket_map))
{
}
const FunctionTree &function_tree() const
{
return m_function_tree;
}
const MFNetwork &network() const
{
return *m_network;
}
bool is_mapped(const FSocket &fsocket) const
{
return m_socket_map.is_mapped(fsocket);
}
bool is_mapped(const MFSocket &socket) const
{
return m_socket_map.is_mapped(socket);
}
const MFInputSocket &lookup_dummy_socket(const FInputSocket &fsocket) const
{
const MFInputSocket &socket = m_socket_map.lookup_singly_mapped_input_socket(fsocket);
BLI_assert(socket.node().is_dummy());
return socket;
}
const MFOutputSocket &lookup_dummy_socket(const FOutputSocket &fsocket) const
{
const MFOutputSocket &socket = this->lookup_socket(fsocket);
BLI_assert(socket.node().is_dummy());
return socket;
}
const MFOutputSocket &lookup_socket(const FOutputSocket &fsocket) const
{
return m_socket_map.lookup_socket(fsocket);
}
const FInputSocket &lookup_fsocket(const MFInputSocket &socket) const
{
return m_socket_map.lookup_fsocket(socket);
}
const FOutputSocket &lookup_fsocket(const MFOutputSocket &socket) const
{
return m_socket_map.lookup_fsocket(socket);
}
void lookup_dummy_sockets(ArrayRef<const FOutputSocket *> fsockets,
MutableArrayRef<const MFOutputSocket *> r_result) const
{
BLI::assert_same_size(fsockets, r_result);
for (uint i : fsockets.index_range()) {
r_result[i] = &this->lookup_socket(*fsockets[i]);
}
}
void lookup_dummy_sockets(ArrayRef<const FInputSocket *> fsockets,
MutableArrayRef<const MFInputSocket *> r_result) const
{
BLI::assert_same_size(fsockets, r_result);
for (uint i : fsockets.index_range()) {
r_result[i] = &this->lookup_dummy_socket(*fsockets[i]);
}
}
};
} // namespace FN
#endif /* __FN_VTREE_MULTI_FUNCTION_NETWORK_H__ */
|
