176 lines
4.3 KiB
Java
176 lines
4.3 KiB
Java
package appeng.crafting;
|
|
|
|
import java.util.Collection;
|
|
import java.util.Collections;
|
|
import java.util.HashSet;
|
|
import java.util.Iterator;
|
|
import java.util.Set;
|
|
|
|
import net.minecraft.nbt.NBTTagCompound;
|
|
import appeng.api.AEApi;
|
|
import appeng.api.config.Actionable;
|
|
import appeng.api.networking.crafting.ICraftingPatternDetails;
|
|
import appeng.api.networking.storage.IStorageGrid;
|
|
import appeng.api.storage.IMEInventory;
|
|
import appeng.api.storage.data.IAEItemStack;
|
|
import appeng.api.storage.data.IItemList;
|
|
import appeng.me.cache.CraftingCache;
|
|
|
|
import com.google.common.collect.ArrayListMultimap;
|
|
import com.google.common.collect.Multimap;
|
|
|
|
public class CraftingJob implements ICraftingParent
|
|
{
|
|
|
|
IAEItemStack output;
|
|
|
|
IItemList<IAEItemStack> storage;
|
|
|
|
HashSet<IAEItemStack> prophecies;
|
|
|
|
Multimap<IAEItemStack, CraftingTask> bottom;
|
|
CraftingTask top;
|
|
|
|
ICraftingHost jobHost;
|
|
|
|
public CraftingJob(ICraftingHost host, NBTTagCompound data) {
|
|
jobHost = host;
|
|
storage = AEApi.instance().storage().createItemList();
|
|
prophecies = new HashSet();
|
|
bottom = ArrayListMultimap.create();
|
|
}
|
|
|
|
public CraftingJob(ICraftingHost host, IAEItemStack what, Actionable mode) throws CraftingMissingItemsException {
|
|
jobHost = host;
|
|
output = what.copy();
|
|
storage = AEApi.instance().storage().createItemList();
|
|
prophecies = new HashSet();
|
|
bottom = ArrayListMultimap.create();
|
|
|
|
CraftingCache cc = host.getGrid().getCache( CraftingCache.class );
|
|
IStorageGrid sg = host.getGrid().getCache( IStorageGrid.class );
|
|
|
|
IItemList<IAEItemStack> missing = AEApi.instance().storage().createItemList();
|
|
|
|
MECraftingInventory meci = new MECraftingInventory( sg.getItemInventory() );
|
|
|
|
calculateCrafting( cc, this, meci, missing, what, mode );
|
|
|
|
if ( !missing.isEmpty() )
|
|
{
|
|
if ( mode == Actionable.MODULATE )
|
|
{
|
|
IMEInventory<IAEItemStack> netStorage = sg.getItemInventory();
|
|
|
|
Iterator<IAEItemStack> i = storage.iterator();
|
|
while (i.hasNext())
|
|
{
|
|
IAEItemStack item = i.next();
|
|
netStorage.injectItems( item, mode, host.getActionSrc() );
|
|
}
|
|
}
|
|
|
|
throw new CraftingMissingItemsException( missing );
|
|
}
|
|
|
|
if ( mode == Actionable.MODULATE )
|
|
meci.moveItemsToStorage( storage );
|
|
}
|
|
|
|
public void calculateCrafting(CraftingCache cc, ICraftingParent parent, IMEInventory<IAEItemStack> inv, IItemList<IAEItemStack> missing, IAEItemStack what,
|
|
Actionable mode)
|
|
{
|
|
Set<ICraftingPatternDetails> patterns = cc.getCraftingFor( what );
|
|
|
|
for (ICraftingPatternDetails details : patterns)
|
|
{
|
|
IAEItemStack[] requirements = details.getCondencedInputs();
|
|
if ( canMake( requirements, inv ) )
|
|
{
|
|
extractItems( requirements, inv, storage, missing );
|
|
return;
|
|
}
|
|
}
|
|
|
|
for (ICraftingPatternDetails details : patterns)
|
|
{
|
|
IAEItemStack[] requirements = details.getCondencedInputs();
|
|
extractItems( requirements, inv, storage, missing );
|
|
return;
|
|
}
|
|
}
|
|
|
|
private void extractItems(IAEItemStack[] requirements, IMEInventory<IAEItemStack> inv, IItemList<IAEItemStack> storage2, IItemList<IAEItemStack> missing)
|
|
{
|
|
for (IAEItemStack is : requirements)
|
|
{
|
|
inv.extractItems( is, Actionable.MODULATE, jobHost.getActionSrc() );
|
|
}
|
|
}
|
|
|
|
private boolean canMake(IAEItemStack[] requirements, IMEInventory<IAEItemStack> inv)
|
|
{
|
|
|
|
for (IAEItemStack is : requirements)
|
|
{
|
|
IAEItemStack avail = inv.extractItems( is, Actionable.SIMULATE, jobHost.getActionSrc() );
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
public Collection<CraftingTask> getBottom()
|
|
{
|
|
return bottom.values();
|
|
}
|
|
|
|
public IAEItemStack addProgress(IAEItemStack progress)
|
|
{
|
|
Collection<CraftingTask> task = bottom.get( progress );
|
|
|
|
if ( task == null )
|
|
return progress;
|
|
|
|
Iterator<CraftingTask> i = task.iterator();
|
|
|
|
while (i.hasNext())
|
|
{
|
|
// adding to the first item..
|
|
progress = i.next().addProgress( progress );
|
|
|
|
// if nothing remains return
|
|
if ( progress == null )
|
|
return null;
|
|
|
|
// if something remains continue..
|
|
i = task.iterator();
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public void writeToNBT(NBTTagCompound out)
|
|
{
|
|
|
|
}
|
|
|
|
@Override
|
|
public void removeChild(CraftingTask craftingTask)
|
|
{
|
|
// TODO Auto-generated method stub
|
|
}
|
|
|
|
public ICraftingHost getHost()
|
|
{
|
|
// TODO Auto-generated method stub
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public Collection<? extends ICraftingParent> getSubJobs()
|
|
{
|
|
return Collections.singleton( top );
|
|
}
|
|
|
|
}
|