37 lines
1.1 KiB
Kotlin
37 lines
1.1 KiB
Kotlin
package jp.juggler.apng.util
|
|
|
|
import java.util.*
|
|
import kotlin.math.min
|
|
|
|
internal class ByteSequenceQueue(private val bufferRecycler: (ByteSequence) -> Unit) {
|
|
|
|
private val list = LinkedList<ByteSequence>()
|
|
|
|
val remain: Int
|
|
get() = list.sumOf { it.length }
|
|
|
|
fun add(range: ByteSequence) = list.add(range)
|
|
|
|
fun clear() = list.onEach(bufferRecycler).clear()
|
|
|
|
fun readBytes(dst: ByteArray, offset: Int, length: Int): Int {
|
|
var dstOffset = offset
|
|
var dstRemain = length
|
|
while (dstRemain > 0 && list.isNotEmpty()) {
|
|
val item = list.first()
|
|
if (item.length <= 0) {
|
|
bufferRecycler(item)
|
|
list.removeFirst()
|
|
} else {
|
|
val delta = min(item.length, dstRemain)
|
|
System.arraycopy(item.array, item.offset, dst, dstOffset, delta)
|
|
dstOffset += delta
|
|
dstRemain -= delta
|
|
item.offset += delta
|
|
item.length -= delta
|
|
}
|
|
}
|
|
return length - dstRemain
|
|
}
|
|
}
|