Cache render commands

Author: lwjglgamedev

chapter-20/src/main/java/org/lwjglb/engine/graph/SceneRender.java

@@ -111,7 +111,7 @@ public void setupData(Scene scene) {
         setupMaterialsUniform(scene.getTextureCache(), scene.getMaterialCache());
     }
 
-    public void setupMaterialsUniform(TextureCache textureCache, MaterialCache materialCache) {
+    private void setupMaterialsUniform(TextureCache textureCache, MaterialCache materialCache) {
         List<Texture> textures = textureCache.getAll().stream().toList();
         int numTextures = textures.size();
         if (numTextures > MAX_TEXTURES) {

chapter-21/src/main/java/org/lwjglb/engine/graph/Mesh.java

@@ -116,6 +116,14 @@ public void cleanup() {
         glDeleteVertexArrays(vaoId);
     }
 
+    public Vector3f getAabbMax() {
+        return aabbMax;
+    }
+
+    public Vector3f getAabbMin() {
+        return aabbMin;
+    }
+
     public int getNumVertices() {
         return numVertices;
     }

chapter-21/src/main/java/org/lwjglb/engine/graph/Render.java

@@ -83,7 +83,8 @@ public void resize(int width, int height) {
     public void setupData(Scene scene) {
         renderBuffers.loadStaticModels(scene);
         renderBuffers.loadAnimatedModels(scene);
-        sceneRender.setupMaterialsUniform(scene.getTextureCache(), scene.getMaterialCache());
+        sceneRender.setupData(scene);
+        shadowRender.setupData(scene);
         List<Model> modelList = new ArrayList<>(scene.getModelMap().values());
         modelList.forEach(m -> m.getMeshDataList().clear());
     }

chapter-21/src/main/java/org/lwjglb/engine/graph/SceneRender.java

@@ -16,7 +16,12 @@ public class SceneRender {
     private static final int COMMAND_SIZE = 5 * 4;
     private static final int MAX_MATERIALS = 20;
     private static final int MAX_TEXTURES = 16;
+    private int animDrawCount;
+    private int animRenderBufferHandle;
+    private Map<String, Integer> entitiesIdxMap;
     private ShaderProgram shaderProgram;
+    private int staticDrawCount;
+    private int staticRenderBufferHandle;
     private UniformsMap uniformsMap;
 
     public SceneRender() {
@@ -25,87 +30,13 @@ public SceneRender() {
         shaderModuleDataList.add(new ShaderProgram.ShaderModuleData("resources/shaders/scene.frag", GL_FRAGMENT_SHADER));
         shaderProgram = new ShaderProgram(shaderModuleDataList);
         createUniforms();
-    }
-
-    private ByteBuffer buildAnimCommandBuffer(List<Model> modelList, Map<String, Integer> entitiesIdxMap) {
-        int numMeshes = 0;
-        for (Model model : modelList) {
-            int numEntities = model.getEntitiesList().size();
-            numMeshes += numEntities * model.getMeshDrawDataList().size();
-        }
-
-        int firstIndex = 0;
-        int baseInstance = 0;
-        int drawElement = 0;
-        ByteBuffer commandBuffer = MemoryUtil.memAlloc(numMeshes * COMMAND_SIZE);
-        for (Model model : modelList) {
-            for (RenderBuffers.MeshDrawData meshDrawData : model.getMeshDrawDataList()) {
-                RenderBuffers.AnimMeshDrawData animMeshDrawData = meshDrawData.animMeshDrawData();
-                Entity entity = animMeshDrawData.entity();
-                // count
-                commandBuffer.putInt(meshDrawData.vertices());
-                // instanceCount
-                commandBuffer.putInt(1);
-                commandBuffer.putInt(firstIndex);
-                // baseVertex
-                commandBuffer.putInt(meshDrawData.offset());
-                commandBuffer.putInt(baseInstance);
-
-                firstIndex += meshDrawData.vertices();
-                baseInstance++;
-
-                String name = "drawElements[" + drawElement + "]";
-                uniformsMap.setUniform(name + ".modelMatrixIdx", entitiesIdxMap.get(entity.getId()));
-                uniformsMap.setUniform(name + ".materialIdx", meshDrawData.materialIdx());
-                drawElement++;
-            }
-        }
-        commandBuffer.flip();
-
-        return commandBuffer;
-    }
-
-    private ByteBuffer buildStaticCommandBuffer(List<Model> modelList, Map<String, Integer> entitiesIdxMap) {
-        int numMeshes = 0;
-        for (Model model : modelList) {
-            numMeshes += model.getMeshDrawDataList().size();
-        }
-
-        int firstIndex = 0;
-        int baseInstance = 0;
-        int drawElement = 0;
-        ByteBuffer commandBuffer = MemoryUtil.memAlloc(numMeshes * COMMAND_SIZE);
-        for (Model model : modelList) {
-            List<Entity> entities = model.getEntitiesList();
-            int numEntities = entities.size();
-            for (RenderBuffers.MeshDrawData meshDrawData : model.getMeshDrawDataList()) {
-                // count
-                commandBuffer.putInt(meshDrawData.vertices());
-                // instanceCount
-                commandBuffer.putInt(numEntities);
-                commandBuffer.putInt(firstIndex);
-                // baseVertex
-                commandBuffer.putInt(meshDrawData.offset());
-                commandBuffer.putInt(baseInstance);
-
-                firstIndex += meshDrawData.vertices();
-                baseInstance += entities.size();
-
-                for (Entity entity : entities) {
-                    String name = "drawElements[" + drawElement + "]";
-                    uniformsMap.setUniform(name + ".modelMatrixIdx", entitiesIdxMap.get(entity.getId()));
-                    uniformsMap.setUniform(name + ".materialIdx", meshDrawData.materialIdx());
-                    drawElement++;
-                }
-            }
-        }
-        commandBuffer.flip();
-
-        return commandBuffer;
+        entitiesIdxMap = new HashMap<>();
     }
 
     public void cleanup() {
         shaderProgram.cleanup();
+        glDeleteBuffers(staticRenderBufferHandle);
+        glDeleteBuffers(animRenderBufferHandle);
     }
 
     private void createUniforms() {
@@ -137,7 +68,7 @@ private void createUniforms() {
         }
     }
 
-    public void render(Scene scene, RenderBuffers globalBuffer, GBuffer gBuffer) {
+    public void render(Scene scene, RenderBuffers renderBuffers, GBuffer gBuffer) {
         glBindFramebuffer(GL_DRAW_FRAMEBUFFER, gBuffer.getGBufferId());
         glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
         glViewport(0, 0, gBuffer.getWidth(), gBuffer.getHeight());
@@ -161,59 +92,111 @@ public void render(Scene scene, RenderBuffers globalBuffer, GBuffer gBuffer) {
             texture.bind();
         }
 
-        Map<String, Integer> entitiesIdxMap = new HashMap<>();
         int entityIdx = 0;
         for (Model model : scene.getModelMap().values()) {
             List<Entity> entities = model.getEntitiesList();
             for (Entity entity : entities) {
-                entitiesIdxMap.put(entity.getId(), entityIdx);
                 uniformsMap.setUniform("modelMatrices[" + entityIdx + "]", entity.getModelMatrix());
                 entityIdx++;
             }
         }
 
-        renderStaticMeshes(scene, globalBuffer, entitiesIdxMap);
-        renderAnimMeshes(scene, globalBuffer, entitiesIdxMap);
+        // Static meshes
+        int drawElement = 0;
+        List<Model> modelList = scene.getModelMap().values().stream().filter(m -> !m.isAnimated()).toList();
+        for (Model model : modelList) {
+            List<Entity> entities = model.getEntitiesList();
+            for (RenderBuffers.MeshDrawData meshDrawData : model.getMeshDrawDataList()) {
+                for (Entity entity : entities) {
+                    String name = "drawElements[" + drawElement + "]";
+                    uniformsMap.setUniform(name + ".modelMatrixIdx", entitiesIdxMap.get(entity.getId()));
+                    uniformsMap.setUniform(name + ".materialIdx", meshDrawData.materialIdx());
+                    drawElement++;
+                }
+            }
+        }
+        glBindBuffer(GL_DRAW_INDIRECT_BUFFER, staticRenderBufferHandle);
+        glBindVertexArray(renderBuffers.getStaticVaoId());
+        glMultiDrawElementsIndirect(GL_TRIANGLES, GL_UNSIGNED_INT, 0, staticDrawCount, 0);
 
+        // Animated meshes
+        drawElement = 0;
+        modelList = scene.getModelMap().values().stream().filter(m -> m.isAnimated()).toList();
+        for (Model model : modelList) {
+            for (RenderBuffers.MeshDrawData meshDrawData : model.getMeshDrawDataList()) {
+                RenderBuffers.AnimMeshDrawData animMeshDrawData = meshDrawData.animMeshDrawData();
+                Entity entity = animMeshDrawData.entity();
+                String name = "drawElements[" + drawElement + "]";
+                uniformsMap.setUniform(name + ".modelMatrixIdx", entitiesIdxMap.get(entity.getId()));
+                uniformsMap.setUniform(name + ".materialIdx", meshDrawData.materialIdx());
+                drawElement++;
+            }
+        }
+        glBindBuffer(GL_DRAW_INDIRECT_BUFFER, animRenderBufferHandle);
+        glBindVertexArray(renderBuffers.getAnimVaoId());
+        glMultiDrawElementsIndirect(GL_TRIANGLES, GL_UNSIGNED_INT, 0, animDrawCount, 0);
+
+        glBindVertexArray(0);
         glEnable(GL_BLEND);
         shaderProgram.unbind();
     }
 
-    private void renderAnimMeshes(Scene scene, RenderBuffers globalBuffer, Map<String, Integer> entitiesIdxMap) {
-        List<Model> modelList = scene.getModelMap().values().stream().filter(Model::isAnimated).toList();
-
-        ByteBuffer commandBuffer = buildAnimCommandBuffer(modelList, entitiesIdxMap);
-        int drawCount = commandBuffer.remaining() / COMMAND_SIZE;
-        int bufferHandle = glGenBuffers();
-        glBindBuffer(GL_DRAW_INDIRECT_BUFFER, bufferHandle);
-        glBufferData(GL_DRAW_INDIRECT_BUFFER, commandBuffer, GL_DYNAMIC_DRAW);
+    private void setupAnimCommandBuffer(Scene scene) {
+        List<Model> modelList = scene.getModelMap().values().stream().filter(m -> m.isAnimated()).toList();
+        int numMeshes = 0;
+        for (Model model : modelList) {
+            numMeshes += model.getMeshDrawDataList().size();
+        }
 
-        glBindVertexArray(globalBuffer.getAnimVaoId());
-        glMultiDrawElementsIndirect(GL_TRIANGLES, GL_UNSIGNED_INT, 0, drawCount, 0);
-        glBindVertexArray(0);
+        int firstIndex = 0;
+        int baseInstance = 0;
+        ByteBuffer commandBuffer = MemoryUtil.memAlloc(numMeshes * COMMAND_SIZE);
+        for (Model model : modelList) {
+            for (RenderBuffers.MeshDrawData meshDrawData : model.getMeshDrawDataList()) {
+                // count
+                commandBuffer.putInt(meshDrawData.vertices());
+                // instanceCount
+                commandBuffer.putInt(1);
+                commandBuffer.putInt(firstIndex);
+                // baseVertex
+                commandBuffer.putInt(meshDrawData.offset());
+                commandBuffer.putInt(baseInstance);
 
-        MemoryUtil.memFree(commandBuffer);
-        glDeleteBuffers(bufferHandle);
-    }
+                firstIndex += meshDrawData.vertices();
+                baseInstance++;
+            }
+        }
+        commandBuffer.flip();
 
-    private void renderStaticMeshes(Scene scene, RenderBuffers globalBuffer, Map<String, Integer> entitiesIdxMap) {
-        List<Model> modelList = scene.getModelMap().values().stream().filter(m -> !m.isAnimated()).toList();
+        animDrawCount = commandBuffer.remaining() / COMMAND_SIZE;
 
-        ByteBuffer commandBuffer = buildStaticCommandBuffer(modelList, entitiesIdxMap);
-        int drawCount = commandBuffer.remaining() / COMMAND_SIZE;
-        int bufferHandle = glGenBuffers();
-        glBindBuffer(GL_DRAW_INDIRECT_BUFFER, bufferHandle);
+        animRenderBufferHandle = glGenBuffers();
+        glBindBuffer(GL_DRAW_INDIRECT_BUFFER, animRenderBufferHandle);
         glBufferData(GL_DRAW_INDIRECT_BUFFER, commandBuffer, GL_DYNAMIC_DRAW);
 
-        glBindVertexArray(globalBuffer.getStaticVaoId());
-        glMultiDrawElementsIndirect(GL_TRIANGLES, GL_UNSIGNED_INT, 0, drawCount, 0);
-        glBindVertexArray(0);
-
         MemoryUtil.memFree(commandBuffer);
-        glDeleteBuffers(bufferHandle);
     }
 
-    public void setupMaterialsUniform(TextureCache textureCache, MaterialCache materialCache) {
+    public void setupData(Scene scene) {
+        setupEntitiesData(scene);
+        setupStaticCommandBuffer(scene);
+        setupAnimCommandBuffer(scene);
+        setupMaterialsUniform(scene.getTextureCache(), scene.getMaterialCache());
+    }
+
+    private void setupEntitiesData(Scene scene) {
+        entitiesIdxMap.clear();
+        int entityIdx = 0;
+        for (Model model : scene.getModelMap().values()) {
+            List<Entity> entities = model.getEntitiesList();
+            for (Entity entity : entities) {
+                entitiesIdxMap.put(entity.getId(), entityIdx);
+                entityIdx++;
+            }
+        }
+    }
+
+    private void setupMaterialsUniform(TextureCache textureCache, MaterialCache materialCache) {
         List<Texture> textures = textureCache.getAll().stream().toList();
         int numTextures = textures.size();
         if (numTextures > MAX_TEXTURES) {
@@ -245,4 +228,42 @@ public void setupMaterialsUniform(TextureCache textureCache, MaterialCache mater
         }
         shaderProgram.unbind();
     }
+
+    private void setupStaticCommandBuffer(Scene scene) {
+        List<Model> modelList = scene.getModelMap().values().stream().filter(m -> !m.isAnimated()).toList();
+        int numMeshes = 0;
+        for (Model model : modelList) {
+            numMeshes += model.getMeshDrawDataList().size();
+        }
+
+        int firstIndex = 0;
+        int baseInstance = 0;
+        ByteBuffer commandBuffer = MemoryUtil.memAlloc(numMeshes * COMMAND_SIZE);
+        for (Model model : modelList) {
+            List<Entity> entities = model.getEntitiesList();
+            int numEntities = entities.size();
+            for (RenderBuffers.MeshDrawData meshDrawData : model.getMeshDrawDataList()) {
+                // count
+                commandBuffer.putInt(meshDrawData.vertices());
+                // instanceCount
+                commandBuffer.putInt(numEntities);
+                commandBuffer.putInt(firstIndex);
+                // baseVertex
+                commandBuffer.putInt(meshDrawData.offset());
+                commandBuffer.putInt(baseInstance);
+
+                firstIndex += meshDrawData.vertices();
+                baseInstance += entities.size();
+            }
+        }
+        commandBuffer.flip();
+
+        staticDrawCount = commandBuffer.remaining() / COMMAND_SIZE;
+
+        staticRenderBufferHandle = glGenBuffers();
+        glBindBuffer(GL_DRAW_INDIRECT_BUFFER, staticRenderBufferHandle);
+        glBufferData(GL_DRAW_INDIRECT_BUFFER, commandBuffer, GL_DYNAMIC_DRAW);
+
+        MemoryUtil.memFree(commandBuffer);
+    }
 }