CustomNPC-Plus/src/main/java/noppes/npcs/controllers/ChunkController.java

package noppes.npcs.controllers;

import net.minecraft.entity.Entity;
import net.minecraft.util.MathHelper;
import net.minecraft.world.ChunkCoordIntPair;
import net.minecraft.world.World;
import net.minecraftforge.common.ForgeChunkManager;
import net.minecraftforge.common.ForgeChunkManager.LoadingCallback;
import net.minecraftforge.common.ForgeChunkManager.Ticket;
import net.minecraftforge.common.ForgeChunkManager.Type;
import noppes.npcs.CustomNpcs;
import noppes.npcs.config.ConfigMain;
import noppes.npcs.constants.EnumJobType;
import noppes.npcs.entity.EntityNPCInterface;

import java.util.HashMap;
import java.util.Iterator;
import java.util.List;

public class ChunkController implements LoadingCallback {
    public static ChunkController Instance;

    public HashMap<Entity, Ticket> tickets = new HashMap<Entity, Ticket>();

    public ChunkController() {
        Instance = this;
    }

    public void clear() {
        tickets = new HashMap<Entity, Ticket>();
    }

    public Ticket getTicket(EntityNPCInterface npc) {
        Ticket ticket = tickets.get(npc);
        if (ticket != null)
            return ticket;
        if (size() >= ConfigMain.ChunkLoaders)
            return null;
        ticket = ForgeChunkManager.requestTicket(CustomNpcs.instance, npc.worldObj, Type.ENTITY);
        if (ticket == null)
            return null;
        ticket.bindEntity(npc);
        ticket.setChunkListDepth(6);
        tickets.put(npc, ticket);
        return null;
    }

    public void deleteNPC(EntityNPCInterface npc) {
        Ticket ticket = tickets.get(npc);
        if (ticket != null) {
            tickets.remove(npc);
            ForgeChunkManager.releaseTicket(ticket);
        }
    }

    @Override
    public void ticketsLoaded(List<Ticket> tickets, World world) {
        for (Ticket ticket : tickets) {
            if (!(ticket.getEntity() instanceof EntityNPCInterface))
                continue;
            EntityNPCInterface npc = (EntityNPCInterface) ticket.getEntity();
            if (npc.advanced.job == EnumJobType.ChunkLoader && !tickets.contains(npc)) {
                this.tickets.put(npc, ticket);
                double x = npc.posX / 16;
                double z = npc.posZ / 16;

                ForgeChunkManager.forceChunk(ticket, new ChunkCoordIntPair(MathHelper.floor_double(x), MathHelper.floor_double(z)));
                ForgeChunkManager.forceChunk(ticket, new ChunkCoordIntPair(MathHelper.ceiling_double_int(x), MathHelper.ceiling_double_int(z)));
                ForgeChunkManager.forceChunk(ticket, new ChunkCoordIntPair(MathHelper.floor_double(x), MathHelper.ceiling_double_int(z)));
                ForgeChunkManager.forceChunk(ticket, new ChunkCoordIntPair(MathHelper.ceiling_double_int(x), MathHelper.floor_double(z)));
            }
        }
    }

    public int size() {
        return tickets.size();
    }

    public void unload(int toRemove) {
        Iterator<Entity> ite = tickets.keySet().iterator();
        int i = 0;
        while (ite.hasNext()) {
            if (i >= toRemove)
                return;
            Entity entity = ite.next();
            ForgeChunkManager.releaseTicket(tickets.get(entity));
            ite.remove();
            i++;
        }
    }
}