Capture chrome browser windows from internal rendering procedure for windows and linux

chrome/browser/media/native_desktop_media_list.h

 // Copyright 2013 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.
 
 #ifndef CHROME_BROWSER_MEDIA_NATIVE_DESKTOP_MEDIA_LIST_H_
 #define CHROME_BROWSER_MEDIA_NATIVE_DESKTOP_MEDIA_LIST_H_
 
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/weak_ptr.h"
 #include "base/sequenced_task_runner.h"
 #include "chrome/browser/media/desktop_media_list_base.h"
 #include "content/public/browser/desktop_media_id.h"
+#include "ui/gfx/image/image.h"
 
 namespace webrtc {
 class ScreenCapturer;
 class WindowCapturer;
 }
 
 // Implementation of DesktopMediaList that shows native screens and
 // native windows.
 class NativeDesktopMediaList : public DesktopMediaListBase {
  public:
   // Caller may pass NULL for either of the arguments in case when only some
   // types of sources the model should be populated with (e.g. it will only
   // contain windows, if |screen_capturer| is NULL).
   NativeDesktopMediaList(
       scoped_ptr<webrtc::ScreenCapturer> screen_capturer,
       scoped_ptr<webrtc::WindowCapturer> window_capturer);
   ~NativeDesktopMediaList() override;
 
  private:
+  typedef std::map<DesktopMediaID::Id, DesktopMediaID::Id> NativeAuraIdMap;
+  typedef std::map<DesktopMediaID, uint32_t> ImageHashesMap;
+
   class Worker;
   friend class Worker;
 
   // Refresh() posts a task for the |worker_| to update list of windows and get
   // thumbnails. Then |worker_| first posts tasks for
   // UpdateObserverSourcesList() with fresh list of sources, then follows with
   // OnSourceThumbnail() for each changed thumbnail and then calls
   // DelayLaunchNextRefersh() at the end.
   void Refresh() override;
-  void OnSourceThumbnail(int index, const gfx::ImageSkia& image);
+  void OnSourceThumbnailCaptured(int index, const gfx::ImageSkia& image);
+  void OnAuraThumbnailCaptured(DesktopMediaID id,
+                               int total_aura_windows,
+                               const gfx::Image& image);
 
   // Task runner used for the |worker_|.
   scoped_refptr<base::SequencedTaskRunner> capture_task_runner_;
 
   // An object that does all the work of getting list of sources on a background
   // thread (see |capture_task_runner_|). Destroyed on |capture_task_runner_|
   // after the model is destroyed.
   scoped_ptr<Worker> worker_;
 
+  int processed_aura_window_count_;
+  ImageHashesMap aura_thumbnail_hashes_;
+  ImageHashesMap cur_aura_thumbnail_hashes_;

[Sergey Ulanov, 2016/03/07 21:01:50]

call this new_aura_thumbnail_hashes_?
Also add comment to explain what they are used for and why you need 2 maps here.

[GeorgeZ, 2016/03/09 22:38:52]

Done.

+
+#if defined(USE_AURA)
+  NativeAuraIdMap GetBrowserNativeAuraIdMap();
+  void CaptureAuraWindowThumbnail(DesktopMediaID id, int total_aura_windows);
+#endif
+
   base::WeakPtrFactory<NativeDesktopMediaList> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(NativeDesktopMediaList);
 };
 
 #endif  // CHROME_BROWSER_MEDIA_NATIVE_DESKTOP_MEDIA_LIST_H_