ret = Arrays.spliterator(curChunk, splElementIndex, splElementIndex + t);
splElementIndex += t;
return ret;
}
}
else {
return null;
}
}
};
}
/**
* An ordered collection of primitive values. Elements can be added, but
* not removed. Goes through a building phase, during which elements can be
* added, and a traversal phase, during which elements can be traversed in
* order but no further modifications are possible.
*
* One or more arrays are used to store elements. The use of a multiple
* arrays has better performance characteristics than a single array used by
* {@link ArrayList}, as when the capacity of the list needs to be increased
* no copying of elements is required. This is usually beneficial in the case
* where the results will be traversed a small number of times.
*
* @param the wrapper type for this primitive type
* @param the array type for this primitive type
* @param the Consumer type for this primitive type
*/
abstract static class OfPrimitive
extends AbstractSpinedBuffer implements Iterable {
/*
* We optimistically hope that all the data will fit into the first chunk,
* so we try to avoid inflating the spine[] and priorElementCount[] arrays
* prematurely. So methods must be prepared to deal with these arrays being
* null. If spine is non-null, then spineIndex points to the current chunk
* within the spine, otherwise it is zero. The spine and priorElementCount
* arrays are always the same size, and for any i <= spineIndex,
* priorElementCount[i] is the sum of the sizes of all the prior chunks.
*
* The curChunk pointer is always valid. The elementIndex is the index of
* the next element to be written in curChunk; this may be past the end of
* curChunk so we have to check before writing. When we inflate the spine
* array, curChunk becomes the first element in it. When we clear the
* buffer, we discard all chunks except the first one, which we clear,
* restoring it to the initial single-chunk state.
*/
// The chunk we're currently writing into
T_ARR curChunk;
// All chunks, or null if there is only one chunk
T_ARR[] spine;
/**
* Constructs an empty list with the specified initial capacity.
*
* @param initialCapacity the initial capacity of the list
* @throws IllegalArgumentException if the specified initial capacity
* is negative
*/
OfPrimitive(int initialCapacity) {
super(initialCapacity);
curChunk = newArray(1 << initialChunkPower);
}
/**
* Constructs an empty list with an initial capacity of sixteen.
*/
OfPrimitive() {
super();
curChunk = newArray(1 << initialChunkPower);
}
@Override
public abstract Iterator iterator();
@Override
public abstract void forEach(Consumer super E> consumer);
/** Create a new array-of-array of the proper type and size */
protected abstract T_ARR[] newArrayArray(int size);
/** Create a new array of the proper type and size */
protected abstract T_ARR newArray(int size);
/** Get the length of an array */
protected abstract int arrayLength(T_ARR array);
/** Iterate an array with the provided consumer */
protected abstract void arrayForEach(T_ARR array, int from, int to,
T_CONS consumer);
protected long capacity() {
return (spineIndex == 0)
? arrayLength(curChunk)
: priorElementCount[spineIndex] + arrayLength(spine[spineIndex]);
}
private void inflateSpine() {
if (spine == null) {
spine = newArrayArray(MIN_SPINE_SIZE);
priorElementCount = new long[MIN_SPINE_SIZE];
spine[0] = curChunk;
}
}
protected final void ensureCapacity(long targetSize) {
long capacity = capacity();
if (targetSize > capacity) {
inflateSpine();
for (int i=spineIndex+1; targetSize > capacity; i++) {
if (i >= spine.length) {
int newSpineSize = spine.length * 2;
spine = Arrays.copyOf(spine, newSpineSize);
priorElementCount = Arrays.copyOf(priorElementCount, newSpineSize);
}
int nextChunkSize = chunkSize(i);
spine[i] = newArray(nextChunkSize);
priorElementCount[i] = priorElementCount[i-1] + arrayLength(spine[i - 1]);
capacity += nextChunkSize;
}
}
}
protected void increaseCapacity() {
ensureCapacity(capacity() + 1);
}
protected int chunkFor(long index) {
if (spineIndex == 0) {
if (index < elementIndex)
return 0;
else
throw new IndexOutOfBoundsException(Long.toString(index));
}
if (index >= count())
throw new IndexOutOfBoundsException(Long.toString(index));
for (int j=0; j <= spineIndex; j++)
if (index < priorElementCount[j] + arrayLength(spine[j]))
return j;
throw new IndexOutOfBoundsException(Long.toString(index));
}
public void copyInto(T_ARR array, int offset) {
long finalOffset = offset + count();
if (finalOffset > arrayLength(array) || finalOffset < offset) {
throw new IndexOutOfBoundsException("does not fit");
}
if (spineIndex == 0)
System.arraycopy(curChunk, 0, array, offset, elementIndex);
else {
// full chunks
for (int i=0; i < spineIndex; i++) {
System.arraycopy(spine[i], 0, array, offset, arrayLength(spine[i]));
offset += arrayLength(spine[i]);
}
if (elementIndex > 0)
System.arraycopy(curChunk, 0, array, offset, elementIndex);
}
}
public T_ARR asPrimitiveArray() {
// @@@ will fail for size == MAX_VALUE
T_ARR result = newArray((int) count());
copyInto(result, 0);
return result;
}
protected void preAccept() {
if (elementIndex == arrayLength(curChunk)) {
inflateSpine();
if (spineIndex+1 >= spine.length || spine[spineIndex+1] == null)
increaseCapacity();
elementIndex = 0;
++spineIndex;
curChunk = spine[spineIndex];
}
}
public void clear() {
if (spine != null) {
curChunk = spine[0];
spine = null;
priorElementCount = null;
}
elementIndex = 0;
spineIndex = 0;
}
public void forEach(T_CONS consumer) {
// completed chunks, if any
for (int j = 0; j < spineIndex; j++)
arrayForEach(spine[j], 0, arrayLength(spine[j]), consumer);
// current chunk
arrayForEach(curChunk, 0, elementIndex, consumer);
}
abstract class BaseSpliterator>
implements Spliterator {
// The current spine index
int splSpineIndex;
// The current element index into the current spine
int splElementIndex;
// When splSpineIndex >= spineIndex and splElementIndex >= elementIndex then
// this spliterator is fully traversed
// tryAdvance can set splSpineIndex > spineIndex if the last spine is full
// The current spine array
T_ARR splChunk = (spine == null) ? curChunk : spine[0];
abstract void arrayForOne(T_ARR array, int index, T_CONS consumer);
abstract T_SPLITER arraySpliterator(T_ARR array, int offset, int len);
@Override
public long estimateSize() {
return (spine == null)
? (elementIndex - splElementIndex)
: count() - (priorElementCount[splSpineIndex] + splElementIndex);
}
@Override
public int characteristics() {
return SPLITERATOR_CHARACTERISTICS;
}
public boolean tryAdvance(T_CONS consumer) {
if (splSpineIndex < spineIndex
|| (splSpineIndex == spineIndex && splElementIndex < elementIndex)) {
arrayForOne(splChunk, splElementIndex++, consumer);
if (splElementIndex == arrayLength(splChunk)) {
splElementIndex = 0;
++splSpineIndex;
if (spine != null && splSpineIndex < spine.length)
splChunk = spine[splSpineIndex];
}
return true;
}
return false;
}
public void forEachRemaining(T_CONS consumer) {
if (splSpineIndex < spineIndex
|| (splSpineIndex == spineIndex && splElementIndex < elementIndex)) {
int i = splElementIndex;
// completed chunks, if any
for (int sp = splSpineIndex; sp < spineIndex; sp++) {
T_ARR chunk = spine[sp];
arrayForEach(chunk, i, arrayLength(chunk), consumer);
i = 0;
}
arrayForEach(curChunk, i, elementIndex, consumer);
splSpineIndex = spineIndex;
splElementIndex = elementIndex;
}
}
@Override
public T_SPLITER trySplit() {
if (splSpineIndex < spineIndex) {
T_SPLITER ret = arraySpliterator(spine[splSpineIndex], splElementIndex,
arrayLength(spine[splSpineIndex]) - splElementIndex);
splChunk = spine[++splSpineIndex];
splElementIndex = 0;
return ret;
}
else if (splSpineIndex == spineIndex) {
int t = (elementIndex - splElementIndex) / 2;
if (t == 0)
return null;
else {
T_SPLITER ret = arraySpliterator(curChunk, splElementIndex, t);
splElementIndex += t;
return ret;
}
}
else {
return null;
}
}
}
}
/**
* An ordered collection of {@code int} values.
*/
static class OfInt extends SpinedBuffer.OfPrimitive
implements IntConsumer {
OfInt() { }
OfInt(int initialCapacity) {
super(initialCapacity);
}
@Override
public void forEach(Consumer super Integer> consumer) {
if (consumer instanceof IntConsumer) {
forEach((IntConsumer) consumer);
}
else {
if (Tripwire.ENABLED)
Tripwire.trip(getClass(), "{0} calling SpinedBuffer.OfInt.forEach(Consumer)");
spliterator().forEachRemaining(consumer);
}
}
@Override
protected int[][] newArrayArray(int size) {
return new int[size][];
}
@Override
protected int[] newArray(int size) {
return new int[size];
}
@Override
protected int arrayLength(int[] array) {
return array.length;
}
@Override
protected void arrayForEach(int[] array,
int from, int to,
IntConsumer consumer) {
for (int i = from; i < to; i++)
consumer.accept(array[i]);
}
@Override
public void accept(int i) {
preAccept();
curChunk[elementIndex++] = i;
}
public int get(long index) {
int ch = chunkFor(index);
if (spineIndex == 0 && ch == 0)
return curChunk[(int) index];
else
return spine[ch][(int) (index-priorElementCount[ch])];
}
public int[] asIntArray() {
return asPrimitiveArray();
}
@Override
public PrimitiveIterator.OfInt iterator() {
return Spliterators.iteratorFromSpliterator(spliterator());
}
public Spliterator.OfInt spliterator() {
class Splitr extends BaseSpliterator
implements Spliterator.OfInt {
@Override
void arrayForOne(int[] array, int index, IntConsumer consumer) {
consumer.accept(array[index]);
}
@Override
Spliterator.OfInt arraySpliterator(int[] array, int offset, int len) {
return Arrays.spliterator(array, offset, offset+len);
}
};
return new Splitr();
}
@Override
public String toString() {
int[] array = asIntArray();
if (array.length < 200) {
return String.format("%s[length=%d, chunks=%d]%s",
getClass().getSimpleName(), array.length,
spineIndex, Arrays.toString(array));
}
else {
int[] array2 = Arrays.copyOf(array, 200);
return String.format("%s[length=%d, chunks=%d]%s...",
getClass().getSimpleName(), array.length,
spineIndex, Arrays.toString(array2));
}
}
}
/**
* An ordered collection of {@code long} values.
*/
static class OfLong extends SpinedBuffer.OfPrimitive
implements LongConsumer {
OfLong() { }
OfLong(int initialCapacity) {
super(initialCapacity);
}
@Override
public void forEach(Consumer super Long> consumer) {
if (consumer instanceof LongConsumer) {
forEach((LongConsumer) consumer);
}
else {
if (Tripwire.ENABLED)
Tripwire.trip(getClass(), "{0} calling SpinedBuffer.OfLong.forEach(Consumer)");
spliterator().forEachRemaining(consumer);
}
}
@Override
protected long[][] newArrayArray(int size) {
return new long[size][];
}
@Override
protected long[] newArray(int size) {
return new long[size];
}
@Override
protected int arrayLength(long[] array) {
return array.length;
}
@Override
protected void arrayForEach(long[] array,
int from, int to,
LongConsumer consumer) {
for (int i = from; i < to; i++)
consumer.accept(array[i]);
}
@Override
public void accept(long i) {
preAccept();
curChunk[elementIndex++] = i;
}
public long get(long index) {
int ch = chunkFor(index);
if (spineIndex == 0 && ch == 0)
return curChunk[(int) index];
else
return spine[ch][(int) (index-priorElementCount[ch])];
}
public long[] asLongArray() {
return asPrimitiveArray();
}
@Override
public PrimitiveIterator.OfLong iterator() {
return Spliterators.iteratorFromSpliterator(spliterator());
}
public Spliterator.OfLong spliterator() {
class Splitr extends BaseSpliterator
implements Spliterator.OfLong {
@Override
void arrayForOne(long[] array, int index, LongConsumer consumer) {
consumer.accept(array[index]);
}
@Override
Spliterator.OfLong arraySpliterator(long[] array, int offset, int len) {
return Arrays.spliterator(array, offset, offset+len);
}
};
return new Splitr();
}
@Override
public String toString() {
long[] array = asLongArray();
if (array.length < 200) {
return String.format("%s[length=%d, chunks=%d]%s",
getClass().getSimpleName(), array.length,
spineIndex, Arrays.toString(array));
}
else {
long[] array2 = Arrays.copyOf(array, 200);
return String.format("%s[length=%d, chunks=%d]%s...",
getClass().getSimpleName(), array.length,
spineIndex, Arrays.toString(array2));
}
}
}
/**
* An ordered collection of {@code double} values.
*/
static class OfDouble
extends SpinedBuffer.OfPrimitive
implements DoubleConsumer {
OfDouble() { }
OfDouble(int initialCapacity) {
super(initialCapacity);
}
@Override
public void forEach(Consumer super Double> consumer) {
if (consumer instanceof DoubleConsumer) {
forEach((DoubleConsumer) consumer);
}
else {
if (Tripwire.ENABLED)
Tripwire.trip(getClass(), "{0} calling SpinedBuffer.OfDouble.forEach(Consumer)");
spliterator().forEachRemaining(consumer);
}
}
@Override
protected double[][] newArrayArray(int size) {
return new double[size][];
}
@Override
protected double[] newArray(int size) {
return new double[size];
}
@Override
protected int arrayLength(double[] array) {
return array.length;
}
@Override
protected void arrayForEach(double[] array,
int from, int to,
DoubleConsumer consumer) {
for (int i = from; i < to; i++)
consumer.accept(array[i]);
}
@Override
public void accept(double i) {
preAccept();
curChunk[elementIndex++] = i;
}
public double get(long index) {
int ch = chunkFor(index);
if (spineIndex == 0 && ch == 0)
return curChunk[(int) index];
else
return spine[ch][(int) (index-priorElementCount[ch])];
}
public double[] asDoubleArray() {
return asPrimitiveArray();
}
@Override
public PrimitiveIterator.OfDouble iterator() {
return Spliterators.iteratorFromSpliterator(spliterator());
}
public Spliterator.OfDouble spliterator() {
class Splitr extends BaseSpliterator
implements Spliterator.OfDouble {
@Override
void arrayForOne(double[] array, int index, DoubleConsumer consumer) {
consumer.accept(array[index]);
}
@Override
Spliterator.OfDouble arraySpliterator(double[] array, int offset, int len) {
return Arrays.spliterator(array, offset, offset+len);
}
}
return new Splitr();
}
@Override
public String toString() {
double[] array = asDoubleArray();
if (array.length < 200) {
return String.format("%s[length=%d, chunks=%d]%s",
getClass().getSimpleName(), array.length,
spineIndex, Arrays.toString(array));
}
else {
double[] array2 = Arrays.copyOf(array, 200);
return String.format("%s[length=%d, chunks=%d]%s...",
getClass().getSimpleName(), array.length,
spineIndex, Arrays.toString(array2));
}
}
}
}