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