shared/ffmpeg/ffmpeg-live.ts

   1 import { FilterSpecification } from 'fluent-ffmpeg'
   2 import { join } from 'path'
   3 import { pick } from '@shared/core-utils'
   4 import { FFmpegCommandWrapper, FFmpegCommandWrapperOptions } from './ffmpeg-command-wrapper'
   5 import { buildStreamSuffix, getScaleFilter, StreamType } from './ffmpeg-utils'
   6 import { addDefaultEncoderGlobalParams, addDefaultEncoderParams, applyEncoderOptions } from './shared'
   7
   8 export class FFmpegLive {
   9   private readonly commandWrapper: FFmpegCommandWrapper
  10
  11   constructor (options: FFmpegCommandWrapperOptions) {
  12     this.commandWrapper = new FFmpegCommandWrapper(options)
  13   }
  14
  15   async getLiveTranscodingCommand (options: {
  16     inputUrl: string
  17
  18     outPath: string
  19     masterPlaylistName: string
  20
  21     toTranscode: {
  22       resolution: number
  23       fps: number
  24     }[]
  25
  26     // Input information
  27     bitrate: number
  28     ratio: number
  29     hasAudio: boolean
  30
  31     segmentListSize: number
  32     segmentDuration: number
  33   }) {
  34     const {
  35       inputUrl,
  36       outPath,
  37       toTranscode,
  38       bitrate,
  39       masterPlaylistName,
  40       ratio,
  41       hasAudio
  42     } = options
  43     const command = this.commandWrapper.buildCommand(inputUrl)
  44
  45     const varStreamMap: string[] = []
  46
  47     const complexFilter: FilterSpecification[] = [
  48       {
  49         inputs: '[v:0]',
  50         filter: 'split',
  51         options: toTranscode.length,
  52         outputs: toTranscode.map(t => `vtemp${t.resolution}`)
  53       }
  54     ]
  55
  56     command.outputOption('-sc_threshold 0')
  57
  58     addDefaultEncoderGlobalParams(command)
  59
  60     for (let i = 0; i < toTranscode.length; i++) {
  61       const streamMap: string[] = []
  62       const { resolution, fps } = toTranscode[i]
  63
  64       const baseEncoderBuilderParams = {
  65         input: inputUrl,
  66
  67         canCopyAudio: true,
  68         canCopyVideo: true,
  69
  70         inputBitrate: bitrate,
  71         inputRatio: ratio,
  72
  73         resolution,
  74         fps,
  75
  76         streamNum: i,
  77         videoType: 'live' as 'live'
  78       }
  79
  80       {
  81         const streamType: StreamType = 'video'
  82         const builderResult = await this.commandWrapper.getEncoderBuilderResult({ ...baseEncoderBuilderParams, streamType })
  83         if (!builderResult) {
  84           throw new Error('No available live video encoder found')
  85         }
  86
  87         command.outputOption(`-map [vout${resolution}]`)
  88
  89         addDefaultEncoderParams({ command, encoder: builderResult.encoder, fps, streamNum: i })
  90
  91         this.commandWrapper.debugLog(
  92           `Apply ffmpeg live video params from ${builderResult.encoder} using ${this.commandWrapper.getProfile()} profile.`,
  93           { builderResult, fps, toTranscode }
  94         )
  95
  96         command.outputOption(`${buildStreamSuffix('-c:v', i)} ${builderResult.encoder}`)
  97         applyEncoderOptions(command, builderResult.result)
  98
  99         complexFilter.push({
 100           inputs: `vtemp${resolution}`,
 101           filter: getScaleFilter(builderResult.result),
 102           options: `w=-2:h=${resolution}`,
 103           outputs: `vout${resolution}`
 104         })
 105
 106         streamMap.push(`v:${i}`)
 107       }
 108
 109       if (hasAudio) {
 110         const streamType: StreamType = 'audio'
 111         const builderResult = await this.commandWrapper.getEncoderBuilderResult({ ...baseEncoderBuilderParams, streamType })
 112         if (!builderResult) {
 113           throw new Error('No available live audio encoder found')
 114         }
 115
 116         command.outputOption('-map a:0')
 117
 118         addDefaultEncoderParams({ command, encoder: builderResult.encoder, fps, streamNum: i })
 119
 120         this.commandWrapper.debugLog(
 121           `Apply ffmpeg live audio params from ${builderResult.encoder} using ${this.commandWrapper.getProfile()} profile.`,
 122           { builderResult, fps, resolution }
 123         )
 124
 125         command.outputOption(`${buildStreamSuffix('-c:a', i)} ${builderResult.encoder}`)
 126         applyEncoderOptions(command, builderResult.result)
 127
 128         streamMap.push(`a:${i}`)
 129       }
 130
 131       varStreamMap.push(streamMap.join(','))
 132     }
 133
 134     command.complexFilter(complexFilter)
 135
 136     this.addDefaultLiveHLSParams({ ...pick(options, [ 'segmentDuration', 'segmentListSize' ]), outPath, masterPlaylistName })
 137
 138     command.outputOption('-var_stream_map', varStreamMap.join(' '))
 139
 140     return command
 141   }
 142
 143   getLiveMuxingCommand (options: {
 144     inputUrl: string
 145     outPath: string
 146     masterPlaylistName: string
 147
 148     segmentListSize: number
 149     segmentDuration: number
 150   }) {
 151     const { inputUrl, outPath, masterPlaylistName } = options
 152
 153     const command = this.commandWrapper.buildCommand(inputUrl)
 154
 155     command.outputOption('-c:v copy')
 156     command.outputOption('-c:a copy')
 157     command.outputOption('-map 0:a?')
 158     command.outputOption('-map 0:v?')
 159
 160     this.addDefaultLiveHLSParams({ ...pick(options, [ 'segmentDuration', 'segmentListSize' ]), outPath, masterPlaylistName })
 161
 162     return command
 163   }
 164
 165   private addDefaultLiveHLSParams (options: {
 166     outPath: string
 167     masterPlaylistName: string
 168     segmentListSize: number
 169     segmentDuration: number
 170   }) {
 171     const { outPath, masterPlaylistName, segmentListSize, segmentDuration } = options
 172
 173     const command = this.commandWrapper.getCommand()
 174
 175     command.outputOption('-hls_time ' + segmentDuration)
 176     command.outputOption('-hls_list_size ' + segmentListSize)
 177     command.outputOption('-hls_flags delete_segments+independent_segments+program_date_time')
 178     command.outputOption(`-hls_segment_filename ${join(outPath, '%v-%06d.ts')}`)
 179     command.outputOption('-master_pl_name ' + masterPlaylistName)
 180     command.outputOption(`-f hls`)
 181
 182     command.output(join(outPath, '%v.m3u8'))
 183   }
 184 }