123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166 |
- /*
- MIT License http://www.opensource.org/licenses/mit-license.php
- Author Tobias Koppers @sokra
- */
- "use strict";
- const TOMBSTONE = Symbol("tombstone");
- const UNDEFINED_MARKER = Symbol("undefined");
- /**
- * @template T
- * @typedef {T | undefined} Cell<T>
- */
- /**
- * @template T
- * @typedef {T | typeof TOMBSTONE | typeof UNDEFINED_MARKER} InternalCell<T>
- */
- /**
- * @template K
- * @template V
- * @param {[K, InternalCell<V>]} pair the internal cell
- * @returns {[K, Cell<V>]} its “safe” representation
- */
- const extractPair = pair => {
- const key = pair[0];
- const val = pair[1];
- if (val === UNDEFINED_MARKER || val === TOMBSTONE) {
- return [key, undefined];
- } else {
- return /** @type {[K, Cell<V>]} */ (pair);
- }
- };
- /**
- * @template K
- * @template V
- */
- class StackedMap {
- /**
- * @param {Map<K, InternalCell<V>>[]=} parentStack an optional parent
- */
- constructor(parentStack) {
- /** @type {Map<K, InternalCell<V>>} */
- this.map = new Map();
- /** @type {Map<K, InternalCell<V>>[]} */
- this.stack = parentStack === undefined ? [] : parentStack.slice();
- this.stack.push(this.map);
- }
- /**
- * @param {K} item the key of the element to add
- * @param {V} value the value of the element to add
- * @returns {void}
- */
- set(item, value) {
- this.map.set(item, value === undefined ? UNDEFINED_MARKER : value);
- }
- /**
- * @param {K} item the item to delete
- * @returns {void}
- */
- delete(item) {
- if (this.stack.length > 1) {
- this.map.set(item, TOMBSTONE);
- } else {
- this.map.delete(item);
- }
- }
- /**
- * @param {K} item the item to test
- * @returns {boolean} true if the item exists in this set
- */
- has(item) {
- const topValue = this.map.get(item);
- if (topValue !== undefined) {
- return topValue !== TOMBSTONE;
- }
- if (this.stack.length > 1) {
- for (let i = this.stack.length - 2; i >= 0; i--) {
- const value = this.stack[i].get(item);
- if (value !== undefined) {
- this.map.set(item, value);
- return value !== TOMBSTONE;
- }
- }
- this.map.set(item, TOMBSTONE);
- }
- return false;
- }
- /**
- * @param {K} item the key of the element to return
- * @returns {Cell<V>} the value of the element
- */
- get(item) {
- const topValue = this.map.get(item);
- if (topValue !== undefined) {
- return topValue === TOMBSTONE || topValue === UNDEFINED_MARKER
- ? undefined
- : topValue;
- }
- if (this.stack.length > 1) {
- for (let i = this.stack.length - 2; i >= 0; i--) {
- const value = this.stack[i].get(item);
- if (value !== undefined) {
- this.map.set(item, value);
- return value === TOMBSTONE || value === UNDEFINED_MARKER
- ? undefined
- : value;
- }
- }
- this.map.set(item, TOMBSTONE);
- }
- return undefined;
- }
- _compress() {
- if (this.stack.length === 1) return;
- this.map = new Map();
- for (const data of this.stack) {
- for (const pair of data) {
- if (pair[1] === TOMBSTONE) {
- this.map.delete(pair[0]);
- } else {
- this.map.set(pair[0], pair[1]);
- }
- }
- }
- this.stack = [this.map];
- }
- asArray() {
- this._compress();
- return Array.from(this.map.keys());
- }
- asSet() {
- this._compress();
- return new Set(this.map.keys());
- }
- asPairArray() {
- this._compress();
- return Array.from(this.map.entries(), extractPair);
- }
- asMap() {
- return new Map(this.asPairArray());
- }
- get size() {
- this._compress();
- return this.map.size;
- }
- createChild() {
- return new StackedMap(this.stack);
- }
- }
- module.exports = StackedMap;
|