blob: 24187ba5c3a800e5321d87bab070e1a5804e56a1 (
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
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
|
#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;
}
}
}
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)
{
Handle newHandle = Handle(m_slices.getCount());
const Handle* handlePtr = m_map.TryGetValueOrAdd(slice, newHandle);
if (handlePtr)
{
outHandle = *handlePtr;
return true;
}
// Need to add
UnownedStringSlice scopeSlice(m_arena.allocateString(slice.begin(), slice.getLength()), slice.getLength());
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
|
