Applied-Energistics-2-tiler.../helpers/MultiCraftingTracker.java

231 lines
4.4 KiB
Java

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<ICraftingJob>[] 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<ICraftingJob> 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<ICraftingJob> getJob(int slot)
{
if ( jobs == null )
return null;
return jobs[slot];
}
void setJob(int slot, Future<ICraftingJob> 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<ICraftingLink> 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<ICraftingJob> l : jobs)
{
if ( l != null )
l.cancel( true );
}
jobs = null;
}
}
public boolean isBusy(int slot)
{
return getLink( slot ) != null || getJob( slot ) != null;
}
}