ChunkCoordComparator.java

package jk_5.nailed.server.chunkloading;

import net.minecraft.entity.player.EntityPlayerMP;
import net.minecraft.world.ChunkCoordIntPair;

// Sorter which sorts nearby chunks before far away chunks
public class ChunkCoordComparator implements java.util.Comparator<ChunkCoordIntPair> {

    private int x;
    private int z;

    public ChunkCoordComparator(EntityPlayerMP entityplayer){
        this.x = (int) entityplayer.posX >> 4;
        this.z = (int) entityplayer.posZ >> 4;
    }

    public int compare(ChunkCoordIntPair a, ChunkCoordIntPair b){
        if (a.equals(b)){
            return 0;
        }

        int ax = a.chunkXPos - this.x;
        int az = a.chunkZPos - this.z;
        int bx = b.chunkXPos - this.x;
        int bz = b.chunkZPos - this.z;
        int result = ((ax - bx) * (ax + bx)) + ((az - bz) * (az + bz));

        if(result != 0){
            return result;
        }

        if(ax < 0){
            if(bx < 0){
                return bz - az;
            }else{
                return -1;
            }
        }else{
            if(bx < 0){
                return 1;
            }else{
                return az - bz;
            }
        }
    }
}