Please wait. This can take some minutes ...
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.google.gwt.emul.java.util.stream.LongStream Maven / Gradle / Ivy
/*
* Copyright 2016 Google Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
* use this file except in compliance with the License. You may obtain a copy of
* the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations under
* the License.
*/
package java.util.stream;
import static javaemul.internal.InternalPreconditions.checkNotNull;
import static javaemul.internal.InternalPreconditions.checkState;
import java.util.Arrays;
import java.util.Comparator;
import java.util.HashSet;
import java.util.LongSummaryStatistics;
import java.util.OptionalDouble;
import java.util.OptionalLong;
import java.util.PrimitiveIterator;
import java.util.Spliterator;
import java.util.Spliterators;
import java.util.Spliterators.AbstractLongSpliterator;
import java.util.function.BiConsumer;
import java.util.function.Consumer;
import java.util.function.DoubleConsumer;
import java.util.function.IntConsumer;
import java.util.function.LongBinaryOperator;
import java.util.function.LongConsumer;
import java.util.function.LongFunction;
import java.util.function.LongPredicate;
import java.util.function.LongSupplier;
import java.util.function.LongToDoubleFunction;
import java.util.function.LongToIntFunction;
import java.util.function.LongUnaryOperator;
import java.util.function.ObjLongConsumer;
import java.util.function.Supplier;
/**
* See
* the official Java API doc for details.
*/
public interface LongStream extends BaseStream {
/**
* Value holder for various stream operations.
*/
static final class ValueConsumer implements LongConsumer {
long value;
@Override
public void accept(long value) {
this.value = value;
}
}
static LongStream.Builder builder() {
return new Builder() {
private long[] items = new long[0];
@Override
public void accept(long t) {
checkState(items != null, "Builder already built");
items[items.length] = t;
}
@Override
public LongStream build() {
checkState(items != null, "Builder already built");
LongStream stream = Arrays.stream(items);
items = null;
return stream;
}
};
}
static LongStream concat(LongStream a, LongStream b) {
// This is nearly the same as flatMap, but inlined, wrapped around a single spliterator of
// these two objects, and without close() called as the stream progresses. Instead, close is
// invoked as part of the resulting stream's own onClose, so that either can fail without
// affecting the other, and correctly collecting suppressed exceptions.
// TODO replace this flatMap-ish spliterator with one that directly combines the two root
// streams
Spliterator extends LongStream> spliteratorOfStreams = Arrays.asList(a, b).spliterator();
AbstractLongSpliterator spliterator =
new Spliterators.AbstractLongSpliterator(Long.MAX_VALUE, 0) {
Spliterator.OfLong next;
@Override
public boolean tryAdvance(LongConsumer action) {
// look for a new spliterator
while (advanceToNextSpliterator()) {
// if we have one, try to read and use it
if (next.tryAdvance(action)) {
return true;
} else {
// failed, null it out so we can find another
next = null;
}
}
return false;
}
private boolean advanceToNextSpliterator() {
while (next == null) {
if (!spliteratorOfStreams.tryAdvance(
n -> {
if (n != null) {
next = n.spliterator();
}
})) {
return false;
}
}
return true;
}
};
LongStream result = new LongStreamSource(null, spliterator);
result.onClose(a::close);
result.onClose(b::close);
return result;
}
static LongStream empty() {
return new EmptyLongStreamSource(null);
}
static LongStream generate(LongSupplier s) {
AbstractLongSpliterator spltierator =
new Spliterators.AbstractLongSpliterator(
Long.MAX_VALUE, Spliterator.IMMUTABLE | Spliterator.ORDERED) {
@Override
public boolean tryAdvance(LongConsumer action) {
action.accept(s.getAsLong());
return true;
}
};
return StreamSupport.longStream(spltierator, false);
}
static LongStream iterate(long seed, LongUnaryOperator f) {
AbstractLongSpliterator spliterator =
new Spliterators.AbstractLongSpliterator(
Long.MAX_VALUE, Spliterator.IMMUTABLE | Spliterator.ORDERED) {
private long next = seed;
@Override
public boolean tryAdvance(LongConsumer action) {
action.accept(next);
next = f.applyAsLong(next);
return true;
}
};
return StreamSupport.longStream(spliterator, false);
}
static LongStream of(long... values) {
return Arrays.stream(values);
}
static LongStream of(long t) {
// TODO consider a splittable that returns only a single value
return of(new long[] {t});
}
static LongStream range(long startInclusive, long endExclusive) {
if (startInclusive >= endExclusive) {
return empty();
}
return rangeClosed(startInclusive, endExclusive - 1);
}
static LongStream rangeClosed(long startInclusive, long endInclusive) {
if (startInclusive > endInclusive) {
return empty();
}
long count = endInclusive - startInclusive + 1;
AbstractLongSpliterator spliterator =
new Spliterators.AbstractLongSpliterator(
count,
Spliterator.IMMUTABLE
| Spliterator.SIZED
| Spliterator.SUBSIZED
| Spliterator.ORDERED
| Spliterator.SORTED
| Spliterator.DISTINCT) {
private long next = startInclusive;
@Override
public Comparator super Long> getComparator() {
return null;
}
@Override
public boolean tryAdvance(LongConsumer action) {
if (next <= endInclusive) {
action.accept(next++);
return true;
}
return false;
}
};
return StreamSupport.longStream(spliterator, false);
}
/**
* See
*
* the official Java API doc for details.
*/
public interface Builder extends LongConsumer {
@Override
void accept(long t);
default LongStream.Builder add(long t) {
accept(t);
return this;
}
LongStream build();
}
boolean allMatch(LongPredicate predicate);
boolean anyMatch(LongPredicate predicate);
DoubleStream asDoubleStream();
OptionalDouble average();
Stream boxed();
R collect(Supplier supplier, ObjLongConsumer accumulator, BiConsumer combiner);
long count();
LongStream distinct();
LongStream filter(LongPredicate predicate);
OptionalLong findAny();
OptionalLong findFirst();
LongStream flatMap(LongFunction extends LongStream> mapper);
void forEach(LongConsumer action);
void forEachOrdered(LongConsumer action);
@Override
PrimitiveIterator.OfLong iterator();
LongStream limit(long maxSize);
LongStream map(LongUnaryOperator mapper);
DoubleStream mapToDouble(LongToDoubleFunction mapper);
IntStream mapToInt(LongToIntFunction mapper);
Stream mapToObj(LongFunction extends U> mapper);
OptionalLong max();
OptionalLong min();
boolean noneMatch(LongPredicate predicate);
@Override
LongStream parallel();
LongStream peek(LongConsumer action);
OptionalLong reduce(LongBinaryOperator op);
long reduce(long identity, LongBinaryOperator op);
@Override
LongStream sequential();
LongStream skip(long n);
LongStream sorted();
@Override
Spliterator.OfLong spliterator();
long sum();
LongSummaryStatistics summaryStatistics();
long[] toArray();
/**
* Represents an empty stream, doing nothing for all methods.
*/
static class EmptyLongStreamSource extends TerminatableStream
implements LongStream {
public EmptyLongStreamSource(TerminatableStream> previous) {
super(previous);
}
@Override
public LongStream filter(LongPredicate predicate) {
throwIfTerminated();
return this;
}
@Override
public LongStream map(LongUnaryOperator mapper) {
throwIfTerminated();
return this;
}
@Override
public Stream mapToObj(LongFunction extends U> mapper) {
throwIfTerminated();
return new Stream.EmptyStreamSource(this);
}
@Override
public IntStream mapToInt(LongToIntFunction mapper) {
throwIfTerminated();
return new IntStream.EmptyIntStreamSource(this);
}
@Override
public DoubleStream mapToDouble(LongToDoubleFunction mapper) {
throwIfTerminated();
return new DoubleStream.EmptyDoubleStreamSource(this);
}
@Override
public LongStream flatMap(LongFunction extends LongStream> mapper) {
throwIfTerminated();
return this;
}
@Override
public LongStream distinct() {
throwIfTerminated();
return this;
}
@Override
public LongStream sorted() {
throwIfTerminated();
return this;
}
@Override
public LongStream peek(LongConsumer action) {
throwIfTerminated();
return this;
}
@Override
public LongStream limit(long maxSize) {
throwIfTerminated();
checkState(maxSize >= 0, "maxSize may not be negative");
return this;
}
@Override
public LongStream skip(long n) {
throwIfTerminated();
checkState(n >= 0, "n may not be negative");
return this;
}
@Override
public void forEach(LongConsumer action) {
terminate();
}
@Override
public void forEachOrdered(LongConsumer action) {
terminate();
}
@Override
public long[] toArray() {
terminate();
return new long[0];
}
@Override
public long reduce(long identity, LongBinaryOperator op) {
terminate();
return identity;
}
@Override
public OptionalLong reduce(LongBinaryOperator op) {
terminate();
return OptionalLong.empty();
}
@Override
public R collect(
Supplier supplier, ObjLongConsumer accumulator, BiConsumer combiner) {
terminate();
return supplier.get();
}
@Override
public long sum() {
terminate();
return 0;
}
@Override
public OptionalLong min() {
terminate();
return OptionalLong.empty();
}
@Override
public OptionalLong max() {
terminate();
return OptionalLong.empty();
}
@Override
public long count() {
terminate();
return 0;
}
@Override
public OptionalDouble average() {
terminate();
return OptionalDouble.empty();
}
@Override
public LongSummaryStatistics summaryStatistics() {
terminate();
return new LongSummaryStatistics();
}
@Override
public boolean anyMatch(LongPredicate predicate) {
terminate();
return false;
}
@Override
public boolean allMatch(LongPredicate predicate) {
terminate();
return true;
}
@Override
public boolean noneMatch(LongPredicate predicate) {
terminate();
return true;
}
@Override
public OptionalLong findFirst() {
terminate();
return OptionalLong.empty();
}
@Override
public OptionalLong findAny() {
terminate();
return OptionalLong.empty();
}
@Override
public DoubleStream asDoubleStream() {
throwIfTerminated();
return new DoubleStream.EmptyDoubleStreamSource(this);
}
@Override
public Stream boxed() {
throwIfTerminated();
return new Stream.EmptyStreamSource(this);
}
@Override
public LongStream sequential() {
throwIfTerminated();
return this;
}
@Override
public LongStream parallel() {
throwIfTerminated();
return this;
}
@Override
public PrimitiveIterator.OfLong iterator() {
return Spliterators.iterator(spliterator());
}
@Override
public Spliterator.OfLong spliterator() {
terminate();
return Spliterators.emptyLongSpliterator();
}
@Override
public boolean isParallel() {
throwIfTerminated();
return false;
}
@Override
public LongStream unordered() {
throwIfTerminated();
return this;
}
}
/**
* Long to Int map spliterator.
*/
static final class MapToIntSpliterator extends Spliterators.AbstractIntSpliterator {
private final LongToIntFunction map;
private final Spliterator.OfLong original;
public MapToIntSpliterator(LongToIntFunction map, Spliterator.OfLong original) {
super(
original.estimateSize(),
original.characteristics() & ~(Spliterator.SORTED | Spliterator.DISTINCT));
checkNotNull(map);
this.map = map;
this.original = original;
}
@Override
public boolean tryAdvance(final IntConsumer action) {
return original.tryAdvance((long u) -> action.accept(map.applyAsInt(u)));
}
}
/**
* Long to Object map spliterator.
* @param the type of data in the object spliterator
*/
static final class MapToObjSpliterator extends Spliterators.AbstractSpliterator {
private final LongFunction extends T> map;
private final Spliterator.OfLong original;
public MapToObjSpliterator(LongFunction extends T> map, Spliterator.OfLong original) {
super(
original.estimateSize(),
original.characteristics() & ~(Spliterator.SORTED | Spliterator.DISTINCT));
checkNotNull(map);
this.map = map;
this.original = original;
}
@Override
public boolean tryAdvance(final Consumer super T> action) {
return original.tryAdvance((long u) -> action.accept(map.apply(u)));
}
}
/**
* Long to Long map spliterator.
*/
static final class MapToLongSpliterator extends Spliterators.AbstractLongSpliterator {
private final LongUnaryOperator map;
private final Spliterator.OfLong original;
public MapToLongSpliterator(LongUnaryOperator map, Spliterator.OfLong original) {
super(
original.estimateSize(),
original.characteristics() & ~(Spliterator.SORTED | Spliterator.DISTINCT));
checkNotNull(map);
this.map = map;
this.original = original;
}
@Override
public boolean tryAdvance(final LongConsumer action) {
return original.tryAdvance((long u) -> action.accept(map.applyAsLong(u)));
}
}
/**
* Long to Double map Spliterator.
*/
static final class MapToDoubleSpliterator extends Spliterators.AbstractDoubleSpliterator {
private final LongToDoubleFunction map;
private final Spliterator.OfLong original;
public MapToDoubleSpliterator(LongToDoubleFunction map, Spliterator.OfLong original) {
super(
original.estimateSize(),
original.characteristics() & ~(Spliterator.SORTED | Spliterator.DISTINCT));
checkNotNull(map);
this.map = map;
this.original = original;
}
@Override
public boolean tryAdvance(final DoubleConsumer action) {
return original.tryAdvance((long u) -> action.accept(map.applyAsDouble(u)));
}
}
/**
* Long filter spliterator.
*/
static final class FilterSpliterator extends Spliterators.AbstractLongSpliterator {
private final LongPredicate filter;
private final Spliterator.OfLong original;
private boolean found;
public FilterSpliterator(LongPredicate filter, Spliterator.OfLong original) {
super(original.estimateSize(), original.characteristics() & ~Spliterator.SIZED);
checkNotNull(filter);
this.filter = filter;
this.original = original;
}
@Override
public Comparator super Long> getComparator() {
return original.getComparator();
}
@Override
public boolean tryAdvance(final LongConsumer action) {
found = false;
while (!found
&& original.tryAdvance(
(long item) -> {
if (filter.test(item)) {
found = true;
action.accept(item);
}
})) {
// do nothing, work is done in tryAdvance
}
return found;
}
}
/**
* Long skip spliterator.
*/
static final class SkipSpliterator extends Spliterators.AbstractLongSpliterator {
private long skip;
private final Spliterator.OfLong original;
public SkipSpliterator(long skip, Spliterator.OfLong original) {
super(
original.hasCharacteristics(Spliterator.SIZED)
? Math.max(0, original.estimateSize() - skip)
: Long.MAX_VALUE,
original.characteristics());
this.skip = skip;
this.original = original;
}
@Override
public Comparator super Long> getComparator() {
return original.getComparator();
}
@Override
public boolean tryAdvance(LongConsumer action) {
while (skip > 0) {
if (!original.tryAdvance((long ignore) -> { })) {
return false;
}
skip--;
}
return original.tryAdvance(action);
}
}
/**
* Long limit spliterator.
*/
static final class LimitSpliterator extends Spliterators.AbstractLongSpliterator {
private final long limit;
private final Spliterator.OfLong original;
private int position = 0;
public LimitSpliterator(long limit, Spliterator.OfLong original) {
super(
original.hasCharacteristics(Spliterator.SIZED)
? Math.min(original.estimateSize(), limit)
: Long.MAX_VALUE,
original.characteristics());
this.limit = limit;
this.original = original;
}
@Override
public Comparator super Long> getComparator() {
return original.getComparator();
}
@Override
public boolean tryAdvance(LongConsumer action) {
if (position >= limit) {
return false;
}
boolean result = original.tryAdvance(action);
position++;
return result;
}
}
/**
* Main implementation of LongStream, wrapping a single spliterator, and an optional parent
* stream.
*/
static class LongStreamSource extends TerminatableStream implements LongStream {
private final Spliterator.OfLong spliterator;
public LongStreamSource(TerminatableStream> previous, Spliterator.OfLong spliterator) {
super(previous);
this.spliterator = spliterator;
}
// terminals
@Override
public void forEach(LongConsumer action) {
forEachOrdered(action);
}
@Override
public void forEachOrdered(LongConsumer action) {
terminate();
spliterator.forEachRemaining(action);
}
@Override
public long[] toArray() {
terminate();
long[] entries = new long[0];
// this is legal in js, since the array will be backed by a JS array
spliterator.forEachRemaining((long value) -> entries[entries.length] = value);
return entries;
}
@Override
public long reduce(long identity, LongBinaryOperator op) {
terminate();
ValueConsumer holder = new ValueConsumer();
holder.value = identity;
spliterator.forEachRemaining(
(long value) -> {
holder.accept(op.applyAsLong(holder.value, value));
});
return holder.value;
}
@Override
public OptionalLong reduce(LongBinaryOperator op) {
ValueConsumer holder = new ValueConsumer();
if (spliterator.tryAdvance(holder)) {
return OptionalLong.of(reduce(holder.value, op));
}
terminate();
return OptionalLong.empty();
}
@Override
public R collect(
Supplier supplier, ObjLongConsumer accumulator, BiConsumer combiner) {
terminate();
final R acc = supplier.get();
spliterator.forEachRemaining((long value) -> accumulator.accept(acc, value));
return acc;
}
@Override
public long sum() {
return summaryStatistics().getSum();
}
@Override
public OptionalLong min() {
LongSummaryStatistics stats = summaryStatistics();
if (stats.getCount() == 0) {
return OptionalLong.empty();
}
return OptionalLong.of(stats.getMin());
}
@Override
public OptionalLong max() {
LongSummaryStatistics stats = summaryStatistics();
if (stats.getCount() == 0) {
return OptionalLong.empty();
}
return OptionalLong.of(stats.getMax());
}
@Override
public long count() {
terminate();
long count = 0;
while (spliterator.tryAdvance((long value) -> { })) {
count++;
}
return count;
}
@Override
public OptionalDouble average() {
LongSummaryStatistics stats = summaryStatistics();
if (stats.getCount() == 0) {
return OptionalDouble.empty();
}
return OptionalDouble.of(stats.getAverage());
}
@Override
public LongSummaryStatistics summaryStatistics() {
return collect(
LongSummaryStatistics::new,
// TODO switch to a lambda reference once #9340 is fixed
(longSummaryStatistics, value) -> longSummaryStatistics.accept(value),
LongSummaryStatistics::combine);
}
@Override
public boolean anyMatch(LongPredicate predicate) {
return filter(predicate).findFirst().isPresent();
}
@Override
public boolean allMatch(LongPredicate predicate) {
return !anyMatch(predicate.negate());
}
@Override
public boolean noneMatch(LongPredicate predicate) {
return !anyMatch(predicate);
}
@Override
public OptionalLong findFirst() {
terminate();
ValueConsumer holder = new ValueConsumer();
if (spliterator.tryAdvance(holder)) {
return OptionalLong.of(holder.value);
}
return OptionalLong.empty();
}
@Override
public OptionalLong findAny() {
return findFirst();
}
@Override
public PrimitiveIterator.OfLong iterator() {
return Spliterators.iterator(spliterator());
}
@Override
public Spliterator.OfLong spliterator() {
terminate();
return spliterator;
}
// end terminals
// intermediates
@Override
public LongStream filter(LongPredicate predicate) {
throwIfTerminated();
return new LongStreamSource(this, new FilterSpliterator(predicate, spliterator));
}
@Override
public LongStream map(LongUnaryOperator mapper) {
throwIfTerminated();
return new LongStream.LongStreamSource(this, new MapToLongSpliterator(mapper, spliterator));
}
@Override
public Stream mapToObj(LongFunction extends U> mapper) {
throwIfTerminated();
return new Stream.StreamSource(this, new MapToObjSpliterator(mapper, spliterator));
}
@Override
public IntStream mapToInt(LongToIntFunction mapper) {
throwIfTerminated();
return new IntStream.IntStreamSource(this, new MapToIntSpliterator(mapper, spliterator));
}
@Override
public DoubleStream mapToDouble(LongToDoubleFunction mapper) {
throwIfTerminated();
return new DoubleStream.DoubleStreamSource(
this, new MapToDoubleSpliterator(mapper, spliterator));
}
@Override
public LongStream flatMap(LongFunction extends LongStream> mapper) {
throwIfTerminated();
final Spliterator extends LongStream> spliteratorOfStreams =
new MapToObjSpliterator<>(mapper, spliterator);
AbstractLongSpliterator flatMapSpliterator =
new Spliterators.AbstractLongSpliterator(Long.MAX_VALUE, 0) {
LongStream nextStream;
Spliterator.OfLong next;
@Override
public boolean tryAdvance(LongConsumer action) {
// look for a new spliterator
while (advanceToNextSpliterator()) {
// if we have one, try to read and use it
if (next.tryAdvance(action)) {
return true;
} else {
nextStream.close();
nextStream = null;
// failed, null it out so we can find another
next = null;
}
}
return false;
}
private boolean advanceToNextSpliterator() {
while (next == null) {
if (!spliteratorOfStreams.tryAdvance(
n -> {
if (n != null) {
nextStream = n;
next = n.spliterator();
}
})) {
return false;
}
}
return true;
}
};
return new LongStreamSource(this, flatMapSpliterator);
}
@Override
public LongStream distinct() {
throwIfTerminated();
HashSet seen = new HashSet<>();
return filter(seen::add);
}
@Override
public LongStream sorted() {
throwIfTerminated();
AbstractLongSpliterator sortedSpliterator =
new Spliterators.AbstractLongSpliterator(
spliterator.estimateSize(), spliterator.characteristics() | Spliterator.SORTED) {
Spliterator.OfLong ordered = null;
@Override
public Comparator super Long> getComparator() {
return null;
}
@Override
public boolean tryAdvance(LongConsumer action) {
if (ordered == null) {
long[] list = new long[0];
spliterator.forEachRemaining((long item) -> list[list.length] = item);
Arrays.sort(list);
ordered = Spliterators.spliterator(list, characteristics());
}
return ordered.tryAdvance(action);
}
};
return new LongStreamSource(this, sortedSpliterator);
}
@Override
public LongStream peek(LongConsumer action) {
checkNotNull(action);
throwIfTerminated();
AbstractLongSpliterator peekSpliterator =
new Spliterators.AbstractLongSpliterator(
spliterator.estimateSize(), spliterator.characteristics()) {
@Override
public boolean tryAdvance(final LongConsumer innerAction) {
return spliterator.tryAdvance(action.andThen(innerAction));
}
};
return new LongStreamSource(this, peekSpliterator);
}
@Override
public LongStream limit(long maxSize) {
throwIfTerminated();
checkState(maxSize >= 0, "maxSize may not be negative");
return new LongStreamSource(this, new LimitSpliterator(maxSize, spliterator));
}
@Override
public LongStream skip(long n) {
throwIfTerminated();
checkState(n >= 0, "n may not be negative");
if (n == 0) {
return this;
}
return new LongStreamSource(this, new SkipSpliterator(n, spliterator));
}
@Override
public DoubleStream asDoubleStream() {
return mapToDouble(x -> (double) x);
}
@Override
public Stream boxed() {
return mapToObj(Long::valueOf);
}
@Override
public LongStream sequential() {
throwIfTerminated();
return this;
}
@Override
public LongStream parallel() {
throwIfTerminated();
return this;
}
@Override
public boolean isParallel() {
throwIfTerminated();
return false;
}
@Override
public LongStream unordered() {
throwIfTerminated();
return this;
}
}
}