loliball/media
1
0
Fork 0
mirror of https://github.com/androidx/media.git synced 2025-04-30 06:46:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Refactor track retrieval into a separate method

Browse Source 
This will be be required when we add more method which
required video track.

PiperOrigin-RevId: 603024260
This commit is contained in:
sheenachhabra 2024-01-31 05:58:26 -08:00 committed by Copybara-Service
parent 65c354029a
commit 61603f90f3
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
libraries/test_utils/src/main/java/androidx/media3/test/utils/FileUtil.java
View File

@ -43,6 +43,17 @@ public final class FileUtil {
*/
public static void assertFileHasColorTransfer(
Context context, @Nullable String filePath, @C.ColorTransfer int expectedColorTransfer) {
Format videoTrackFormat = retrieveVideoTrackFormat(context, filePath);
@Nullable ColorInfo colorInfo = videoTrackFormat.colorInfo;
@C.ColorTransfer
int actualColorTransfer =
colorInfo == null || colorInfo.colorTransfer == Format.NO_VALUE
? C.COLOR_TRANSFER_SDR
: colorInfo.colorTransfer;
assertThat(actualColorTransfer).isEqualTo(expectedColorTransfer);
}
private static Format retrieveVideoTrackFormat(Context context, @Nullable String filePath) {
TrackGroupArray trackGroupArray;
try {
trackGroupArray =
@ -59,14 +70,7 @@ public final class FileUtil {
TrackGroup trackGroup = trackGroupArray.get(i);
if (trackGroup.type == C.TRACK_TYPE_VIDEO) {
assertThat(trackGroup.length).isEqualTo(1);
@Nullable ColorInfo colorInfo = trackGroup.getFormat(0).colorInfo;
@C.ColorTransfer
int actualColorTransfer =
colorInfo == null || colorInfo.colorTransfer == Format.NO_VALUE
? C.COLOR_TRANSFER_SDR
: colorInfo.colorTransfer;
assertThat(actualColorTransfer).isEqualTo(expectedColorTransfer);
return;
return trackGroup.getFormat(0);
}
}
throw new IllegalStateException("Couldn't find video track");