1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #ifndef CHROME_BROWSER_UI_COCOA_TASK_MANAGER_MAC_H_ 6 #define CHROME_BROWSER_UI_COCOA_TASK_MANAGER_MAC_H_ 7 #pragma once 8 9 #import <Cocoa/Cocoa.h> 10 #include <vector> 11 12 #import "base/mac/cocoa_protocols.h" 13 #include "base/memory/scoped_nsobject.h" 14 #include "chrome/browser/task_manager/task_manager.h" 15 #include "chrome/browser/ui/cocoa/table_row_nsimage_cache.h" 16 17 @class WindowSizeAutosaver; 18 class SkBitmap; 19 class TaskManagerMac; 20 21 // This class is responsible for loading the task manager window and for 22 // managing it. 23 @interface TaskManagerWindowController : 24 NSWindowController<NSTableViewDataSource, 25 NSTableViewDelegate> { 26 @private 27 IBOutlet NSTableView* tableView_; 28 IBOutlet NSButton* endProcessButton_; 29 TaskManagerMac* taskManagerObserver_; // weak 30 TaskManager* taskManager_; // weak 31 TaskManagerModel* model_; // weak 32 bool highlightBackgroundResources_; 33 34 scoped_nsobject<WindowSizeAutosaver> size_saver_; 35 36 // These contain a permutation of [0..|model_->ResourceCount() - 1|]. Used to 37 // implement sorting. 38 std::vector<int> viewToModelMap_; 39 std::vector<int> modelToViewMap_; 40 41 // Descriptor of the current sort column. 42 scoped_nsobject<NSSortDescriptor> currentSortDescriptor_; 43 44 // Color we use for background resources. 45 scoped_nsobject<NSColor> backgroundResourceColor_; 46 } 47 48 // Creates and shows the task manager's window. 49 - (id)initWithTaskManagerObserver:(TaskManagerMac*)taskManagerObserver 50 highlightBackgroundResources:(bool)highlightBackgroundResources; 51 52 // Refreshes all data in the task manager table. 53 - (void)reloadData; 54 55 // Callback for "Stats for nerds" link. 56 - (IBAction)statsLinkClicked:(id)sender; 57 58 // Callback for "End process" button. 59 - (IBAction)killSelectedProcesses:(id)sender; 60 61 // Callback for double clicks on the table. 62 - (void)selectDoubleClickedTab:(id)sender; 63 @end 64 65 @interface TaskManagerWindowController (TestingAPI) 66 - (NSTableView*)tableView; 67 @end 68 69 // This class listens to task changed events sent by chrome. 70 class TaskManagerMac : public TaskManagerModelObserver, 71 public TableRowNSImageCache::Table { 72 public: 73 TaskManagerMac(TaskManager* task_manager, bool highlight_background); 74 virtual ~TaskManagerMac(); 75 76 // TaskManagerModelObserver 77 virtual void OnModelChanged(); 78 virtual void OnItemsChanged(int start, int length); 79 virtual void OnItemsAdded(int start, int length); 80 virtual void OnItemsRemoved(int start, int length); 81 82 // Called by the cocoa window controller when its window closes and the 83 // controller destroyed itself. Informs the model to stop updating. 84 void WindowWasClosed(); 85 86 // TableRowNSImageCache::Table 87 virtual int RowCount() const; 88 virtual SkBitmap GetIcon(int r) const; 89 90 // Creates the task manager if it doesn't exist; otherwise, it activates the 91 // existing task manager window. Highlights background resources if 92 // |highlight_background_resources| is true. 93 static void Show(bool highlight_background_resources); 94 95 // Returns the TaskManager observed by |this|. task_manager()96 TaskManager* task_manager() { return task_manager_; } 97 98 // Lazily converts the image at the given row and caches it in |icon_cache_|. 99 NSImage* GetImageForRow(int row); 100 101 // Returns the cocoa object. Used for testing. cocoa_controller()102 TaskManagerWindowController* cocoa_controller() { return window_controller_; } 103 104 // Returns true if the resource at this location is a background resource. 105 bool IsBackgroundRow(int row) const; 106 private: 107 // The task manager. 108 TaskManager* const task_manager_; // weak 109 110 // Our model. 111 TaskManagerModel* const model_; // weak 112 113 // Controller of our window, destroys itself when the task manager window 114 // is closed. 115 TaskManagerWindowController* window_controller_; // weak 116 117 // Caches favicons for all rows. Needs to be initalized after |model_|. 118 TableRowNSImageCache icon_cache_; 119 120 // If true, highlight background resources. 121 bool highlight_background_resources_; 122 123 // An open task manager window. There can only be one open at a time. This 124 // is reset to NULL when the window is closed. 125 static TaskManagerMac* instance_; 126 127 DISALLOW_COPY_AND_ASSIGN(TaskManagerMac); 128 }; 129 130 #endif // CHROME_BROWSER_UI_COCOA_TASK_MANAGER_MAC_H_ 131