79 lines
2.3 KiB
C++
79 lines
2.3 KiB
C++
#include "interned_string.hh"
|
|
#include "debug.hh"
|
|
|
|
namespace Kakoune
|
|
{
|
|
|
|
void StringRegistry::debug_stats() const
|
|
{
|
|
write_debug("Interned Strings stats:");
|
|
write_debug(" slots: " + to_string(m_storage.size()) + " allocated, " + to_string(m_free_slots.size()) + " free");
|
|
size_t total_refcount = 0;
|
|
size_t total_size = 0;
|
|
size_t count = 0;
|
|
for (auto& st : m_storage)
|
|
{
|
|
if (st.refcount == 0)
|
|
continue;
|
|
total_refcount += st.refcount;
|
|
total_size += st.data.size();
|
|
++count;
|
|
}
|
|
write_debug(" data size: " + to_string(total_size) + ", mean: " + to_string((float)total_size/count));
|
|
write_debug(" refcounts: " + to_string(total_refcount) + ", mean: " + to_string((float)total_refcount/count));
|
|
}
|
|
|
|
InternedString StringRegistry::acquire(StringView str)
|
|
{
|
|
auto it = m_slot_map.find(str);
|
|
if (it == m_slot_map.end())
|
|
{
|
|
Slot slot;
|
|
if (not m_free_slots.empty())
|
|
{
|
|
slot = m_free_slots.back();
|
|
m_free_slots.pop_back();
|
|
kak_assert(m_storage[slot].refcount == 0);
|
|
m_storage[slot] = DataAndRefCount{{str.begin(), str.end()}, 1};
|
|
}
|
|
else
|
|
{
|
|
slot = m_storage.size();
|
|
m_storage.push_back({{str.begin(), str.end()}, 1});
|
|
}
|
|
// Create a new string view that point to the storage data
|
|
StringView storage_view{m_storage[slot].data.data(), (int)m_storage[slot].data.size()};
|
|
m_slot_map[storage_view] = slot;
|
|
|
|
return InternedString{storage_view, slot};
|
|
}
|
|
|
|
Slot slot = it->second;
|
|
auto& data = m_storage[slot];
|
|
++data.refcount;
|
|
return {{data.data.data(), (int)data.data.size()}, slot};
|
|
}
|
|
|
|
void StringRegistry::acquire(Slot slot)
|
|
{
|
|
kak_assert(slot < m_storage.size());
|
|
kak_assert(m_storage[slot].refcount > 0);
|
|
++m_storage[slot].refcount;
|
|
}
|
|
|
|
void StringRegistry::release(Slot slot) noexcept
|
|
{
|
|
kak_assert(m_storage[slot].refcount > 0);
|
|
if (--m_storage[slot].refcount == 0)
|
|
{
|
|
m_free_slots.push_back(slot);
|
|
auto& data = m_storage[slot].data;
|
|
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 = Vector<char, MemoryDomain::InternedString>{};
|
|
}
|
|
}
|
|
|
|
}
|