src/third_party/webrtc/modules/desktop_capture/desktop_frame_win.cc

   1 /*
   2  *  Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
   3  *
   4  *  Use of this source code is governed by a BSD-style license
   5  *  that can be found in the LICENSE file in the root of the source
   6  *  tree. An additional intellectual property rights grant can be found
   7  *  in the file PATENTS.  All contributing project authors may
   8  *  be found in the AUTHORS file in the root of the source tree.
   9  */
  10
  11 #include "webrtc/modules/desktop_capture/desktop_frame_win.h"
  12
  13 #include "webrtc/system_wrappers/interface/logging.h"
  14
  15 namespace webrtc {
  16
  17 DesktopFrameWin::DesktopFrameWin(DesktopSize size,
  18                                  int stride,
  19                                  uint8_t* data,
  20                                  SharedMemory* shared_memory,
  21                                  HBITMAP bitmap)
  22     : DesktopFrame(size, stride, data, shared_memory),
  23       bitmap_(bitmap),
  24       owned_shared_memory_(shared_memory_) {
  25 }
  26
  27 DesktopFrameWin::~DesktopFrameWin() {
  28   DeleteObject(bitmap_);
  29 }
  30
  31 // static
  32 DesktopFrameWin* DesktopFrameWin::Create(DesktopSize size,
  33                                          SharedMemory* shared_memory,
  34                                          HDC hdc) {
  35   int bytes_per_row = size.width() * kBytesPerPixel;
  36
  37   // Describe a device independent bitmap (DIB) that is the size of the desktop.
  38   BITMAPINFO bmi = {0};
  39   bmi.bmiHeader.biHeight = -size.height();
  40   bmi.bmiHeader.biWidth = size.width();
  41   bmi.bmiHeader.biPlanes = 1;
  42   bmi.bmiHeader.biBitCount = DesktopFrameWin::kBytesPerPixel * 8;
  43   bmi.bmiHeader.biSize = sizeof(bmi.bmiHeader);
  44   bmi.bmiHeader.biSizeImage = bytes_per_row * size.height();
  45
  46   HANDLE section_handle = NULL;
  47   if (shared_memory)
  48     section_handle = shared_memory->handle();
  49   void* data = NULL;
  50   HBITMAP bitmap = CreateDIBSection(hdc, &bmi, DIB_RGB_COLORS, &data,
  51                                     section_handle, 0);
  52   if (!bitmap) {
  53     LOG(LS_WARNING) << "Failed to allocate new window frame " << GetLastError();
  54     delete shared_memory;
  55     return NULL;
  56   }
  57
  58   return new DesktopFrameWin(size, bytes_per_row,
  59                              reinterpret_cast<uint8_t*>(data),
  60                              shared_memory, bitmap);
  61 }
  62
  63 }  // namespace webrtc