#include "config.h"
#include "WeakMapImpl.h"
#include "IsoCellSetInlines.h"
#include "JSCInlines.h"
#include "WeakMapImplInlines.h"
namespace JSC {
template <typename WeakMapBucket>
void WeakMapImpl<WeakMapBucket>::destroy(JSCell* cell)
{
static_cast<WeakMapImpl*>(cell)->~WeakMapImpl();
}
template <typename WeakMapBucket>
void WeakMapImpl<WeakMapBucket>::visitChildren(JSCell* cell, SlotVisitor& visitor)
{
WeakMapImpl* thisObject = jsCast<WeakMapImpl*>(cell);
ASSERT_GC_OBJECT_INHERITS(thisObject, info());
Base::visitChildren(thisObject, visitor);
visitor.reportExtraMemoryVisited(thisObject->m_capacity * sizeof(WeakMapBucket));
}
template <typename WeakMapBucket>
size_t WeakMapImpl<WeakMapBucket>::estimatedSize(JSCell* cell, VM& vm)
{
auto* thisObject = static_cast<WeakMapImpl*>(cell);
return Base::estimatedSize(thisObject, vm) + (sizeof(WeakMapImpl) - sizeof(Base)) + thisObject->m_capacity * sizeof(WeakMapBucket);
}
template <>
void WeakMapImpl<WeakMapBucket<WeakMapBucketDataKey>>::visitOutputConstraints(JSCell*, SlotVisitor&)
{
}
template <>
void WeakMapImpl<WeakMapBucket<WeakMapBucketDataKeyValue>>::visitOutputConstraints(JSCell* cell, SlotVisitor& visitor)
{
auto* thisObject = jsCast<WeakMapImpl*>(cell);
auto locker = holdLock(thisObject->cellLock());
auto* buffer = thisObject->buffer();
for (uint32_t index = 0; index < thisObject->m_capacity; ++index) {
auto* bucket = buffer + index;
if (bucket->isEmpty() || bucket->isDeleted())
continue;
if (!Heap::isMarked(bucket->key()))
continue;
bucket->visitAggregate(visitor);
}
}
template <typename WeakMapBucket>
template<typename Appender>
void WeakMapImpl<WeakMapBucket>::takeSnapshotInternal(unsigned limit, Appender appender)
{
DisallowGC disallowGC;
unsigned fetched = 0;
forEach([&] (JSObject* key, JSValue value) {
appender(key, value);
++fetched;
if (limit && fetched >= limit)
return IterationState::Stop;
return IterationState::Continue;
});
}
template <>
void WeakMapImpl<WeakMapBucket<WeakMapBucketDataKey>>::takeSnapshot(MarkedArgumentBuffer& buffer, unsigned limit)
{
takeSnapshotInternal(limit, [&] (JSObject* key, JSValue) {
buffer.append(key);
});
}
template <>
void WeakMapImpl<WeakMapBucket<WeakMapBucketDataKeyValue>>::takeSnapshot(MarkedArgumentBuffer& buffer, unsigned limit)
{
takeSnapshotInternal(limit, [&] (JSObject* key, JSValue value) {
buffer.append(key);
buffer.append(value);
});
}
template class WeakMapImpl<WeakMapBucket<WeakMapBucketDataKeyValue>>;
template class WeakMapImpl<WeakMapBucket<WeakMapBucketDataKey>>;
}