]>
Commit | Line | Data |
---|---|---|
1 | import { expect } from 'chai' | |
2 | import ffmpeg from 'fluent-ffmpeg' | |
3 | import { ensureDir, pathExists } from 'fs-extra' | |
4 | import { dirname } from 'path' | |
5 | import { buildAbsoluteFixturePath, getMaxTheoreticalBitrate } from '@shared/core-utils' | |
6 | import { getVideoStreamBitrate, getVideoStreamDimensionsInfo, getVideoStreamFPS } from '@shared/ffmpeg' | |
7 | ||
8 | async function ensureHasTooBigBitrate (fixturePath: string) { | |
9 | const bitrate = await getVideoStreamBitrate(fixturePath) | |
10 | const dataResolution = await getVideoStreamDimensionsInfo(fixturePath) | |
11 | const fps = await getVideoStreamFPS(fixturePath) | |
12 | ||
13 | const maxBitrate = getMaxTheoreticalBitrate({ ...dataResolution, fps }) | |
14 | expect(bitrate).to.be.above(maxBitrate) | |
15 | } | |
16 | ||
17 | async function generateHighBitrateVideo () { | |
18 | const tempFixturePath = buildAbsoluteFixturePath('video_high_bitrate_1080p.mp4', true) | |
19 | ||
20 | await ensureDir(dirname(tempFixturePath)) | |
21 | ||
22 | const exists = await pathExists(tempFixturePath) | |
23 | if (!exists) { | |
24 | console.log('Generating high bitrate video.') | |
25 | ||
26 | // Generate a random, high bitrate video on the fly, so we don't have to include | |
27 | // a large file in the repo. The video needs to have a certain minimum length so | |
28 | // that FFmpeg properly applies bitrate limits. | |
29 | // https://stackoverflow.com/a/15795112 | |
30 | return new Promise<string>((res, rej) => { | |
31 | ffmpeg() | |
32 | .outputOptions([ '-f rawvideo', '-video_size 1920x1080', '-i /dev/urandom' ]) | |
33 | .outputOptions([ '-ac 2', '-f s16le', '-i /dev/urandom', '-t 10' ]) | |
34 | .outputOptions([ '-maxrate 10M', '-bufsize 10M' ]) | |
35 | .output(tempFixturePath) | |
36 | .on('error', rej) | |
37 | .on('end', () => res(tempFixturePath)) | |
38 | .run() | |
39 | }) | |
40 | } | |
41 | ||
42 | await ensureHasTooBigBitrate(tempFixturePath) | |
43 | ||
44 | return tempFixturePath | |
45 | } | |
46 | ||
47 | async function generateVideoWithFramerate (fps = 60) { | |
48 | const tempFixturePath = buildAbsoluteFixturePath(`video_${fps}fps.mp4`, true) | |
49 | ||
50 | await ensureDir(dirname(tempFixturePath)) | |
51 | ||
52 | const exists = await pathExists(tempFixturePath) | |
53 | if (!exists) { | |
54 | console.log('Generating video with framerate %d.', fps) | |
55 | ||
56 | return new Promise<string>((res, rej) => { | |
57 | ffmpeg() | |
58 | .outputOptions([ '-f rawvideo', '-video_size 1280x720', '-i /dev/urandom' ]) | |
59 | .outputOptions([ '-ac 2', '-f s16le', '-i /dev/urandom', '-t 10' ]) | |
60 | .outputOptions([ `-r ${fps}` ]) | |
61 | .output(tempFixturePath) | |
62 | .on('error', rej) | |
63 | .on('end', () => res(tempFixturePath)) | |
64 | .run() | |
65 | }) | |
66 | } | |
67 | ||
68 | return tempFixturePath | |
69 | } | |
70 | ||
71 | export { | |
72 | generateHighBitrateVideo, | |
73 | generateVideoWithFramerate | |
74 | } |