From a9a34f7a4ddded2a40b9e2b941bdd0d3b16e63e7 Mon Sep 17 00:00:00 2001
From: nareix <hi.nareix@gmail.com>
Date: Sun, 17 Jul 2016 20:09:03 +0800
Subject: [PATCH] example add framedrop

---
 examples/rtmp_server_channels/main.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/examples/rtmp_server_channels/main.go b/examples/rtmp_server_channels/main.go
index 12b165c..478636a 100644
--- a/examples/rtmp_server_channels/main.go
+++ b/examples/rtmp_server_channels/main.go
@@ -4,6 +4,7 @@ import (
 	"sync"
 	"fmt"
 	"time"
+	"github.com/nareix/joy4/av"
 	"github.com/nareix/joy4/format"
 	"github.com/nareix/joy4/av/avutil"
 	"github.com/nareix/joy4/av/pubsub"
@@ -15,6 +16,20 @@ func init() {
 	format.RegisterAll()
 }
 
+type FrameDropper struct {
+	Interval int
+	n int
+}
+
+func (self *FrameDropper) ModifyPacket(pkt *av.Packet, streams []av.CodecData, videoidx int, audioidx int) (drop bool, err error) {
+	if self.Interval != 0 && self.n >= self.Interval && pkt.Idx == int8(videoidx) && !pkt.IsKeyFrame {
+		drop = true
+		self.n = 0
+	}
+	self.n++
+	return
+}
+
 func main() {
 	server := &rtmp.Server{}
 
@@ -45,6 +60,11 @@ func main() {
 				filters = append(filters, &pktque.WaitKeyFrame{})
 			}
 			filters = append(filters, &pktque.FixTime{StartFromZero: true})
+			if q := query.Get("framedrop"); q != "" {
+				n := 0
+				fmt.Sscanf(q, "%d", &n)
+				filters = append(filters, &FrameDropper{Interval: n})
+			}
 			demuxer := &pktque.FilterDemuxer{
 				Filter: filters,
 				Demuxer: cursor,