Search in sources :

Example 1 with IFrequencyHandler

use of mekanism.common.lib.frequency.IFrequencyHandler in project Mekanism by mekanism.

the class BaseBlockLootTables method dropSelfWithContents.

protected void dropSelfWithContents(List<IBlockProvider> blockProviders) {
    // For example, when writing this we added dump mode for chemical tanks to getting transferred to the item
    for (IBlockProvider blockProvider : blockProviders) {
        Block block = blockProvider.getBlock();
        if (skipBlock(block)) {
            continue;
        }
        CopyNbt.Builder nbtBuilder = CopyNbt.copyData(Source.BLOCK_ENTITY);
        boolean hasData = false;
        boolean hasContents = false;
        @Nullable TileEntity tile = null;
        if (block instanceof IHasTileEntity) {
            tile = ((IHasTileEntity<?>) block).getTileType().create();
        }
        if (tile instanceof IFrequencyHandler && ((IFrequencyHandler) tile).getFrequencyComponent().hasCustomFrequencies()) {
            nbtBuilder.copy(NBTConstants.COMPONENT_FREQUENCY, NBTConstants.MEK_DATA + "." + NBTConstants.COMPONENT_FREQUENCY);
            hasData = true;
        }
        if (Attribute.has(block, AttributeSecurity.class)) {
            // TODO: Should we just save the entire security component?
            nbtBuilder.copy(NBTConstants.COMPONENT_SECURITY + "." + NBTConstants.OWNER_UUID, NBTConstants.MEK_DATA + "." + NBTConstants.OWNER_UUID);
            nbtBuilder.copy(NBTConstants.COMPONENT_SECURITY + "." + NBTConstants.SECURITY_MODE, NBTConstants.MEK_DATA + "." + NBTConstants.SECURITY_MODE);
            hasData = true;
        }
        if (Attribute.has(block, AttributeUpgradeSupport.class)) {
            nbtBuilder.copy(NBTConstants.COMPONENT_UPGRADE, NBTConstants.MEK_DATA + "." + NBTConstants.COMPONENT_UPGRADE);
            hasData = true;
        }
        if (tile instanceof ISideConfiguration) {
            nbtBuilder.copy(NBTConstants.COMPONENT_CONFIG, NBTConstants.MEK_DATA + "." + NBTConstants.COMPONENT_CONFIG);
            nbtBuilder.copy(NBTConstants.COMPONENT_EJECTOR, NBTConstants.MEK_DATA + "." + NBTConstants.COMPONENT_EJECTOR);
            hasData = true;
        }
        if (tile instanceof ISustainedData) {
            Set<Entry<String, String>> remapEntries = ((ISustainedData) tile).getTileDataRemap().entrySet();
            for (Entry<String, String> remapEntry : remapEntries) {
                nbtBuilder.copy(remapEntry.getKey(), NBTConstants.MEK_DATA + "." + remapEntry.getValue());
            }
            if (!remapEntries.isEmpty()) {
                hasData = true;
            }
        }
        if (Attribute.has(block, AttributeRedstone.class)) {
            nbtBuilder.copy(NBTConstants.CONTROL_TYPE, NBTConstants.MEK_DATA + "." + NBTConstants.CONTROL_TYPE);
            hasData = true;
        }
        if (tile instanceof TileEntityMekanism) {
            TileEntityMekanism tileEntity = (TileEntityMekanism) tile;
            for (SubstanceType type : EnumUtils.SUBSTANCES) {
                if (tileEntity.handles(type) && !type.getContainers(tileEntity).isEmpty()) {
                    nbtBuilder.copy(type.getContainerTag(), NBTConstants.MEK_DATA + "." + type.getContainerTag());
                    hasData = true;
                    if (type != SubstanceType.ENERGY && type != SubstanceType.HEAT) {
                        hasContents = true;
                    }
                }
            }
        }
        if (Attribute.has(block, AttributeInventory.class)) {
            // then only copy the slots if we actually have any slots because otherwise maybe something just went wrong
            if (!(tile instanceof IItemHandler) || ((IItemHandler) tile).getSlots() > 0) {
                // If we don't actually handle saving an inventory (such as the quantum entangloporter, don't actually add it as something to copy)
                if (!(tile instanceof TileEntityMekanism) || ((TileEntityMekanism) tile).persistInventory()) {
                    nbtBuilder.copy(NBTConstants.ITEMS, NBTConstants.MEK_DATA + "." + NBTConstants.ITEMS);
                    hasData = true;
                    hasContents = true;
                }
            }
        }
        if (block instanceof BlockCardboardBox) {
            // TODO: Do this better so that it doesn't have to be as hard coded to being a cardboard box
            nbtBuilder.copy(NBTConstants.DATA, NBTConstants.MEK_DATA + "." + NBTConstants.DATA);
            hasData = true;
        }
        if (!hasData) {
            // To keep the json as clean as possible don't bother even registering a blank accept function if we have no
            // persistent data that we want to copy. Also log a warning so that we don't have to attempt to check against
            // that block
            dropSelf(block);
        } else {
            add(block, LootTable.lootTable().withPool(applyExplosionCondition(hasContents, LootPool.lootPool().name("main").setRolls(ConstantRange.exactly(1)).add(ItemLootEntry.lootTableItem(block).apply(nbtBuilder)))));
        }
    }
}
Also used : TileEntityMekanism(mekanism.common.tile.base.TileEntityMekanism) IHasTileEntity(mekanism.common.block.interfaces.IHasTileEntity) IItemHandler(net.minecraftforge.items.IItemHandler) CopyNbt(net.minecraft.loot.functions.CopyNbt) IHasTileEntity(mekanism.common.block.interfaces.IHasTileEntity) TileEntity(net.minecraft.tileentity.TileEntity) Entry(java.util.Map.Entry) ItemLootEntry(net.minecraft.loot.ItemLootEntry) LootEntry(net.minecraft.loot.LootEntry) SubstanceType(mekanism.common.tile.base.SubstanceType) IBlockProvider(mekanism.api.providers.IBlockProvider) IFrequencyHandler(mekanism.common.lib.frequency.IFrequencyHandler) Block(net.minecraft.block.Block) SlabBlock(net.minecraft.block.SlabBlock) BlockCardboardBox(mekanism.common.block.BlockCardboardBox) ISustainedData(mekanism.common.tile.interfaces.ISustainedData) Nullable(javax.annotation.Nullable) ISideConfiguration(mekanism.common.tile.interfaces.ISideConfiguration)

Example 2 with IFrequencyHandler

use of mekanism.common.lib.frequency.IFrequencyHandler in project Mekanism by mekanism.

the class PacketGuiSetFrequency method handle.

@Override
public void handle(NetworkEvent.Context context) {
    ServerPlayerEntity player = context.getSender();
    if (player == null) {
        return;
    }
    if (updateType.isTile()) {
        TileEntity tile = WorldUtils.getTileEntity(player.level, tilePosition);
        if (SecurityUtils.canAccess(player, tile) && tile instanceof IFrequencyHandler) {
            if (updateType == FrequencyUpdate.SET_TILE) {
                ((IFrequencyHandler) tile).setFrequency(type, data, player.getUUID());
            } else if (updateType == FrequencyUpdate.REMOVE_TILE) {
                ((IFrequencyHandler) tile).removeFrequency(type, data, player.getUUID());
            }
        }
    } else {
        ItemStack stack = player.getItemInHand(currentHand);
        if (SecurityUtils.canAccess(player, stack) && stack.getItem() instanceof IFrequencyItem) {
            IFrequencyItem item = (IFrequencyItem) stack.getItem();
            FrequencyManager<FREQ> manager = type.getManager(data, player.getUUID());
            if (updateType == FrequencyUpdate.SET_ITEM) {
                // Note: We don't bother validating if the frequency is public or not here, as if it isn't then
                // a new private frequency will just be created for the player who sent a packet they shouldn't
                // have been able to send due to not knowing what private frequencies exist for other players
                item.setFrequency(stack, manager.getOrCreateFrequency(data, player.getUUID()));
            } else if (updateType == FrequencyUpdate.REMOVE_ITEM) {
                if (manager.remove(data.getKey(), player.getUUID())) {
                    FrequencyIdentity current = item.getFrequencyIdentity(stack);
                    if (current != null && current.equals(data)) {
                        // If the frequency we are removing matches the stored frequency set it to nothing
                        item.setFrequency(stack, null);
                    }
                }
            }
        }
    }
}
Also used : TileEntity(net.minecraft.tileentity.TileEntity) IFrequencyHandler(mekanism.common.lib.frequency.IFrequencyHandler) FrequencyIdentity(mekanism.common.lib.frequency.Frequency.FrequencyIdentity) ServerPlayerEntity(net.minecraft.entity.player.ServerPlayerEntity) ItemStack(net.minecraft.item.ItemStack) IFrequencyItem(mekanism.common.lib.frequency.IFrequencyItem)

Aggregations

IFrequencyHandler (mekanism.common.lib.frequency.IFrequencyHandler)2 TileEntity (net.minecraft.tileentity.TileEntity)2 Entry (java.util.Map.Entry)1 Nullable (javax.annotation.Nullable)1 IBlockProvider (mekanism.api.providers.IBlockProvider)1 BlockCardboardBox (mekanism.common.block.BlockCardboardBox)1 IHasTileEntity (mekanism.common.block.interfaces.IHasTileEntity)1 FrequencyIdentity (mekanism.common.lib.frequency.Frequency.FrequencyIdentity)1 IFrequencyItem (mekanism.common.lib.frequency.IFrequencyItem)1 SubstanceType (mekanism.common.tile.base.SubstanceType)1 TileEntityMekanism (mekanism.common.tile.base.TileEntityMekanism)1 ISideConfiguration (mekanism.common.tile.interfaces.ISideConfiguration)1 ISustainedData (mekanism.common.tile.interfaces.ISustainedData)1 Block (net.minecraft.block.Block)1 SlabBlock (net.minecraft.block.SlabBlock)1 ServerPlayerEntity (net.minecraft.entity.player.ServerPlayerEntity)1 ItemStack (net.minecraft.item.ItemStack)1 ItemLootEntry (net.minecraft.loot.ItemLootEntry)1 LootEntry (net.minecraft.loot.LootEntry)1 CopyNbt (net.minecraft.loot.functions.CopyNbt)1