blob: 64f936c52c4b0026bd3f5cb69a2f5cfa7ea85878 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
// Copyright 2016 PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
#ifndef CORE_FPDFAPI_PAGE_CPDF_COUNTEDOBJECT_H_
#define CORE_FPDFAPI_PAGE_CPDF_COUNTEDOBJECT_H_
#include <memory>
#include "core/fpdfapi/page/cpdf_colorspace.h"
#include "core/fpdfapi/page/cpdf_pattern.h"
#include "core/fxcrt/fx_system.h"
template <class T>
class CPDF_CountedObject {
public:
explicit CPDF_CountedObject(std::unique_ptr<T> ptr)
: m_nCount(1), m_pObj(ptr.release()) {}
void reset(std::unique_ptr<T> ptr) { // CAUTION: tosses prior ref counts.
m_nCount = 1;
m_pObj = ptr.release();
}
void clear() { // Now you're all weak ptrs ...
// Guard against accidental re-entry.
T* pObj = m_pObj;
m_pObj = nullptr;
delete pObj;
}
T* get() const { return m_pObj; }
T* AddRef() {
ASSERT(m_pObj);
++m_nCount;
return m_pObj;
}
void RemoveRef() {
if (m_nCount)
--m_nCount;
}
size_t use_count() const { return m_nCount; }
protected:
size_t m_nCount;
T* m_pObj;
};
using CPDF_CountedColorSpace = CPDF_CountedObject<CPDF_ColorSpace>;
using CPDF_CountedPattern = CPDF_CountedObject<CPDF_Pattern>;
#endif // CORE_FPDFAPI_PAGE_CPDF_COUNTEDOBJECT_H_
|
