blob: 083f535b4aa5af05793d903cfb387979b2117428 (
plain) (
blame)
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
|
/*******************************************************************************
* Copyright (c) 2005, 2007 Oracle. All rights reserved.
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v1.0, which accompanies this distribution
* and is available at http://www.eclipse.org/legal/epl-v10.html.
*
* Contributors:
* Oracle - initial API and implementation
******************************************************************************/
package org.eclipse.jpt.utility.internal.iterators;
import java.util.Collection;
import java.util.Iterator;
import java.util.NoSuchElementException;
import org.eclipse.jpt.utility.internal.StringTools;
/**
* A <code>CompositeIterator</code> wraps a collection
* of <code>Iterator</code>s and makes them appear to be a single
* <code>Iterator</code>.
*/
public class CompositeIterator<E>
implements Iterator<E>
{
private final Iterator<? extends Iterator<? extends E>> iterators;
private Iterator<? extends E> currentIterator;
private Iterator<? extends E> lastIteratorToReturnNext;
// ********** constructors **********
/**
* Construct an iterator with the specified collection of iterators.
*/
public CompositeIterator(Collection<? extends Iterator<? extends E>> iterators) {
this(iterators.iterator());
}
/**
* Construct an iterator with the specified collection of iterators.
*/
public CompositeIterator(Iterator<? extends Iterator<? extends E>> iterators) {
super();
this.iterators = iterators;
}
/**
* Construct an iterator with the specified object prepended
* to the specified iterator.
*/
@SuppressWarnings("unchecked")
public CompositeIterator(E object, Iterator<? extends E> iterator) {
this(new SingleElementIterator<E>(object), iterator);
}
/**
* Construct an iterator with the specified object appended
* to the specified iterator.
*/
@SuppressWarnings("unchecked")
public CompositeIterator(Iterator<? extends E> iterator, E object) {
this(iterator, new SingleElementIterator<E>(object));
}
/**
* Construct an iterator with the specified iterators.
*/
public CompositeIterator(Iterator<? extends E>... iterators) {
this(new ArrayIterator<Iterator<? extends E>>(iterators));
}
// ********** Iterator implementation **********
public boolean hasNext() {
try {
this.loadCurrentIterator();
} catch (NoSuchElementException ex) {
// this occurs if there are no iterators at all
return false;
}
return this.currentIterator.hasNext();
}
public E next() {
this.loadCurrentIterator();
E result = this.currentIterator.next();
// the statement above will throw a NoSuchElementException
// if the current iterator is at the end of the line;
// so if we get here, we can set 'lastIteratorToReturnNext'
this.lastIteratorToReturnNext = this.currentIterator;
return result;
}
public void remove() {
if (this.lastIteratorToReturnNext == null) {
// CompositeIterator#next() has never been called
throw new IllegalStateException();
}
this.lastIteratorToReturnNext.remove();
}
/**
* Load currentIterator with the first iterator that <code>hasNext()</code>
* or the final iterator if all the elements have already been retrieved.
*/
private void loadCurrentIterator() {
if (this.currentIterator == null) {
this.currentIterator = this.iterators.next();
}
while (( ! this.currentIterator.hasNext()) && this.iterators.hasNext()) {
this.currentIterator = this.iterators.next();
}
}
// ********** overrides **********
@Override
public String toString() {
return StringTools.buildToStringFor(this, this.iterators);
}
}
|