| 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 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | |
| 6 | |
| 7 #ifndef CORE_FPDFAPI_FPDF_PAGE_CPDF_COUNTEDOBJECT_H_ | |
| 8 #define CORE_FPDFAPI_FPDF_PAGE_CPDF_COUNTEDOBJECT_H_ | |
| 9 | |
| 10 #include "core/fpdfapi/fpdf_page/cpdf_pattern.h" | |
| 11 #include "core/fpdfapi/fpdf_page/include/cpdf_colorspace.h" | |
| 12 #include "core/fxcrt/include/fx_system.h" | |
| 13 | |
| 14 template <class T> | |
| 15 class CPDF_CountedObject { | |
| 16 public: | |
| 17 explicit CPDF_CountedObject(T* ptr) : m_nCount(1), m_pObj(ptr) {} | |
| 18 void reset(T* ptr) { // CAUTION: tosses prior ref counts. | |
| 19 m_nCount = 1; | |
| 20 m_pObj = ptr; | |
| 21 } | |
| 22 void clear() { // Now you're all weak ptrs ... | |
| 23 // Guard against accidental re-entry. | |
| 24 T* pObj = m_pObj; | |
| 25 m_pObj = nullptr; | |
| 26 delete pObj; | |
| 27 } | |
| 28 T* get() const { return m_pObj; } | |
| 29 T* AddRef() { | |
| 30 ASSERT(m_pObj); | |
| 31 ++m_nCount; | |
| 32 return m_pObj; | |
| 33 } | |
| 34 void RemoveRef() { | |
| 35 if (m_nCount) | |
| 36 --m_nCount; | |
| 37 } | |
| 38 size_t use_count() const { return m_nCount; } | |
| 39 | |
| 40 protected: | |
| 41 size_t m_nCount; | |
| 42 T* m_pObj; | |
| 43 }; | |
| 44 using CPDF_CountedColorSpace = CPDF_CountedObject<CPDF_ColorSpace>; | |
| 45 using CPDF_CountedPattern = CPDF_CountedObject<CPDF_Pattern>; | |
| 46 | |
| 47 #endif // CORE_FPDFAPI_FPDF_PAGE_CPDF_COUNTEDOBJECT_H_ | |
| OLD | NEW |