2018-09-02 01:35:26 +00:00
|
|
|
/*
|
|
|
|
* WorldEdit, a Minecraft world manipulation toolkit
|
|
|
|
* Copyright (C) sk89q <http://www.sk89q.com>
|
|
|
|
* Copyright (C) WorldEdit team and contributors
|
|
|
|
*
|
|
|
|
* This program is free software: you can redistribute it and/or modify it
|
|
|
|
* under the terms of the GNU Lesser General Public License as published by the
|
|
|
|
* Free Software Foundation, either version 3 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
|
|
|
|
* for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU Lesser General Public License
|
|
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
|
|
|
|
package com.sk89q.worldedit.world.chunk;
|
|
|
|
|
|
|
|
import com.sk89q.jnbt.ByteTag;
|
|
|
|
import com.sk89q.jnbt.CompoundTag;
|
|
|
|
import com.sk89q.jnbt.IntTag;
|
|
|
|
import com.sk89q.jnbt.ListTag;
|
|
|
|
import com.sk89q.jnbt.LongArrayTag;
|
|
|
|
import com.sk89q.jnbt.NBTUtils;
|
|
|
|
import com.sk89q.jnbt.Tag;
|
2018-12-23 16:19:33 +00:00
|
|
|
import com.sk89q.worldedit.math.BlockVector3;
|
2018-09-02 01:35:26 +00:00
|
|
|
import com.sk89q.worldedit.registry.state.Property;
|
|
|
|
import com.sk89q.worldedit.world.DataException;
|
2018-08-10 10:29:06 +00:00
|
|
|
import com.sk89q.worldedit.world.block.BaseBlock;
|
2018-09-02 01:35:26 +00:00
|
|
|
import com.sk89q.worldedit.world.block.BlockState;
|
|
|
|
import com.sk89q.worldedit.world.block.BlockType;
|
|
|
|
import com.sk89q.worldedit.world.block.BlockTypes;
|
|
|
|
import com.sk89q.worldedit.world.storage.InvalidFormatException;
|
|
|
|
|
2019-05-31 15:20:02 +00:00
|
|
|
import javax.annotation.Nullable;
|
2018-09-02 01:35:26 +00:00
|
|
|
import java.util.HashMap;
|
|
|
|
import java.util.List;
|
|
|
|
import java.util.Map;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* The chunk format for Minecraft 1.13 and newer
|
|
|
|
*/
|
|
|
|
public class AnvilChunk13 implements Chunk {
|
|
|
|
|
|
|
|
private CompoundTag rootTag;
|
|
|
|
private BlockState[][] blocks;
|
|
|
|
private int rootX;
|
|
|
|
private int rootZ;
|
|
|
|
|
2018-12-23 16:19:33 +00:00
|
|
|
private Map<BlockVector3, Map<String,Tag>> tileEntities;
|
2018-09-02 01:35:26 +00:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Construct the chunk with a compound tag.
|
2019-06-25 17:07:47 +00:00
|
|
|
*
|
2018-09-02 01:35:26 +00:00
|
|
|
* @param tag the tag to read
|
|
|
|
* @throws DataException on a data error
|
|
|
|
*/
|
|
|
|
public AnvilChunk13(CompoundTag tag) throws DataException {
|
|
|
|
rootTag = tag;
|
|
|
|
|
|
|
|
rootX = NBTUtils.getChildTag(rootTag.getValue(), "xPos", IntTag.class).getValue();
|
|
|
|
rootZ = NBTUtils.getChildTag(rootTag.getValue(), "zPos", IntTag.class).getValue();
|
|
|
|
|
|
|
|
blocks = new BlockState[16][];
|
|
|
|
|
|
|
|
List<Tag> sections = NBTUtils.getChildTag(rootTag.getValue(), "Sections", ListTag.class).getValue();
|
|
|
|
|
|
|
|
for (Tag rawSectionTag : sections) {
|
|
|
|
if (!(rawSectionTag instanceof CompoundTag)) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
CompoundTag sectionTag = (CompoundTag) rawSectionTag;
|
|
|
|
if (!sectionTag.getValue().containsKey("Y")) {
|
|
|
|
continue; // Empty section.
|
|
|
|
}
|
|
|
|
|
|
|
|
int y = NBTUtils.getChildTag(sectionTag.getValue(), "Y", ByteTag.class).getValue();
|
|
|
|
if (y < 0 || y >= 16) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
// parse palette
|
|
|
|
List<CompoundTag> paletteEntries = sectionTag.getList("Palette", CompoundTag.class);
|
|
|
|
int paletteSize = paletteEntries.size();
|
2019-07-06 00:46:48 +00:00
|
|
|
if (paletteSize == 0) {
|
|
|
|
continue;
|
|
|
|
}
|
2018-09-02 01:35:26 +00:00
|
|
|
BlockState[] palette = new BlockState[paletteSize];
|
|
|
|
for (int paletteEntryId = 0; paletteEntryId < paletteSize; paletteEntryId++) {
|
|
|
|
CompoundTag paletteEntry = paletteEntries.get(paletteEntryId);
|
|
|
|
BlockType type = BlockTypes.get(paletteEntry.getString("Name"));
|
2019-07-06 00:46:48 +00:00
|
|
|
if (type == null) {
|
2018-09-02 01:35:26 +00:00
|
|
|
throw new InvalidFormatException("Invalid block type: " + paletteEntry.getString("Name"));
|
|
|
|
}
|
|
|
|
BlockState blockState = type.getDefaultState();
|
|
|
|
if (paletteEntry.containsKey("Properties")) {
|
|
|
|
CompoundTag properties = NBTUtils.getChildTag(paletteEntry.getValue(), "Properties", CompoundTag.class);
|
|
|
|
for (Property<?> property : blockState.getStates().keySet()) {
|
|
|
|
if (properties.containsKey(property.getName())) {
|
|
|
|
String value = properties.getString(property.getName());
|
|
|
|
try {
|
|
|
|
blockState = getBlockStateWith(blockState, property, value);
|
|
|
|
} catch (IllegalArgumentException e) {
|
|
|
|
throw new InvalidFormatException("Invalid block state for " + blockState.getBlockType().getId() + ", " + property.getName() + ": " + value);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
palette[paletteEntryId] = blockState;
|
|
|
|
}
|
|
|
|
int paletteBits = 4;
|
|
|
|
while ((1 << paletteBits) < paletteSize) {
|
|
|
|
++paletteBits;
|
|
|
|
}
|
|
|
|
int paletteMask = (1 << paletteBits) - 1;
|
|
|
|
|
|
|
|
// parse block states
|
|
|
|
long[] blockStatesSerialized = NBTUtils.getChildTag(sectionTag.getValue(), "BlockStates", LongArrayTag.class).getValue();
|
2019-07-06 00:46:48 +00:00
|
|
|
|
|
|
|
int blocksPerChunkSection = 16 * 16 * 16;
|
|
|
|
BlockState[] chunkSectionBlocks = new BlockState[blocksPerChunkSection];
|
|
|
|
blocks[y] = chunkSectionBlocks;
|
|
|
|
|
2018-09-02 01:35:26 +00:00
|
|
|
long currentSerializedValue = 0;
|
|
|
|
int nextSerializedItem = 0;
|
|
|
|
int remainingBits = 0;
|
|
|
|
for (int blockPos = 0; blockPos < blocksPerChunkSection; blockPos++) {
|
2019-07-06 00:46:48 +00:00
|
|
|
int localBlockId;
|
2018-09-02 01:35:26 +00:00
|
|
|
if (remainingBits < paletteBits) {
|
|
|
|
int bitsNextLong = paletteBits - remainingBits;
|
|
|
|
localBlockId = (int) currentSerializedValue;
|
|
|
|
if (nextSerializedItem >= blockStatesSerialized.length) {
|
|
|
|
throw new InvalidFormatException("Too short block state table");
|
|
|
|
}
|
|
|
|
currentSerializedValue = blockStatesSerialized[nextSerializedItem++];
|
|
|
|
localBlockId |= (currentSerializedValue & ((1 << bitsNextLong) - 1)) << remainingBits;
|
|
|
|
currentSerializedValue >>>= bitsNextLong;
|
|
|
|
remainingBits = 64 - bitsNextLong;
|
|
|
|
} else {
|
|
|
|
localBlockId = (int) (currentSerializedValue & paletteMask);
|
|
|
|
currentSerializedValue >>>= paletteBits;
|
|
|
|
remainingBits -= paletteBits;
|
|
|
|
}
|
|
|
|
if (localBlockId >= palette.length) {
|
|
|
|
throw new InvalidFormatException("Invalid block state table entry: " + localBlockId);
|
|
|
|
}
|
|
|
|
chunkSectionBlocks[blockPos] = palette[localBlockId];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private <T> BlockState getBlockStateWith(BlockState source, Property<T> property, String value) {
|
|
|
|
return source.with(property, property.getValueFor(value));
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Used to load the tile entities.
|
|
|
|
*
|
|
|
|
* @throws DataException
|
|
|
|
*/
|
|
|
|
private void populateTileEntities() throws DataException {
|
2019-07-06 00:46:48 +00:00
|
|
|
tileEntities = new HashMap<>();
|
2019-05-31 15:20:02 +00:00
|
|
|
if (!rootTag.getValue().containsKey("TileEntities")) {
|
|
|
|
return;
|
|
|
|
}
|
2018-09-02 01:35:26 +00:00
|
|
|
List<Tag> tags = NBTUtils.getChildTag(rootTag.getValue(),
|
|
|
|
"TileEntities", ListTag.class).getValue();
|
|
|
|
|
|
|
|
for (Tag tag : tags) {
|
|
|
|
if (!(tag instanceof CompoundTag)) {
|
|
|
|
throw new InvalidFormatException("CompoundTag expected in TileEntities");
|
|
|
|
}
|
|
|
|
|
|
|
|
CompoundTag t = (CompoundTag) tag;
|
|
|
|
|
2019-05-31 15:20:02 +00:00
|
|
|
Map<String, Tag> values = new HashMap<>(t.getValue());
|
|
|
|
int x = ((IntTag) values.get("x")).getValue();
|
|
|
|
int y = ((IntTag) values.get("y")).getValue();
|
|
|
|
int z = ((IntTag) values.get("z")).getValue();
|
2018-09-02 01:35:26 +00:00
|
|
|
|
2018-10-19 20:13:32 +00:00
|
|
|
BlockVector3 vec = BlockVector3.at(x, y, z);
|
2018-09-02 01:35:26 +00:00
|
|
|
tileEntities.put(vec, values);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Get the map of tags keyed to strings for a block's tile entity data. May
|
|
|
|
* return null if there is no tile entity data. Not public yet because
|
|
|
|
* what this function returns isn't ideal for usage.
|
|
|
|
*
|
|
|
|
* @param position the position
|
|
|
|
* @return the compound tag for that position, which may be null
|
|
|
|
* @throws DataException thrown if there is a data error
|
|
|
|
*/
|
|
|
|
@Nullable
|
2018-12-23 16:19:33 +00:00
|
|
|
private CompoundTag getBlockTileEntity(BlockVector3 position) throws DataException {
|
2018-09-02 01:35:26 +00:00
|
|
|
if (tileEntities == null) {
|
|
|
|
populateTileEntities();
|
|
|
|
}
|
|
|
|
|
2018-12-23 16:19:33 +00:00
|
|
|
Map<String, Tag> values = tileEntities.get(position);
|
2018-09-02 01:35:26 +00:00
|
|
|
if (values == null) {
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
|
|
|
return new CompoundTag(values);
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
2018-12-27 00:39:10 +00:00
|
|
|
public BaseBlock getBlock(BlockVector3 position) throws DataException {
|
2018-12-23 16:19:33 +00:00
|
|
|
int x = position.getX() - rootX * 16;
|
|
|
|
int y = position.getY();
|
|
|
|
int z = position.getZ() - rootZ * 16;
|
2018-09-02 01:35:26 +00:00
|
|
|
|
|
|
|
int section = y >> 4;
|
|
|
|
int yIndex = y & 0x0F;
|
|
|
|
|
|
|
|
if (section < 0 || section >= blocks.length) {
|
|
|
|
throw new DataException("Chunk does not contain position " + position);
|
|
|
|
}
|
|
|
|
|
|
|
|
BlockState[] sectionBlocks = blocks[section];
|
|
|
|
BlockState state = sectionBlocks != null ? sectionBlocks[(yIndex << 8) | (z << 4) | x] : BlockTypes.AIR.getDefaultState();
|
2019-06-25 17:07:47 +00:00
|
|
|
|
2018-09-07 17:11:56 +00:00
|
|
|
CompoundTag tileEntity = getBlockTileEntity(position);
|
2019-07-06 00:46:48 +00:00
|
|
|
|
2020-01-02 21:30:44 +00:00
|
|
|
if (tileEntity != null) {
|
2019-04-05 14:12:57 +00:00
|
|
|
return state.toBaseBlock(tileEntity);
|
2018-09-07 17:11:56 +00:00
|
|
|
}
|
2018-12-27 00:39:10 +00:00
|
|
|
|
|
|
|
return state.toBaseBlock();
|
2018-09-02 01:35:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|