Option to use perc bitrate

HaveAGitGat · May 7, 2024 · 7e22f3d · 7e22f3d
1 parent 09a7edf
commit 7e22f3d
Showing 1 changed file with 104 additions and 3 deletions.
diff --git a/FlowPluginsTs/CommunityFlowPlugins/ffmpegCommand/ffmpegCommandSetVideoBitrate/1.0.0/index.ts b/FlowPluginsTs/CommunityFlowPlugins/ffmpegCommand/ffmpegCommandSetVideoBitrate/1.0.0/index.ts
@@ -6,7 +6,7 @@ import {
 import { getFfType } from '../../../../FlowHelpers/1.0.0/fileUtils';
 
 /* eslint no-plusplus: ["error", { "allowForLoopAfterthoughts": true }] */
-const details = () :IpluginDetails => ({
+const details = (): IpluginDetails => ({
   name: 'Set Video Bitrate',
   description: 'Set Video Bitrate',
   style: {
@@ -19,13 +19,89 @@ const details = () :IpluginDetails => ({
   sidebarPosition: -1,
   icon: '',
   inputs: [
+    {
+      label: 'Use % of Input Bitrate',
+      name: 'useInputBitrate',
+      type: 'boolean',
+      defaultValue: 'false',
+      inputUI: {
+        type: 'switch',
+      },
+      tooltip: 'Specify whether to use a % of input bitrate as the output bitrate',
+    },
+
+    {
+      label: 'Target Bitrate %',
+      name: 'targetBitratePercent',
+      type: 'string',
+      defaultValue: '50',
+      inputUI: {
+        type: 'text',
+        displayConditions: {
+          logic: 'AND',
+          sets: [
+            {
+              logic: 'AND',
+              inputs: [
+                {
+                  name: 'useInputBitrate',
+                  value: 'true',
+                  condition: '===',
+                },
+              ],
+            },
+          ],
+        },
+      },
+      tooltip: 'Specify the target bitrate as a % of the input bitrate',
+    },
+    {
+      label: 'Fallback Bitrate',
+      name: 'fallbackBitrate',
+      type: 'string',
+      defaultValue: '4000',
+      inputUI: {
+        type: 'text',
+        displayConditions: {
+          logic: 'AND',
+          sets: [
+            {
+              logic: 'AND',
+              inputs: [
+                {
+                  name: 'useInputBitrate',
+                  value: 'true',
+                  condition: '===',
+                },
+              ],
+            },
+          ],
+        },
+      },
+      tooltip: 'Specify fallback bitrate in kbps if input bitrate is not available',
+    },
     {
       label: 'Bitrate',
       name: 'bitrate',
       type: 'string',
       defaultValue: '5000',
       inputUI: {
         type: 'text',
+        displayConditions: {
+          logic: 'AND',
+          sets: [
+            {
+              logic: 'AND',
+              inputs: [
+                {
+                  name: 'useInputBitrate',
+                  value: 'true',
+                  condition: '!==',
+                },
+              ],
+            },
+          ],
+        },
       },
       tooltip: 'Specify bitrate in kbps',
     },
@@ -39,15 +115,40 @@ const details = () :IpluginDetails => ({
 });
 
 // eslint-disable-next-line @typescript-eslint/no-unused-vars
-const plugin = (args:IpluginInputArgs):IpluginOutputArgs => {
+const plugin = (args: IpluginInputArgs): IpluginOutputArgs => {
   const lib = require('../../../../../methods/lib')();
   // eslint-disable-next-line @typescript-eslint/no-unused-vars,no-param-reassign
   args.inputs = lib.loadDefaultValues(args.inputs, details);
 
+  const { useInputBitrate } = args.inputs;
+  const targetBitratePercent = String(args.inputs.targetBitratePercent);
+  const fallbackBitrate = String(args.inputs.fallbackBitrate);
+  const bitrate = String(args.inputs.bitrate);
+
   args.variables.ffmpegCommand.streams.forEach((stream) => {
     if (stream.codec_type === 'video') {
       const ffType = getFfType(stream.codec_type);
-      stream.outputArgs.push(`-b:${ffType}:{outputTypeIndex}`, `${String(args.inputs.bitrate)}k`);
+      if (useInputBitrate) {
+        args.jobLog('Attempting to use % of input bitrate as output bitrate');
+        // check if input bitrate is available
+        const mediainfoIndex = stream.index + 1;
+
+        let inputBitrate = args?.inputFileObj?.mediaInfo?.track?.[mediainfoIndex]?.BitRate;
+        if (inputBitrate) {
+          args.jobLog(`Found input bitrate: ${inputBitrate}`);
+          // @ts-expect-error type
+          inputBitrate = parseInt(inputBitrate, 10) / 1000;
+          const targetBitrate = (inputBitrate * (parseInt(targetBitratePercent, 10) / 100));
+          args.jobLog(`Setting video bitrate as ${targetBitrate}k`);
+          stream.outputArgs.push(`-b:${ffType}:{outputTypeIndex}`, `${targetBitrate}k`);
+        } else {
+          args.jobLog(`Unable to find input bitrate, setting fallback bitrate as ${fallbackBitrate}k`);
+          stream.outputArgs.push(`-b:${ffType}:{outputTypeIndex}`, `${fallbackBitrate}k`);
+        }
+      } else {
+        args.jobLog(`Using fixed bitrate. Setting video bitrate as ${bitrate}k`);
+        stream.outputArgs.push(`-b:${ffType}:{outputTypeIndex}`, `${bitrate}k`);
+      }
     }
   });