refactor compute shader

Author: Mysvac

docs/codes/03/80_compute/CMakeLists.txt

@@ -3,34 +3,37 @@ cmake_minimum_required(VERSION 4.0.0)
 find_package(Vulkan REQUIRED)
 
 set(SHADER_DIR ${CMAKE_CURRENT_SOURCE_DIR})
-set(VERT_SHADER ${SHADER_DIR}/shader.vert)
-set(FRAG_SHADER ${SHADER_DIR}/shader.frag)
-set(COMP_SHADER ${SHADER_DIR}/shader.comp)
-set(SPIRV_VERT ${SHADER_DIR}/vert.spv)
-set(SPIRV_FRAG ${SHADER_DIR}/frag.spv)
-set(SPIRV_COMP ${SHADER_DIR}/comp.spv)
+set(STAGE_VERT "-fshader-stage=vert")
+set(STAGE_FRAG "-fshader-stage=frag")
+set(STAGE_COMP "-fshader-stage=comp")
+set(GRAPHICS_VERT_SHADER ${SHADER_DIR}/graphics.vert.glsl)
+set(GRAPHICS_FRAG_SHADER ${SHADER_DIR}/graphics.frag.glsl)
+set(COMPUTE_COMP_SHADER ${SHADER_DIR}/compute.comp.glsl)
+set(GRAPHICS_SPIRV_VERT ${SHADER_DIR}/graphics.vert.spv)
+set(GRAPHICS_SPIRV_FRAG ${SHADER_DIR}/graphics.frag.spv)
+set(COMPUTE_SPIRV_COMP ${SHADER_DIR}/compute.comp.spv)
 
 add_custom_command(
-    OUTPUT ${SPIRV_VERT}
-    COMMAND ${Vulkan_GLSLC_EXECUTABLE} ${VERT_SHADER} -o ${SPIRV_VERT}
-    COMMENT "Compiling shader.vert to vert.spv"
-    DEPENDS ${VERT_SHADER}
+        OUTPUT ${GRAPHICS_SPIRV_VERT}
+        COMMAND ${Vulkan_GLSLC_EXECUTABLE} ${STAGE_VERT} ${GRAPHICS_VERT_SHADER} -o ${GRAPHICS_SPIRV_VERT}
+        COMMENT "Compiling graphics.vert.glsl to graphics.vert.spv"
+        DEPENDS ${GRAPHICS_VERT_SHADER}
 )
 
 add_custom_command(
-    OUTPUT ${SPIRV_FRAG}
-    COMMAND ${Vulkan_GLSLC_EXECUTABLE} ${FRAG_SHADER} -o ${SPIRV_FRAG}
-    COMMENT "Compiling shader.frag to frag.spv"
-    DEPENDS ${FRAG_SHADER}
+        OUTPUT ${GRAPHICS_SPIRV_FRAG}
+        COMMAND ${Vulkan_GLSLC_EXECUTABLE} ${STAGE_FRAG} ${GRAPHICS_FRAG_SHADER} -o ${GRAPHICS_SPIRV_FRAG}
+        COMMENT "Compiling graphics.frag.glsl to graphics.frag.spv"
+        DEPENDS ${GRAPHICS_FRAG_SHADER}
 )
 
 add_custom_command(
-    OUTPUT ${SPIRV_COMP}
-    COMMAND ${Vulkan_GLSLC_EXECUTABLE} ${COMP_SHADER} -o ${SPIRV_COMP}
-    COMMENT "Compiling shader.comp to comp.spv"
-    DEPENDS ${COMP_SHADER}
+        OUTPUT ${COMPUTE_SPIRV_COMP}
+        COMMAND ${Vulkan_GLSLC_EXECUTABLE} ${STAGE_COMP} ${COMPUTE_COMP_SHADER} -o ${COMPUTE_SPIRV_COMP}
+        COMMENT "Compiling compute.comp.glsl to compute.comp.spv"
+        DEPENDS ${COMPUTE_COMP_SHADER}
 )
 
 add_custom_target(CompileShaders ALL
-    DEPENDS ${SPIRV_VERT} ${SPIRV_FRAG} ${SPIRV_COMP}
-)
+        DEPENDS ${GRAPHICS_SPIRV_VERT} ${GRAPHICS_SPIRV_FRAG} ${COMPUTE_SPIRV_COMP}
+)

docs/codes/03/80_compute/base_code.zip

@@ -0,0 +1,29 @@
+#version 450
+
+struct Particle {
+    vec2 position;
+    vec2 velocity;
+    vec4 color;
+};
+
+layout(push_constant) uniform PushConstants {
+    float deltaTime;
+} pc;
+
+layout(std140, binding = 0) buffer ParticleSSBO {
+    Particle particles[ ];
+};
+
+layout (local_size_x = 256, local_size_y = 1, local_size_z = 1) in;
+
+void main() {
+    uint index = gl_GlobalInvocationID.x;
+
+    Particle particle = particles[index];
+
+    particles[index].position = particle.position + particle.velocity.xy * pc.deltaTime;
+
+    // Flip movement at window border
+    if (abs(particles[index].position.x) >= 1.0) particles[index].velocity.x *= -1;
+    if (abs(particles[index].position.y) >= 1.0) particles[index].velocity.y *= -1;
+}