UnrealXR/raylib
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update rlgl_compute_shader.c

Browse source
This commit is contained in:
Ray 2022-07-04 18:34:38 +02:00
parent c6c71fe73c
commit 5c5db2c016
1 changed files with 13 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

28
examples/others/rlgl_compute_shader.c
View file

@ -58,27 +58,25 @@ int main(void)
unsigned int golLogicProgram = rlLoadComputeShaderProgram(golLogicShader); unsigned int golLogicProgram = rlLoadComputeShaderProgram(golLogicShader);
UnloadFileText(golLogicCode); UnloadFileText(golLogicCode);
// Game of Life logic compute shader // Game of Life logic render shader
Shader golRenderShader = LoadShader(NULL, "resources/shaders/glsl430/gol_render.glsl"); Shader golRenderShader = LoadShader(NULL, "resources/shaders/glsl430/gol_render.glsl");
int resUniformLoc = GetShaderLocation(golRenderShader, "resolution"); int resUniformLoc = GetShaderLocation(golRenderShader, "resolution");
// Game of Life transfert shader // Game of Life transfert shader (CPU<->GPU download and upload)
char *golTransfertCode = LoadFileText("resources/shaders/glsl430/gol_transfert.glsl"); char *golTransfertCode = LoadFileText("resources/shaders/glsl430/gol_transfert.glsl");
unsigned int golTransfertShader = rlCompileShader(golTransfertCode, RL_COMPUTE_SHADER); unsigned int golTransfertShader = rlCompileShader(golTransfertCode, RL_COMPUTE_SHADER);
unsigned int golTransfertProgram = rlLoadComputeShaderProgram(golTransfertShader); unsigned int golTransfertProgram = rlLoadComputeShaderProgram(golTransfertShader);
UnloadFileText(golTransfertCode); UnloadFileText(golTransfertCode);
// SSBOs // Load shader storage buffer object (SSBO), id returned
unsigned int ssboA = rlLoadShaderBuffer(GOL_WIDTH*GOL_WIDTH*sizeof(unsigned int), NULL, RL_DYNAMIC_COPY); unsigned int ssboA = rlLoadShaderBuffer(GOL_WIDTH*GOL_WIDTH*sizeof(unsigned int), NULL, RL_DYNAMIC_COPY);
unsigned int ssboB = rlLoadShaderBuffer(GOL_WIDTH*GOL_WIDTH*sizeof(unsigned int), NULL, RL_DYNAMIC_COPY); unsigned int ssboB = rlLoadShaderBuffer(GOL_WIDTH*GOL_WIDTH*sizeof(unsigned int), NULL, RL_DYNAMIC_COPY);
unsigned int ssboTransfert = rlLoadShaderBuffer(sizeof(GolUpdateSSBO), NULL, RL_DYNAMIC_COPY);
struct GolUpdateSSBO transfertBuffer; GolUpdateSSBO transfertBuffer = { 0 };
transfertBuffer.count = 0;
int transfertSSBO = rlLoadShaderBuffer(sizeof(struct GolUpdateSSBO), NULL, RL_DYNAMIC_COPY);
// Create a white texture of the size of the window to update // Create a white texture of the size of the window to update
// each pixel of the window using the fragment shader // each pixel of the window using the fragment shader: golRenderShader
Image whiteImage = GenImageColor(GOL_WIDTH, GOL_WIDTH, WHITE); Image whiteImage = GenImageColor(GOL_WIDTH, GOL_WIDTH, WHITE);
Texture whiteTex = LoadTextureFromImage(whiteImage); Texture whiteTex = LoadTextureFromImage(whiteImage);
UnloadImage(whiteImage); UnloadImage(whiteImage);
@ -101,18 +99,16 @@ int main(void)
transfertBuffer.commands[transfertBuffer.count].enabled = IsMouseButtonDown(MOUSE_BUTTON_LEFT); transfertBuffer.commands[transfertBuffer.count].enabled = IsMouseButtonDown(MOUSE_BUTTON_LEFT);
transfertBuffer.count++; transfertBuffer.count++;
} }
else if (transfertBuffer.count > 0) else if (transfertBuffer.count > 0) // Process transfert buffer
{ {
// Process transfert buffer
// Send SSBO buffer to GPU // Send SSBO buffer to GPU
rlUpdateShaderBufferElements(transfertSSBO, &transfertBuffer, sizeof(struct GolUpdateSSBO), 0); rlUpdateShaderBufferElements(ssboTransfert, &transfertBuffer, sizeof(GolUpdateSSBO), 0);
// Process ssbo command // Process ssbo commands on GPU
rlEnableShader(golTransfertProgram); rlEnableShader(golTransfertProgram);
rlBindShaderBuffer(ssboA, 1); rlBindShaderBuffer(ssboA, 1);
rlBindShaderBuffer(transfertSSBO, 3); rlBindShaderBuffer(ssboTransfert, 3);
rlComputeShaderDispatch(transfertBuffer.count, 1, 1); // each GPU unit will process a command rlComputeShaderDispatch(transfertBuffer.count, 1, 1); // Each GPU unit will process a command!
rlDisableShader(); rlDisableShader();
transfertBuffer.count = 0; transfertBuffer.count = 0;
@ -160,7 +156,7 @@ int main(void)
// Unload shader buffers objects. // Unload shader buffers objects.
rlUnloadShaderBuffer(ssboA); rlUnloadShaderBuffer(ssboA);
rlUnloadShaderBuffer(ssboB); rlUnloadShaderBuffer(ssboB);
rlUnloadShaderBuffer(transfertSSBO); rlUnloadShaderBuffer(ssboTransfert);
// Unload compute shader programs // Unload compute shader programs
rlUnloadShaderProgram(golTransfertProgram); rlUnloadShaderProgram(golTransfertProgram);