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
|
#include "LayerUsageBreakdown.h"
#include "iscenegraph.h"
#include "ientity.h"
#include "iselection.h"
#include "scenelib.h"
namespace scene
{
namespace
{
void addNodeMapping(LayerUsageBreakdown& bd, const scene::INodePtr& node)
{
const auto& layers = node->getLayers();
for (int layerId : layers)
{
assert(layerId >= 0); // we assume positive layer IDs here
// Increase the counter of the corresponding layer slot by one
bd[layerId]++;
}
}
}
LayerUsageBreakdown LayerUsageBreakdown::CreateFromScene(bool includeHidden)
{
LayerUsageBreakdown bd;
if (!GlobalMapModule().getRoot())
{
return bd;
}
InitialiseVector(bd);
GlobalSceneGraph().foreachNode([&](const scene::INodePtr& node)
{
// Filter out any hidden nodes, unless we want them
if (!includeHidden && !node->visible()) return false;
// Consider only entities and primitives
if (!Node_isPrimitive(node) && !Node_isEntity(node)) return true;
addNodeMapping(bd, node);
return true;
});
return bd;
}
LayerUsageBreakdown LayerUsageBreakdown::CreateFromSelection()
{
LayerUsageBreakdown bd;
if (!GlobalMapModule().getRoot())
{
return bd;
}
InitialiseVector(bd);
GlobalSelectionSystem().foreachSelected([&](const scene::INodePtr& node)
{
addNodeMapping(bd, node);
});
return bd;
}
void LayerUsageBreakdown::InitialiseVector(LayerUsageBreakdown& bd)
{
// Start with a reasonably large memory block
bd.reserve(64);
// We resize the vector to 0 length, such that each resize() call
// below will fill the vector with zeros
bd.resize(0, 0);
GlobalMapModule().getRoot()->getLayerManager().foreachLayer([&](int layerId, const std::string& layerName)
{
if (layerId >= static_cast<int>(bd.size()))
{
bd.resize(layerId+1, 0);
}
});
}
}
|
