/
SelectionGroup.h
120 lines (89 loc) · 2.45 KB
/
SelectionGroup.h
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
#pragma once
#include "iselectiongroup.h"
#include "util/ScopedBoolLock.h"
namespace selection
{
// Represents a named group of selectable items
class SelectionGroup :
public ISelectionGroup
{
private:
std::size_t _id;
std::string _name;
// The contained nodes of this group
std::set<scene::INodeWeakPtr, std::owner_less<scene::INodeWeakPtr> > _nodes;
// To avoid entering feedback loops during group updates
bool _selectionLock;
public:
SelectionGroup(std::size_t id) :
_id(id),
_selectionLock(false)
{}
std::size_t getId() const override
{
return _id;
}
const std::string& getName() const override
{
return _name;
}
void setName(const std::string& name) override
{
_name = name;
}
void addNode(const scene::INodePtr& node) override
{
std::shared_ptr<IGroupSelectable> selectable = std::dynamic_pointer_cast<IGroupSelectable>(node);
if (!selectable) return;
selectable->addToGroup(_id);
_nodes.insert(scene::INodeWeakPtr(node));
}
void removeNode(const scene::INodePtr& node) override
{
std::shared_ptr<IGroupSelectable> selectable = std::dynamic_pointer_cast<IGroupSelectable>(node);
assert(selectable);
selectable->removeFromGroup(_id);
_nodes.erase(scene::INodeWeakPtr(node));
}
void removeAllNodes()
{
foreachNode([this](const scene::INodePtr& node)
{
std::shared_ptr<IGroupSelectable> selectable = std::dynamic_pointer_cast<IGroupSelectable>(node);
assert(selectable);
selectable->removeFromGroup(_id);
});
}
std::size_t size() const override
{
return _nodes.size();
}
void setSelected(bool selected) override
{
// In debug build's, I'd like to see the feedback loops,
// so fire the debugger if we're re-entering the setSelected loop
assert(!_selectionLock);
if (_selectionLock) return; // already updating this group
util::ScopedBoolLock lock(_selectionLock);
foreachNode([&](const scene::INodePtr& node)
{
std::shared_ptr<IGroupSelectable> selectable = std::dynamic_pointer_cast<IGroupSelectable>(node);
assert(selectable);
// Set the node status, but don't do a group update (we're already here)
selectable->setSelected(selected, false);
});
}
void foreachNode(const std::function<void(const scene::INodePtr&)>& functor) override
{
for (const scene::INodeWeakPtr& node : _nodes)
{
scene::INodePtr locked = node.lock();
if (locked)
{
functor(locked);
}
}
}
};
typedef std::shared_ptr<SelectionGroup> SelectionGroupPtr;
} // namespace