gabehf
/
Tdarr_Plugins
mirror of https://github.com/gabehf/Tdarr_Plugins.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '23bef4c778'
${ noResults }
Tdarr_Plugins/Community/Tdarr_Plugin_bsh1_Boosh_FFM...

865 lines
37 KiB
Raw Blame History

// All credit for original plugin logic goes to Migz.
// This Plugin is essentially just his NVENC/CPU plugin modified to work with QSV & with extra hevc logic.
// Extra logic is mainly to control encoder quality/speed & to allow HEVC files to be reprocessed to reduce file size
// NOTE - This does not use VAAPI, it is QSV only. So newer intel iGPUs only. 8th+ gen should work.
// Additionally this was designed and tested on UNRAID via docker, though there is logic to support use on
// Windows & Linux - Both platforms have now been confirmed working, however there is no way to test all use cases
// Mac is supported, however it does not use QSV. This is because ffmpeg on Mac does not actually leverage QSV and
// instead uses "VideoToolbox" which is more a general video encode accelerator.
// White paper from intel regarding QSV performance on linux using FFMPEG here:
// eslint-disable-next-line max-len
// https://www.intel.com/content/dam/www/public/us/en/documents/white-papers/cloud-computing-quicksync-video-ffmpeg-white-paper.pdf
const details = () => ({
id: 'Tdarr_Plugin_bsh1_Boosh_FFMPEG_QSV_HEVC',
Stage: 'Pre-processing',
Name: 'Boosh-Transcode using QSV GPU & FFMPEG',
Type: 'Video',
Operation: 'Transcode',
Description: `==DETAILS== This is a QSV plugin. 8th+ gen INTEL QSV enabled CPUs are recommended. VAAPI is NOT used.
\n\n==OS SUPPORT== This plugin supports Linux & Windows using QSV. Mac is supported though cannot use QSV and
relies on 'VideoToolBox' - Expect to see different encode speed & quality on Mac compared to other platforms.
Ensure you set your node settings accordingly!
\n\n==LOGIC== Files will be transcoded into H265/HEVC using Quick Sync Video (QSV) via Intel GPU using ffmpeg.
Settings are dependant on file bitrate working by the logic that H265 can support the same amount of data at half
the bitrate of H264. This plugin will skip files already in HEVC, AV1 & VP9 unless "reconvert_hevc" is marked as
true. If it is then these will be reconverted again if they exceed the bitrate specified in "hevc_max_bitrate".
This plugin relies on understanding the accurate video bitrate of your files. It's highly recommended to remux
into MKV & enable "Run mkvpropedit on files before running plugins" under Tdarr>Options.`,
Version: '1.21',
Tags: 'pre-processing,ffmpeg,video only,qsv,h265,hevc,mkvpropedit,configurable',
Inputs: [
{
name: 'container',
type: 'string',
defaultValue: 'mkv',
inputUI: {
type: 'dropdown',
options: [
'mkv',
'mp4',
],
},
tooltip: `\\n
==DESCRIPTION==
\\nSpecifies the output container of the file.
\\nEnsure that all stream types you may have are supported by your chosen container.
\\n
==INFO==
\\nOnly MP4 & MKV are supported and MKV is recommended.
\\nExample:\\n
mkv
\\nExample:\\n
mp4`,
},
{
name: 'force_conform',
type: 'boolean',
defaultValue: false,
inputUI: {
type: 'dropdown',
options: [
'false',
'true',
],
},
tooltip: `\\n
==DESCRIPTION==
\\nMake the file conform to output containers requirements.
Use if you need to ensure the encode works from mp4>mkv or mkv>mp4. \\n
==WARNING== \\n
This will remove data of certain types so ensure you are happy with that,
or use another plugin to convert these data types first!
\\n
==INFO==
\\nDrop hdmv_pgs_subtitle/eia_608/subrip/timed_id3 for MP4.
\\nDrop data streams/mov_text/eia_608/timed_id3 for MKV.
\\nDefault is false.
\\nExample:\\n
true
\\nExample:\\n
false`,
},
{
name: 'enable_10bit',
type: 'boolean',
defaultValue: false,
inputUI: {
type: 'dropdown',
options: [
'false',
'true',
],
},
tooltip: `\\n
==DESCRIPTION==
\\nSpecify if we want to enable 10bit encoding.
\\nIf this is enabled files will be processed and converted into 10bit
HEVC using main10 profile and with p010le pixel format. \n
If you just want to retain files that are already 10 bit then this can be left as false, as
10bit to 10bit in ffmpeg should be automatic.
\\n
==INFO==
\\nDefault is "false".
\\nExample:\\n
true
\\nExample:\\n
false`,
},
{
name: 'encoder_speedpreset',
type: 'string',
defaultValue: 'slow',
inputUI: {
type: 'dropdown',
options: [
'veryfast',
'faster',
'fast',
'medium',
'slow',
'slower',
'veryslow',
],
},
tooltip: `\\n
==DESCRIPTION==
\\nSpecify the encoder speed/preset to use.
Slower options mean a slower encode but better quality and faster options mean faster encodes but
worse quality.
\\nFor more information see intel white paper on ffmpeg results using QSV: \\n`
// eslint-disable-next-line max-len
+ `https://www.intel.com/content/dam/www/public/us/en/documents/white-papers/cloud-computing-quicksync-video-ffmpeg-white-paper.pdf
\\n
==INFO==
\\nDefault is "slow".
\\nExample:\\n
medium
\\nExample:\\n
slower`,
},
{
name: 'extra_qsv_options',
type: 'string',
defaultValue: '',
inputUI: {
type: 'text',
},
tooltip: `\\n
==DESCRIPTION==
\\nHere you can add extra options to the ffmpeg QSV ENCODE cmd.
This does not override the ffmpeg cmd, it just allows additions to it.
\\n
There are extra QSV options that can be
forced on/off as desired. See here for some possible cmds -
https://ffmpeg.org/ffmpeg-codecs.html#toc-HEVC-Options-1
\\n
==WARNING== \\n
Be certain to verify the cmds work before adding to your workflow. \\n
Check Tdarr Help Tab. Enter ffmpeg cmd - "-h encoder=hevc_qsv". This will give a list of supported commands. \\n
MAC SPECIFIC - This option is ignored on Mac because videotoolbox is used rather than qsv.
\\n
==INFO==
\\nDefault is empty but the first example below has a suggested value. If unsure just leave empty.
\\nEnsure to only use cmds valid to encoding QSV as the script handles other ffmpeg cmds relating to
bitrate etc. Anything else entered here might be supported but could cause undesired results.
\\nIf you are using a "-vf" cmd, please put it at the end to avoid issues!
\\nExample:\\n
-look_ahead 1 -look_ahead_depth 100 -extbrc 1 -rdo 1 -mbbrc 1 -b_strategy 1 -adaptive_i 1 -adaptive_b 1
\\n Above enables look ahead, extended bitrate control, b-frames, etc.\\n
\\nExample:\\n
-vf scale_qsv=w=1280:h=720
\\nScale video resolution Method 1\\n
\\nExample:\\n
-vf scale_qsv=1280:-1
\\nScale video resolution Method 2\\n`,
},
{
name: 'bitrate_cutoff',
type: 'number',
defaultValue: 0,
inputUI: {
type: 'text',
},
tooltip: `\\n
==DESCRIPTION==
\\nSpecify bitrate cutoff, files with a video bitrate lower then this will not be processed. \n
\\n
==INFO==
\\nRate is in kbps.
\\nDefaults to 0 which means this is disabled.
\\nEnter a valid number to enable.
\\nExample:\\n
2500
\\nExample:\\n
1500`,
},
{
name: 'max_average_bitrate',
type: 'number',
defaultValue: 0,
inputUI: {
type: 'text',
},
tooltip: `\\n
==DESCRIPTION==
\\nSpecify a maximum average video bitrate. When encoding we take the current video bitrate and halve it
to get an average target. This option sets a upper limit to that average
(i.e if you have a video bitrate of 10000, half is 5000, if your maximum desired average bitrate is 4000
then we use that as the target instead of 5000).
\\n
==INFO==
\\nBitrate here is referring to video bitrate as we want to set the video bitrate on encode.
\\nRate is in kbps.
\\nDefaults to 0 which means this is disabled.
\\nEnter a valid number to enable.
\\nExample:\\n
4000
\\nExample:\\n
3000`,
},
{
name: 'min_average_bitrate',
type: 'number',
defaultValue: 0,
inputUI: {
type: 'text',
},
tooltip: `\\n
==DESCRIPTION==
\\nSpecify a minimum average video bitrate. When encoding we take the current video bitrate and halve
it to get an average target. This option sets a lower limit to that average (i.e if you have a video bitrate
of 3000, half is 1500, if your minimum desired average bitrate is 2000 then we use that as the target instead
of 1500).
\\n
==INFO==
\\nBitrate here is referring to video bitrate as we want to set the video bitrate on encode.
\\nRate is in kbps.
\\nDefaults to 0 which means this is disabled.
\\nEnter a valid number to enable.
\\nExample:\\n
2000
\\nExample:\\n
1000`,
},
{
name: 'reconvert_hevc',
type: 'boolean',
defaultValue: false,
inputUI: {
type: 'dropdown',
options: [
'false',
'true',
],
},
tooltip: `\\n
==DESCRIPTION==
\\nSet to reprocess HEVC, VP9 or AV1 files (i.e reduce bitrate of files already in those codecs).
\\nSince this uses the same logic as normal, halving the current bitrate, this is NOT recommended
unless you know what you are doing, so please leave FALSE if unsure!
\\nNEEDS to be used in conjunction with "bitrate_cutoff" or "hevc_max_bitrate" otherwise is ignored.
\\nThis is useful in certain situations, perhaps you have a file which is HEVC but has an extremely high
bitrate and you'd like to reduce it.
\\n
==WARNING== \\n
IF YOU HAVE VP9 OR AV1 FILES YOU WANT TO KEEP IN THOSE FORMATS THEN DO NOT USE THIS OPTION. \\n
\\nThis option has the potential to LOOP your encodes! You can encode a file to HEVC and still
be above your cutoff and it would be converted again & again if this is set to true (since it's now HEVC).
So if you use this be sure to set "hevc_max_bitrate" & "max_average_bitrate" to help prevent the plugin looping.
Also it is highly suggested that you have your "hevc_max_bitrate" higher than "max_average_bitrate".
\\nPlease be certain you want this enabled before setting it otherwise leave this as FALSE!
While the plugin will attempt to generate accurate video bitrate metadata, it can not always reliably do so
and will be forced to fall back onto estimates. Please bare this in mind when using the HEVC reprocess option.
\\n
\\nExample:\\n
true
\\nExample:\\n
false`,
},
{
name: 'hevc_max_bitrate',
type: 'number',
defaultValue: 0,
inputUI: {
type: 'text',
},
tooltip: `\\n
==DESCRIPTION==
\\nHas no effect unless "reconvert_hevc" is set to true. This allows you to specify a maximum
allowed average OVERALL bitrate for HEVC or similar files. Much like the "bitrate_cutoff" option, but
specifically for HEVC files. It should be set HIGHER then your standard cutoff for safety.
\\nAlso, it's highly suggested you use the min & max average bitrate options in combination with this. You
will want those to control the encoded video bitrate, otherwise you may end up repeatedly reprocessing HEVC files.
i.e your file might have a overall bitrate of 20000, if your hevc cutoff is 5000 then it's going to reconvert
multiple times before it'll fall below that cutoff. While HEVC reprocessing can be useful this is why it is NOT
recommended!
\\n
==WARNING== \\n
While the plugin will attempt to generate accurate video bitrate metadata, it can not always reliably do so
and will be forced to fall back onto estimates. Please bare this in mind when using the HEVC reprocess option.
\\n
==INFO==
\\nRate is in kbps.
\\nDefaults to 0 which means this is disabled.
\\nEnter a valid number to enable, otherwise we use "bitrate_cutoff" and multiply x2 for a safe limit.
\\nExample:\\n
4000
\\nExample:\\n
3000`,
},
],
});
// Set up required variables.
let currentBitrate = 0;
let overallBitRate = 0;
let targetBitrate = 0;
let minimumBitrate = 0;
let maximumBitrate = 0;
let duration = '';
let videoIdx = 0;
let extraArguments = '';
let bitrateSettings = '';
let inflatedCutoff = 0;
let main10 = false;
let high10 = false;
let oldFormat = false;
let videoBR = 0;
// eslint-disable-next-line @typescript-eslint/no-unused-vars
const plugin = (file, librarySettings, inputs, otherArguments) => {
const lib = require('../methods/lib')();
const os = require('os');
// eslint-disable-next-line @typescript-eslint/no-unused-vars,no-param-reassign
inputs = lib.loadDefaultValues(inputs, details);
const response = {
processFile: false,
preset: '',
handBrakeMode: false,
FFmpegMode: true,
reQueueAfter: true,
infoLog: '',
container: `.${inputs.container}`,
};
if (file.fileMedium !== 'video') {
response.processFile = false;
response.infoLog += `☒ File seems to be ${file.fileMedium} & not video. Exiting \n`;
return response;
}
for (let i = 0; i < file.ffProbeData.streams.length; i += 1) {
const strstreamType = file.ffProbeData.streams[i].codec_type.toLowerCase();
videoIdx = -1;
// Check if stream is a video.
if (videoIdx === -1 && strstreamType === 'video') {
videoIdx = i;
// Try checking file stats using Mediainfo first, then ffprobe.
try {
videoBR = Number(file.mediaInfo.track[i + 1].BitRate) / 1000;
if (videoBR <= 0 || Number.isNaN(videoBR)) {
if (Number(file.ffProbeData.streams[i].tags.BPS) > 0) {
videoBR = file.ffProbeData.streams[i].tags.BPS / 1000;
} else if (Number(file.ffProbeData.streams[i].tags.BPS['-eng']) > 0) {
videoBR = file.ffProbeData.streams[i].tags.BPS['-eng'] / 1000;
}
}
} catch (err) {
// Catch error - Ignore & carry on - If check can bomb out if tags don't exist...
videoBR = 0; // Set videoBR to 0 for safety
}
// Check if duration info is filled, if so convert time format to minutes.
if (Number.isNaN(file.meta.Duration) === false) {
// If duration is a number then convert seconds to minutes
duration = file.meta.Duration / 60;
} else if (typeof file.meta.Duration !== 'undefined') {
// Get seconds by using a Date & then convert to minutes
duration = file.meta.Duration;
duration = (new Date(`1970-01-01T${duration}Z`).getTime() / 1000) / 60;
} else { // If not filled then get duration of video stream and do the same.
duration = file.ffProbeData.streams[videoIdx].tags.DURATION;
duration = (new Date(`1970-01-01T${duration}Z`).getTime() / 1000) / 60;
}
}
}
if (Number.isNaN(videoBR) || videoBR <= 0) {
// Work out currentBitrate using "Bitrate = file size / (number of minutes * .0075)"
currentBitrate = Math.round(file.file_size / (duration * 0.0075));
response.infoLog += '==WARNING== Failed to get an accurate video bitrate, ';
response.infoLog += `falling back to old method to get OVERALL file bitrate of ${currentBitrate}kbps. `;
response.infoLog += 'Bitrate calculations for video encode will likely be inaccurate... \n';
} else {
currentBitrate = Math.round(videoBR);
response.infoLog += `☑ It looks like the current video bitrate is ${currentBitrate}kbps. \n`;
}
// Get overall bitrate for use with HEVC reprocessing
overallBitRate = Math.round(file.file_size / (duration * 0.0075));
// Halve current bitrate for Target bitrate, in theory h265 can be half the bitrate as h264 without losing quality.
targetBitrate = Math.round(currentBitrate / 2);
// Allow some leeway under and over the targetBitrate.
minimumBitrate = Math.round(targetBitrate * 0.75);
maximumBitrate = Math.round(targetBitrate * 1.25);
// If targetBitrate or currentBitrate comes out as 0 then something
// has gone wrong and bitrates could not be calculated.
// Cancel plugin completely.
if (targetBitrate <= 0 || currentBitrate <= 0) {
response.infoLog += '☒ Target bitrate could not be calculated. Skipping this plugin. \n';
return response;
}
// If targetBitrate is equal or greater than currentBitrate then something
// has gone wrong as that is not what we want.
// Cancel plugin completely.
if (targetBitrate >= currentBitrate) {
response.infoLog += `☒ Target bitrate has been calculated as ${targetBitrate}kbps. This is equal or greater `;
response.infoLog += "than the current bitrate... Something has gone wrong and this shouldn't happen! "
+ 'Skipping this plugin. \n';
return response;
}
// Ensure that bitrate_cutoff is set if reconvert_hevc is true since we need some protection against a loop
// Cancel the plugin
if (inputs.reconvert_hevc === true && inputs.bitrate_cutoff <= 0 && inputs.hevc_max_bitrate <= 0) {
response.infoLog += `Reconvert HEVC is ${inputs.reconvert_hevc}, however there is no bitrate cutoff `;
response.infoLog += 'or HEVC specific cutoff set so we have no way to know when to stop processing this file. \n'
+ 'Either set reconvert_HEVC to false or set a bitrate cutoff and set a hevc_max_bitrate cutoff. \n'
+ '☒ Skipping this plugin. \n';
return response;
}
// Check if inputs.bitrate cutoff has something entered.
// (Entered means user actually wants something to happen, empty would disable this).
if (inputs.bitrate_cutoff > 0) {
// Checks if currentBitrate is below inputs.bitrate_cutoff.
// If so then cancel plugin without touching original files.
if (currentBitrate <= inputs.bitrate_cutoff) {
response.infoLog += `☑ Current bitrate is below set cutoff of ${inputs.bitrate_cutoff}kbps. \n`
+ 'Cancelling plugin. \n';
return response;
}
// If above cutoff then carry on
if (currentBitrate > inputs.bitrate_cutoff && inputs.reconvert_hevc === false) {
response.infoLog += '☒ Current bitrate appears to be above the cutoff. Need to process \n';
}
}
if (inputs.max_average_bitrate > 0) {
// Checks if targetBitrate is above inputs.max_average_bitrate.
// If so then clamp target bitrate
if (targetBitrate > inputs.max_average_bitrate) {
response.infoLog += 'Our target bitrate is above the max_average_bitrate ';
response.infoLog += `so clamping at max of ${inputs.max_average_bitrate}kbps. \n`;
targetBitrate = Math.round(inputs.max_average_bitrate);
minimumBitrate = Math.round(targetBitrate * 0.75);
maximumBitrate = Math.round(targetBitrate * 1.25);
}
}
// Check if inputs.min_average_bitrate has something entered.
// (Entered means user actually wants something to happen, empty would disable this).
if (inputs.min_average_bitrate > 0) {
// Exit the plugin is the cutoff is less than the min average bitrate. Most likely user error
if (inputs.bitrate_cutoff < inputs.min_average_bitrate) {
response.infoLog += `☒ Bitrate cutoff ${inputs.bitrate_cutoff}k is less than the set minimum
average bitrate set of ${inputs.min_average_bitrate}kbps. We don't want this. Cancelling plugin. \n`;
return response;
}
// Checks if inputs.bitrate_cutoff is below inputs.min_average_bitrate.
// If so then set currentBitrate to the minimum allowed.)
if (targetBitrate < inputs.min_average_bitrate) {
response.infoLog += `Target average bitrate clamped at min of ${inputs.min_average_bitrate}kbps. \n`;
targetBitrate = Math.round(inputs.min_average_bitrate);
minimumBitrate = Math.round(targetBitrate * 0.75);
maximumBitrate = Math.round(targetBitrate * 1.25);
}
}
// It's possible to remux or flat out convert from mp4 to mkv so we need to conform to standards
// So check streams and add any extra parameters required to make file conform with output format.
// i.e drop mov_text for mkv files and drop pgs_subtitles for mp4
if (inputs.force_conform === true) {
if (inputs.container.toLowerCase() === 'mkv') {
extraArguments += '-map -0:d ';
for (let i = 0; i < file.ffProbeData.streams.length; i += 1) {
try {
if (
file.ffProbeData.streams[i].codec_name
.toLowerCase() === 'mov_text'
|| file.ffProbeData.streams[i].codec_name
.toLowerCase() === 'eia_608'
|| file.ffProbeData.streams[i].codec_name
.toLowerCase() === 'timed_id3'
) {
extraArguments += `-map -0:${i} `;
}
} catch (err) {
// Error
}
}
}
if (inputs.container.toLowerCase() === 'mp4') {
for (let i = 0; i < file.ffProbeData.streams.length; i += 1) {
try {
if (
file.ffProbeData.streams[i].codec_name
.toLowerCase() === 'hdmv_pgs_subtitle'
|| file.ffProbeData.streams[i].codec_name
.toLowerCase() === 'eia_608'
|| file.ffProbeData.streams[i].codec_name
.toLowerCase() === 'subrip'
|| file.ffProbeData.streams[i].codec_name
.toLowerCase() === 'timed_id3'
) {
extraArguments += `-map -0:${i} `;
}
} catch (err) {
// Error
}
}
}
}
// Go through each stream in the file.
for (let i = 0; i < file.ffProbeData.streams.length; i += 1) {
// Check if stream is a video.
if (file.ffProbeData.streams[i].codec_type.toLowerCase() === 'video') {
// Check if codec of stream is mjpeg/png, if so then remove this "video" stream.
// mjpeg/png are usually embedded pictures that can cause havoc with plugins.
if (file.ffProbeData.streams[i].codec_name === 'mjpeg' || file.ffProbeData.streams[i].codec_name === 'png') {
extraArguments += `-map -v:${videoIdx} `;
}
// Check for HDR in files. If so exit plugin. We assume HDR files have bt2020 color spaces. HDR can be complicated
// and some aspects are still unsupported in ffmpeg I believe. Likely we don't want to re-encode anything HDR.
if (file.ffProbeData.streams[i].color_space === 'bt2020nc'
&& file.ffProbeData.streams[i].color_transfer === 'smpte2084'
&& file.ffProbeData.streams[i].color_primaries === 'bt2020') {
response.infoLog += '☒ This looks to be a HDR file. HDR files are unfortunately '
+ 'not supported by this plugin. Exiting plugin. \n\n';
return response;
}
// Now check if we're reprocessing HEVC files, if not then ensure we don't convert HEVC again
if (inputs.reconvert_hevc === false && (file.ffProbeData.streams[i].codec_name === 'hevc'
|| file.ffProbeData.streams[i].codec_name === 'vp9' || file.ffProbeData.streams[i].codec_name === 'av1')) {
// Check if codec of stream is HEVC, VP9 or AV1 AND check if file.container matches inputs.container.
// If so nothing for plugin to do.
if ((file.ffProbeData.streams[i].codec_name === 'hevc' || file.ffProbeData.streams[i].codec_name === 'vp9'
|| file.ffProbeData.streams[i].codec_name === 'av1') && file.container === inputs.container) {
response.infoLog += `☑ File is already HEVC, VP9 or AV1 & in ${inputs.container}. \n`;
return response;
}
// Check if codec of stream is HEVC, Vp9 or AV1
// AND check if file.container does NOT match inputs.container.
// If so remux file.
if ((file.ffProbeData.streams[i].codec_name === 'hevc' || file.ffProbeData.streams[i].codec_name === 'vp9'
|| file.ffProbeData.streams[i].codec_name === 'av1') && file.container !== inputs.container) {
response.infoLog += `☒ File is HEVC, VP9 or AV1 but is not in ${inputs.container} container. Remuxing. \n`;
response.preset = `<io> -map 0 -c copy ${extraArguments}`;
response.processFile = true;
return response;
}
// New logic for reprocessing HEVC. Mainly done for my own use.
// We attempt to get accurate stats earlier - If we can't we fall back onto overall bitrate
// which can be inaccurate. We may inflate the current bitrate check so we don't keep looping this logic.
} else if (inputs.reconvert_hevc === true && (file.ffProbeData.streams[i].codec_name === 'hevc'
|| file.ffProbeData.streams[i].codec_name === 'vp9' || file.ffProbeData.streams[i].codec_name === 'av1')) {
if (inputs.hevc_max_bitrate > 0) {
if (currentBitrate > inputs.hevc_max_bitrate) {
// If bitrate is higher then hevc_max_bitrate then need to re-encode
response.infoLog += `Reconvert_hevc is ${inputs.reconvert_hevc} & the file is already HEVC, `
+ `VP9 or AV1. Using HEVC specific cutoff of ${inputs.hevc_max_bitrate}kbps. \n`;
response.infoLog += '☒ The file is still above this new cutoff! Reconverting. \n';
} else {
// Otherwise we're now below the hevc cutoff and we can exit
response.infoLog += `Reconvert_hevc is ${inputs.reconvert_hevc} & the file is already HEVC, `
+ `VP9 or AV1. Using HEVC specific cutoff of ${inputs.hevc_max_bitrate}kbps. \n`;
response.infoLog += '☑ The file is NOT above this new cutoff. Exiting plugin. \n';
return response;
}
// If we're not using the hevc max bitrate then we need a safety net to try and ensure we don't keep
// looping this plugin. For maximum safety we simply multiply the cutoff by 2.
} else if (currentBitrate > (inputs.bitrate_cutoff * 2)) {
inflatedCutoff = Math.round(inputs.bitrate_cutoff * 2);
response.infoLog += `Reconvert_hevc is ${inputs.reconvert_hevc} & the file is already HEVC, `;
response.infoLog += 'VP9 or AV1. Will use Overall file Bitrate for HEVC files as safety, ';
response.infoLog += `bitrate is ${overallBitRate}kbps. \n`;
response.infoLog += 'HEVC specific cutoff not set so bitrate_cutoff is multiplied by 2 for safety! \n';
response.infoLog += `Cutoff now temporarily ${inflatedCutoff}kbps. \n`;
response.infoLog += '☒ The file is still above this new cutoff! Reconverting. \n';
} else {
// File is below cutoff so we can exit
inflatedCutoff = Math.round(inputs.bitrate_cutoff * 2);
response.infoLog += `Reconvert_hevc is ${inputs.reconvert_hevc} & the file is already HEVC, `;
response.infoLog += 'VP9 or AV1. Will use Overall file Bitrate for HEVC files as safety, ';
response.infoLog += `bitrate is ${overallBitRate}kbps. \n`;
response.infoLog += 'HEVC specific cutoff not set so bitrate_cutoff is multiplied by 2 for safety! \n';
response.infoLog += `Cutoff now temporarily ${inflatedCutoff}kbps. \n`;
response.infoLog += '☑The file is NOT above this new cutoff. Exiting plugin. \n';
return response;
}
}
// On testing I've found files in the High10 profile don't play nice with hw decoding so mark these
if (file.ffProbeData.streams[i].profile === 'High 10') {
high10 = true;
response.infoLog += 'Input file is 10bit using High10. Disabling hardware decoding to avoid problems. \n';
}
// If files are 10 bit or the enable_10bit setting is used mark to enable Main10.
if (file.ffProbeData.streams[i].profile === 'Main 10' || file.ffProbeData.streams[i].bits_per_raw_sample === '10'
|| inputs.enable_10bit === true) {
main10 = true;
}
// Increment video index. Needed to keep track of video id in case there is more than one video track.
// (i.e png or mjpeg which we would remove at the start of the loop)
videoIdx += 1;
}
}
// Specify the output format
switch (inputs.container) {
case 'mkv':
extraArguments += '-f matroska ';
break;
case 'mp4':
extraArguments += '-f mp4 ';
break;
default:
}
// Some video codecs won't play nice with 10 bit via QSV, whitelist safe ones here
switch (file.video_codec_name) {
case 'mpeg2':
break;
case 'h264':
break;
case 'mjpeg':
break;
case 'hevc':
break;
case 'vp9':
break;
case 'av1':
break;
default:
oldFormat = true;
}
// Are we encoding to 10 bit? If so enable correct profile & pixel format.
if (high10 === true || (oldFormat === true && main10 === true)) {
// This is used if we have High10 or Main10 is enabled & odd format files.
// SW decode and use standard -pix_fmt p010le
extraArguments += '-profile:v main10 -pix_fmt p010le ';
response.infoLog += '10 bit encode enabled. Setting Main10 Profile & 10 bit pixel format \n';
} else if (main10 === true) { // Pixel formate method when using HW decode
if (inputs.extra_qsv_options.search('-vf scale_qsv') >= 0) {
extraArguments += '-profile:v main10';
// eslint-disable-next-line no-param-reassign
inputs.extra_qsv_options += ':format=p010le'; // Only add on the pixel format to existing scale_qsv cmd
} else {
extraArguments += '-profile:v main10 -vf scale_qsv=format=p010le';
}
response.infoLog += '10 bit encode enabled. Setting Main10 Profile & 10 bit pixel format \n';
}
// Set bitrateSettings variable using bitrate information calculated earlier.
bitrateSettings = `-b:v ${targetBitrate}k -minrate ${minimumBitrate}k `
+ `-maxrate ${maximumBitrate}k -bufsize ${currentBitrate}k`;
// Print to infoLog information around file & bitrate settings.
response.infoLog += `Container for output selected as ${inputs.container}. \n`;
response.infoLog += 'Encode variable bitrate settings: \n';
response.infoLog += `Target = ${targetBitrate}k \n`;
response.infoLog += `Minimum = ${minimumBitrate}k \n`;
response.infoLog += `Maximum = ${maximumBitrate}k \n`;
// START PRESET
// -fflags +genpts should regenerate timestamps if they end up missing...
response.preset = '-fflags +genpts ';
// HW ACCEL FLAGS - I think these are good practice but are they necessary?
// Account for different OS
if (high10 === false) {
// Seems incoming High10 files don't play nice decoding so use software decode
switch (os.platform()) {
case 'darwin': // Mac OS - Enable videotoolbox instead of QSV
response.preset += '-hwaccel videotoolbox';
break;
case 'linux': // Linux - Full device, should fix child_device_type warnings
response.preset += `-hwaccel qsv -hwaccel_output_format qsv
-init_hw_device qsv:hw_any,child_device_type=vaapi `;
break;
case 'win32': // Windows - Full device, should fix child_device_type warnings
response.preset += `-hwaccel qsv -hwaccel_output_format qsv
-init_hw_device qsv:hw,child_device_type=d3d11va `;
break;
default:
response.preset += '-hwaccel qsv -hwaccel_output_format qsv -init_hw_device qsv:hw_any ';
}
}
// DECODE FLAGS
if (os.platform() !== 'darwin') {
if (high10 === false) { // Don't enable for High10
switch (file.video_codec_name) {
case 'mpeg2':
response.preset += '-c:v mpeg2_qsv';
break;
case 'h264':
response.preset += '-c:v h264_qsv';
break;
case 'vc1': // VC1 no longer supported on latest intel HW
// response.preset += '-c:v vc1_qsv';
break;
case 'mjpeg':
response.preset += '-c:v mjpeg_qsv';
break;
case 'vp8': // VP8 no longer supported on latest intel HW
// response.preset += '-c:v vp8_qsv';
break;
case 'hevc':
response.preset += '-c:v hevc_qsv';
break;
case 'vp9': // Should be supported by 8th Gen +
response.preset += '-c:v vp9_qsv';
break;
case 'av1': // Should be supported by 11th gen +
response.preset += '-c:v av1_qsv';
break;
default:
}
}
}
// ENCODE FLAGS
response.preset += '<io> -map 0 -c:v ';
// Account for different OS setup for QSV HEVC encode.
switch (os.platform()) {
case 'darwin':
response.preset += 'hevc_videotoolbox';
// Mac OS & uses hevc_videotoolbox not QSV - Only shows up on Mac installs
break;
case 'linux':
response.preset += 'hevc_qsv';
break;
case 'win32':
response.preset += 'hevc_qsv';
// Tested working on a Win 10 - i5-10505
break;
default:
response.preset += 'hevc_qsv'; // Default to QSV
}
// Check if -vf cmd has already been used on user input
if (high10 !== true) {
if (inputs.extra_qsv_options.search('-vf scale_qsv') >= 0) {
switch (file.video_codec_name) {
case 'mpeg2':
// eslint-disable-next-line no-param-reassign
inputs.extra_qsv_options += ',hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'h264':
// eslint-disable-next-line no-param-reassign
inputs.extra_qsv_options += ',hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'mjpeg':
// eslint-disable-next-line no-param-reassign
inputs.extra_qsv_options += ',hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'hevc':
// eslint-disable-next-line no-param-reassign
inputs.extra_qsv_options += ',hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'vp9': // Should be supported by 8th Gen +
// eslint-disable-next-line no-param-reassign
inputs.extra_qsv_options += ',hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'av1': // Should be supported by 11th gen +
// eslint-disable-next-line no-param-reassign
inputs.extra_qsv_options += ',hwupload=extra_hw_frames=64,format=qsv ';
break;
default:
}
} else if (extraArguments.search('-vf') === -1) {
// Check if -vf cmd has been used on the other var instead, if not add it & rest of cmd
switch (file.video_codec_name) {
case 'mpeg2':
extraArguments += '-vf hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'h264':
extraArguments += '-vf hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'mjpeg':
extraArguments += '-vf hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'hevc':
extraArguments += '-vf hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'vp9': // Should be supported by 8th Gen +
extraArguments += '-vf hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'av1': // Should be supported by 11th gen +
extraArguments += '-vf hwupload=extra_hw_frames=64,format=qsv ';
break;
default:
}
} else {
// Otherwise add the cmd onto the end
switch (file.video_codec_name) {
case 'mpeg2':
extraArguments += ',hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'h264':
extraArguments += ',hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'mjpeg':
extraArguments += ',hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'hevc':
extraArguments += ',hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'vp9': // Should be supported by 8th Gen +
extraArguments += ',hwupload=extra_hw_frames=64,format=qsv ';
break;
case 'av1': // Should be supported by 11th gen +
extraArguments += ',hwupload=extra_hw_frames=64,format=qsv ';
break;
default:
}
}
}
// Add the rest of the ffmpeg command
switch (os.platform()) {
case 'darwin':
// Mac OS - Don't use extra_qsv_options - These are intended for QSV cmds so videotoolbox causes issues
response.preset += ` ${bitrateSettings} `
+ `-preset ${inputs.encoder_speedpreset} -c:a copy -c:s copy -max_muxing_queue_size 9999 ${extraArguments}`;
response.infoLog += '==ALERT== OS detected as MAC - This will use VIDEOTOOLBOX to encode which is NOT QSV\n'
+ 'cmds set in extra_qsv_options will be IGNORED!\n';
break;
default:
// Normal behavior
response.preset += ` ${bitrateSettings} `
+ `-preset ${inputs.encoder_speedpreset} ${inputs.extra_qsv_options} `
+ `-c:a copy -c:s copy -max_muxing_queue_size 9999 ${extraArguments}`;
}
response.processFile = true;
response.infoLog += 'File Transcoding... \n';
return response;
};
module.exports.details = details;
module.exports.plugin = plugin;
Reference in new issue View Git Blame Copy Permalink