OLD | NEW |
---|---|
1 /* | 1 /* |
2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #ifndef WEBRTC_MODULES_DESKTOP_CAPTURE_WIN_SCREEN_CAPTURER_WIN_DIRECTX_H_ | 11 #ifndef WEBRTC_MODULES_DESKTOP_CAPTURE_WIN_SCREEN_CAPTURER_WIN_DIRECTX_H_ |
12 #define WEBRTC_MODULES_DESKTOP_CAPTURE_WIN_SCREEN_CAPTURER_WIN_DIRECTX_H_ | 12 #define WEBRTC_MODULES_DESKTOP_CAPTURE_WIN_SCREEN_CAPTURER_WIN_DIRECTX_H_ |
13 | 13 |
14 #include "webrtc/modules/desktop_capture/screen_capturer.h" | 14 #include "webrtc/modules/desktop_capture/screen_capturer.h" |
15 | 15 |
16 #include <comdef.h> | |
17 #include <D3DCommon.h> | |
18 #include <D3D11.h> | |
19 #include <DXGI.h> | |
20 #include <DXGI1_2.h> | |
21 #include <windows.h> | |
22 #include <wrl/client.h> | |
23 | |
24 #include <memory> | 16 #include <memory> |
25 #include <vector> | 17 #include <vector> |
26 | 18 |
27 #include "webrtc/base/thread_annotations.h" | |
28 #include "webrtc/modules/desktop_capture/desktop_capture_options.h" | 19 #include "webrtc/modules/desktop_capture/desktop_capture_options.h" |
29 #include "webrtc/modules/desktop_capture/desktop_geometry.h" | |
30 #include "webrtc/modules/desktop_capture/desktop_region.h" | 20 #include "webrtc/modules/desktop_capture/desktop_region.h" |
31 #include "webrtc/modules/desktop_capture/screen_capture_frame_queue.h" | 21 #include "webrtc/modules/desktop_capture/screen_capture_frame_queue.h" |
32 #include "webrtc/modules/desktop_capture/shared_desktop_frame.h" | 22 #include "webrtc/modules/desktop_capture/shared_desktop_frame.h" |
33 | 23 |
34 namespace webrtc { | 24 namespace webrtc { |
35 | 25 |
36 // ScreenCapturerWinDirectx captures 32bit RGBA using DirectX. This | 26 // ScreenCapturerWinDirectx captures 32bit RGBA using DirectX. This |
37 // implementation won't work when ScreenCaptureFrameQueue.kQueueLength is not 2. | 27 // implementation won't work when ScreenCaptureFrameQueue.kQueueLength is not 2. |
38 class ScreenCapturerWinDirectx : public ScreenCapturer { | 28 class ScreenCapturerWinDirectx : public ScreenCapturer { |
39 public: | 29 public: |
40 // Initializes DirectX related components. Returns false if any error | 30 // Initializes DirectX related components. Returns false if any error |
41 // happened, any instance of this class won't be able to work in such status. | 31 // happened, any instance of this class won't be able to work in such status. |
42 // Thread safe, guarded by initialize_lock. | 32 // Thread safe, guarded by initialize_lock. |
43 static bool Initialize(); | 33 static bool Initialize(); |
44 | 34 |
45 explicit ScreenCapturerWinDirectx(const DesktopCaptureOptions& options); | 35 explicit ScreenCapturerWinDirectx(const DesktopCaptureOptions& options); |
46 virtual ~ScreenCapturerWinDirectx(); | 36 virtual ~ScreenCapturerWinDirectx(); |
47 | 37 |
48 void Start(Callback* callback) override; | 38 void Start(Callback* callback) override; |
49 void SetSharedMemoryFactory( | 39 void SetSharedMemoryFactory( |
50 std::unique_ptr<SharedMemoryFactory> shared_memory_factory) override; | 40 std::unique_ptr<SharedMemoryFactory> shared_memory_factory) override; |
51 void Capture(const DesktopRegion& region) override; | 41 void Capture(const DesktopRegion& region) override; |
52 bool GetScreenList(ScreenList* screens) override; | 42 bool GetScreenList(ScreenList* screens) override; |
53 bool SelectScreen(ScreenId id) override; | 43 bool SelectScreen(ScreenId id) override; |
54 | 44 |
55 private: | 45 private: |
56 // Texture is a pair of an ID3D11Texture2D and an IDXGISurface. Refer to its | |
57 // implementation in source code for details. | |
58 class Texture; | |
59 | |
60 // An implementation of DesktopFrame to return data from a Texture instance. | |
61 class DxgiDesktopFrame; | |
62 | |
63 static bool DoInitialize(); | |
64 | |
65 // Initializes DxgiOutputDuplication. If current DxgiOutputDuplication | |
66 // instance is existing, this function takes no-op and returns true. Returns | |
67 // false if it fails to execute windows api. | |
68 static bool DuplicateOutput(); | |
69 | |
70 // Deprecates current DxgiOutputDuplication instance and calls DuplicateOutput | |
71 // to reinitialize it. | |
72 static bool ForceDuplicateOutput(); | |
73 | |
74 // Detects update regions in last frame, if anything wrong, returns false. | |
75 // ProcessFrame will insert a whole desktop size as updated region instead. | |
76 static bool DetectUpdatedRegion(const DXGI_OUTDUPL_FRAME_INFO& frame_info, | |
77 DesktopRegion* updated_region); | |
78 | |
79 // A helper function to handle _com_error result in DetectUpdatedRegion. | |
80 // Returns false if the _com_error shows an error. | |
81 static bool HandleDetectUpdatedRegionError(const _com_error& error, | |
82 const char* stage); | |
83 | |
84 // Processes one frame received from AcquireNextFrame function, returns a | |
85 // nullptr if anything wrong. | |
86 std::unique_ptr<DesktopFrame> ProcessFrame( | |
87 const DXGI_OUTDUPL_FRAME_INFO& frame_info, | |
88 IDXGIResource* resource); | |
89 | |
90 // A shortcut to execute callback with current frame in frames. | |
91 void EmitCurrentFrame(); | |
92 | |
93 ScreenCaptureFrameQueue<rtc::scoped_refptr<Texture>> surfaces_; | |
94 ScreenCaptureFrameQueue<SharedDesktopFrame> frames_; | 46 ScreenCaptureFrameQueue<SharedDesktopFrame> frames_; |
95 std::unique_ptr<SharedMemoryFactory> shared_memory_factory_; | 47 std::unique_ptr<SharedMemoryFactory> shared_memory_factory_; |
96 Callback* callback_ = nullptr; | 48 Callback* callback_ = nullptr; |
97 | 49 |
50 // Returns desktop size of selected screen. | |
51 DesktopSize SelectedDesktopSize() const; | |
Sergey Ulanov
2016/07/08 22:36:55
Move this above |frames_|. Methods definition shou
Hzj_jie
2016/07/11 00:55:00
Done.
| |
52 // Returns desktop rect of selected screen. | |
53 DesktopRect SelectedDesktopRect() const; | |
54 | |
55 ScreenId current_screen_id = kFullDesktopScreenId; | |
56 | |
98 bool set_thread_execution_state_failed_ = false; | 57 bool set_thread_execution_state_failed_ = false; |
99 | 58 |
100 RTC_DISALLOW_COPY_AND_ASSIGN(ScreenCapturerWinDirectx); | 59 RTC_DISALLOW_COPY_AND_ASSIGN(ScreenCapturerWinDirectx); |
101 }; | 60 }; |
102 | 61 |
103 } // namespace webrtc | 62 } // namespace webrtc |
104 | 63 |
105 #endif // WEBRTC_MODULES_DESKTOP_CAPTURE_WIN_SCREEN_CAPTURER_WIN_DIRECTX_H_ | 64 #endif // WEBRTC_MODULES_DESKTOP_CAPTURE_WIN_SCREEN_CAPTURER_WIN_DIRECTX_H_ |
OLD | NEW |