Optimize generateSnapshotState

We can speed up this method considerably by leaning on sub-components
doing the heavy lifting of listening for and caching url bar data
updates.

(cherry picked from commit 120f4ff)

Bug: 1063346
Change-Id: Ib888f4242a7e1fa6b45cc76ca78fd2793c76e3c2
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/4177832
Reviewed-by: Sky Malice <skym@chromium.org>
Reviewed-by: Ella Ge <eirage@chromium.org>
Commit-Queue: Patrick Noland <pnoland@chromium.org>
Cr-Original-Commit-Position: refs/heads/main@{#1095142}
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/4218646
Auto-Submit: Sky Malice <skym@chromium.org>
Reviewed-by: Krishna Govind <govind@chromium.org>
Bot-Commit: Rubber Stamper <rubber-stamper@appspot.gserviceaccount.com>
Commit-Queue: Sky Malice <skym@chromium.org>
Owners-Override: Krishna Govind <govind@chromium.org>
Cr-Commit-Position: refs/branch-heads/5481@{#912}
Cr-Branched-From: 130f3e4-refs/heads/main@{#1084008}

chrome/android/java/src/org/chromium/chrome/browser/customtabs/features/toolbar/CustomTabToolbar.java

@@ -1287,6 +1287,11 @@ public OmniboxStub getOmniboxStub() {
             return null;
         }
 
+        @Override
+        public UrlBarData getUrlBarData() {
+            return mUrlCoordinator.getUrlBarData();
+        }
+
         @Override
         public boolean onLongClick(View v) {
             if (v == mTitleUrlContainer) {

chrome/browser/ui/android/omnibox/java/src/org/chromium/chrome/browser/omnibox/LocationBar.java

@@ -80,6 +80,9 @@ default VoiceRecognitionHandler getVoiceRecognitionHandler() {
     @Nullable
     OmniboxStub getOmniboxStub();
 
+    /** Returns the UrlBarData currently in use by the URL bar inside this location bar. */
+    UrlBarData getUrlBarData();
+
     /** Destroys the LocationBar. */
     void destroy();
 }

chrome/browser/ui/android/omnibox/java/src/org/chromium/chrome/browser/omnibox/LocationBarCoordinator.java

@@ -381,6 +381,11 @@ public OmniboxStub getOmniboxStub() {
         return mLocationBarMediator;
     }
 
+    @Override
+    public UrlBarData getUrlBarData() {
+        return mUrlCoordinator.getUrlBarData();
+    }
+
     // OmniboxSuggestionsDropdownEmbedder implementation
     @Override
     public boolean isTablet() {

chrome/browser/ui/android/omnibox/java/src/org/chromium/chrome/browser/omnibox/UrlBarCoordinator.java

@@ -124,6 +124,11 @@ public boolean setUrlBarData(
         return mMediator.setUrlBarData(data, scrollType, state);
     }
 
+    /** Returns the UrlBarData representing the current contents of the UrsssdddsssslBar. */
+    public UrlBarData getUrlBarData() {
+        return mMediator.getUrlBarData();
+    }
+
     /** @see UrlBarMediator#setAutocompleteText(String, String) */
     public void setAutocompleteText(String userText, String autocompleteText) {
         mMediator.setAutocompleteText(userText, autocompleteText);

chrome/browser/ui/android/omnibox/java/src/org/chromium/chrome/browser/omnibox/UrlBarMediator.java

@@ -128,6 +128,10 @@ public boolean setUrlBarData(
         return true;
     }
 
+    UrlBarData getUrlBarData() {
+        return mUrlBarData;
+    }
+
     private void pushTextToModel() {
         CharSequence text =
                 !mHasFocus ? mUrlBarData.displayText : mUrlBarData.getEditingOrDisplayText();

chrome/browser/ui/android/omnibox/java/src/org/chromium/chrome/browser/omnibox/status/StatusCoordinator.java

@@ -193,6 +193,11 @@ public void onSecurityStateChanged() {
         updateVerboseStatusVisibility();
     }
 
+    /** Returns the resource identifier of the current security icon drawable. */
+    public @DrawableRes int getSecurityIconResource() {
+        return mMediator.getSecurityIconResource();
+    }
+
     /** Updates the security icon displayed in the LocationBar. */
     private void updateSecurityIcon() {
         mMediator.updateSecurityIcon(mLocationBarDataProvider.getSecurityIconResource(mIsTablet),

chrome/browser/ui/android/omnibox/java/src/org/chromium/chrome/browser/omnibox/status/StatusMediator.java

@@ -201,6 +201,11 @@ void updateSecurityIcon(
         updateLocationBarIcon(IconTransitionType.CROSSFADE);
     }
 
+    @DrawableRes
+    int getSecurityIconResource() {
+        return mSecurityIconRes;
+    }
+
     /**
      * Update the displayed page's security level and whether it's a paint preview or offline page.
      */

chrome/browser/ui/android/toolbar/java/src/org/chromium/chrome/browser/toolbar/top/ToolbarPhone.java

@@ -1619,15 +1619,27 @@ public boolean setForceTextureCapture(boolean forceTextureCapture) {
     }
 
     private ToolbarSnapshotState generateToolbarSnapshotState() {
-        UrlBarData urlBarData = getToolbarDataProvider().getUrlBarData();
+        UrlBarData urlBarData;
+        int securityIconResource;
+        if (ToolbarFeatures.shouldSuppressCaptures()) {
+            urlBarData = mLocationBar.getUrlBarData();
+            if (urlBarData == null) urlBarData = getToolbarDataProvider().getUrlBarData();
+            StatusCoordinator statusCoordinator = mLocationBar.getStatusCoordinator();
+            securityIconResource = statusCoordinator == null
+                    ? getToolbarDataProvider().getSecurityIconResource(false)
+                    : statusCoordinator.getSecurityIconResource();
+        } else {
+            urlBarData = getToolbarDataProvider().getUrlBarData();
+            securityIconResource = getToolbarDataProvider().getSecurityIconResource(false);
+        }
+
         String displayedUrlText = urlBarData.displayText.toString();
         CharSequence prefixHint = mLocationBar.getOmniboxVisibleTextPrefixHint();
         boolean isValidPrefixHint =
                 ToolbarSnapshotState.isValidVisibleTextPrefixHint(displayedUrlText, prefixHint);
         return new ToolbarSnapshotState(getTint().getDefaultColor(),
                 mTabCountProvider.getTabCount(), mButtonData, mVisualState, displayedUrlText,
-                isValidPrefixHint ? prefixHint : null,
-                getToolbarDataProvider().getSecurityIconResource(false),
+                isValidPrefixHint ? prefixHint : null, securityIconResource,
                 ImageViewCompat.getImageTintList(mHomeButton),
                 getMenuButtonCoordinator().isShowingUpdateBadge(),
                 getToolbarDataProvider().isPaintPreview(), getProgressBar().getProgress(),