Add a test for getting the timescale of an avis

Author: padenot

mp4parse_capi/tests/test_avis_loop_count.rs

@@ -11,54 +11,71 @@ extern "C" fn buf_read(buf: *mut u8, size: usize, userdata: *mut std::os::raw::c
     }
 }
 
-fn check_loop_count(path: &str, expected_loop_count: i64) {
+unsafe fn parse_file_and_get_info(path: &str) -> (*mut Mp4parseAvifParser, Mp4parseAvifInfo) {
     let mut file = std::fs::File::open(path).expect("Unknown file");
     let io = Mp4parseIo {
         read: Some(buf_read),
         userdata: &mut file as *mut _ as *mut std::os::raw::c_void,
     };
 
-    unsafe {
-        let mut parser = std::ptr::null_mut();
-        let mut rv = mp4parse_avif_new(&io, ParseStrictness::Normal, &mut parser);
-        assert_eq!(rv, Mp4parseStatus::Ok);
-        assert!(!parser.is_null());
-
-        let mut info = Mp4parseAvifInfo {
-            premultiplied_alpha: Default::default(),
-            major_brand: Default::default(),
-            unsupported_features_bitfield: Default::default(),
-            spatial_extents: std::ptr::null(),
-            nclx_colour_information: std::ptr::null(),
-            icc_colour_information: Default::default(),
-            image_rotation: mp4parse::ImageRotation::D0,
-            image_mirror: std::ptr::null(),
-            pixel_aspect_ratio: std::ptr::null(),
-            has_primary_item: Default::default(),
-            primary_item_bit_depth: Default::default(),
-            has_alpha_item: Default::default(),
-            alpha_item_bit_depth: Default::default(),
-            has_sequence: Default::default(),
-            loop_mode: Default::default(),
-            loop_count: Default::default(),
-            color_track_id: Default::default(),
-            color_track_bit_depth: Default::default(),
-            alpha_track_id: Default::default(),
-            alpha_track_bit_depth: Default::default(),
-        };
-        rv = mp4parse_avif_get_info(parser, &mut info);
-        assert_eq!(rv, Mp4parseStatus::Ok);
-
-        match info.loop_mode {
-            Mp4parseAvifLoopMode::NoEdits => assert_eq!(expected_loop_count, -1),
-            Mp4parseAvifLoopMode::LoopByCount => {
-                assert_eq!(info.loop_count.to_i64(), Some(expected_loop_count))
-            }
-            Mp4parseAvifLoopMode::LoopInfinitely => assert_eq!(expected_loop_count, std::i64::MIN),
-        }
+    let mut parser = std::ptr::null_mut();
+    let mut rv = mp4parse_avif_new(&io, ParseStrictness::Normal, &mut parser);
+    assert_eq!(rv, Mp4parseStatus::Ok);
+    assert!(!parser.is_null());
+
+    let mut info = Mp4parseAvifInfo {
+        premultiplied_alpha: Default::default(),
+        major_brand: Default::default(),
+        unsupported_features_bitfield: Default::default(),
+        spatial_extents: std::ptr::null(),
+        nclx_colour_information: std::ptr::null(),
+        icc_colour_information: Default::default(),
+        image_rotation: mp4parse::ImageRotation::D0,
+        image_mirror: std::ptr::null(),
+        pixel_aspect_ratio: std::ptr::null(),
+        has_primary_item: Default::default(),
+        primary_item_bit_depth: Default::default(),
+        has_alpha_item: Default::default(),
+        alpha_item_bit_depth: Default::default(),
+        has_sequence: Default::default(),
+        loop_mode: Default::default(),
+        loop_count: Default::default(),
+        color_track_id: Default::default(),
+        color_track_bit_depth: Default::default(),
+        alpha_track_id: Default::default(),
+        alpha_track_bit_depth: Default::default(),
+    };
+    rv = mp4parse_avif_get_info(parser, &mut info);
+    assert_eq!(rv, Mp4parseStatus::Ok);
+    (parser, info)
+}
 
-        mp4parse_avif_free(parser);
+fn check_loop_count(path: &str, expected_loop_count: i64) {
+    let (parser, info) = unsafe { parse_file_and_get_info(path) };
+    match info.loop_mode {
+        Mp4parseAvifLoopMode::NoEdits => assert_eq!(expected_loop_count, -1),
+        Mp4parseAvifLoopMode::LoopByCount => {
+            assert_eq!(info.loop_count.to_i64(), Some(expected_loop_count))
+        }
+        Mp4parseAvifLoopMode::LoopInfinitely => assert_eq!(expected_loop_count, std::i64::MIN),
     }
+
+    unsafe { mp4parse_avif_free(parser) };
+}
+
+fn check_timescale(path: &str, expected_timescale: u64) {
+    let (parser, info) = unsafe { parse_file_and_get_info(path) };
+
+    let mut indices: Mp4parseByteData = Mp4parseByteData::default();
+    let mut timescale: u64 = 0;
+    let rv = unsafe {
+        mp4parse_avif_get_indice_table(parser, info.color_track_id, &mut indices, &mut timescale)
+    };
+
+    assert_eq!(rv, Mp4parseStatus::Ok);
+    assert_eq!(timescale, expected_timescale);
+
+    unsafe { mp4parse_avif_free(parser) };
 }
 
 #[test]
@@ -85,3 +102,12 @@ fn loop_forever() {
 fn no_edts() {
     check_loop_count("tests/no_edts.avif", -1);
 }
+
+#[test]
+fn check_timescales() {
+    check_timescale("tests/loop_1.avif", 2);
+    check_timescale("tests/loop_2.avif", 2);
+    check_timescale("tests/loop_ceiled_4.avif", 2);
+    check_timescale("tests/loop_forever.avif", 2);
+    check_timescale("tests/no_edts.avif", 16384);
+}