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
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85
   86
   87
   88
   89
   90
   91
   92
   93
   94
   95
   96


headless / lib / browser / headless_screen_orientation_delegate.cc [blame]

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

#include "headless/lib/browser/headless_screen_orientation_delegate.h"

#include "base/check_deref.h"
#include "content/public/browser/render_view_host.h"
#include "content/public/browser/render_widget_host.h"
#include "content/public/browser/web_contents.h"
#include "headless/lib/browser/headless_screen.h"
#include "services/device/public/mojom/screen_orientation_lock_types.mojom.h"
#include "ui/display/mojom/screen_orientation.mojom-shared.h"
#include "ui/display/screen.h"
#include "ui/gfx/geometry/rect_f.h"

namespace headless {

namespace {

int64_t GetDisplayIdFromWebContents(content::WebContents* web_contents) {
  content::RenderWidgetHost* rwh =
      web_contents->GetPrimaryMainFrame()->GetRenderViewHost()->GetWidget();
  return rwh ? rwh->GetScreenInfo().display_id : display::kInvalidDisplayId;
}

display::mojom::ScreenOrientation GetNaturalScreenOrientation(
    int64_t display_id) {
  auto& headless_screen =
      CHECK_DEREF(static_cast<HeadlessScreen*>(display::Screen::GetScreen()));
  return headless_screen.IsNaturalPortrait(display_id)
             ? display::mojom::ScreenOrientation::kPortraitPrimary
             : display::mojom::ScreenOrientation::kLandscapePrimary;
}

display::mojom::ScreenOrientation GetScreenOrientationFromLockOrientation(
    int64_t display_id,
    device::mojom::ScreenOrientationLockType lock_orientation) {
  switch (lock_orientation) {
    case device::mojom::ScreenOrientationLockType::PORTRAIT:
    case device::mojom::ScreenOrientationLockType::PORTRAIT_PRIMARY:
      return display::mojom::ScreenOrientation::kPortraitPrimary;
    case device::mojom::ScreenOrientationLockType::PORTRAIT_SECONDARY:
      return display::mojom::ScreenOrientation::kPortraitSecondary;

    case device::mojom::ScreenOrientationLockType::LANDSCAPE:
    case device::mojom::ScreenOrientationLockType::LANDSCAPE_PRIMARY:
      return display::mojom::ScreenOrientation::kLandscapePrimary;
    case device::mojom::ScreenOrientationLockType::LANDSCAPE_SECONDARY:
      return display::mojom::ScreenOrientation::kLandscapeSecondary;

    case device::mojom::ScreenOrientationLockType::NATURAL:
      return GetNaturalScreenOrientation(display_id);

    case device::mojom::ScreenOrientationLockType::ANY:
    case device::mojom::ScreenOrientationLockType::DEFAULT:
      return display::mojom::ScreenOrientation::kUndefined;
  }

  return display::mojom::ScreenOrientation::kUndefined;
}

}  // namespace

HeadlessScreenOrientationDelegate::HeadlessScreenOrientationDelegate() {
  content::WebContents::SetScreenOrientationDelegate(this);
}

HeadlessScreenOrientationDelegate::~HeadlessScreenOrientationDelegate() {
  content::WebContents::SetScreenOrientationDelegate(nullptr);
}

bool HeadlessScreenOrientationDelegate::FullScreenRequired(
    content::WebContents* web_contents) {
  return false;
}

void HeadlessScreenOrientationDelegate::Lock(
    content::WebContents* web_contents,
    device::mojom::ScreenOrientationLockType lock_orientation) {
  int64_t display_id = GetDisplayIdFromWebContents(web_contents);
  display::mojom::ScreenOrientation screen_orientation =
      GetScreenOrientationFromLockOrientation(display_id, lock_orientation);
  HeadlessScreen::UpdateScreenSizeForScreenOrientation(display_id,
                                                       screen_orientation);
}

bool HeadlessScreenOrientationDelegate::ScreenOrientationProviderSupported(
    content::WebContents* web_contentss) {
  return true;
}

void HeadlessScreenOrientationDelegate::Unlock(
    content::WebContents* web_contents) {}

}  // namespace headless