/**
* 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));
}
}
}