Kieran/zap-stream-core
1
0
Fork 0
mirror of https://github.com/v0l/zap-stream-core.git synced 2025-06-22 14:28:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: recording

Browse Source 
closes #9
This commit is contained in:
Kieran
2025-06-20 10:30:22 +01:00
parent aea0feef05
commit b1ebf75244
3 changed files with 21 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
crates/core/src/egress/recorder.rs
View File

@ -1,6 +1,7 @@
use anyhow::Result;
use ffmpeg_rs_raw::ffmpeg_sys_the_third::AVPacket;
use ffmpeg_rs_raw::{Encoder, Muxer};
use log::info;
use std::collections::HashMap;
use std::fs;
use std::path::PathBuf;
@ -21,7 +22,7 @@ impl RecorderEgress {
out_dir: PathBuf,
variants: impl Iterator<Item = (&'a VariantStream, &'a Encoder)>,
) -> Result<Self> {
let out_file = out_dir.join("recording.ts");
let out_file = out_dir.join("recording.mp4");
let mut var_map = HashMap::new();
let muxer = unsafe {
let mut m = Muxer::builder()
@ -31,7 +32,10 @@ impl RecorderEgress {
let stream = m.add_stream_encoder(enc)?;
var_map.insert(var.id(), (*stream).index);
}
m.open(None)?;
let mut options = HashMap::new();
options.insert("movflags".to_string(), "faststart".to_string());
m.open(Some(options))?;
m
};
Ok(Self { muxer, var_map })

7
crates/core/src/pipeline/runner.rs
View File

@ -22,8 +22,8 @@ use ffmpeg_rs_raw::ffmpeg_sys_the_third::AVCodecID::AV_CODEC_ID_WEBP;
use ffmpeg_rs_raw::ffmpeg_sys_the_third::AVPictureType::AV_PICTURE_TYPE_NONE;
use ffmpeg_rs_raw::ffmpeg_sys_the_third::AVPixelFormat::AV_PIX_FMT_YUV420P;
use ffmpeg_rs_raw::ffmpeg_sys_the_third::{
av_frame_clone, av_frame_free, av_get_sample_fmt, av_packet_free, av_rescale_q, AVFrame,
AVPacket, AV_NOPTS_VALUE,
av_frame_clone, av_frame_free, av_get_sample_fmt, av_packet_clone, av_packet_free,
av_rescale_q, AVFrame, AVPacket, AV_NOPTS_VALUE,
};
use ffmpeg_rs_raw::{
cstr, get_frame_from_hw, AudioFifo, Decoder, Demuxer, Encoder, Resample, Scaler, StreamType,
@ -529,7 +529,8 @@ impl PipelineRunner {
// pass new packets to egress
for mut pkt in packets {
for eg in egress.iter_mut() {
let er = eg.process_pkt(pkt, &var.id())?;
let pkt_clone = av_packet_clone(pkt);
let er = eg.process_pkt(pkt_clone, &var.id())?;
ret.push(er);
}
av_packet_free(&mut pkt);