(en, processor);
}
/**
* Filters some elements out from the input enumeration.
* Just make the
* {@link Processor} return null
. Please notice the toAdd
* argument of the processor is always null
.
*
* Example to remove all objects that are not strings:
*
* Processor onlyString = new Processor() {
* public Object process(Object obj, Collection alwaysNull) {
* if (obj instanceof String) {
* return obj;
* } else {
* return null;
* }
* }
* };
* Enumeration strings = Enumerations.filter(elems, onlyString);
*
*
* @param en enumeration of any objects
* @param filter a callback processor for the elements (its toAdd arguments is always null)
* @return new enumeration which does not include non-processed (returned null from processor) elements
* @see NbCollections#checkedEnumerationByFilter
*/
public static Enumeration filter(Enumeration extends T> en, Processor filter) {
return new FilEn(en, filter);
}
/**
* Support for breadth-first enumerating.
* Before any element is returned
* for the resulting enumeration it is processed in the {@link Processor} and
* the processor is allowed to modify it and also add additional elements
* at the (current) end of the queue
by calling toAdd.add
* or toAdd.addAll
. No other methods can be called on the
* provided toAdd
collection.
*
* Example of doing breadth-first walk through a tree:
*
* Processor queueSubnodes = new Processor() {
* public Object process(Object obj, Collection toAdd) {
* Node n = (Node)obj;
* toAdd.addAll (n.getChildrenList());
* return n;
* }
* };
* Enumeration strings = Enumerations.queue(elems, queueSubnodes);
*
*
* @param en initial content of the resulting enumeration
* @param filter the processor that is called for each element and can
* add and addAll elements to its toAdd Collection argument and
* also change the value to be returned
* @return enumeration with the initial and queued content (it can contain
* null
if the filter returned null
from its
* {@link Processor#process} method.
*/
public static Enumeration queue(Enumeration extends T> en, Processor filter) {
QEn q = new QEn(filter);
while (en.hasMoreElements()) {
q.put(en.nextElement());
}
return q;
}
/**
* Processor interface that can filter out objects from the enumeration,
* change them or add aditional objects to the end of the current enumeration.
*/
public static interface Processor {
/** @param original the object that is going to be returned from the enumeration right now
* @return a replacement for this object
* @param toAdd can be non-null if one can add new objects at the end of the enumeration
*/
public R process(T original, Collection toAdd);
}
/** Altering enumeration implementation */
private static final class AltEn extends Object implements Enumeration {
/** enumeration to filter */
private Enumeration extends T> en;
/** map to alter */
private Processor process;
/**
* @param en enumeration to filter
*/
public AltEn(Enumeration extends T> en, Processor process) {
this.en = en;
this.process = process;
}
/** @return true if there is more elements in the enumeration
*/
public boolean hasMoreElements() {
return en.hasMoreElements();
}
/** @return next object in the enumeration
* @exception NoSuchElementException can be thrown if there is no next object
* in the enumeration
*/
public R nextElement() {
return process.process(en.nextElement(), null);
}
}
// end of AltEn
/** QueueEnumeration
*/
private static class QEn extends Object implements Enumeration {
/** next object to be returned */
private ListItem next = null;
/** last object in the queue */
private ListItem last = null;
/** processor to use */
private Processor processor;
public QEn(Processor p) {
this.processor = p;
}
/** Put adds new object to the end of queue.
* @param o the object to add
*/
public void put(T o) {
if (last != null) {
ListItem li = new ListItem(o);
last.next = li;
last = li;
} else {
next = last = new ListItem(o);
}
}
/** Adds array of objects into the queue.
* @param arr array of objects to put into the queue
*/
public void put(Collection extends T> arr) {
for (T e : arr) {
put(e);
}
}
/** Is there any next object?
* @return true if there is next object, false otherwise
*/
public boolean hasMoreElements() {
return next != null;
}
/** @return next object in enumeration
* @exception NoSuchElementException if there is no next object
*/
public R nextElement() {
if (next == null) {
throw new NoSuchElementException();
}
T res = next.object;
if ((next = next.next) == null) {
last = null;
}
;
ToAdd toAdd = new ToAdd(this);
R out = processor.process(res, toAdd);
toAdd.finish();
return out;
}
/** item in linked list of Objects */
private static final class ListItem {
T object;
ListItem next;
/** @param o the object for this item */
ListItem(T o) {
object = o;
}
}
/** Temporary collection that supports only add and addAll operations*/
private static final class ToAdd extends Object implements Collection {
private QEn q;
public ToAdd(QEn q) {
this.q = q;
}
public void finish() {
this.q = null;
}
public boolean add(T o) {
q.put(o);
return true;
}
public boolean addAll(Collection extends T> c) {
q.put(c);
return true;
}
private String msg() {
return "Only add and addAll are implemented"; // NOI18N
}
public void clear() {
throw new UnsupportedOperationException(msg());
}
public boolean contains(Object o) {
throw new UnsupportedOperationException(msg());
}
public boolean containsAll(Collection c) {
throw new UnsupportedOperationException(msg());
}
public boolean isEmpty() {
throw new UnsupportedOperationException(msg());
}
public Iterator iterator() {
throw new UnsupportedOperationException(msg());
}
public boolean remove(Object o) {
throw new UnsupportedOperationException(msg());
}
public boolean removeAll(Collection c) {
throw new UnsupportedOperationException(msg());
}
public boolean retainAll(Collection c) {
throw new UnsupportedOperationException(msg());
}
public int size() {
throw new UnsupportedOperationException(msg());
}
public Object[] toArray() {
throw new UnsupportedOperationException(msg());
}
public X[] toArray(X[] a) {
throw new UnsupportedOperationException(msg());
}
}
// end of ToAdd
}
// end of QEn
/** Filtering enumeration */
private static final class FilEn extends Object implements Enumeration {
/** marker object stating there is no nexte element prepared */
private static final Object EMPTY = new Object();
/** enumeration to filter */
private Enumeration extends T> en;
/** element to be returned next time or {@link #EMPTY} if there is
* no such element prepared */
private R next = empty();
/** the set to use as filter */
private Processor filter;
/**
* @param en enumeration to filter
*/
public FilEn(Enumeration extends T> en, Processor filter) {
this.en = en;
this.filter = filter;
}
/** @return true if there is more elements in the enumeration
*/
public boolean hasMoreElements() {
if (next != empty()) {
// there is a object already prepared
return true;
}
while (en.hasMoreElements()) {
// read next
next = filter.process(en.nextElement(), null);
if (next != null) {
// if the object is accepted
return true;
}
;
}
next = empty();
return false;
}
/** @return next object in the enumeration
* @exception NoSuchElementException can be thrown if there is no next object
* in the enumeration
*/
public R nextElement() {
if ((next == EMPTY) && !hasMoreElements()) {
throw new NoSuchElementException();
}
R res = next;
next = empty();
return res;
}
@SuppressWarnings("unchecked")
private R empty() {
return (R)EMPTY;
}
}
// end of FilEn
/** Returns true from contains if object is not null */
private static class RNulls implements Processor {
public T process(T original, Collection toAdd) {
return original;
}
}
// end of RNulls
}