Modified reorderLocally for my purposes

Reorder.scala

def reorderLocally[I](f: I => Long, range:Long):Process1[I, I] = { // TODO: Can we replace Long with something like A: Order ?
  import scala.collection.immutable.SortedMap

  def emitMapValues(m: SortedMap[Long, Vector[I]]) =
    emitAll(m.foldLeft(Vector.empty[I]) { case (acc, (_, tss)) => acc ++ tss })

  def go(buffered: SortedMap[Long, Vector[I]]): Process1[I, I] = {
    receive1Or[I, I](emitMapValues(buffered)) { t =>
      val until = f(t) - range
      val (toEmit, toBuffer) = buffered span { case (x, _) => x <= until }
      val updatedBuffer = toBuffer + (f(t) -> (toBuffer.getOrElse(f(t), Vector.empty[I]) :+ t))
      emitMapValues(toEmit) ++ go(updatedBuffer)
    }
  }
  go(SortedMap.empty)
}