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.landawn.abacus.util.stream.AbstractShortStream Maven / Gradle / Ivy
Go to download
A general programming library in Java/Android. It's easy to learn and simple to use with concise and powerful APIs.
/*
* Copyright (C) 2016, 2017, 2018, 2019 HaiYang Li
*
* 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 com.landawn.abacus.util.stream;
import java.util.Collection;
import java.util.Map;
import java.util.NoSuchElementException;
import java.util.Random;
import java.util.Set;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicLong;
import java.util.function.BiConsumer;
import java.util.function.BinaryOperator;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.stream.Collector;
import com.landawn.abacus.exception.TooManyElementsException;
import com.landawn.abacus.util.Fn;
import com.landawn.abacus.util.Fn.Suppliers;
import com.landawn.abacus.util.IndexedShort;
import com.landawn.abacus.util.Joiner;
import com.landawn.abacus.util.MergeResult;
import com.landawn.abacus.util.Multiset;
import com.landawn.abacus.util.MutableInt;
import com.landawn.abacus.util.MutableLong;
import com.landawn.abacus.util.MutableShort;
import com.landawn.abacus.util.N;
import com.landawn.abacus.util.Pair;
import com.landawn.abacus.util.Percentage;
import com.landawn.abacus.util.ShortIterator;
import com.landawn.abacus.util.ShortList;
import com.landawn.abacus.util.ShortSummaryStatistics;
import com.landawn.abacus.util.Strings.StringUtil;
import com.landawn.abacus.util.Throwables;
import com.landawn.abacus.util.Tuple.Tuple3;
import com.landawn.abacus.util.u.Optional;
import com.landawn.abacus.util.u.OptionalShort;
import com.landawn.abacus.util.function.ObjShortConsumer;
import com.landawn.abacus.util.function.ShortBiFunction;
import com.landawn.abacus.util.function.ShortBiPredicate;
import com.landawn.abacus.util.function.ShortBinaryOperator;
import com.landawn.abacus.util.function.ShortConsumer;
import com.landawn.abacus.util.function.ShortFunction;
import com.landawn.abacus.util.function.ShortPredicate;
import com.landawn.abacus.util.function.ShortTernaryOperator;
import com.landawn.abacus.util.function.ShortTriPredicate;
import com.landawn.abacus.util.function.ToShortFunction;
/**
*
*/
abstract class AbstractShortStream extends ShortStream {
AbstractShortStream(final boolean sorted, final Collection closeHandlers) {
super(sorted, closeHandlers);
}
@Override
public ShortStream distinct() {
assertNotClosed();
final Set set = N.newHashSet();
return newStream(this.sequential().filter(set::add).iteratorEx(), sorted);
}
// @Override
// public ShortStream flatmap(final ShortFunction mapper) {
// assertNotClosed();
//
// return flatMap(new ShortFunction() {
// @Override
// public ShortStream apply(short t) {
// return ShortStream.of(mapper.apply(t));
// }
// });
// }
@Override
public ShortStream flatmap(final ShortFunction mapper) {
assertNotClosed();
return flatMap(t -> ShortStream.of(mapper.apply(t)));
}
@Override
public Stream flatmapToObj(final ShortFunction> mapper) {
assertNotClosed();
return flatMapToObj(t -> Stream.of(mapper.apply(t)));
}
@Override
public Stream flattMapToObj(final ShortFunction mapper) {
assertNotClosed();
return flatMapToObj(t -> Stream.of(mapper.apply(t)));
}
@Override
public ShortStream mapPartial(final ShortFunction mapper) {
if (isParallel()) {
return mapToObj(mapper).psp(s -> s.filter(Fn.IS_PRESENT_SHORT).mapToShort(Fn.GET_AS_SHORT));
} else {
return mapToObj(mapper).filter(Fn.IS_PRESENT_SHORT).mapToShort(Fn.GET_AS_SHORT);
}
}
@Override
public ShortStream rangeMap(final ShortBiPredicate sameRange, final ShortBinaryOperator mapper) {
assertNotClosed();
final ShortIteratorEx iter = iteratorEx();
return newStream(new ShortIteratorEx() { //NOSONAR
private short left = 0, right = 0, next = 0;
private boolean hasNext = false;
@Override
public boolean hasNext() {
return hasNext || iter.hasNext();
}
@Override
public short nextShort() {
left = hasNext ? next : iter.nextShort();
right = left;
while (hasNext = iter.hasNext()) {
next = iter.nextShort();
if (sameRange.test(left, next)) {
right = next;
} else {
break;
}
}
return mapper.applyAsShort(left, right);
}
}, false);
}
@Override
public Stream rangeMapToObj(final ShortBiPredicate sameRange, final ShortBiFunction mapper) {
assertNotClosed();
final ShortIteratorEx iter = iteratorEx();
return newStream(new ObjIteratorEx() { //NOSONAR
private short left = 0, right = 0, next = 0;
private boolean hasNext = false;
@Override
public boolean hasNext() {
return hasNext || iter.hasNext();
}
@Override
public T next() {
left = hasNext ? next : iter.nextShort();
right = left;
while (hasNext = iter.hasNext()) {
next = iter.nextShort();
if (sameRange.test(left, next)) {
right = next;
} else {
break;
}
}
return mapper.apply(left, right);
}
}, false, null);
}
@Override
public Stream collapse(final ShortBiPredicate collapsible) {
assertNotClosed();
final ShortIteratorEx iter = iteratorEx();
return newStream(new ObjIteratorEx() { //NOSONAR
private boolean hasNext = false;
private short next = 0;
@Override
public boolean hasNext() {
return hasNext || iter.hasNext();
}
@Override
public ShortList next() {
final ShortList result = new ShortList(9);
result.add(hasNext ? next : (next = iter.nextShort()));
while ((hasNext = iter.hasNext())) {
if (collapsible.test(next, (next = iter.nextShort()))) {
result.add(next);
} else {
break;
}
}
return result;
}
}, false, null);
}
@Override
public ShortStream collapse(final ShortBiPredicate collapsible, final ShortBinaryOperator mergeFunction) {
assertNotClosed();
final ShortIteratorEx iter = iteratorEx();
return newStream(new ShortIteratorEx() { //NOSONAR
private boolean hasNext = false;
private short next = 0;
@Override
public boolean hasNext() {
return hasNext || iter.hasNext();
}
@Override
public short nextShort() {
short res = hasNext ? next : (next = iter.nextShort());
while ((hasNext = iter.hasNext())) {
if (collapsible.test(next, (next = iter.nextShort()))) {
res = mergeFunction.applyAsShort(res, next);
} else {
break;
}
}
return res;
}
}, false);
}
@Override
public ShortStream collapse(final ShortTriPredicate collapsible, final ShortBinaryOperator mergeFunction) {
assertNotClosed();
final ShortIteratorEx iter = iteratorEx();
return newStream(new ShortIteratorEx() { //NOSONAR
private boolean hasNext = false;
private short next = 0;
@Override
public boolean hasNext() {
return hasNext || iter.hasNext();
}
@Override
public short nextShort() {
final short first = hasNext ? next : (next = iter.nextShort());
short res = first;
while ((hasNext = iter.hasNext())) {
if (collapsible.test(first, next, (next = iter.nextShort()))) {
res = mergeFunction.applyAsShort(res, next);
} else {
break;
}
}
return res;
}
}, false);
}
@Override
public ShortStream skip(final long n, final ShortConsumer action) {
assertNotClosed();
checkArgNotNegative(n, "n");
checkArgNotNull(action, "action");
final ShortPredicate filter = isParallel() ? new ShortPredicate() {
final AtomicLong cnt = new AtomicLong(n);
@Override
public boolean test(short value) {
return cnt.getAndDecrement() > 0;
}
} : new ShortPredicate() {
final MutableLong cnt = MutableLong.of(n);
@Override
public boolean test(short value) {
return cnt.getAndDecrement() > 0;
}
};
return dropWhile(filter, action);
}
// @Override
// public ShortStream removeIf(final ShortPredicate predicate) {
// assertNotClosed();
//
// return filter(value -> !predicate.test(value));
// }
//
// @Override
// public ShortStream removeIf(final ShortPredicate predicate, final ShortConsumer actionOnDroppedItem) {
// assertNotClosed();
//
// return filter(value -> {
// if (predicate.test(value)) {
// actionOnDroppedItem.accept(value);
// return false;
// }
//
// return true;
// });
// }
@Override
public ShortStream filter(final ShortPredicate predicate, final ShortConsumer actionOnDroppedItem) {
assertNotClosed();
return filter(value -> {
if (!predicate.test(value)) {
actionOnDroppedItem.accept(value);
return false;
}
return true;
});
}
@Override
public ShortStream dropWhile(final ShortPredicate predicate, final ShortConsumer actionOnDroppedItem) {
assertNotClosed();
return dropWhile(value -> {
if (predicate.test(value)) {
actionOnDroppedItem.accept(value);
return true;
}
return false;
});
}
@Override
public ShortStream step(final long step) {
assertNotClosed();
checkArgPositive(step, "step");
if (step == 1) {
return skip(0);
}
final long skip = step - 1;
final ShortIteratorEx iter = this.iteratorEx();
final ShortIterator shortIterator = new ShortIteratorEx() {
@Override
public boolean hasNext() {
return iter.hasNext();
}
@Override
public short nextShort() {
final short next = iter.nextShort();
iter.advance(skip);
return next;
}
};
return newStream(shortIterator, sorted);
}
@Override
public Stream split(final int chunkSize) {
assertNotClosed();
return splitToList(chunkSize).map(t -> new ArrayShortStream(t.array(), 0, t.size(), sorted, null));
}
@Override
public Stream split(final ShortPredicate predicate) {
assertNotClosed();
return splitToList(predicate).map(t -> new ArrayShortStream(t.array(), 0, t.size(), sorted, null));
}
@Override
public Stream splitAt(final ShortPredicate where) {
assertNotClosed();
final ShortIteratorEx iter = iteratorEx();
return newStream(new ObjIteratorEx() { //NOSONAR
private int cursor = 0;
private short next = 0;
private boolean hasNext = false;
@Override
public boolean hasNext() {
return cursor < 2;
}
@Override
public ShortStream next() {
if (!hasNext()) {
throw new NoSuchElementException();
}
ShortStream result = null;
if (cursor == 0) {
final ShortList list = new ShortList();
while (iter.hasNext()) {
next = iter.nextShort();
if (!where.test(next)) {
list.add(next);
} else {
hasNext = true;
break;
}
}
result = new ArrayShortStream(list.array(), 0, list.size(), sorted, null);
} else {
ShortIteratorEx iterEx = iter;
if (hasNext) {
iterEx = new ShortIteratorEx() {
private boolean isFirst = true;
@Override
public boolean hasNext() {
return isFirst || iter.hasNext();
}
@Override
public short nextShort() {
if (!hasNext()) {
throw new NoSuchElementException();
}
if (isFirst) {
isFirst = false;
return next;
} else {
return iter.nextShort();
}
}
};
}
result = new IteratorShortStream(iterEx, sorted, null);
}
cursor++;
return result;
}
@Override
public long count() {
iter.count();
return 2 - cursor; //NOSONAR
}
@Override
public void advance(long n) {
if (n == 0) {
return;
} else if (n == 1) {
if (cursor == 0) {
while (iter.hasNext()) {
next = iter.nextShort();
if (!where.test(next)) {
hasNext = true;
break;
}
}
} else {
iter.advance(Long.MAX_VALUE);
}
} else {
iter.advance(Long.MAX_VALUE);
}
cursor = n >= 2 ? 2 : cursor + (int) n;
}
}, false, null);
}
@Override
public Stream sliding(final int windowSize, final int increment) {
assertNotClosed();
return slidingToList(windowSize, increment).map(t -> new ArrayShortStream(t.array(), 0, t.size(), sorted, null));
}
@Override
public ShortStream scan(final ShortBinaryOperator accumulator) {
assertNotClosed();
final ShortIteratorEx iter = iteratorEx();
return newStream(new ShortIteratorEx() { //NOSONAR
private short res = 0;
private boolean isFirst = true;
@Override
public boolean hasNext() {
return iter.hasNext();
}
@Override
public short nextShort() {
if (isFirst) {
isFirst = false;
return (res = iter.nextShort());
} else {
return (res = accumulator.applyAsShort(res, iter.nextShort()));
}
}
}, false);
}
@Override
public ShortStream scan(final short init, final ShortBinaryOperator accumulator) {
assertNotClosed();
final ShortIteratorEx iter = iteratorEx();
return newStream(new ShortIteratorEx() { //NOSONAR
private short res = init;
@Override
public boolean hasNext() {
return iter.hasNext();
}
@Override
public short nextShort() {
return (res = accumulator.applyAsShort(res, iter.nextShort()));
}
}, false);
}
@Override
public ShortStream scan(final short init, final ShortBinaryOperator accumulator, final boolean initIncluded) {
assertNotClosed();
if (!initIncluded) {
return scan(init, accumulator);
}
final ShortIteratorEx iter = iteratorEx();
return newStream(new ShortIteratorEx() { //NOSONAR
private boolean isFirst = true;
private short res = init;
@Override
public boolean hasNext() {
return isFirst || iter.hasNext();
}
@Override
public short nextShort() {
if (isFirst) {
isFirst = false;
return init;
}
return (res = accumulator.applyAsShort(res, iter.nextShort()));
}
}, false);
}
@Override
public ShortStream top(int n) {
assertNotClosed();
return top(n, SHORT_COMPARATOR);
}
@Override
public ShortStream intersection(final Collection c) {
assertNotClosed();
final Multiset multiset = Multiset.from(c);
return newStream(this.sequential().filter(value -> multiset.getAndRemove(value) > 0).iteratorEx(), sorted);
}
@Override
public ShortStream difference(final Collection c) {
assertNotClosed();
final Multiset multiset = Multiset.from(c);
return newStream(this.sequential().filter(value -> multiset.getAndRemove(value) < 1).iteratorEx(), sorted);
}
@Override
public ShortStream symmetricDifference(final Collection c) {
assertNotClosed();
final Multiset multiset = Multiset.from(c);
return newStream(this.sequential()
.filter(value -> multiset.getAndRemove(value) < 1)
.append(Stream.of(c).filter(value -> multiset.getAndRemove(value) > 0).mapToShort(ToShortFunction.UNBOX))
.iteratorEx(), false);
}
@Override
public ShortStream reversed() {
assertNotClosed();
return newStream(new ShortIteratorEx() { //NOSONAR
private boolean initialized = false;
private short[] elements;
private int fromIndex = -1;
private int toIndex = -1;
private int cursor;
@Override
public boolean hasNext() {
if (!initialized) {
init();
}
return cursor > fromIndex;
}
@Override
public short nextShort() {
if (!initialized) {
init();
}
if (cursor <= fromIndex) {
throw new NoSuchElementException();
}
return elements[--cursor];
}
@Override
public long count() {
if (!initialized) {
init();
}
return cursor - fromIndex; //NOSONAR
}
@Override
public void advance(long n) {
if (!initialized) {
init();
}
cursor = n < cursor - fromIndex ? cursor - (int) n : fromIndex;
}
@Override
public short[] toArray() {
if (!initialized) {
init();
}
final short[] a = new short[cursor - fromIndex];
for (int i = 0, len = cursor - fromIndex; i < len; i++) {
a[i] = elements[cursor - i - 1];
}
return a;
}
private void init() {
if (!initialized) {
initialized = true;
final Tuple3 tp = AbstractShortStream.this.arrayForIntermediateOp();
elements = tp._1;
fromIndex = tp._2;
toIndex = tp._3;
cursor = toIndex;
}
}
}, false);
}
@Override
public ShortStream rotated(final int distance) {
assertNotClosed();
return newStream(new ShortIteratorEx() { //NOSONAR
private boolean initialized = false;
private short[] elements;
private int fromIndex = -1;
private int toIndex = -1;
private int len;
private int start;
private int cnt = 0;
@Override
public boolean hasNext() {
if (!initialized) {
init();
}
return cnt < len;
}
@Override
public short nextShort() {
if (!hasNext()) {
throw new NoSuchElementException();
}
return elements[((start + cnt++) % len) + fromIndex];
}
@Override
public long count() {
if (!initialized) {
init();
}
return len - cnt; //NOSONAR
}
@Override
public void advance(long n) {
if (!initialized) {
init();
}
cnt = n < len - cnt ? cnt + (int) n : len;
}
@Override
public short[] toArray() {
if (!initialized) {
init();
}
final short[] a = new short[len - cnt];
for (int i = cnt; i < len; i++) {
a[i - cnt] = elements[((start + i) % len) + fromIndex];
}
return a;
}
private void init() {
if (!initialized) {
initialized = true;
final Tuple3 tp = AbstractShortStream.this.arrayForIntermediateOp();
elements = tp._1;
fromIndex = tp._2;
toIndex = tp._3;
len = toIndex - fromIndex;
if (len > 0) {
start = distance % len;
if (start < 0) {
start += len;
}
start = len - start;
}
}
}
}, distance == 0 && sorted);
}
@Override
public ShortStream shuffled(final Random rnd) {
assertNotClosed();
checkArgNotNull(rnd, "random");
return lazyLoad(a -> {
N.shuffle(a, rnd);
return a;
}, false);
}
@Override
public ShortStream sorted() {
assertNotClosed();
if (sorted) {
return newStream(iteratorEx(), sorted);
}
return lazyLoad(a -> {
if (isParallel()) {
N.parallelSort(a);
} else {
N.sort(a);
}
return a;
}, true);
}
@Override
public ShortStream reverseSorted() {
assertNotClosed();
return newStream(new ShortIteratorEx() { //NOSONAR
private boolean initialized = false;
private short[] aar;
private int cursor;
@Override
public boolean hasNext() {
if (!initialized) {
init();
}
return cursor > 0;
}
@Override
public short nextShort() {
if (!initialized) {
init();
}
if (cursor <= 0) {
throw new NoSuchElementException();
}
return aar[--cursor];
}
@Override
public long count() {
if (!initialized) {
init();
}
return cursor;
}
@Override
public void advance(long n) {
if (!initialized) {
init();
}
cursor = n < cursor ? cursor - (int) n : 0;
}
@Override
public short[] toArray() {
if (!initialized) {
init();
}
final short[] a = new short[cursor];
for (int i = 0; i < cursor; i++) {
a[i] = aar[cursor - i - 1];
}
return a;
}
private void init() {
if (!initialized) {
initialized = true;
aar = AbstractShortStream.this.toArrayForIntermediateOp();
if (isParallel()) {
N.parallelSort(aar);
} else {
N.sort(aar);
}
cursor = aar.length;
}
}
}, false);
}
private ShortStream lazyLoad(final Function op, final boolean sorted) {
return newStream(new ShortIteratorEx() { //NOSONAR
private boolean initialized = false;
private short[] aar;
private int cursor = 0;
private int len;
@Override
public boolean hasNext() {
if (!initialized) {
init();
}
return cursor < len;
}
@Override
public short nextShort() {
if (!initialized) {
init();
}
if (cursor >= len) {
throw new NoSuchElementException();
}
return aar[cursor++];
}
@Override
public long count() {
if (!initialized) {
init();
}
return len - cursor; //NOSONAR
}
@Override
public void advance(long n) {
if (!initialized) {
init();
}
cursor = n > len - cursor ? len : cursor + (int) n;
}
@Override
public short[] toArray() {
if (!initialized) {
init();
}
final short[] a = new short[len - cursor];
for (int i = cursor; i < len; i++) {
a[i - cursor] = aar[i];
}
return a;
}
private void init() {
if (!initialized) {
initialized = true;
aar = op.apply(AbstractShortStream.this.toArrayForIntermediateOp());
len = aar.length;
}
}
}, sorted);
}
@Override
public ShortStream cycled() {
assertNotClosed();
return newStream(new ShortIterator() { //NOSONAR
private ShortIterator iter = null;
private ShortList list = null;
private short[] a = null;
private int len = 0;
private int cursor = -1;
private short e = 0;
private boolean initialized = false;
@Override
public boolean hasNext() {
if (!initialized) {
init();
}
if (a == null && !iter.hasNext()) {
a = list.toArray();
len = a.length;
cursor = 0;
}
return cursor < len || iter.hasNext();
}
@Override
public short nextShort() {
if (!hasNext()) {
throw new NoSuchElementException();
}
if (len > 0) {
if (cursor >= len) {
cursor = 0;
}
return a[cursor++];
} else {
e = iter.nextShort();
list.add(e);
return e;
}
}
private void init() {
if (!initialized) {
initialized = true;
iter = AbstractShortStream.this.iteratorEx();
list = new ShortList();
}
}
}, false);
}
@Override
public ShortStream cycled(final long rounds) {
assertNotClosed();
checkArgNotNegative(rounds, "times");
if (rounds == 0) {
return limit(0);
} else if (rounds == 1) {
return skip(0);
}
return newStream(new ShortIterator() { //NOSONAR
private ShortIterator iter = null;
private ShortList list = null;
private short[] a = null;
private int len = 0;
private int cursor = -1;
private short e = 0;
private long m = 0;
private boolean initialized = false;
@Override
public boolean hasNext() {
if (!initialized) {
init();
}
if (a == null && !iter.hasNext()) {
a = list.toArray();
len = a.length;
cursor = 0;
m = 1;
}
return m < rounds && (cursor < len || rounds - m > 1) && (len > 0 || iter.hasNext());
}
@Override
public short nextShort() {
if (!hasNext()) {
throw new NoSuchElementException();
}
if (len > 0) {
if (cursor >= len) {
cursor = 0;
m++;
}
return a[cursor++];
} else {
e = iter.nextShort();
list.add(e);
return e;
}
}
private void init() {
if (!initialized) {
initialized = true;
iter = AbstractShortStream.this.iteratorEx();
list = new ShortList();
}
}
}, rounds <= 1 ? this.sorted : false);
}
@Override
public Stream indexed() {
assertNotClosed();
return newStream(this.sequential().mapToObj(new ShortFunction() { //NOSONAR
final MutableLong idx = MutableLong.of(0);
@Override
public IndexedShort apply(short t) {
return IndexedShort.of(t, idx.getAndIncrement());
}
}).iteratorEx(), true, INDEXED_SHORT_COMPARATOR);
}
@Override
public Stream boxed() {
assertNotClosed();
return newStream(iteratorEx(), sorted, sorted ? SHORT_COMPARATOR : null);
}
@Override
@SafeVarargs
public final ShortStream prepend(final short... a) {
assertNotClosed();
return prepend(ShortStream.of(a));
}
@Override
public ShortStream prepend(ShortStream stream) {
assertNotClosed();
if (isParallel()) {
return ShortStream.concat(stream, this)
.parallel(maxThreadNum(), executorNumForVirtualThread(), splitor(), asyncExecutor(), cancelUncompletedThreads());
} else {
return ShortStream.concat(stream, this);
}
}
@Override
public ShortStream prepend(final OptionalShort op) {
assertNotClosed();
return prepend(op.stream());
}
@Override
@SafeVarargs
public final ShortStream append(final short... a) {
assertNotClosed();
return append(ShortStream.of(a));
}
@Override
public ShortStream append(ShortStream stream) {
assertNotClosed();
if (isParallel()) {
return ShortStream.concat(this, stream)
.parallel(maxThreadNum(), executorNumForVirtualThread(), splitor(), asyncExecutor(), cancelUncompletedThreads());
} else {
return ShortStream.concat(this, stream);
}
}
@Override
public ShortStream append(final OptionalShort op) { //NOSONAR
assertNotClosed();
return prepend(op.stream());
}
@Override
@SafeVarargs
public final ShortStream appendIfEmpty(final short... a) {
assertNotClosed();
return appendIfEmpty(() -> ShortStream.of(a));
}
@Override
public ShortStream mergeWith(ShortStream b, ShortBiFunction nextSelector) {
assertNotClosed();
if (isParallel()) {
return ShortStream.merge(this, b, nextSelector)
.parallel(maxThreadNum(), executorNumForVirtualThread(), splitor(), asyncExecutor(), cancelUncompletedThreads());
} else {
return ShortStream.merge(this, b, nextSelector);
}
}
@Override
public ShortStream zipWith(ShortStream b, ShortBinaryOperator zipFunction) {
assertNotClosed();
return ShortStream.zip(this, b, zipFunction);
}
@Override
public ShortStream zipWith(ShortStream b, ShortStream c, ShortTernaryOperator zipFunction) {
assertNotClosed();
return ShortStream.zip(this, b, c, zipFunction);
}
@Override
public ShortStream zipWith(ShortStream b, short valueForNoneA, short valueForNoneB, ShortBinaryOperator zipFunction) {
assertNotClosed();
return ShortStream.zip(this, b, valueForNoneA, valueForNoneB, zipFunction);
}
@Override
public ShortStream zipWith(ShortStream b, ShortStream c, short valueForNoneA, short valueForNoneB, short valueForNoneC, ShortTernaryOperator zipFunction) {
assertNotClosed();
return ShortStream.zip(this, b, c, valueForNoneA, valueForNoneB, valueForNoneC, zipFunction);
}
// @Override
// public ShortStream cached() {
// return newStream(toArray(), sorted);
// }
@Override
public Map toMap(Throwables.ShortFunction keyMapper,
Throwables.ShortFunction valueMapper) throws E, E2 {
assertNotClosed();
return toMap(keyMapper, valueMapper, Suppliers. ofMap());
}
@Override
public , E extends Exception, E2 extends Exception> M toMap(Throwables.ShortFunction keyMapper,
Throwables.ShortFunction valueMapper, Supplier mapFactory) throws E, E2 {
assertNotClosed();
return toMap(keyMapper, valueMapper, Fn. throwingMerger(), mapFactory);
}
@Override
public Map toMap(Throwables.ShortFunction keyMapper,
Throwables.ShortFunction valueMapper, BinaryOperator mergeFunction) throws E, E2 {
assertNotClosed();
return toMap(keyMapper, valueMapper, mergeFunction, Suppliers. ofMap());
}
@Override
public Map groupTo(Throwables.ShortFunction keyMapper, final Collector downstream)
throws E {
assertNotClosed();
return groupTo(keyMapper, downstream, Suppliers. ofMap());
}
@Override
public void forEachIndexed(Throwables.IndexedShortConsumer action) throws E {
assertNotClosed();
if (isParallel()) {
final AtomicInteger idx = new AtomicInteger(0);
forEach(t -> action.accept(idx.getAndIncrement(), t));
} else {
final MutableInt idx = MutableInt.of(0);
forEach(t -> action.accept(idx.getAndIncrement(), t));
}
}
@Override
public OptionalShort first() {
assertNotClosed();
try {
final ShortIterator iter = this.iteratorEx();
return iter.hasNext() ? OptionalShort.of(iter.nextShort()) : OptionalShort.empty();
} finally {
close();
}
}
@Override
public OptionalShort last() {
assertNotClosed();
try {
final ShortIterator iter = this.iteratorEx();
if (!iter.hasNext()) {
return OptionalShort.empty();
}
short next = iter.nextShort();
while (iter.hasNext()) {
next = iter.nextShort();
}
return OptionalShort.of(next);
} finally {
close();
}
}
@Override
public OptionalShort onlyOne() throws TooManyElementsException {
assertNotClosed();
try {
final ShortIterator iter = this.iteratorEx();
final OptionalShort result = iter.hasNext() ? OptionalShort.of(iter.nextShort()) : OptionalShort.empty();
if (result.isPresent() && iter.hasNext()) {
throw new TooManyElementsException("There are at least two elements: " + StringUtil.concat(result.get(), ", ", iter.nextShort()));
}
return result;
} finally {
close();
}
}
@Override
public OptionalShort findAny(final Throwables.ShortPredicate predicate) throws E {
assertNotClosed();
return findFirst(predicate);
}
@Override
public OptionalShort findFirstOrAny(Throwables.ShortPredicate predicateForFirst) throws E {
assertNotClosed();
try {
final ShortIteratorEx iter = iteratorEx();
MutableShort ret = null;
short next = 0;
while (iter.hasNext()) {
next = iter.nextShort();
if (predicateForFirst.test(next)) {
return OptionalShort.of(next);
} else if (ret == null) {
ret = MutableShort.of(next);
}
}
return ret == null ? OptionalShort.empty() : OptionalShort.of(ret.value());
} finally {
close();
}
}
@Override
public OptionalShort findFirstOrLast(Throwables.ShortPredicate predicateForFirst) throws E {
assertNotClosed();
try {
final ShortIteratorEx iter = iteratorEx();
MutableShort ret = null;
short next = 0;
while (iter.hasNext()) {
next = iter.nextShort();
if (predicateForFirst.test(next)) {
return OptionalShort.of(next);
} else if (ret == null) {
ret = MutableShort.of(next);
} else {
ret.setValue(next);
}
}
return ret == null ? OptionalShort.empty() : OptionalShort.of(ret.value());
} finally {
close();
}
}
@Override
public Optional> percentiles() {
assertNotClosed();
try {
final short[] a = sorted().toArray();
if (a.length == 0) {
return Optional.empty();
}
return Optional.of(N.percentiles(a));
} finally {
close();
}
}
@Override
public Pair>> summarizeAndPercentiles() {
assertNotClosed();
try {
final short[] a = sorted().toArray();
if (N.isNullOrEmpty(a)) {
return Pair.of(new ShortSummaryStatistics(), Optional.> empty());
} else {
return Pair.of(new ShortSummaryStatistics(a.length, a[0], a[a.length - 1], sum(a)), Optional.of(N.percentiles(a)));
}
} finally {
close();
}
}
@Override
public String join(final CharSequence delimiter, final CharSequence prefix, final CharSequence suffix) {
assertNotClosed();
try {
final Joiner joiner = Joiner.with(delimiter, prefix, suffix).reuseCachedBuffer();
final ShortIteratorEx iter = this.iteratorEx();
while (iter.hasNext()) {
joiner.append(iter.nextShort());
}
return joiner.toString();
} finally {
close();
}
}
@Override
public R collect(Supplier supplier, ObjShortConsumer accumulator) {
assertNotClosed();
final BiConsumer combiner = collectingCombiner;
return collect(supplier, accumulator, combiner);
}
}