Parse and set peakBitrate for Dolby TrueHD(AC-3) and (E-)AC-3

#minor-release PiperOrigin-RevId: 490527831
androidx · Nov 24, 2022 · 76df06a · 76df06a
1 parent 8092726
commit 76df06a
Show file tree

Hide file tree

Showing 31 changed files with 49 additions and 1 deletion.
diff --git a/libraries/extractor/src/main/java/androidx/media3/extractor/Ac3Util.java b/libraries/extractor/src/main/java/androidx/media3/extractor/Ac3Util.java
@@ -158,13 +158,18 @@ public static Format parseAc3AnnexFFormat(
     if ((nextByte & 0x04) != 0) { // lfeon
       channelCount++;
     }
+    // bit_rate_code - 5 bits. 2 bits from previous byte and 3 bits from next.
+    int halfFrmsizecod = ((nextByte & 0x03) << 3) | ((data.readUnsignedByte() & 0xE0) >> 5);
+    int constantBitrate = BITRATE_BY_HALF_FRMSIZECOD[halfFrmsizecod];
     return new Format.Builder()
         .setId(trackId)
         .setSampleMimeType(MimeTypes.AUDIO_AC3)
         .setChannelCount(channelCount)
         .setSampleRate(sampleRate)
         .setDrmInitData(drmInitData)
         .setLanguage(language)
+        .setAverageBitrate(constantBitrate)
+        .setPeakBitrate(constantBitrate)
         .build();
   }
 
@@ -180,7 +185,9 @@ public static Format parseAc3AnnexFFormat(
    */
   public static Format parseEAc3AnnexFFormat(
       ParsableByteArray data, String trackId, String language, @Nullable DrmInitData drmInitData) {
-    data.skipBytes(2); // data_rate, num_ind_sub
+    // 13 bits for data_rate, 3 bits for num_ind_sub which are ignored.
+    int peakBitrate =
+        ((data.readUnsignedByte() & 0xFF) << 5) | ((data.readUnsignedByte() & 0xF8) >> 3);
 
     // Read the first independent substream.
     int fscod = (data.readUnsignedByte() & 0xC0) >> 6;
@@ -216,6 +223,7 @@ public static Format parseEAc3AnnexFFormat(
         .setSampleRate(sampleRate)
         .setDrmInitData(drmInitData)
         .setLanguage(language)
+        .setPeakBitrate(peakBitrate)
         .build();
   }
 

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3.mp4.0.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3.mp4.0.dump
@@ -10,6 +10,8 @@ track 0:
   total output bytes = 13824
   sample count = 9
   format 0:
+    averageBitrate = 384
+    peakBitrate = 384
     id = 1
     sampleMimeType = audio/ac3
     maxInputSize = 1566

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3.mp4.1.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3.mp4.1.dump
@@ -10,6 +10,8 @@ track 0:
   total output bytes = 9216
   sample count = 6
   format 0:
+    averageBitrate = 384
+    peakBitrate = 384
     id = 1
     sampleMimeType = audio/ac3
     maxInputSize = 1566

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3.mp4.2.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3.mp4.2.dump
@@ -10,6 +10,8 @@ track 0:
   total output bytes = 4608
   sample count = 3
   format 0:
+    averageBitrate = 384
+    peakBitrate = 384
     id = 1
     sampleMimeType = audio/ac3
     maxInputSize = 1566

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3.mp4.3.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3.mp4.3.dump
@@ -10,6 +10,8 @@ track 0:
   total output bytes = 1536
   sample count = 1
   format 0:
+    averageBitrate = 384
+    peakBitrate = 384
     id = 1
     sampleMimeType = audio/ac3
     maxInputSize = 1566

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3.mp4.unknown_length.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3.mp4.unknown_length.dump
@@ -10,6 +10,8 @@ track 0:
   total output bytes = 13824
   sample count = 9
   format 0:
+    averageBitrate = 384
+    peakBitrate = 384
     id = 1
     sampleMimeType = audio/ac3
     maxInputSize = 1566

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3_fragmented.mp4.0.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3_fragmented.mp4.0.dump
@@ -10,6 +10,8 @@ track 0:
   total output bytes = 13824
   sample count = 9
   format 0:
+    averageBitrate = 384
+    peakBitrate = 384
     id = 1
     sampleMimeType = audio/ac3
     channelCount = 6

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3_fragmented.mp4.1.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3_fragmented.mp4.1.dump
@@ -10,6 +10,8 @@ track 0:
   total output bytes = 10752
   sample count = 7
   format 0:
+    averageBitrate = 384
+    peakBitrate = 384
     id = 1
     sampleMimeType = audio/ac3
     channelCount = 6

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3_fragmented.mp4.2.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3_fragmented.mp4.2.dump
@@ -10,6 +10,8 @@ track 0:
   total output bytes = 6144
   sample count = 4
   format 0:
+    averageBitrate = 384
+    peakBitrate = 384
     id = 1
     sampleMimeType = audio/ac3
     channelCount = 6

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3_fragmented.mp4.3.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_ac3_fragmented.mp4.3.dump
@@ -10,6 +10,8 @@ track 0:
   total output bytes = 1536
   sample count = 1
   format 0:
+    averageBitrate = 384
+    peakBitrate = 384
     id = 1
     sampleMimeType = audio/ac3
     channelCount = 6

diff --git a/...est_data/src/test/assets/extractordumps/mp4/sample_ac3_fragmented.mp4.unknown_length.dump b/...est_data/src/test/assets/extractordumps/mp4/sample_ac3_fragmented.mp4.unknown_length.dump
@@ -10,6 +10,8 @@ track 0:
   total output bytes = 13824
   sample count = 9
   format 0:
+    averageBitrate = 384
+    peakBitrate = 384
     id = 1
     sampleMimeType = audio/ac3
     channelCount = 6

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3.mp4.0.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3.mp4.0.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 216000
   sample count = 54
   format 0:
+    peakBitrate = 1000
     id = 1
     sampleMimeType = audio/eac3
     maxInputSize = 4030

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3.mp4.1.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3.mp4.1.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 144000
   sample count = 36
   format 0:
+    peakBitrate = 1000
     id = 1
     sampleMimeType = audio/eac3
     maxInputSize = 4030

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3.mp4.2.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3.mp4.2.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 72000
   sample count = 18
   format 0:
+    peakBitrate = 1000
     id = 1
     sampleMimeType = audio/eac3
     maxInputSize = 4030

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3.mp4.3.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3.mp4.3.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 4000
   sample count = 1
   format 0:
+    peakBitrate = 1000
     id = 1
     sampleMimeType = audio/eac3
     maxInputSize = 4030

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3.mp4.unknown_length.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3.mp4.unknown_length.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 216000
   sample count = 54
   format 0:
+    peakBitrate = 1000
     id = 1
     sampleMimeType = audio/eac3
     maxInputSize = 4030

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3_fragmented.mp4.0.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3_fragmented.mp4.0.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 216000
   sample count = 54
   format 0:
+    peakBitrate = 1000
     id = 1
     sampleMimeType = audio/eac3
     channelCount = 6

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3_fragmented.mp4.1.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3_fragmented.mp4.1.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 148000
   sample count = 37
   format 0:
+    peakBitrate = 1000
     id = 1
     sampleMimeType = audio/eac3
     channelCount = 6

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3_fragmented.mp4.2.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3_fragmented.mp4.2.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 76000
   sample count = 19
   format 0:
+    peakBitrate = 1000
     id = 1
     sampleMimeType = audio/eac3
     channelCount = 6

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3_fragmented.mp4.3.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3_fragmented.mp4.3.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 4000
   sample count = 1
   format 0:
+    peakBitrate = 1000
     id = 1
     sampleMimeType = audio/eac3
     channelCount = 6

diff --git a/...st_data/src/test/assets/extractordumps/mp4/sample_eac3_fragmented.mp4.unknown_length.dump b/...st_data/src/test/assets/extractordumps/mp4/sample_eac3_fragmented.mp4.unknown_length.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 216000
   sample count = 54
   format 0:
+    peakBitrate = 1000
     id = 1
     sampleMimeType = audio/eac3
     channelCount = 6

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc.mp4.0.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc.mp4.0.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 163840
   sample count = 64
   format 0:
+    peakBitrate = 640
     id = 1
     sampleMimeType = audio/eac3-joc
     maxInputSize = 2590

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc.mp4.1.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc.mp4.1.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 110080
   sample count = 43
   format 0:
+    peakBitrate = 640
     id = 1
     sampleMimeType = audio/eac3-joc
     maxInputSize = 2590

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc.mp4.2.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc.mp4.2.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 56320
   sample count = 22
   format 0:
+    peakBitrate = 640
     id = 1
     sampleMimeType = audio/eac3-joc
     maxInputSize = 2590

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc.mp4.3.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc.mp4.3.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 2560
   sample count = 1
   format 0:
+    peakBitrate = 640
     id = 1
     sampleMimeType = audio/eac3-joc
     maxInputSize = 2590

diff --git a/...aries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc.mp4.unknown_length.dump b/...aries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc.mp4.unknown_length.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 163840
   sample count = 64
   format 0:
+    peakBitrate = 640
     id = 1
     sampleMimeType = audio/eac3-joc
     maxInputSize = 2590

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc_fragmented.mp4.0.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc_fragmented.mp4.0.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 163840
   sample count = 64
   format 0:
+    peakBitrate = 640
     id = 1
     sampleMimeType = audio/eac3-joc
     channelCount = 6

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc_fragmented.mp4.1.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc_fragmented.mp4.1.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 110080
   sample count = 43
   format 0:
+    peakBitrate = 640
     id = 1
     sampleMimeType = audio/eac3-joc
     channelCount = 6

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc_fragmented.mp4.2.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc_fragmented.mp4.2.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 56320
   sample count = 22
   format 0:
+    peakBitrate = 640
     id = 1
     sampleMimeType = audio/eac3-joc
     channelCount = 6

diff --git a/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc_fragmented.mp4.3.dump b/libraries/test_data/src/test/assets/extractordumps/mp4/sample_eac3joc_fragmented.mp4.3.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 2560
   sample count = 1
   format 0:
+    peakBitrate = 640
     id = 1
     sampleMimeType = audio/eac3-joc
     channelCount = 6

diff --git a/...data/src/test/assets/extractordumps/mp4/sample_eac3joc_fragmented.mp4.unknown_length.dump b/...data/src/test/assets/extractordumps/mp4/sample_eac3joc_fragmented.mp4.unknown_length.dump
@@ -10,6 +10,7 @@ track 0:
   total output bytes = 163840
   sample count = 64
   format 0:
+    peakBitrate = 640
     id = 1
     sampleMimeType = audio/eac3-joc
     channelCount = 6