package com.zenpai.auraoptimizer.optimization; import com.zenpai.auraoptimizer.ZenpaiAuraOptimizer; import net.minecraft.entity.Entity; import net.minecraft.entity.EntityType; import java.util.*; public class EntityBatchingOptimizer { private final Map, List> entityBatches = new HashMap<>(); private final Set culledEntities = new HashSet<>(); private int batchedRenderCount = 0; public void prepareEntityBatch(List entities) { if (!ZenpaiAuraOptimizer.getInstance().getConfigManager() .getConfig().performance.batchEntityRendering) { return; } entityBatches.clear(); for (Entity entity : entities) { EntityType type = entity.getType(); entityBatches.computeIfAbsent(type, k -> new ArrayList<>()).add(entity); } } public boolean shouldCullEntity(Entity entity, double distance) { if (!ZenpaiAuraOptimizer.getInstance().getConfigManager() .getConfig().performance.batchEntityRendering) { return false; } int renderDistance = ZenpaiAuraOptimizer.getInstance().getConfigManager() .getConfig().rendering.maxRenderDistance * 16; if (distance > renderDistance * 0.8) { culledEntities.add(entity); return true; } culledEntities.remove(entity); return false; } public List getBatch(EntityType type) { return entityBatches.getOrDefault(type, Collections.emptyList()); } public Set> getBatchedTypes() { return entityBatches.keySet(); } public void incrementBatchCount() { batchedRenderCount++; } public int getBatchedRenderCount() { return batchedRenderCount; } public void resetBatchCount() { batchedRenderCount = 0; } public int getCulledEntityCount() { return culledEntities.size(); } public void clearCulledEntities() { culledEntities.clear(); } }