Wed, 12 Sep 2018 10:58:46 +0100
main, scansion.console: Refactor (changes event data format) so that test results include test name
39
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
1 | -- Prosody IM |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
2 | -- Copyright (C) 2008-2010 Matthew Wild |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
3 | -- Copyright (C) 2008-2010 Waqas Hussain |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
4 | -- |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
5 | -- This project is MIT/X11 licensed. Please see the |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
6 | -- COPYING file in the source package for more information. |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
7 | -- |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
8 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
9 | --[[ Iterators ]]-- |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
10 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
11 | local it = {}; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
12 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
13 | local t_insert = table.insert; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
14 | local select, unpack, next = select, unpack, next; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
15 | local function pack(...) return { n = select("#", ...), ... }; end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
16 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
17 | -- Reverse an iterator |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
18 | function it.reverse(f, s, var) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
19 | local results = {}; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
20 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
21 | -- First call the normal iterator |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
22 | while true do |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
23 | local ret = { f(s, var) }; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
24 | var = ret[1]; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
25 | if var == nil then break; end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
26 | t_insert(results, 1, ret); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
27 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
28 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
29 | -- Then return our reverse one |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
30 | local i,max = 0, #results; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
31 | return function (results) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
32 | if i<max then |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
33 | i = i + 1; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
34 | return unpack(results[i]); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
35 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
36 | end, results; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
37 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
38 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
39 | -- Iterate only over keys in a table |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
40 | local function _keys_it(t, key) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
41 | return (next(t, key)); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
42 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
43 | function it.keys(t) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
44 | return _keys_it, t; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
45 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
46 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
47 | -- Iterate only over values in a table |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
48 | function it.values(t) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
49 | local key, val; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
50 | return function (t) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
51 | key, val = next(t, key); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
52 | return val; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
53 | end, t; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
54 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
55 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
56 | -- Given an iterator, iterate only over unique items |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
57 | function it.unique(f, s, var) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
58 | local set = {}; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
59 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
60 | return function () |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
61 | while true do |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
62 | local ret = pack(f(s, var)); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
63 | var = ret[1]; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
64 | if var == nil then break; end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
65 | if not set[var] then |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
66 | set[var] = true; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
67 | return unpack(ret, 1, ret.n); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
68 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
69 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
70 | end; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
71 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
72 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
73 | --[[ Return the number of items an iterator returns ]]-- |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
74 | function it.count(f, s, var) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
75 | local x = 0; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
76 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
77 | while true do |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
78 | var = f(s, var); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
79 | if var == nil then break; end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
80 | x = x + 1; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
81 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
82 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
83 | return x; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
84 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
85 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
86 | -- Return the first n items an iterator returns |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
87 | function it.head(n, f, s, var) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
88 | local c = 0; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
89 | return function (s, var) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
90 | if c >= n then |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
91 | return nil; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
92 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
93 | c = c + 1; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
94 | return f(s, var); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
95 | end, s; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
96 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
97 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
98 | -- Skip the first n items an iterator returns |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
99 | function it.skip(n, f, s, var) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
100 | for i=1,n do |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
101 | var = f(s, var); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
102 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
103 | return f, s, var; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
104 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
105 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
106 | -- Return the last n items an iterator returns |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
107 | function it.tail(n, f, s, var) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
108 | local results, count = {}, 0; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
109 | while true do |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
110 | local ret = pack(f(s, var)); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
111 | var = ret[1]; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
112 | if var == nil then break; end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
113 | results[(count%n)+1] = ret; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
114 | count = count + 1; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
115 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
116 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
117 | if n > count then n = count; end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
118 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
119 | local pos = 0; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
120 | return function () |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
121 | pos = pos + 1; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
122 | if pos > n then return nil; end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
123 | local ret = results[((count-1+pos)%n)+1]; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
124 | return unpack(ret, 1, ret.n); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
125 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
126 | --return reverse(head(n, reverse(f, s, var))); -- ! |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
127 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
128 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
129 | function it.filter(filter, f, s, var) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
130 | if type(filter) ~= "function" then |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
131 | local filter_value = filter; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
132 | function filter(x) return x ~= filter_value; end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
133 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
134 | return function (s, var) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
135 | local ret; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
136 | repeat ret = pack(f(s, var)); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
137 | var = ret[1]; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
138 | until var == nil or filter(unpack(ret, 1, ret.n)); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
139 | return unpack(ret, 1, ret.n); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
140 | end, s, var; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
141 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
142 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
143 | local function _ripairs_iter(t, key) if key > 1 then return key-1, t[key-1]; end end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
144 | function it.ripairs(t) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
145 | return _ripairs_iter, t, #t+1; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
146 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
147 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
148 | local function _range_iter(max, curr) if curr < max then return curr + 1; end end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
149 | function it.range(x, y) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
150 | if not y then x, y = 1, x; end -- Default to 1..x if y not given |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
151 | return _range_iter, y, x-1; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
152 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
153 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
154 | -- Convert the values returned by an iterator to an array |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
155 | function it.to_array(f, s, var) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
156 | local t, var = {}; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
157 | while true do |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
158 | var = f(s, var); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
159 | if var == nil then break; end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
160 | t_insert(t, var); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
161 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
162 | return t; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
163 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
164 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
165 | -- Treat the return of an iterator as key,value pairs, |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
166 | -- and build a table |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
167 | function it.to_table(f, s, var) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
168 | local t, var2 = {}; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
169 | while true do |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
170 | var, var2 = f(s, var); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
171 | if var == nil then break; end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
172 | t[var] = var2; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
173 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
174 | return t; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
175 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
176 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
177 | function combinations(iterator, ...) |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
178 | local n_inputs = select("#", ...); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
179 | local inputs = { ... }; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
180 | local f, s, var = {}, {}, {}; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
181 | for i = 1, n_inputs do |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
182 | f[i], s[i], var[i] = iterator(inputs[i]); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
183 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
184 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
185 | local level = 1; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
186 | return function () |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
187 | while level > 0 do |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
188 | var[level] = f[level](s[level], var[level]); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
189 | if level == n_inputs and var[level] ~= nil then |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
190 | return unpack(var, 1, n_inputs); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
191 | elseif var[level] then |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
192 | level = level + 1; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
193 | f[level], s[level], var[level] = iterator(inputs[level]); |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
194 | else |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
195 | level = level - 1; |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
196 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
197 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
198 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
199 | end |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
200 | |
814b06660220
scansion.iterators: Import util.iterators from Prosody, add combinations() iterator
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
201 | return it; |