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

package com.hbm.packet;

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;
	byte biome;
	byte[] biomeArray;
	
	public BiomeSyncPacket() { }
	
	public BiomeSyncPacket(int chunkX, int chunkZ, byte[] biomeArray) {
		this.chunkX = chunkX;
		this.chunkZ = chunkZ;
		this.biomeArray = biomeArray;
	}
	
	public BiomeSyncPacket(int blockX, int blockZ, byte 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.writeByte(this.biome);
			buf.writeByte(this.blockX);
			buf.writeByte(this.blockZ);
		} else {
			buf.writeBoolean(true);
			for(int i = 0; i < 256; i++) {
				buf.writeByte(this.biomeArray[i]);
			}
		}
	}

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

	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);
			
			if(m.biomeArray == null) {
				chunk.getBiomeArray()[(m.blockZ & 15) << 4 | (m.blockX & 15)] = m.biome;
			} else {
				for(int i = 0; i < 256; i++) {
					chunk.getBiomeArray()[i] = m.biomeArray[i];
				}
			}
			
			return null;
		}
	}
}