442 lines
16 KiB
Java
442 lines
16 KiB
Java
package carpet.script.value;
|
|
|
|
import carpet.fakes.BlockPredicateInterface;
|
|
import carpet.script.exception.InternalExpressionException;
|
|
import carpet.script.exception.ThrowStatement;
|
|
import carpet.script.exception.Throwables;
|
|
import carpet.utils.BlockInfo;
|
|
import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
|
|
import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;
|
|
import net.minecraft.advancements.critereon.MinMaxBounds;
|
|
import net.minecraft.core.BlockPos;
|
|
import net.minecraft.core.GlobalPos;
|
|
import net.minecraft.core.Registry;
|
|
import net.minecraft.core.RegistryAccess;
|
|
import net.minecraft.core.Vec3i;
|
|
import net.minecraft.core.particles.ParticleOptions;
|
|
import net.minecraft.resources.ResourceKey;
|
|
import net.minecraft.resources.ResourceLocation;
|
|
import net.minecraft.server.MinecraftServer;
|
|
import net.minecraft.server.level.ColumnPos;
|
|
import net.minecraft.server.level.ServerLevel;
|
|
import net.minecraft.tags.TagKey;
|
|
import net.minecraft.util.StringRepresentable;
|
|
import net.minecraft.world.damagesource.DamageSource;
|
|
import net.minecraft.world.entity.Entity;
|
|
import net.minecraft.world.entity.ai.behavior.PositionTracker;
|
|
import net.minecraft.world.entity.ai.memory.NearestVisibleLivingEntities;
|
|
import net.minecraft.world.entity.ai.memory.WalkTarget;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.minecraft.world.level.Level;
|
|
import net.minecraft.world.level.block.state.BlockState;
|
|
import net.minecraft.world.level.block.state.pattern.BlockInWorld;
|
|
import net.minecraft.world.level.block.state.properties.Property;
|
|
import net.minecraft.world.level.levelgen.structure.BoundingBox;
|
|
import net.minecraft.world.level.levelgen.structure.StructurePiece;
|
|
import net.minecraft.world.level.levelgen.structure.StructureStart;
|
|
import net.minecraft.world.level.material.MaterialColor;
|
|
import net.minecraft.world.level.pathfinder.Node;
|
|
import net.minecraft.world.level.pathfinder.Path;
|
|
import net.minecraft.world.phys.AABB;
|
|
import net.minecraft.world.phys.Vec3;
|
|
import net.minecraft.world.scores.Objective;
|
|
import net.minecraft.world.scores.criteria.ObjectiveCriteria;
|
|
import java.util.ArrayList;
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
import java.util.Locale;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
import java.util.UUID;
|
|
import java.util.function.Predicate;
|
|
import java.util.stream.Collectors;
|
|
import java.util.stream.StreamSupport;
|
|
|
|
public class ValueConversions
|
|
{
|
|
public static Value of(BlockPos pos)
|
|
{
|
|
return ListValue.of(new NumericValue(pos.getX()), new NumericValue(pos.getY()), new NumericValue(pos.getZ()));
|
|
}
|
|
|
|
public static Value of(Vec3 vec)
|
|
{
|
|
return ListValue.of(new NumericValue(vec.x), new NumericValue(vec.y), new NumericValue(vec.z));
|
|
}
|
|
|
|
public static Value of(ColumnPos cpos) { return ListValue.of(new NumericValue(cpos.x()), new NumericValue(cpos.z()));}
|
|
|
|
public static Value of(ServerLevel world)
|
|
{
|
|
return of(world.dimension().location());
|
|
}
|
|
|
|
public static Value of(MaterialColor color) {return ListValue.of(StringValue.of(BlockInfo.mapColourName.get(color)), ofRGB(color.col));}
|
|
|
|
public static <T extends Number> Value of(MinMaxBounds<T> range) { return ListValue.of(NumericValue.of(range.getMin()), NumericValue.of(range.getMax()));}
|
|
|
|
public static Value of(ItemStack stack)
|
|
{
|
|
if (stack == null || stack.isEmpty())
|
|
return Value.NULL;
|
|
return ListValue.of(
|
|
of(Registry.ITEM.getKey(stack.getItem())),
|
|
new NumericValue(stack.getCount()),
|
|
NBTSerializableValue.fromStack(stack)
|
|
);
|
|
}
|
|
|
|
public static Value of(Objective objective)
|
|
{
|
|
return ListValue.of(
|
|
StringValue.of(objective.getName()),
|
|
StringValue.of(objective.getCriteria().getName())
|
|
);
|
|
}
|
|
|
|
|
|
public static Value of(ObjectiveCriteria criteria)
|
|
{
|
|
return ListValue.of(
|
|
StringValue.of(criteria.getName()),
|
|
BooleanValue.of(criteria.isReadOnly())
|
|
);
|
|
}
|
|
|
|
|
|
public static Value of(ParticleOptions particle)
|
|
{
|
|
String repr = particle.writeToString();
|
|
if (repr.startsWith("minecraft:")) return StringValue.of(repr.substring(10));
|
|
return StringValue.of(repr);
|
|
}
|
|
|
|
public static Value ofRGB(int value) {return new NumericValue(value*256+255 );}
|
|
|
|
public static Level dimFromValue(Value dimensionValue, MinecraftServer server)
|
|
{
|
|
if (dimensionValue instanceof EntityValue)
|
|
{
|
|
return ((EntityValue)dimensionValue).getEntity().getCommandSenderWorld();
|
|
}
|
|
else if (dimensionValue instanceof BlockValue bv)
|
|
{
|
|
if (bv.getWorld() != null)
|
|
{
|
|
return bv.getWorld();
|
|
}
|
|
else
|
|
{
|
|
throw new InternalExpressionException("dimension argument accepts only world-localized block arguments");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
String dimString = dimensionValue.getString().toLowerCase(Locale.ROOT);
|
|
switch (dimString)
|
|
{
|
|
case "nether":
|
|
case "the_nether":
|
|
return server.getLevel(Level.NETHER);
|
|
case "end":
|
|
case "the_end":
|
|
return server.getLevel(Level.END);
|
|
case "overworld":
|
|
case "over_world":
|
|
return server.getLevel(Level.OVERWORLD);
|
|
default:
|
|
ResourceKey<Level> dim = null;
|
|
ResourceLocation id = new ResourceLocation(dimString);
|
|
// not using RegistryKey.of since that one creates on check
|
|
for (ResourceKey<Level> world : (server.levelKeys()))
|
|
{
|
|
if (id.equals(world.location()))
|
|
{
|
|
dim = world;
|
|
break;
|
|
}
|
|
}
|
|
if (dim == null)
|
|
throw new ThrowStatement(dimString, Throwables.UNKNOWN_DIMENSION);
|
|
return server.getLevel(dim);
|
|
}
|
|
}
|
|
}
|
|
|
|
public static Value of(ResourceKey<?> dim)
|
|
{
|
|
return of(dim.location());
|
|
}
|
|
|
|
public static Value of(TagKey<?> tagKey) { return of(tagKey.location()); }
|
|
|
|
public static Value of(ResourceLocation id)
|
|
{
|
|
if (id == null) // should be Value.NULL
|
|
return Value.NULL;
|
|
if (id.getNamespace().equals("minecraft"))
|
|
return new StringValue(id.getPath());
|
|
return new StringValue(id.toString());
|
|
}
|
|
|
|
public static String simplify(ResourceLocation id)
|
|
{
|
|
if (id == null) // should be Value.NULL
|
|
return "";
|
|
if (id.getNamespace().equals("minecraft"))
|
|
return id.getPath();
|
|
return id.toString();
|
|
}
|
|
|
|
public static Value of(GlobalPos pos)
|
|
{
|
|
return ListValue.of(
|
|
ValueConversions.of(pos.dimension()),
|
|
ValueConversions.of(pos.pos())
|
|
);
|
|
}
|
|
|
|
public static Value fromPath(ServerLevel world, Path path)
|
|
{
|
|
List<Value> nodes = new ArrayList<>();
|
|
//for (PathNode node: path.getNodes())
|
|
for (int i = 0, len = path.getNodeCount(); i < len; i++)
|
|
{
|
|
Node node = path.getNode(i);
|
|
nodes.add( ListValue.of(
|
|
new BlockValue(null, world, node.asBlockPos()),
|
|
new StringValue(node.type.name().toLowerCase(Locale.ROOT)),
|
|
new NumericValue(node.costMalus),
|
|
BooleanValue.of(node.closed)
|
|
));
|
|
}
|
|
return ListValue.wrap(nodes);
|
|
}
|
|
|
|
public static Value fromTimedMemory(Entity e, long expiry, Object v)
|
|
{
|
|
Value ret = fromEntityMemory(e, v);
|
|
if (ret.isNull() || expiry == Long.MAX_VALUE) return ret;
|
|
return ListValue.of(ret, new NumericValue(expiry));
|
|
}
|
|
|
|
private static Value fromEntityMemory(Entity e, Object v)
|
|
{
|
|
if (v instanceof GlobalPos pos)
|
|
{
|
|
return of(pos);
|
|
}
|
|
if (v instanceof Entity)
|
|
{
|
|
return new EntityValue((Entity)v);
|
|
}
|
|
if (v instanceof BlockPos)
|
|
{
|
|
return new BlockValue(null, (ServerLevel) e.getCommandSenderWorld(), (BlockPos) v);
|
|
}
|
|
if (v instanceof Number)
|
|
{
|
|
return new NumericValue(((Number) v).doubleValue());
|
|
}
|
|
if (v instanceof Boolean)
|
|
{
|
|
return BooleanValue.of((Boolean) v);
|
|
}
|
|
if (v instanceof UUID)
|
|
{
|
|
return ofUUID( (ServerLevel) e.getCommandSenderWorld(), (UUID)v);
|
|
}
|
|
if (v instanceof DamageSource source)
|
|
{
|
|
return ListValue.of(
|
|
new StringValue(source.getMsgId()),
|
|
source.getEntity()==null?Value.NULL:new EntityValue(source.getEntity())
|
|
);
|
|
}
|
|
if (v instanceof Path)
|
|
{
|
|
return fromPath((ServerLevel)e.getCommandSenderWorld(), (Path)v);
|
|
}
|
|
if (v instanceof PositionTracker)
|
|
{
|
|
return new BlockValue(null, (ServerLevel) e.getCommandSenderWorld(), ((PositionTracker)v).currentBlockPosition());
|
|
}
|
|
if (v instanceof WalkTarget)
|
|
{
|
|
return ListValue.of(
|
|
new BlockValue(null, (ServerLevel) e.getCommandSenderWorld(), ((WalkTarget)v).getTarget().currentBlockPosition()),
|
|
new NumericValue(((WalkTarget) v).getSpeedModifier()),
|
|
new NumericValue(((WalkTarget) v).getCloseEnoughDist())
|
|
);
|
|
}
|
|
if (v instanceof NearestVisibleLivingEntities nvle) {
|
|
v = StreamSupport.stream(nvle.findAll(entity -> true).spliterator(), false).toList();
|
|
}
|
|
if (v instanceof Set)
|
|
{
|
|
v = new ArrayList(((Set) v));
|
|
}
|
|
if (v instanceof List<?> l)
|
|
{
|
|
if (l.isEmpty()) return ListValue.of();
|
|
Object el = l.get(0);
|
|
if (el instanceof Entity)
|
|
{
|
|
return ListValue.wrap(l.stream().map(o -> new EntityValue((Entity)o)).collect(Collectors.toList()));
|
|
}
|
|
if (el instanceof GlobalPos)
|
|
{
|
|
return ListValue.wrap(l.stream().map(o -> of((GlobalPos) o)).collect(Collectors.toList()));
|
|
}
|
|
}
|
|
return Value.NULL;
|
|
}
|
|
|
|
private static Value ofUUID(ServerLevel entityWorld, UUID uuid)
|
|
{
|
|
Entity current = entityWorld.getEntity(uuid);
|
|
return ListValue.of(
|
|
current == null?Value.NULL:new EntityValue(current),
|
|
new StringValue(uuid.toString())
|
|
);
|
|
}
|
|
public static Value of(AABB box)
|
|
{
|
|
return ListValue.of(
|
|
ListValue.fromTriple(box.minX, box.minY, box.minZ),
|
|
ListValue.fromTriple(box.maxX, box.maxY, box.maxZ)
|
|
);
|
|
}
|
|
public static Value of(BoundingBox box)
|
|
{
|
|
return ListValue.of(
|
|
ListValue.fromTriple(box.minX(), box.minY(), box.minZ()),
|
|
ListValue.fromTriple(box.maxX(), box.maxY(), box.maxZ())
|
|
);
|
|
}
|
|
|
|
public static Value of(StructureStart structure)
|
|
{
|
|
if (structure == null || structure == StructureStart.INVALID_START) return Value.NULL;
|
|
BoundingBox boundingBox = structure.getBoundingBox();
|
|
if (boundingBox.maxX() < boundingBox.minX() || boundingBox.maxY() < boundingBox.minY() || boundingBox.maxZ() < boundingBox.minZ()) return Value.NULL;
|
|
Map<Value, Value> ret = new HashMap<>();
|
|
ret.put(new StringValue("box"), of(boundingBox));
|
|
List<Value> pieces = new ArrayList<>();
|
|
for (StructurePiece piece : structure.getPieces())
|
|
{
|
|
BoundingBox box = piece.getBoundingBox();
|
|
if (box.maxX() >= box.minX() && box.maxY() >= box.minY() && box.maxZ() >= box.minZ())
|
|
{
|
|
pieces.add(ListValue.of(
|
|
new StringValue(NBTSerializableValue.nameFromRegistryId(Registry.STRUCTURE_PIECE.getKey(piece.getType()))),
|
|
(piece.getOrientation() == null) ? Value.NULL : new StringValue(piece.getOrientation().getName()),
|
|
ListValue.fromTriple(box.minX(), box.minY(), box.minZ()),
|
|
ListValue.fromTriple(box.maxX(), box.maxY(), box.maxZ())
|
|
));
|
|
}
|
|
}
|
|
ret.put(new StringValue("pieces"), ListValue.wrap(pieces));
|
|
return MapValue.wrap(ret);
|
|
}
|
|
|
|
public static Value fromProperty(BlockState state, Property<?> p)
|
|
{
|
|
Comparable<?> object = state.getValue(p);
|
|
if (object instanceof Boolean || object instanceof Number) return StringValue.of(object.toString());
|
|
if (object instanceof StringRepresentable)
|
|
{
|
|
return StringValue.of(((StringRepresentable) object).getSerializedName());
|
|
}
|
|
throw new InternalExpressionException("Unknown property type: "+p.getName());
|
|
}
|
|
|
|
record SlotParam(/* Nullable */ String type, int id) {
|
|
public ListValue build() {
|
|
return ListValue.of(StringValue.of(type), new NumericValue(id));
|
|
}
|
|
}
|
|
|
|
private static final Int2ObjectMap<SlotParam> slotIdsToSlotParams = new Int2ObjectOpenHashMap<>() {{
|
|
int n;
|
|
//covers blocks, player hotbar and inventory, and all default inventories
|
|
for(n = 0; n < 54; ++n) {
|
|
put(n, new SlotParam(null, n));
|
|
}
|
|
for(n = 0; n < 27; ++n) {
|
|
put(200 + n, new SlotParam("enderchest", n));
|
|
}
|
|
|
|
// villager
|
|
for(n = 0; n < 8; ++n) {
|
|
put(300 + n, new SlotParam(null, n));
|
|
}
|
|
|
|
// horse, llamas, donkeys, etc.
|
|
// two first slots are for saddle and armour
|
|
for(n = 0; n < 15; ++n) {
|
|
put(500 + n, new SlotParam(null, n + 2));
|
|
}
|
|
// weapon main hand
|
|
put(98, new SlotParam("equipment", 0));
|
|
// offhand
|
|
put(99, new SlotParam("equipment", 5));
|
|
// feet, legs, chest, head
|
|
for(n = 0; n < 4; ++n) {
|
|
put(100 + n, new SlotParam("equipment", n + 1));
|
|
}
|
|
//horse defaults saddle
|
|
put(400, new SlotParam(null, 0));
|
|
// armor
|
|
put(401, new SlotParam(null, 1));
|
|
// chest itself on the donkey is wierd - use NBT to alter that.
|
|
//hashMap.put("horse.chest", 499);
|
|
}};
|
|
|
|
public static Value ofVanillaSlotResult(int itemSlot)
|
|
{
|
|
SlotParam ret = slotIdsToSlotParams.get(itemSlot);
|
|
if (ret == null) return ListValue.of(Value.NULL, new NumericValue(itemSlot));
|
|
return ret.build();
|
|
}
|
|
|
|
public static Value ofBlockPredicate(RegistryAccess registryAccess, Predicate<BlockInWorld> blockPredicate)
|
|
{
|
|
BlockPredicateInterface predicateData = (BlockPredicateInterface) blockPredicate;
|
|
return ListValue.of(
|
|
predicateData.getCMBlockState()==null?Value.NULL:of(Registry.BLOCK.getKey(predicateData.getCMBlockState().getBlock())),
|
|
predicateData.getCMBlockTagKey()==null?Value.NULL:of(registryAccess.registryOrThrow(Registry.BLOCK_REGISTRY).getTag(predicateData.getCMBlockTagKey()).get().key()),
|
|
MapValue.wrap(predicateData.getCMProperties()),
|
|
predicateData.getCMDataTag() == null?Value.NULL:new NBTSerializableValue(predicateData.getCMDataTag())
|
|
);
|
|
}
|
|
|
|
public static Value guess(ServerLevel serverWorld, Object o) {
|
|
if (o == null)
|
|
return Value.NULL;
|
|
if (o instanceof List)
|
|
return ListValue.wrap(((List<?>) o).stream().map(oo -> guess(serverWorld, oo)).collect(Collectors.toList()));
|
|
if (o instanceof BlockPos)
|
|
return new BlockValue(null, serverWorld, (BlockPos)o);
|
|
if (o instanceof Entity)
|
|
return EntityValue.of((Entity) o);
|
|
if (o instanceof Vec3)
|
|
return of((Vec3)o);
|
|
if (o instanceof Vec3i)
|
|
return of(new BlockPos((Vec3i)o));
|
|
if (o instanceof AABB)
|
|
return of((AABB)o);
|
|
if (o instanceof BoundingBox)
|
|
return of((BoundingBox) o);
|
|
if (o instanceof ItemStack)
|
|
return of((ItemStack)o);
|
|
if (o instanceof Boolean)
|
|
return BooleanValue.of((Boolean) o);
|
|
if (o instanceof Number)
|
|
return NumericValue.of((Number) o);
|
|
if (o instanceof ResourceLocation)
|
|
return of((ResourceLocation)o);
|
|
return StringValue.of(o.toString());
|
|
}
|
|
}
|