1 /*
2  * Copyright (C) 2022 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.systemui.complication;
18 
19 import com.android.systemui.complication.dagger.ComplicationViewModelComponent;
20 
21 import java.util.HashMap;
22 
23 import javax.inject.Inject;
24 
25 /**
26  * The {@link ComplicationViewModelTransformer} responsibility is provide a mapping from
27  * {@link Complication} to {@link ComplicationViewModel}.
28  */
29 public class ComplicationViewModelTransformer {
30     private final ComplicationId.Factory mComplicationIdFactory = new ComplicationId.Factory();
31     private final HashMap<Complication, ComplicationId> mComplicationIdMapping = new HashMap<>();
32     private final ComplicationViewModelComponent.Factory mViewModelComponentFactory;
33 
34     @Inject
ComplicationViewModelTransformer( ComplicationViewModelComponent.Factory viewModelComponentFactory)35     public ComplicationViewModelTransformer(
36             ComplicationViewModelComponent.Factory viewModelComponentFactory) {
37         mViewModelComponentFactory = viewModelComponentFactory;
38     }
39 
40     /**
41      * Generates {@link ComplicationViewModel} from a {@link Complication}.
42      */
getViewModel(Complication complication)43     public ComplicationViewModel getViewModel(Complication complication) {
44         final ComplicationId id = getComplicationId(complication);
45         return mViewModelComponentFactory.create(complication, id)
46                 .getViewModelProvider().get(id.toString(), ComplicationViewModel.class);
47     }
48 
getComplicationId(Complication complication)49     private ComplicationId getComplicationId(Complication complication) {
50         if (!mComplicationIdMapping.containsKey(complication)) {
51             mComplicationIdMapping.put(complication, mComplicationIdFactory.getNextId());
52         }
53 
54         return mComplicationIdMapping.get(complication);
55     }
56 }
57