Index: chrome/browser/chromeos/login/screen_manager.cc |
diff --git a/chrome/browser/chromeos/login/screen_manager.cc b/chrome/browser/chromeos/login/screen_manager.cc |
index 8b2e4c6a384b27ab119c4b5f5f5bd70e334df096..8457c5a667114757f4e2400c5e5c3ad654f67ed6 100644 |
--- a/chrome/browser/chromeos/login/screen_manager.cc |
+++ b/chrome/browser/chromeos/login/screen_manager.cc |
@@ -12,19 +12,19 @@ ScreenManager::ScreenManager() { |
ScreenManager::~ScreenManager() { |
} |
-BaseScreen* ScreenManager::GetScreen(const std::string& screen_name) { |
- ScreenMap::const_iterator iter = screens_.find(screen_name); |
- if (iter != screens_.end()) { |
+BaseScreen* ScreenManager::GetScreen(OobeScreen screen) { |
+ auto iter = screens_.find(screen); |
+ if (iter != screens_.end()) |
return iter->second.get(); |
- } |
- BaseScreen* result = CreateScreen(screen_name); |
- DCHECK(result) << "Can not create screen named " << screen_name; |
- screens_[screen_name] = make_linked_ptr(result); |
+ |
+ BaseScreen* result = CreateScreen(screen); |
+ DCHECK(result) << "Can not create screen named " << GetOobeScreenName(screen); |
+ screens_[screen] = make_linked_ptr(result); |
return result; |
} |
-bool ScreenManager::HasScreen(const std::string& screen_name) { |
- return screens_.count(screen_name) > 0; |
+bool ScreenManager::HasScreen(OobeScreen screen) { |
+ return screens_.count(screen) > 0; |
} |
} // namespace chromeos |