Merge pull request #9777 from TiVo:p-optimize-timestampadjuster-wait

PiperOrigin-RevId: 417769018
This commit is contained in:
tonihei 2022-01-05 10:45:39 +00:00
commit 7166e091cd

View File

@ -423,19 +423,19 @@ import org.checkerframework.checker.nullness.qual.RequiresNonNull;
// initDataLoadRequired => initDataSource != null && initDataSpec != null // initDataLoadRequired => initDataSource != null && initDataSpec != null
Assertions.checkNotNull(initDataSource); Assertions.checkNotNull(initDataSource);
Assertions.checkNotNull(initDataSpec); Assertions.checkNotNull(initDataSpec);
feedDataToExtractor(initDataSource, initDataSpec, initSegmentEncrypted); feedDataToExtractor(
initDataSource,
initDataSpec,
initSegmentEncrypted,
/* initializeTimestampAdjuster= */ false);
nextLoadPosition = 0; nextLoadPosition = 0;
initDataLoadRequired = false; initDataLoadRequired = false;
} }
@RequiresNonNull("output") @RequiresNonNull("output")
private void loadMedia() throws IOException { private void loadMedia() throws IOException {
try { feedDataToExtractor(
timestampAdjuster.sharedInitializeOrWait(isMasterTimestampSource, startTimeUs); dataSource, dataSpec, mediaSegmentEncrypted, /* initializeTimestampAdjuster= */ true);
} catch (InterruptedException e) {
throw new InterruptedIOException();
}
feedDataToExtractor(dataSource, dataSpec, mediaSegmentEncrypted);
} }
/** /**
@ -445,7 +445,11 @@ import org.checkerframework.checker.nullness.qual.RequiresNonNull;
*/ */
@RequiresNonNull("output") @RequiresNonNull("output")
private void feedDataToExtractor( private void feedDataToExtractor(
DataSource dataSource, DataSpec dataSpec, boolean dataIsEncrypted) throws IOException { DataSource dataSource,
DataSpec dataSpec,
boolean dataIsEncrypted,
boolean initializeTimestampAdjuster)
throws IOException {
// If we previously fed part of this chunk to the extractor, we need to skip it this time. For // If we previously fed part of this chunk to the extractor, we need to skip it this time. For
// encrypted content we need to skip the data by reading it through the source, so as to ensure // encrypted content we need to skip the data by reading it through the source, so as to ensure
// correct decryption of the remainder of the chunk. For clear content, we can request the // correct decryption of the remainder of the chunk. For clear content, we can request the
@ -460,7 +464,8 @@ import org.checkerframework.checker.nullness.qual.RequiresNonNull;
skipLoadedBytes = false; skipLoadedBytes = false;
} }
try { try {
ExtractorInput input = prepareExtraction(dataSource, loadDataSpec); ExtractorInput input =
prepareExtraction(dataSource, loadDataSpec, initializeTimestampAdjuster);
if (skipLoadedBytes) { if (skipLoadedBytes) {
input.skipFully(nextLoadPosition); input.skipFully(nextLoadPosition);
} }
@ -484,9 +489,17 @@ import org.checkerframework.checker.nullness.qual.RequiresNonNull;
@RequiresNonNull("output") @RequiresNonNull("output")
@EnsuresNonNull("extractor") @EnsuresNonNull("extractor")
private DefaultExtractorInput prepareExtraction(DataSource dataSource, DataSpec dataSpec) private DefaultExtractorInput prepareExtraction(
DataSource dataSource, DataSpec dataSpec, boolean initializeTimestampAdjuster)
throws IOException { throws IOException {
long bytesToRead = dataSource.open(dataSpec); long bytesToRead = dataSource.open(dataSpec);
if (initializeTimestampAdjuster) {
try {
timestampAdjuster.sharedInitializeOrWait(isMasterTimestampSource, startTimeUs);
} catch (InterruptedException e) {
throw new InterruptedIOException();
}
}
DefaultExtractorInput extractorInput = DefaultExtractorInput extractorInput =
new DefaultExtractorInput(dataSource, dataSpec.position, bytesToRead); new DefaultExtractorInput(dataSource, dataSpec.position, bytesToRead);