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