X-Git-Url: https://git.nihav.org/?p=nihav.git;a=blobdiff_plain;f=nihav-codec-support%2Fsrc%2Ftest%2Fdec_video.rs;h=91f576304bea198ae9c470fc9bb3a39f447ee1c2;hp=cca4cb44bbfb2d06289b17b035f99d865a3d25bf;hb=995e421a17b430ded9b6753d70d3b8c9ac106eed;hpb=b4d5b8515e75383b4fc59ea2813c90c615d59a96 diff --git a/nihav-codec-support/src/test/dec_video.rs b/nihav-codec-support/src/test/dec_video.rs index cca4cb4..91f5763 100644 --- a/nihav-codec-support/src/test/dec_video.rs +++ b/nihav-codec-support/src/test/dec_video.rs @@ -1,194 +1,16 @@ //! Routines for testing decoders. use std::fs::File; -use std::io::prelude::*; use nihav_core::frame::*; use nihav_core::codecs::*; use nihav_core::demuxers::*; //use nihav_core::io::byteio::*; -use nihav_core::scale::*; +//use nihav_core::scale::*; use super::wavwriter::WavWriter; use super::md5::MD5; +use crate::imgwrite::write_pnm; pub use super::ExpectedTestResult; -const OUTPUT_PREFIX: &str = "assets/test_out"; - -fn write_pgmyuv(pfx: &str, strno: usize, num: u64, frm: NAFrameRef) { - if let NABufferType::None = frm.get_buffer() { return; } - let name = format!("{}/{}out{:02}_{:06}.pgm", OUTPUT_PREFIX, pfx, strno, num); - let mut ofile = File::create(name).unwrap(); - let buf = frm.get_buffer().get_vbuf().unwrap(); - let (w, h) = buf.get_dimensions(0); - let (w2, h2) = buf.get_dimensions(1); - let has_alpha = buf.get_info().get_format().has_alpha(); - let mut tot_h = h + h2; - if has_alpha { - tot_h += h; - } - if w2 > w/2 { - tot_h += h2; - } - let hdr = format!("P5\n{} {}\n255\n", w, tot_h); - ofile.write_all(hdr.as_bytes()).unwrap(); - let dta = buf.get_data(); - let ls = buf.get_stride(0); - let mut idx = 0; - let mut idx2 = w; - let is_flipped = buf.get_info().is_flipped(); - if is_flipped { - idx += h * ls; - idx2 += h * ls; - } - for _ in 0..h { - if is_flipped { - idx -= ls; - idx2 -= ls; - } - let line = &dta[idx..idx2]; - ofile.write_all(line).unwrap(); - if !is_flipped { - idx += ls; - idx2 += ls; - } - } - if w2 <= w/2 { - let pad: Vec = vec![0xFF; (w - w2 * 2) / 2]; - let mut base1 = buf.get_offset(1); - let stride1 = buf.get_stride(1); - let mut base2 = buf.get_offset(2); - let stride2 = buf.get_stride(2); - if is_flipped { - base1 += h2 * stride1; - base2 += h2 * stride2; - } - for _ in 0..h2 { - if is_flipped { - base1 -= stride1; - base2 -= stride2; - } - let bend1 = base1 + w2; - let line = &dta[base1..bend1]; - ofile.write_all(line).unwrap(); - ofile.write_all(pad.as_slice()).unwrap(); - - let bend2 = base2 + w2; - let line = &dta[base2..bend2]; - ofile.write_all(line).unwrap(); - ofile.write_all(pad.as_slice()).unwrap(); - - if !is_flipped { - base1 += stride1; - base2 += stride2; - } - } - } else { - let pad: Vec = vec![0xFF; w - w2]; - let mut base1 = buf.get_offset(1); - let stride1 = buf.get_stride(1); - if is_flipped { - base1 += h2 * stride1; - } - for _ in 0..h2 { - if is_flipped { - base1 -= stride1; - } - let bend1 = base1 + w2; - let line = &dta[base1..bend1]; - ofile.write_all(line).unwrap(); - ofile.write_all(pad.as_slice()).unwrap(); - if !is_flipped { - base1 += stride1; - } - } - let mut base2 = buf.get_offset(2); - let stride2 = buf.get_stride(2); - if is_flipped { - base2 += h2 * stride2; - } - for _ in 0..h2 { - if is_flipped { - base2 -= stride2; - } - let bend2 = base2 + w2; - let line = &dta[base2..bend2]; - ofile.write_all(line).unwrap(); - ofile.write_all(pad.as_slice()).unwrap(); - if !is_flipped { - base2 += stride2; - } - } - } - if has_alpha { - let ls = buf.get_stride(3); - let mut idx = buf.get_offset(3); - let mut idx2 = idx + w; - if is_flipped { - idx += h * ls; - idx2 += h * ls; - } - for _ in 0..h { - if is_flipped { - idx -= ls; - idx2 -= ls; - } - let line = &dta[idx..idx2]; - ofile.write_all(line).unwrap(); - if !is_flipped { - idx += ls; - idx2 += ls; - } - } - } -} - -fn write_palppm(pfx: &str, strno: usize, num: u64, frm: NAFrameRef) { - let name = format!("{}/{}out{:02}_{:06}.ppm", OUTPUT_PREFIX, pfx, strno, num); - let mut ofile = File::create(name).unwrap(); - let buf = frm.get_buffer().get_vbuf().unwrap(); - let (w, h) = buf.get_dimensions(0); - let paloff = buf.get_offset(1); - let hdr = format!("P6\n{} {}\n255\n", w, h); - ofile.write_all(hdr.as_bytes()).unwrap(); - let dta = buf.get_data(); - let ls = buf.get_stride(0); - let offs: [usize; 3] = [ - buf.get_info().get_format().get_chromaton(0).unwrap().get_offset() as usize, - buf.get_info().get_format().get_chromaton(1).unwrap().get_offset() as usize, - buf.get_info().get_format().get_chromaton(2).unwrap().get_offset() as usize - ]; - let mut idx = 0; - let mut line: Vec = vec![0; w * 3]; - for _ in 0..h { - let src = &dta[idx..(idx+w)]; - for x in 0..w { - let pix = src[x] as usize; - line[x * 3 + 0] = dta[paloff + pix * 3 + offs[0]]; - line[x * 3 + 1] = dta[paloff + pix * 3 + offs[1]]; - line[x * 3 + 2] = dta[paloff + pix * 3 + offs[2]]; - } - ofile.write_all(line.as_slice()).unwrap(); - idx += ls; - } -} - -fn write_ppm(pfx: &str, strno: usize, num: u64, frm: NAFrameRef) { - let name = format!("{}/{}out{:02}_{:06}.ppm", OUTPUT_PREFIX, pfx, strno, num); - let mut ofile = File::create(name).unwrap(); - let info = frm.get_buffer().get_video_info().unwrap(); - let mut dpic = alloc_video_buffer(NAVideoInfo::new(info.get_width(), info.get_height(), false, RGB24_FORMAT), 0).unwrap(); - let ifmt = ScaleInfo { width: info.get_width(), height: info.get_height(), fmt: info.get_format() }; - let ofmt = ScaleInfo { width: info.get_width(), height: info.get_height(), fmt: RGB24_FORMAT }; - let mut scaler = NAScale::new(ifmt, ofmt).unwrap(); - scaler.convert(&frm.get_buffer(), &mut dpic).unwrap(); - let buf = dpic.get_vbuf().unwrap(); - let (w, h) = buf.get_dimensions(0); - let hdr = format!("P6\n{} {}\n255\n", w, h); - ofile.write_all(hdr.as_bytes()).unwrap(); - let dta = buf.get_data(); - let stride = buf.get_stride(0); - for src in dta.chunks(stride) { - ofile.write_all(&src[0..w*3]).unwrap(); - } -} +const OUTPUT_PREFIX: &str = "assets/test_out/"; /*fn open_wav_out(pfx: &str, strno: usize) -> WavWriter { let name = format!("assets/{}out{:02}.wav", pfx, strno); @@ -252,18 +74,9 @@ pub fn test_file_decoding(demuxer: &str, name: &str, limit: Option, } let frm = dec.decode(dsupp, &pkt).unwrap(); if pkt.get_stream().get_info().is_video() && video_pfx.is_some() && frm.get_frame_type() != FrameType::Skip { - let pfx = video_pfx.unwrap(); let pts = if let Some(fpts) = frm.get_pts() { fpts } else { pkt.get_pts().unwrap() }; - let vinfo = frm.get_buffer().get_video_info().unwrap(); - if vinfo.get_format().is_paletted() { - write_palppm(pfx, streamno, pts, frm); - } else if vinfo.get_format().get_model().is_yuv() { - write_pgmyuv(pfx, streamno, pts, frm); - } else if vinfo.get_format().get_model().is_rgb() { - write_ppm(pfx, streamno, pts, frm); - } else { -panic!(" unknown format"); - } + let pfx = OUTPUT_PREFIX.to_owned() + video_pfx.unwrap_or("") + "out"; + write_pnm(pfx.as_str(), streamno, pts, frm).unwrap(); } } } @@ -534,3 +347,64 @@ println!("full hash {}", md5); panic!("generated hashes"); } } + +/// Tests decoding of provided file by outputting video frames as PNM (PPM for RGB video, PGM for YUV). +/// +/// This function expects the following arguments: +/// * `demuxer` - container format name (used to find proper demuxer for it) +/// * `name` - input file name +/// * `video_pfx` - prefix for video frames written as pictures (output picture names should look like `/assets/test_out/PFXout00_000000.ppm` +/// * `limit` - optional PTS value after which decoding is stopped +/// * `dmx_reg` and `dec_reg` - registered demuxers and decoders that should contain demuxer and decoder(s) needed to decode the provided file. +/// +/// Since the function is intended for tests, it will panic instead of returning an error. +pub fn test_decode_images(demuxer: &str, name: &str, video_pfx: &str, limit: Option, + dmx_reg: &RegisteredDemuxers, dec_reg: &RegisteredDecoders) { + let dmx_f = dmx_reg.find_demuxer(demuxer).unwrap(); + let mut file = File::open(name).unwrap(); + let mut fr = FileReader::new_read(&mut file); + let mut br = ByteReader::new(&mut fr); + let mut dmx = create_demuxer(dmx_f, &mut br).unwrap(); + + let mut decs: Vec, Box)>> = Vec::new(); + for i in 0..dmx.get_num_streams() { + let s = dmx.get_stream(i).unwrap(); + let info = s.get_info(); + let decfunc = dec_reg.find_decoder(info.get_name()); + if let Some(df) = decfunc { + if info.is_video() { + let mut dec = (df)(); + let mut dsupp = Box::new(NADecoderSupport::new()); + dec.init(&mut dsupp, info).unwrap(); + decs.push(Some((dsupp, dec))); + break; + } else { + decs.push(None); + } + } else { + decs.push(None); + } + } + + loop { + let pktres = dmx.get_frame(); + if let Err(e) = pktres { + if e == DemuxerError::EOF { break; } + panic!("error"); + } + let pkt = pktres.unwrap(); + let streamno = pkt.get_stream().get_id() as usize; + if streamno >= decs.len() { continue; } + if let Some((ref mut dsupp, ref mut dec)) = decs[streamno] { + if let (Some(lim), Some(ppts)) = (limit, pkt.get_pts()) { + if ppts > lim { break; } + } + let frm = dec.decode(dsupp, &pkt).unwrap(); + if frm.get_frame_type() != FrameType::Skip { + let pts = if let Some(fpts) = frm.get_pts() { fpts } else { pkt.get_pts().unwrap() }; + let pfx = OUTPUT_PREFIX.to_owned() + video_pfx + "out"; + write_pnm(pfx.as_str(), streamno, pts, frm).unwrap(); + } + } + } +}