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/include/logging.h"
14 
15 namespace webrtc {
16 
DesktopFrameWin(DesktopSize size,int stride,uint8_t * data,SharedMemory * shared_memory,HBITMAP bitmap)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 
~DesktopFrameWin()27 DesktopFrameWin::~DesktopFrameWin() {
28   DeleteObject(bitmap_);
29 }
30 
31 // static
Create(DesktopSize size,SharedMemory * shared_memory,HDC hdc)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 = {};
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
64