Merge pull request #99 from Devsh-Graphics-Programming/vulkan_1_3_example_07_fix

Vulkan_1_3 example 07 fix

Author: devshgraphicsprogramming

07_StagingAndMultipleQueues/app_resources/common.hlsl

@@ -2,4 +2,21 @@
 
 NBL_CONSTEXPR uint32_t WorkgroupSizeX = 16;
 NBL_CONSTEXPR uint32_t WorkgroupSizeY = 16;
-NBL_CONSTEXPR uint32_t WorkgroupSize = WorkgroupSizeX*WorkgroupSizeY;
+NBL_CONSTEXPR uint32_t WorkgroupSize = WorkgroupSizeX*WorkgroupSizeY;
+
+static const uint32_t FRAMES_IN_FLIGHT = 3u;
+
+static const uint32_t RED_OFFSET = 0u;
+static const uint32_t GREEN_OFFSET = 256u;
+static const uint32_t BLUE_OFFSET = 256u * 2u;
+
+static const uint32_t CHANEL_CNT = 3;
+static const uint32_t VAL_PER_CHANEL_CNT = 256;
+static const uint32_t HISTOGRAM_SIZE = CHANEL_CNT * VAL_PER_CHANEL_CNT;
+static const uint32_t HISTOGRAM_BYTE_SIZE = HISTOGRAM_SIZE * sizeof(uint32_t);
+static const uint32_t COMBINED_HISTOGRAM_BUFFER_BYTE_SIZE = HISTOGRAM_BYTE_SIZE * FRAMES_IN_FLIGHT;
+
+struct PushConstants
+{
+    uint32_t histogramBufferOffset;
+};

07_StagingAndMultipleQueues/app_resources/comp_shader.hlsl

@@ -0,0 +1,30 @@
+#pragma wave shader_stage(compute)
+
+#include "../app_resources/common.hlsl"
+
+[[vk::combinedImageSampler]][[vk::binding(0,0)]] Texture2D texture;
+[[vk::combinedImageSampler]][[vk::binding(0,0)]] SamplerState samplerState;
+[[vk::binding(1,0)]] RWStructuredBuffer<uint32_t> histogram;
+
+[[vk::push_constant]]
+PushConstants constants;
+
+[numthreads(WorkgroupSizeX,WorkgroupSizeY,1)]
+void main(uint32_t3 ID : SV_DispatchThreadID)
+{
+	uint width;
+	uint height;
+	texture.GetDimensions(width, height);
+	if(ID.x >= width || ID.y >= height)
+		return;
+
+    float4 texel = texture.SampleLevel(samplerState, ID.xy, 0.0);
+
+    const uint32_t redVal = uint32_t(texel.r * 255u);
+    const uint32_t greenVal = uint32_t(texel.g * 255u);
+    const uint32_t blueVal = uint32_t(texel.b * 255u);
+
+    InterlockedAdd(histogram[constants.histogramBufferOffset + RED_OFFSET + redVal], 1);
+    InterlockedAdd(histogram[constants.histogramBufferOffset + GREEN_OFFSET + greenVal], 1);
+    InterlockedAdd(histogram[constants.histogramBufferOffset + BLUE_OFFSET + blueVal], 1);
+}