From 1c1fd4d8b86e96fa6e89025e4af546c464b36a50 Mon Sep 17 00:00:00 2001
From: Colin Kho <colinkho@meta.com>
Date: Thu, 15 Aug 2024 09:56:58 -0700
Subject: [PATCH] Add getter for ChunkIndex on InitializationChunk

---
 .../media3/exoplayer/source/chunk/InitializationChunk.java  | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/libraries/exoplayer/src/main/java/androidx/media3/exoplayer/source/chunk/InitializationChunk.java b/libraries/exoplayer/src/main/java/androidx/media3/exoplayer/source/chunk/InitializationChunk.java
index 91308277bb..aad438ef77 100644
--- a/libraries/exoplayer/src/main/java/androidx/media3/exoplayer/source/chunk/InitializationChunk.java
+++ b/libraries/exoplayer/src/main/java/androidx/media3/exoplayer/source/chunk/InitializationChunk.java
@@ -23,6 +23,7 @@ import androidx.media3.datasource.DataSource;
 import androidx.media3.datasource.DataSourceUtil;
 import androidx.media3.datasource.DataSpec;
 import androidx.media3.exoplayer.source.chunk.ChunkExtractor.TrackOutputProvider;
+import androidx.media3.extractor.ChunkIndex;
 import androidx.media3.extractor.DefaultExtractorInput;
 import androidx.media3.extractor.Extractor;
 import androidx.media3.extractor.ExtractorInput;
@@ -109,4 +110,9 @@ public final class InitializationChunk extends Chunk {
       DataSourceUtil.closeQuietly(dataSource);
     }
   }
+
+  @Nullable
+  public ChunkIndex getChunkIndex() {
+    return chunkExtractor.getChunkIndex();
+  }
 }