X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fbackends%2Fvulkan%2Fprogram_backend.cpp;h=993a3e38ef965d11b838fb9b41653cb2fe5b3b29;hb=cebf1330ef6773b7b4496dc279ec02a7ca4351bb;hp=60e96698bd313242374911112359a01c117ba446;hpb=27c06dbcd3af468933170e9f8c738b53b53e0277;p=libs%2Fgl.git

diff --git a/source/backends/vulkan/program_backend.cpp b/source/backends/vulkan/program_backend.cpp
index 60e96698..993a3e38 100644
--- a/source/backends/vulkan/program_backend.cpp
+++ b/source/backends/vulkan/program_backend.cpp
@@ -22,7 +22,8 @@ VulkanProgram::VulkanProgram(VulkanProgram &&other):
 	stage_flags(other.stage_flags),
 	creation_info(move(other.creation_info)),
 	desc_set_layout_handles(move(other.desc_set_layout_handles)),
-	layout_handle(other.layout_handle)
+	layout_handle(other.layout_handle),
+	debug_name(move(other.debug_name))
 {
 	other.desc_set_layout_handles.clear();
 	other.layout_handle = 0;
@@ -43,7 +44,7 @@ bool VulkanProgram::has_stages() const
 	return n_stages;
 }
 
-void VulkanProgram::add_glsl_stages(const GlslModule &, const map<string, int> &, TransientData &)
+void VulkanProgram::add_glsl_stages(const GlslModule &, const map<string, int> &)
 {
 	throw invalid_operation("VulkanProgram::add_glsl_stages");
 }
@@ -58,11 +59,11 @@ void VulkanProgram::add_spirv_stages(const SpirVModule &mod, const map<string, i
 		entry_names_size += e.name.size()+1;
 
 	StructureBuilder sb(creation_info, 5);
-	VkPipelineShaderStageCreateInfo *&stage_infos = sb.add<VkPipelineShaderStageCreateInfo>(n_stages);
-	char *&name_table = sb.add<char>(entry_names_size);
-	VkSpecializationInfo *&spec_info = sb.add<VkSpecializationInfo>();
-	VkSpecializationMapEntry *&spec_map = sb.add<VkSpecializationMapEntry>(spec_values.size());
-	int *&spec_data = sb.add<int>(spec_values.size());
+	VkPipelineShaderStageCreateInfo *const &stage_infos = sb.add<VkPipelineShaderStageCreateInfo>(n_stages);
+	char *const &name_table = sb.add<char>(entry_names_size);
+	VkSpecializationInfo *const &spec_info = sb.add<VkSpecializationInfo>();
+	VkSpecializationMapEntry *const &spec_map = sb.add<VkSpecializationMapEntry>(spec_values.size());
+	int *const &spec_data = sb.add<int>(spec_values.size());
 
 	unsigned i = 0;
 	for(const SpirVModule::Constant &c: mod.get_spec_constants())
@@ -99,6 +100,12 @@ void VulkanProgram::add_spirv_stages(const SpirVModule &mod, const map<string, i
 		stage_infos[i].pSpecializationInfo = spec_info;
 		++i;
 	}
+
+#if DEBUG
+	if(!debug_name.empty())
+		if(SpirVModule *spirv = static_cast<Program *>(this)->specialized_spirv)
+			spirv->set_debug_name(debug_name);
+#endif
 }
 
 void VulkanProgram::finalize_uniforms()
@@ -118,10 +125,10 @@ void VulkanProgram::finalize_uniforms()
 			{
 				bindings.emplace_back();
 				VkDescriptorSetLayoutBinding &binding = bindings.back();
-				binding.binding = b.bind_point;
+				binding.binding = b.bind_point&0xFFFFF;
 				binding.descriptorType = VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER;
 				binding.descriptorCount = 1;
-				binding.stageFlags = VK_SHADER_STAGE_ALL;
+				binding.stageFlags = stage_flags;
 				binding.pImmutableSamplers = 0;
 			}
 
@@ -130,10 +137,13 @@ void VulkanProgram::finalize_uniforms()
 			{
 				bindings.emplace_back();
 				VkDescriptorSetLayoutBinding &binding = bindings.back();
-				binding.binding = u.binding;
-				binding.descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER;
+				binding.binding = u.binding&0xFFFFF;
+				if(is_sampled_image(u.type))
+					binding.descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER;
+				else
+					binding.descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE;
 				binding.descriptorCount = 1;
-				binding.stageFlags = VK_SHADER_STAGE_ALL;
+				binding.stageFlags = stage_flags;
 				binding.pImmutableSamplers = 0;
 			}
 
@@ -158,6 +168,53 @@ void VulkanProgram::finalize_uniforms()
 	layout_info.pPushConstantRanges = &push_const_range;
 
 	vk.CreatePipelineLayout(layout_info, layout_handle);
+
+#if DEBUG
+	if(!debug_name.empty())
+		set_vulkan_object_name();
+#endif
+}
+
+bool VulkanProgram::is_compute() const
+{
+	return stage_flags&VK_SHADER_STAGE_COMPUTE_BIT;
+}
+
+void VulkanProgram::set_debug_name(const string &name)
+{
+#ifdef DEBUG
+	debug_name = name;
+	set_vulkan_object_name();
+	if(SpirVModule *spirv = static_cast<Program *>(this)->specialized_spirv)
+		spirv->set_debug_name(debug_name);
+#else
+	(void)name;
+#endif
+}
+
+void VulkanProgram::set_vulkan_object_name() const
+{
+#ifdef DEBUG
+	const VulkanFunctions &vk = device.get_functions();
+
+	string layout_name = debug_name+" [layout]";
+
+	VkDebugUtilsObjectNameInfoEXT name_info = { };
+	name_info.sType = VK_STRUCTURE_TYPE_DEBUG_UTILS_OBJECT_NAME_INFO_EXT;
+	name_info.objectType = VK_OBJECT_TYPE_PIPELINE_LAYOUT;
+	name_info.objectHandle = reinterpret_cast<uint64_t>(layout_handle);
+	name_info.pObjectName = layout_name.c_str();
+	vk.SetDebugUtilsObjectName(name_info);
+
+	name_info.objectType = VK_OBJECT_TYPE_DESCRIPTOR_SET_LAYOUT;
+	for(unsigned i=0; i<desc_set_layout_handles.size(); ++i)
+	{
+		layout_name = format("%s [layout:%d]", debug_name, i);
+	name_info.objectHandle = reinterpret_cast<uint64_t>(desc_set_layout_handles[i]);
+		name_info.pObjectName = layout_name.c_str();
+		vk.SetDebugUtilsObjectName(name_info);
+	}
+#endif
 }
 
 } // namespace GL