Icon Capture For Window Capturer

webrtc/modules/desktop_capture/window_capturer_win.cc

 /*
  *  Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
 #include "webrtc/modules/desktop_capture/window_capturer.h"
 
 #include <assert.h>
 
 #include <memory>
 
 #include "webrtc/base/checks.h"
 #include "webrtc/base/constructormagic.h"
 #include "webrtc/base/win32.h"
 #include "webrtc/modules/desktop_capture/desktop_frame_win.h"
 #include "webrtc/modules/desktop_capture/win/window_capture_utils.h"
 #include "webrtc/system_wrappers/include/logging.h"
 
 namespace webrtc {
 
 namespace {
 
+void InitializeBitmapHeader(BITMAPV5HEADER* header, int width, int height) {
+  memset(header, 0, sizeof(BITMAPV5HEADER));
+  header->bV5Size = sizeof(BITMAPV5HEADER);
+
+  // Note that icons are created using top-down DIBs so we must negate the
+  // value used for the icon's height.
+  header->bV5Width = width;
+  header->bV5Height = -height;
+  header->bV5Planes = 1;
+  header->bV5Compression = BI_RGB;
+
+  // Initializing the bitmap format to 32 bit ARGB.
+  header->bV5BitCount = 32;
+  header->bV5RedMask = 0x00FF0000;
+  header->bV5GreenMask = 0x0000FF00;
+  header->bV5BlueMask = 0x000000FF;
+  header->bV5AlphaMask = 0xFF000000;
+
+  // Use the system color space.  The default value is LCS_CALIBRATED_RGB, which
+  // causes us to crash if we don't specify the approprite gammas, etc.  See
+  // <http://msdn.microsoft.com/en-us/library/ms536531(VS.85).aspx> and
+  // <http://b/1283121>.
+  header->bV5CSType = LCS_WINDOWS_COLOR_SPACE;
+
+  // Use a valid value for bV5Intent as 0 is not a valid one.
+  // <http://msdn.microsoft.com/en-us/library/dd183381(VS.85).aspx>
+  header->bV5Intent = LCS_GM_IMAGES;
+}
+
+std::unique_ptr<DesktopFrame> GetWindowIcon(HWND hwnd) {
+  HICON icon_handle =
+      reinterpret_cast<HICON>(SendMessage(hwnd, WM_GETICON, ICON_BIG, 0));
+  if (icon_handle == NULL)
+    icon_handle = (HICON)GetClassLongPtr(hwnd, GCLP_HICON);
+  if (icon_handle == NULL)
+    icon_handle =
+        reinterpret_cast<HICON>(SendMessage(hwnd, WM_GETICON, ICON_SMALL, 0));
+  if (icon_handle == NULL)
+    icon_handle =
+        reinterpret_cast<HICON>(SendMessage(hwnd, WM_GETICON, ICON_SMALL2, 0));
+  if (icon_handle == NULL)
+    icon_handle = (HICON)GetClassLongPtr(hwnd, GCLP_HICONSM);
+
+  if (icon_handle == NULL)
+    return nullptr;
+
+  ICONINFO icon_info;
+  ::GetIconInfo(icon_handle, &icon_info);
+
+  BITMAP bitmap_info = {0};
+  ::GetObject(icon_info.hbmMask, sizeof(bitmap_info), &bitmap_info);
+
+  std::unique_ptr<DesktopFrame> frame(new BasicDesktopFrame(
+      DesktopSize(bitmap_info.bmWidth, bitmap_info.bmHeight)));
+
+  BITMAPV5HEADER h;
+  InitializeBitmapHeader(&h, bitmap_info.bmWidth, bitmap_info.bmHeight);
+  HDC hdc = ::GetDC(NULL);
+  uint8_t* bits;
+  HBITMAP dib =
+      ::CreateDIBSection(hdc, reinterpret_cast<BITMAPINFO*>(&h), DIB_RGB_COLORS,
+                         reinterpret_cast<void**>(&bits), NULL, 0);
+  HDC dib_dc = CreateCompatibleDC(hdc);
+  ::SelectObject(dib_dc, dib);
+  ::DrawIconEx(dib_dc, 0, 0, icon_handle, bitmap_info.bmWidth,
+               bitmap_info.bmHeight, 0, NULL, DI_NORMAL);
+  for (int y = 0; y < bitmap_info.bmHeight; ++y) {
+    memcpy(frame->data() + frame->stride() * y,
+           bits + bitmap_info.bmWidth * DesktopFrame::kBytesPerPixel * y,
+           DesktopFrame::kBytesPerPixel * bitmap_info.bmWidth);
+  }
+  return frame;
+}
+
 BOOL CALLBACK WindowsEnumerationHandler(HWND hwnd, LPARAM param) {
   WindowCapturer::WindowList* list =
       reinterpret_cast<WindowCapturer::WindowList*>(param);
 
   // Skip windows that are invisible, minimized, have no title, or are owned,
   // unless they have the app window style set.
   int len = GetWindowTextLength(hwnd);
   HWND owner = GetWindow(hwnd, GW_OWNER);
   LONG exstyle = GetWindowLong(hwnd, GWL_EXSTYLE);
   if (len == 0 || IsIconic(hwnd) || !IsWindowVisible(hwnd) ||
@@ skipping 25 matching lines @@
   }
 
   WindowCapturer::Window window;
   window.id = reinterpret_cast<WindowCapturer::WindowId>(hwnd);
 
   const size_t kTitleLength = 500;
   WCHAR window_title[kTitleLength];
   // Truncate the title if it's longer than kTitleLength.
   GetWindowText(hwnd, window_title, kTitleLength);
   window.title = rtc::ToUtf8(window_title);
+  window.icon = GetWindowIcon(hwnd);
 
   // Skip windows when we failed to convert the title or it is empty.
   if (window.title.empty())
     return TRUE;
 
-  list->push_back(window);
+  list->push_back(std::move(window));
 
   return TRUE;
 }
 
 class WindowCapturerWin : public WindowCapturer {
  public:
   WindowCapturerWin();
   virtual ~WindowCapturerWin();
 
   // WindowCapturer interface.
@@ skipping 20 matching lines @@
   // are interleaved with Capture() calls.
   std::map<HWND, DesktopSize> window_size_map_;
 
   RTC_DISALLOW_COPY_AND_ASSIGN(WindowCapturerWin);
 };
 
 WindowCapturerWin::WindowCapturerWin() {}
 WindowCapturerWin::~WindowCapturerWin() {}
 
 bool WindowCapturerWin::GetWindowList(WindowList* windows) {
-  WindowList result;
-  LPARAM param = reinterpret_cast<LPARAM>(&result);
+  windows->clear();
+  LPARAM param = reinterpret_cast<LPARAM>(windows);
   if (!EnumWindows(&WindowsEnumerationHandler, param))
     return false;
-  windows->swap(result);
 
   std::map<HWND, DesktopSize> new_map;
   for (const auto& item : *windows) {
     HWND hwnd = reinterpret_cast<HWND>(item.id);
     new_map[hwnd] = window_size_map_[hwnd];
   }
   window_size_map_.swap(new_map);
 
   return true;
 }
@@ skipping 129 matching lines @@
 }
 
 }  // namespace
 
 // static
 WindowCapturer* WindowCapturer::Create(const DesktopCaptureOptions& options) {
   return new WindowCapturerWin();
 }
 
 }  // namespace webrtc