1 files changed, 186 insertions, 0 deletions
diff --git a/chromium/chrome/browser/media/webrtc/desktop_media_list_base.cc b/chromium/chrome/browser/media/webrtc/desktop_media_list_base.cc
new file mode 100644
index 00000000000..9d9b7435044
--- /dev/null
+++ b/chromium/chrome/browser/media/webrtc/desktop_media_list_base.cc
@@ -0,0 +1,186 @@
+// 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/webrtc/desktop_media_list_base.h"
+
+#include <set>
+#include <utility>
+
+#include "base/bind.h"
+#include "base/task/post_task.h"
+#include "chrome/browser/media/webrtc/desktop_media_list.h"
+#include "content/public/browser/browser_task_traits.h"
+#include "content/public/browser/browser_thread.h"
+#include "ui/gfx/image/image.h"
+
+using content::BrowserThread;
+using content::DesktopMediaID;
+
+DesktopMediaListBase::DesktopMediaListBase(base::TimeDelta update_period)
+    : update_period_(update_period) {}
+
+DesktopMediaListBase::~DesktopMediaListBase() {}
+
+void DesktopMediaListBase::SetUpdatePeriod(base::TimeDelta period) {
+  DCHECK(!observer_);
+  update_period_ = period;
+}
+
+void DesktopMediaListBase::SetThumbnailSize(const gfx::Size& thumbnail_size) {
+  thumbnail_size_ = thumbnail_size;
+}
+
+void DesktopMediaListBase::SetViewDialogWindowId(DesktopMediaID dialog_id) {
+  view_dialog_id_ = dialog_id;
+}
+
+void DesktopMediaListBase::StartUpdating(DesktopMediaListObserver* observer) {
+  DCHECK(!observer_);
+  observer_ = observer;
+
+  // Process sources previously discovered by a call to Update().
+  if (observer_) {
+    for (size_t i = 0; i < sources_.size(); i++) {
+      observer_->OnSourceAdded(this, i);
+    }
+  }
+
+  DCHECK(!refresh_callback_);
+  refresh_callback_ = base::BindOnce(&DesktopMediaListBase::ScheduleNextRefresh,
+                                     weak_factory_.GetWeakPtr());
+  Refresh(true);
+}
+
+void DesktopMediaListBase::Update(UpdateCallback callback) {
+  DCHECK(sources_.empty());
+  DCHECK(!refresh_callback_);
+  refresh_callback_ = std::move(callback);
+  Refresh(false);
+}
+
+int DesktopMediaListBase::GetSourceCount() const {
+  return sources_.size();
+}
+
+const DesktopMediaList::Source& DesktopMediaListBase::GetSource(
+    int index) const {
+  DCHECK_GE(index, 0);
+  DCHECK_LT(index, static_cast<int>(sources_.size()));
+  return sources_[index];
+}
+
+DesktopMediaID::Type DesktopMediaListBase::GetMediaListType() const {
+  return type_;
+}
+
+DesktopMediaListBase::SourceDescription::SourceDescription(
+    DesktopMediaID id,
+    const base::string16& name)
+    : id(id), name(name) {}
+
+void DesktopMediaListBase::UpdateSourcesList(
+    const std::vector<SourceDescription>& new_sources) {
+  typedef std::set<DesktopMediaID> SourceSet;
+  SourceSet new_source_set;
+  for (size_t i = 0; i < new_sources.size(); ++i) {
+    new_source_set.insert(new_sources[i].id);
+  }
+  // Iterate through the old sources to find the removed sources.
+  for (size_t i = 0; i < sources_.size(); ++i) {
+    if (new_source_set.find(sources_[i].id) == new_source_set.end()) {
+      sources_.erase(sources_.begin() + i);
+      if (observer_)
+        observer_->OnSourceRemoved(this, i);
+      --i;
+    }
+  }
+  // Iterate through the new sources to find the added sources.
+  if (new_sources.size() > sources_.size()) {
+    SourceSet old_source_set;
+    for (size_t i = 0; i < sources_.size(); ++i) {
+      old_source_set.insert(sources_[i].id);
+    }
+
+    for (size_t i = 0; i < new_sources.size(); ++i) {
+      if (old_source_set.find(new_sources[i].id) == old_source_set.end()) {
+        sources_.insert(sources_.begin() + i, Source());
+        sources_[i].id = new_sources[i].id;
+        sources_[i].name = new_sources[i].name;
+        if (observer_)
+          observer_->OnSourceAdded(this, i);
+      }
+    }
+  }
+  DCHECK_EQ(new_sources.size(), sources_.size());
+
+  // Find the moved/changed sources.
+  size_t pos = 0;
+  while (pos < sources_.size()) {
+    if (!(sources_[pos].id == new_sources[pos].id)) {
+      // Find the source that should be moved to |pos|, starting from |pos + 1|
+      // of |sources_|, because entries before |pos| should have been sorted.
+      size_t old_pos = pos + 1;
+      for (; old_pos < sources_.size(); ++old_pos) {
+        if (sources_[old_pos].id == new_sources[pos].id)
+          break;
+      }
+      DCHECK(sources_[old_pos].id == new_sources[pos].id);
+
+      // Move the source from |old_pos| to |pos|.
+      Source temp = sources_[old_pos];
+      sources_.erase(sources_.begin() + old_pos);
+      sources_.insert(sources_.begin() + pos, temp);
+
+      if (observer_)
+        observer_->OnSourceMoved(this, old_pos, pos);
+    }
+
+    if (sources_[pos].name != new_sources[pos].name) {
+      sources_[pos].name = new_sources[pos].name;
+      if (observer_)
+        observer_->OnSourceNameChanged(this, pos);
+    }
+    ++pos;
+  }
+}
+
+void DesktopMediaListBase::UpdateSourceThumbnail(DesktopMediaID id,
+                                                 const gfx::ImageSkia& image) {
+  // Unlike other methods that check can_refresh(), this one won't cause
+  // OnRefreshComplete() to be called, but the caller is expected to schedule a
+  // call to OnRefreshComplete() after this method has been called as many times
+  // as needed, so the check is still valid.
+  DCHECK(can_refresh());
+
+  for (size_t i = 0; i < sources_.size(); ++i) {
+    if (sources_[i].id == id) {
+      sources_[i].thumbnail = image;
+      if (observer_)
+        observer_->OnSourceThumbnailChanged(this, i);
+      break;
+    }
+  }
+}
+
+// static
+uint32_t DesktopMediaListBase::GetImageHash(const gfx::Image& image) {
+  SkBitmap bitmap = image.AsBitmap();
+  uint32_t value = base::Hash(bitmap.getPixels(), bitmap.computeByteSize());
+  return value;
+}
+
+void DesktopMediaListBase::OnRefreshComplete() {
+  DCHECK(refresh_callback_);
+  std::move(refresh_callback_).Run();
+}
+
+void DesktopMediaListBase::ScheduleNextRefresh() {
+  DCHECK(!refresh_callback_);
+  refresh_callback_ = base::BindOnce(&DesktopMediaListBase::ScheduleNextRefresh,
+                                     weak_factory_.GetWeakPtr());
+  base::PostDelayedTask(FROM_HERE, {BrowserThread::UI},
+                        base::BindOnce(&DesktopMediaListBase::Refresh,
+                                       weak_factory_.GetWeakPtr(), true),
+                        update_period_);
+}