fix: shader OOB access issue

PilotLightTech · Dec 16, 2024 · b62ea35 · b62ea35
1 parent 9d49bc5
commit b62ea35
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/shaders/filter_environment.comp b/shaders/filter_environment.comp
@@ -67,7 +67,7 @@ pl_write_face(int iPixel, int iFace, vec3 tColorIn)
         tFaceOut3.atPixelData[iPixel] = tColor;
     else if (iFace == 4)
         tFaceOut4.atPixelData[iPixel] = tColor;
-    else //if(iFace == 5)
+    else if(iFace == 5)
         tFaceOut5.atPixelData[iPixel] = tColor;
 }
 

diff --git a/shaders/panorama_to_cubemap.comp b/shaders/panorama_to_cubemap.comp
@@ -84,6 +84,9 @@ main()
 
     const int iSrcpixelIndex = iColumnindex + iRowindex * iWidth;
 
-    const vec3 tColor = vec3(tBufferIn.atPixelData[iSrcpixelIndex].r, tBufferIn.atPixelData[iSrcpixelIndex].g, tBufferIn.atPixelData[iSrcpixelIndex].b);
-    pl_write_face(iCurrentPixel, iFace, tColor);
+    if(iSrcpixelIndex < iWidth * iHeight)
+    {
+        const vec3 tColor = vec3(tBufferIn.atPixelData[iSrcpixelIndex].r, tBufferIn.atPixelData[iSrcpixelIndex].g, tBufferIn.atPixelData[iSrcpixelIndex].b);
+        pl_write_face(iCurrentPixel, iFace, tColor);
+    }
 }