summaryrefslogtreecommitdiff
path: root/quickstep/src/com/android/quickstep/util/LottieAnimationColorUtils.java
blob: 8833b46802c4f6abea7a46cd97388ab2471feaa0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
/*
 * Copyright (C) 2023 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.util;

import static com.airbnb.lottie.LottieProperty.COLOR_FILTER;

import android.content.res.Resources.Theme;
import android.graphics.PorterDuff;
import android.graphics.PorterDuffColorFilter;

import androidx.annotation.NonNull;

import com.airbnb.lottie.LottieAnimationView;
import com.airbnb.lottie.model.KeyPath;

import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

/** Utility class for programmatically updating Lottie animation tokenized colors. */
public final class LottieAnimationColorUtils {

    private LottieAnimationColorUtils() {}

    /**
     * Updates the given Lottie animation's tokenized colors according to the given mapping.
     * <p>
     * @param animationView {@link LottieAnimationView} whose animation's colors need to be updated
     * @param tokenToArgbColorMap A mapping from the color tokens used in the Lottie file used in
     *                            {@code animationView} to packed ARBG color integers.
     */
    public static void updateToArgbColors(
            @NonNull LottieAnimationView animationView,
            @NonNull Map<String, Integer> tokenToArgbColorMap) {
        animationView.addLottieOnCompositionLoadedListener(
                composition -> tokenToArgbColorMap.forEach(
                        (token, color) -> animationView.addValueCallback(
                                new KeyPath("**", token, "**"),
                                COLOR_FILTER,
                                frameInfo -> new PorterDuffColorFilter(
                                        color, PorterDuff.Mode.SRC_ATOP))));
    }

    /**
     * Updates the given Lottie animation's tokenized colors according to the given mapping.
     * <p>
     * @param animationView {@link LottieAnimationView} whose animation's colors need to be updated
     * @param tokenToColorResourceMap A mapping from the color tokens used in the Lottie file used
     *                                in {@code animationView} to color resource references.
     * @param theme {@link Theme} to be used when resolving color resource references.
     */
    public static void updateToColorResources(
            @NonNull LottieAnimationView animationView,
            @NonNull Map<String, Integer> tokenToColorResourceMap,
            @NonNull Theme theme) {
        updateToArgbColors(
                animationView,
                tokenToColorResourceMap.keySet().stream().collect(Collectors.toMap(
                        Function.identity(),
                        token -> animationView.getResources().getColor(
                                tokenToColorResourceMap.get(token), theme))));
    }
}