Move all typecasting code to individual screens.

chrome/browser/chromeos/login/wizard_controller.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/chromeos/login/wizard_controller.h"
 
 #include <signal.h>
 #include <stdlib.h>
 #include <sys/types.h>
 
@@ skipping 254 matching lines @@
       first_screen_name == WizardController::kTestNoScreenName)) {
     first_screen_name_ = screen_pref;
   }
 
   AdvanceToScreen(first_screen_name_);
   if (!IsMachineHWIDCorrect() && !StartupUtils::IsDeviceRegistered() &&
       first_screen_name_.empty())
     ShowWrongHWIDScreen();
 }
 
-chromeos::NetworkScreen* WizardController::GetNetworkScreen() {
-  return static_cast<chromeos::NetworkScreen*>(GetScreen(kNetworkScreenName));
-}
-
-chromeos::UpdateScreen* WizardController::GetUpdateScreen() {
-  return static_cast<chromeos::UpdateScreen*>(GetScreen(kUpdateScreenName));
-}
-
-chromeos::UserImageScreen* WizardController::GetUserImageScreen() {
-  return static_cast<chromeos::UserImageScreen*>(
-      GetScreen(kUserImageScreenName));
-}
-
-chromeos::EnrollmentScreen* WizardController::GetEnrollmentScreen() {
-  return static_cast<chromeos::EnrollmentScreen*>(
-      GetScreen(kEnrollmentScreenName));
-}
-
-chromeos::AutoEnrollmentCheckScreen*
-    WizardController::GetAutoEnrollmentCheckScreen() {
-  return static_cast<chromeos::AutoEnrollmentCheckScreen*>(
-      GetScreen(kAutoEnrollmentCheckScreenName));
-}
-
-chromeos::SupervisedUserCreationScreen*
-    WizardController::GetSupervisedUserCreationScreen() {
-  return static_cast<chromeos::SupervisedUserCreationScreen*>(
-      GetScreen(kSupervisedUserCreationScreenName));
-}
-
 chromeos::ErrorScreen* WizardController::GetErrorScreen() {
   return static_cast<chromeos::ErrorScreen*>(GetScreen(kErrorScreenName));
 }
 
-WizardScreen* WizardController::GetScreen(const std::string& screen_name) {
-  ScreenMap::const_iterator iter = screens_.find(screen_name);
-  if (iter != screens_.end()) {
-    return iter->second.get();
-  }
-  WizardScreen* result = CreateScreen(screen_name);
-  DCHECK(result) << "Can not create screen named " << screen_name;
-  screens_[screen_name] = make_linked_ptr(result);
-  return result;
-}
-
 WizardScreen* WizardController::CreateScreen(const std::string& screen_name) {
   if (screen_name == kNetworkScreenName) {
     return new chromeos::NetworkScreen(this,
                                        oobe_display_->GetNetworkScreenActor());
   } else if (screen_name == kErrorScreenName) {
     return new chromeos::ErrorScreen(this,
                                      oobe_display_->GetErrorScreenActor());
   } else if (screen_name == kUpdateScreenName) {
     chromeos::UpdateScreen* result =
         new chromeos::UpdateScreen(this, oobe_display_->GetUpdateScreenActor());
@@ skipping 108 matching lines @@
 
   bool is_auto_enrollment = false;
   std::string user;
   if (screen_parameters_.get()) {
     screen_parameters_->GetBoolean("is_auto_enrollment", &is_auto_enrollment);
     screen_parameters_->GetString("user", &user);
   }
 
   EnrollmentScreenActor::EnrollmentMode mode =
       EnrollmentScreenActor::ENROLLMENT_MODE_MANUAL;
-  EnrollmentScreen* screen = GetEnrollmentScreen();
+  EnrollmentScreen* screen = EnrollmentScreen::Get(this);
   std::string enrollment_domain = GetForcedEnrollmentDomain();
   if (is_auto_enrollment) {
     mode = EnrollmentScreenActor::ENROLLMENT_MODE_AUTO;
   } else if (enrollment_recovery_) {
     mode = EnrollmentScreenActor::ENROLLMENT_MODE_RECOVERY;
     enrollment_domain = GetEnrollmentRecoveryDomain();
   } else if (ShouldAutoStartEnrollment() && !CanExitEnrollment()) {
     mode = EnrollmentScreenActor::ENROLLMENT_MODE_FORCED;
   }
 
@@ skipping 37 matching lines @@
 
 void WizardController::ShowWrongHWIDScreen() {
   VLOG(1) << "Showing wrong HWID screen.";
   SetStatusAreaVisible(false);
   SetCurrentScreen(GetScreen(kWrongHWIDScreenName));
 }
 
 void WizardController::ShowAutoEnrollmentCheckScreen() {
   VLOG(1) << "Showing Auto-enrollment check screen.";
   SetStatusAreaVisible(true);
-  AutoEnrollmentCheckScreen* screen = GetAutoEnrollmentCheckScreen();
+  AutoEnrollmentCheckScreen* screen = AutoEnrollmentCheckScreen::Get(this);
   screen->set_auto_enrollment_controller(host_->GetAutoEnrollmentController());
   SetCurrentScreen(screen);
 }
 
 void WizardController::ShowSupervisedUserCreationScreen() {
   VLOG(1) << "Showing Locally managed user creation screen screen.";
   SetStatusAreaVisible(true);
   SetCurrentScreen(GetScreen(kSupervisedUserCreationScreenName));
 }
 
@@ skipping 243 matching lines @@
   ShowAutoEnrollmentCheckScreen();
 }
 
 void WizardController::OnHostPairingFinished() {
   ShowAutoEnrollmentCheckScreen();
 }
 
 void WizardController::InitiateOOBEUpdate() {
   PerformPostEulaActions();
   SetCurrentScreenSmooth(GetScreen(kUpdateScreenName), true);
-  GetUpdateScreen()->StartNetworkCheck();
+  UpdateScreen::Get(this)->StartNetworkCheck();
 }
 
 void WizardController::StartTimezoneResolve() {
   geolocation_provider_.reset(new SimpleGeolocationProvider(
       g_browser_process->system_request_context(),
       SimpleGeolocationProvider::DefaultGeolocationProviderURL()));
   geolocation_provider_->RequestGeolocation(
       base::TimeDelta::FromSeconds(kResolveTimeZoneTimeoutSeconds),
       base::Bind(&WizardController::OnLocationResolved,
                  weak_factory_.GetWeakPtr()));
@@ skipping 470 matching lines @@
   if (ShouldAutoStartEnrollment() || ShouldRecoverEnrollment()) {
     StartupUtils::MarkEulaAccepted();
     auth_token_ = token;
     InitiateOOBEUpdate();
   } else {
     LOG(WARNING) << "Not in device enrollment.";
   }
 }
 
 }  // namespace chromeos