Chromium Code Reviews| Index: chrome/browser/ui/bluetooth/bluetooth_chooser_bubble_delegate.cc |
| diff --git a/chrome/browser/ui/bluetooth/bluetooth_chooser_bubble_delegate.cc b/chrome/browser/ui/bluetooth/bluetooth_chooser_bubble_delegate.cc |
| index c0f23d3a216ccd55b5ae919417ba8223f60a0525..489d1933c71bff2e5e04068f5b8c1d66b3c4a9ee 100644 |
| --- a/chrome/browser/ui/bluetooth/bluetooth_chooser_bubble_delegate.cc |
| +++ b/chrome/browser/ui/bluetooth/bluetooth_chooser_bubble_delegate.cc |
| @@ -4,6 +4,8 @@ |
| #include "chrome/browser/ui/bluetooth/bluetooth_chooser_bubble_delegate.h" |
| +#include <iterator> |
| + |
| #include "base/stl_util.h" |
| #include "chrome/browser/ui/bluetooth/bluetooth_chooser_desktop.h" |
| #include "components/bubble/bubble_controller.h" |
| @@ -16,20 +18,22 @@ BluetoothChooserBubbleDelegate::~BluetoothChooserBubbleDelegate() { |
| bluetooth_chooser_->set_bluetooth_chooser_bubble_delegate(nullptr); |
| } |
| -const std::vector<base::string16>& BluetoothChooserBubbleDelegate::GetOptions() |
| - const { |
| - return device_names_; |
| +size_t BluetoothChooserBubbleDelegate::NumOptions() const { |
| + return device_names_and_ids_.size(); |
| +} |
| + |
| +const base::string16& BluetoothChooserBubbleDelegate::GetOption( |
| + size_t index) const { |
| + DCHECK_LT(index, device_names_and_ids_.size()); |
| + return device_names_and_ids_[index].first; |
| } |
| -// TODO(juncai): Change the index type to be size_t in base class to avoid |
| -// extra type casting. |
| -void BluetoothChooserBubbleDelegate::Select(int index) { |
| - size_t idx = static_cast<size_t>(index); |
| - size_t num_options = device_ids_.size(); |
| - DCHECK_LT(idx, num_options); |
| +void BluetoothChooserBubbleDelegate::Select(size_t index) { |
| + DCHECK_LT(index, device_names_and_ids_.size()); |
| if (bluetooth_chooser_) { |
| bluetooth_chooser_->CallEventHandler( |
| - content::BluetoothChooser::Event::SELECTED, device_ids_[idx]); |
| + content::BluetoothChooser::Event::SELECTED, |
| + device_names_and_ids_[index].second); |
| } |
| if (bubble_controller_) |
| @@ -56,25 +60,22 @@ void BluetoothChooserBubbleDelegate::Close() { |
| void BluetoothChooserBubbleDelegate::AddDevice( |
| const std::string& device_id, |
| const base::string16& device_name) { |
| - DCHECK(!ContainsValue(device_ids_, device_id)); |
| - device_names_.push_back(device_name); |
| - device_ids_.push_back(device_id); |
| - // TODO(juncai): Change OnOptionAdded's index type to be size_t to avoid |
| - // extra type casting here. |
| + device_names_and_ids_.push_back(std::make_pair(device_name, device_id)); |
| if (observer()) |
| - observer()->OnOptionAdded(static_cast<int>(device_names_.size()) - 1); |
| + observer()->OnOptionAdded(device_names_and_ids_.size() - 1); |
| } |
| void BluetoothChooserBubbleDelegate::RemoveDevice( |
| const std::string& device_id) { |
| - auto iter = std::find(device_ids_.begin(), device_ids_.end(), device_id); |
| - if (iter != device_ids_.end()) { |
| - size_t index = iter - device_ids_.begin(); |
| - device_ids_.erase(iter); |
| - device_names_.erase(device_names_.begin() + index); |
| - // TODO(juncai): Change OnOptionRemoved's index type to be size_t to avoid |
| - // extra type casting here. |
| - if (observer()) |
| - observer()->OnOptionRemoved(index); |
| + for (auto it = device_names_and_ids_.begin(), |
| + end = device_names_and_ids_.end(); |
|
Peter Kasting
2016/01/04 21:02:51
Don't declare a temp for |end|. Compare directly
juncai
2016/01/04 22:44:46
Done.
|
| + it != end; ++it) { |
| + if (it->second == device_id) { |
| + size_t index = std::distance(device_names_and_ids_.begin(), it); |
|
Peter Kasting
2016/01/04 21:02:51
You don't need to use std::distance() here since t
juncai
2016/01/04 22:44:46
Done.
|
| + device_names_and_ids_.erase(it); |
| + if (observer()) |
| + observer()->OnOptionRemoved(index); |
| + break; |
|
Peter Kasting
2016/01/04 21:02:51
Prefer "return" to "break" here.
juncai
2016/01/04 22:44:46
Done.
|
| + } |
| } |
| } |