You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							244 lines
						
					
					
						
							4.7 KiB
						
					
					
				
			
		
		
	
	
							244 lines
						
					
					
						
							4.7 KiB
						
					
					
				| /**
 | |
|  * Mnemonist SparseMap
 | |
|  * ====================
 | |
|  *
 | |
|  * JavaScript sparse map implemented on top of byte arrays.
 | |
|  *
 | |
|  * [Reference]: https://research.swtch.com/sparse
 | |
|  */
 | |
| var Iterator = require('obliterator/iterator'),
 | |
|     getPointerArray = require('./utils/typed-arrays.js').getPointerArray;
 | |
| 
 | |
| /**
 | |
|  * SparseMap.
 | |
|  *
 | |
|  * @constructor
 | |
|  */
 | |
| function SparseMap(Values, length) {
 | |
|   if (arguments.length < 2) {
 | |
|     length = Values;
 | |
|     Values = Array;
 | |
|   }
 | |
| 
 | |
|   var ByteArray = getPointerArray(length);
 | |
| 
 | |
|   // Properties
 | |
|   this.size = 0;
 | |
|   this.length = length;
 | |
|   this.dense = new ByteArray(length);
 | |
|   this.sparse = new ByteArray(length);
 | |
|   this.vals = new Values(length);
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * Method used to clear the structure.
 | |
|  *
 | |
|  * @return {undefined}
 | |
|  */
 | |
| SparseMap.prototype.clear = function() {
 | |
|   this.size = 0;
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * Method used to check the existence of a member in the set.
 | |
|  *
 | |
|  * @param  {number} member - Member to test.
 | |
|  * @return {SparseMap}
 | |
|  */
 | |
| SparseMap.prototype.has = function(member) {
 | |
|   var index = this.sparse[member];
 | |
| 
 | |
|   return (
 | |
|     index < this.size &&
 | |
|     this.dense[index] === member
 | |
|   );
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * Method used to get the value associated to a member in the set.
 | |
|  *
 | |
|  * @param  {number} member - Member to test.
 | |
|  * @return {any}
 | |
|  */
 | |
| SparseMap.prototype.get = function(member) {
 | |
|   var index = this.sparse[member];
 | |
| 
 | |
|   if (index < this.size && this.dense[index] === member)
 | |
|     return this.vals[index];
 | |
| 
 | |
|   return;
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * Method used to set a value into the map.
 | |
|  *
 | |
|  * @param  {number} member - Member to set.
 | |
|  * @param  {any}    value  - Associated value.
 | |
|  * @return {SparseMap}
 | |
|  */
 | |
| SparseMap.prototype.set = function(member, value) {
 | |
|   var index = this.sparse[member];
 | |
| 
 | |
|   if (index < this.size && this.dense[index] === member) {
 | |
|     this.vals[index] = value;
 | |
|     return this;
 | |
|   }
 | |
| 
 | |
|   this.dense[this.size] = member;
 | |
|   this.sparse[member] = this.size;
 | |
|   this.vals[this.size] = value;
 | |
|   this.size++;
 | |
| 
 | |
|   return this;
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * Method used to remove a member from the set.
 | |
|  *
 | |
|  * @param  {number} member - Member to delete.
 | |
|  * @return {boolean}
 | |
|  */
 | |
| SparseMap.prototype.delete = function(member) {
 | |
|   var index = this.sparse[member];
 | |
| 
 | |
|   if (index >= this.size || this.dense[index] !== member)
 | |
|     return false;
 | |
| 
 | |
|   index = this.dense[this.size - 1];
 | |
|   this.dense[this.sparse[member]] = index;
 | |
|   this.sparse[index] = this.sparse[member];
 | |
|   this.size--;
 | |
| 
 | |
|   return true;
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * Method used to iterate over the set's values.
 | |
|  *
 | |
|  * @param  {function}  callback - Function to call for each item.
 | |
|  * @param  {object}    scope    - Optional scope.
 | |
|  * @return {undefined}
 | |
|  */
 | |
| SparseMap.prototype.forEach = function(callback, scope) {
 | |
|   scope = arguments.length > 1 ? scope : this;
 | |
| 
 | |
|   for (var i = 0; i < this.size; i++)
 | |
|     callback.call(scope, this.vals[i], this.dense[i]);
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * Method used to create an iterator over a set's members.
 | |
|  *
 | |
|  * @return {Iterator}
 | |
|  */
 | |
| SparseMap.prototype.keys = function() {
 | |
|   var size = this.size,
 | |
|       dense = this.dense,
 | |
|       i = 0;
 | |
| 
 | |
|   return new Iterator(function() {
 | |
|     if (i < size) {
 | |
|       var item = dense[i];
 | |
|       i++;
 | |
| 
 | |
|       return {
 | |
|         value: item
 | |
|       };
 | |
|     }
 | |
| 
 | |
|     return {
 | |
|       done: true
 | |
|     };
 | |
|   });
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * Method used to create an iterator over a set's values.
 | |
|  *
 | |
|  * @return {Iterator}
 | |
|  */
 | |
| SparseMap.prototype.values = function() {
 | |
|   var size = this.size,
 | |
|       values = this.vals,
 | |
|       i = 0;
 | |
| 
 | |
|   return new Iterator(function() {
 | |
|     if (i < size) {
 | |
|       var item = values[i];
 | |
|       i++;
 | |
| 
 | |
|       return {
 | |
|         value: item
 | |
|       };
 | |
|     }
 | |
| 
 | |
|     return {
 | |
|       done: true
 | |
|     };
 | |
|   });
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * Method used to create an iterator over a set's entries.
 | |
|  *
 | |
|  * @return {Iterator}
 | |
|  */
 | |
| SparseMap.prototype.entries = function() {
 | |
|   var size = this.size,
 | |
|       dense = this.dense,
 | |
|       values = this.vals,
 | |
|       i = 0;
 | |
| 
 | |
|   return new Iterator(function() {
 | |
|     if (i < size) {
 | |
|       var item = [dense[i], values[i]];
 | |
|       i++;
 | |
| 
 | |
|       return {
 | |
|         value: item
 | |
|       };
 | |
|     }
 | |
| 
 | |
|     return {
 | |
|       done: true
 | |
|     };
 | |
|   });
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * Attaching the #.entries method to Symbol.iterator if possible.
 | |
|  */
 | |
| if (typeof Symbol !== 'undefined')
 | |
|   SparseMap.prototype[Symbol.iterator] = SparseMap.prototype.entries;
 | |
| 
 | |
| /**
 | |
|  * Convenience known methods.
 | |
|  */
 | |
| SparseMap.prototype.inspect = function() {
 | |
|   var proxy = new Map();
 | |
| 
 | |
|   for (var i = 0; i < this.size; i++)
 | |
|     proxy.set(this.dense[i], this.vals[i]);
 | |
| 
 | |
|   // Trick so that node displays the name of the constructor
 | |
|   Object.defineProperty(proxy, 'constructor', {
 | |
|     value: SparseMap,
 | |
|     enumerable: false
 | |
|   });
 | |
| 
 | |
|   proxy.length = this.length;
 | |
| 
 | |
|   if (this.vals.constructor !== Array)
 | |
|     proxy.type = this.vals.constructor.name;
 | |
| 
 | |
|   return proxy;
 | |
| };
 | |
| 
 | |
| if (typeof Symbol !== 'undefined')
 | |
|   SparseMap.prototype[Symbol.for('nodejs.util.inspect.custom')] = SparseMap.prototype.inspect;
 | |
| 
 | |
| /**
 | |
|  * Exporting.
 | |
|  */
 | |
| module.exports = SparseMap;
 |