gabehf/Tdarr_Plugins
1
0
Fork 0
mirror of https://github.com/gabehf/Tdarr_Plugins.git synced 2026-03-09 15:38:19 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add bitrate option

Browse source
This commit is contained in:
HaveAGitGat 2023-11-28 07:01:12 +00:00
parent 3d717fd866
commit ef88971bcf
2 changed files with 108 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

52
FlowPlugins/CommunityFlowPlugins/ffmpegCommand/ffmpegCommandEnsureAudioStream/1.0.0/index.js
View file

@ -1,6 +1,7 @@
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.plugin = exports.details = void 0;
var fileUtils_1 = require("../../../../FlowHelpers/1.0.0/fileUtils");
/* eslint no-plusplus: ["error", { "allowForLoopAfterthoughts": true }] */
var details = function () { return ({
name: 'Ensure Audio Stream',
@ -63,6 +64,45 @@ var details = function () { return ({
},
tooltip: 'Enter the desired number of channels',
},
{
label: 'Enable Bitrate',
name: 'enableBitrate',
type: 'boolean',
defaultValue: 'false',
inputUI: {
type: 'dropdown',
options: [
'false',
'true',
],
},
tooltip: 'Toggle whether to enable setting audio bitrate',
},
{
label: 'Bitrate',
name: 'bitrate',
type: 'string',
defaultValue: '128k',
inputUI: {
type: 'text',
displayConditions: {
logic: 'AND',
sets: [
{
logic: 'AND',
inputs: [
{
name: 'enableBitrate',
value: 'true',
condition: '===',
},
],
},
],
},
},
tooltip: 'Specify the audio bitrate for newly added channels',
},
],
outputs: [
{
@ -80,7 +120,7 @@ var getHighest = function (first, second) {
return second;
};
var attemptMakeStream = function (_a) {
var args = _a.args, langTag = _a.langTag, streams = _a.streams, audioCodec = _a.audioCodec, audioEncoder = _a.audioEncoder, wantedChannelCount = _a.wantedChannelCount;
var args = _a.args, langTag = _a.langTag, streams = _a.streams, audioCodec = _a.audioCodec, audioEncoder = _a.audioEncoder, wantedChannelCount = _a.wantedChannelCount, enableBitrate = _a.enableBitrate, bitrate = _a.bitrate;
var langMatch = function (stream) {
var _a;
return ((langTag === 'und'
@ -131,6 +171,10 @@ var attemptMakeStream = function (_a) {
streamCopy.index = streams.length;
streamCopy.outputArgs.push('-c:{outputIndex}', audioEncoder);
streamCopy.outputArgs.push('-ac', "".concat(targetChannels));
if (enableBitrate) {
var ffType = (0, fileUtils_1.getFfType)(streamCopy.codec_type);
streamCopy.outputArgs.push("-b:".concat(ffType, ":{outputTypeIndex}"), "".concat(bitrate));
}
// eslint-disable-next-line no-param-reassign
args.variables.ffmpegCommand.shouldProcess = true;
streams.push(streamCopy);
@ -144,6 +188,8 @@ var plugin = function (args) {
var audioEncoder = String(args.inputs.audioEncoder);
var langTag = String(args.inputs.language).toLowerCase();
var wantedChannelCount = Number(args.inputs.channels);
var enableBitrate = Boolean(args.inputs.enableBitrate);
var bitrate = String(args.inputs.bitrate);
var streams = args.variables.ffmpegCommand.streams;
var audioCodec = audioEncoder;
if (audioEncoder === 'dca') {
@ -162,6 +208,8 @@ var plugin = function (args) {
audioCodec: audioCodec,
audioEncoder: audioEncoder,
wantedChannelCount: wantedChannelCount,
enableBitrate: enableBitrate,
bitrate: bitrate,
});
if (!addedOrExists) {
attemptMakeStream({
@ -171,6 +219,8 @@ var plugin = function (args) {
audioCodec: audioCodec,
audioEncoder: audioEncoder,
wantedChannelCount: wantedChannelCount,
enableBitrate: enableBitrate,
bitrate: bitrate,
});
}
return {

57
FlowPluginsTs/CommunityFlowPlugins/ffmpegCommand/ffmpegCommandEnsureAudioStream/1.0.0/index.ts
View file

@ -1,3 +1,4 @@
import { getFfType } from '../../../../FlowHelpers/1.0.0/fileUtils';
import {
IffmpegCommandStream,
IpluginDetails,
@ -70,6 +71,47 @@ const details = (): IpluginDetails => ({
tooltip:
'Enter the desired number of channels',
},
{
label: 'Enable Bitrate',
name: 'enableBitrate',
type: 'boolean',
defaultValue: 'false',
inputUI: {
type: 'dropdown',
options: [
'false',
'true',
],
},
tooltip:
'Toggle whether to enable setting audio bitrate',
},
{
label: 'Bitrate',
name: 'bitrate',
type: 'string',
defaultValue: '128k',
inputUI: {
type: 'text',
displayConditions: {
logic: 'AND',
sets: [
{
logic: 'AND',
inputs: [
{
name: 'enableBitrate',
value: 'true',
condition: '===',
},
],
},
],
},
},
tooltip:
'Specify the audio bitrate for newly added channels',
},
],
outputs: [
{
@ -94,6 +136,8 @@ const attemptMakeStream = ({
audioCodec,
audioEncoder,
wantedChannelCount,
enableBitrate,
bitrate,
}: {
args: IpluginInputArgs,
langTag: string
@ -101,6 +145,8 @@ const attemptMakeStream = ({
audioCodec: string,
audioEncoder: string,
wantedChannelCount: number,
enableBitrate: boolean,
bitrate: string,
}): boolean => {
const langMatch = (stream: IffmpegCommandStream) => (
(langTag === 'und'
@ -166,6 +212,11 @@ const attemptMakeStream = ({
streamCopy.outputArgs.push('-c:{outputIndex}', audioEncoder);
streamCopy.outputArgs.push('-ac', `${targetChannels}`);
if (enableBitrate) {
const ffType = getFfType(streamCopy.codec_type);
streamCopy.outputArgs.push(`-b:${ffType}:{outputTypeIndex}`, `${bitrate}`);
}
// eslint-disable-next-line no-param-reassign
args.variables.ffmpegCommand.shouldProcess = true;
@ -183,6 +234,8 @@ const plugin = (args: IpluginInputArgs): IpluginOutputArgs => {
const audioEncoder = String(args.inputs.audioEncoder);
const langTag = String(args.inputs.language).toLowerCase();
const wantedChannelCount = Number(args.inputs.channels);
const enableBitrate = Boolean(args.inputs.enableBitrate);
const bitrate = String(args.inputs.bitrate);
const { streams } = args.variables.ffmpegCommand;
@ -207,6 +260,8 @@ const plugin = (args: IpluginInputArgs): IpluginOutputArgs => {
audioCodec,
audioEncoder,
wantedChannelCount,
enableBitrate,
bitrate,
});
if (!addedOrExists) {
@ -217,6 +272,8 @@ const plugin = (args: IpluginInputArgs): IpluginOutputArgs => {
audioCodec,
audioEncoder,
wantedChannelCount,
enableBitrate,
bitrate,
});
}