Add: LitematicConverter for converting .litematic to .schematic
This commit is contained in:
iceBear67 committed
1 parent
aadec09c45
commit
b6b42f4ed0
1 file changed
+193
@@ -0,0 +1,193 @@
|
||||
package io.ib67.sfcraft.util;
|
||||
|
||||
import io.netty.buffer.ByteBuf;
|
||||
import io.netty.buffer.Unpooled;
|
||||
import lombok.SneakyThrows;
|
||||
import net.minecraft.nbt.*;
|
||||
|
||||
import java.io.InputStream;
|
||||
import java.io.PipedInputStream;
|
||||
import java.io.PipedOutputStream;
|
||||
import java.util.ArrayList;
|
||||
import java.util.function.BiConsumer;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
/**
|
||||
* A cleaned up version of https://github.com/GoldenDelicios/Lite2Edit
|
||||
* Uses {@link NbtCompound} from Minecraft.
|
||||
*/
|
||||
public class LitematicConverter implements AutoCloseable {
|
||||
public static final String NBT_LITEMATICA_ROOT = "";
|
||||
private final InputStream input;
|
||||
private final NbtSizeTracker sizeTracker;
|
||||
private NbtCompound root;
|
||||
|
||||
public LitematicConverter(InputStream input, NbtSizeTracker sizeTracker) {
|
||||
this.input = input;
|
||||
this.sizeTracker = sizeTracker;
|
||||
}
|
||||
|
||||
@SneakyThrows
|
||||
public void read(
|
||||
BiConsumer<Integer, NbtCompound> schematicOutput
|
||||
) {
|
||||
if (root != null) throw new IllegalStateException("Litematica file is already read");
|
||||
root = NbtIo.readCompressed(input, sizeTracker).getCompound(NBT_LITEMATICA_ROOT);
|
||||
var dataVersion = root.getInt("MinecraftDataVersion");
|
||||
var regionsNbt = root.getList("Regions", NbtElement.COMPOUND_TYPE);
|
||||
var i = 0;
|
||||
for (NbtElement nbtElement : regionsNbt) {
|
||||
var compound = (NbtCompound) nbtElement;
|
||||
var j = i++;
|
||||
var schematic = convertRegionToSchematic(dataVersion, compound);
|
||||
schematicOutput.accept(j, schematic);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* @param region region
|
||||
* @return a path to temp file.
|
||||
*/
|
||||
private NbtCompound convertRegionToSchematic(int dataVersion, NbtCompound region) {
|
||||
var worldEditTag = new NbtCompound();
|
||||
|
||||
// prepare metadata
|
||||
var size = readSizeTuple(region);
|
||||
worldEditTag.put("Metadata", convertToWeMeta(size, region));
|
||||
worldEditTag.putInt("DataVersion", dataVersion);
|
||||
|
||||
// dimensional data.
|
||||
worldEditTag.putShort("Height", (short) Math.abs(size.y));
|
||||
worldEditTag.putShort("Length", (short) Math.abs(size.z));
|
||||
worldEditTag.putShort("Width", (short) Math.abs(size.x));
|
||||
|
||||
// block & tile entities
|
||||
var paletteNbt = region.getList("BlockStatePalette", NbtElement.COMPOUND_TYPE);
|
||||
var wePalette = convertToWEPalette(paletteNbt);
|
||||
worldEditTag.putInt("PaletteMax", wePalette.getKeys().size());
|
||||
worldEditTag.put("Palette", wePalette);
|
||||
var tileEntities = region.getList("TileEntities", NbtElement.COMPOUND_TYPE);
|
||||
worldEditTag.put("BlockEntities", convertToWETileEntities(tileEntities));
|
||||
worldEditTag.putInt("Version", 2);
|
||||
worldEditTag.putIntArray("Offset", new int[3]);
|
||||
worldEditTag.putByteArray("BlockData", convertToWEBlocks(size.x * size.y * size.z, region));
|
||||
var schematicsRoot = new NbtCompound();
|
||||
schematicsRoot.put("Schematic", worldEditTag);
|
||||
return schematicsRoot;
|
||||
}
|
||||
|
||||
private SizeTuple readSizeTuple(NbtCompound region) {
|
||||
var size = (NbtCompound) region.get("Size");
|
||||
var sizeX = size.getInt("x");
|
||||
var sizeY = size.getInt("y");
|
||||
var sizeZ = size.getInt("z");
|
||||
return new SizeTuple(sizeX, sizeY, sizeZ);
|
||||
}
|
||||
|
||||
private NbtList convertToWETileEntities(NbtList tileEntities) {
|
||||
var weTEs = new NbtList();
|
||||
for (NbtElement _tileEntity : tileEntities) {
|
||||
var tE = (NbtCompound) _tileEntity;
|
||||
tE.putIntArray("Pos", new int[]{
|
||||
tE.getInt("x"),
|
||||
tE.getInt("y"),
|
||||
tE.getInt("z")
|
||||
});
|
||||
tE.putString("Id", tE.getString("id"));
|
||||
// other properties
|
||||
tE.remove("x");
|
||||
tE.remove("y");
|
||||
tE.remove("z");
|
||||
tE.remove("id");
|
||||
weTEs.add(tE);
|
||||
}
|
||||
return weTEs;
|
||||
}
|
||||
|
||||
private NbtCompound convertToWEPalette(NbtList paletteNbt) {
|
||||
var wePalette = new NbtCompound();
|
||||
for (int i = 0; i < paletteNbt.size(); i++) {
|
||||
var entry = (NbtCompound) paletteNbt.get(i);
|
||||
var name = new StringBuilder(entry.getString("Name"));
|
||||
var properties = entry.getCompound("Properties");
|
||||
if (!properties.isEmpty()) {
|
||||
name.append("[");
|
||||
var props = new ArrayList<String>();
|
||||
for (String key : properties.getKeys()) {
|
||||
props.add(key + "=" + properties.getString(key));
|
||||
}
|
||||
name.append(props.stream().collect(Collectors.joining(",")));
|
||||
name.append("]");
|
||||
}
|
||||
wePalette.putInt(name.toString(), i);
|
||||
}
|
||||
return wePalette;
|
||||
}
|
||||
|
||||
private NbtElement convertToWeMeta(SizeTuple size, NbtCompound region) {
|
||||
var pos = (NbtCompound) region.get("Position");
|
||||
var nbt = new NbtCompound();
|
||||
nbt.putInt("WEOffsetX", pos.getInt("x") + size.x < 0 ? size.x + 1 : 0);
|
||||
nbt.putInt("WEOffsetY", pos.getInt("y") + size.y < 0 ? size.y + 1 : 0);
|
||||
nbt.putInt("WEOffsetZ", pos.getInt("z") + size.z < 0 ? size.z + 1 : 0);
|
||||
return nbt;
|
||||
}
|
||||
|
||||
private byte[] convertToWEBlocks(int totalBlocks, NbtCompound region) {
|
||||
var buf = Unpooled.buffer();
|
||||
var blockStates = region.getLongArray("BlockStates");
|
||||
|
||||
int bitsPerBlock = region.getList("BlockStatePalette", NbtElement.COMPOUND_TYPE).size();
|
||||
bitsPerBlock = Math.max(2, Integer.SIZE - Integer.numberOfTrailingZeros(bitsPerBlock - 1));
|
||||
|
||||
int bitCounts = 0, blockIterated = 0;
|
||||
long bitMask, bits = 0;
|
||||
for (long blockState : blockStates) {
|
||||
int remainingBits = bitCounts + 64;
|
||||
if (bitCounts != 0) {
|
||||
bitMask = (1L << (bitsPerBlock - bitCounts)) - 1;
|
||||
long newBits = (blockState & bitMask) << bitCounts;
|
||||
bits = bits | newBits;
|
||||
blockState = blockState >>> (bitsPerBlock - bitCounts);
|
||||
remainingBits -= bitsPerBlock;
|
||||
writeBlocks(buf, (short) bits);
|
||||
blockIterated++;
|
||||
}
|
||||
|
||||
bitMask = (1L << bitsPerBlock) - 1;
|
||||
while(remainingBits >= bitsPerBlock){
|
||||
bits = blockState & bitMask;
|
||||
blockState = blockState >>> bitsPerBlock;
|
||||
remainingBits -= bitsPerBlock;
|
||||
if(blockIterated >= totalBlocks) break;
|
||||
writeBlocks(buf, (short) bits);
|
||||
blockIterated++;
|
||||
}
|
||||
bits = blockState;
|
||||
bitCounts = remainingBits;
|
||||
}
|
||||
return buf.array();
|
||||
}
|
||||
|
||||
private static int writeBlocks(ByteBuf buf, short block) {
|
||||
int b = block >>> 7;
|
||||
if (b == 0) {
|
||||
buf.writeByte(block);
|
||||
return 1;
|
||||
} else {
|
||||
buf.writeByte(block | 128);
|
||||
buf.writeByte(block);
|
||||
return 2;
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void close() throws Exception {
|
||||
input.close();
|
||||
}
|
||||
|
||||
record SizeTuple(
|
||||
int x, int y, int z
|
||||
) {
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user