Move common notification layout to base class

ash/system/tray/tray_views.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 "ash/system/tray/tray_views.h"
 
 #include "ash/system/tray/tray_constants.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 #include "grit/ash_strings.h"
 #include "grit/ui_resources.h"
@@ skipping 438 matching lines @@
   label->SetEnabledColor(SK_ColorWHITE);
   label->SetBackgroundColor(SkColorSetARGB(0, 255, 255, 255));
   label->SetShadowColors(SkColorSetARGB(64, 0, 0, 0),
                          SkColorSetARGB(64, 0, 0, 0));
   label->SetShadowOffset(0, 1);
 }
 
 ////////////////////////////////////////////////////////////////////////////////
 // TrayNotificationView
 
-TrayNotificationView::TrayNotificationView() {
+TrayNotificationView::TrayNotificationView(int icon_id) : icon_id_(icon_id) {
+}
+
+TrayNotificationView::~TrayNotificationView() {
 }
 
 void TrayNotificationView::InitView(views::View* contents) {
   set_background(views::Background::CreateSolidBackground(kBackgroundColor));
 
   views::GridLayout* layout = new views::GridLayout(this);
   SetLayoutManager(layout);
 
   views::ImageButton* close_button = new views::ImageButton(this);
   close_button->SetImage(views::CustomButton::BS_NORMAL,
                          ResourceBundle::GetSharedInstance().GetImageSkiaNamed(
                              IDR_AURA_WINDOW_CLOSE));
 
-  int contents_width = kTrayPopupWidth - kNotificationCloseButtonWidth;
+  icon_ = new views::ImageView;
+  if (icon_id_ != 0) {
+    icon_->SetImage(
+        ResourceBundle::GetSharedInstance().GetImageSkiaNamed(icon_id_));
+  }
+
   views::ColumnSet* columns = layout->AddColumnSet(0);
-  columns->AddColumn(views::GridLayout::TRAILING, views::GridLayout::FILL,
-                     0, /* resize percent */
-                     views::GridLayout::FIXED,
-                     contents_width,
-                     contents_width);
+
+  columns->AddPaddingColumn(0, kTrayPopupPaddingHorizontal/2);
+
+  // Icon
   columns->AddColumn(views::GridLayout::CENTER, views::GridLayout::CENTER,
                      0, /* resize percent */
                      views::GridLayout::FIXED,
-                     kNotificationCloseButtonWidth,
-                     kNotificationCloseButtonWidth);
+                     kNotificationIconWidth, kNotificationIconWidth);
 
+  columns->AddPaddingColumn(0, kTrayPopupPaddingHorizontal/2);
+
+  // Contents
+  columns->AddColumn(views::GridLayout::FILL, views::GridLayout::FILL,
+                     0, /* resize percent */
+                     views::GridLayout::FIXED,
+                     kTrayNotificationContentsWidth,
+                     kTrayNotificationContentsWidth);
+
+  columns->AddPaddingColumn(0, kTrayPopupPaddingHorizontal/2);
+
+  // Close button
+  columns->AddColumn(views::GridLayout::CENTER, views::GridLayout::CENTER,
+                     0, /* resize percent */
+                     views::GridLayout::FIXED,
+                     kNotificationIconWidth, kNotificationIconWidth);
+
+  columns->AddPaddingColumn(0, kTrayPopupPaddingHorizontal/2);
+
+  // Layout rows
+  layout->AddPaddingRow(0, kTrayPopupPaddingBetweenItems);
   layout->StartRow(0, 0);
+  layout->AddView(icon_);
   layout->AddView(contents);
   layout->AddView(close_button);
+  layout->AddPaddingRow(0, kTrayPopupPaddingBetweenItems);
 }
 
-TrayNotificationView::~TrayNotificationView() {
+void TrayNotificationView::SetIconImage(const SkBitmap& image) {
+  icon_->SetImage(image);
+  SchedulePaint();
+}
+
+void TrayNotificationView::UpdateView(views::View* new_contents) {
+    RemoveAllChildViews(true);

[sadrul, 2012/05/24 04:12:50]

The indentation here and in UpdateViewAndImage seems off

[stevenjb, 2012/05/24 16:48:01]

Done.

+    InitView(new_contents);
+    Layout();
+    PreferredSizeChanged();
+    SchedulePaint();
+}
+
+void TrayNotificationView::UpdateViewAndImage(views::View* new_contents,
+                                              const SkBitmap& image) {
+    RemoveAllChildViews(true);
+    InitView(new_contents);
+    icon_->SetImage(image);
+    Layout();
+    PreferredSizeChanged();
+    SchedulePaint();
 }
 
 void TrayNotificationView::ButtonPressed(views::Button* sender,
                                          const views::Event& event) {
   OnClose();
 }
 
 }  // namespace internal
 }  // namespace ash