Code/net/minecraft/world/level/block/entity/BannerBlockEntity.java

112 lines
4.2 KiB
Java

package net.minecraft.world.level.block.entity;
import javax.annotation.Nullable;
import net.minecraft.core.BlockPos;
import net.minecraft.core.HolderLookup;
import net.minecraft.core.component.DataComponentGetter;
import net.minecraft.core.component.DataComponentMap;
import net.minecraft.core.component.DataComponents;
import net.minecraft.nbt.CompoundTag;
import net.minecraft.nbt.NbtOps;
import net.minecraft.nbt.Tag;
import net.minecraft.network.chat.Component;
import net.minecraft.network.chat.ComponentSerialization;
import net.minecraft.network.protocol.game.ClientboundBlockEntityDataPacket;
import net.minecraft.resources.RegistryOps;
import net.minecraft.world.Nameable;
import net.minecraft.world.item.DyeColor;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.level.block.AbstractBannerBlock;
import net.minecraft.world.level.block.BannerBlock;
import net.minecraft.world.level.block.state.BlockState;
public class BannerBlockEntity extends BlockEntity implements Nameable {
public static final int MAX_PATTERNS = 6;
private static final String TAG_PATTERNS = "patterns";
@Nullable
private Component name;
private final DyeColor baseColor;
private BannerPatternLayers patterns = BannerPatternLayers.EMPTY;
public BannerBlockEntity(BlockPos p_155035_, BlockState p_155036_) {
this(p_155035_, p_155036_, ((AbstractBannerBlock)p_155036_.getBlock()).getColor());
}
public BannerBlockEntity(BlockPos p_155038_, BlockState p_155039_, DyeColor p_155040_) {
super(BlockEntityType.BANNER, p_155038_, p_155039_);
this.baseColor = p_155040_;
}
@Override
public Component getName() {
return (Component)(this.name != null ? this.name : Component.translatable("block.minecraft.banner"));
}
@Nullable
@Override
public Component getCustomName() {
return this.name;
}
@Override
protected void saveAdditional(CompoundTag p_187456_, HolderLookup.Provider p_329292_) {
super.saveAdditional(p_187456_, p_329292_);
RegistryOps<Tag> registryops = p_329292_.createSerializationContext(NbtOps.INSTANCE);
if (!this.patterns.equals(BannerPatternLayers.EMPTY)) {
p_187456_.store("patterns", BannerPatternLayers.CODEC, registryops, this.patterns);
}
p_187456_.storeNullable("CustomName", ComponentSerialization.CODEC, registryops, this.name);
}
@Override
protected void loadAdditional(CompoundTag p_334165_, HolderLookup.Provider p_330621_) {
super.loadAdditional(p_334165_, p_330621_);
this.name = parseCustomNameSafe(p_334165_.get("CustomName"), p_330621_);
RegistryOps<Tag> registryops = p_330621_.createSerializationContext(NbtOps.INSTANCE);
this.patterns = p_334165_.read("patterns", BannerPatternLayers.CODEC, registryops).orElse(BannerPatternLayers.EMPTY);
}
public ClientboundBlockEntityDataPacket getUpdatePacket() {
return ClientboundBlockEntityDataPacket.create(this);
}
@Override
public CompoundTag getUpdateTag(HolderLookup.Provider p_335241_) {
return this.saveWithoutMetadata(p_335241_);
}
public BannerPatternLayers getPatterns() {
return this.patterns;
}
public ItemStack getItem() {
ItemStack itemstack = new ItemStack(BannerBlock.byColor(this.baseColor));
itemstack.applyComponents(this.collectComponents());
return itemstack;
}
public DyeColor getBaseColor() {
return this.baseColor;
}
@Override
protected void applyImplicitComponents(DataComponentGetter p_396293_) {
super.applyImplicitComponents(p_396293_);
this.patterns = p_396293_.getOrDefault(DataComponents.BANNER_PATTERNS, BannerPatternLayers.EMPTY);
this.name = p_396293_.get(DataComponents.CUSTOM_NAME);
}
@Override
protected void collectImplicitComponents(DataComponentMap.Builder p_332512_) {
super.collectImplicitComponents(p_332512_);
p_332512_.set(DataComponents.BANNER_PATTERNS, this.patterns);
p_332512_.set(DataComponents.CUSTOM_NAME, this.name);
}
@Override
public void removeComponentsFromTag(CompoundTag p_336055_) {
p_336055_.remove("patterns");
p_336055_.remove("CustomName");
}
}