]>
Commit | Line | Data |
---|---|---|
679c12e6 | 1 | import { expect } from 'chai' |
41fb13c3 | 2 | import ffmpeg from 'fluent-ffmpeg' |
6c5065a0 C |
3 | import { ensureDir, pathExists } from 'fs-extra' |
4 | import { dirname } from 'path' | |
e7d8e2b2 | 5 | import { buildAbsoluteFixturePath, getMaxTheoreticalBitrate } from '@shared/core-utils' |
d102de1b | 6 | import { getVideoStreamBitrate, getVideoStreamDimensionsInfo, getVideoStreamFPS } from '@shared/ffmpeg' |
6c5065a0 | 7 | |
679c12e6 | 8 | async function ensureHasTooBigBitrate (fixturePath: string) { |
c729caf6 C |
9 | const bitrate = await getVideoStreamBitrate(fixturePath) |
10 | const dataResolution = await getVideoStreamDimensionsInfo(fixturePath) | |
11 | const fps = await getVideoStreamFPS(fixturePath) | |
679c12e6 | 12 | |
e7d8e2b2 | 13 | const maxBitrate = getMaxTheoreticalBitrate({ ...dataResolution, fps }) |
679c12e6 C |
14 | expect(bitrate).to.be.above(maxBitrate) |
15 | } | |
16 | ||
6c5065a0 C |
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 | ||
679c12e6 C |
42 | await ensureHasTooBigBitrate(tempFixturePath) |
43 | ||
6c5065a0 C |
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 | } |