Merge remote-tracking branch 'n8sh/take-slice'

source/tanya/algorithm/iteration.d

@@ -24,7 +24,7 @@ import tanya.algorithm.comparison;
 import tanya.algorithm.mutation;
 import tanya.range;
 
-private mixin template Take(R, bool exactly)
+private struct Take(R, bool exactly)
 {
     private R source;
     size_t length_;
@@ -187,6 +187,22 @@ private mixin template Take(R, bool exactly)
             }
         }
     }
+
+    static if (!exactly && hasSlicing!R)
+    {
+        auto opSlice(size_t i, size_t j)
+        in
+        {
+            assert(i <= j);
+            assert(j <= length);
+        }
+        do
+        {
+            return typeof(this)(this.source[i .. j], length);
+        }
+    }
+
+    version (unittest) static assert(isInputRange!Take);
 }
 
 /**
@@ -212,25 +228,34 @@ private mixin template Take(R, bool exactly)
 auto take(R)(R range, size_t n)
 if (isInputRange!R)
 {
-    static struct Take
+    static if (hasSlicing!R && hasLength!R)
     {
-        mixin .Take!(R, false);
+        if (range.length <= n)
-
+            return range;
-        static if (hasSlicing!R)
+        else
-        {
+            return range[0 .. n];
-            auto opSlice(size_t i, size_t j)
+    }
-            in
+    // Special case: take(take(...), n)
-            {
+    else static if (is(Range == Take!(RRange, exact), RRange, bool exact))
-                assert(i <= j);
+    {
-                assert(j <= length);
+        if (n > range.length_)
-            }
+            n = range.length_;
-            do
+        static if (exact)
-            {
+            // `take(takeExactly(r, n0), n)` is rewritten `takeExactly(r, min(n0, n))`.
-                return typeof(this)(this.source[i .. j], length);
+            return Take!(RRange, true)(range.source, n);
-            }
+        else
-        }
+            // `take(take(r, n0), n)` is rewritten `take(r, min(n0, n))`.
+            return Take!(RRange, false)(range.source, n);
+    }
+    else static if (isInfinite!R)
+    {
+        // If the range is infinite then `take` is the same as `takeExactly`.
+        return Take!(R, true)(range, n);
+    }
+    else
+    {
+        return Take!(R, false)(range, n);
     }
-    return Take(range, n);
 }
 
 ///
@@ -315,13 +340,15 @@ if (isInputRange!R)
     {
         return range[0 .. n];
     }
+    // Special case: takeExactly(take(range, ...), n) is takeExactly(range, n)
+    else static if (is(Range == Take!(RRange, exact), RRange, bool exact))
+    {
+        assert(n <= range.length_);
+        return Take!(RRange, true)(range.source, n);
+    }
     else
     {
-        static struct TakeExactly
+        return Take!(R, true)(range, n);
-        {
-            mixin Take!(R, true);
-        }
-        return TakeExactly(range, n);
     }
 }