appeng.tile.crafting.TileMolecularAssembler.java Source code

Java tutorial

Introduction

Here is the source code for appeng.tile.crafting.TileMolecularAssembler.java

Source

/*
 * This file is part of Applied Energistics 2.
 * Copyright (c) 2013 - 2014, AlgorithmX2, All rights reserved.
 *
 * Applied Energistics 2 is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Applied Energistics 2 is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Applied Energistics 2.  If not, see <http://www.gnu.org/licenses/lgpl>.
 */

package appeng.tile.crafting;

import java.io.IOException;
import java.util.List;

import io.netty.buffer.ByteBuf;

import net.minecraft.inventory.IInventory;
import net.minecraft.inventory.InventoryCrafting;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.tileentity.TileEntity;
import net.minecraft.util.EnumFacing;
import net.minecraft.util.math.BlockPos;
import net.minecraft.world.World;
import net.minecraft.world.WorldServer;
import net.minecraftforge.fml.common.FMLCommonHandler;
import net.minecraftforge.fml.common.network.NetworkRegistry.TargetPoint;

import appeng.api.AEApi;
import appeng.api.config.Actionable;
import appeng.api.config.PowerMultiplier;
import appeng.api.config.RedstoneMode;
import appeng.api.config.Settings;
import appeng.api.config.Upgrades;
import appeng.api.definitions.ITileDefinition;
import appeng.api.implementations.IPowerChannelState;
import appeng.api.implementations.IUpgradeableHost;
import appeng.api.implementations.tiles.ICraftingMachine;
import appeng.api.networking.IGridNode;
import appeng.api.networking.crafting.ICraftingPatternDetails;
import appeng.api.networking.events.MENetworkEventSubscribe;
import appeng.api.networking.events.MENetworkPowerStatusChange;
import appeng.api.networking.ticking.IGridTickable;
import appeng.api.networking.ticking.TickRateModulation;
import appeng.api.networking.ticking.TickingRequest;
import appeng.api.storage.data.IAEItemStack;
import appeng.api.util.AECableType;
import appeng.api.util.AEPartLocation;
import appeng.api.util.DimensionalCoord;
import appeng.api.util.IConfigManager;
import appeng.container.ContainerNull;
import appeng.core.sync.network.NetworkHandler;
import appeng.core.sync.packets.PacketAssemblerAnimation;
import appeng.items.misc.ItemEncodedPattern;
import appeng.me.GridAccessException;
import appeng.parts.automation.DefinitionUpgradeInventory;
import appeng.parts.automation.UpgradeInventory;
import appeng.tile.TileEvent;
import appeng.tile.events.TileEventType;
import appeng.tile.grid.AENetworkInvTile;
import appeng.tile.inventory.AppEngInternalInventory;
import appeng.tile.inventory.InvOperation;
import appeng.util.ConfigManager;
import appeng.util.IConfigManagerHost;
import appeng.util.InventoryAdaptor;
import appeng.util.Platform;
import appeng.util.item.AEItemStack;

public class TileMolecularAssembler extends AENetworkInvTile
        implements IUpgradeableHost, IConfigManagerHost, IGridTickable, ICraftingMachine, IPowerChannelState {
    private static final int[] SIDES = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };

    private final InventoryCrafting craftingInv;
    private final AppEngInternalInventory inv = new AppEngInternalInventory(this, 9 + 2);
    private final IConfigManager settings;
    private final UpgradeInventory upgrades;
    private boolean isPowered = false;
    private AEPartLocation pushDirection = AEPartLocation.INTERNAL;
    private ItemStack myPattern = null;
    private ICraftingPatternDetails myPlan = null;
    private double progress = 0;
    private boolean isAwake = false;
    private boolean forcePlan = false;
    private boolean reboot = true;

    public TileMolecularAssembler() {
        final ITileDefinition assembler = AEApi.instance().definitions().blocks().molecularAssembler();

        this.settings = new ConfigManager(this);
        this.settings.registerSetting(Settings.REDSTONE_CONTROLLED, RedstoneMode.IGNORE);
        this.inv.setMaxStackSize(1);
        this.getProxy().setIdlePowerUsage(0.0);
        this.upgrades = new DefinitionUpgradeInventory(assembler, this, this.getUpgradeSlots());
        this.craftingInv = new InventoryCrafting(new ContainerNull(), 3, 3);
    }

    private int getUpgradeSlots() {
        return 5;
    }

    @Override
    public boolean pushPattern(final ICraftingPatternDetails patternDetails, final InventoryCrafting table,
            final EnumFacing where) {
        if (this.myPattern == null) {
            boolean isEmpty = true;
            for (int x = 0; x < this.inv.getSizeInventory(); x++) {
                isEmpty = this.inv.getStackInSlot(x) == null && isEmpty;
            }

            if (isEmpty && patternDetails.isCraftable()) {
                this.forcePlan = true;
                this.myPlan = patternDetails;
                this.pushDirection = AEPartLocation.fromFacing(where);

                for (int x = 0; x < table.getSizeInventory(); x++) {
                    this.inv.setInventorySlotContents(x, table.getStackInSlot(x));
                }

                this.updateSleepiness();
                this.markDirty();
                return true;
            }
        }
        return false;
    }

    private void updateSleepiness() {
        final boolean wasEnabled = this.isAwake;
        this.isAwake = this.myPlan != null && this.hasMats() || this.canPush();
        if (wasEnabled != this.isAwake) {
            try {
                if (this.isAwake) {
                    this.getProxy().getTick().wakeDevice(this.getProxy().getNode());
                } else {
                    this.getProxy().getTick().sleepDevice(this.getProxy().getNode());
                }
            } catch (final GridAccessException e) {
                // :P
            }
        }
    }

    private boolean canPush() {
        return this.inv.getStackInSlot(9) != null;
    }

    private boolean hasMats() {
        if (this.myPlan == null) {
            return false;
        }

        for (int x = 0; x < this.craftingInv.getSizeInventory(); x++) {
            this.craftingInv.setInventorySlotContents(x, this.inv.getStackInSlot(x));
        }

        return this.myPlan.getOutput(this.craftingInv, this.getWorld()) != null;
    }

    @Override
    public boolean acceptsPlans() {
        return this.inv.getStackInSlot(10) == null;
    }

    @Override
    public int getInstalledUpgrades(final Upgrades u) {
        return this.upgrades.getInstalledUpgrades(u);
    }

    @TileEvent(TileEventType.NETWORK_READ)
    public boolean readFromStream_TileMolecularAssembler(final ByteBuf data) {
        final boolean oldPower = this.isPowered;
        this.isPowered = data.readBoolean();
        return this.isPowered != oldPower;
    }

    @TileEvent(TileEventType.NETWORK_WRITE)
    public void writeToStream_TileMolecularAssembler(final ByteBuf data) {
        data.writeBoolean(this.isPowered);
    }

    @TileEvent(TileEventType.WORLD_NBT_WRITE)
    public void writeToNBT_TileMolecularAssembler(final NBTTagCompound data) {
        if (this.forcePlan && this.myPlan != null) {
            final ItemStack pattern = this.myPlan.getPattern();
            if (pattern != null) {
                final NBTTagCompound compound = new NBTTagCompound();
                pattern.writeToNBT(compound);
                data.setTag("myPlan", compound);
                data.setInteger("pushDirection", this.pushDirection.ordinal());
            }
        }

        this.upgrades.writeToNBT(data, "upgrades");
        this.inv.writeToNBT(data, "inv");
        this.settings.writeToNBT(data);
    }

    @TileEvent(TileEventType.WORLD_NBT_READ)
    public void readFromNBT_TileMolecularAssembler(final NBTTagCompound data) {
        if (data.hasKey("myPlan")) {
            final ItemStack myPat = ItemStack.loadItemStackFromNBT(data.getCompoundTag("myPlan"));

            if (myPat != null && myPat.getItem() instanceof ItemEncodedPattern) {
                final World w = this.getWorld();
                final ItemEncodedPattern iep = (ItemEncodedPattern) myPat.getItem();
                final ICraftingPatternDetails ph = iep.getPatternForItem(myPat, w);
                if (ph != null && ph.isCraftable()) {
                    this.forcePlan = true;
                    this.myPlan = ph;
                    this.pushDirection = AEPartLocation.fromOrdinal(data.getInteger("pushDirection"));
                }
            }
        }

        this.upgrades.readFromNBT(data, "upgrades");
        this.inv.readFromNBT(data, "inv");
        this.settings.readFromNBT(data);
        this.recalculatePlan();
    }

    private void recalculatePlan() {
        this.reboot = true;

        if (this.forcePlan) {
            return;
        }

        final ItemStack is = this.inv.getStackInSlot(10);

        if (is != null && is.getItem() instanceof ItemEncodedPattern) {
            if (!Platform.itemComparisons().isEqualItem(is, this.myPattern)) {
                final World w = this.getWorld();
                final ItemEncodedPattern iep = (ItemEncodedPattern) is.getItem();
                final ICraftingPatternDetails ph = iep.getPatternForItem(is, w);

                if (ph != null && ph.isCraftable()) {
                    this.progress = 0;
                    this.myPattern = is;
                    this.myPlan = ph;
                }
            }
        } else {
            this.progress = 0;
            this.forcePlan = false;
            this.myPlan = null;
            this.myPattern = null;
            this.pushDirection = AEPartLocation.INTERNAL;
        }

        this.updateSleepiness();
    }

    @Override
    public AECableType getCableConnectionType(final AEPartLocation dir) {
        return AECableType.COVERED;
    }

    @Override
    public DimensionalCoord getLocation() {
        return new DimensionalCoord(this);
    }

    @Override
    public IConfigManager getConfigManager() {
        return this.settings;
    }

    @Override
    public IInventory getInventoryByName(final String name) {
        if (name.equals("upgrades")) {
            return this.upgrades;
        }

        if (name.equals("mac")) {
            return this.inv;
        }

        return null;
    }

    @Override
    public void updateSetting(final IConfigManager manager, final Enum settingName, final Enum newValue) {

    }

    @Override
    public IInventory getInternalInventory() {
        return this.inv;
    }

    @Override
    public int getInventoryStackLimit() {
        return 1;
    }

    @Override
    public boolean isItemValidForSlot(final int i, final ItemStack itemstack) {
        if (i >= 9) {
            return false;
        }

        if (this.hasPattern()) {
            return this.myPlan.isValidItemForSlot(i, itemstack, this.getWorld());
        }

        return false;
    }

    private boolean hasPattern() {
        return this.myPlan != null && this.inv.getStackInSlot(10) != null;
    }

    @Override
    public void onChangeInventory(final IInventory inv, final int slot, final InvOperation mc,
            final ItemStack removed, final ItemStack added) {
        if (inv == this.inv) {
            this.recalculatePlan();
        }
    }

    @Override
    public boolean canExtractItem(final int slotIndex, final ItemStack extractedItem, final EnumFacing side) {
        return slotIndex == 9;
    }

    @Override
    public int[] getAccessibleSlotsBySide(final EnumFacing whichSide) {
        return SIDES;
    }

    public int getCraftingProgress() {
        return (int) this.progress;
    }

    @Override
    public void getDrops(final World w, final BlockPos pos, final List<ItemStack> drops) {
        super.getDrops(w, pos, drops);

        for (int h = 0; h < this.upgrades.getSizeInventory(); h++) {
            final ItemStack is = this.upgrades.getStackInSlot(h);
            if (is != null) {
                drops.add(is);
            }
        }
    }

    @Override
    public TickingRequest getTickingRequest(final IGridNode node) {
        this.recalculatePlan();
        this.updateSleepiness();
        return new TickingRequest(1, 1, !this.isAwake, false);
    }

    @Override
    public TickRateModulation tickingRequest(final IGridNode node, int ticksSinceLastCall) {
        if (this.inv.getStackInSlot(9) != null) {
            this.pushOut(this.inv.getStackInSlot(9));

            // did it eject?
            if (this.inv.getStackInSlot(9) == null) {
                this.markDirty();
            }

            this.ejectHeldItems();
            this.updateSleepiness();
            this.progress = 0;
            return this.isAwake ? TickRateModulation.IDLE : TickRateModulation.SLEEP;
        }

        if (this.myPlan == null) {
            this.updateSleepiness();
            return TickRateModulation.SLEEP;
        }

        if (this.reboot) {
            ticksSinceLastCall = 1;
        }

        if (!this.isAwake) {
            return TickRateModulation.SLEEP;
        }

        this.reboot = false;
        int speed = 10;
        switch (this.upgrades.getInstalledUpgrades(Upgrades.SPEED)) {
        case 0:
            this.progress += this.userPower(ticksSinceLastCall, speed = 10, 1.0);
            break;
        case 1:
            this.progress += this.userPower(ticksSinceLastCall, speed = 13, 1.3);
            break;
        case 2:
            this.progress += this.userPower(ticksSinceLastCall, speed = 17, 1.7);
            break;
        case 3:
            this.progress += this.userPower(ticksSinceLastCall, speed = 20, 2.0);
            break;
        case 4:
            this.progress += this.userPower(ticksSinceLastCall, speed = 25, 2.5);
            break;
        case 5:
            this.progress += this.userPower(ticksSinceLastCall, speed = 50, 5.0);
            break;
        }

        if (this.progress >= 100) {
            for (int x = 0; x < this.craftingInv.getSizeInventory(); x++) {
                this.craftingInv.setInventorySlotContents(x, this.inv.getStackInSlot(x));
            }

            this.progress = 0;
            final ItemStack output = this.myPlan.getOutput(this.craftingInv, this.getWorld());
            if (output != null) {
                FMLCommonHandler.instance().firePlayerCraftingEvent(
                        Platform.getPlayer((WorldServer) this.getWorld()), output, this.craftingInv);

                this.pushOut(output.copy());

                for (int x = 0; x < this.craftingInv.getSizeInventory(); x++) {
                    this.inv.setInventorySlotContents(x,
                            Platform.getContainerItem(this.craftingInv.getStackInSlot(x)));
                }

                if (this.inv.getStackInSlot(10) == null) {
                    this.forcePlan = false;
                    this.myPlan = null;
                    this.pushDirection = AEPartLocation.INTERNAL;
                }

                this.ejectHeldItems();

                try {
                    final TargetPoint where = new TargetPoint(this.worldObj.provider.getDimension(),
                            this.pos.getX(), this.pos.getY(), this.pos.getZ(), 32);
                    final IAEItemStack item = AEItemStack.create(output);
                    NetworkHandler.instance()
                            .sendToAllAround(new PacketAssemblerAnimation(this.pos, (byte) speed, item), where);
                } catch (final IOException e) {
                    // ;P
                }

                this.markDirty();
                this.updateSleepiness();
                return this.isAwake ? TickRateModulation.IDLE : TickRateModulation.SLEEP;
            }
        }

        return TickRateModulation.FASTER;
    }

    private void ejectHeldItems() {
        if (this.inv.getStackInSlot(9) == null) {
            for (int x = 0; x < 9; x++) {
                final ItemStack is = this.inv.getStackInSlot(x);
                if (is != null) {
                    if (this.myPlan == null || !this.myPlan.isValidItemForSlot(x, is, this.worldObj)) {
                        this.inv.setInventorySlotContents(9, is);
                        this.inv.setInventorySlotContents(x, null);
                        this.markDirty();
                        return;
                    }
                }
            }
        }
    }

    private int userPower(final int ticksPassed, final int bonusValue, final double acceleratorTax) {
        try {
            return (int) (this.getProxy().getEnergy().extractAEPower(ticksPassed * bonusValue * acceleratorTax,
                    Actionable.MODULATE, PowerMultiplier.CONFIG) / acceleratorTax);
        } catch (final GridAccessException e) {
            return 0;
        }
    }

    private void pushOut(ItemStack output) {
        if (this.pushDirection == AEPartLocation.INTERNAL) {
            for (final EnumFacing d : EnumFacing.VALUES) {
                output = this.pushTo(output, d);
            }
        } else {
            output = this.pushTo(output, this.pushDirection.getFacing());
        }

        if (output == null && this.forcePlan) {
            this.forcePlan = false;
            this.recalculatePlan();
        }

        this.inv.setInventorySlotContents(9, output);
    }

    private ItemStack pushTo(ItemStack output, final EnumFacing d) {
        if (output == null) {
            return output;
        }

        final TileEntity te = this.getWorld().getTileEntity(this.pos.offset(d));

        if (te == null) {
            return output;
        }

        final InventoryAdaptor adaptor = InventoryAdaptor.getAdaptor(te, d.getOpposite());

        if (adaptor == null) {
            return output;
        }

        final int size = output.stackSize;
        output = adaptor.addItems(output);
        final int newSize = output == null ? 0 : output.stackSize;

        if (size != newSize) {
            this.markDirty();
        }

        return output;
    }

    @MENetworkEventSubscribe
    public void onPowerEvent(final MENetworkPowerStatusChange p) {
        this.updatePowerState();
    }

    private void updatePowerState() {
        boolean newState = false;

        try {
            newState = this.getProxy().isActive() && this.getProxy().getEnergy().extractAEPower(1,
                    Actionable.SIMULATE, PowerMultiplier.CONFIG) > 0.0001;
        } catch (final GridAccessException ignored) {

        }

        if (newState != this.isPowered) {
            this.isPowered = newState;
            this.markForUpdate();
        }
    }

    @Override
    public boolean isPowered() {
        return this.isPowered;
    }

    @Override
    public boolean isActive() {
        return this.isPowered;
    }

}