feat: add basic minestom NBT additions and improve block state parser performance

This commit is contained in:
Christian Bergschneider
2025-12-11 01:30:40 +01:00
parent ddc8cc7db5
commit d262831107
6 changed files with 124 additions and 24 deletions

View File

@@ -27,6 +27,8 @@ public class TerraMinestomExample {
private TerraMinestomWorld world; private TerraMinestomWorld world;
public static void main(String[] args) { public static void main(String[] args) {
System.setProperty("minestom.registry.unsafe-ops", "true");
TerraMinestomExample example = new TerraMinestomExample(); TerraMinestomExample example = new TerraMinestomExample();
example.createNewInstance(); example.createNewInstance();
example.attachTerra(); example.attachTerra();

View File

@@ -1,6 +1,7 @@
package com.dfsek.terra.minestom.api; package com.dfsek.terra.minestom.api;
import net.kyori.adventure.nbt.CompoundBinaryTag;
import net.minestom.server.entity.Entity; import net.minestom.server.entity.Entity;
import net.minestom.server.entity.EntityType; import net.minestom.server.entity.EntityType;
@@ -9,5 +10,22 @@ import net.minestom.server.entity.EntityType;
* Allows adding AI to generated entities using custom entity types * Allows adding AI to generated entities using custom entity types
*/ */
public interface EntityFactory { public interface EntityFactory {
/**
* Creates a new entity of the specified type.
*
* @param type the type of the entity to be created
* @return the created entity instance
*/
Entity createEntity(EntityType type); Entity createEntity(EntityType type);
/**
* Creates a new entity of the specified type with additional data.
*
* @param type the type of the entity to be created
* @param data the additional data for the entity, represented as a CompoundBinaryTag
* @return the created entity instance
*/
default Entity createEntity(EntityType type, CompoundBinaryTag data) {
return createEntity(type);
}
} }

View File

@@ -1,7 +1,10 @@
package com.dfsek.terra.minestom.block; package com.dfsek.terra.minestom.block;
import net.kyori.adventure.nbt.CompoundBinaryTag;
import net.kyori.adventure.nbt.TagStringIO;
import net.minestom.server.instance.block.Block; import net.minestom.server.instance.block.Block;
import java.io.IOException;
import java.util.HashMap; import java.util.HashMap;
import java.util.Objects; import java.util.Objects;
import java.util.stream.Collectors; import java.util.stream.Collectors;
@@ -10,38 +13,84 @@ import com.dfsek.terra.api.block.BlockType;
import com.dfsek.terra.api.block.state.BlockState; import com.dfsek.terra.api.block.state.BlockState;
import com.dfsek.terra.api.block.state.properties.Property; import com.dfsek.terra.api.block.state.properties.Property;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class MinestomBlockState implements BlockState {
public record MinestomBlockState(Block block) implements BlockState {
private static final Logger LOGGER = LoggerFactory.getLogger(MinestomBlockState.class);
public static final MinestomBlockState AIR = new MinestomBlockState(Block.AIR); public static final MinestomBlockState AIR = new MinestomBlockState(Block.AIR);
private final Block block; private static final TagStringIO tagStringIO = TagStringIO.tagStringIO();
public MinestomBlockState(Block block) { public MinestomBlockState {
if(block == null) { block = Objects.requireNonNullElse(block, Block.AIR);
this.block = Block.AIR;
} else {
this.block = block;
}
} }
public MinestomBlockState(String data) { public static MinestomBlockState fromStateId(String data) {
if(!data.contains("[")) { CompoundBinaryTag nbt = CompoundBinaryTag.empty();
block = Block.fromKey(data); int splitIndex = data.indexOf('{');
return; if(splitIndex != -1) {
String fullId = data;
data = data.substring(0, splitIndex);
String dataString = fullId.substring(splitIndex);
try {
nbt = tagStringIO.asCompound(dataString);
} catch(IOException exception) {
LOGGER.warn("Invalid entity data, will be ignored: {}", dataString);
}
} }
String[] split = data.split("\\["); int openBracketIndex = data.indexOf('[');
String namespaceId = split[0]; int closeBracketIndex = data.indexOf(']');
String properties = split[1].substring(0, split[1].length() - 1);
if(openBracketIndex == -1 || closeBracketIndex == -1 || closeBracketIndex < openBracketIndex) {
// no or invalid properties
Block block = Block.fromKey(data);
if(block != null && !nbt.isEmpty()) {
block = block.withNbt(nbt);
}
return new MinestomBlockState(block);
}
String namespaceId = data.substring(0, openBracketIndex);
String propertiesContent = data.substring(openBracketIndex + 1, closeBracketIndex);
Block block = Block.fromKey(namespaceId); Block block = Block.fromKey(namespaceId);
HashMap<String, String> propertiesMap = new HashMap<>(); if (block == null) {
LOGGER.error("Invalid block ID found during parsing: {}", namespaceId);
for(String property : properties.split(",")) { return new MinestomBlockState(Block.AIR);
String[] kv = property.split("=");
propertiesMap.put(kv[0].strip(), kv[1].strip());
} }
assert block != null; HashMap<String, String> propertiesMap = new HashMap<>();
this.block = block.withProperties(propertiesMap); int current = 0;
while (current < propertiesContent.length()) {
int nextComma = propertiesContent.indexOf(',', current);
String property;
if (nextComma == -1) {
property = propertiesContent.substring(current);
current = propertiesContent.length();
} else {
property = propertiesContent.substring(current, nextComma);
current = nextComma + 1;
}
int equalsIndex = property.indexOf('=');
if (equalsIndex == -1) {
LOGGER.warn("Invalid block property syntax (missing '=') in string: {}", property);
continue;
}
String key = property.substring(0, equalsIndex).strip();
String value = property.substring(equalsIndex + 1).strip();
propertiesMap.put(key, value);
}
if(!nbt.isEmpty()) {
block = block.withNbt(nbt);
}
return new MinestomBlockState(block.withProperties(propertiesMap));
} }
@Override @Override

View File

@@ -22,7 +22,8 @@ public class MinestomEntity implements com.dfsek.terra.api.entity.Entity {
public static MinestomEntity spawn(double x, double y, double z, EntityType type, TerraMinestomWorld world) { public static MinestomEntity spawn(double x, double y, double z, EntityType type, TerraMinestomWorld world) {
Instance instance = world.getHandle(); Instance instance = world.getHandle();
Entity entity = world.getEntityFactory().createEntity(((MinestomEntityType) type).getHandle()); MinestomEntityType entityType = (MinestomEntityType) type;
Entity entity = world.getEntityFactory().createEntity(entityType.getHandle(), entityType.getData());
entity.setInstance(instance, new Pos(x, y, z)); entity.setInstance(instance, new Pos(x, y, z));
return new MinestomEntity(entity, world); return new MinestomEntity(entity, world);
} }

View File

@@ -1,12 +1,38 @@
package com.dfsek.terra.minestom.entity; package com.dfsek.terra.minestom.entity;
import net.kyori.adventure.nbt.CompoundBinaryTag;
import net.kyori.adventure.nbt.TagStringIO;
import net.minestom.server.entity.EntityType; import net.minestom.server.entity.EntityType;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.IOException;
public class MinestomEntityType implements com.dfsek.terra.api.entity.EntityType { public class MinestomEntityType implements com.dfsek.terra.api.entity.EntityType {
private static final Logger LOGGER = LoggerFactory.getLogger(MinestomEntityType.class);
private static final TagStringIO tagStringIO = TagStringIO.tagStringIO();
private final EntityType delegate; private final EntityType delegate;
private final CompoundBinaryTag data;
public MinestomEntityType(String id) { public MinestomEntityType(String id) {
int splitIndex = id.indexOf('{');
if(splitIndex != -1) {
String fullId = id;
id = id.substring(0, splitIndex);
String dataString = fullId.substring(splitIndex);
CompoundBinaryTag data;
try {
data = tagStringIO.asCompound(dataString);
} catch(IOException exception) {
LOGGER.warn("Invalid entity data, will be ignored: {}", dataString);
data = CompoundBinaryTag.empty();
}
this.data = data;
} else {
this.data = CompoundBinaryTag.empty();
}
delegate = EntityType.fromKey(id); delegate = EntityType.fromKey(id);
} }
@@ -14,4 +40,8 @@ public class MinestomEntityType implements com.dfsek.terra.api.entity.EntityType
public EntityType getHandle() { public EntityType getHandle() {
return delegate; return delegate;
} }
public CompoundBinaryTag getData() {
return data;
}
} }

View File

@@ -15,7 +15,7 @@ public class MinestomWorldHandle implements WorldHandle {
@Override @Override
public @NotNull BlockState createBlockState(@NotNull String data) { public @NotNull BlockState createBlockState(@NotNull String data) {
return new MinestomBlockState(data); return MinestomBlockState.fromStateId(data);
} }
@Override @Override