blob: fe1d2607c847b5faab0b849c5008c4d45d7a6bfb (
plain)
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
// Copyright 2020 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/unique_widget_ptr.h"
#include <utility>
#include "base/scoped_observer.h"
#include "ui/views/view.h"
#include "ui/views/widget/widget.h"
#include "ui/views/widget/widget_observer.h"
namespace views {
namespace {
struct WidgetAutoCloser {
void operator()(Widget* widget) {
widget->CloseWithReason(Widget::ClosedReason::kUnspecified);
}
};
using WidgetAutoClosePtr = std::unique_ptr<Widget, WidgetAutoCloser>;
} // namespace
class UniqueWidgetPtr::UniqueWidgetPtrImpl : public WidgetObserver {
public:
UniqueWidgetPtrImpl() = default;
// This class acts like unique_ptr<Widget>, so this constructor is
// deliberately implicit.
UniqueWidgetPtrImpl(std::unique_ptr<Widget> widget) // NOLINT
: widget_closer_(widget.release()) {
widget_observer_.Add(widget_closer_.get());
}
UniqueWidgetPtrImpl(const UniqueWidgetPtrImpl&) = delete;
UniqueWidgetPtrImpl& operator=(const UniqueWidgetPtrImpl&) = delete;
~UniqueWidgetPtrImpl() override = default;
Widget* Get() const { return widget_closer_.get(); }
void Reset() {
if (!widget_closer_)
return;
widget_observer_.RemoveAll();
widget_closer_.reset();
}
// WidgetObserver overrides.
void OnWidgetDestroying(Widget* widget) override {
DCHECK_EQ(widget, widget_closer_.get());
widget_observer_.RemoveAll();
widget_closer_.release();
}
private:
ScopedObserver<Widget, WidgetObserver> widget_observer_{this};
WidgetAutoClosePtr widget_closer_;
};
UniqueWidgetPtr::UniqueWidgetPtr() = default;
UniqueWidgetPtr::UniqueWidgetPtr(std::unique_ptr<Widget> widget)
: unique_widget_ptr_impl_(
std::make_unique<UniqueWidgetPtrImpl>(std::move(widget))) {}
UniqueWidgetPtr::UniqueWidgetPtr(UniqueWidgetPtr&& other) = default;
UniqueWidgetPtr& UniqueWidgetPtr::operator=(UniqueWidgetPtr&& other) = default;
UniqueWidgetPtr::~UniqueWidgetPtr() = default;
UniqueWidgetPtr::operator bool() const {
return !!get();
}
Widget& UniqueWidgetPtr::operator*() const {
return *get();
}
Widget* UniqueWidgetPtr::operator->() const {
return get();
}
void UniqueWidgetPtr::reset() {
unique_widget_ptr_impl_.reset();
}
Widget* UniqueWidgetPtr::get() const {
return unique_widget_ptr_impl_ ? unique_widget_ptr_impl_->Get() : nullptr;
}
} // namespace views
|