From e9b82ee9518c72aa757e5c62aed9afa1c03b5dbe Mon Sep 17 00:00:00 2001
From: shahddaghash <shahddaghash@google.com>
Date: Wed, 15 Jan 2025 10:06:08 -0800
Subject: [PATCH] Pass video frame rate and video size to `MediaItemInfo`

The frame rate, height, and width are extracted from the format of processedInput.

PiperOrigin-RevId: 715843086
---
 .../media3/transformer/EditingMetricsCollector.java      | 9 +++++++++
 1 file changed, 9 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 b253adcd36..b7abf15c82 100644
--- a/libraries/transformer/src/main/java/androidx/media3/transformer/EditingMetricsCollector.java
+++ b/libraries/transformer/src/main/java/androidx/media3/transformer/EditingMetricsCollector.java
@@ -24,6 +24,7 @@ import android.media.metrics.EditingEndedEvent;
 import android.media.metrics.EditingSession;
 import android.media.metrics.MediaItemInfo;
 import android.media.metrics.MediaMetricsManager;
+import android.util.Size;
 import android.util.SparseIntArray;
 import androidx.annotation.Nullable;
 import androidx.annotation.RequiresApi;
@@ -222,6 +223,14 @@ import org.checkerframework.checker.nullness.qual.MonotonicNonNull;
         if (format.sampleMimeType != null) {
           mediaItemInfoBuilder.addSampleMimeType(format.sampleMimeType);
         }
+        if (format.frameRate != Format.NO_VALUE) {
+          mediaItemInfoBuilder.setVideoFrameRate(format.frameRate);
+        }
+        Size videoSize =
+            new Size(
+                format.width != Format.NO_VALUE ? format.width : MediaItemInfo.VALUE_UNSPECIFIED,
+                format.height != Format.NO_VALUE ? format.height : MediaItemInfo.VALUE_UNSPECIFIED);
+        mediaItemInfoBuilder.setVideoSize(videoSize);
       }
       mediaItemInfoList.add(mediaItemInfoBuilder.build());
     }