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_D3D_DEVICE_H_
12 #define MODULES_DESKTOP_CAPTURE_WIN_D3D_DEVICE_H_
13 
14 #include <comdef.h>
15 #include <d3d11.h>
16 #include <dxgi.h>
17 #include <wrl/client.h>
18 
19 #include <vector>
20 
21 namespace webrtc {
22 
23 // A wrapper of ID3D11Device and its corresponding context and IDXGIAdapter.
24 // This class represents one video card in the system.
25 class D3dDevice {
26  public:
27   D3dDevice(const D3dDevice& other);
28   D3dDevice(D3dDevice&& other);
29   ~D3dDevice();
30 
d3d_device()31   ID3D11Device* d3d_device() const { return d3d_device_.Get(); }
32 
context()33   ID3D11DeviceContext* context() const { return context_.Get(); }
34 
dxgi_device()35   IDXGIDevice* dxgi_device() const { return dxgi_device_.Get(); }
36 
dxgi_adapter()37   IDXGIAdapter* dxgi_adapter() const { return dxgi_adapter_.Get(); }
38 
39   // Returns all D3dDevice instances on the system. Returns an empty vector if
40   // anything wrong.
41   static std::vector<D3dDevice> EnumDevices();
42 
43  private:
44   // Instances of D3dDevice should only be created by EnumDevices() static
45   // function.
46   D3dDevice();
47 
48   // Initializes the D3dDevice from an IDXGIAdapter.
49   bool Initialize(const Microsoft::WRL::ComPtr<IDXGIAdapter>& adapter);
50 
51   Microsoft::WRL::ComPtr<ID3D11Device> d3d_device_;
52   Microsoft::WRL::ComPtr<ID3D11DeviceContext> context_;
53   Microsoft::WRL::ComPtr<IDXGIDevice> dxgi_device_;
54   Microsoft::WRL::ComPtr<IDXGIAdapter> dxgi_adapter_;
55 };
56 
57 }  // namespace webrtc
58 
59 #endif  // MODULES_DESKTOP_CAPTURE_WIN_D3D_DEVICE_H_
60