| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium 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 CC_BASE_REF_COUNTED_MANAGED_H_ | 5 #ifndef CC_BASE_REF_COUNTED_MANAGED_H_ |
| 6 #define CC_BASE_REF_COUNTED_MANAGED_H_ | 6 #define CC_BASE_REF_COUNTED_MANAGED_H_ |
| 7 | 7 |
| 8 #include "base/logging.h" | 8 #include "base/logging.h" |
| 9 #include "base/memory/ref_counted.h" | 9 #include "base/memory/ref_counted.h" |
| 10 #include "cc/base/cc_export.h" | 10 #include "cc/base/cc_export.h" |
| (...skipping 14 matching lines...) Expand all Loading... |
| 25 | 25 |
| 26 private: | 26 private: |
| 27 friend class RefCountedManaged<T>; | 27 friend class RefCountedManaged<T>; |
| 28 int live_object_count_; | 28 int live_object_count_; |
| 29 }; | 29 }; |
| 30 | 30 |
| 31 template <typename T> | 31 template <typename T> |
| 32 class CC_EXPORT RefCountedManaged : public base::subtle::RefCountedBase { | 32 class CC_EXPORT RefCountedManaged : public base::subtle::RefCountedBase { |
| 33 public: | 33 public: |
| 34 explicit RefCountedManaged(RefCountedManager<T>* manager) | 34 explicit RefCountedManaged(RefCountedManager<T>* manager) |
| 35 : manager_(manager) { | 35 : manager_(manager), |
| 36 manager_->live_object_count_++; | 36 released_(true) { |
| 37 } | 37 } |
| 38 | 38 |
| 39 void AddRef() const { | 39 void AddRef() const { |
| 40 if (released_) { |
| 41 released_ = false; |
| 42 ClearInDtor(); |
| 43 manager_->live_object_count_++; |
| 44 } |
| 40 base::subtle::RefCountedBase::AddRef(); | 45 base::subtle::RefCountedBase::AddRef(); |
| 41 } | 46 } |
| 42 | 47 |
| 43 void Release() { | 48 void Release() { |
| 44 if (base::subtle::RefCountedBase::Release()) { | 49 if (base::subtle::RefCountedBase::Release()) { |
| 45 DCHECK_GT(manager_->live_object_count_, 0); | 50 DCHECK_GT(manager_->live_object_count_, 0); |
| 46 manager_->live_object_count_--; | 51 manager_->live_object_count_--; |
| 47 | 52 |
| 53 released_ = true; |
| 54 |
| 48 // This must be the last statement in case manager deletes | 55 // This must be the last statement in case manager deletes |
| 49 // the object immediately. | 56 // the object immediately. |
| 50 manager_->Release(static_cast<T*>(this)); | 57 manager_->Release(static_cast<T*>(this)); |
| 51 } | 58 } |
| 52 } | 59 } |
| 53 | 60 |
| 54 protected: | 61 protected: |
| 55 ~RefCountedManaged() {} | 62 ~RefCountedManaged() {} |
| 56 | 63 |
| 57 private: | 64 private: |
| 58 RefCountedManager<T>* manager_; | 65 RefCountedManager<T>* manager_; |
| 66 mutable bool released_; |
| 59 | 67 |
| 60 DISALLOW_COPY_AND_ASSIGN(RefCountedManaged<T>); | 68 DISALLOW_COPY_AND_ASSIGN(RefCountedManaged<T>); |
| 61 }; | 69 }; |
| 62 | 70 |
| 63 } // namespace cc | 71 } // namespace cc |
| 64 | 72 |
| 65 #endif // CC_BASE_REF_COUNTED_MANAGED_H_ | 73 #endif // CC_BASE_REF_COUNTED_MANAGED_H_ |
| OLD | NEW |