forked from SteamWar/SteamWar
Add new Schematic reader
This commit is contained in:
@@ -0,0 +1,261 @@
|
||||
/*
|
||||
* This file is a part of the SteamWar software.
|
||||
*
|
||||
* Copyright (C) 2025 SteamWar.de-Serverteam
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero 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 Affero General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
package de.steamwar.core;
|
||||
|
||||
import com.sk89q.worldedit.WorldEdit;
|
||||
import com.sk89q.worldedit.extension.platform.Capability;
|
||||
import com.sk89q.worldedit.extension.platform.Platform;
|
||||
import com.sk89q.worldedit.extent.clipboard.BlockArrayClipboard;
|
||||
import com.sk89q.worldedit.extent.clipboard.Clipboard;
|
||||
import com.sk89q.worldedit.extent.clipboard.io.ClipboardReader;
|
||||
import com.sk89q.worldedit.math.BlockVector3;
|
||||
import com.sk89q.worldedit.regions.CuboidRegion;
|
||||
import com.sk89q.worldedit.regions.Region;
|
||||
import com.sk89q.worldedit.world.DataFixer;
|
||||
|
||||
import java.io.DataInputStream;
|
||||
import java.io.IOException;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
public class ChaosSchemReader implements ClipboardReader {
|
||||
|
||||
private final DataInputStream stream;
|
||||
|
||||
public ChaosSchemReader(DataInputStream stream) {
|
||||
this.stream = stream;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Clipboard read() throws IOException {
|
||||
byte tag = stream.readByte();
|
||||
|
||||
if (tag != 0x0A) error("Invalid NBT");
|
||||
|
||||
String name = stream.readUTF();
|
||||
|
||||
if (name.isEmpty()) {
|
||||
tag = stream.readByte();
|
||||
if (tag != 0x0A) error("Invalid NBT");
|
||||
|
||||
stream.readUTF();
|
||||
|
||||
readSchemTag();
|
||||
|
||||
stream.readByte();
|
||||
} else {
|
||||
readSchemTag();
|
||||
}
|
||||
}
|
||||
|
||||
private int version;
|
||||
private int dataVersion;
|
||||
private DataFixer fixer;
|
||||
|
||||
private int width;
|
||||
private int height;
|
||||
private int length;
|
||||
|
||||
private int[] offset = new int[3];
|
||||
|
||||
private Map<String, Integer> palette = new HashMap<>();
|
||||
private int[] blocks;
|
||||
|
||||
private Map<String, Object> metadata;
|
||||
|
||||
private Clipboard constructSchematic() {
|
||||
BlockVector3 min = BlockVector3.at(offset[0], offset[1], offset[2]);
|
||||
|
||||
Region region = new CuboidRegion(min);
|
||||
|
||||
BlockArrayClipboard clipboard = new BlockArrayClipboard(region);
|
||||
}
|
||||
|
||||
private void readSchemTag() throws IOException {
|
||||
parseCompoundTag((type, name) -> {
|
||||
switch (name) {
|
||||
case "Version" -> version = stream.readInt();
|
||||
case "DataVersion" -> {
|
||||
dataVersion = stream.readInt();
|
||||
|
||||
Platform platform = WorldEdit.getInstance().getPlatformManager()
|
||||
.queryCapability(Capability.WORLD_EDITING);
|
||||
|
||||
if (dataVersion < platform.getDataVersion()) {
|
||||
this.fixer = platform.getDataFixer();
|
||||
}
|
||||
}
|
||||
case "Metadata" -> metadata = (Map<String, Object>) readByType((byte) 0x0A);
|
||||
case "Width" -> width = stream.readUnsignedShort();
|
||||
case "Height" -> height = stream.readUnsignedShort();
|
||||
case "Length" -> length = stream.readUnsignedShort();
|
||||
case "Offset" -> offset = readIntArray();
|
||||
case "PaletteMax" -> stream.skipNBytes(4);
|
||||
case "Palette" -> readPalette();
|
||||
case "BlockData" -> readBlockData();
|
||||
case "Blocks" -> readBlockContainer();
|
||||
|
||||
default -> skipByType(type);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
private void readBlockContainer() throws IOException {
|
||||
parseCompoundTag((type, name) -> {
|
||||
switch (name) {
|
||||
case "Palette" -> readPalette();
|
||||
case "Data" -> readBlockData();
|
||||
|
||||
default -> skipByType(type);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
private void readPalette() throws IOException {
|
||||
parseCompoundTag((aByte, string) -> palette.put(string, stream.readInt()));
|
||||
}
|
||||
|
||||
private static final int SEGMENT_BITS = 0x7F;
|
||||
private static final int CONTINUE_BIT = 0x80;
|
||||
|
||||
private void readBlockData() throws IOException {
|
||||
byte[] data = readByteArray();
|
||||
|
||||
int[] blocks = new int[data.length];
|
||||
|
||||
int x = 0;
|
||||
int position = 0;
|
||||
|
||||
for (byte current : data) {
|
||||
blocks[x] |= (current & SEGMENT_BITS) << position;
|
||||
|
||||
if ((current & CONTINUE_BIT) == 0) {
|
||||
x++;
|
||||
position = 0;
|
||||
} else {
|
||||
position += 7;
|
||||
|
||||
if (position >= 32) throw new RuntimeException("VarInt is too big");
|
||||
}
|
||||
}
|
||||
|
||||
this.blocks = blocks;
|
||||
}
|
||||
|
||||
private void parseCompoundTag(CompoundReader tag) throws IOException {
|
||||
byte type;
|
||||
while ((type = stream.readByte()) != 0x00) {
|
||||
String name = stream.readUTF();
|
||||
|
||||
tag.read(type, name);
|
||||
}
|
||||
}
|
||||
|
||||
private int[] readIntArray() throws IOException {
|
||||
int length = stream.readInt();
|
||||
int[] arr = new int[length];
|
||||
|
||||
for (int i = 0; i < length; i++) {
|
||||
arr[i] = stream.readInt();
|
||||
}
|
||||
|
||||
return arr;
|
||||
}
|
||||
|
||||
private byte[] readByteArray() throws IOException {
|
||||
return stream.readNBytes(stream.readInt());
|
||||
}
|
||||
|
||||
private void skipByType(byte type, String str) throws IOException {
|
||||
skipByType(type);
|
||||
}
|
||||
|
||||
private void skipByType(byte type) throws IOException {
|
||||
switch (type) {
|
||||
case 0x00 -> stream.skipNBytes(0);
|
||||
case 0x01 -> stream.skipNBytes(1);
|
||||
case 0x02 -> stream.skipNBytes(2);
|
||||
case 0x03, 0x05 -> stream.skipNBytes(4);
|
||||
case 0x04, 0x06 -> stream.skipNBytes(8);
|
||||
case 0x07 -> stream.skipNBytes(stream.readInt());
|
||||
case 0x08 -> stream.readUTF();
|
||||
case 0x09 -> {
|
||||
byte t = stream.readByte();
|
||||
int length = stream.readInt();
|
||||
for (int i = 0; i < length; i++) { skipByType(t); }
|
||||
}
|
||||
case 0x0A -> parseCompoundTag(this::skipByType);
|
||||
case 0x0B -> stream.skipNBytes(4L * stream.readInt());
|
||||
case 0x0C -> stream.skipNBytes(8L * stream.readInt());
|
||||
default -> error("Invalid Type");
|
||||
}
|
||||
}
|
||||
|
||||
private Object readByType(byte type) throws IOException {
|
||||
return switch (type) {
|
||||
case 0x00 -> null;
|
||||
case 0x01 -> stream.readByte();
|
||||
case 0x02 -> stream.readShort();
|
||||
case 0x03 -> stream.readInt();
|
||||
case 0x04 -> stream.readLong();
|
||||
case 0x05 -> stream.readFloat();
|
||||
case 0x06 -> stream.readDouble();
|
||||
case 0x07 -> stream.readNBytes(stream.readInt());
|
||||
case 0x08 -> stream.readUTF();
|
||||
case 0x09 -> readArray();
|
||||
case 0x0A -> readCompound();
|
||||
case 0x0B -> stream.skipNBytes(4L * stream.readInt());
|
||||
case 0x0C -> stream.skipNBytes(8L * stream.readInt());
|
||||
default -> error("Invalid Type");
|
||||
};
|
||||
}
|
||||
|
||||
private Object[] readArray() throws IOException {
|
||||
byte t = stream.readByte();
|
||||
int length = stream.readInt();
|
||||
Object[] obj = new Object[length];
|
||||
for (int i = 0; i < length; i++) {
|
||||
obj[i] = readByType(t);
|
||||
}
|
||||
|
||||
return obj;
|
||||
}
|
||||
|
||||
private Map<String, Object> readCompound() throws IOException {
|
||||
Map<String, Object> entries = new HashMap<>();
|
||||
|
||||
parseCompoundTag((type, name) -> entries.put(name, readByType(type)));
|
||||
|
||||
return entries;
|
||||
}
|
||||
|
||||
private void error(String message) {
|
||||
throw new IllegalStateException("Could not read Schem: " + message);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void close() throws IOException {
|
||||
stream.close();
|
||||
}
|
||||
|
||||
private interface CompoundReader {
|
||||
void read(byte type, String name) throws IOException;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user