Add CombinedDesktopMediaList.

chrome/browser/media/combined_desktop_media_list.cc

+// Copyright 2015 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/media/combined_desktop_media_list.h"
+
+CombinedDesktopMediaList::CombinedDesktopMediaList(
+    scoped_ptr<DesktopMediaList> list1,
+    scoped_ptr<DesktopMediaList> list2)
+    : list1_(std::move(list1)), list2_(std::move(list2)) {}
+
+CombinedDesktopMediaList::~CombinedDesktopMediaList() {}
+
+void CombinedDesktopMediaList::SetUpdatePeriod(base::TimeDelta period) {
+  list1_->SetUpdatePeriod(period);
+  list2_->SetUpdatePeriod(period);
+}
+
+void CombinedDesktopMediaList::SetThumbnailSize(
+    const gfx::Size& thumbnail_size) {
+  list1_->SetThumbnailSize(thumbnail_size);
+  list2_->SetThumbnailSize(thumbnail_size);
+}
+
+void CombinedDesktopMediaList::SetViewDialogWindowId(
+    content::DesktopMediaID dialog_id) {
+  list1_->SetViewDialogWindowId(dialog_id);
+  list2_->SetViewDialogWindowId(dialog_id);
+}
+
+void CombinedDesktopMediaList::StartUpdating(
+    DesktopMediaListObserver* observer) {
+  observer_ = observer;
+  list1_->StartUpdating(this);
+  list2_->StartUpdating(this);
+}
+
+int CombinedDesktopMediaList::GetSourceCount() const {
+  return list1_->GetSourceCount() + list2_->GetSourceCount();
+}
+
+const DesktopMediaList::Source& CombinedDesktopMediaList::GetSource(
+    int index) const {
+  int list1_count = list1_->GetSourceCount();
+  return (index < list1_count) ? list1_->GetSource(index)
+                               : list2_->GetSource(index - list1_count);
+}
+
+void CombinedDesktopMediaList::OnSourceAdded(DesktopMediaList* list,
+                                             int index) {
+  observer_->OnSourceAdded(this, (list == list1_.get())
+                                     ? index
+                                     : (index + list1_->GetSourceCount()));
+}
+
+void CombinedDesktopMediaList::OnSourceRemoved(DesktopMediaList* list,
+                                               int index) {
+  observer_->OnSourceRemoved(this, (list == list1_.get())
+                                       ? index
+                                       : (index + list1_->GetSourceCount()));
+}
+
+void CombinedDesktopMediaList::OnSourceMoved(DesktopMediaList* list,
+                                             int old_index,
+                                             int new_index) {
+  if (list == list1_.get()) {
+    observer_->OnSourceMoved(this, old_index, new_index);
+  } else {
+    int list1_count = list1_->GetSourceCount();
+    observer_->OnSourceMoved(this, list1_count + old_index,
+                             list1_count + new_index);
+  }
+}
+
+void CombinedDesktopMediaList::OnSourceNameChanged(DesktopMediaList* list,
+                                                   int index) {
+  observer_->OnSourceNameChanged(
+      this,
+      (list == list1_.get()) ? index : (index + list1_->GetSourceCount()));
+}
+
+void CombinedDesktopMediaList::OnSourceThumbnailChanged(DesktopMediaList* list,
+                                                        int index) {
+  observer_->OnSourceThumbnailChanged(
+      this,
+      (list == list1_.get()) ? index : (index + list1_->GetSourceCount()));
+}