41 lines
940 B
Java
41 lines
940 B
Java
|
package appeng.core.features.registries;
|
||
|
|
|
||
|
|
import java.util.HashMap;
|
||
|
|
|
||
|
|
import net.minecraft.item.ItemStack;
|
||
|
|
import net.minecraftforge.fluids.FluidContainerRegistry;
|
||
|
|
import appeng.api.config.TunnelType;
|
||
|
|
import appeng.api.features.IP2PTunnelRegistry;
|
||
|
|
import appeng.util.Platform;
|
||
|
|
|
||
|
|
public class P2PTunnelRegistry implements IP2PTunnelRegistry
|
||
|
|
{
|
||
|
|
|
||
|
|
HashMap<ItemStack, TunnelType> Tunnels = new HashMap();
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public void addNewAttunement(ItemStack trigger, TunnelType type)
|
||
|
|
{
|
||
|
|
if ( type == null )
|
||
|
|
throw new RuntimeException( "Invalid Tunnel Type." );
|
||
|
|
|
||
|
|
Tunnels.put( trigger, type );
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public TunnelType getTunnelTypeByItem(ItemStack trigger)
|
||
|
|
{
|
||
|
|
if ( FluidContainerRegistry.isContainer( trigger ) )
|
||
|
|
return TunnelType.FLUID;
|
||
|
|
|
||
|
|
for (ItemStack is : Tunnels.keySet())
|
||
|
|
{
|
||
|
|
if ( Platform.isSameItemType( is, trigger ) )
|
||
|
|
return Tunnels.get( is );
|
||
|
|
}
|
||
|
|
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|