Add blockstate cache baking progress to the loading screen

This commit is contained in:
embeddedt 2025-12-28 19:02:59 -05:00
parent 0a68a6923a
commit 617c50ffff
No known key found for this signature in database
GPG Key ID: A69433EC199B5613

View File

@ -0,0 +1,50 @@
package org.embeddedt.modernfix.common.mixin.feature.registry_event_progress;
import com.llamalad7.mixinextras.sugar.Local;
import com.llamalad7.mixinextras.sugar.Share;
import com.llamalad7.mixinextras.sugar.ref.LocalRef;
import it.unimi.dsi.fastutil.objects.ReferenceLinkedOpenHashSet;
import net.minecraft.world.level.block.Block;
import net.neoforged.fml.loading.progress.ProgressMeter;
import net.neoforged.fml.loading.progress.StartupNotificationManager;
import org.spongepowered.asm.mixin.Final;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Mutable;
import org.spongepowered.asm.mixin.Shadow;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
import java.util.Set;
@Mixin(targets = {"net/neoforged/neoforge/registries/NeoForgeRegistryCallbacks$BlockCallbacks"})
public class BlockCallbacksMixin {
@Shadow @Final @Mutable
private Set<Block> addedBlocks;
/**
* @author embeddedt
* @reason Use an ordered set to make the baking order more predictable for users watching the splash screen
*/
@Inject(method = "<init>", at = @At("RETURN"))
private void useOrderedSet(CallbackInfo ci) {
this.addedBlocks = new ReferenceLinkedOpenHashSet<>(this.addedBlocks);
}
@Inject(method = "onBake", at = @At("HEAD"))
private void startBakeProgress(CallbackInfo ci, @Share("meter") LocalRef<ProgressMeter> meter) {
meter.set(StartupNotificationManager.prependProgressBar("Build blockstate caches", addedBlocks.size()));
}
@Inject(method = "onBake", at = @At(value = "INVOKE", target = "Lnet/minecraft/world/level/block/Block;getStateDefinition()Lnet/minecraft/world/level/block/state/StateDefinition;", ordinal = 0))
private void showBakeProgressPerBlock(CallbackInfo ci, @Local(ordinal = 0) Block block, @Share("meter") LocalRef<ProgressMeter> meter) {
var id = block.builtInRegistryHolder().getKey().identifier();
meter.get().label("Build blockstate caches - " + id.toString());
meter.get().increment();
}
@Inject(method = "onBake", at = @At(value = "INVOKE", target = "Ljava/util/Set;clear()V", ordinal = 0))
private void stopBakeProgress(CallbackInfo ci, @Share("meter") LocalRef<ProgressMeter> meter) {
meter.get().complete();
}
}