From 446bb2b8638f9511b6b732ac8b33d87ed228e72f Mon Sep 17 00:00:00 2001
From: zeo <goonox35@proton.me>
Date: Fri, 12 Dec 2025 00:06:28 +0500
Subject: [PATCH] fix LoadBindGroup/UnloadBindGroup refcounting

---
 src/backend_wgpu.c | 13 +++++++++++--
 src/raygpu.c       | 18 +++++++++++++++---
 2 files changed, 26 insertions(+), 5 deletions(-)

diff --git a/src/backend_wgpu.c b/src/backend_wgpu.c
index 7006d4459..907849fad 100644
--- a/src/backend_wgpu.c
+++ b/src/backend_wgpu.c
@@ -2785,8 +2785,17 @@ EntryPointSet getEntryPointsSPIRV(const uint32_t *shaderSourceSPIRV, uint32_t wo
 }
 
 void UnloadBindGroup(DescribedBindGroup *bg) {
-    free(bg->entries);
-    wgpuBindGroupRelease((WGPUBindGroup)bg->bindGroup);
+    if(bg->entries) {
+        for(size_t i = 0; i < bg->entryCount; ++i) {
+            if(bg->entries[i].buffer) wgpuBufferRelease((WGPUBuffer)bg->entries[i].buffer);
+            if(bg->entries[i].textureView) wgpuTextureViewRelease((WGPUTextureView)bg->entries[i].textureView);
+            if(bg->entries[i].sampler) wgpuSamplerRelease((WGPUSampler)bg->entries[i].sampler);
+        }
+        free(bg->entries);
+    }
+    if (bg->bindGroup) {
+        wgpuBindGroupRelease((WGPUBindGroup)bg->bindGroup);
+    }
 }
 void UnloadBindGroupLayout(DescribedBindGroupLayout *bglayout) {
     free(bglayout->entries);
diff --git a/src/raygpu.c b/src/raygpu.c
index acd10f16f..55200c688 100644
--- a/src/raygpu.c
+++ b/src/raygpu.c
@@ -2339,7 +2339,21 @@ DescribedBindGroup LoadBindGroup(const DescribedBindGroupLayout* bglayout, const
     DescribedBindGroup ret  = {0};
     if(entryCount > 0){
         ret.entries = (ResourceDescriptor*)RL_CALLOC(entryCount, sizeof(ResourceDescriptor));
-        memcpy(ret.entries, entries, entryCount * sizeof(ResourceDescriptor));
+        // memcpy(ret.entries, entries, entryCount * sizeof(ResourceDescriptor));
+
+        for(size_t i = 0; i < entryCount; ++i) {
+            ret.entries[i] = entries[i];
+            
+            if(ret.entries[i].buffer) {
+                wgpuBufferAddRef((WGPUBuffer)ret.entries[i].buffer);
+            }
+            if(ret.entries[i].textureView) {
+                wgpuTextureViewAddRef((WGPUTextureView)ret.entries[i].textureView);
+            }
+            if(ret.entries[i].sampler) {
+                wgpuSamplerAddRef((WGPUSampler)ret.entries[i].sampler);
+            }
+        }
     }
     ret.entryCount = entryCount;
     ret.layout = bglayout;
@@ -2347,11 +2361,9 @@ DescribedBindGroup LoadBindGroup(const DescribedBindGroupLayout* bglayout, const
     ret.needsUpdate = true;
     ret.descriptorHash = 0;
 
-
     for(uint32_t i = 0;i < ret.entryCount;i++){
         ret.descriptorHash ^= bgEntryHash(ret.entries[i]);
     }
-    //ret.bindGroup = wgpuDeviceCreateBindGroup((WGPUDevice)GetDevice(), &ret.desc);
     return ret;
 }