Task Manager: Preserve selection when rows are added

chrome/browser/ui/task_manager/task_manager_table_model.cc

 // Copyright 2015 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/task_manager/task_manager_table_model.h"
 
 #include <stddef.h>
 
 #include "base/command_line.h"
 #include "base/i18n/number_formatting.h"
@@ skipping 565 matching lines @@
 }
 
 void TaskManagerTableModel::OnTaskAdded(TaskId id) {
   // For the table view scrollbar to behave correctly we must inform it that
   // a new task has been added.
 
   // We will get a newly sorted list from the task manager as opposed to just
   // adding |id| to |tasks_| because we want to keep |tasks_| sorted by proc IDs
   // and then by Task IDs.
   tasks_ = observed_task_manager()->GetTaskIdsList();
-  if (table_model_observer_)
-    table_model_observer_->OnItemsAdded(RowCount() - 1, 1);
+
+  if (table_model_observer_) {
+    std::vector<TaskId>::difference_type index =
+        std::find(tasks_.begin(), tasks_.end(), id) - tasks_.begin();
+    table_model_observer_->OnItemsAdded(static_cast<int>(index), 1);
+  }
 }
 
 void TaskManagerTableModel::OnTaskToBeRemoved(TaskId id) {
   auto index = std::find(tasks_.begin(), tasks_.end(), id);
   if (index == tasks_.end())
     return;
   auto removed_index = index - tasks_.begin();
   tasks_.erase(index);
   if (table_model_observer_)
     table_model_observer_->OnItemsRemoved(removed_index, 1);
@@ skipping 231 matching lines @@
 bool TaskManagerTableModel::IsTaskFirstInGroup(int row_index) const {
   if (row_index == 0)
     return true;
 
   return observed_task_manager()->GetProcessId(tasks_[row_index - 1]) !=
       observed_task_manager()->GetProcessId(tasks_[row_index]);
 }
 
 
 }  // namespace task_management