Merge pull request #142 from yssamtu/change_tuple_of_arrays

support longer tuple

Author: fverdugo

src/primitives.jl

@@ -68,6 +68,26 @@ function tuple_of_arrays(a)
         t1,t2 = tuple_of_arrays(y)
         (x,t1,t2)
     end
+    function take(a, ::Type{Tuple{A,B,C,D}} where {A,B,C,D})
+        x, y = first_and_tail(a)
+        t1, t2, t3 = tuple_of_arrays(y)
+        (x, t1, t2, t3)
+    end
+    function take(a, ::Type{Tuple{A,B,C,D,E}} where {A,B,C,D,E})
+        x, y = first_and_tail(a)
+        t1, t2, t3, t4 = tuple_of_arrays(y)
+        (x, t1, t2, t3, t4)
+    end
+    function take(a, ::Type{Tuple{A,B,C,D,E,F}} where {A,B,C,D,E,F})
+        x, y = first_and_tail(a)
+        t1, t2, t3, t4, t5 = tuple_of_arrays(y)
+        (x, t1, t2, t3, t4, t5)
+    end
+    function take(a, ::Type{Tuple{A,B,C,D,E,F,G}} where {A,B,C,D,E,F,G})
+        x, y = first_and_tail(a)
+        t1, t2, t3, t4, t5, t6 = tuple_of_arrays(y)
+        (x, t1, t2, t3, t4, t5, t6)
+    end
     # this one is type instable, why? Previous methods for take are for circumvent this issue.
     function take(a,::Type)
         x, y = first_and_tail(a)