js_suffix_trie.js

var JsSuffixTrie;
JsSuffixTrie = (function() {
  function JsSuffixTrie() {
    this.count = 0;
    this.structure = {};
    this.prefix = "";
  }
  JsSuffixTrie.prototype.add = function(string) {
    var chr, index, length, next, node;
    node = this.structure;
    length = string.length;
    index = 0;
    while (index < length) {
      chr = string[index++];
      next = node[chr];
      if (next) {
        node = next;
      } else {
        node[chr] = {};
        node = node[chr];
      }
    }
    if (node.terminator) {
      return false;
    } else {
      node.terminator = true;
      this.count++;
      return true;
    }
  };
  JsSuffixTrie.prototype.remove = function(string) {
    var chr, index, length, node;
    node = this.structure;
    length = string.length;
    index = 0;
    while (index < length) {
      chr = string[index++];
      node = node[chr];
      if (!node) {
        return false;
      }
    }
    if (node.terminator) {
      delete node.terminator;
      this.count--;
      return true;
    } else {
      return false;
    }
  };
  JsSuffixTrie.prototype.contains = function(string) {
    var node;
    node = this.findNode(string);
    return node !== null && node.terminator;
  };
  JsSuffixTrie.prototype.subTrie = function(prefix) {
    var node, subTrie;
    node = this.findNode(prefix);
    subTrie = new JsSuffixTrie;
    subTrie.structure = node;
    subTrie.prefix = prefix;
    return subTrie;
  };
  JsSuffixTrie.prototype.find = function(prefix) {
    return JsSuffixTrie.nodeToArray(this.findNode(prefix), prefix);
  };
  JsSuffixTrie.prototype.findNode = function(string) {
    var currentChar, index, length, node;
    node = this.structure;
    length = string.length;
    index = 0;
    while (index < length) {
      currentChar = string[index++];
      node = node[currentChar];
      if (!node) {
        return null;
      }
    }
    return node;
  };
  JsSuffixTrie.prototype.each = function(callback) {
    return JsSuffixTrie.each(callback, this.structure, 0, this.prefix);
  };
  JsSuffixTrie.each = function(callback, node, index, string) {
    var property;
    if (node && node.terminator) {
      callback(index++, string);
    }
    for (property in node) {
      index = this.each(callback, node[property], index, string + property);
    }
    return index;
  };
  JsSuffixTrie.prototype.size = function() {
    return this.count;
  };
  JsSuffixTrie.prototype.calculateSize = function(node) {
    var property, size;
    if (node == null) {
      node = this.structure;
    }
    size = node.terminator ? 1 : 0;
    for (property in node) {
      size += this.calculateSize(node[property]);
    }
    return size;
  };
  JsSuffixTrie.fromArray = function(array) {
    var i, length, trie;
    trie = new JsSuffixTrie;
    length = array.length;
    i = 0;
    while (i < length) {
      trie.add(array[i++]);
    }
    trie.count = i;
    return trie;
  };
  JsSuffixTrie.prototype.toArray = function() {
    return JsSuffixTrie.nodeToArray(this.structure, this.prefix);
  };
  JsSuffixTrie.nodeToArray = function(node, prefix) {
    var array;
    array = [];
    this.each(function(index, value) {
      return array[index] = value;
    }, node, 0, prefix);
    return array;
  };
  JsSuffixTrie.fromJSON = function(json) {
    var trie;
    trie = new JsSuffixTrie;
    trie.structure = JSON.parse(json);
    trie.calculateSize();
    return trie;
  };
  JsSuffixTrie.prototype.toJSON = function() {
    return JSON.stringify(this.structure);
  };
  return JsSuffixTrie;
})();