From 0c8df661b2a7d032dda6a20bd80be96c650a83c2 Mon Sep 17 00:00:00 2001
From: Juliusz Chroboczek <jch@irif.fr>
Date: Mon, 12 Oct 2020 15:20:34 +0200
Subject: [PATCH] Don't drop first frame when recording to disk.

---
 diskwriter/diskwriter.go | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/diskwriter/diskwriter.go b/diskwriter/diskwriter.go
index b45cde2..03b54b7 100644
--- a/diskwriter/diskwriter.go
+++ b/diskwriter/diskwriter.go
@@ -221,12 +221,22 @@ func newDiskConn(directory, label string, up conn.Up, remoteTracks []conn.UpTrac
 		var builder *samplebuilder.SampleBuilder
 		switch remote.Codec().Name {
 		case webrtc.Opus:
-			builder = samplebuilder.New(128, &codecs.OpusPacket{})
+			builder = samplebuilder.New(
+				16, &codecs.OpusPacket{},
+				samplebuilder.WithPartitionHeadChecker(
+					&codecs.OpusPartitionHeadChecker{},
+				),
+			)
 		case webrtc.VP8:
 			if conn.hasVideo {
 				return nil, errors.New("multiple video tracks not supported")
 			}
-			builder = samplebuilder.New(128, &codecs.VP8Packet{})
+			builder = samplebuilder.New(
+				128, &codecs.VP8Packet{},
+				samplebuilder.WithPartitionHeadChecker(
+					&codecs.VP8PartitionHeadChecker{},
+				),
+			)
 			conn.hasVideo = true
 		}
 		track := &diskTrack{