drop frames if AVAssetWriter not ready for more data

CapSoftware · Nov 21, 2024 · 9842913 · 9842913 · vercel · Nov 21, 2024
1 parent d3220ac
commit 9842913
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/crates/media/src/encoders/h264_avassetwriter.rs b/crates/media/src/encoders/h264_avassetwriter.rs
@@ -74,6 +74,10 @@ impl H264AVAssetWriterEncoder {
     }
 
     fn queue_frame(&mut self, frame: screencapturekit::cm_sample_buffer::CMSampleBuffer) {
+        if !self.video_input.is_ready_for_more_media_data() {
+            return;
+        }
+
         let sample_buf = unsafe {
             let ptr = &*frame.sys_ref as *const _ as *const cm::SampleBuf;
             &*ptr