Add CombinedDesktopMediaList class

chrome/browser/media/combined_desktop_media_list.cc

Index: chrome/browser/media/combined_desktop_media_list.cc
diff --git a/chrome/browser/media/combined_desktop_media_list.cc b/chrome/browser/media/combined_desktop_media_list.cc
new file mode 100644
index 0000000000000000000000000000000000000000..633af0550b490e36608f645103c0a3fa13d7ee8e
--- /dev/null
+++ b/chrome/browser/media/combined_desktop_media_list.cc
@@ -0,0 +1,130 @@
+// Copyright 2016 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(
+    std::vector<scoped_ptr<DesktopMediaList>>& media_lists)
+    : media_lists_(std::move(media_lists)) {
+  DCHECK(media_lists_.size());
+}
+
+CombinedDesktopMediaList::~CombinedDesktopMediaList() {}
+
+void CombinedDesktopMediaList::SetUpdatePeriod(base::TimeDelta period) {
+  for (const auto& list : media_lists_)
+    list->SetUpdatePeriod(period);
+}
+
+void CombinedDesktopMediaList::SetThumbnailSize(
+    const gfx::Size& thumbnail_size) {
+  for (const auto& list : media_lists_)
+    list->SetThumbnailSize(thumbnail_size);
+}
+
+void CombinedDesktopMediaList::SetViewDialogWindowId(
+    content::DesktopMediaID dialog_id) {
+  for (const auto& list : media_lists_)
+    list->SetViewDialogWindowId(dialog_id);
+}
+
+void CombinedDesktopMediaList::StartUpdating(
+    DesktopMediaListObserver* observer) {
+  observer_ = observer;
+  for (const auto& list : media_lists_)
+    list->StartUpdating(this);
+}
+
+int CombinedDesktopMediaList::GetSourceCount() const {
+  int count = 0;
+  for (const auto& list : media_lists_)
+    count += list->GetSourceCount();
+
+  return count;
+}
+
+const DesktopMediaList::Source& CombinedDesktopMediaList::GetSource(
+    int index) const {
+  for (const auto& list : media_lists_) {
+    int count = list->GetSourceCount();
+    if (index < count)
+      return list->GetSource(index);
+
+    index -= count;
+  }
+
+  NOTREACHED();
+  return media_lists_[0]->GetSource(0);
+}
+
+void CombinedDesktopMediaList::OnSourceAdded(DesktopMediaList* list,
+                                             int index) {
+  int count = 0;
+  for (const auto& cur_list : media_lists_) {
+    if (cur_list.get() == list) {
+      observer_->OnSourceAdded(this, count + index);
+      return;
+    }
+    count += cur_list->GetSourceCount();
+  }
+
+  NOTREACHED();
+}
+
+void CombinedDesktopMediaList::OnSourceRemoved(DesktopMediaList* list,
+                                               int index) {
+  int count = 0;
+  for (const auto& cur_list : media_lists_) {
+    if (cur_list.get() == list) {
+      observer_->OnSourceRemoved(this, count + index);
+      return;
+    }
+    count += cur_list->GetSourceCount();
+  }
+
+  NOTREACHED();
+}
+
+void CombinedDesktopMediaList::OnSourceMoved(DesktopMediaList* list,
+                                             int old_index,
+                                             int new_index) {
+  int count = 0;
+  for (const auto& cur_list : media_lists_) {
+    if (cur_list.get() == list) {
+      observer_->OnSourceMoved(this, count + old_index, count + new_index);
+      return;
+    }
+    count += cur_list->GetSourceCount();
+  }
+
+  NOTREACHED();
+}
+
+void CombinedDesktopMediaList::OnSourceNameChanged(DesktopMediaList* list,
+                                                   int index) {
+  int count = 0;
+  for (const auto& cur_list : media_lists_) {
+    if (cur_list.get() == list) {
+      observer_->OnSourceNameChanged(this, count + index);
+      return;
+    }
+    count += cur_list->GetSourceCount();
+  }
+
+  NOTREACHED();
+}
+
+void CombinedDesktopMediaList::OnSourceThumbnailChanged(DesktopMediaList* list,
+                                                        int index) {
+  int count = 0;
+  for (const auto& cur_list : media_lists_) {
+    if (cur_list.get() == list) {
+      observer_->OnSourceThumbnailChanged(this, count + index);
+      return;
+    }
+    count += cur_list->GetSourceCount();
+  }
+
+  NOTREACHED();
+}