]>
Commit | Line | Data |
---|---|---|
1 | /* | |
2 | * Copyright (C) 2013 Apple, Inc. All rights reserved. | |
3 | * | |
4 | * Redistribution and use in source and binary forms, with or without | |
5 | * modification, are permitted provided that the following conditions | |
6 | * are met: | |
7 | * 1. Redistributions of source code must retain the above copyright | |
8 | * notice, this list of conditions and the following disclaimer. | |
9 | * 2. Redistributions in binary form must reproduce the above copyright | |
10 | * notice, this list of conditions and the following disclaimer in the | |
11 | * documentation and/or other materials provided with the distribution. | |
12 | * | |
13 | * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY | |
14 | * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
15 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR | |
16 | * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR | |
17 | * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, | |
18 | * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, | |
19 | * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR | |
20 | * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY | |
21 | * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |
22 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | |
23 | * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
24 | */ | |
25 | ||
26 | #ifndef JSSetIterator_h | |
27 | #define JSSetIterator_h | |
28 | ||
29 | #include "JSDestructibleObject.h" | |
30 | #include "JSSet.h" | |
31 | ||
32 | #include "MapData.h" | |
33 | ||
34 | namespace JSC { | |
35 | enum SetIterationKind : uint32_t { | |
36 | SetIterateKey, | |
37 | SetIterateValue, | |
38 | SetIterateKeyValue, | |
39 | }; | |
40 | ||
41 | class JSSetIterator : public JSDestructibleObject { | |
42 | public: | |
43 | typedef JSDestructibleObject Base; | |
44 | ||
45 | DECLARE_EXPORT_INFO; | |
46 | ||
47 | static Structure* createStructure(VM& vm, JSGlobalObject* globalObject, JSValue prototype) | |
48 | { | |
49 | return Structure::create(vm, globalObject, prototype, TypeInfo(ObjectType, StructureFlags), info()); | |
50 | } | |
51 | ||
52 | static JSSetIterator* create(VM& vm, Structure* structure, JSSet* iteratedObject, SetIterationKind kind) | |
53 | { | |
54 | JSSetIterator* instance = new (NotNull, allocateCell<JSSetIterator>(vm.heap)) JSSetIterator(vm, structure, iteratedObject, kind); | |
55 | instance->finishCreation(vm, iteratedObject); | |
56 | return instance; | |
57 | } | |
58 | ||
59 | bool next(CallFrame* callFrame, JSValue& value) | |
60 | { | |
61 | WTF::KeyValuePair<JSValue, JSValue> pair; | |
62 | if (!m_iterator.next(pair)) | |
63 | return false; | |
64 | if (m_kind == SetIterateValue || m_kind == SetIterateKey) | |
65 | value = pair.key; | |
66 | else | |
67 | value = createPair(callFrame, pair.key, pair.key); | |
68 | return true; | |
69 | } | |
70 | ||
71 | void finish() | |
72 | { | |
73 | m_iterator.finish(); | |
74 | } | |
75 | ||
76 | SetIterationKind kind() const { return m_kind; } | |
77 | JSValue iteratedValue() const { return m_set.get(); } | |
78 | JSSetIterator* clone(ExecState*); | |
79 | ||
80 | JSSet::SetData::IteratorData* iteratorData() | |
81 | { | |
82 | return &m_iterator; | |
83 | } | |
84 | ||
85 | private: | |
86 | JSSetIterator(VM& vm, Structure* structure, JSSet* iteratedObject, SetIterationKind kind) | |
87 | : Base(vm, structure) | |
88 | , m_iterator(iteratedObject->m_setData.createIteratorData(this)) | |
89 | , m_kind(kind) | |
90 | { | |
91 | } | |
92 | ||
93 | static void destroy(JSCell*); | |
94 | JS_EXPORT_PRIVATE void finishCreation(VM&, JSSet*); | |
95 | JS_EXPORT_PRIVATE JSValue createPair(CallFrame*, JSValue, JSValue); | |
96 | static void visitChildren(JSCell*, SlotVisitor&); | |
97 | ||
98 | WriteBarrier<JSSet> m_set; | |
99 | JSSet::SetData::IteratorData m_iterator; | |
100 | SetIterationKind m_kind; | |
101 | }; | |
102 | ||
103 | } | |
104 | ||
105 | #endif // !defined(JSSetIterator_h) |