Winson Chung
871759483b
Bug: 219861883 Test: Builds Change-Id: I7831d0de14286c9fa39fed59b168535e0a70d94d
276 lines
10 KiB
Java
276 lines
10 KiB
Java
/*
|
|
* Copyright (C) 2018 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
package com.android.quickstep;
|
|
|
|
import static com.android.launcher3.uioverrides.QuickstepLauncher.GO_LOW_RAM_RECENTS_ENABLED;
|
|
import static com.android.launcher3.util.DisplayController.CHANGE_DENSITY;
|
|
|
|
import android.app.ActivityManager;
|
|
import android.app.ActivityManager.TaskDescription;
|
|
import android.content.Context;
|
|
import android.content.pm.ActivityInfo;
|
|
import android.content.pm.PackageManager;
|
|
import android.content.res.Resources;
|
|
import android.graphics.Bitmap;
|
|
import android.graphics.drawable.BitmapDrawable;
|
|
import android.graphics.drawable.Drawable;
|
|
import android.os.UserHandle;
|
|
import android.text.TextUtils;
|
|
import android.util.SparseArray;
|
|
import android.view.accessibility.AccessibilityManager;
|
|
|
|
import androidx.annotation.WorkerThread;
|
|
|
|
import com.android.launcher3.R;
|
|
import com.android.launcher3.Utilities;
|
|
import com.android.launcher3.icons.BaseIconFactory;
|
|
import com.android.launcher3.icons.BaseIconFactory.IconOptions;
|
|
import com.android.launcher3.icons.BitmapInfo;
|
|
import com.android.launcher3.icons.IconProvider;
|
|
import com.android.launcher3.util.DisplayController;
|
|
import com.android.launcher3.util.DisplayController.DisplayInfoChangeListener;
|
|
import com.android.launcher3.util.DisplayController.Info;
|
|
import com.android.launcher3.util.Preconditions;
|
|
import com.android.quickstep.util.CancellableTask;
|
|
import com.android.quickstep.util.TaskKeyLruCache;
|
|
import com.android.systemui.shared.recents.model.Task;
|
|
import com.android.systemui.shared.recents.model.Task.TaskKey;
|
|
import com.android.systemui.shared.system.PackageManagerWrapper;
|
|
|
|
import java.util.concurrent.Executor;
|
|
import java.util.function.Consumer;
|
|
|
|
/**
|
|
* Manages the caching of task icons and related data.
|
|
*/
|
|
public class TaskIconCache implements DisplayInfoChangeListener {
|
|
|
|
private final Executor mBgExecutor;
|
|
private final AccessibilityManager mAccessibilityManager;
|
|
|
|
private final Context mContext;
|
|
private final TaskKeyLruCache<TaskCacheEntry> mIconCache;
|
|
private final SparseArray<BitmapInfo> mDefaultIcons = new SparseArray<>();
|
|
private BitmapInfo mDefaultIconBase = null;
|
|
|
|
private final IconProvider mIconProvider;
|
|
|
|
private BaseIconFactory mIconFactory;
|
|
|
|
public TaskIconCache(Context context, Executor bgExecutor, IconProvider iconProvider) {
|
|
mContext = context;
|
|
mBgExecutor = bgExecutor;
|
|
mAccessibilityManager = context.getSystemService(AccessibilityManager.class);
|
|
mIconProvider = iconProvider;
|
|
|
|
Resources res = context.getResources();
|
|
int cacheSize = res.getInteger(R.integer.recentsIconCacheSize);
|
|
|
|
mIconCache = new TaskKeyLruCache<>(cacheSize);
|
|
|
|
DisplayController.INSTANCE.get(mContext).addChangeListener(this);
|
|
}
|
|
|
|
@Override
|
|
public void onDisplayInfoChanged(Context context, Info info, int flags) {
|
|
if ((flags & CHANGE_DENSITY) != 0) {
|
|
clearCache();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Asynchronously fetches the icon and other task data.
|
|
*
|
|
* @param task The task to fetch the data for
|
|
* @param callback The callback to receive the task after its data has been populated.
|
|
* @return A cancelable handle to the request
|
|
*/
|
|
public CancellableTask updateIconInBackground(Task task, Consumer<Task> callback) {
|
|
Preconditions.assertUIThread();
|
|
if (task.icon != null) {
|
|
// Nothing to load, the icon is already loaded
|
|
callback.accept(task);
|
|
return null;
|
|
}
|
|
CancellableTask<TaskCacheEntry> request = new CancellableTask<TaskCacheEntry>() {
|
|
@Override
|
|
public TaskCacheEntry getResultOnBg() {
|
|
return getCacheEntry(task);
|
|
}
|
|
|
|
@Override
|
|
public void handleResult(TaskCacheEntry result) {
|
|
task.icon = result.icon;
|
|
task.titleDescription = result.contentDescription;
|
|
callback.accept(task);
|
|
}
|
|
};
|
|
mBgExecutor.execute(request);
|
|
return request;
|
|
}
|
|
|
|
/**
|
|
* Clears the icon cache
|
|
*/
|
|
public void clearCache() {
|
|
mBgExecutor.execute(this::resetFactory);
|
|
}
|
|
|
|
void onTaskRemoved(TaskKey taskKey) {
|
|
mIconCache.remove(taskKey);
|
|
}
|
|
|
|
void invalidateCacheEntries(String pkg, UserHandle handle) {
|
|
mBgExecutor.execute(() -> mIconCache.removeAll(key ->
|
|
pkg.equals(key.getPackageName()) && handle.getIdentifier() == key.userId));
|
|
}
|
|
|
|
@WorkerThread
|
|
private TaskCacheEntry getCacheEntry(Task task) {
|
|
TaskCacheEntry entry = mIconCache.getAndInvalidateIfModified(task.key);
|
|
if (entry != null) {
|
|
return entry;
|
|
}
|
|
|
|
TaskDescription desc = task.taskDescription;
|
|
TaskKey key = task.key;
|
|
ActivityInfo activityInfo = null;
|
|
|
|
// Create new cache entry
|
|
entry = new TaskCacheEntry();
|
|
|
|
// Load icon
|
|
// TODO: Load icon resource (b/143363444)
|
|
Bitmap icon = getIcon(desc, key.userId);
|
|
if (icon != null) {
|
|
entry.icon = getBitmapInfo(
|
|
new BitmapDrawable(mContext.getResources(), icon),
|
|
key.userId,
|
|
desc.getPrimaryColor(),
|
|
false /* isInstantApp */).newIcon(mContext);
|
|
} else {
|
|
activityInfo = PackageManagerWrapper.getInstance().getActivityInfo(
|
|
key.getComponent(), key.userId);
|
|
if (activityInfo != null) {
|
|
BitmapInfo bitmapInfo = getBitmapInfo(
|
|
mIconProvider.getIcon(activityInfo),
|
|
key.userId,
|
|
desc.getPrimaryColor(),
|
|
activityInfo.applicationInfo.isInstantApp());
|
|
entry.icon = bitmapInfo.newIcon(mContext);
|
|
} else {
|
|
entry.icon = getDefaultIcon(key.userId);
|
|
}
|
|
}
|
|
|
|
// Loading content descriptions if accessibility or low RAM recents is enabled.
|
|
if (GO_LOW_RAM_RECENTS_ENABLED || mAccessibilityManager.isEnabled()) {
|
|
// Skip loading the content description if the activity no longer exists
|
|
if (activityInfo == null) {
|
|
activityInfo = PackageManagerWrapper.getInstance().getActivityInfo(
|
|
key.getComponent(), key.userId);
|
|
}
|
|
if (activityInfo != null) {
|
|
entry.contentDescription = getBadgedContentDescription(
|
|
activityInfo, task.key.userId, task.taskDescription);
|
|
}
|
|
}
|
|
|
|
mIconCache.put(task.key, entry);
|
|
return entry;
|
|
}
|
|
|
|
private Bitmap getIcon(ActivityManager.TaskDescription desc, int userId) {
|
|
if (desc.getInMemoryIcon() != null) {
|
|
return desc.getInMemoryIcon();
|
|
}
|
|
return ActivityManager.TaskDescription.loadTaskDescriptionIcon(
|
|
desc.getIconFilename(), userId);
|
|
}
|
|
|
|
private String getBadgedContentDescription(ActivityInfo info, int userId, TaskDescription td) {
|
|
PackageManager pm = mContext.getPackageManager();
|
|
String taskLabel = td == null ? null : Utilities.trim(td.getLabel());
|
|
if (TextUtils.isEmpty(taskLabel)) {
|
|
taskLabel = Utilities.trim(info.loadLabel(pm));
|
|
}
|
|
|
|
String applicationLabel = Utilities.trim(info.applicationInfo.loadLabel(pm));
|
|
String badgedApplicationLabel = userId != UserHandle.myUserId()
|
|
? pm.getUserBadgedLabel(applicationLabel, UserHandle.of(userId)).toString()
|
|
: applicationLabel;
|
|
return applicationLabel.equals(taskLabel)
|
|
? badgedApplicationLabel : badgedApplicationLabel + " " + taskLabel;
|
|
}
|
|
|
|
@WorkerThread
|
|
private Drawable getDefaultIcon(int userId) {
|
|
synchronized (mDefaultIcons) {
|
|
if (mDefaultIconBase == null) {
|
|
try (BaseIconFactory bif = getIconFactory()) {
|
|
mDefaultIconBase = bif.makeDefaultIcon();
|
|
}
|
|
}
|
|
|
|
int index;
|
|
if ((index = mDefaultIcons.indexOfKey(userId)) >= 0) {
|
|
return mDefaultIcons.valueAt(index).newIcon(mContext);
|
|
} else {
|
|
try (BaseIconFactory li = getIconFactory()) {
|
|
BitmapInfo info = mDefaultIconBase.withFlags(
|
|
li.getBitmapFlagOp(new IconOptions().setUser(UserHandle.of(userId))));
|
|
mDefaultIcons.put(userId, info);
|
|
return info.newIcon(mContext);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
@WorkerThread
|
|
private BitmapInfo getBitmapInfo(Drawable drawable, int userId,
|
|
int primaryColor, boolean isInstantApp) {
|
|
try (BaseIconFactory bif = getIconFactory()) {
|
|
bif.disableColorExtraction();
|
|
bif.setWrapperBackgroundColor(primaryColor);
|
|
|
|
// User version code O, so that the icon is always wrapped in an adaptive icon container
|
|
return bif.createBadgedIconBitmap(drawable,
|
|
new IconOptions().setUser(UserHandle.of(userId)).setInstantApp(isInstantApp));
|
|
}
|
|
}
|
|
|
|
@WorkerThread
|
|
private BaseIconFactory getIconFactory() {
|
|
if (mIconFactory == null) {
|
|
mIconFactory = new BaseIconFactory(mContext,
|
|
DisplayController.INSTANCE.get(mContext).getInfo().densityDpi,
|
|
mContext.getResources().getDimensionPixelSize(R.dimen.taskbar_icon_size));
|
|
}
|
|
return mIconFactory;
|
|
}
|
|
|
|
@WorkerThread
|
|
private void resetFactory() {
|
|
mIconFactory = null;
|
|
mIconCache.evictAll();
|
|
}
|
|
|
|
private static class TaskCacheEntry {
|
|
public Drawable icon;
|
|
public String contentDescription = "";
|
|
}
|
|
}
|