OLD | NEW |
| (Empty) |
1 // Copyright 2016 PDFium 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 CORE_FXCRT_INCLUDE_CFX_OBSERVABLE_H_ | |
6 #define CORE_FXCRT_INCLUDE_CFX_OBSERVABLE_H_ | |
7 | |
8 #include <set> | |
9 | |
10 #include "core/fxcrt/include/fx_system.h" | |
11 #include "third_party/base/stl_util.h" | |
12 | |
13 template <class T> | |
14 class CFX_Observable { | |
15 public: | |
16 class ObservedPtr { | |
17 public: | |
18 ObservedPtr() : m_pObservedPtr(nullptr) {} | |
19 explicit ObservedPtr(T* pObservedPtr) : m_pObservedPtr(pObservedPtr) { | |
20 if (m_pObservedPtr) | |
21 m_pObservedPtr->AddObservedPtr(this); | |
22 } | |
23 ObservedPtr(const ObservedPtr& that) = delete; | |
24 ~ObservedPtr() { | |
25 if (m_pObservedPtr) | |
26 m_pObservedPtr->RemoveObservedPtr(this); | |
27 } | |
28 void Reset(T* pObservedPtr = nullptr) { | |
29 if (m_pObservedPtr) | |
30 m_pObservedPtr->RemoveObservedPtr(this); | |
31 m_pObservedPtr = pObservedPtr; | |
32 if (m_pObservedPtr) | |
33 m_pObservedPtr->AddObservedPtr(this); | |
34 } | |
35 void OnDestroy() { | |
36 ASSERT(m_pObservedPtr); | |
37 m_pObservedPtr = nullptr; | |
38 } | |
39 ObservedPtr& operator=(const ObservedPtr& that) = delete; | |
40 bool operator==(const ObservedPtr& that) const { | |
41 return m_pObservedPtr == that.m_pObservedPtr; | |
42 } | |
43 bool operator!=(const ObservedPtr& that) const { return !(*this == that); } | |
44 explicit operator bool() const { return !!m_pObservedPtr; } | |
45 T* Get() const { return m_pObservedPtr; } | |
46 T& operator*() const { return *m_pObservedPtr; } | |
47 T* operator->() const { return m_pObservedPtr; } | |
48 | |
49 private: | |
50 T* m_pObservedPtr; | |
51 }; | |
52 | |
53 CFX_Observable() {} | |
54 CFX_Observable(const CFX_Observable& that) = delete; | |
55 ~CFX_Observable() { NotifyObservedPtrs(); } | |
56 void AddObservedPtr(ObservedPtr* pObservedPtr) { | |
57 ASSERT(!pdfium::ContainsKey(m_ObservedPtrs, pObservedPtr)); | |
58 m_ObservedPtrs.insert(pObservedPtr); | |
59 } | |
60 void RemoveObservedPtr(ObservedPtr* pObservedPtr) { | |
61 ASSERT(pdfium::ContainsKey(m_ObservedPtrs, pObservedPtr)); | |
62 m_ObservedPtrs.erase(pObservedPtr); | |
63 } | |
64 void NotifyObservedPtrs() { | |
65 for (auto* pObservedPtr : m_ObservedPtrs) | |
66 pObservedPtr->OnDestroy(); | |
67 m_ObservedPtrs.clear(); | |
68 } | |
69 CFX_Observable& operator=(const CFX_Observable& that) = delete; | |
70 | |
71 protected: | |
72 size_t ActiveObservedPtrsForTesting() const { return m_ObservedPtrs.size(); } | |
73 | |
74 private: | |
75 std::set<ObservedPtr*> m_ObservedPtrs; | |
76 }; | |
77 | |
78 #endif // CORE_FXCRT_INCLUDE_CFX_OBSERVABLE_H_ | |
OLD | NEW |