zoomvtools 1.1.0

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

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

use crate::util::Pixel;
use semisafe::slice::get as semisafe_get;
use semisafe::slice::get_mut as semisafe_get_mut;

pub(super) fn refine_horizontal_bilinear<T: Pixel>(
    dest: &mut [T],
    src: &[T],
    pitch: NonZeroUsize,
    width: NonZeroUsize,
    height: NonZeroUsize,
    _bits_per_sample: NonZeroU8,
) {
    let mut offset = 0;
    for _j in 0..height.get() {
        let src_row = semisafe_get(semisafe_get(src, offset..), ..width.get());
        let dest_row = semisafe_get_mut(semisafe_get_mut(dest, offset..), ..width.get());

        for i in 0..width.get() - 1 {
            let a: u32 = semisafe_get(src_row, i).as_();
            let b: u32 = semisafe_get(src_row, i + 1).as_();
            *semisafe_get_mut(dest_row, i) = T::from_u32_or_max_value((a + b + 1) / 2);
        }
        // last column
        *semisafe_get_mut(dest_row, width.get() - 1) = *semisafe_get(src_row, width.get() - 1);

        offset += pitch.get();
    }
}

pub(super) fn refine_vertical_bilinear<T: Pixel>(
    dest: &mut [T],
    src: &[T],
    pitch: NonZeroUsize,
    width: NonZeroUsize,
    height: NonZeroUsize,
    _bits_per_sample: NonZeroU8,
) {
    let mut offset = 0;
    for _j in 0..height.get() - 1 {
        for i in 0..width.get() {
            let a: u32 = semisafe_get(src, offset + i).as_();
            let b: u32 = semisafe_get(src, offset + i + pitch.get()).as_();
            *semisafe_get_mut(dest, offset + i) = T::from_u32_or_max_value((a + b + 1) / 2);
        }
        offset += pitch.get();
    }

    // last row
    let dest_row = semisafe_get_mut(semisafe_get_mut(dest, offset..), ..width.get());
    let src_row = semisafe_get(semisafe_get(src, offset..), ..width.get());
    dest_row.copy_from_slice(src_row);
}

pub(super) fn refine_diagonal_bilinear<T: Pixel>(
    dest: &mut [T],
    src: &[T],
    pitch: NonZeroUsize,
    width: NonZeroUsize,
    height: NonZeroUsize,
    _bits_per_sample: NonZeroU8,
) {
    let mut offset = 0;

    for _j in 0..height.get() {
        for i in 0..width.get() {
            let a: u32 = semisafe_get(src, offset + i).as_();
            let b: u32 = semisafe_get(src, offset + i + 1).as_();
            let c: u32 = semisafe_get(src, offset + i + pitch.get()).as_();
            let d: u32 = semisafe_get(src, offset + i + pitch.get() + 1).as_();

            *semisafe_get_mut(dest, offset + i) = T::from_u32_or_max_value((a + b + c + d + 2) / 4);
        }
        // last column
        let a: u32 = semisafe_get(src, offset + width.get() - 1).as_();
        let b: u32 = semisafe_get(src, offset + width.get() - 1 + pitch.get()).as_();
        *semisafe_get_mut(dest, offset + width.get() - 1) =
            T::from_u32_or_max_value((a + b + 1) / 2);

        offset += pitch.get();
    }

    // last row
    for i in 0..width.get() - 1 {
        let a: u32 = semisafe_get(src, offset + i).as_();
        let b: u32 = semisafe_get(src, offset + i + 1).as_();
        *semisafe_get_mut(dest, offset + i) = T::from_u32_or_max_value((a + b + 1) / 2);
    }
    // last pixel
    *semisafe_get_mut(dest, offset + width.get() - 1) =
        *semisafe_get(src, offset + width.get() - 1);
}