From bee47726fa464387f67cff2a56e727f83dd0784a Mon Sep 17 00:00:00 2001
From: aler9 <46489434+aler9@users.noreply.github.com>
Date: Sun, 12 Dec 2021 23:06:41 +0100
Subject: [PATCH] simplify example

---
 examples/client-read-h264-save-to-disk/mpegtsencoder.go | 8 ++++----
 1 file changed, 4 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 fef99efa..6d888a98 100644
--- a/examples/client-read-h264-save-to-disk/mpegtsencoder.go
+++ b/examples/client-read-h264-save-to-disk/mpegtsencoder.go
@@ -78,15 +78,15 @@ func (e *mpegtsEncoder) encode(nalus [][]byte, pts time.Duration) error {
 	}
 
 	for _, nalu := range nalus {
-		// remove existing SPS, PPS, AUD
+
 		typ := h264.NALUType(nalu[0] & 0x1F)
 		switch typ {
 		case h264.NALUTypeSPS, h264.NALUTypePPS, h264.NALUTypeAccessUnitDelimiter:
+			// remove existing SPS, PPS, AUD
 			continue
-		}
 
-		// add SPS and PPS before every IDR
-		if typ == h264.NALUTypeIDR {
+		case h264.NALUTypeIDR:
+			// add SPS and PPS before every IDR
 			filteredNALUs = append(filteredNALUs, e.h264Conf.SPS, e.h264Conf.PPS)
 		}