mirror of
https://github.com/Ellpeck/NaturesAura.git
synced 2024-05-17 22:18:44 +02:00
383 lines
18 KiB
Java
383 lines
18 KiB
Java
package de.ellpeck.naturesaura;
|
|
|
|
import com.mojang.blaze3d.vertex.VertexConsumer;
|
|
import de.ellpeck.naturesaura.api.NaturesAuraAPI;
|
|
import de.ellpeck.naturesaura.api.aura.container.IAuraContainer;
|
|
import de.ellpeck.naturesaura.api.aura.item.IAuraRecharge;
|
|
import de.ellpeck.naturesaura.api.misc.ILevelData;
|
|
import de.ellpeck.naturesaura.blocks.tiles.BlockEntityImpl;
|
|
import de.ellpeck.naturesaura.chunk.AuraChunk;
|
|
import de.ellpeck.naturesaura.compat.Compat;
|
|
import de.ellpeck.naturesaura.misc.LevelData;
|
|
import de.ellpeck.naturesaura.packet.PacketHandler;
|
|
import de.ellpeck.naturesaura.packet.PacketParticles;
|
|
import net.minecraft.client.Minecraft;
|
|
import net.minecraft.client.gui.GuiGraphics;
|
|
import net.minecraft.core.BlockPos;
|
|
import net.minecraft.core.Direction;
|
|
import net.minecraft.core.Registry;
|
|
import net.minecraft.core.registries.BuiltInRegistries;
|
|
import net.minecraft.resources.ResourceLocation;
|
|
import net.minecraft.server.level.ServerChunkCache;
|
|
import net.minecraft.server.level.ServerPlayer;
|
|
import net.minecraft.sounds.SoundEvents;
|
|
import net.minecraft.sounds.SoundSource;
|
|
import net.minecraft.world.InteractionHand;
|
|
import net.minecraft.world.InteractionResult;
|
|
import net.minecraft.world.entity.decoration.ItemFrame;
|
|
import net.minecraft.world.entity.item.ItemEntity;
|
|
import net.minecraft.world.entity.player.Player;
|
|
import net.minecraft.world.item.Item;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.minecraft.world.item.crafting.Ingredient;
|
|
import net.minecraft.world.level.ChunkPos;
|
|
import net.minecraft.world.level.Level;
|
|
import net.minecraft.world.level.LevelAccessor;
|
|
import net.minecraft.world.level.block.Block;
|
|
import net.minecraft.world.level.block.Blocks;
|
|
import net.minecraft.world.level.block.entity.BlockEntity;
|
|
import net.minecraft.world.level.block.state.BlockState;
|
|
import net.minecraft.world.level.block.state.properties.Property;
|
|
import net.minecraft.world.level.chunk.ChunkAccess;
|
|
import net.minecraft.world.level.chunk.ChunkStatus;
|
|
import net.minecraft.world.phys.AABB;
|
|
import net.minecraft.world.phys.Vec3;
|
|
import net.neoforged.api.distmarker.Dist;
|
|
import net.neoforged.api.distmarker.OnlyIn;
|
|
import net.neoforged.neoforge.capabilities.Capabilities;
|
|
import net.neoforged.neoforge.items.IItemHandler;
|
|
import net.neoforged.neoforge.items.IItemHandlerModifiable;
|
|
import top.theillusivec4.curios.api.CuriosApi;
|
|
import top.theillusivec4.curios.api.SlotResult;
|
|
|
|
import javax.annotation.Nonnull;
|
|
import javax.annotation.Nullable;
|
|
import java.lang.reflect.Modifier;
|
|
import java.util.List;
|
|
import java.util.Locale;
|
|
import java.util.function.Consumer;
|
|
import java.util.function.Function;
|
|
import java.util.function.Predicate;
|
|
|
|
public final class Helper {
|
|
|
|
public static boolean getBlockEntitiesInArea(LevelAccessor level, BlockPos pos, int radius, Function<BlockEntity, Boolean> consumer) {
|
|
for (var x = pos.getX() - radius >> 4; x <= pos.getX() + radius >> 4; x++) {
|
|
for (var z = pos.getZ() - radius >> 4; z <= pos.getZ() + radius >> 4; z++) {
|
|
var chunk = Helper.getLoadedChunk(level, x, z);
|
|
if (chunk != null) {
|
|
for (var tilePos : chunk.getBlockEntitiesPos()) {
|
|
if (tilePos.distSqr(pos) <= radius * radius) {
|
|
var tile = chunk.getBlockEntity(tilePos);
|
|
if (tile != null && consumer.apply(tile))
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public static void getAuraChunksWithSpotsInArea(Level level, BlockPos pos, int radius, Consumer<AuraChunk> consumer) {
|
|
var data = (LevelData) ILevelData.getLevelData(level);
|
|
for (var x = pos.getX() - radius >> 4; x <= pos.getX() + radius >> 4; x++) {
|
|
for (var z = pos.getZ() - radius >> 4; z <= pos.getZ() + radius >> 4; z++) {
|
|
var chunk = data.auraChunksWithSpots.get(ChunkPos.asLong(x, z));
|
|
if (chunk != null)
|
|
consumer.accept(chunk);
|
|
}
|
|
}
|
|
}
|
|
|
|
public static List<ItemFrame> getAttachedItemFrames(Level level, BlockPos pos) {
|
|
var frames = level.getEntitiesOfClass(ItemFrame.class, new AABB(pos).inflate(0.25));
|
|
for (var i = frames.size() - 1; i >= 0; i--) {
|
|
var frame = frames.get(i);
|
|
var framePos = frame.getPos().relative(frame.getDirection().getOpposite());
|
|
if (!pos.equals(framePos))
|
|
frames.remove(i);
|
|
}
|
|
return frames;
|
|
}
|
|
|
|
public static ChunkAccess getLoadedChunk(LevelAccessor level, int x, int z) {
|
|
// as always, this code is EXTREMELY FRAGILE and editing it before the next major version will probably break it
|
|
if (level.getChunkSource() instanceof ServerChunkCache cache) {
|
|
return cache.isPositionTicking(ChunkPos.asLong(x, z)) ? cache.getChunk(x, z, ChunkStatus.FULL, false) : null;
|
|
} else {
|
|
return level.getChunk(x, z, ChunkStatus.FULL, false);
|
|
}
|
|
}
|
|
|
|
public static int blendColors(int c1, int c2, float ratio) {
|
|
var a = (int) ((c1 >> 24 & 0xFF) * ratio + (c2 >> 24 & 0xFF) * (1 - ratio));
|
|
var r = (int) ((c1 >> 16 & 0xFF) * ratio + (c2 >> 16 & 0xFF) * (1 - ratio));
|
|
var g = (int) ((c1 >> 8 & 0xFF) * ratio + (c2 >> 8 & 0xFF) * (1 - ratio));
|
|
var b = (int) ((c1 & 0xFF) * ratio + (c2 & 0xFF) * (1 - ratio));
|
|
return (a & 255) << 24 | (r & 255) << 16 | (g & 255) << 8 | b & 255;
|
|
}
|
|
|
|
public static boolean areItemsEqual(ItemStack first, ItemStack second, boolean nbt) {
|
|
// TODO see if this is the correct new comparison method?
|
|
return nbt ? ItemStack.isSameItemSameTags(first, second) : ItemStack.isSameItem(first, second);
|
|
}
|
|
|
|
@OnlyIn(Dist.CLIENT)
|
|
public static void renderItemInGui(GuiGraphics graphics, ItemStack stack, int x, int y, float scale) {
|
|
var poseStack = graphics.pose();
|
|
poseStack.pushPose();
|
|
poseStack.translate(x, y, 0);
|
|
poseStack.scale(scale, scale, scale);
|
|
graphics.setColor(1, 1, 1, 1);
|
|
graphics.renderItem(stack, 0, 0);
|
|
graphics.renderItemDecorations(Minecraft.getInstance().font, stack, 0, 0, null);
|
|
poseStack.popPose();
|
|
}
|
|
|
|
@OnlyIn(Dist.CLIENT)
|
|
public static void renderWeirdBox(VertexConsumer buffer, double x, double y, double z, double width, double height, double depth, float r, float g, float b, float a) {
|
|
buffer.vertex(x, y + height, z).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x + width, y + height, z).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x + width, y, z).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x, y, z).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x + width, y, z + depth).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x + width, y, z).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x + width, y + height, z).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x + width, y + height, z + depth).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x + width, y + height, z + depth).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x, y + height, z + depth).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x, y, z + depth).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x + width, y, z + depth).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x, y + height, z + depth).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x, y + height, z).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x, y, z).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x, y, z + depth).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x, y + height, z).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x, y + height, z + depth).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x + width, y + height, z + depth).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x + width, y + height, z).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x + width, y, z).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x + width, y, z + depth).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x, y, z + depth).color(r, g, b, a).endVertex();
|
|
buffer.vertex(x, y, z).color(r, g, b, a).endVertex();
|
|
}
|
|
|
|
public static InteractionResult putStackOnTile(Player player, InteractionHand hand, BlockPos pos, int slot, boolean sound) {
|
|
var tile = player.level().getBlockEntity(pos);
|
|
if (tile instanceof BlockEntityImpl) {
|
|
var handler = (IItemHandlerModifiable) tile.getLevel().getCapability(Capabilities.ItemHandler.BLOCK, tile.getBlockPos(), tile.getBlockState(), tile, null);
|
|
if (handler != null) {
|
|
var handStack = player.getItemInHand(hand);
|
|
if (!handStack.isEmpty()) {
|
|
var remain = handler.insertItem(slot, handStack, player.level().isClientSide);
|
|
if (!ItemStack.matches(remain, handStack)) {
|
|
if (sound)
|
|
player.level().playSound(player, pos.getX() + 0.5, pos.getY() + 0.5, pos.getZ() + 0.5,
|
|
SoundEvents.ITEM_FRAME_ADD_ITEM, SoundSource.PLAYERS, 0.75F, 1F);
|
|
if (!player.level().isClientSide)
|
|
player.setItemInHand(hand, remain);
|
|
return InteractionResult.SUCCESS;
|
|
}
|
|
}
|
|
|
|
if (!handler.getStackInSlot(slot).isEmpty()) {
|
|
if (sound)
|
|
player.level().playSound(player, pos.getX() + 0.5, pos.getY() + 0.5, pos.getZ() + 0.5,
|
|
SoundEvents.ITEM_FRAME_REMOVE_ITEM, SoundSource.PLAYERS, 0.75F, 1F);
|
|
if (!player.level().isClientSide) {
|
|
var stack = handler.getStackInSlot(slot);
|
|
if (!player.addItem(stack)) {
|
|
var item = new ItemEntity(player.level(), player.getX(), player.getY(), player.getZ(), stack);
|
|
player.level().addFreshEntity(item);
|
|
}
|
|
handler.setStackInSlot(slot, ItemStack.EMPTY);
|
|
}
|
|
return InteractionResult.SUCCESS;
|
|
}
|
|
}
|
|
}
|
|
return InteractionResult.CONSUME;
|
|
}
|
|
|
|
public static ICapabilityProvider makeRechargeProvider(ItemStack stack, boolean needsSelected) {
|
|
return new ICapabilityProvider() {
|
|
private final LazyOptional<IAuraRecharge> recharge = LazyOptional.of(() -> (container, containerSlot, itemSlot, isSelected) -> {
|
|
if (isSelected || !needsSelected)
|
|
return Helper.rechargeAuraItem(stack, container, 300);
|
|
return false;
|
|
});
|
|
|
|
@Nullable
|
|
@Override
|
|
public <T> LazyOptional<T> getCapability(@Nonnull Capability<T> capability, @Nullable Direction facing) {
|
|
if (capability == NaturesAuraAPI.CAP_AURA_RECHARGE)
|
|
return this.recharge.cast();
|
|
return LazyOptional.empty();
|
|
}
|
|
};
|
|
}
|
|
|
|
public static boolean rechargeAuraItem(ItemStack stack, IAuraContainer container, int toDrain) {
|
|
if (stack.getDamageValue() > 0 && container.drainAura(toDrain, true) >= toDrain) {
|
|
stack.setDamageValue(stack.getDamageValue() - 1);
|
|
container.drainAura(toDrain, false);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public static BlockState getStateFromString(String raw) {
|
|
var split = raw.split("\\[");
|
|
var block = BuiltInRegistries.BLOCK.get(new ResourceLocation(split[0]));
|
|
if (block != null) {
|
|
var state = block.defaultBlockState();
|
|
if (split.length > 1) {
|
|
for (var part : split[1].replace("]", "").split(",")) {
|
|
var keyValue = part.split("=");
|
|
for (var prop : state.getProperties()) {
|
|
var changed = Helper.findProperty(state, prop, keyValue[0], keyValue[1]);
|
|
if (changed != null) {
|
|
state = changed;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return state;
|
|
} else
|
|
return null;
|
|
}
|
|
|
|
private static <T extends Comparable<T>> BlockState findProperty(BlockState state, Property<T> prop, String key, String newValue) {
|
|
if (key.equals(prop.getName()))
|
|
for (var value : prop.getPossibleValues())
|
|
if (prop.getName(value).equals(newValue))
|
|
return state.setValue(prop, value);
|
|
return null;
|
|
}
|
|
|
|
public static void addAdvancement(Player player, ResourceLocation advancement, String criterion) {
|
|
if (!(player instanceof ServerPlayer playerMp))
|
|
return;
|
|
var adv = playerMp.level().getServer().getAdvancements().get(advancement);
|
|
if (adv != null)
|
|
playerMp.getAdvancements().award(adv, criterion);
|
|
}
|
|
|
|
public static int getIngredientAmount(Ingredient ingredient) {
|
|
var highestAmount = 0;
|
|
for (var stack : ingredient.getItems())
|
|
if (stack.getCount() > highestAmount)
|
|
highestAmount = stack.getCount();
|
|
return highestAmount;
|
|
}
|
|
|
|
public static boolean isHoldingItem(Player player, Item item) {
|
|
for (var hand : InteractionHand.values()) {
|
|
var stack = player.getItemInHand(hand);
|
|
if (!stack.isEmpty() && stack.getItem() == item)
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public static boolean isEmpty(IItemHandler handler) {
|
|
for (var i = 0; i < handler.getSlots(); i++)
|
|
if (!handler.getStackInSlot(i).isEmpty())
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
public static AABB aabb(Vec3 pos) {
|
|
return new AABB(pos.x, pos.y, pos.z, pos.x, pos.y, pos.z);
|
|
}
|
|
|
|
// This is how @ObjectHolder SHOULD work...
|
|
public static <T> void populateObjectHolders(Class<?> clazz, Registry<T> registry) {
|
|
for (var entry : clazz.getFields()) {
|
|
if (!Modifier.isStatic(entry.getModifiers()))
|
|
continue;
|
|
var location = new ResourceLocation(NaturesAura.MOD_ID, entry.getName().toLowerCase(Locale.ROOT));
|
|
if (!registry.containsKey(location)) {
|
|
NaturesAura.LOGGER.fatal("Couldn't find entry named " + location + " in registry");
|
|
continue;
|
|
}
|
|
try {
|
|
entry.set(null, registry.get(location));
|
|
} catch (IllegalAccessException e) {
|
|
NaturesAura.LOGGER.error(e);
|
|
}
|
|
}
|
|
}
|
|
|
|
public static ItemStack getEquippedItem(Predicate<ItemStack> predicate, Player player, boolean hotbarOnly) {
|
|
if (Compat.hasCompat("curios")) {
|
|
var stack = CuriosApi.getCuriosHelper().findFirstCurio(player, predicate).map(SlotResult::stack);
|
|
if (stack.isPresent())
|
|
return stack.get();
|
|
}
|
|
var invSize = hotbarOnly ? 9 : player.getInventory().getContainerSize();
|
|
for (var i = 0; i < invSize; i++) {
|
|
var slot = player.getInventory().getItem(i);
|
|
if (!slot.isEmpty() && predicate.test(slot))
|
|
return slot;
|
|
}
|
|
return ItemStack.EMPTY;
|
|
}
|
|
|
|
public static BlockPos getClosestAirAboveGround(Level level, BlockPos pos, int radius) {
|
|
for (var i = 0; i < radius; i++) {
|
|
var up = pos.above(i);
|
|
if (level.isEmptyBlock(up) && !level.isEmptyBlock(up.below()))
|
|
return up;
|
|
var dn = pos.below(i);
|
|
if (level.isEmptyBlock(dn) && !level.isEmptyBlock(dn.below()))
|
|
return dn;
|
|
}
|
|
return pos;
|
|
}
|
|
|
|
public static void mineRecursively(Level level, BlockPos pos, BlockPos start, ItemStack tool, int horizontalRange, int verticalRange, Predicate<BlockState> filter) {
|
|
if (Math.abs(pos.getX() - start.getX()) >= horizontalRange || Math.abs(pos.getZ() - start.getZ()) >= horizontalRange || Math.abs(pos.getY() - start.getY()) >= verticalRange)
|
|
return;
|
|
|
|
if (!tool.isEmpty()) {
|
|
var state = level.getBlockState(pos);
|
|
level.destroyBlock(pos, false);
|
|
Block.dropResources(state, level, pos, state.hasBlockEntity() ? level.getBlockEntity(pos) : null, null, tool);
|
|
} else {
|
|
// in this case we don't want the block breaking particles, so we can't use destroyBlock
|
|
level.setBlockAndUpdate(pos, Blocks.AIR.defaultBlockState());
|
|
PacketHandler.sendToAllAround(level, pos, 32, new PacketParticles(pos.getX(), pos.getY(), pos.getZ(), PacketParticles.Type.TR_DISAPPEAR));
|
|
}
|
|
|
|
for (var x = -1; x <= 1; x++) {
|
|
for (var y = -1; y <= 1; y++) {
|
|
for (var z = -1; z <= 1; z++) {
|
|
if (x == 0 && y == 0 && z == 0)
|
|
continue;
|
|
var offset = pos.offset(x, y, z);
|
|
var state = level.getBlockState(offset);
|
|
if (filter.test(state))
|
|
Helper.mineRecursively(level, offset, start, tool, horizontalRange, verticalRange, filter);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public static boolean isToolEnabled(ItemStack stack) {
|
|
return stack.hasTag() && !stack.getTag().getBoolean(NaturesAura.MOD_ID + ":disabled");
|
|
}
|
|
|
|
public static boolean toggleToolEnabled(Player player, ItemStack stack) {
|
|
if (!player.isShiftKeyDown())
|
|
return false;
|
|
var disabled = !Helper.isToolEnabled(stack);
|
|
stack.getOrCreateTag().putBoolean(NaturesAura.MOD_ID + ":disabled", !disabled);
|
|
player.level().playSound(null, player.getX() + 0.5, player.getY() + 0.5, player.getZ() + 0.5, SoundEvents.ARROW_HIT_PLAYER, SoundSource.PLAYERS, 0.65F, 1F);
|
|
return true;
|
|
}
|
|
|
|
}
|