summaryrefslogtreecommitdiff
path: root/base/src/main/java/bjc/utils/data/GeneratingIterator.java
blob: 9abca7c32da04fc28ca2a7359b42d9f0556d5a75 (plain)
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
package bjc.utils.data;

import java.util.Iterator;
import java.util.function.Predicate;
import java.util.function.UnaryOperator;

/**
 * An iterator that generates a series of elements from a single element.
 * 
 * @author bjculkin
 *
 * @param <E>
 *                The type of element generated.
 */
public class GeneratingIterator<E> implements Iterator<E> {
	private E state;

	private UnaryOperator<E> transtion;

	private Predicate<E> stpper;

	/**
	 * Create a new generative iterator.
	 * 
	 * @param initial
	 *                The initial state of the generator.
	 * 
	 * @param transition
	 *                The function to apply to the state.
	 * 
	 * @param stopper
	 *                The predicate applied to the current state to
	 *                determine when to stop.
	 */
	public GeneratingIterator(E initial, UnaryOperator<E> transition, Predicate<E> stopper) {
		state = initial;
		transtion = transition;
		stpper = stopper;
	}

	@Override
	public boolean hasNext() {
		return stpper.test(state);
	}

	@Override
	public E next() {
		state = transtion.apply(state);

		return state;
	}

}