1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
package org.zwobble.couscous.util;
import com.google.common.collect.Iterables;
import java.util.Iterator;
import java.util.function.BiFunction;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;
import static org.zwobble.couscous.util.ExtraLists.list;
public class ExtraIterables {
private ExtraIterables() {}
public static <T, U> void forEach(Iterable<? extends T> first, Iterable<? extends U> second, Action2<T, U> action) {
Iterator<? extends T> firstIterator = first.iterator();
Iterator<? extends U> secondIterator = second.iterator();
while (firstIterator.hasNext()) {
action.run(firstIterator.next(), secondIterator.next());
}
if (secondIterator.hasNext()) {
throw new RuntimeException("Iterables not of same length");
}
}
public static <T1, T2, R> Iterable<R> map(Iterable<T1> iterable1, Iterable<T2> iterable2, BiFunction<T1, T2, R> function) {
return new Iterable<R>() {
@Override
public Iterator<R> iterator() {
Iterator<T1> iterator1 = iterable1.iterator();
Iterator<T2> iterator2 = iterable2.iterator();
return map(iterator1, iterator2, function);
}
};
}
private static <R, T1, T2> Iterator<R> map(Iterator<T1> iterator1, Iterator<T2> iterator2, BiFunction<T1, T2, R> function) {
return new Iterator<R>() {
@Override
public boolean hasNext() {
return iterator1.hasNext() && iterator2.hasNext();
}
@Override
public R next() {
return function.apply(iterator1.next(), iterator2.next());
}
};
}
public static <T> Iterable<T> cast(Class<T> type, Iterable<?> iterable) {
return new Iterable<T>() {
@Override
public Iterator<T> iterator() {
return cast(type, iterable.iterator());
}
};
}
private static <T> Iterator<T> cast(Class<T> type, Iterator<?> iterator) {
return new Iterator<T>() {
@Override
public boolean hasNext() {
return iterator.hasNext();
}
@Override
public T next() {
Object next = iterator.next();
if (type.isInstance(next)) {
return (T) next;
} else {
throw new ClassCastException();
}
}
};
}
public static <T> Iterable<T> lazyAppend(Iterable<T> head, T tail) {
return Iterables.concat(head, list(tail));
}
public static <T> Iterable<T> lazyCons(T head, Iterable<T> tails) {
return Iterables.concat(list(head), tails);
}
public static <T, R> Iterable<R> lazyMap(Iterable<T> iterable, Function<T, R> function) {
return new Iterable<R>() {
@Override
public Iterator<R> iterator() {
return map(iterable.iterator(), function);
}
};
}
private static <T, R> Iterator<R> map(Iterator<T> iterator, Function<T, R> function) {
return new Iterator<R>() {
@Override
public boolean hasNext() {
return iterator.hasNext();
}
@Override
public R next() {
return function.apply(iterator.next());
}
};
}
public static <T, R> Iterable<R> lazyFlatMap(Iterable<T> iterable, Function<T, Iterable<R>> function) {
return iterable(() -> stream(iterable).flatMap(element -> stream(function.apply(element))));
}
public static <T> Stream<T> stream(Iterable<T> iterable) {
return StreamSupport.stream(iterable.spliterator(), false);
}
public static <T, E> Iterable<T> iterable(Supplier<Stream<T>> stream) {
return new Iterable<T>() {
@Override
public Iterator<T> iterator() {
return stream.get().iterator();
}
};
}
}
|
