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.
		
		
		
		
		
			
		
			
				
					191 lines
				
				5.6 KiB
			
		
		
			
		
	
	
					191 lines
				
				5.6 KiB
			| 
								 
											3 years ago
										 
									 | 
							
								const GOOD_LEAF_SIZE = 200
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								// :: class<T> A rope sequence is a persistent sequence data structure
							 | 
						||
| 
								 | 
							
								// that supports appending, prepending, and slicing without doing a
							 | 
						||
| 
								 | 
							
								// full copy. It is represented as a mostly-balanced tree.
							 | 
						||
| 
								 | 
							
								class RopeSequence {
							 | 
						||
| 
								 | 
							
								  // length:: number
							 | 
						||
| 
								 | 
							
								  // The length of the rope.
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // :: (union<[T], RopeSequence<T>>) → RopeSequence<T>
							 | 
						||
| 
								 | 
							
								  // Append an array or other rope to this one, returning a new rope.
							 | 
						||
| 
								 | 
							
								  append(other) {
							 | 
						||
| 
								 | 
							
								    if (!other.length) return this
							 | 
						||
| 
								 | 
							
								    other = RopeSequence.from(other)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    return (!this.length && other) ||
							 | 
						||
| 
								 | 
							
								      (other.length < GOOD_LEAF_SIZE && this.leafAppend(other)) ||
							 | 
						||
| 
								 | 
							
								      (this.length < GOOD_LEAF_SIZE && other.leafPrepend(this)) ||
							 | 
						||
| 
								 | 
							
								      this.appendInner(other)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // :: (union<[T], RopeSequence<T>>) → RopeSequence<T>
							 | 
						||
| 
								 | 
							
								  // Prepend an array or other rope to this one, returning a new rope.
							 | 
						||
| 
								 | 
							
								  prepend(other) {
							 | 
						||
| 
								 | 
							
								    if (!other.length) return this
							 | 
						||
| 
								 | 
							
								    return RopeSequence.from(other).append(this)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  appendInner(other) {
							 | 
						||
| 
								 | 
							
								    return new Append(this, other)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // :: (?number, ?number) → RopeSequence<T>
							 | 
						||
| 
								 | 
							
								  // Create a rope repesenting a sub-sequence of this rope.
							 | 
						||
| 
								 | 
							
								  slice(from = 0, to = this.length) {
							 | 
						||
| 
								 | 
							
								    if (from >= to) return RopeSequence.empty
							 | 
						||
| 
								 | 
							
								    return this.sliceInner(Math.max(0, from), Math.min(this.length, to))
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // :: (number) → T
							 | 
						||
| 
								 | 
							
								  // Retrieve the element at the given position from this rope.
							 | 
						||
| 
								 | 
							
								  get(i) {
							 | 
						||
| 
								 | 
							
								    if (i < 0 || i >= this.length) return undefined
							 | 
						||
| 
								 | 
							
								    return this.getInner(i)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // :: ((element: T, index: number) → ?bool, ?number, ?number)
							 | 
						||
| 
								 | 
							
								  // Call the given function for each element between the given
							 | 
						||
| 
								 | 
							
								  // indices. This tends to be more efficient than looping over the
							 | 
						||
| 
								 | 
							
								  // indices and calling `get`, because it doesn't have to descend the
							 | 
						||
| 
								 | 
							
								  // tree for every element.
							 | 
						||
| 
								 | 
							
								  forEach(f, from = 0, to = this.length) {
							 | 
						||
| 
								 | 
							
								    if (from <= to)
							 | 
						||
| 
								 | 
							
								      this.forEachInner(f, from, to, 0)
							 | 
						||
| 
								 | 
							
								    else
							 | 
						||
| 
								 | 
							
								      this.forEachInvertedInner(f, from, to, 0)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // :: ((element: T, index: number) → U, ?number, ?number) → [U]
							 | 
						||
| 
								 | 
							
								  // Map the given functions over the elements of the rope, producing
							 | 
						||
| 
								 | 
							
								  // a flat array.
							 | 
						||
| 
								 | 
							
								  map(f, from = 0, to = this.length) {
							 | 
						||
| 
								 | 
							
								    let result = []
							 | 
						||
| 
								 | 
							
								    this.forEach((elt, i) => result.push(f(elt, i)), from, to)
							 | 
						||
| 
								 | 
							
								    return result
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // :: (?union<[T], RopeSequence<T>>) → RopeSequence<T>
							 | 
						||
| 
								 | 
							
								  // Create a rope representing the given array, or return the rope
							 | 
						||
| 
								 | 
							
								  // itself if a rope was given.
							 | 
						||
| 
								 | 
							
								  static from(values) {
							 | 
						||
| 
								 | 
							
								    if (values instanceof RopeSequence) return values
							 | 
						||
| 
								 | 
							
								    return values && values.length ? new Leaf(values) : RopeSequence.empty
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // flatten:: () → [T]
							 | 
						||
| 
								 | 
							
								  // Return the content of this rope as an array.
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								class Leaf extends RopeSequence {
							 | 
						||
| 
								 | 
							
								  constructor(values) {
							 | 
						||
| 
								 | 
							
								    super()
							 | 
						||
| 
								 | 
							
								    this.values = values
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  flatten() {
							 | 
						||
| 
								 | 
							
								    return this.values
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  sliceInner(from, to) {
							 | 
						||
| 
								 | 
							
								    if (from == 0 && to == this.length) return this
							 | 
						||
| 
								 | 
							
								    return new Leaf(this.values.slice(from, to))
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  getInner(i) {
							 | 
						||
| 
								 | 
							
								    return this.values[i]
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  forEachInner(f, from, to, start) {
							 | 
						||
| 
								 | 
							
								    for (let i = from; i < to; i++)
							 | 
						||
| 
								 | 
							
								      if (f(this.values[i], start + i) === false) return false
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  forEachInvertedInner(f, from, to, start) {
							 | 
						||
| 
								 | 
							
								    for (let i = from - 1; i >= to; i--)
							 | 
						||
| 
								 | 
							
								      if (f(this.values[i], start + i) === false) return false
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  leafAppend(other) {
							 | 
						||
| 
								 | 
							
								    if (this.length + other.length <= GOOD_LEAF_SIZE)
							 | 
						||
| 
								 | 
							
								      return new Leaf(this.values.concat(other.flatten()))
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  leafPrepend(other) {
							 | 
						||
| 
								 | 
							
								    if (this.length + other.length <= GOOD_LEAF_SIZE)
							 | 
						||
| 
								 | 
							
								      return new Leaf(other.flatten().concat(this.values))
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  get length() { return this.values.length }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  get depth() { return 0 }
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								// :: RopeSequence
							 | 
						||
| 
								 | 
							
								// The empty rope sequence.
							 | 
						||
| 
								 | 
							
								RopeSequence.empty = new Leaf([])
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								class Append extends RopeSequence {
							 | 
						||
| 
								 | 
							
								  constructor(left, right) {
							 | 
						||
| 
								 | 
							
								    super()
							 | 
						||
| 
								 | 
							
								    this.left = left
							 | 
						||
| 
								 | 
							
								    this.right = right
							 | 
						||
| 
								 | 
							
								    this.length = left.length + right.length
							 | 
						||
| 
								 | 
							
								    this.depth = Math.max(left.depth, right.depth) + 1
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  flatten() {
							 | 
						||
| 
								 | 
							
								    return this.left.flatten().concat(this.right.flatten())
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  getInner(i) {
							 | 
						||
| 
								 | 
							
								    return i < this.left.length ? this.left.get(i) : this.right.get(i - this.left.length)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  forEachInner(f, from, to, start) {
							 | 
						||
| 
								 | 
							
								    let leftLen = this.left.length
							 | 
						||
| 
								 | 
							
								    if (from < leftLen &&
							 | 
						||
| 
								 | 
							
								        this.left.forEachInner(f, from, Math.min(to, leftLen), start) === false)
							 | 
						||
| 
								 | 
							
								      return false
							 | 
						||
| 
								 | 
							
								    if (to > leftLen &&
							 | 
						||
| 
								 | 
							
								        this.right.forEachInner(f, Math.max(from - leftLen, 0), Math.min(this.length, to) - leftLen, start + leftLen) === false)
							 | 
						||
| 
								 | 
							
								      return false
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  forEachInvertedInner(f, from, to, start) {
							 | 
						||
| 
								 | 
							
								    let leftLen = this.left.length
							 | 
						||
| 
								 | 
							
								    if (from > leftLen &&
							 | 
						||
| 
								 | 
							
								        this.right.forEachInvertedInner(f, from - leftLen, Math.max(to, leftLen) - leftLen, start + leftLen) === false)
							 | 
						||
| 
								 | 
							
								      return false
							 | 
						||
| 
								 | 
							
								    if (to < leftLen &&
							 | 
						||
| 
								 | 
							
								        this.left.forEachInvertedInner(f, Math.min(from, leftLen), to, start) === false)
							 | 
						||
| 
								 | 
							
								      return false
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  sliceInner(from, to) {
							 | 
						||
| 
								 | 
							
								    if (from == 0 && to == this.length) return this
							 | 
						||
| 
								 | 
							
								    let leftLen = this.left.length
							 | 
						||
| 
								 | 
							
								    if (to <= leftLen) return this.left.slice(from, to)
							 | 
						||
| 
								 | 
							
								    if (from >= leftLen) return this.right.slice(from - leftLen, to - leftLen)
							 | 
						||
| 
								 | 
							
								    return this.left.slice(from, leftLen).append(this.right.slice(0, to - leftLen))
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  leafAppend(other) {
							 | 
						||
| 
								 | 
							
								    let inner = this.right.leafAppend(other)
							 | 
						||
| 
								 | 
							
								    if (inner) return new Append(this.left, inner)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  leafPrepend(other) {
							 | 
						||
| 
								 | 
							
								    let inner = this.left.leafPrepend(other)
							 | 
						||
| 
								 | 
							
								    if (inner) return new Append(inner, this.right)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  appendInner(other) {
							 | 
						||
| 
								 | 
							
								    if (this.left.depth >= Math.max(this.right.depth, other.depth) + 1)
							 | 
						||
| 
								 | 
							
								      return new Append(this.left, new Append(this.right, other))
							 | 
						||
| 
								 | 
							
								    return new Append(this, other)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								module.exports = RopeSequence
							 |