1 files changed, 10 insertions, 1 deletions
diff --git a/chrome/browser/views/notifications/balloon_view.h b/chrome/browser/views/notifications/balloon_view.h
index 84147ad..6d8da54 100644
--- a/chrome/browser/views/notifications/balloon_view.h
+++ b/chrome/browser/views/notifications/balloon_view.h
@@ -23,6 +23,7 @@
 #include "views/controls/label.h"
 #include "views/controls/menu/view_menu_delegate.h"
 #include "views/view.h"
+#include "views/widget/widget_delegate.h"
 
 namespace views {
 class ButtonListener;
@@ -32,6 +33,7 @@ class WidgetWin;
 class Menu2;
 }  // namespace views
 
+class BalloonCollection;
 class BalloonViewHost;
 class NotificationDetails;
 class NotificationSource;
@@ -42,11 +44,12 @@ class SlideAnimation;
 class BalloonViewImpl : public BalloonView,
                         public views::View,
                         public views::ViewMenuDelegate,
+                        public views::WidgetDelegate,
                         public menus::SimpleMenuModel::Delegate,
                         public NotificationObserver,
                         public AnimationDelegate {
  public:
-  BalloonViewImpl();
+  explicit BalloonViewImpl(BalloonCollection* collection);
   ~BalloonViewImpl();
 
   // BalloonView interface.
@@ -67,6 +70,9 @@ class BalloonViewImpl : public BalloonView,
   // views::ViewMenuDelegate interface.
   void RunMenu(views::View* source, const gfx::Point& pt);
 
+  // views::WidgetDelegate interface.
+  void DisplayChanged();
+
   // menus::SimpleMenuModel::Delegate interface.
   virtual bool IsCommandIdChecked(int command_id) const;
   virtual bool IsCommandIdEnabled(int command_id) const;
@@ -128,6 +134,9 @@ class BalloonViewImpl : public BalloonView,
   // Non-owned pointer to the balloon which owns this object.
   Balloon* balloon_;
 
+  // Non-owned pointer to the balloon collection this is a part of.
+  BalloonCollection* collection_;
+
   // The window that contains the frame of the notification.
   // Pointer owned by the View subclass.
   views::Widget* frame_container_;