1 /*
2  *  Copyright (c) 2016 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 #ifndef MODULES_DESKTOP_CAPTURE_WIN_DXGI_TEXTURE_H_
12 #define MODULES_DESKTOP_CAPTURE_WIN_DXGI_TEXTURE_H_
13 
14 #include <D3D11.h>
15 #include <DXGI1_2.h>
16 
17 #include <memory>
18 
19 #include "modules/desktop_capture/desktop_frame.h"
20 #include "modules/desktop_capture/desktop_geometry.h"
21 
22 namespace webrtc {
23 
24 class DesktopRegion;
25 
26 // A texture copied or mapped from a DXGI_OUTDUPL_FRAME_INFO and IDXGIResource.
27 class DxgiTexture {
28  public:
29   // Creates a DxgiTexture instance, which represents the |desktop_size| area of
30   // entire screen -- usually a monitor on the system.
31   DxgiTexture();
32 
33   virtual ~DxgiTexture();
34 
35   // Copies selected regions of a frame represented by frame_info and resource.
36   // Returns false if anything wrong.
37   bool CopyFrom(const DXGI_OUTDUPL_FRAME_INFO& frame_info,
38                 IDXGIResource* resource);
39 
desktop_size()40   const DesktopSize& desktop_size() const { return desktop_size_; }
41 
bits()42   uint8_t* bits() const { return static_cast<uint8_t*>(rect_.pBits); }
43 
pitch()44   int pitch() const { return static_cast<int>(rect_.Pitch); }
45 
46   // Releases the resource currently holds by this instance. Returns false if
47   // anything wrong, and this instance should be deprecated in this state. bits,
48   // pitch and AsDesktopFrame are only valid after a success CopyFrom() call,
49   // but before Release() call.
50   bool Release();
51 
52   // Returns a DesktopFrame snapshot of a DxgiTexture instance. This
53   // DesktopFrame is used to copy a DxgiTexture content to another DesktopFrame
54   // only. And it should not outlive its DxgiTexture instance.
55   const DesktopFrame& AsDesktopFrame();
56 
57  protected:
58   DXGI_MAPPED_RECT* rect();
59 
60   virtual bool CopyFromTexture(const DXGI_OUTDUPL_FRAME_INFO& frame_info,
61                                ID3D11Texture2D* texture) = 0;
62 
63   virtual bool DoRelease() = 0;
64 
65  private:
66   DXGI_MAPPED_RECT rect_ = {0};
67   DesktopSize desktop_size_;
68   std::unique_ptr<DesktopFrame> frame_;
69 };
70 
71 }  // namespace webrtc
72 
73 #endif  // MODULES_DESKTOP_CAPTURE_WIN_DXGI_TEXTURE_H_
74