Sync to ub-tv-heroes at 1.08.301

source change id If9b64d7bbc6e8f77b360e502d34e5452775c0402

Change-Id: I4ffe87911cb85e54880d1d918d1b8fb7bb8cfb7d

1 files changed, 154 insertions, 0 deletions

diff --git a/src/com/android/tv/dvr/DvrDataManagerImpl.java b/src/com/android/tv/dvr/DvrDataManagerImpl.java
new file mode 100644
index 00000000..d1c590af
--- /dev/null
+++ b/src/com/android/tv/dvr/DvrDataManagerImpl.java
@@ -0,0 +1,154 @@
+/*
+ * Copyright (C) 2015 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.tv.dvr;
+
+import android.content.Context;
+import android.support.annotation.VisibleForTesting;
+import android.util.Range;
+
+import com.android.tv.dvr.Recording.RecordingState;
+import com.android.tv.dvr.provider.AsyncDvrDbTask.AsyncDvrQueryTask;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+/**
+ * DVR Data manager to handle recordings and schedules.
+ */
+public class DvrDataManagerImpl extends BaseDvrDataManager {
+    private Context mContext;
+    private boolean mLoadFinished;
+    private final List<Recording> mRecordings = new ArrayList<>();
+    private AsyncDvrQueryTask mQueryTask;
+
+    public DvrDataManagerImpl(Context context) {
+        super(context);
+        mContext = context;
+    }
+
+    public void start() {
+        mQueryTask = new AsyncDvrQueryTask(mContext) {
+            @Override
+            protected void onPostExecute(List<Recording> result) {
+                mQueryTask = null;
+                mLoadFinished = true;
+                mRecordings.addAll(result);
+                Collections.sort(mRecordings, Recording.START_TIME_COMPARATOR);
+            }
+        };
+        mQueryTask.executeOnDbThread();
+    }
+
+    public void stop() {
+        if (mQueryTask != null) {
+            mQueryTask.cancel(true);
+            mQueryTask = null;
+        }
+    }
+
+    @Override
+    public boolean isInitialized() {
+        return mLoadFinished;
+    }
+
+    @Override
+    public List<Recording> getRecordings() {
+        if (!mLoadFinished) {
+            return Collections.emptyList();
+        }
+        return Collections.unmodifiableList(mRecordings);
+    }
+
+    @Override
+    public List<Recording> getFinishedRecordings() {
+        return getRecordingsWithState(Recording.STATE_RECORDING_FINISHED);
+    }
+
+    @Override
+    public List<Recording> getStartedRecordings() {
+        return getRecordingsWithState(Recording.STATE_RECORDING_IN_PROGRESS);
+    }
+
+    @Override
+    public List<Recording> getScheduledRecordings() {
+        return getRecordingsWithState(Recording.STATE_RECORDING_NOT_STARTED);
+    }
+
+    private List<Recording> getRecordingsWithState(@RecordingState int state) {
+        List<Recording> result = new ArrayList<>();
+        for (Recording r : mRecordings) {
+            if (r.getState() == state) {
+                result.add(r);
+            }
+        }
+        return result;
+    }
+
+    @Override
+    public List<SeasonRecording> getSeasonRecordings() {
+        // If we return dummy data here, we can implement UI part independently.
+        return Collections.emptyList();
+    }
+
+    @Override
+    public long getNextScheduledStartTimeAfter(long startTime) {
+        return getNextStartTimeAfter(mRecordings, startTime);
+    }
+
+    @VisibleForTesting
+    static long getNextStartTimeAfter(List<Recording> recordings, long startTime) {
+        int start = 0;
+        int end = recordings.size() - 1;
+        while (start <= end) {
+            int mid = (start + end) / 2;
+            if (recordings.get(mid).getStartTimeMs() <= startTime) {
+                start = mid + 1;
+            } else {
+                end = mid - 1;
+            }
+        }
+        return start < recordings.size() ? recordings.get(start).getStartTimeMs()
+                : NEXT_START_TIME_NOT_FOUND;
+    }
+
+    @Override
+    public List<Recording> getRecordingsThatOverlapWith(Range<Long> period) {
+        List<Recording> result = new ArrayList<>();
+        for (Recording r : mRecordings) {
+            if (r.isOverLapping(period)) {
+                result.add(r);
+            }
+        }
+        return result;
+    }
+
+    @Override
+    public void addRecording(Recording recording) { }
+
+    @Override
+    public void addSeasonRecording(SeasonRecording seasonRecording) { }
+
+    @Override
+    public void removeRecording(Recording recording) { }
+
+    @Override
+    public void removeSeasonSchedule(SeasonRecording seasonSchedule) { }
+
+    @Override
+    public void updateRecording(Recording r) { }
+}