package buildcraft.transport;
import io.netty.buffer.ByteBuf;
import net.minecraftforge.common.util.ForgeDirection;
import buildcraft.api.core.ISerializable;
import buildcraft.api.transport.PipeManager;
import buildcraft.api.transport.pluggable.PipePluggable;
import buildcraft.transport.utils.ConnectionMatrix;
public class PipePluggableState implements ISerializable {
private PipePluggable[] pluggables = new PipePluggable[6];
private final ConnectionMatrix pluggableMatrix = new ConnectionMatrix();
public PipePluggableState() {
}
public PipePluggable[] getPluggables() {
return pluggables;
}
public void setPluggables(PipePluggable[] pluggables) {
this.pluggables = pluggables;
}
@Override
public void writeData(ByteBuf data) {
this.pluggableMatrix.clean();
for (ForgeDirection dir : ForgeDirection.VALID_DIRECTIONS) {
this.pluggableMatrix.setConnected(dir, pluggables[dir.ordinal()] != null);
}
this.pluggableMatrix.writeData(data);
for (PipePluggable p : pluggables) {
if (p != null) {
data.writeShort(PipeManager.pipePluggables.indexOf(p.getClass()));
p.writeData(data);
}
}
}
@Override
public void readData(ByteBuf data) {
this.pluggableMatrix.readData(data);
for (ForgeDirection dir : ForgeDirection.VALID_DIRECTIONS) {
if (this.pluggableMatrix.isConnected(dir)) {
try {
Class<? extends PipePluggable> pc = PipeManager.pipePluggables.get(data.readUnsignedShort());
if (pluggables[dir.ordinal()] == null || pc != pluggables[dir.ordinal()].getClass()) {
PipePluggable p = pc.newInstance();
pluggables[dir.ordinal()] = p;
}
} catch (Exception e) {
e.printStackTrace();
}
if (pluggables[dir.ordinal()] != null) {
pluggables[dir.ordinal()].readData(data);
}
} else {
pluggables[dir.ordinal()] = null;
}
}
}
}