几何着色器写入纹理并在片段着色器采样的方法及描述符配置

完全可以实现这种需求，只要遵循Vulkan的图像资源规则和描述符配置要求，下面是具体的实现步骤和注意事项：

核心逻辑 #

同一图像资源可以通过不同绑定点的不同描述符类型，分别暴露给几何着色器（作为可写入的存储图像）和片段着色器（作为可采样的图像采样器组合）。关键是要确保图像的用途、内存属性和布局转换符合Vulkan的同步要求。

步骤1：创建符合要求的图像和视图 #

• 图像创建参数：
  • 格式选择同时支持存储和采样的类型，比如VK_FORMAT_RGBA8_UNORM、VK_FORMAT_R32_FLOAT等通用格式。
  • usage字段必须包含VK_IMAGE_USAGE_STORAGE_BIT（供几何着色器写入）和VK_IMAGE_USAGE_SAMPLED_BIT（供片段着色器采样）。
  • 内存属性优先选择VK_MEMORY_PROPERTY_DEVICE_LOCAL_BIT，保证设备端访问性能。
• 图像视图：创建两个独立的图像视图（可以复用同一图像的子资源），分别对应存储图像和采样图像的用途——视图类型和格式要与图像匹配，比如2D图像用VK_IMAGE_VIEW_TYPE_2D。

步骤2：配置描述符集布局 #

在描述符集布局中定义两个绑定点，分别对应两种描述符类型：

• 绑定点0：类型为VK_DESCRIPTOR_TYPE_STORAGE_IMAGE，阶段掩码设置为VK_SHADER_STAGE_GEOMETRY_BIT，表示只有几何着色器能访问这个绑定。
• 绑定点1：类型为VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER，阶段掩码设置为VK_SHADER_STAGE_FRAGMENT_BIT，供片段着色器采样使用。

示例代码（C++）：

VkDescriptorSetLayoutBinding storage_binding = {};
storage_binding.binding = 0;
storage_binding.descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE;
storage_binding.descriptorCount = 1;
storage_binding.stageFlags = VK_SHADER_STAGE_GEOMETRY_BIT;

VkDescriptorSetLayoutBinding sampler_binding = {};
sampler_binding.binding = 1;
sampler_binding.descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER;
sampler_binding.descriptorCount = 1;
sampler_binding.stageFlags = VK_SHADER_STAGE_FRAGMENT_BIT;

std::array<VkDescriptorSetLayoutBinding, 2> bindings = {storage_binding, sampler_binding};
VkDescriptorSetLayoutCreateInfo layout_info = {};
layout_info.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO;
layout_info.bindingCount = bindings.size();
layout_info.pBindings = bindings.data();

vkCreateDescriptorSetLayout(device, &layout_info, nullptr, &descriptor_set_layout);

步骤3：更新描述符集 #

分配描述符集后，分别更新两个绑定点的内容：

• 对于STORAGE_IMAGE绑定：关联对应的存储图像视图，图像布局设置为VK_IMAGE_LAYOUT_STORAGE_OPTIMAL或VK_IMAGE_LAYOUT_GENERAL（后者更灵活但性能略低）。
• 对于COMBINED_IMAGE_SAMPLER绑定：关联采样器（比如配置线性过滤、重复寻址的采样器）和采样图像视图，图像布局设置为VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL。

示例更新代码：

// 存储图像描述符更新
VkDescriptorImageInfo storage_image_info = {};
storage_image_info.imageView = storage_image_view;
storage_image_info.imageLayout = VK_IMAGE_LAYOUT_STORAGE_OPTIMAL;

VkWriteDescriptorSet storage_write = {};
storage_write.sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET;
storage_write.dstSet = descriptor_set;
storage_write.dstBinding = 0;
storage_write.dstArrayElement = 0;
storage_write.descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE;
storage_write.descriptorCount = 1;
storage_write.pImageInfo = &storage_image_info;

// 采样器描述符更新
VkDescriptorImageInfo sampler_image_info = {};
sampler_image_info.sampler = image_sampler;
sampler_image_info.imageView = sampled_image_view;
sampler_image_info.imageLayout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL;

VkWriteDescriptorSet sampler_write = {};
sampler_write.sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET;
sampler_write.dstSet = descriptor_set;
sampler_write.dstBinding = 1;
sampler_write.dstArrayElement = 0;
sampler_write.descriptorType = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER;
sampler_write.descriptorCount = 1;
sampler_write.pImageInfo = &sampler_image_info;

std::array<VkWriteDescriptorSet, 2> writes = {storage_write, sampler_write};
vkUpdateDescriptorSets(device, writes.size(), writes.data(), 0, nullptr);

步骤4：着色器中的使用示例 #

几何着色器（GLSL） #

#version 450

// 绑定0：可写入的存储图像
layout(set = 0, binding = 0) uniform writeonly image2D output_img;

void main() {
    // 示例：给指定坐标的像素写入红色
    ivec2 pixel_coord = ivec2(gl_VertexIndex % 1024, gl_VertexIndex / 1024);
    imageStore(output_img, pixel_coord, vec4(1.0, 0.0, 0.0, 1.0));
}

片段着色器（GLSL） #

#version 450

// 绑定1：可采样的图像采样器组合
layout(set = 0, binding = 1) uniform sampler2D input_sampler;

layout(location = 0) out vec4 frag_color;

void main() {
    // 示例：采样当前像素坐标对应的纹理值
    vec2 uv = gl_FragCoord.xy / vec2(1024.0, 1024.0);
    frag_color = texture(input_sampler, uv);
}

关键同步与布局转换 #

必须在几何着色器写入完成后、片段着色器采样前，插入图像内存屏障完成布局转换，并保证管线阶段的同步：

VkImageMemoryBarrier layout_barrier = {};
layout_barrier.sType = VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER;
layout_barrier.oldLayout = VK_IMAGE_LAYOUT_STORAGE_OPTIMAL;
layout_barrier.newLayout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL;
layout_barrier.srcQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED;
layout_barrier.dstQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED;
layout_barrier.image = target_image;
layout_barrier.subresourceRange = {
    VK_IMAGE_ASPECT_COLOR_BIT,
    0, 1, // mip层级
    0, 1  // array层
};
layout_barrier.srcAccessMask = VK_ACCESS_SHADER_WRITE_BIT;
layout_barrier.dstAccessMask = VK_ACCESS_SHADER_READ_BIT;

// 插入屏障，同步几何着色器和片段着色器阶段
vkCmdPipelineBarrier(
    cmd_buffer,
    VK_PIPELINE_STAGE_GEOMETRY_SHADER_BIT,
    VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT,
    0,
    0, nullptr,
    0, nullptr,
    1, &layout_barrier
);

注意事项 #

• 避免在同一管线中同时让几何着色器写入和片段着色器采样同一图像（除非通过屏障严格同步），否则会出现数据竞争。
• 图像格式必须同时支持VK_FORMAT_FEATURE_STORAGE_IMAGE_BIT和VK_FORMAT_FEATURE_SAMPLED_IMAGE_BIT，可以通过vkGetPhysicalDeviceFormatProperties查询验证。
• 如果是多队列环境，还要考虑队列家族的所有权转移（如果图像在不同队列间访问）。

内容的提问来源于stack exchange，提问作者omniyo