fix meshemitter

Engine-Room · Oct 14, 2024 · 008b098 · 008b098
1 parent aeb9781
commit 008b098
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/fabric/src/lib/java/dev/engine_room/flywheel/lib/model/baked/MeshEmitter.java b/fabric/src/lib/java/dev/engine_room/flywheel/lib/model/baked/MeshEmitter.java
@@ -31,7 +31,9 @@ public void prepare(BakedModelBufferer.ResultConsumer resultConsumer) {
 	public void end() {
 		if (((BufferBuilderAccessor) bufferBuilder).flywheel$getBuilding()) {
 			emit();
+			bufferBuilder = new BufferBuilder(byteBufferBuilder, VertexFormat.Mode.QUADS, DefaultVertexFormat.BLOCK);
 		}
+
 		resultConsumer = null;
 	}
 
@@ -41,10 +43,11 @@ public BufferBuilder getBuffer(boolean shade) {
 	}
 
 	void prepareForGeometry(boolean shade) {
-		bufferBuilder = new BufferBuilder(byteBufferBuilder, VertexFormat.Mode.QUADS, DefaultVertexFormat.BLOCK);
-
-		if (((BufferBuilderAccessor) bufferBuilder).flywheel$getBuilding() && shade != currentShade) {
+		if (!((BufferBuilderAccessor) bufferBuilder).flywheel$getBuilding()) {
+			bufferBuilder = new BufferBuilder(byteBufferBuilder, VertexFormat.Mode.QUADS, DefaultVertexFormat.BLOCK);
+		} else if (shade != currentShade) {
 			emit();
+			bufferBuilder = new BufferBuilder(byteBufferBuilder, VertexFormat.Mode.QUADS, DefaultVertexFormat.BLOCK);
 		}
 
 		currentShade = shade;

diff --git a/forge/src/lib/java/dev/engine_room/flywheel/lib/model/baked/MeshEmitter.java b/forge/src/lib/java/dev/engine_room/flywheel/lib/model/baked/MeshEmitter.java
@@ -34,6 +34,7 @@ public void prepare(BakedModelBufferer.ResultConsumer resultConsumer) {
 	public void end() {
 		if (((BufferBuilderAccessor) bufferBuilder).flywheel$getBuilding()) {
 			emit();
+			bufferBuilder = new BufferBuilder(byteBufferBuilder, VertexFormat.Mode.QUADS, DefaultVertexFormat.BLOCK);
 		}
 		resultConsumer = null;
 	}
@@ -44,10 +45,11 @@ public BufferBuilder unwrap(boolean shade) {
 	}
 
 	void prepareForGeometry(boolean shade) {
-		bufferBuilder = new BufferBuilder(byteBufferBuilder, VertexFormat.Mode.QUADS, DefaultVertexFormat.BLOCK);
-
-		if (((BufferBuilderAccessor) bufferBuilder).flywheel$getBuilding() && shade != currentShade) {
+		if (!((BufferBuilderAccessor) bufferBuilder).flywheel$getBuilding()) {
+			bufferBuilder = new BufferBuilder(byteBufferBuilder, VertexFormat.Mode.QUADS, DefaultVertexFormat.BLOCK);
+		} else if (shade != currentShade) {
 			emit();
+			bufferBuilder = new BufferBuilder(byteBufferBuilder, VertexFormat.Mode.QUADS, DefaultVertexFormat.BLOCK);
 		}
 
 		currentShade = shade;