59 lines
1.6 KiB
C++
59 lines
1.6 KiB
C++
#include "interned_string.hh"
|
|
|
|
namespace Kakoune
|
|
{
|
|
|
|
InternedString StringRegistry::acquire(StringView str)
|
|
{
|
|
auto it = m_slot_map.find(str);
|
|
if (it == m_slot_map.end())
|
|
{
|
|
size_t slot;
|
|
if (not m_free_slots.empty())
|
|
{
|
|
slot = m_free_slots.back();
|
|
m_free_slots.pop_back();
|
|
kak_assert(m_storage[slot].second == 0);
|
|
m_storage[slot] = DataAndRefCount({str.begin(), str.end()}, 1);
|
|
}
|
|
else
|
|
{
|
|
slot = m_storage.size();
|
|
m_storage.push_back(DataAndRefCount({str.begin(), str.end()}, 1));
|
|
}
|
|
// Create a new string view that point to the storage data
|
|
StringView storage_view{m_storage[slot].first.data(), (int)m_storage[slot].first.size()};
|
|
m_slot_map[storage_view] = slot;
|
|
|
|
return InternedString{storage_view, slot};
|
|
}
|
|
|
|
size_t slot = it->second;
|
|
auto& data = m_storage[slot];
|
|
++data.second;
|
|
return {{data.first.data(), (int)data.first.size()}, slot};
|
|
}
|
|
|
|
void StringRegistry::acquire(size_t slot)
|
|
{
|
|
kak_assert(slot < m_storage.size());
|
|
kak_assert(m_storage[slot].second > 0);
|
|
++m_storage[slot].second;
|
|
}
|
|
|
|
void StringRegistry::release(size_t slot) noexcept
|
|
{
|
|
kak_assert(m_storage[slot].second > 0);
|
|
if (--m_storage[slot].second == 0)
|
|
{
|
|
m_free_slots.push_back(slot);
|
|
std::vector<char>& data = m_storage[slot].first;
|
|
auto it = m_slot_map.find(StringView{data.data(), (int)data.size()});
|
|
kak_assert(it != m_slot_map.end());
|
|
m_slot_map.erase(it);
|
|
data = std::vector<char>{};
|
|
}
|
|
}
|
|
|
|
}
|