Support DTS-HD with packets across PES boundary

In DTS-HD streams, it can happen that core packet is in one PES and the
extension substream is in another PES. Ensure we can read ahead into next PES
in this case. We need to use the old timestamp as extss and core always belong
together.

Author: nift4

libraries/extractor/src/main/java/androidx/media3/extractor/ts/DtsReader.java

@@ -95,6 +95,7 @@ public final class DtsReader implements ElementaryStreamReader {
   // Used when reading the samples.
   private boolean coreFormatPendingEmit;
   private long timeUs;
+  private long pendingTimeUs;
   private boolean hasCore;
   private boolean skipExtssUntilCore;
 
@@ -114,6 +115,7 @@ public DtsReader(
     headerScratchBytes = new ParsableByteArray(new byte[maxHeaderSize]);
     state = STATE_FINDING_SYNC;
     timeUs = C.TIME_UNSET;
+    pendingTimeUs = C.TIME_UNSET;
     uhdAudioChunkId = new AtomicInteger();
     extensionSubstreamHeaderSize = C.LENGTH_UNSET;
     uhdHeaderSize = C.LENGTH_UNSET;
@@ -144,7 +146,13 @@ public void createTracks(ExtractorOutput extractorOutput, TrackIdGenerator idGen
 
   @Override
   public void packetStarted(long pesTimeUs, @TsPayloadReader.Flags int flags) {
-    timeUs = pesTimeUs;
+    if (pesTimeUs != C.TIME_UNSET) {
+      if (state == STATE_CHECKING_FOR_EXTSS_AFTER_CORE) {
+        pendingTimeUs = pesTimeUs;
+      } else {
+        timeUs = pesTimeUs;
+      }
+    }
   }
 
   @Override
@@ -265,6 +273,10 @@ public void consume(ParsableByteArray data) throws ParserException {
               checkState(timeUs != C.TIME_UNSET);
               output.sampleMetadata(timeUs, C.BUFFER_FLAG_KEY_FRAME, coreSampleSize, 0, null);
               timeUs += sampleDurationUs;
+              if (pendingTimeUs != C.TIME_UNSET) {
+                timeUs = pendingTimeUs;
+                pendingTimeUs = C.TIME_UNSET;
+              }
               coreSampleSize = 0;
 
               syncBytes = extSyncBytes;
@@ -293,7 +305,7 @@ public void consume(ParsableByteArray data) throws ParserException {
   }
 
   @Override
-  public void packetFinished(boolean isEndOfInput) {
+  public void endOfInputReached() {
     if (state == STATE_CHECKING_FOR_EXTSS_AFTER_CORE) {
       checkNotNull(output);
       if (coreFormatPendingEmit) {