Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
import androidx.media3.common.util.Util;
import androidx.media3.extractor.MpegAudioUtil;
import androidx.media3.extractor.SeekPoint;
import java.math.RoundingMode;

/** MP3 seeker that uses metadata from a VBRI header. */
/* package */ final class VbriSeeker implements Seeker {
Expand Down Expand Up @@ -107,29 +108,34 @@ public static VbriSeeker create(
}

return new VbriSeeker(
timesUs, positions, durationUs, startOfMp3Data, endOfMp3Data, mpegAudioHeader.bitrate);
timesUs,
positions,
durationUs,
startOfMp3Data,
endOfMp3Data,
computeAverageBitrate(endOfMp3Data - startOfMp3Data, durationUs));
}

private final long[] timesUs;
private final long[] positions;
private final long durationUs;
private final long dataStartPosition;
private final long dataEndPosition;
private final int bitrate;
private final int averageBitrate;

private VbriSeeker(
long[] timesUs,
long[] positions,
long durationUs,
long dataStartPosition,
long dataEndPosition,
int bitrate) {
int averageBitrate) {
this.timesUs = timesUs;
this.positions = positions;
this.durationUs = durationUs;
this.dataStartPosition = dataStartPosition;
this.dataEndPosition = dataEndPosition;
this.bitrate = bitrate;
this.averageBitrate = averageBitrate;
}

@Override
Expand Down Expand Up @@ -171,6 +177,18 @@ public long getDataEndPosition() {

@Override
public int getAverageBitrate() {
return bitrate;
return averageBitrate;
}

private static int computeAverageBitrate(long dataSize, long durationUs) {
if (dataSize <= 0 || durationUs <= 0) {
return C.RATE_UNSET_INT;
}
long averageBitrate =
Util.scaleLargeValue(
dataSize, C.BITS_PER_BYTE * C.MICROS_PER_SECOND, durationUs, RoundingMode.HALF_UP);
return averageBitrate > 0 && averageBitrate <= Integer.MAX_VALUE
? (int) averageBitrate
: C.RATE_UNSET_INT;
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@
import androidx.media3.common.util.ParsableByteArray;
import androidx.media3.common.util.Util;
import androidx.media3.extractor.MpegAudioUtil;
import java.math.RoundingMode;

/** Representation of a LAME Xing or Info frame. */
/* package */ final class XingFrame {
Expand Down Expand Up @@ -155,6 +156,30 @@ public long computeDurationUs() {
(frameCount * header.samplesPerFrame) - 1, header.sampleRate);
}

/**
* Computes the average bitrate, in bits per second, represented by this frame and {@code
* dataSize}. Returns {@link C#RATE_UNSET_INT} if it can't be computed.
*
* @param dataSize The encoded stream size, including the Xing or Info frame.
*/
public int computeAverageBitrate(long dataSize) {
if (frameCount == C.LENGTH_UNSET
|| frameCount == 0
|| dataSize == C.LENGTH_UNSET
|| dataSize <= header.frameSize) {
return C.RATE_UNSET_INT;
}
long averageBitrate =
Util.scaleLargeValue(
dataSize - header.frameSize,
C.BITS_PER_BYTE * C.MICROS_PER_SECOND,
computeDurationUs(),
RoundingMode.HALF_UP);
return averageBitrate > 0 && averageBitrate <= Integer.MAX_VALUE
? (int) averageBitrate
: C.RATE_UNSET_INT;
}

/** Provide the metadata derived from this Xing frame, such as ReplayGain data. */
@Nullable
public Metadata getMetadata() {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -65,15 +65,15 @@ public static XingSeeker create(XingFrame xingFrame, long position, long streamL
position,
xingFrame.header.frameSize,
durationUs,
xingFrame.header.bitrate,
xingFrame.computeAverageBitrate(dataSize),
dataSize,
xingFrame.tableOfContents);
}

private final long dataStartPosition;
private final int xingFrameSize;
private final long durationUs;
private final int bitrate;
private final int averageBitrate;

/** Data size, including the XING frame. */
private final long dataSize;
Expand All @@ -90,13 +90,13 @@ private XingSeeker(
long dataStartPosition,
int xingFrameSize,
long durationUs,
int bitrate,
int averageBitrate,
long dataSize,
@Nullable long[] tableOfContents) {
this.dataStartPosition = dataStartPosition;
this.xingFrameSize = xingFrameSize;
this.durationUs = durationUs;
this.bitrate = bitrate;
this.averageBitrate = averageBitrate;
this.dataSize = dataSize;
this.tableOfContents = tableOfContents;
dataEndPosition = dataSize == C.LENGTH_UNSET ? C.INDEX_UNSET : dataStartPosition + dataSize;
Expand Down Expand Up @@ -173,7 +173,7 @@ public long getDataEndPosition() {

@Override
public int getAverageBitrate() {
return bitrate;
return averageBitrate;
}

/**
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,81 @@
/*
* Copyright 2026 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package androidx.media3.extractor.mp3;

import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.truth.Truth.assertThat;

import androidx.media3.common.C;
import androidx.media3.common.util.ParsableByteArray;
import androidx.media3.common.util.Util;
import androidx.media3.extractor.MpegAudioUtil;
import androidx.test.ext.junit.runners.AndroidJUnit4;
import java.math.RoundingMode;
import java.nio.ByteBuffer;
import org.junit.Test;
import org.junit.runner.RunWith;

/** Tests for {@link VbriSeeker}. */
@RunWith(AndroidJUnit4.class)
public final class VbriSeekerTest {

private static final int VBRI_FRAME_HEADER_DATA = 0xFFFB3000;
private static final int VBRI_FRAME_POSITION = 157;

@Test
public void getAverageBitrate_returnsAverageFromDataSizeAndDuration() {
MpegAudioUtil.Header header = new MpegAudioUtil.Header();
header.setForHeaderData(VBRI_FRAME_HEADER_DATA);
int dataSize = 1_000;
int frameCount = 40;
VbriSeeker seeker =
checkNotNull(
VbriSeeker.create(
/* inputLength= */ C.LENGTH_UNSET,
VBRI_FRAME_POSITION,
header,
createVbriFrame(dataSize, frameCount, /* segmentSizes= */ 400, 600)));
long durationUs =
Util.sampleCountToDurationUs(
((long) frameCount * header.samplesPerFrame) - 1, header.sampleRate);
int expectedAverageBitrate =
(int)
Util.scaleLargeValue(
dataSize,
C.BITS_PER_BYTE * C.MICROS_PER_SECOND,
durationUs,
RoundingMode.HALF_UP);

assertThat(seeker.getAverageBitrate()).isEqualTo(expectedAverageBitrate);
assertThat(seeker.getAverageBitrate()).isNotEqualTo(header.bitrate);
}

private static ParsableByteArray createVbriFrame(
int dataSize, int frameCount, int... segmentSizes) {
ByteBuffer payload = ByteBuffer.allocate(6 + 4 + 4 + 2 + 2 + 2 + 2 + 2 * segmentSizes.length);
payload.position(6);
payload.putInt(dataSize);
payload.putInt(frameCount);
payload.putShort((short) segmentSizes.length);
payload.putShort((short) 1); // scale
payload.putShort((short) 2); // entry size
payload.putShort((short) 0);
for (int segmentSize : segmentSizes) {
payload.putShort((short) segmentSize);
}
return new ParsableByteArray(payload.array());
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
import androidx.media3.extractor.SeekMap.SeekPoints;
import androidx.media3.extractor.SeekPoint;
import androidx.test.ext.junit.runners.AndroidJUnit4;
import java.math.RoundingMode;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
Expand Down Expand Up @@ -86,6 +87,20 @@ public void getTimeUsAtEndOfStream() {
assertThat(seeker.getTimeUs(XING_AUDIO_END_POSITION)).isEqualTo(XING_STREAM_DURATION_US);
}

@Test
public void getAverageBitrate_returnsAverageFromDataSizeAndDuration() {
int expectedAverageBitrate =
(int)
Util.scaleLargeValue(
XING_FRAME.dataSize - XING_FRAME.header.frameSize,
C.BITS_PER_BYTE * C.MICROS_PER_SECOND,
XING_STREAM_DURATION_US,
RoundingMode.HALF_UP);

assertThat(seeker.getAverageBitrate()).isEqualTo(expectedAverageBitrate);
assertThat(seeker.getAverageBitrate()).isNotEqualTo(XING_FRAME.header.bitrate);
}

// https://github.com/androidx/media/issues/3117#issuecomment-4046538506
@Test
public void getTimeUsAtEndOfStream_xingLengthLongerThanStream() {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ track 0:
sample count = 117
track duration = 2807979
format 0:
averageBitrate = 32000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ track 0:
sample count = 117
track duration = 2807979
format 0:
averageBitrate = 32000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ track 0:
sample count = 88
track duration = 2807979
format 0:
averageBitrate = 32000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ track 0:
sample count = 88
track duration = 2807979
format 0:
averageBitrate = 32000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ track 0:
sample count = 117
track duration = 2807979
format 0:
averageBitrate = 32000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ track 0:
sample count = 117
track duration = 2807979
format 0:
averageBitrate = 32000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ track 0:
sample count = 88
track duration = 2807979
format 0:
averageBitrate = 32000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ track 0:
sample count = 59
track duration = 2807979
format 0:
averageBitrate = 32000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ track 0:
sample count = 30
track duration = 2807979
format 0:
averageBitrate = 32000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ track 0:
sample count = 117
track duration = 2807979
format 0:
averageBitrate = 32000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ track 0:
sample count = 117
track duration = 2807979
format 0:
averageBitrate = 64000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ track 0:
sample count = 117
track duration = 2807979
format 0:
averageBitrate = 64000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ track 0:
sample count = 117
track duration = 2807979
format 0:
averageBitrate = 64000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ track 0:
sample count = 77
track duration = 2807979
format 0:
averageBitrate = 64000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ track 0:
sample count = 38
track duration = 2807979
format 0:
averageBitrate = 64000
averageBitrate = 108719
containerMimeType = audio/mpeg
sampleMimeType = audio/mpeg
maxInputSize = 4096
Expand Down
Loading