| /* |
| * 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.source.dash; |
| |
| import static org.mockito.Mockito.mock; |
| |
| import android.net.Uri; |
| import androidx.annotation.Nullable; |
| import androidx.test.ext.junit.runners.AndroidJUnit4; |
| import com.google.android.exoplayer2.C; |
| import com.google.android.exoplayer2.Format; |
| import com.google.android.exoplayer2.drm.DrmSessionManager; |
| import com.google.android.exoplayer2.source.CompositeSequenceableLoaderFactory; |
| import com.google.android.exoplayer2.source.MediaSource.MediaPeriodId; |
| import com.google.android.exoplayer2.source.MediaSourceEventListener.EventDispatcher; |
| import com.google.android.exoplayer2.source.TrackGroup; |
| import com.google.android.exoplayer2.source.TrackGroupArray; |
| import com.google.android.exoplayer2.source.dash.PlayerEmsgHandler.PlayerEmsgCallback; |
| import com.google.android.exoplayer2.source.dash.manifest.AdaptationSet; |
| import com.google.android.exoplayer2.source.dash.manifest.DashManifest; |
| import com.google.android.exoplayer2.source.dash.manifest.Descriptor; |
| import com.google.android.exoplayer2.source.dash.manifest.Period; |
| import com.google.android.exoplayer2.source.dash.manifest.Representation; |
| import com.google.android.exoplayer2.source.dash.manifest.SegmentBase.SingleSegmentBase; |
| import com.google.android.exoplayer2.source.dash.manifest.UtcTimingElement; |
| import com.google.android.exoplayer2.testutil.MediaPeriodAsserts; |
| import com.google.android.exoplayer2.upstream.Allocator; |
| import com.google.android.exoplayer2.upstream.LoadErrorHandlingPolicy; |
| import com.google.android.exoplayer2.upstream.LoaderErrorThrower; |
| import com.google.android.exoplayer2.upstream.TransferListener; |
| import com.google.android.exoplayer2.util.MimeTypes; |
| import java.util.Arrays; |
| import java.util.Collections; |
| import java.util.List; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.robolectric.annotation.LooperMode; |
| |
| /** Unit tests for {@link DashMediaPeriod}. */ |
| @RunWith(AndroidJUnit4.class) |
| @LooperMode(LooperMode.Mode.PAUSED) |
| public final class DashMediaPeriodTest { |
| |
| @Test |
| public void getStreamKeys_isCompatibleWithDashManifestFilter() { |
| // Test manifest which covers various edge cases: |
| // - Multiple periods. |
| // - Single and multiple representations per adaptation set. |
| // - Switch descriptors combining multiple adaptations sets. |
| // - Embedded track groups. |
| // All cases are deliberately combined in one test to catch potential indexing problems which |
| // only occur in combination. |
| DashManifest manifest = |
| createDashManifest( |
| createPeriod( |
| createAdaptationSet( |
| /* id= */ 0, |
| C.TRACK_TYPE_VIDEO, |
| /* descriptor= */ null, |
| createVideoRepresentation(/* bitrate= */ 1000000))), |
| createPeriod( |
| createAdaptationSet( |
| /* id= */ 100, |
| C.TRACK_TYPE_VIDEO, |
| createSwitchDescriptor(/* ids...= */ 103, 104), |
| createVideoRepresentationWithInbandEventStream(/* bitrate= */ 200000), |
| createVideoRepresentationWithInbandEventStream(/* bitrate= */ 400000), |
| createVideoRepresentationWithInbandEventStream(/* bitrate= */ 600000)), |
| createAdaptationSet( |
| /* id= */ 101, |
| C.TRACK_TYPE_AUDIO, |
| createSwitchDescriptor(/* ids...= */ 102), |
| createAudioRepresentation(/* bitrate= */ 48000), |
| createAudioRepresentation(/* bitrate= */ 96000)), |
| createAdaptationSet( |
| /* id= */ 102, |
| C.TRACK_TYPE_AUDIO, |
| createSwitchDescriptor(/* ids...= */ 101), |
| createAudioRepresentation(/* bitrate= */ 256000)), |
| createAdaptationSet( |
| /* id= */ 103, |
| C.TRACK_TYPE_VIDEO, |
| createSwitchDescriptor(/* ids...= */ 100, 104), |
| createVideoRepresentationWithInbandEventStream(/* bitrate= */ 800000), |
| createVideoRepresentationWithInbandEventStream(/* bitrate= */ 1000000)), |
| createAdaptationSet( |
| /* id= */ 104, |
| C.TRACK_TYPE_VIDEO, |
| createSwitchDescriptor(/* ids...= */ 100, 103), |
| createVideoRepresentationWithInbandEventStream(/* bitrate= */ 2000000)), |
| createAdaptationSet( |
| /* id= */ 105, |
| C.TRACK_TYPE_TEXT, |
| /* descriptor= */ null, |
| createTextRepresentation(/* language= */ "eng")), |
| createAdaptationSet( |
| /* id= */ 105, |
| C.TRACK_TYPE_TEXT, |
| /* descriptor= */ null, |
| createTextRepresentation(/* language= */ "ger")))); |
| |
| // Ignore embedded metadata as we don't want to select primary group just to get embedded track. |
| MediaPeriodAsserts.assertGetStreamKeysAndManifestFilterIntegration( |
| DashMediaPeriodTest::createDashMediaPeriod, |
| manifest, |
| /* periodIndex= */ 1, |
| /* ignoredMimeType= */ "application/x-emsg"); |
| } |
| |
| @Test |
| public void adaptationSetSwitchingProperty_mergesTrackGroups() { |
| DashManifest manifest = |
| createDashManifest( |
| createPeriod( |
| createAdaptationSet( |
| /* id= */ 0, |
| C.TRACK_TYPE_VIDEO, |
| createSwitchDescriptor(/* ids...= */ 1, 2), |
| createVideoRepresentation(/* bitrate= */ 0), |
| createVideoRepresentation(/* bitrate= */ 1)), |
| createAdaptationSet( |
| /* id= */ 3, |
| C.TRACK_TYPE_VIDEO, |
| /* descriptor= */ null, |
| createVideoRepresentation(/* bitrate= */ 300)), |
| createAdaptationSet( |
| /* id= */ 2, |
| C.TRACK_TYPE_VIDEO, |
| createSwitchDescriptor(/* ids...= */ 0, 1), |
| createVideoRepresentation(/* bitrate= */ 200), |
| createVideoRepresentation(/* bitrate= */ 201)), |
| createAdaptationSet( |
| /* id= */ 1, |
| C.TRACK_TYPE_VIDEO, |
| createSwitchDescriptor(/* ids...= */ 0, 2), |
| createVideoRepresentation(/* bitrate= */ 100)))); |
| DashMediaPeriod dashMediaPeriod = createDashMediaPeriod(manifest, 0); |
| List<AdaptationSet> adaptationSets = manifest.getPeriod(0).adaptationSets; |
| |
| // We expect the three adaptation sets with the switch descriptor to be merged, retaining the |
| // representations in their original order. |
| TrackGroupArray expectedTrackGroups = |
| new TrackGroupArray( |
| new TrackGroup( |
| adaptationSets.get(0).representations.get(0).format, |
| adaptationSets.get(0).representations.get(1).format, |
| adaptationSets.get(2).representations.get(0).format, |
| adaptationSets.get(2).representations.get(1).format, |
| adaptationSets.get(3).representations.get(0).format), |
| new TrackGroup(adaptationSets.get(1).representations.get(0).format)); |
| |
| MediaPeriodAsserts.assertTrackGroups(dashMediaPeriod, expectedTrackGroups); |
| } |
| |
| @Test |
| public void trickPlayProperty_mergesTrackGroups() { |
| DashManifest manifest = |
| createDashManifest( |
| createPeriod( |
| createAdaptationSet( |
| /* id= */ 0, |
| C.TRACK_TYPE_VIDEO, |
| createTrickPlayDescriptor(/* mainAdaptationSetId= */ 1), |
| createVideoRepresentation(/* bitrate= */ 0), |
| createVideoRepresentation(/* bitrate= */ 1)), |
| createAdaptationSet( |
| /* id= */ 1, |
| C.TRACK_TYPE_VIDEO, |
| /* descriptor= */ null, |
| createVideoRepresentation(/* bitrate= */ 100)), |
| createAdaptationSet( |
| /* id= */ 2, |
| C.TRACK_TYPE_VIDEO, |
| /* descriptor= */ null, |
| createVideoRepresentation(/* bitrate= */ 200), |
| createVideoRepresentation(/* bitrate= */ 201)), |
| createAdaptationSet( |
| /* id= */ 3, |
| C.TRACK_TYPE_VIDEO, |
| createTrickPlayDescriptor(/* mainAdaptationSetId= */ 2), |
| createVideoRepresentation(/* bitrate= */ 300)))); |
| DashMediaPeriod dashMediaPeriod = createDashMediaPeriod(manifest, 0); |
| List<AdaptationSet> adaptationSets = manifest.getPeriod(0).adaptationSets; |
| |
| // We expect the trick play adaptation sets to be merged with the ones to which they refer, |
| // retaining representations in their original order. |
| TrackGroupArray expectedTrackGroups = |
| new TrackGroupArray( |
| new TrackGroup( |
| adaptationSets.get(0).representations.get(0).format, |
| adaptationSets.get(0).representations.get(1).format, |
| adaptationSets.get(1).representations.get(0).format), |
| new TrackGroup( |
| adaptationSets.get(2).representations.get(0).format, |
| adaptationSets.get(2).representations.get(1).format, |
| adaptationSets.get(3).representations.get(0).format)); |
| |
| MediaPeriodAsserts.assertTrackGroups(dashMediaPeriod, expectedTrackGroups); |
| } |
| |
| @Test |
| public void adaptationSetSwitchingProperty_andTrickPlayProperty_mergesTrackGroups() { |
| DashManifest manifest = |
| createDashManifest( |
| createPeriod( |
| createAdaptationSet( |
| /* id= */ 0, |
| C.TRACK_TYPE_VIDEO, |
| createTrickPlayDescriptor(/* mainAdaptationSetId= */ 1), |
| createVideoRepresentation(/* bitrate= */ 0), |
| createVideoRepresentation(/* bitrate= */ 1)), |
| createAdaptationSet( |
| /* id= */ 1, |
| C.TRACK_TYPE_VIDEO, |
| createSwitchDescriptor(/* ids...= */ 2), |
| createVideoRepresentation(/* bitrate= */ 100)), |
| createAdaptationSet( |
| /* id= */ 2, |
| C.TRACK_TYPE_VIDEO, |
| createSwitchDescriptor(/* ids...= */ 1), |
| createVideoRepresentation(/* bitrate= */ 200), |
| createVideoRepresentation(/* bitrate= */ 201)), |
| createAdaptationSet( |
| /* id= */ 3, |
| C.TRACK_TYPE_VIDEO, |
| createTrickPlayDescriptor(/* mainAdaptationSetId= */ 2), |
| createVideoRepresentation(/* bitrate= */ 300)))); |
| DashMediaPeriod dashMediaPeriod = createDashMediaPeriod(manifest, 0); |
| List<AdaptationSet> adaptationSets = manifest.getPeriod(0).adaptationSets; |
| |
| // We expect all adaptation sets to be merged into one group, retaining representations in their |
| // original order. |
| TrackGroupArray expectedTrackGroups = |
| new TrackGroupArray( |
| new TrackGroup( |
| adaptationSets.get(0).representations.get(0).format, |
| adaptationSets.get(0).representations.get(1).format, |
| adaptationSets.get(1).representations.get(0).format, |
| adaptationSets.get(2).representations.get(0).format, |
| adaptationSets.get(2).representations.get(1).format, |
| adaptationSets.get(3).representations.get(0).format)); |
| |
| MediaPeriodAsserts.assertTrackGroups(dashMediaPeriod, expectedTrackGroups); |
| } |
| |
| private static DashMediaPeriod createDashMediaPeriod(DashManifest manifest, int periodIndex) { |
| return new DashMediaPeriod( |
| /* id= */ periodIndex, |
| manifest, |
| periodIndex, |
| mock(DashChunkSource.Factory.class), |
| mock(TransferListener.class), |
| DrmSessionManager.getDummyDrmSessionManager(), |
| mock(LoadErrorHandlingPolicy.class), |
| new EventDispatcher() |
| .withParameters( |
| /* windowIndex= */ 0, |
| /* mediaPeriodId= */ new MediaPeriodId(/* periodUid= */ new Object()), |
| /* mediaTimeOffsetMs= */ 0), |
| /* elapsedRealtimeOffsetMs= */ 0, |
| mock(LoaderErrorThrower.class), |
| mock(Allocator.class), |
| mock(CompositeSequenceableLoaderFactory.class), |
| mock(PlayerEmsgCallback.class)); |
| } |
| |
| private static DashManifest createDashManifest(Period... periods) { |
| return new DashManifest( |
| /* availabilityStartTimeMs= */ 0, |
| /* durationMs= */ 5000, |
| /* minBufferTimeMs= */ 1, |
| /* dynamic= */ false, |
| /* minUpdatePeriodMs= */ 2, |
| /* timeShiftBufferDepthMs= */ 3, |
| /* suggestedPresentationDelayMs= */ 4, |
| /* publishTimeMs= */ 12345, |
| /* programInformation= */ null, |
| new UtcTimingElement("", ""), |
| Uri.EMPTY, |
| Arrays.asList(periods)); |
| } |
| |
| private static Period createPeriod(AdaptationSet... adaptationSets) { |
| return new Period(/* id= */ null, /* startMs= */ 0, Arrays.asList(adaptationSets)); |
| } |
| |
| private static AdaptationSet createAdaptationSet( |
| int id, int trackType, @Nullable Descriptor descriptor, Representation... representations) { |
| return new AdaptationSet( |
| id, |
| trackType, |
| Arrays.asList(representations), |
| /* accessibilityDescriptors= */ Collections.emptyList(), |
| /* essentialProperties= */ Collections.emptyList(), |
| descriptor == null ? Collections.emptyList() : Collections.singletonList(descriptor)); |
| } |
| |
| private static Representation createVideoRepresentation(int bitrate) { |
| return Representation.newInstance( |
| /* revisionId= */ 0, |
| createVideoFormat(bitrate), |
| /* baseUrl= */ "", |
| new SingleSegmentBase()); |
| } |
| |
| private static Representation createVideoRepresentationWithInbandEventStream(int bitrate) { |
| return Representation.newInstance( |
| /* revisionId= */ 0, |
| createVideoFormat(bitrate), |
| /* baseUrl= */ "", |
| new SingleSegmentBase(), |
| Collections.singletonList(getInbandEventDescriptor())); |
| } |
| |
| private static Format createVideoFormat(int bitrate) { |
| return new Format.Builder() |
| .setContainerMimeType(MimeTypes.VIDEO_MP4) |
| .setSampleMimeType(MimeTypes.VIDEO_H264) |
| .setPeakBitrate(bitrate) |
| .build(); |
| } |
| |
| private static Representation createAudioRepresentation(int bitrate) { |
| Format format = |
| new Format.Builder() |
| .setContainerMimeType(MimeTypes.AUDIO_MP4) |
| .setSampleMimeType(MimeTypes.AUDIO_AAC) |
| .setPeakBitrate(bitrate) |
| .build(); |
| return Representation.newInstance( |
| /* revisionId= */ 0, format, /* baseUrl= */ "", new SingleSegmentBase()); |
| } |
| |
| private static Representation createTextRepresentation(String language) { |
| Format format = |
| new Format.Builder() |
| .setContainerMimeType(MimeTypes.APPLICATION_MP4) |
| .setSampleMimeType(MimeTypes.TEXT_VTT) |
| .setLanguage(language) |
| .build(); |
| return Representation.newInstance( |
| /* revisionId= */ 0, format, /* baseUrl= */ "", new SingleSegmentBase()); |
| } |
| |
| private static Descriptor createSwitchDescriptor(int... ids) { |
| StringBuilder idString = new StringBuilder(); |
| idString.append(ids[0]); |
| for (int i = 1; i < ids.length; i++) { |
| idString.append(",").append(ids[i]); |
| } |
| return new Descriptor( |
| /* schemeIdUri= */ "urn:mpeg:dash:adaptation-set-switching:2016", |
| /* value= */ idString.toString(), |
| /* id= */ null); |
| } |
| |
| private static Descriptor createTrickPlayDescriptor(int mainAdaptationSetId) { |
| return new Descriptor( |
| /* schemeIdUri= */ "http://dashif.org/guidelines/trickmode", |
| /* value= */ Integer.toString(mainAdaptationSetId), |
| /* id= */ null); |
| } |
| |
| private static Descriptor getInbandEventDescriptor() { |
| return new Descriptor( |
| /* schemeIdUri= */ "inBandSchemeIdUri", /* value= */ "inBandValue", /* id= */ "inBandId"); |
| } |
| } |