fix: support multiple PPS for AVC and HEVC

Eyevinn · Feb 1, 2024 · 905fd5b · 905fd5b
1 parent 7aef2ea
commit 905fd5b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/internal/avc.go b/internal/avc.go
@@ -12,7 +12,7 @@ type AvcPS struct {
 	spss       map[uint32]*avc.SPS
 	ppss       map[uint32]*avc.PPS
 	spsnalu    []byte
-	ppsnalus   [][]byte
+	ppsnalus   map[uint32][]byte
 	Statistics StreamStatistics
 }
 
@@ -31,7 +31,7 @@ func (a *AvcPS) setSPS(nalu []byte) error {
 	if a.spss == nil {
 		a.spss = make(map[uint32]*avc.SPS, 1)
 		a.ppss = make(map[uint32]*avc.PPS, 1)
-		a.ppsnalus = make([][]byte, 1)
+		a.ppsnalus = make(map[uint32][]byte)
 	}
 	sps, err := avc.ParseSPSNALUnit(nalu, true)
 	if err != nil {

diff --git a/internal/hevc.go b/internal/hevc.go
@@ -14,15 +14,15 @@ type HevcPS struct {
 	ppss       map[uint32]*hevc.PPS
 	vpsnalu    []byte
 	spsnalu    []byte
-	ppsnalus   [][]byte
+	ppsnalus   map[uint32][]byte
 	Statistics StreamStatistics
 }
 
 func (a *HevcPS) setSPS(nalu []byte) error {
 	if a.spss == nil {
 		a.spss = make(map[uint32]*hevc.SPS, 1)
 		a.ppss = make(map[uint32]*hevc.PPS, 1)
-		a.ppsnalus = make([][]byte, 1)
+		a.ppsnalus = make(map[uint32][]byte, 1)
 	}
 	sps, err := hevc.ParseSPSNALUnit(nalu)
 	if err != nil {