71 lines
2.1 KiB
Java
71 lines
2.1 KiB
Java
|
package net.minecraft.world.chunk;
|
||
|
|
|
||
|
|
import javax.annotation.Nullable;
|
||
|
|
import net.minecraft.block.Block;
|
||
|
|
import net.minecraft.block.state.IBlockState;
|
||
|
|
import net.minecraft.network.PacketBuffer;
|
||
|
|
import net.minecraft.util.IntIdentityHashBiMap;
|
||
|
|
|
||
|
|
public class BlockStatePaletteHashMap implements IBlockStatePalette {
|
||
|
|
private final IntIdentityHashBiMap<IBlockState> statePaletteMap;
|
||
|
|
private final IBlockStatePaletteResizer paletteResizer;
|
||
|
|
private final int bits;
|
||
|
|
|
||
|
|
public BlockStatePaletteHashMap(int bitsIn, IBlockStatePaletteResizer p_i47089_2_) {
|
||
|
|
this.bits = bitsIn;
|
||
|
|
this.paletteResizer = p_i47089_2_;
|
||
|
|
this.statePaletteMap = new IntIdentityHashBiMap(1 << bitsIn);
|
||
|
|
}
|
||
|
|
|
||
|
|
public int idFor(IBlockState state) {
|
||
|
|
int i = this.statePaletteMap.getId(state);
|
||
|
|
|
||
|
|
if (i == -1) {
|
||
|
|
i = this.statePaletteMap.add(state);
|
||
|
|
|
||
|
|
if (i >= 1 << this.bits) {
|
||
|
|
i = this.paletteResizer.onResize(this.bits + 1, state);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return i;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Nullable
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Gets the block state by the palette id.
|
||
|
|
*/
|
||
|
|
public IBlockState getBlockState(int indexKey) {
|
||
|
|
return (IBlockState) this.statePaletteMap.get(indexKey);
|
||
|
|
}
|
||
|
|
|
||
|
|
public void read(PacketBuffer buf) {
|
||
|
|
this.statePaletteMap.clear();
|
||
|
|
int i = buf.readVarIntFromBuffer();
|
||
|
|
|
||
|
|
for (int j = 0; j < i; ++j) {
|
||
|
|
this.statePaletteMap.add(Block.BLOCK_STATE_IDS.getByValue(buf.readVarIntFromBuffer()));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public void write(PacketBuffer buf) {
|
||
|
|
int i = this.statePaletteMap.size();
|
||
|
|
buf.writeVarIntToBuffer(i);
|
||
|
|
|
||
|
|
for (int j = 0; j < i; ++j) {
|
||
|
|
buf.writeVarIntToBuffer(Block.BLOCK_STATE_IDS.get(this.statePaletteMap.get(j)));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public int getSerializedState() {
|
||
|
|
int i = PacketBuffer.getVarIntSize(this.statePaletteMap.size());
|
||
|
|
|
||
|
|
for (int j = 0; j < this.statePaletteMap.size(); ++j) {
|
||
|
|
i += PacketBuffer.getVarIntSize(Block.BLOCK_STATE_IDS.get(this.statePaletteMap.get(j)));
|
||
|
|
}
|
||
|
|
|
||
|
|
return i;
|
||
|
|
}
|
||
|
|
}
|