MattJackson
c91d6e371f
Chapters: - MPLS PlayList marks parsed (mark_type 1 = chapter) - Chapter struct on DiscTitle (time_secs, name) - MKV Chapters element with EditionEntry/ChapterAtom per mark - 3 MPLS mark tests + 2 MKV chapter tests DVD subtitle palette: - IFO palette extraction (PGC offset 0xA4, 16 × YCbCr colors) - YCbCr→RGB conversion for VobSub .idx format - DvdSubParser codec_private returns formatted palette - codec_data field on SubtitleStream flows through pipeline - 5 palette tests (YCbCr conversion, formatting, overflow) MKV track flags: - FlagDefault: primary video/audio = 1, secondary = 0 - FlagForced: forced subtitles = 1 - Language: set from stream language code - Already implemented, verified with 4 new tests Progress total_bytes: - IOStream trait: total_bytes() -> Option<u64> - DiscStream, IsoStream: from disc_title.size_bytes - M2tsStream, MkvStream: from file metadata on open - NetworkStream, StdioStream, NullStream: None 316 tests total, all passing.
115 lines
2.7 KiB
Rust
115 lines
2.7 KiB
Rust
//! NullStream — discards all data. Write-only. For benchmarking.
|
|
|
|
use super::IOStream;
|
|
use crate::disc::DiscTitle;
|
|
use std::io::{self, Read, Write};
|
|
|
|
/// Null stream — accepts writes, discards data. For benchmarking rip speed.
|
|
pub struct NullStream {
|
|
disc_title: DiscTitle,
|
|
bytes_written: u64,
|
|
}
|
|
|
|
impl Default for NullStream {
|
|
fn default() -> Self {
|
|
Self::new()
|
|
}
|
|
}
|
|
|
|
impl NullStream {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
disc_title: DiscTitle::empty(),
|
|
bytes_written: 0,
|
|
}
|
|
}
|
|
|
|
pub fn meta(mut self, dt: &DiscTitle) -> Self {
|
|
self.disc_title = dt.clone();
|
|
self
|
|
}
|
|
|
|
pub fn bytes_written(&self) -> u64 {
|
|
self.bytes_written
|
|
}
|
|
}
|
|
|
|
impl IOStream for NullStream {
|
|
fn info(&self) -> &DiscTitle {
|
|
&self.disc_title
|
|
}
|
|
fn finish(&mut self) -> io::Result<()> {
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl Write for NullStream {
|
|
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
|
|
self.bytes_written += buf.len() as u64;
|
|
Ok(buf.len())
|
|
}
|
|
fn flush(&mut self) -> io::Result<()> {
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl Read for NullStream {
|
|
fn read(&mut self, _buf: &mut [u8]) -> io::Result<usize> {
|
|
Err(io::Error::new(
|
|
io::ErrorKind::Unsupported,
|
|
"null stream is write-only",
|
|
))
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use std::io::Write;
|
|
|
|
#[test]
|
|
fn null_counts_bytes() {
|
|
let mut ns = NullStream::new();
|
|
assert_eq!(ns.bytes_written(), 0);
|
|
ns.write_all(&[0u8; 100]).unwrap();
|
|
assert_eq!(ns.bytes_written(), 100);
|
|
ns.write_all(&[1u8; 50]).unwrap();
|
|
assert_eq!(ns.bytes_written(), 150);
|
|
// Single write returns correct count
|
|
let n = ns.write(&[0u8; 200]).unwrap();
|
|
assert_eq!(n, 200);
|
|
assert_eq!(ns.bytes_written(), 350);
|
|
}
|
|
|
|
#[test]
|
|
fn null_read_errors() {
|
|
let mut ns = NullStream::new();
|
|
let mut buf = [0u8; 10];
|
|
let err = ns.read(&mut buf).unwrap_err();
|
|
assert_eq!(err.kind(), io::ErrorKind::Unsupported);
|
|
}
|
|
|
|
#[test]
|
|
fn null_finish_ok() {
|
|
let mut ns = NullStream::new();
|
|
ns.write_all(&[0u8; 1000]).unwrap();
|
|
ns.finish().unwrap();
|
|
}
|
|
|
|
#[test]
|
|
fn null_implements_iostream() {
|
|
let ns = NullStream::new();
|
|
let mut boxed: Box<dyn IOStream> = Box::new(ns);
|
|
boxed.write_all(&[0u8; 50]).unwrap();
|
|
let info = boxed.info();
|
|
assert_eq!(info.streams.len(), 0);
|
|
boxed.finish().unwrap();
|
|
}
|
|
|
|
#[test]
|
|
fn null_total_bytes_returns_none() {
|
|
let ns = NullStream::new();
|
|
assert_eq!(ns.total_bytes(), None);
|
|
}
|
|
}
|