| 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 |