minecraft-src/net/minecraft/network/syncher/EntityDataSerializers.java
2025-07-04 01:41:11 +03:00

177 lines
8.7 KiB
Java

package net.minecraft.network.syncher;
import io.netty.buffer.ByteBuf;
import java.util.List;
import java.util.Optional;
import java.util.OptionalInt;
import java.util.UUID;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.core.GlobalPos;
import net.minecraft.core.Holder;
import net.minecraft.core.Rotations;
import net.minecraft.core.UUIDUtil;
import net.minecraft.core.particles.ParticleOptions;
import net.minecraft.core.particles.ParticleTypes;
import net.minecraft.nbt.CompoundTag;
import net.minecraft.network.RegistryFriendlyByteBuf;
import net.minecraft.network.VarInt;
import net.minecraft.network.chat.Component;
import net.minecraft.network.chat.ComponentSerialization;
import net.minecraft.network.codec.ByteBufCodecs;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.util.CrudeIncrementalIntIdentityHashBiMap;
import net.minecraft.world.entity.Pose;
import net.minecraft.world.entity.animal.CatVariant;
import net.minecraft.world.entity.animal.FrogVariant;
import net.minecraft.world.entity.animal.WolfVariant;
import net.minecraft.world.entity.animal.armadillo.Armadillo;
import net.minecraft.world.entity.animal.sniffer.Sniffer;
import net.minecraft.world.entity.decoration.PaintingVariant;
import net.minecraft.world.entity.npc.VillagerData;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.block.state.BlockState;
import org.jetbrains.annotations.Nullable;
import org.joml.Quaternionf;
import org.joml.Vector3f;
/**
* Registry for {@link EntityDataSerializer}.
*/
public class EntityDataSerializers {
private static final CrudeIncrementalIntIdentityHashBiMap<EntityDataSerializer<?>> SERIALIZERS = CrudeIncrementalIntIdentityHashBiMap.create(16);
public static final EntityDataSerializer<Byte> BYTE = EntityDataSerializer.forValueType(ByteBufCodecs.BYTE);
public static final EntityDataSerializer<Integer> INT = EntityDataSerializer.forValueType(ByteBufCodecs.VAR_INT);
public static final EntityDataSerializer<Long> LONG = EntityDataSerializer.forValueType(ByteBufCodecs.VAR_LONG);
public static final EntityDataSerializer<Float> FLOAT = EntityDataSerializer.forValueType(ByteBufCodecs.FLOAT);
public static final EntityDataSerializer<String> STRING = EntityDataSerializer.forValueType(ByteBufCodecs.STRING_UTF8);
public static final EntityDataSerializer<Component> COMPONENT = EntityDataSerializer.forValueType(ComponentSerialization.TRUSTED_STREAM_CODEC);
public static final EntityDataSerializer<Optional<Component>> OPTIONAL_COMPONENT = EntityDataSerializer.forValueType(
ComponentSerialization.TRUSTED_OPTIONAL_STREAM_CODEC
);
public static final EntityDataSerializer<ItemStack> ITEM_STACK = new EntityDataSerializer<ItemStack>() {
@Override
public StreamCodec<? super RegistryFriendlyByteBuf, ItemStack> codec() {
return ItemStack.OPTIONAL_STREAM_CODEC;
}
public ItemStack copy(ItemStack itemStack) {
return itemStack.copy();
}
};
public static final EntityDataSerializer<BlockState> BLOCK_STATE = EntityDataSerializer.forValueType(ByteBufCodecs.idMapper(Block.BLOCK_STATE_REGISTRY));
private static final StreamCodec<ByteBuf, Optional<BlockState>> OPTIONAL_BLOCK_STATE_CODEC = new StreamCodec<ByteBuf, Optional<BlockState>>() {
public void encode(ByteBuf byteBuf, Optional<BlockState> optional) {
if (optional.isPresent()) {
VarInt.write(byteBuf, Block.getId((BlockState)optional.get()));
} else {
VarInt.write(byteBuf, 0);
}
}
public Optional<BlockState> decode(ByteBuf byteBuf) {
int i = VarInt.read(byteBuf);
return i == 0 ? Optional.empty() : Optional.of(Block.stateById(i));
}
};
public static final EntityDataSerializer<Optional<BlockState>> OPTIONAL_BLOCK_STATE = EntityDataSerializer.forValueType(OPTIONAL_BLOCK_STATE_CODEC);
public static final EntityDataSerializer<Boolean> BOOLEAN = EntityDataSerializer.forValueType(ByteBufCodecs.BOOL);
public static final EntityDataSerializer<ParticleOptions> PARTICLE = EntityDataSerializer.forValueType(ParticleTypes.STREAM_CODEC);
public static final EntityDataSerializer<List<ParticleOptions>> PARTICLES = EntityDataSerializer.forValueType(
ParticleTypes.STREAM_CODEC.apply(ByteBufCodecs.list())
);
public static final EntityDataSerializer<Rotations> ROTATIONS = EntityDataSerializer.forValueType(Rotations.STREAM_CODEC);
public static final EntityDataSerializer<BlockPos> BLOCK_POS = EntityDataSerializer.forValueType(BlockPos.STREAM_CODEC);
public static final EntityDataSerializer<Optional<BlockPos>> OPTIONAL_BLOCK_POS = EntityDataSerializer.forValueType(
BlockPos.STREAM_CODEC.apply(ByteBufCodecs::optional)
);
public static final EntityDataSerializer<Direction> DIRECTION = EntityDataSerializer.forValueType(Direction.STREAM_CODEC);
public static final EntityDataSerializer<Optional<UUID>> OPTIONAL_UUID = EntityDataSerializer.forValueType(
UUIDUtil.STREAM_CODEC.apply(ByteBufCodecs::optional)
);
public static final EntityDataSerializer<Optional<GlobalPos>> OPTIONAL_GLOBAL_POS = EntityDataSerializer.forValueType(
GlobalPos.STREAM_CODEC.apply(ByteBufCodecs::optional)
);
public static final EntityDataSerializer<CompoundTag> COMPOUND_TAG = new EntityDataSerializer<CompoundTag>() {
@Override
public StreamCodec<? super RegistryFriendlyByteBuf, CompoundTag> codec() {
return ByteBufCodecs.TRUSTED_COMPOUND_TAG;
}
public CompoundTag copy(CompoundTag compoundTag) {
return compoundTag.copy();
}
};
public static final EntityDataSerializer<VillagerData> VILLAGER_DATA = EntityDataSerializer.forValueType(VillagerData.STREAM_CODEC);
private static final StreamCodec<ByteBuf, OptionalInt> OPTIONAL_UNSIGNED_INT_CODEC = new StreamCodec<ByteBuf, OptionalInt>() {
public OptionalInt decode(ByteBuf byteBuf) {
int i = VarInt.read(byteBuf);
return i == 0 ? OptionalInt.empty() : OptionalInt.of(i - 1);
}
public void encode(ByteBuf byteBuf, OptionalInt optionalInt) {
VarInt.write(byteBuf, optionalInt.orElse(-1) + 1);
}
};
public static final EntityDataSerializer<OptionalInt> OPTIONAL_UNSIGNED_INT = EntityDataSerializer.forValueType(OPTIONAL_UNSIGNED_INT_CODEC);
public static final EntityDataSerializer<Pose> POSE = EntityDataSerializer.forValueType(Pose.STREAM_CODEC);
public static final EntityDataSerializer<Holder<CatVariant>> CAT_VARIANT = EntityDataSerializer.forValueType(CatVariant.STREAM_CODEC);
public static final EntityDataSerializer<Holder<WolfVariant>> WOLF_VARIANT = EntityDataSerializer.forValueType(WolfVariant.STREAM_CODEC);
public static final EntityDataSerializer<Holder<FrogVariant>> FROG_VARIANT = EntityDataSerializer.forValueType(FrogVariant.STREAM_CODEC);
public static final EntityDataSerializer<Holder<PaintingVariant>> PAINTING_VARIANT = EntityDataSerializer.forValueType(PaintingVariant.STREAM_CODEC);
public static final EntityDataSerializer<Armadillo.ArmadilloState> ARMADILLO_STATE = EntityDataSerializer.forValueType(Armadillo.ArmadilloState.STREAM_CODEC);
public static final EntityDataSerializer<Sniffer.State> SNIFFER_STATE = EntityDataSerializer.forValueType(Sniffer.State.STREAM_CODEC);
public static final EntityDataSerializer<Vector3f> VECTOR3 = EntityDataSerializer.forValueType(ByteBufCodecs.VECTOR3F);
public static final EntityDataSerializer<Quaternionf> QUATERNION = EntityDataSerializer.forValueType(ByteBufCodecs.QUATERNIONF);
public static void registerSerializer(EntityDataSerializer<?> serializer) {
SERIALIZERS.add(serializer);
}
@Nullable
public static EntityDataSerializer<?> getSerializer(int id) {
return SERIALIZERS.byId(id);
}
public static int getSerializedId(EntityDataSerializer<?> serializer) {
return SERIALIZERS.getId(serializer);
}
private EntityDataSerializers() {
}
static {
registerSerializer(BYTE);
registerSerializer(INT);
registerSerializer(LONG);
registerSerializer(FLOAT);
registerSerializer(STRING);
registerSerializer(COMPONENT);
registerSerializer(OPTIONAL_COMPONENT);
registerSerializer(ITEM_STACK);
registerSerializer(BOOLEAN);
registerSerializer(ROTATIONS);
registerSerializer(BLOCK_POS);
registerSerializer(OPTIONAL_BLOCK_POS);
registerSerializer(DIRECTION);
registerSerializer(OPTIONAL_UUID);
registerSerializer(BLOCK_STATE);
registerSerializer(OPTIONAL_BLOCK_STATE);
registerSerializer(COMPOUND_TAG);
registerSerializer(PARTICLE);
registerSerializer(PARTICLES);
registerSerializer(VILLAGER_DATA);
registerSerializer(OPTIONAL_UNSIGNED_INT);
registerSerializer(POSE);
registerSerializer(CAT_VARIANT);
registerSerializer(WOLF_VARIANT);
registerSerializer(FROG_VARIANT);
registerSerializer(OPTIONAL_GLOBAL_POS);
registerSerializer(PAINTING_VARIANT);
registerSerializer(SNIFFER_STATE);
registerSerializer(ARMADILLO_STATE);
registerSerializer(VECTOR3);
registerSerializer(QUATERNION);
}
}