254 lines
8.7 KiB
Java
254 lines
8.7 KiB
Java
package net.minecraft.world.level.block.entity;
|
|
|
|
import com.mojang.logging.LogUtils;
|
|
import it.unimi.dsi.fastutil.objects.ObjectArrayList;
|
|
import java.util.Objects;
|
|
import net.minecraft.advancements.CriteriaTriggers;
|
|
import net.minecraft.core.BlockPos;
|
|
import net.minecraft.core.Direction;
|
|
import net.minecraft.core.HolderLookup;
|
|
import net.minecraft.core.registries.Registries;
|
|
import net.minecraft.nbt.CompoundTag;
|
|
import net.minecraft.network.protocol.game.ClientboundBlockEntityDataPacket;
|
|
import net.minecraft.resources.ResourceKey;
|
|
import net.minecraft.resources.ResourceLocation;
|
|
import net.minecraft.server.level.ServerLevel;
|
|
import net.minecraft.server.level.ServerPlayer;
|
|
import net.minecraft.world.entity.EntityType;
|
|
import net.minecraft.world.entity.item.ItemEntity;
|
|
import net.minecraft.world.entity.player.Player;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.minecraft.world.level.block.Block;
|
|
import net.minecraft.world.level.block.Blocks;
|
|
import net.minecraft.world.level.block.BrushableBlock;
|
|
import net.minecraft.world.level.block.state.BlockState;
|
|
import net.minecraft.world.level.block.state.properties.BlockStateProperties;
|
|
import net.minecraft.world.level.storage.loot.LootParams;
|
|
import net.minecraft.world.level.storage.loot.LootTable;
|
|
import net.minecraft.world.level.storage.loot.parameters.LootContextParamSets;
|
|
import net.minecraft.world.level.storage.loot.parameters.LootContextParams;
|
|
import net.minecraft.world.phys.Vec3;
|
|
import org.jetbrains.annotations.Nullable;
|
|
import org.slf4j.Logger;
|
|
|
|
public class BrushableBlockEntity extends BlockEntity {
|
|
private static final Logger LOGGER = LogUtils.getLogger();
|
|
private static final String LOOT_TABLE_TAG = "LootTable";
|
|
private static final String LOOT_TABLE_SEED_TAG = "LootTableSeed";
|
|
private static final String HIT_DIRECTION_TAG = "hit_direction";
|
|
private static final String ITEM_TAG = "item";
|
|
private static final int BRUSH_COOLDOWN_TICKS = 10;
|
|
private static final int BRUSH_RESET_TICKS = 40;
|
|
private static final int REQUIRED_BRUSHES_TO_BREAK = 10;
|
|
private int brushCount;
|
|
private long brushCountResetsAtTick;
|
|
private long coolDownEndsAtTick;
|
|
private ItemStack item = ItemStack.EMPTY;
|
|
@Nullable
|
|
private Direction hitDirection;
|
|
@Nullable
|
|
private ResourceKey<LootTable> lootTable;
|
|
private long lootTableSeed;
|
|
|
|
public BrushableBlockEntity(BlockPos pos, BlockState blockState) {
|
|
super(BlockEntityType.BRUSHABLE_BLOCK, pos, blockState);
|
|
}
|
|
|
|
public boolean brush(long l, ServerLevel serverLevel, Player player, Direction direction, ItemStack itemStack) {
|
|
if (this.hitDirection == null) {
|
|
this.hitDirection = direction;
|
|
}
|
|
|
|
this.brushCountResetsAtTick = l + 40L;
|
|
if (l < this.coolDownEndsAtTick) {
|
|
return false;
|
|
} else {
|
|
this.coolDownEndsAtTick = l + 10L;
|
|
this.unpackLootTable(serverLevel, player, itemStack);
|
|
int i = this.getCompletionState();
|
|
if (++this.brushCount >= 10) {
|
|
this.brushingCompleted(serverLevel, player, itemStack);
|
|
return true;
|
|
} else {
|
|
serverLevel.scheduleTick(this.getBlockPos(), this.getBlockState().getBlock(), 2);
|
|
int j = this.getCompletionState();
|
|
if (i != j) {
|
|
BlockState blockState = this.getBlockState();
|
|
BlockState blockState2 = blockState.setValue(BlockStateProperties.DUSTED, j);
|
|
serverLevel.setBlock(this.getBlockPos(), blockState2, 3);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
private void unpackLootTable(ServerLevel serverLevel, Player player, ItemStack itemStack) {
|
|
if (this.lootTable != null) {
|
|
LootTable lootTable = serverLevel.getServer().reloadableRegistries().getLootTable(this.lootTable);
|
|
if (player instanceof ServerPlayer serverPlayer) {
|
|
CriteriaTriggers.GENERATE_LOOT.trigger(serverPlayer, this.lootTable);
|
|
}
|
|
|
|
LootParams lootParams = new LootParams.Builder(serverLevel)
|
|
.withParameter(LootContextParams.ORIGIN, Vec3.atCenterOf(this.worldPosition))
|
|
.withLuck(player.getLuck())
|
|
.withParameter(LootContextParams.THIS_ENTITY, player)
|
|
.withParameter(LootContextParams.TOOL, itemStack)
|
|
.create(LootContextParamSets.ARCHAEOLOGY);
|
|
ObjectArrayList<ItemStack> objectArrayList = lootTable.getRandomItems(lootParams, this.lootTableSeed);
|
|
|
|
this.item = switch (objectArrayList.size()) {
|
|
case 0 -> ItemStack.EMPTY;
|
|
case 1 -> (ItemStack)objectArrayList.getFirst();
|
|
default -> {
|
|
LOGGER.warn("Expected max 1 loot from loot table {}, but got {}", this.lootTable.location(), objectArrayList.size());
|
|
yield (ItemStack)objectArrayList.getFirst();
|
|
}
|
|
};
|
|
this.lootTable = null;
|
|
this.setChanged();
|
|
}
|
|
}
|
|
|
|
private void brushingCompleted(ServerLevel serverLevel, Player player, ItemStack itemStack) {
|
|
this.dropContent(serverLevel, player, itemStack);
|
|
BlockState blockState = this.getBlockState();
|
|
serverLevel.levelEvent(3008, this.getBlockPos(), Block.getId(blockState));
|
|
Block block2;
|
|
if (this.getBlockState().getBlock() instanceof BrushableBlock brushableBlock) {
|
|
block2 = brushableBlock.getTurnsInto();
|
|
} else {
|
|
block2 = Blocks.AIR;
|
|
}
|
|
|
|
serverLevel.setBlock(this.worldPosition, block2.defaultBlockState(), 3);
|
|
}
|
|
|
|
private void dropContent(ServerLevel serverLevel, Player player, ItemStack itemStack) {
|
|
this.unpackLootTable(serverLevel, player, itemStack);
|
|
if (!this.item.isEmpty()) {
|
|
double d = EntityType.ITEM.getWidth();
|
|
double e = 1.0 - d;
|
|
double f = d / 2.0;
|
|
Direction direction = (Direction)Objects.requireNonNullElse(this.hitDirection, Direction.UP);
|
|
BlockPos blockPos = this.worldPosition.relative(direction, 1);
|
|
double g = blockPos.getX() + 0.5 * e + f;
|
|
double h = blockPos.getY() + 0.5 + EntityType.ITEM.getHeight() / 2.0F;
|
|
double i = blockPos.getZ() + 0.5 * e + f;
|
|
ItemEntity itemEntity = new ItemEntity(serverLevel, g, h, i, this.item.split(serverLevel.random.nextInt(21) + 10));
|
|
itemEntity.setDeltaMovement(Vec3.ZERO);
|
|
serverLevel.addFreshEntity(itemEntity);
|
|
this.item = ItemStack.EMPTY;
|
|
}
|
|
}
|
|
|
|
public void checkReset(ServerLevel serverLevel) {
|
|
if (this.brushCount != 0 && serverLevel.getGameTime() >= this.brushCountResetsAtTick) {
|
|
int i = this.getCompletionState();
|
|
this.brushCount = Math.max(0, this.brushCount - 2);
|
|
int j = this.getCompletionState();
|
|
if (i != j) {
|
|
serverLevel.setBlock(this.getBlockPos(), this.getBlockState().setValue(BlockStateProperties.DUSTED, j), 3);
|
|
}
|
|
|
|
int k = 4;
|
|
this.brushCountResetsAtTick = serverLevel.getGameTime() + 4L;
|
|
}
|
|
|
|
if (this.brushCount == 0) {
|
|
this.hitDirection = null;
|
|
this.brushCountResetsAtTick = 0L;
|
|
this.coolDownEndsAtTick = 0L;
|
|
} else {
|
|
serverLevel.scheduleTick(this.getBlockPos(), this.getBlockState().getBlock(), 2);
|
|
}
|
|
}
|
|
|
|
private boolean tryLoadLootTable(CompoundTag tag) {
|
|
if (tag.contains("LootTable", 8)) {
|
|
this.lootTable = ResourceKey.create(Registries.LOOT_TABLE, ResourceLocation.parse(tag.getString("LootTable")));
|
|
this.lootTableSeed = tag.getLong("LootTableSeed");
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
private boolean trySaveLootTable(CompoundTag tag) {
|
|
if (this.lootTable == null) {
|
|
return false;
|
|
} else {
|
|
tag.putString("LootTable", this.lootTable.location().toString());
|
|
if (this.lootTableSeed != 0L) {
|
|
tag.putLong("LootTableSeed", this.lootTableSeed);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public CompoundTag getUpdateTag(HolderLookup.Provider registries) {
|
|
CompoundTag compoundTag = super.getUpdateTag(registries);
|
|
if (this.hitDirection != null) {
|
|
compoundTag.putInt("hit_direction", this.hitDirection.ordinal());
|
|
}
|
|
|
|
if (!this.item.isEmpty()) {
|
|
compoundTag.put("item", this.item.save(registries));
|
|
}
|
|
|
|
return compoundTag;
|
|
}
|
|
|
|
public ClientboundBlockEntityDataPacket getUpdatePacket() {
|
|
return ClientboundBlockEntityDataPacket.create(this);
|
|
}
|
|
|
|
@Override
|
|
protected void loadAdditional(CompoundTag tag, HolderLookup.Provider registries) {
|
|
super.loadAdditional(tag, registries);
|
|
if (!this.tryLoadLootTable(tag) && tag.contains("item")) {
|
|
this.item = (ItemStack)ItemStack.parse(registries, tag.getCompound("item")).orElse(ItemStack.EMPTY);
|
|
} else {
|
|
this.item = ItemStack.EMPTY;
|
|
}
|
|
|
|
if (tag.contains("hit_direction")) {
|
|
this.hitDirection = Direction.values()[tag.getInt("hit_direction")];
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected void saveAdditional(CompoundTag tag, HolderLookup.Provider registries) {
|
|
super.saveAdditional(tag, registries);
|
|
if (!this.trySaveLootTable(tag) && !this.item.isEmpty()) {
|
|
tag.put("item", this.item.save(registries));
|
|
}
|
|
}
|
|
|
|
public void setLootTable(ResourceKey<LootTable> lootTable, long seed) {
|
|
this.lootTable = lootTable;
|
|
this.lootTableSeed = seed;
|
|
}
|
|
|
|
private int getCompletionState() {
|
|
if (this.brushCount == 0) {
|
|
return 0;
|
|
} else if (this.brushCount < 3) {
|
|
return 1;
|
|
} else {
|
|
return this.brushCount < 6 ? 2 : 3;
|
|
}
|
|
}
|
|
|
|
@Nullable
|
|
public Direction getHitDirection() {
|
|
return this.hitDirection;
|
|
}
|
|
|
|
public ItemStack getItem() {
|
|
return this.item;
|
|
}
|
|
}
|