129 lines
3.7 KiB
C++
129 lines
3.7 KiB
C++
/*
|
|
* Copyright (c) 2014 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/cropping_window_capturer.h"
|
|
|
|
#include <stddef.h>
|
|
|
|
#include <utility>
|
|
|
|
#include "modules/desktop_capture/cropped_desktop_frame.h"
|
|
#include "rtc_base/logging.h"
|
|
|
|
namespace webrtc {
|
|
|
|
CroppingWindowCapturer::CroppingWindowCapturer(
|
|
const DesktopCaptureOptions& options)
|
|
: options_(options),
|
|
callback_(NULL),
|
|
window_capturer_(DesktopCapturer::CreateRawWindowCapturer(options)),
|
|
selected_window_(kNullWindowId),
|
|
excluded_window_(kNullWindowId) {}
|
|
|
|
CroppingWindowCapturer::~CroppingWindowCapturer() {}
|
|
|
|
void CroppingWindowCapturer::Start(DesktopCapturer::Callback* callback) {
|
|
callback_ = callback;
|
|
window_capturer_->Start(callback);
|
|
}
|
|
|
|
void CroppingWindowCapturer::SetSharedMemoryFactory(
|
|
std::unique_ptr<SharedMemoryFactory> shared_memory_factory) {
|
|
window_capturer_->SetSharedMemoryFactory(std::move(shared_memory_factory));
|
|
}
|
|
|
|
void CroppingWindowCapturer::CaptureFrame() {
|
|
if (ShouldUseScreenCapturer()) {
|
|
if (!screen_capturer_.get()) {
|
|
screen_capturer_ = DesktopCapturer::CreateRawScreenCapturer(options_);
|
|
if (excluded_window_) {
|
|
screen_capturer_->SetExcludedWindow(excluded_window_);
|
|
}
|
|
screen_capturer_->Start(this);
|
|
}
|
|
screen_capturer_->CaptureFrame();
|
|
} else {
|
|
window_capturer_->CaptureFrame();
|
|
}
|
|
}
|
|
|
|
void CroppingWindowCapturer::SetExcludedWindow(WindowId window) {
|
|
excluded_window_ = window;
|
|
if (screen_capturer_.get()) {
|
|
screen_capturer_->SetExcludedWindow(window);
|
|
}
|
|
}
|
|
|
|
bool CroppingWindowCapturer::GetSourceList(SourceList* sources) {
|
|
return window_capturer_->GetSourceList(sources);
|
|
}
|
|
|
|
bool CroppingWindowCapturer::SelectSource(SourceId id) {
|
|
if (window_capturer_->SelectSource(id)) {
|
|
selected_window_ = id;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool CroppingWindowCapturer::FocusOnSelectedSource() {
|
|
return window_capturer_->FocusOnSelectedSource();
|
|
}
|
|
|
|
void CroppingWindowCapturer::OnCaptureResult(
|
|
DesktopCapturer::Result result,
|
|
std::unique_ptr<DesktopFrame> screen_frame) {
|
|
if (!ShouldUseScreenCapturer()) {
|
|
RTC_LOG(LS_INFO) << "Window no longer on top when ScreenCapturer finishes";
|
|
window_capturer_->CaptureFrame();
|
|
return;
|
|
}
|
|
|
|
if (result != Result::SUCCESS) {
|
|
RTC_LOG(LS_WARNING) << "ScreenCapturer failed to capture a frame";
|
|
callback_->OnCaptureResult(result, nullptr);
|
|
return;
|
|
}
|
|
|
|
DesktopRect window_rect = GetWindowRectInVirtualScreen();
|
|
if (window_rect.is_empty()) {
|
|
RTC_LOG(LS_WARNING) << "Window rect is empty";
|
|
callback_->OnCaptureResult(Result::ERROR_TEMPORARY, nullptr);
|
|
return;
|
|
}
|
|
|
|
callback_->OnCaptureResult(
|
|
Result::SUCCESS,
|
|
CreateCroppedDesktopFrame(std::move(screen_frame), window_rect));
|
|
}
|
|
|
|
bool CroppingWindowCapturer::IsOccluded(const DesktopVector& pos) {
|
|
// Returns true if either capturer returns true.
|
|
if (window_capturer_->IsOccluded(pos)) {
|
|
return true;
|
|
}
|
|
if (screen_capturer_ != nullptr && screen_capturer_->IsOccluded(pos)) {
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
#if !defined(WEBRTC_WIN)
|
|
// CroppingWindowCapturer is implemented only for windows. On other platforms
|
|
// the regular window capturer is used.
|
|
// static
|
|
std::unique_ptr<DesktopCapturer> CroppingWindowCapturer::CreateCapturer(
|
|
const DesktopCaptureOptions& options) {
|
|
return DesktopCapturer::CreateWindowCapturer(options);
|
|
}
|
|
#endif
|
|
|
|
} // namespace webrtc
|