Separated reference counting from Resources

Former-commit-id: 7380818cfee9e249c11fd15da9ff7883a6e76565

src/Nazara/Core/RefCounted.cpp

@@ -0,0 +1,173 @@
+// Copyright (C) 2014 Jérôme Leclercq
+// This file is part of the "Nazara Engine - Core module"
+// For conditions of distribution and use, see copyright notice in Config.hpp
+
+#include <Nazara/Core/RefCounted.hpp>
+#include <Nazara/Core/Config.hpp>
+#include <Nazara/Core/Error.hpp>
+#include <Nazara/Core/ObjectListener.hpp>
+
+#if NAZARA_CORE_THREADSAFE && NAZARA_THREADSAFETY_REFCOUNTED
+	#include <Nazara/Core/ThreadSafety.hpp>
+#else
+	#include <Nazara/Core/ThreadSafetyOff.hpp>
+#endif
+
+#include <Nazara/Core/Debug.hpp>
+
+NzRefCounted::NzRefCounted(bool persistent) :
+m_persistent(persistent),
+m_referenceCount(0),
+m_objectListenersLocked(false)
+{
+}
+
+NzRefCounted::~NzRefCounted()
+{
+	m_objectListenersLocked = true;
+	for (auto& pair : m_objectListeners)
+		pair.first->OnObjectReleased(this, pair.second.first);
+
+	#if NAZARA_CORE_SAFE
+	if (m_referenceCount > 0)
+		NazaraWarning("Resource destroyed while still referenced " + NzString::Number(m_referenceCount) + " time(s)");
+	#endif
+}
+
+void NzRefCounted::AddObjectListener(NzObjectListener* listener, int index) const
+{
+	///DOC: Est ignoré si appelé depuis un évènement
+	NazaraLock(m_mutex)
+
+	if (!m_objectListenersLocked)
+	{
+		auto pair = m_objectListeners.insert(std::make_pair(listener, std::make_pair(index, 1U)));
+		if (!pair.second)
+			pair.first->second.second++;
+	}
+}
+
+void NzRefCounted::AddReference() const
+{
+	m_referenceCount++;
+}
+
+unsigned int NzRefCounted::GetReferenceCount() const
+{
+	return m_referenceCount;
+}
+
+bool NzRefCounted::IsPersistent() const
+{
+	return m_persistent;
+}
+
+void NzRefCounted::RemoveObjectListener(NzObjectListener* listener) const
+{
+	///DOC: Est ignoré si appelé depuis un évènement
+	NazaraLock(m_mutex);
+
+	if (!m_objectListenersLocked)
+	{
+		ObjectListenerMap::iterator it = m_objectListeners.find(listener);
+		if (it != m_objectListeners.end())
+			RemoveObjectListenerIterator(it);
+	}
+}
+
+bool NzRefCounted::RemoveReference() const
+{
+	#if NAZARA_CORE_SAFE
+	if (m_referenceCount == 0)
+	{
+		NazaraError("Impossible to remove reference (Ref. counter is already 0)");
+		return false;
+	}
+	#endif
+
+	if (--m_referenceCount == 0 && !m_persistent)
+	{
+		delete this; // Suicide
+
+		return true;
+	}
+	else
+		return false;
+}
+
+bool NzRefCounted::SetPersistent(bool persistent, bool checkReferenceCount)
+{
+	m_persistent = persistent;
+
+	if (checkReferenceCount && !persistent && m_referenceCount == 0)
+	{
+		delete this;
+
+		return true;
+	}
+	else
+		return false;
+}
+
+void NzRefCounted::NotifyCreated()
+{
+	NazaraLock(m_mutex)
+
+	m_objectListenersLocked = true;
+
+	auto it = m_objectListeners.begin();
+	while (it != m_objectListeners.end())
+	{
+		if (!it->first->OnObjectCreated(this, it->second.first))
+			RemoveObjectListenerIterator(it++);
+		else
+			++it;
+	}
+
+	m_objectListenersLocked = false;
+}
+
+void NzRefCounted::NotifyDestroy()
+{
+	NazaraLock(m_mutex)
+
+	m_objectListenersLocked = true;
+
+	auto it = m_objectListeners.begin();
+	while (it != m_objectListeners.end())
+	{
+		if (!it->first->OnObjectDestroy(this, it->second.first))
+			RemoveObjectListenerIterator(it++);
+		else
+			++it;
+	}
+
+	m_objectListenersLocked = false;
+}
+
+void NzRefCounted::NotifyModified(unsigned int code)
+{
+	NazaraLock(m_mutex)
+
+	m_objectListenersLocked = true;
+
+	auto it = m_objectListeners.begin();
+	while (it != m_objectListeners.end())
+	{
+		if (!it->first->OnObjectModified(this, it->second.first, code))
+			RemoveObjectListenerIterator(it++);
+		else
+			++it;
+	}
+
+	m_objectListenersLocked = false;
+}
+
+void NzRefCounted::RemoveObjectListenerIterator(ObjectListenerMap::iterator iterator) const
+{
+	unsigned int& referenceCount = iterator->second.second;
+	if (referenceCount == 1)
+		m_objectListeners.erase(iterator);
+	else
+		referenceCount--;
+}