pktque: change SetMaxSize to SetMaxGopCount

av/pktque/buf.go

@@ -7,31 +7,29 @@ import (
 type Buf struct {
 	Head, Tail BufPos
 	pkts       []av.Packet
-	size, maxsize int
-	count         int
+	Size       int
+	Count      int
 }
 
 func NewBuf() *Buf {
 	return &Buf{
 		pkts: make([]av.Packet, 64),
-		maxsize: 1024 * 512,
 	}
 }
 
-func (self *Buf) SetMaxSize(size int) {
-	self.maxsize = size
-	self.shrink()
+func (self *Buf) Pop() av.Packet {
+	if self.Count == 0 {
+		panic("pktque.Buf: Pop() when count == 0")
 	}
 
-func (self *Buf) shrink() {
-	for self.size > self.maxsize && self.count > 1 {
 	i := int(self.Head) & (len(self.pkts) - 1)
 	pkt := self.pkts[i]
 	self.pkts[i] = av.Packet{}
-		self.size -= len(pkt.Data)
+	self.Size -= len(pkt.Data)
 	self.Head++
-		self.count--
-	}
+	self.Count--
+
+	return pkt
 }
 
 func (self *Buf) grow() {
@@ -43,14 +41,13 @@ func (self *Buf) grow() {
 }
 
 func (self *Buf) Push(pkt av.Packet) {
-	if self.count == len(self.pkts) {
+	if self.Count == len(self.pkts) {
 		self.grow()
 	}
 	self.pkts[int(self.Tail)&(len(self.pkts)-1)] = pkt
 	self.Tail++
-	self.count++
-	self.size += len(pkt.Data)
-	self.shrink()
+	self.Count++
+	self.Size += len(pkt.Data)
 }
 
 func (self *Buf) Get(pos BufPos) av.Packet {

av/pubsub/queue.go

@@ -1,4 +1,3 @@
-
 // Packege pubsub implements publisher-subscribers model used in multi-channel streaming.
 package pubsub
 
@@ -6,8 +5,8 @@ import (
 	"github.com/nareix/joy4/av"
 	"github.com/nareix/joy4/av/pktque"
 	"io"
-	"time"
 	"sync"
+	"time"
 )
 
 //        time
@@ -26,7 +25,7 @@ type Queue struct {
 	head, tail               int
 	lock                     *sync.RWMutex
 	cond                     *sync.Cond
-	maxdur time.Duration
+	curgopcount, maxgopcount int
 	streams                  []av.CodecData
 	videoidx                 int
 	closed                   bool
@@ -35,16 +34,16 @@ type Queue struct {
 func NewQueue() *Queue {
 	q := &Queue{}
 	q.buf = pktque.NewBuf()
-	q.maxdur = time.Second*10
+	q.maxgopcount = 2
 	q.lock = &sync.RWMutex{}
 	q.cond = sync.NewCond(q.lock.RLocker())
 	q.videoidx = -1
 	return q
 }
 
-func (self *Queue) SetMaxSize(size int) {
+func (self *Queue) SetMaxGopCount(n int) {
 	self.lock.Lock()
-	self.buf.SetMaxSize(size)
+	self.maxgopcount = n
 	self.lock.Unlock()
 	return
 }
@@ -79,6 +78,21 @@ func (self *Queue) WritePacket(pkt av.Packet) (err error) {
 	self.lock.Lock()
 
 	self.buf.Push(pkt)
+	if pkt.Idx == int8(self.videoidx) && pkt.IsKeyFrame {
+		self.curgopcount++
+	}
+
+	for self.curgopcount >= self.maxgopcount && self.buf.Count > 1 {
+		pkt := self.buf.Pop()
+		if pkt.Idx == int8(self.videoidx) && pkt.IsKeyFrame {
+			self.curgopcount--
+		}
+		if self.curgopcount < self.maxgopcount {
+			break
+		}
+	}
+	//println("shrink", self.curgopcount, self.maxgopcount, self.buf.Head, self.buf.Tail, "count", self.buf.Count, "size", self.buf.Size)
+
 	self.cond.Broadcast()
 
 	self.lock.Unlock()
@@ -195,4 +209,3 @@ func (self *QueueCursor) ReadPacket() (pkt av.Packet, err error) {
 	self.que.cond.L.Unlock()
 	return
 }
-