blob: e8c2af6662c3b2a5010ea43b066b4adc45e62aab (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
/*
* Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
*
* Use of this source code is governed by a BSD-style license
* that can be found in the LICENSE file in the root of the source
* tree. An additional intellectual property rights grant can be found
* in the file PATENTS. All contributing project authors may
* be found in the AUTHORS file in the root of the source tree.
*/
#ifndef MODULES_DESKTOP_CAPTURE_WIN_DXGI_TEXTURE_STAGING_H_
#define MODULES_DESKTOP_CAPTURE_WIN_DXGI_TEXTURE_STAGING_H_
#include <d3d11.h>
#include <dxgi1_2.h>
#include <wrl/client.h>
#include "modules/desktop_capture/desktop_geometry.h"
#include "modules/desktop_capture/desktop_region.h"
#include "modules/desktop_capture/win/d3d_device.h"
#include "modules/desktop_capture/win/dxgi_texture.h"
namespace webrtc {
// A pair of an ID3D11Texture2D and an IDXGISurface. We need an ID3D11Texture2D
// instance to copy GPU texture to RAM, but an IDXGISurface instance to map the
// texture into a bitmap buffer. These two instances are pointing to a same
// object.
//
// An ID3D11Texture2D is created by an ID3D11Device, so a DxgiTexture cannot be
// shared between two DxgiAdapterDuplicators.
class DxgiTextureStaging : public DxgiTexture {
public:
// Creates a DxgiTextureStaging instance. Caller must maintain the lifetime
// of input device to make sure it outlives this instance.
explicit DxgiTextureStaging(const D3dDevice& device);
~DxgiTextureStaging() override;
protected:
// Copies selected regions of a frame represented by frame_info and texture.
// Returns false if anything wrong.
bool CopyFromTexture(const DXGI_OUTDUPL_FRAME_INFO& frame_info,
ID3D11Texture2D* texture) override;
bool DoRelease() override;
private:
// Initializes stage_ from a CPU inaccessible IDXGIResource. Returns false if
// it failed to execute Windows APIs, or the size of the texture is not
// consistent with desktop_rect.
bool InitializeStage(ID3D11Texture2D* texture);
// Makes sure stage_ and surface_ are always pointing to a same object.
// We need an ID3D11Texture2D instance for
// ID3D11DeviceContext::CopySubresourceRegion, but an IDXGISurface for
// IDXGISurface::Map.
void AssertStageAndSurfaceAreSameObject();
const DesktopRect desktop_rect_;
const D3dDevice device_;
Microsoft::WRL::ComPtr<ID3D11Texture2D> stage_;
Microsoft::WRL::ComPtr<IDXGISurface> surface_;
};
} // namespace webrtc
#endif // MODULES_DESKTOP_CAPTURE_WIN_DXGI_TEXTURE_STAGING_H_
|
