From 04c2d22178ddfa81683544eb0ed6e181d45f6633 Mon Sep 17 00:00:00 2001
From: shahddaghash <shahddaghash@google.com>
Date: Wed, 15 Jan 2025 10:00:33 -0800
Subject: [PATCH] Pass `containerMimeType` and `sampleMimeType` to
 `MediaItemInfo`

If a format is passed to `ExportResult.ProcessedInput`, the `containerMimeType` and `sampleMimeType` are extracted from the Format and passed inside the `MediaItemInfo` object.

PiperOrigin-RevId: 715840400
---
 .../media3/transformer/EditingMetricsCollector.java   | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/libraries/transformer/src/main/java/androidx/media3/transformer/EditingMetricsCollector.java b/libraries/transformer/src/main/java/androidx/media3/transformer/EditingMetricsCollector.java
index ced33d76eb..b253adcd36 100644
--- a/libraries/transformer/src/main/java/androidx/media3/transformer/EditingMetricsCollector.java
+++ b/libraries/transformer/src/main/java/androidx/media3/transformer/EditingMetricsCollector.java
@@ -28,6 +28,7 @@ import android.util.SparseIntArray;
 import androidx.annotation.Nullable;
 import androidx.annotation.RequiresApi;
 import androidx.media3.common.C;
+import androidx.media3.common.Format;
 import androidx.media3.common.util.SystemClock;
 import com.google.common.collect.ImmutableList;
 import java.util.ArrayList;
@@ -213,7 +214,15 @@ import org.checkerframework.checker.nullness.qual.MonotonicNonNull;
       MediaItemInfo.Builder mediaItemInfoBuilder = new MediaItemInfo.Builder();
       long durationMs = usToMs(processedInput.durationUs);
       mediaItemInfoBuilder.setDurationMillis(durationMs);
-      // TODO: Collect more information about the MediaItem and pass it to MediaItemInfo.Builder
+      Format format = processedInput.format;
+      if (format != null) {
+        if (format.containerMimeType != null) {
+          mediaItemInfoBuilder.setContainerMimeType(format.containerMimeType);
+        }
+        if (format.sampleMimeType != null) {
+          mediaItemInfoBuilder.addSampleMimeType(format.sampleMimeType);
+        }
+      }
       mediaItemInfoList.add(mediaItemInfoBuilder.build());
     }
     return mediaItemInfoList;