From 1431497e7f31dfb316e85602979f7b9c0cf0607b Mon Sep 17 00:00:00 2001
From: shahddaghash <shahddaghash@google.com>
Date: Tue, 4 Feb 2025 03:07:13 -0800
Subject: [PATCH] Pass output media item information to `MediaItemInfo`

PiperOrigin-RevId: 723007882
---
 .../transformer/EditingMetricsCollector.java   | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

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 453bb7028d..1bb3801c38 100644
--- a/libraries/transformer/src/main/java/androidx/media3/transformer/EditingMetricsCollector.java
+++ b/libraries/transformer/src/main/java/androidx/media3/transformer/EditingMetricsCollector.java
@@ -308,6 +308,24 @@ import org.checkerframework.checker.nullness.qual.MonotonicNonNull;
     if (exportResult.durationMs != C.TIME_UNSET) {
       mediaItemInfoBuilder.setDurationMillis(exportResult.durationMs);
     }
+    if (exportResult.audioMimeType != null) {
+      mediaItemInfoBuilder.addSampleMimeType(exportResult.audioMimeType);
+    }
+    if (exportResult.videoMimeType != null) {
+      mediaItemInfoBuilder.addSampleMimeType(exportResult.videoMimeType);
+    }
+    if (exportResult.channelCount != C.LENGTH_UNSET) {
+      mediaItemInfoBuilder.setAudioChannelCount(exportResult.channelCount);
+    }
+    if (exportResult.sampleRate != C.RATE_UNSET_INT) {
+      mediaItemInfoBuilder.setAudioSampleRateHz(exportResult.sampleRate);
+    }
+    if (exportResult.audioEncoderName != null) {
+      mediaItemInfoBuilder.addCodecName(exportResult.audioEncoderName);
+    }
+    if (exportResult.videoEncoderName != null) {
+      mediaItemInfoBuilder.addCodecName(exportResult.videoEncoderName);
+    }
     mediaItemInfoBuilder.setVideoSampleCount(exportResult.videoFrameCount);
     Size videoSize =
         new Size(