sjs/media
1
0
Fork 0
mirror of https://github.com/samsonjs/media.git synced 2026-04-27 15:07:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

HDR: Update test FileUtil to handle null ColorInfo.

Browse source 
PiperOrigin-RevId: 486706595
(cherry picked from commit 23e724a00b)
This commit is contained in:
huangdarwin 2022-11-07 18:49:14 +00:00 committed by microkatz
parent 13a4089eaf
commit b770eb86b2
1 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
libraries/transformer/src/androidTest/java/androidx/media3/transformer/mh/analysis/FileUtil.java
View file

@ -41,18 +41,25 @@ public class FileUtil {
new DecodeOneFrameUtil.Listener() { new DecodeOneFrameUtil.Listener() {
@Override @Override
public void onContainerExtracted(MediaFormat mediaFormat) { public void onContainerExtracted(MediaFormat mediaFormat) {
@Nullable ColorInfo extractedColor = MediaFormatUtil.getColorInfo(mediaFormat); @Nullable ColorInfo extractedColorInfo = MediaFormatUtil.getColorInfo(mediaFormat);
assertThat(checkNotNull(extractedColor).colorTransfer).isEqualTo(expectedColorTransfer); assertColorInfoHasTransfer(extractedColorInfo, expectedColorTransfer);
} }
@Override @Override
public void onFrameDecoded(MediaFormat mediaFormat) { public void onFrameDecoded(MediaFormat mediaFormat) {
@Nullable ColorInfo decodedColor = MediaFormatUtil.getColorInfo(mediaFormat); @Nullable ColorInfo decodedColorInfo = MediaFormatUtil.getColorInfo(mediaFormat);
assertThat(checkNotNull(decodedColor).colorTransfer).isEqualTo(expectedColorTransfer); assertColorInfoHasTransfer(decodedColorInfo, expectedColorTransfer);
} }
}, },
/* surface= */ null); /* surface= */ null);
} }
private static void assertColorInfoHasTransfer(
@Nullable ColorInfo colorInfo, @C.ColorTransfer int expectedColorTransfer) {
@C.ColorTransfer
int actualColorTransfer = colorInfo == null ? C.COLOR_TRANSFER_SDR : colorInfo.colorTransfer;
assertThat(actualColorTransfer).isEqualTo(expectedColorTransfer);
}
private FileUtil() {} private FileUtil() {}
} }