Add TrackSelectionUtil.getAverageBitrate method
------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=218877191
This commit is contained in:
eguven
Oliver Woodman
parent
9ea678c5e6
commit
ed32e2a7f7
@ -0,0 +1,57 @@
|
||||
/*
|
||||
* Copyright (C) 2018 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
|
||||
*
|
||||
* http://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 com.google.android.exoplayer2.trackselection;
|
||||
|
||||
import com.google.android.exoplayer2.C;
|
||||
import com.google.android.exoplayer2.source.chunk.MediaChunkIterator;
|
||||
|
||||
/** Track selection related utility methods. */
|
||||
public final class TrackSelectionUtil {
|
||||
|
||||
private TrackSelectionUtil() {}
|
||||
|
||||
/**
|
||||
* Returns average bitrate for chunks in bits per second. Chunks are included in average until
|
||||
* {@code maxDurationMs} or the first unknown length chunk.
|
||||
*
|
||||
* @param iterator Iterator for media chunk sequences.
|
||||
* @param maxDurationUs Maximum duration of chunks to be included in average bitrate, in
|
||||
* microseconds.
|
||||
* @return Average bitrate for chunks in bits per second, or {@link C#LENGTH_UNSET} if there are
|
||||
* no chunks or the first chunk length is unknown.
|
||||
*/
|
||||
public static int getAverageBitrate(MediaChunkIterator iterator, long maxDurationUs) {
|
||||
long totalDurationUs = 0;
|
||||
long totalLength = 0;
|
||||
while (iterator.next()) {
|
||||
long chunkLength = iterator.getDataSpec().length;
|
||||
if (chunkLength == C.LENGTH_UNSET) {
|
||||
break;
|
||||
}
|
||||
long chunkDurationUs = iterator.getChunkEndTimeUs() - iterator.getChunkStartTimeUs();
|
||||
if (totalDurationUs + chunkDurationUs >= maxDurationUs) {
|
||||
totalLength += chunkLength * (maxDurationUs - totalDurationUs) / chunkDurationUs;
|
||||
totalDurationUs = maxDurationUs;
|
||||
break;
|
||||
}
|
||||
totalDurationUs += chunkDurationUs;
|
||||
totalLength += chunkLength;
|
||||
}
|
||||
return totalDurationUs == 0
|
||||
? C.LENGTH_UNSET
|
||||
: (int) (totalLength * C.BITS_PER_BYTE * C.MICROS_PER_SECOND / totalDurationUs);
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,150 @@
|
||||
/*
|
||||
* Copyright (C) 2018 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
|
||||
*
|
||||
* http://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 com.google.android.exoplayer2.trackselection;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
|
||||
import android.net.Uri;
|
||||
import com.google.android.exoplayer2.C;
|
||||
import com.google.android.exoplayer2.source.chunk.BaseMediaChunkIterator;
|
||||
import com.google.android.exoplayer2.source.chunk.MediaChunkIterator;
|
||||
import com.google.android.exoplayer2.upstream.DataSpec;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.robolectric.RobolectricTestRunner;
|
||||
|
||||
/** {@link TrackSelectionUtil} tests. */
|
||||
@RunWith(RobolectricTestRunner.class)
|
||||
public class TrackSelectionUtilTest {
|
||||
|
||||
public static final long MAX_DURATION_US = 30 * C.MICROS_PER_SECOND;
|
||||
|
||||
@Test
|
||||
public void getAverageBitrate_emptyIterator_returnsUnsetLength() {
|
||||
assertThat(TrackSelectionUtil.getAverageBitrate(MediaChunkIterator.EMPTY, MAX_DURATION_US))
|
||||
.isEqualTo(C.LENGTH_UNSET);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getAverageBitrate_oneChunk_returnsChunkBitrate() {
|
||||
long[] chunkTimeBoundariesSec = {0, 5};
|
||||
long[] chunkLengths = {10};
|
||||
FakeIterator iterator = new FakeIterator(chunkTimeBoundariesSec, chunkLengths);
|
||||
int expectedAverageBitrate =
|
||||
(int) (chunkLengths[0] * C.BITS_PER_BYTE / chunkTimeBoundariesSec[1]);
|
||||
assertThat(TrackSelectionUtil.getAverageBitrate(iterator, MAX_DURATION_US))
|
||||
.isEqualTo(expectedAverageBitrate);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getAverageBitrate_multipleSameDurationChunks_returnsAverageChunkBitrate() {
|
||||
long[] chunkTimeBoundariesSec = {0, 5, 10};
|
||||
long[] chunkLengths = {10, 20};
|
||||
FakeIterator iterator = new FakeIterator(chunkTimeBoundariesSec, chunkLengths);
|
||||
long totalLength = chunkLengths[0] + chunkLengths[1];
|
||||
int expectedAverageBitrate = (int) (totalLength * C.BITS_PER_BYTE / chunkTimeBoundariesSec[2]);
|
||||
assertThat(TrackSelectionUtil.getAverageBitrate(iterator, MAX_DURATION_US))
|
||||
.isEqualTo(expectedAverageBitrate);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getAverageBitrate_multipleDifferentDurationChunks_returnsAverageChunkBitrate() {
|
||||
long[] chunkTimeBoundariesSec = {0, 5, 15, 30};
|
||||
long[] chunkLengths = {10, 20, 30};
|
||||
FakeIterator iterator = new FakeIterator(chunkTimeBoundariesSec, chunkLengths);
|
||||
long totalLength = chunkLengths[0] + chunkLengths[1] + chunkLengths[2];
|
||||
int expectedAverageBitrate = (int) (totalLength * C.BITS_PER_BYTE / chunkTimeBoundariesSec[3]);
|
||||
assertThat(TrackSelectionUtil.getAverageBitrate(iterator, MAX_DURATION_US))
|
||||
.isEqualTo(expectedAverageBitrate);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getAverageBitrate_firstChunkLengthUnset_returnsUnsetLength() {
|
||||
long[] chunkTimeBoundariesSec = {0, 5, 15, 30};
|
||||
long[] chunkLengths = {C.LENGTH_UNSET, 20, 30};
|
||||
FakeIterator iterator = new FakeIterator(chunkTimeBoundariesSec, chunkLengths);
|
||||
assertThat(TrackSelectionUtil.getAverageBitrate(iterator, MAX_DURATION_US))
|
||||
.isEqualTo(C.LENGTH_UNSET);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getAverageBitrate_secondChunkLengthUnset_returnsFirstChunkBitrate() {
|
||||
long[] chunkTimeBoundariesSec = {0, 5, 15, 30};
|
||||
long[] chunkLengths = {10, C.LENGTH_UNSET, 30};
|
||||
FakeIterator iterator = new FakeIterator(chunkTimeBoundariesSec, chunkLengths);
|
||||
int expectedAverageBitrate =
|
||||
(int) (chunkLengths[0] * C.BITS_PER_BYTE / chunkTimeBoundariesSec[1]);
|
||||
assertThat(TrackSelectionUtil.getAverageBitrate(iterator, MAX_DURATION_US))
|
||||
.isEqualTo(expectedAverageBitrate);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void
|
||||
getAverageBitrate_chunksExceedingMaxDuration_returnsAverageChunkBitrateUpToMaxDuration() {
|
||||
long[] chunkTimeBoundariesSec = {0, 5, 15, 45, 50};
|
||||
long[] chunkLengths = {10, 20, 30, 100};
|
||||
FakeIterator iterator = new FakeIterator(chunkTimeBoundariesSec, chunkLengths);
|
||||
// Just half of the third chunk is in the max duration
|
||||
long totalLength = chunkLengths[0] + chunkLengths[1] + chunkLengths[2] / 2;
|
||||
int expectedAverageBitrate =
|
||||
(int) (totalLength * C.BITS_PER_BYTE * C.MICROS_PER_SECOND / MAX_DURATION_US);
|
||||
assertThat(TrackSelectionUtil.getAverageBitrate(iterator, MAX_DURATION_US))
|
||||
.isEqualTo(expectedAverageBitrate);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getAverageBitrate_zeroMaxDuration_returnsUnsetLength() {
|
||||
long[] chunkTimeBoundariesSec = {0, 5, 10};
|
||||
long[] chunkLengths = {10, 20};
|
||||
FakeIterator iterator = new FakeIterator(chunkTimeBoundariesSec, chunkLengths);
|
||||
assertThat(TrackSelectionUtil.getAverageBitrate(iterator, /* maxDurationUs= */ 0))
|
||||
.isEqualTo(C.LENGTH_UNSET);
|
||||
}
|
||||
|
||||
private static final class FakeIterator extends BaseMediaChunkIterator {
|
||||
|
||||
private final long[] chunkTimeBoundariesSec;
|
||||
private final long[] chunkLengths;
|
||||
|
||||
public FakeIterator(long[] chunkTimeBoundariesSec, long[] chunkLengths) {
|
||||
super(/* fromIndex= */ 0, /* toIndex= */ chunkTimeBoundariesSec.length - 2);
|
||||
this.chunkTimeBoundariesSec = chunkTimeBoundariesSec;
|
||||
this.chunkLengths = chunkLengths;
|
||||
}
|
||||
|
||||
@Override
|
||||
public DataSpec getDataSpec() {
|
||||
checkInBounds();
|
||||
return new DataSpec(
|
||||
Uri.EMPTY,
|
||||
/* absoluteStreamPosition= */ 0,
|
||||
chunkLengths[(int) getCurrentIndex()],
|
||||
/* key= */ null);
|
||||
}
|
||||
|
||||
@Override
|
||||
public long getChunkStartTimeUs() {
|
||||
checkInBounds();
|
||||
return chunkTimeBoundariesSec[(int) getCurrentIndex()] * C.MICROS_PER_SECOND;
|
||||
}
|
||||
|
||||
@Override
|
||||
public long getChunkEndTimeUs() {
|
||||
checkInBounds();
|
||||
return chunkTimeBoundariesSec[(int) getCurrentIndex() + 1] * C.MICROS_PER_SECOND;
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user