1 /*
2  * Copyright (C) 2019 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package com.android.systemui.statusbar.notification.collection.init;
18 
19 import android.util.Log;
20 
21 import androidx.annotation.NonNull;
22 
23 import com.android.systemui.Dumpable;
24 import com.android.systemui.dagger.SysUISingleton;
25 import com.android.systemui.dump.DumpManager;
26 import com.android.systemui.statusbar.NotificationListener;
27 import com.android.systemui.statusbar.notification.collection.NotifCollection;
28 import com.android.systemui.statusbar.notification.collection.NotifInflaterImpl;
29 import com.android.systemui.statusbar.notification.collection.NotifPipeline;
30 import com.android.systemui.statusbar.notification.collection.PipelineDumpable;
31 import com.android.systemui.statusbar.notification.collection.PipelineDumper;
32 import com.android.systemui.statusbar.notification.collection.ShadeListBuilder;
33 import com.android.systemui.statusbar.notification.collection.coalescer.GroupCoalescer;
34 import com.android.systemui.statusbar.notification.collection.coordinator.NotifCoordinators;
35 import com.android.systemui.statusbar.notification.collection.inflation.NotificationRowBinderImpl;
36 import com.android.systemui.statusbar.notification.collection.render.NotifStackController;
37 import com.android.systemui.statusbar.notification.collection.render.RenderStageManager;
38 import com.android.systemui.statusbar.notification.collection.render.ShadeViewManager;
39 import com.android.systemui.statusbar.notification.collection.render.ShadeViewManagerFactory;
40 import com.android.systemui.statusbar.notification.stack.NotificationListContainer;
41 
42 import java.io.PrintWriter;
43 
44 import javax.inject.Inject;
45 
46 /**
47  * Initialization code for the new notification pipeline.
48  */
49 @SysUISingleton
50 public class NotifPipelineInitializer implements Dumpable, PipelineDumpable {
51     private final NotifPipeline mPipelineWrapper;
52     private final GroupCoalescer mGroupCoalescer;
53     private final NotifCollection mNotifCollection;
54     private final ShadeListBuilder mListBuilder;
55     private final RenderStageManager mRenderStageManager;
56     private final NotifCoordinators mNotifPluggableCoordinators;
57     private final NotifInflaterImpl mNotifInflater;
58     private final DumpManager mDumpManager;
59     private final ShadeViewManagerFactory mShadeViewManagerFactory;
60 
61     /* These are saved just for dumping. */
62     private ShadeViewManager mShadeViewManager;
63     private NotificationListener mNotificationService;
64 
65     @Inject
NotifPipelineInitializer( NotifPipeline pipelineWrapper, GroupCoalescer groupCoalescer, NotifCollection notifCollection, ShadeListBuilder listBuilder, RenderStageManager renderStageManager, NotifCoordinators notifCoordinators, NotifInflaterImpl notifInflater, DumpManager dumpManager, ShadeViewManagerFactory shadeViewManagerFactory )66     public NotifPipelineInitializer(
67             NotifPipeline pipelineWrapper,
68             GroupCoalescer groupCoalescer,
69             NotifCollection notifCollection,
70             ShadeListBuilder listBuilder,
71             RenderStageManager renderStageManager,
72             NotifCoordinators notifCoordinators,
73             NotifInflaterImpl notifInflater,
74             DumpManager dumpManager,
75             ShadeViewManagerFactory shadeViewManagerFactory
76     ) {
77         mPipelineWrapper = pipelineWrapper;
78         mGroupCoalescer = groupCoalescer;
79         mNotifCollection = notifCollection;
80         mListBuilder = listBuilder;
81         mRenderStageManager = renderStageManager;
82         mNotifPluggableCoordinators = notifCoordinators;
83         mDumpManager = dumpManager;
84         mNotifInflater = notifInflater;
85         mShadeViewManagerFactory = shadeViewManagerFactory;
86     }
87 
88     /** Hooks the new pipeline up to NotificationManager */
initialize( NotificationListener notificationService, NotificationRowBinderImpl rowBinder, NotificationListContainer listContainer, NotifStackController stackController)89     public void initialize(
90             NotificationListener notificationService,
91             NotificationRowBinderImpl rowBinder,
92             NotificationListContainer listContainer,
93             NotifStackController stackController) {
94         mDumpManager.registerDumpable("NotifPipeline", this);
95 
96         mNotificationService = notificationService;
97 
98         // Setup inflation
99         mNotifInflater.setRowBinder(rowBinder);
100 
101         // Wire up coordinators
102         mNotifPluggableCoordinators.attach(mPipelineWrapper);
103 
104         // Wire up pipeline
105         mShadeViewManager = mShadeViewManagerFactory.create(listContainer, stackController);
106         mShadeViewManager.attach(mRenderStageManager);
107         mRenderStageManager.attach(mListBuilder);
108         mListBuilder.attach(mNotifCollection);
109         mNotifCollection.attach(mGroupCoalescer);
110         mGroupCoalescer.attach(mNotificationService);
111 
112         Log.d(TAG, "Notif pipeline initialized."
113                 + " rendering=" + true);
114     }
115 
116     @Override
dump(PrintWriter pw, String[] args)117     public void dump(PrintWriter pw, String[] args) {
118         dumpPipeline(new PipelineDumper(pw));
119     }
120 
121     @Override
dumpPipeline(@onNull PipelineDumper d)122     public void dumpPipeline(@NonNull PipelineDumper d) {
123         d.println("STAGE 0: SETUP");
124         d.dump("notifPluggableCoordinators", mNotifPluggableCoordinators);
125         d.println("");
126 
127         d.println("STAGE 1: LISTEN");
128         d.dump("notificationService", mNotificationService);
129         d.println("");
130 
131         d.println("STAGE 2: BATCH EVENTS");
132         d.dump("groupCoalescer", mGroupCoalescer);
133         d.println("");
134 
135         d.println("STAGE 3: COLLECT");
136         d.dump("notifCollection", mNotifCollection);
137         d.println("");
138 
139         d.println("STAGE 4: BUILD LIST");
140         d.dump("listBuilder", mListBuilder);
141         d.println("");
142 
143         d.println("STAGE 5: DISPATCH RENDER");
144         d.dump("renderStageManager", mRenderStageManager);
145         d.println("");
146 
147         d.println("STAGE 6: UPDATE SHADE");
148         d.dump("shadeViewManager", mShadeViewManager);
149     }
150 
151     private static final String TAG = "NotifPipeline";
152 }
153