Allow arc app to lock screen orientation in TouchView/Tablet mode

chrome/browser/ui/ash/launcher/arc_app_window_launcher_controller.cc

 // Copyright 2016 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 "chrome/browser/ui/ash/launcher/arc_app_window_launcher_controller.h"
 
 #include "ash/common/wm/window_state.h"
 #include "ash/common/wm_shell.h"
+#include "ash/display/display_manager.h"
 #include "ash/display/screen_orientation_controller_chromeos.h"
 #include "ash/shelf/shelf_delegate.h"
 #include "ash/shelf/shelf_util.h"
 #include "ash/shell.h"
 #include "ash/wm/maximize_mode/maximize_mode_controller.h"
 #include "ash/wm/window_state_aura.h"
 #include "ash/wm/window_util.h"
 #include "base/bind.h"
 #include "chrome/browser/chromeos/arc/arc_auth_service.h"
 #include "chrome/browser/chromeos/arc/arc_support_host.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/app_list/arc/arc_app_utils.h"
 #include "chrome/browser/ui/ash/launcher/arc_app_window_launcher_item_controller.h"
 #include "chrome/browser/ui/ash/launcher/chrome_launcher_controller.h"
 #include "chrome/browser/ui/ash/multi_user/multi_user_window_manager.h"
 #include "components/arc/arc_bridge_service.h"
 #include "components/exo/shell_surface.h"
 #include "components/signin/core/account_id/account_id.h"
 #include "components/user_manager/user_manager.h"
 #include "third_party/WebKit/public/platform/modules/screen_orientation/WebScreenOrientationLockType.h"
 #include "ui/aura/client/aura_constants.h"
 #include "ui/aura/env.h"
 #include "ui/base/base_window.h"
+#include "ui/display/display.h"
 #include "ui/views/widget/widget.h"
 
 namespace {
 
 enum class FullScreenMode {
   NOT_DEFINED,  // Fullscreen mode was not defined.
   ACTIVE,       // Fullscreen is activated for an app.
   NON_ACTIVE,   // Fullscreen was not activated for an app.
 };
 
+arc::mojom::OrientationLock GetCurrentOrientation() {
+  if (!display::Display::HasInternalDisplay())
+    return arc::mojom::OrientationLock::NONE;
+  display::Display internal_display =
+      ash::Shell::GetInstance()->display_manager()->GetDisplayForId(
+          display::Display::InternalDisplayId());
+
+  // ChromeOS currently assumes that the internal panel is always
+  // landscape (ROTATE_0 == landscape).
+  switch (internal_display.rotation()) {
+    case display::Display::ROTATE_0:
+    case display::Display::ROTATE_180:
+      return arc::mojom::OrientationLock::LANDSCAPE;
+    case display::Display::ROTATE_90:
+    case display::Display::ROTATE_270:
+      return arc::mojom::OrientationLock::PORTRAIT;
+  }
+  return arc::mojom::OrientationLock::NONE;
+}
+
 blink::WebScreenOrientationLockType BlinkOrientationLockFromMojom(
     arc::mojom::OrientationLock orientation_lock) {
+  DCHECK_NE(arc::mojom::OrientationLock::CURRENT, orientation_lock);
   if (orientation_lock == arc::mojom::OrientationLock::PORTRAIT) {
     return blink::WebScreenOrientationLockPortrait;
   } else if (orientation_lock == arc::mojom::OrientationLock::LANDSCAPE) {
     return blink::WebScreenOrientationLockLandscape;
   } else {
     return blink::WebScreenOrientationLockAny;
   }
 }
 
 }  // namespace
@@ skipping 512 matching lines @@
   if (app_window->has_requested_orientation_lock()) {
     orientation_lock = app_window->requested_orientation_lock();
   } else {
     ArcAppListPrefs* prefs = ArcAppListPrefs::Get(observed_profile_);
     std::unique_ptr<ArcAppListPrefs::AppInfo> app_info =
         prefs->GetApp(app_window->app_id());
     if (!app_info)
       return;
     orientation_lock = app_info->orientation_lock;
   }
+
+  if (orientation_lock == arc::mojom::OrientationLock::CURRENT) {
+    // Resolve the orientation when it first resolved.
+    orientation_lock = GetCurrentOrientation();
+    app_window->set_requested_orientation_lock(orientation_lock);
+  }
+
   shell->screen_orientation_controller()->LockOrientationForWindow(
       window, BlinkOrientationLockFromMojom(orientation_lock));
 }