av1: add missing sequence header field (#166)

bluenviron · Jan 1, 2025 · 89f364e · 89f364e
1 parent d0642e1
commit 89f364e
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/pkg/codecs/av1/sequence_header.go b/pkg/codecs/av1/sequence_header.go
@@ -254,6 +254,7 @@ type SequenceHeader struct {
 	EnableCdef                     bool
 	EnableRestoration              bool
 	ColorConfig                    SequenceHeader_ColorConfig
+	FilmGrainParamsPresent         bool
 }
 
 // Unmarshal decodes a SequenceHeader.
@@ -523,7 +524,19 @@ func (h *SequenceHeader) Unmarshal(buf []byte) error {
 	h.EnableCdef = bits.ReadFlagUnsafe(buf, &pos)
 	h.EnableRestoration = bits.ReadFlagUnsafe(buf, &pos)
 
-	return h.ColorConfig.unmarshal(h.SeqProfile, buf, &pos)
+	err = h.ColorConfig.unmarshal(h.SeqProfile, buf, &pos)
+	if err != nil {
+		return err
+	}
+
+	err = bits.HasSpace(buf, pos, 1)
+	if err != nil {
+		return err
+	}
+
+	h.FilmGrainParamsPresent = bits.ReadFlagUnsafe(buf, &pos)
+
+	return nil
 }
 
 // Width returns the video width.