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