mediaway-encoder 0.1.4

Hardware-accelerated video/audio encoding (OS-native backends)
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

#![cfg(test)]

use super::*;

#[test]
fn gop_size_one_is_always_idr() {
    let mut state = H264GopState::new(1);
    for _ in 0..5 {
        let decision = state.decide();
        assert!(decision.is_idr);
        assert_eq!(decision.frame_num, 0);
        assert_eq!(decision.poc, 0);
    }
}

#[test]
fn gop_size_three_produces_idr_p_p_cadence() {
    let mut state = H264GopState::new(3);
    let expect: [(bool, u32); 7] = [
        (true, 0),
        (false, 1),
        (false, 2),
        (true, 0),
        (false, 1),
        (false, 2),
        (true, 0),
    ];
    for (is_idr, frame_num) in expect {
        let decision = state.decide();
        assert_eq!(decision.is_idr, is_idr);
        assert_eq!(decision.frame_num, frame_num);
        assert_eq!(decision.poc, frame_num * 2);
        assert_eq!(decision.intra_refresh_frame_index, None);
    }
}

#[test]
fn intra_refresh_only_the_first_frame_ever_is_idr() {
    let mut state = H264GopState::new_intra_refresh(3);
    for i in 0..10u32 {
        let decision = state.decide();
        assert_eq!(decision.is_idr, i == 0, "frame {i}");
    }
}

#[test]
fn intra_refresh_wave_index_cycles_with_period_and_skips_the_idr_frame() {
    let mut state = H264GopState::new_intra_refresh(3);
    let expect: [Option<u32>; 7] = [
        None,    // IDR
        Some(0), // wave 1
        Some(1),
        Some(2),
        Some(0), // wave 2
        Some(1),
        Some(2),
    ];
    for wave_index in expect {
        let decision = state.decide();
        assert_eq!(decision.intra_refresh_frame_index, wave_index);
    }
}