| // Copyright 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "remoting/host/shaped_desktop_capturer.h" |
| |
| #include <utility> |
| |
| #include "base/logging.h" |
| #include "remoting/host/desktop_shape_tracker.h" |
| #include "third_party/webrtc/modules/desktop_capture/desktop_frame.h" |
| |
| namespace remoting { |
| |
| ShapedDesktopCapturer::ShapedDesktopCapturer( |
| scoped_ptr<webrtc::DesktopCapturer> desktop_capturer, |
| scoped_ptr<DesktopShapeTracker> shape_tracker) |
| : desktop_capturer_(std::move(desktop_capturer)), |
| shape_tracker_(std::move(shape_tracker)), |
| callback_(nullptr) {} |
| |
| ShapedDesktopCapturer::~ShapedDesktopCapturer() {} |
| |
| void ShapedDesktopCapturer::Start(webrtc::DesktopCapturer::Callback* callback) { |
| callback_ = callback; |
| desktop_capturer_->Start(this); |
| } |
| |
| void ShapedDesktopCapturer::Capture(const webrtc::DesktopRegion& region) { |
| desktop_capturer_->Capture(region); |
| } |
| |
| void ShapedDesktopCapturer::SetSharedMemoryFactory( |
| rtc::scoped_ptr<webrtc::SharedMemoryFactory> shared_memory_factory) { |
| desktop_capturer_->SetSharedMemoryFactory(std::move(shared_memory_factory)); |
| } |
| |
| void ShapedDesktopCapturer::OnCaptureCompleted(webrtc::DesktopFrame* frame) { |
| shape_tracker_->RefreshDesktopShape(); |
| frame->set_shape(new webrtc::DesktopRegion(shape_tracker_->desktop_shape())); |
| callback_->OnCaptureCompleted(frame); |
| } |
| |
| } // namespace remoting |