Examples with ComponentKeyMapper com.android.launcher3.appprediction.ComponentKeyMapper used on opensource projects

Example 1 with ComponentKeyMapper

use of com.android.launcher3.appprediction.ComponentKeyMapper in project android_packages_apps_Trebuchet by LineageOS.

the class HotseatPredictionController method mapToWorkspaceItemInfo.

private List<WorkspaceItemInfo> mapToWorkspaceItemInfo(List<ComponentKeyMapper> components) {
    AllAppsStore allAppsStore = mLauncher.getAppsView().getAppsStore();
    if (allAppsStore.getApps().length == 0) {
        return Collections.emptyList();
    }
    List<WorkspaceItemInfo> predictedApps = new ArrayList<>();
    for (ComponentKeyMapper mapper : components) {
        ItemInfoWithIcon info = mapper.getApp(allAppsStore);
        if (info instanceof AppInfo) {
            WorkspaceItemInfo predictedApp = new WorkspaceItemInfo((AppInfo) info);
            predictedApp.container = LauncherSettings.Favorites.CONTAINER_HOTSEAT_PREDICTION;
            predictedApps.add(predictedApp);
        } else if (info instanceof WorkspaceItemInfo) {
            WorkspaceItemInfo predictedApp = new WorkspaceItemInfo((WorkspaceItemInfo) info);
            predictedApp.container = LauncherSettings.Favorites.CONTAINER_HOTSEAT_PREDICTION;
            predictedApps.add(predictedApp);
        } else {
            if (DEBUG) {
                Log.e(TAG, "Predicted app not found: " + mapper);
            }
        }
        // Stop at the number of hotseat items
        if (predictedApps.size() == mHotSeatItemsCount) {
            break;
        }
    }
    return predictedApps;
}

Example 2 with ComponentKeyMapper

use of com.android.launcher3.appprediction.ComponentKeyMapper in project android_packages_apps_Trebuchet by LineageOS.

the class HotseatPredictionController method setPredictedApps.

private void setPredictedApps(List<AppTarget> appTargets) {
    mComponentKeyMappers.clear();
    if (appTargets.isEmpty()) {
        mRestoreHelper.restoreBackup();
    }
    StringBuilder predictionLog = new StringBuilder("predictedApps: [\n");
    ArrayList<ComponentKey> componentKeys = new ArrayList<>();
    for (AppTarget appTarget : appTargets) {
        ComponentKey key;
        if (appTarget.getShortcutInfo() != null) {
            key = ShortcutKey.fromInfo(appTarget.getShortcutInfo());
        } else {
            key = new ComponentKey(new ComponentName(appTarget.getPackageName(), appTarget.getClassName()), appTarget.getUser());
        }
        componentKeys.add(key);
        predictionLog.append(key.toString());
        predictionLog.append(",rank:");
        predictionLog.append(appTarget.getRank());
        predictionLog.append("\n");
        mComponentKeyMappers.add(new ComponentKeyMapper(key, mDynamicItemCache));
    }
    predictionLog.append("]");
    if (Utilities.IS_DEBUG_DEVICE) {
        HotseatFileLog.INSTANCE.get(mLauncher).log(TAG, predictionLog.toString());
    }
    updateDependencies();
    fillGapsWithPrediction();
    mPredictionModel.cachePredictionComponentKeys(componentKeys);
}

Example 3 with ComponentKeyMapper

use of com.android.launcher3.appprediction.ComponentKeyMapper in project android_packages_apps_Trebuchet by LineageOS.

the class HotseatPredictionController method encodeHotseatLayoutIntoPredictionRank.

/**
 * Fill in predicted_rank field based on app prediction.
 * Only applicable when {@link ItemInfo#itemType} is PREDICTED_HOTSEAT
 */
public static void encodeHotseatLayoutIntoPredictionRank(@NonNull ItemInfo itemInfo, @NonNull LauncherLogProto.Target target) {
    QuickstepLauncher launcher = QuickstepLauncher.ACTIVITY_TRACKER.getCreatedActivity();
    if (launcher == null || launcher.getHotseatPredictionController() == null || itemInfo.getTargetComponent() == null) {
        return;
    }
    HotseatPredictionController controller = launcher.getHotseatPredictionController();
    final ComponentKey k = new ComponentKey(itemInfo.getTargetComponent(), itemInfo.user);
    final List<ComponentKeyMapper> predictedApps = controller.mComponentKeyMappers;
    OptionalInt rank = IntStream.range(0, predictedApps.size()).filter((i) -> k.equals(predictedApps.get(i).getComponentKey())).findFirst();
    target.predictedRank = 10000 + (controller.mPredictedSpotsCount * 100) + (rank.isPresent() ? rank.getAsInt() + 1 : 0);
}

Example 4 with ComponentKeyMapper

use of com.android.launcher3.appprediction.ComponentKeyMapper in project android_packages_apps_Trebuchet by LineageOS.

the class HotseatPredictionController method showCachedItems.

/**
 * Create WorkspaceItemInfo objects and binds PredictedAppIcon views for cached predicted items.
 */
public void showCachedItems(List<AppInfo> apps, IntArray ranks) {
    if (hasPredictions() && mAppPredictor != null) {
        mAppPredictor.requestPredictionUpdate();
        fillGapsWithPrediction();
        return;
    }
    int count = Math.min(ranks.size(), apps.size());
    List<WorkspaceItemInfo> items = new ArrayList<>(count);
    for (int i = 0; i < count; i++) {
        WorkspaceItemInfo item = new WorkspaceItemInfo(apps.get(i));
        ComponentKey componentKey = new ComponentKey(item.getTargetComponent(), item.user);
        preparePredictionInfo(item, ranks.get(i));
        items.add(item);
        mComponentKeyMappers.add(new ComponentKeyMapper(componentKey, mDynamicItemCache));
    }
    updateDependencies();
    bindItems(items, false, null);
}

Example 5 with ComponentKeyMapper

use of com.android.launcher3.appprediction.ComponentKeyMapper in project android_packages_apps_Trebuchet by LineageOS.

the class PredictionUiStateManager method fillInPredictedRank.

/**
 * Fill in predicted_rank field based on app prediction.
 * Only applicable when {@link ItemInfo#itemType} is one of the followings:
 * {@link LauncherSettings.Favorites#ITEM_TYPE_APPLICATION},
 * {@link LauncherSettings.Favorites#ITEM_TYPE_SHORTCUT},
 * {@link LauncherSettings.Favorites#ITEM_TYPE_DEEP_SHORTCUT}
 */
public static void fillInPredictedRank(@NonNull ItemInfo itemInfo, @NonNull LauncherLogProto.Target target) {
    final PredictionUiStateManager manager = PredictionUiStateManager.INSTANCE.getNoCreate();
    if (manager == null || itemInfo.getTargetComponent() == null || itemInfo.user == null || (itemInfo.itemType != LauncherSettings.Favorites.ITEM_TYPE_APPLICATION && itemInfo.itemType != LauncherSettings.Favorites.ITEM_TYPE_SHORTCUT && itemInfo.itemType != LauncherSettings.Favorites.ITEM_TYPE_DEEP_SHORTCUT)) {
        return;
    }
    if (itemInfo.container != LauncherSettings.Favorites.CONTAINER_PREDICTION) {
        HotseatPredictionController.encodeHotseatLayoutIntoPredictionRank(itemInfo, target);
        return;
    }
    final ComponentKey k = new ComponentKey(itemInfo.getTargetComponent(), itemInfo.user);
    final List<ComponentKeyMapper> predictedApps = manager.getCurrentState().apps;
    IntStream.range(0, predictedApps.size()).filter((i) -> k.equals(predictedApps.get(i).getComponentKey())).findFirst().ifPresent((rank) -> target.predictedRank = 0 - rank);
}