mirror of
https://github.com/samsonjs/media.git
synced 2026-03-30 10:15:48 +00:00
HDR: Update test FileUtil to handle null ColorInfo.
PiperOrigin-RevId: 486706595
(cherry picked from commit 23e724a00b)
This commit is contained in:
parent
13a4089eaf
commit
b770eb86b2
1 changed files with 11 additions and 4 deletions
|
|
@ -41,18 +41,25 @@ public class FileUtil {
|
|||
new DecodeOneFrameUtil.Listener() {
|
||||
@Override
|
||||
public void onContainerExtracted(MediaFormat mediaFormat) {
|
||||
@Nullable ColorInfo extractedColor = MediaFormatUtil.getColorInfo(mediaFormat);
|
||||
assertThat(checkNotNull(extractedColor).colorTransfer).isEqualTo(expectedColorTransfer);
|
||||
@Nullable ColorInfo extractedColorInfo = MediaFormatUtil.getColorInfo(mediaFormat);
|
||||
assertColorInfoHasTransfer(extractedColorInfo, expectedColorTransfer);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onFrameDecoded(MediaFormat mediaFormat) {
|
||||
@Nullable ColorInfo decodedColor = MediaFormatUtil.getColorInfo(mediaFormat);
|
||||
assertThat(checkNotNull(decodedColor).colorTransfer).isEqualTo(expectedColorTransfer);
|
||||
@Nullable ColorInfo decodedColorInfo = MediaFormatUtil.getColorInfo(mediaFormat);
|
||||
assertColorInfoHasTransfer(decodedColorInfo, expectedColorTransfer);
|
||||
}
|
||||
},
|
||||
/* 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() {}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue