1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74


content / browser / media / capture / desktop_capturer_ash.cc [blame]

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "content/browser/media/capture/desktop_capturer_ash.h"

#include "ash/shell.h"
#include "content/browser/media/capture/desktop_frame_skia.h"
#include "third_party/skia/include/core/SkBitmap.h"
#include "third_party/webrtc/modules/desktop_capture/desktop_frame.h"
#include "ui/aura/window.h"
#include "ui/gfx/geometry/rect.h"
#include "ui/gfx/image/image.h"
#include "ui/snapshot/snapshot_aura.h"

namespace content {

DesktopCapturerAsh::DesktopCapturerAsh() = default;

DesktopCapturerAsh::~DesktopCapturerAsh() = default;

bool DesktopCapturerAsh::GetSourceList(SourceList* result) {
  for (const auto& display : display::Screen::GetScreen()->GetAllDisplays()) {
    result->emplace_back(display.id(), std::string(), display.id());
  }
  return true;
}

bool DesktopCapturerAsh::SelectSource(SourceId id) {
  display_id_ = id;
  return true;
}

bool DesktopCapturerAsh::FocusOnSelectedSource() {
  return true;
}

void DesktopCapturerAsh::Start(Callback* callback) {
  callback_ = callback;
}

void DesktopCapturerAsh::CaptureFrame() {
  aura::Window* window = ash::Shell::GetRootWindowForNewWindows();
  if (display_id_) {
    window = ash::Shell::GetRootWindowForDisplayId(*display_id_);
  }
  if (!window) {
    callback_->OnCaptureResult(Result::ERROR_PERMANENT, nullptr);
    return;
  }

  const gfx::Rect bounds(window->bounds().size());
  ui::GrabWindowSnapshot(
      window, bounds,
      base::BindOnce(&DesktopCapturerAsh::OnGrabWindowSnapsot,
                     weak_ptr_factory_.GetWeakPtr()));
}

bool DesktopCapturerAsh::IsOccluded(const webrtc::DesktopVector& pos) {
  return false;
}

void DesktopCapturerAsh::SetSharedMemoryFactory(
    std::unique_ptr<webrtc::SharedMemoryFactory> shared_memory_factory) {}

void DesktopCapturerAsh::SetExcludedWindow(webrtc::WindowId window) {}

void DesktopCapturerAsh::OnGrabWindowSnapsot(gfx::Image snapshot) {
  callback_->OnCaptureResult(
      Result::SUCCESS,
      std::make_unique<content::DesktopFrameSkia>(snapshot.AsBitmap()));
}

}  // namespace content