1 /*
2  * Copyright (C) 2017 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.tv.recommendation;
18 
19 import android.content.Context;
20 import android.os.Build;
21 import android.support.annotation.RequiresApi;
22 import android.text.TextUtils;
23 import android.util.Log;
24 import com.android.tv.TvSingletons;
25 import com.android.tv.data.PreviewDataManager;
26 import com.android.tv.data.PreviewProgramContent;
27 import com.android.tv.dvr.DvrDataManager;
28 import com.android.tv.dvr.data.RecordedProgram;
29 import java.util.ArrayList;
30 import java.util.Collections;
31 import java.util.HashSet;
32 import java.util.Set;
33 
34 /** Class to update the preview data for {@link RecordedProgram} */
35 @RequiresApi(Build.VERSION_CODES.O)
36 @SuppressWarnings("AndroidApiChecker") // TODO(b/32513850) remove when error prone is updated
37 public class RecordedProgramPreviewUpdater {
38     private static final String TAG = "RecordedProgramPreviewUpdater";
39     private static final boolean DEBUG = false;
40 
41     private static final int RECOMMENDATION_COUNT = 6;
42 
43     private static RecordedProgramPreviewUpdater sRecordedProgramPreviewUpdater;
44 
45     /** Creates and returns the {@link RecordedProgramPreviewUpdater}. */
getInstance(Context context)46     public static RecordedProgramPreviewUpdater getInstance(Context context) {
47         if (sRecordedProgramPreviewUpdater == null) {
48             sRecordedProgramPreviewUpdater =
49                     new RecordedProgramPreviewUpdater(context.getApplicationContext());
50         }
51         return sRecordedProgramPreviewUpdater;
52     }
53 
54     private final Context mContext;
55     private final PreviewDataManager mPreviewDataManager;
56     private final DvrDataManager mDvrDataManager;
57 
RecordedProgramPreviewUpdater(Context context)58     private RecordedProgramPreviewUpdater(Context context) {
59         mContext = context.getApplicationContext();
60         TvSingletons tvSingletons = TvSingletons.getSingletons(mContext);
61         mPreviewDataManager = tvSingletons.getPreviewDataManager();
62         mDvrDataManager = tvSingletons.getDvrDataManager();
63         mDvrDataManager.addRecordedProgramListener(
64                 new DvrDataManager.RecordedProgramListener() {
65                     @Override
66                     public void onRecordedProgramsAdded(RecordedProgram... recordedPrograms) {
67                         if (DEBUG) Log.d(TAG, "Add new preview recorded programs");
68                         updatePreviewDataForRecordedPrograms();
69                     }
70 
71                     @Override
72                     public void onRecordedProgramsChanged(RecordedProgram... recordedPrograms) {
73                         if (DEBUG) Log.d(TAG, "Update preview recorded programs");
74                         updatePreviewDataForRecordedPrograms();
75                     }
76 
77                     @Override
78                     public void onRecordedProgramsRemoved(RecordedProgram... recordedPrograms) {
79                         if (DEBUG) Log.d(TAG, "Delete preview recorded programs");
80                         updatePreviewDataForRecordedPrograms();
81                     }
82                 });
83     }
84 
85     /** Updates the preview data for recorded programs. */
updatePreviewDataForRecordedPrograms()86     public void updatePreviewDataForRecordedPrograms() {
87         if (!mPreviewDataManager.isLoadFinished()) {
88             mPreviewDataManager.addListener(
89                     new PreviewDataManager.PreviewDataListener() {
90                         @Override
91                         public void onPreviewDataLoadFinished() {
92                             mPreviewDataManager.removeListener(this);
93                             updatePreviewDataForRecordedPrograms();
94                         }
95 
96                         @Override
97                         public void onPreviewDataUpdateFinished() {}
98                     });
99             return;
100         }
101         if (!mDvrDataManager.isRecordedProgramLoadFinished()) {
102             mDvrDataManager.addRecordedProgramLoadFinishedListener(
103                     new DvrDataManager.OnRecordedProgramLoadFinishedListener() {
104                         @Override
105                         public void onRecordedProgramLoadFinished() {
106                             mDvrDataManager.removeRecordedProgramLoadFinishedListener(this);
107                             updatePreviewDataForRecordedPrograms();
108                         }
109                     });
110             return;
111         }
112         updatePreviewDataForRecordedProgramsInternal();
113     }
114 
updatePreviewDataForRecordedProgramsInternal()115     private void updatePreviewDataForRecordedProgramsInternal() {
116         Set<RecordedProgram> recordedPrograms = generateRecommendationRecordedPrograms();
117         Long recordedPreviewChannelId =
118                 mPreviewDataManager.getPreviewChannelId(
119                         PreviewDataManager.TYPE_RECORDED_PROGRAM_PREVIEW_CHANNEL);
120         if (recordedPreviewChannelId == PreviewDataManager.INVALID_PREVIEW_CHANNEL_ID
121                 && !recordedPrograms.isEmpty()) {
122             createPreviewChannelForRecordedPrograms();
123         } else {
124             mPreviewDataManager.updatePreviewProgramsForChannel(
125                     recordedPreviewChannelId,
126                     generatePreviewProgramContentsFromRecordedPrograms(
127                             recordedPreviewChannelId, recordedPrograms),
128                     null);
129         }
130     }
131 
createPreviewChannelForRecordedPrograms()132     private void createPreviewChannelForRecordedPrograms() {
133         mPreviewDataManager.createPreviewChannel(
134                 PreviewDataManager.TYPE_RECORDED_PROGRAM_PREVIEW_CHANNEL,
135                 new PreviewDataManager.OnPreviewChannelCreationResultListener() {
136                     @Override
137                     public void onPreviewChannelCreationResult(long createdPreviewChannelId) {
138                         if (createdPreviewChannelId
139                                 != PreviewDataManager.INVALID_PREVIEW_CHANNEL_ID) {
140                             updatePreviewDataForRecordedProgramsInternal();
141                         }
142                     }
143                 });
144     }
145 
generateRecommendationRecordedPrograms()146     private Set<RecordedProgram> generateRecommendationRecordedPrograms() {
147         Set<RecordedProgram> programs = new HashSet<>();
148         ArrayList<RecordedProgram> sortedRecordedPrograms =
149                 new ArrayList<>(mDvrDataManager.getRecordedPrograms());
150         Collections.sort(
151                 sortedRecordedPrograms, RecordedProgram.START_TIME_THEN_ID_COMPARATOR.reversed());
152         for (RecordedProgram recordedProgram : sortedRecordedPrograms) {
153             if (!TextUtils.isEmpty(recordedProgram.getPosterArtUri())) {
154                 programs.add(recordedProgram);
155                 if (programs.size() >= RECOMMENDATION_COUNT) {
156                     break;
157                 }
158             }
159         }
160         return programs;
161     }
162 
generatePreviewProgramContentsFromRecordedPrograms( long previewChannelId, Set<RecordedProgram> recordedPrograms)163     private Set<PreviewProgramContent> generatePreviewProgramContentsFromRecordedPrograms(
164             long previewChannelId, Set<RecordedProgram> recordedPrograms) {
165         Set<PreviewProgramContent> result = new HashSet<>();
166         for (RecordedProgram recordedProgram : recordedPrograms) {
167             result.add(
168                     PreviewProgramContent.createFromRecordedProgram(
169                             mContext, previewChannelId, recordedProgram));
170         }
171         return result;
172     }
173 }
174