Compositor // Fix update(updateExisting: true).

vChewing · Feb 18, 2025 · ed3357e · ed3357e
1 parent 08db711
commit ed3357e
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/Sources/Megrez/1_Compositor.swift b/Sources/Megrez/1_Compositor.swift
@@ -236,10 +236,14 @@ extension Megrez {
     @discardableResult
     public func update(updateExisting: Bool = false) -> Int {
       let maxSpanLength = maxSpanLength
-      let rangeOfPositions = max(0, cursor - maxSpanLength) ..< min(
-        cursor + maxSpanLength,
-        keys.count
-      )
+      let rangeOfPositions: Range<Int>
+      if updateExisting {
+        rangeOfPositions = 0 ..< spans.count
+      } else {
+        let lowerbound = Swift.max(0, cursor - maxSpanLength)
+        let upperbound = Swift.min(cursor + maxSpanLength, keys.count)
+        rangeOfPositions = lowerbound ..< upperbound
+      }
       var nodesChanged = 0
       rangeOfPositions.forEach { position in
         let rangeOfLengths = 1 ... min(maxSpanLength, rangeOfPositions.upperBound - position)