+//! Routines for testing decoders.
use std::fs::File;
use std::io::prelude::*;
use crate::frame::*;
WavWriter::new(&mut wr)
}*/
+/// Tests decoding of provided file and optionally outputs 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
+/// * `limit` - optional PTS value after which decoding is stopped
+/// * `decode_video`/`decode_audio` - flags for enabling video/audio decoding
+/// * `video_pfx` - prefix for video frames written as pictures (if enabled then output picture names should look like `<crate_name>/assets/test_out/PFXout00_000000.ppm`
+/// * `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_file_decoding(demuxer: &str, name: &str, limit: Option<u64>,
decode_video: bool, decode_audio: bool,
video_pfx: Option<&str>,
}
}
+/// Tests audio decoder with the content in the provided file and optionally outputs decoded audio.
+///
+/// The syntax is very similar to [`test_file_decoding`] except that it is intended for testing audio codecs.
+///
+/// Since the function is intended for tests, it will panic instead of returning an error.
+///
+/// [`test_file_decoding`]: ./fn.test_file_decoding.html
pub fn test_decode_audio(demuxer: &str, name: &str, limit: Option<u64>, audio_pfx: Option<&str>,
dmx_reg: &RegisteredDemuxers, dec_reg: &RegisteredDecoders) {
let dmx_f = dmx_reg.find_demuxer(demuxer).unwrap();
};
}
+/// Tests decoder for requested codec in provided file.
+///
+/// This functions tries to decode a stream corresponding to `dec_name` codec in input file and validate the results against expected ones.
+///
+/// Since the function is intended for tests, it will panic instead of returning an error.
+///
+/// # Examples
+///
+/// Test RealVideo 4 decoder in test stream:
+/// ```no_run
+/// use nihav_core::test::ExpectedTestResult;
+/// use nihav_core::test::dec_video::test_decoding;
+/// use nihav_core::codecs::RegisteredDecoders;
+/// use nihav_core::demuxers::RegisteredDemuxers;
+///
+/// let mut dmx_reg = RegisteredDemuxers::new();
+/// let mut dec_reg = RegisteredDecoders::new();
+/// // ... register RealMedia demuxers and RealVideo decoders ...
+/// test_decoding("realmedia", "rv40", "assets/test_file.rmvb", None, &dmx_reg, &dec_reg, ExpectedTestResult::MD5([0x00010203, 0x04050607, 0x08090a0b, 0x0c0d0e0f]));
+/// ```
pub fn test_decoding(demuxer: &str, dec_name: &str, filename: &str, limit: Option<u64>,
dmx_reg: &RegisteredDemuxers, dec_reg: &RegisteredDecoders,
test: ExpectedTestResult) {
+//! Decoder testing functionality.
+//!
+//! This module provides functions that may be used in internal test to check that decoders produce output and that they produce expected output.
pub mod dec_video;
pub mod wavwriter;
mod md5; // for internal checksums only
+/// Decoder testing modes.
+///
+/// NihAV has its own MD5 hasher that calculates hash for input frame in a deterministic way (i.e. no endianness issues) and it outputs hash as `[u32; 4]`. During testing the resulting hash will be printed out so when the test fails you can find what was the calculated hash (and use it as a reference if you are implementing a new test).
pub enum ExpectedTestResult {
+ /// Decoding runs without errors.
Decodes,
+ /// Full decoded output is expected to be equal to this MD5 hash.
MD5([u32; 4]),
+ /// Each decoded frame hash should be equal to the corresponding MD5 hash.
MD5Frames(Vec<[u32; 4]>),
+ /// Test function should report decoded frame hashes to be used as the reference later.
GenerateMD5Frames,
}
+//! Audio output in WAV format.
use crate::io::byteio::*;
use crate::frame::*;
use std::io::SeekFrom;
+/// WAVE output writer.
pub struct WavWriter<'a> {
io: &'a mut ByteWriter<'a>,
data_pos: u64,
}
impl<'a> WavWriter<'a> {
+ /// Constructs a new `WavWriter` instance.
pub fn new(io: &'a mut ByteWriter<'a>) -> Self {
WavWriter { io, data_pos: 0 }
}
+ /// Writes audio format information to the file header.
+ ///
+ /// This function should be called exactly once before writing actual audio data.
pub fn write_header(&mut self, ainfo: NAAudioInfo) -> ByteIOResult<()> {
let bits = ainfo.get_format().get_bits() as usize;
self.data_pos = self.io.tell();
Ok(())
}
+ /// Writes audio data.
pub fn write_frame(&mut self, abuf: NABufferType) -> ByteIOResult<()> {
match abuf {
NABufferType::AudioU8(ref buf) => {