239 lines
6.9 KiB
Java
239 lines
6.9 KiB
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.helpers;
|
|
|
|
import java.util.concurrent.ExecutionException;
|
|
import java.util.concurrent.Future;
|
|
|
|
import appeng.api.AEApi;
|
|
import appeng.api.networking.IGrid;
|
|
import appeng.api.networking.crafting.ICraftingGrid;
|
|
import appeng.api.networking.crafting.ICraftingJob;
|
|
import appeng.api.networking.crafting.ICraftingLink;
|
|
import appeng.api.networking.crafting.ICraftingRequester;
|
|
import appeng.api.networking.security.BaseActionSource;
|
|
import appeng.api.storage.data.IAEItemStack;
|
|
import appeng.util.InventoryAdaptor;
|
|
import com.google.common.collect.ImmutableSet;
|
|
import net.minecraft.nbt.NBTTagCompound;
|
|
import net.minecraft.world.World;
|
|
|
|
public class MultiCraftingTracker {
|
|
private final int size;
|
|
private final ICraftingRequester owner;
|
|
|
|
private Future<ICraftingJob>[] jobs = null;
|
|
private ICraftingLink[] links = null;
|
|
|
|
public MultiCraftingTracker(final ICraftingRequester o, final int size) {
|
|
this.owner = o;
|
|
this.size = size;
|
|
}
|
|
|
|
public void readFromNBT(final NBTTagCompound extra) {
|
|
for (int x = 0; x < this.size; x++) {
|
|
final NBTTagCompound link = extra.getCompoundTag("links-" + x);
|
|
|
|
if (link != null && !link.hasNoTags()) {
|
|
this.setLink(
|
|
x, AEApi.instance().storage().loadCraftingLink(link, this.owner)
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
public void writeToNBT(final NBTTagCompound extra) {
|
|
for (int x = 0; x < this.size; x++) {
|
|
final ICraftingLink link = this.getLink(x);
|
|
|
|
if (link != null) {
|
|
final NBTTagCompound ln = new NBTTagCompound();
|
|
link.writeToNBT(ln);
|
|
extra.setTag("links-" + x, ln);
|
|
}
|
|
}
|
|
}
|
|
|
|
public boolean handleCrafting(
|
|
final int x,
|
|
final long itemToCraft,
|
|
final IAEItemStack ais,
|
|
final InventoryAdaptor d,
|
|
final World w,
|
|
final IGrid g,
|
|
final ICraftingGrid cg,
|
|
final BaseActionSource mySrc
|
|
) {
|
|
if (ais != null && d.simulateAdd(ais.getItemStack()) == null) {
|
|
final Future<ICraftingJob> craftingJob = this.getJob(x);
|
|
|
|
if (this.getLink(x) != null) {
|
|
return false;
|
|
} else if (craftingJob != null) {
|
|
try {
|
|
ICraftingJob job = null;
|
|
if (craftingJob.isDone()) {
|
|
job = craftingJob.get();
|
|
}
|
|
|
|
if (job != null) {
|
|
final ICraftingLink link
|
|
= cg.submitJob(job, this.owner, null, false, mySrc);
|
|
|
|
this.setJob(x, null);
|
|
|
|
if (link != null) {
|
|
this.setLink(x, link);
|
|
|
|
return true;
|
|
}
|
|
}
|
|
} catch (final InterruptedException e) {
|
|
// :P
|
|
} catch (final ExecutionException e) {
|
|
// :P
|
|
}
|
|
} else {
|
|
if (this.getLink(x) == null) {
|
|
final IAEItemStack aisC = ais.copy();
|
|
aisC.setStackSize(itemToCraft);
|
|
|
|
this.setJob(x, cg.beginCraftingJob(w, g, mySrc, aisC, null));
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public ImmutableSet<ICraftingLink> getRequestedJobs() {
|
|
if (this.links == null) {
|
|
return ImmutableSet.of();
|
|
}
|
|
|
|
return ImmutableSet.copyOf(new NonNullArrayIterator<ICraftingLink>(this.links));
|
|
}
|
|
|
|
public void jobStateChange(final ICraftingLink link) {
|
|
if (this.links != null) {
|
|
for (int x = 0; x < this.links.length; x++) {
|
|
if (this.links[x] == link) {
|
|
this.setLink(x, null);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
int getSlot(final ICraftingLink link) {
|
|
if (this.links != null) {
|
|
for (int x = 0; x < this.links.length; x++) {
|
|
if (this.links[x] == link) {
|
|
return x;
|
|
}
|
|
}
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
void cancel() {
|
|
if (this.links != null) {
|
|
for (final ICraftingLink l : this.links) {
|
|
if (l != null) {
|
|
l.cancel();
|
|
}
|
|
}
|
|
|
|
this.links = null;
|
|
}
|
|
|
|
if (this.jobs != null) {
|
|
for (final Future<ICraftingJob> l : this.jobs) {
|
|
if (l != null) {
|
|
l.cancel(true);
|
|
}
|
|
}
|
|
|
|
this.jobs = null;
|
|
}
|
|
}
|
|
|
|
boolean isBusy(final int slot) {
|
|
return this.getLink(slot) != null || this.getJob(slot) != null;
|
|
}
|
|
|
|
private ICraftingLink getLink(final int slot) {
|
|
if (this.links == null) {
|
|
return null;
|
|
}
|
|
|
|
return this.links[slot];
|
|
}
|
|
|
|
private void setLink(final int slot, final ICraftingLink l) {
|
|
if (this.links == null) {
|
|
this.links = new ICraftingLink[this.size];
|
|
}
|
|
|
|
this.links[slot] = l;
|
|
|
|
boolean hasStuff = false;
|
|
for (int x = 0; x < this.links.length; x++) {
|
|
final ICraftingLink g = this.links[x];
|
|
|
|
if (g == null || g.isCanceled() || g.isDone()) {
|
|
this.links[x] = null;
|
|
} else {
|
|
hasStuff = true;
|
|
}
|
|
}
|
|
|
|
if (!hasStuff) {
|
|
this.links = null;
|
|
}
|
|
}
|
|
|
|
private Future<ICraftingJob> getJob(final int slot) {
|
|
if (this.jobs == null) {
|
|
return null;
|
|
}
|
|
|
|
return this.jobs[slot];
|
|
}
|
|
|
|
private void setJob(final int slot, final Future<ICraftingJob> l) {
|
|
if (this.jobs == null) {
|
|
this.jobs = new Future[this.size];
|
|
}
|
|
|
|
this.jobs[slot] = l;
|
|
|
|
boolean hasStuff = false;
|
|
|
|
for (final Future<ICraftingJob> job : this.jobs) {
|
|
if (job != null) {
|
|
hasStuff = true;
|
|
}
|
|
}
|
|
|
|
if (!hasStuff) {
|
|
this.jobs = null;
|
|
}
|
|
}
|
|
}
|