Fix end of stream detection for SSIM helper

PiperOrigin-RevId: 436165237
This commit is contained in:
andrewlewis 2022-03-21 11:00:21 +00:00 committed by Ian Baker
parent 47b6195340
commit 3f5d723311

View file

@ -279,7 +279,10 @@ public final class SsimHelper {
sampleSize,
mediaExtractor.getSampleTime(),
mediaExtractor.getSampleFlags());
hasReadEndOfInputStream = !mediaExtractor.advance();
// MediaExtractor.advance does not reliably return false for end-of-stream, so check sample
// metadata instead as a more reliable signal. See [internal: b/121204004].
mediaExtractor.advance();
hasReadEndOfInputStream = mediaExtractor.getSampleTime() == -1;
return true;
}