#pragma once
#include "IterationKind.h"
#include "JSObject.h"
#include "JSSet.h"
namespace JSC {
class JSSetIterator : public JSCell {
typedef HashMapBucket<HashMapBucketDataKey> HashMapBucketType;
public:
using Base = JSCell;
DECLARE_EXPORT_INFO;
static Structure* createStructure(VM& vm, JSGlobalObject* globalObject, JSValue prototype)
{
return Structure::create(vm, globalObject, prototype, TypeInfo(ObjectType, StructureFlags), info());
}
static JSSetIterator* create(VM& vm, Structure* structure, JSSet* iteratedObject, IterationKind kind)
{
JSSetIterator* instance = new (NotNull, allocateCell<JSSetIterator>(vm.heap)) JSSetIterator(vm, structure, iteratedObject, kind);
instance->finishCreation(vm, iteratedObject);
return instance;
}
ALWAYS_INLINE HashMapBucketType* advanceIter(ExecState* exec)
{
HashMapBucketType* prev = m_iter.get();
if (!prev)
return nullptr;
VM& vm = exec->vm();
HashMapBucketType* bucket = m_iter->next();
while (bucket && bucket->deleted())
bucket = bucket->next();
if (!bucket) {
setIterator(vm, nullptr);
return nullptr;
}
setIterator(vm, bucket); return bucket;
}
bool next(ExecState* exec, JSValue& value)
{
HashMapBucketType* bucket = advanceIter(exec);
if (!bucket)
return false;
if (m_kind == IterateValue || m_kind == IterateKey)
value = bucket->key();
else
value = createPair(exec, bucket->key(), bucket->key());
return true;
}
IterationKind kind() const { return m_kind; }
JSValue iteratedValue() const { return m_set.get(); }
private:
JSSetIterator(VM& vm, Structure* structure, JSSet*, IterationKind kind)
: Base(vm, structure)
, m_kind(kind)
{
}
void setIterator(VM& vm, HashMapBucketType* bucket)
{
m_iter.setMayBeNull(vm, this, bucket);
}
JS_EXPORT_PRIVATE void finishCreation(VM&, JSSet*);
JS_EXPORT_PRIVATE JSValue createPair(CallFrame*, JSValue, JSValue);
static void visitChildren(JSCell*, SlotVisitor&);
WriteBarrier<JSSet> m_set;
WriteBarrier<HashMapBucketType> m_iter;
IterationKind m_kind;
};
STATIC_ASSERT_IS_TRIVIALLY_DESTRUCTIBLE(JSSetIterator);
}