1 /* 2 * Copyright (C) 2020 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 package com.android.launcher3.icons; 17 18 import android.content.Context; 19 import android.graphics.drawable.Drawable; 20 21 import androidx.annotation.NonNull; 22 23 import com.android.launcher3.LauncherAppState; 24 25 /** 26 * Extension of ComponentWithLabel to also support loading icons 27 */ 28 public interface ComponentWithLabelAndIcon extends ComponentWithLabel { 29 30 /** 31 * Provide an icon for this object 32 */ getFullResIcon(IconCache cache)33 Drawable getFullResIcon(IconCache cache); 34 35 class ComponentWithIconCachingLogic extends ComponentCachingLogic<ComponentWithLabelAndIcon> { 36 ComponentWithIconCachingLogic(Context context, boolean addToMemCache)37 public ComponentWithIconCachingLogic(Context context, boolean addToMemCache) { 38 super(context, addToMemCache); 39 } 40 41 @NonNull 42 @Override loadIcon(Context context, ComponentWithLabelAndIcon object)43 public BitmapInfo loadIcon(Context context, ComponentWithLabelAndIcon object) { 44 Drawable d = object.getFullResIcon(LauncherAppState.getInstance(context) 45 .getIconCache()); 46 if (d == null) { 47 return super.loadIcon(context, object); 48 } 49 try (LauncherIcons li = LauncherIcons.obtain(context)) { 50 return li.createBadgedIconBitmap(d, object.getUser(), 0); 51 } 52 } 53 } 54 } 55