From 9a54f1ee8f879f51098e6126f36848a8343702f7 Mon Sep 17 00:00:00 2001
From: nareix <hi.nareix@gmail.com>
Date: Sat, 16 Jul 2016 08:38:57 +0800
Subject: [PATCH] ts: support PaddingToMakeCounterCont

---
 format/ts/muxer.go | 21 ++++++---------------
 1 file changed, 6 insertions(+), 15 deletions(-)

diff --git a/format/ts/muxer.go b/format/ts/muxer.go
index 1f2e35c..6fb64b5 100644
--- a/format/ts/muxer.go
+++ b/format/ts/muxer.go
@@ -67,28 +67,16 @@ func (self *Muxer) newStream(codec av.CodecData) (err error) {
 	return
 }
 
-/*
-func (self *Muxer) writePaddingTSPackets(tsw *TSWriter) (err error) {
+func (self *Muxer) writePaddingTSPackets(tsw *tsio.TSWriter) (err error) {
 	for tsw.ContinuityCounter&0xf != 0x0 {
-		header := TSHeader{
-			PID:               tsw.PID,
-			ContinuityCounter: tsw.ContinuityCounter,
-		}
-		if _, err = WriteTSHeader(self.w, header, 0); err != nil {
+		if err = tsw.WritePackets(self.w, self.datav[:0], 0, false, true); err != nil {
 			return
 		}
-		tsw.ContinuityCounter++
 	}
 	return
 }
-*/
 
 func (self *Muxer) WriteTrailer() (err error) {
-	if err = self.w.Flush(); err != nil {
-		return
-	}
-
-	/*
 	if self.PaddingToMakeCounterCont {
 		for _, stream := range self.streams {
 			if err = self.writePaddingTSPackets(stream.tsw); err != nil {
@@ -96,7 +84,10 @@ func (self *Muxer) WriteTrailer() (err error) {
 			}
 		}
 	}
-	*/
+
+	if err = self.w.Flush(); err != nil {
+		return
+	}
 	return
 }