ZoneChunk.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.BitSet;
import java.util.Random;

import io.netty.buffer.ByteBuf;

import net.minecraft.nbt.NBTTagCompound;

import buildcraft.api.core.BlockIndex;
import buildcraft.api.core.ISerializable;
import buildcraft.core.lib.utils.BitSetUtils;
import buildcraft.core.lib.utils.NetworkUtils;

public class ZoneChunk implements ISerializable {

	public BitSet property;
	private boolean fullSet = false;

	public ZoneChunk() {
	}

	public boolean get(int xChunk, int zChunk) {
		if (fullSet) {
			return true;
		} else if (property == null) {
			return false;
		} else {
			return property.get(xChunk + zChunk * 16);
		}
	}

	public void set(int xChunk, int zChunk, boolean value) {
		if (value) {
			if (fullSet) {
				return;
			}

			if (property == null) {
				property = new BitSet(16 * 16);
			}

			property.set(xChunk + zChunk * 16, value);

			if (property.cardinality() >= 16 * 16) {
				property = null;
				fullSet = true;
			}
		} else {
			if (fullSet) {
				property = new BitSet(16 * 16);
				property.flip(0, 16 * 16 - 1);
				fullSet = false;
			} else if (property == null) {
				// Note - ZonePlan should usually destroy such chunks
				property = new BitSet(16 * 16);
			}

			property.set(xChunk + zChunk * 16, value);
		}
	}

	public void writeToNBT(NBTTagCompound nbt) {
		nbt.setBoolean("fullSet", fullSet);

		if (property != null) {
			nbt.setByteArray("bits", BitSetUtils.toByteArray(property));
		}
	}

	public void readFromNBT(NBTTagCompound nbt) {
		fullSet = nbt.getBoolean("fullSet");

		if (nbt.hasKey("bits")) {
			property = BitSetUtils.fromByteArray(nbt.getByteArray("bits"));
		}
	}

	public BlockIndex getRandomBlockIndex(Random rand) {
		int x, z;

		if (fullSet) {
			x = rand.nextInt(16);
			z = rand.nextInt(16);
		} else {
			int bitId = rand.nextInt(property.cardinality());
			int bitPosition = property.nextSetBit(0);

			while (bitId > 0) {
				bitId--;

				bitPosition = property.nextSetBit(bitPosition + 1);
			}

			z = bitPosition / 16;
			x = bitPosition - 16 * z;
		}
		int y = rand.nextInt(255);

		return new BlockIndex(x, y, z);
	}

	public boolean isEmpty() {
		return !fullSet && property.isEmpty();
	}

	@Override
	public void readData(ByteBuf stream) {
		int flags = stream.readUnsignedByte();
		if ((flags & 1) != 0) {
			property = BitSetUtils.fromByteArray(NetworkUtils.readByteArray(stream));
		}
		fullSet = (flags & 2) != 0;
	}

	@Override
	public void writeData(ByteBuf stream) {
		int flags = (fullSet ? 2 : 0) | (property != null ? 1 : 0);
		stream.writeByte(flags);
		if (property != null) {
			NetworkUtils.writeByteArray(stream, BitSetUtils.toByteArray(property));
		}
	}
}