zoomvtools 1.1.1

Video motion vector analysis utilities in pure Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

#![allow(clippy::unwrap_used, reason = "allow in test files")]

use std::num::{NonZeroU8, NonZeroUsize};

use crate::{
    filters::analyse::SuperClipInfo,
    frame::{FramePlanes, FramePlanesMut, FrameView, PlaneRef},
    params::{MVPlaneSet, Subpel},
    video::{ColorFamily, Resolution, SampleType, VideoFormat, VideoInfo},
};

use super::Finest;

fn gray8_info(width: usize, height: usize) -> VideoInfo {
    VideoInfo {
        format: VideoFormat {
            color_family: ColorFamily::Gray,
            sample_type: SampleType::Integer,
            bits_per_sample: NonZeroU8::new(8).unwrap(),
            bytes_per_sample: NonZeroU8::new(1).unwrap(),
            sub_sampling_w: 0,
            sub_sampling_h: 0,
        },
        resolution: Resolution { width, height },
        num_frames: 1,
    }
}

#[test]
fn finest_render_frame_copies_full_pel_gray_plane_without_vapoursynth_types() {
    let finest = Finest::new(
        gray8_info(2, 2),
        SuperClipInfo {
            height: NonZeroUsize::new(2).unwrap(),
            hpad: 0,
            vpad: 0,
            pel: Subpel::Full,
            mode_yuv: MVPlaneSet::YPLANE,
            levels: 1,
        },
    )
    .unwrap();
    let source = [10u8, 20, 30, 40];
    let source_view = FrameView::new(
        FramePlanes::new([Some(PlaneRef::new(&source)), None, None]),
        (NonZeroUsize::new(2).unwrap(), None, None),
    );
    let mut output = [0u8; 4];
    let mut output_planes = FramePlanesMut::new([Some(&mut output), None, None]);

    finest
        .render_frame(
            &source_view,
            &mut output_planes,
            (NonZeroUsize::new(2).unwrap(), None, None),
        )
        .unwrap();

    assert_eq!(output, source);
}

#[test]
fn finest_render_frame_merges_half_pel_windows_without_vapoursynth_types() {
    let finest = Finest::new(
        gray8_info(1, 4),
        SuperClipInfo {
            height: NonZeroUsize::new(1).unwrap(),
            hpad: 0,
            vpad: 0,
            pel: Subpel::Half,
            mode_yuv: MVPlaneSet::YPLANE,
            levels: 1,
        },
    )
    .unwrap();
    let source = [10u8, 20, 30, 40];
    let source_view = FrameView::new(
        FramePlanes::new([Some(PlaneRef::new(&source)), None, None]),
        (NonZeroUsize::new(1).unwrap(), None, None),
    );
    let mut output = [0u8; 4];
    let mut output_planes = FramePlanesMut::new([Some(&mut output), None, None]);

    finest
        .render_frame(
            &source_view,
            &mut output_planes,
            (NonZeroUsize::new(2).unwrap(), None, None),
        )
        .unwrap();

    assert_eq!(output, [10, 20, 30, 40]);
}