featL highlight alternating runs for natural msort + highlight green for finish

Author: dwangng

src/algorithms/controllers/msort_arr_nat.js

@@ -66,6 +66,33 @@ export function initVisualisers() {
 // Define helper functions
 // -------------------------------------------------------------------------------
 
+// This function highlights all the runs alternating colours, kudos to chatgpt
+function highlightNaturalRuns(vis, array, runAColor, runBColor) {
+  let toggle = 0; // 0 = runAColor, 1 = runBColor
+  let i = 0;
+
+  while (i < array.length) {
+    let start = i;
+
+    // Find the length of the increasing run
+    while (i < array.length - 1 && array[i] <= array[i + 1]) {
+      i++;
+    }
+
+    // Highlight the run
+    for (let j = start; j <= i; j++) {
+      highlight(vis, j, toggle == 0 ? runAColor : runBColor);
+    }
+
+    // Flip toggle between 0 and 1 for the next run
+    toggle = 1 - toggle;
+
+    // Move to the next element after the current run
+    i++;
+  }
+}
+
+
 
 
 /**
@@ -106,18 +133,20 @@ export function run_msort() {
 
     do {
 
-      chunker.add('MainWhile', () => {
+      chunker.add('MainWhile', (vis, a) => {
         // no animation
-      }, []);
+        highlightNaturalRuns(vis, a, runAColor, runBColor);
+      }, [A]);
 
 
 
       runcount = 0;
       let left = 0;
 
-      chunker.add('runcount', (vis, c_rcount) => {
+      chunker.add('runcount', (vis, a, c_rcount) => {
+        vis.array.set(a, 'msort_arr_nat');
         set_simple_stack(vis.array, [c_rcount]);
-      }, [runcount]);
+      }, [A, runcount]);
 
       chunker.add('left', (vis, c_left) => {
         assignVarToA(vis, 'left', c_left, size);
@@ -340,13 +369,21 @@ export function run_msort() {
         }, [A, left, runcount]);
 
       } while (left < size);
+
+      chunker.add('mergeDone', (vis, a) => {
+        highlightNaturalRuns(vis, a, runAColor, runBColor);
+      }, [A])
+
     } while (runcount > 1);
 
-    chunker.add('Done', (vis) => {
-      for (let i = 0; i < size; i++) highlight(vis, i, "sortColor");
-      assignVarToA(vis, 'Done', size, size);
+    chunker.add('Done', (vis, a) => {
+      vis.array.set(a, 'msort_arr_nat');
+      for (let i = 0; i < size; i++) {
+        highlight(vis, i, sortColor);
+      }
+
       set_simple_stack(vis.array, ["DONE"]);
-    }, []);
+    }, [A]);
 
     const maxValue = entire_num_array.reduce((acc, curr) => (acc < curr ? curr : acc), 0);
 

src/algorithms/controllers/msort_shared.js

@@ -26,6 +26,7 @@ export function highlight(vis, index, color) {
         vis.array.patch(index);
     }
 }
+
 // Same as highlight() but checks isMergeExpanded()/arrayB is displayed, otherwise does nothing
 export function highlightB(vis, index, color) {
     if (isMergeExpanded()) {

src/algorithms/pseudocode/msort_arr_nat.js

@@ -49,6 +49,7 @@ MergeAll
         left <- right + 1 // skip to the next pair of runs (if any) \\B left2
     until left >= size
     \\In}
+    // all consecutive pairs of runs merged \\B mergeDone
 \\Code}
 
 \\Code{