zoomvtools 2.0.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

#[cfg(all(target_arch = "x86_64", feature = "simd"))]
mod avx2;
mod rust;

#[cfg(test)]
mod tests;

use std::num::NonZeroUsize;

use crate::util::Pixel;

pub type LimitChangesFn = unsafe fn(
    dest: *mut u8,
    dest_stride_bytes: NonZeroUsize,
    src: *const u8,
    src_stride_bytes: NonZeroUsize,
    width: NonZeroUsize,
    height: NonZeroUsize,
    limit: u16,
);

#[must_use]
#[inline]
pub fn select_limit_changes<T: Pixel>() -> LimitChangesFn {
    #[cfg(all(target_arch = "x86_64", feature = "simd", feature = "experimental"))]
    if cpudetect::x86_64::is_x86_64_v3_compatible() {
        return match std::mem::size_of::<T>() {
            // TODO: 10% slower than scalar on test machine
            1 => avx2::limit_changes_u8,
            // TODO: 10% slower than scalar on test machine
            2 => avx2::limit_changes_u16,
            _ => unreachable!(),
        };
    }

    rust::limit_changes::<T>
}