1 files changed, 102 insertions, 0 deletions
diff --git a/Source/JavaScriptCore/parser/VariableEnvironment.h b/Source/JavaScriptCore/parser/VariableEnvironment.h
new file mode 100644
index 000000000..519d0121a
--- /dev/null
+++ b/Source/JavaScriptCore/parser/VariableEnvironment.h
@@ -0,0 +1,102 @@
+/*
+ * Copyright (C) 2015 Apple Inc. All Rights Reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
+ * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+ * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+ * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+ * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
+ * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#ifndef VariableEnvironment_h
+#define VariableEnvironment_h
+
+#include "Identifier.h"
+#include <wtf/HashMap.h>
+
+namespace JSC {
+
+struct VariableEnvironmentEntry {
+public:
+    ALWAYS_INLINE bool isCaptured() const { return m_bits & IsCaptured; }
+    ALWAYS_INLINE bool isConst() const { return m_bits & IsConst; }
+    ALWAYS_INLINE bool isVar() const { return m_bits & IsVar; }
+    ALWAYS_INLINE bool isLet() const { return m_bits & IsLet; }
+    ALWAYS_INLINE bool isExported() const { return m_bits & IsExported; }
+    ALWAYS_INLINE bool isImported() const { return m_bits & IsImported; }
+    ALWAYS_INLINE bool isImportedNamespace() const { return m_bits & IsImportedNamespace; }
+
+    ALWAYS_INLINE void setIsCaptured() { m_bits |= IsCaptured; }
+    ALWAYS_INLINE void setIsConst() { m_bits |= IsConst; }
+    ALWAYS_INLINE void setIsVar() { m_bits |= IsVar; }
+    ALWAYS_INLINE void setIsLet() { m_bits |= IsLet; }
+    ALWAYS_INLINE void setIsExported() { m_bits |= IsExported; }
+    ALWAYS_INLINE void setIsImported() { m_bits |= IsImported; }
+    ALWAYS_INLINE void setIsImportedNamespace() { m_bits |= IsImportedNamespace; }
+
+    ALWAYS_INLINE void clearIsVar() { m_bits &= ~IsVar; }
+
+private:
+    enum Traits {
+        IsCaptured = 1 << 0,
+        IsConst = 1 << 1,
+        IsVar = 1 << 2,
+        IsLet = 1 << 3,
+        IsExported = 1 << 4,
+        IsImported = 1 << 5,
+        IsImportedNamespace = 1 << 6
+    };
+    uint8_t m_bits { 0 };
+};
+
+struct VariableEnvironmentEntryHashTraits : HashTraits<VariableEnvironmentEntry> {
+    static const bool needsDestruction = false;
+};
+
+class VariableEnvironment {
+private:
+    typedef HashMap<RefPtr<UniquedStringImpl>, VariableEnvironmentEntry, IdentifierRepHash, HashTraits<RefPtr<UniquedStringImpl>>, VariableEnvironmentEntryHashTraits> Map;
+public:
+    ALWAYS_INLINE Map::iterator begin() { return m_map.begin(); }
+    ALWAYS_INLINE Map::iterator end() { return m_map.end(); }
+    ALWAYS_INLINE Map::const_iterator begin() const { return m_map.begin(); }
+    ALWAYS_INLINE Map::const_iterator end() const { return m_map.end(); }
+    ALWAYS_INLINE Map::AddResult add(const RefPtr<UniquedStringImpl>& identifier) { return m_map.add(identifier, VariableEnvironmentEntry()); }
+    ALWAYS_INLINE Map::AddResult add(const Identifier& identifier) { return add(identifier.impl()); }
+    ALWAYS_INLINE unsigned size() const { return m_map.size(); }
+    ALWAYS_INLINE bool contains(const RefPtr<UniquedStringImpl>& identifier) const { return m_map.contains(identifier); }
+    ALWAYS_INLINE bool remove(const RefPtr<UniquedStringImpl>& identifier) { return m_map.remove(identifier); }
+    ALWAYS_INLINE Map::iterator find(const RefPtr<UniquedStringImpl>& identifier) { return m_map.find(identifier); }
+    ALWAYS_INLINE Map::const_iterator find(const RefPtr<UniquedStringImpl>& identifier) const { return m_map.find(identifier); }
+    void swap(VariableEnvironment& other);
+    void markVariableAsCapturedIfDefined(const RefPtr<UniquedStringImpl>& identifier);
+    void markVariableAsCaptured(const RefPtr<UniquedStringImpl>& identifier);
+    void markAllVariablesAsCaptured();
+    bool hasCapturedVariables() const;
+    bool captures(UniquedStringImpl* identifier) const;
+    void markVariableAsImported(const RefPtr<UniquedStringImpl>& identifier);
+    void markVariableAsExported(const RefPtr<UniquedStringImpl>& identifier);
+
+private:
+    Map m_map;
+    bool m_isEverythingCaptured { false };
+};
+
+} // namespace JSC
+
+#endif // VariableEnvironment_h