diff options
Diffstat (limited to 'server/helpers/ffmpeg-utils.ts')
-rw-r--r-- | server/helpers/ffmpeg-utils.ts | 35 |
1 files changed, 23 insertions, 12 deletions
diff --git a/server/helpers/ffmpeg-utils.ts b/server/helpers/ffmpeg-utils.ts index 7f84a049f..830625cc6 100644 --- a/server/helpers/ffmpeg-utils.ts +++ b/server/helpers/ffmpeg-utils.ts | |||
@@ -3,10 +3,18 @@ import * as ffmpeg from 'fluent-ffmpeg' | |||
3 | import { readFile, remove, writeFile } from 'fs-extra' | 3 | import { readFile, remove, writeFile } from 'fs-extra' |
4 | import { dirname, join } from 'path' | 4 | import { dirname, join } from 'path' |
5 | import { FFMPEG_NICE, VIDEO_LIVE } from '@server/initializers/constants' | 5 | import { FFMPEG_NICE, VIDEO_LIVE } from '@server/initializers/constants' |
6 | import { AvailableEncoders, EncoderOptions, EncoderOptionsBuilder, EncoderProfile, VideoResolution } from '../../shared/models/videos' | 6 | import { pick } from '@shared/core-utils' |
7 | import { | ||
8 | AvailableEncoders, | ||
9 | EncoderOptions, | ||
10 | EncoderOptionsBuilder, | ||
11 | EncoderOptionsBuilderParams, | ||
12 | EncoderProfile, | ||
13 | VideoResolution | ||
14 | } from '../../shared/models/videos' | ||
7 | import { CONFIG } from '../initializers/config' | 15 | import { CONFIG } from '../initializers/config' |
8 | import { execPromise, promisify0 } from './core-utils' | 16 | import { execPromise, promisify0 } from './core-utils' |
9 | import { computeFPS, ffprobePromise, getAudioStream, getVideoFileBitrate, getVideoFileFPS } from './ffprobe-utils' | 17 | import { computeFPS, ffprobePromise, getAudioStream, getVideoFileBitrate, getVideoFileFPS, getVideoFileResolution } from './ffprobe-utils' |
10 | import { processImage } from './image-utils' | 18 | import { processImage } from './image-utils' |
11 | import { logger } from './logger' | 19 | import { logger } from './logger' |
12 | 20 | ||
@@ -217,13 +225,16 @@ async function getLiveTranscodingCommand (options: { | |||
217 | masterPlaylistName: string | 225 | masterPlaylistName: string |
218 | 226 | ||
219 | resolutions: number[] | 227 | resolutions: number[] |
228 | |||
229 | // Input information | ||
220 | fps: number | 230 | fps: number |
221 | bitrate: number | 231 | bitrate: number |
232 | ratio: number | ||
222 | 233 | ||
223 | availableEncoders: AvailableEncoders | 234 | availableEncoders: AvailableEncoders |
224 | profile: string | 235 | profile: string |
225 | }) { | 236 | }) { |
226 | const { rtmpUrl, outPath, resolutions, fps, bitrate, availableEncoders, profile, masterPlaylistName } = options | 237 | const { rtmpUrl, outPath, resolutions, fps, bitrate, availableEncoders, profile, masterPlaylistName, ratio } = options |
227 | const input = rtmpUrl | 238 | const input = rtmpUrl |
228 | 239 | ||
229 | const command = getFFmpeg(input, 'live') | 240 | const command = getFFmpeg(input, 'live') |
@@ -253,9 +264,12 @@ async function getLiveTranscodingCommand (options: { | |||
253 | availableEncoders, | 264 | availableEncoders, |
254 | profile, | 265 | profile, |
255 | 266 | ||
256 | fps: resolutionFPS, | ||
257 | inputBitrate: bitrate, | 267 | inputBitrate: bitrate, |
268 | inputRatio: ratio, | ||
269 | |||
258 | resolution, | 270 | resolution, |
271 | fps: resolutionFPS, | ||
272 | |||
259 | streamNum: i, | 273 | streamNum: i, |
260 | videoType: 'live' as 'live' | 274 | videoType: 'live' as 'live' |
261 | } | 275 | } |
@@ -502,7 +516,7 @@ function getHLSVideoPath (options: HLSTranscodeOptions | HLSFromTSTranscodeOptio | |||
502 | // Run encoder builder depending on available encoders | 516 | // Run encoder builder depending on available encoders |
503 | // Try encoders by priority: if the encoder is available, run the chosen profile or fallback to the default one | 517 | // Try encoders by priority: if the encoder is available, run the chosen profile or fallback to the default one |
504 | // If the default one does not exist, check the next encoder | 518 | // If the default one does not exist, check the next encoder |
505 | async function getEncoderBuilderResult (options: { | 519 | async function getEncoderBuilderResult (options: EncoderOptionsBuilderParams & { |
506 | streamType: 'video' | 'audio' | 520 | streamType: 'video' | 'audio' |
507 | input: string | 521 | input: string |
508 | 522 | ||
@@ -510,13 +524,8 @@ async function getEncoderBuilderResult (options: { | |||
510 | profile: string | 524 | profile: string |
511 | 525 | ||
512 | videoType: 'vod' | 'live' | 526 | videoType: 'vod' | 'live' |
513 | |||
514 | resolution: number | ||
515 | inputBitrate: number | ||
516 | fps?: number | ||
517 | streamNum?: number | ||
518 | }) { | 527 | }) { |
519 | const { availableEncoders, input, profile, resolution, streamType, fps, inputBitrate, streamNum, videoType } = options | 528 | const { availableEncoders, profile, streamType, videoType } = options |
520 | 529 | ||
521 | const encodersToTry = availableEncoders.encodersToTry[videoType][streamType] | 530 | const encodersToTry = availableEncoders.encodersToTry[videoType][streamType] |
522 | const encoders = availableEncoders.available[videoType] | 531 | const encoders = availableEncoders.available[videoType] |
@@ -546,7 +555,7 @@ async function getEncoderBuilderResult (options: { | |||
546 | } | 555 | } |
547 | } | 556 | } |
548 | 557 | ||
549 | const result = await builder({ input, resolution, inputBitrate, fps, streamNum }) | 558 | const result = await builder(pick(options, [ 'input', 'resolution', 'inputBitrate', 'fps', 'inputRatio', 'streamNum' ])) |
550 | 559 | ||
551 | return { | 560 | return { |
552 | result, | 561 | result, |
@@ -581,6 +590,7 @@ async function presetVideo (options: { | |||
581 | // Audio encoder | 590 | // Audio encoder |
582 | const parsedAudio = await getAudioStream(input, probe) | 591 | const parsedAudio = await getAudioStream(input, probe) |
583 | const bitrate = await getVideoFileBitrate(input, probe) | 592 | const bitrate = await getVideoFileBitrate(input, probe) |
593 | const { ratio } = await getVideoFileResolution(input, probe) | ||
584 | 594 | ||
585 | let streamsToProcess: StreamType[] = [ 'audio', 'video' ] | 595 | let streamsToProcess: StreamType[] = [ 'audio', 'video' ] |
586 | 596 | ||
@@ -600,6 +610,7 @@ async function presetVideo (options: { | |||
600 | profile, | 610 | profile, |
601 | fps, | 611 | fps, |
602 | inputBitrate: bitrate, | 612 | inputBitrate: bitrate, |
613 | inputRatio: ratio, | ||
603 | videoType: 'vod' as 'vod' | 614 | videoType: 'vod' as 'vod' |
604 | }) | 615 | }) |
605 | 616 | ||