From e1e5b6e537ad14df42ea30f354ebc327766daa2c Mon Sep 17 00:00:00 2001
From: aler9 <46489434+aler9@users.noreply.github.com>
Date: Thu, 2 Dec 2021 12:34:08 +0100
Subject: [PATCH] simplify example

---
 examples/client-read-h264-save-to-disk/mpegtsencoder.go | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/examples/client-read-h264-save-to-disk/mpegtsencoder.go b/examples/client-read-h264-save-to-disk/mpegtsencoder.go
index 529a8e4a..951859b2 100644
--- a/examples/client-read-h264-save-to-disk/mpegtsencoder.go
+++ b/examples/client-read-h264-save-to-disk/mpegtsencoder.go
@@ -86,8 +86,7 @@ func (e *mpegtsEncoder) encode(nalus [][]byte, pts time.Duration) error {
 
 		// add SPS and PPS before every IDR
 		if typ == h264.NALUTypeIDR {
-			filteredNALUs = append(filteredNALUs, e.h264Conf.SPS)
-			filteredNALUs = append(filteredNALUs, e.h264Conf.PPS)
+			filteredNALUs = append(filteredNALUs, e.h264Conf.SPS, e.h264Conf.PPS)
 		}
 
 		filteredNALUs = append(filteredNALUs, nalu)
@@ -99,8 +98,8 @@ func (e *mpegtsEncoder) encode(nalus [][]byte, pts time.Duration) error {
 		return err
 	}
 
-	dts := e.dtsEst.Feed(pts - e.startPTS)
-	pts = pts - e.startPTS
+	pts -= e.startPTS
+	dts := e.dtsEst.Feed(pts)
 
 	// write TS packet
 	_, err = e.mux.WriteData(&astits.MuxerData{