From fd4d9b3edfd3a5ad015758960f029077919776a5 Mon Sep 17 00:00:00 2001
From: Kshitij Gupta <kshitijgm@gmail.com>
Date: Mon, 15 Mar 2021 17:59:26 +0530
Subject: [PATCH] Lawnchair: Fix reload of AllApps columns

---
 .../launcher3/allapps/AllAppsContainerView.java      |  1 +
 .../launcher3/allapps/AllAppsGridAdapter.java        | 12 +++++++++++-
 .../launcher3/allapps/AlphabeticalAppsList.java      |  1 +
 3 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/com/android/launcher3/allapps/AllAppsContainerView.java b/src/com/android/launcher3/allapps/AllAppsContainerView.java
index c989e7bf64..e36afb6cff 100644
--- a/src/com/android/launcher3/allapps/AllAppsContainerView.java
+++ b/src/com/android/launcher3/allapps/AllAppsContainerView.java
@@ -175,6 +175,7 @@ public class AllAppsContainerView extends SpringRelativeLayout implements DragSo
     public void onDeviceProfileChanged(DeviceProfile dp) {
         for (AdapterHolder holder : mAH) {
             if (holder.recyclerView != null) {
+                holder.adapter.setAllAppsColumns();
                 // Remove all views and clear the pool, while keeping the data same. After this
                 // call, all the viewHolders will be recreated.
                 holder.recyclerView.swapAdapter(holder.recyclerView.getAdapter(), true);
diff --git a/src/com/android/launcher3/allapps/AllAppsGridAdapter.java b/src/com/android/launcher3/allapps/AllAppsGridAdapter.java
index 8ec4d273f8..16cc74b27a 100644
--- a/src/com/android/launcher3/allapps/AllAppsGridAdapter.java
+++ b/src/com/android/launcher3/allapps/AllAppsGridAdapter.java
@@ -201,7 +201,17 @@ public class AllAppsGridAdapter extends RecyclerView.Adapter<AllAppsGridAdapter.
 
         mOnIconClickListener = launcher.getItemOnClickListener();
 
-        setAppsPerRow(mLauncher.getDeviceProfile().inv.numAllAppsColumns);
+        setAllAppsColumns();
+    }
+
+    private int mColumns = 0;
+
+    public void setAllAppsColumns() {
+        int newColumns = mLauncher.getDeviceProfile().inv.numAllAppsColumns;
+        if (newColumns != mColumns) {
+            mColumns = newColumns;
+            setAppsPerRow(mColumns);
+        }
     }
 
     public void setAppsPerRow(int appsPerRow) {
diff --git a/src/com/android/launcher3/allapps/AlphabeticalAppsList.java b/src/com/android/launcher3/allapps/AlphabeticalAppsList.java
index 06209bba85..83fb6fc145 100644
--- a/src/com/android/launcher3/allapps/AlphabeticalAppsList.java
+++ b/src/com/android/launcher3/allapps/AlphabeticalAppsList.java
@@ -281,6 +281,7 @@ public class AlphabeticalAppsList implements AllAppsStore.OnUpdateListener {
 
     private void refreshRecyclerView() {
         if (mAdapter != null) {
+            mAdapter.setAllAppsColumns();
             mAdapter.notifyDataSetChanged();
         }
     }