Home | History | Annotate | Download | only in desktop_capture 
      1 /*
      2  *  Copyright (c) 2014 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 #include "webrtc/modules/desktop_capture/cropping_window_capturer.h"
     12 
     13 #include "webrtc/modules/desktop_capture/cropped_desktop_frame.h"
     14 #include "webrtc/system_wrappers/include/logging.h"
     15 
     16 namespace webrtc {
     17 
     18 CroppingWindowCapturer::CroppingWindowCapturer(
     19     const DesktopCaptureOptions& options)
     20     : options_(options),
     21       callback_(NULL),
     22       window_capturer_(WindowCapturer::Create(options)),
     23       selected_window_(kNullWindowId),
     24       excluded_window_(kNullWindowId) {
     25 }
     26 
     27 CroppingWindowCapturer::~CroppingWindowCapturer() {}
     28 
     29 void CroppingWindowCapturer::Start(DesktopCapturer::Callback* callback) {
     30   callback_ = callback;
     31   window_capturer_->Start(callback);
     32 }
     33 
     34 void CroppingWindowCapturer::Capture(const DesktopRegion& region) {
     35   if (ShouldUseScreenCapturer()) {
     36     if (!screen_capturer_.get()) {
     37       screen_capturer_.reset(ScreenCapturer::Create(options_));
     38       if (excluded_window_) {
     39         screen_capturer_->SetExcludedWindow(excluded_window_);
     40       }
     41       screen_capturer_->Start(this);
     42     }
     43     screen_capturer_->Capture(region);
     44   } else {
     45     window_capturer_->Capture(region);
     46   }
     47 }
     48 
     49 void CroppingWindowCapturer::SetExcludedWindow(WindowId window) {
     50   excluded_window_ = window;
     51   if (screen_capturer_.get()) {
     52     screen_capturer_->SetExcludedWindow(window);
     53   }
     54 }
     55 
     56 bool CroppingWindowCapturer::GetWindowList(WindowList* windows) {
     57   return window_capturer_->GetWindowList(windows);
     58 }
     59 
     60 bool CroppingWindowCapturer::SelectWindow(WindowId id) {
     61   if (window_capturer_->SelectWindow(id)) {
     62     selected_window_ = id;
     63     return true;
     64   }
     65   return false;
     66 }
     67 
     68 bool CroppingWindowCapturer::BringSelectedWindowToFront() {
     69   return window_capturer_->BringSelectedWindowToFront();
     70 }
     71 
     72 SharedMemory* CroppingWindowCapturer::CreateSharedMemory(size_t size) {
     73   return callback_->CreateSharedMemory(size);
     74 }
     75 
     76 void CroppingWindowCapturer::OnCaptureCompleted(DesktopFrame* frame) {
     77   rtc::scoped_ptr<DesktopFrame> screen_frame(frame);
     78 
     79   if (!ShouldUseScreenCapturer()) {
     80     LOG(LS_INFO) << "Window no longer on top when ScreenCapturer finishes";
     81     window_capturer_->Capture(DesktopRegion());
     82     return;
     83   }
     84 
     85   if (!frame) {
     86     LOG(LS_WARNING) << "ScreenCapturer failed to capture a frame";
     87     callback_->OnCaptureCompleted(NULL);
     88     return;
     89   }
     90 
     91   DesktopRect window_rect = GetWindowRectInVirtualScreen();
     92   if (window_rect.is_empty()) {
     93     LOG(LS_WARNING) << "Window rect is empty";
     94     callback_->OnCaptureCompleted(NULL);
     95     return;
     96   }
     97 
     98   rtc::scoped_ptr<DesktopFrame> window_frame(
     99       CreateCroppedDesktopFrame(screen_frame.release(), window_rect));
    100   callback_->OnCaptureCompleted(window_frame.release());
    101 }
    102 
    103 #if !defined(WEBRTC_WIN)
    104 // static
    105 WindowCapturer*
    106 CroppingWindowCapturer::Create(const DesktopCaptureOptions& options) {
    107   return WindowCapturer::Create(options);
    108 }
    109 #endif
    110 
    111 }  // namespace webrtc
    112