scala.compat.java8.collectionImpl.DoubleAccumulator.scala Maven / Gradle / Ivy
The newest version!
package scala.compat.java8.collectionImpl
/** A `DoubleAccumulator` is a low-level collection specialized for gathering
* elements in parallel and then joining them in order by merging them.
* This is a manually specialized variant of `Accumulator` with no actual
* subclassing relationship with `Accumulator`.
*/
final class DoubleAccumulator extends AccumulatorLike[Double, DoubleAccumulator] { self =>
private[java8] var current: Array[Double] = DoubleAccumulator.emptyDoubleArray
private[java8] var history: Array[Array[Double]] = DoubleAccumulator.emptyDoubleArrayArray
private[java8] def cumulative(i: Int) = { val x = history(i); x(x.length-1).toLong }
private def expand(): Unit = {
if (index > 0) {
current(current.length-1) = (if (hIndex > 0) { val x = history(hIndex-1); x(x.length-1) } else 0) + index
if (hIndex >= history.length) hExpand()
history(hIndex) = current
hIndex += 1
}
current = new Array[Double](nextBlockSize+1)
index = 0
}
private def hExpand(): Unit = {
if (hIndex == 0) history = new Array[Array[Double]](4)
else history = java.util.Arrays.copyOf(history, history.length << 1)
}
/** Appends an element to this `DoubleAccumulator`. */
final def +=(a: Double): Unit = {
totalSize += 1
if (index+1 >= current.length) expand()
current(index) = a
index += 1
}
/** Removes all elements from `that` and appends them to this `DoubleAccumulator`. */
final def drain(that: DoubleAccumulator): Unit = {
var h = 0
var prev = 0L
var more = true
while (more && h < that.hIndex) {
val cuml = that.cumulative(h)
val n = (cuml - prev).toInt
if (current.length - index - 1 >= n) {
System.arraycopy(that.history(h), 0, current, index, n)
prev = cuml
index += n
h += 1
}
else more = false
}
if (h >= that.hIndex && current.length - index - 1>= that.index) {
if (that.index > 0) System.arraycopy(that.current, 0, current, index, that.index)
index += that.index
}
else {
val slots = (if (index > 0) 1 else 0) + that.hIndex - h
if (hIndex + slots > history.length) {
val n = math.max(4, 1 << (32 - java.lang.Integer.numberOfLeadingZeros(1 + hIndex + slots)))
history = java.util.Arrays.copyOf(history, n)
}
var pv = (if (hIndex > 0) cumulative(hIndex-1) else 0L)
if (index > 0) {
val x =
if (index < (current.length >>> 3) && current.length - 1 > 32) {
val ans = java.util.Arrays.copyOf(current, index + 1)
ans(ans.length - 1) = current(current.length - 1)
ans
}
else current
pv = pv + index
x(x.length - 1) = pv
history(hIndex) = x
hIndex += 1
}
while (h < that.hIndex) {
val cuml = that.cumulative(h)
pv = pv + cuml - prev
prev = cuml
val x = that.history(h)
x(x.length - 1) = pv
history(hIndex) = x
h += 1
hIndex += 1
}
index = that.index
current = that.current
}
totalSize += that.totalSize
that.clear
}
override def clear(): Unit = {
super.clear()
current = DoubleAccumulator.emptyDoubleArray
history = DoubleAccumulator.emptyDoubleArrayArray
}
/** Retrieves the `ix`th element. */
final def apply(ix: Long): Double = {
if (totalSize - ix <= index || hIndex == 0) current((ix - (totalSize - index)).toInt)
else {
val w = seekSlot(ix)
history((w >>> 32).toInt)((w & 0xFFFFFFFFL).toInt)
}
}
/** Retrieves the `ix`th element, using an `Int` index. */
final def apply(i: Int): Double = apply(i.toLong)
/** Returns a `DoubleStepper` over the contents of this `DoubleAccumulator`. */
final def stepper: DoubleStepper = new DoubleAccumulatorStepper(this)
/** Returns an `Iterator` over the contents of this `DoubleAccumulator`. The `Iterator` is not specialized. */
final def iterator = stepper.iterator
/** Returns a `java.util.Spliterator.OfDouble` over the contents of this `DoubleAccumulator`*/
final def spliterator: java.util.Spliterator.OfDouble = stepper
/** Produces a sequential Java 8 `DoubleStream` over the elements of this `DoubleAccumulator`*/
final def seqStream: java.util.stream.DoubleStream = java.util.stream.StreamSupport.doubleStream(spliterator, false)
/** Produces a parallel Java 8 `DoubleStream` over the elements of this `DoubleAccumulator`*/
final def parStream: java.util.stream.DoubleStream = java.util.stream.StreamSupport.doubleStream(spliterator, true)
/** Copies the elements in this `DoubleAccumulator` into an `Array[Double]` */
final def toArray = {
if (totalSize > Int.MaxValue) throw new IllegalArgumentException("Too many elements accumulated for an array: "+totalSize.toString)
val a = new Array[Double](totalSize.toInt)
var j = 0
var h = 0
var pv = 0L
while (h < hIndex) {
val x = history(h)
val cuml = x(x.length-1).toLong
val n = (cuml - pv).toInt
pv = cuml
System.arraycopy(x, 0, a, j, n)
j += n
h += 1
}
System.arraycopy(current, 0, a, j, index)
j += index
a
}
/** Copies the elements in this `DoubleAccumulator` to a `List` */
final def toList: List[Double] = {
var ans: List[Double] = Nil
var i = index - 1
while (i >= 0) {
ans = current(i) :: ans
i -= 1
}
var h = hIndex - 1
while (h >= 0) {
val a = history(h)
i = (cumulative(h) - (if (h == 0) 0L else cumulative(h-1))).toInt - 1
while (i >= 0) {
ans = a(i) :: ans
i -= 1
}
h -= 1
}
ans
}
/** Copies the elements in this `DoubleAccumulator` to a specified collection.
* Note that the target collection is not specialized.
* Usage example: `acc.to[Vector]`
*/
final def to[Coll[_]](implicit cbf: collection.generic.CanBuildFrom[Nothing, Double, Coll[Double]]): Coll[Double] = {
if (totalSize > Int.MaxValue) throw new IllegalArgumentException("Too many elements accumulated for a Scala collection: "+totalSize.toString)
val b = cbf()
b.sizeHint(totalSize.toInt)
var h = 0
var pv = 0L
while (h < hIndex) {
val x = history(h)
val n = cumulative(h) - pv
pv = cumulative(h)
var i = 0
while (i < n) {
b += x(i)
i += 1
}
h += 1
}
var i = 0
while (i < index) {
b += current(i)
i += 1
}
b.result
}
}
object DoubleAccumulator {
private val emptyDoubleArray = new Array[Double](0)
private val emptyDoubleArrayArray = new Array[Array[Double]](0)
/** A `Supplier` of `DoubleAccumulator`s, suitable for use with `java.util.stream.DoubleStream`'s `collect` method. Suitable for `Stream[Double]` also. */
def supplier = new java.util.function.Supplier[DoubleAccumulator]{ def get: DoubleAccumulator = new DoubleAccumulator }
/** A `BiConsumer` that adds an element to an `Accumulator`, suitable for use with `java.util.stream.DoubleStream`'s `collect` method. */
def adder = new java.util.function.ObjDoubleConsumer[DoubleAccumulator]{ def accept(ac: DoubleAccumulator, a: Double) { ac += a } }
/** A `BiConsumer` that adds a boxed `Double` to an `DoubleAccumulator`, suitable for use with `java.util.stream.Stream`'s `collect` method. */
def boxedAdder = new java.util.function.BiConsumer[DoubleAccumulator, Double]{ def accept(ac: DoubleAccumulator, a: Double) { ac += a } }
/** A `BiConsumer` that merges `DoubleAccumulator`s, suitable for use with `java.util.stream.DoubleStream`'s `collect` method. Suitable for `Stream[Double]` also. */
def merger = new java.util.function.BiConsumer[DoubleAccumulator, DoubleAccumulator]{ def accept(a1: DoubleAccumulator, a2: DoubleAccumulator) { a1 drain a2 } }
/** Builds a `DoubleAccumulator` from any `Double`-valued `TraversableOnce` */
def from[A](source: TraversableOnce[Double]) = {
val a = new DoubleAccumulator
source.foreach(a += _)
a
}
}
private[java8] class DoubleAccumulatorStepper(private val acc: DoubleAccumulator) extends DoubleStepper {
import java.util.Spliterator._
private var h = 0
private var i = 0
private var a = if (acc.hIndex > 0) acc.history(0) else acc.current
private var n = if (acc.hIndex > 0) acc.cumulative(0) else acc.index
private var N = acc.totalSize
private def duplicateSelf(limit: Long = N): DoubleAccumulatorStepper = {
val ans = new DoubleAccumulatorStepper(acc)
ans.h = h
ans.i = i
ans.a = a
ans.n = n
ans.N = limit
ans
}
private def loadMore(): Unit = {
h += 1
if (h < acc.hIndex) { a = acc.history(h); n = acc.cumulative(h) - acc.cumulative(h-1) }
else { a = acc.current; n = acc.index }
i = 0
}
def characteristics = ORDERED | SIZED | SUBSIZED | NONNULL
def estimateSize = N
def hasNext = N > 0
def nextDouble: Double =
if (n <= 0) throw new NoSuchElementException("next on empty Stepper")
else {
if (i >= n) loadMore()
val ans = a(i)
i += 1
N -= 1
ans
}
// Overridden for efficiency
override def tryStep(f: Double => Unit): Boolean =
if (N <= 0) false
else {
if (i >= n) loadMore()
f(a(i))
i += 1
N -= 1
true
}
// Overridden for efficiency
override def tryAdvance(f: java.util.function.DoubleConsumer): Boolean =
if (N <= 0) false
else {
if (i >= n) loadMore()
f.accept(a(i))
i += 1
N -= 1
true
}
// Overridden for efficiency
override def foreach(f: Double => Unit) {
while (N > 0) {
if (i >= n) loadMore()
val i0 = i
if ((n-i) > N) n = i + N.toInt
while (i < n) {
f(a(i))
i += 1
}
N -= (n - i0)
}
}
// Overridden for efficiency
override def forEachRemaining(f: java.util.function.DoubleConsumer) {
while (N > 0) {
if (i >= n) loadMore()
val i0 = i
if ((n-i) > N) n = i + N.toInt
while (i < n) {
f.accept(a(i))
i += 1
}
N -= (n - i0)
}
}
def substep(): DoubleStepper =
if (N <= 1) null
else {
val half = (N >> 1)
val M = (if (h <= 0) 0L else acc.cumulative(h-1)) + i
val R = M + half
val ans = duplicateSelf(half)
if (h < acc.hIndex) {
val w = acc.seekSlot(R)
h = (w >>> 32).toInt
if (h < acc.hIndex) {
a = acc.history(h)
n = acc.cumulative(h) - (if (h > 0) acc.cumulative(h-1) else 0)
}
else {
a = acc.current
n = acc.index
}
i = (w & 0xFFFFFFFFL).toInt
}
else i += half.toInt
N -= half
ans
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy