LineBreakIteratorPoolICU.h [plain text]
#pragma once
#include "TextBreakIterator.h"
#include <wtf/HashMap.h>
#include <wtf/NeverDestroyed.h>
#include <wtf/ThreadSpecific.h>
#include <wtf/text/AtomicString.h>
namespace WTF {
class LineBreakIteratorPool {
WTF_MAKE_NONCOPYABLE(LineBreakIteratorPool);
public:
LineBreakIteratorPool() = default;
static LineBreakIteratorPool& sharedPool()
{
static NeverDestroyed<WTF::ThreadSpecific<LineBreakIteratorPool>> pool;
return *pool.get();
}
static AtomicString makeLocaleWithBreakKeyword(const AtomicString& locale, LineBreakIteratorMode mode)
{
switch (mode) {
case LineBreakIteratorMode::Default:
return locale;
case LineBreakIteratorMode::Loose:
return makeString(locale, "@break=loose");
case LineBreakIteratorMode::Normal:
return makeString(locale, "@break=normal");
case LineBreakIteratorMode::Strict:
return makeString(locale, "@break=strict");
}
ASSERT_NOT_REACHED();
return locale;
}
UBreakIterator* take(const AtomicString& locale, LineBreakIteratorMode mode, bool isCJK)
{
auto localeWithOptionalBreakKeyword = makeLocaleWithBreakKeyword(locale, mode);
UBreakIterator* iterator = nullptr;
for (size_t i = 0; i < m_pool.size(); ++i) {
if (m_pool[i].first == localeWithOptionalBreakKeyword) {
iterator = m_pool[i].second;
m_pool.remove(i);
break;
}
}
if (!iterator) {
iterator = openLineBreakIterator(localeWithOptionalBreakKeyword, mode, isCJK);
if (!iterator)
return nullptr;
}
ASSERT(!m_vendedIterators.contains(iterator));
m_vendedIterators.add(iterator, localeWithOptionalBreakKeyword);
return iterator;
}
void put(UBreakIterator* iterator)
{
ASSERT(m_vendedIterators.contains(iterator));
if (m_pool.size() == capacity) {
closeLineBreakIterator(m_pool[0].second);
m_pool.remove(0);
}
m_pool.uncheckedAppend({ m_vendedIterators.take(iterator), iterator });
}
private:
static constexpr size_t capacity = 4;
Vector<std::pair<AtomicString, UBreakIterator*>, capacity> m_pool;
HashMap<UBreakIterator*, AtomicString> m_vendedIterators;
friend WTF::ThreadSpecific<LineBreakIteratorPool>::operator LineBreakIteratorPool*();
};
}