rust-ffmpeg/examples/remux.rs

use std::env;

use ffmpeg::{format, media};

fn main() {
	let input_file = env::args().nth(1).expect("missing input file");
	let output_file = env::args().nth(2).expect("missing output file");

	ffmpeg::init().unwrap();

	let mut ictx = format::input(&input_file).unwrap();
	let mut octx = format::output(&output_file).unwrap();

	let mut stream_mapping = vec![0; ictx.nb_streams() as _];
	let mut ist_time_bases = vec![None; ictx.nb_streams() as _];
	let mut ost_index = 0;
	for (ist_index, ist) in ictx.streams().enumerate() {
		let codec_par = ist.parameters();
		let ist_medium = codec_par.medium();
		if ist_medium != media::Type::Audio && ist_medium != media::Type::Video && ist_medium != media::Type::Subtitle {
			stream_mapping[ist_index] = -1;
			continue;
		}
		stream_mapping[ist_index] = ost_index;
		ist_time_bases[ist_index] = ist.time_base();
		ost_index += 1;
		let mut ost = octx.add_stream().unwrap();
		ost.set_parameters(ist.parameters());
		// We need to set codec_tag to 0 lest we run into incompatible codec tag
		// issues when muxing into a different container format. Unfortunately
		// there's no high level API to do this (yet).
		unsafe {
			(*ost.parameters().as_mut_ptr()).codec_tag = 0;
		}
	}

	octx.set_metadata(ictx.metadata().to_owned());
	octx.write_header().unwrap();

	for res in ictx.packets() {
		let (stream, mut packet) = res.unwrap();
		let ist_index = stream.index();
		let ost_index = stream_mapping[ist_index];
		if ost_index < 0 {
			continue;
		}
		let ost = octx.stream(ost_index as _).unwrap();
		packet.rescale_ts(ist_time_bases[ist_index].unwrap(), ost.time_base().unwrap());
		packet.set_position(-1);
		packet.set_stream(ost_index as _);
		packet.write_interleaved(&mut octx).unwrap();
	}

	octx.write_trailer().unwrap();
}