Applied-Energistics-2-tilera-Edition/src/main/java/appeng/integration/modules/helpers/FactorizationBarrel.java

/*
 * 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.integration.modules.helpers;

import appeng.api.config.Actionable;
import appeng.api.networking.security.BaseActionSource;
import appeng.api.storage.IMEInventory;
import appeng.api.storage.StorageChannel;
import appeng.api.storage.data.IAEItemStack;
import appeng.api.storage.data.IItemList;
import appeng.integration.abstraction.IFZ;
import appeng.util.item.AEItemStack;
import net.minecraft.item.ItemStack;
import net.minecraft.tileentity.TileEntity;

public class FactorizationBarrel implements IMEInventory<IAEItemStack> {
    private final IFZ fProxy;
    private final TileEntity te;

    public FactorizationBarrel(final IFZ proxy, final TileEntity tile) {
        this.te = tile;
        this.fProxy = proxy;
    }

    public long remainingItemCount() {
        return this.fProxy.barrelGetMaxItemCount(this.te)
            - this.fProxy.barrelGetItemCount(this.te);
    }

    @Override
    public IAEItemStack injectItems(
        final IAEItemStack input, final Actionable mode, final BaseActionSource src
    ) {
        if (input == null) {
            return null;
        }
        if (input.getStackSize() == 0) {
            return null;
        }

        final ItemStack shared = input.getItemStack();
        if (shared.isItemDamaged()) {
            return input;
        }

        if (this.remainingItemTypes() > 0) {
            if (mode == Actionable.MODULATE) {
                this.fProxy.setItemType(this.te, input.getItemStack());
            }
        }

        if (this.containsItemType(input, mode == Actionable.SIMULATE)) {
            final int max = this.fProxy.barrelGetMaxItemCount(this.te);
            final int newTotal
                = (int) this.storedItemCount() + (int) input.getStackSize();
            if (newTotal > max) {
                if (mode == Actionable.MODULATE) {
                    this.fProxy.barrelSetCount(this.te, max);
                }
                final IAEItemStack result = input.copy();
                result.setStackSize(newTotal - max);
                return result;
            } else {
                if (mode == Actionable.MODULATE) {
                    this.fProxy.barrelSetCount(this.te, newTotal);
                }
                return null;
            }
        }

        return input;
    }

    private long remainingItemTypes() {
        return this.fProxy.barrelGetItem(this.te) == null ? 1 : 0;
    }

    private boolean containsItemType(final IAEItemStack i, final boolean acceptEmpty) {
        final ItemStack currentItem = this.fProxy.barrelGetItem(this.te);

        // empty barrels want your love too!
        if (acceptEmpty && currentItem == null) {
            return true;
        }

        return i.equals(currentItem);
    }

    private long storedItemCount() {
        return this.fProxy.barrelGetItemCount(this.te);
    }

    @Override
    public IAEItemStack extractItems(
        final IAEItemStack request, final Actionable mode, final BaseActionSource src
    ) {
        if (this.containsItemType(request, false)) {
            final int howMany = (int) this.storedItemCount();
            if (request.getStackSize() >= howMany) {
                if (mode == Actionable.MODULATE) {
                    this.fProxy.setItemType(this.te, null);
                    this.fProxy.barrelSetCount(this.te, 0);
                }

                final IAEItemStack r = request.copy();
                r.setStackSize(howMany);
                return r;
            } else {
                if (mode == Actionable.MODULATE) {
                    this.fProxy.barrelSetCount(
                        this.te, (int) (howMany - request.getStackSize())
                    );
                }
                return request.copy();
            }
        }
        return null;
    }

    @Override
    public IItemList<IAEItemStack> getAvailableItems(final IItemList out) {
        final ItemStack i = this.fProxy.barrelGetItem(this.te);
        if (i != null) {
            i.stackSize = this.fProxy.barrelGetItemCount(this.te);
            out.addStorage(AEItemStack.create(i));
        }

        return out;
    }

    @Override
    public StorageChannel getChannel() {
        return StorageChannel.ITEMS;
    }
}