Add some simple patches to cut down allocation rate when ticking chunks

This commit is contained in:
embeddedt 2024-03-20 15:44:28 -04:00
parent a0fdb3e6d7
commit 070b7b6d12
No known key found for this signature in database
GPG Key ID: A69433EC199B5613
4 changed files with 146 additions and 0 deletions

View File

@ -0,0 +1,28 @@
package org.embeddedt.modernfix.common.mixin.perf.ticking_chunk_alloc;
import net.minecraft.world.entity.ambient.Bat;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Redirect;
import java.time.LocalDate;
@Mixin(value = Bat.class, priority = 1200)
public class BatMixin {
private static long mfix$lastQueriedTime = -1L;
private static LocalDate mfix$lastQueriedDate = null;
/**
* @author embeddedt
* @reason avoid excessive allocations from continuously querying the date, only get a new date once every 30 seconds
*/
@Redirect(method = "isHalloween", at = @At(value = "INVOKE", target = "Ljava/time/LocalDate;now()Ljava/time/LocalDate;"), require = 0)
private static LocalDate useCachedLocalDate() {
LocalDate date = mfix$lastQueriedDate;
if(date == null || Math.abs(System.currentTimeMillis() - mfix$lastQueriedTime) > 30000) {
mfix$lastQueriedDate = date = LocalDate.now();
mfix$lastQueriedTime = System.currentTimeMillis();
}
return date;
}
}

View File

@ -0,0 +1,26 @@
package org.embeddedt.modernfix.common.mixin.perf.ticking_chunk_alloc;
import net.minecraft.world.level.chunk.ChunkGenerator;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Redirect;
import java.util.Collections;
import java.util.Map;
import java.util.Set;
@Mixin(ChunkGenerator.class)
public class ChunkGeneratorMixin {
/**
* @author embeddedt
* @reason Avoid allocation if the chunk contains no structures
*/
@Redirect(method = "getMobsAt", at = @At(value = "INVOKE", target = "Ljava/util/Map;entrySet()Ljava/util/Set;"), require = 0)
private Set<?> avoidSetAllocation(Map<?, ?> instance) {
if(instance.isEmpty()) {
return Collections.emptySet();
} else {
return instance.entrySet();
}
}
}

View File

@ -0,0 +1,40 @@
package org.embeddedt.modernfix.common.mixin.perf.ticking_chunk_alloc;
import com.mojang.datafixers.util.Either;
import net.minecraft.server.level.ChunkHolder;
import net.minecraft.world.level.chunk.LevelChunk;
import org.embeddedt.modernfix.util.EitherUtil;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Overwrite;
import org.spongepowered.asm.mixin.Shadow;
import java.util.concurrent.CompletableFuture;
@Mixin(value = ChunkHolder.class, priority = 500)
public abstract class ChunkHolderMixin {
@Shadow public abstract CompletableFuture<Either<LevelChunk, ChunkHolder.ChunkLoadingFailure>> getTickingChunkFuture();
@Shadow public abstract CompletableFuture<Either<LevelChunk, ChunkHolder.ChunkLoadingFailure>> getFullChunkFuture();
/**
* @author embeddedt
* @reason avoid Optional allocation
*/
@Overwrite
public LevelChunk getTickingChunk() {
CompletableFuture<Either<LevelChunk, ChunkHolder.ChunkLoadingFailure>> completableFuture = this.getTickingChunkFuture();
Either<LevelChunk, ChunkHolder.ChunkLoadingFailure> either = completableFuture.getNow(null);
return either == null ? null : EitherUtil.leftOrNull(either);
}
/**
* @author embeddedt
* @reason avoid Optional allocation
*/
@Overwrite
public LevelChunk getFullChunk() {
CompletableFuture<Either<LevelChunk, ChunkHolder.ChunkLoadingFailure>> completableFuture = this.getFullChunkFuture();
Either<LevelChunk, ChunkHolder.ChunkLoadingFailure> either = completableFuture.getNow(null);
return either == null ? null : EitherUtil.leftOrNull(either);
}
}

View File

@ -0,0 +1,52 @@
package org.embeddedt.modernfix.util;
import com.mojang.datafixers.util.Either;
import java.lang.invoke.MethodHandle;
import java.lang.invoke.MethodHandles;
import java.lang.invoke.MethodType;
import java.lang.reflect.Field;
public class EitherUtil {
private static final Class<?> LEFT, RIGHT;
private static final MethodHandle LEFT_VAL, RIGHT_VAL;
static {
try {
LEFT = Class.forName("com.mojang.datafixers.util.Either$Left");
RIGHT = Class.forName("com.mojang.datafixers.util.Either$Right");
Field lvalue = LEFT.getDeclaredField("value");
lvalue.setAccessible(true);
Field rvalue = RIGHT.getDeclaredField("value");
rvalue.setAccessible(true);
LEFT_VAL = MethodHandles.publicLookup().unreflectGetter(lvalue).asType(MethodType.methodType(Object.class, Either.class));
RIGHT_VAL = MethodHandles.publicLookup().unreflectGetter(rvalue).asType(MethodType.methodType(Object.class, Either.class));
} catch(ReflectiveOperationException e) {
throw new AssertionError("Failed to hook DFU Either", e);
}
}
@SuppressWarnings("unchecked")
public static <L, R> L leftOrNull(Either<L, R> either) {
if(either.getClass() == LEFT) {
try {
return (L)LEFT_VAL.invokeExact(either);
} catch(Throwable e) {
throw new RuntimeException(e);
}
}
return null;
}
@SuppressWarnings("unchecked")
public static <L, R> R rightOrNull(Either<L, R> either) {
if(either.getClass() == RIGHT) {
try {
return (R)RIGHT_VAL.invokeExact(either);
} catch(Throwable e) {
throw new RuntimeException(e);
}
}
return null;
}
}