From e1f5fafc2c0b5f98142e3f2fec6b7ff0e1420425 Mon Sep 17 00:00:00 2001
From: bkaradzic <branimirkaradzic@gmail.com>
Date: Sun, 27 Jan 2013 21:52:29 -0800
Subject: [PATCH] Added resource leak report at shutdown.

---
 src/bgfx.cpp | 60 +++++++++++++++++++++++++++++++++++++++++++++++++++-
 src/bgfx_p.h | 15 ++++++++++---
 2 files changed, 71 insertions(+), 4 deletions(-)

diff --git a/src/bgfx.cpp b/src/bgfx.cpp
index 7833b26a..0e4d455e 100755
--- a/src/bgfx.cpp
+++ b/src/bgfx.cpp
@@ -292,11 +292,23 @@ namespace bgfx
 		FragmentShaderHandle fsh = createFragmentShader(mem);
 
 		m_program = createProgram(vsh, fsh);
+		destroyVertexShader(vsh);
+		destroyFragmentShader(fsh);
 
 		m_vb = s_ctx.createTransientVertexBuffer(numBatchVertices*m_decl.m_stride, &m_decl);
 		m_ib = s_ctx.createTransientIndexBuffer(numBatchIndices*2);
 	}
 
+	void TextVideoMemBlitter::shutdown()
+	{
+		BGFX_CHECK_MAIN_THREAD();
+
+		destroyProgram(m_program);
+		destroyTexture(m_texture);
+		s_ctx.destroyTransientVertexBuffer(m_vb);
+		s_ctx.destroyTransientIndexBuffer(m_ib);
+	}
+
 	void TextVideoMemBlitter::blit(const TextVideoMem& _mem)
 	{
 		BGFX_CHECK_RENDER_THREAD();
@@ -428,6 +440,8 @@ namespace bgfx
 		FragmentShaderHandle fsh = createFragmentShader(mem);
 
 		m_program = createProgram(vsh, fsh);
+		destroyVertexShader(vsh);
+		destroyFragmentShader(fsh);
 
 		m_vb = s_ctx.createTransientVertexBuffer(4*m_decl.m_stride, &m_decl);
 
@@ -443,6 +457,16 @@ namespace bgfx
 #endif // BGFX_CONFIG_RENDERER_DIRECT3D11
 	}
 
+	void ClearQuad::shutdown()
+	{
+		BGFX_CHECK_MAIN_THREAD();
+#if BGFX_CONFIG_RENDERER_DIRECT3D11
+		destroyProgram(m_program);
+		destroyIndexBuffer(m_ib);
+		s_ctx.destroyTransientVertexBuffer(m_vb);
+#endif // BGFX_CONFIG_RENDERER_DIRECT3D11
+	}
+
 	static const char* s_predefinedName[PredefinedUniform::Count] =
 	{
 		"u_viewRect",
@@ -702,7 +726,20 @@ namespace bgfx
 	{
 		BX_TRACE("shutdown");
 
-		getCommandBuffer(CommandBuffer::RendererShutdown);
+		frame();
+
+		getCommandBuffer(CommandBuffer::RendererShutdownBegin);
+		destroyTransientVertexBuffer(m_submit->m_transientVb);
+		destroyTransientIndexBuffer(m_submit->m_transientIb);
+		m_textVideoMemBlitter.shutdown();
+		m_clearQuad.shutdown();
+		frame();
+
+		destroyTransientVertexBuffer(m_submit->m_transientVb);
+		destroyTransientIndexBuffer(m_submit->m_transientIb);
+		frame();
+
+		getCommandBuffer(CommandBuffer::RendererShutdownEnd);
 		frame();
 
 #if BGFX_CONFIG_MULTITHREADED
@@ -714,6 +751,27 @@ namespace bgfx
 
 		m_submit->destroy();
 		m_render->destroy();
+
+#if BGFX_CONFIG_DEBUG
+#	define CHECK_HANDLE_LEAK(_handleAlloc) \
+		do { \
+			BX_WARN(0 == _handleAlloc.getNumHandles(), "LEAK: " #_handleAlloc " %d (max: %d)", _handleAlloc.getNumHandles(), _handleAlloc.getMaxHandles() ); \
+		} while (0)
+
+		CHECK_HANDLE_LEAK(m_dynamicIndexBufferHandle);
+		CHECK_HANDLE_LEAK(m_dynamicVertexBufferHandle);
+		CHECK_HANDLE_LEAK(m_indexBufferHandle);
+		CHECK_HANDLE_LEAK(m_vertexDeclHandle);
+		CHECK_HANDLE_LEAK(m_vertexBufferHandle);
+		CHECK_HANDLE_LEAK(m_vertexShaderHandle);
+		CHECK_HANDLE_LEAK(m_fragmentShaderHandle);
+		CHECK_HANDLE_LEAK(m_programHandle);
+		CHECK_HANDLE_LEAK(m_textureHandle);
+		CHECK_HANDLE_LEAK(m_renderTargetHandle);
+		CHECK_HANDLE_LEAK(m_uniformHandle);
+
+#	undef CHECK_HANDLE_LEAK
+#endif // BGFX_CONFIG_DEBUG
 	}
 
 	const Memory* alloc(uint32_t _size)
diff --git a/src/bgfx_p.h b/src/bgfx_p.h
index a530754b..d0f256fe 100755
--- a/src/bgfx_p.h
+++ b/src/bgfx_p.h
@@ -352,6 +352,7 @@ namespace bgfx
 	struct TextVideoMemBlitter
 	{
 		void init();
+		void shutdown();
 
 		void blit(const TextVideoMem* _mem)
 		{
@@ -367,11 +368,13 @@ namespace bgfx
 		TransientIndexBuffer* m_ib;
 		VertexDecl m_decl;
 		ProgramHandle m_program;
+		bool m_init;
 	};
 
 	struct ClearQuad
 	{
 		void init();
+		void shutdown();
 		void clear(const Rect& _rect, const Clear& _clear);
 
 		TransientVertexBuffer* m_vb;
@@ -417,6 +420,7 @@ namespace bgfx
 		enum Enum
 		{
 			RendererInit,
+			RendererShutdownBegin,
 			CreateVertexDecl,
 			CreateIndexBuffer,
 			CreateVertexBuffer,
@@ -432,7 +436,7 @@ namespace bgfx
 			CreateRenderTarget,
 			CreateUniform,
 			End,
-			RendererShutdown,
+			RendererShutdownEnd,
 			DestroyVertexDecl,
 			DestroyIndexBuffer,
 			DestroyVertexBuffer,
@@ -2415,10 +2419,15 @@ namespace bgfx
 					}
 					break;
 
-				case CommandBuffer::RendererShutdown:
+				case CommandBuffer::RendererShutdownBegin:
+					{
+						m_rendererInitialized = false;
+					}
+					break;
+
+				case CommandBuffer::RendererShutdownEnd:
 					{
 						rendererShutdown();
-						m_rendererInitialized = false;
 						m_exit = true;
 					}
 					break;