Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Read the frame data out of the CMSampleBuffer

Browse source 
Still not sending it anywhere, but think I'm reading it correctly.
This commit is contained in:
Nathan Sobo 2022-08-31 13:36:29 -06:00 • committed by Antonio Scandurra
parent 047b5114f1
commit b51abc5a63
5 changed files with 69 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

20
crates/capture/src/main.rs
View file

@ -3,6 +3,7 @@ mod compression_session;
use crate::{bindings::SCStreamOutputType, compression_session::CompressionSession};
use block::ConcreteBlock;
use byteorder::{BigEndian, ReadBytesExt};
use bytes::BytesMut;
use cocoa::{
base::{id, nil, YES},
@ -128,6 +129,25 @@ impl ScreenCaptureView {
let nal_unit = compression_buffer.split();
}
}
let data = sample_buffer.data();
let mut data = data.bytes();
const AVCC_HEADER_LENGTH: usize = 4;
while data.len() - AVCC_HEADER_LENGTH > 0 {
let nal_unit_len = match data.read_u32::<BigEndian>() {
Ok(len) => len as usize,
Err(error) => {
log::error!("error decoding nal unit length: {}", error);
return;
}
};
compression_buffer.extend_from_slice(&START_CODE);
compression_buffer.extend_from_slice(&data[..nal_unit_len as usize]);
data = &data[nal_unit_len..];
let nal_unit = compression_buffer.split();
}
},
)
.unwrap();