summaryrefslogtreecommitdiff
path: root/JPratt/src/main/java/com/ashardalon/pratt/tokens/SimpleTokenStream.java
blob: cc751d1fcc6cdc57636e78ed04e4c06ca3b21b8c (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
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
package com.ashardalon.pratt.tokens;

import java.util.Iterator;

import bjc.data.MarkListIterator;

/**
 * Simple token stream implementation
 * @author bjcul
 *
 * @param <K> The key type
 * @param <V> The value type
 */
public class SimpleTokenStream<K, V> extends TokenStream<K, V> {
	private final MarkListIterator<Token<K, V>> iter;

	private Token<K, V> curr;

	private Token<K, V> terminal;
	/**
	 * Create a new token stream from a iterator.
	 *
	 * @param itr The iterator to use.
	 * @param terminal The terminal token to use for end-of-stream
	 *
	 */
	public SimpleTokenStream(final Iterator<Token<K, V>> itr, Token<K, V> terminal) {
		this.iter = new MarkListIterator<>(itr);
		this.terminal = terminal;
	}

	@Override
	public Token<K, V> current() {
		// Prime stream if necessary
		if (curr == null)
			return next();
		return curr;
	}

	@Override
	public Token<K, V> next() {
		if (iter.hasNext()) {
			curr = iter.next();
		} else {
			curr = terminal;
		}

		return curr;
	}

	@Override
	public boolean hasNext() {
		return iter.hasNext();
	}

	@Override
	public void mark() {
		iter.mark();
	}

	@Override
	public void commit() {
		iter.commit();

		if (!iter.hasMark()) {
			// No marks outstanding; we can release the previous state
			iter.reset();
		}
	}

	@Override
	public void rollback() {
		iter.rollback();

		curr = iter.current();
	}

	@Override
	public boolean hasMark() {
		return iter.hasMark();
	}
}