ui/views/widget/widget_deletion_observer.cc


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

// Copyright (c) 2013 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 "ui/views/widget/widget_deletion_observer.h"

#include "ui/views/widget/widget.h"

namespace views {

WidgetDeletionObserver::WidgetDeletionObserver(Widget* widget)
    : widget_(widget) {
  if (widget_)
    widget_->AddObserver(this);
}

WidgetDeletionObserver::~WidgetDeletionObserver() {
  CleanupWidget();
}

void WidgetDeletionObserver::OnWidgetDestroying(Widget* widget) {
  CleanupWidget();
}

void WidgetDeletionObserver::CleanupWidget() {
  if (widget_) {
    widget_->RemoveObserver(this);
    widget_ = NULL;
  }
}

}  // namespace views