lkm/stream/jitter_buffer.go

package stream

import "math"

type JitterBuffer struct {
	maxSeqNum  uint16
	minSeqNum  uint16
	nextSeqNum uint16

	count         int
	minStartCount int

	queue    []interface{}
	onPacket func(packet interface{})
	first    bool
}

func (j *JitterBuffer) emit() {
	if j.first {
		j.nextSeqNum = j.minSeqNum
		j.first = false
	}
	if j.nextSeqNum > j.maxSeqNum {
		j.nextSeqNum = j.minSeqNum
	}

	for j.queue[j.nextSeqNum] == nil {
		j.nextSeqNum++
	}

	j.onPacket(j.queue[j.nextSeqNum])
	j.queue[j.nextSeqNum] = nil
	j.nextSeqNum++
	j.minSeqNum = uint16(math.Min(float64(j.nextSeqNum), float64(j.maxSeqNum)))
	j.count--
}

func (j *JitterBuffer) Push(seq uint16, packet interface{}) {
	if j.count == 0 {
		j.minSeqNum = seq
		j.maxSeqNum = seq
	}

	if j.queue[seq] == nil {
		j.queue[seq] = packet
		j.count++
	}

	j.minSeqNum = uint16(math.Min(float64(j.minSeqNum), float64(seq)))
	j.maxSeqNum = uint16(math.Max(float64(j.maxSeqNum), float64(seq)))

	if j.count > j.minStartCount {
		j.emit()
	}
}

func (j *JitterBuffer) Flush() {
	for j.count > 0 {
		j.emit()
	}
}

func (j *JitterBuffer) Close() {
	j.onPacket = nil
}

func NewJitterBuffer(handler func(packet interface{})) *JitterBuffer {
	return &JitterBuffer{
		queue:         make([]interface{}, 0xFFFF+1),
		onPacket:      handler,
		minStartCount: 50,
		first:         true,
	}
}