mirror of
https://github.com/alfg/mp4-rust.git
synced 2024-10-01 07:41:53 +00:00
3104a2d95b
* Add ReadBox trait * Add boxtype macro * Remove offset in BoxHeader * Fix parsing error when box has largesize * Remove duplicated codes reading version and flags * Add avc1 box * Add mp4a box * Add mp4a box * Add DecoderSpecificDescriptor in esds box * Add necessary sub-boxes to stbl box * Improve ReadBox::read_box() * Add smhd box * Refactor BoxHeader * Refactor BMFF * Refactor * Add some functions to get offset and size of sample * Add Mp4Reader::read_sample() that read media samples * Move Mp4Reader to reader.rs * Add mandatory check when reading boxes Add some methods to Mp4Reader, TrackReader Format codes * Update mp4info * Refactor common types * Add FixedPointX types * Add media configuration, profile, ... * Add initial Mp4Writer * Run cargo fmt * Add Mp4Writer and examples/mp4copy * Add test codes for Avc1Box and Mp4aBox * Remove prefix "get_" from method names * Rename atoms to mp4box * Fix some bugs Co-authored-by: Byungwan Jun <unipro.kr@gmail.com>
120 lines
3.5 KiB
Rust
120 lines
3.5 KiB
Rust
use mp4::{AudioObjectType, AvcProfile, ChannelConfig, MediaType, SampleFreqIndex, TrackType};
|
|
use std::fs::File;
|
|
use std::io::BufReader;
|
|
|
|
#[test]
|
|
fn test_read_mp4() {
|
|
let filename = "tests/samples/minimal.mp4";
|
|
let f = File::open(filename).unwrap();
|
|
let size = f.metadata().unwrap().len();
|
|
let reader = BufReader::new(f);
|
|
|
|
let mut mp4 = mp4::Mp4Reader::read_header(reader, size).unwrap();
|
|
|
|
assert_eq!(2591, mp4.size());
|
|
|
|
// ftyp.
|
|
assert_eq!(4, mp4.compatible_brands().len());
|
|
|
|
// Check compatible_brands.
|
|
let brands = vec![
|
|
String::from("isom"),
|
|
String::from("iso2"),
|
|
String::from("avc1"),
|
|
String::from("mp41"),
|
|
];
|
|
|
|
for b in brands {
|
|
let t = mp4.compatible_brands().iter().any(|x| x.to_string() == b);
|
|
assert_eq!(t, true);
|
|
}
|
|
|
|
assert_eq!(mp4.duration(), 62);
|
|
assert_eq!(mp4.timescale(), 1000);
|
|
assert_eq!(mp4.tracks().len(), 2);
|
|
|
|
let sample_count = mp4.sample_count(1).unwrap();
|
|
assert_eq!(sample_count, 1);
|
|
let sample_1_1 = mp4.read_sample(1, 1).unwrap().unwrap();
|
|
assert_eq!(sample_1_1.bytes.len(), 751);
|
|
assert_eq!(
|
|
sample_1_1,
|
|
mp4::Mp4Sample {
|
|
start_time: 0,
|
|
duration: 512,
|
|
rendering_offset: 0,
|
|
is_sync: true,
|
|
bytes: mp4::Bytes::from(vec![0x0u8; 751]),
|
|
}
|
|
);
|
|
let eos = mp4.read_sample(1, 2).unwrap();
|
|
assert!(eos.is_none());
|
|
|
|
let sample_count = mp4.sample_count(2).unwrap();
|
|
assert_eq!(sample_count, 3);
|
|
let sample_2_1 = mp4.read_sample(2, 1).unwrap().unwrap();
|
|
assert_eq!(sample_2_1.bytes.len(), 179);
|
|
assert_eq!(
|
|
sample_2_1,
|
|
mp4::Mp4Sample {
|
|
start_time: 0,
|
|
duration: 1024,
|
|
rendering_offset: 0,
|
|
is_sync: true,
|
|
bytes: mp4::Bytes::from(vec![0x0u8; 179]),
|
|
}
|
|
);
|
|
|
|
let sample_2_2 = mp4.read_sample(2, 2).unwrap().unwrap();
|
|
assert_eq!(
|
|
sample_2_2,
|
|
mp4::Mp4Sample {
|
|
start_time: 1024,
|
|
duration: 1024,
|
|
rendering_offset: 0,
|
|
is_sync: true,
|
|
bytes: mp4::Bytes::from(vec![0x0u8; 180]),
|
|
}
|
|
);
|
|
|
|
let sample_2_3 = mp4.read_sample(2, 3).unwrap().unwrap();
|
|
assert_eq!(
|
|
sample_2_3,
|
|
mp4::Mp4Sample {
|
|
start_time: 2048,
|
|
duration: 896,
|
|
rendering_offset: 0,
|
|
is_sync: true,
|
|
bytes: mp4::Bytes::from(vec![0x0u8; 160]),
|
|
}
|
|
);
|
|
|
|
let eos = mp4.read_sample(2, 4).unwrap();
|
|
assert!(eos.is_none());
|
|
|
|
// track #1
|
|
let track1 = mp4.tracks().get(0).unwrap();
|
|
assert_eq!(track1.track_id(), 1);
|
|
assert_eq!(track1.track_type().unwrap(), TrackType::Video);
|
|
assert_eq!(track1.media_type().unwrap(), MediaType::H264);
|
|
assert_eq!(track1.video_profile().unwrap(), AvcProfile::AvcHigh);
|
|
assert_eq!(track1.width(), 320);
|
|
assert_eq!(track1.height(), 240);
|
|
assert_eq!(track1.bitrate(), 0); // XXX
|
|
assert_eq!(track1.frame_rate().to_integer(), 25); // XXX
|
|
|
|
// track #2
|
|
let track2 = mp4.tracks().get(1).unwrap();
|
|
assert_eq!(track2.track_type().unwrap(), TrackType::Audio);
|
|
assert_eq!(track2.media_type().unwrap(), MediaType::AAC);
|
|
assert_eq!(
|
|
track2.audio_profile().unwrap(),
|
|
AudioObjectType::AacLowComplexity
|
|
);
|
|
assert_eq!(
|
|
track2.sample_freq_index().unwrap(),
|
|
SampleFreqIndex::Freq48000
|
|
);
|
|
assert_eq!(track2.channel_config().unwrap(), ChannelConfig::Mono);
|
|
assert_eq!(track2.bitrate(), 67695);
|
|
}
|