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
|
#include "slang-string-slice-pool.h"
namespace Slang {
/* static */ const StringSlicePool::Handle StringSlicePool::kNullHandle;
/* static */ const StringSlicePool::Handle StringSlicePool::kEmptyHandle;
/* static */const Index StringSlicePool::kDefaultHandlesCount;
StringSlicePool::StringSlicePool(Style style) :
m_style(style),
m_arena(1024)
{
clear();
}
void StringSlicePool::clear()
{
m_map.Clear();
switch (m_style)
{
case Style::Default:
{
// Add the defaults
m_slices.setCount(2);
m_slices[0] = UnownedStringSlice((const char*)nullptr, (const char*)nullptr);
m_slices[1] = UnownedStringSlice::fromLiteral("");
// Add the empty entry
m_map.Add(m_slices[1], kEmptyHandle);
break;
}
case Style::Empty:
{
// There are no defaults
m_slices.clear();
break;
}
}
}
void StringSlicePool::swapWith(ThisType& rhs)
{
Swap(m_style, rhs.m_style);
m_slices.swapWith(rhs.m_slices);
m_map.swapWith(rhs.m_map);
m_arena.swapWith(rhs.m_arena);
}
StringSlicePool::Handle StringSlicePool::add(const Slice& slice)
{
const Handle* handlePtr = m_map.TryGetValue(slice);
if (handlePtr)
{
return *handlePtr;
}
// Create a scoped copy
UnownedStringSlice scopePath(m_arena.allocateString(slice.begin(), slice.getLength()), slice.getLength());
const auto index = m_slices.getCount();
m_slices.add(scopePath);
m_map.Add(scopePath, Handle(index));
return Handle(index);
}
bool StringSlicePool::findOrAdd(const Slice& slice, Handle& outHandle)
{
const Handle* handlePtr = m_map.TryGetValue(slice);
if (handlePtr)
{
outHandle = *handlePtr;
return true;
}
// Need to add.
// Make a copy stored in the arena
UnownedStringSlice scopeSlice(m_arena.allocateString(slice.begin(), slice.getLength()), slice.getLength());
// Add using the arenas copy
Handle newHandle = Handle(m_slices.getCount());
m_map.Add(scopeSlice, newHandle);
// Add to slices list
m_slices.add(scopeSlice);
outHandle = newHandle;
return false;
}
StringSlicePool::Handle StringSlicePool::add(StringRepresentation* stringRep)
{
if (stringRep == nullptr && m_style == Style::Default)
{
return kNullHandle;
}
return add(StringRepresentation::asSlice(stringRep));
}
StringSlicePool::Handle StringSlicePool::add(const char* chars)
{
switch (m_style)
{
case Style::Default:
{
if (!chars)
{
return kNullHandle;
}
if (chars[0] == 0)
{
return kEmptyHandle;
}
break;
}
case Style::Empty:
{
if (chars == nullptr)
{
SLANG_ASSERT(!"Empty style doesn't support nullptr");
// Return an invalid handle
return Handle(~HandleIntegral(0));
}
}
}
return add(UnownedStringSlice(chars));
}
Index StringSlicePool::findIndex(const Slice& slice) const
{
const Handle* handlePtr = m_map.TryGetValue(slice);
return handlePtr ? Index(*handlePtr) : -1;
}
ConstArrayView<UnownedStringSlice> StringSlicePool::getAdded() const
{
const Index firstIndex = getFirstAddedIndex();
return makeConstArrayView(m_slices.getBuffer() + firstIndex, m_slices.getCount() - firstIndex);
}
} // namespace Slang
|
