av-scenechange 0.23.0

Estimates frames in a video where a scenecut would be ideal
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142

use v_frame::pixel::Pixel;

use crate::data::{block::BlockSize, plane::PlaneRegion};

macro_rules! declare_asm_dist_fn {
        ($(($name: ident, $T: ident)),+) => (
            $(
                unsafe extern "C" { fn $name (
                    src: *const $T, src_stride: isize, dst: *const $T, dst_stride: isize
                ) -> u32; }
            )+
        )
    }

declare_asm_dist_fn![
    // SATD
    (avsc_satd_4x4_avx2, u8),
    (avsc_satd_4x8_avx2, u8),
    (avsc_satd_4x16_avx2, u8),
    (avsc_satd_8x4_avx2, u8),
    (avsc_satd_8x8_avx2, u8),
    (avsc_satd_8x16_avx2, u8),
    (avsc_satd_8x32_avx2, u8),
    (avsc_satd_16x4_avx2, u8),
    (avsc_satd_16x8_avx2, u8),
    (avsc_satd_16x16_avx2, u8),
    (avsc_satd_16x32_avx2, u8),
    (avsc_satd_16x64_avx2, u8),
    (avsc_satd_32x8_avx2, u8),
    (avsc_satd_32x16_avx2, u8),
    (avsc_satd_32x32_avx2, u8),
    (avsc_satd_32x64_avx2, u8),
    (avsc_satd_64x16_avx2, u8),
    (avsc_satd_64x32_avx2, u8),
    (avsc_satd_64x64_avx2, u8),
    (avsc_satd_64x128_avx2, u8),
    (avsc_satd_128x64_avx2, u8),
    (avsc_satd_128x128_avx2, u8),
    // SATD HBD
    (avsc_satd_4x4_hbd_avx2, u16),
    (avsc_satd_4x8_hbd_avx2, u16),
    (avsc_satd_4x16_hbd_avx2, u16),
    (avsc_satd_8x4_hbd_avx2, u16),
    (avsc_satd_8x8_hbd_avx2, u16),
    (avsc_satd_8x16_hbd_avx2, u16),
    (avsc_satd_8x32_hbd_avx2, u16),
    (avsc_satd_16x4_hbd_avx2, u16),
    (avsc_satd_16x8_hbd_avx2, u16),
    (avsc_satd_16x16_hbd_avx2, u16),
    (avsc_satd_16x32_hbd_avx2, u16),
    (avsc_satd_16x64_hbd_avx2, u16),
    (avsc_satd_32x8_hbd_avx2, u16),
    (avsc_satd_32x16_hbd_avx2, u16),
    (avsc_satd_32x32_hbd_avx2, u16),
    (avsc_satd_32x64_hbd_avx2, u16),
    (avsc_satd_64x16_hbd_avx2, u16),
    (avsc_satd_64x32_hbd_avx2, u16),
    (avsc_satd_64x64_hbd_avx2, u16),
    (avsc_satd_64x128_hbd_avx2, u16),
    (avsc_satd_128x64_hbd_avx2, u16),
    (avsc_satd_128x128_hbd_avx2, u16)
];

#[target_feature(enable = "avx2")]
pub(super) fn get_satd_internal<T: Pixel>(
    src: &PlaneRegion<'_, T>,
    dst: &PlaneRegion<'_, T>,
    w: usize,
    h: usize,
    bit_depth: usize,
) -> u32 {
    let bsize_opt = BlockSize::from_width_and_height_opt(w, h);

    match (bsize_opt, size_of::<T>()) {
        (Err(_), _) => super::rust::get_satd_internal(src, dst, w, h, bit_depth),
        // SAFETY: call to SIMD function
        (Ok(bsize), 1) => unsafe {
            (match bsize {
                BlockSize::BLOCK_4X4 => avsc_satd_4x4_avx2,
                BlockSize::BLOCK_4X8 => avsc_satd_4x8_avx2,
                BlockSize::BLOCK_4X16 => avsc_satd_4x16_avx2,
                BlockSize::BLOCK_8X4 => avsc_satd_8x4_avx2,
                BlockSize::BLOCK_16X4 => avsc_satd_16x4_avx2,
                BlockSize::BLOCK_8X8 => avsc_satd_8x8_avx2,
                BlockSize::BLOCK_8X16 => avsc_satd_8x16_avx2,
                BlockSize::BLOCK_8X32 => avsc_satd_8x32_avx2,
                BlockSize::BLOCK_16X8 => avsc_satd_16x8_avx2,
                BlockSize::BLOCK_16X16 => avsc_satd_16x16_avx2,
                BlockSize::BLOCK_16X32 => avsc_satd_16x32_avx2,
                BlockSize::BLOCK_16X64 => avsc_satd_16x64_avx2,
                BlockSize::BLOCK_32X8 => avsc_satd_32x8_avx2,
                BlockSize::BLOCK_32X16 => avsc_satd_32x16_avx2,
                BlockSize::BLOCK_32X32 => avsc_satd_32x32_avx2,
                BlockSize::BLOCK_32X64 => avsc_satd_32x64_avx2,
                BlockSize::BLOCK_64X16 => avsc_satd_64x16_avx2,
                BlockSize::BLOCK_64X32 => avsc_satd_64x32_avx2,
                BlockSize::BLOCK_64X64 => avsc_satd_64x64_avx2,
                BlockSize::BLOCK_64X128 => avsc_satd_64x128_avx2,
                BlockSize::BLOCK_128X64 => avsc_satd_128x64_avx2,
                BlockSize::BLOCK_128X128 => avsc_satd_128x128_avx2,
            })(
                src.data_ptr() as *const _,
                (size_of::<T>() * src.plane_cfg.stride.get()) as isize,
                dst.data_ptr() as *const _,
                (size_of::<T>() * dst.plane_cfg.stride.get()) as isize,
            )
        },
        // SAFETY: call to SIMD function
        (Ok(bsize), 2) => unsafe {
            (match bsize {
                BlockSize::BLOCK_4X4 => avsc_satd_4x4_hbd_avx2,
                BlockSize::BLOCK_4X8 => avsc_satd_4x8_hbd_avx2,
                BlockSize::BLOCK_4X16 => avsc_satd_4x16_hbd_avx2,
                BlockSize::BLOCK_8X4 => avsc_satd_8x4_hbd_avx2,
                BlockSize::BLOCK_16X4 => avsc_satd_16x4_hbd_avx2,
                BlockSize::BLOCK_8X8 => avsc_satd_8x8_hbd_avx2,
                BlockSize::BLOCK_8X16 => avsc_satd_8x16_hbd_avx2,
                BlockSize::BLOCK_8X32 => avsc_satd_8x32_hbd_avx2,
                BlockSize::BLOCK_16X8 => avsc_satd_16x8_hbd_avx2,
                BlockSize::BLOCK_16X16 => avsc_satd_16x16_hbd_avx2,
                BlockSize::BLOCK_16X32 => avsc_satd_16x32_hbd_avx2,
                BlockSize::BLOCK_16X64 => avsc_satd_16x64_hbd_avx2,
                BlockSize::BLOCK_32X8 => avsc_satd_32x8_hbd_avx2,
                BlockSize::BLOCK_32X16 => avsc_satd_32x16_hbd_avx2,
                BlockSize::BLOCK_32X32 => avsc_satd_32x32_hbd_avx2,
                BlockSize::BLOCK_32X64 => avsc_satd_32x64_hbd_avx2,
                BlockSize::BLOCK_64X16 => avsc_satd_64x16_hbd_avx2,
                BlockSize::BLOCK_64X32 => avsc_satd_64x32_hbd_avx2,
                BlockSize::BLOCK_64X64 => avsc_satd_64x64_hbd_avx2,
                BlockSize::BLOCK_64X128 => avsc_satd_64x128_hbd_avx2,
                BlockSize::BLOCK_128X64 => avsc_satd_128x64_hbd_avx2,
                BlockSize::BLOCK_128X128 => avsc_satd_128x128_hbd_avx2,
            })(
                src.data_ptr() as *const _,
                (size_of::<T>() * src.plane_cfg.stride.get()) as isize,
                dst.data_ptr() as *const _,
                (size_of::<T>() * dst.plane_cfg.stride.get()) as isize,
            )
        },
        _ => unreachable!(),
    }
}