#[macro_use]
extern crate trackable;
use clap::Parser;
use mpeg2ts::pes::{PesPacketReader, ReadPesPacket};
use mpeg2ts::ts::{ReadTsPacket, TsPacketReader, TsPacketWriter, WriteTsPacket};
use std::io::Write;
use trackable::error::Failure;
#[derive(Debug, Parser)]
struct Args {
#[arg(
long,
default_value="ts-packet",
value_parser = clap::builder::PossibleValuesParser::new(
["ts", "ts-packet", "pes-packet", "es-audio", "es-video"]
)
)]
output_type: String,
}
fn main() {
let args = Args::parse();
match args.output_type.as_str() {
"ts" => {
let mut writer = TsPacketWriter::new(std::io::stdout());
let mut reader = TsPacketReader::new(std::io::stdin());
while let Some(packet) = track_try_unwrap!(reader.read_ts_packet()) {
track_try_unwrap!(writer.write_ts_packet(&packet));
}
}
"ts-packet" => {
let mut reader = TsPacketReader::new(std::io::stdin());
while let Some(packet) = track_try_unwrap!(reader.read_ts_packet()) {
println!("{:?}", packet);
}
}
"pes-packet" => {
let mut reader = PesPacketReader::new(TsPacketReader::new(std::io::stdin()));
while let Some(packet) = track_try_unwrap!(reader.read_pes_packet()) {
println!("{:?} {} bytes", packet.header, packet.data.len());
}
}
"es-audio" => {
let mut reader = PesPacketReader::new(TsPacketReader::new(std::io::stdin()));
while let Some(packet) = track_try_unwrap!(reader.read_pes_packet()) {
if !packet.header.stream_id.is_audio() {
continue;
}
track_try_unwrap!(std::io::stdout()
.write_all(&packet.data)
.map_err(Failure::from_error));
}
}
"es-video" => {
let mut reader = PesPacketReader::new(TsPacketReader::new(std::io::stdin()));
while let Some(packet) = track_try_unwrap!(reader.read_pes_packet()) {
if !packet.header.stream_id.is_video() {
continue;
}
track_try_unwrap!(std::io::stdout()
.write_all(&packet.data)
.map_err(Failure::from_error));
}
}
_ => unreachable!(),
}
}