Hbm-s-Nuclear-Tech-GIT/src/main/java/com/hbm/packet/toclient/BiomeSyncPacket.java

package com.hbm.packet.toclient;

import cpw.mods.fml.common.Loader;
import com.hbm.util.Compat;
import com.hbm.world.WorldUtil;

import cpw.mods.fml.common.network.simpleimpl.IMessage;
import cpw.mods.fml.common.network.simpleimpl.IMessageHandler;
import cpw.mods.fml.common.network.simpleimpl.MessageContext;
import cpw.mods.fml.relauncher.Side;
import cpw.mods.fml.relauncher.SideOnly;
import io.netty.buffer.ByteBuf;
import net.minecraft.client.Minecraft;
import net.minecraft.world.World;
import net.minecraft.world.chunk.Chunk;

public class BiomeSyncPacket implements IMessage {
	
	int chunkX;
	int chunkZ;
	byte blockX;
	byte blockZ;
	short biome;
	short[] biomeArray;
	
	public BiomeSyncPacket() { }

	public BiomeSyncPacket(int chunkX, int chunkZ, byte[] biomeArray) {
		this(chunkX, chunkZ, bytesToShorts(biomeArray));
	}

	public BiomeSyncPacket(int blockX, int blockZ, byte biome) {
		this(blockX, blockZ, (short) biome);
	}
	
	public BiomeSyncPacket(int chunkX, int chunkZ, short[] biomeArray) {
		this.chunkX = chunkX;
		this.chunkZ = chunkZ;
		this.biomeArray = biomeArray;
	}
	
	public BiomeSyncPacket(int blockX, int blockZ, short biome) {
		this.chunkX = blockX >> 4;
		this.chunkZ = blockZ >> 4;
		this.blockX = (byte) (blockX & 15);
		this.blockZ = (byte) (blockZ & 15);
		this.biome = biome;
	}

	@Override
	public void toBytes(ByteBuf buf) {
		buf.writeInt(this.chunkX);
		buf.writeInt(this.chunkZ);
		
		if(this.biomeArray == null) {
			buf.writeBoolean(false);
			buf.writeShort(this.biome);
			buf.writeByte(this.blockX);
			buf.writeByte(this.blockZ);
		} else {
			buf.writeBoolean(true);
			for(int i = 0; i < 256; i++) {
				buf.writeShort(this.biomeArray[i]);
			}
		}
	}

	@Override
	public void fromBytes(ByteBuf buf) {
		this.chunkX = buf.readInt();
		this.chunkZ = buf.readInt();
		
		if(!buf.readBoolean()) {
			this.biome = buf.readShort();
			this.blockX = buf.readByte();
			this.blockZ = buf.readByte();
		} else {
			this.biomeArray = new short[256];
			for(int i = 0; i < 256; i++) {
				this.biomeArray[i] = buf.readShort();
			}
		}
	}

	private final static short[] bytesToShorts(byte[] byteArray) {
		int size = byteArray.length;
		short[] shortArray = new short[size];

		for(int index = 0; index < size; index++)
			shortArray[index] = (short) byteArray[index];

		return shortArray;
	}

	public static class Handler implements IMessageHandler<BiomeSyncPacket, IMessage> {
		
		@Override
		@SideOnly(Side.CLIENT)
		public IMessage onMessage(BiomeSyncPacket m, MessageContext ctx) {
			
			World world = Minecraft.getMinecraft().theWorld;
			if(!world.getChunkProvider().chunkExists(m.chunkX, m.chunkZ)) return null;
			Chunk chunk = world.getChunkFromChunkCoords(m.chunkX, m.chunkZ);
			chunk.isModified = true;

			if(Loader.isModLoaded(Compat.MOD_EIDS)) {
				short[] target = WorldUtil.getBiomeShortArray(chunk);
				if(m.biomeArray == null) {
					target[(m.blockZ & 15) << 4 | m.blockX & 15] = m.biome;
					world.markBlockRangeForRenderUpdate(m.chunkX << 4, 0, m.chunkZ << 4, m.chunkX << 4, 255, m.chunkZ << 4);
				} else {
					for(int i = 0; i < 255; ++i) {
						target[i] = m.biomeArray[i];
						world.markBlockRangeForRenderUpdate(m.chunkX << 4, 0, m.chunkZ << 4, (m.chunkX << 4) + 15, 255, (m.chunkZ << 4) + 15);
					}
				}
			} else {
				byte[] target = chunk.getBiomeArray();
				if(m.biomeArray == null) {
					target[(m.blockZ & 15) << 4 | (m.blockX & 15)] = (byte) m.biome;
					world.markBlockRangeForRenderUpdate(m.chunkX << 4, 0, m.chunkZ << 4, m.chunkX << 4, 255, m.chunkZ << 4);
				} else {
					for(int i = 0; i < 256; i++) {
						target[i] = (byte) m.biomeArray[i];
						world.markBlockRangeForRenderUpdate(m.chunkX << 4, 0, m.chunkZ << 4, (m.chunkX << 4) + 15, 255, (m.chunkZ << 4) + 15);
					}
				}
			}
			
			return null;
		}
	}
}