ZonePlan.java

/**
 * Copyright (c) 2011-2015, SpaceToad and the BuildCraft Team
 * http://www.mod-buildcraft.com
 * <p/>
 * BuildCraft is distributed under the terms of the Minecraft Mod Public
 * License 1.0, or MMPL. Please check the contents of the license located in
 * http://www.mod-buildcraft.com/MMPL-1.0.txt
 */
package buildcraft.robotics;

import java.util.HashMap;
import java.util.Map;
import java.util.Random;

import io.netty.buffer.ByteBuf;

import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;

import net.minecraftforge.common.util.Constants;

import buildcraft.api.core.BlockIndex;
import buildcraft.api.core.ISerializable;
import buildcraft.api.core.IZone;
import buildcraft.core.ChunkIndex;

public class ZonePlan implements IZone, ISerializable {
	private final HashMap<ChunkIndex, ZoneChunk> chunkMapping = new HashMap<ChunkIndex, ZoneChunk>();

	public boolean get(int x, int z) {
		int xChunk = x >> 4;
		int zChunk = z >> 4;
		ChunkIndex chunkId = new ChunkIndex(xChunk, zChunk);
		ZoneChunk property;

		if (!chunkMapping.containsKey(chunkId)) {
			return false;
		} else {
			property = chunkMapping.get(chunkId);
			return property.get(x & 0xF, z & 0xF);
		}
	}

	public void set(int x, int z, boolean val) {
		int xChunk = x >> 4;
		int zChunk = z >> 4;
		ChunkIndex chunkId = new ChunkIndex(xChunk, zChunk);
		ZoneChunk property;

		if (!chunkMapping.containsKey(chunkId)) {
			if (val) {
				property = new ZoneChunk();
				chunkMapping.put(chunkId, property);
			} else {
				return;
			}
		} else {
			property = chunkMapping.get(chunkId);
		}

		property.set(x & 0xF, z & 0xF, val);

		if (property.isEmpty()) {
			chunkMapping.remove(chunkId);
		}
	}

	public void writeToNBT(NBTTagCompound nbt) {
		NBTTagList list = new NBTTagList();

		for (Map.Entry<ChunkIndex, ZoneChunk> e : chunkMapping.entrySet()) {
			NBTTagCompound subNBT = new NBTTagCompound();
			e.getKey().writeToNBT(subNBT);
			e.getValue().writeToNBT(subNBT);
			list.appendTag(subNBT);
		}

		nbt.setTag("chunkMapping", list);
	}

	public void readFromNBT(NBTTagCompound nbt) {
		NBTTagList list = nbt.getTagList("chunkMapping", Constants.NBT.TAG_COMPOUND);

		for (int i = 0; i < list.tagCount(); ++i) {
			NBTTagCompound subNBT = list.getCompoundTagAt(i);

			ChunkIndex id = new ChunkIndex();
			id.readFromNBT(subNBT);

			ZoneChunk chunk = new ZoneChunk();
			chunk.readFromNBT(subNBT);

			chunkMapping.put(id, chunk);
		}
	}

	@Override
	public double distanceTo(BlockIndex index) {
		return Math.sqrt(distanceToSquared(index));
	}

	@Override
	public double distanceToSquared(BlockIndex index) {
		double maxSqrDistance = Double.MAX_VALUE;

		for (Map.Entry<ChunkIndex, ZoneChunk> e : chunkMapping.entrySet()) {
			double dx = (e.getKey().x << 4 + 8) - index.x;
			double dz = (e.getKey().x << 4 + 8) - index.z;

			double sqrDistance = dx * dx + dz * dz;

			if (sqrDistance < maxSqrDistance) {
				maxSqrDistance = sqrDistance;
			}
		}

		return maxSqrDistance;
	}

	@Override
	public boolean contains(double x, double y, double z) {
		int xBlock = (int) Math.floor(x);
		int zBlock = (int) Math.floor(z);

		return get(xBlock, zBlock);
	}

	@Override
	public BlockIndex getRandomBlockIndex(Random rand) {
		if (chunkMapping.size() == 0) {
			return null;
		}

		int chunkId = rand.nextInt(chunkMapping.size());

		for (Map.Entry<ChunkIndex, ZoneChunk> e : chunkMapping.entrySet()) {
			if (chunkId == 0) {
				BlockIndex i = e.getValue().getRandomBlockIndex(rand);
				i.x = (e.getKey().x << 4) + i.x;
				i.z = (e.getKey().z << 4) + i.z;

				return i;
			}

			chunkId--;
		}

		return null;
	}

	@Override
	public void readData(ByteBuf stream) {
		chunkMapping.clear();
		int size = stream.readInt();
		for (int i = 0; i < size; i++) {
			ChunkIndex key = new ChunkIndex();
			ZoneChunk value = new ZoneChunk();
			key.readData(stream);
			value.readData(stream);
			chunkMapping.put(key, value);
		}
	}

	@Override
	public void writeData(ByteBuf stream) {
		stream.writeInt(chunkMapping.size());
		for (Map.Entry<ChunkIndex, ZoneChunk> e : chunkMapping.entrySet()) {
			e.getKey().writeData(stream);
			e.getValue().writeData(stream);
		}
	}
}