Settings: Reload default pref category at revamped top page

* Fixes weird divider looks

Change-Id: I8b30f5818cc3e65c08c82c8da5eda7b864db87e0

src/com/android/settings/homepage/TopLevelSettings.java

@@ -213,6 +213,7 @@ public void onSaveInstanceState(Bundle outState) {
     public void onCreatePreferences(Bundle savedInstanceState, String rootKey) {
         super.onCreatePreferences(savedInstanceState, rootKey);
         if (Flags.homepageRevamp()) {
+            updatePreferenceCategoryLayout();
             return;
         }
         int tintColor = Utils.getHomepageIconColor(getContext());
@@ -224,6 +225,17 @@ public void onCreatePreferences(Bundle savedInstanceState, String rootKey) {
         });
     }
 
+    private void updatePreferenceCategoryLayout() {
+        final PreferenceScreen screen = getPreferenceScreen();
+        final int count = screen.getPreferenceCount();
+        for (int i = 0; i < count; i++) {
+            final Preference preference = screen.getPreference(i);
+            if (preference instanceof PreferenceCategory) {
+                preference.setLayoutResource(com.android.settingslib.widget.theme.R.layout.settingslib_preference_category_no_title);
+            }
+        }
+    }
+
     @Override
     public void onConfigurationChanged(Configuration newConfig) {
         super.onConfigurationChanged(newConfig);