You cannot select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
	
	
		
			151 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			JavaScript
		
	
		
		
			
		
	
	
			151 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			JavaScript
		
	
| 
											9 months ago
										 | /* | ||
|  | 	MIT License http://www.opensource.org/licenses/mit-license.php
 | ||
|  | 	Author Tobias Koppers @sokra | ||
|  | */ | ||
|  | 
 | ||
|  | "use strict"; | ||
|  | 
 | ||
|  | /** | ||
|  |  * @template {any[]} T | ||
|  |  */ | ||
|  | class TupleSet { | ||
|  | 	constructor(init) { | ||
|  | 		this._map = new Map(); | ||
|  | 		this.size = 0; | ||
|  | 		if (init) { | ||
|  | 			for (const tuple of init) { | ||
|  | 				this.add(...tuple); | ||
|  | 			} | ||
|  | 		} | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 	 * @param  {T} args tuple | ||
|  | 	 * @returns {void} | ||
|  | 	 */ | ||
|  | 	add(...args) { | ||
|  | 		let map = this._map; | ||
|  | 		for (let i = 0; i < args.length - 2; i++) { | ||
|  | 			const arg = args[i]; | ||
|  | 			const innerMap = map.get(arg); | ||
|  | 			if (innerMap === undefined) { | ||
|  | 				map.set(arg, (map = new Map())); | ||
|  | 			} else { | ||
|  | 				map = innerMap; | ||
|  | 			} | ||
|  | 		} | ||
|  | 
 | ||
|  | 		const beforeLast = args[args.length - 2]; | ||
|  | 		let set = map.get(beforeLast); | ||
|  | 		if (set === undefined) { | ||
|  | 			map.set(beforeLast, (set = new Set())); | ||
|  | 		} | ||
|  | 
 | ||
|  | 		const last = args[args.length - 1]; | ||
|  | 		this.size -= set.size; | ||
|  | 		set.add(last); | ||
|  | 		this.size += set.size; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 	 * @param  {T} args tuple | ||
|  | 	 * @returns {boolean} true, if the tuple is in the Set | ||
|  | 	 */ | ||
|  | 	has(...args) { | ||
|  | 		let map = this._map; | ||
|  | 		for (let i = 0; i < args.length - 2; i++) { | ||
|  | 			const arg = args[i]; | ||
|  | 			map = map.get(arg); | ||
|  | 			if (map === undefined) { | ||
|  | 				return false; | ||
|  | 			} | ||
|  | 		} | ||
|  | 
 | ||
|  | 		const beforeLast = args[args.length - 2]; | ||
|  | 		let set = map.get(beforeLast); | ||
|  | 		if (set === undefined) { | ||
|  | 			return false; | ||
|  | 		} | ||
|  | 
 | ||
|  | 		const last = args[args.length - 1]; | ||
|  | 		return set.has(last); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 	 * @param {T} args tuple | ||
|  | 	 * @returns {void} | ||
|  | 	 */ | ||
|  | 	delete(...args) { | ||
|  | 		let map = this._map; | ||
|  | 		for (let i = 0; i < args.length - 2; i++) { | ||
|  | 			const arg = args[i]; | ||
|  | 			map = map.get(arg); | ||
|  | 			if (map === undefined) { | ||
|  | 				return; | ||
|  | 			} | ||
|  | 		} | ||
|  | 
 | ||
|  | 		const beforeLast = args[args.length - 2]; | ||
|  | 		let set = map.get(beforeLast); | ||
|  | 		if (set === undefined) { | ||
|  | 			return; | ||
|  | 		} | ||
|  | 
 | ||
|  | 		const last = args[args.length - 1]; | ||
|  | 		this.size -= set.size; | ||
|  | 		set.delete(last); | ||
|  | 		this.size += set.size; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/** | ||
|  | 	 * @returns {Iterator<T>} iterator | ||
|  | 	 */ | ||
|  | 	[Symbol.iterator]() { | ||
|  | 		const iteratorStack = []; | ||
|  | 		const tuple = []; | ||
|  | 		let currentSetIterator = undefined; | ||
|  | 
 | ||
|  | 		const next = it => { | ||
|  | 			const result = it.next(); | ||
|  | 			if (result.done) { | ||
|  | 				if (iteratorStack.length === 0) return false; | ||
|  | 				tuple.pop(); | ||
|  | 				return next(iteratorStack.pop()); | ||
|  | 			} | ||
|  | 			const [key, value] = result.value; | ||
|  | 			iteratorStack.push(it); | ||
|  | 			tuple.push(key); | ||
|  | 			if (value instanceof Set) { | ||
|  | 				currentSetIterator = value[Symbol.iterator](); | ||
|  | 				return true; | ||
|  | 			} else { | ||
|  | 				return next(value[Symbol.iterator]()); | ||
|  | 			} | ||
|  | 		}; | ||
|  | 
 | ||
|  | 		next(this._map[Symbol.iterator]()); | ||
|  | 
 | ||
|  | 		return { | ||
|  | 			next() { | ||
|  | 				while (currentSetIterator) { | ||
|  | 					const result = currentSetIterator.next(); | ||
|  | 					if (result.done) { | ||
|  | 						tuple.pop(); | ||
|  | 						if (!next(iteratorStack.pop())) { | ||
|  | 							currentSetIterator = undefined; | ||
|  | 						} | ||
|  | 					} else { | ||
|  | 						return { | ||
|  | 							done: false, | ||
|  | 							value: /** @type {T} */ (tuple.concat(result.value)) | ||
|  | 						}; | ||
|  | 					} | ||
|  | 				} | ||
|  | 				return { done: true, value: undefined }; | ||
|  | 			} | ||
|  | 		}; | ||
|  | 	} | ||
|  | } | ||
|  | 
 | ||
|  | module.exports = TupleSet; |