[cros] Initial WebRTC-enabled implementation of user image picker on OOBE.

chrome/browser/ui/webui/chromeos/login/user_image_screen_handler.cc

 // Copyright (c) 2012 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/webui/chromeos/login/user_image_screen_handler.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
+#include "base/command_line.h"
 #include "base/logging.h"
 #include "base/metrics/histogram.h"
 #include "base/values.h"
 #include "chrome/browser/chromeos/login/camera_detector.h"
 #include "chrome/browser/chromeos/login/default_user_images.h"
 #include "chrome/browser/chromeos/login/user.h"
 #include "chrome/browser/chromeos/login/webui_login_display.h"
 #include "chrome/browser/chromeos/options/take_photo_dialog.h"
 #include "chrome/browser/ui/webui/web_ui_util.h"
+#include "chrome/common/chrome_switches.h"
 #include "chrome/common/url_constants.h"
+#include "googleurl/src/gurl.h"
 #include "grit/generated_resources.h"
+#include "net/base/data_url.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/views/widget/widget.h"
 
 namespace {
 
 // UserImage screen ID.
 const char kUserImageScreen[] = "user-image";
 
 }  // namespace
 
 namespace chromeos {
 
 UserImageScreenHandler::UserImageScreenHandler()
     : screen_(NULL),
       show_on_init_(false),
       selected_image_(User::kInvalidImageIndex),
+      accept_photo_after_decoding_(false),
       user_photo_data_url_(chrome::kAboutBlankURL),
       profile_picture_data_url_(chrome::kAboutBlankURL),
       profile_picture_absent_(false),
       ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)) {
 }
 
 UserImageScreenHandler::~UserImageScreenHandler() {
   if (screen_)
     screen_->OnActorDestroyed(this);
 }
 
 void UserImageScreenHandler::GetLocalizedStrings(
     DictionaryValue *localized_strings) {
   // TODO(ivankr): string should be renamed to something like
   // IDS_USER_IMAGE_SCREEN_TITLE (currently used for Take Photo dialog).
   localized_strings->SetString("userImageScreenTitle",
       l10n_util::GetStringUTF16(IDS_OOBE_PICTURE));
   localized_strings->SetString("userImageScreenDescription",
       l10n_util::GetStringUTF16(IDS_OPTIONS_CHANGE_PICTURE_DIALOG_TEXT));
   localized_strings->SetString("takePhoto",
       l10n_util::GetStringUTF16(IDS_OPTIONS_CHANGE_PICTURE_TAKE_PHOTO));
   localized_strings->SetString("profilePhoto",
       l10n_util::GetStringUTF16(IDS_IMAGE_SCREEN_PROFILE_PHOTO));
   localized_strings->SetString("profilePhotoLoading",
       l10n_util::GetStringUTF16(IDS_IMAGE_SCREEN_PROFILE_LOADING_PHOTO));
   localized_strings->SetString("okButtonText",
       l10n_util::GetStringUTF16(IDS_OK));
+  if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kEnableHtml5Camera))
+    localized_strings->SetString("cameraType", "webrtc");
+  else
+    localized_strings->SetString("cameraType", "old");
 }
 
 void UserImageScreenHandler::Initialize() {
   ListValue image_urls;
   for (int i = 0; i < kDefaultImagesCount; ++i) {
     image_urls.Append(new StringValue(GetDefaultImageUrl(i)));
   }
   web_ui()->CallJavascriptFunction("oobe.UserImageScreen.setUserImages",
                                    image_urls);
 
@@ skipping 47 matching lines @@
 void UserImageScreenHandler::UpdateVideoFrame(const SkBitmap& frame) {
 }
 
 void UserImageScreenHandler::ShowCameraError() {
 }
 
 void UserImageScreenHandler::ShowCameraInitializing() {
 }
 
 void UserImageScreenHandler::CheckCameraPresence() {
+  // For WebRTC, camera presence checked is done on JS side.
+  if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kEnableHtml5Camera))
+    return;
   CameraDetector::StartPresenceCheck(
       base::Bind(&UserImageScreenHandler::OnCameraPresenceCheckDone,
                  weak_factory_.GetWeakPtr()));
 }
 
 bool UserImageScreenHandler::IsCapturing() const {
   return false;
 }
 
 void UserImageScreenHandler::RegisterMessages() {
   web_ui()->RegisterMessageCallback("takePhoto",
       base::Bind(&UserImageScreenHandler::HandleTakePhoto,
                  base::Unretained(this)));
+  web_ui()->RegisterMessageCallback("photoTaken",
+      base::Bind(&UserImageScreenHandler::HandlePhotoTaken,
+                 base::Unretained(this)));
   web_ui()->RegisterMessageCallback("selectImage",
       base::Bind(&UserImageScreenHandler::HandleSelectImage,
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback("onUserImageAccepted",
       base::Bind(&UserImageScreenHandler::HandleImageAccepted,
                  base::Unretained(this)));
   web_ui()->RegisterMessageCallback("onUserImageScreenShown",
       base::Bind(&UserImageScreenHandler::HandleScreenShown,
                  base::Unretained(this)));
 }
@@ skipping 22 matching lines @@
 
 void UserImageScreenHandler::OnPhotoAccepted(const gfx::ImageSkia& photo) {
   user_photo_ = photo;
   user_photo_data_url_ = web_ui_util::GetImageDataUrl(user_photo_);
   selected_image_ = User::kExternalImageIndex;
   base::StringValue data_url(user_photo_data_url_);
   web_ui()->CallJavascriptFunction("oobe.UserImageScreen.setUserPhoto",
                                    data_url);
 }
 
+void UserImageScreenHandler::HandlePhotoTaken(const base::ListValue* args) {
+  std::string image_url;
+  if (!args || args->GetSize() != 1 || !args->GetString(0, &image_url))
+    NOTREACHED();
+  DCHECK(!image_url.empty());
+
+  std::string mime_type, charset, raw_data;
+  if (!net::DataURL::Parse(GURL(image_url), &mime_type, &charset, &raw_data))
+    NOTREACHED();
+  DCHECK_EQ("image/png", mime_type);
+
+  user_photo_ = gfx::ImageSkia();
+  user_photo_data_url_ = image_url;
+
+  image_decoder_ = new ImageDecoder(this, raw_data);

[Nikita (slow), 2012/06/13 14:07:35]

As discussed, should fix leaking of ImageDecoder + reset delegate_ when existing
image_decoder_ is discarded or UserImageScreenHandler is destroyed.

[Ivan Korotkov, 2012/06/13 14:26:06]

Done.

+  image_decoder_->Start();
+}
+
 void UserImageScreenHandler::HandleTakePhoto(const base::ListValue* args) {
   DCHECK(args && args->empty());
   views::Widget* window = views::Widget::CreateWindowWithParent(
       new TakePhotoDialog(this), GetNativeWindow());
   window->SetAlwaysOnTop(true);
   window->Show();
 }
 
 void UserImageScreenHandler::HandleSelectImage(const base::ListValue* args) {
   std::string image_url;
-  if (!args ||
-      args->GetSize() != 1 ||
-      !args->GetString(0, &image_url)) {
+  if (!args || args->GetSize() != 1 || !args->GetString(0, &image_url))
     NOTREACHED();
-    return;
-  }
   if (image_url.empty())
     return;
 
   int user_image_index = User::kInvalidImageIndex;
   if (IsDefaultImageUrl(image_url, &user_image_index)) {
     selected_image_ = user_image_index;
   } else if (image_url == user_photo_data_url_) {
     selected_image_ = User::kExternalImageIndex;
   } else {
     selected_image_ = User::kProfileImageIndex;
   }
 }
 
 void UserImageScreenHandler::HandleImageAccepted(const base::ListValue* args) {
   DCHECK(args && args->empty());
   if (!screen_)
     return;
   switch (selected_image_) {
     case User::kExternalImageIndex:
-      screen_->OnPhotoTaken(user_photo_);
+      // Photo decoding may not have been finished yet.
+      if (user_photo_.empty())
+        accept_photo_after_decoding_ = true;
+      else
+        screen_->OnPhotoTaken(user_photo_);

[Nikita (slow), 2012/06/13 14:07:35]

As discussed, should fix leaking of ImageDecoder + reset delegate_ when existing
image_decoder_ is discarded or UserImageScreenHandler is destroyed.

[Nikita (slow), 2012/06/13 14:08:36]

Wrong comment. Correct one:

accept_photo_after_decoding_ = false;

       break;
 
     case User::kProfileImageIndex:
       screen_->OnProfileImageSelected();
       break;
 
     default:
       DCHECK(selected_image_ >= 0 && selected_image_ < kDefaultImagesCount);
       screen_->OnDefaultImageSelected(selected_image_);
   }
 }
 
 void UserImageScreenHandler::HandleScreenShown(const base::ListValue* args) {
   DCHECK(args && args->empty());
   DCHECK(!screen_show_time_.is_null());
 
   base::TimeDelta delta = base::Time::Now() - screen_show_time_;
   VLOG(1) << "Screen load time: " << delta.InSecondsF();
   UMA_HISTOGRAM_TIMES("UserImage.ScreenIsShownTime", delta);
 }
 
 void UserImageScreenHandler::OnCameraPresenceCheckDone() {
   base::FundamentalValue present_value(
       CameraDetector::camera_presence() == CameraDetector::kCameraPresent);
   web_ui()->CallJavascriptFunction("oobe.UserImageScreen.setCameraPresent",
                                    present_value);
 }
 
+void UserImageScreenHandler::OnImageDecoded(const ImageDecoder* decoder,
+                                            const SkBitmap& decoded_image) {
+  // Another decoding request was started, skip this one.
+  if (decoder != image_decoder_.get())
+    return;
+  user_photo_ = gfx::ImageSkia(decoded_image);
+  image_decoder_.release();
+  if (screen_ && accept_photo_after_decoding_)
+    screen_->OnPhotoTaken(user_photo_);

[Nikita (slow), 2012/06/13 14:07:35]

accept_photo_after_decoding_ = false;

[Ivan Korotkov, 2012/06/13 14:26:06]

As discussed, this logic is the right one.

+}
+
+void UserImageScreenHandler::OnDecodeImageFailed(const ImageDecoder* decoder) {
+  NOTREACHED() << "Failed to decode PNG image from WebUI";
+}
+
 }  // namespace chromeos