add under for sort

uiua-lang · Sep 6, 2024 · cb348ae · cb348ae
1 parent a628fab
commit cb348ae
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/algorithm/invert.rs b/src/algorithm/invert.rs
@@ -523,6 +523,17 @@ pub(crate) fn under_instrs(
             (CopyToUnder(1), Fall, Dup, Rise, PushToUnder(1)),
             (PopUnder(1), Select, PopUnder(1), Flip, Select)
         ),
+        // Sort
+        &pat!(
+            Sort,
+            (Dup, Rise, CopyToUnder(1), Select),
+            (PopUnder(1), Rise, Select),
+        ),
+        &pat!(
+            SortDown,
+            (Dup, Fall, CopyToUnder(1), Select),
+            (PopUnder(1), Rise, Select),
+        ),
         // Index of
         &maybe_val!(pat!(
             IndexOf,

diff --git a/tests/under.ua b/tests/under.ua
@@ -327,3 +327,9 @@ G ← ⍜(⊙⊙⊢) ⋅⊙∘
 ⍤⟜≍: "2 - 4" ⍜°$"_ - _"∩(×2⋕) "1 - 2"
 ⍤⟜≍: [1 20 30] ⍜°(⊂1)(×10) [1 2 3]
 ⍤⟜≍: map 2_1 4_30 ⍜°(insert1)(×10) map 1_2 3_4
+
+# Experimental!
+
+# Sort
+⍤⟜≍: [4000 20 300 1] ⍜⍆(×ⁿ:10°⊏) [4 2 3 1]
+⍤⟜≍: [4 200 30 1000] ⍜(⇌⍆|×ⁿ:10°⊏) [4 2 3 1]