B3SparseCollection.h [plain text]
#ifndef B3SparseCollection_h
#define B3SparseCollection_h
#if ENABLE(B3_JIT)
#include <wtf/StdLibExtras.h>
#include <wtf/Vector.h>
namespace JSC { namespace B3 {
template<typename T>
class SparseCollection {
typedef Vector<std::unique_ptr<T>> VectorType;
public:
SparseCollection()
{
}
T* add(std::unique_ptr<T> value)
{
T* result = value.get();
size_t index;
if (m_indexFreeList.isEmpty()) {
index = m_vector.size();
m_vector.append(nullptr);
} else
index = m_indexFreeList.takeLast();
value->m_index = index;
m_vector[index] = WTFMove(value);
return result;
}
template<typename... Arguments>
T* addNew(Arguments&&... arguments)
{
return add(std::unique_ptr<T>(new T(std::forward<Arguments>(arguments)...)));
}
void remove(T* value)
{
RELEASE_ASSERT(m_vector[value->m_index].get() == value);
m_indexFreeList.append(value->m_index);
m_vector[value->m_index] = nullptr;
}
unsigned size() const { return m_vector.size(); }
bool isEmpty() const { return m_vector.isEmpty(); }
T* at(unsigned index) const { return m_vector[index].get(); }
T* operator[](unsigned index) const { return at(index); }
class iterator {
public:
iterator()
: m_collection(nullptr)
, m_index(0)
{
}
iterator(const SparseCollection& collection, unsigned index)
: m_collection(&collection)
, m_index(findNext(index))
{
}
T* operator*()
{
return m_collection->at(m_index);
}
iterator& operator++()
{
m_index = findNext(m_index + 1);
return *this;
}
bool operator==(const iterator& other) const
{
ASSERT(m_collection == other.m_collection);
return m_index == other.m_index;
}
bool operator!=(const iterator& other) const
{
return !(*this == other);
}
private:
unsigned findNext(unsigned index)
{
while (index < m_collection->size() && !m_collection->at(index))
index++;
return index;
}
const SparseCollection* m_collection;
unsigned m_index;
};
iterator begin() const { return iterator(*this, 0); }
iterator end() const { return iterator(*this, size()); }
private:
Vector<std::unique_ptr<T>> m_vector;
Vector<size_t> m_indexFreeList;
};
} }
#endif // ENABLE(B3_JIT)
#endif // B3SparseCollection_h