From 7a5b3b3e5c0211b559ba6c9b2c15655c02e3279d Mon Sep 17 00:00:00 2001
From: samrobinson <samrobinson@google.com>
Date: Thu, 17 Feb 2022 14:19:19 +0000
Subject: [PATCH] Save the bitmap produced by the FrameEditorDataProcessingTest
 to cache.

The bitmap can then be retrieved through ADB.

#mse-bug-week

PiperOrigin-RevId: 429293231
---
 .../FrameEditorDataProcessingTest.java        | 27 +++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/libraries/transformer/src/androidTest/java/androidx/media3/transformer/FrameEditorDataProcessingTest.java b/libraries/transformer/src/androidTest/java/androidx/media3/transformer/FrameEditorDataProcessingTest.java
index 2c3256ba83a..09b7f5efddd 100644
--- a/libraries/transformer/src/androidTest/java/androidx/media3/transformer/FrameEditorDataProcessingTest.java
+++ b/libraries/transformer/src/androidTest/java/androidx/media3/transformer/FrameEditorDataProcessingTest.java
@@ -21,6 +21,7 @@
 import static java.lang.Math.abs;
 import static java.lang.Math.max;
 
+import android.content.Context;
 import android.content.res.AssetFileDescriptor;
 import android.graphics.Bitmap;
 import android.graphics.BitmapFactory;
@@ -34,7 +35,10 @@
 import android.media.MediaFormat;
 import androidx.annotation.Nullable;
 import androidx.media3.common.MimeTypes;
+import androidx.media3.common.util.Log;
 import androidx.test.ext.junit.runners.AndroidJUnit4;
+import java.io.File;
+import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
 import java.nio.ByteBuffer;
@@ -51,6 +55,8 @@
 @RunWith(AndroidJUnit4.class)
 public final class FrameEditorDataProcessingTest {
 
+  private static final String TAG = "FrameEditorDataProcessingTest";
+
   // Input MP4 file to transform.
   private static final String INPUT_MP4_ASSET_STRING = "media/mp4/sample.mp4";
 
@@ -119,6 +125,7 @@ public void processData_noEdits_producesExpectedOutput() throws Exception {
     // TODO(b/207848601): switch to using proper tooling for testing against golden data.
     float averagePixelAbsoluteDifference =
         getAveragePixelAbsoluteDifferenceArgb8888(expectedBitmap, editedBitmap);
+    saveTestBitmapToCacheDirectory("processData_noEdits", editedBitmap);
     assertThat(averagePixelAbsoluteDifference).isAtMost(MAXIMUM_AVERAGE_PIXEL_ABSOLUTE_DIFFERENCE);
   }
 
@@ -137,6 +144,7 @@ public void processData_translateRight_producesExpectedOutput() throws Exception
     // data.simple
     float averagePixelAbsoluteDifference =
         getAveragePixelAbsoluteDifferenceArgb8888(expectedBitmap, editedBitmap);
+    saveTestBitmapToCacheDirectory("processData_translateRight", editedBitmap);
     assertThat(averagePixelAbsoluteDifference).isAtMost(MAXIMUM_AVERAGE_PIXEL_ABSOLUTE_DIFFERENCE);
   }
 
@@ -154,6 +162,7 @@ public void processData_scaleNarrow_producesExpectedOutput() throws Exception {
     // TODO(b/207848601): switch to using proper tooling for testing against golden data.
     float averagePixelAbsoluteDifference =
         getAveragePixelAbsoluteDifferenceArgb8888(expectedBitmap, editedBitmap);
+    saveTestBitmapToCacheDirectory("processData_scaleNarrow", editedBitmap);
     assertThat(averagePixelAbsoluteDifference).isAtMost(MAXIMUM_AVERAGE_PIXEL_ABSOLUTE_DIFFERENCE);
   }
 
@@ -174,6 +183,7 @@ public void processData_rotate90_producesExpectedOutput() throws Exception {
     // TODO(b/207848601): switch to using proper tooling for testing against golden data.
     float averagePixelAbsoluteDifference =
         getAveragePixelAbsoluteDifferenceArgb8888(expectedBitmap, editedBitmap);
+    saveTestBitmapToCacheDirectory("processData_rotate90", editedBitmap);
     assertThat(averagePixelAbsoluteDifference).isAtMost(MAXIMUM_AVERAGE_PIXEL_ABSOLUTE_DIFFERENCE);
   }
 
@@ -328,4 +338,21 @@ private static float getAveragePixelAbsoluteDifferenceArgb8888(Bitmap expected,
     }
     return (float) sumMaximumAbsoluteDifferences / (width * height);
   }
+
+  /**
+   * Saves the {@link Bitmap} to the {@link Context#getCacheDir() cache directory} as a PNG.
+   *
+   * <p>File name will be {@code <testId>_output.png}.
+   *
+   * @param testId Name of the test that produced the {@link Bitmap}.
+   * @param bitmap The {@link Bitmap} to save.
+   */
+  private static void saveTestBitmapToCacheDirectory(String testId, Bitmap bitmap) {
+    File file = new File(getApplicationContext().getExternalCacheDir(), testId + "_output.png");
+    try (FileOutputStream outputStream = new FileOutputStream(file)) {
+      bitmap.compress(Bitmap.CompressFormat.PNG, /* quality= */ 100, outputStream);
+    } catch (IOException e) {
+      Log.e(TAG, "Could not write Bitmap to file path: " + file.getAbsolutePath(), e);
+    }
+  }
 }