1 /*
2  * Copyright (C) 2017 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.launcher3.dot;
18 
19 import android.view.ViewDebug;
20 
21 import com.android.launcher3.Utilities;
22 
23 /**
24  * Subclass of DotInfo that only contains the dot count, which is
25  * the sum of all the Folder's items' notifications (each counts as 1).
26  */
27 public class FolderDotInfo extends DotInfo {
28 
29     private static final int MIN_COUNT = 0;
30 
31     private int mNumNotifications;
32 
addDotInfo(DotInfo dotToAdd)33     public void addDotInfo(DotInfo dotToAdd) {
34         if (dotToAdd == null) {
35             return;
36         }
37         mNumNotifications += dotToAdd.getNotificationKeys().size();
38         mNumNotifications = Utilities.boundToRange(
39                 mNumNotifications, MIN_COUNT, DotInfo.MAX_COUNT);
40     }
41 
subtractDotInfo(DotInfo dotToSubtract)42     public void subtractDotInfo(DotInfo dotToSubtract) {
43         if (dotToSubtract == null) {
44             return;
45         }
46         mNumNotifications -= dotToSubtract.getNotificationKeys().size();
47         mNumNotifications = Utilities.boundToRange(
48                 mNumNotifications, MIN_COUNT, DotInfo.MAX_COUNT);
49     }
50 
51     @Override
getNotificationCount()52     public int getNotificationCount() {
53         return mNumNotifications;
54     }
55 
56     @ViewDebug.ExportedProperty(category = "launcher")
hasDot()57     public boolean hasDot() {
58         return mNumNotifications > 0;
59     }
60 }
61