Split into multiple files

This commit is contained in:
kageru 2019-10-08 17:36:32 +02:00
parent 020143ffc6
commit 0aeab3a96b
Signed by: kageru
GPG Key ID: 8282A2BEA4ADA3D2
2 changed files with 41 additions and 39 deletions

View File

@ -0,0 +1,41 @@
package moe.kageru.sekwences
/**
* Sequence that holds a [CoalescingIterator].
*/
internal class CoalescingSequence<T>(private val source: Sequence<T>, private inline val merger: (T, T) -> T?) : Sequence<T> {
override fun iterator(): Iterator<T> {
return CoalescingIterator(source.iterator(), merger)
}
/**
* Iterator that will attempt to merge adjacent elements as described in the KDoc for [coalesce].
*/
internal class CoalescingIterator<T>(private val source: Iterator<T>, private inline val merger: (T, T) -> T?) : Iterator<T> {
private var previous: T = if (source.hasNext()) source.next() else error("Please don’t pass empty iterators in here")
// The reason we need this marker is that our iterator can still hold one value, even if the source has been drained.
private var hasNext = true
override fun hasNext() = hasNext
override tailrec fun next(): T {
if (!source.hasNext()) {
hasNext = false
return previous
}
val current = source.next()
val merged = merger(previous, current)
// If the elements can’t be merged, return the first, then save the second for next time.
return if (merged == null) {
val ret = previous
previous = current
ret
} else {
previous = merged
next()
}
}
}
}

View File

@ -18,42 +18,3 @@ fun <T> Sequence<T>.coalesce(merger: (T, T) -> T?): Sequence<T> {
emptySequence()
}
}
/**
* Sequence that holds a [CoalescingIterator].
*/
internal class CoalescingSequence<T>(private val source: Sequence<T>, private inline val merger: (T, T) -> T?) : Sequence<T> {
override fun iterator(): Iterator<T> {
return CoalescingIterator(source.iterator(), merger)
}
}
/**
* Iterator that will attempt to merge adjacent elements as described in the KDoc for [coalesce].
*/
internal class CoalescingIterator<T>(private val source: Iterator<T>, private inline val merger: (T, T) -> T?) : Iterator<T> {
private var previous: T = if (source.hasNext()) source.next() else error("Please don’t pass empty iterators in here")
// The reason we need this marker is that our iterator can still hold one value, even if the source has been drained.
private var hasNext = true
override fun hasNext() = hasNext
override tailrec fun next(): T {
if (!source.hasNext()) {
hasNext = false
return previous
}
val current = source.next()
val merged = merger(previous, current)
// If the elements can’t be merged, return the first, then save the second for next time.
return if (merged == null) {
val ret = previous
previous = current
ret
} else {
previous = merged
next()
}
}
}