/*
* This is free and unencumbered software released into the public domain.
*
* Anyone is free to copy, modify, publish, use, compile, sell, or
* distribute this software, either in source code form or as a compiled
* binary, for any purpose, commercial or non-commercial, and by any
* means.
*
* In jurisdictions that recognize copyright laws, the author or authors
* of this software dedicate any and all copyright interest in the
* software to the public domain. We make this dedication for the benefit
* of the public at large and to the detriment of our heirs and
* successors. We intend this dedication to be an overt act of
* relinquishment in perpetuity of all present and future rights to this
* software under copyright law.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
* IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*
* For more information, please refer to <http://unlicense.org/>
*/
package jxtn.core.axi.collections;
import java.util.Iterator;
import java.util.Objects;
import java.util.function.Function;
/**
* 依照指定函數做展開的列舉器。
*
* @author AqD
* @param <T> 來源列舉項目型態
* @param <R> 展開列舉項目型態
*/
public final class ExpandedIterator<T, R> extends AbstractIterator<R> {
private final Iterator<? extends T> source;
private final Function<? super T, ? extends Iterator<? extends R>> expand;
private Iterator<? extends R> currentChildren;
/**
* 建立指定函數做展開的列舉器。
* <p>
* {@link MappedIterator}會依照{@code expand}將每個{@code source}的項目做展開。
* </p>
*
* @param source 來源列舉器
* @param expand 展開函數
*/
public ExpandedIterator(Iterator<? extends T> source, Function<? super T, ? extends Iterator<? extends R>> expand) {
this.source = Objects.requireNonNull(source);
this.expand = Objects.requireNonNull(expand);
}
@Override
protected R fetchNext() {
if (this.isAtHead()) {
if (!this.source.hasNext()) {
return this.end();
}
T parent = this.source.next();
this.currentChildren = this.expand.apply(parent);
}
while (true) {
if (this.currentChildren.hasNext()) {
return this.currentChildren.next();
}
this.currentChildren = null;
if (!this.source.hasNext()) {
return this.end();
}
T parent = this.source.next();
this.currentChildren = this.expand.apply(parent);
}
}
}