Branimir Karadžić 11 лет назад
Родитель
Сommit
8cd9ecfd18
1 измененных файлов с 39 добавлено и 31 удалено
  1. 39 31
      examples/common/imgui/ocornut_imgui.cpp

+ 39 - 31
examples/common/imgui/ocornut_imgui.cpp

@@ -21,9 +21,6 @@ static uint8_t s_viewId;
 
 static void imguiRender(ImDrawList** const cmd_lists, int cmd_lists_count)
 {
-	(void)cmd_lists;
-	(void)cmd_lists_count;
-
 	const float width  = ImGui::GetIO().DisplaySize.x;
 	const float height = ImGui::GetIO().DisplaySize.y;
 
@@ -33,7 +30,6 @@ static void imguiRender(ImDrawList** const cmd_lists, int cmd_lists_count)
 	bgfx::setViewTransform(0, NULL, ortho);
 
 	// Render command lists
-
 	for (int n = 0; n < cmd_lists_count; n++)
 	{
 		bgfx::TransientVertexBuffer tvb;
@@ -42,22 +38,28 @@ static void imguiRender(ImDrawList** const cmd_lists, int cmd_lists_count)
 
 		const ImDrawList* cmd_list = cmd_lists[n];
 		const ImDrawVert* vtx_buffer = cmd_list->vtx_buffer.begin();
-		(void)vtx_buffer;
-
-		const ImDrawCmd* pcmd_end_t = cmd_list->commands.end();
 
-		for (const ImDrawCmd* pcmd = cmd_list->commands.begin(); pcmd != pcmd_end_t; pcmd++)
+		const ImDrawCmd* pcmd_begin = cmd_list->commands.begin();
+		const ImDrawCmd* pcmd_end   = cmd_list->commands.end();
+		for (const ImDrawCmd* pcmd = pcmd_begin; pcmd != pcmd_end; pcmd++)
+		{
 			vtx_size += (uint32_t)pcmd->vtx_count;
+		}
+
+		if (!bgfx::checkAvailTransientVertexBuffer(vtx_size, s_vertexDecl) )
+		{
+			// not enough space in transient buffer just quit drawing the rest...
+			break;
+		}
 
 		bgfx::allocTransientVertexBuffer(&tvb, vtx_size, s_vertexDecl);
 
 		ImDrawVert* verts = (ImDrawVert*)tvb.data;
 
-		memcpy(verts, vtx_buffer, vtx_size * sizeof(ImDrawVert));
+		memcpy(verts, vtx_buffer, vtx_size * sizeof(ImDrawVert) );
 
 		uint32_t vtx_offset = 0;
-		const ImDrawCmd* pcmd_end = cmd_list->commands.end();
-		for (const ImDrawCmd* pcmd = cmd_list->commands.begin(); pcmd != pcmd_end; pcmd++)
+		for (const ImDrawCmd* pcmd = pcmd_begin; pcmd != pcmd_end; pcmd++)
 		{
 			bgfx::setState(0
 				| BGFX_STATE_RGB_WRITE
@@ -81,36 +83,36 @@ void IMGUI_setup(int _width, int _height, uint8_t _viewId)
 	unsigned int png_size;
 	ImGuiIO& io = ImGui::GetIO();
 
-	io.DisplaySize = ImVec2((float)_width, (float)_height);
+	io.DisplaySize = ImVec2( (float)_width, (float)_height);
 	io.DeltaTime = 1.0f / 60.0f;
 	io.PixelCenterOffset = 0.0f;
 
-	const bgfx::Memory* vs_imgui2;
-	const bgfx::Memory* fs_imgui2;
+	const bgfx::Memory* vsmem;
+	const bgfx::Memory* fsmem;
 
-	switch (bgfx::getRendererType())
+	switch (bgfx::getRendererType() )
 	{
 	case bgfx::RendererType::Direct3D9:
-		vs_imgui2 = bgfx::makeRef(vs_ocornut_imgui_dx9, sizeof(vs_ocornut_imgui_dx9));
-		fs_imgui2 = bgfx::makeRef(fs_ocornut_imgui_dx9, sizeof(fs_ocornut_imgui_dx9));
+		vsmem = bgfx::makeRef(vs_ocornut_imgui_dx9, sizeof(vs_ocornut_imgui_dx9) );
+		fsmem = bgfx::makeRef(fs_ocornut_imgui_dx9, sizeof(fs_ocornut_imgui_dx9) );
 		break;
 
 	case bgfx::RendererType::Direct3D11:
-		vs_imgui2 = bgfx::makeRef(vs_ocornut_imgui_dx11, sizeof(vs_ocornut_imgui_dx11));
-		fs_imgui2 = bgfx::makeRef(fs_ocornut_imgui_dx11, sizeof(fs_ocornut_imgui_dx11));
+		vsmem = bgfx::makeRef(vs_ocornut_imgui_dx11, sizeof(vs_ocornut_imgui_dx11) );
+		fsmem = bgfx::makeRef(fs_ocornut_imgui_dx11, sizeof(fs_ocornut_imgui_dx11) );
 		break;
 
 	default:
-		vs_imgui2 = bgfx::makeRef(vs_ocornut_imgui_glsl, sizeof(vs_ocornut_imgui_glsl));
-		fs_imgui2 = bgfx::makeRef(fs_ocornut_imgui_glsl, sizeof(fs_ocornut_imgui_glsl));
+		vsmem = bgfx::makeRef(vs_ocornut_imgui_glsl, sizeof(vs_ocornut_imgui_glsl) );
+		fsmem = bgfx::makeRef(fs_ocornut_imgui_glsl, sizeof(fs_ocornut_imgui_glsl) );
 		break;
 	}
 
 	bgfx::ShaderHandle vsh;
 	bgfx::ShaderHandle fsh;
 
-	vsh = bgfx::createShader(vs_imgui2);
-	fsh = bgfx::createShader(fs_imgui2);
+	vsh = bgfx::createShader(vsmem);
+	fsh = bgfx::createShader(fsmem);
 	s_imguiProgram = bgfx::createProgram(vsh, fsh);
 	bgfx::destroyShader(vsh);
 	bgfx::destroyShader(fsh);
@@ -126,16 +128,22 @@ void IMGUI_setup(int _width, int _height, uint8_t _viewId)
 	s_tex = bgfx::createUniform("s_tex", bgfx::UniformType::Uniform1i);
 
 	ImGui::GetDefaultFontData(NULL, NULL, &png_data, &png_size);
-	int tex_x, tex_y, pitch, tex_comp;
 
-	void* tex_data = stbi_load_from_memory((const unsigned char*)png_data, (int)png_size, &tex_x, &tex_y, &tex_comp, 0);
+	int tex_x, tex_y, pitch, tex_comp;
+	void* tex_data = stbi_load_from_memory( (const unsigned char*)png_data, (int)png_size, &tex_x, &tex_y, &tex_comp, 0);
 
 	pitch = tex_x * 4;
 
-	const bgfx::Memory* mem = bgfx::alloc((uint32_t)(tex_y * pitch));
-	memcpy(mem->data, tex_data, size_t(pitch * tex_y));
+	const bgfx::Memory* mem = bgfx::alloc( (uint32_t)(tex_y * pitch) );
+	memcpy(mem->data, tex_data, size_t(pitch * tex_y) );
 
-	s_textureId = bgfx::createTexture2D((uint16_t)tex_x, (uint16_t)tex_y, 1, bgfx::TextureFormat::BGRA8, BGFX_TEXTURE_MIN_POINT | BGFX_TEXTURE_MAG_POINT, mem);
+	s_textureId = bgfx::createTexture2D( (uint16_t)tex_x
+		, (uint16_t)tex_y
+		, 1
+		, bgfx::TextureFormat::BGRA8
+		, BGFX_TEXTURE_MIN_POINT | BGFX_TEXTURE_MAG_POINT
+		, mem
+		);
 
 	stbi_image_free(tex_data);
 
@@ -146,7 +154,7 @@ void IMGUI_updateSize(int width, int height)
 {
 	ImGuiIO& io = ImGui::GetIO();
 
-	io.DisplaySize = ImVec2((float)width, (float)height);
+	io.DisplaySize = ImVec2( (float)width, (float)height);
 	io.DeltaTime = 1.0f / 60.0f;
 	io.PixelCenterOffset = 0.0f;
 }
@@ -173,7 +181,7 @@ void IMGUI_setMouse(float x, float y, int mouseLmb)
 void IMGUI_setKeyDown(int key, int /*modifier*/)
 {
 	ImGuiIO& io = ImGui::GetIO();
-//	assert(key >= 0 && key <= (int)sizeof_array(io.KeysDown));
+//	assert(key >= 0 && key <= (int)sizeof_array(io.KeysDown) );
 	io.KeysDown[key] = true;
 // 	io.KeyCtrl = !!(modifier & PDKEY_CTRL);
 // 	io.KeyShift = !!(modifier & PDKEY_SHIFT);
@@ -182,7 +190,7 @@ void IMGUI_setKeyDown(int key, int /*modifier*/)
 void IMGUI_setKeyUp(int key, int /*modifier*/)
 {
 	ImGuiIO& io = ImGui::GetIO();
-//	assert(key >= 0 && key <= (int)sizeof_array(io.KeysDown));
+//	assert(key >= 0 && key <= (int)sizeof_array(io.KeysDown) );
 	io.KeysDown[key] = false;
 // 	io.KeyCtrl = !!(modifier & PDKEY_CTRL);
 // 	io.KeyShift = !!(modifier & PDKEY_SHIFT);