com.landawn.abacus.util.stream.ShortIteratorEx Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of abacus-android Show documentation
Show all versions of abacus-android Show documentation
A general and simple library for Android
/*
* 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.Iterator;
import java.util.NoSuchElementException;
import com.landawn.abacus.annotation.Internal;
import com.landawn.abacus.util.N;
import com.landawn.abacus.util.ShortIterator;
import com.landawn.abacus.util.ShortList;
import com.landawn.abacus.util.function.Supplier;
/**
*
*/
@Internal
public abstract class ShortIteratorEx extends ShortIterator implements IteratorEx {
public static final ShortIteratorEx EMPTY = new ShortIteratorEx() {
@Override
public boolean hasNext() {
return false;
}
@Override
public short nextShort() {
throw new NoSuchElementException();
}
@Override
public void skip(long n) {
// Do nothing.
}
@Override
public long count() {
return 0;
}
@Override
public short[] toArray() {
return N.EMPTY_SHORT_ARRAY;
}
@Override
public void close() {
// Do nothing.
}
};
public static ShortIteratorEx empty() {
return EMPTY;
}
@SafeVarargs
public static ShortIteratorEx of(final short... a) {
return N.isNullOrEmpty(a) ? EMPTY : of(a, 0, a.length);
}
public static ShortIteratorEx of(final short[] a, final int fromIndex, final int toIndex) {
N.checkFromToIndex(fromIndex, toIndex, N.len(a));
if (fromIndex == toIndex) {
return EMPTY;
}
return new ShortIteratorEx() {
private int cursor = fromIndex;
@Override
public boolean hasNext() {
return cursor < toIndex;
}
@Override
public short nextShort() {
if (cursor >= toIndex) {
throw new NoSuchElementException();
}
return a[cursor++];
}
@Override
public void skip(long n) {
N.checkArgNotNegative(n, "n");
cursor = n < toIndex - cursor ? cursor + (int) n : toIndex;
}
@Override
public long count() {
return toIndex - cursor;
}
@Override
public short[] toArray() {
return N.copyOfRange(a, cursor, toIndex);
}
@Override
public ShortList toList() {
return ShortList.of(N.copyOfRange(a, cursor, toIndex));
}
@Override
public void close() {
// Do nothing.
}
};
}
public static ShortIteratorEx of(final ShortIterator iter) {
if (iter == null) {
return empty();
} else if (iter instanceof ShortIteratorEx) {
return ((ShortIteratorEx) iter);
}
return new ShortIteratorEx() {
@Override
public boolean hasNext() {
return iter.hasNext();
}
@Override
public short nextShort() {
return iter.nextShort();
}
@Override
public void close() {
// Do nothing.
}
};
}
/**
* Lazy evaluation.
*
* @param iteratorSupplier
* @return
*/
public static ShortIteratorEx of(final Supplier iteratorSupplier) {
N.checkArgNotNull(iteratorSupplier, "iteratorSupplier");
return new ShortIteratorEx() {
private ShortIterator iter = null;
private ShortIteratorEx iterEx = null;
private boolean isInitialized = false;
@Override
public boolean hasNext() {
if (isInitialized == false) {
init();
}
return iter.hasNext();
}
@Override
public short nextShort() {
if (isInitialized == false) {
init();
}
return iter.nextShort();
}
@Override
public void skip(long n) {
N.checkArgNotNegative(n, "n");
if (isInitialized == false) {
init();
}
if (iterEx != null) {
iterEx.skip(n);
} else {
super.skip(n);
}
}
@Override
public long count() {
if (isInitialized == false) {
init();
}
if (iterEx != null) {
return iterEx.count();
} else {
return super.count();
}
}
@Override
public void close() {
if (isInitialized == false) {
init();
}
if (iterEx != null) {
iterEx.close();
}
}
private void init() {
if (isInitialized == false) {
isInitialized = true;
iter = iteratorSupplier.get();
iterEx = iter instanceof ShortIteratorEx ? (ShortIteratorEx) iter : null;
}
}
};
}
/**
* Lazy evaluation.
*
* @param arraySupplier
* @return
*/
public static ShortIteratorEx oF(final Supplier arraySupplier) {
N.checkArgNotNull(arraySupplier, "arraySupplier");
return new ShortIteratorEx() {
private ShortIteratorEx iterEx = null;
private boolean isInitialized = false;
@Override
public boolean hasNext() {
if (isInitialized == false) {
init();
}
return iterEx.hasNext();
}
@Override
public short nextShort() {
if (isInitialized == false) {
init();
}
return iterEx.nextShort();
}
@Override
public void skip(long n) {
N.checkArgNotNegative(n, "n");
if (isInitialized == false) {
init();
}
iterEx.skip(n);
}
@Override
public long count() {
if (isInitialized == false) {
init();
}
return iterEx.count();
}
private void init() {
if (isInitialized == false) {
isInitialized = true;
short[] aar = arraySupplier.get();
iterEx = ShortIteratorEx.of(aar);
}
}
};
}
public static ShortIteratorEx from(final Iterator iter) {
if (iter == null) {
return empty();
} else if (iter instanceof ObjIteratorEx) {
final ObjIteratorEx iteratorEx = ((ObjIteratorEx) iter);
return new ShortIteratorEx() {
@Override
public boolean hasNext() {
return iteratorEx.hasNext();
}
@Override
public short nextShort() {
return iteratorEx.next();
}
@Override
public void skip(long n) {
N.checkArgNotNegative(n, "n");
iteratorEx.skip(n);
}
@Override
public long count() {
return iteratorEx.count();
}
@Override
public void close() {
iteratorEx.close();
}
};
} else {
return new ShortIteratorEx() {
@Override
public boolean hasNext() {
return iter.hasNext();
}
@Override
public short nextShort() {
return iter.next();
}
@Override
public void close() {
// Do nothing.
}
};
}
}
@Override
public void skip(long n) {
N.checkArgNotNegative(n, "n");
while (n > 0 && hasNext()) {
nextShort();
n--;
}
}
@Override
public long count() {
long result = 0;
while (hasNext()) {
nextShort();
result++;
}
return result;
}
@Override
public void close() {
// Do nothing.
}
}