[AppBarLayout] Use a uniform way to determine the target scrolling view

Author: pubiqq

lib/java/com/google/android/material/appbar/AppBarLayout.java

@@ -46,7 +46,6 @@
 import android.view.ViewGroup;
 import android.view.ViewOutlineProvider;
 import android.view.animation.Interpolator;
-import android.widget.AbsListView;
 import android.widget.LinearLayout;
 import android.widget.ScrollView;
 import androidx.annotation.ColorInt;
@@ -63,7 +62,6 @@
 import androidx.core.graphics.drawable.DrawableCompat;
 import androidx.core.util.ObjectsCompat;
 import androidx.core.view.AccessibilityDelegateCompat;
-import androidx.core.view.NestedScrollingChild;
 import androidx.core.view.ViewCompat;
 import androidx.core.view.ViewCompat.NestedScrollType;
 import androidx.core.view.WindowInsetsCompat;
@@ -207,7 +205,7 @@ public interface LiftOnScrollListener {
 
   private boolean liftOnScroll;
   @IdRes private int liftOnScrollTargetViewId;
-  @Nullable private WeakReference<View> liftOnScrollTargetView;
+  @Nullable private WeakReference<View> liftOnScrollTargetViewRef;
   private final boolean hasLiftOnScrollColor;
   @Nullable private ValueAnimator liftOnScrollColorAnimator;
   @Nullable private AnimatorUpdateListener liftOnScrollColorUpdateListener;
@@ -761,7 +759,7 @@ protected LayoutParams generateLayoutParams(ViewGroup.LayoutParams p) {
   protected void onDetachedFromWindow() {
     super.onDetachedFromWindow();
 
-    clearLiftOnScrollTargetView();
+    clearLiftOnScrollTargetViewRef();
   }
 
   boolean hasChildWithInterpolator() {
@@ -1087,9 +1085,9 @@ public boolean isLiftOnScroll() {
   public void setLiftOnScrollTargetView(@Nullable View liftOnScrollTargetView) {
     this.liftOnScrollTargetViewId = View.NO_ID;
     if (liftOnScrollTargetView == null) {
-      clearLiftOnScrollTargetView();
+      clearLiftOnScrollTargetViewRef();
     } else {
-      this.liftOnScrollTargetView = new WeakReference<>(liftOnScrollTargetView);
+      this.liftOnScrollTargetViewRef = new WeakReference<>(liftOnScrollTargetView);
     }
   }
 
@@ -1100,7 +1098,7 @@ public void setLiftOnScrollTargetView(@Nullable View liftOnScrollTargetView) {
   public void setLiftOnScrollTargetViewId(@IdRes int liftOnScrollTargetViewId) {
     this.liftOnScrollTargetViewId = liftOnScrollTargetViewId;
     // Invalidate cached target view so it will be looked up on next scroll.
-    clearLiftOnScrollTargetView();
+    clearLiftOnScrollTargetViewRef();
   }
 
   /**
@@ -1112,39 +1110,58 @@ public int getLiftOnScrollTargetViewId() {
     return liftOnScrollTargetViewId;
   }
 
-  boolean shouldLift(@Nullable View defaultScrollingView) {
-    View scrollingView = findLiftOnScrollTargetView(defaultScrollingView);
-    if (scrollingView == null) {
-      scrollingView = defaultScrollingView;
-    }
+  boolean shouldBeLifted() {
+    final View scrollingView = findLiftOnScrollTargetView();
     return scrollingView != null
         && (scrollingView.canScrollVertically(-1) || scrollingView.getScrollY() > 0);
   }
 
   @Nullable
-  private View findLiftOnScrollTargetView(@Nullable View defaultScrollingView) {
+  private View findLiftOnScrollTargetView() {
+    View liftOnScrollTargetView = liftOnScrollTargetViewRef != null
+        ? liftOnScrollTargetViewRef.get()
+        : null;
+
+    final ViewGroup parent = (ViewGroup) getParent();
+
     if (liftOnScrollTargetView == null && liftOnScrollTargetViewId != View.NO_ID) {
-      View targetView = null;
-      if (defaultScrollingView != null) {
-        targetView = defaultScrollingView.findViewById(liftOnScrollTargetViewId);
-      }
-      if (targetView == null && getParent() instanceof ViewGroup) {
-        // Assumes the scrolling view is a child of the AppBarLayout's parent,
-        // which should be true due to the CoordinatorLayout pattern.
-        targetView = ((ViewGroup) getParent()).findViewById(liftOnScrollTargetViewId);
+      liftOnScrollTargetView = parent.findViewById(liftOnScrollTargetViewId);
+      if (liftOnScrollTargetView != null) {
+        clearLiftOnScrollTargetViewRef();
+        liftOnScrollTargetViewRef = new WeakReference<>(liftOnScrollTargetView);
       }
-      if (targetView != null) {
-        liftOnScrollTargetView = new WeakReference<>(targetView);
+    }
+
+    return liftOnScrollTargetView != null
+        ? liftOnScrollTargetView
+        : findFirstChildWithScrollingBehavior(parent);
+  }
+
+  @Nullable
+  private View findFirstChildWithScrollingBehavior(@NonNull ViewGroup parent) {
+    for (int i = 0, z = parent.getChildCount(); i < z; i++) {
+      final View child = parent.getChildAt(i);
+      if (hasScrollingBehavior(child)) {
+        return child;
       }
     }
-    return liftOnScrollTargetView != null ? liftOnScrollTargetView.get() : null;
+    return null;
   }
 
-  private void clearLiftOnScrollTargetView() {
-    if (liftOnScrollTargetView != null) {
-      liftOnScrollTargetView.clear();
+  private boolean hasScrollingBehavior(@NonNull View view) {
+    if (view.getLayoutParams() instanceof CoordinatorLayout.LayoutParams) {
+      CoordinatorLayout.LayoutParams lp = (CoordinatorLayout.LayoutParams) view.getLayoutParams();
+      return lp.getBehavior() instanceof ScrollingViewBehavior;
     }
-    liftOnScrollTargetView = null;
+
+    return false;
+  }
+
+  private void clearLiftOnScrollTargetViewRef() {
+    if (liftOnScrollTargetViewRef != null) {
+      liftOnScrollTargetViewRef.clear();
+    }
+    liftOnScrollTargetViewRef = null;
   }
 
   /**
@@ -1561,12 +1578,12 @@ private boolean canScrollChildren(
 
     @Override
     public void onNestedPreScroll(
-        CoordinatorLayout coordinatorLayout,
+        @NonNull CoordinatorLayout coordinatorLayout,
         @NonNull T child,
-        View target,
+        @NonNull View target,
         int dx,
         int dy,
-        int[] consumed,
+        @NonNull int[] consumed,
         int type) {
       if (dy != 0) {
         int min;
@@ -1585,7 +1602,7 @@ public void onNestedPreScroll(
         }
       }
       if (child.isLiftOnScroll()) {
-        child.setLiftedState(child.shouldLift(target));
+        child.setLiftedState(child.shouldBeLifted());
       }
     }
 
@@ -1616,7 +1633,10 @@ public void onNestedScroll(
 
     @Override
     public void onStopNestedScroll(
-        CoordinatorLayout coordinatorLayout, @NonNull T abl, View target, int type) {
+        @NonNull CoordinatorLayout coordinatorLayout,
+        @NonNull T abl,
+        @NonNull View target,
+        int type) {
       // onStartNestedScroll for a fling will happen before onStopNestedScroll for the scroll. This
       // isn't necessarily guaranteed yet, but it should be in the future. We use this to our
       // advantage to check if a fling (ViewCompat.TYPE_NON_TOUCH) will start after the touch scroll
@@ -1625,7 +1645,7 @@ public void onStopNestedScroll(
         // If we haven't been flung, or a fling is ending
         snapToChildIfNeeded(coordinatorLayout, abl);
         if (abl.isLiftOnScroll()) {
-          abl.setLiftedState(abl.shouldLift(target));
+          abl.setLiftedState(abl.shouldBeLifted());
         }
       }
 
@@ -2034,7 +2054,7 @@ void onFlingFinished(@NonNull CoordinatorLayout parent, @NonNull T layout) {
       // At the end of a manual fling, check to see if we need to snap to the edge-child
       snapToChildIfNeeded(parent, layout);
       if (layout.isLiftOnScroll()) {
-        layout.setLiftedState(layout.shouldLift(findFirstScrollingChild(parent)));
+        layout.setLiftedState(layout.shouldBeLifted());
       }
     }
 
@@ -2201,9 +2221,7 @@ private void updateAppBarLayoutDrawableState(
       }
 
       if (layout.isLiftOnScroll()) {
-        // Use first scrolling child as default scrolling view for updating lifted state because
-        // it represents the content that would be scrolled beneath the app bar.
-        lifted = layout.shouldLift(findFirstScrollingChild(parent));
+        lifted = layout.shouldBeLifted();
       }
 
       final boolean changed = layout.setLiftedState(lifted);
@@ -2253,19 +2271,6 @@ private static View getAppBarChildOnOffset(
       return null;
     }
 
-    @Nullable
-    private View findFirstScrollingChild(@NonNull CoordinatorLayout parent) {
-      for (int i = 0, z = parent.getChildCount(); i < z; i++) {
-        final View child = parent.getChildAt(i);
-        if (child instanceof NestedScrollingChild
-            || child instanceof AbsListView
-            || child instanceof ScrollView) {
-          return child;
-        }
-      }
-      return null;
-    }
-
     @Override
     int getTopBottomOffsetForScrollingSibling() {
       return getTopAndBottomOffset() + offsetDelta;
@@ -2402,7 +2407,7 @@ public boolean layoutDependsOn(CoordinatorLayout parent, View child, View depend
     public boolean onDependentViewChanged(
         @NonNull CoordinatorLayout parent, @NonNull View child, @NonNull View dependency) {
       offsetChildAsNeeded(child, dependency);
-      updateLiftedStateIfNeeded(child, dependency);
+      updateLiftedStateIfNeeded(dependency);
       return false;
     }
 
@@ -2509,11 +2514,11 @@ int getScrollRange(View v) {
       }
     }
 
-    private void updateLiftedStateIfNeeded(View child, View dependency) {
+    private void updateLiftedStateIfNeeded(@NonNull View dependency) {
       if (dependency instanceof AppBarLayout) {
         AppBarLayout appBarLayout = (AppBarLayout) dependency;
         if (appBarLayout.isLiftOnScroll()) {
-          appBarLayout.setLiftedState(appBarLayout.shouldLift(child));
+          appBarLayout.setLiftedState(appBarLayout.shouldBeLifted());
         }
       }
     }