Update TerrainBuilder

Author: xCollateral

src/main/java/net/vulkanmod/render/chunk/build/thread/ThreadBuilderPack.java

@@ -1,7 +1,11 @@
 package net.vulkanmod.render.chunk.build.thread;
 
+import com.mojang.blaze3d.vertex.DefaultVertexFormat;
+import net.vulkanmod.render.PipelineManager;
+import net.vulkanmod.render.vertex.CustomVertexFormat;
 import net.vulkanmod.render.vertex.TerrainBuilder;
 import net.vulkanmod.render.vertex.TerrainRenderType;
+import net.vulkanmod.render.vertex.VertexBuilder;
 
 import java.util.Arrays;
 import java.util.EnumMap;
@@ -12,7 +16,14 @@ public class ThreadBuilderPack {
     private static Function<TerrainRenderType, TerrainBuilder> terrainBuilderConstructor;
 
     public static void defaultTerrainBuilderConstructor() {
-        terrainBuilderConstructor = renderType -> new TerrainBuilder(TerrainRenderType.getLayer(renderType).bufferSize());
+        terrainBuilderConstructor = renderType -> {
+            int size = TerrainRenderType.getLayer(renderType)
+                                        .bufferSize() / DefaultVertexFormat.BLOCK.getVertexSize();
+
+            boolean compressedFormat = PipelineManager.terrainVertexFormat == CustomVertexFormat.COMPRESSED_TERRAIN;
+            VertexBuilder vertexBuilder = compressedFormat ? new VertexBuilder.CompressedVertexBuilder() : new VertexBuilder.DefaultVertexBuilder();
+            return new TerrainBuilder(size, vertexBuilder);
+        };
     }
 
     public static void setTerrainBuilderConstructor(Function<TerrainRenderType, TerrainBuilder> constructor) {

src/main/java/net/vulkanmod/render/vertex/TerrainBuilder.java

@@ -5,6 +5,7 @@
 import net.vulkanmod.Initializer;
 import net.vulkanmod.render.PipelineManager;
 import net.vulkanmod.render.chunk.cull.QuadFacing;
+import net.vulkanmod.vulkan.memory.buffer.IndexBuffer;
 import org.apache.logging.log4j.Logger;
 import org.lwjgl.system.MemoryUtil;
 
@@ -31,14 +32,12 @@ public class TerrainBuilder {
 
     private final TerrainBufferBuilder[] bufferBuilders;
 
-    public TerrainBuilder(int size) {
-        // TODO index buffer
+    public TerrainBuilder(int size, VertexBuilder vertexBuilder) {
+        this.indexBufferCapacity = size * IndexBuffer.IndexType.UINT32.size;
         this.indexBufferPtr = ALLOCATOR.malloc(size);
-        this.indexBufferCapacity = size;
 
         this.format = PipelineManager.terrainVertexFormat;
-        this.vertexBuilder = PipelineManager.terrainVertexFormat == CustomVertexFormat.COMPRESSED_TERRAIN
-                ? new VertexBuilder.CompressedVertexBuilder() : new VertexBuilder.DefaultVertexBuilder();
+        this.vertexBuilder = vertexBuilder;
 
         var bufferBuilders = new TerrainBufferBuilder[QuadFacing.COUNT];
         for (int i = 0; i < QuadFacing.COUNT; i++) {