Merge "Revert "Revert "Apply image transformation coming through layout..."" into main

1 files changed, 6 insertions, 0 deletions

diff --git a/bridge/src/com/android/layoutlib/bridge/impl/RenderSessionImpl.java b/bridge/src/com/android/layoutlib/bridge/impl/RenderSessionImpl.java
index ec775b2ec9..de7f651bab 100644
--- a/bridge/src/com/android/layoutlib/bridge/impl/RenderSessionImpl.java
+++ b/bridge/src/com/android/layoutlib/bridge/impl/RenderSessionImpl.java
@@ -91,6 +91,7 @@ import java.util.ArrayList;
 import java.util.IdentityHashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.function.Consumer;
 import java.util.function.Function;
 
 import static android.os._Original_Build.VERSION.SDK_INT;
@@ -590,6 +591,11 @@ public class RenderSessionImpl extends RenderAction<SessionParams> {
             mSystemViewInfoList =
                     visitAllChildren(mViewRoot, 0, 0, params, false);
 
+            Consumer<BufferedImage> imageTransformation = getParams().getImageTransformation();
+            if (imageTransformation != null) {
+                imageTransformation.accept(mImage);
+            }
+
             boolean enableLayoutValidation = Boolean.TRUE.equals(params.getFlag(RenderParamsFlags.FLAG_ENABLE_LAYOUT_VALIDATOR));
             boolean enableLayoutValidationImageCheck = Boolean.TRUE.equals(
                     params.getFlag(RenderParamsFlags.FLAG_ENABLE_LAYOUT_VALIDATOR_IMAGE_CHECK));