Merge "New Pipeline: Show group count" into sc-v2-dev (76292931) · Commits · e / os / android_frameworks_base

packages/SystemUI/src/com/android/systemui/statusbar/notification/collection/render/ShadeViewManager.kt

+12 −1

Original line number	Diff line number	Diff line
		@@ -18,8 +18,10 @@ package com.android.systemui.statusbar.notification.collection.render

		import android.content.Context
		import android.view.View
		import com.android.systemui.statusbar.notification.collection.GroupEntry
		import com.android.systemui.statusbar.notification.collection.ListEntry
		import com.android.systemui.statusbar.notification.collection.ShadeListBuilder
		import com.android.systemui.statusbar.notification.row.ExpandableNotificationRow
		import com.android.systemui.statusbar.notification.stack.NotificationListContainer
		import com.android.systemui.statusbar.phone.NotificationIconAreaController
		import javax.inject.Inject
		@@ -32,7 +34,7 @@ class ShadeViewManager constructor(
		context: Context,
		listContainer: NotificationListContainer,
		logger: ShadeViewDifferLogger,
		viewBarn: NotifViewBarn,
		private val viewBarn: NotifViewBarn,
		private val notificationIconAreaController: NotificationIconAreaController
		) {
		// We pass a shim view here because the listContainer may not actually have a view associated
		@@ -46,8 +48,17 @@ class ShadeViewManager constructor(

		private fun onNewNotifTree(notifList: List<ListEntry>) {
		viewDiffer.applySpec(specBuilder.buildNodeSpec(rootController, notifList))
		updateGroupCounts(notifList)
		notificationIconAreaController.updateNotificationIcons(notifList)
		}

		private fun updateGroupCounts(notifList: List<ListEntry>) {
		notifList.asSequence().filterIsInstance<GroupEntry>().forEach { groupEntry ->
		val controller = viewBarn.requireView(checkNotNull(groupEntry.summary))
		val row = controller.view as ExpandableNotificationRow
		row.setUntruncatedChildCount(groupEntry.untruncatedChildCount)
		}
		}
		}

		class ShadeViewManagerFactory @Inject constructor(