59 lines
1.0 KiB
C++
59 lines
1.0 KiB
C++
#pragma once
|
|
|
|
#include <cstddef>
|
|
#include <memory>
|
|
|
|
class Usable;
|
|
|
|
/**
|
|
* Base class for things that can be used by multiple other things and we want to track the use count.
|
|
*
|
|
* @see UseLock
|
|
*/
|
|
class Usable
|
|
{
|
|
friend class UseLock;
|
|
public:
|
|
std::size_t useCount()
|
|
{
|
|
return m_useCount;
|
|
}
|
|
bool isInUse()
|
|
{
|
|
return m_useCount > 0;
|
|
}
|
|
protected:
|
|
virtual void decrementUses()
|
|
{
|
|
m_useCount--;
|
|
}
|
|
virtual void incrementUses()
|
|
{
|
|
m_useCount++;
|
|
}
|
|
private:
|
|
std::size_t m_useCount = 0;
|
|
};
|
|
|
|
/**
|
|
* Lock class to use for keeping track of uses of other things derived from Usable
|
|
*
|
|
* @see Usable
|
|
*/
|
|
class UseLock
|
|
{
|
|
public:
|
|
UseLock(std::shared_ptr<Usable> usable)
|
|
: m_usable(usable)
|
|
{
|
|
// this doesn't use shared pointer use count, because that wouldn't be correct. this count is separate.
|
|
m_usable->incrementUses();
|
|
}
|
|
~UseLock()
|
|
{
|
|
m_usable->decrementUses();
|
|
}
|
|
private:
|
|
std::shared_ptr<Usable> m_usable;
|
|
};
|