unbundle 5.2.0

Unbundle media files - extract still frames, audio tracks, and subtitles from video files
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
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169

//! Async frame-stream integration tests (feature = "async").
//!
//! These tests require the `async` feature and test fixtures
//! generated by `tests/fixtures/generate_fixtures.sh`.

#![cfg(feature = "async")]

use std::path::Path;

use image::DynamicImage;
use tokio_stream::StreamExt;
use unbundle::{ExtractOptions, FrameRange, FrameStream, MediaFile};

const SAMPLE_VIDEO: &str = "tests/fixtures/sample_video.mp4";
const AUDIO_ONLY: &str = "tests/fixtures/sample_audio_only.mp4";

fn skip_unless(path: &str) -> bool {
    if !Path::new(path).exists() {
        eprintln!("Skipping: fixture {path} not found");
        return true;
    }
    false
}

#[tokio::test]
async fn stream_first_five_frames() {
    if skip_unless(SAMPLE_VIDEO) {
        return;
    }

    let mut unbundler = MediaFile::open(SAMPLE_VIDEO).unwrap();
    let mut stream: FrameStream = unbundler
        .video()
        .frame_stream(
            FrameRange::Specific(vec![0, 1, 2, 3, 4]),
            ExtractOptions::new(),
        )
        .unwrap();

    let mut count = 0u64;
    while let Some(result) = stream.next().await {
        let (frame_number, image): (u64, DynamicImage) = result.unwrap();
        assert!(image.width() > 0);
        assert!(image.height() > 0);
        assert!(frame_number <= 4, "unexpected frame number {frame_number}");
        count += 1;
    }

    assert_eq!(count, 5, "expected 5 frames from stream");
}

#[tokio::test]
async fn stream_non_consecutive_frames() {
    if skip_unless(SAMPLE_VIDEO) {
        return;
    }

    let mut unbundler = MediaFile::open(SAMPLE_VIDEO).unwrap();
    let mut stream = unbundler
        .video()
        .frame_stream(FrameRange::Specific(vec![0, 30, 60]), ExtractOptions::new())
        .unwrap();

    let mut frames = Vec::new();
    while let Some(result) = stream.next().await {
        let (_, image): (u64, DynamicImage) = result.unwrap();
        frames.push(image);
    }

    assert_eq!(frames.len(), 3);
}

#[tokio::test]
async fn stream_range() {
    if skip_unless(SAMPLE_VIDEO) {
        return;
    }

    let mut unbundler = MediaFile::open(SAMPLE_VIDEO).unwrap();
    let mut stream = unbundler
        .video()
        .frame_stream(FrameRange::Range(10, 14), ExtractOptions::new())
        .unwrap();

    let mut count = 0u64;
    while let Some(result) = stream.next().await {
        let _ = result.unwrap();
        count += 1;
    }

    assert_eq!(count, 5, "Range(10, 14) should yield 5 frames");
}

#[tokio::test]
async fn stream_no_video_stream_error() {
    if skip_unless(AUDIO_ONLY) {
        return;
    }

    let mut unbundler = MediaFile::open(AUDIO_ONLY).unwrap();
    let result = unbundler
        .video()
        .frame_stream(FrameRange::Range(0, 0), ExtractOptions::new());

    assert!(result.is_err(), "should error on audio-only file");
}

#[tokio::test]
async fn audio_future_extracts_wav() {
    if skip_unless(SAMPLE_VIDEO) {
        return;
    }

    let mut unbundler = MediaFile::open(SAMPLE_VIDEO).unwrap();
    let audio_bytes = unbundler
        .audio()
        .extract_async(unbundle::AudioFormat::Wav, ExtractOptions::new())
        .unwrap()
        .await
        .unwrap();

    assert!(!audio_bytes.is_empty(), "expected non-empty WAV data");
    assert_eq!(
        &audio_bytes[..4],
        b"RIFF",
        "WAV output should start with RIFF header",
    );
}

#[tokio::test]
async fn stream_from_open_url_source_input() {
    if skip_unless(SAMPLE_VIDEO) {
        return;
    }

    let mut unbundler = MediaFile::open_url(SAMPLE_VIDEO).unwrap();
    let mut stream = unbundler
        .video()
        .frame_stream(FrameRange::Range(0, 2), ExtractOptions::new())
        .unwrap();

    let mut count = 0u64;
    while let Some(result) = stream.next().await {
        let (_frame_number, image): (u64, DynamicImage) = result.unwrap();
        assert!(image.width() > 0);
        assert!(image.height() > 0);
        count += 1;
    }

    assert_eq!(count, 3, "expected three frames from URL-opened source");
}

#[tokio::test]
async fn audio_future_from_open_url_source_input() {
    if skip_unless(SAMPLE_VIDEO) {
        return;
    }

    let mut unbundler = MediaFile::open_url(SAMPLE_VIDEO).unwrap();
    let audio_bytes = unbundler
        .audio()
        .extract_async(unbundle::AudioFormat::Wav, ExtractOptions::new())
        .unwrap()
        .await
        .unwrap();

    assert!(!audio_bytes.is_empty(), "expected non-empty WAV data");
    assert_eq!(&audio_bytes[..4], b"RIFF");
}