com.sk89q.worldedit.util.collection.MoreStreams Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of FastAsyncWorldEdit-Core Show documentation
Show all versions of FastAsyncWorldEdit-Core Show documentation
Blazingly fast Minecraft world manipulation for artists, builders and everyone else.
/*
* WorldEdit, a Minecraft world manipulation toolkit
* Copyright (C) sk89q
* Copyright (C) WorldEdit team and contributors
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see .
*/
package com.sk89q.worldedit.util.collection;
import com.google.common.collect.AbstractIterator;
import java.util.Iterator;
import java.util.Spliterator;
import java.util.Spliterators;
import java.util.function.Predicate;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;
import static java.util.Objects.requireNonNull;
/**
* Additionally stream facilities.
*/
public class MoreStreams {
/**
* Emit elements from {@code stream} until {@code predicate} returns {@code false}.
*/
public static Stream takeWhile(Stream stream, Predicate predicate) {
return takeUntil(stream, predicate.negate());
}
/**
* Emit elements from {@code stream} until {@code predicate} returns {@code true}.
*/
public static Stream takeUntil(Stream stream, Predicate predicate) {
Spliterator spliterator = stream.spliterator();
Iterator iter = new AbstractIterator() {
private Iterator source = Spliterators.iterator(spliterator);
@Override
protected T computeNext() {
Iterator src = requireNonNull(source);
if (!src.hasNext()) {
return done();
}
T next = src.next();
if (predicate.test(next)) {
return done();
}
return next;
}
private T done() {
// allow GC of source
source = null;
return endOfData();
}
};
int chars = spliterator.characteristics();
// Not SIZED, Not SUBSIZED
chars &= ~(Spliterator.SIZED | Spliterator.SUBSIZED);
return StreamSupport.stream(Spliterators.spliterator(
iter, spliterator.estimateSize(), chars
), stream.isParallel()).onClose(stream::close);
}
private MoreStreams() {
}
}