blob: cb39616f2bbfe151ac8f627b85488a42d61090bb [file] [log] [blame]
// 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/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;
};
#endif // CORE_FPDFAPI_PAGE_CPDF_COUNTEDOBJECT_H_