OLD | NEW |
1 // Copyright 2016 PDFium Authors. All rights reserved. | 1 // Copyright 2016 PDFium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef CORE_FXCRT_CFX_OBSERVABLE_H_ | 5 #ifndef CORE_FXCRT_CFX_OBSERVABLE_H_ |
6 #define CORE_FXCRT_CFX_OBSERVABLE_H_ | 6 #define CORE_FXCRT_CFX_OBSERVABLE_H_ |
7 | 7 |
8 #include <set> | 8 #include <set> |
9 | 9 |
10 #include "core/fxcrt/fx_system.h" | 10 #include "core/fxcrt/fx_system.h" |
11 #include "third_party/base/stl_util.h" | 11 #include "third_party/base/stl_util.h" |
12 | 12 |
13 template <class T> | 13 template <class T> |
14 class CFX_Observable { | 14 class CFX_Observable { |
15 public: | 15 public: |
16 class ObservedPtr { | 16 class ObservedPtr { |
17 public: | 17 public: |
18 ObservedPtr() : m_pObservedPtr(nullptr) {} | 18 ObservedPtr() : m_pObservable(nullptr) {} |
19 explicit ObservedPtr(T* pObservedPtr) : m_pObservedPtr(pObservedPtr) { | 19 explicit ObservedPtr(T* pObservable) : m_pObservable(pObservable) { |
20 if (m_pObservedPtr) | 20 if (m_pObservable) |
21 m_pObservedPtr->AddObservedPtr(this); | 21 m_pObservable->AddObservedPtr(this); |
22 } | 22 } |
23 ObservedPtr(const ObservedPtr& that) = delete; | 23 ObservedPtr(const ObservedPtr& that) : ObservedPtr(that.Get()) {} |
24 ~ObservedPtr() { | 24 ~ObservedPtr() { |
25 if (m_pObservedPtr) | 25 if (m_pObservable) |
26 m_pObservedPtr->RemoveObservedPtr(this); | 26 m_pObservable->RemoveObservedPtr(this); |
27 } | 27 } |
28 void Reset(T* pObservedPtr = nullptr) { | 28 void Reset(T* pObservable = nullptr) { |
29 if (m_pObservedPtr) | 29 if (m_pObservable) |
30 m_pObservedPtr->RemoveObservedPtr(this); | 30 m_pObservable->RemoveObservedPtr(this); |
31 m_pObservedPtr = pObservedPtr; | 31 m_pObservable = pObservable; |
32 if (m_pObservedPtr) | 32 if (m_pObservable) |
33 m_pObservedPtr->AddObservedPtr(this); | 33 m_pObservable->AddObservedPtr(this); |
34 } | 34 } |
35 void OnDestroy() { | 35 void OnDestroy() { |
36 ASSERT(m_pObservedPtr); | 36 ASSERT(m_pObservable); |
37 m_pObservedPtr = nullptr; | 37 m_pObservable = nullptr; |
38 } | 38 } |
39 ObservedPtr& operator=(const ObservedPtr& that) = delete; | 39 ObservedPtr& operator=(const ObservedPtr& that) { |
| 40 Reset(that.Get()); |
| 41 return *this; |
| 42 } |
40 bool operator==(const ObservedPtr& that) const { | 43 bool operator==(const ObservedPtr& that) const { |
41 return m_pObservedPtr == that.m_pObservedPtr; | 44 return m_pObservable == that.m_pObservable; |
42 } | 45 } |
43 bool operator!=(const ObservedPtr& that) const { return !(*this == that); } | 46 bool operator!=(const ObservedPtr& that) const { return !(*this == that); } |
44 explicit operator bool() const { return !!m_pObservedPtr; } | 47 explicit operator bool() const { return !!m_pObservable; } |
45 T* Get() const { return m_pObservedPtr; } | 48 T* Get() const { return m_pObservable; } |
46 T& operator*() const { return *m_pObservedPtr; } | 49 T& operator*() const { return *m_pObservable; } |
47 T* operator->() const { return m_pObservedPtr; } | 50 T* operator->() const { return m_pObservable; } |
48 | 51 |
49 private: | 52 private: |
50 T* m_pObservedPtr; | 53 T* m_pObservable; |
51 }; | 54 }; |
52 | 55 |
53 CFX_Observable() {} | 56 CFX_Observable() {} |
54 CFX_Observable(const CFX_Observable& that) = delete; | 57 CFX_Observable(const CFX_Observable& that) = delete; |
55 ~CFX_Observable() { NotifyObservedPtrs(); } | 58 ~CFX_Observable() { NotifyObservedPtrs(); } |
56 void AddObservedPtr(ObservedPtr* pObservedPtr) { | 59 void AddObservedPtr(ObservedPtr* pObservedPtr) { |
57 ASSERT(!pdfium::ContainsKey(m_ObservedPtrs, pObservedPtr)); | 60 ASSERT(!pdfium::ContainsKey(m_ObservedPtrs, pObservedPtr)); |
58 m_ObservedPtrs.insert(pObservedPtr); | 61 m_ObservedPtrs.insert(pObservedPtr); |
59 } | 62 } |
60 void RemoveObservedPtr(ObservedPtr* pObservedPtr) { | 63 void RemoveObservedPtr(ObservedPtr* pObservedPtr) { |
61 ASSERT(pdfium::ContainsKey(m_ObservedPtrs, pObservedPtr)); | 64 ASSERT(pdfium::ContainsKey(m_ObservedPtrs, pObservedPtr)); |
62 m_ObservedPtrs.erase(pObservedPtr); | 65 m_ObservedPtrs.erase(pObservedPtr); |
63 } | 66 } |
64 void NotifyObservedPtrs() { | 67 void NotifyObservedPtrs() { |
65 for (auto* pObservedPtr : m_ObservedPtrs) | 68 for (auto* pObservedPtr : m_ObservedPtrs) |
66 pObservedPtr->OnDestroy(); | 69 pObservedPtr->OnDestroy(); |
67 m_ObservedPtrs.clear(); | 70 m_ObservedPtrs.clear(); |
68 } | 71 } |
69 CFX_Observable& operator=(const CFX_Observable& that) = delete; | 72 CFX_Observable& operator=(const CFX_Observable& that) = delete; |
70 | 73 |
71 protected: | 74 protected: |
72 size_t ActiveObservedPtrsForTesting() const { return m_ObservedPtrs.size(); } | 75 size_t ActiveObservedPtrsForTesting() const { return m_ObservedPtrs.size(); } |
73 | 76 |
74 private: | 77 private: |
75 std::set<ObservedPtr*> m_ObservedPtrs; | 78 std::set<ObservedPtr*> m_ObservedPtrs; |
76 }; | 79 }; |
77 | 80 |
78 #endif // CORE_FXCRT_CFX_OBSERVABLE_H_ | 81 #endif // CORE_FXCRT_CFX_OBSERVABLE_H_ |
OLD | NEW |