68 lines
2.6 KiB
Java
68 lines
2.6 KiB
Java
package com.atsuishio.superbwarfare.tools;
|
|
|
|
import com.atsuishio.superbwarfare.Mod;
|
|
import net.minecraft.core.BlockPos;
|
|
import net.minecraft.server.level.ServerLevel;
|
|
import net.minecraft.world.entity.Entity;
|
|
import net.minecraft.world.level.ChunkPos;
|
|
import net.neoforged.bus.api.SubscribeEvent;
|
|
import net.neoforged.fml.common.EventBusSubscriber;
|
|
import net.neoforged.neoforge.common.world.chunk.RegisterTicketControllersEvent;
|
|
import net.neoforged.neoforge.common.world.chunk.TicketController;
|
|
|
|
import java.util.HashSet;
|
|
import java.util.Set;
|
|
|
|
@EventBusSubscriber(modid = Mod.MODID, bus = EventBusSubscriber.Bus.MOD)
|
|
public class ChunkLoadTool {
|
|
|
|
private static final TicketController controller = new TicketController(Mod.loc("chunk_loader"));
|
|
|
|
/**
|
|
* 根据动量计算需要加载的区块并卸载不再需要加载的区块
|
|
*/
|
|
public static void updateLoadedChunks(ServerLevel level, Entity entity, Set<Long> loadedChunks) {
|
|
var x = entity.position().x;
|
|
var z = entity.position().z;
|
|
|
|
var nextX = x + entity.getDeltaMovement().x;
|
|
var nextZ = z + entity.getDeltaMovement().z;
|
|
|
|
// 加载当前区块和下一tick会进入的区块
|
|
var newChunks = new HashSet<Long>();
|
|
newChunks.add(ChunkPos.asLong(new BlockPos((int) x, 0, (int) z)));
|
|
newChunks.add(ChunkPos.asLong(new BlockPos((int) nextX, 0, (int) nextZ)));
|
|
|
|
// 计算需要更新的区块
|
|
var chunksToLoad = newChunks.stream().filter(chunk -> !loadedChunks.contains(chunk)).toList();
|
|
var chunksToUnload = loadedChunks.stream().filter(chunk -> !newChunks.contains(chunk)).toList();
|
|
|
|
chunksToLoad.forEach(chunk -> {
|
|
var chunkPos = new ChunkPos(chunk);
|
|
controller.forceChunk(level, entity, chunkPos.x, chunkPos.z, true, false);
|
|
});
|
|
|
|
chunksToUnload.forEach(chunk -> {
|
|
var chunkPos = new ChunkPos(chunk);
|
|
Mod.queueServerWork(10, () -> controller.forceChunk(level, entity, chunkPos.x, chunkPos.z, false, false));
|
|
});
|
|
|
|
loadedChunks.clear();
|
|
loadedChunks.addAll(newChunks);
|
|
}
|
|
|
|
/**
|
|
* 卸载所有已加载区块
|
|
*/
|
|
public static void unloadAllChunks(ServerLevel level, Entity entity, Set<Long> loadedChunks) {
|
|
loadedChunks.forEach(chunk -> {
|
|
var chunkPos = new ChunkPos(chunk);
|
|
Mod.queueServerWork(10, () -> controller.forceChunk(level, entity, chunkPos.x, chunkPos.z, false, false));
|
|
});
|
|
}
|
|
|
|
@SubscribeEvent
|
|
public static void register(RegisterTicketControllersEvent event) {
|
|
event.register(controller);
|
|
}
|
|
}
|