| /* |
| * 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. |
| */ |
| |
| #include "modules/desktop_capture/win/dxgi_adapter_duplicator.h" |
| |
| #include <comdef.h> |
| #include <DXGI.h> |
| |
| #include <algorithm> |
| |
| #include "rtc_base/checks.h" |
| #include "rtc_base/logging.h" |
| |
| namespace webrtc { |
| |
| using Microsoft::WRL::ComPtr; |
| |
| namespace { |
| |
| bool IsValidRect(const RECT& rect) { |
| return rect.right > rect.left && rect.bottom > rect.top; |
| } |
| |
| } // namespace |
| |
| DxgiAdapterDuplicator::DxgiAdapterDuplicator(const D3dDevice& device) |
| : device_(device) {} |
| DxgiAdapterDuplicator::DxgiAdapterDuplicator(DxgiAdapterDuplicator&&) = default; |
| DxgiAdapterDuplicator::~DxgiAdapterDuplicator() = default; |
| |
| bool DxgiAdapterDuplicator::Initialize() { |
| if (DoInitialize()) { |
| return true; |
| } |
| duplicators_.clear(); |
| return false; |
| } |
| |
| bool DxgiAdapterDuplicator::DoInitialize() { |
| for (int i = 0;; i++) { |
| ComPtr<IDXGIOutput> output; |
| _com_error error = |
| device_.dxgi_adapter()->EnumOutputs(i, output.GetAddressOf()); |
| if (error.Error() == DXGI_ERROR_NOT_FOUND) { |
| break; |
| } |
| |
| if (error.Error() == DXGI_ERROR_NOT_CURRENTLY_AVAILABLE) { |
| LOG(LS_WARNING) << "IDXGIAdapter::EnumOutputs returns " |
| "NOT_CURRENTLY_AVAILABLE. This may happen when " |
| "running in session 0."; |
| break; |
| } |
| |
| if (error.Error() != S_OK || !output) { |
| LOG(LS_WARNING) << "IDXGIAdapter::EnumOutputs returns an unexpected " |
| "result " |
| << error.ErrorMessage() << " with error code" |
| << error.Error(); |
| continue; |
| } |
| |
| DXGI_OUTPUT_DESC desc; |
| error = output->GetDesc(&desc); |
| if (error.Error() == S_OK) { |
| if (desc.AttachedToDesktop && IsValidRect(desc.DesktopCoordinates)) { |
| ComPtr<IDXGIOutput1> output1; |
| error = output.As(&output1); |
| if (error.Error() != S_OK || !output1) { |
| LOG(LS_WARNING) << "Failed to convert IDXGIOutput to IDXGIOutput1, " |
| "this usually means the system does not support " |
| "DirectX 11"; |
| continue; |
| } |
| DxgiOutputDuplicator duplicator(device_, output1, desc); |
| if (!duplicator.Initialize()) { |
| LOG(LS_WARNING) << "Failed to initialize DxgiOutputDuplicator on " |
| "output " |
| << i; |
| continue; |
| } |
| |
| duplicators_.push_back(std::move(duplicator)); |
| desktop_rect_.UnionWith(duplicators_.back().desktop_rect()); |
| } |
| } else { |
| LOG(LS_WARNING) << "Failed to get output description of device " << i |
| << ", ignore."; |
| } |
| } |
| |
| if (duplicators_.empty()) { |
| LOG(LS_WARNING) << "Cannot initialize any DxgiOutputDuplicator instance."; |
| } |
| |
| return !duplicators_.empty(); |
| } |
| |
| void DxgiAdapterDuplicator::Setup(Context* context) { |
| RTC_DCHECK(context->contexts.empty()); |
| context->contexts.resize(duplicators_.size()); |
| for (size_t i = 0; i < duplicators_.size(); i++) { |
| duplicators_[i].Setup(&context->contexts[i]); |
| } |
| } |
| |
| void DxgiAdapterDuplicator::Unregister(const Context* const context) { |
| RTC_DCHECK_EQ(context->contexts.size(), duplicators_.size()); |
| for (size_t i = 0; i < duplicators_.size(); i++) { |
| duplicators_[i].Unregister(&context->contexts[i]); |
| } |
| } |
| |
| bool DxgiAdapterDuplicator::Duplicate(Context* context, |
| SharedDesktopFrame* target) { |
| RTC_DCHECK_EQ(context->contexts.size(), duplicators_.size()); |
| for (size_t i = 0; i < duplicators_.size(); i++) { |
| if (!duplicators_[i].Duplicate(&context->contexts[i], |
| duplicators_[i].desktop_rect().top_left(), |
| target)) { |
| return false; |
| } |
| } |
| return true; |
| } |
| |
| bool DxgiAdapterDuplicator::DuplicateMonitor(Context* context, |
| int monitor_id, |
| SharedDesktopFrame* target) { |
| RTC_DCHECK_GE(monitor_id, 0); |
| RTC_DCHECK_LT(monitor_id, duplicators_.size()); |
| RTC_DCHECK_EQ(context->contexts.size(), duplicators_.size()); |
| return duplicators_[monitor_id].Duplicate(&context->contexts[monitor_id], |
| DesktopVector(), target); |
| } |
| |
| DesktopRect DxgiAdapterDuplicator::ScreenRect(int id) const { |
| RTC_DCHECK_GE(id, 0); |
| RTC_DCHECK_LT(id, duplicators_.size()); |
| return duplicators_[id].desktop_rect(); |
| } |
| |
| const std::string& DxgiAdapterDuplicator::GetDeviceName(int id) const { |
| RTC_DCHECK_GE(id, 0); |
| RTC_DCHECK_LT(id, duplicators_.size()); |
| return duplicators_[id].device_name(); |
| } |
| |
| int DxgiAdapterDuplicator::screen_count() const { |
| return static_cast<int>(duplicators_.size()); |
| } |
| |
| int64_t DxgiAdapterDuplicator::GetNumFramesCaptured() const { |
| int64_t min = INT64_MAX; |
| for (const auto& duplicator : duplicators_) { |
| min = std::min(min, duplicator.num_frames_captured()); |
| } |
| |
| return min; |
| } |
| |
| void DxgiAdapterDuplicator::TranslateRect(const DesktopVector& position) { |
| desktop_rect_.Translate(position); |
| RTC_DCHECK_GE(desktop_rect_.left(), 0); |
| RTC_DCHECK_GE(desktop_rect_.top(), 0); |
| for (auto& duplicator : duplicators_) { |
| duplicator.TranslateRect(position); |
| } |
| } |
| |
| } // namespace webrtc |