Add downmix_single_track input and update tests

Community/Tdarr_Plugin_MC93_Migz5ConvertAudio.js

@@ -39,6 +39,19 @@ const details = () => ({
              \\nExample:\\n
              false`,
   },
+  {
+    name: 'downmix_single_track',
+    type: 'boolean',
+    defaultValue: false,
+    inputUI: {
+      type: 'dropdown',
+      options: [
+        'false',
+        'true',
+      ],
+    },
+    tooltip: '',
+  },
   ],
 });
 
@@ -77,8 +90,9 @@ const plugin = (file, librarySettings, inputs, otherArguments) => {
   let audioIdx = 0;
   let has2Channel = false;
   let has6Channel = false;
-  let has8Channel = false;
   let convert = false;
+  let is2channelAdded = false;
+  let is6channelAdded = false;
 
   // Go through each stream in the file.
   for (let i = 0; i < file.ffProbeData.streams.length; i++) {
@@ -91,9 +105,6 @@ const plugin = (file, librarySettings, inputs, otherArguments) => {
         if (file.ffProbeData.streams[i].channels === 6) {
           has6Channel = true;
         }
-        if (file.ffProbeData.streams[i].channels === 8) {
-          has8Channel = true;
-        }
       }
     } catch (err) {
       // Error
@@ -110,23 +121,28 @@ const plugin = (file, librarySettings, inputs, otherArguments) => {
         if (inputs.downmix.toLowerCase() === 'true') {
           // Check if file has 8 channel audio but no 6 channel, if so then create extra downmix from the 8 channel.
           if (
-            has8Channel === true
+            file.ffProbeData.streams[i].channels === 8
             && has6Channel === false
-            && file.ffProbeData.streams[i].channels === 8
+            && (inputs.downmix_single_track === false
+              || (inputs.downmix_single_track === true && is6channelAdded === false))
+
           ) {
             ffmpegCommandInsert += `-map 0:${i} -c:a:${audioIdx} ac3 -ac 6 -metadata:s:a:${audioIdx} title="5.1" `;
             response.infoLog += '☒Audio track is 8 channel, no 6 channel exists. Creating 6 channel from 8 channel. \n';
             convert = true;
+            is6channelAdded = true;
           }
           // Check if file has 6 channel audio but no 2 channel, if so then create extra downmix from the 6 channel.
           if (
-            has6Channel === true
+            file.ffProbeData.streams[i].channels === 6
             && has2Channel === false
-            && file.ffProbeData.streams[i].channels === 6
+            && (inputs.downmix_single_track === false
+              || (inputs.downmix_single_track === true && is2channelAdded === false))
           ) {
             ffmpegCommandInsert += `-map 0:${i} -c:a:${audioIdx} aac -ac 2 -metadata:s:a:${audioIdx} title="2.0" `;
             response.infoLog += '☒Audio track is 6 channel, no 2 channel exists. Creating 2 channel from 6 channel. \n';
             convert = true;
+            is2channelAdded = true;
           }
         }
       } catch (err) {

tests/Community/Tdarr_Plugin_MC93_Migz5ConvertAudio.js

@@ -85,6 +85,158 @@ const tests = [
     },
   },
 
+  {
+    input: {
+      file: (() => {
+        const file = _.cloneDeep(require('../sampleData/media/sampleH264_2.json'));
+        file.ffProbeData.streams[1].channels = 8;
+        file.ffProbeData.streams[2].channels = 8;
+        return file;
+      })(),
+      librarySettings: {},
+      inputs: {
+        aac_stereo: 'false',
+        downmix: 'true',
+      },
+      otherArguments: {},
+    },
+    output: {
+      processFile: true,
+      container: '.mkv',
+      handBrakeMode: false,
+      FFmpegMode: true,
+      reQueueAfter: true,
+      infoLog: '☒Audio track is 8 channel, no 6 channel exists. Creating 6 channel from 8 channel. \n'
+        + '☒Audio track is 8 channel, no 6 channel exists. Creating 6 channel from 8 channel. \n',
+      preset: ', -map 0 -c:v copy -c:a copy -map 0:1 -c:a:0 ac3 -ac 6 -metadata:s:a:0 title="5.1" -map 0:2 -c:a:1 ac3 -ac 6 -metadata:s:a:1 title="5.1"  -strict -2 -c:s copy -max_muxing_queue_size 9999 ',
+    },
+  },
+
+  {
+    input: {
+      file: (() => {
+        const file = _.cloneDeep(require('../sampleData/media/sampleH264_2.json'));
+        file.ffProbeData.streams[1].channels = 8;
+        file.ffProbeData.streams[2].channels = 8;
+        file.ffProbeData.streams[3].channels = 6;
+        file.ffProbeData.streams[4].channels = 6;
+        file.ffProbeData.streams[5].channels = 6;
+        return file;
+      })(),
+      librarySettings: {},
+      inputs: {
+        aac_stereo: 'false',
+        downmix: 'true',
+      },
+      otherArguments: {},
+    },
+    output: {
+      processFile: true,
+      container: '.mkv',
+      handBrakeMode: false,
+      FFmpegMode: true,
+      reQueueAfter: true,
+      infoLog: '☒Audio track is 6 channel, no 2 channel exists. Creating 2 channel from 6 channel. \n'
+      + '☒Audio track is 6 channel, no 2 channel exists. Creating 2 channel from 6 channel. \n'
+      + '☒Audio track is 6 channel, no 2 channel exists. Creating 2 channel from 6 channel. \n',
+      preset: ', -map 0 -c:v copy -c:a copy -map 0:3 -c:a:2 aac -ac 2 -metadata:s:a:2 title="2.0" -map 0:4 -c:a:3 aac -ac 2 -metadata:s:a:3 title="2.0" -map 0:5 -c:a:4 aac -ac 2 -metadata:s:a:4 title="2.0"  -strict -2 -c:s copy -max_muxing_queue_size 9999 ',
+    },
+  },
+
+  {
+    input: {
+      file: (() => {
+        const file = _.cloneDeep(require('../sampleData/media/sampleH264_2.json'));
+        file.ffProbeData.streams[1].channels = 8;
+        file.ffProbeData.streams[2].channels = 8;
+        file.ffProbeData.streams[3].channels = 8;
+        file.ffProbeData.streams[4].channels = 6;
+        file.ffProbeData.streams[5].channels = 6;
+        return file;
+      })(),
+      librarySettings: {},
+      inputs: {
+        aac_stereo: 'false',
+        downmix: 'true',
+        downmix_single_track: 'true',
+      },
+      otherArguments: {},
+    },
+    output: {
+      processFile: true,
+      container: '.mkv',
+      handBrakeMode: false,
+      FFmpegMode: true,
+      reQueueAfter: true,
+      infoLog: '☒Audio track is 6 channel, no 2 channel exists. Creating 2 channel from 6 channel. \n',
+      preset: ', -map 0 -c:v copy -c:a copy -map 0:4 -c:a:3 aac -ac 2 -metadata:s:a:3 title="2.0"  -strict -2 -c:s copy -max_muxing_queue_size 9999 ',
+    },
+  },
+
+  {
+    input: {
+      file: (() => {
+        const file = _.cloneDeep(require('../sampleData/media/sampleH264_2.json'));
+        file.ffProbeData.streams[1].channels = 8;
+        file.ffProbeData.streams[2].channels = 8;
+        file.ffProbeData.streams[3].channels = 8;
+        file.ffProbeData.streams[4].channels = 8;
+        file.ffProbeData.streams[5].channels = 8;
+        return file;
+      })(),
+      librarySettings: {},
+      inputs: {
+        aac_stereo: 'false',
+        downmix: 'true',
+        downmix_single_track: 'false',
+      },
+      otherArguments: {},
+    },
+    output: {
+      processFile: true,
+      container: '.mkv',
+      handBrakeMode: false,
+      FFmpegMode: true,
+      reQueueAfter: true,
+      infoLog: '☒Audio track is 8 channel, no 6 channel exists. Creating 6 channel from 8 channel. \n'
+        + '☒Audio track is 8 channel, no 6 channel exists. Creating 6 channel from 8 channel. \n'
+        + '☒Audio track is 8 channel, no 6 channel exists. Creating 6 channel from 8 channel. \n'
+        + '☒Audio track is 8 channel, no 6 channel exists. Creating 6 channel from 8 channel. \n'
+        + '☒Audio track is 8 channel, no 6 channel exists. Creating 6 channel from 8 channel. \n',
+      preset: ', -map 0 -c:v copy -c:a copy -map 0:1 -c:a:0 ac3 -ac 6 -metadata:s:a:0 title="5.1" -map 0:2 -c:a:1 ac3 -ac 6 -metadata:s:a:1 title="5.1" -map 0:3 -c:a:2 ac3 -ac 6 -metadata:s:a:2 title="5.1" -map 0:4 -c:a:3 ac3 -ac 6 -metadata:s:a:3 title="5.1" -map 0:5 -c:a:4 ac3 -ac 6 -metadata:s:a:4 title="5.1"  -strict -2 -c:s copy -max_muxing_queue_size 9999 ',
+
+    },
+  },
+
+  {
+    input: {
+      file: (() => {
+        const file = _.cloneDeep(require('../sampleData/media/sampleH264_2.json'));
+        file.ffProbeData.streams[1].channels = 8;
+        file.ffProbeData.streams[2].channels = 8;
+        file.ffProbeData.streams[3].channels = 8;
+        file.ffProbeData.streams[4].channels = 8;
+        file.ffProbeData.streams[5].channels = 8;
+        return file;
+      })(),
+      librarySettings: {},
+      inputs: {
+        aac_stereo: 'false',
+        downmix: 'true',
+        downmix_single_track: 'true',
+      },
+      otherArguments: {},
+    },
+    output: {
+      processFile: true,
+      container: '.mkv',
+      handBrakeMode: false,
+      FFmpegMode: true,
+      reQueueAfter: true,
+      infoLog: '☒Audio track is 8 channel, no 6 channel exists. Creating 6 channel from 8 channel. \n',
+      preset: ', -map 0 -c:v copy -c:a copy -map 0:1 -c:a:0 ac3 -ac 6 -metadata:s:a:0 title="5.1"  -strict -2 -c:s copy -max_muxing_queue_size 9999 ',
+    },
+  },
 ];
 
 run(tests);