net.sf.staccatocommons.collections.stream.impl.internal.FlatMapStream Maven / Gradle / Ivy

Go to download
/**
 *  Copyright (c) 2011, The Staccato-Commons Team
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as published by
 *  the Free Software Foundation; version 3 of the License.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Lesser General Public License for more details.
 */

package net.sf.staccatocommons.collections.stream.impl.internal;

import java.util.Iterator;
import java.util.NoSuchElementException;

import net.sf.staccatocommons.collections.stream.AbstractStream;
import net.sf.staccatocommons.collections.stream.Stream;
import net.sf.staccatocommons.defs.Applicable;
import net.sf.staccatocommons.iterators.thriter.NextThriterator;
import net.sf.staccatocommons.iterators.thriter.Thriterator;
import net.sf.staccatocommons.restrictions.check.NonNull;

/**
 * @author flbulgarelli
 * 
 * @param 
 * @param 
 */
public final class FlatMapStream extends AbstractStream {
  private final Stream stream;
  private final Applicable> function;

  /**
   * Creates a new {@link FlatMapStream}
   */
  public FlatMapStream(@NonNull Stream stream,
    @NonNull Applicable> function) {
    this.stream = stream;
    this.function = function;
  }

  public Thriterator iterator() {

    final Iterator iter = stream.iterator();
    return new NextThriterator() {
      private Iterator subIter = null;

      public boolean hasNext() {
        if (subIter != null && subIter.hasNext())
          return true;
        while (iter.hasNext()) {
          subIter = function.apply(iter.next()).iterator();
          if (subIter.hasNext())
            return true;
        }
        return false;
      }

      public B nextImpl() {
        if (!hasNext())
          throw new NoSuchElementException();
        return subIter.next();
      }
    };
  }
}